venko
/
wraith-lang
forked from cosmonaut/wraith-lang
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

improve field storage pattern

generics
cosmonaut 2021-04-20 19:07:11 -07:00
parent 2708dfbbed
commit 180583d772
1 changed files with 36 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
compiler.c
View File

@ -70,6 +70,26 @@ static void AddStructField(LLVMBuilderRef builder, LLVMValueRef wStructPointer,
} }
} }
static void MarkStructFieldForWrite(LLVMValueRef wStructPointer, LLVMValueRef value)
{
uint32_t i, j;
for (i = 0; i < structFieldMapCount; i += 1)
{
if (structFieldMaps[i].structPointer == wStructPointer)
{
for (j = 0; j < structFieldMaps[i].fieldCount; j += 1)
{
if (structFieldMaps[i].fields[j].value == value)
{
structFieldMaps[i].fields[j].needsWrite = 1;
break;
}
}
}
}
}
static LLVMValueRef GetStructFieldPointer(LLVMValueRef wStructPointer, LLVMValueRef value) static LLVMValueRef GetStructFieldPointer(LLVMValueRef wStructPointer, LLVMValueRef value)
{ {
uint32_t i, j; uint32_t i, j;
@ -91,14 +111,26 @@ static LLVMValueRef GetStructFieldPointer(LLVMValueRef wStructPointer, LLVMValue
return NULL; return NULL;
} }
static void RemoveStruct(LLVMValueRef wStructPointer) static void RemoveStruct(LLVMBuilderRef builder, LLVMValueRef wStructPointer)
{ {
uint32_t i; uint32_t i, j;
for (i = 0; i < structFieldMapCount; i += 1) for (i = 0; i < structFieldMapCount; i += 1)
{ {
if (structFieldMaps[i].structPointer == wStructPointer) if (structFieldMaps[i].structPointer == wStructPointer)
{ {
for (j = 0; j < structFieldMaps[i].fieldCount; j += 1)
{
if (structFieldMaps[i].fields[j].needsWrite)
{
LLVMBuildStore(
builder,
structFieldMaps[i].fields[j].value,
structFieldMaps[i].fields[j].valuePointer
);
}
}
free(structFieldMaps[i].fields); free(structFieldMaps[i].fields);
structFieldMaps[i].fields = NULL; structFieldMaps[i].fields = NULL;
structFieldMaps[i].fieldCount = 0; structFieldMaps[i].fieldCount = 0;
@ -281,11 +313,7 @@ static void CompileAssignment(LLVMValueRef wStructValue, LLVMBuilderRef builder,
LLVMValueRef result = CompileExpression(wStructValue, builder, function, assignmentStatement->children[1]); LLVMValueRef result = CompileExpression(wStructValue, builder, function, assignmentStatement->children[1]);
LLVMValueRef identifier = CompileExpression(wStructValue, builder, function, assignmentStatement->children[0]); LLVMValueRef identifier = CompileExpression(wStructValue, builder, function, assignmentStatement->children[0]);
fieldPointer = GetStructFieldPointer(wStructValue, identifier); MarkStructFieldForWrite(wStructValue, identifier);
if (fieldPointer != NULL)
{
LLVMBuildStore(builder, result, fieldPointer);
}
} }
static uint8_t CompileStatement(LLVMValueRef wStructValue, LLVMBuilderRef builder, LLVMValueRef function, Node *statement) static uint8_t CompileStatement(LLVMValueRef wStructValue, LLVMBuilderRef builder, LLVMValueRef function, Node *statement)
@ -360,7 +388,7 @@ static void CompileFunction(
hasReturn |= CompileStatement(wStructPointer, builder, function, functionBody->children[i]); hasReturn |= CompileStatement(wStructPointer, builder, function, functionBody->children[i]);
} }
AddNamedVariable(functionSignature->children[0]->value.string, function); RemoveStruct(builder, wStructPointer);
if (LLVMGetTypeKind(returnType) == LLVMVoidTypeKind && !hasReturn) if (LLVMGetTypeKind(returnType) == LLVMVoidTypeKind && !hasReturn)
{ {
@ -370,8 +398,6 @@ static void CompileFunction(
{ {
fprintf(stderr, "Return statement not provided!"); fprintf(stderr, "Return statement not provided!");
} }
RemoveStruct(wStructPointer);
} }
static void CompileStruct(LLVMModuleRef module, LLVMContextRef context, Node *node) static void CompileStruct(LLVMModuleRef module, LLVMContextRef context, Node *node)