123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216 |
- #include "entity.cpp"
- #include "types.cpp"
- void check_expr(Checker *c, Operand *operand, AstNode *expression);
- bool is_operand_value(Operand o) {
- switch (o.mode) {
- case Addressing_Value:
- case Addressing_Variable:
- case Addressing_Immutable:
- case Addressing_Constant:
- case Addressing_MapIndex:
- return true;
- }
- return false;
- }
- bool is_operand_nil(Operand o) {
- return o.mode == Addressing_Value && o.type == t_untyped_nil;
- }
- bool is_operand_undef(Operand o) {
- return o.mode == Addressing_Value && o.type == t_untyped_undef;
- }
- gb_global Scope *universal_scope = nullptr;
- void scope_reset(Scope *scope) {
- if (scope == nullptr) return;
- scope->first_child = nullptr;
- scope->last_child = nullptr;
- map_clear (&scope->elements);
- array_clear (&scope->shared);
- ptr_set_clear(&scope->implicit);
- ptr_set_clear(&scope->imported);
- ptr_set_clear(&scope->exported);
- }
- i32 is_scope_an_ancestor(Scope *parent, Scope *child) {
- i32 i = 0;
- while (child != nullptr) {
- if (parent == child) {
- return i;
- }
- child = child->parent;
- i++;
- }
- return -1;
- }
- void entity_graph_node_set_destroy(EntityGraphNodeSet *s) {
- if (s->hashes.data != nullptr) {
- ptr_set_destroy(s);
- }
- }
- void entity_graph_node_set_add(EntityGraphNodeSet *s, EntityGraphNode *n) {
- if (s->hashes.data == nullptr) {
- ptr_set_init(s, heap_allocator());
- }
- ptr_set_add(s, n);
- }
- bool entity_graph_node_set_exists(EntityGraphNodeSet *s, EntityGraphNode *n) {
- return ptr_set_exists(s, n);
- }
- void entity_graph_node_set_remove(EntityGraphNodeSet *s, EntityGraphNode *n) {
- ptr_set_remove(s, n);
- }
- void entity_graph_node_destroy(EntityGraphNode *n, gbAllocator a) {
- entity_graph_node_set_destroy(&n->pred);
- entity_graph_node_set_destroy(&n->succ);
- gb_free(a, n);
- }
- int entity_graph_node_cmp(EntityGraphNode **data, isize i, isize j) {
- EntityGraphNode *x = data[i];
- EntityGraphNode *y = data[j];
- isize a = x->entity->order_in_src;
- isize b = y->entity->order_in_src;
- if (x->dep_count < y->dep_count) return -1;
- if (x->dep_count > y->dep_count) return +1;
- return a < b ? -1 : b > a;
- }
- void entity_graph_node_swap(EntityGraphNode **data, isize i, isize j) {
- EntityGraphNode *x = data[i];
- EntityGraphNode *y = data[j];
- data[i] = y;
- data[j] = x;
- x->index = j;
- y->index = i;
- }
- void import_graph_node_set_destroy(ImportGraphNodeSet *s) {
- if (s->hashes.data != nullptr) {
- ptr_set_destroy(s);
- }
- }
- void import_graph_node_set_add(ImportGraphNodeSet *s, ImportGraphNode *n) {
- if (s->hashes.data == nullptr) {
- ptr_set_init(s, heap_allocator());
- }
- ptr_set_add(s, n);
- }
- bool import_graph_node_set_exists(ImportGraphNodeSet *s, ImportGraphNode *n) {
- return ptr_set_exists(s, n);
- }
- void import_graph_node_set_remove(ImportGraphNodeSet *s, ImportGraphNode *n) {
- ptr_set_remove(s, n);
- }
- ImportGraphNode *import_graph_node_create(gbAllocator a, Scope *scope) {
- ImportGraphNode *n = gb_alloc_item(a, ImportGraphNode);
- n->scope = scope;
- n->path = scope->file->tokenizer.fullpath;
- n->file_id = scope->file->id;
- return n;
- }
- void import_graph_node_destroy(ImportGraphNode *n, gbAllocator a) {
- import_graph_node_set_destroy(&n->pred);
- import_graph_node_set_destroy(&n->succ);
- gb_free(a, n);
- }
- int import_graph_node_cmp(ImportGraphNode **data, isize i, isize j) {
- ImportGraphNode *x = data[i];
- ImportGraphNode *y = data[j];
- GB_ASSERT(x != y);
- GB_ASSERT(x->scope != y->scope);
- bool xg = x->scope->is_global;
- bool yg = y->scope->is_global;
- if (xg != yg) return xg ? -1 : +1;
- if (xg && yg) return x->file_id < y->file_id ? +1 : -1;
- if (x->dep_count < y->dep_count) return -1;
- if (x->dep_count > y->dep_count) return +1;
- return 0;
- }
- void import_graph_node_swap(ImportGraphNode **data, isize i, isize j) {
- ImportGraphNode *x = data[i];
- ImportGraphNode *y = data[j];
- data[i] = y;
- data[j] = x;
- x->index = j;
- y->index = i;
- }
- GB_COMPARE_PROC(ast_node_cmp) {
- AstNode *x = *cast(AstNode **)a;
- AstNode *y = *cast(AstNode **)b;
- Token i = ast_node_token(x);
- Token j = ast_node_token(y);
- return token_pos_cmp(i.pos, j.pos);
- }
- void init_declaration_info(DeclInfo *d, Scope *scope, DeclInfo *parent) {
- d->parent = parent;
- d->scope = scope;
- ptr_set_init(&d->deps, heap_allocator());
- array_init (&d->labels, heap_allocator());
- }
- DeclInfo *make_declaration_info(gbAllocator a, Scope *scope, DeclInfo *parent) {
- DeclInfo *d = gb_alloc_item(a, DeclInfo);
- init_declaration_info(d, scope, parent);
- return d;
- }
- void destroy_declaration_info(DeclInfo *d) {
- ptr_set_destroy(&d->deps);
- array_free(&d->labels);
- }
- bool decl_info_has_init(DeclInfo *d) {
- if (d->init_expr != nullptr) {
- return true;
- }
- if (d->proc_lit != nullptr) {
- switch (d->proc_lit->kind) {
- case_ast_node(pl, ProcLit, d->proc_lit);
- if (pl->body != nullptr) {
- return true;
- }
- case_end;
- }
- }
- return false;
- }
- Scope *create_scope(Scope *parent, gbAllocator allocator) {
- Scope *s = gb_alloc_item(allocator, Scope);
- s->parent = parent;
- map_init(&s->elements, heap_allocator());
- array_init(&s->shared, heap_allocator());
- ptr_set_init(&s->implicit, heap_allocator());
- ptr_set_init(&s->imported, heap_allocator());
- ptr_set_init(&s->exported, heap_allocator());
- if (parent != nullptr && parent != universal_scope) {
- DLIST_APPEND(parent->first_child, parent->last_child, s);
- }
- return s;
- }
- Scope *create_scope_from_file(Checker *c, AstFile *f) {
- GB_ASSERT(f != nullptr);
- Scope *s = create_scope(c->global_scope, c->allocator);
- array_init(&s->delayed_file_decls, heap_allocator());
- s->file = f;
- f->scope = s;
- s->is_file = true;
- if (f->tokenizer.fullpath == c->parser->init_fullpath) {
- s->is_init = true;
- } else {
- s->is_init = f->file_kind == ImportedFile_Init;
- }
- s->is_global = f->is_global_scope;
- if (s->is_global) array_add(&c->global_scope->shared, s);
- if (s->is_init || s->is_global) {
- s->has_been_imported = true;
- }
- return s;
- }
- void destroy_scope(Scope *scope) {
- for_array(i, scope->elements.entries) {
- Entity *e =scope->elements.entries[i].value;
- if (e->kind == Entity_Variable) {
- if (!(e->flags & EntityFlag_Used)) {
- #if 0
- warning(e->token, "Unused variable '%.*s'", LIT(e->token.string));
- #endif
- }
- }
- }
- for (Scope *child = scope->first_child; child != nullptr; child = child->next) {
- destroy_scope(child);
- }
- map_destroy(&scope->elements);
- array_free(&scope->shared);
- array_free(&scope->delayed_file_decls);
- array_free(&scope->delayed_asserts);
- ptr_set_destroy(&scope->implicit);
- ptr_set_destroy(&scope->imported);
- ptr_set_destroy(&scope->exported);
- // NOTE(bill): No need to free scope as it "should" be allocated in an arena (except for the global scope)
- }
- void add_scope(Checker *c, AstNode *node, Scope *scope) {
- GB_ASSERT(node != nullptr);
- GB_ASSERT(scope != nullptr);
- scope->node = node;
- node->scope = scope;
- }
- void check_open_scope(Checker *c, AstNode *node) {
- node = unparen_expr(node);
- GB_ASSERT(node->kind == AstNode_Invalid ||
- is_ast_node_stmt(node) ||
- is_ast_node_type(node));
- Scope *scope = create_scope(c->context.scope, c->allocator);
- add_scope(c, node, scope);
- switch (node->kind) {
- case AstNode_ProcType:
- scope->is_proc = true;
- break;
- case AstNode_StructType:
- case AstNode_EnumType:
- case AstNode_UnionType:
- scope->is_struct = true;
- break;
- }
- c->context.scope = scope;
- c->context.stmt_state_flags |= StmtStateFlag_bounds_check;
- }
- void check_close_scope(Checker *c) {
- c->context.scope = c->context.scope->parent;
- }
- Entity *current_scope_lookup_entity(Scope *s, String name) {
- HashKey key = hash_string(name);
- Entity **found = map_get(&s->elements, key);
- if (found) {
- return *found;
- }
- for_array(i, s->shared) {
- Scope *shared = s->shared[i];
- Entity **found = map_get(&shared->elements, key);
- if (found) {
- Entity *e = *found;
- if (e->kind == Entity_Variable &&
- !e->scope->is_file &&
- !e->scope->is_global) {
- continue;
- }
- if (e->scope != shared) {
- // Do not return imported entities even #include ones
- continue;
- }
- return e;
- }
- }
- return nullptr;
- }
- void scope_lookup_parent_entity(Scope *scope, String name, Scope **scope_, Entity **entity_) {
- bool gone_thru_proc = false;
- bool gone_thru_file = false;
- HashKey key = hash_string(name);
- for (Scope *s = scope; s != nullptr; s = s->parent) {
- Entity **found = map_get(&s->elements, key);
- if (found) {
- Entity *e = *found;
- if (gone_thru_proc) {
- // IMPORTANT TODO(bill): Is this correct?!
- if (e->kind == Entity_Label) {
- continue;
- }
- if (e->kind == Entity_Variable &&
- !e->scope->is_file &&
- !e->scope->is_global) {
- continue;
- }
- }
- if (entity_) *entity_ = e;
- if (scope_) *scope_ = s;
- return;
- }
- if (s->is_proc) {
- gone_thru_proc = true;
- } else {
- // Check shared scopes - i.e. other files @ global scope
- for_array(i, s->shared) {
- Scope *shared = s->shared[i];
- Entity **found = map_get(&shared->elements, key);
- if (found) {
- Entity *e = *found;
- if (e->kind == Entity_Variable &&
- !e->scope->is_file &&
- !e->scope->is_global) {
- continue;
- }
- if (e->scope != shared) {
- // Do not return imported entities even #include ones
- continue;
- }
- if ((e->kind == Entity_ImportName ||
- e->kind == Entity_LibraryName)
- && gone_thru_file) {
- continue;
- }
- if (entity_) *entity_ = e;
- if (scope_) *scope_ = shared;
- return;
- }
- }
- }
- if (s->is_file) {
- gone_thru_file = true;
- }
- }
- if (entity_) *entity_ = nullptr;
- if (scope_) *scope_ = nullptr;
- }
- Entity *scope_lookup_entity(Scope *s, String name) {
- Entity *entity = nullptr;
- scope_lookup_parent_entity(s, name, nullptr, &entity);
- return entity;
- }
- Entity *scope_insert_entity(Scope *s, Entity *entity) {
- String name = entity->token.string;
- if (name == "") {
- return nullptr;
- }
- HashKey key = hash_string(name);
- Entity **found = map_get(&s->elements, key);
- if (found) {
- return *found;
- }
- map_set(&s->elements, key, entity);
- if (entity->scope == nullptr) {
- entity->scope = s;
- }
- return nullptr;
- }
- GB_COMPARE_PROC(entity_variable_pos_cmp) {
- Entity *x = *cast(Entity **)a;
- Entity *y = *cast(Entity **)b;
- return token_pos_cmp(x->token.pos, y->token.pos);
- }
- void check_scope_usage(Checker *c, Scope *scope) {
- // TODO(bill): Use this?
- #if 0
- Array<Entity *> unused = {};
- array_init(&unused, heap_allocator());
- defer (array_free(&unused));
- for_array(i, scope->elements.entries) {
- Entity *e = scope->elements.entries[i].value;
- if (e != nullptr && e->kind == Entity_Variable && (e->flags&EntityFlag_Used) == 0) {
- array_add(&unused, e);
- }
- }
- gb_sort_array(unused.data, unused.count, entity_variable_pos_cmp);
- for_array(i, unused) {
- Entity *e = unused[i];
- error(e->token, "'%.*s' declared but not used", LIT(e->token.string));
- }
- for (Scope *child = scope->first_child;
- child != nullptr;
- child = child->next) {
- if (!child->is_proc && !child->is_struct && !child->is_file) {
- check_scope_usage(c, child);
- }
- }
- #endif
- }
- void add_dependency(DeclInfo *d, Entity *e) {
- ptr_set_add(&d->deps, e);
- }
- void add_preload_dependency(Checker *c, char *name) {
- String n = make_string_c(name);
- Entity *e = scope_lookup_entity(c->global_scope, n);
- GB_ASSERT(e != nullptr);
- ptr_set_add(&c->context.decl->deps, e);
- }
- void add_declaration_dependency(Checker *c, Entity *e) {
- if (e == nullptr) {
- return;
- }
- if (c->context.decl != nullptr) {
- // DeclInfo *decl = decl_info_of_entity(&c->info, e);
- add_dependency(c->context.decl, e);
- }
- }
- Entity *add_global_entity(Entity *entity) {
- String name = entity->token.string;
- if (gb_memchr(name.text, ' ', name.len)) {
- return entity; // NOTE(bill): 'untyped thing'
- }
- if (scope_insert_entity(universal_scope, entity)) {
- compiler_error("double declaration");
- }
- entity->state = EntityState_Resolved;
- return entity;
- }
- void add_global_constant(String name, Type *type, ExactValue value) {
- Entity *entity = alloc_entity(Entity_Constant, nullptr, make_token_ident(name), type);
- entity->Constant.value = value;
- add_global_entity(entity);
- }
- void add_global_string_constant(String name, String value) {
- add_global_constant(name, t_untyped_string, exact_value_string(value));
- }
- void add_global_type_entity(String name, Type *type) {
- add_global_entity(alloc_entity_type_name(nullptr, make_token_ident(name), type));
- }
- void init_universal_scope(void) {
- BuildContext *bc = &build_context;
- // NOTE(bill): No need to free these
- gbAllocator a = heap_allocator();
- universal_scope = create_scope(nullptr, a);
- // Types
- for (isize i = 0; i < gb_count_of(basic_types); i++) {
- add_global_type_entity(basic_types[i].Basic.name, &basic_types[i]);
- }
- add_global_type_entity(str_lit("byte"), &basic_types[Basic_u8]);
- // Constants
- add_global_constant(str_lit("true"), t_untyped_bool, exact_value_bool(true));
- add_global_constant(str_lit("false"), t_untyped_bool, exact_value_bool(false));
- add_global_entity(alloc_entity_nil(str_lit("nil"), t_untyped_nil));
- add_global_entity(alloc_entity_library_name(universal_scope,
- make_token_ident(str_lit("__llvm_core")), t_invalid,
- str_lit(""), str_lit("__llvm_core")));
- // TODO(bill): Set through flags in the compiler
- add_global_string_constant(str_lit("ODIN_OS"), bc->ODIN_OS);
- add_global_string_constant(str_lit("ODIN_ARCH"), bc->ODIN_ARCH);
- add_global_string_constant(str_lit("ODIN_ENDIAN"), bc->ODIN_ENDIAN);
- add_global_string_constant(str_lit("ODIN_VENDOR"), bc->ODIN_VENDOR);
- add_global_string_constant(str_lit("ODIN_VERSION"), bc->ODIN_VERSION);
- add_global_string_constant(str_lit("ODIN_ROOT"), bc->ODIN_ROOT);
- add_global_constant(str_lit("ODIN_DEBUG"), t_untyped_bool, exact_value_bool(bc->ODIN_DEBUG));
- // Builtin Procedures
- for (isize i = 0; i < gb_count_of(builtin_procs); i++) {
- BuiltinProcId id = cast(BuiltinProcId)i;
- String name = builtin_procs[i].name;
- if (name != "") {
- Entity *entity = alloc_entity(Entity_Builtin, nullptr, make_token_ident(name), t_invalid);
- entity->Builtin.id = id;
- add_global_entity(entity);
- }
- }
- t_u8_ptr = alloc_type_pointer(t_u8);
- t_int_ptr = alloc_type_pointer(t_int);
- t_i64_ptr = alloc_type_pointer(t_i64);
- t_f64_ptr = alloc_type_pointer(t_f64);
- t_u8_slice = alloc_type_slice(t_u8);
- t_string_slice = alloc_type_slice(t_string);
- }
- void init_checker_info(CheckerInfo *i) {
- gbAllocator a = heap_allocator();
- map_init(&i->types, a);
- array_init(&i->definitions, a);
- array_init(&i->entities, a);
- map_init(&i->untyped, a);
- map_init(&i->foreigns, a);
- map_init(&i->gen_procs, a);
- map_init(&i->gen_types, a);
- array_init(&i->type_info_types, a);
- map_init(&i->type_info_map, a);
- map_init(&i->files, a);
- array_init(&i->variable_init_order, a);
- }
- void destroy_checker_info(CheckerInfo *i) {
- map_destroy(&i->types);
- array_free(&i->definitions);
- array_free(&i->entities);
- map_destroy(&i->untyped);
- map_destroy(&i->foreigns);
- map_destroy(&i->gen_procs);
- map_destroy(&i->gen_types);
- array_free(&i->type_info_types);
- map_destroy(&i->type_info_map);
- map_destroy(&i->files);
- array_free(&i->variable_init_order);
- }
- void init_checker(Checker *c, Parser *parser) {
- if (global_error_collector.count > 0) {
- gb_exit(1);
- }
- gbAllocator a = heap_allocator();
- c->parser = parser;
- init_checker_info(&c->info);
- gb_mutex_init(&c->mutex);
- array_init(&c->proc_stack, a);
- array_init(&c->procs, a);
- // NOTE(bill): Is this big enough or too small?
- isize item_size = gb_max3(gb_size_of(Entity), gb_size_of(Type), gb_size_of(Scope));
- isize total_token_count = 0;
- for_array(i, c->parser->files) {
- AstFile *f = c->parser->files[i];
- total_token_count += f->tokens.count;
- }
- isize arena_size = 2 * item_size * total_token_count;
- gb_arena_init_from_allocator(&c->tmp_arena, a, arena_size);
- gb_arena_init_from_allocator(&c->arena, a, arena_size);
- // c->allocator = pool_allocator(&c->pool);
- c->allocator = heap_allocator();
- // c->allocator = gb_arena_allocator(&c->arena);
- c->tmp_allocator = gb_arena_allocator(&c->tmp_arena);
- c->global_scope = create_scope(universal_scope, c->allocator);
- map_init(&c->file_scopes, heap_allocator());
- ptr_set_init(&c->checked_files, heap_allocator());
- array_init(&c->file_order, heap_allocator(), 0, c->parser->files.count);
- // Init context
- c->context.scope = c->global_scope;
- c->context.type_path = new_checker_type_path();
- c->context.type_level = 0;
- }
- void destroy_checker(Checker *c) {
- destroy_checker_info(&c->info);
- gb_mutex_destroy(&c->mutex);
- destroy_scope(c->global_scope);
- array_free(&c->proc_stack);
- array_free(&c->procs);
- gb_arena_free(&c->tmp_arena);
- map_destroy(&c->file_scopes);
- ptr_set_destroy(&c->checked_files);
- array_free(&c->file_order);
- destroy_checker_type_path(c->context.type_path);
- }
- Entity *entity_of_ident(CheckerInfo *i, AstNode *identifier) {
- if (identifier->kind == AstNode_Ident) {
- return identifier->Ident.entity;
- }
- return nullptr;
- }
- TypeAndValue type_and_value_of_expr(CheckerInfo *i, AstNode *expr) {
- TypeAndValue result = {};
- TypeAndValue *found = map_get(&i->types, hash_node(expr));
- if (found) result = *found;
- return result;
- }
- Type *type_of_expr(CheckerInfo *i, AstNode *expr) {
- TypeAndValue tav = type_and_value_of_expr(i, expr);
- if (tav.mode != Addressing_Invalid) {
- return tav.type;
- }
- if (expr->kind == AstNode_Ident) {
- Entity *entity = entity_of_ident(i, expr);
- if (entity) {
- return entity->type;
- }
- }
- return nullptr;
- }
- Entity *implicit_entity_of_node(CheckerInfo *i, AstNode *clause) {
- // Entity **found = map_get(&i->implicits, hash_node(clause));
- // if (found != nullptr) {
- // return *found;
- // }
- if (clause->kind == AstNode_CaseClause) {
- return clause->CaseClause.implicit_entity;
- }
- return nullptr;
- }
- bool is_entity_implicitly_imported(Entity *import_name, Entity *e) {
- GB_ASSERT(import_name->kind == Entity_ImportName);
- return ptr_set_exists(&import_name->ImportName.scope->implicit, e);
- }
- // Will return nullptr if not found
- Entity *entity_of_node(CheckerInfo *i, AstNode *expr) {
- expr = unparen_expr(expr);
- switch (expr->kind) {
- case_ast_node(ident, Ident, expr);
- return entity_of_ident(i, expr);
- case_end;
- case_ast_node(se, SelectorExpr, expr);
- AstNode *s = unselector_expr(se->selector);
- if (s->kind == AstNode_Ident) {
- return entity_of_ident(i, s);
- }
- case_end;
- case_ast_node(cc, CaseClause, expr);
- return cc->implicit_entity;
- case_end;
- }
- return nullptr;
- }
- DeclInfo *decl_info_of_entity(CheckerInfo *i, Entity *e) {
- if (e != nullptr) {
- return e->decl_info;
- }
- return nullptr;
- }
- DeclInfo *decl_info_of_ident(CheckerInfo *i, AstNode *ident) {
- return decl_info_of_entity(i, entity_of_ident(i, ident));
- }
- AstFile *ast_file_of_filename(CheckerInfo *i, String filename) {
- AstFile **found = map_get(&i->files, hash_string(filename));
- if (found != nullptr) {
- return *found;
- }
- return nullptr;
- }
- Scope *scope_of_node(CheckerInfo *i, AstNode *node) {
- return node->scope;
- }
- ExprInfo *check_get_expr_info(CheckerInfo *i, AstNode *expr) {
- return map_get(&i->untyped, hash_node(expr));
- }
- void check_set_expr_info(CheckerInfo *i, AstNode *expr, ExprInfo info) {
- map_set(&i->untyped, hash_node(expr), info);
- }
- void check_remove_expr_info(CheckerInfo *i, AstNode *expr) {
- map_remove(&i->untyped, hash_node(expr));
- }
- isize type_info_index(CheckerInfo *info, Type *type, bool error_on_failure) {
- type = default_type(type);
- if (type == t_llvm_bool) {
- type = t_bool;
- }
- isize entry_index = -1;
- HashKey key = hash_type(type);
- isize *found_entry_index = map_get(&info->type_info_map, key);
- if (found_entry_index) {
- entry_index = *found_entry_index;
- }
- if (entry_index < 0) {
- // NOTE(bill): Do manual search
- // TODO(bill): This is O(n) and can be very slow
- for_array(i, info->type_info_map.entries){
- auto *e = &info->type_info_map.entries[i];
- Type *prev_type = cast(Type *)e->key.ptr;
- if (are_types_identical(prev_type, type)) {
- entry_index = e->value;
- // NOTE(bill): Add it to the search map
- map_set(&info->type_info_map, key, entry_index);
- break;
- }
- }
- }
- if (error_on_failure && entry_index < 0) {
- compiler_error("TypeInfo for '%s' could not be found", type_to_string(type));
- }
- return entry_index;
- }
- void add_untyped(CheckerInfo *i, AstNode *expression, bool lhs, AddressingMode mode, Type *type, ExactValue value) {
- if (expression == nullptr) {
- return;
- }
- if (mode == Addressing_Invalid) {
- return;
- }
- if (mode == Addressing_Constant && type == t_invalid) {
- compiler_error("add_untyped - invalid type: %s", type_to_string(type));
- }
- map_set(&i->untyped, hash_node(expression), make_expr_info(mode, type, value, lhs));
- }
- void add_type_and_value(CheckerInfo *i, AstNode *expression, AddressingMode mode, Type *type, ExactValue value) {
- if (expression == nullptr) {
- return;
- }
- if (mode == Addressing_Invalid) {
- return;
- }
- if (mode == Addressing_Constant && type == t_invalid) {
- compiler_error("add_type_and_value - invalid type: %s", type_to_string(type));
- }
- TypeAndValue tv = {};
- tv.type = type;
- tv.value = value;
- tv.mode = mode;
- map_set(&i->types, hash_node(expression), tv);
- }
- void add_entity_definition(CheckerInfo *i, AstNode *identifier, Entity *entity) {
- GB_ASSERT(identifier != nullptr);
- GB_ASSERT(identifier->kind == AstNode_Ident);
- if (is_blank_ident(identifier)) {
- return;
- }
- if (identifier->Ident.entity != nullptr) {
- // NOTE(bill): Identifier has already been handled
- return;
- }
- identifier->Ident.entity = entity;
- entity->identifier = identifier;
- array_add(&i->definitions, entity);
- }
- bool add_entity(Checker *c, Scope *scope, AstNode *identifier, Entity *entity) {
- if (scope == nullptr) {
- return false;
- }
- String name = entity->token.string;
- if (!is_blank_ident(name)) {
- Entity *ie = scope_insert_entity(scope, entity);
- if (ie != nullptr) {
- TokenPos pos = ie->token.pos;
- Entity *up = ie->using_parent;
- if (up != nullptr) {
- if (pos == up->token.pos) {
- // NOTE(bill): Error should have been handled already
- return false;
- }
- error(entity->token,
- "Redeclaration of '%.*s' in this scope through 'using'\n"
- "\tat %.*s(%td:%td)",
- LIT(name),
- LIT(up->token.pos.file), up->token.pos.line, up->token.pos.column);
- return false;
- } else {
- if (pos == entity->token.pos) {
- // NOTE(bill): Error should have been handled already
- return false;
- }
- error(entity->token,
- "Redeclaration of '%.*s' in this scope\n"
- "\tat %.*s(%td:%td)",
- LIT(name),
- LIT(pos.file), pos.line, pos.column);
- return false;
- }
- }
- }
- if (identifier != nullptr) {
- add_entity_definition(&c->info, identifier, entity);
- }
- return true;
- }
- void add_entity_use(Checker *c, AstNode *identifier, Entity *entity) {
- GB_ASSERT(identifier != nullptr);
- if (identifier->kind != AstNode_Ident) {
- return;
- }
- if (entity == nullptr) {
- return;
- }
- if (entity->identifier == nullptr) {
- entity->identifier = identifier;
- }
- identifier->Ident.entity = entity;
- add_declaration_dependency(c, entity); // TODO(bill): Should this be here?
- String dmsg = entity->deprecated_message;
- if (dmsg.len > 0) {
- warning(identifier, "%.*s is deprecated: %.*s", LIT(entity->token.string), LIT(dmsg));
- }
- }
- void add_entity_and_decl_info(Checker *c, AstNode *identifier, Entity *e, DeclInfo *d) {
- GB_ASSERT(identifier->kind == AstNode_Ident);
- GB_ASSERT(e != nullptr && d != nullptr);
- GB_ASSERT(identifier->Ident.token.string == e->token.string);
- if (e->scope != nullptr) {
- add_entity(c, e->scope, identifier, e);
- }
- add_entity_definition(&c->info, identifier, e);
- GB_ASSERT(e->decl_info == nullptr);
- e->decl_info = d;
- array_add(&c->info.entities, e);
- e->order_in_src = c->info.entities.count;
- }
- void add_implicit_entity(Checker *c, AstNode *clause, Entity *e) {
- GB_ASSERT(clause != nullptr);
- GB_ASSERT(e != nullptr);
- GB_ASSERT(clause->kind == AstNode_CaseClause);
- clause->CaseClause.implicit_entity = e;
- }
- void add_type_info_type(Checker *c, Type *t) {
- if (t == nullptr) {
- return;
- }
- t = default_type(t);
- if (is_type_bit_field_value(t)) {
- t = default_bit_field_value_type(t);
- }
- if (is_type_untyped(t)) {
- return; // Could be nil
- }
- if (is_type_polymorphic(base_type(t))) {
- return;
- }
- if (map_get(&c->info.type_info_map, hash_type(t)) != nullptr) {
- // Types have already been added
- return;
- }
- bool prev = false;
- isize ti_index = -1;
- for_array(i, c->info.type_info_map.entries) {
- auto *e = &c->info.type_info_map.entries[i];
- Type *prev_type = cast(Type *)e->key.ptr;
- if (are_types_identical(t, prev_type)) {
- // Duplicate entry
- ti_index = e->value;
- prev = true;
- break;
- }
- }
- if (ti_index < 0) {
- // Unique entry
- // NOTE(bill): map entries grow linearly and in order
- ti_index = c->info.type_info_types.count;
- array_add(&c->info.type_info_types, t);
- }
- map_set(&c->info.type_info_map, hash_type(t), ti_index);
- if (prev) {
- // NOTE(bill): If a previous one exists already, no need to continue
- return;
- }
- // Add nested types
- if (t->kind == Type_Named) {
- // NOTE(bill): Just in case
- add_type_info_type(c, t->Named.base);
- return;
- }
- Type *bt = base_type(t);
- add_type_info_type(c, bt);
- switch (bt->kind) {
- case Type_Basic:
- switch (bt->Basic.kind) {
- case Basic_string:
- add_type_info_type(c, t_u8_ptr);
- add_type_info_type(c, t_int);
- break;
- case Basic_any:
- add_type_info_type(c, t_type_info_ptr);
- add_type_info_type(c, t_rawptr);
- break;
- case Basic_complex64:
- add_type_info_type(c, t_type_info_float);
- add_type_info_type(c, t_f32);
- break;
- case Basic_complex128:
- add_type_info_type(c, t_type_info_float);
- add_type_info_type(c, t_f64);
- break;
- }
- break;
- case Type_Pointer:
- add_type_info_type(c, bt->Pointer.elem);
- break;
- case Type_Array:
- add_type_info_type(c, bt->Array.elem);
- add_type_info_type(c, alloc_type_pointer(bt->Array.elem));
- add_type_info_type(c, t_int);
- break;
- case Type_DynamicArray:
- add_type_info_type(c, bt->DynamicArray.elem);
- add_type_info_type(c, alloc_type_pointer(bt->DynamicArray.elem));
- add_type_info_type(c, t_int);
- add_type_info_type(c, t_allocator);
- break;
- case Type_Slice:
- add_type_info_type(c, bt->Slice.elem);
- add_type_info_type(c, alloc_type_pointer(bt->Slice.elem));
- add_type_info_type(c, t_int);
- break;
- case Type_Enum:
- add_type_info_type(c, bt->Enum.base_type);
- break;
- case Type_Union:
- add_type_info_type(c, t_int);
- add_type_info_type(c, t_type_info_ptr);
- for_array(i, bt->Union.variants) {
- add_type_info_type(c, bt->Union.variants[i]);
- }
- break;
- case Type_Struct:
- if (bt->Struct.scope != nullptr) {
- for_array(i, bt->Struct.scope->elements.entries) {
- Entity *e = bt->Struct.scope->elements.entries[i].value;
- add_type_info_type(c, e->type);
- }
- }
- for_array(i, bt->Struct.fields) {
- Entity *f = bt->Struct.fields[i];
- add_type_info_type(c, f->type);
- }
- break;
- case Type_Map:
- init_map_internal_types(bt);
- add_type_info_type(c, bt->Map.key);
- add_type_info_type(c, bt->Map.value);
- add_type_info_type(c, bt->Map.generated_struct_type);
- break;
- case Type_Tuple:
- for_array(i, bt->Tuple.variables) {
- Entity *var = bt->Tuple.variables[i];
- add_type_info_type(c, var->type);
- }
- break;
- case Type_Proc:
- add_type_info_type(c, bt->Proc.params);
- add_type_info_type(c, bt->Proc.results);
- break;
- }
- }
- void check_procedure_later(Checker *c, ProcedureInfo info) {
- GB_ASSERT(info.decl != nullptr);
- array_add(&c->procs, info);
- }
- void check_procedure_later(Checker *c, AstFile *file, Token token, DeclInfo *decl, Type *type, AstNode *body, u64 tags) {
- ProcedureInfo info = {};
- info.file = file;
- info.token = token;
- info.decl = decl;
- info.type = type;
- info.body = body;
- info.tags = tags;
- check_procedure_later(c, info);
- }
- void push_procedure(Checker *c, Type *type) {
- array_add(&c->proc_stack, type);
- }
- void pop_procedure(Checker *c) {
- array_pop(&c->proc_stack);
- }
- Type *const curr_procedure_type(Checker *c) {
- isize count = c->proc_stack.count;
- if (count > 0) {
- return c->proc_stack[count-1];
- }
- return nullptr;
- }
- void add_curr_ast_file(Checker *c, AstFile *file) {
- if (file != nullptr) {
- TokenPos zero_pos = {};
- global_error_collector.prev = zero_pos;
- c->curr_ast_file = file;
- c->context.decl = file->decl_info;
- c->context.scope = file->scope;
- c->context.file_scope = file->scope;
- }
- }
- void add_dependency_to_map(PtrSet<Entity *> *map, CheckerInfo *info, Entity *entity) {
- if (entity == nullptr) {
- return;
- }
- String name = entity->token.string;
- if (entity->type != nullptr &&
- is_type_polymorphic(entity->type)) {
- DeclInfo *decl = decl_info_of_entity(info, entity);
- if (decl != nullptr && decl->gen_proc_type == nullptr) {
- return;
- }
- }
- if (ptr_set_exists(map, entity)) {
- return;
- }
- ptr_set_add(map, entity);
- DeclInfo *decl = decl_info_of_entity(info, entity);
- if (decl == nullptr) {
- return;
- }
- for_array(i, decl->deps.entries) {
- Entity *e = decl->deps.entries[i].ptr;
- add_dependency_to_map(map, info, e);
- if (e->kind == Entity_Procedure && e->Procedure.is_foreign) {
- Entity *fl = e->Procedure.foreign_library;
- if (fl != nullptr) {
- GB_ASSERT_MSG(fl->kind == Entity_LibraryName &&
- fl->LibraryName.used,
- "%.*s", LIT(name));
- add_dependency_to_map(map, info, fl);
- }
- }
- if (e->kind == Entity_Variable && e->Variable.is_foreign) {
- Entity *fl = e->Variable.foreign_library;
- if (fl != nullptr) {
- GB_ASSERT_MSG(fl->kind == Entity_LibraryName &&
- fl->LibraryName.used,
- "%.*s", LIT(name));
- add_dependency_to_map(map, info, fl);
- }
- }
- }
- }
- PtrSet<Entity *> generate_minimum_dependency_set(Checker *c, Entity *start) {
- CheckerInfo *info = &c->info;
- PtrSet<Entity *> map = {}; // Key: Entity *
- ptr_set_init(&map, heap_allocator());
- String required_entities[] = {
- str_lit("__mem_zero"),
- str_lit("__init_context"),
- str_lit("default_allocator"),
- str_lit("make_source_code_location"),
- str_lit("__bounds_check_error"),
- str_lit("__slice_expr_error"),
- str_lit("__dynamic_array_expr_error"),
- str_lit("__args__"),
- str_lit("__type_table"),
- };
- for (isize i = 0; i < gb_count_of(required_entities); i++) {
- add_dependency_to_map(&map, info, scope_lookup_entity(c->global_scope, required_entities[i]));
- }
- for_array(i, info->definitions) {
- Entity *e = info->definitions[i];
- // if (e->scope->is_global && !is_type_poly_proc(e->type)) { // TODO(bill): is the check enough?
- if (e->scope->is_global) { // TODO(bill): is the check enough?
- if (e->type == nullptr || !is_type_poly_proc(e->type)) {
- if (e->kind == Entity_TypeName) {
- add_dependency_to_map(&map, info, e);
- } else {
- // add_dependency_to_map(&map, info, e);
- }
- }
- } else if (e->kind == Entity_Procedure && e->Procedure.is_export) {
- add_dependency_to_map(&map, info, e);
- } else if (e->kind == Entity_Variable && e->Procedure.is_export) {
- add_dependency_to_map(&map, info, e);
- }
- }
- add_dependency_to_map(&map, info, start);
- return map;
- }
- bool is_entity_a_dependency(Entity *e) {
- if (e == nullptr) return false;
- switch (e->kind) {
- case Entity_Procedure:
- case Entity_Variable:
- case Entity_Constant:
- return true;
- }
- return false;
- }
- Array<EntityGraphNode *> generate_entity_dependency_graph(CheckerInfo *info) {
- gbAllocator a = heap_allocator();
- Map<EntityGraphNode *> M = {}; // Key: Entity *
- map_init(&M, a, info->entities.count);
- defer (map_destroy(&M));
- for_array(i, info->entities) {
- Entity *e = info->entities[i];
- DeclInfo *d = e->decl_info;
- if (is_entity_a_dependency(e)) {
- EntityGraphNode *n = gb_alloc_item(a, EntityGraphNode);
- n->entity = e;
- map_set(&M, hash_pointer(e), n);
- }
- }
- // Calculate edges for graph M
- for_array(i, M.entries) {
- Entity * e = cast(Entity *)M.entries[i].key.ptr;
- EntityGraphNode *n = M.entries[i].value;
- DeclInfo *decl = decl_info_of_entity(info, e);
- if (decl != nullptr) {
- for_array(j, decl->deps.entries) {
- auto entry = decl->deps.entries[j];
- Entity *dep = entry.ptr;
- if (dep && is_entity_a_dependency(dep)) {
- EntityGraphNode **m_ = map_get(&M, hash_pointer(dep));
- if (m_ != nullptr) {
- EntityGraphNode *m = *m_;
- entity_graph_node_set_add(&n->succ, m);
- entity_graph_node_set_add(&m->pred, n);
- }
- }
- }
- }
- }
- auto G = array_make<EntityGraphNode *>(a, 0, M.entries.count);
- for_array(i, M.entries) {
- auto *entry = &M.entries[i];
- auto *e = cast(Entity *)entry->key.ptr;
- EntityGraphNode *n = entry->value;
- if (e->kind == Entity_Procedure) {
- // Connect each pred 'p' of 'n' with each succ 's' and from
- // the procedure node
- for_array(j, n->pred.entries) {
- EntityGraphNode *p = n->pred.entries[j].ptr;
- // Ignore self-cycles
- if (p != n) {
- // Each succ 's' of 'n' becomes a succ of 'p', and
- // each pred 'p' of 'n' becomes a pred of 's'
- for_array(k, n->succ.entries) {
- EntityGraphNode *s = n->succ.entries[k].ptr;
- // Ignore self-cycles
- if (s != n) {
- entity_graph_node_set_add(&p->succ, s);
- entity_graph_node_set_add(&s->pred, p);
- // Remove edge to 'n'
- entity_graph_node_set_remove(&s->pred, n);
- }
- }
- // Remove edge to 'n'
- entity_graph_node_set_remove(&p->succ, n);
- }
- }
- } else {
- array_add(&G, n);
- }
- }
- for_array(i, G) {
- EntityGraphNode *n = G[i];
- n->index = i;
- n->dep_count = n->succ.entries.count;
- GB_ASSERT(n->dep_count >= 0);
- }
- return G;
- }
- Entity *find_core_entity(Checker *c, String name) {
- Entity *e = current_scope_lookup_entity(c->global_scope, name);
- if (e == nullptr) {
- compiler_error("Could not find type declaration for '%.*s'\n"
- "Is '_preload.odin' missing from the 'core' directory relative to odin.exe?", LIT(name));
- // NOTE(bill): This will exit the program as it's cannot continue without it!
- }
- return e;
- }
- Type *find_core_type(Checker *c, String name) {
- Entity *e = current_scope_lookup_entity(c->global_scope, name);
- if (e == nullptr) {
- compiler_error("Could not find type declaration for '%.*s'\n"
- "Is '_preload.odin' missing from the 'core' directory relative to odin.exe?", LIT(name));
- // NOTE(bill): This will exit the program as it's cannot continue without it!
- }
- return e->type;
- }
- CheckerTypePath *new_checker_type_path() {
- gbAllocator a = heap_allocator();
- auto *tp = gb_alloc_item(a, CheckerTypePath);
- array_init(tp, a, 0, 16);
- return tp;
- }
- void destroy_checker_type_path(CheckerTypePath *tp) {
- array_free(tp);
- gb_free(heap_allocator(), tp);
- }
- void check_type_path_push(Checker *c, Entity *e) {
- GB_ASSERT(c->context.type_path != nullptr);
- GB_ASSERT(e != nullptr);
- array_add(c->context.type_path, e);
- }
- Entity *check_type_path_pop(Checker *c) {
- GB_ASSERT(c->context.type_path != nullptr);
- return array_pop(c->context.type_path);
- }
- void check_entity_decl(Checker *c, Entity *e, DeclInfo *d, Type *named_type);
- Array<Entity *> proc_group_entities(Checker *c, Operand o) {
- Array<Entity *> procs = {};
- if (o.mode == Addressing_ProcGroup) {
- GB_ASSERT(o.proc_group != nullptr);
- if (o.proc_group->kind == Entity_ProcGroup) {
- check_entity_decl(c, o.proc_group, nullptr, nullptr);
- return o.proc_group->ProcGroup.entities;
- }
- }
- return procs;
- }
- void init_preload(Checker *c) {
- if (t_type_info == nullptr) {
- Entity *type_info_entity = find_core_entity(c, str_lit("Type_Info"));
- t_type_info = type_info_entity->type;
- t_type_info_ptr = alloc_type_pointer(t_type_info);
- GB_ASSERT(is_type_struct(type_info_entity->type));
- TypeStruct *tis = &base_type(type_info_entity->type)->Struct;
- Entity *type_info_enum_value = find_core_entity(c, str_lit("Type_Info_Enum_Value"));
- t_type_info_enum_value = type_info_enum_value->type;
- t_type_info_enum_value_ptr = alloc_type_pointer(t_type_info_enum_value);
- GB_ASSERT(tis->fields.count == 3);
- Entity *type_info_variant = tis->fields[2];
- Type *tiv_type = type_info_variant->type;
- GB_ASSERT(is_type_union(tiv_type));
- t_type_info_named = find_core_type(c, str_lit("Type_Info_Named"));
- t_type_info_integer = find_core_type(c, str_lit("Type_Info_Integer"));
- t_type_info_rune = find_core_type(c, str_lit("Type_Info_Rune"));
- t_type_info_float = find_core_type(c, str_lit("Type_Info_Float"));
- t_type_info_complex = find_core_type(c, str_lit("Type_Info_Complex"));
- t_type_info_string = find_core_type(c, str_lit("Type_Info_String"));
- t_type_info_boolean = find_core_type(c, str_lit("Type_Info_Boolean"));
- t_type_info_any = find_core_type(c, str_lit("Type_Info_Any"));
- t_type_info_pointer = find_core_type(c, str_lit("Type_Info_Pointer"));
- t_type_info_procedure = find_core_type(c, str_lit("Type_Info_Procedure"));
- t_type_info_array = find_core_type(c, str_lit("Type_Info_Array"));
- t_type_info_dynamic_array = find_core_type(c, str_lit("Type_Info_Dynamic_Array"));
- t_type_info_slice = find_core_type(c, str_lit("Type_Info_Slice"));
- t_type_info_tuple = find_core_type(c, str_lit("Type_Info_Tuple"));
- t_type_info_struct = find_core_type(c, str_lit("Type_Info_Struct"));
- t_type_info_union = find_core_type(c, str_lit("Type_Info_Union"));
- t_type_info_enum = find_core_type(c, str_lit("Type_Info_Enum"));
- t_type_info_map = find_core_type(c, str_lit("Type_Info_Map"));
- t_type_info_bit_field = find_core_type(c, str_lit("Type_Info_Bit_Field"));
- t_type_info_named_ptr = alloc_type_pointer(t_type_info_named);
- t_type_info_integer_ptr = alloc_type_pointer(t_type_info_integer);
- t_type_info_rune_ptr = alloc_type_pointer(t_type_info_rune);
- t_type_info_float_ptr = alloc_type_pointer(t_type_info_float);
- t_type_info_complex_ptr = alloc_type_pointer(t_type_info_complex);
- t_type_info_string_ptr = alloc_type_pointer(t_type_info_string);
- t_type_info_boolean_ptr = alloc_type_pointer(t_type_info_boolean);
- t_type_info_any_ptr = alloc_type_pointer(t_type_info_any);
- t_type_info_pointer_ptr = alloc_type_pointer(t_type_info_pointer);
- t_type_info_procedure_ptr = alloc_type_pointer(t_type_info_procedure);
- t_type_info_array_ptr = alloc_type_pointer(t_type_info_array);
- t_type_info_dynamic_array_ptr = alloc_type_pointer(t_type_info_dynamic_array);
- t_type_info_slice_ptr = alloc_type_pointer(t_type_info_slice);
- t_type_info_tuple_ptr = alloc_type_pointer(t_type_info_tuple);
- t_type_info_struct_ptr = alloc_type_pointer(t_type_info_struct);
- t_type_info_union_ptr = alloc_type_pointer(t_type_info_union);
- t_type_info_enum_ptr = alloc_type_pointer(t_type_info_enum);
- t_type_info_map_ptr = alloc_type_pointer(t_type_info_map);
- t_type_info_bit_field_ptr = alloc_type_pointer(t_type_info_bit_field);
- }
- if (t_allocator == nullptr) {
- Entity *e = find_core_entity(c, str_lit("Allocator"));
- t_allocator = e->type;
- t_allocator_ptr = alloc_type_pointer(t_allocator);
- }
- if (t_context == nullptr) {
- Entity *e = find_core_entity(c, str_lit("Context"));
- e_context = e;
- t_context = e->type;
- t_context_ptr = alloc_type_pointer(t_context);
- }
- if (t_source_code_location == nullptr) {
- Entity *e = find_core_entity(c, str_lit("Source_Code_Location"));
- t_source_code_location = e->type;
- t_source_code_location_ptr = alloc_type_pointer(t_allocator);
- }
- if (t_map_key == nullptr) {
- Entity *e = find_core_entity(c, str_lit("__Map_Key"));
- t_map_key = e->type;
- }
- if (t_map_header == nullptr) {
- Entity *e = find_core_entity(c, str_lit("__Map_Header"));
- t_map_header = e->type;
- }
- {
- String _global = str_lit("_global");
- Entity *type_info_entity = find_core_entity(c, str_lit("Type_Info"));
- Scope *preload_scope = type_info_entity->scope;
- Entity *e = alloc_entity_import_name(preload_scope, make_token_ident(_global), t_invalid,
- str_lit(""), _global,
- preload_scope);
- add_entity(c, universal_scope, nullptr, e);
- }
- c->done_preload = true;
- }
- DECL_ATTRIBUTE_PROC(foreign_block_decl_attribute) {
- if (name == "default_calling_convention") {
- if (value.kind == ExactValue_String) {
- auto cc = string_to_calling_convention(value.value_string);
- if (cc == ProcCC_Invalid) {
- error(elem, "Unknown procedure calling convention: '%.*s'\n", LIT(value.value_string));
- } else {
- c->context.foreign_context.default_cc = cc;
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- } else if (name == "link_prefix") {
- if (value.kind == ExactValue_String) {
- String link_prefix = value.value_string;
- if (!is_foreign_name_valid(link_prefix)) {
- error(elem, "Invalid link prefix: '%.*s'\n", LIT(link_prefix));
- } else {
- c->context.foreign_context.link_prefix = link_prefix;
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- }
- return false;
- }
- DECL_ATTRIBUTE_PROC(proc_decl_attribute) {
- if (name == "link_name") {
- if (value.kind == ExactValue_String) {
- ac->link_name = value.value_string;
- if (!is_foreign_name_valid(ac->link_name)) {
- error(elem, "Invalid link name: %.*s", LIT(ac->link_name));
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- } else if (name == "link_prefix") {
- if (value.kind == ExactValue_String) {
- ac->link_prefix = value.value_string;
- if (!is_foreign_name_valid(ac->link_prefix)) {
- error(elem, "Invalid link prefix: %.*s", LIT(ac->link_prefix));
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- } else if (name == "deprecated") {
- if (value.kind == ExactValue_String) {
- String msg = value.value_string;
- if (msg.len == 0) {
- error(elem, "Deprecation message cannot be an empty string");
- } else {
- ac->deprecated_message = msg;
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- }
- return false;
- }
- DECL_ATTRIBUTE_PROC(var_decl_attribute) {
- if (c->context.curr_proc_decl != nullptr) {
- error(elem, "Only a variable at file scope can have a '%.*s'", LIT(name));
- return true;
- }
- if (name == "link_name") {
- if (value.kind == ExactValue_String) {
- ac->link_name = value.value_string;
- if (!is_foreign_name_valid(ac->link_name)) {
- error(elem, "Invalid link name: %.*s", LIT(ac->link_name));
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- } else if (name == "link_prefix") {
- if (value.kind == ExactValue_String) {
- ac->link_prefix = value.value_string;
- if (!is_foreign_name_valid(ac->link_prefix)) {
- error(elem, "Invalid link prefix: %.*s", LIT(ac->link_prefix));
- }
- } else {
- error(elem, "Expected a string value for '%.*s'", LIT(name));
- }
- return true;
- } else if (name == "thread_local") {
- if (ac->init_expr_list_count > 0) {
- error(elem, "A thread local variable declaration cannot have initialization values");
- } else if (c->context.foreign_context.curr_library || c->context.foreign_context.in_export) {
- error(elem, "A foreign block variable cannot be thread local");
- } else if (value.kind == ExactValue_Invalid) {
- ac->thread_local_model = str_lit("default");
- } else if (value.kind == ExactValue_String) {
- String model = value.value_string;
- if (model == "localdynamic" ||
- model == "initialexec" ||
- model == "localexec") {
- ac->thread_local_model = model;
- } else {
- error(elem, "Invalid thread local model '%.*s'", LIT(model));
- }
- } else {
- error(elem, "Expected either no value or a string for '%.*s'", LIT(name));
- }
- return true;
- }
- return false;
- }
- #include "check_expr.cpp"
- #include "check_type.cpp"
- #include "check_decl.cpp"
- #include "check_stmt.cpp"
- void check_decl_attributes(Checker *c, Array<AstNode *> attributes, DeclAttributeProc *proc, AttributeContext *ac) {
- if (attributes.count == 0) return;
- String original_link_prefix = {};
- if (ac) {
- original_link_prefix = ac->link_prefix;
- }
- StringSet set = {};
- string_set_init(&set, heap_allocator());
- defer (string_set_destroy(&set));
- for_array(i, attributes) {
- AstNode *attr = attributes[i];
- if (attr->kind != AstNode_Attribute) continue;
- for_array(j, attr->Attribute.elems) {
- AstNode *elem = attr->Attribute.elems[j];
- String name = {};
- AstNode *value = nullptr;
- switch (elem->kind) {
- case_ast_node(i, Ident, elem);
- name = i->token.string;
- case_end;
- case_ast_node(fv, FieldValue, elem);
- GB_ASSERT(fv->field->kind == AstNode_Ident);
- name = fv->field->Ident.token.string;
- value = fv->value;
- case_end;
- default:
- error(elem, "Invalid attribute element");
- continue;
- }
- ExactValue ev = {};
- if (value != nullptr) {
- Operand op = {};
- check_expr(c, &op, value);
- if (op.mode) {
- if (op.mode != Addressing_Constant) {
- error(value, "An attribute element must be constant");
- } else {
- ev = op.value;
- }
- }
- }
- if (string_set_exists(&set, name)) {
- error(elem, "Previous declaration of '%.*s'", LIT(name));
- continue;
- } else {
- string_set_add(&set, name);
- }
- if (!proc(c, elem, name, ev, ac)) {
- error(elem, "Unknown attribute element name '%.*s'", LIT(name));
- }
- }
- }
- if (ac) {
- if (ac->link_prefix.text == original_link_prefix.text) {
- if (ac->link_name.len > 0) {
- ac->link_prefix.text = nullptr;
- ac->link_prefix.len = 0;
- }
- }
- }
- }
- bool check_arity_match(Checker *c, AstNodeValueDecl *vd, bool is_global) {
- isize lhs = vd->names.count;
- isize rhs = vd->values.count;
- if (rhs == 0) {
- if (vd->type == nullptr) {
- error(vd->names[0], "Missing type or initial expression");
- return false;
- }
- } else if (lhs < rhs) {
- if (lhs < vd->values.count) {
- AstNode *n = vd->values[lhs];
- gbString str = expr_to_string(n);
- error(n, "Extra initial expression '%s'", str);
- gb_string_free(str);
- } else {
- error(vd->names[0], "Extra initial expression");
- }
- return false;
- } else if (lhs > rhs) {
- if (!is_global && rhs != 1) {
- AstNode *n = vd->names[rhs];
- gbString str = expr_to_string(n);
- error(n, "Missing expression for '%s'", str);
- gb_string_free(str);
- return false;
- } else if (is_global) {
- AstNode *n = vd->values[rhs-1];
- error(n, "Expected %td expressions on the right hand side, got %td", lhs, rhs);
- return false;
- }
- }
- return true;
- }
- void check_collect_entities_from_when_stmt(Checker *c, AstNodeWhenStmt *ws) {
- Operand operand = {Addressing_Invalid};
- if (!ws->is_cond_determined) {
- check_expr(c, &operand, ws->cond);
- if (operand.mode != Addressing_Invalid && !is_type_boolean(operand.type)) {
- error(ws->cond, "Non-boolean condition in 'when' statement");
- }
- if (operand.mode != Addressing_Constant) {
- error(ws->cond, "Non-constant condition in 'when' statement");
- }
- ws->is_cond_determined = true;
- ws->determined_cond = operand.value.kind == ExactValue_Bool && operand.value.value_bool;
- }
- if (ws->body == nullptr || ws->body->kind != AstNode_BlockStmt) {
- error(ws->cond, "Invalid body for 'when' statement");
- } else {
- if (ws->determined_cond) {
- check_collect_entities(c, ws->body->BlockStmt.stmts);
- } else if (ws->else_stmt) {
- switch (ws->else_stmt->kind) {
- case AstNode_BlockStmt:
- check_collect_entities(c, ws->else_stmt->BlockStmt.stmts);
- break;
- case AstNode_WhenStmt:
- check_collect_entities_from_when_stmt(c, &ws->else_stmt->WhenStmt);
- break;
- default:
- error(ws->else_stmt, "Invalid 'else' statement in 'when' statement");
- break;
- }
- }
- }
- }
- void check_collect_value_decl(Checker *c, AstNode *decl) {
- ast_node(vd, ValueDecl, decl);
- if (vd->been_handled) return;
- vd->been_handled = true;
- if (vd->is_mutable) {
- if (!c->context.scope->is_file) {
- // NOTE(bill): local scope -> handle later and in order
- return;
- }
- // NOTE(bill): You need to store the entity information here unline a constant declaration
- isize entity_cap = vd->names.count;
- isize entity_count = 0;
- Entity **entities = gb_alloc_array(c->allocator, Entity *, entity_cap);
- DeclInfo *di = nullptr;
- if (vd->values.count > 0) {
- di = make_declaration_info(heap_allocator(), c->context.scope, c->context.decl);
- di->entities = entities;
- di->type_expr = vd->type;
- di->init_expr = vd->values[0];
- di->init_expr_list = vd->values;
- }
- for_array(i, vd->names) {
- AstNode *name = vd->names[i];
- AstNode *value = nullptr;
- if (i < vd->values.count) {
- value = vd->values[i];
- }
- if (name->kind != AstNode_Ident) {
- error(name, "A declaration's name must be an identifier, got %.*s", LIT(ast_node_strings[name->kind]));
- continue;
- }
- Entity *e = alloc_entity_variable(c->context.scope, name->Ident.token, nullptr, false);
- e->identifier = name;
- if (vd->is_using) {
- vd->is_using = false; // NOTE(bill): This error will be only caught once
- error(name, "'using' is not allowed at the file scope");
- }
- AstNode *fl = c->context.foreign_context.curr_library;
- if (fl != nullptr) {
- GB_ASSERT(fl->kind == AstNode_Ident);
- e->Variable.is_foreign = true;
- e->Variable.foreign_library_ident = fl;
- e->Variable.link_prefix = c->context.foreign_context.link_prefix;
- } else if (c->context.foreign_context.in_export) {
- e->Variable.is_export = true;
- }
- entities[entity_count++] = e;
- DeclInfo *d = di;
- if (d == nullptr || i > 0) {
- AstNode *init_expr = value;
- d = make_declaration_info(heap_allocator(), e->scope, c->context.decl);
- d->type_expr = vd->type;
- d->init_expr = init_expr;
- }
- d->attributes = vd->attributes;
- add_entity_and_decl_info(c, name, e, d);
- }
- if (di != nullptr) {
- di->entity_count = entity_count;
- }
- check_arity_match(c, vd, true);
- } else {
- for_array(i, vd->names) {
- AstNode *name = vd->names[i];
- if (name->kind != AstNode_Ident) {
- error(name, "A declaration's name must be an identifier, got %.*s", LIT(ast_node_strings[name->kind]));
- continue;
- }
- AstNode *init = unparen_expr(vd->values[i]);
- if (init == nullptr) {
- error(name, "Expected a value for this constant value declaration");
- continue;
- }
- Token token = name->Ident.token;
- AstNode *fl = c->context.foreign_context.curr_library;
- DeclInfo *d = make_declaration_info(c->allocator, c->context.scope, c->context.decl);
- Entity *e = nullptr;
- d->attributes = vd->attributes;
- if (is_ast_node_type(init) ||
- (vd->type != nullptr && vd->type->kind == AstNode_TypeType)) {
- e = alloc_entity_type_name(d->scope, token, nullptr);
- if (vd->type != nullptr) {
- error(name, "A type declaration cannot have an type parameter");
- }
- d->type_expr = init;
- d->init_expr = init;
- } else if (init->kind == AstNode_ProcLit) {
- if (c->context.scope->is_struct) {
- error(name, "Procedure declarations are not allowed within a struct");
- continue;
- }
- ast_node(pl, ProcLit, init);
- e = alloc_entity_procedure(d->scope, token, nullptr, pl->tags);
- if (fl != nullptr) {
- GB_ASSERT(fl->kind == AstNode_Ident);
- e->Procedure.foreign_library_ident = fl;
- e->Procedure.is_foreign = true;
- GB_ASSERT(pl->type->kind == AstNode_ProcType);
- auto cc = pl->type->ProcType.calling_convention;
- if (cc == ProcCC_ForeignBlockDefault) {
- cc = ProcCC_CDecl;
- if (c->context.foreign_context.default_cc > 0) {
- cc = c->context.foreign_context.default_cc;
- }
- }
- e->Procedure.link_prefix = c->context.foreign_context.link_prefix;
- GB_ASSERT(cc != ProcCC_Invalid);
- pl->type->ProcType.calling_convention = cc;
- } else if (c->context.foreign_context.in_export) {
- e->Procedure.is_export = true;
- }
- d->proc_lit = init;
- d->type_expr = pl->type;
- } else if (init->kind == AstNode_ProcGroup) {
- ast_node(pg, ProcGroup, init);
- e = alloc_entity_proc_group(d->scope, token, nullptr);
- if (fl != nullptr) {
- error(name, "Procedure groups are not allowed within a foreign block");
- }
- d->init_expr = init;
- } else {
- e = alloc_entity_constant(d->scope, token, nullptr, empty_exact_value);
- d->type_expr = vd->type;
- d->init_expr = init;
- }
- e->identifier = name;
- if (e->kind != Entity_Procedure) {
- if (fl != nullptr || c->context.foreign_context.in_export) {
- AstNodeKind kind = init->kind;
- error(name, "Only procedures and variables are allowed to be in a foreign block, got %.*s", LIT(ast_node_strings[kind]));
- if (kind == AstNode_ProcType) {
- gb_printf_err("\tDid you forget to append '---' to the procedure?\n");
- }
- }
- }
- add_entity_and_decl_info(c, name, e, d);
- }
- check_arity_match(c, vd, true);
- }
- }
- void check_add_foreign_block_decl(Checker *c, AstNode *decl) {
- ast_node(fb, ForeignBlockDecl, decl);
- if (fb->been_handled) return;
- fb->been_handled = true;
- AstNode *foreign_library = fb->foreign_library;
- CheckerContext prev_context = c->context;
- if (foreign_library->kind == AstNode_Ident) {
- c->context.foreign_context.curr_library = foreign_library;
- } else if (foreign_library->kind == AstNode_Implicit && foreign_library->Implicit.kind == Token_export) {
- c->context.foreign_context.in_export = true;
- } else {
- error(foreign_library, "Foreign block name must be an identifier or 'export'");
- c->context.foreign_context.curr_library = nullptr;
- }
- check_decl_attributes(c, fb->attributes, foreign_block_decl_attribute, nullptr);
- c->context.collect_delayed_decls = true;
- check_collect_entities(c, fb->decls);
- c->context = prev_context;
- }
- // NOTE(bill): If file_scopes == nullptr, this will act like a local scope
- void check_collect_entities(Checker *c, Array<AstNode *> nodes) {
- for_array(decl_index, nodes) {
- AstNode *decl = nodes[decl_index];
- if (!is_ast_node_decl(decl) && !is_ast_node_when_stmt(decl)) {
- continue;
- }
- switch (decl->kind) {
- case_ast_node(bd, BadDecl, decl);
- case_end;
- case_ast_node(ws, WhenStmt, decl);
- // Will be handled later
- case_end;
- case_ast_node(vd, ValueDecl, decl);
- check_collect_value_decl(c, decl);
- case_end;
- case_ast_node(id, ImportDecl, decl);
- if (!c->context.scope->is_file) {
- error(decl, "import declarations are only allowed in the file scope");
- // NOTE(bill): _Should_ be caught by the parser
- // TODO(bill): Better error handling if it isn't
- continue;
- }
- if (c->context.collect_delayed_decls) {
- check_add_import_decl(c, id);
- }
- case_end;
- case_ast_node(ed, ExportDecl, decl);
- if (!c->context.scope->is_file) {
- error(decl, "export declarations are only allowed in the file scope");
- // NOTE(bill): _Should_ be caught by the parser
- // TODO(bill): Better error handling if it isn't
- continue;
- }
- if (c->context.collect_delayed_decls) {
- check_add_export_decl(c, ed);
- }
- case_end;
- case_ast_node(fl, ForeignImportDecl, decl);
- if (!c->context.scope->is_file) {
- error(decl, "%.*s declarations are only allowed in the file scope", LIT(fl->token.string));
- // NOTE(bill): _Should_ be caught by the parser
- // TODO(bill): Better error handling if it isn't
- continue;
- }
- check_add_foreign_import_decl(c, decl);
- case_end;
- case_ast_node(fb, ForeignBlockDecl, decl);
- check_add_foreign_block_decl(c, decl);
- case_end;
- default:
- if (c->context.scope->is_file) {
- error(decl, "Only declarations are allowed at file scope");
- }
- break;
- }
- }
- // NOTE(bill): 'when' stmts need to be handled after the other as the condition may refer to something
- // declared after this stmt in source
- if (!c->context.scope->is_file || c->context.collect_delayed_decls) {
- for_array(i, nodes) {
- AstNode *node = nodes[i];
- switch (node->kind) {
- case_ast_node(ws, WhenStmt, node);
- check_collect_entities_from_when_stmt(c, ws);
- case_end;
- }
- }
- }
- }
- void check_all_global_entities(Checker *c) {
- Scope *prev_file = nullptr;
- bool processing_preload = true;
- for_array(i, c->info.entities) {
- Entity *e = c->info.entities[i];
- DeclInfo *d = e->decl_info;
- if (d->scope != e->scope) {
- continue;
- }
- if (!d->scope->has_been_imported) {
- // NOTE(bill): All of these unchecked entities could mean a lot of unused allocations
- // TODO(bill): Should this be worried about?
- continue;
- }
- AstFile *file = d->scope->file;
- add_curr_ast_file(c, file);
- if (e->token.string == "main") {
- if (e->kind != Entity_Procedure) {
- if (e->scope->is_init) {
- error(e->token, "'main' is reserved as the entry point procedure in the initial scope");
- continue;
- }
- } else if (e->scope->is_global) {
- error(e->token, "'main' is reserved as the entry point procedure in the initial scope");
- continue;
- }
- }
- CheckerContext prev_context = c->context;
- c->context.decl = d;
- c->context.scope = d->scope;
- check_entity_decl(c, e, d, nullptr);
- c->context = prev_context;
- if (!d->scope->is_global) {
- processing_preload = false;
- }
- if (!processing_preload) {
- init_preload(c);
- }
- }
- }
- bool is_string_an_identifier(String s) {
- isize offset = 0;
- if (s.len < 1) {
- return false;
- }
- while (offset < s.len) {
- bool ok = false;
- Rune r = -1;
- isize size = gb_utf8_decode(s.text+offset, s.len-offset, &r);
- if (offset == 0) {
- ok = rune_is_letter(r);
- } else {
- ok = rune_is_letter(r) || rune_is_digit(r);
- }
- if (!ok) {
- return false;
- }
- offset += size;
- }
- return offset == s.len;
- }
- String path_to_entity_name(String name, String fullpath) {
- if (name.len != 0) {
- return name;
- }
- // NOTE(bill): use file name (without extension) as the identifier
- // If it is a valid identifier
- String filename = fullpath;
- isize slash = 0;
- isize dot = 0;
- for (isize i = filename.len-1; i >= 0; i--) {
- u8 c = filename[i];
- if (c == '/' || c == '\\') {
- break;
- }
- slash = i;
- }
- filename = substring(filename, slash, filename.len);
- dot = filename.len;
- while (dot --> 0) {
- u8 c = filename[dot];
- if (c == '.') {
- break;
- }
- }
- filename = substring(filename, 0, dot);
- if (is_string_an_identifier(filename)) {
- return filename;
- } else {
- return str_lit("_");
- }
- }
- void add_import_dependency_node(Checker *c, AstNode *decl, Map<ImportGraphNode *> *M) {
- Scope *parent_file_scope = decl->file->scope;
- switch (decl->kind) {
- case_ast_node(id, ImportDecl, decl);
- String path = id->fullpath;
- HashKey key = hash_string(path);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) {
- for_array(scope_index, c->file_scopes.entries) {
- Scope *scope = c->file_scopes.entries[scope_index].value;
- gb_printf_err("%.*s\n", LIT(scope->file->tokenizer.fullpath));
- }
- Token token = ast_node_token(decl);
- gb_printf_err("%.*s(%td:%td)\n", LIT(token.pos.file), token.pos.line, token.pos.column);
- GB_PANIC("Unable to find scope for file: %.*s", LIT(path));
- }
- Scope *scope = *found;
- GB_ASSERT(scope != nullptr);
- id->file = scope->file;
- ImportGraphNode **found_node = nullptr;
- ImportGraphNode *m = nullptr;
- ImportGraphNode *n = nullptr;
- found_node = map_get(M, hash_pointer(scope));
- GB_ASSERT(found_node != nullptr);
- m = *found_node;
- found_node = map_get(M, hash_pointer(parent_file_scope));
- GB_ASSERT(found_node != nullptr);
- n = *found_node;
- // TODO(bill): How should the edges be attched for 'import'?
- import_graph_node_set_add(&n->succ, m);
- import_graph_node_set_add(&m->pred, n);
- ptr_set_add(&m->scope->imported, n->scope);
- if (id->is_using) {
- ptr_set_add(&m->scope->exported, n->scope);
- }
- case_end;
- case_ast_node(ed, ExportDecl, decl);
- String path = ed->fullpath;
- HashKey key = hash_string(path);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) {
- for_array(scope_index, c->file_scopes.entries) {
- Scope *scope = c->file_scopes.entries[scope_index].value;
- gb_printf_err("%.*s\n", LIT(scope->file->tokenizer.fullpath));
- }
- Token token = ast_node_token(decl);
- gb_printf_err("%.*s(%td:%td)\n", LIT(token.pos.file), token.pos.line, token.pos.column);
- GB_PANIC("Unable to find scope for file: %.*s", LIT(path));
- }
- Scope *scope = *found;
- GB_ASSERT(scope != nullptr);
- ed->file = scope->file;
- ImportGraphNode **found_node = nullptr;
- ImportGraphNode *m = nullptr;
- ImportGraphNode *n = nullptr;
- found_node = map_get(M, hash_pointer(scope));
- GB_ASSERT(found_node != nullptr);
- m = *found_node;
- found_node = map_get(M, hash_pointer(parent_file_scope));
- GB_ASSERT(found_node != nullptr);
- n = *found_node;
- import_graph_node_set_add(&n->succ, m);
- import_graph_node_set_add(&m->pred, n);
- ptr_set_add(&m->scope->exported, n->scope);
- case_end;
- case_ast_node(ws, WhenStmt, decl);
- if (ws->body != nullptr) {
- auto stmts = ws->body->BlockStmt.stmts;
- for_array(i, stmts) {
- add_import_dependency_node(c, stmts[i], M);
- }
- }
- if (ws->else_stmt != nullptr) {
- switch (ws->else_stmt->kind) {
- case AstNode_BlockStmt: {
- auto stmts = ws->else_stmt->BlockStmt.stmts;
- for_array(i, stmts) {
- add_import_dependency_node(c, stmts[i], M);
- }
- break;
- }
- case AstNode_WhenStmt:
- add_import_dependency_node(c, ws->else_stmt, M);
- break;
- }
- }
- case_end;
- }
- }
- Array<ImportGraphNode *> generate_import_dependency_graph(Checker *c) {
- gbAllocator a = heap_allocator();
- Map<ImportGraphNode *> M = {}; // Key: Scope *
- map_init(&M, a);
- defer (map_destroy(&M));
- for_array(i, c->parser->files) {
- Scope *scope = c->parser->files[i]->scope;
- ImportGraphNode *n = import_graph_node_create(heap_allocator(), scope);
- map_set(&M, hash_pointer(scope), n);
- }
- // Calculate edges for graph M
- for_array(i, c->parser->files) {
- AstFile *f = c->parser->files[i];
- for_array(j, f->decls) {
- AstNode *decl = f->decls[j];
- add_import_dependency_node(c, decl, &M);
- }
- }
- Array<ImportGraphNode *> G = {};
- array_init(&G, a);
- for_array(i, M.entries) {
- array_add(&G, M.entries[i].value);
- }
- for_array(i, G) {
- ImportGraphNode *n = G[i];
- n->index = i;
- n->dep_count = n->succ.entries.count;
- GB_ASSERT(n->dep_count >= 0);
- }
- return G;
- }
- struct ImportPathItem {
- Scope * scope;
- AstNode *decl;
- };
- Array<ImportPathItem> find_import_path(Checker *c, Scope *start, Scope *end, PtrSet<Scope *> *visited) {
- Array<ImportPathItem> empty_path = {};
- if (ptr_set_exists(visited, start)) {
- return empty_path;
- }
- ptr_set_add(visited, start);
- String path = start->file->tokenizer.fullpath;
- HashKey key = hash_string(path);
- Scope **found = map_get(&c->file_scopes, key);
- if (found) {
- AstFile *f = (*found)->file;
- GB_ASSERT(f != nullptr);
- for_array(i, f->imports_and_exports) {
- Scope *s = nullptr;
- AstNode *decl = f->imports_and_exports[i];
- if (decl->kind == AstNode_ExportDecl) {
- s = decl->ExportDecl.file->scope;
- } else if (decl->kind == AstNode_ImportDecl) {
- if (!decl->ImportDecl.is_using) {
- // continue;
- }
- s = decl->ImportDecl.file->scope;
- } else {
- continue;
- }
- GB_ASSERT(s != nullptr);
- ImportPathItem item = {s, decl};
- if (s == end) {
- auto path = array_make<ImportPathItem>(heap_allocator());
- array_add(&path, item);
- return path;
- }
- auto next_path = find_import_path(c, s, end, visited);
- if (next_path.count > 0) {
- array_add(&next_path, item);
- return next_path;
- }
- }
- }
- return empty_path;
- }
- void check_add_import_decl(Checker *c, AstNodeImportDecl *id) {
- if (id->been_handled) return;
- id->been_handled = true;
- Scope *parent_scope = c->context.scope;
- GB_ASSERT(parent_scope->is_file);
- Token token = id->relpath;
- HashKey key = hash_string(id->fullpath);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) {
- for_array(scope_index, c->file_scopes.entries) {
- Scope *scope = c->file_scopes.entries[scope_index].value;
- gb_printf_err("%.*s\n", LIT(scope->file->tokenizer.fullpath));
- }
- gb_printf_err("%.*s(%td:%td)\n", LIT(token.pos.file), token.pos.line, token.pos.column);
- GB_PANIC("Unable to find scope for file: %.*s", LIT(id->fullpath));
- }
- Scope *scope = *found;
- if (scope->is_global) {
- error(token, "Importing a #shared_global_scope is disallowed and unnecessary");
- return;
- }
- if (ptr_set_exists(&parent_scope->imported, scope)) {
- // error(token, "Multiple import of the same file within this scope");
- } else {
- ptr_set_add(&parent_scope->imported, scope);
- }
- if (id->using_in_list.count == 0) {
- String import_name = path_to_entity_name(id->import_name.string, id->fullpath);
- if (is_blank_ident(import_name)) {
- if (id->is_using) {
- // TODO(bill): Should this be a warning?
- } else {
- error(token, "File name, %.*s, cannot be use as an import name as it is not a valid identifier", LIT(id->import_name.string));
- }
- } else {
- GB_ASSERT(id->import_name.pos.line != 0);
- id->import_name.string = import_name;
- Entity *e = alloc_entity_import_name(parent_scope, id->import_name, t_invalid,
- id->fullpath, id->import_name.string,
- scope);
- add_entity(c, parent_scope, nullptr, e);
- }
- }
- if (id->is_using) {
- if (parent_scope->is_global) {
- error(id->import_name, "#shared_global_scope imports cannot use using");
- return;
- }
- // NOTE(bill): Add imported entities to this file's scope
- if (id->using_in_list.count > 0) {
- for_array(list_index, id->using_in_list) {
- AstNode *node = id->using_in_list[list_index];
- ast_node(ident, Ident, node);
- String name = ident->token.string;
- Entity *e = scope_lookup_entity(scope, name);
- if (e == nullptr) {
- if (is_blank_ident(name)) {
- error(node, "'_' cannot be used as a value");
- } else {
- error(node, "Undeclared name in this importation: '%.*s'", LIT(name));
- }
- continue;
- }
- if (e->scope == parent_scope) continue;
- bool implicit_is_found = ptr_set_exists(&scope->implicit, e);
- if (is_entity_exported(e) && !implicit_is_found) {
- add_entity_use(c, node, e);
- bool ok = add_entity(c, parent_scope, e->identifier, e);
- if (ok) ptr_set_add(&parent_scope->implicit, e);
- } else {
- error(node, "'%.*s' is exported from this scope", LIT(name));
- continue;
- }
- }
- } else {
- for_array(elem_index, scope->elements.entries) {
- Entity *e = scope->elements.entries[elem_index].value;
- if (e->scope == parent_scope) continue;
- bool implicit_is_found = ptr_set_exists(&scope->implicit, e);
- if (is_entity_exported(e) && !implicit_is_found) {
- Entity *prev = scope_lookup_entity(parent_scope, e->token.string);
- // if (prev) gb_printf_err("%.*s\n", LIT(prev->token.string));
- bool ok = add_entity(c, parent_scope, e->identifier, e);
- if (ok) ptr_set_add(&parent_scope->implicit, e);
- }
- }
- }
- }
- ptr_set_add(&c->checked_files, scope->file);
- scope->has_been_imported = true;
- }
- void check_add_export_decl(Checker *c, AstNodeExportDecl *ed) {
- if (ed->been_handled) return;
- ed->been_handled = true;
- Scope *parent_scope = c->context.scope;
- GB_ASSERT(parent_scope->is_file);
- Token token = ed->relpath;
- HashKey key = hash_string(ed->fullpath);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) {
- for_array(scope_index, c->file_scopes.entries) {
- Scope *scope = c->file_scopes.entries[scope_index].value;
- gb_printf_err("%.*s\n", LIT(scope->file->tokenizer.fullpath));
- }
- gb_printf_err("%.*s(%td:%td)\n", LIT(token.pos.file), token.pos.line, token.pos.column);
- GB_PANIC("Unable to find scope for file: %.*s", LIT(ed->fullpath));
- }
- Scope *scope = *found;
- if (scope->is_global) {
- error(token, "Exporting a #shared_global_scope is disallowed and unnecessary");
- return;
- }
- if (parent_scope->is_global) {
- error(ed->token, "'export' cannot be used on #shared_global_scope");
- return;
- }
- if (ptr_set_exists(&parent_scope->imported, scope)) {
- // error(token, "Multiple import of the same file within this scope");
- } else {
- ptr_set_add(&parent_scope->imported, scope);
- }
- if (ed->using_in_list.count > 0) {
- for_array(list_index, ed->using_in_list) {
- AstNode *node = ed->using_in_list[list_index];
- ast_node(ident, Ident, node);
- String name = ident->token.string;
- Entity *e = scope_lookup_entity(scope, name);
- if (e == nullptr) {
- if (is_blank_ident(name)) {
- error(node, "'_' cannot be used as a value");
- } else {
- error(node, "Undeclared name in this importation: '%.*s'", LIT(name));
- }
- continue;
- }
- if (e->scope == parent_scope) continue;
- if (is_entity_exported(e)) {
- add_entity(c, parent_scope, e->identifier, e);
- } else {
- error(node, "'%.*s' is exported from this scope", LIT(name));
- continue;
- }
- }
- } else {
- // NOTE(bill): Add imported entities to this file's scope
- for_array(elem_index, scope->elements.entries) {
- Entity *e = scope->elements.entries[elem_index].value;
- String name = e->token.string;
- if (e->scope == parent_scope) continue;
- DeclInfo *d = e->decl_info;
- // NOTE(bill): The enum #export must be evaluated before use
- if (d != nullptr && d->init_expr != nullptr) {
- AstNode *expr = unparen_expr(d->init_expr);
- if (expr->kind == AstNode_EnumType && expr->EnumType.is_export) {
- check_entity_decl(c, e, d, nullptr);
- }
- }
- if (is_entity_kind_exported(e->kind)) {
- add_entity(c, parent_scope, e->identifier, e);
- }
- }
- }
- ptr_set_add(&c->checked_files, scope->file);
- scope->has_been_imported = true;
- }
- void check_add_foreign_import_decl(Checker *c, AstNode *decl) {
- ast_node(fl, ForeignImportDecl, decl);
- if (fl->been_handled) return;
- fl->been_handled = true;
- Scope *parent_scope = c->context.scope;
- GB_ASSERT(parent_scope->is_file);
- String fullpath = fl->fullpath;
- String library_name = path_to_entity_name(fl->library_name.string, fullpath);
- if (is_blank_ident(library_name)) {
- error(fl->token, "File name, %.*s, cannot be as a library name as it is not a valid identifier", LIT(fl->library_name.string));
- return;
- }
- if (fl->collection_name != "system") {
- char *c_str = gb_alloc_array(heap_allocator(), char, fullpath.len+1);
- defer (gb_free(heap_allocator(), c_str));
- gb_memcopy(c_str, fullpath.text, fullpath.len);
- c_str[fullpath.len] = '\0';
- gbFile f = {};
- gbFileError file_err = gb_file_open(&f, c_str);
- defer (gb_file_close(&f));
- switch (file_err) {
- case gbFileError_Invalid:
- error(decl, "Invalid file or cannot be found ('%.*s')", LIT(fullpath));
- return;
- case gbFileError_NotExists:
- error(decl, "File cannot be found ('%.*s')", LIT(fullpath));
- return;
- }
- }
- GB_ASSERT(fl->library_name.pos.line != 0);
- fl->library_name.string = library_name;
- Entity *e = alloc_entity_library_name(parent_scope, fl->library_name, t_invalid,
- fl->fullpath, library_name);
- add_entity(c, parent_scope, nullptr, e);
- }
- bool collect_checked_files_from_import_decl_list(Checker *c, Array<AstNode *> decls) {
- bool new_files = false;
- for_array(i, decls) {
- AstNode *decl = decls[i];
- switch (decl->kind) {
- case_ast_node(id, ImportDecl, decl);
- HashKey key = hash_string(id->fullpath);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) continue;
- Scope *s = *found;
- if (!ptr_set_exists(&c->checked_files, s->file)) {
- new_files = true;
- ptr_set_add(&c->checked_files, s->file);
- }
- case_end;
- case_ast_node(ed, ExportDecl, decl);
- HashKey key = hash_string(ed->fullpath);
- Scope **found = map_get(&c->file_scopes, key);
- if (found == nullptr) continue;
- Scope *s = *found;
- if (!ptr_set_exists(&c->checked_files, s->file)) {
- new_files = true;
- ptr_set_add(&c->checked_files, s->file);
- }
- case_end;
- }
- }
- return new_files;
- }
- bool collect_checked_files_from_when_stmt(Checker *c, AstNodeWhenStmt *ws) {
- Operand operand = {Addressing_Invalid};
- if (!ws->is_cond_determined) {
- check_expr(c, &operand, ws->cond);
- if (operand.mode != Addressing_Invalid && !is_type_boolean(operand.type)) {
- error(ws->cond, "Non-boolean condition in 'when' statement");
- }
- if (operand.mode != Addressing_Constant) {
- error(ws->cond, "Non-constant condition in 'when' statement");
- }
- ws->is_cond_determined = true;
- ws->determined_cond = operand.value.kind == ExactValue_Bool && operand.value.value_bool;
- }
- if (ws->body == nullptr || ws->body->kind != AstNode_BlockStmt) {
- error(ws->cond, "Invalid body for 'when' statement");
- } else {
- if (ws->determined_cond) {
- return collect_checked_files_from_import_decl_list(c, ws->body->BlockStmt.stmts);
- } else if (ws->else_stmt) {
- switch (ws->else_stmt->kind) {
- case AstNode_BlockStmt:
- return collect_checked_files_from_import_decl_list(c, ws->else_stmt->BlockStmt.stmts);
- case AstNode_WhenStmt:
- return collect_checked_files_from_when_stmt(c, &ws->else_stmt->WhenStmt);
- default:
- error(ws->else_stmt, "Invalid 'else' statement in 'when' statement");
- break;
- }
- }
- }
- return false;
- }
- void check_delayed_file_import_entity(Checker *c, AstNode *decl) {
- Scope *parent_scope = c->context.scope;
- GB_ASSERT(parent_scope->is_file);
- switch (decl->kind) {
- case_ast_node(ws, WhenStmt, decl);
- check_collect_entities_from_when_stmt(c, ws);
- case_end;
- case_ast_node(id, ImportDecl, decl);
- check_add_import_decl(c, id);
- case_end;
- case_ast_node(ed, ExportDecl, decl);
- check_add_export_decl(c, ed);
- case_end;
- case_ast_node(fl, ForeignImportDecl, decl);
- check_add_foreign_import_decl(c, decl);
- case_end;
- }
- }
- // NOTE(bill): Returns true if a new file is present
- bool collect_file_decls(Checker *c, Array<AstNode *> decls);
- bool collect_file_decls_from_when_stmt(Checker *c, AstNodeWhenStmt *ws);
- bool collect_file_decls_from_when_stmt(Checker *c, AstNodeWhenStmt *ws) {
- Operand operand = {Addressing_Invalid};
- if (!ws->is_cond_determined) {
- check_expr(c, &operand, ws->cond);
- if (operand.mode != Addressing_Invalid && !is_type_boolean(operand.type)) {
- error(ws->cond, "Non-boolean condition in 'when' statement");
- }
- if (operand.mode != Addressing_Constant) {
- error(ws->cond, "Non-constant condition in 'when' statement");
- }
- ws->is_cond_determined = true;
- ws->determined_cond = operand.value.kind == ExactValue_Bool && operand.value.value_bool;
- }
- if (ws->body == nullptr || ws->body->kind != AstNode_BlockStmt) {
- error(ws->cond, "Invalid body for 'when' statement");
- } else {
- if (ws->determined_cond) {
- return collect_file_decls(c, ws->body->BlockStmt.stmts);
- } else if (ws->else_stmt) {
- switch (ws->else_stmt->kind) {
- case AstNode_BlockStmt:
- return collect_file_decls(c, ws->else_stmt->BlockStmt.stmts);
- case AstNode_WhenStmt:
- return collect_file_decls_from_when_stmt(c, &ws->else_stmt->WhenStmt);
- default:
- error(ws->else_stmt, "Invalid 'else' statement in 'when' statement");
- break;
- }
- }
- }
- return false;
- }
- bool collect_file_decls(Checker *c, Array<AstNode *> decls) {
- Scope *parent_scope = c->context.scope;
- GB_ASSERT(parent_scope->is_file);
- if (collect_checked_files_from_import_decl_list(c, decls)) {
- return true;
- }
- for_array(i, decls) {
- AstNode *decl = decls[i];
- switch (decl->kind) {
- case_ast_node(vd, ValueDecl, decl);
- check_collect_value_decl(c, decl);
- case_end;
- case_ast_node(id, ImportDecl, decl);
- check_add_import_decl(c, id);
- case_end;
- case_ast_node(ed, ExportDecl, decl);
- check_add_export_decl(c, ed);
- case_end;
- case_ast_node(fl, ForeignImportDecl, decl);
- check_add_foreign_import_decl(c, decl);
- case_end;
- case_ast_node(fb, ForeignBlockDecl, decl);
- check_add_foreign_block_decl(c, decl);
- case_end;
- case_ast_node(ws, WhenStmt, decl);
- if (!ws->is_cond_determined) {
- if (collect_checked_files_from_when_stmt(c, ws)) {
- return true;
- }
- CheckerContext prev_context = c->context;
- defer (c->context = prev_context);
- c->context.collect_delayed_decls = true;
- if (collect_file_decls_from_when_stmt(c, ws)) {
- return true;
- }
- } else {
- CheckerContext prev_context = c->context;
- defer (c->context = prev_context);
- c->context.collect_delayed_decls = true;
- if (collect_file_decls_from_when_stmt(c, ws)) {
- return true;
- }
- }
- case_end;
- case_ast_node(ce, CallExpr, decl);
- if (ce->proc->kind == AstNode_BasicDirective &&
- ce->proc->BasicDirective.name == "assert") {
- Operand o = {};
- check_expr(c, &o, decl);
- }
- case_end;
- }
- }
- return false;
- }
- void check_import_entities(Checker *c) {
- Array<ImportGraphNode *> dep_graph = generate_import_dependency_graph(c);
- defer ({
- for_array(i, dep_graph) {
- import_graph_node_destroy(dep_graph[i], heap_allocator());
- }
- array_free(&dep_graph);
- });
- // NOTE(bill): Priority queue
- auto pq = priority_queue_create(dep_graph, import_graph_node_cmp, import_graph_node_swap);
- PtrSet<Scope *> emitted = {};
- ptr_set_init(&emitted, heap_allocator());
- defer (ptr_set_destroy(&emitted));
- while (pq.queue.count > 0) {
- ImportGraphNode *n = priority_queue_pop(&pq);
- Scope *s = n->scope;
- if (n->dep_count > 0) {
- PtrSet<Scope *> visited = {};
- ptr_set_init(&visited, heap_allocator());
- defer (ptr_set_destroy(&visited));
- auto path = find_import_path(c, s, s, &visited);
- defer (array_free(&path));
- // TODO(bill): This needs better TokenPos finding
- auto const fn = [](ImportPathItem item) -> String {
- Scope *s = item.scope;
- return remove_directory_from_path(s->file->tokenizer.fullpath);
- };
- if (path.count == 1) {
- // TODO(bill): Should this be allowed or disabled?
- #if 0
- ImportPathItem item = path[0];
- String filename = fn(item);
- error(item.decl, "Self importation of '%.*s'", LIT(filename));
- #endif
- } else if (path.count > 0) {
- ImportPathItem item = path[path.count-1];
- String filename = fn(item);
- error(item.decl, "Cyclic importation of '%.*s'", LIT(filename));
- for (isize i = 0; i < path.count; i++) {
- error(item.decl, "'%.*s' refers to", LIT(filename));
- item = path[i];
- filename = fn(item);
- }
- error(item.decl, "'%.*s'", LIT(filename));
- }
- }
- for_array(i, n->pred.entries) {
- ImportGraphNode *p = n->pred.entries[i].ptr;
- p->dep_count = gb_max(p->dep_count-1, 0);
- // p->dep_count -= 1;
- priority_queue_fix(&pq, p->index);
- }
- if (s == nullptr) {
- continue;
- }
- if (ptr_set_exists(&emitted, s)) {
- continue;
- }
- ptr_set_add(&emitted, s);
- array_add(&c->file_order, n);
- }
- for_array(file_index, c->parser->files) {
- AstFile *f = c->parser->files[file_index];
- Scope *s = f->scope;
- if (s->is_init || s->is_global) {
- ptr_set_add(&c->checked_files, f);
- }
- }
- // for_array(file_index, c->file_order) {
- // ImportGraphNode *node = c->file_order[file_index];
- // AstFile *f = node->scope->file;
- // gb_printf_err("--- %.*s -> %td\n", LIT(f->fullpath), node->succ.entries.count);
- // }
- for (;;) {
- bool new_files = false;
- for_array(file_index, c->file_order) {
- ImportGraphNode *node = c->file_order[file_index];
- AstFile *f = node->scope->file;
- if (!ptr_set_exists(&c->checked_files, f)) {
- continue;
- }
- CheckerContext prev_context = c->context;
- defer (c->context = prev_context);
- add_curr_ast_file(c, f);
- new_files |= collect_checked_files_from_import_decl_list(c, f->decls);
- }
- if (new_files) break;
- }
- for (isize file_index = 0; file_index < c->file_order.count; file_index += 1) {
- ImportGraphNode *node = c->file_order[file_index];
- AstFile *f = node->scope->file;
- if (!ptr_set_exists(&c->checked_files, f)) {
- continue;
- }
- CheckerContext prev_context = c->context;
- defer (c->context = prev_context);
- c->context.collect_delayed_decls = true;
- add_curr_ast_file(c, f);
- bool new_files = collect_file_decls(c, f->decls);
- if (new_files) {
- // TODO(bill): Only start from the lowest new file
- file_index = -1;
- continue;
- }
- }
- // gb_printf_err("End here!\n");
- // gb_exit(1);
- }
- Array<Entity *> find_entity_path(Entity *start, Entity *end, Map<Entity *> *visited = nullptr) {
- Map<Entity *> visited_ = {};
- bool made_visited = false;
- if (visited == nullptr) {
- made_visited = true;
- map_init(&visited_, heap_allocator());
- visited = &visited_;
- }
- defer (if (made_visited) {
- map_destroy(&visited_);
- });
- Array<Entity *> empty_path = {};
- HashKey key = hash_pointer(start);
- if (map_get(visited, key) != nullptr) {
- return empty_path;
- }
- map_set(visited, key, start);
- DeclInfo *decl = start->decl_info;
- if (decl) {
- for_array(i, decl->deps.entries) {
- Entity *dep = decl->deps.entries[i].ptr;
- if (dep == end) {
- auto path = array_make<Entity *>(heap_allocator());
- array_add(&path, dep);
- return path;
- }
- auto next_path = find_entity_path(dep, end, visited);
- if (next_path.count > 0) {
- array_add(&next_path, dep);
- return next_path;
- }
- }
- }
- return empty_path;
- }
- void calculate_global_init_order(Checker *c) {
- #if 0
- Timings timings = {};
- timings_init(&timings, str_lit("calculate_global_init_order"), 16);
- defer ({
- timings_print_all(&timings);
- timings_destroy(&timings);
- });
- #define TIME_SECTION(str) timings_start_section(&timings, str_lit(str))
- #else
- #define TIME_SECTION(str)
- #endif
- CheckerInfo *info = &c->info;
- TIME_SECTION("generate entity dependency graph");
- Array<EntityGraphNode *> dep_graph = generate_entity_dependency_graph(info);
- defer ({
- for_array(i, dep_graph) {
- entity_graph_node_destroy(dep_graph[i], heap_allocator());
- }
- array_free(&dep_graph);
- });
- TIME_SECTION("priority queue create");
- // NOTE(bill): Priority queue
- auto pq = priority_queue_create(dep_graph, entity_graph_node_cmp, entity_graph_node_swap);
- PtrSet<DeclInfo *> emitted = {};
- ptr_set_init(&emitted, heap_allocator());
- defer (ptr_set_destroy(&emitted));
- TIME_SECTION("queue sort");
- while (pq.queue.count > 0) {
- EntityGraphNode *n = priority_queue_pop(&pq);
- Entity *e = n->entity;
- if (n->dep_count > 0) {
- auto path = find_entity_path(e, e);
- defer (array_free(&path));
- if (path.count > 0) {
- Entity *e = path[0];
- error(e->token, "Cyclic initialization of '%.*s'", LIT(e->token.string));
- for (isize i = path.count-1; i >= 0; i--) {
- error(e->token, "\t'%.*s' refers to", LIT(e->token.string));
- e = path[i];
- }
- error(e->token, "\t'%.*s'", LIT(e->token.string));
- }
- }
- for_array(i, n->pred.entries) {
- EntityGraphNode *p = n->pred.entries[i].ptr;
- p->dep_count -= 1;
- priority_queue_fix(&pq, p->index);
- }
- if (e == nullptr || e->kind != Entity_Variable) {
- continue;
- }
- DeclInfo *d = decl_info_of_entity(info, e);
- if (ptr_set_exists(&emitted, d)) {
- continue;
- }
- ptr_set_add(&emitted, d);
- if (d->entities == nullptr) {
- d->entities = gb_alloc_array(c->allocator, Entity *, 1);
- d->entities[0] = e;
- d->entity_count = 1;
- }
- array_add(&info->variable_init_order, d);
- }
- if (false) {
- gb_printf("Variable Initialization Order:\n");
- for_array(i, info->variable_init_order) {
- DeclInfo *d = info->variable_init_order[i];
- for (isize j = 0; j < d->entity_count; j++) {
- Entity *e = d->entities[j];
- if (j == 0) gb_printf("\t");
- if (j > 0) gb_printf(", ");
- gb_printf("'%.*s' %td", LIT(e->token.string), e->order_in_src);
- }
- gb_printf("\n");
- }
- gb_printf("\n");
- }
- #undef TIME_SECTION
- }
- void check_parsed_files(Checker *c) {
- #if 0
- Timings timings = {};
- timings_init(&timings, str_lit("check_parsed_files"), 16);
- defer ({
- timings_print_all(&timings);
- timings_destroy(&timings);
- });
- #define TIME_SECTION(str) timings_start_section(&timings, str_lit(str))
- #else
- #define TIME_SECTION(str)
- #endif
- TIME_SECTION("map full filepaths to scope");
- add_type_info_type(c, t_invalid);
- // Map full filepaths to Scopes
- for_array(i, c->parser->files) {
- AstFile *f = c->parser->files[i];
- Scope *scope = create_scope_from_file(c, f);
- f->decl_info = make_declaration_info(c->allocator, f->scope, c->context.decl);
- HashKey key = hash_string(f->tokenizer.fullpath);
- map_set(&c->file_scopes, key, scope);
- map_set(&c->info.files, key, f);
- if (scope->is_init) {
- c->info.init_scope = scope;
- }
- }
- TIME_SECTION("collect entities");
- // Collect Entities
- for_array(i, c->parser->files) {
- AstFile *f = c->parser->files[i];
- CheckerContext prev_context = c->context;
- add_curr_ast_file(c, f);
- check_collect_entities(c, f->decls);
- c->context = prev_context;
- }
- TIME_SECTION("import entities");
- check_import_entities(c);
- TIME_SECTION("check all global entities");
- check_all_global_entities(c);
- TIME_SECTION("init preload");
- init_preload(c); // NOTE(bill): This could be setup previously through the use of 'type_info_of'
- TIME_SECTION("check procedure bodies");
- // Check procedure bodies
- // NOTE(bill): Nested procedures bodies will be added to this "queue"
- for_array(i, c->procs) {
- ProcedureInfo *pi = &c->procs[i];
- if (pi->type == nullptr) {
- continue;
- }
- CheckerContext prev_context = c->context;
- defer (c->context = prev_context);
- TypeProc *pt = &pi->type->Proc;
- String name = pi->token.string;
- if (pt->is_polymorphic) {
- GB_ASSERT_MSG(pt->is_poly_specialized, "%.*s", LIT(name));
- }
- add_curr_ast_file(c, pi->file);
- bool bounds_check = (pi->tags & ProcTag_bounds_check) != 0;
- bool no_bounds_check = (pi->tags & ProcTag_no_bounds_check) != 0;
- if (bounds_check) {
- c->context.stmt_state_flags |= StmtStateFlag_bounds_check;
- c->context.stmt_state_flags &= ~StmtStateFlag_no_bounds_check;
- } else if (no_bounds_check) {
- c->context.stmt_state_flags |= StmtStateFlag_no_bounds_check;
- c->context.stmt_state_flags &= ~StmtStateFlag_bounds_check;
- }
- check_proc_body(c, pi->token, pi->decl, pi->type, pi->body);
- }
- TIME_SECTION("generate minimum dependency set");
- c->info.minimum_dependency_set = generate_minimum_dependency_set(c, c->info.entry_point);
- TIME_SECTION("calculate global init order");
- // Calculate initialization order of global variables
- calculate_global_init_order(c);
- TIME_SECTION("add untyped expression values");
- // Add untyped expression values
- for_array(i, c->info.untyped.entries) {
- auto *entry = &c->info.untyped.entries[i];
- HashKey key = entry->key;
- AstNode *expr = cast(AstNode *)key.ptr;
- ExprInfo *info = &entry->value;
- if (info != nullptr && expr != nullptr) {
- if (is_type_typed(info->type)) {
- compiler_error("%s (type %s) is typed!", expr_to_string(expr), type_to_string(info->type));
- }
- add_type_and_value(&c->info, expr, info->mode, info->type, info->value);
- }
- }
- // TODO(bill): Check for unused imports (and remove) or even warn/err
- // TODO(bill): Any other checks?
- TIME_SECTION("add type information");
- // Add "Basic" type information
- for (isize i = 0; i < Basic_COUNT; i++) {
- Type *t = &basic_types[i];
- if (t->Basic.size > 0 &&
- (t->Basic.flags & BasicFlag_LLVM) == 0) {
- add_type_info_type(c, t);
- }
- }
- // NOTE(bill): Check for illegal cyclic type declarations
- for_array(i, c->info.definitions) {
- Entity *e = c->info.definitions[i];
- if (e->kind == Entity_TypeName && e->type != nullptr) {
- // i64 size = type_size_of(c->allocator, e->type);
- i64 align = type_align_of(e->type);
- if (align > 0 && ptr_set_exists(&c->info.minimum_dependency_set, e)) {
- add_type_info_type(c, e->type);
- }
- }
- }
- TIME_SECTION("check entry point");
- if (!build_context.is_dll) {
- Scope *s = c->info.init_scope;
- GB_ASSERT(s != nullptr);
- GB_ASSERT(s->is_init);
- Entity *e = current_scope_lookup_entity(s, str_lit("main"));
- if (e == nullptr) {
- Token token = {};
- if (s->file->tokens.count > 0) {
- token = s->file->tokens[0];
- } else {
- token.pos.file = s->file->tokenizer.fullpath;
- token.pos.line = 1;
- token.pos.column = 1;
- }
- error(token, "Undefined entry point procedure 'main'");
- }
- }
- #undef TIME_SECTION
- }
|