84 lines
2.7 KiB
C
84 lines
2.7 KiB
C
#include "typeutils.h"
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
void ConvertIdCustomsToGenerics(IdNode *node) {
|
|
uint32_t i;
|
|
switch(node->type)
|
|
{
|
|
case UnorderedScope:
|
|
case OrderedScope:
|
|
case Struct:
|
|
/* FIXME: This case will need to be modified to handle type parameters over structs. */
|
|
for (i = 0; i < node->childCount; i += 1) {
|
|
ConvertIdCustomsToGenerics(node->children[i]);
|
|
}
|
|
return;
|
|
|
|
case Variable: {
|
|
TypeTag *varType = node->typeTag;
|
|
if (varType->type == Custom) {
|
|
IdNode *x = LookupId(node->parent, node, varType->value.customType);
|
|
if (x != NULL && x->type == GenericType) {
|
|
varType->type = Generic;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
case Function: {
|
|
TypeTag *funcType = node->typeTag;
|
|
if (funcType->type == Custom) {
|
|
/* For functions we have to handle the type lookup manually since the generic type
|
|
* identifiers are declared as children of the function's IdNode. */
|
|
for (i = 0; i < node->childCount; i += 1) {
|
|
IdNode *child = node->children[i];
|
|
if (child->type == GenericType && strcmp(child->name, funcType->value.customType) == 0) {
|
|
funcType->type = Generic;
|
|
}
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < node->childCount; i += 1) {
|
|
ConvertIdCustomsToGenerics(node->children[i]);
|
|
}
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* FIXME: This function will need to be modified to handle type parameters over structs. */
|
|
void ConvertASTCustomsToGenerics(Node *node) {
|
|
uint32_t i;
|
|
switch (node->syntaxKind)
|
|
{
|
|
case Declaration:
|
|
{
|
|
Node *type = node->declaration.type->type.typeNode;
|
|
Node *id = node->declaration.identifier;
|
|
if (id->typeTag->type == Generic && type->syntaxKind == CustomTypeNode) {
|
|
free(node->declaration.type);
|
|
node->declaration.type = MakeGenericTypeNode(id->typeTag->value.genericType);
|
|
}
|
|
return;
|
|
}
|
|
|
|
case FunctionSignature:
|
|
{
|
|
Node *id = node->functionSignature.identifier;
|
|
Node *type = node->functionSignature.type;
|
|
if (id->typeTag->type == Generic && type->syntaxKind == CustomTypeNode) {
|
|
free(node->functionSignature.type);
|
|
node->functionSignature.type = MakeGenericTypeNode(id->typeTag->value.genericType);
|
|
}
|
|
ConvertASTCustomsToGenerics(node->functionSignature.arguments);
|
|
return;
|
|
}
|
|
|
|
default:
|
|
recurse(node, *ConvertASTCustomsToGenerics);
|
|
}
|
|
}
|