IPC code gen improvements (#6352)
* HipcGenerator: skip do-nothing call to MemoryMarshal.Cast<byte, byte>() in generated code * HipcGenerator: fix method name typos * HipcGenerator: make generated methods use stackalloc for `isBufferMapAlias` bool array * HipcGenerator: make generated GetCommandHandlers() method return a FrozenDictionary<int, CommandHandler> * HipcGenerator: return `FrozenDictionary<,>.Empty` when there are no command implementations, otherwise create `FrozenDictionary` from a `IEnumerable<KeyValuePair<,>>`` instead of a `Dictionary<,>``
This commit is contained in:
parent
d4d0a48bfe
commit
08384ee5a8
2 changed files with 54 additions and 42 deletions
|
@ -74,6 +74,7 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
generator.AppendLine("using Ryujinx.Horizon.Sdk.Sf.Cmif;");
|
||||
generator.AppendLine("using Ryujinx.Horizon.Sdk.Sf.Hipc;");
|
||||
generator.AppendLine("using System;");
|
||||
generator.AppendLine("using System.Collections.Frozen;");
|
||||
generator.AppendLine("using System.Collections.Generic;");
|
||||
generator.AppendLine("using System.Runtime.CompilerServices;");
|
||||
generator.AppendLine("using System.Runtime.InteropServices;");
|
||||
|
@ -115,17 +116,24 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
private static void GenerateMethodTable(CodeGenerator generator, Compilation compilation, CommandInterface commandInterface)
|
||||
{
|
||||
generator.EnterScope($"public IReadOnlyDictionary<int, CommandHandler> GetCommandHandlers()");
|
||||
generator.EnterScope($"return new Dictionary<int, CommandHandler>()");
|
||||
|
||||
if (commandInterface.CommandImplementations.Count == 0)
|
||||
{
|
||||
generator.AppendLine("return FrozenDictionary<int, CommandHandler>.Empty;");
|
||||
}
|
||||
else
|
||||
{
|
||||
generator.EnterScope($"return FrozenDictionary.ToFrozenDictionary(new []");
|
||||
|
||||
foreach (var method in commandInterface.CommandImplementations)
|
||||
{
|
||||
foreach (var commandId in GetAttributeAguments(compilation, method, TypeCommandAttribute, 0))
|
||||
foreach (var commandId in GetAttributeArguments(compilation, method, TypeCommandAttribute, 0))
|
||||
{
|
||||
string[] args = new string[method.ParameterList.Parameters.Count];
|
||||
|
||||
if (args.Length == 0)
|
||||
{
|
||||
generator.AppendLine($"{{ {commandId}, new CommandHandler({method.Identifier.Text}, Array.Empty<CommandArg>()) }},");
|
||||
generator.AppendLine($"KeyValuePair.Create({commandId}, new CommandHandler({method.Identifier.Text}, Array.Empty<CommandArg>())),");
|
||||
}
|
||||
else
|
||||
{
|
||||
|
@ -140,8 +148,8 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
|
||||
if (argType == CommandArgType.Buffer)
|
||||
{
|
||||
string bufferFlags = GetFirstAttributeAgument(compilation, parameter, TypeBufferAttribute, 0);
|
||||
string bufferFixedSize = GetFirstAttributeAgument(compilation, parameter, TypeBufferAttribute, 1);
|
||||
string bufferFlags = GetFirstAttributeArgument(compilation, parameter, TypeBufferAttribute, 0);
|
||||
string bufferFixedSize = GetFirstAttributeArgument(compilation, parameter, TypeBufferAttribute, 1);
|
||||
|
||||
if (bufferFixedSize != null)
|
||||
{
|
||||
|
@ -166,16 +174,18 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
args[index++] = arg;
|
||||
}
|
||||
|
||||
generator.AppendLine($"{{ {commandId}, new CommandHandler({method.Identifier.Text}, {string.Join(", ", args)}) }},");
|
||||
generator.AppendLine($"KeyValuePair.Create({commandId}, new CommandHandler({method.Identifier.Text}, {string.Join(", ", args)})),");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
generator.LeaveScope(";");
|
||||
generator.LeaveScope(");");
|
||||
}
|
||||
|
||||
generator.LeaveScope();
|
||||
}
|
||||
|
||||
private static IEnumerable<string> GetAttributeAguments(Compilation compilation, SyntaxNode syntaxNode, string attributeName, int argIndex)
|
||||
private static IEnumerable<string> GetAttributeArguments(Compilation compilation, SyntaxNode syntaxNode, string attributeName, int argIndex)
|
||||
{
|
||||
ISymbol symbol = compilation.GetSemanticModel(syntaxNode.SyntaxTree).GetDeclaredSymbol(syntaxNode);
|
||||
|
||||
|
@ -188,9 +198,9 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
}
|
||||
}
|
||||
|
||||
private static string GetFirstAttributeAgument(Compilation compilation, SyntaxNode syntaxNode, string attributeName, int argIndex)
|
||||
private static string GetFirstAttributeArgument(Compilation compilation, SyntaxNode syntaxNode, string attributeName, int argIndex)
|
||||
{
|
||||
return GetAttributeAguments(compilation, syntaxNode, attributeName, argIndex).FirstOrDefault();
|
||||
return GetAttributeArguments(compilation, syntaxNode, attributeName, argIndex).FirstOrDefault();
|
||||
}
|
||||
|
||||
private static void GenerateMethod(CodeGenerator generator, Compilation compilation, MethodDeclarationSyntax method)
|
||||
|
@ -233,7 +243,7 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
|
||||
if (buffersCount != 0)
|
||||
{
|
||||
generator.AppendLine($"bool[] {IsBufferMapAliasVariableName} = new bool[{method.ParameterList.Parameters.Count}];");
|
||||
generator.AppendLine($"Span<bool> {IsBufferMapAliasVariableName} = stackalloc bool[{method.ParameterList.Parameters.Count}];");
|
||||
generator.AppendLine();
|
||||
|
||||
generator.AppendLine($"{ResultVariableName} = processor.ProcessBuffers(ref context, {IsBufferMapAliasVariableName}, runtimeMetadata);");
|
||||
|
@ -719,7 +729,9 @@ namespace Ryujinx.Horizon.Generators.Hipc
|
|||
|
||||
private static string GenerateSpanCast(string targetType, string input)
|
||||
{
|
||||
return $"MemoryMarshal.Cast<byte, {targetType}>({input})";
|
||||
return targetType == "byte"
|
||||
? input
|
||||
: $"MemoryMarshal.Cast<byte, {targetType}>({input})";
|
||||
}
|
||||
|
||||
private static bool HasAttribute(Compilation compilation, ParameterSyntax parameterSyntax, string fullAttributeName)
|
||||
|
|
|
@ -127,7 +127,7 @@ namespace Ryujinx.Horizon.Sdk.Sf
|
|||
return _bufferRanges[argIndex];
|
||||
}
|
||||
|
||||
public Result ProcessBuffers(ref ServiceDispatchContext context, bool[] isBufferMapAlias, ServerMessageRuntimeMetadata runtimeMetadata)
|
||||
public Result ProcessBuffers(ref ServiceDispatchContext context, scoped Span<bool> isBufferMapAlias, ServerMessageRuntimeMetadata runtimeMetadata)
|
||||
{
|
||||
bool mapAliasBuffersValid = true;
|
||||
|
||||
|
@ -246,7 +246,7 @@ namespace Ryujinx.Horizon.Sdk.Sf
|
|||
return mode == HipcBufferMode.Normal;
|
||||
}
|
||||
|
||||
public void SetOutBuffers(HipcMessageData response, bool[] isBufferMapAlias)
|
||||
public void SetOutBuffers(HipcMessageData response, ReadOnlySpan<bool> isBufferMapAlias)
|
||||
{
|
||||
int recvPointerIndex = 0;
|
||||
|
||||
|
|
Reference in a new issue