| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886 |
- typedef struct AstNode AstNode;
- typedef struct Scope Scope;
- typedef struct DeclInfo DeclInfo;
- typedef enum ParseFileError {
- ParseFile_None,
- ParseFile_WrongExtension,
- ParseFile_InvalidFile,
- ParseFile_EmptyFile,
- ParseFile_Permission,
- ParseFile_NotFound,
- ParseFile_InvalidToken,
- ParseFile_Count,
- } ParseFileError;
- typedef Array(AstNode *) AstNodeArray;
- typedef struct AstFile {
- i32 id;
- gbArena arena;
- Tokenizer tokenizer;
- Array(Token) tokens;
- isize curr_token_index;
- Token curr_token;
- Token prev_token; // previous non-comment
- // >= 0: In Expression
- // < 0: In Control Clause
- // NOTE(bill): Used to prevent type literals in control clauses
- isize expr_level;
- AstNodeArray decls;
- bool is_global_scope;
- AstNode * curr_proc;
- isize scope_level;
- Scope * scope; // NOTE(bill): Created in checker
- DeclInfo * decl_info; // NOTE(bill): Created in checker
- // TODO(bill): Error recovery
- #define PARSER_MAX_FIX_COUNT 6
- isize fix_count;
- TokenPos fix_prev_pos;
- } AstFile;
- typedef struct ImportedFile {
- String path;
- String rel_path;
- TokenPos pos; // #import
- } ImportedFile;
- typedef struct Parser {
- String init_fullpath;
- Array(AstFile) files;
- Array(ImportedFile) imports;
- gbAtomic32 import_index;
- isize total_token_count;
- isize total_line_count;
- gbMutex mutex;
- } Parser;
- typedef enum ProcTag {
- ProcTag_bounds_check = 1<<0,
- ProcTag_no_bounds_check = 1<<1,
- ProcTag_foreign = 1<<10,
- ProcTag_export = 1<<11,
- ProcTag_link_name = 1<<12,
- ProcTag_inline = 1<<13,
- ProcTag_no_inline = 1<<14,
- // ProcTag_dll_import = 1<<15,
- // ProcTag_dll_export = 1<<16,
- } ProcTag;
- typedef enum ProcCallingConvention {
- ProcCC_Odin = 0,
- ProcCC_C = 1,
- ProcCC_Std = 2,
- ProcCC_Fast = 3,
- ProcCC_Invalid,
- } ProcCallingConvention;
- typedef enum VarDeclFlag {
- VarDeclFlag_using = 1<<0,
- VarDeclFlag_immutable = 1<<1,
- VarDeclFlag_thread_local = 1<<2,
- } VarDeclFlag;
- typedef enum StmtStateFlag {
- StmtStateFlag_bounds_check = 1<<0,
- StmtStateFlag_no_bounds_check = 1<<1,
- } StmtStateFlag;
- typedef enum FieldFlag {
- FieldFlag_ellipsis = 1<<0,
- FieldFlag_using = 1<<1,
- FieldFlag_no_alias = 1<<2,
- FieldFlag_immutable = 1<<3,
- FieldFlag_Signature = FieldFlag_ellipsis|FieldFlag_using|FieldFlag_no_alias|FieldFlag_immutable,
- } FieldListTag;
- AstNodeArray make_ast_node_array(AstFile *f) {
- AstNodeArray a;
- // array_init(&a, gb_arena_allocator(&f->arena));
- array_init(&a, heap_allocator());
- return a;
- }
- // NOTE(bill): This massive define is so it is possible to create a discriminated union (and extra debug info)
- // for the AstNode. I personally prefer discriminated unions over subtype polymorphism as I can preallocate
- // all the nodes and even memcpy in a different kind of node
- #define AST_NODE_KINDS \
- AST_NODE_KIND(Ident, "identifier", Token) \
- AST_NODE_KIND(Implicit, "implicit", Token) \
- AST_NODE_KIND(BasicLit, "basic literal", Token) \
- AST_NODE_KIND(BasicDirective, "basic directive", struct { \
- Token token; \
- String name; \
- }) \
- AST_NODE_KIND(Ellipsis, "ellipsis", struct { \
- Token token; \
- AstNode *expr; \
- }) \
- AST_NODE_KIND(ProcLit, "procedure literal", struct { \
- AstNode *type; \
- AstNode *body; \
- u64 tags; \
- AstNode *foreign_library; \
- String foreign_name; \
- String link_name; \
- }) \
- AST_NODE_KIND(CompoundLit, "compound literal", struct { \
- AstNode *type; \
- AstNodeArray elems; \
- Token open, close; \
- }) \
- AST_NODE_KIND(Alias, "alias", struct { \
- Token token; \
- AstNode *expr; \
- }) \
- AST_NODE_KIND(_ExprBegin, "", i32) \
- AST_NODE_KIND(BadExpr, "bad expression", struct { Token begin, end; }) \
- AST_NODE_KIND(TagExpr, "tag expression", struct { Token token, name; AstNode *expr; }) \
- AST_NODE_KIND(RunExpr, "run expression", struct { Token token, name; AstNode *expr; }) \
- AST_NODE_KIND(UnaryExpr, "unary expression", struct { Token op; AstNode *expr; }) \
- AST_NODE_KIND(BinaryExpr, "binary expression", struct { Token op; AstNode *left, *right; } ) \
- AST_NODE_KIND(ParenExpr, "parentheses expression", struct { AstNode *expr; Token open, close; }) \
- AST_NODE_KIND(SelectorExpr, "selector expression", struct { Token token; AstNode *expr, *selector; }) \
- AST_NODE_KIND(IndexExpr, "index expression", struct { AstNode *expr, *index; Token open, close; }) \
- AST_NODE_KIND(DerefExpr, "dereference expression", struct { Token op; AstNode *expr; }) \
- AST_NODE_KIND(SliceExpr, "slice expression", struct { \
- AstNode *expr; \
- Token open, close; \
- bool index3; \
- AstNode *low, *high, *max; \
- }) \
- AST_NODE_KIND(CallExpr, "call expression", struct { \
- AstNode * proc; \
- AstNodeArray args; \
- Token open; \
- Token close; \
- Token ellipsis; \
- }) \
- AST_NODE_KIND(MacroCallExpr, "macro call expression", struct { \
- AstNode * macro; \
- Token bang; \
- AstNodeArray args; \
- Token open; \
- Token close; \
- }) \
- AST_NODE_KIND(CastExpr, "cast expression", struct { Token token; AstNode *type, *expr; Token open, close; }) \
- AST_NODE_KIND(FieldValue, "field value", struct { Token eq; AstNode *field, *value; }) \
- AST_NODE_KIND(TernaryExpr, "ternary expression", struct { AstNode *cond, *x, *y; }) \
- AST_NODE_KIND(IntervalExpr, "interval expression", struct { Token op; AstNode *left, *right; }) \
- AST_NODE_KIND(_ExprEnd, "", i32) \
- AST_NODE_KIND(_StmtBegin, "", i32) \
- AST_NODE_KIND(BadStmt, "bad statement", struct { Token begin, end; }) \
- AST_NODE_KIND(EmptyStmt, "empty statement", struct { Token token; }) \
- AST_NODE_KIND(ExprStmt, "expression statement", struct { AstNode *expr; } ) \
- AST_NODE_KIND(TagStmt, "tag statement", struct { \
- Token token; \
- Token name; \
- AstNode *stmt; \
- }) \
- AST_NODE_KIND(AssignStmt, "assign statement", struct { \
- Token op; \
- AstNodeArray lhs, rhs; \
- }) \
- AST_NODE_KIND(IncDecStmt, "increment decrement statement", struct { \
- Token op; \
- AstNode *expr; \
- }) \
- AST_NODE_KIND(_ComplexStmtBegin, "", i32) \
- AST_NODE_KIND(BlockStmt, "block statement", struct { \
- AstNodeArray stmts; \
- Token open, close; \
- }) \
- AST_NODE_KIND(IfStmt, "if statement", struct { \
- Token token; \
- AstNode *init; \
- AstNode *cond; \
- AstNode *body; \
- AstNode *else_stmt; \
- }) \
- AST_NODE_KIND(WhenStmt, "when statement", struct { \
- Token token; \
- AstNode *cond; \
- AstNode *body; \
- AstNode *else_stmt; \
- }) \
- AST_NODE_KIND(ReturnStmt, "return statement", struct { \
- Token token; \
- AstNodeArray results; \
- }) \
- AST_NODE_KIND(ForStmt, "for statement", struct { \
- Token token; \
- AstNode *label; \
- AstNode *init; \
- AstNode *cond; \
- AstNode *post; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(RangeStmt, "range statement", struct { \
- Token token; \
- AstNode *label; \
- AstNode *value; \
- AstNode *index; \
- Token in_token; \
- AstNode *expr; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(CaseClause, "case clause", struct { \
- Token token; \
- AstNodeArray list; \
- AstNodeArray stmts; \
- }) \
- AST_NODE_KIND(MatchStmt, "match statement", struct { \
- Token token; \
- AstNode *label; \
- AstNode *init; \
- AstNode *tag; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(TypeMatchStmt, "type match statement", struct { \
- Token token; \
- AstNode *label; \
- AstNode *tag; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(DeferStmt, "defer statement", struct { Token token; AstNode *stmt; }) \
- AST_NODE_KIND(BranchStmt, "branch statement", struct { Token token; AstNode *label; }) \
- AST_NODE_KIND(UsingStmt, "using statement", struct { \
- Token token; \
- AstNodeArray list; \
- }) \
- AST_NODE_KIND(AsmOperand, "assembly operand", struct { \
- Token string; \
- AstNode *operand; \
- }) \
- AST_NODE_KIND(AsmStmt, "assembly statement", struct { \
- Token token; \
- bool is_volatile; \
- Token open, close; \
- Token code_string; \
- AstNode *output_list; \
- AstNode *input_list; \
- AstNode *clobber_list; \
- isize output_count, input_count, clobber_count; \
- }) \
- AST_NODE_KIND(PushAllocator, "push_allocator statement", struct { \
- Token token; \
- AstNode *expr; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(PushContext, "push_context statement", struct { \
- Token token; \
- AstNode *expr; \
- AstNode *body; \
- }) \
- AST_NODE_KIND(_ComplexStmtEnd, "", i32) \
- AST_NODE_KIND(_StmtEnd, "", i32) \
- AST_NODE_KIND(_DeclBegin, "", i32) \
- AST_NODE_KIND(BadDecl, "bad declaration", struct { Token begin, end; }) \
- AST_NODE_KIND(ValueDecl, "value declaration", struct { \
- bool is_var; \
- AstNodeArray names; \
- AstNode * type; \
- AstNodeArray values; \
- u32 flags; \
- }) \
- AST_NODE_KIND(ImportDecl, "import declaration", struct { \
- Token token; \
- bool is_import; \
- Token relpath; \
- String fullpath; \
- Token import_name; \
- AstNode *cond; \
- AstNode *note; \
- }) \
- AST_NODE_KIND(ForeignLibrary, "foreign library", struct { \
- Token token, filepath; \
- Token library_name; \
- String base_dir; \
- AstNode *cond; \
- bool is_system; \
- }) \
- AST_NODE_KIND(Label, "label", struct { \
- Token token; \
- AstNode *name; \
- }) \
- AST_NODE_KIND(_DeclEnd, "", i32) \
- AST_NODE_KIND(Field, "field", struct { \
- AstNodeArray names; \
- AstNode * type; \
- u32 flags; \
- }) \
- AST_NODE_KIND(FieldList, "field list", struct { \
- Token token; \
- AstNodeArray list; \
- }) \
- AST_NODE_KIND(UnionField, "union field", struct { \
- AstNode *name; \
- AstNode *list; \
- }) \
- AST_NODE_KIND(_TypeBegin, "", i32) \
- AST_NODE_KIND(HelperType, "type", struct { \
- Token token; \
- AstNode *type; \
- }) \
- AST_NODE_KIND(ProcType, "procedure type", struct { \
- Token token; \
- AstNode *params; \
- AstNode *results; \
- u64 tags; \
- ProcCallingConvention calling_convention; \
- }) \
- AST_NODE_KIND(PointerType, "pointer type", struct { \
- Token token; \
- AstNode *type; \
- }) \
- AST_NODE_KIND(ArrayType, "array type", struct { \
- Token token; \
- AstNode *count; \
- AstNode *elem; \
- }) \
- AST_NODE_KIND(DynamicArrayType, "dynamic array type", struct { \
- Token token; \
- AstNode *elem; \
- }) \
- AST_NODE_KIND(VectorType, "vector type", struct { \
- Token token; \
- AstNode *count; \
- AstNode *elem; \
- }) \
- AST_NODE_KIND(StructType, "struct type", struct { \
- Token token; \
- AstNodeArray fields; \
- isize field_count; \
- bool is_packed; \
- bool is_ordered; \
- AstNode *align; \
- }) \
- AST_NODE_KIND(UnionType, "union type", struct { \
- Token token; \
- AstNodeArray fields; \
- isize field_count; \
- AstNodeArray variants; \
- }) \
- AST_NODE_KIND(RawUnionType, "raw union type", struct { \
- Token token; \
- AstNodeArray fields; \
- isize field_count; \
- }) \
- AST_NODE_KIND(EnumType, "enum type", struct { \
- Token token; \
- AstNode *base_type; \
- AstNodeArray fields; /* FieldValue */ \
- }) \
- AST_NODE_KIND(MapType, "map type", struct { \
- Token token; \
- AstNode *count; \
- AstNode *key; \
- AstNode *value; \
- }) \
- AST_NODE_KIND(_TypeEnd, "", i32)
- typedef enum AstNodeKind {
- AstNode_Invalid,
- #define AST_NODE_KIND(_kind_name_, ...) GB_JOIN2(AstNode_, _kind_name_),
- AST_NODE_KINDS
- #undef AST_NODE_KIND
- AstNode_Count,
- } AstNodeKind;
- String const ast_node_strings[] = {
- {cast(u8 *)"invalid node", gb_size_of("invalid node")},
- #define AST_NODE_KIND(_kind_name_, name, ...) {cast(u8 *)name, gb_size_of(name)-1},
- AST_NODE_KINDS
- #undef AST_NODE_KIND
- };
- #define AST_NODE_KIND(_kind_name_, name, ...) typedef __VA_ARGS__ GB_JOIN2(AstNode, _kind_name_);
- AST_NODE_KINDS
- #undef AST_NODE_KIND
- typedef struct AstNode {
- AstNodeKind kind;
- u32 stmt_state_flags;
- union {
- #define AST_NODE_KIND(_kind_name_, name, ...) GB_JOIN2(AstNode, _kind_name_) _kind_name_;
- AST_NODE_KINDS
- #undef AST_NODE_KIND
- };
- } AstNode;
- #define ast_node(n_, Kind_, node_) GB_JOIN2(AstNode, Kind_) *n_ = &(node_)->Kind_; GB_ASSERT((node_)->kind == GB_JOIN2(AstNode_, Kind_))
- #define case_ast_node(n_, Kind_, node_) case GB_JOIN2(AstNode_, Kind_): { ast_node(n_, Kind_, node_);
- #ifndef case_end
- #define case_end } break;
- #endif
- gb_inline bool is_ast_node_expr(AstNode *node) {
- return gb_is_between(node->kind, AstNode__ExprBegin+1, AstNode__ExprEnd-1);
- }
- gb_inline bool is_ast_node_stmt(AstNode *node) {
- return gb_is_between(node->kind, AstNode__StmtBegin+1, AstNode__StmtEnd-1);
- }
- gb_inline bool is_ast_node_complex_stmt(AstNode *node) {
- return gb_is_between(node->kind, AstNode__ComplexStmtBegin+1, AstNode__ComplexStmtEnd-1);
- }
- gb_inline bool is_ast_node_decl(AstNode *node) {
- return gb_is_between(node->kind, AstNode__DeclBegin+1, AstNode__DeclEnd-1);
- }
- gb_inline bool is_ast_node_type(AstNode *node) {
- return gb_is_between(node->kind, AstNode__TypeBegin+1, AstNode__TypeEnd-1);
- }
- gb_inline bool is_ast_node_when_stmt(AstNode *node) {
- return node->kind == AstNode_WhenStmt;
- }
- Token ast_node_token(AstNode *node) {
- switch (node->kind) {
- case AstNode_Ident: return node->Ident;
- case AstNode_Implicit: return node->Implicit;
- case AstNode_BasicLit: return node->BasicLit;
- case AstNode_BasicDirective: return node->BasicDirective.token;
- case AstNode_ProcLit: return ast_node_token(node->ProcLit.type);
- case AstNode_CompoundLit:
- if (node->CompoundLit.type != NULL) {
- return ast_node_token(node->CompoundLit.type);
- }
- return node->CompoundLit.open;
- case AstNode_Alias: return node->Alias.token;
- case AstNode_TagExpr: return node->TagExpr.token;
- case AstNode_RunExpr: return node->RunExpr.token;
- case AstNode_BadExpr: return node->BadExpr.begin;
- case AstNode_UnaryExpr: return node->UnaryExpr.op;
- case AstNode_BinaryExpr: return ast_node_token(node->BinaryExpr.left);
- case AstNode_ParenExpr: return node->ParenExpr.open;
- case AstNode_CallExpr: return ast_node_token(node->CallExpr.proc);
- case AstNode_MacroCallExpr: return ast_node_token(node->MacroCallExpr.macro);
- case AstNode_SelectorExpr:
- if (node->SelectorExpr.selector != NULL) {
- return ast_node_token(node->SelectorExpr.selector);
- }
- return node->SelectorExpr.token;
- case AstNode_IndexExpr: return node->IndexExpr.open;
- case AstNode_SliceExpr: return node->SliceExpr.open;
- case AstNode_Ellipsis: return node->Ellipsis.token;
- case AstNode_CastExpr: return node->CastExpr.token;
- case AstNode_FieldValue: return node->FieldValue.eq;
- case AstNode_DerefExpr: return node->DerefExpr.op;
- case AstNode_TernaryExpr: return ast_node_token(node->TernaryExpr.cond);
- case AstNode_IntervalExpr: return ast_node_token(node->IntervalExpr.left);
- case AstNode_BadStmt: return node->BadStmt.begin;
- case AstNode_EmptyStmt: return node->EmptyStmt.token;
- case AstNode_ExprStmt: return ast_node_token(node->ExprStmt.expr);
- case AstNode_TagStmt: return node->TagStmt.token;
- case AstNode_AssignStmt: return node->AssignStmt.op;
- case AstNode_IncDecStmt: return ast_node_token(node->IncDecStmt.expr);
- case AstNode_BlockStmt: return node->BlockStmt.open;
- case AstNode_IfStmt: return node->IfStmt.token;
- case AstNode_WhenStmt: return node->WhenStmt.token;
- case AstNode_ReturnStmt: return node->ReturnStmt.token;
- case AstNode_ForStmt: return node->ForStmt.token;
- case AstNode_RangeStmt: return node->RangeStmt.token;
- case AstNode_MatchStmt: return node->MatchStmt.token;
- case AstNode_CaseClause: return node->CaseClause.token;
- case AstNode_DeferStmt: return node->DeferStmt.token;
- case AstNode_BranchStmt: return node->BranchStmt.token;
- case AstNode_UsingStmt: return node->UsingStmt.token;
- case AstNode_AsmStmt: return node->AsmStmt.token;
- case AstNode_PushAllocator: return node->PushAllocator.token;
- case AstNode_PushContext: return node->PushContext.token;
- case AstNode_BadDecl: return node->BadDecl.begin;
- case AstNode_ValueDecl: return ast_node_token(node->ValueDecl.names.e[0]);
- case AstNode_ImportDecl: return node->ImportDecl.token;
- case AstNode_ForeignLibrary: return node->ForeignLibrary.token;
- case AstNode_Label: return node->Label.token;
- case AstNode_Field:
- if (node->Field.names.count > 0) {
- return ast_node_token(node->Field.names.e[0]);
- }
- return ast_node_token(node->Field.type);
- case AstNode_FieldList:
- return node->FieldList.token;
- case AstNode_UnionField:
- return ast_node_token(node->UnionField.name);
- case AstNode_HelperType: return node->HelperType.token;
- case AstNode_ProcType: return node->ProcType.token;
- case AstNode_PointerType: return node->PointerType.token;
- case AstNode_ArrayType: return node->ArrayType.token;
- case AstNode_DynamicArrayType: return node->DynamicArrayType.token;
- case AstNode_VectorType: return node->VectorType.token;
- case AstNode_StructType: return node->StructType.token;
- case AstNode_UnionType: return node->UnionType.token;
- case AstNode_RawUnionType: return node->RawUnionType.token;
- case AstNode_EnumType: return node->EnumType.token;
- case AstNode_MapType: return node->MapType.token;
- }
- return empty_token;
- }
- void error_node(AstNode *node, char *fmt, ...) {
- va_list va;
- va_start(va, fmt);
- error_va(ast_node_token(node), fmt, va);
- va_end(va);
- }
- void warning_node(AstNode *node, char *fmt, ...) {
- va_list va;
- va_start(va, fmt);
- warning_va(ast_node_token(node), fmt, va);
- va_end(va);
- }
- void syntax_error_node(AstNode *node, char *fmt, ...) {
- va_list va;
- va_start(va, fmt);
- syntax_error_va(ast_node_token(node), fmt, va);
- va_end(va);
- }
- bool ast_node_expect(AstNode *node, AstNodeKind kind) {
- if (node->kind != kind) {
- error_node(node, "Expected %.*s, got %.*s", LIT(ast_node_strings[node->kind]));
- return false;
- }
- return true;
- }
- // NOTE(bill): And this below is why is I/we need a new language! Discriminated unions are a pain in C/C++
- AstNode *make_ast_node(AstFile *f, AstNodeKind kind) {
- gbArena *arena = &f->arena;
- if (gb_arena_size_remaining(arena, GB_DEFAULT_MEMORY_ALIGNMENT) <= gb_size_of(AstNode)) {
- // NOTE(bill): If a syntax error is so bad, just quit!
- gb_exit(1);
- }
- AstNode *node = gb_alloc_item(gb_arena_allocator(arena), AstNode);
- node->kind = kind;
- return node;
- }
- AstNode *ast_bad_expr(AstFile *f, Token begin, Token end) {
- AstNode *result = make_ast_node(f, AstNode_BadExpr);
- result->BadExpr.begin = begin;
- result->BadExpr.end = end;
- return result;
- }
- AstNode *ast_tag_expr(AstFile *f, Token token, Token name, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_TagExpr);
- result->TagExpr.token = token;
- result->TagExpr.name = name;
- result->TagExpr.expr = expr;
- return result;
- }
- AstNode *ast_run_expr(AstFile *f, Token token, Token name, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_RunExpr);
- result->RunExpr.token = token;
- result->RunExpr.name = name;
- result->RunExpr.expr = expr;
- return result;
- }
- AstNode *ast_tag_stmt(AstFile *f, Token token, Token name, AstNode *stmt) {
- AstNode *result = make_ast_node(f, AstNode_TagStmt);
- result->TagStmt.token = token;
- result->TagStmt.name = name;
- result->TagStmt.stmt = stmt;
- return result;
- }
- AstNode *ast_unary_expr(AstFile *f, Token op, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_UnaryExpr);
- result->UnaryExpr.op = op;
- result->UnaryExpr.expr = expr;
- return result;
- }
- AstNode *ast_binary_expr(AstFile *f, Token op, AstNode *left, AstNode *right) {
- AstNode *result = make_ast_node(f, AstNode_BinaryExpr);
- if (left == NULL) {
- syntax_error(op, "No lhs expression for binary expression `%.*s`", LIT(op.string));
- left = ast_bad_expr(f, op, op);
- }
- if (right == NULL) {
- syntax_error(op, "No rhs expression for binary expression `%.*s`", LIT(op.string));
- right = ast_bad_expr(f, op, op);
- }
- result->BinaryExpr.op = op;
- result->BinaryExpr.left = left;
- result->BinaryExpr.right = right;
- return result;
- }
- AstNode *ast_paren_expr(AstFile *f, AstNode *expr, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_ParenExpr);
- result->ParenExpr.expr = expr;
- result->ParenExpr.open = open;
- result->ParenExpr.close = close;
- return result;
- }
- AstNode *ast_call_expr(AstFile *f, AstNode *proc, AstNodeArray args, Token open, Token close, Token ellipsis) {
- AstNode *result = make_ast_node(f, AstNode_CallExpr);
- result->CallExpr.proc = proc;
- result->CallExpr.args = args;
- result->CallExpr.open = open;
- result->CallExpr.close = close;
- result->CallExpr.ellipsis = ellipsis;
- return result;
- }
- AstNode *ast_macro_call_expr(AstFile *f, AstNode *macro, Token bang, AstNodeArray args, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_MacroCallExpr);
- result->MacroCallExpr.macro = macro;
- result->MacroCallExpr.bang = bang;
- result->MacroCallExpr.args = args;
- result->MacroCallExpr.open = open;
- result->MacroCallExpr.close = close;
- return result;
- }
- AstNode *ast_selector_expr(AstFile *f, Token token, AstNode *expr, AstNode *selector) {
- AstNode *result = make_ast_node(f, AstNode_SelectorExpr);
- result->SelectorExpr.expr = expr;
- result->SelectorExpr.selector = selector;
- return result;
- }
- AstNode *ast_index_expr(AstFile *f, AstNode *expr, AstNode *index, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_IndexExpr);
- result->IndexExpr.expr = expr;
- result->IndexExpr.index = index;
- result->IndexExpr.open = open;
- result->IndexExpr.close = close;
- return result;
- }
- AstNode *ast_slice_expr(AstFile *f, AstNode *expr, Token open, Token close, bool index3, AstNode *low, AstNode *high, AstNode *max) {
- AstNode *result = make_ast_node(f, AstNode_SliceExpr);
- result->SliceExpr.expr = expr;
- result->SliceExpr.open = open;
- result->SliceExpr.close = close;
- result->SliceExpr.index3 = index3;
- result->SliceExpr.low = low;
- result->SliceExpr.high = high;
- result->SliceExpr.max = max;
- return result;
- }
- AstNode *ast_deref_expr(AstFile *f, AstNode *expr, Token op) {
- AstNode *result = make_ast_node(f, AstNode_DerefExpr);
- result->DerefExpr.expr = expr;
- result->DerefExpr.op = op;
- return result;
- }
- AstNode *ast_interval_expr(AstFile *f, Token op, AstNode *left, AstNode *right) {
- AstNode *result = make_ast_node(f, AstNode_IntervalExpr);
- result->IntervalExpr.op = op;
- result->IntervalExpr.left = left;
- result->IntervalExpr.right = right;
- return result;
- }
- AstNode *ast_ident(AstFile *f, Token token) {
- AstNode *result = make_ast_node(f, AstNode_Ident);
- result->Ident = token;
- return result;
- }
- AstNode *ast_implicit(AstFile *f, Token token) {
- AstNode *result = make_ast_node(f, AstNode_Implicit);
- result->Implicit = token;
- return result;
- }
- AstNode *ast_basic_lit(AstFile *f, Token basic_lit) {
- AstNode *result = make_ast_node(f, AstNode_BasicLit);
- result->BasicLit = basic_lit;
- return result;
- }
- AstNode *ast_basic_directive(AstFile *f, Token token, String name) {
- AstNode *result = make_ast_node(f, AstNode_BasicDirective);
- result->BasicDirective.token = token;
- result->BasicDirective.name = name;
- return result;
- }
- AstNode *ast_ellipsis(AstFile *f, Token token, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_Ellipsis);
- result->Ellipsis.token = token;
- result->Ellipsis.expr = expr;
- return result;
- }
- AstNode *ast_proc_lit(AstFile *f, AstNode *type, AstNode *body, u64 tags, AstNode *foreign_library, String foreign_name, String link_name) {
- AstNode *result = make_ast_node(f, AstNode_ProcLit);
- result->ProcLit.type = type;
- result->ProcLit.body = body;
- result->ProcLit.tags = tags;
- result->ProcLit.foreign_library = foreign_library;
- result->ProcLit.foreign_name = foreign_name;
- result->ProcLit.link_name = link_name;
- return result;
- }
- AstNode *ast_field_value(AstFile *f, AstNode *field, AstNode *value, Token eq) {
- AstNode *result = make_ast_node(f, AstNode_FieldValue);
- result->FieldValue.field = field;
- result->FieldValue.value = value;
- result->FieldValue.eq = eq;
- return result;
- }
- AstNode *ast_cast_expr(AstFile *f, Token token, AstNode *type, AstNode *expr, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_CastExpr);
- result->CastExpr.token = token;
- result->CastExpr.type = type;
- result->CastExpr.expr = expr;
- result->CastExpr.open = open;
- result->CastExpr.close = close;
- return result;
- }
- AstNode *ast_compound_lit(AstFile *f, AstNode *type, AstNodeArray elems, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_CompoundLit);
- result->CompoundLit.type = type;
- result->CompoundLit.elems = elems;
- result->CompoundLit.open = open;
- result->CompoundLit.close = close;
- return result;
- }
- AstNode *ast_alias(AstFile *f, Token token, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_Alias);
- result->Alias.token = token;
- result->Alias.expr = expr;
- return result;
- }
- AstNode *ast_ternary_expr(AstFile *f, AstNode *cond, AstNode *x, AstNode *y) {
- AstNode *result = make_ast_node(f, AstNode_TernaryExpr);
- result->TernaryExpr.cond = cond;
- result->TernaryExpr.x = x;
- result->TernaryExpr.y = y;
- return result;
- }
- AstNode *ast_bad_stmt(AstFile *f, Token begin, Token end) {
- AstNode *result = make_ast_node(f, AstNode_BadStmt);
- result->BadStmt.begin = begin;
- result->BadStmt.end = end;
- return result;
- }
- AstNode *ast_empty_stmt(AstFile *f, Token token) {
- AstNode *result = make_ast_node(f, AstNode_EmptyStmt);
- result->EmptyStmt.token = token;
- return result;
- }
- AstNode *ast_expr_stmt(AstFile *f, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_ExprStmt);
- result->ExprStmt.expr = expr;
- return result;
- }
- AstNode *ast_assign_stmt(AstFile *f, Token op, AstNodeArray lhs, AstNodeArray rhs) {
- AstNode *result = make_ast_node(f, AstNode_AssignStmt);
- result->AssignStmt.op = op;
- result->AssignStmt.lhs = lhs;
- result->AssignStmt.rhs = rhs;
- return result;
- }
- AstNode *ast_inc_dec_stmt(AstFile *f, Token op, AstNode *expr) {
- AstNode *result = make_ast_node(f, AstNode_IncDecStmt);
- result->IncDecStmt.op = op;
- result->IncDecStmt.expr = expr;
- return result;
- }
- AstNode *ast_block_stmt(AstFile *f, AstNodeArray stmts, Token open, Token close) {
- AstNode *result = make_ast_node(f, AstNode_BlockStmt);
- result->BlockStmt.stmts = stmts;
- result->BlockStmt.open = open;
- result->BlockStmt.close = close;
- return result;
- }
- AstNode *ast_if_stmt(AstFile *f, Token token, AstNode *init, AstNode *cond, AstNode *body, AstNode *else_stmt) {
- AstNode *result = make_ast_node(f, AstNode_IfStmt);
- result->IfStmt.token = token;
- result->IfStmt.init = init;
- result->IfStmt.cond = cond;
- result->IfStmt.body = body;
- result->IfStmt.else_stmt = else_stmt;
- return result;
- }
- AstNode *ast_when_stmt(AstFile *f, Token token, AstNode *cond, AstNode *body, AstNode *else_stmt) {
- AstNode *result = make_ast_node(f, AstNode_WhenStmt);
- result->WhenStmt.token = token;
- result->WhenStmt.cond = cond;
- result->WhenStmt.body = body;
- result->WhenStmt.else_stmt = else_stmt;
- return result;
- }
- AstNode *ast_return_stmt(AstFile *f, Token token, AstNodeArray results) {
- AstNode *result = make_ast_node(f, AstNode_ReturnStmt);
- result->ReturnStmt.token = token;
- result->ReturnStmt.results = results;
- return result;
- }
- AstNode *ast_for_stmt(AstFile *f, Token token, AstNode *init, AstNode *cond, AstNode *post, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_ForStmt);
- result->ForStmt.token = token;
- result->ForStmt.init = init;
- result->ForStmt.cond = cond;
- result->ForStmt.post = post;
- result->ForStmt.body = body;
- return result;
- }
- AstNode *ast_range_stmt(AstFile *f, Token token, AstNode *value, AstNode *index, Token in_token, AstNode *expr, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_RangeStmt);
- result->RangeStmt.token = token;
- result->RangeStmt.value = value;
- result->RangeStmt.index = index;
- result->RangeStmt.in_token = in_token;
- result->RangeStmt.expr = expr;
- result->RangeStmt.body = body;
- return result;
- }
- AstNode *ast_match_stmt(AstFile *f, Token token, AstNode *init, AstNode *tag, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_MatchStmt);
- result->MatchStmt.token = token;
- result->MatchStmt.init = init;
- result->MatchStmt.tag = tag;
- result->MatchStmt.body = body;
- return result;
- }
- AstNode *ast_type_match_stmt(AstFile *f, Token token, AstNode *tag, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_TypeMatchStmt);
- result->TypeMatchStmt.token = token;
- result->TypeMatchStmt.tag = tag;
- result->TypeMatchStmt.body = body;
- return result;
- }
- AstNode *ast_case_clause(AstFile *f, Token token, AstNodeArray list, AstNodeArray stmts) {
- AstNode *result = make_ast_node(f, AstNode_CaseClause);
- result->CaseClause.token = token;
- result->CaseClause.list = list;
- result->CaseClause.stmts = stmts;
- return result;
- }
- AstNode *ast_defer_stmt(AstFile *f, Token token, AstNode *stmt) {
- AstNode *result = make_ast_node(f, AstNode_DeferStmt);
- result->DeferStmt.token = token;
- result->DeferStmt.stmt = stmt;
- return result;
- }
- AstNode *ast_branch_stmt(AstFile *f, Token token, AstNode *label) {
- AstNode *result = make_ast_node(f, AstNode_BranchStmt);
- result->BranchStmt.token = token;
- result->BranchStmt.label = label;
- return result;
- }
- AstNode *ast_using_stmt(AstFile *f, Token token, AstNodeArray list) {
- AstNode *result = make_ast_node(f, AstNode_UsingStmt);
- result->UsingStmt.token = token;
- result->UsingStmt.list = list;
- return result;
- }
- AstNode *ast_asm_operand(AstFile *f, Token string, AstNode *operand) {
- AstNode *result = make_ast_node(f, AstNode_AsmOperand);
- result->AsmOperand.string = string;
- result->AsmOperand.operand = operand;
- return result;
- }
- AstNode *ast_asm_stmt(AstFile *f, Token token, bool is_volatile, Token open, Token close, Token code_string,
- AstNode *output_list, AstNode *input_list, AstNode *clobber_list,
- isize output_count, isize input_count, isize clobber_count) {
- AstNode *result = make_ast_node(f, AstNode_AsmStmt);
- result->AsmStmt.token = token;
- result->AsmStmt.is_volatile = is_volatile;
- result->AsmStmt.open = open;
- result->AsmStmt.close = close;
- result->AsmStmt.code_string = code_string;
- result->AsmStmt.output_list = output_list;
- result->AsmStmt.input_list = input_list;
- result->AsmStmt.clobber_list = clobber_list;
- result->AsmStmt.output_count = output_count;
- result->AsmStmt.input_count = input_count;
- result->AsmStmt.clobber_count = clobber_count;
- return result;
- }
- AstNode *ast_push_allocator(AstFile *f, Token token, AstNode *expr, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_PushAllocator);
- result->PushAllocator.token = token;
- result->PushAllocator.expr = expr;
- result->PushAllocator.body = body;
- return result;
- }
- AstNode *ast_push_context(AstFile *f, Token token, AstNode *expr, AstNode *body) {
- AstNode *result = make_ast_node(f, AstNode_PushContext);
- result->PushContext.token = token;
- result->PushContext.expr = expr;
- result->PushContext.body = body;
- return result;
- }
- AstNode *ast_bad_decl(AstFile *f, Token begin, Token end) {
- AstNode *result = make_ast_node(f, AstNode_BadDecl);
- result->BadDecl.begin = begin;
- result->BadDecl.end = end;
- return result;
- }
- AstNode *ast_field(AstFile *f, AstNodeArray names, AstNode *type, u32 flags) {
- AstNode *result = make_ast_node(f, AstNode_Field);
- result->Field.names = names;
- result->Field.type = type;
- result->Field.flags = flags;
- return result;
- }
- AstNode *ast_field_list(AstFile *f, Token token, AstNodeArray list) {
- AstNode *result = make_ast_node(f, AstNode_FieldList);
- result->FieldList.token = token;
- result->FieldList.list = list;
- return result;
- }
- AstNode *ast_union_field(AstFile *f, AstNode *name, AstNode *list) {
- AstNode *result = make_ast_node(f, AstNode_UnionField);
- result->UnionField.name = name;
- result->UnionField.list = list;
- return result;
- }
- AstNode *ast_helper_type(AstFile *f, Token token, AstNode *type) {
- AstNode *result = make_ast_node(f, AstNode_HelperType);
- result->HelperType.token = token;
- result->HelperType.type = type;
- return result;
- }
- AstNode *ast_proc_type(AstFile *f, Token token, AstNode *params, AstNode *results, u64 tags, ProcCallingConvention calling_convention) {
- AstNode *result = make_ast_node(f, AstNode_ProcType);
- result->ProcType.token = token;
- result->ProcType.params = params;
- result->ProcType.results = results;
- result->ProcType.tags = tags;
- result->ProcType.calling_convention = calling_convention;
- return result;
- }
- AstNode *ast_pointer_type(AstFile *f, Token token, AstNode *type) {
- AstNode *result = make_ast_node(f, AstNode_PointerType);
- result->PointerType.token = token;
- result->PointerType.type = type;
- return result;
- }
- AstNode *ast_array_type(AstFile *f, Token token, AstNode *count, AstNode *elem) {
- AstNode *result = make_ast_node(f, AstNode_ArrayType);
- result->ArrayType.token = token;
- result->ArrayType.count = count;
- result->ArrayType.elem = elem;
- return result;
- }
- AstNode *ast_dynamic_array_type(AstFile *f, Token token, AstNode *elem) {
- AstNode *result = make_ast_node(f, AstNode_DynamicArrayType);
- result->DynamicArrayType.token = token;
- result->DynamicArrayType.elem = elem;
- return result;
- }
- AstNode *ast_vector_type(AstFile *f, Token token, AstNode *count, AstNode *elem) {
- AstNode *result = make_ast_node(f, AstNode_VectorType);
- result->VectorType.token = token;
- result->VectorType.count = count;
- result->VectorType.elem = elem;
- return result;
- }
- AstNode *ast_struct_type(AstFile *f, Token token, AstNodeArray fields, isize field_count,
- bool is_packed, bool is_ordered, AstNode *align) {
- AstNode *result = make_ast_node(f, AstNode_StructType);
- result->StructType.token = token;
- result->StructType.fields = fields;
- result->StructType.field_count = field_count;
- result->StructType.is_packed = is_packed;
- result->StructType.is_ordered = is_ordered;
- result->StructType.align = align;
- return result;
- }
- AstNode *ast_union_type(AstFile *f, Token token, AstNodeArray fields, isize field_count, AstNodeArray variants) {
- AstNode *result = make_ast_node(f, AstNode_UnionType);
- result->UnionType.token = token;
- result->UnionType.fields = fields;
- result->UnionType.field_count = field_count;
- result->UnionType.variants = variants;
- return result;
- }
- AstNode *ast_raw_union_type(AstFile *f, Token token, AstNodeArray fields, isize field_count) {
- AstNode *result = make_ast_node(f, AstNode_RawUnionType);
- result->RawUnionType.token = token;
- result->RawUnionType.fields = fields;
- result->RawUnionType.field_count = field_count;
- return result;
- }
- AstNode *ast_enum_type(AstFile *f, Token token, AstNode *base_type, AstNodeArray fields) {
- AstNode *result = make_ast_node(f, AstNode_EnumType);
- result->EnumType.token = token;
- result->EnumType.base_type = base_type;
- result->EnumType.fields = fields;
- return result;
- }
- AstNode *ast_map_type(AstFile *f, Token token, AstNode *count, AstNode *key, AstNode *value) {
- AstNode *result = make_ast_node(f, AstNode_MapType);
- result->MapType.token = token;
- result->MapType.count = count;
- result->MapType.key = key;
- result->MapType.value = value;
- return result;
- }
- AstNode *ast_value_decl(AstFile *f, bool is_var, AstNodeArray names, AstNode *type, AstNodeArray values) {
- AstNode *result = make_ast_node(f, AstNode_ValueDecl);
- result->ValueDecl.is_var = is_var;
- result->ValueDecl.names = names;
- result->ValueDecl.type = type;
- result->ValueDecl.values = values;
- return result;
- }
- AstNode *ast_import_decl(AstFile *f, Token token, bool is_import, Token relpath, Token import_name, AstNode *cond) {
- AstNode *result = make_ast_node(f, AstNode_ImportDecl);
- result->ImportDecl.token = token;
- result->ImportDecl.is_import = is_import;
- result->ImportDecl.relpath = relpath;
- result->ImportDecl.import_name = import_name;
- result->ImportDecl.cond = cond;
- return result;
- }
- AstNode *ast_foreign_library(AstFile *f, Token token, Token filepath, Token library_name, AstNode *cond, bool is_system) {
- AstNode *result = make_ast_node(f, AstNode_ForeignLibrary);
- result->ForeignLibrary.token = token;
- result->ForeignLibrary.filepath = filepath;
- result->ForeignLibrary.library_name = library_name;
- result->ForeignLibrary.cond = cond;
- result->ForeignLibrary.is_system = is_system;
- return result;
- }
- AstNode *ast_label_decl(AstFile *f, Token token, AstNode *name) {
- AstNode *result = make_ast_node(f, AstNode_Label);
- result->Label.token = token;
- result->Label.name = name;
- return result;
- }
- bool next_token(AstFile *f) {
- Token prev = f->curr_token;
- if (f->curr_token_index+1 < f->tokens.count) {
- if (f->curr_token.kind != Token_Comment) {
- f->prev_token = f->curr_token;
- }
- f->curr_token_index++;
- f->curr_token = f->tokens.e[f->curr_token_index];
- if (f->curr_token.kind == Token_Comment) {
- return next_token(f);
- }
- return true;
- }
- syntax_error(f->curr_token, "Token is EOF");
- return false;
- }
- TokenKind look_ahead_token_kind(AstFile *f, isize amount) {
- GB_ASSERT(amount > 0);
- TokenKind kind = Token_Invalid;
- isize index = f->curr_token_index;
- while (amount > 0) {
- index++;
- kind = f->tokens.e[index].kind;
- if (kind != Token_Comment) {
- amount--;
- }
- }
- return kind;
- }
- Token expect_token(AstFile *f, TokenKind kind) {
- Token prev = f->curr_token;
- if (prev.kind != kind) {
- String p = token_strings[prev.kind];
- syntax_error(f->curr_token, "Expected `%.*s`, got `%.*s`",
- LIT(token_strings[kind]),
- LIT(token_strings[prev.kind]));
- }
- next_token(f);
- return prev;
- }
- Token expect_token_after(AstFile *f, TokenKind kind, char *msg) {
- Token prev = f->curr_token;
- if (prev.kind != kind) {
- String p = token_strings[prev.kind];
- syntax_error(f->curr_token, "Expected `%.*s` after %s, got `%.*s`",
- LIT(token_strings[kind]),
- msg,
- LIT(p));
- }
- next_token(f);
- return prev;
- }
- Token expect_operator(AstFile *f) {
- Token prev = f->curr_token;
- if (!gb_is_between(prev.kind, Token__OperatorBegin+1, Token__OperatorEnd-1)) {
- syntax_error(f->curr_token, "Expected an operator, got `%.*s`",
- LIT(token_strings[prev.kind]));
- }
- next_token(f);
- return prev;
- }
- Token expect_keyword(AstFile *f) {
- Token prev = f->curr_token;
- if (!gb_is_between(prev.kind, Token__KeywordBegin+1, Token__KeywordEnd-1)) {
- syntax_error(f->curr_token, "Expected a keyword, got `%.*s`",
- LIT(token_strings[prev.kind]));
- }
- next_token(f);
- return prev;
- }
- bool allow_token(AstFile *f, TokenKind kind) {
- Token prev = f->curr_token;
- if (prev.kind == kind) {
- next_token(f);
- return true;
- }
- return false;
- }
- bool is_blank_ident(String str) {
- if (str.len == 1) {
- return str.text[0] == '_';
- }
- return false;
- }
- // NOTE(bill): Go to next statement to prevent numerous error messages popping up
- void fix_advance_to_next_stmt(AstFile *f) {
- // TODO(bill): fix_advance_to_next_stmt
- #if 1
- for (;;) {
- Token t = f->curr_token;
- switch (t.kind) {
- case Token_EOF:
- case Token_Semicolon:
- return;
- case Token_if:
- case Token_when:
- case Token_return:
- case Token_match:
- case Token_defer:
- case Token_asm:
- case Token_using:
- // case Token_thread_local:
- case Token_no_alias:
- // case Token_immutable:
- case Token_break:
- case Token_continue:
- case Token_fallthrough:
- case Token_push_allocator:
- case Token_push_context:
- case Token_Hash:
- {
- if (token_pos_eq(t.pos, f->fix_prev_pos) &&
- f->fix_count < PARSER_MAX_FIX_COUNT) {
- f->fix_count++;
- return;
- }
- if (token_pos_cmp(f->fix_prev_pos, t.pos) < 0) {
- f->fix_prev_pos = t.pos;
- f->fix_count = 0; // NOTE(bill): Reset
- return;
- }
- // NOTE(bill): Reaching here means there is a parsing bug
- } break;
- }
- next_token(f);
- }
- #endif
- }
- Token expect_closing(AstFile *f, TokenKind kind, String context) {
- if (f->curr_token.kind != kind &&
- f->curr_token.kind == Token_Semicolon &&
- str_eq(f->curr_token.string, str_lit("\n"))) {
- error(f->curr_token, "Missing `,` before newline in %.*s", LIT(context));
- next_token(f);
- }
- return expect_token(f, kind);
- }
- bool is_semicolon_optional_for_node(AstFile *f, AstNode *s) {
- if (s == NULL) {
- return false;
- }
- switch (s->kind) {
- case AstNode_HelperType:
- return is_semicolon_optional_for_node(f, s->HelperType.type);
- case AstNode_PointerType:
- return is_semicolon_optional_for_node(f, s->PointerType.type);
- case AstNode_StructType:
- case AstNode_UnionType:
- case AstNode_RawUnionType:
- case AstNode_EnumType:
- return true;
- case AstNode_ProcLit:
- return s->ProcLit.body != NULL;
- case AstNode_ValueDecl:
- if (!s->ValueDecl.is_var) {
- if (s->ValueDecl.values.count > 0) {
- AstNode *last = s->ValueDecl.values.e[s->ValueDecl.values.count-1];
- return is_semicolon_optional_for_node(f, last);
- }
- }
- break;
- }
- return false;
- }
- void expect_semicolon(AstFile *f, AstNode *s) {
- if (allow_token(f, Token_Semicolon)) {
- return;
- }
- Token prev_token = f->prev_token;
- switch (f->curr_token.kind) {
- case Token_EOF:
- return;
- }
- if (s != NULL) {
- if (prev_token.pos.line != f->curr_token.pos.line) {
- if (is_semicolon_optional_for_node(f, s)) {
- return;
- }
- } else {
- // switch (s->kind) {
- // case AstNode_GiveExpr:
- // if (f->curr_token.kind == Token_CloseBrace) {
- // return;
- // }
- // break;
- // }
- }
- syntax_error(prev_token, "Expected `;` after %.*s, got %.*s",
- LIT(ast_node_strings[s->kind]), LIT(token_strings[prev_token.kind]));
- } else {
- syntax_error(prev_token, "Expected `;`");
- }
- fix_advance_to_next_stmt(f);
- }
- AstNode * parse_expr(AstFile *f, bool lhs);
- AstNode * parse_proc_type(AstFile *f, AstNode **foreign_library, String *foreign_name, String *link_name);
- AstNodeArray parse_stmt_list(AstFile *f);
- AstNode * parse_stmt(AstFile *f);
- AstNode * parse_body(AstFile *f);
- AstNode *parse_ident(AstFile *f) {
- Token token = f->curr_token;
- if (token.kind == Token_Ident) {
- next_token(f);
- } else {
- token.string = str_lit("_");
- expect_token(f, Token_Ident);
- }
- return ast_ident(f, token);
- }
- AstNode *parse_tag_expr(AstFile *f, AstNode *expression) {
- Token token = expect_token(f, Token_Hash);
- Token name = expect_token(f, Token_Ident);
- return ast_tag_expr(f, token, name, expression);
- }
- AstNode *unparen_expr(AstNode *node) {
- for (;;) {
- if (node == NULL) {
- return NULL;
- }
- if (node->kind != AstNode_ParenExpr) {
- return node;
- }
- node = node->ParenExpr.expr;
- }
- }
- AstNode *parse_value(AstFile *f);
- AstNodeArray parse_element_list(AstFile *f) {
- AstNodeArray elems = make_ast_node_array(f);
- while (f->curr_token.kind != Token_CloseBrace &&
- f->curr_token.kind != Token_EOF) {
- AstNode *elem = parse_value(f);
- if (f->curr_token.kind == Token_Eq) {
- Token eq = expect_token(f, Token_Eq);
- AstNode *value = parse_value(f);
- elem = ast_field_value(f, elem, value, eq);
- }
- array_add(&elems, elem);
- if (!allow_token(f, Token_Comma)) {
- break;
- }
- }
- return elems;
- }
- AstNode *parse_literal_value(AstFile *f, AstNode *type) {
- AstNodeArray elems = {0};
- Token open = expect_token(f, Token_OpenBrace);
- f->expr_level++;
- if (f->curr_token.kind != Token_CloseBrace) {
- elems = parse_element_list(f);
- }
- f->expr_level--;
- Token close = expect_closing(f, Token_CloseBrace, str_lit("compound literal"));
- return ast_compound_lit(f, type, elems, open, close);
- }
- AstNode *parse_value(AstFile *f) {
- if (f->curr_token.kind == Token_OpenBrace) {
- return parse_literal_value(f, NULL);
- }
- AstNode *value = parse_expr(f, false);
- return value;
- }
- AstNode *parse_type_or_ident(AstFile *f);
- void check_proc_add_tag(AstFile *f, AstNode *tag_expr, u64 *tags, ProcTag tag, String tag_name) {
- if (*tags & tag) {
- syntax_error_node(tag_expr, "Procedure tag already used: %.*s", LIT(tag_name));
- }
- *tags |= tag;
- }
- bool is_foreign_name_valid(String name) {
- // TODO(bill): is_foreign_name_valid
- if (name.len == 0)
- return false;
- isize offset = 0;
- while (offset < name.len) {
- Rune rune;
- isize remaining = name.len - offset;
- isize width = gb_utf8_decode(name.text+offset, remaining, &rune);
- if (rune == GB_RUNE_INVALID && width == 1) {
- return false;
- } else if (rune == GB_RUNE_BOM && remaining > 0) {
- return false;
- }
- if (offset == 0) {
- switch (rune) {
- case '-':
- case '$':
- case '.':
- case '_':
- break;
- default:
- if (!gb_char_is_alpha(cast(char)rune))
- return false;
- break;
- }
- } else {
- switch (rune) {
- case '-':
- case '$':
- case '.':
- case '_':
- break;
- default:
- if (!gb_char_is_alphanumeric(cast(char)rune)) {
- return false;
- }
- break;
- }
- }
- offset += width;
- }
- return true;
- }
- void parse_proc_tags(AstFile *f, u64 *tags, AstNode **foreign_library_token, String *foreign_name, String *link_name, ProcCallingConvention *calling_convention) {
- // TODO(bill): Add this to procedure literals too
- GB_ASSERT(tags != NULL);
- GB_ASSERT(link_name != NULL);
- GB_ASSERT(link_name != NULL);
- ProcCallingConvention cc = ProcCC_Invalid;
- while (f->curr_token.kind == Token_Hash) {
- AstNode *tag_expr = parse_tag_expr(f, NULL);
- ast_node(te, TagExpr, tag_expr);
- String tag_name = te->name.string;
- #define ELSE_IF_ADD_TAG(name) \
- else if (str_eq(tag_name, str_lit(#name))) { \
- check_proc_add_tag(f, tag_expr, tags, ProcTag_##name, tag_name); \
- }
- if (str_eq(tag_name, str_lit("foreign"))) {
- check_proc_add_tag(f, tag_expr, tags, ProcTag_foreign, tag_name);
- *foreign_library_token = parse_ident(f);
- if (f->curr_token.kind == Token_String) {
- *foreign_name = f->curr_token.string;
- // TODO(bill): Check if valid string
- if (!is_foreign_name_valid(*foreign_name)) {
- syntax_error_node(tag_expr, "Invalid alternative foreign procedure name: `%.*s`", LIT(*foreign_name));
- }
- next_token(f);
- }
- } else if (str_eq(tag_name, str_lit("link_name"))) {
- check_proc_add_tag(f, tag_expr, tags, ProcTag_link_name, tag_name);
- if (f->curr_token.kind == Token_String) {
- *link_name = f->curr_token.string;
- // TODO(bill): Check if valid string
- if (!is_foreign_name_valid(*link_name)) {
- syntax_error_node(tag_expr, "Invalid alternative link procedure name `%.*s`", LIT(*link_name));
- }
- next_token(f);
- } else {
- expect_token(f, Token_String);
- }
- }
- ELSE_IF_ADD_TAG(export)
- ELSE_IF_ADD_TAG(bounds_check)
- ELSE_IF_ADD_TAG(no_bounds_check)
- ELSE_IF_ADD_TAG(inline)
- ELSE_IF_ADD_TAG(no_inline)
- // ELSE_IF_ADD_TAG(dll_import)
- // ELSE_IF_ADD_TAG(dll_export)
- else if (str_eq(tag_name, str_lit("cc_odin"))) {
- if (cc == ProcCC_Invalid) {
- cc = ProcCC_Odin;
- } else {
- syntax_error_node(tag_expr, "Multiple calling conventions for procedure type");
- }
- } else if (str_eq(tag_name, str_lit("cc_c"))) {
- if (cc == ProcCC_Invalid) {
- cc = ProcCC_C;
- } else {
- syntax_error_node(tag_expr, "Multiple calling conventions for procedure type");
- }
- } else if (str_eq(tag_name, str_lit("cc_std"))) {
- if (cc == ProcCC_Invalid) {
- cc = ProcCC_Std;
- } else {
- syntax_error_node(tag_expr, "Multiple calling conventions for procedure type");
- }
- } else if (str_eq(tag_name, str_lit("cc_fast"))) {
- if (cc == ProcCC_Invalid) {
- cc = ProcCC_Fast;
- } else {
- syntax_error_node(tag_expr, "Multiple calling conventions for procedure type");
- }
- } else {
- syntax_error_node(tag_expr, "Unknown procedure tag #%.*s\n", LIT(tag_name));
- }
- #undef ELSE_IF_ADD_TAG
- }
- if (cc == ProcCC_Invalid) {
- if ((*tags) & ProcTag_foreign) {
- cc = ProcCC_C;
- } else {
- cc = ProcCC_Odin;
- }
- }
- if (calling_convention) {
- *calling_convention = cc;
- }
- if ((*tags & ProcTag_foreign) && (*tags & ProcTag_export)) {
- syntax_error(f->curr_token, "You cannot apply both #foreign and #export to a procedure");
- }
- if ((*tags & ProcTag_inline) && (*tags & ProcTag_no_inline)) {
- syntax_error(f->curr_token, "You cannot apply both #inline and #no_inline to a procedure");
- }
- if ((*tags & ProcTag_bounds_check) && (*tags & ProcTag_no_bounds_check)) {
- syntax_error(f->curr_token, "You cannot apply both #bounds_check and #no_bounds_check to a procedure");
- }
- if (((*tags & ProcTag_bounds_check) || (*tags & ProcTag_no_bounds_check)) && (*tags & ProcTag_foreign)) {
- syntax_error(f->curr_token, "You cannot apply both #bounds_check or #no_bounds_check to a procedure without a body");
- }
- }
- AstNodeArray parse_lhs_expr_list(AstFile *f);
- AstNodeArray parse_rhs_expr_list(AstFile *f);
- AstNode * parse_simple_stmt (AstFile *f, bool in_stmt_ok);
- AstNode * parse_type (AstFile *f);
- AstNode *convert_stmt_to_expr(AstFile *f, AstNode *statement, String kind) {
- if (statement == NULL) {
- return NULL;
- }
- if (statement->kind == AstNode_ExprStmt) {
- return statement->ExprStmt.expr;
- }
- syntax_error(f->curr_token, "Expected `%.*s`, found a simple statement.", LIT(kind));
- return ast_bad_expr(f, f->curr_token, f->tokens.e[f->curr_token_index+1]);
- }
- // AstNode *parse_block_expr(AstFile *f) {
- // AstNodeArray stmts = {0};
- // Token open, close;
- // open = expect_token(f, Token_OpenBrace);
- // f->expr_level++;
- // stmts = parse_stmt_list(f);
- // f->expr_level--;
- // close = expect_token(f, Token_CloseBrace);
- // return ast_block_expr(f, stmts, open, close);
- // }
- // AstNode *parse_if_expr(AstFile *f) {
- // if (f->curr_proc == NULL) {
- // syntax_error(f->curr_token, "You cannot use an if expression in the file scope");
- // return ast_bad_stmt(f, f->curr_token, f->curr_token);
- // }
- // Token token = expect_token(f, Token_if);
- // AstNode *init = NULL;
- // AstNode *cond = NULL;
- // AstNode *body = NULL;
- // AstNode *else_expr = NULL;
- // isize prev_level = f->expr_level;
- // f->expr_level = -1;
- // if (allow_token(f, Token_Semicolon)) {
- // cond = parse_expr(f, false);
- // } else {
- // init = parse_simple_stmt(f, false);
- // if (allow_token(f, Token_Semicolon)) {
- // cond = parse_expr(f, false);
- // } else {
- // cond = convert_stmt_to_expr(f, init, str_lit("boolean expression"));
- // init = NULL;
- // }
- // }
- // f->expr_level = prev_level;
- // if (cond == NULL) {
- // syntax_error(f->curr_token, "Expected condition for if statement");
- // }
- // body = parse_block_expr(f);
- // if (allow_token(f, Token_else)) {
- // switch (f->curr_token.kind) {
- // case Token_if:
- // else_expr = parse_if_expr(f);
- // break;
- // case Token_OpenBrace:
- // else_expr = parse_block_expr(f);
- // break;
- // default:
- // syntax_error(f->curr_token, "Expected if expression block statement");
- // else_expr = ast_bad_expr(f, f->curr_token, f->tokens.e[f->curr_token_index+1]);
- // break;
- // }
- // } else {
- // syntax_error(f->curr_token, "An if expression must have an else clause");
- // return ast_bad_stmt(f, f->curr_token, f->tokens.e[f->curr_token_index+1]);
- // }
- // return ast_if_expr(f, token, init, cond, body, else_expr);
- // }
- AstNode *parse_operand(AstFile *f, bool lhs) {
- AstNode *operand = NULL; // Operand
- switch (f->curr_token.kind) {
- case Token_Ident:
- operand = parse_ident(f);
- if (!lhs) {
- // TODO(bill): Handle?
- }
- return operand;
- case Token_context:
- return ast_implicit(f, expect_token(f, Token_context));
- case Token_Integer:
- case Token_Float:
- case Token_Imag:
- case Token_Rune:
- operand = ast_basic_lit(f, f->curr_token);
- next_token(f);
- return operand;
- case Token_String: {
- Token token = f->curr_token;
- next_token(f);
- if (f->curr_token.kind == Token_String) {
- // NOTE(bill): Allow neighbouring string literals to be merge together to
- // become one big string
- String s = f->curr_token.string;
- Array(u8) data;
- array_init_reserve(&data, heap_allocator(), token.string.len+s.len);
- gb_memmove(data.e, token.string.text, token.string.len);
- data.count += token.string.len;
- while (f->curr_token.kind == Token_String) {
- String s = f->curr_token.string;
- isize old_count = data.count;
- array_resize(&data, data.count + s.len);
- gb_memmove(data.e+old_count, s.text, s.len);
- next_token(f);
- }
- token.string = make_string(data.e, data.count);
- array_add(&f->tokenizer.allocated_strings, token.string);
- }
- return ast_basic_lit(f, token);
- }
- case Token_OpenParen: {
- Token open, close;
- // NOTE(bill): Skip the Paren Expression
- open = expect_token(f, Token_OpenParen);
- f->expr_level++;
- operand = parse_expr(f, false);
- f->expr_level--;
- close = expect_token(f, Token_CloseParen);
- return ast_paren_expr(f, operand, open, close);
- }
- case Token_Hash: {
- Token token = expect_token(f, Token_Hash);
- Token name = expect_token(f, Token_Ident);
- if (str_eq(name.string, str_lit("run"))) {
- AstNode *expr = parse_expr(f, false);
- operand = ast_run_expr(f, token, name, expr);
- if (unparen_expr(expr)->kind != AstNode_CallExpr) {
- error_node(expr, "#run can only be applied to procedure calls");
- operand = ast_bad_expr(f, token, f->curr_token);
- }
- warning(token, "#run is not yet implemented");
- } else if (str_eq(name.string, str_lit("file"))) { return ast_basic_directive(f, token, name.string);
- } else if (str_eq(name.string, str_lit("line"))) { return ast_basic_directive(f, token, name.string);
- } else if (str_eq(name.string, str_lit("procedure"))) { return ast_basic_directive(f, token, name.string);
- } else if (str_eq(name.string, str_lit("type"))) { return ast_helper_type(f, token, parse_type(f));
- } else if (!lhs && str_eq(name.string, str_lit("alias"))) { return ast_alias(f, token, parse_expr(f, false));
- } else {
- operand = ast_tag_expr(f, token, name, parse_expr(f, false));
- }
- return operand;
- }
- // Parse Procedure Type or Literal
- case Token_proc: {
- Token token = f->curr_token;
- AstNode *foreign_library = NULL;
- String foreign_name = {0};
- String link_name = {0};
- AstNode *type = parse_proc_type(f, &foreign_library, &foreign_name, &link_name);
- u64 tags = type->ProcType.tags;
- if (f->curr_token.kind == Token_OpenBrace) {
- if ((tags & ProcTag_foreign) != 0) {
- syntax_error(token, "A procedure tagged as `#foreign` cannot have a body");
- }
- AstNode *curr_proc = f->curr_proc;
- AstNode *body = NULL;
- f->curr_proc = type;
- body = parse_body(f);
- f->curr_proc = curr_proc;
- return ast_proc_lit(f, type, body, tags, foreign_library, foreign_name, link_name);
- }
- if ((tags & ProcTag_foreign) != 0) {
- return ast_proc_lit(f, type, NULL, tags, foreign_library, foreign_name, link_name);
- }
- if (tags != 0) {
- syntax_error(token, "A procedure type cannot have tags");
- }
- return type;
- }
- // case Token_if:
- // if (!lhs && f->expr_level >= 0) {
- // return parse_if_expr(f);
- // }
- // break;
- // case Token_OpenBrace:
- // if (!lhs && f->expr_level >= 0) {
- // return parse_block_expr(f);
- // }
- // break;
- default: {
- AstNode *type = parse_type_or_ident(f);
- if (type != NULL) {
- // TODO(bill): Is this correct???
- // NOTE(bill): Sanity check as identifiers should be handled already
- TokenPos pos = ast_node_token(type).pos;
- GB_ASSERT_MSG(type->kind != AstNode_Ident, "Type cannot be identifier %.*s(%td:%td)", LIT(pos.file), pos.line, pos.column);
- return type;
- }
- break;
- }
- }
- Token begin = f->curr_token;
- syntax_error(begin, "Expected an operand");
- fix_advance_to_next_stmt(f);
- return ast_bad_expr(f, begin, f->curr_token);
- }
- bool is_literal_type(AstNode *node) {
- node = unparen_expr(node);
- switch (node->kind) {
- case AstNode_BadExpr:
- case AstNode_Ident:
- case AstNode_SelectorExpr:
- case AstNode_ArrayType:
- case AstNode_VectorType:
- case AstNode_StructType:
- case AstNode_DynamicArrayType:
- case AstNode_MapType:
- return true;
- }
- return false;
- }
- AstNode *parse_call_expr(AstFile *f, AstNode *operand) {
- AstNodeArray args = make_ast_node_array(f);
- Token open_paren, close_paren;
- Token ellipsis = {0};
- f->expr_level++;
- open_paren = expect_token(f, Token_OpenParen);
- while (f->curr_token.kind != Token_CloseParen &&
- f->curr_token.kind != Token_EOF &&
- ellipsis.pos.line == 0) {
- if (f->curr_token.kind == Token_Comma) {
- syntax_error(f->curr_token, "Expected an expression not a ,");
- }
- if (f->curr_token.kind == Token_Ellipsis) {
- ellipsis = f->curr_token;
- next_token(f);
- }
- AstNode *arg = parse_expr(f, false);
- array_add(&args, arg);
- if (!allow_token(f, Token_Comma)) {
- break;
- }
- }
- f->expr_level--;
- close_paren = expect_closing(f, Token_CloseParen, str_lit("argument list"));
- return ast_call_expr(f, operand, args, open_paren, close_paren, ellipsis);
- }
- AstNode *parse_macro_call_expr(AstFile *f, AstNode *operand) {
- AstNodeArray args = make_ast_node_array(f);
- Token bang, open_paren, close_paren;
- bang = expect_token(f, Token_Not);
- f->expr_level++;
- open_paren = expect_token(f, Token_OpenParen);
- while (f->curr_token.kind != Token_CloseParen &&
- f->curr_token.kind != Token_EOF) {
- if (f->curr_token.kind == Token_Comma) {
- syntax_error(f->curr_token, "Expected an expression not a ,");
- }
- AstNode *arg = parse_expr(f, false);
- array_add(&args, arg);
- if (!allow_token(f, Token_Comma)) {
- break;
- }
- }
- f->expr_level--;
- close_paren = expect_closing(f, Token_CloseParen, str_lit("argument list"));
- return ast_macro_call_expr(f, operand, bang, args, open_paren, close_paren);
- }
- AstNode *parse_atom_expr(AstFile *f, bool lhs) {
- AstNode *operand = parse_operand(f, lhs);
- bool loop = true;
- while (loop) {
- switch (f->curr_token.kind) {
- case Token_OpenParen:
- operand = parse_call_expr(f, operand);
- break;
- case Token_Not:
- operand = parse_macro_call_expr(f, operand);
- break;
- case Token_Period: {
- Token token = f->curr_token;
- next_token(f);
- switch (f->curr_token.kind) {
- case Token_Ident:
- operand = ast_selector_expr(f, token, operand, parse_ident(f));
- break;
- // case Token_Integer:
- // operand = ast_selector_expr(f, token, operand, parse_expr(f, lhs));
- // break;
- default:
- syntax_error(f->curr_token, "Expected a selector");
- next_token(f);
- operand = ast_bad_expr(f, ast_node_token(operand), f->curr_token);
- // operand = ast_selector_expr(f, f->curr_token, operand, NULL);
- break;
- }
- } break;
- case Token_OpenBracket: {
- if (lhs) {
- // TODO(bill): Handle this
- }
- Token open = {0}, close = {0}, interval = {0};
- AstNode *indices[3] = {0};
- isize ellipsis_count = 0;
- Token ellipses[2] = {0};
- f->expr_level++;
- open = expect_token(f, Token_OpenBracket);
- if (f->curr_token.kind != Token_Ellipsis) {
- indices[0] = parse_expr(f, false);
- }
- bool is_index = true;
- while (f->curr_token.kind == Token_Ellipsis && ellipsis_count < gb_count_of(ellipses)) {
- ellipses[ellipsis_count++] = f->curr_token;
- next_token(f);
- if (f->curr_token.kind != Token_Ellipsis &&
- f->curr_token.kind != Token_CloseBracket &&
- f->curr_token.kind != Token_EOF) {
- indices[ellipsis_count] = parse_expr(f, false);
- }
- }
- f->expr_level--;
- close = expect_token(f, Token_CloseBracket);
- if (ellipsis_count > 0) {
- bool index3 = false;
- if (ellipsis_count == 2) {
- index3 = true;
- // 2nd and 3rd index must be present
- if (indices[1] == NULL) {
- error(ellipses[0], "2nd index required in 3-index slice expression");
- indices[1] = ast_bad_expr(f, ellipses[0], ellipses[1]);
- }
- if (indices[2] == NULL) {
- error(ellipses[1], "3rd index required in 3-index slice expression");
- indices[2] = ast_bad_expr(f, ellipses[1], close);
- }
- }
- operand = ast_slice_expr(f, operand, open, close, index3, indices[0], indices[1], indices[2]);
- } else {
- operand = ast_index_expr(f, operand, indices[0], open, close);
- }
- } break;
- case Token_Pointer: // Deference
- operand = ast_deref_expr(f, operand, expect_token(f, Token_Pointer));
- break;
- case Token_OpenBrace:
- if (!lhs && is_literal_type(operand) && f->expr_level >= 0) {
- operand = parse_literal_value(f, operand);
- } else {
- loop = false;
- }
- break;
- default:
- loop = false;
- break;
- }
- lhs = false; // NOTE(bill): 'tis not lhs anymore
- }
- return operand;
- }
- AstNode *parse_unary_expr(AstFile *f, bool lhs) {
- switch (f->curr_token.kind) {
- case Token_cast:
- case Token_transmute:
- case Token_down_cast:
- case Token_union_cast:
- {
- Token token = f->curr_token; next_token(f);
- Token open = expect_token(f, Token_OpenParen);
- AstNode *type = parse_type(f);
- Token close = expect_token(f, Token_CloseParen);
- AstNode *expr = parse_unary_expr(f, lhs);
- return ast_cast_expr(f, token, type, expr, open, close);
- } break;
- case Token_Pointer: {
- Token op = f->curr_token;
- next_token(f);
- AstNode *expr = parse_unary_expr(f, lhs);
- if (is_ast_node_type(expr)) {
- return ast_pointer_type(f, op, expr);
- }
- return ast_unary_expr(f, op, expr);
- } break;
- // case Token_Maybe:
- case Token_Add:
- case Token_Sub:
- case Token_Not:
- case Token_Xor: {
- Token op = f->curr_token;
- next_token(f);
- return ast_unary_expr(f, op, parse_unary_expr(f, lhs));
- } break;
- }
- return parse_atom_expr(f, lhs);
- }
- // NOTE(bill): result == priority
- i32 token_precedence(TokenKind t) {
- switch (t) {
- case Token_Question:
- return 1;
- case Token_CmpOr:
- return 2;
- case Token_CmpAnd:
- return 3;
- case Token_CmpEq:
- case Token_NotEq:
- case Token_Lt:
- case Token_Gt:
- case Token_LtEq:
- case Token_GtEq:
- return 4;
- case Token_Add:
- case Token_Sub:
- case Token_Or:
- case Token_Xor:
- return 5;
- case Token_Mul:
- case Token_Quo:
- case Token_Mod:
- case Token_And:
- case Token_AndNot:
- case Token_Shl:
- case Token_Shr:
- return 6;
- }
- return 0;
- }
- AstNode *parse_binary_expr(AstFile *f, bool lhs, i32 prec_in) {
- AstNode *expr = parse_unary_expr(f, lhs);
- for (i32 prec = token_precedence(f->curr_token.kind); prec >= prec_in; prec--) {
- for (;;) {
- Token op = f->curr_token;
- i32 op_prec = token_precedence(op.kind);
- if (op_prec != prec) {
- // NOTE(bill): This will also catch operators that are not valid "binary" operators
- break;
- }
- expect_operator(f); // NOTE(bill): error checks too
- if (op.kind == Token_Question) {
- AstNode *cond = expr;
- // Token_Question
- AstNode *x = parse_expr(f, lhs);
- Token token_c = expect_token(f, Token_Colon);
- AstNode *y = parse_expr(f, lhs);
- expr = ast_ternary_expr(f, cond, x, y);
- } else {
- AstNode *right = parse_binary_expr(f, false, prec+1);
- if (!right) {
- syntax_error(op, "Expected expression on the right-hand side of the binary operator");
- }
- expr = ast_binary_expr(f, op, expr, right);
- }
- lhs = false;
- }
- }
- return expr;
- }
- AstNode *parse_expr(AstFile *f, bool lhs) {
- return parse_binary_expr(f, lhs, 0+1);
- }
- AstNodeArray parse_expr_list(AstFile *f, bool lhs) {
- AstNodeArray list = make_ast_node_array(f);
- for (;;) {
- AstNode *e = parse_expr(f, lhs);
- array_add(&list, e);
- if (f->curr_token.kind != Token_Comma ||
- f->curr_token.kind == Token_EOF) {
- break;
- }
- next_token(f);
- }
- return list;
- }
- AstNodeArray parse_lhs_expr_list(AstFile *f) {
- return parse_expr_list(f, true);
- }
- AstNodeArray parse_rhs_expr_list(AstFile *f) {
- return parse_expr_list(f, false);
- }
- AstNodeArray parse_ident_list(AstFile *f) {
- AstNodeArray list = make_ast_node_array(f);
- do {
- array_add(&list, parse_ident(f));
- if (f->curr_token.kind != Token_Comma ||
- f->curr_token.kind == Token_EOF) {
- break;
- }
- next_token(f);
- } while (true);
- return list;
- }
- AstNode *parse_type_attempt(AstFile *f) {
- AstNode *type = parse_type_or_ident(f);
- if (type != NULL) {
- // TODO(bill): Handle?
- }
- return type;
- }
- AstNode *parse_type(AstFile *f) {
- AstNode *type = parse_type_attempt(f);
- if (type == NULL) {
- Token token = f->curr_token;
- syntax_error(token, "Expected a type");
- next_token(f);
- return ast_bad_expr(f, token, f->curr_token);
- }
- return type;
- }
- AstNode *parse_value_decl(AstFile *f, AstNodeArray lhs) {
- AstNode *type = NULL;
- AstNodeArray values = {0};
- bool is_mutable = true;
- if (allow_token(f, Token_Colon)) {
- type = parse_type_attempt(f);
- } else if (f->curr_token.kind != Token_Eq &&
- f->curr_token.kind != Token_Semicolon) {
- syntax_error(f->curr_token, "Expected a type separator `:` or `=`");
- }
- switch (f->curr_token.kind) {
- case Token_Colon:
- is_mutable = false;
- /*fallthrough*/
- case Token_Eq:
- next_token(f);
- values = parse_rhs_expr_list(f);
- if (values.count > lhs.count) {
- syntax_error(f->curr_token, "Too many values on the right hand side of the declaration");
- } else if (values.count < lhs.count && !is_mutable) {
- syntax_error(f->curr_token, "All constant declarations must be defined");
- } else if (values.count == 0) {
- syntax_error(f->curr_token, "Expected an expression for this declaration");
- }
- break;
- }
- if (is_mutable) {
- if (type == NULL && values.count == 0) {
- syntax_error(f->curr_token, "Missing variable type or initialization");
- return ast_bad_decl(f, f->curr_token, f->curr_token);
- }
- } else {
- if (type == NULL && values.count == 0 && lhs.count > 0) {
- syntax_error(f->curr_token, "Missing constant value");
- return ast_bad_decl(f, f->curr_token, f->curr_token);
- }
- }
- if (values.e == NULL) {
- values = make_ast_node_array(f);
- }
- AstNodeArray specs = {0};
- array_init_reserve(&specs, heap_allocator(), 1);
- return ast_value_decl(f, is_mutable, lhs, type, values);
- }
- AstNode *parse_simple_stmt(AstFile *f, bool in_stmt_ok) {
- AstNodeArray lhs = parse_lhs_expr_list(f);
- Token token = f->curr_token;
- switch (token.kind) {
- case Token_Eq:
- case Token_AddEq:
- case Token_SubEq:
- case Token_MulEq:
- case Token_QuoEq:
- case Token_ModEq:
- case Token_AndEq:
- case Token_OrEq:
- case Token_XorEq:
- case Token_ShlEq:
- case Token_ShrEq:
- case Token_AndNotEq:
- case Token_CmpAndEq:
- case Token_CmpOrEq:
- {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a simple statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- next_token(f);
- AstNodeArray rhs = parse_rhs_expr_list(f);
- if (rhs.count == 0) {
- syntax_error(token, "No right-hand side in assignment statement.");
- return ast_bad_stmt(f, token, f->curr_token);
- }
- return ast_assign_stmt(f, token, lhs, rhs);
- } break;
- case Token_in:
- if (in_stmt_ok) {
- allow_token(f, Token_in);
- AstNode *expr = parse_expr(f, false);
- switch (f->curr_token.kind) {
- case Token_Ellipsis: {
- Token op = f->curr_token;
- next_token(f);
- AstNode *right = parse_expr(f, false);
- expr = ast_interval_expr(f, op, expr, right);
- } break;
- }
- AstNodeArray rhs = {0};
- array_init_count(&rhs, heap_allocator(), 1);
- rhs.e[0] = expr;
- return ast_assign_stmt(f, token, lhs, rhs);
- }
- break;
- case Token_Colon:
- return parse_value_decl(f, lhs);
- }
- if (lhs.count > 1) {
- syntax_error(token, "Expected 1 expression");
- return ast_bad_stmt(f, token, f->curr_token);
- }
- switch (token.kind) {
- case Token_Inc:
- case Token_Dec:
- next_token(f);
- return ast_inc_dec_stmt(f, token, lhs.e[0]);
- }
- return ast_expr_stmt(f, lhs.e[0]);
- }
- AstNode *parse_block_stmt(AstFile *f, b32 is_when) {
- if (!is_when && f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a block statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- return parse_body(f);
- }
- AstNode *parse_field_list(AstFile *f, isize *name_count_, u32 allowed_flags, TokenKind follow);
- AstNode *parse_results(AstFile *f) {
- if (!allow_token(f, Token_ArrowRight)) {
- return NULL;
- }
- if (f->curr_token.kind != Token_OpenParen) {
- Token begin_token = f->curr_token;
- AstNodeArray empty_names = {0};
- AstNodeArray list = make_ast_node_array(f);
- AstNode *type = parse_type(f);
- array_add(&list, ast_field(f, empty_names, type, 0));
- return ast_field_list(f, begin_token, list);
- }
- AstNode *list = NULL;
- expect_token(f, Token_OpenParen);
- list = parse_field_list(f, NULL, 0, Token_CloseParen);
- expect_token_after(f, Token_CloseParen, "parameter list");
- return list;
- }
- AstNode *parse_proc_type(AstFile *f, AstNode **foreign_library_, String *foreign_name_, String *link_name_) {
- AstNode *params = {0};
- AstNode *results = {0};
- Token proc_token = expect_token(f, Token_proc);
- expect_token(f, Token_OpenParen);
- params = parse_field_list(f, NULL, FieldFlag_Signature, Token_CloseParen);
- expect_token_after(f, Token_CloseParen, "parameter list");
- results = parse_results(f);
- u64 tags = 0;
- String foreign_name = {0};
- String link_name = {0};
- AstNode *foreign_library = NULL;
- ProcCallingConvention cc = ProcCC_Odin;
- parse_proc_tags(f, &tags, &foreign_library, &foreign_name, &link_name, &cc);
- if (foreign_library_) *foreign_library_ = foreign_library;
- if (foreign_name_) *foreign_name_ = foreign_name;
- if (link_name_) *link_name_ = link_name;
- return ast_proc_type(f, proc_token, params, results, tags, cc);
- }
- AstNode *parse_var_type(AstFile *f, bool allow_ellipsis) {
- if (allow_ellipsis && f->curr_token.kind == Token_Ellipsis) {
- Token tok = f->curr_token;
- next_token(f);
- AstNode *type = parse_type_or_ident(f);
- if (type == NULL) {
- error(tok, "variadic field missing type after `...`");
- type = ast_bad_expr(f, tok, f->curr_token);
- }
- return ast_ellipsis(f, tok, type);
- }
- AstNode *type = parse_type_attempt(f);
- if (type == NULL) {
- Token tok = f->curr_token;
- error(tok, "Expected a type");
- type = ast_bad_expr(f, tok, f->curr_token);
- }
- return type;
- }
- bool is_token_field_prefix(TokenKind kind) {
- switch (kind) {
- case Token_using:
- case Token_no_alias:
- case Token_immutable:
- return true;
- }
- return false;
- }
- u32 parse_field_prefixes(AstFile *f) {
- i32 using_count = 0;
- i32 no_alias_count = 0;
- i32 immutable_count = 0;
- while (is_token_field_prefix(f->curr_token.kind)) {
- switch (f->curr_token.kind) {
- case Token_using: using_count += 1; next_token(f); break;
- case Token_no_alias: no_alias_count += 1; next_token(f); break;
- case Token_immutable: immutable_count += 1; next_token(f); break;
- }
- }
- if (using_count > 1) syntax_error(f->curr_token, "Multiple `using` in this field list");
- if (no_alias_count > 1) syntax_error(f->curr_token, "Multiple `no_alias` in this field list");
- if (immutable_count > 1) syntax_error(f->curr_token, "Multiple `immutable` in this field list");
- u32 field_flags = 0;
- if (using_count > 0) field_flags |= FieldFlag_using;
- if (no_alias_count > 0) field_flags |= FieldFlag_no_alias;
- if (immutable_count > 0) field_flags |= FieldFlag_immutable;
- return field_flags;
- }
- u32 check_field_prefixes(AstFile *f, isize name_count, u32 allowed_flags, u32 set_flags) {
- if (name_count > 1 && (set_flags&FieldFlag_using)) {
- syntax_error(f->curr_token, "Cannot apply `using` to more than one of the same type");
- set_flags &= ~FieldFlag_using;
- }
- if ((allowed_flags&FieldFlag_using) == 0 && (set_flags&FieldFlag_using)) {
- syntax_error(f->curr_token, "`using` is not allowed within this field list");
- set_flags &= ~FieldFlag_using;
- }
- if ((allowed_flags&FieldFlag_no_alias) == 0 && (set_flags&FieldFlag_no_alias)) {
- syntax_error(f->curr_token, "`no_alias` is not allowed within this field list");
- set_flags &= ~FieldFlag_no_alias;
- }
- if ((allowed_flags&FieldFlag_immutable) == 0 && (set_flags&FieldFlag_immutable)) {
- syntax_error(f->curr_token, "`immutable` is not allowed within this field list");
- set_flags &= ~FieldFlag_immutable;
- }
- return set_flags;
- }
- typedef struct AstNodeAndFlags {
- AstNode *node;
- u32 flags;
- } AstNodeAndFlags;
- typedef Array(AstNodeAndFlags) AstNodeAndFlagsArray;
- AstNodeArray convert_to_ident_list(AstFile *f, AstNodeAndFlagsArray list, bool ignore_flags) {
- AstNodeArray idents = {0};
- array_init_reserve(&idents, heap_allocator(), list.count);
- // Convert to ident list
- for_array(i, list) {
- AstNode *ident = list.e[i].node;
- if (!ignore_flags) {
- if (i != 0) {
- error_node(ident, "Illegal use of prefixes in parameter list");
- }
- }
- switch (ident->kind) {
- case AstNode_Ident:
- case AstNode_BadExpr:
- break;
- default:
- error_node(ident, "Expected an identifier");
- ident = ast_ident(f, blank_token);
- break;
- }
- array_add(&idents, ident);
- }
- return idents;
- }
- bool parse_expect_field_separator(AstFile *f, AstNode *param) {
- Token token = f->curr_token;
- if (allow_token(f, Token_Comma)) {
- return true;
- }
- if (token.kind == Token_Semicolon) {
- next_token(f);
- error(f->curr_token, "Expected a comma, got a semicolon");
- return true;
- }
- return false;
- }
- AstNode *parse_field_list(AstFile *f, isize *name_count_, u32 allowed_flags, TokenKind follow) {
- TokenKind separator = Token_Comma;
- Token start_token = f->curr_token;
- AstNodeArray params = make_ast_node_array(f);
- AstNodeAndFlagsArray list = {0}; array_init(&list, heap_allocator()); // LEAK(bill):
- isize total_name_count = 0;
- bool allow_ellipsis = allowed_flags&FieldFlag_ellipsis;
- while (f->curr_token.kind != follow &&
- f->curr_token.kind != Token_Colon &&
- f->curr_token.kind != Token_EOF) {
- u32 flags = parse_field_prefixes(f);
- AstNode *param = parse_var_type(f, allow_ellipsis);
- AstNodeAndFlags naf = {param, flags};
- array_add(&list, naf);
- if (f->curr_token.kind != Token_Comma) {
- break;
- }
- next_token(f);
- }
- if (f->curr_token.kind == Token_Colon) {
- AstNodeArray names = convert_to_ident_list(f, list, true); // Copy for semantic reasons
- if (names.count == 0) {
- syntax_error(f->curr_token, "Empty field declaration");
- }
- u32 set_flags = 0;
- if (list.count > 0) {
- set_flags = list.e[0].flags;
- }
- set_flags = check_field_prefixes(f, names.count, allowed_flags, set_flags);
- total_name_count += names.count;
- expect_token_after(f, Token_Colon, "field list");
- AstNode *type = parse_var_type(f, allow_ellipsis);
- AstNode *param = ast_field(f, names, type, set_flags);
- array_add(¶ms, param);
- parse_expect_field_separator(f, type);
- while (f->curr_token.kind != follow &&
- f->curr_token.kind != Token_EOF) {
- u32 set_flags = parse_field_prefixes(f);
- AstNodeArray names = parse_ident_list(f);
- if (names.count == 0) {
- syntax_error(f->curr_token, "Empty field declaration");
- break;
- }
- set_flags = check_field_prefixes(f, names.count, allowed_flags, set_flags);
- total_name_count += names.count;
- expect_token_after(f, Token_Colon, "field list");
- AstNode *type = parse_var_type(f, allow_ellipsis);
- AstNode *param = ast_field(f, names, type, set_flags);
- array_add(¶ms, param);
- if (!parse_expect_field_separator(f, param)) {
- break;
- }
- }
- if (name_count_) *name_count_ = total_name_count;
- return ast_field_list(f, start_token, params);
- }
- for_array(i, list) {
- AstNodeArray names = {0};
- AstNode *type = list.e[i].node;
- Token token = blank_token;
- array_init_count(&names, heap_allocator(), 1);
- token.pos = ast_node_token(type).pos;
- names.e[0] = ast_ident(f, token);
- u32 flags = check_field_prefixes(f, list.count, allowed_flags, list.e[i].flags);
- AstNode *param = ast_field(f, names, list.e[i].node, flags);
- array_add(¶ms, param);
- }
- if (name_count_) *name_count_ = total_name_count;
- return ast_field_list(f, start_token, params);
- }
- AstNode *parse_record_fields(AstFile *f, isize *field_count_, u32 flags, String context) {
- return parse_field_list(f, field_count_, flags, Token_CloseBrace);
- }
- AstNode *parse_type_or_ident(AstFile *f) {
- switch (f->curr_token.kind) {
- case Token_Ident:
- {
- AstNode *e = parse_ident(f);
- while (f->curr_token.kind == Token_Period) {
- Token token = f->curr_token;
- next_token(f);
- AstNode *sel = parse_ident(f);
- e = ast_selector_expr(f, token, e, sel);
- }
- if (f->curr_token.kind == Token_OpenParen) {
- // HACK NOTE(bill): For type_of_val(expr) et al.
- e = parse_call_expr(f, e);
- }
- return e;
- }
- case Token_Hash: {
- Token hash_token = expect_token(f, Token_Hash);
- Token name = expect_token(f, Token_Ident);
- String tag = name.string;
- if (str_eq(tag, str_lit("type"))) {
- AstNode *type = parse_type(f);
- return ast_helper_type(f, hash_token, type);
- }
- syntax_error(name, "Expected `type` after #");
- return ast_bad_expr(f, hash_token, f->curr_token);
- }
- case Token_Pointer: {
- Token token = expect_token(f, Token_Pointer);
- AstNode *elem = parse_type(f);
- return ast_pointer_type(f, token, elem);
- }
- case Token_OpenBracket: {
- Token token = expect_token(f, Token_OpenBracket);
- AstNode *count_expr = NULL;
- bool is_vector = false;
- if (f->curr_token.kind == Token_Ellipsis) {
- count_expr = ast_unary_expr(f, expect_token(f, Token_Ellipsis), NULL);
- } else if (f->curr_token.kind == Token_vector) {
- next_token(f);
- if (f->curr_token.kind != Token_CloseBracket) {
- f->expr_level++;
- count_expr = parse_expr(f, false);
- f->expr_level--;
- } else {
- syntax_error(f->curr_token, "Vector type missing count");
- }
- is_vector = true;
- } else if (f->curr_token.kind == Token_dynamic) {
- next_token(f);
- expect_token(f, Token_CloseBracket);
- return ast_dynamic_array_type(f, token, parse_type(f));
- } else if (f->curr_token.kind != Token_CloseBracket) {
- f->expr_level++;
- count_expr = parse_expr(f, false);
- f->expr_level--;
- }
- expect_token(f, Token_CloseBracket);
- if (is_vector) {
- return ast_vector_type(f, token, count_expr, parse_type(f));
- }
- return ast_array_type(f, token, count_expr, parse_type(f));
- }
- case Token_map: {
- Token token = expect_token(f, Token_map);
- AstNode *count = NULL;
- AstNode *key = NULL;
- AstNode *value = NULL;
- Token open = expect_token_after(f, Token_OpenBracket, "map");
- key = parse_expr(f, true);
- if (allow_token(f, Token_Comma)) {
- count = key;
- key = parse_type(f);
- }
- Token close = expect_token(f, Token_CloseBracket);
- value = parse_type(f);
- return ast_map_type(f, token, count, key, value);
- } break;
- case Token_struct: {
- Token token = expect_token(f, Token_struct);
- bool is_packed = false;
- bool is_ordered = false;
- AstNode *align = NULL;
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- while (allow_token(f, Token_Hash)) {
- Token tag = expect_token_after(f, Token_Ident, "#");
- if (str_eq(tag.string, str_lit("packed"))) {
- if (is_packed) {
- syntax_error(tag, "Duplicate struct tag `#%.*s`", LIT(tag.string));
- }
- is_packed = true;
- } else if (str_eq(tag.string, str_lit("ordered"))) {
- if (is_ordered) {
- syntax_error(tag, "Duplicate struct tag `#%.*s`", LIT(tag.string));
- }
- is_ordered = true;
- } else if (str_eq(tag.string, str_lit("align"))) {
- if (align) {
- syntax_error(tag, "Duplicate struct tag `#%.*s`", LIT(tag.string));
- }
- align = parse_expr(f, true);
- } else {
- syntax_error(tag, "Invalid struct tag `#%.*s`", LIT(tag.string));
- }
- }
- f->expr_level = prev_level;
- if (is_packed && is_ordered) {
- syntax_error(token, "`#ordered` is not needed with `#packed` which implies ordering");
- }
- Token open = expect_token_after(f, Token_OpenBrace, "struct");
- isize decl_count = 0;
- AstNode *fields = parse_record_fields(f, &decl_count, FieldFlag_using, str_lit("struct"));
- Token close = expect_token(f, Token_CloseBrace);
- AstNodeArray decls = {0};
- if (fields != NULL) {
- GB_ASSERT(fields->kind == AstNode_FieldList);
- decls = fields->FieldList.list;
- }
- return ast_struct_type(f, token, decls, decl_count, is_packed, is_ordered, align);
- } break;
- case Token_union: {
- Token token = expect_token(f, Token_union);
- Token open = expect_token_after(f, Token_OpenBrace, "union");
- AstNodeArray decls = make_ast_node_array(f);
- AstNodeArray variants = make_ast_node_array(f);
- isize total_decl_name_count = 0;
- while (f->curr_token.kind != Token_CloseBrace &&
- f->curr_token.kind != Token_EOF) {
- u32 decl_flags = parse_field_prefixes(f);
- if (decl_flags != 0) {
- AstNodeArray names = parse_ident_list(f);
- if (names.count == 0) {
- syntax_error(f->curr_token, "Empty field declaration");
- }
- u32 set_flags = check_field_prefixes(f, names.count, FieldFlag_using, decl_flags);
- total_decl_name_count += names.count;
- expect_token_after(f, Token_Colon, "field list");
- AstNode *type = parse_var_type(f, false);
- array_add(&decls, ast_field(f, names, type, set_flags));
- } else {
- AstNodeArray names = parse_ident_list(f);
- if (names.count == 0) {
- break;
- }
- if (names.count > 1 || f->curr_token.kind == Token_Colon) {
- u32 set_flags = check_field_prefixes(f, names.count, FieldFlag_using, decl_flags);
- total_decl_name_count += names.count;
- expect_token_after(f, Token_Colon, "field list");
- AstNode *type = parse_var_type(f, false);
- array_add(&decls, ast_field(f, names, type, set_flags));
- } else {
- AstNode *name = names.e[0];
- Token open = expect_token(f, Token_OpenBrace);
- isize decl_count = 0;
- AstNode *list = parse_record_fields(f, &decl_count, FieldFlag_using, str_lit("union"));
- Token close = expect_token(f, Token_CloseBrace);
- array_add(&variants, ast_union_field(f, name, list));
- }
- }
- if (f->curr_token.kind != Token_Comma) {
- break;
- }
- next_token(f);
- }
- Token close = expect_token(f, Token_CloseBrace);
- return ast_union_type(f, token, decls, total_decl_name_count, variants);
- }
- case Token_raw_union: {
- Token token = expect_token(f, Token_raw_union);
- Token open = expect_token_after(f, Token_OpenBrace, "raw_union");
- isize decl_count = 0;
- AstNode *fields = parse_record_fields(f, &decl_count, FieldFlag_using, str_lit("raw_union"));
- Token close = expect_token(f, Token_CloseBrace);
- AstNodeArray decls = {0};
- if (fields != NULL) {
- GB_ASSERT(fields->kind == AstNode_FieldList);
- decls = fields->FieldList.list;
- }
- return ast_raw_union_type(f, token, decls, decl_count);
- }
- case Token_enum: {
- Token token = expect_token(f, Token_enum);
- AstNode *base_type = NULL;
- if (f->curr_token.kind != Token_OpenBrace) {
- base_type = parse_type(f);
- }
- Token open = expect_token(f, Token_OpenBrace);
- AstNodeArray values = parse_element_list(f);
- Token close = expect_token(f, Token_CloseBrace);
- return ast_enum_type(f, token, base_type, values);
- }
- case Token_proc: {
- Token token = f->curr_token;
- AstNode *pt = parse_proc_type(f, NULL, NULL, NULL);
- if (pt->ProcType.tags != 0) {
- syntax_error(token, "A procedure type cannot have tags");
- }
- return pt;
- }
- case Token_OpenParen: {
- Token open = expect_token(f, Token_OpenParen);
- AstNode *type = parse_type(f);
- Token close = expect_token(f, Token_CloseParen);
- return ast_paren_expr(f, type, open, close);
- } break;
- }
- // No type found
- return NULL;
- }
- AstNode *parse_body(AstFile *f) {
- AstNodeArray stmts = {0};
- Token open, close;
- isize prev_expr_level = f->expr_level;
- // NOTE(bill): The body may be within an expression so reset to zero
- f->expr_level = 0;
- open = expect_token(f, Token_OpenBrace);
- stmts = parse_stmt_list(f);
- close = expect_token(f, Token_CloseBrace);
- f->expr_level = prev_expr_level;
- return ast_block_stmt(f, stmts, open, close);
- }
- AstNode *parse_if_stmt(AstFile *f) {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use an if statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- Token token = expect_token(f, Token_if);
- AstNode *init = NULL;
- AstNode *cond = NULL;
- AstNode *body = NULL;
- AstNode *else_stmt = NULL;
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- if (allow_token(f, Token_Semicolon)) {
- cond = parse_expr(f, false);
- } else {
- init = parse_simple_stmt(f, false);
- if (allow_token(f, Token_Semicolon)) {
- cond = parse_expr(f, false);
- } else {
- cond = convert_stmt_to_expr(f, init, str_lit("boolean expression"));
- init = NULL;
- }
- }
- f->expr_level = prev_level;
- if (cond == NULL) {
- syntax_error(f->curr_token, "Expected condition for if statement");
- }
- body = parse_block_stmt(f, false);
- if (allow_token(f, Token_else)) {
- switch (f->curr_token.kind) {
- case Token_if:
- else_stmt = parse_if_stmt(f);
- break;
- case Token_OpenBrace:
- else_stmt = parse_block_stmt(f, false);
- break;
- default:
- syntax_error(f->curr_token, "Expected if statement block statement");
- else_stmt = ast_bad_stmt(f, f->curr_token, f->tokens.e[f->curr_token_index+1]);
- break;
- }
- }
- return ast_if_stmt(f, token, init, cond, body, else_stmt);
- }
- AstNode *parse_when_stmt(AstFile *f) {
- Token token = expect_token(f, Token_when);
- AstNode *cond = NULL;
- AstNode *body = NULL;
- AstNode *else_stmt = NULL;
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- cond = parse_expr(f, false);
- f->expr_level = prev_level;
- if (cond == NULL) {
- syntax_error(f->curr_token, "Expected condition for when statement");
- }
- body = parse_block_stmt(f, true);
- if (allow_token(f, Token_else)) {
- switch (f->curr_token.kind) {
- case Token_when:
- else_stmt = parse_when_stmt(f);
- break;
- case Token_OpenBrace:
- else_stmt = parse_block_stmt(f, true);
- break;
- default:
- syntax_error(f->curr_token, "Expected when statement block statement");
- else_stmt = ast_bad_stmt(f, f->curr_token, f->tokens.e[f->curr_token_index+1]);
- break;
- }
- }
- return ast_when_stmt(f, token, cond, body, else_stmt);
- }
- AstNode *parse_return_stmt(AstFile *f) {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a return statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- if (f->expr_level > 0) {
- syntax_error(f->curr_token, "You cannot use a return statement within an expression");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- Token token = expect_token(f, Token_return);
- AstNodeArray results;
- if (f->curr_token.kind != Token_Semicolon && f->curr_token.kind != Token_CloseBrace) {
- results = parse_rhs_expr_list(f);
- } else {
- results = make_ast_node_array(f);
- }
- expect_semicolon(f, results.e[0]);
- return ast_return_stmt(f, token, results);
- }
- // AstNode *parse_give_stmt(AstFile *f) {
- // if (f->curr_proc == NULL) {
- // syntax_error(f->curr_token, "You cannot use a give statement in the file scope");
- // return ast_bad_stmt(f, f->curr_token, f->curr_token);
- // }
- // if (f->expr_level == 0) {
- // syntax_error(f->curr_token, "A give statement must be used within an expression");
- // return ast_bad_stmt(f, f->curr_token, f->curr_token);
- // }
- // Token token = expect_token(f, Token_give);
- // AstNodeArray results;
- // if (f->curr_token.kind != Token_Semicolon && f->curr_token.kind != Token_CloseBrace) {
- // results = parse_rhs_expr_list(f);
- // } else {
- // results = make_ast_node_array(f);
- // }
- // AstNode *ge = ast_give_expr(f, token, results);
- // expect_semicolon(f, ge);
- // return ast_expr_stmt(f, ge);
- // }
- AstNode *parse_for_stmt(AstFile *f) {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a for statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- Token token = expect_token(f, Token_for);
- AstNode *init = NULL;
- AstNode *cond = NULL;
- AstNode *post = NULL;
- AstNode *body = NULL;
- bool is_range = false;
- if (f->curr_token.kind != Token_OpenBrace) {
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- if (f->curr_token.kind != Token_Semicolon) {
- cond = parse_simple_stmt(f, true);
- if (cond->kind == AstNode_AssignStmt && cond->AssignStmt.op.kind == Token_in) {
- is_range = true;
- }
- }
- if (!is_range && f->curr_token.kind == Token_Semicolon) {
- next_token(f);
- init = cond;
- cond = NULL;
- if (f->curr_token.kind != Token_Semicolon) {
- cond = parse_simple_stmt(f, false);
- }
- expect_semicolon(f, cond);
- if (f->curr_token.kind != Token_OpenBrace) {
- post = parse_simple_stmt(f, false);
- }
- }
- f->expr_level = prev_level;
- }
- body = parse_block_stmt(f, false);
- if (is_range) {
- GB_ASSERT(cond->kind == AstNode_AssignStmt);
- Token in_token = cond->AssignStmt.op;
- AstNode *value = NULL;
- AstNode *index = NULL;
- switch (cond->AssignStmt.lhs.count) {
- case 1:
- value = cond->AssignStmt.lhs.e[0];
- break;
- case 2:
- value = cond->AssignStmt.lhs.e[0];
- index = cond->AssignStmt.lhs.e[1];
- break;
- default:
- error_node(cond, "Expected at 1 or 2 identifiers");
- return ast_bad_stmt(f, token, f->curr_token);
- }
- AstNode *rhs = NULL;
- if (cond->AssignStmt.rhs.count > 0) {
- rhs = cond->AssignStmt.rhs.e[0];
- }
- return ast_range_stmt(f, token, value, index, in_token, rhs, body);
- }
- cond = convert_stmt_to_expr(f, cond, str_lit("boolean expression"));
- return ast_for_stmt(f, token, init, cond, post, body);
- }
- AstNode *parse_case_clause(AstFile *f) {
- Token token = f->curr_token;
- AstNodeArray list = make_ast_node_array(f);
- if (allow_token(f, Token_case)) {
- list = parse_rhs_expr_list(f);
- } else {
- expect_token(f, Token_default);
- }
- expect_token(f, Token_Colon); // TODO(bill): Is this the best syntax?
- // expect_token(f, Token_ArrowRight); // TODO(bill): Is this the best syntax?
- AstNodeArray stmts = parse_stmt_list(f);
- return ast_case_clause(f, token, list, stmts);
- }
- AstNode *parse_type_case_clause(AstFile *f) {
- Token token = f->curr_token;
- AstNodeArray list = make_ast_node_array(f);
- if (allow_token(f, Token_case)) {
- for (;;) {
- AstNode *t = parse_type(f);
- array_add(&list, t);
- if (f->curr_token.kind != Token_Comma ||
- f->curr_token.kind == Token_EOF) {
- break;
- }
- next_token(f);
- }
- } else {
- expect_token(f, Token_default);
- }
- expect_token(f, Token_Colon); // TODO(bill): Is this the best syntax?
- // expect_token(f, Token_ArrowRight); // TODO(bill): Is this the best syntax?
- AstNodeArray stmts = parse_stmt_list(f);
- return ast_case_clause(f, token, list, stmts);
- }
- AstNode *parse_match_stmt(AstFile *f) {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a match statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- Token token = expect_token(f, Token_match);
- AstNode *init = NULL;
- AstNode *tag = NULL;
- AstNode *body = NULL;
- Token open, close;
- bool is_type_match = false;
- if (f->curr_token.kind != Token_OpenBrace) {
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- tag = parse_simple_stmt(f, true);
- if (tag->kind == AstNode_AssignStmt && tag->AssignStmt.op.kind == Token_in) {
- is_type_match = true;
- } else {
- if (allow_token(f, Token_Semicolon)) {
- init = tag;
- tag = NULL;
- if (f->curr_token.kind != Token_OpenBrace) {
- tag = parse_simple_stmt(f, false);
- }
- }
- }
- f->expr_level = prev_level;
- }
- open = expect_token(f, Token_OpenBrace);
- AstNodeArray list = make_ast_node_array(f);
- while (f->curr_token.kind == Token_case ||
- f->curr_token.kind == Token_default) {
- if (is_type_match) {
- array_add(&list, parse_type_case_clause(f));
- } else {
- array_add(&list, parse_case_clause(f));
- }
- }
- close = expect_token(f, Token_CloseBrace);
- body = ast_block_stmt(f, list, open, close);
- if (!is_type_match) {
- tag = convert_stmt_to_expr(f, tag, str_lit("match expression"));
- return ast_match_stmt(f, token, init, tag, body);
- } else {
- return ast_type_match_stmt(f, token, tag, body);
- }
- }
- AstNode *parse_defer_stmt(AstFile *f) {
- if (f->curr_proc == NULL) {
- syntax_error(f->curr_token, "You cannot use a defer statement in the file scope");
- return ast_bad_stmt(f, f->curr_token, f->curr_token);
- }
- Token token = expect_token(f, Token_defer);
- AstNode *stmt = parse_stmt(f);
- switch (stmt->kind) {
- case AstNode_EmptyStmt:
- syntax_error(token, "Empty statement after defer (e.g. `;`)");
- break;
- case AstNode_DeferStmt:
- syntax_error(token, "You cannot defer a defer statement");
- stmt = stmt->DeferStmt.stmt;
- break;
- case AstNode_ReturnStmt:
- syntax_error(token, "You cannot a return statement");
- break;
- }
- return ast_defer_stmt(f, token, stmt);
- }
- AstNode *parse_asm_stmt(AstFile *f) {
- Token token = expect_token(f, Token_asm);
- bool is_volatile = false;
- Token open, close, code_string;
- open = expect_token(f, Token_OpenBrace);
- code_string = expect_token(f, Token_String);
- AstNode *output_list = NULL;
- AstNode *input_list = NULL;
- AstNode *clobber_list = NULL;
- isize output_count = 0;
- isize input_count = 0;
- isize clobber_count = 0;
- // TODO(bill): Finish asm statement and determine syntax
- // if (f->curr_token.kind != Token_CloseBrace) {
- // expect_token(f, Token_Colon);
- // }
- close = expect_token(f, Token_CloseBrace);
- return ast_asm_stmt(f, token, is_volatile, open, close, code_string,
- output_list, input_list, clobber_list,
- output_count, input_count, clobber_count);
- }
- AstNode *parse_stmt(AstFile *f) {
- AstNode *s = NULL;
- Token token = f->curr_token;
- switch (token.kind) {
- // Operands
- case Token_context:
- case Token_Ident:
- case Token_Integer:
- case Token_Float:
- case Token_Imag:
- case Token_Rune:
- case Token_String:
- case Token_OpenParen:
- // Unary Operators
- case Token_Add:
- case Token_Sub:
- case Token_Xor:
- case Token_Not:
- s = parse_simple_stmt(f, false);
- expect_semicolon(f, s);
- return s;
- case Token_if: return parse_if_stmt(f);
- case Token_when: return parse_when_stmt(f);
- case Token_for: return parse_for_stmt(f);
- case Token_match: return parse_match_stmt(f);
- case Token_defer: return parse_defer_stmt(f);
- case Token_asm: return parse_asm_stmt(f);
- case Token_return: return parse_return_stmt(f);
- // case Token_give: return parse_give_stmt(f);
- case Token_break:
- case Token_continue:
- case Token_fallthrough: {
- AstNode *label = NULL;
- next_token(f);
- if (token.kind != Token_fallthrough &&
- f->curr_token.kind == Token_Ident) {
- label = parse_ident(f);
- }
- s = ast_branch_stmt(f, token, label);
- expect_semicolon(f, s);
- return s;
- }
- case Token_using: {
- // TODO(bill): Make using statements better
- Token token = expect_token(f, Token_using);
- AstNodeArray list = parse_lhs_expr_list(f);
- if (list.count == 0) {
- syntax_error(token, "Illegal use of `using` statement");
- expect_semicolon(f, NULL);
- return ast_bad_stmt(f, token, f->curr_token);
- }
- if (f->curr_token.kind != Token_Colon) {
- expect_semicolon(f, list.e[list.count-1]);
- return ast_using_stmt(f, token, list);
- }
- AstNode *decl = parse_simple_stmt(f, false);
- expect_semicolon(f, decl);
- if (decl->kind == AstNode_ValueDecl) {
- if (!decl->ValueDecl.is_var) {
- syntax_error(token, "`using` may not be applied to constant declarations");
- return decl;
- }
- if (f->curr_proc == NULL) {
- syntax_error(token, "`using` is not allowed at the file scope");
- } else {
- decl->ValueDecl.flags |= VarDeclFlag_using;
- }
- return decl;
- }
- syntax_error(token, "Illegal use of `using` statement");
- return ast_bad_stmt(f, token, f->curr_token);
- } break;
- #if 1
- case Token_immutable: {
- Token token = expect_token(f, Token_immutable);
- AstNode *node = parse_stmt(f);
- if (node->kind == AstNode_ValueDecl) {
- if (!node->ValueDecl.is_var) {
- syntax_error(token, "`immutable` may not be applied to constant declarations");
- } else {
- node->ValueDecl.flags |= VarDeclFlag_immutable;
- }
- return node;
- }
- syntax_error(token, "`immutable` may only be applied to a variable declaration");
- return ast_bad_stmt(f, token, f->curr_token);
- } break;
- #endif
- case Token_push_allocator: {
- next_token(f);
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- AstNode *expr = parse_expr(f, false);
- f->expr_level = prev_level;
- AstNode *body = parse_block_stmt(f, false);
- return ast_push_allocator(f, token, expr, body);
- } break;
- case Token_push_context: {
- next_token(f);
- isize prev_level = f->expr_level;
- f->expr_level = -1;
- AstNode *expr = parse_expr(f, false);
- f->expr_level = prev_level;
- AstNode *body = parse_block_stmt(f, false);
- return ast_push_context(f, token, expr, body);
- } break;
- case Token_Hash: {
- AstNode *s = NULL;
- Token hash_token = expect_token(f, Token_Hash);
- Token name = expect_token(f, Token_Ident);
- String tag = name.string;
- if (str_eq(tag, str_lit("label"))) {
- AstNode *name = parse_ident(f);
- AstNode *label = ast_label_decl(f, token, name);
- AstNode *stmt = parse_stmt(f);
- #define _SET_LABEL(Kind_, label_) case GB_JOIN2(AstNode_, Kind_): (stmt->Kind_).label = label_; break
- switch (stmt->kind) {
- _SET_LABEL(ForStmt, label);
- _SET_LABEL(RangeStmt, label);
- _SET_LABEL(MatchStmt, label);
- _SET_LABEL(TypeMatchStmt, label);
- default:
- syntax_error(token, "#label cannot only be applied to a loop or match statement");
- break;
- }
- #undef _SET_LABEL
- return stmt;
- } else if (str_eq(tag, str_lit("import"))) {
- AstNode *cond = NULL;
- Token import_name = {0};
- switch (f->curr_token.kind) {
- case Token_Period:
- import_name = f->curr_token;
- import_name.kind = Token_Ident;
- next_token(f);
- break;
- case Token_Ident:
- import_name = f->curr_token;
- next_token(f);
- break;
- default:
- import_name.pos = f->curr_token.pos;
- break;
- }
- if (str_eq(import_name.string, str_lit("_"))) {
- syntax_error(import_name, "Illegal #import name: `_`");
- }
- Token file_path = expect_token_after(f, Token_String, "#import");
- if (allow_token(f, Token_when)) {
- cond = parse_expr(f, false);
- }
- AstNode *decl = NULL;
- if (f->curr_proc != NULL) {
- syntax_error(import_name, "You cannot use `#import` within a procedure. This must be done at the file scope");
- decl = ast_bad_decl(f, import_name, file_path);
- } else {
- decl = ast_import_decl(f, hash_token, true, file_path, import_name, cond);
- }
- expect_semicolon(f, decl);
- return decl;
- } else if (str_eq(tag, str_lit("load"))) {
- AstNode *cond = NULL;
- Token file_path = expect_token_after(f, Token_String, "#load");
- Token import_name = file_path;
- import_name.string = str_lit(".");
- if (allow_token(f, Token_when)) {
- cond = parse_expr(f, false);
- }
- AstNode *decl = NULL;
- if (f->curr_proc != NULL) {
- syntax_error(import_name, "You cannot use `#load` within a procedure. This must be done at the file scope");
- decl = ast_bad_decl(f, import_name, file_path);
- } else {
- decl = ast_import_decl(f, hash_token, false, file_path, import_name, cond);
- }
- expect_semicolon(f, decl);
- return decl;
- } else if (str_eq(tag, str_lit("shared_global_scope"))) {
- if (f->curr_proc == NULL) {
- f->is_global_scope = true;
- s = ast_empty_stmt(f, f->curr_token);
- } else {
- syntax_error(token, "You cannot use #shared_global_scope within a procedure. This must be done at the file scope");
- s = ast_bad_decl(f, token, f->curr_token);
- }
- expect_semicolon(f, s);
- return s;
- } else if (str_eq(tag, str_lit("foreign_system_library"))) {
- AstNode *cond = NULL;
- Token lib_name = {0};
- switch (f->curr_token.kind) {
- case Token_Ident:
- lib_name = f->curr_token;
- next_token(f);
- break;
- default:
- lib_name.pos = f->curr_token.pos;
- break;
- }
- if (str_eq(lib_name.string, str_lit("_"))) {
- syntax_error(lib_name, "Illegal #foreign_library name: `_`");
- }
- Token file_path = expect_token(f, Token_String);
- if (allow_token(f, Token_when)) {
- cond = parse_expr(f, false);
- }
- if (f->curr_proc == NULL) {
- s = ast_foreign_library(f, hash_token, file_path, lib_name, cond, true);
- } else {
- syntax_error(token, "You cannot use #foreign_system_library within a procedure. This must be done at the file scope");
- s = ast_bad_decl(f, token, file_path);
- }
- expect_semicolon(f, s);
- return s;
- } else if (str_eq(tag, str_lit("foreign_library"))) {
- AstNode *cond = NULL;
- Token lib_name = {0};
- switch (f->curr_token.kind) {
- case Token_Ident:
- lib_name = f->curr_token;
- next_token(f);
- break;
- default:
- lib_name.pos = f->curr_token.pos;
- break;
- }
- if (str_eq(lib_name.string, str_lit("_"))) {
- syntax_error(lib_name, "Illegal #foreign_library name: `_`");
- }
- Token file_path = expect_token(f, Token_String);
- if (allow_token(f, Token_when)) {
- cond = parse_expr(f, false);
- }
- if (f->curr_proc == NULL) {
- s = ast_foreign_library(f, hash_token, file_path, lib_name, cond, false);
- } else {
- syntax_error(token, "You cannot use #foreign_library within a procedure. This must be done at the file scope");
- s = ast_bad_decl(f, token, file_path);
- }
- expect_semicolon(f, s);
- return s;
- } else if (str_eq(tag, str_lit("thread_local"))) {
- AstNode *s = parse_stmt(f);
- if (s->kind == AstNode_ValueDecl) {
- if (!s->ValueDecl.is_var) {
- syntax_error(token, "`thread_local` may not be applied to constant declarations");
- }
- if (f->curr_proc != NULL) {
- syntax_error(token, "`thread_local` is only allowed at the file scope");
- } else {
- s->ValueDecl.flags |= VarDeclFlag_thread_local;
- }
- return s;
- }
- syntax_error(token, "`thread_local` may only be applied to a variable declaration");
- return ast_bad_stmt(f, token, f->curr_token);
- } else if (str_eq(tag, str_lit("bounds_check"))) {
- s = parse_stmt(f);
- s->stmt_state_flags |= StmtStateFlag_bounds_check;
- if ((s->stmt_state_flags & StmtStateFlag_no_bounds_check) != 0) {
- syntax_error(token, "#bounds_check and #no_bounds_check cannot be applied together");
- }
- return s;
- } else if (str_eq(tag, str_lit("no_bounds_check"))) {
- s = parse_stmt(f);
- s->stmt_state_flags |= StmtStateFlag_no_bounds_check;
- if ((s->stmt_state_flags & StmtStateFlag_bounds_check) != 0) {
- syntax_error(token, "#bounds_check and #no_bounds_check cannot be applied together");
- }
- return s;
- }
- if (str_eq(tag, str_lit("include"))) {
- syntax_error(token, "#include is not a valid import declaration kind. Use #load instead");
- s = ast_bad_stmt(f, token, f->curr_token);
- } else {
- syntax_error(token, "Unknown tag directive used: `%.*s`", LIT(tag));
- s = ast_bad_stmt(f, token, f->curr_token);
- }
- fix_advance_to_next_stmt(f);
- return s;
- } break;
- case Token_OpenBrace:
- return parse_block_stmt(f, false);
- case Token_Semicolon:
- s = ast_empty_stmt(f, token);
- next_token(f);
- return s;
- }
- syntax_error(token,
- "Expected a statement, got `%.*s`",
- LIT(token_strings[token.kind]));
- fix_advance_to_next_stmt(f);
- return ast_bad_stmt(f, token, f->curr_token);
- }
- AstNodeArray parse_stmt_list(AstFile *f) {
- AstNodeArray list = make_ast_node_array(f);
- while (f->curr_token.kind != Token_case &&
- f->curr_token.kind != Token_default &&
- f->curr_token.kind != Token_CloseBrace &&
- f->curr_token.kind != Token_EOF) {
- AstNode *stmt = parse_stmt(f);
- if (stmt && stmt->kind != AstNode_EmptyStmt) {
- array_add(&list, stmt);
- if (stmt->kind == AstNode_ExprStmt &&
- stmt->ExprStmt.expr != NULL &&
- stmt->ExprStmt.expr->kind == AstNode_ProcLit) {
- syntax_error_node(stmt, "Procedure literal evaluated but not used");
- }
- }
- }
- return list;
- }
- ParseFileError init_ast_file(AstFile *f, String fullpath) {
- fullpath = string_trim_whitespace(fullpath); // Just in case
- if (!string_has_extension(fullpath, str_lit("odin"))) {
- return ParseFile_WrongExtension;
- }
- TokenizerInitError err = init_tokenizer(&f->tokenizer, fullpath);
- if (err == TokenizerInit_None) {
- array_init(&f->tokens, heap_allocator());
- {
- for (;;) {
- Token token = tokenizer_get_token(&f->tokenizer);
- if (token.kind == Token_Invalid) {
- return ParseFile_InvalidToken;
- }
- array_add(&f->tokens, token);
- if (token.kind == Token_EOF) {
- break;
- }
- }
- }
- f->curr_token_index = 0;
- f->prev_token = f->tokens.e[f->curr_token_index];
- f->curr_token = f->tokens.e[f->curr_token_index];
- // NOTE(bill): Is this big enough or too small?
- isize arena_size = gb_size_of(AstNode);
- arena_size *= 2*f->tokens.count;
- gb_arena_init_from_allocator(&f->arena, heap_allocator(), arena_size);
- f->curr_proc = NULL;
- return ParseFile_None;
- }
- switch (err) {
- case TokenizerInit_NotExists:
- return ParseFile_NotFound;
- case TokenizerInit_Permission:
- return ParseFile_Permission;
- case TokenizerInit_Empty:
- return ParseFile_EmptyFile;
- }
- return ParseFile_InvalidFile;
- }
- void destroy_ast_file(AstFile *f) {
- gb_arena_free(&f->arena);
- array_free(&f->tokens);
- gb_free(heap_allocator(), f->tokenizer.fullpath.text);
- destroy_tokenizer(&f->tokenizer);
- }
- bool init_parser(Parser *p) {
- array_init(&p->files, heap_allocator());
- array_init(&p->imports, heap_allocator());
- gb_mutex_init(&p->mutex);
- return true;
- }
- void destroy_parser(Parser *p) {
- // TODO(bill): Fix memory leak
- for_array(i, p->files) {
- destroy_ast_file(&p->files.e[i]);
- }
- #if 0
- for_array(i, p->imports) {
- // gb_free(heap_allocator(), p->imports[i].text);
- }
- #endif
- array_free(&p->files);
- array_free(&p->imports);
- gb_mutex_destroy(&p->mutex);
- }
- // NOTE(bill): Returns true if it's added
- bool try_add_import_path(Parser *p, String path, String rel_path, TokenPos pos) {
- gb_mutex_lock(&p->mutex);
- path = string_trim_whitespace(path);
- rel_path = string_trim_whitespace(rel_path);
- for_array(i, p->imports) {
- String import = p->imports.e[i].path;
- if (str_eq(import, path)) {
- return false;
- }
- }
- ImportedFile item;
- item.path = path;
- item.rel_path = rel_path;
- item.pos = pos;
- array_add(&p->imports, item);
- gb_mutex_unlock(&p->mutex);
- return true;
- }
- gb_global Rune illegal_import_runes[] = {
- '"', '\'', '`', ' ', '\t', '\r', '\n', '\v', '\f',
- '\\', // NOTE(bill): Disallow windows style filepaths
- '!', '$', '%', '^', '&', '*', '(', ')', '=', '+',
- '[', ']', '{', '}',
- ';', ':', '#',
- '|', ',', '<', '>', '?',
- };
- bool is_import_path_valid(String path) {
- if (path.len > 0) {
- u8 *start = path.text;
- u8 *end = path.text + path.len;
- u8 *curr = start;
- while (curr < end) {
- isize width = 1;
- Rune r = curr[0];
- if (r >= 0x80) {
- width = gb_utf8_decode(curr, end-curr, &r);
- if (r == GB_RUNE_INVALID && width == 1) {
- return false;
- }
- else if (r == GB_RUNE_BOM && curr-start > 0) {
- return false;
- }
- }
- for (isize i = 0; i < gb_count_of(illegal_import_runes); i++) {
- if (r == illegal_import_runes[i]) {
- return false;
- }
- }
- curr += width;
- }
- return true;
- }
- return false;
- }
- void parse_setup_file_decls(Parser *p, AstFile *f, String base_dir, AstNodeArray decls) {
- for_array(i, decls) {
- AstNode *node = decls.e[i];
- if (!is_ast_node_decl(node) &&
- node->kind != AstNode_BadStmt &&
- node->kind != AstNode_EmptyStmt) {
- // NOTE(bill): Sanity check
- syntax_error_node(node, "Only declarations are allowed at file scope %.*s", LIT(ast_node_strings[node->kind]));
- } else if (node->kind == AstNode_ImportDecl) {
- ast_node(id, ImportDecl, node);
- String file_str = id->relpath.string;
- if (!is_import_path_valid(file_str)) {
- if (id->is_import) {
- syntax_error_node(node, "Invalid import path: `%.*s`", LIT(file_str));
- } else {
- syntax_error_node(node, "Invalid include path: `%.*s`", LIT(file_str));
- }
- // NOTE(bill): It's a naughty name
- decls.e[i] = ast_bad_decl(f, id->relpath, id->relpath);
- continue;
- }
- gbAllocator allocator = heap_allocator(); // TODO(bill): Change this allocator
- String rel_path = get_fullpath_relative(allocator, base_dir, file_str);
- String import_file = rel_path;
- if (!gb_file_exists(cast(char *)rel_path.text)) { // NOTE(bill): This should be null terminated
- String abs_path = get_fullpath_core(allocator, file_str);
- if (gb_file_exists(cast(char *)abs_path.text)) {
- import_file = abs_path;
- }
- }
- id->fullpath = import_file;
- try_add_import_path(p, import_file, file_str, ast_node_token(node).pos);
- } else if (node->kind == AstNode_ForeignLibrary) {
- AstNodeForeignLibrary *fl = &node->ForeignLibrary;
- String file_str = fl->filepath.string;
- if (!is_import_path_valid(file_str)) {
- if (fl->is_system) {
- syntax_error_node(node, "Invalid `foreign_system_library` path");
- } else {
- syntax_error_node(node, "Invalid `foreign_library` path");
- }
- // NOTE(bill): It's a naughty name
- f->decls.e[i] = ast_bad_decl(f, fl->token, fl->token);
- continue;
- }
- fl->base_dir = base_dir;
- }
- }
- }
- void parse_file(Parser *p, AstFile *f) {
- String filepath = f->tokenizer.fullpath;
- String base_dir = filepath;
- for (isize i = filepath.len-1; i >= 0; i--) {
- if (base_dir.text[i] == '\\' ||
- base_dir.text[i] == '/') {
- break;
- }
- base_dir.len--;
- }
- while (f->curr_token.kind == Token_Comment) {
- next_token(f);
- }
- f->decls = parse_stmt_list(f);
- parse_setup_file_decls(p, f, base_dir, f->decls);
- }
- ParseFileError parse_files(Parser *p, char *init_filename) {
- char *fullpath_str = gb_path_get_full_name(heap_allocator(), init_filename);
- String init_fullpath = make_string_c(fullpath_str);
- TokenPos init_pos = {0};
- ImportedFile init_imported_file = {init_fullpath, init_fullpath, init_pos};
- {
- String s = get_fullpath_core(heap_allocator(), str_lit("_preload.odin"));
- ImportedFile runtime_file = {s, s, init_pos};
- array_add(&p->imports, runtime_file);
- }
- {
- String s = get_fullpath_core(heap_allocator(), str_lit("_soft_numbers.odin"));
- ImportedFile runtime_file = {s, s, init_pos};
- array_add(&p->imports, runtime_file);
- }
- array_add(&p->imports, init_imported_file);
- p->init_fullpath = init_fullpath;
- for_array(i, p->imports) {
- ImportedFile imported_file = p->imports.e[i];
- String import_path = imported_file.path;
- String import_rel_path = imported_file.rel_path;
- TokenPos pos = imported_file.pos;
- AstFile file = {0};
- ParseFileError err = init_ast_file(&file, import_path);
- if (err != ParseFile_None) {
- if (err == ParseFile_EmptyFile) {
- return ParseFile_None;
- }
- if (pos.line != 0) {
- gb_printf_err("%.*s(%td:%td) ", LIT(pos.file), pos.line, pos.column);
- }
- gb_printf_err("Failed to parse file: %.*s\n\t", LIT(import_rel_path));
- switch (err) {
- case ParseFile_WrongExtension:
- gb_printf_err("Invalid file extension: File must have the extension `.odin`");
- break;
- case ParseFile_InvalidFile:
- gb_printf_err("Invalid file or cannot be found");
- break;
- case ParseFile_Permission:
- gb_printf_err("File permissions problem");
- break;
- case ParseFile_NotFound:
- gb_printf_err("File cannot be found (`%.*s`)", LIT(import_path));
- break;
- case ParseFile_InvalidToken:
- gb_printf_err("Invalid token found in file");
- break;
- }
- gb_printf_err("\n");
- return err;
- }
- parse_file(p, &file);
- {
- gb_mutex_lock(&p->mutex);
- file.id = p->files.count;
- array_add(&p->files, file);
- p->total_line_count += file.tokenizer.line_count;
- gb_mutex_unlock(&p->mutex);
- }
- }
- for_array(i, p->files) {
- p->total_token_count += p->files.e[i].tokens.count;
- }
- return ParseFile_None;
- }
|