1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#include "cps/CodeGen.h"
#include "cps/Operator.h"
using namespace cps;
using namespace llvm;
const TypeInfo *CodeGen::requireStringType() {
const TypeInfo *Type = resolveType("STRING");
if (!Type || !Type->IsClass) {
reportError("String literals require a visible STRING class; include the string library or omit -no-std");
return nullptr;
}
return Type;
}
Value *CodeGen::emitStringFromCString(Value *Text) {
const TypeInfo *Type = requireStringType();
if (!Type || !Text) return nullptr;
const FuncSig *Constructor = nullptr;
for (const FuncSig *Sig : FuncGen->getOverloads(Type->Name + ".NEW"))
if (Sig->Params.size() == 1 && !Sig->Params[0].IsArray && !Sig->Params[0].IsByRef &&
resolveType("C_STRING")->sameType(resolveType(Sig->Params[0].TypeName)))
Constructor = Sig;
if (!Constructor) {
reportError("STRING requires a constructor NEW(Text : C_STRING)");
return nullptr;
}
if (!checkOperatorAccess(*Constructor)) return nullptr;
auto *Storage = CreateEntryBlockAlloca(Builder->GetInsertBlock()->getParent(),
Type->LLVMType, "string_value");
Builder->CreateStore(Types->getZeroValue(Type->Name), Storage);
if (!emitFieldDefaults(Type, Storage, DiagnosticLine)) return nullptr;
if (!FuncGen->emitCall(*Constructor, {Storage, Text})) return nullptr;
// The consumer takes ownership, just as with an ordinary class result.
return Builder->CreateLoad(Type->LLVMType, Storage, "string");
}
// STRING -> CHAR: the string must hold exactly one character.
Value *CodeGen::emitCharFromString(ExprAST *Expr, int Line) {
const TypeInfo *Type = requireStringType();
if (!Type) return nullptr;
const TypeInfo *Source = getExprTypeInfo(Expr);
Value *Text = emitCoercedExpr(Expr, Type);
if (!Text) return nullptr;
if (!isObjectLValue(Expr) || !Source || !Source->sameType(Type))
materializeOperatorReceiver(Text, Type);
auto *Receiver = CreateEntryBlockAlloca(Builder->GetInsertBlock()->getParent(),
Type->LLVMType, "string_view");
Builder->CreateStore(Text, Receiver);
const FuncSig *Method = resolveMethod(Type->Name, "_LENGTH", {});
if (!Method || !resolveType("INTEGER")->sameType(resolveType(Method->ReturnTypeName))) {
reportError("STRING requires a method _LENGTH() RETURNS INTEGER");
return nullptr;
}
Value *Length = emitMethodCall(Type->Name, Receiver, "_LENGTH", {}, Line);
if (!Length) return nullptr;
Value *Wrong = Builder->CreateICmpNE(Length, ConstantInt::get(Length->getType(), 1), "char_len_bad");
Value *Msg = Builder->CreateGlobalString("[Fatal] line %d: STRING of length %lld cannot be converted to CHAR\n",
"err_string_to_char", 0, TheModule.get());
RuntimeChecker->emitCheck(Wrong, Msg, {RuntimeChecker->lineValue(Line), Length});
Value *Ptr = emitMethodCall(Type->Name, Receiver, "C_STR", {}, Line);
if (!Ptr) return nullptr;
return Builder->CreateLoad(resolveType("CHAR")->LLVMType, Ptr, "char_value");
}
Value *CodeGen::emitStringCString(Value *Value) {
const TypeInfo *Type = requireStringType();
if (!Type || !Value) return nullptr;
const FuncSig *Method = resolveMethod(Type->Name, "C_STR", {});
if (!Method || !resolveType("C_STRING")->sameType(resolveType(Method->ReturnTypeName))) {
reportError("STRING requires a method C_STR() RETURNS C_STRING");
return nullptr;
}
auto *Receiver = CreateEntryBlockAlloca(Builder->GetInsertBlock()->getParent(),
Type->LLVMType, "string_view");
Builder->CreateStore(Value, Receiver);
return emitMethodCall(Type->Name, Receiver, "C_STR", {}, DiagnosticLine);
}
Value *CodeGen::emitCStringExpr(ExprAST *Expr) {
const TypeInfo *Source = getExprTypeInfo(Expr);
const TypeInfo *Type = requireStringType();
if (!Type || !Source || isWholeArrayVar(Expr) ||
(!Source->sameType(Type) && !findImplicitConstructor(Source, Type))) {
reportError("Expected a STRING value or an IMPLICIT conversion to STRING");
return nullptr;
}
Value *Value = emitCoercedExpr(Expr, Type);
if (!Value) return nullptr;
if (!isObjectLValue(Expr) || !Source->sameType(Type)) materializeOperatorReceiver(Value, Type);
return emitStringCString(Value);
}
bool CodeGen::containsString(const TypeInfo *Type) const {
if (!Type) return false;
if (Type->isString()) return true;
if (Type->asRecord())
for (const auto &[Name, Field] : Type->Record.Fields)
if (containsString(resolveType(Field.TypeName))) return true;
return false;
}
|
