pimhwang
/
CppSharp
mirror of https://github.com/mono/CppSharp.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

Refactor and extract common code in NAPI source generator.

instantiate-types-nested-templates
Joao Matos 5 years ago committed by João Matos
parent
1ddecb1dbd
commit
06d271806d
2 changed files with 283 additions and 228 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 181
      src/Generator/Generators/NAPI/NAPIHelpers.cs
  2. 256
      src/Generator/Generators/NAPI/NAPISources.cs

181
src/Generator/Generators/NAPI/NAPIHelpers.cs
Unescape View File

@ -1,9 +1,12 @@ @@ -1,9 +1,12 @@
using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using CppSharp.AST;
using CppSharp.AST.Extensions;
using CppSharp.Generators.C;
using CppSharp.Generators.NAPI;
using CppSharp.Passes;
namespace CppSharp.Generators.Cpp
@ -84,6 +87,184 @@ namespace CppSharp.Generators.Cpp @@ -84,6 +87,184 @@ namespace CppSharp.Generators.Cpp
return;
}
}
public virtual MarshalPrinter<MarshalContext> GetMarshalManagedToNativePrinter(MarshalContext ctx)
{
return new NAPIMarshalManagedToNativePrinter(ctx);
}
public virtual MarshalPrinter<MarshalContext> GetMarshalNativeToManagedPrinter(MarshalContext ctx)
{
return new NAPIMarshalNativeToManagedPrinter(ctx);
}
public struct ParamMarshal
{
public string Name;
public string Prefix;
public Parameter Param;
public MarshalContext Context;
}
public virtual List<ParamMarshal> GenerateFunctionParamsMarshal(IEnumerable<Parameter> @params,
Function function = null)
{
var marshals = new List<ParamMarshal>();
var paramIndex = 0;
foreach (var param in @params)
{
marshals.Add(GenerateFunctionParamMarshal(param, paramIndex, function));
paramIndex++;
}
return marshals;
}
public virtual ParamMarshal GenerateFunctionParamMarshal(Parameter param, int paramIndex,
Function function = null)
{
var paramMarshal = new ParamMarshal { Name = param.Name, Param = param };
var argName = Generator.GeneratedIdentifier("arg") + paramIndex.ToString(CultureInfo.InvariantCulture);
Parameter effectiveParam = param;
var isRef = param.IsOut || param.IsInOut;
var paramType = param.Type;
// TODO: Use same name between generators
var typeArgName = $"args[{paramIndex}]";
if (this is QuickJSInvokes)
typeArgName = $"argv[{paramIndex}]";
var ctx = new MarshalContext(Context, CurrentIndentation)
{
Parameter = effectiveParam,
ParameterIndex = paramIndex,
ArgName = typeArgName,
Function = function
};
paramMarshal.Context = ctx;
var marshal = GetMarshalManagedToNativePrinter(ctx);
effectiveParam.Visit(marshal);
if (string.IsNullOrEmpty(marshal.Context.Return))
throw new Exception($"Cannot marshal argument of function '{function.QualifiedOriginalName}'");
if (isRef)
{
var type = paramType.Visit(CTypePrinter);
if (param.IsInOut)
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
{
Write(marshal.Context.Before);
NeedNewLine();
}
WriteLine($"{type} {argName} = {marshal.Context.Return};");
}
else
WriteLine($"{type} {argName};");
}
else
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
{
Write(marshal.Context.Before);
NeedNewLine();
}
NewLineIfNeeded();
WriteLine($"auto {marshal.Context.VarPrefix}{argName} = {marshal.Context.Return};");
paramMarshal.Prefix = marshal.Context.ArgumentPrefix;
}
paramMarshal.Name = argName;
return paramMarshal;
}
public virtual void GenerateFunctionCallReturnMarshal(Function function)
{
var ctx = new MarshalContext(Context, CurrentIndentation)
{
ArgName = Helpers.ReturnIdentifier,
ReturnVarName = Helpers.ReturnIdentifier,
ReturnType = function.ReturnType
};
// TODO: Move this into the marshaler
if (function.ReturnType.Type.Desugar().IsClass())
ctx.ArgName = $"&{ctx.ArgName}";
var marshal = GetMarshalNativeToManagedPrinter(ctx);
function.ReturnType.Visit(marshal);
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
{
Write(marshal.Context.Before);
NewLine();
}
WriteLine($"return {marshal.Context.Return};");
}
public virtual void GenerateFunctionParamsMarshalCleanups(List<ParamMarshal> @params)
{
var marshalers = new List<MarshalPrinter<MarshalContext>>();
PushBlock();
{
foreach (var paramInfo in @params)
{
var param = paramInfo.Param;
if (param.Usage != ParameterUsage.Out && param.Usage != ParameterUsage.InOut)
continue;
if (param.Type.IsPointer() && !param.Type.GetFinalPointee().IsPrimitiveType())
param.QualifiedType = new QualifiedType(param.Type.GetFinalPointee());
var ctx = new MarshalContext(Context, CurrentIndentation)
{
ArgName = paramInfo.Name,
ReturnVarName = paramInfo.Name,
ReturnType = param.QualifiedType
};
var marshal = GetMarshalNativeToManagedPrinter(ctx);
marshalers.Add(marshal);
param.Visit(marshal);
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
Write(marshal.Context.Before);
WriteLine($"{param.Name} = {marshal.Context.Return};");
}
}
PopBlock(NewLineKind.IfNotEmpty);
PushBlock();
{
foreach (var marshal in marshalers)
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Cleanup))
Write(marshal.Context.Cleanup);
}
foreach (var marshal in @params)
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Cleanup))
Write(marshal.Context.Cleanup);
}
}
PopBlock(NewLineKind.IfNotEmpty);
}
}
/// <summary>

256
src/Generator/Generators/NAPI/NAPISources.cs
Unescape View File

@ -11,6 +11,7 @@ using CppSharp.Generators.NAPI; @@ -11,6 +11,7 @@ using CppSharp.Generators.NAPI;
using CppSharp.Passes;
using CppSharp.Utils.FSM;
using static CppSharp.Generators.Cpp.NAPISources;
using Type = CppSharp.AST.Type;
namespace CppSharp.Generators.Cpp
{
@ -415,7 +416,6 @@ namespace CppSharp.Generators.Cpp @@ -415,7 +416,6 @@ namespace CppSharp.Generators.Cpp
PushBlock(BlockKind.Function, function);
GenerateFunctionCallback(@group);
GenerateNativeCall(group);
NewLine();
// TODO:
@ -440,7 +440,6 @@ namespace CppSharp.Generators.Cpp @@ -440,7 +440,6 @@ namespace CppSharp.Generators.Cpp
PushBlock(BlockKind.Method);
GenerateFunctionCallback(@group.OfType<Function>().ToList());
GenerateNativeCall(@group);
NewLine();
WriteLine("return _this;");
@ -449,11 +448,6 @@ namespace CppSharp.Generators.Cpp @@ -449,11 +448,6 @@ namespace CppSharp.Generators.Cpp
PopBlock(NewLineKind.BeforeNextBlock);
}
public virtual void GenerateNativeCall(IEnumerable<Function> group)
{
}
public virtual void GenerateFunctionCallback(List<Function> @group)
{
var function = @group.First();
@ -479,15 +473,8 @@ namespace CppSharp.Generators.Cpp @@ -479,15 +473,8 @@ namespace CppSharp.Generators.Cpp
NewLine();
// Handle the zero arguments case right away if one exists.
var zeroParamsOverload = @group.SingleOrDefault(f => f.Parameters.Count == 0);
if (zeroParamsOverload != null && @group.Count > 1)
{
var index = @group.FindIndex(f => f == zeroParamsOverload);
WriteLine($"if (argc == 0)");
WriteLineIndent($"goto overload{index};");
NewLine();
}
CheckZeroArguments(@group);
NewLineIfNeeded();
// Check if the arguments are in the expected range.
CheckArgumentsRange(@group);
@ -496,16 +483,17 @@ namespace CppSharp.Generators.Cpp @@ -496,16 +483,17 @@ namespace CppSharp.Generators.Cpp
var needsArguments = @group.Any(f => f.Parameters.Any(p => p.IsGenerated));
if (needsArguments)
{
//WriteLine("void* data;");
WriteLine("status = napi_get_cb_info(env, info, &argc, args, nullptr, nullptr);");
WriteLine("assert(status == napi_ok);");
NewLine();
// Next we need to disambiguate which overload to call based on:
// 1. Number of arguments passed to the method
// 2. Type of arguments
CheckArgumentsTypes(@group);
WriteLine("for (size_t i = 0; i < argc; i++)");
WriteOpenBraceAndIndent();
{
WriteLine("status = napi_typeof(env, args[i], &types[i]);");
WriteLine("assert(status == napi_ok);");
}
UnindentAndWriteCloseBrace();
NewLine();
}
@ -530,6 +518,19 @@ namespace CppSharp.Generators.Cpp @@ -530,6 +518,19 @@ namespace CppSharp.Generators.Cpp
{
var stateMachine = CalculateOverloadStates(@group);
CheckArgumentsOverload(@group, stateMachine);
// Error state.
Unindent();
WriteLine($"error:");
Indent();
WriteLine("status = napi_throw_type_error(env, nullptr, \"Unsupported argument type\");");
WriteLine("assert(status == napi_ok);");
NewLine();
WriteLine("return nullptr;");
NewLine();
GenerateOverloadCalls(@group, stateMachine);
}
else
@ -547,7 +548,20 @@ namespace CppSharp.Generators.Cpp @@ -547,7 +548,20 @@ namespace CppSharp.Generators.Cpp
}
}
private void CheckArgumentsRange(IEnumerable<Function> @group)
public virtual void CheckZeroArguments(List<Function> @group)
{
var zeroParamsOverload = @group.SingleOrDefault(f => f.Parameters.Count == 0);
if (zeroParamsOverload == null || @group.Count <= 1)
return;
var index = @group.FindIndex(f => f == zeroParamsOverload);
WriteLine($"if (argc == 0)");
WriteLineIndent($"goto overload{index};");
NeedNewLine();
}
public virtual void CheckArgumentsRange(IEnumerable<Function> @group)
{
var enumerable = @group as List<Function> ?? @group.ToList();
var (minArgs, maxArgs) = (enumerable.Min(m => m.Parameters.Count),
@ -557,26 +571,17 @@ namespace CppSharp.Generators.Cpp @@ -557,26 +571,17 @@ namespace CppSharp.Generators.Cpp
WriteLine($"if ({rangeCheck})");
WriteOpenBraceAndIndent();
{
WriteLine("status = napi_throw_type_error(env, nullptr, \"Unsupported number of arguments\");");
WriteLine("assert(status == napi_ok);");
NewLine();
WriteLine("return nullptr;");
UnindentAndWriteCloseBrace();
}
private void CheckArgumentsTypes(IEnumerable<Function> @group)
{
WriteLine("for (size_t i = 0; i < argc; i++)");
WriteOpenBraceAndIndent();
WriteLine("status = napi_typeof(env, args[i], &types[i]);");
WriteLine("assert(status == napi_ok);");
UnindentAndWriteCloseBrace();
}
private void CheckArgumentsOverload(IList<Function> @group, DFSM stateMachine)
public virtual void CheckArgumentsOverload(IList<Function> @group, DFSM stateMachine)
{
var typeCheckStates = stateMachine.Q.Except(stateMachine.F).ToList();
var finalStates = stateMachine.F;
@ -610,12 +615,7 @@ namespace CppSharp.Generators.Cpp @@ -610,12 +615,7 @@ namespace CppSharp.Generators.Cpp
var type = uniqueTypes[(int) transition.Symbol];
var typeChecker = new NAPITypeCheckGen(paramIndex);
type.Visit(typeChecker);
var condition = typeChecker.Generate();
if (string.IsNullOrWhiteSpace(condition))
throw new NotSupportedException();
var condition = GenerateTypeCheckForParameter(paramIndex, type);
WriteLine($"if ({condition})");
@ -632,21 +632,21 @@ namespace CppSharp.Generators.Cpp @@ -632,21 +632,21 @@ namespace CppSharp.Generators.Cpp
NeedNewLine();
}
NewLineIfNeeded();
}
// Error state.
Unindent();
WriteLine($"error:");
Indent();
public virtual string GenerateTypeCheckForParameter(int paramIndex, Type type)
{
var typeChecker = new NAPITypeCheckGen(paramIndex);
type.Visit(typeChecker);
WriteLine("status = napi_throw_type_error(env, nullptr, \"Unsupported argument type\");");
WriteLine("assert(status == napi_ok);");
NewLine();
var condition = typeChecker.Generate();
if (string.IsNullOrWhiteSpace(condition))
throw new NotSupportedException();
WriteLine("return nullptr;");
NewLine();
return condition;
}
private void GenerateOverloadCalls(IList<Function> @group, DFSM stateMachine)
public virtual void GenerateOverloadCalls(IList<Function> @group, DFSM stateMachine)
{
// Final states.
for (var i = 0; i < stateMachine.F.Count; i++)
@ -661,9 +661,9 @@ namespace CppSharp.Generators.Cpp @@ -661,9 +661,9 @@ namespace CppSharp.Generators.Cpp
Indent();
WriteOpenBraceAndIndent();
{
GenerateFunctionCall(function);
}
UnindentAndWriteCloseBrace();
NeedNewLine();
}
@ -672,9 +672,12 @@ namespace CppSharp.Generators.Cpp @@ -672,9 +672,12 @@ namespace CppSharp.Generators.Cpp
public virtual void GenerateFunctionCall(Function function)
{
var @params = GenerateFunctionParamsMarshal(function.Parameters, function);
var method = function as Method;
var isVoidReturn = function.ReturnType.Type.IsPrimitiveType(PrimitiveType.Void);
var needsReturn = !function.ReturnType.Type.IsPrimitiveType(PrimitiveType.Void);
if (needsReturn)
PushBlock();
{
if (!isVoidReturn)
{
CTypePrinter.PushContext(TypePrinterContextKind.Native);
var returnType = function.ReturnType.Visit(CTypePrinter);
@ -683,9 +686,7 @@ namespace CppSharp.Generators.Cpp @@ -683,9 +686,7 @@ namespace CppSharp.Generators.Cpp
Write($"{returnType} {Helpers.ReturnIdentifier} = ");
}
var method = function as Method;
var @class = function.Namespace as Class;
var property = method?.AssociatedDeclaration as Property;
var field = property?.Field;
if (field != null)
@ -719,58 +720,23 @@ namespace CppSharp.Generators.Cpp @@ -719,58 +720,23 @@ namespace CppSharp.Generators.Cpp
GenerateFunctionParams(@params);
WriteLine(");");
}
foreach(var paramInfo in @params)
{
var param = paramInfo.Param;
if(param.Usage != ParameterUsage.Out && param.Usage != ParameterUsage.InOut)
continue;
if (param.Type.IsPointer() && !param.Type.GetFinalPointee().IsPrimitiveType())
param.QualifiedType = new QualifiedType(param.Type.GetFinalPointee());
var ctx = new MarshalContext(Context, CurrentIndentation)
{
ArgName = paramInfo.Name,
ReturnVarName = paramInfo.Name,
ReturnType = param.QualifiedType
};
var marshal = GetMarshalNativeToManagedPrinter(ctx);
param.Visit(marshal);
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
Write(marshal.Context.Before);
WriteLine($"{param.Name} = {marshal.Context.Return};");
}
PopBlock(NewLineKind.IfNotEmpty);
GenerateFunctionParamsMarshalCleanups(@params);
var isCtor = method != null && method.IsConstructor;
var needsReturn = !isVoidReturn || (this is QuickJSInvokes && !isCtor);
if (needsReturn)
{
NewLine();
GenerateFunctionCallReturnMarshal(function);
}
}
public virtual void GenerateFunctionCallReturnMarshal(Function function)
{
var ctx = new MarshalContext(Context, CurrentIndentation)
{
ArgName = Helpers.ReturnIdentifier,
ReturnVarName = Helpers.ReturnIdentifier,
ReturnType = function.ReturnType
};
var marshal = GetMarshalNativeToManagedPrinter(ctx);
function.ReturnType.Visit(marshal);
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
if (isCtor)
{
Write(marshal.Context.Before);
NewLine();
WriteLine("goto wrap;");
}
WriteLine($"return {marshal.Context.Return};");
}
public bool IsNativeFunctionOrStaticMethod(Function function)
@ -788,98 +754,6 @@ namespace CppSharp.Generators.Cpp @@ -788,98 +754,6 @@ namespace CppSharp.Generators.Cpp
return method.IsStatic || method.Conversion != MethodConversionKind.None;
}
public struct ParamMarshal
{
public string Name;
public string Prefix;
public Parameter Param;
}
public List<ParamMarshal> GenerateFunctionParamsMarshal(IEnumerable<Parameter> @params,
Function function = null)
{
var marshals = new List<ParamMarshal>();
var paramIndex = 0;
foreach (var param in @params)
{
marshals.Add(GenerateFunctionParamMarshal(param, paramIndex, function));
paramIndex++;
}
return marshals;
}
public virtual MarshalPrinter<MarshalContext> GetMarshalManagedToNativePrinter(MarshalContext ctx)
{
return new NAPIMarshalManagedToNativePrinter(ctx);
}
public virtual MarshalPrinter<MarshalContext> GetMarshalNativeToManagedPrinter(MarshalContext ctx)
{
return new NAPIMarshalNativeToManagedPrinter(ctx);
}
public virtual ParamMarshal GenerateFunctionParamMarshal(Parameter param, int paramIndex,
Function function = null)
{
var paramMarshal = new ParamMarshal { Name = param.Name, Param = param };
var argName = Generator.GeneratedIdentifier("arg") + paramIndex.ToString(CultureInfo.InvariantCulture);
Parameter effectiveParam = param;
var isRef = param.IsOut || param.IsInOut;
var paramType = param.Type;
var ctx = new MarshalContext(Context, CurrentIndentation)
{
Parameter = effectiveParam,
ParameterIndex = paramIndex,
ArgName = argName,
Function = function
};
var marshal = GetMarshalManagedToNativePrinter(ctx);
effectiveParam.Visit(marshal);
if (string.IsNullOrEmpty(marshal.Context.Return))
throw new Exception($"Cannot marshal argument of function '{function.QualifiedOriginalName}'");
if (isRef)
{
var type = paramType.Visit(CTypePrinter);
if (param.IsInOut)
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
{
Write(marshal.Context.Before);
NeedNewLine();
}
WriteLine($"{type} {argName} = {marshal.Context.Return};");
}
else
WriteLine($"{type} {argName};");
}
else
{
if (!string.IsNullOrWhiteSpace(marshal.Context.Before))
{
Write(marshal.Context.Before);
NeedNewLine();
}
NewLineIfNeeded();
WriteLine($"auto {marshal.Context.VarPrefix}{argName} = {marshal.Context.Return};");
paramMarshal.Prefix = marshal.Context.ArgumentPrefix;
}
paramMarshal.Name = argName;
return paramMarshal;
}
public void GenerateFunctionParams(List<ParamMarshal> @params)
{
var names = @params.Select(param =>
@ -889,7 +763,7 @@ namespace CppSharp.Generators.Cpp @@ -889,7 +763,7 @@ namespace CppSharp.Generators.Cpp
Write(string.Join(", ", names));
}
private static DFSM CalculateOverloadStates(IEnumerable<Function> group)
public static DFSM CalculateOverloadStates(IEnumerable<Function> group)
{
var functionGroup = group.ToList();

Write Preview
Loading…
Cancel
Save