raylib building
This commit is contained in:
nub31
@@ -16,6 +16,9 @@ public class LlvmSharpGenerator
|
||||
private readonly Dictionary<string, LLVMValueRef> _functions = new();
|
||||
private readonly Dictionary<string, LLVMValueRef> _locals = new();
|
||||
private readonly Stack<(LLVMBasicBlockRef breakBlock, LLVMBasicBlockRef continueBlock)> _loopStack = new();
|
||||
private readonly Stack<Scope> _scopes = new();
|
||||
|
||||
private Scope CurrentScope => _scopes.Peek();
|
||||
|
||||
public void Emit(List<TopLevelNode> topLevelNodes, ModuleRepository repository, string sourceFileName, string outputPath)
|
||||
{
|
||||
@@ -32,50 +35,97 @@ public class LlvmSharpGenerator
|
||||
_functions.Clear();
|
||||
_locals.Clear();
|
||||
_loopStack.Clear();
|
||||
_scopes.Clear();
|
||||
|
||||
var stringType = _context.CreateNamedStruct("nub.string");
|
||||
stringType.StructSetBody([LLVMTypeRef.Int64, LLVMTypeRef.CreatePointer(LLVMTypeRef.Int8, 0)], false);
|
||||
_structTypes["nub.string"] = stringType;
|
||||
|
||||
// note(nub31): Declare all structs and functions
|
||||
foreach (var module in repository.GetAll())
|
||||
{
|
||||
foreach (var structType in module.StructTypes)
|
||||
{
|
||||
var structName = StructName(structType.Module, structType.Name);
|
||||
var llvmStructType = _context.CreateNamedStruct(structName);
|
||||
_structTypes[structName] = llvmStructType;
|
||||
}
|
||||
}
|
||||
var llvmStructType = _context.CreateNamedStruct(StructName(structType.Module, structType.Name));
|
||||
llvmStructType.StructSetBody(structType.Fields.Select(f => MapType(f.Type)).ToArray(), structType.Packed);
|
||||
_structTypes[StructName(structType.Module, structType.Name)] = llvmStructType;
|
||||
|
||||
foreach (var module in repository.GetAll())
|
||||
{
|
||||
foreach (var structType in module.StructTypes)
|
||||
{
|
||||
var structName = StructName(structType.Module, structType.Name);
|
||||
var llvmStructType = _structTypes[structName];
|
||||
var fieldTypes = structType.Fields.Select(f => MapType(f.Type)).ToArray();
|
||||
llvmStructType.StructSetBody(fieldTypes, false);
|
||||
}
|
||||
}
|
||||
var constructorType = LLVMTypeRef.CreateFunction(LLVMTypeRef.Void, [LLVMTypeRef.CreatePointer(llvmStructType, 0)]);
|
||||
var constructor = _llvmModule.AddFunction(StructConstructorName(structType.Module, structType.Name), constructorType);
|
||||
|
||||
_functions[StructConstructorName(structType.Module, structType.Name)] = constructor;
|
||||
}
|
||||
|
||||
foreach (var module in repository.GetAll())
|
||||
{
|
||||
foreach (var prototype in module.FunctionPrototypes)
|
||||
{
|
||||
CreateFunctionDeclaration(prototype, module.Name);
|
||||
var funcName = FuncName(module.Name, prototype.NameToken.Value, prototype.ExternSymbolToken?.Value);
|
||||
|
||||
var paramTypes = prototype.Parameters.Select(p => MapType(p.Type)).ToArray();
|
||||
var funcType = LLVMTypeRef.CreateFunction(MapType(prototype.ReturnType), paramTypes);
|
||||
var func = _llvmModule.AddFunction(funcName, funcType);
|
||||
|
||||
func.FunctionCallConv = (uint)LLVMCallConv.LLVMCCallConv;
|
||||
|
||||
_functions[funcName] = func;
|
||||
}
|
||||
}
|
||||
|
||||
// note(nub31): Define struct constructors
|
||||
foreach (var structNode in topLevelNodes.OfType<StructNode>())
|
||||
{
|
||||
EmitStructConstructor(structNode);
|
||||
var structType = _structTypes[StructName(_module, structNode.NameToken.Value)];
|
||||
var constructor = _functions[StructConstructorName(_module, structNode.NameToken.Value)];
|
||||
|
||||
var entryBlock = constructor.AppendBasicBlock("entry");
|
||||
_builder.PositionAtEnd(entryBlock);
|
||||
|
||||
var selfParam = constructor.GetParam(0);
|
||||
selfParam.Name = "self";
|
||||
|
||||
_locals.Clear();
|
||||
|
||||
foreach (var field in structNode.Fields)
|
||||
{
|
||||
if (field.Value == null) continue;
|
||||
|
||||
var index = structNode.StructType.GetFieldIndex(field.NameToken.Value);
|
||||
var fieldPtr = _builder.BuildStructGEP2(structType, selfParam, (uint)index);
|
||||
EmitExpressionInto(field.Value, fieldPtr);
|
||||
}
|
||||
|
||||
_builder.BuildRetVoid();
|
||||
}
|
||||
|
||||
// note(nub31): Define function bodies
|
||||
foreach (var funcNode in topLevelNodes.OfType<FuncNode>())
|
||||
{
|
||||
if (funcNode.Body != null)
|
||||
if (funcNode.Body == null) continue;
|
||||
|
||||
var funcName = FuncName(_module, funcNode.Prototype.NameToken.Value, funcNode.Prototype.ExternSymbolToken?.Value);
|
||||
var func = _functions[funcName];
|
||||
|
||||
var entryBlock = func.AppendBasicBlock("entry");
|
||||
_builder.PositionAtEnd(entryBlock);
|
||||
|
||||
_locals.Clear();
|
||||
|
||||
for (uint i = 0; i < funcNode.Prototype.Parameters.Count; i++)
|
||||
{
|
||||
EmitFunction(funcNode);
|
||||
var param = func.GetParam(i);
|
||||
var paramNode = funcNode.Prototype.Parameters[(int)i];
|
||||
var alloca = _builder.BuildAlloca(MapType(paramNode.Type), paramNode.NameToken.Value);
|
||||
_builder.BuildStore(param, alloca);
|
||||
_locals[paramNode.NameToken.Value] = alloca;
|
||||
}
|
||||
|
||||
EmitBlock(funcNode.Body);
|
||||
|
||||
if (funcNode.Prototype.ReturnType is NubVoidType)
|
||||
{
|
||||
if (_builder.InsertBlock.Terminator.Handle == IntPtr.Zero)
|
||||
{
|
||||
_builder.BuildRetVoid();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -89,94 +139,16 @@ public class LlvmSharpGenerator
|
||||
_builder.Dispose();
|
||||
}
|
||||
|
||||
private void CreateFunctionDeclaration(FuncPrototypeNode prototype, string moduleName)
|
||||
{
|
||||
var funcName = FuncName(moduleName, prototype.NameToken.Value, prototype.ExternSymbolToken?.Value);
|
||||
|
||||
var paramTypes = prototype.Parameters.Select(p => MapType(p.Type)).ToArray();
|
||||
var returnType = MapType(prototype.ReturnType);
|
||||
|
||||
var funcType = LLVMTypeRef.CreateFunction(returnType, paramTypes);
|
||||
var func = _llvmModule.AddFunction(funcName, funcType);
|
||||
|
||||
func.FunctionCallConv = (uint)LLVMCallConv.LLVMCCallConv;
|
||||
|
||||
for (var i = 0; i < prototype.Parameters.Count; i++)
|
||||
{
|
||||
func.GetParam((uint)i).Name = prototype.Parameters[i].NameToken.Value;
|
||||
}
|
||||
|
||||
_functions[funcName] = func;
|
||||
}
|
||||
|
||||
private void EmitStructConstructor(StructNode structNode)
|
||||
{
|
||||
var structType = _structTypes[StructName(_module, structNode.NameToken.Value)];
|
||||
var ptrType = LLVMTypeRef.CreatePointer(structType, 0);
|
||||
|
||||
var funcType = LLVMTypeRef.CreateFunction(LLVMTypeRef.Void, [ptrType]);
|
||||
var funcName = StructConstructorName(_module, structNode.NameToken.Value);
|
||||
var func = _llvmModule.AddFunction(funcName, funcType);
|
||||
func.FunctionCallConv = (uint)LLVMCallConv.LLVMCCallConv;
|
||||
|
||||
var entryBlock = func.AppendBasicBlock("entry");
|
||||
_builder.PositionAtEnd(entryBlock);
|
||||
|
||||
var selfParam = func.GetParam(0);
|
||||
selfParam.Name = "self";
|
||||
|
||||
_locals.Clear();
|
||||
|
||||
foreach (var field in structNode.Fields)
|
||||
{
|
||||
if (field.Value != null)
|
||||
{
|
||||
var index = structNode.StructType.GetFieldIndex(field.NameToken.Value);
|
||||
var fieldPtr = _builder.BuildStructGEP2(structType, selfParam, (uint)index);
|
||||
EmitExpressionInto(field.Value, fieldPtr);
|
||||
}
|
||||
}
|
||||
|
||||
_builder.BuildRetVoid();
|
||||
_functions[funcName] = func;
|
||||
}
|
||||
|
||||
private void EmitFunction(FuncNode funcNode)
|
||||
{
|
||||
var funcName = FuncName(_module, funcNode.Prototype.NameToken.Value, funcNode.Prototype.ExternSymbolToken?.Value);
|
||||
var func = _functions[funcName];
|
||||
|
||||
var entryBlock = func.AppendBasicBlock("entry");
|
||||
_builder.PositionAtEnd(entryBlock);
|
||||
|
||||
_locals.Clear();
|
||||
|
||||
for (uint i = 0; i < funcNode.Prototype.Parameters.Count; i++)
|
||||
{
|
||||
var param = func.GetParam(i);
|
||||
var paramNode = funcNode.Prototype.Parameters[(int)i];
|
||||
var alloca = _builder.BuildAlloca(MapType(paramNode.Type), paramNode.NameToken.Value);
|
||||
_builder.BuildStore(param, alloca);
|
||||
_locals[paramNode.NameToken.Value] = alloca;
|
||||
}
|
||||
|
||||
EmitBlock(funcNode.Body!);
|
||||
|
||||
if (funcNode.Prototype.ReturnType is NubVoidType)
|
||||
{
|
||||
if (_builder.InsertBlock.Terminator.Handle == IntPtr.Zero)
|
||||
{
|
||||
_builder.BuildRetVoid();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void EmitBlock(BlockNode blockNode)
|
||||
{
|
||||
_scopes.Push(new Scope());
|
||||
foreach (var statement in blockNode.Statements)
|
||||
{
|
||||
EmitStatement(statement);
|
||||
}
|
||||
|
||||
EmitScopeExit();
|
||||
_scopes.Pop();
|
||||
}
|
||||
|
||||
private void EmitStatement(StatementNode statement)
|
||||
@@ -195,6 +167,13 @@ public class LlvmSharpGenerator
|
||||
case ContinueNode:
|
||||
EmitContinue();
|
||||
break;
|
||||
case DeferNode deferNode:
|
||||
CurrentScope.Defer(() => EmitStatement(deferNode.Statement));
|
||||
break;
|
||||
case ForConstArrayNode forConstArrayNode:
|
||||
throw new NotImplementedException();
|
||||
case ForSliceNode forSliceNode:
|
||||
throw new NotImplementedException();
|
||||
case IfNode ifNode:
|
||||
EmitIf(ifNode);
|
||||
break;
|
||||
@@ -211,7 +190,7 @@ public class LlvmSharpGenerator
|
||||
EmitWhile(whileNode);
|
||||
break;
|
||||
default:
|
||||
throw new NotImplementedException($"Statement type {statement.GetType()} not implemented");
|
||||
throw new ArgumentOutOfRangeException(nameof(statement));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -270,10 +249,12 @@ public class LlvmSharpGenerator
|
||||
if (returnNode.Value != null)
|
||||
{
|
||||
var value = EmitExpression(returnNode.Value);
|
||||
EmitScopeExit();
|
||||
_builder.BuildRet(value);
|
||||
}
|
||||
else
|
||||
{
|
||||
EmitScopeExit();
|
||||
_builder.BuildRetVoid();
|
||||
}
|
||||
}
|
||||
@@ -512,7 +493,10 @@ public class LlvmSharpGenerator
|
||||
{
|
||||
var funcPtr = EmitExpression(funcCall.Expression);
|
||||
var args = funcCall.Parameters.Select(x => EmitExpression(x)).ToArray();
|
||||
return _builder.BuildCall2(MapType(funcCall.Expression.Type), funcPtr, args, funcCall.Type is NubVoidType ? "" : "call");
|
||||
|
||||
var functionType = (NubFuncType)funcCall.Expression.Type;
|
||||
var llvmFunctionType = LLVMTypeRef.CreateFunction(MapType(functionType.ReturnType), functionType.Parameters.Select(MapType).ToArray());
|
||||
return _builder.BuildCall2(llvmFunctionType, funcPtr, args, funcCall.Type is NubVoidType ? "" : "call");
|
||||
}
|
||||
|
||||
private LLVMValueRef EmitStructFieldAccess(StructFieldAccessNode field)
|
||||
@@ -733,13 +717,14 @@ public class LlvmSharpGenerator
|
||||
NubBoolType => LLVMTypeRef.Int1,
|
||||
NubIntType intType => LLVMTypeRef.CreateInt((uint)intType.Width),
|
||||
NubFloatType floatType => floatType.Width == 32 ? LLVMTypeRef.Float : LLVMTypeRef.Double,
|
||||
NubFuncType funcType => LLVMTypeRef.CreateFunction(MapType(funcType.ReturnType), funcType.Parameters.Select(MapType).ToArray()),
|
||||
NubFuncType funcType => LLVMTypeRef.CreatePointer(LLVMTypeRef.CreateFunction(MapType(funcType.ReturnType), funcType.Parameters.Select(MapType).ToArray()), 0),
|
||||
NubPointerType ptrType => LLVMTypeRef.CreatePointer(MapType(ptrType.BaseType), 0),
|
||||
NubSliceType nubSliceType => MapSliceType(nubSliceType),
|
||||
NubStringType => _structTypes["nub.string"],
|
||||
NubArrayType arrType => LLVMTypeRef.CreatePointer(MapType(arrType.ElementType), 0),
|
||||
NubConstArrayType constArr => LLVMTypeRef.CreateArray(MapType(constArr.ElementType), (uint)constArr.Size),
|
||||
NubStructType structType => _structTypes[StructName(structType.Module, structType.Name)],
|
||||
NubEnumType enumType => MapType(enumType.UnderlyingType),
|
||||
NubVoidType => LLVMTypeRef.Void,
|
||||
_ => throw new ArgumentOutOfRangeException(nameof(type), type, null)
|
||||
};
|
||||
@@ -778,4 +763,28 @@ public class LlvmSharpGenerator
|
||||
|
||||
return $"{module}.{name}";
|
||||
}
|
||||
|
||||
private void EmitScopeExit()
|
||||
{
|
||||
var deferredActions = CurrentScope.GetDeferredActions();
|
||||
while (deferredActions.TryPop(out var action))
|
||||
{
|
||||
action.Invoke();
|
||||
}
|
||||
}
|
||||
|
||||
private class Scope
|
||||
{
|
||||
private readonly Stack<Action> _deferredActions = [];
|
||||
|
||||
public Stack<Action> GetDeferredActions()
|
||||
{
|
||||
return _deferredActions;
|
||||
}
|
||||
|
||||
public void Defer(Action action)
|
||||
{
|
||||
_deferredActions.Push(action);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user