func (v *Codegen) addNamedTypeReference(n *ast.TypeReference, gcon *ast.GenericContext) { outer := gcon gcon = ast.NewGenericContextFromTypeReference(n) gcon.Outer = outer v.addNamedType(n.BaseType.(*ast.NamedType), ast.TypeReferenceMangledName(ast.MANGLE_ARK_UNSTABLE, n, gcon), gcon) }
func (v *Codegen) typeRefToLLVMTypeWithGenericContext(typ *ast.TypeReference, gcon *ast.GenericContext) llvm.Type { switch nt := typ.BaseType.(type) { case *ast.NamedType: switch nt.ActualType().(type) { case ast.StructType, ast.EnumType: v.addNamedTypeReference(typ, gcon) lt := v.namedTypeLookup[ast.TypeReferenceMangledName(ast.MANGLE_ARK_UNSTABLE, typ, gcon)] return lt default: return v.typeToLLVMType(nt.Type, gcon) } } return v.typeToLLVMType(typ.BaseType, gcon) }
func (v *Codegen) typeToLLVMType(typ ast.Type, gcon *ast.GenericContext) llvm.Type { switch typ := typ.(type) { case ast.PrimitiveType: return v.primitiveTypeToLLVMType(typ) case ast.FunctionType: return v.functionTypeToLLVMType(typ, true, gcon) case ast.StructType: return v.structTypeToLLVMType(typ, gcon) case ast.PointerType: return llvm.PointerType(v.typeRefToLLVMTypeWithOuter(typ.Addressee, gcon), 0) case ast.ArrayType: return v.arrayTypeToLLVMType(typ, gcon) case ast.TupleType: return v.tupleTypeToLLVMType(typ, gcon) case ast.EnumType: return v.enumTypeToLLVMType(typ, gcon) case ast.ReferenceType: return llvm.PointerType(v.typeRefToLLVMTypeWithOuter(typ.Referrer, gcon), 0) case *ast.NamedType: switch typ.Type.(type) { case ast.StructType, ast.EnumType: // If something seems wrong with the code and thie problems seems to come from here, // make sure the type doesn't need generics arguments as well. // This here ignores them. name := ast.TypeReferenceMangledName(ast.MANGLE_ARK_UNSTABLE, &ast.TypeReference{BaseType: typ}, gcon) v.addNamedType(typ, name, gcon) lt := v.namedTypeLookup[name] return lt default: return v.typeToLLVMType(typ.Type, gcon) } case *ast.SubstitutionType: if gcon == nil { panic("gcon == nil when getting substitution type") } if gcon.GetSubstitutionType(typ) == nil { panic("missing generic map entry for type " + typ.TypeName() + " " + fmt.Sprintf("(%p)", typ)) } return v.typeRefToLLVMTypeWithOuter(gcon.GetSubstitutionType(typ), gcon) default: log.Debugln("codegen", "Type was %s (%s)", typ.TypeName(), reflect.TypeOf(typ)) panic("Unimplemented type category in LLVM codegen") } }
func (v *Codegen) addEnumType(typ ast.EnumType, name string, gcon *ast.GenericContext) { if _, ok := v.namedTypeLookup[name]; ok { return } if typ.Simple { // TODO: Handle other integer size, maybe dynamic depending on max value? v.namedTypeLookup[name] = llvm.IntType(32) } else { enum := v.curFile.LlvmModule.Context().StructCreateNamed(name) v.namedTypeLookup[name] = enum for _, member := range typ.Members { if named, ok := member.Type.(*ast.NamedType); ok { v.addNamedType(named, ast.TypeReferenceMangledName(ast.MANGLE_ARK_UNSTABLE, &ast.TypeReference{BaseType: member.Type}, gcon), gcon) } } enum.StructSetBody(v.enumTypeToLLVMTypeFields(typ, gcon), false) } }