Nikita Tsukanov
4 years ago
20 changed files with 821 additions and 750 deletions
@ -1,79 +0,0 @@ |
|||
using System; |
|||
using System.Linq; |
|||
using System.Reflection; |
|||
using System.Runtime.InteropServices; |
|||
using Avalonia.Platform.Interop; |
|||
|
|||
namespace Avalonia.OpenGL |
|||
{ |
|||
public class GlInterfaceBase : GlInterfaceBase<object> |
|||
{ |
|||
public GlInterfaceBase(Func<string, IntPtr> getProcAddress) : base(getProcAddress, null) |
|||
{ |
|||
} |
|||
|
|||
public GlInterfaceBase(Func<Utf8Buffer, IntPtr> nativeGetProcAddress) : base(nativeGetProcAddress, null) |
|||
{ |
|||
} |
|||
} |
|||
|
|||
public class GlInterfaceBase<TContext> |
|||
{ |
|||
private readonly Func<string, IntPtr> _getProcAddress; |
|||
public GlInterfaceBase(Func<string, IntPtr> getProcAddress, TContext context) |
|||
{ |
|||
_getProcAddress = getProcAddress; |
|||
foreach (var prop in this.GetType().GetProperties()) |
|||
{ |
|||
var attrs = prop.GetCustomAttributes() |
|||
.Where(a => |
|||
a is IGlEntryPointAttribute || a is IGlEntryPointAttribute<TContext>) |
|||
.ToList(); |
|||
if(attrs.Count == 0) |
|||
continue; |
|||
|
|||
var isOptional = prop.GetCustomAttribute<GlOptionalEntryPoint>() != null; |
|||
|
|||
var fieldName = $"<{prop.Name}>k__BackingField"; |
|||
var field = prop.DeclaringType.GetField(fieldName, |
|||
BindingFlags.Instance | BindingFlags.NonPublic); |
|||
if (field == null) |
|||
throw new InvalidProgramException($"Expected property {prop.Name} to have {fieldName}"); |
|||
|
|||
|
|||
IntPtr proc = IntPtr.Zero; |
|||
foreach (var attr in attrs) |
|||
{ |
|||
if (attr is IGlEntryPointAttribute<TContext> typed) |
|||
proc = typed.GetProcAddress(context, getProcAddress); |
|||
else if (attr is IGlEntryPointAttribute untyped) |
|||
proc = untyped.GetProcAddress(getProcAddress); |
|||
if (proc != IntPtr.Zero) |
|||
break; |
|||
} |
|||
|
|||
if (proc != IntPtr.Zero) |
|||
field.SetValue(this, Marshal.GetDelegateForFunctionPointer(proc, prop.PropertyType)); |
|||
else if (!isOptional) |
|||
throw new OpenGlException("Unable to find a suitable GL function for " + prop.Name); |
|||
} |
|||
} |
|||
|
|||
protected static Func<string, IntPtr> ConvertNative(Func<Utf8Buffer, IntPtr> func) => |
|||
(proc) => |
|||
{ |
|||
using (var u = new Utf8Buffer(proc)) |
|||
{ |
|||
var rv = func(u); |
|||
return rv; |
|||
} |
|||
}; |
|||
|
|||
public GlInterfaceBase(Func<Utf8Buffer, IntPtr> nativeGetProcAddress, TContext context) : this(ConvertNative(nativeGetProcAddress), context) |
|||
{ |
|||
|
|||
} |
|||
|
|||
public IntPtr GetProcAddress(string proc) => _getProcAddress(proc); |
|||
} |
|||
} |
|||
@ -0,0 +1,346 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Collections.Immutable; |
|||
using System.Diagnostics; |
|||
using System.IO; |
|||
using System.Linq; |
|||
using System.Text; |
|||
using System.Threading; |
|||
using Microsoft.CodeAnalysis; |
|||
using Microsoft.CodeAnalysis.CSharp; |
|||
using Microsoft.CodeAnalysis.CSharp.Syntax; |
|||
|
|||
namespace Generator; |
|||
|
|||
[Generator(LanguageNames.CSharp)] |
|||
public class GetProcAddressInitializationGenerator : IIncrementalGenerator |
|||
{ |
|||
const string GetProcAddressFullName = "global::Avalonia.SourceGenerator.GetProcAddressAttribute"; |
|||
|
|||
public void Initialize(IncrementalGeneratorInitializationContext context) |
|||
{ |
|||
/* |
|||
Console.WriteLine("PID: " + Process.GetCurrentProcess().Id); |
|||
|
|||
File.WriteAllText("/tmp/pid", "PID: " + Process.GetCurrentProcess().Id); |
|||
while (!Debugger.IsAttached) |
|||
{ |
|||
Thread.Sleep(1000); |
|||
}*/ |
|||
var allMethodsWithAttributes = context.SyntaxProvider |
|||
.CreateSyntaxProvider( |
|||
static (s, _) => s is MethodDeclarationSyntax |
|||
{ |
|||
AttributeLists.Count: > 0, |
|||
} md && md.Modifiers.Any(m=>m.IsKind(SyntaxKind.PartialKeyword)), |
|||
static (context, _) => |
|||
(IMethodSymbol)context.SemanticModel.GetDeclaredSymbol(context.Node)!); |
|||
|
|||
var fieldsWithAttribute = allMethodsWithAttributes |
|||
.Where(s => s.HasAttributeWithFullyQualifiedName(GetProcAddressFullName)); |
|||
|
|||
var all = fieldsWithAttribute.Collect(); |
|||
context.RegisterSourceOutput(all, static (context, methods) => |
|||
{ |
|||
foreach (var typeGroup in methods.GroupBy(f => f.ContainingType)) |
|||
{ |
|||
var nextContext = 0; |
|||
var contexts = new Dictionary<string, int>(); |
|||
|
|||
string GetContextNameFromIndex(int c) => "context" + (c == 0 ? "" : c); |
|||
string GetContextName(string type) |
|||
{ |
|||
if (contexts.TryGetValue(type, out var idx)) |
|||
return GetContextNameFromIndex(idx); |
|||
if (nextContext != 0) |
|||
idx += nextContext; |
|||
nextContext++; |
|||
return GetContextNameFromIndex(contexts[type] = idx); |
|||
} |
|||
|
|||
var classBuilder = new StringBuilder(); |
|||
if (typeGroup.Key.ContainingNamespace != null) |
|||
classBuilder |
|||
.AppendLine("using System;") |
|||
.Append("namespace ") |
|||
.Append(typeGroup.Key.ContainingNamespace) |
|||
.AppendLine(";"); |
|||
classBuilder |
|||
.Append("unsafe partial class ") |
|||
.AppendLine(typeGroup.Key.Name) |
|||
.AppendLine("{"); |
|||
var initializeBody = new StringBuilder() |
|||
.Pad(2) |
|||
.AppendLine("var addr = IntPtr.Zero;"); |
|||
|
|||
foreach (var method in typeGroup) |
|||
{ |
|||
var isOptional = false; |
|||
var first = true; |
|||
var fieldName = "_addr_" + method.Name; |
|||
var delegateType = BuildDelegateType(method); |
|||
|
|||
void AppendNextAddr() |
|||
{ |
|||
if (first) |
|||
{ |
|||
first = false; |
|||
initializeBody.Pad(2); |
|||
} |
|||
else |
|||
initializeBody |
|||
.Pad(2) |
|||
.Append("if(addr == IntPtr.Zero) "); |
|||
} |
|||
|
|||
initializeBody |
|||
.Pad(2).Append("// Initializing ").AppendLine(method.Name) |
|||
.Pad(2) |
|||
.AppendLine("addr = IntPtr.Zero;"); |
|||
foreach (var attr in method.GetAttributes()) |
|||
{ |
|||
if (attr.AttributeClass?.HasFullyQualifiedName(GetProcAddressFullName) == true) |
|||
{ |
|||
string? primaryName = null; |
|||
foreach (var arg in attr.ConstructorArguments) |
|||
{ |
|||
if (arg.Value is string name) |
|||
primaryName = name; |
|||
if (arg.Value is bool opt) |
|||
isOptional = opt; |
|||
} |
|||
|
|||
if (primaryName != null) |
|||
{ |
|||
AppendNextAddr(); |
|||
initializeBody |
|||
.Append("addr = getProcAddress(\"") |
|||
.Append(primaryName) |
|||
.AppendLine("\");"); |
|||
} |
|||
} |
|||
else |
|||
{ |
|||
if (attr.AttributeClass != null |
|||
&& attr.AttributeClass.MemberNames.Contains("GetProcAddress")) |
|||
{ |
|||
var getProcMethod = attr.AttributeClass.GetMembers() |
|||
.FirstOrDefault(m => m.Name == "GetProcAddress") as IMethodSymbol; |
|||
if (getProcMethod == null || !getProcMethod.IsStatic || getProcMethod.Parameters.Length < 2) |
|||
continue; |
|||
var contextName = |
|||
GetContextName(getProcMethod |
|||
.Parameters[1].Type |
|||
.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat)); |
|||
AppendNextAddr(); |
|||
initializeBody |
|||
.Append("addr = ") |
|||
.Append(attr.AttributeClass.GetFullyQualifiedName()) |
|||
.Append(".GetProcAddress(") |
|||
.Append("getProcAddress, ") |
|||
.Append(contextName); |
|||
|
|||
var syntaxNode = (AttributeSyntax)attr.ApplicationSyntaxReference.GetSyntax(); |
|||
foreach (var arg in syntaxNode.ArgumentList.Arguments) |
|||
initializeBody.Append(", ").Append(arg.GetText()); |
|||
initializeBody.AppendLine(");"); |
|||
} |
|||
} |
|||
} |
|||
|
|||
if (!isOptional) |
|||
{ |
|||
initializeBody |
|||
.Pad(2) |
|||
.Append("if (addr == IntPtr.Zero) throw new System.EntryPointNotFoundException(\"") |
|||
.Append(fieldName).AppendLine("\");"); |
|||
} |
|||
|
|||
initializeBody |
|||
.Pad(2) |
|||
.Append(fieldName) |
|||
.Append(" = (") |
|||
.Append(delegateType) |
|||
.AppendLine(")addr;"); |
|||
|
|||
classBuilder |
|||
.Pad(1) |
|||
.Append(delegateType); |
|||
classBuilder |
|||
.Append(fieldName) |
|||
.AppendLine(";"); |
|||
|
|||
classBuilder |
|||
.Pad(1) |
|||
.Append("public partial ") |
|||
.Append(method.ReturnType.GetFullyQualifiedName()) |
|||
.Append(" ") |
|||
.Append(method.Name) |
|||
.Append("("); |
|||
var firstArg = true; |
|||
foreach (var p in method.Parameters) |
|||
{ |
|||
if (firstArg) |
|||
firstArg = false; |
|||
else |
|||
classBuilder.Append(", "); |
|||
AppendRefKind(classBuilder, p.RefKind); |
|||
classBuilder |
|||
.Append(p.Type.GetFullyQualifiedName()) |
|||
.Append(" @") |
|||
.Append(p.Name); |
|||
} |
|||
classBuilder |
|||
.AppendLine(")") |
|||
.Pad(1) |
|||
.AppendLine("{"); |
|||
if (isOptional) |
|||
classBuilder |
|||
.Pad(2) |
|||
.Append("if (") |
|||
.Append(fieldName) |
|||
.Append(" == null) throw new System.EntryPointNotFoundException(\"") |
|||
.Append(method.Name) |
|||
.AppendLine("\");"); |
|||
|
|||
foreach(var p in method.Parameters) |
|||
if (NeedsPin(p.Type)) |
|||
classBuilder.Pad(2) |
|||
.Append("fixed(") |
|||
.Append(MapToNative(p.Type)) |
|||
.Append(" @__p_") |
|||
.Append(p.Name) |
|||
.Append(" = ") |
|||
.Append(p.Name) |
|||
.AppendLine(")"); |
|||
|
|||
classBuilder.Pad(2); |
|||
if (!method.ReturnsVoid) |
|||
classBuilder.Append("return "); |
|||
|
|||
var invokeBuilder = new StringBuilder(); |
|||
|
|||
invokeBuilder |
|||
.Append(fieldName) |
|||
.Append("("); |
|||
firstArg = true; |
|||
foreach (var p in method.Parameters) |
|||
{ |
|||
if (firstArg) |
|||
firstArg = false; |
|||
else |
|||
invokeBuilder.Append(", "); |
|||
AppendRefKind(invokeBuilder, p.RefKind); |
|||
invokeBuilder |
|||
.Append("@") |
|||
.Append(ConvertToNative(p.Name, p.Type)); |
|||
} |
|||
|
|||
invokeBuilder.Append(")"); |
|||
classBuilder.Append(ConvertToManaged(method.ReturnType, invokeBuilder.ToString())); |
|||
|
|||
classBuilder.AppendLine(";").Pad(1).AppendLine("}"); |
|||
if (isOptional) |
|||
classBuilder |
|||
.Pad(1) |
|||
.Append("public bool Is") |
|||
.Append(method.Name) |
|||
.Append("Available => ") |
|||
.Append(fieldName) |
|||
.AppendLine(" != null;"); |
|||
} |
|||
|
|||
classBuilder |
|||
.Pad(1) |
|||
.Append("void Initialize(Func<string, IntPtr> getProcAddress"); |
|||
foreach (var kv in contexts.OrderBy(x => x.Value)) |
|||
{ |
|||
classBuilder |
|||
.Append(", ") |
|||
.Append(kv.Key) |
|||
.Append(" ") |
|||
.Append(GetContextNameFromIndex(kv.Value)); |
|||
} |
|||
|
|||
classBuilder.AppendLine(")").Pad(1).AppendLine("{"); |
|||
classBuilder.Append(initializeBody.ToString()); |
|||
classBuilder.Append("}\n}"); |
|||
|
|||
|
|||
context.AddSource(typeGroup.Key.GetFullyQualifiedName().Replace(":", ""), classBuilder.ToString()); |
|||
} |
|||
}); |
|||
|
|||
|
|||
} |
|||
|
|||
static StringBuilder AppendRefKind(StringBuilder sb, RefKind kind) |
|||
{ |
|||
if (kind == RefKind.Ref) |
|||
sb.Append("ref "); |
|||
if (kind == RefKind.Out) |
|||
sb.Append("out "); |
|||
return sb; |
|||
} |
|||
|
|||
static bool NeedsPin(ITypeSymbol type) |
|||
{ |
|||
if (type.TypeKind == TypeKind.Array) |
|||
return true; |
|||
return false; |
|||
} |
|||
|
|||
static string ConvertToNative(string name, ITypeSymbol type) |
|||
{ |
|||
if (NeedsPin(type)) |
|||
return "__p_" + name; |
|||
if (IsBool(type)) |
|||
return $"{name} ? 1 : 0"; |
|||
return name; |
|||
} |
|||
|
|||
static string ConvertToManaged(ITypeSymbol type, string expr) |
|||
{ |
|||
if (IsBool(type)) |
|||
return expr + " != 0"; |
|||
return expr; |
|||
} |
|||
|
|||
static bool IsBool(ITypeSymbol type) => type.GetFullyQualifiedName() == "global::System.Boolean" || |
|||
type.GetFullyQualifiedName() == "bool"; |
|||
|
|||
static string MapToNative(ITypeSymbol type) |
|||
{ |
|||
if (type.TypeKind == TypeKind.Array) |
|||
return ((IArrayTypeSymbol)type).ElementType.GetFullyQualifiedName() + "*"; |
|||
if (IsBool(type)) |
|||
return "int"; |
|||
return type.GetFullyQualifiedName(); |
|||
} |
|||
|
|||
static string BuildDelegateType(IMethodSymbol method) |
|||
{ |
|||
StringBuilder name = new("delegate* unmanaged[Stdcall]<"); |
|||
var firstArg = true; |
|||
|
|||
void AppendArg(string a, RefKind kind) |
|||
{ |
|||
if (firstArg) |
|||
firstArg = false; |
|||
else |
|||
name.Append(","); |
|||
AppendRefKind(name, kind); |
|||
name.Append(a); |
|||
} |
|||
|
|||
foreach (var p in method.Parameters) |
|||
{ |
|||
AppendArg(MapToNative(p.Type), p.RefKind); |
|||
} |
|||
|
|||
AppendArg(MapToNative(method.ReturnType), RefKind.None); |
|||
name.Append(">"); |
|||
return name.ToString(); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,31 @@ |
|||
using System.Collections.Immutable; |
|||
using System.Text; |
|||
using Microsoft.CodeAnalysis; |
|||
|
|||
namespace Generator; |
|||
|
|||
static class Helpers |
|||
{ |
|||
public static StringBuilder Pad(this StringBuilder sb, int count) => sb.Append(' ', count * 4); |
|||
|
|||
public static string GetFullyQualifiedName(this ISymbol symbol) |
|||
{ |
|||
return symbol.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat); |
|||
} |
|||
|
|||
public static bool HasFullyQualifiedName(this ISymbol symbol, string name) |
|||
{ |
|||
return symbol.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat) == name; |
|||
} |
|||
|
|||
public static bool HasAttributeWithFullyQualifiedName(this ISymbol symbol, string name) |
|||
{ |
|||
ImmutableArray<AttributeData> attributes = symbol.GetAttributes(); |
|||
|
|||
foreach (AttributeData attribute in attributes) |
|||
if (attribute.AttributeClass?.HasFullyQualifiedName(name) == true) |
|||
return true; |
|||
|
|||
return false; |
|||
} |
|||
} |
|||
Loading…
Reference in new issue