123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873 |
- #include "BfCompiler.h"
- #include "BfSystem.h"
- #include "BfParser.h"
- #include "BfCodeGen.h"
- #include "BfExprEvaluator.h"
- #include <fcntl.h>
- #include "BfConstResolver.h"
- #include "BfMangler.h"
- #include "BeefySysLib/util/PerfTimer.h"
- #include "BeefySysLib/util/BeefPerf.h"
- #include "BfSourceClassifier.h"
- #include "BfAutoComplete.h"
- #include "BfDemangler.h"
- #include "BfResolvePass.h"
- #include "BfFixits.h"
- #include "BfIRCodeGen.h"
- #include "BfDefBuilder.h"
- #pragma warning(push)
- #pragma warning(disable:4141)
- #pragma warning(disable:4146)
- #pragma warning(disable:4291)
- #pragma warning(disable:4244)
- #pragma warning(disable:4267)
- #pragma warning(disable:4624)
- #pragma warning(disable:4800)
- #pragma warning(disable:4996)
- #include "llvm/IR/Module.h"
- #include "llvm/IR/Constants.h"
- #include "llvm/IR/GlobalValue.h"
- #include "llvm/IR/GlobalVariable.h"
- #include "llvm/ADT/ArrayRef.h"
- #include "llvm/IR/InlineAsm.h"
- #include "llvm/Support/FileSystem.h"
- #include "BeefySysLib/util/AllocDebug.h"
- #pragma warning(pop)
- USING_NS_BF;
- bool BfModule::AddDeferredCallEntry(BfDeferredCallEntry* deferredCallEntry, BfScopeData* scopeData)
- {
- if (((mCompiler->mIsResolveOnly) || (mBfIRBuilder->mIgnoreWrites)) && (deferredCallEntry->mDeferredBlock == NULL))
- {
- // For resolve entries, we only keep deferred blocks because we need to process them later so we can
- // resolve inside of them. This is also required for lambda bind scan-pass
- delete deferredCallEntry;
- return false;
- }
- if (mBfIRBuilder->mIgnoreWrites)
- {
- deferredCallEntry->mIgnored = true;
- scopeData->mDeferredCallEntries.PushBack(deferredCallEntry);
- return true;
- }
- // We don't need to do a "clear handlers" if we're just adding another dyn to an existing dyn list
- bool isDyn = mCurMethodState->mCurScope->IsDyn(scopeData);
- if (!isDyn)
- {
- mCurMethodState->mCurScope->ClearHandlers(scopeData);
- if (!mBfIRBuilder->mIgnoreWrites)
- {
- if ((IsTargetingBeefBackend()) && (deferredCallEntry->mModuleMethodInstance.mMethodInstance != NULL) &&
- (!mContext->IsSentinelMethod(deferredCallEntry->mModuleMethodInstance.mMethodInstance)))
- {
- SizedArray<BfIRType, 8> origParamTypes;
- BfIRType origReturnType;
- deferredCallEntry->mModuleMethodInstance.mMethodInstance->GetIRFunctionInfo(this, origReturnType, origParamTypes);
- BF_ASSERT(origParamTypes.size() == deferredCallEntry->mScopeArgs.size());
- for (int paramIdx = 0; paramIdx < (int)deferredCallEntry->mScopeArgs.size(); paramIdx++)
- {
- auto scopeArg = deferredCallEntry->mScopeArgs[paramIdx];
- if ((scopeArg.IsConst()) || (scopeArg.IsFake()))
- continue;
- auto prevInsertBlock = mBfIRBuilder->GetInsertBlock();
- mBfIRBuilder->SetInsertPoint(mCurMethodState->mIRHeadBlock);
- auto allocaInst = mBfIRBuilder->CreateAlloca(origParamTypes[paramIdx]);
- mBfIRBuilder->ClearDebugLocation(allocaInst);
- mBfIRBuilder->SetInsertPoint(prevInsertBlock);
- if (WantsLifetimes())
- mBfIRBuilder->CreateLifetimeStart(allocaInst);
- mBfIRBuilder->CreateStore(scopeArg, allocaInst);
- deferredCallEntry->mScopeArgs[paramIdx] = allocaInst;
- if (WantsLifetimes())
- scopeData->mDeferredLifetimeEnds.push_back(allocaInst);
- }
- deferredCallEntry->mArgsNeedLoad = true;
- }
- }
- scopeData->mDeferredCallEntries.PushFront(deferredCallEntry);
- return true;
- }
- bool isLooped = mCurMethodState->mCurScope->IsLooped(scopeData);
- BfDeferredCallEntry* listEntry = NULL;
- if (!scopeData->mDeferredCallEntries.IsEmpty())
- {
- listEntry = scopeData->mDeferredCallEntries.mHead;
- if (!listEntry->IsDynList())
- listEntry = NULL;
- }
- auto deferredCallEntryType = ResolveTypeDef(mCompiler->mDeferredCallTypeDef);
- AddDependency(deferredCallEntryType, mCurTypeInstance, BfDependencyMap::DependencyFlag_Allocates);
- mBfIRBuilder->PopulateType(deferredCallEntryType);
- auto deferredCallEntryTypePtr = CreatePointerType(deferredCallEntryType);
-
- UpdateSrcPos(mCurMethodInstance->mMethodDef->GetRefNode(), BfSrcPosFlag_NoSetDebugLoc);
-
- if (listEntry == NULL)
- {
- listEntry = new BfDeferredCallEntry();
- if (!mBfIRBuilder->mIgnoreWrites)
- {
- listEntry->mDynCallTail = CreateAlloca(deferredCallEntryTypePtr, false, "deferredCallTail");
- if (WantsLifetimes())
- scopeData->mDeferredLifetimeEnds.push_back(listEntry->mDynCallTail);
- auto prevInsertBlock = mBfIRBuilder->GetInsertBlock();
- mBfIRBuilder->SaveDebugLocation();
- mBfIRBuilder->SetInsertPointAtStart(mCurMethodState->mIRInitBlock);
- auto scopeHead = &mCurMethodState->mHeadScope;
- mBfIRBuilder->SetCurrentDebugLocation(mCurFilePosition.mCurLine + 1, 0, scopeHead->mDIScope, BfIRMDNode());
- if (WantsLifetimes())
- mBfIRBuilder->CreateLifetimeStart(listEntry->mDynCallTail);
- auto storeInst = mBfIRBuilder->CreateStore(GetDefaultValue(deferredCallEntryTypePtr), listEntry->mDynCallTail);
- mBfIRBuilder->ClearDebugLocation(storeInst);
- if (WantsDebugInfo())
- {
- auto deferredCallEntryType = ResolveTypeDef(mCompiler->mDeferredCallTypeDef);
- auto deferredCallEntryTypePtr = CreatePointerType(deferredCallEntryType);
- String varName = StrFormat("__deferred%d", mCurMethodState->mDeferredLoopListCount);
- mBfIRBuilder->SetInsertPoint(mCurMethodState->mIRInitBlock);
- mBfIRBuilder->SetCurrentDebugLocation(mCurFilePosition.mCurLine + 1, 0, scopeHead->mDIScope, BfIRMDNode());
- mBfIRBuilder->CreateStatementStart();
- //TODO: Make this work for LLVM - we need a proper debug location
- //if (IsTargetingBeefBackend())
- {
- auto diVariable = mBfIRBuilder->DbgCreateAutoVariable(scopeHead->mDIScope, varName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, mBfIRBuilder->DbgGetType(deferredCallEntryTypePtr));
- mBfIRBuilder->DbgInsertDeclare(listEntry->mDynCallTail, diVariable);
- }
- mCurMethodState->mDeferredLoopListCount++;
- }
- mBfIRBuilder->SetInsertPoint(prevInsertBlock);
- mBfIRBuilder->RestoreDebugLocation();
- }
- mCurMethodState->mCurScope->ClearHandlers(scopeData);
- scopeData->mDeferredCallEntries.PushFront(listEntry);
- }
- BfIRValue deferredAlloca;
- SizedArray<BfType*, 4> types;
- SizedArray<String, 8> memberNames;
- int instAlign = 1;
- int dataPos = 0;
- int instSize = 0;
- Array<int> memberPositions;
- String typeName;
- BfDeferredMethodCallData* deferredMethodCallData = NULL;
- if (deferredCallEntry->mDeferredBlock != NULL)
- {
- int blockId = -deferredCallEntry->mDeferredBlock->GetSrcStart();
- auto deferType = deferredCallEntryType;
- BfIRType deferIRType;
- auto int64Type = GetPrimitiveType(BfTypeCode_Int64);
-
- types.push_back(int64Type);
- memberNames.push_back("__methodId");
- types.push_back(deferredCallEntryTypePtr);
- memberNames.push_back("__next");
- for (auto& capture : deferredCallEntry->mCaptures)
- {
- BfType* type = capture.mValue.mType;
- types.push_back(type);
- memberNames.push_back(capture.mName);
- }
- SizedArray<BfIRType, 4> llvmTypes;
- SizedArray<BfIRMDNode, 8> diFieldTypes;
-
- typeName = StrFormat("_BF_DeferredData_%s", BfTypeUtils::HashEncode64(blockId).c_str());
-
- auto valueType = ResolveTypeDef(mCompiler->mValueTypeTypeDef);
- llvmTypes.push_back(mBfIRBuilder->MapType(valueType));
- //int dataPos = 0;
- for (int i = 0; i < (int)memberNames.size(); i++)
- {
- auto type = types[i];
- auto memberName = memberNames[i];
- if (!type->IsValuelessType())
- {
- llvmTypes.push_back(mBfIRBuilder->MapType(type));
- instAlign = BF_MAX(instAlign, (int)type->mAlign);
- int alignSize = (int)type->mAlign;
- int dataSize = type->mSize;
- if (alignSize > 1)
- dataPos = (dataPos + (alignSize - 1)) & ~(alignSize - 1);
- memberPositions.push_back(dataPos);
- dataPos += type->mSize;
- }
- }
- instSize = dataPos;
- deferIRType = mBfIRBuilder->CreateStructType(typeName);
- mBfIRBuilder->StructSetBody(deferIRType, llvmTypes, false);
-
- auto prevInsertPoint = mBfIRBuilder->GetInsertBlock();
- if (!isLooped)
- mBfIRBuilder->SetInsertPoint(mCurMethodState->mIRHeadBlock);
- deferredAlloca = mBfIRBuilder->CreateAlloca(deferIRType);
- mBfIRBuilder->SetAllocaAlignment(deferredAlloca, instAlign);
- mBfIRBuilder->SetAllocaNoChkStkHint(deferredAlloca);
- if (!isLooped)
- mBfIRBuilder->SetInsertPoint(prevInsertPoint);
- auto gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, 2); // mNext
- auto prevVal = mBfIRBuilder->CreateLoad(listEntry->mDynCallTail);
- mBfIRBuilder->CreateStore(prevVal, gepInstance);
- gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, 1); // mMethodId
- mBfIRBuilder->CreateStore(GetConstValue64(blockId), gepInstance);
- if (!deferredCallEntry->mCaptures.empty())
- {
- int dataIdx = 3;
- for (int captureIdx = 0; captureIdx < (int)deferredCallEntry->mCaptures.size(); captureIdx++)
- {
- auto& capture = deferredCallEntry->mCaptures[captureIdx];
- if (!capture.mValue.mType->IsValuelessType())
- {
- auto gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, dataIdx);
- mBfIRBuilder->CreateStore(capture.mValue.mValue, gepInstance);
- dataIdx++;
- }
- }
- }
- mBfIRBuilder->CreateStore(mBfIRBuilder->CreateBitCast(deferredAlloca, mBfIRBuilder->MapType(deferredCallEntryTypePtr)), listEntry->mDynCallTail);
- deferredCallEntry->mDeferredAlloca = deferredAlloca;
- listEntry->mDynList.PushFront(deferredCallEntry);
- }
- else
- {
- auto& llvmArgs = deferredCallEntry->mScopeArgs;
- auto moduleMethodInstance = deferredCallEntry->mModuleMethodInstance;
- auto methodInstance = moduleMethodInstance.mMethodInstance;
- auto methodDef = methodInstance->mMethodDef;
- auto owningType = methodInstance->mMethodInstanceGroup->mOwner;
- auto voidType = GetPrimitiveType(BfTypeCode_None);
- auto voidPtrType = CreatePointerType(voidType);
- BfDeferredMethodCallData** deferredMethodCallDataPtr = NULL;
- if (mDeferredMethodCallData.TryGetValue(methodInstance, &deferredMethodCallDataPtr))
- {
- deferredMethodCallData = *deferredMethodCallDataPtr;
- }
- else
- {
- deferredMethodCallData = new BfDeferredMethodCallData();
- mDeferredMethodCallData[methodInstance] = deferredMethodCallData;
- if (mDeferredMethodIds.Add(methodInstance->mIdHash))
- {
- deferredMethodCallData->mMethodId = methodInstance->mIdHash;
- }
- else
- {
- // Try to create a hash that will hopefully be globally unique. If it isn't then it just means we will end up with two
- // conflicting debug info definitions for the same name, which is not an error but may cause a debugger to show the
- // wrong one to the user. Does not affect runtime correctness.
- int64 checkId = Hash64(mModuleName.c_str(), (int)mModuleName.length(), mDeferredMethodIds.size());
- while (true)
- {
- if (!mDeferredMethodIds.Contains(checkId))
- break;
- checkId += 0x100;
- }
- mDeferredMethodIds.Add(checkId);
- deferredMethodCallData->mMethodId = checkId;
- }
- auto int64Type = GetPrimitiveType(BfTypeCode_Int64);
- auto methodDef = moduleMethodInstance.mMethodInstance->mMethodDef;
- auto thisType = moduleMethodInstance.mMethodInstance->mMethodInstanceGroup->mOwner;
-
- types.push_back(int64Type);
- memberNames.push_back("__methodId");
- types.push_back(deferredCallEntryTypePtr);
- memberNames.push_back("__next");
- if (!methodDef->mIsStatic)
- {
- types.push_back(thisType);
- memberNames.push_back("__this");
- }
- for (int paramIdx = 0; paramIdx < (int)methodInstance->GetParamCount(); paramIdx++)
- {
- if (methodInstance->IsParamSkipped(paramIdx))
- paramIdx++;
- types.push_back(methodInstance->GetParamType(paramIdx));
- memberNames.push_back(methodInstance->GetParamName(paramIdx));
- }
- SizedArray<BfIRType, 4> llvmTypes;
- //String typeName;
- typeName += StrFormat("_BF_DeferredData_%s", BfTypeUtils::HashEncode64(deferredMethodCallData->mMethodId).c_str());
- BfLogSysM("Building type: %s from methodInstance:%p\n", typeName.c_str(), methodInstance);
- //Array<int> memberPositions;
- //int instAlign = 1;
- //int dataPos = 0;
- BF_ASSERT(types.size() == memberNames.size());
- for (int i = 0; i < (int)types.size(); i++)
- {
- auto type = types[i];
- auto memberName = memberNames[i];
- if (!type->IsValuelessType())
- {
- llvmTypes.push_back(mBfIRBuilder->MapType(type));
- instAlign = BF_MAX(instAlign, (int)type->mAlign);
- int alignSize = (int)type->mAlign;
- int dataSize = type->mSize;
- if (alignSize > 1)
- dataPos = (dataPos + (alignSize - 1)) & ~(alignSize - 1);
- memberPositions.push_back(dataPos);
- dataPos += type->mSize;
- }
- }
- instSize = dataPos;
- deferredMethodCallData->mAlign = instAlign;
- deferredMethodCallData->mSize = instSize;
- deferredMethodCallData->mDeferType = mBfIRBuilder->CreateStructType(typeName);
- mBfIRBuilder->StructSetBody(deferredMethodCallData->mDeferType, llvmTypes, false);
- deferredMethodCallData->mDeferTypePtr = mBfIRBuilder->GetPointerTo(deferredMethodCallData->mDeferType);
- }
- auto deferType = deferredMethodCallData->mDeferType;
- auto prevInsertPoint = mBfIRBuilder->GetInsertBlock();
- if (!isLooped)
- mBfIRBuilder->SetInsertPoint(mCurMethodState->mIRHeadBlock);
- deferredAlloca = mBfIRBuilder->CreateAlloca(BfIRType(deferredMethodCallData->mDeferType));
- mBfIRBuilder->ClearDebugLocation(deferredAlloca);
- mBfIRBuilder->SetAllocaAlignment(deferredAlloca, deferredMethodCallData->mAlign);
- mBfIRBuilder->SetAllocaNoChkStkHint(deferredAlloca);
- if (!isLooped)
- mBfIRBuilder->SetInsertPoint(prevInsertPoint);
- auto gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, 1);
- auto prevVal = mBfIRBuilder->CreateLoad(listEntry->mDynCallTail);
- mBfIRBuilder->CreateStore(prevVal, gepInstance);
- gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, 0);
- mBfIRBuilder->CreateStore(GetConstValue64(deferredMethodCallData->mMethodId), gepInstance);
- int dataIdx = 2;
- int argIdx = 0;
- if (!methodDef->mIsStatic)
- {
- gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, 2);
- if (owningType->IsStruct())
- {
- if ((!methodDef->mIsMutating) && (owningType->IsSplattable()))
- {
- BfTypedValue splatVal(llvmArgs[0], owningType, BfTypedValueKind_ThisSplatHead);
- BfTypedValue aggVal = AggregateSplat(splatVal, &llvmArgs[0]);
- aggVal = LoadValue(aggVal);
- mBfIRBuilder->CreateStore(aggVal.mValue, gepInstance);
- BfTypeUtils::SplatIterate([&](BfType* checkType) { argIdx++; }, owningType);
- }
- else
- {
- auto thisArg = mBfIRBuilder->CreateLoad(llvmArgs[0]);
- mBfIRBuilder->CreateStore(thisArg, gepInstance);
- argIdx++;
- }
- }
- else
- {
- mBfIRBuilder->CreateStore(llvmArgs[0], gepInstance);
- argIdx++;
- }
- dataIdx++;
- }
- for (int paramIdx = 0; paramIdx < (int)methodInstance->GetParamCount(); paramIdx++, dataIdx++)
- {
- if (methodInstance->IsParamSkipped(paramIdx))
- paramIdx++;
- auto paramType = methodInstance->GetParamType(paramIdx);
- bool paramIsSplat = methodInstance->GetParamIsSplat(paramIdx);
- gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredAlloca, 0, dataIdx/*, methodDef->mParams[paramIdx]->mName*/);
- if (paramType->IsStruct())
- {
- if (paramIsSplat)
- {
- BfTypedValue splatVal(llvmArgs[argIdx], paramType, BfTypedValueKind_SplatHead);
- BfTypedValue aggVal = AggregateSplat(splatVal, &llvmArgs[argIdx]);
- aggVal = LoadValue(aggVal);
- mBfIRBuilder->CreateStore(aggVal.mValue, gepInstance);
- BfTypeUtils::SplatIterate([&](BfType* checkType) { argIdx++; }, paramType);
- }
- else
- {
- auto val = mBfIRBuilder->CreateLoad(llvmArgs[argIdx]);
- mBfIRBuilder->CreateStore(val, gepInstance);
- argIdx++;
- }
- }
- else
- {
- mBfIRBuilder->CreateStore(llvmArgs[argIdx], gepInstance);
- argIdx++;
- }
- }
- mBfIRBuilder->CreateStore(mBfIRBuilder->CreateBitCast(deferredAlloca, mBfIRBuilder->MapType(deferredCallEntryTypePtr)), listEntry->mDynCallTail);
-
- deferredCallEntry->mDeferredAlloca = deferredAlloca;
- listEntry->mDynList.PushFront(deferredCallEntry);
- }
- if ((mBfIRBuilder->DbgHasInfo()) && (mCompiler->mOptions.mEmitDebugInfo) && (mCurMethodState->mCurScope->mDIScope))
- {
- auto int64Type = GetPrimitiveType(BfTypeCode_Int64);
- auto moduleMethodInstance = deferredCallEntry->mModuleMethodInstance;
- auto methodInstance = moduleMethodInstance.mMethodInstance;
- BfIRMDNode deferDIType;
- if ((deferredMethodCallData != NULL) && (deferredMethodCallData->mDeferDIType))
- {
- deferDIType = deferredMethodCallData->mDeferDIType;
- }
- else
- {
- BfIRMDNode diForwardDecl = NULL;
- SizedArray<BfIRMDNode, 8> diFieldTypes;
- if ((mBfIRBuilder->DbgHasInfo()) && (mHasFullDebugInfo))
- {
- String dbgTypeName;
- if (mCompiler->mOptions.IsCodeView())
- dbgTypeName += "_bf::";
- dbgTypeName += typeName;
- diForwardDecl = mBfIRBuilder->DbgCreateReplaceableCompositeType(llvm::dwarf::DW_TAG_structure_type, dbgTypeName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mFileInstance->mDIFile,
- mCurFilePosition.mCurLine, instSize * 8, instAlign * 8);
- if (methodInstance != NULL)
- {
- // We make a fake member to get inserted into the DbgModule data so we can show what method this deferred call goes to
- StringT<128> mangledName;
- BfMangler::Mangle(mangledName, mCompiler->GetMangleKind(), methodInstance);
- auto memberType = mBfIRBuilder->DbgCreateMemberType(diForwardDecl, mangledName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine,
- 0, 0, -1, 0, mBfIRBuilder->DbgGetType(int64Type));
- diFieldTypes.push_back(memberType);
- }
- }
- for (int i = 0; i < (int)types.size(); i++)
- {
- auto type = types[i];
- auto memberName = memberNames[i];
- if ((mBfIRBuilder->DbgHasInfo()) && (mHasFullDebugInfo))
- {
- int memberFlags = 0;
- auto memberType = mBfIRBuilder->DbgCreateMemberType(diForwardDecl, memberName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine,
- type->mSize * 8, type->mAlign * 8, memberPositions[i] * 8, memberFlags, mBfIRBuilder->DbgGetType(type));
- diFieldTypes.push_back(memberType);
- }
- }
- int diFlags = 0;
- mBfIRBuilder->DbgMakePermanent(diForwardDecl, BfIRMDNode(), diFieldTypes);
- deferDIType = mBfIRBuilder->DbgCreatePointerType(diForwardDecl);
- if (deferredMethodCallData != NULL)
- deferredMethodCallData->mDeferDIType = deferDIType;
- }
-
- // We don't actually want to see this, and it doesn't emit properly in LLVM CodeView anyway - it only accepts static allocs,
- // not dynamic allocas
- String varName = StrFormat("$__deferredCall_%d", mCurMethodState->mDeferredLoopListEntryCount);
- mCurMethodState->mDeferredLoopListEntryCount++;
- auto diVariable = mBfIRBuilder->DbgCreateAutoVariable(mCurMethodState->mCurScope->mDIScope,
- varName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, deferDIType);
- mBfIRBuilder->DbgInsertDeclare(deferredAlloca, diVariable);
- }
- return true;
- }
- void BfModule::AddDeferredBlock(BfBlock* block, BfScopeData* scopeData, Array<BfDeferredCapture>* captures)
- {
- BfDeferredCallEntry* deferredCallEntry = new BfDeferredCallEntry();
- deferredCallEntry->mDeferredBlock = block;
- if (captures != NULL)
- deferredCallEntry->mCaptures = *captures;
- AddDeferredCallEntry(deferredCallEntry, scopeData); \
- }
- BfDeferredCallEntry* BfModule::AddDeferredCall(const BfModuleMethodInstance& moduleMethodInstance, SizedArrayImpl<BfIRValue>& llvmArgs, BfScopeData* scopeData, BfAstNode* srcNode, bool bypassVirtual, bool doNullCheck)
- {
- BfDeferredCallEntry* deferredCallEntry = new BfDeferredCallEntry();
- BF_ASSERT(moduleMethodInstance);
- deferredCallEntry->mModuleMethodInstance = moduleMethodInstance;
- for (auto arg : llvmArgs)
- {
- deferredCallEntry->mScopeArgs.push_back(arg);
- }
- deferredCallEntry->mSrcNode = srcNode;
- deferredCallEntry->mBypassVirtual = bypassVirtual;
- deferredCallEntry->mDoNullCheck = doNullCheck;
- if (!AddDeferredCallEntry(deferredCallEntry, scopeData))
- return NULL;
- return deferredCallEntry;
- }
- void BfModule::EmitDeferredCall(BfModuleMethodInstance moduleMethodInstance, SizedArrayImpl<BfIRValue>& llvmArgs, BfDeferredBlockFlags flags)
- {
- if (moduleMethodInstance.mMethodInstance == mContext->mValueTypeDeinitSentinel)
- {
- BF_ASSERT(llvmArgs.size() == 3);
- auto sizeConstant = mBfIRBuilder->GetConstant(llvmArgs[1]);
- int clearSize = BF_MIN(sizeConstant->mInt32, 32);
-
- auto alignConstant = mBfIRBuilder->GetConstant(llvmArgs[2]);
- int clearAlign = alignConstant->mInt32;
- mBfIRBuilder->CreateMemSet(llvmArgs[0], GetConstValue8(0xDD), GetConstValue(clearSize), clearAlign);
- return;
- }
- auto methodInstance = moduleMethodInstance.mMethodInstance;
- auto methodOwner = methodInstance->mMethodInstanceGroup->mOwner;
- bool isDtor = methodInstance->mMethodDef->mMethodType == BfMethodType_Dtor;
- bool isScopeDtor = isDtor && ((flags & BfDeferredBlockFlag_BypassVirtual) != 0);
- if ((isDtor) && (methodInstance->GetParamCount() != 0))
- {
- // Dtor declared with params
- AssertErrorState();
- return;
- }
- BfIRBlock nullLabel;
- BfIRBlock notNullLabel;
- if ((flags & BfDeferredBlockFlag_DoNullChecks) != 0)
- {
- nullLabel = mBfIRBuilder->CreateBlock("deferred.isNull");
- notNullLabel = mBfIRBuilder->CreateBlock("deferred.notNull");
- auto notNullVal = mBfIRBuilder->CreateIsNotNull(llvmArgs[0]);
- mBfIRBuilder->CreateCondBr(notNullVal, notNullLabel, nullLabel);
- mBfIRBuilder->AddBlock(notNullLabel);
- mBfIRBuilder->SetInsertPoint(notNullLabel);
- }
- bool skipAccessCheck = false;
- if ((flags & BfDeferredBlockFlag_SkipObjectAccessCheck) != 0)
- skipAccessCheck = true;
- if ((!methodInstance->mMethodDef->mIsStatic) && (methodOwner->IsObjectOrInterface()) && (!isScopeDtor) &&
- (!skipAccessCheck))
- {
- EmitObjectAccessCheck(BfTypedValue(llvmArgs[0], methodOwner));
- }
- BfExprEvaluator expressionEvaluator(this);
- expressionEvaluator.CreateCall(moduleMethodInstance.mMethodInstance, moduleMethodInstance.mFunc, ((flags & BfDeferredBlockFlag_BypassVirtual) != 0), llvmArgs);
- if ((flags & BfDeferredBlockFlag_DoNullChecks) != 0)
- {
- mBfIRBuilder->CreateBr(nullLabel);
- mBfIRBuilder->AddBlock(nullLabel);
- mBfIRBuilder->SetInsertPoint(nullLabel);
- if (!mBfIRBuilder->mIgnoreWrites)
- {
- if ((flags & BfDeferredBlockFlag_MoveNewBlocksToEnd) != 0)
- {
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(notNullLabel);
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(nullLabel);
- }
- }
- }
- }
- void BfModule::EmitDeferredCall(BfDeferredCallEntry& deferredCallEntry)
- {
- if ((mCompiler->mIsResolveOnly) && (deferredCallEntry.mHandlerCount > 0))
- {
- // We only want to process deferred blocks once, otherwise it could significantly slow down autocompletion
- return;
- }
- deferredCallEntry.mHandlerCount++;
- if (deferredCallEntry.IsDynList())
- {
- EmitDeferredCallProcessor(deferredCallEntry.mDynList, deferredCallEntry.mDynCallTail);
- return;
- }
- if (deferredCallEntry.mDeferredBlock != NULL)
- {
- // Only show warnings on the first pass
- // For errors, show on the first pass OR as long as we haven't gotten any errors within this method. I'm not sure if there's a case
- // where the first emission succeeds but a subsequent one would fail, but we leave this logic to handle that possibility
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, (deferredCallEntry.mHandlerCount > 1) && (mCurMethodInstance->mHasFailed));
- SetAndRestoreValue<bool> prevIgnoreWarnings(mIgnoreWarnings, (deferredCallEntry.mHandlerCount > 1));
- BfScopeData scopeData;
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- for (auto& capture : deferredCallEntry.mCaptures)
- {
- BfLocalVariable* localVar = new BfLocalVariable();
- localVar->mIsReadOnly = true;
- localVar->mIsAssigned = true;
- localVar->mReadFromId = 0;
- localVar->mName = capture.mName;
- localVar->mValue = capture.mValue.mValue;
- localVar->mResolvedType = capture.mValue.mType;
- if ((mBfIRBuilder->DbgHasInfo()) && (!localVar->mResolvedType->IsValuelessType()))
- {
- auto addr = CreateAlloca(localVar->mResolvedType);
- mBfIRBuilder->CreateAlignedStore(localVar->mValue, addr, localVar->mResolvedType->mAlign);
- localVar->mAddr = addr;
- }
- AddLocalVariableDef(localVar, true);
- }
-
- VisitEmbeddedStatement(deferredCallEntry.mDeferredBlock, NULL, BfEmbeddedStatementFlags_IsDeferredBlock);
- RestoreScopeState();
- return;
- }
- auto moduleMethodInstance = deferredCallEntry.mModuleMethodInstance;
- auto args = deferredCallEntry.mScopeArgs;
- if (deferredCallEntry.mArgsNeedLoad)
- {
- for (auto& arg : args)
- {
- if (!arg.IsConst())
- arg = mBfIRBuilder->CreateLoad(arg);
- }
- }
- if (deferredCallEntry.mCastThis)
- {
- args[0] = mBfIRBuilder->CreateBitCast(args[0], mBfIRBuilder->MapTypeInstPtr(deferredCallEntry.mModuleMethodInstance.mMethodInstance->GetOwner()));
- }
- BfDeferredBlockFlags flags = BfDeferredBlockFlag_None;
- if (deferredCallEntry.mBypassVirtual)
- flags = (BfDeferredBlockFlags)(flags | BfDeferredBlockFlag_BypassVirtual);
- if (deferredCallEntry.mDoNullCheck)
- flags = (BfDeferredBlockFlags)(flags | BfDeferredBlockFlag_DoNullChecks | BfDeferredBlockFlag_SkipObjectAccessCheck | BfDeferredBlockFlag_MoveNewBlocksToEnd);
- EmitDeferredCall(deferredCallEntry.mModuleMethodInstance, args, flags);
- }
- void BfModule::EmitDeferredCallProcessor(SLIList<BfDeferredCallEntry*>& callEntries, BfIRValue callTail)
- {
- int64 collisionId = 0;
- struct _CallInfo
- {
- BfModuleMethodInstance mModuleMethodInstance;
- bool mBypassVirtual;
- };
- //typedef std::map<int64, _CallInfo> MapType;
- //MapType methodInstanceMap;
- Dictionary<int64, _CallInfo> methodInstanceMap;
- int blockCount = 0;
- HashSet<BfMethodInstance*> nullCheckMethodSet;
- BfDeferredCallEntry* deferredCallEntry = callEntries.mHead;
- while (deferredCallEntry != NULL)
- {
- BfModuleMethodInstance moduleMethodInstance = deferredCallEntry->mModuleMethodInstance;
- int64 methodId = 0;
- if (moduleMethodInstance.mMethodInstance != NULL)
- {
- int64 idHash = moduleMethodInstance.mMethodInstance->mIdHash;
- auto deferredMethodCallData = mDeferredMethodCallData[moduleMethodInstance.mMethodInstance];
- BF_ASSERT(deferredMethodCallData->mMethodId != 0);
- //methodInstanceMap[deferredMethodCallData->mMethodId] = moduleMethodInstance;
- _CallInfo* callInfo = NULL;
- if (methodInstanceMap.TryAdd(deferredMethodCallData->mMethodId, NULL, &callInfo))
- {
- callInfo->mModuleMethodInstance = moduleMethodInstance;
- callInfo->mBypassVirtual = deferredCallEntry->mBypassVirtual;
- }
- else
- {
- // Only bypass virtual if ALL these calls are devirtualized
- callInfo->mBypassVirtual &= deferredCallEntry->mBypassVirtual;
- }
- }
- else
- blockCount++;
- if (deferredCallEntry->mDoNullCheck)
- nullCheckMethodSet.Add(deferredCallEntry->mModuleMethodInstance.mMethodInstance);
- deferredCallEntry = deferredCallEntry->mNext;
- }
- bool moveBlocks = mCurMethodState->mCurScope != mCurMethodState->mTailScope;
- auto valueScopeStart = ValueScopeStart();
- if (valueScopeStart)
- mBfIRBuilder->SetName(valueScopeStart, "deferredScopeVal");
- BfIRBlock condBB = mBfIRBuilder->CreateBlock("deferCall.cond", true);
- if (moveBlocks)
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(condBB);
- mBfIRBuilder->CreateBr(condBB);
- auto deferredCallEntryType = ResolveTypeDef(mCompiler->mDeferredCallTypeDef);
- auto deferredCallEntryTypePtr = CreatePointerType(deferredCallEntryType);
- BfIRBlock bodyBB = mBfIRBuilder->CreateBlock("deferCall.body");
- if (moveBlocks)
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(bodyBB);
- BfIRBlock endBB = mBfIRBuilder->CreateBlock("deferCall.end");
- if (moveBlocks)
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(endBB);
- BfIRBlock exitBB = endBB;
- BfIRValue deferredCallTail;
- mBfIRBuilder->SetInsertPoint(condBB);
- deferredCallTail = mBfIRBuilder->CreateLoad(callTail);
- auto isNotNull = mBfIRBuilder->CreateIsNotNull(deferredCallTail);
- ValueScopeEnd(valueScopeStart);
- mBfIRBuilder->CreateCondBr(isNotNull, bodyBB, exitBB);
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
-
- BfIRValue switchInst;
- bool wantsSwitch = ((int)methodInstanceMap.size() + blockCount) > 1;
- if (blockCount > 0)
- {
- // A block may embed a switch so we need a switch whenever we have blocks
- wantsSwitch = true;
- }
- if (mCurMethodState->mCancelledDeferredCall)
- wantsSwitch = true;
- if (wantsSwitch)
- {
- if (IsTargetingBeefBackend())
- deferredCallTail = mBfIRBuilder->CreateLoad(callTail);
- auto idPtr = mBfIRBuilder->CreateInBoundsGEP(deferredCallTail, 0, 1); // mMethodId
- auto id = mBfIRBuilder->CreateLoad(idPtr);
- switchInst = mBfIRBuilder->CreateSwitch(id, exitBB, (int)methodInstanceMap.size());
- ValueScopeEnd(valueScopeStart);
- }
- BfDeferredCallEntry* prevHead = callEntries.mHead;
- BfDeferredCallEntry* prevCallEntry = NULL;
- HashSet<BfDeferredCallEntry*> handledSet;
- deferredCallEntry = callEntries.mHead;
- while (deferredCallEntry != NULL)
- {
- auto block = deferredCallEntry->mDeferredBlock;
- if (block == NULL)
- {
- deferredCallEntry = deferredCallEntry->mNext;
- continue;
- }
- int blockId = -block->GetSrcStart();
- //auto itr = handledSet.insert(deferredCallEntry);
- //if (!itr.second)
- if (!handledSet.Add(deferredCallEntry))
- {
- // Already handled, can happen if we defer again within the block
- deferredCallEntry = deferredCallEntry->mNext;
- continue;
- }
- if (switchInst)
- {
- String caseName = StrFormat("deferCall.%s", BfTypeUtils::HashEncode64(blockId).c_str());
- auto caseBB = mBfIRBuilder->CreateBlock(caseName, true);
- if (moveBlocks)
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(caseBB);
- mBfIRBuilder->AddSwitchCase(switchInst, GetConstValue64(blockId), caseBB);
- mBfIRBuilder->SetInsertPoint(caseBB);
- }
- // Update .mDeferredAlloca to use the deferredCallTail
- if (IsTargetingBeefBackend())
- deferredCallTail = mBfIRBuilder->CreateLoad(callTail);
- auto nextPtr = mBfIRBuilder->CreateInBoundsGEP(deferredCallTail, 0, 2); // mNext
- auto next = mBfIRBuilder->CreateLoad(nextPtr);
- mBfIRBuilder->CreateStore(next, callTail);
- deferredCallEntry->mDeferredAlloca = mBfIRBuilder->CreateBitCast(deferredCallTail, mBfIRBuilder->GetType(deferredCallEntry->mDeferredAlloca));
- int dataIdx = 3;
- // Update .mCaptures to contain the stored values at the time the defer occurred
- for (int captureIdx = 0; captureIdx < (int)deferredCallEntry->mCaptures.size(); captureIdx++)
- {
- auto& capture = deferredCallEntry->mCaptures[captureIdx];
- if (!capture.mValue.mType->IsValuelessType())
- {
- auto gepInstance = mBfIRBuilder->CreateInBoundsGEP(deferredCallEntry->mDeferredAlloca, 0, dataIdx);
- capture.mValue.mValue = mBfIRBuilder->CreateLoad(gepInstance);
- dataIdx++;
- }
- }
- auto prevHead = callEntries.mHead;
- EmitDeferredCall(*deferredCallEntry);
- ValueScopeEnd(valueScopeStart);
- mBfIRBuilder->CreateBr(condBB);
- if (prevHead != callEntries.mHead)
- {
- // The list changed, start over and ignore anything we've already handled
- deferredCallEntry = callEntries.mHead;
- }
- else
- deferredCallEntry = deferredCallEntry->mNext;
- }
- // Blocks may have added new method types, so rebuild map
- if (blockCount > 0)
- {
- deferredCallEntry = callEntries.mHead;
- while (deferredCallEntry != NULL)
- {
- BfModuleMethodInstance moduleMethodInstance = deferredCallEntry->mModuleMethodInstance;
- if (moduleMethodInstance.mMethodInstance != NULL)
- {
- auto deferredMethodCallData = mDeferredMethodCallData[moduleMethodInstance.mMethodInstance];
- //methodInstanceMap.insert(MapType::value_type(deferredMethodCallData->mMethodId, moduleMethodInstance));
- _CallInfo* callInfo = NULL;
- if (methodInstanceMap.TryAdd(deferredMethodCallData->mMethodId, NULL, &callInfo))
- {
- callInfo->mModuleMethodInstance = moduleMethodInstance;
- callInfo->mBypassVirtual = deferredCallEntry->mBypassVirtual;
- }
- }
- deferredCallEntry = deferredCallEntry->mNext;
- }
- }
- BfExprEvaluator exprEvaluator(this);
- //for (auto itr = methodInstanceMap.begin(); itr != methodInstanceMap.end(); ++itr)
- for (auto& callInfoKV : methodInstanceMap)
- {
- auto moduleMethodInstance = callInfoKV.mValue.mModuleMethodInstance;
- bool bypassVirtual = callInfoKV.mValue.mBypassVirtual;
- auto methodInstance = moduleMethodInstance.mMethodInstance;
- auto methodDef = methodInstance->mMethodDef;
- auto methodOwner = methodInstance->mMethodInstanceGroup->mOwner;
- BfIRValue deferredCallInst = deferredCallTail;
- auto deferredMethodCallData = mDeferredMethodCallData[methodInstance];
- int64 methodId = deferredMethodCallData->mMethodId;
- if (switchInst)
- {
- String caseName = StrFormat("deferCall.%s", BfTypeUtils::HashEncode64(methodId).c_str());
- auto caseBB = mBfIRBuilder->CreateBlock(caseName, true);
- if (moveBlocks)
- mCurMethodState->mCurScope->mAtEndBlocks.push_back(caseBB);
- mBfIRBuilder->AddSwitchCase(switchInst, GetConstValue64(methodId), caseBB);
- mBfIRBuilder->SetInsertPoint(caseBB);
- }
-
- if (IsTargetingBeefBackend())
- deferredCallTail = mBfIRBuilder->CreateLoad(callTail);
- auto nextPtr = mBfIRBuilder->CreateInBoundsGEP(deferredCallTail, 0, 2); // mNext
- auto next = mBfIRBuilder->CreateLoad(nextPtr);
- mBfIRBuilder->CreateStore(next, callTail);
- deferredCallInst = mBfIRBuilder->CreateBitCast(deferredCallTail, deferredMethodCallData->mDeferTypePtr);
- int paramIdx = 0;
- if (!methodDef->mIsStatic)
- paramIdx = -1;
- SizedArray<BfIRValue, 8> llvmArgs;
- for (int argIdx = 0; paramIdx < methodInstance->GetParamCount(); argIdx++, paramIdx++)
- {
- auto argPtr = mBfIRBuilder->CreateInBoundsGEP(deferredCallInst, 0, argIdx + 2);
- bool isStruct = false;
- bool doSplat = methodInstance->GetParamIsSplat(paramIdx);;
- BfTypedValue typedVal;
- if (paramIdx == -1)
- {
- typedVal = BfTypedValue(argPtr, methodOwner, true);
- }
- else
- {
- auto paramType = methodInstance->GetParamType(paramIdx);
- typedVal = BfTypedValue(argPtr, paramType, true);
- }
- if (doSplat)
- {
- exprEvaluator.SplatArgs(typedVal, llvmArgs);
- continue;
- }
- if ((argIdx == 0) && (!methodDef->mIsStatic))
- {
- // 'this'
- isStruct = methodOwner->IsStruct();
- }
- else
- {
- while (methodInstance->IsParamSkipped(paramIdx))
- paramIdx++;
- if (paramIdx >= methodInstance->GetParamCount())
- break;
- auto paramType = methodInstance->GetParamType(paramIdx);
- isStruct = paramType->IsStruct();
- }
- if (isStruct)
- {
- llvmArgs.push_back(argPtr);
- }
- else
- {
- auto arg = mBfIRBuilder->CreateLoad(argPtr);
- llvmArgs.push_back(arg);
- }
- }
- BfDeferredBlockFlags flags = BfDeferredBlockFlag_None;
- if (moveBlocks)
- flags = (BfDeferredBlockFlags)(flags | BfDeferredBlockFlag_MoveNewBlocksToEnd);
- if (nullCheckMethodSet.Contains(moduleMethodInstance.mMethodInstance))
- flags = (BfDeferredBlockFlags)(flags | BfDeferredBlockFlag_DoNullChecks | BfDeferredBlockFlag_SkipObjectAccessCheck);
- if (bypassVirtual)
- flags = (BfDeferredBlockFlags)(flags | BfDeferredBlockFlag_BypassVirtual);
- EmitDeferredCall(moduleMethodInstance, llvmArgs, flags);
- ValueScopeEnd(valueScopeStart);
- mBfIRBuilder->CreateBr(condBB);
- }
-
- if (endBB)
- {
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- }
- }
- void BfModule::TryInitVar(BfAstNode* checkNode, BfLocalVariable* localVar, BfTypedValue initValue, BfTypedValue& checkResult)
- {
- BF_ASSERT(!localVar->mAddr);
- localVar->mAddr = AllocLocalVariable(localVar->mResolvedType, localVar->mName);
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- auto varType = localVar->mResolvedType;
- AddDependency(varType, mCurTypeInstance, BfDependencyMap::DependencyFlag_ExprTypeReference);
- if (!initValue)
- {
- AssertErrorState();
- checkResult = BfTypedValue(mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), boolType);
- return;
- }
- auto initType = initValue.mType;
- bool isDynamicCast = false;
- if (varType->IsGenericParam())
- {
- auto genericParamType = (BfGenericParamType*) varType;
- auto genericParam = GetGenericParamInstance(genericParamType);
- auto typeConstraint = genericParam->mTypeConstraint;
- if ((typeConstraint == NULL) && (genericParam->mGenericParamFlags & BfGenericParamFlag_Class))
- typeConstraint = mContext->mBfObjectType;
- if (typeConstraint != NULL)
- varType = typeConstraint;
- initValue = GetDefaultTypedValue(varType);
- }
- BfTypeInstance* srcTypeInstance = initValue.mType->ToTypeInstance();
- BfTypeInstance* varTypeInstance = varType->ToTypeInstance();
- if (CanImplicitlyCast(initValue, varType))
- {
- if ((!varType->IsPointer()) && (!varType->IsObjectOrInterface()))
- {
- if (!IsInSpecializedSection())
- {
- if (initValue.mType != varType)
- Warn(BfWarning_CS0472_ValueTypeNullCompare, StrFormat("Variable declaration is always 'true' because static cast cannot fail and a value of type '%s' can never be null",
- TypeToString(varType).c_str()), checkNode);
- else
- Warn(BfWarning_CS0472_ValueTypeNullCompare, StrFormat("Variable declaration is always 'true' because a value of type '%s' can never be null",
- TypeToString(varType).c_str()), checkNode);
- }
- }
- }
- // else if ((initType->IsInterface()) || (initType == mContext->mBfObjectType))
- // {
- // // Interface or System.Object -> *
- // isDynamicCast = true;
- // }
- // else if ((srcTypeInstance != NULL) && (varTypeInstance != NULL) &&
- // ((srcTypeInstance->IsObject()) && (TypeIsSubTypeOf(varTypeInstance, srcTypeInstance))))
- // {
- // // Class downcast
- // isDynamicCast = true;
- // }
- // else if ((!CanImplicitlyCast(GetFakeTypedValue(varType), initType)) && (!initType->IsGenericParam()))
- // {
- // if (!IsInSpecializedSection())
- // {
- // Fail(StrFormat("Cannot convert type '%s' to '%s' via any conversion",
- // TypeToString(initValue.mType).c_str(), TypeToString(varType).c_str()), checkNode);
- // }
- // }
- if (!isDynamicCast)
- {
- //initValue = Cast(checkNode, initValue, varType, (BfCastFlags)(BfCastFlags_Explicit | BfCastFlags_SilentFail));
- initValue = Cast(checkNode, initValue, varType);
- if (!initValue)
- {
- checkResult = BfTypedValue(GetConstValue(0, boolType), boolType);
- }
- else
- {
- if (localVar->mAddr)
- {
- initValue = LoadValue(initValue);
- mBfIRBuilder->CreateAlignedStore(initValue.mValue, localVar->mAddr, initValue.mType->mAlign);
- }
- if ((varType->IsPointer()) || (varType->IsObjectOrInterface()))
- {
- checkResult = BfTypedValue(mBfIRBuilder->CreateIsNotNull(initValue.mValue), boolType);
- }
- else
- {
- checkResult = BfTypedValue(GetConstValue(1, boolType), boolType);
- }
- }
- return;
- }
- if (mCompiler->IsAutocomplete())
- {
- auto allocaResult = mBfIRBuilder->CreateAlloca(mBfIRBuilder->MapType(boolType));
- auto val = mBfIRBuilder->CreateLoad(allocaResult);
- checkResult = BfTypedValue(val, boolType);
- return;
- }
- int wantTypeId = 0;
- if (!varType->IsGenericParam())
- wantTypeId = varType->mTypeId;
- auto objectType = mContext->mBfObjectType;
- PopulateType(objectType, BfPopulateType_Full);
- initValue = LoadValue(initValue);
- auto prevBB = mBfIRBuilder->GetInsertBlock();
- auto matchBB = mBfIRBuilder->CreateBlock("is.match");
- auto endBB = mBfIRBuilder->CreateBlock("is.done");
- BfIRValue boolResult = CreateAlloca(boolType);
- mBfIRBuilder->CreateStore(mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), boolResult);
- EmitDynamicCastCheck(initValue, varType, matchBB, endBB);
- AddBasicBlock(matchBB);
- mBfIRBuilder->CreateStore(mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 1), boolResult);
- mBfIRBuilder->CreateBr(endBB);
- AddBasicBlock(endBB);
- checkResult = BfTypedValue(mBfIRBuilder->CreateLoad(boolResult), boolType);
- }
- BfLocalVariable* BfModule::HandleVariableDeclaration(BfVariableDeclaration* varDecl, BfExprEvaluator* exprEvaluator)
- {
- if (mCurMethodState == NULL)
- {
- Fail("Invalid variable declaration", varDecl);
- return NULL;
- }
- BfAutoComplete* bfAutocomplete = NULL;
-
- // Just a check
- mBfIRBuilder->GetInsertBlock();
- if (mCompiler->mResolvePassData != NULL)
- bfAutocomplete = mCompiler->mResolvePassData->mAutoComplete;
- if (bfAutocomplete != NULL)
- bfAutocomplete->CheckTypeRef(varDecl->mTypeRef, true, true);
- // if (varDecl->mNameNode == NULL)
- // {
- // // Minimum parsing requirements
- // auto resolvedType = ResolveTypeRef(varDecl->mTypeRef);
- // if (resolvedType != NULL)
- // AddDependency(resolvedType, mCurTypeInstance, BfDependencyMap::DependencyFlag_LocalUsage);
- // AssertErrorState();
- // return NULL;
- // }
- bool isConst = (varDecl->mModSpecifier != NULL) && (varDecl->mModSpecifier->GetToken() == BfToken_Const);
- bool isReadOnly = (varDecl->mModSpecifier != NULL) && (varDecl->mModSpecifier->GetToken() == BfToken_ReadOnly);
- BfLocalVariable* localDef = new BfLocalVariable();
- if (varDecl->mNameNode != NULL)
- {
- varDecl->mNameNode->ToString(localDef->mName);
- localDef->mNameNode = BfNodeDynCast<BfIdentifierNode>(varDecl->mNameNode);
- }
- else
- {
- localDef->mName = "val";
- }
- bool handledExprBoolResult = false;
- bool handledVarInit = false;
- bool handledVarStore = false;
- BfType* unresolvedType = NULL;
- BfType* resolvedType = NULL;
- BfTypedValue initValue;
- bool hadVarType = false;
- bool isLet = varDecl->mTypeRef->IsA<BfLetTypeReference>();
- bool initHandled = false;
- auto _DoConditionalInit = [&](BfType* expectedType)
- {
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- auto _EmitCond = [&](BfIRValue condVal, BfTypedValue initValue)
- {
- initValue = Cast(varDecl->mInitializer, initValue, expectedType);
- if (!initValue)
- return;
- initHandled = true;
- if (localDef->mIsReadOnly)
- {
- if ((initValue.IsReadOnly()) ||
- (initValue.mKind == BfTypedValueKind_TempAddr))
- {
- localDef->mAddr = initValue.mValue;
- exprEvaluator->mResult = BfTypedValue(condVal, boolType);
- return;
- }
- }
- localDef->mAddr = AllocLocalVariable(resolvedType, localDef->mName);
- auto doAssignBlock = mBfIRBuilder->CreateBlock("assign");
- auto skipAssignBlock = mBfIRBuilder->CreateBlock("skipAssign");
- auto insertBlock = mBfIRBuilder->GetInsertBlock();
- mBfIRBuilder->CreateCondBr(condVal, doAssignBlock, skipAssignBlock);
- mBfIRBuilder->AddBlock(doAssignBlock);
- mBfIRBuilder->SetInsertPoint(doAssignBlock);
- initValue = LoadValue(initValue);
- mBfIRBuilder->CreateStore(initValue.mValue, localDef->mAddr);
- mBfIRBuilder->CreateBr(skipAssignBlock);
- mBfIRBuilder->AddBlock(skipAssignBlock);
- mBfIRBuilder->SetInsertPoint(skipAssignBlock);
- auto phiVal = mBfIRBuilder->CreatePhi(mBfIRBuilder->MapType(boolType), 2);
- mBfIRBuilder->AddPhiIncoming(phiVal, mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), insertBlock);
- mBfIRBuilder->AddPhiIncoming(phiVal, mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 1), doAssignBlock);
- exprEvaluator->mResult = BfTypedValue(phiVal, boolType);
- };
- bool handled = false;
- if ((initValue) && (initValue.mType->IsPayloadEnum()))
- {
- auto typeInst = initValue.mType->ToTypeInstance();
- PopulateType(typeInst);
- BfType* outType = NULL;
- int tagId = -1;
- if (typeInst->GetResultInfo(outType, tagId))
- {
- int dscDataIdx = -1;
- auto dscType = typeInst->GetDiscriminatorType(&dscDataIdx);
- BfIRValue dscVal = ExtractValue(initValue, dscDataIdx);
- auto eqVal = mBfIRBuilder->CreateCmpEQ(dscVal, GetConstValue(tagId, dscType));
- exprEvaluator->mResult = BfTypedValue(eqVal, boolType);
-
- if (!outType->IsValuelessType())
- {
- auto outPtrType = CreatePointerType(outType);
- initValue = MakeAddressable(initValue);
- auto payloadVal = mBfIRBuilder->CreateBitCast(initValue.mValue, mBfIRBuilder->MapType(outPtrType));
- auto payload = BfTypedValue(payloadVal, outType, true);
- if ((initValue.mKind == BfTypedValueKind_ReadOnlyAddr) ||
- (initValue.mKind == BfTypedValueKind_TempAddr) ||
- (initValue.mKind == BfTypedValueKind_ReadOnlyTempAddr))
- payload.mKind = initValue.mKind;
- _EmitCond(eqVal, payload);
- }
- handled = true;
- }
- }
- if (handled)
- {
- handledExprBoolResult = true;
- handledVarInit = true;
- handledVarStore = true;
- }
- else if ((initValue) && (initValue.mType->IsNullable()))
- {
- auto underlyingType = initValue.mType->GetUnderlyingType();
- exprEvaluator->mResult = BfTypedValue(ExtractValue(initValue, 2), boolType);
- handledExprBoolResult = true;
- if (!resolvedType->IsNullable())
- {
- if (initValue.IsAddr())
- initValue = BfTypedValue(mBfIRBuilder->CreateInBoundsGEP(initValue.mValue, 0, 1), initValue.mType->GetUnderlyingType(), true);
- else
- initValue = BfTypedValue(mBfIRBuilder->CreateExtractValue(initValue.mValue, 1), initValue.mType->GetUnderlyingType());
- }
- if ((initValue) && (!initValue.mType->IsValuelessType()))
- {
- _EmitCond(exprEvaluator->mResult.mValue, initValue);
- handledVarStore = true;
- }
- handledVarInit = true;
- }
- else if (initValue)
- {
- BfAstNode* refNode = varDecl;
- if (varDecl->mInitializer != NULL)
- refNode = varDecl->mInitializer;
- TryInitVar(refNode, localDef, initValue, exprEvaluator->mResult);
- handledExprBoolResult = true;
- handledVarInit = true;
- handledVarStore = true;
- }
- };
- if ((varDecl->mTypeRef->IsA<BfVarTypeReference>()) || (isLet))
- {
- hadVarType = true;
- if (varDecl->mInitializer == NULL)
- {
- if (!isLet)
- {
- BfLocalVarEntry* shadowEntry;
- if (mCurMethodState->mLocalVarSet.TryGet(BfLocalVarEntry(localDef), &shadowEntry))
- {
- auto prevLocal = shadowEntry->mLocalVar;
- if (prevLocal->mLocalVarIdx >= mCurMethodState->GetLocalStartIdx())
- {
- BfExprEvaluator exprEvaluator(this);
- initValue = exprEvaluator.LoadLocal(prevLocal);
- resolvedType = initValue.mType;
- unresolvedType = resolvedType;
- localDef->mLocalVarId = prevLocal->mLocalVarId;
- localDef->mIsAssigned = true;
- localDef->mIsShadow = true;
- exprEvaluator.mResultLocalVarRefNode = varDecl->mNameNode;
- exprEvaluator.mResultLocalVar = prevLocal;
- exprEvaluator.CheckResultForReading(initValue);
- if (bfAutocomplete != NULL)
- bfAutocomplete->CheckVarResolution(varDecl->mTypeRef, resolvedType);
- }
- }
- }
- if (!initValue)
- {
- Fail("Implicitly-typed variables must be initialized", varDecl);
- initValue = GetDefaultTypedValue(mContext->mBfObjectType);
- }
- }
- else
- {
- if (isConst)
- {
- BfConstResolver constResolver(this);
- initValue = constResolver.Resolve(varDecl->mInitializer);
- }
- else
- {
- BfExprEvaluator valExprEvaluator(this);
- valExprEvaluator.mAllowReadOnlyReference = isLet;
- initValue = CreateValueFromExpression(valExprEvaluator, varDecl->mInitializer, NULL, (BfEvalExprFlags)(BfEvalExprFlags_AllowSplat | BfEvalExprFlags_AllowRefExpr | BfEvalExprFlags_VariableDeclaration));
- if ((exprEvaluator != NULL) && (initValue))
- {
- if (initValue.mType->IsNullable())
- {
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- initValue = LoadValue(initValue);
- exprEvaluator->mResult = BfTypedValue(mBfIRBuilder->CreateExtractValue(initValue.mValue, 2), boolType);
- handledExprBoolResult = true;
- initValue = BfTypedValue(mBfIRBuilder->CreateExtractValue(initValue.mValue, 1), initValue.mType->GetUnderlyingType());
- }
- else
- {
- auto typeInst = initValue.mType->ToTypeInstance();
- if (typeInst != NULL)
- {
- PopulateType(typeInst);
- BfType* outType = NULL;
- int tagId = -1;
- if (typeInst->GetResultInfo(outType, tagId))
- {
- handledExprBoolResult = true;
- unresolvedType = outType;
- resolvedType = outType;
- isReadOnly = isLet;
- localDef->mIsReadOnly = isLet;
- _DoConditionalInit(outType);
- }
- }
- }
- }
- }
- }
- if (!initValue)
- {
- initValue = GetDefaultTypedValue(GetPrimitiveType(BfTypeCode_Var));
- }
- if (initValue.mType->IsNull())
- {
- Fail("Implicitly-typed variables cannot be initialized to 'null'", varDecl->mInitializer);
- initValue = GetDefaultTypedValue(mContext->mBfObjectType);
- }
- if (unresolvedType == NULL)
- unresolvedType = initValue.mType;
- resolvedType = unresolvedType;
- if (initValue.IsTempAddr())
- {
- // Take over value
- localDef->mAddr = initValue.mValue;
- handledVarInit = true;
- if (isLet)
- {
- localDef->mValue = mBfIRBuilder->CreateLoad(localDef->mAddr);
- }
- }
- if (bfAutocomplete != NULL)
- bfAutocomplete->CheckVarResolution(varDecl->mTypeRef, resolvedType);
- }
- else
- {
- unresolvedType = ResolveTypeRef(varDecl->mTypeRef, BfPopulateType_Data, (BfResolveTypeRefFlags)(BfResolveTypeRefFlag_NoResolveGenericParam | BfResolveTypeRefFlag_AllowRef));
- if (unresolvedType == NULL)
- unresolvedType = mContext->mBfObjectType; // Fake an object
- //resolvedType = ResolveGenericType(unresolvedType);
- resolvedType = unresolvedType;
- if (unresolvedType == NULL)
- resolvedType = mContext->mBfObjectType; // Fake an object
- }
- auto _CheckConst = [&]
- {
- if (initValue.mValue.IsConst())
- {
- auto constant = mBfIRBuilder->GetConstant(initValue.mValue);
- // NullPtr is standin for GlobalVar during autocomplete
- if ((constant->mConstType == BfConstType_GlobalVar) ||
- (constant->mTypeCode == BfTypeCode_NullPtr))
- {
- // Not really constant
- // localNeedsAddr = false;
- // isConst = false;
- // initHandled = true;
- // localDef->mValue = initValue.mValue;
- isConst = false;
- }
- }
- };
- PopulateType(resolvedType);
- AddDependency(resolvedType, mCurTypeInstance, BfDependencyMap::DependencyFlag_LocalUsage);
-
- localDef->mResolvedType = resolvedType;
- localDef->mIsReadOnly = isReadOnly;
-
- if (!initHandled)
- {
- if (isLet)
- {
- localDef->mIsReadOnly = true;
- if (initValue)
- {
- if ((initValue.mValue) && (initValue.mValue.IsConst()))
- {
- isConst = true;
- }
- }
- }
- }
- _CheckConst();
-
- bool localNeedsAddr = false;
- bool allowValueAccess = true;
- if (mHasFullDebugInfo)
- {
- //if (!IsTargetingBeefBackend())
- if (!isConst)
- localNeedsAddr = true;
- /*if (mCurMethodInstance->mMethodDef->mName != "Boop2")
- dbgNeedsAddr = true;*/
- }
-
- // This is required because of lifetime and LLVM domination rules for certain instances of variable declarations in binary conditionals.
- // IE: if ((something) && (let a = somethingElse))
- if ((exprEvaluator != NULL) && (!isConst))
- {
- localNeedsAddr = true;
- allowValueAccess = false;
- }
-
- if ((varDecl->mEqualsNode != NULL) && (mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL) && (!initHandled))
- {
- mCompiler->mResolvePassData->mAutoComplete->CheckEmptyStart(varDecl->mEqualsNode, resolvedType);
- }
- BfIRInitType initType = BfIRInitType_NotSet;
-
- if (varDecl->mInitializer != NULL)
- {
- initType = BfIRInitType_NotNeeded_AliveOnDecl;
- if ((!initValue) && (!initHandled))
- {
- if (isConst)
- {
- BfConstResolver constResolver(this);
- initValue = constResolver.Resolve(varDecl->mInitializer, resolvedType);
- if (!initValue)
- initValue = GetDefaultTypedValue(resolvedType);
- }
- else if (varDecl->mInitializer->IsA<BfUninitializedExpression>())
- {
- // Fake 'is assigned'
- }
- else
- { //
- auto expectedType = resolvedType;
- if (expectedType->IsRef())
- expectedType = expectedType->GetUnderlyingType();
- BfExprEvaluator valExprEvaluator(this);
- valExprEvaluator.mAllowReadOnlyReference = isReadOnly;
- initValue = CreateValueFromExpression(valExprEvaluator, varDecl->mInitializer, expectedType, (BfEvalExprFlags)(BfEvalExprFlags_NoCast | BfEvalExprFlags_AllowRefExpr | BfEvalExprFlags_VariableDeclaration));
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- if (exprEvaluator != NULL)
- _DoConditionalInit(expectedType);
- if ((!handledVarInit) && (initValue))
- initValue = Cast(varDecl->mInitializer, initValue, resolvedType, BfCastFlags_PreferAddr);
- // Why did we remove this?
- // if ((valExprEvaluator.mResultIsTempComposite) && (initValue.IsAddr()))
- // {
- // BF_ASSERT(initValue.mType->IsComposite());
- // localDef->mAddr = initValue.mValue;
- // handledVarInit = true;
- // handledVarStore = true;
- // }
- }
- }
- if ((!handledVarInit) && (!isConst))
- {
- if (initValue)
- {
- // Handled later
- }
- else if (varDecl->mInitializer->IsA<BfUninitializedExpression>())
- {
- // Fake 'is assigned'
- initType = BfIRInitType_Uninitialized;
- }
- else
- {
- AssertErrorState();
- }
- }
- localDef->mIsAssigned = true;
- }
- else
- {
- if (isConst)
- {
- Fail("Const locals must be initialized", varDecl->mModSpecifier);
- initValue = GetDefaultTypedValue(resolvedType);
- }
- else if (isReadOnly)
- {
- Fail("Readonly locals must be initialized", varDecl->mModSpecifier);
- initValue = GetDefaultTypedValue(resolvedType);
- }
- else if (auto refTypeRef = BfNodeDynCast<BfRefTypeRef>(varDecl->mTypeRef))
- {
- Fail("Ref locals must be initialized", refTypeRef->mRefToken);
- }
- else
- {
- BF_ASSERT(!localDef->mResolvedType->IsRef());
- }
- }
- _CheckConst();
- if ((initValue.mKind == BfTypedValueKind_TempAddr) && (!initHandled))
- {
- BF_ASSERT(initValue.IsAddr());
- BF_ASSERT(initValue.mType->IsComposite());
- handledVarInit = true;
- handledVarStore = true;
- if (!localDef->mAddr)
- {
- localDef->mAddr = initValue.mValue;
- if (localDef->mIsReadOnly)
- localDef->mValue = mBfIRBuilder->CreateLoad(localDef->mAddr);
- }
- if (WantsLifetimes())
- mCurMethodState->mCurScope->mDeferredLifetimeEnds.push_back(localDef->mAddr);
- }
- if ((!localDef->mAddr) && (!isConst) && ((!localDef->mIsReadOnly) || (localNeedsAddr)))
- {
- if ((exprEvaluator != NULL) && (exprEvaluator->mResultIsTempComposite))
- {
- //TODO: Can we remove this one?
- BF_ASSERT(initValue.IsAddr());
- BF_ASSERT(initValue.mType->IsComposite());
- localDef->mAddr = initValue.mValue;
- }
- else
- localDef->mAddr = AllocLocalVariable(resolvedType, localDef->mName);
- }
- bool wantsStore = false;
- if ((initValue) && (!handledVarStore) && (!isConst) && (!initHandled))
- {
- initValue = LoadValue(initValue);
- if (initValue.IsSplat())
- {
- if (!localDef->mAddr)
- localDef->mAddr = AllocLocalVariable(resolvedType, localDef->mName);
- AggregateSplatIntoAddr(initValue, localDef->mAddr);
- initHandled = true;
- }
- else
- {
- initValue = AggregateSplat(initValue);
- localDef->mValue = initValue.mValue;
- if ((localDef->mAddr) && (!localDef->mResolvedType->IsValuelessType()))
- {
- if (!initValue.mType->IsVar())
- wantsStore = true;
- }
- else
- {
- BF_ASSERT(isReadOnly || isLet || initValue.mType->IsValuelessType() || (mBfIRBuilder->mIgnoreWrites));
- }
- }
- }
- if ((localDef->mIsReadOnly) && (!isConst) && (!localDef->mValue) && (!initHandled))
- {
- if (!resolvedType->IsValuelessType())
- {
- AssertErrorState();
- initValue = GetDefaultTypedValue(resolvedType);
- localDef->mValue = initValue.mValue;
- }
- }
- if ((!localDef->mAddr) && (!isConst) && ((!localDef->mIsReadOnly) || (localNeedsAddr)))
- {
- localDef->mAddr = AllocLocalVariable(resolvedType, localDef->mName);
- }
- if ((exprEvaluator != NULL) && (!handledExprBoolResult))
- {
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- if (!initValue)
- {
- AssertErrorState();
- exprEvaluator->mResult = BfTypedValue(mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), boolType);
- }
- else if ((resolvedType->IsPointer()) || (resolvedType->IsObjectOrInterface()))
- {
- exprEvaluator->mResult = BfTypedValue(mBfIRBuilder->CreateIsNotNull(initValue.mValue), boolType);
- }
- else
- {
- // Always true
- if (!IsInSpecializedSection())
- Warn(BfWarning_CS0472_ValueTypeNullCompare, StrFormat("Variable declaration is always 'true' since a value of type '%s' can never be null",
- TypeToString(initValue.mType).c_str()), varDecl);
- exprEvaluator->mResult = BfTypedValue(mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 1), boolType);
- }
- }
- if ((unresolvedType->IsGenericParam()) && (resolvedType->IsValuelessType()) && (mHasFullDebugInfo))
- {
- // We do this in order to be able to bind to lines that contain valueless variable declarations in some generics
- // We don't need to do this in non-generics because the breakpoint will just move to the next line that actually has instructions
- EmitEnsureInstructionAt();
- }
- if ((resolvedType->IsVoid()) && (!IsInSpecializedSection()))
- {
- Warn(0, StrFormat("Variable '%s' is declared as 'void'", localDef->mName.c_str()), varDecl->mTypeRef);
- }
- if ((!handledVarInit) && (isConst))
- localDef->mConstValue = initValue.mValue;
- if (!allowValueAccess)
- localDef->mValue = BfIRValue();
- if (!localDef->mIsShadow)
- CheckVariableDef(localDef);
- ValidateAllocation(localDef->mResolvedType, varDecl->mTypeRef);
- if ((exprEvaluator == NULL) && (varDecl->GetSourceData() != NULL))
- UpdateSrcPos(varDecl);
- localDef->Init();
- if (localDef->mConstValue)
- initType = BfIRInitType_NotNeeded;
- BfLocalVariable* localVar = AddLocalVariableDef(localDef, true, false, BfIRValue(), initType);
- if (wantsStore)
- mBfIRBuilder->CreateStore(initValue.mValue, localVar->mAddr);
- return localVar;
- }
- BfLocalVariable* BfModule::HandleVariableDeclaration(BfVariableDeclaration* varDecl, BfTypedValue val, bool updateSrcLoc, bool forceAddr)
- {
- if (varDecl->mEqualsNode != NULL)
- Fail("Unexpected initialization", varDecl->mEqualsNode);
- if (varDecl->mInitializer != NULL)
- CreateValueFromExpression(varDecl->mInitializer);
- auto isLet = varDecl->mTypeRef->IsA<BfLetTypeReference>();
- auto isVar = varDecl->mTypeRef->IsA<BfVarTypeReference>();
- bool isRef = false;
- if (auto varRefTypeReference = BfNodeDynCast<BfVarRefTypeReference>(varDecl->mTypeRef))
- {
- BF_ASSERT(val.IsAddr());
- isRef = true;
- isLet = varRefTypeReference->mVarToken->GetToken() == BfToken_Let;
- isVar = varRefTypeReference->mVarToken->GetToken() == BfToken_Var;
- }
- else
- {
- BF_ASSERT(!val.IsAddr());
- }
- auto autoComplete = mCompiler->GetAutoComplete();
- if ((autoComplete != NULL) && ((isLet) || (isVar)))
- autoComplete->CheckVarResolution(varDecl->mTypeRef, val.mType);
- // BfType* type = val.mType;
- // if (type == NULL)
- // {
- // type = ResolveTypeRef(varDecl->mTypeRef);
- // if (type == NULL)
- // type = mContext->mBfObjectType;
- // }
- BfType* type = NULL;
- if ((isLet) || (isVar))
- {
- type = val.mType;
- }
- else
- {
- type = ResolveTypeRef(varDecl->mTypeRef);
- }
- if (type == NULL)
- type = mContext->mBfObjectType;
- if ((type->IsVar()) || (type->IsLet()))
- {
-
- }
- if (isRef)
- {
- type = CreateRefType(type);
- }
- BfLocalVariable* localDef = new BfLocalVariable();
- if (varDecl->mNameNode != NULL)
- varDecl->mNameNode->ToString(localDef->mName);
- localDef->mNameNode = BfNodeDynCast<BfIdentifierNode>(varDecl->mNameNode);
- localDef->mResolvedType = type;
- localDef->mIsAssigned = true;
- localDef->mValue = val.mValue;
- if (isLet)
- {
- localDef->mIsReadOnly = true;
- }
- if ((!localDef->mIsReadOnly) || (mHasFullDebugInfo) || (forceAddr))
- {
- localDef->mAddr = AllocLocalVariable(localDef->mResolvedType, localDef->mName);
- if ((val.mValue) && (!localDef->mResolvedType->IsValuelessType()))
- {
- if (val.IsSplat())
- AggregateSplatIntoAddr(val, localDef->mAddr);
- else
- mBfIRBuilder->CreateAlignedStore(val.mValue, localDef->mAddr, localDef->mResolvedType->mAlign);
- }
- if (forceAddr)
- localDef->mValue = BfIRValue();
- }
- CheckVariableDef(localDef);
- if ((updateSrcLoc) && (varDecl->GetSourceData() != NULL))
- UpdateSrcPos(varDecl);
- localDef->Init();
- return AddLocalVariableDef(localDef, true);
- }
- void BfModule::CheckTupleVariableDeclaration(BfTupleExpression* tupleExpr, BfType* initType)
- {
- if (initType == NULL)
- return;
- BfTupleType* initTupleType = NULL;
- if ((initType != NULL) && (initType->IsTuple()))
- initTupleType = (BfTupleType*)initType;
- if (initTupleType != NULL)
- {
- mBfIRBuilder->PopulateType(initTupleType);
- AddDependency(initTupleType, mCurTypeInstance, BfDependencyMap::DependencyFlag_LocalUsage);
- int paramCountDiff = (int)tupleExpr->mValues.size() - (int)initTupleType->mFieldInstances.size();
- if (paramCountDiff > 0)
- {
- Fail(StrFormat("Too many variable names, expected %d fewer.", paramCountDiff), tupleExpr->mValues[(int)initTupleType->mFieldInstances.size()]);
- }
- else if (paramCountDiff < 0)
- {
- BfAstNode* refNode = tupleExpr->mCloseParen;
- if (refNode == NULL)
- refNode = tupleExpr;
- Fail(StrFormat("Too few variable names, expected %d more.", -paramCountDiff), refNode);
- }
- }
- else
- {
- Fail(StrFormat("Value result type '%s' must be a tuple type to be applicable for tuple decomposition", TypeToString(initType).c_str()), tupleExpr);
- }
- }
- void BfModule::HandleTupleVariableDeclaration(BfVariableDeclaration* varDecl, BfTupleExpression* tupleExpr, BfTypedValue initTupleValue, bool isReadOnly, bool isConst, bool forceAddr, BfIRBlock* declBlock)
- {
- BfTupleType* initTupleType = NULL;
- if ((initTupleValue) && (initTupleValue.mType->IsTuple()))
- initTupleType = (BfTupleType*)initTupleValue.mType;
- CheckTupleVariableDeclaration(tupleExpr, initTupleValue.mType);
-
- for (int varIdx = 0; varIdx < (int)tupleExpr->mValues.size(); varIdx++)
- {
- BfType* resolvedType = NULL;
- BfTypedValue initValue;
- if ((initTupleType != NULL) && (varIdx < (int)initTupleType->mFieldInstances.size()))
- {
- auto fieldInstance = &initTupleType->mFieldInstances[varIdx];
- auto fieldDef = fieldInstance->GetFieldDef();
- resolvedType = fieldInstance->GetResolvedType();
- if (fieldInstance->mDataIdx != -1)
- {
- if (initTupleValue.IsAddr())
- {
- initValue = BfTypedValue(mBfIRBuilder->CreateInBoundsGEP(initTupleValue.mValue, 0, fieldInstance->mDataIdx), resolvedType, true);
- initValue = LoadValue(initValue);
- }
- else
- initValue = BfTypedValue(mBfIRBuilder->CreateExtractValue(initTupleValue.mValue, fieldInstance->mDataIdx), resolvedType);
- }
- BfTupleNameNode* tupleNameNode = NULL;
- if (varIdx < (int)tupleExpr->mNames.size())
- tupleNameNode = tupleExpr->mNames[varIdx];
- if (!fieldDef->IsUnnamedTupleField())
- {
- if (tupleNameNode != NULL)
- {
- if (fieldDef->mName != tupleNameNode->mNameNode->ToString())
- {
- Fail(StrFormat("Mismatched tuple field name, expected '%s'", fieldDef->mName.c_str()), tupleNameNode->mNameNode);
- }
- }
- }
- else if ((tupleNameNode != NULL) && (tupleNameNode->mNameNode != NULL))
- {
- Fail(StrFormat("Unexpected tuple field name, expected unnamed tuple field", fieldDef->mName.c_str()), tupleNameNode->mNameNode);
- }
- }
- else
- {
- resolvedType = mContext->mBfObjectType;
- initValue = GetDefaultTypedValue(resolvedType);
- }
- BfExpression* varNameNode = tupleExpr->mValues[varIdx];
- if (!varNameNode->IsExact<BfIdentifierNode>())
- {
- if (BfTupleExpression* innerTupleExpr = BfNodeDynCast<BfTupleExpression>(varNameNode))
- {
- HandleTupleVariableDeclaration(varDecl, innerTupleExpr, initValue, isReadOnly, isConst, false, declBlock);
- }
- else if (!varNameNode->IsExact<BfUninitializedExpression>())
- Fail("Variable name expected", varNameNode);
- continue;
- }
- bool initHandled = false;
- BfLocalVariable* localDef = new BfLocalVariable();
- varNameNode->ToString(localDef->mName);
- localDef->mNameNode = BfNodeDynCast<BfIdentifierNode>(varNameNode);
- localDef->mResolvedType = resolvedType;
- localDef->mReadFromId = 0; // Don't give usage errors for binds
- if (isReadOnly)
- {
- localDef->mIsReadOnly = true;
- if ((initValue) && (initValue.mValue.IsConst()))
- {
- isConst = true;
- }
- }
- CheckVariableDef(localDef);
- if ((!isConst) && ((forceAddr) || (!localDef->mIsReadOnly) || (mHasFullDebugInfo)))
- {
- localDef->mAddr = AllocLocalVariable(resolvedType, localDef->mName);
- }
- if ((varDecl != NULL) && (varDecl->mEqualsNode != NULL) && (mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL) && (!initHandled))
- {
- mCompiler->mResolvePassData->mAutoComplete->CheckEmptyStart(varDecl->mEqualsNode, resolvedType);
- }
- if ((varDecl == NULL) || (varDecl->mInitializer != NULL))
- {
- if ((!isConst) && (!initHandled))
- {
- if (initValue)
- {
- if (!forceAddr)
- localDef->mValue = initValue.mValue;
- if (localDef->mAddr)
- {
- mBfIRBuilder->CreateStore(initValue.mValue, localDef->mAddr);
- }
- }
- else if ((varDecl == NULL) || (varDecl->mInitializer->IsA<BfUninitializedExpression>()))
- {
- // Fake 'is assigned'
- }
- else
- {
- AssertErrorState();
- }
- }
- localDef->mIsAssigned = true;
- }
- else if ((varDecl != NULL) && (varDecl->mInitializer == NULL))
- {
- if (auto refTypeRef = BfNodeDynCast<BfRefTypeRef>(varDecl->mTypeRef))
- {
- Fail("Ref variables must be initialized", refTypeRef->mRefToken);
- }
- else
- {
- BF_ASSERT(!localDef->mResolvedType->IsRef());
- }
- }
- if (isConst)
- localDef->mConstValue = initValue.mValue;
- CheckVariableDef(localDef);
- if ((varDecl != NULL) && (varDecl->GetSourceData() != NULL))
- UpdateSrcPos(varDecl);
- localDef->Init();
- auto defBlock = mBfIRBuilder->GetInsertBlock();
- if (declBlock != NULL)
- mBfIRBuilder->SetInsertPoint(*declBlock);
- AddLocalVariableDef(localDef, true, false, BfIRValue(), BfIRInitType_NotNeeded);
- if (declBlock != NULL)
- mBfIRBuilder->SetInsertPoint(defBlock);
- }
- }
- void BfModule::HandleTupleVariableDeclaration(BfVariableDeclaration* varDecl)
- {
- BfAutoComplete* bfAutocomplete = mCompiler->GetAutoComplete();
- if (bfAutocomplete != NULL)
- bfAutocomplete->CheckTypeRef(varDecl->mTypeRef, true);
- BfTupleExpression* tupleExpr = BfNodeDynCast<BfTupleExpression>(varDecl->mNameNode);
- bool isConst = (varDecl->mModSpecifier != NULL) && (varDecl->mModSpecifier->GetToken() == BfToken_Const);
- bool isReadOnly = (varDecl->mModSpecifier != NULL) && (varDecl->mModSpecifier->GetToken() == BfToken_ReadOnly);
- BfTypedValue initTupleValue;
- bool hadVarType = false;
- bool isLet = varDecl->mTypeRef->IsA<BfLetTypeReference>();
- bool isVar = varDecl->mTypeRef->IsA<BfVarTypeReference>();
- bool wasVarOrLet = isVar || isLet;
-
- if ((!isLet) && (!isVar))
- {
- ResolveTypeRef(varDecl->mTypeRef);
- Fail("'var' or 'let' expected", varDecl->mTypeRef);
- isVar = true;
- }
- if ((isVar) || (isLet))
- {
- hadVarType = true;
- if (varDecl->mInitializer == NULL)
- {
- Fail("Implicitly-typed variables must be initialized", varDecl);
- initTupleValue = GetDefaultTypedValue(mContext->mBfObjectType);
- }
- else
- {
- if (isConst)
- {
- BfConstResolver constResolver(this);
- initTupleValue = constResolver.Resolve(varDecl->mInitializer);
- }
- else
- {
- initTupleValue = CreateValueFromExpression(varDecl->mInitializer, NULL);
- }
- }
- initTupleValue = LoadValue(initTupleValue);
-
- if ((bfAutocomplete != NULL) && (wasVarOrLet))
- bfAutocomplete->CheckVarResolution(varDecl->mTypeRef, initTupleValue.mType);
- }
- bool isCompatible = false;
- if (initTupleValue)
- HandleTupleVariableDeclaration(varDecl, tupleExpr, initTupleValue, isLet || isReadOnly, isConst, false);
- else
- AssertErrorState();
- }
- void BfModule::HandleCaseEnumMatch_Tuple(BfTypedValue tupleVal, const BfSizedArray<BfExpression*>& arguments, BfAstNode* tooFewRef, BfIRValue phiVal, BfIRBlock& matchedBlock, BfIRBlock falseBlock, bool& hadConditional, bool clearOutOnMismatch)
- {
- SetAndRestoreValue<bool> prevInCondBlock(mCurMethodState->mInConditionalBlock);
- auto tupleType = tupleVal.mType->ToTypeInstance();
- struct DeferredAssign
- {
- BfExpression* mExpr;
- BfTypedValue mArgValue;
- BfTypedValue mTupleElement;
- int mFieldIdx;
- };
- Array<DeferredAssign> deferredAssigns;
- auto autoComplete = mCompiler->GetAutoComplete();
- for (int tupleFieldIdx = 0; tupleFieldIdx < (int)tupleType->mFieldInstances.size(); tupleFieldIdx++)
- {
- auto tupleFieldInstance = &tupleType->mFieldInstances[tupleFieldIdx];
- if (tupleFieldIdx >= arguments.size())
- {
- BfError* error = Fail(StrFormat("Not enough parameters specified, expected %d more.", tupleType->mFieldInstances.size() - (int)arguments.size()), tooFewRef);
- break;
- }
- BfTypedValue tupleElement;
- if (tupleFieldInstance->mDataIdx >= 0)
- {
- tupleElement = ExtractValue(tupleVal, tupleFieldInstance, tupleFieldInstance->mDataIdx);
- }
- else
- tupleElement = GetDefaultTypedValue(tupleFieldInstance->GetResolvedType());
- auto expr = BfNodeDynCast<BfExpression>(arguments[tupleFieldIdx]);
- if (auto varDecl = BfNodeDynCast<BfVariableDeclaration>(expr))
- {
- bool isVarOrLet = (varDecl->mTypeRef->IsExact<BfLetTypeReference>()) || (varDecl->mTypeRef->IsExact<BfVarTypeReference>());
- bool isRef = false;
- if (varDecl->mTypeRef->IsExact<BfVarRefTypeReference>())
- {
- isVarOrLet = true;
- isRef = true;
- }
- if (!isVarOrLet)
- {
- auto wantType = ResolveTypeRef(varDecl->mTypeRef);
- if (wantType == NULL)
- wantType = mContext->mBfObjectType;
- if (wantType != NULL)
- tupleElement = Cast(varDecl->mTypeRef, tupleElement, wantType);
- if (!tupleElement)
- tupleElement = GetDefaultTypedValue(wantType);
- }
- PopulateType(tupleElement.mType);
- if (!isRef)
- tupleElement = LoadValue(tupleElement);
- auto localVar = HandleVariableDeclaration(varDecl, tupleElement, false, true);
- localVar->mReadFromId = 0; // Don't give usage errors for binds
- continue;
- }
- if (auto uninitExpr = BfNodeDynCast<BfUninitializedExpression>(expr))
- {
- continue;
- }
- if (tupleFieldInstance->mDataIdx >= 0)
- {
- if (auto tupleExpr = BfNodeDynCast<BfTupleExpression>(expr))
- {
- if (tupleElement.mType->IsTuple())
- {
- BfAstNode* tooFewRef = tupleExpr->mCloseParen;
- if (tupleExpr->mValues.size() > 0)
- tooFewRef = tupleExpr->mValues[tupleExpr->mValues.size() - 1];
- if (tooFewRef == NULL)
- tooFewRef = tupleExpr->mOpenParen;
- HandleCaseEnumMatch_Tuple(tupleElement, tupleExpr->mValues, tooFewRef, phiVal, matchedBlock, falseBlock, hadConditional, clearOutOnMismatch);
- continue;
- }
- }
- }
- if (expr == NULL)
- {
- // Error would have occured in the parser
- //AssertErrorState();
- }
- else
- {
- mCurMethodState->mInConditionalBlock = true;
- auto tupleElementAddr = tupleElement;
- BfTypedValue exprResult;
- if (auto invocationExpr = BfNodeDynCast<BfInvocationExpression>(expr))
- {
- if (auto memberRefExpr = BfNodeDynCast<BfMemberReferenceExpression>(invocationExpr->mTarget))
- {
- if (memberRefExpr->mTarget == NULL)
- {
- if (tupleElement.mType->IsPayloadEnum())
- {
- auto intType = GetPrimitiveType(BfTypeCode_Int32);
- BfTypedValue enumTagVal;
- if (tupleElement.IsAddr())
- {
- enumTagVal = BfTypedValue(mBfIRBuilder->CreateInBoundsGEP(tupleElement.mValue, 0, 2), intType, true);
- enumTagVal = LoadValue(enumTagVal);
- }
- else
- enumTagVal = BfTypedValue(mBfIRBuilder->CreateExtractValue(tupleElement.mValue, 2), intType, false);
-
- int uncondTagId = -1;
- bool hadConditional = false;
- exprResult = TryCaseEnumMatch(tupleElementAddr, enumTagVal, expr, NULL, NULL, NULL, uncondTagId, hadConditional, clearOutOnMismatch);
- }
- }
- }
- }
- if (!exprResult)
- {
- tupleElement = LoadValue(tupleElement);
- BfExprEvaluator exprEvaluator(this);
- exprEvaluator.mExpectingType = tupleFieldInstance->GetResolvedType();
- exprEvaluator.mBfEvalExprFlags = BfEvalExprFlags_AllowOutExpr;
- exprEvaluator.Evaluate(expr);
- auto argValue = exprEvaluator.mResult;
- if (!argValue)
- continue;
- if (argValue.mType->IsRef())
- {
- auto refType = (BfRefType*)argValue.mType;
- if (refType->mRefKind != BfRefType::RefKind_Out)
- {
- BfAstNode* refNode = expr;
- if (auto unaryOperatorExpr = BfNodeDynCast<BfUnaryOperatorExpression>(expr))
- refNode = unaryOperatorExpr->mOpToken;
- Fail("Only 'out' refs can be used to assign to an existing value", refNode);
- }
- DeferredAssign deferredAssign = { expr, argValue, tupleElement, tupleFieldIdx };
- deferredAssigns.push_back(deferredAssign);
- exprEvaluator.MarkResultAssigned();
- continue;
- }
- if (!argValue.mType->IsValueType())
- argValue = LoadValue(argValue);
- argValue = Cast(expr, argValue, tupleFieldInstance->GetResolvedType());
- if (!argValue)
- continue;
-
- exprEvaluator.PerformBinaryOperation(expr, expr, BfBinaryOp_Equality, expr, BfBinOpFlag_NoClassify, tupleElement, argValue);
- exprResult = exprEvaluator.mResult;
- }
-
- if (exprResult)
- {
- hadConditional = true;
- if (phiVal)
- {
- auto insertBlock = mBfIRBuilder->GetInsertBlock();
- mBfIRBuilder->AddPhiIncoming(phiVal, mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), insertBlock);
- }
- matchedBlock = mBfIRBuilder->CreateBlock("match", false);
-
- mBfIRBuilder->CreateCondBr(exprResult.mValue, matchedBlock, falseBlock);
- mBfIRBuilder->AddBlock(matchedBlock);
- mBfIRBuilder->SetInsertPoint(matchedBlock);
- }
- }
- }
- if (!deferredAssigns.empty())
- mBfIRBuilder->SetInsertPoint(matchedBlock);
- // We assign these only after the value checks succeed
- for (auto& deferredAssign : deferredAssigns)
- {
- auto argValue = RemoveRef(deferredAssign.mArgValue);
- auto tupleElement = Cast(deferredAssign.mExpr, deferredAssign.mTupleElement, argValue.mType);
- if (!tupleElement)
- continue;
- mBfIRBuilder->CreateStore(tupleElement.mValue, argValue.mValue);
- }
- if ((clearOutOnMismatch) && (!deferredAssigns.IsEmpty()))
- {
- auto curInsertPoint = mBfIRBuilder->GetInsertBlock();
- mBfIRBuilder->SetInsertPoint(falseBlock);
- for (auto& deferredAssign : deferredAssigns)
- {
- auto tupleFieldInstance = &tupleType->mFieldInstances[deferredAssign.mFieldIdx];
- // We have to re-process the expr because we haven't done it in this branch, and then clear the result out
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, mHadBuildError); // Don't fail twice
- BfExprEvaluator exprEvaluator(this);
- exprEvaluator.mExpectingType = tupleFieldInstance->GetResolvedType();
- exprEvaluator.mBfEvalExprFlags = BfEvalExprFlags_AllowOutExpr;
- exprEvaluator.Evaluate(deferredAssign.mExpr);
- auto argValue = exprEvaluator.mResult;
- if (!argValue)
- continue;
- mBfIRBuilder->CreateMemSet(argValue.mValue, GetConstValue8(0), GetConstValue(argValue.mType->mSize), GetConstValue(argValue.mType->mAlign));
- }
- mBfIRBuilder->SetInsertPoint(curInsertPoint);
- }
- if (arguments.size() > tupleType->mFieldInstances.size())
- {
- for (int i = (int)tupleType->mFieldInstances.size(); i < (int)arguments.size(); i++)
- {
- // For autocomplete and such
- auto expr = arguments[i];
- if (expr != NULL)
- CreateValueFromExpression(expr);
- }
- BfAstNode* errorRef = arguments[(int)tupleType->mFieldInstances.size()];
- BfError* error = Fail(StrFormat("Too many arguments, expected %d fewer.", arguments.size() - tupleType->mFieldInstances.size()), errorRef);
- }
- }
- BfTypedValue BfModule::TryCaseEnumMatch(BfTypedValue enumVal, BfTypedValue tagVal, BfExpression* expr, BfIRBlock* eqBlock, BfIRBlock* notEqBlock, BfIRBlock* matchBlock, int& tagId, bool& hadConditional, bool clearOutOnMismatch)
- {
- auto invocationExpr = BfNodeDynCast<BfInvocationExpression>(expr);
- if (invocationExpr == NULL)
- return BfTypedValue();
- auto activeTypeDef = GetActiveTypeDef();
- BfType* targetType = NULL;
- BfIdentifierNode* nameNode = NULL;
- BfTokenNode* dotNode = NULL;
- if (auto memberRefExpr = BfNodeDynCast<BfMemberReferenceExpression>(invocationExpr->mTarget))
- {
- if (memberRefExpr->mTarget == NULL)
- {
- targetType = enumVal.mType;
- }
- else if (auto typeRef = BfNodeDynCast<BfTypeReference>(memberRefExpr->mTarget))
- {
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, true);
- targetType = ResolveTypeRef(typeRef);
- }
- else if (auto identifier = BfNodeDynCast<BfIdentifierNode>(memberRefExpr->mTarget))
- {
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, true);
- targetType = ResolveTypeRef(identifier, NULL);
- }
- if (auto nameIdentifier = BfNodeDynCast<BfIdentifierNode>(memberRefExpr->mMemberName))
- {
- dotNode = memberRefExpr->mDotToken;
- nameNode = nameIdentifier;
- }
- else
- return BfTypedValue();
- }
- else if (auto qualifiedNameNode = BfNodeDynCast<BfQualifiedNameNode>(invocationExpr->mTarget))
- {
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, true);
- targetType = ResolveTypeRef(qualifiedNameNode->mLeft, NULL);
- nameNode = qualifiedNameNode->mRight;
- }
- else
- return BfTypedValue();
- // These may have been colorized as methods, so change that
- SetElementType(nameNode, BfSourceElementType_Normal);
- if ((targetType == NULL) || (!targetType->IsPayloadEnum()))
- return BfTypedValue();
- auto enumType = targetType->ToTypeInstance();
- PopulateType(enumType);
- StringT<128> enumCaseName;
- if (nameNode != NULL)
- nameNode->ToString(enumCaseName);
- auto tagType = GetPrimitiveType(BfTypeCode_Int32);
- if (enumVal.mType != enumType)
- {
- Fail(StrFormat("Cannot match enum type '%s' with type '%s'",
- TypeToString(enumVal.mType).c_str(), TypeToString(enumType).c_str()));
- enumVal = GetDefaultTypedValue(enumType);
- tagVal = GetDefaultTypedValue(tagType);
- }
- for (int fieldIdx = 0; fieldIdx < (int)enumType->mFieldInstances.size(); fieldIdx++)
- {
- auto fieldInstance = &enumType->mFieldInstances[fieldIdx];
- auto fieldDef = fieldInstance->GetFieldDef();
- if (fieldDef == NULL)
- continue;
- if ((fieldInstance->mIsEnumPayloadCase) && (fieldDef->mName == enumCaseName))
- {
- if ((!enumType->IsTypeMemberIncluded(fieldDef->mDeclaringType, activeTypeDef, this)) ||
- (!enumType->IsTypeMemberAccessible(fieldDef->mDeclaringType, activeTypeDef)))
- continue;
- auto resolvePassData = mCompiler->mResolvePassData;
- if (resolvePassData != NULL)
- {
- if (resolvePassData->mGetSymbolReferenceKind == BfGetSymbolReferenceKind_Field)
- resolvePassData->HandleFieldReference(nameNode, enumType->mTypeDef, fieldDef);
- String filter;
- auto autoComplete = resolvePassData->mAutoComplete;
- if ((autoComplete != NULL) && (autoComplete->InitAutocomplete(dotNode, nameNode, filter)))
- autoComplete->AddEnumTypeMembers(enumType, enumCaseName, false, enumType == mCurTypeInstance);
- }
- BF_ASSERT(fieldInstance->mResolvedType->IsTuple());
- auto tupleType = (BfTupleType*)fieldInstance->mResolvedType;
- PopulateType(tupleType);
- mBfIRBuilder->PopulateType(tupleType);
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- tagId = -fieldInstance->mDataIdx - 1;
- auto startBlock = mBfIRBuilder->GetInsertBlock();
- auto dscrType = enumType->GetDiscriminatorType();
- BfIRValue eqResult = mBfIRBuilder->CreateCmpEQ(tagVal.mValue, mBfIRBuilder->CreateConst(dscrType->mTypeDef->mTypeCode, tagId));
- BfIRBlock falseBlock;
- BfIRBlock doneBlock;
- if (notEqBlock != NULL)
- doneBlock = *notEqBlock;
- else
- doneBlock = mBfIRBuilder->CreateBlock("caseDone", false);
- if (clearOutOnMismatch)
- {
- falseBlock = mBfIRBuilder->CreateBlock("caseNotEq", false);
- mBfIRBuilder->AddBlock(falseBlock);
- }
- BfIRBlock matchedBlock = mBfIRBuilder->CreateBlock("caseMatch", false);
- if (matchBlock != NULL)
- *matchBlock = matchedBlock;
- mBfIRBuilder->CreateCondBr(eqResult, matchedBlock, falseBlock ? falseBlock : doneBlock);
-
- mBfIRBuilder->AddBlock(matchedBlock);
-
- mBfIRBuilder->SetInsertPoint(doneBlock);
- BfIRValue phiVal;
- if (eqBlock == NULL)
- phiVal = mBfIRBuilder->CreatePhi(mBfIRBuilder->MapType(boolType), 1 + (int)tupleType->mFieldInstances.size());
- if (phiVal)
- {
- auto falseVal = mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0);
- if (falseBlock)
- mBfIRBuilder->AddPhiIncoming(phiVal, falseVal, falseBlock);
- else
- mBfIRBuilder->AddPhiIncoming(phiVal, falseVal, startBlock);
- }
- mBfIRBuilder->SetInsertPoint(matchedBlock);
-
- BfTypedValue tupleVal;
- if (!enumVal.IsAddr())
- {
- auto unionInnerType = enumType->GetUnionInnerType();
- if (unionInnerType == tupleType)
- {
- tupleVal = ExtractValue(enumVal, NULL, 1);
- }
- }
- if (!tupleVal)
- {
- if (!tupleType->IsValuelessType())
- {
- tupleVal = ExtractValue(enumVal, NULL, 1);
- tupleVal = Cast(NULL, tupleVal, tupleType, BfCastFlags_Force);
- }
- else
- tupleVal = GetDefaultTypedValue(tupleType);
- }
- BfAstNode* tooFewRef = invocationExpr->mCloseParen;
- if ((tooFewRef == NULL) && (!invocationExpr->mCommas.IsEmpty()))
- tooFewRef = invocationExpr->mCommas[invocationExpr->mCommas.size() - 1];
- else if (tooFewRef == NULL)
- tooFewRef = invocationExpr->mOpenParen;
- ///
- auto autoComplete = mCompiler->GetAutoComplete();
- bool wasCapturingMethodInfo = false;
- if (autoComplete != NULL)
- {
- wasCapturingMethodInfo = autoComplete->mIsCapturingMethodMatchInfo;
- autoComplete->CheckInvocation(invocationExpr, invocationExpr->mOpenParen, invocationExpr->mCloseParen, invocationExpr->mCommas);
- if (autoComplete->mIsCapturingMethodMatchInfo)
- {
- autoComplete->mMethodMatchInfo->mInstanceList.Clear();
- auto methodMatchInfo = autoComplete->mMethodMatchInfo;
- BfAutoComplete::MethodMatchEntry methodMatchEntry;
- methodMatchEntry.mTypeInstance = enumType;
- methodMatchEntry.mCurMethodInstance = mCurMethodInstance;
- methodMatchEntry.mPayloadEnumField = fieldInstance;
- autoComplete->mMethodMatchInfo->mInstanceList.push_back(methodMatchEntry);
- methodMatchInfo->mBestIdx = 0;
- methodMatchInfo->mMostParamsMatched = 0;
- int cursorIdx = invocationExpr->GetParser()->mCursorIdx;
- if ((invocationExpr->mCloseParen == NULL) || (cursorIdx <= invocationExpr->mCloseParen->GetSrcStart()))
- {
- int paramIdx = 0;
- for (int commaIdx = 0; commaIdx < (int)invocationExpr->mCommas.size(); commaIdx++)
- {
- auto commaNode = invocationExpr->mCommas[commaIdx];
- if ((commaNode != NULL) && (cursorIdx >= commaNode->GetSrcStart()))
- paramIdx = commaIdx + 1;
- }
- bool isEmpty = true;
- if (paramIdx < (int)invocationExpr->mArguments.size())
- {
- auto paramNode = invocationExpr->mArguments[paramIdx];
- if (paramNode != NULL)
- isEmpty = false;
- }
- if (isEmpty)
- {
- if (paramIdx < (int)tupleType->mFieldInstances.size())
- {
- auto fieldDef = tupleType->mFieldInstances[paramIdx].GetFieldDef();
- String insertStr;
- if (fieldDef->IsUnnamedTupleField())
- insertStr = "p";
- insertStr += fieldDef->mName;
-
- insertStr.Insert(0, "let ");
- autoComplete->mEntriesSet.Clear();
- autoComplete->AddEntry(AutoCompleteEntry("paramName", insertStr));
- autoComplete->mInsertStartIdx = cursorIdx;
- autoComplete->mInsertEndIdx = cursorIdx;
- }
- }
- }
- }
- }
- defer
- (
- if (autoComplete != NULL)
- autoComplete->mIsCapturingMethodMatchInfo = (wasCapturingMethodInfo) && (!autoComplete->mIsCapturingMethodMatchInfo);
- );
- ///
- HandleCaseEnumMatch_Tuple(tupleVal, invocationExpr->mArguments, tooFewRef, falseBlock ? BfIRValue() : phiVal, matchedBlock, falseBlock ? falseBlock : doneBlock, hadConditional, clearOutOnMismatch);
- if (phiVal)
- {
- auto trueVal = mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 1);
- mBfIRBuilder->AddPhiIncoming(phiVal, trueVal, matchedBlock);
- }
- if (eqBlock != NULL)
- mBfIRBuilder->CreateBr(*eqBlock);
- else
- mBfIRBuilder->CreateBr(doneBlock);
- if (falseBlock)
- {
- mBfIRBuilder->SetInsertPoint(falseBlock);
- mBfIRBuilder->CreateBr(doneBlock);
- //mBfIRBuilder->AddPhiIncoming(phiVal, mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0), falseBlock);
- }
- mBfIRBuilder->AddBlock(doneBlock);
- mBfIRBuilder->SetInsertPoint(doneBlock);
- if (phiVal)
- return BfTypedValue(phiVal, boolType);
- else
- return GetDefaultTypedValue(boolType);
- }
- }
- return BfTypedValue();
- }
- BfTypedValue BfModule::HandleCaseBind(BfTypedValue enumVal, const BfTypedValue& tagVal, BfEnumCaseBindExpression* bindExpr, BfIRBlock* eqBlock, BfIRBlock* notEqBlock, BfIRBlock* matchBlock, int* outEnumIdx)
- {
- BfTypeInstance* tupleType = NULL;
- auto activeTypeDef = GetActiveTypeDef();
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- BfIRValue eqResult;
- if (bindExpr->mEnumMemberExpr != NULL)
- {
- int enumIdx = -1;
- String findName;
- BfType* type = NULL;
- BfAstNode* targetNode = NULL;
- BfAstNode* nameNode = NULL;
- BfAstNode* dotNode = NULL;
- if (auto memberExpr = BfNodeDynCast<BfMemberReferenceExpression>(bindExpr->mEnumMemberExpr))
- {
- dotNode = memberExpr->mDotToken;
- if (memberExpr->mMemberName != NULL)
- {
- nameNode = memberExpr->mMemberName;
- findName = memberExpr->mMemberName->ToString();
- if (memberExpr->mTarget == NULL)
- {
- type = enumVal.mType;
- }
- else if (auto typeRef = BfNodeDynCast<BfTypeReference>(memberExpr->mTarget))
- {
- type = ResolveTypeRef(typeRef);
- }
- }
- targetNode = memberExpr->mTarget;
- }
- else if (auto identiferNode = BfNodeDynCast<BfIdentifierNode>(bindExpr->mEnumMemberExpr))
- {
- if (mCurTypeInstance->IsPayloadEnum())
- {
- nameNode = identiferNode;
- findName = nameNode->ToString();
- targetNode = identiferNode;
- type = mCurTypeInstance;
- }
- else
- {
- Fail("Expected a qualified enum case name. Consider prefixing name with a dot to infer enum type name.", bindExpr->mEnumMemberExpr);
- }
- }
- if (!findName.empty())
- {
- if (type != NULL)
- {
- if (type != enumVal.mType)
- {
- Fail(StrFormat("Enum case type '%s' does not match compared value of type '%s'",
- TypeToString(enumVal.mType).c_str(), TypeToString(type).c_str()), targetNode);
- }
- if (type->IsEnum())
- {
- auto enumType = (BfTypeInstance*)type;
- for (auto fieldInstance : enumType->mFieldInstances)
- {
- auto fieldDef = fieldInstance.GetFieldDef();
- if ((fieldDef != NULL) && (fieldDef->IsEnumCaseEntry()) && (fieldDef->mName == findName))
- {
- if ((!enumType->IsTypeMemberIncluded(fieldDef->mDeclaringType, activeTypeDef, this)) ||
- (!enumType->IsTypeMemberAccessible(fieldDef->mDeclaringType, activeTypeDef)))
- continue;
- auto resolvePassData = mCompiler->mResolvePassData;
- if (resolvePassData != NULL)
- {
- if (resolvePassData->mGetSymbolReferenceKind == BfGetSymbolReferenceKind_Field)
- resolvePassData->HandleFieldReference(nameNode, enumType->mTypeDef, fieldDef);
- String filter;
- auto autoComplete = resolvePassData->mAutoComplete;
- if ((autoComplete != NULL) && (autoComplete->InitAutocomplete(dotNode, nameNode, filter)))
- autoComplete->AddEnumTypeMembers(enumType, findName, false, enumType == mCurTypeInstance);
- }
- enumIdx = -fieldInstance.mDataIdx - 1;
- if (outEnumIdx != NULL)
- *outEnumIdx = enumIdx;
- if (fieldInstance.mIsEnumPayloadCase)
- tupleType = fieldInstance.mResolvedType->ToTypeInstance();
- }
- }
- if (enumIdx == -1)
- {
- Fail("Enum case not found", nameNode);
- }
- }
- else
- {
- Fail(StrFormat("Type '%s' is not an enum type", TypeToString(type).c_str()), targetNode);
- }
- }
- }
- BF_ASSERT(tagVal.mType->IsPrimitiveType());
- eqResult = mBfIRBuilder->CreateCmpEQ(tagVal.mValue, mBfIRBuilder->CreateConst(((BfPrimitiveType*)tagVal.mType)->mTypeDef->mTypeCode, enumIdx));
- }
- else
- {
- eqResult = mBfIRBuilder->CreateConst(BfTypeCode_Boolean, 0);
- }
- BfIRBlock falseBlock;
- if (notEqBlock != NULL)
- {
- falseBlock = *notEqBlock;
- }
- else
- {
- falseBlock = mBfIRBuilder->CreateBlock("notEqBlock", false);
- }
- auto mainBlock = mBfIRBuilder->GetInsertBlock();
- BfIRBlock trueBlock = mBfIRBuilder->CreateBlock("eqBlock", false);
- if (matchBlock != NULL)
- *matchBlock = trueBlock;
- mBfIRBuilder->AddBlock(trueBlock);
- mBfIRBuilder->SetInsertPoint(trueBlock);
- if ((tupleType != NULL) && (bindExpr->mBindNames != NULL))
- {
- BfIRValue valueScopeStart;
- if (IsTargetingBeefBackend())
- valueScopeStart = mBfIRBuilder->CreateValueScopeStart();
- bool isVar = bindExpr->mBindToken->GetToken() == BfToken_Var;
- bool isLet = bindExpr->mBindToken->GetToken() == BfToken_Let;
- BfTypedValue tupleVal;
- if (enumVal.IsAddr())
- {
- auto ptrVal = mBfIRBuilder->CreateInBoundsGEP(enumVal.mValue, 0, 1);
- tupleVal = BfTypedValue(mBfIRBuilder->CreateBitCast(ptrVal, mBfIRBuilder->MapTypeInstPtr(tupleType)), tupleType, true);
- }
- else
- {
- auto unionInnerType = enumVal.mType->ToTypeInstance()->GetUnionInnerType();
- tupleVal = ExtractValue(enumVal, NULL, 1);
- if (unionInnerType != tupleType)
- {
- tupleVal = MakeAddressable(tupleVal);
- tupleVal = BfTypedValue(mBfIRBuilder->CreateBitCast(tupleVal.mValue, mBfIRBuilder->MapTypeInstPtr(tupleType)), tupleType, true);
- }
- }
- HandleTupleVariableDeclaration(NULL, bindExpr->mBindNames, tupleVal, isLet, false, true /*, &mainBlock*/);
- auto autoComplete = mCompiler->GetAutoComplete();
- if ((autoComplete != NULL) && ((isVar || isLet)))
- autoComplete->CheckVarResolution(bindExpr->mBindToken, tupleType);
- if (valueScopeStart)
- mBfIRBuilder->CreateValueScopeSoftEnd(valueScopeStart);
- }
- if (eqBlock != NULL)
- mBfIRBuilder->CreateBr(*eqBlock);
- else
- mBfIRBuilder->CreateBr(falseBlock);
- // Don't create the condBr until now, so HandleTupleVariableDeclaration can create the variable declarations in mainBlock--
- // we need them there since the code that uses the new variables is created outside the eqBlock
- mBfIRBuilder->SetInsertPoint(mainBlock);
- mBfIRBuilder->CreateCondBr(eqResult, trueBlock, falseBlock);
- mBfIRBuilder->AddBlock(falseBlock);
- mBfIRBuilder->SetInsertPoint(falseBlock);
- return BfTypedValue(eqResult, boolType);
- }
- void BfModule::AddBasicBlock(BfIRBlock bb, bool activate)
- {
- mBfIRBuilder->AddBlock(bb);
- if (activate)
- mBfIRBuilder->SetInsertPoint(bb);
- }
- void BfModule::VisitEmbeddedStatement(BfAstNode* stmt, BfExprEvaluator* exprEvaluator, BfEmbeddedStatementFlags flags)
- {
- auto block = BfNodeDynCast<BfBlock>(stmt);
- BfLabelNode* labelNode = NULL;
- if (block == NULL)
- {
- auto labeledBlock = BfNodeDynCast<BfLabeledBlock>(stmt);
- if (labeledBlock != NULL)
- {
- block = labeledBlock->mBlock;
- labelNode = labeledBlock->mLabelNode;
- }
- }
- BfAstNode* openBrace = NULL;
- BfAstNode* closeBrace = NULL;
- if (block != NULL)
- {
- openBrace = block->mOpenBrace;
- closeBrace = block->mCloseBrace;
- if (openBrace == NULL)
- {
- auto checkScope = mCurMethodState->mCurScope;
- while ((checkScope != NULL) && (closeBrace == NULL))
- {
- closeBrace = checkScope->mCloseNode;
- checkScope = checkScope->mPrevScope;
- }
- BF_ASSERT(closeBrace != NULL);
- }
- }
- if ((block != NULL) && (openBrace != NULL))
- UpdateSrcPos(openBrace);
- if (mCurMethodState != NULL)
- {
- bool isIgnore = mBfIRBuilder->mIgnoreWrites;
- SetAndRestoreValue<bool> prevEmbedded(mCurMethodState->mIsEmbedded, block == NULL);
- mCurMethodState->mInHeadScope = false;
- BfScopeData scopeData;
- if (IsTargetingBeefBackend())
- scopeData.mValueScopeStart = mBfIRBuilder->CreateValueScopeStart();
- mCurMethodState->AddScope(&scopeData);
- if (block != NULL)
- {
- mCurMethodState->mCurScope->mAstBlock = block;
- mCurMethodState->mCurScope->mCloseNode = closeBrace;
- }
- if (labelNode != NULL)
- scopeData.mLabelNode = labelNode->mLabel;
- NewScopeState(block != NULL);
- mCurMethodState->mCurScope->mOuterIsConditional = (flags & BfEmbeddedStatementFlags_IsConditional) != 0;
- mCurMethodState->mCurScope->mIsDeferredBlock = (flags & BfEmbeddedStatementFlags_IsDeferredBlock) != 0;
- mCurMethodState->mCurScope->mExprEvaluator = exprEvaluator;
-
- //
- {
- SetAndRestoreValue<bool> inDeferredBlock(mCurMethodState->mInDeferredBlock, mCurMethodState->mInDeferredBlock || mCurMethodState->mCurScope->mIsDeferredBlock);
- if (block != NULL)
- {
- if (labelNode != NULL)
- VisitCodeBlock(block, BfIRBlock(), BfIRBlock(), BfIRBlock(), false, NULL, labelNode);
- else
- VisitCodeBlock(block);
- }
- else
- VisitChild(stmt);
- }
- if ((block != NULL) && (closeBrace != NULL))
- {
- UpdateSrcPos(closeBrace);
- if (!mCurMethodState->mLeftBlockUncond)
- EmitEnsureInstructionAt();
- }
- if (block != NULL)
- {
- BfAutoParentNodeEntry autoParentNodeEntry(this, block);
- RestoreScopeState();
- }
- else
- RestoreScopeState();
- BF_ASSERT(isIgnore == mBfIRBuilder->mIgnoreWrites);
- }
- else
- {
- if (block != NULL)
- VisitCodeBlock(block);
- else
- VisitChild(stmt);
- }
- }
- void BfModule::VisitCodeBlock(BfBlock* block, BfIRBlock continueBlock, BfIRBlock breakBlock, BfIRBlock fallthroughBlock, bool defaultBreak, bool* hadReturn, BfLabelNode* labelNode, bool closeScope)
- {
- BfBreakData breakData;
- breakData.mIRContinueBlock = continueBlock;
- breakData.mIRBreakBlock = breakBlock;
- breakData.mIRFallthroughBlock = fallthroughBlock;
- breakData.mScope = mCurMethodState->mCurScope;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- Visit(block);
- if (closeScope)
- RestoreScopeState();
- if ((!mCurMethodState->mLeftBlockUncond) && (defaultBreak))
- {
- mBfIRBuilder->CreateBr(breakBlock);
- }
- if (hadReturn != NULL)
- {
- *hadReturn = mCurMethodState->mHadReturn;
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- }
- }
- void BfModule::VisitCodeBlock(BfBlock* block)
- {
- BP_ZONE("BfModule::VisitCodeBlock");
- BfAutoParentNodeEntry autoParentNodeEntry(this, block);
- BfIRBlock prevInsertBlock;
- bool hadReturn = false;
-
- BfIRBlock postExitBlock;
- int startLocalMethod = 0; // was -1
- auto rootMethodState = mCurMethodState->GetRootMethodState();
- bool allowLocalMethods = mCurMethodInstance != NULL;
- //int startDeferredLocalIdx = (int)rootMethodState->mDeferredLocalMethods.size();
-
- int curLocalMethodIdx = -1;
- // Scan for any local method declarations
- if (allowLocalMethods)
- {
- startLocalMethod = (int)mCurMethodState->mLocalMethods.size();
- curLocalMethodIdx = startLocalMethod;
- auto itr = block->begin();
- while (itr != block->end())
- {
- BfAstNode* child = *itr;
- if (auto localMethodDecl = BfNodeDynCastExact<BfLocalMethodDeclaration>(child))
- {
- BfLocalMethod* localMethod;
- auto rootMethodState = mCurMethodState->GetRootMethodState();
-
- String methodName;
- if (localMethodDecl->mMethodDeclaration->mNameNode != NULL)
- {
- methodName = GetLocalMethodName(localMethodDecl->mMethodDeclaration->mNameNode->ToString(), localMethodDecl->mMethodDeclaration->mOpenParen, mCurMethodState, mCurMethodState->mMixinState);
- BfLocalMethod** localMethodPtr = NULL;
- if (rootMethodState->mLocalMethodCache.TryGetValue(methodName, &localMethodPtr))
- {
- localMethod = *localMethodPtr;
- }
- else
- {
- localMethod = new BfLocalMethod();
- localMethod->mSystem = mSystem;
- localMethod->mModule = this;
- localMethod->mMethodDeclaration = localMethodDecl->mMethodDeclaration;
- localMethod->mSource = mCurTypeInstance->mTypeDef->mSource;
- localMethod->mSource->mRefCount++;
- if (mCurMethodState->mClosureState != NULL)
- localMethod->mOuterLocalMethod = mCurMethodState->mClosureState->mLocalMethod;
- auto autoComplete = mCompiler->GetAutoComplete();
- if ((autoComplete != NULL) && (autoComplete->mResolveType == BfResolveType_Autocomplete))
- {
- auto autoComplete = mCompiler->mResolvePassData->mAutoComplete;
- if (!autoComplete->IsAutocompleteNode(localMethod->mMethodDeclaration))
- localMethod->mDeclOnly = true;
- }
- if (localMethod->mMethodDeclaration->mNameNode != NULL)
- localMethod->mMethodName = localMethod->mMethodDeclaration->mNameNode->ToString();
- localMethod->mExpectedFullName = methodName;
- rootMethodState->mLocalMethodCache[methodName] = localMethod;
- mContext->mLocalMethodGraveyard.push_back(localMethod);
- }
- BF_ASSERT(mCurMethodState->mCurScope != NULL);
- localMethod->mDeclDIScope = mCurMethodState->mCurScope->mDIScope;
- localMethod->mDeclMethodState = mCurMethodState;
- localMethod->mDeclMixinState = mCurMethodState->mMixinState;
- if (localMethod->mDeclMixinState != NULL)
- localMethod->mDeclMixinState->mHasDeferredUsage = true;
- mCurMethodState->mLocalMethods.push_back(localMethod);
- String* namePtr;
- if (!mCurMethodState->mLocalMethodMap.TryAdd(localMethod->mMethodName, &namePtr, &localMethodPtr))
- {
- BF_ASSERT(localMethod != *localMethodPtr);
- localMethod->mNextWithSameName = *localMethodPtr;
- }
- *localMethodPtr = localMethod;
- }
- }
- ++itr;
- }
- }
- bool wantsAllLocalMethods = true;
- auto autoComplete = mCompiler->GetAutoComplete();
- if (autoComplete != NULL)
- {
- // If we only need reasoning "at the cursor" then we don't need all local methods
- if ((!autoComplete->mIsAutoComplete) ||
- (autoComplete->mResolveType == BfResolveType_GetCurrentLocation) ||
- (autoComplete->mResolveType == BfResolveType_GetFixits) ||
- (autoComplete->mResolveType == BfResolveType_GetVarType) ||
- (autoComplete->mResolveType == BfResolveType_GetSymbolInfo) ||
- (autoComplete->mResolveType == BfResolveType_ShowFileSymbolReferences))
- wantsAllLocalMethods = false;
- }
- /*if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mResolveType == BfResolveType_ShowFileSymbolReferences))
- {
- if (mCompiler->mResolvePassData->mSymbolReferenceLocalIdx != -1)
- {
- // We need to reproduce the behavior when we found the symbol - only process autocomplete nodes, otherwise local method ids will be wrong
- // when we have local methods that were skipped the first time but not the second
- wantsAllLocalMethods = false;
- }
- }*/
- // Handle statements
- auto itr = block->begin();
- while (itr != block->end())
- {
- BfAstNode* child = *itr;
- if (auto localMethodDecl = BfNodeDynCastExact<BfLocalMethodDeclaration>(child))
- {
- /*if ((mCurMethodInstance != NULL) && (mCurMethodInstance->mMethodDef->mMethodType == BfMethodType_Mixin))
- Fail("Mixins cannot contain local methods", child);*/
- if (!allowLocalMethods)
- {
- Fail("Invalid use of local methods", child);
- }
- else if (localMethodDecl->mMethodDeclaration->mNameNode != NULL)
- {
- BfLocalMethod* localMethod = mCurMethodState->mLocalMethods[curLocalMethodIdx];
- BF_ASSERT(localMethod->mMethodDeclaration == localMethodDecl->mMethodDeclaration);
- if ((wantsAllLocalMethods) || (autoComplete->IsAutocompleteNode(localMethod->mMethodDeclaration)))
- {
- if (!mCurMethodInstance->IsSpecializedGenericMethodOrType())
- GetLocalMethodInstance(localMethod, BfTypeVector(), NULL, true); // Only necessary on unspecialized pass
- }
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- mCompiler->mResolvePassData->mAutoComplete->CheckMethod(localMethod->mMethodDeclaration, true);
- curLocalMethodIdx++;
- }
- ++itr;
- continue;
- }
- if ((mCurMethodState != NULL) && (mCurMethodState->mLeftBlockUncond)) // mLeftBlock is cleared after conditional block is completed
- {
- if (mCurMethodState->mHadReturn)
- hadReturn = true;
-
- if ((!postExitBlock) && (!mCurMethodState->mInPostReturn))
- {
- Warn(BfWarning_CS0162_UnreachableCode, "Unreachable code", child);
- prevInsertBlock = mBfIRBuilder->GetInsertBlock();
- mCurMethodState->mInPostReturn = true;
- postExitBlock = mBfIRBuilder->CreateBlock("tempPostExit", true);
- mBfIRBuilder->SetInsertPoint(postExitBlock);
- }
- }
- if (itr.IsLast())
- {
- // if (auto exprStmt = BfNodeDynCast<BfExpressionStatement>(child))
- // {
- // BF_FATAL("Happens?");
- //
- // auto expr = exprStmt->mExpression;
- // if (exprStmt->IsMissingSemicolon())
- // {
- // if (mCurMethodState != NULL)
- // {
- // if (mCurMethodState->mCurScope->mExprEvaluator != NULL)
- // {
- // // Evaluate last child as an expression
- // mCurMethodState->mCurScope->mExprEvaluator->VisitChild(expr);
- // mCurMethodState->mCurScope->mExprEvaluator->FinishExpressionResult();
- // break;
- // }
- // else if (mCurMethodState->InMainMixinScope())
- // {
- // mCurMethodState->mMixinState->mResultExpr = expr;
- // break;
- // }
- // else if ((mCurMethodInstance->IsMixin()) && (mCurMethodState->mCurScope == &mCurMethodState->mHeadScope))
- // {
- // // Silently allow...
- // }
- // else
- // {
- // FailAfter("Expression block cannot be used here. Consider adding semicolon if a statement was intended.", expr);
- // }
- // }
- // }
- // }
- // else
- if (auto expr = BfNodeDynCast<BfExpression>(child))
- {
- if (expr->IsExpression())
- {
- if (mCurMethodState != NULL)
- {
- if (mCurMethodState->mCurScope->mExprEvaluator != NULL)
- {
- // Evaluate last child as an expression
- mCurMethodState->mCurScope->mExprEvaluator->VisitChild(expr);
- mCurMethodState->mCurScope->mExprEvaluator->FinishExpressionResult();
- break;
- }
- else if (mCurMethodState->InMainMixinScope())
- {
- mCurMethodState->mMixinState->mResultExpr = expr;
- break;
- }
- else if ((mCurMethodInstance->IsMixin()) && (mCurMethodState->mCurScope == &mCurMethodState->mHeadScope))
- {
- // Silently allow...
- }
- else
- {
- FailAfter("Expression block cannot be used here. Consider adding semicolon if a statement was intended.", expr);
- }
- }
- }
- }
- }
- UpdateSrcPos(child);
- BfAutoParentNodeEntry autoParentNode(this, child);
- child->Accept(this);
-
- mSystem->CheckLockYield();
- ++itr;
- }
- if (mCurMethodState != NULL)
- {
- // Any local method that hasn't been called needs to be processed now
- for (int localMethodIdx = startLocalMethod; localMethodIdx < (int)mCurMethodState->mLocalMethods.size(); localMethodIdx++)
- {
- auto localMethod = mCurMethodState->mLocalMethods[localMethodIdx];
- if ((wantsAllLocalMethods) || (autoComplete->IsAutocompleteNode(localMethod->mMethodDeclaration)))
- {
- //??
- auto moduleMethodInstance = GetLocalMethodInstance(localMethod, BfTypeVector(), NULL, true);
- }
- mSystem->CheckLockYield();
- }
- while ((int)mCurMethodState->mLocalMethods.size() > startLocalMethod)
- {
- auto localMethod = mCurMethodState->mLocalMethods.back();
-
- #if _DEBUG
- BfLocalMethod** localMethodPtr = NULL;
- mCurMethodState->mLocalMethodMap.TryGetValue(localMethod->mMethodName, &localMethodPtr);
- BF_ASSERT(*localMethodPtr == localMethod);
- #endif
- if (localMethod->mNextWithSameName == NULL)
- mCurMethodState->mLocalMethodMap.Remove(localMethod->mMethodName);
- else
- {
- mCurMethodState->mLocalMethodMap[localMethod->mMethodName] = localMethod->mNextWithSameName;
- localMethod->mNextWithSameName = NULL;
- }
-
- mCurMethodState->mLocalMethods.pop_back();
- }
- if (postExitBlock)
- {
- if (hadReturn)
- mCurMethodState->SetHadReturn(true);
- mCurMethodState->mLeftBlockUncond = true;
- mCurMethodState->mInPostReturn = false;
- mBfIRBuilder->DropBlocks(postExitBlock);
- if (prevInsertBlock)
- mBfIRBuilder->SetInsertPoint(prevInsertBlock);
- }
- }
- }
- void BfModule::Visit(BfAstNode* astNode)
- {
- AssertErrorState();
- }
- void BfModule::Visit(BfIdentifierNode* identifierNode)
- {
- Visit((BfExpression*)identifierNode);
- }
- void BfModule::Visit(BfTypeReference* typeRef)
- {
- Visit((BfAstNode*)typeRef);
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- mCompiler->mResolvePassData->mAutoComplete->CheckTypeRef(typeRef, true);
- }
- void BfModule::Visit(BfEmptyStatement* astNode)
- {
- }
- void BfModule::Visit(BfTryStatement* tryStmt)
- {
- Fail("Exceptions not supported", tryStmt->mTryToken);
- VisitChild(tryStmt->mStatement);
- }
- void BfModule::Visit(BfCatchStatement* catchStmt)
- {
- Fail("Exceptions not supported", catchStmt->mCatchToken);
- }
- void BfModule::Visit(BfFinallyStatement* finallyStmt)
- {
- Fail("Exceptions not supported", finallyStmt->mFinallyToken);
- VisitChild(finallyStmt->mStatement);
- }
- void BfModule::Visit(BfCheckedStatement* checkedStmt)
- {
- Fail("'checked' not supported", checkedStmt->mCheckedToken);
- VisitChild(checkedStmt->mStatement);
- }
- void BfModule::Visit(BfUncheckedStatement* uncheckedStmt)
- {
- VisitChild(uncheckedStmt->mStatement);
- }
- void BfModule::DoIfStatement(BfIfStatement* ifStmt, bool includeTrueStmt, bool includeFalseStmt)
- {
- if (ifStmt->mCondition == NULL)
- {
- AssertErrorState();
- return;
- }
- //TODO: Only conditionally create the scopeData here if we create a variable inside the condition statement
- UpdateSrcPos(ifStmt);
- BfScopeData newScope;
- newScope.mOuterIsConditional = true;
- if (ifStmt->mLabelNode != NULL)
- newScope.mLabelNode = ifStmt->mLabelNode->mLabel;
- mCurMethodState->AddScope(&newScope);
- NewScopeState();
- BfBreakData breakData;
- breakData.mScope = &newScope;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- BfDeferredLocalAssignData deferredLocalAssignData(mCurMethodState->mCurScope);
- deferredLocalAssignData.mIsIfCondition = true;
- deferredLocalAssignData.ExtendFrom(mCurMethodState->mDeferredLocalAssignData, true);
- deferredLocalAssignData.mVarIdBarrier = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignData);
- BfAutoParentNodeEntry autoParentNodeEntry(this, ifStmt);
- BfTypedValue condValue = CreateValueFromExpression(ifStmt->mCondition, boolType);
- deferredLocalAssignData.mIsIfCondition = false;
- // The "extend chain" is only valid for the conditional -- since that expression may contain unconditionally executed and
- // conditionally executed code (in the case of "(GetVal(out a) && GetVal(out b))" for example
- mCurMethodState->mDeferredLocalAssignData->BreakExtendChain();
- if (!condValue)
- {
- AssertErrorState();
- condValue = BfTypedValue(GetDefaultValue(boolType), boolType);
- }
-
- BfIRBlock trueBB;
- BfIRBlock falseBB;
- bool isConstBranch = false;
- bool constResult = false;
- if (condValue.mValue.IsConst())
- {
- auto constant = mBfIRBuilder->GetConstant(condValue.mValue);
- if ((constant != NULL) && (constant->mTypeCode == BfTypeCode_Boolean))
- {
- isConstBranch = true;
- constResult = constant->mBool;
- }
- }
- if (!isConstBranch)
- {
- //trueBB = mBfIRBuilder->CreateBlock(StrFormat("if.then_%d", mBfIRBuilder->mStream.GetSize()), true);
- trueBB = mBfIRBuilder->CreateBlock("if.then", true);
- falseBB = (ifStmt->mFalseStatement == NULL) ? BfIRBlock() : mBfIRBuilder->CreateBlock("if.else");
- }
- else
- EmitEnsureInstructionAt();
- auto contBB = mBfIRBuilder->CreateBlock("if.end");
- if (!isConstBranch)
- {
- mBfIRBuilder->CreateCondBr(condValue.mValue, trueBB, (falseBB) ? falseBB : contBB);
- }
- // TRUE statement
- bool ignoredLastBlock = true;
- if (includeTrueStmt)
- {
- SetAndRestoreValue<bool> ignoreWrites(mBfIRBuilder->mIgnoreWrites);
- if (trueBB)
- mBfIRBuilder->SetInsertPoint(trueBB);
- if ((isConstBranch) && (constResult != true))
- mBfIRBuilder->mIgnoreWrites = true;
- else
- ignoredLastBlock = false;
- VisitEmbeddedStatement(ifStmt->mTrueStatement);
- }
- prevDLA.Restore();
- bool trueHadReturn = mCurMethodState->mHadReturn;
- // We restore the scopeData before the False block because we don't want variables created in the if condition to
- // be visible in the false section
- //RestoreScopeState();
- RestoreScoreState_LocalVariables();
- if ((!mCurMethodState->mLeftBlockUncond) && (!ignoredLastBlock))
- mBfIRBuilder->CreateBr_NoCollapse(contBB);
- if (mCurMethodState->mLeftBlockUncond)
- mCurMethodState->mLeftBlockCond = true;
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->SetHadReturn(false);
-
- bool falseHadReturn = false;
- if (ifStmt->mFalseStatement != NULL)
- {
- BfDeferredLocalAssignData falseDeferredLocalAssignData(mCurMethodState->mCurScope);
- if (falseBB)
- {
- mBfIRBuilder->AddBlock(falseBB);
- mBfIRBuilder->SetInsertPoint(falseBB);
- }
-
- ignoredLastBlock = true;
- //
- {
- SetAndRestoreValue<bool> ignoreWrites(mBfIRBuilder->mIgnoreWrites);
- if ((isConstBranch) && (constResult != false))
- mBfIRBuilder->mIgnoreWrites = true;
- else
- ignoredLastBlock = false;
- falseDeferredLocalAssignData.ExtendFrom(mCurMethodState->mDeferredLocalAssignData);
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &falseDeferredLocalAssignData);
- if (includeFalseStmt)
- VisitEmbeddedStatement(ifStmt->mFalseStatement, NULL, BfEmbeddedStatementFlags_IsConditional);
- }
- if ((!mCurMethodState->mLeftBlockUncond) && (!ignoredLastBlock))
- {
-
- if (IsTargetingBeefBackend())
- {
- // If we don't do this, then with:
- // if (a) { } else if (b) { }
- // Then we hit the closing second brace even if 'b' is false
- //SetIllegalSrcPos();
- //BfIRBuilder->ClearDebugLocation();
- }
- auto br = mBfIRBuilder->CreateBr_NoCollapse(contBB);
- //mBfIRBuilder->ClearDebugLocation(br);
- }
- falseHadReturn = mCurMethodState->mHadReturn;
- if (mCurMethodState->mLeftBlockUncond)
- mCurMethodState->mLeftBlockCond = true;
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->SetHadReturn(false);
- deferredLocalAssignData.SetIntersection(falseDeferredLocalAssignData);
- mCurMethodState->ApplyDeferredLocalAssignData(deferredLocalAssignData);
- }
- else
- {
- // If we had a const-ignored if statement with no else
- if (ignoredLastBlock)
- {
- if (!mCurMethodState->mLeftBlockUncond)
- mBfIRBuilder->CreateBr_NoCollapse(contBB);
- }
- }
- mBfIRBuilder->AddBlock(contBB);
- mBfIRBuilder->SetInsertPoint(contBB);
-
- if (isConstBranch)
- mCurMethodState->SetHadReturn(constResult ? trueHadReturn : falseHadReturn);
- else
- mCurMethodState->SetHadReturn(trueHadReturn && falseHadReturn);
- mCurMethodState->mLeftBlockUncond = mCurMethodState->mHadReturn;
- if (mCurMethodState->mHadReturn)
- {
- mBfIRBuilder->EraseFromParent(contBB);
- }
- else
- {
- mBfIRBuilder->SetInsertPoint(contBB);
- }
- RestoreScopeState();
- }
- void BfModule::Visit(BfIfStatement* ifStmt)
- {
- DoIfStatement(ifStmt, true, true);
- }
- void BfModule::Visit(BfVariableDeclaration* varDecl)
- {
- BP_ZONE("BfModule::Visit(BfVariableDeclaration)");
- UpdateSrcPos(varDecl);
- if (mCurMethodState->mIsEmbedded)
- Fail("Variable declarations must be wrapped in a block statement", varDecl);
- BfTupleExpression* tupleVariableDeclaration = BfNodeDynCast<BfTupleExpression>(varDecl->mNameNode);
- if (tupleVariableDeclaration != NULL)
- {
- HandleTupleVariableDeclaration(varDecl);
- }
- else
- HandleVariableDeclaration(varDecl);
- }
- void BfModule::Visit(BfLocalMethodDeclaration* methodDecl)
- {
- Fail("Local method declarations must be wrapped in a block statement", methodDecl->mMethodDeclaration->mNameNode);
- }
- void BfModule::Visit(BfExpression* expression)
- {
- UpdateSrcPos(expression);
- BfExprEvaluator exprEvaluator(this);
- exprEvaluator.mUsedAsStatement = true;
- exprEvaluator.Evaluate(expression);
- }
- void BfModule::Visit(BfExpressionStatement* expressionStmt)
- {
- expressionStmt->mExpression->Accept(this);
- }
- void BfModule::Visit(BfThrowStatement* throwStmt)
- {
- if (throwStmt->mExpression == NULL)
- {
- AssertErrorState();
- return;
- }
- UpdateSrcPos(throwStmt->mThrowToken);
- auto throwValue = CreateValueFromExpression(throwStmt->mExpression);
- if (throwValue)
- {
- //TODO: Actually call some sort of 'ExceptionThrown' function
- auto internalType = ResolveTypeDef(mCompiler->mInternalTypeDef);
- PopulateType(internalType);
- auto moduleMethodInstance = GetMethodByName(internalType->ToTypeInstance(), "Throw");
- if (!moduleMethodInstance)
- Fail("Internal error: System.Internal doesn't contain Throw method");
- else
- {
- auto exType = moduleMethodInstance.mMethodInstance->GetParamType(0);
- auto exTypedValue = Cast(throwStmt->mExpression, throwValue, exType);
- if ((exTypedValue) && (!mCompiler->IsSkippingExtraResolveChecks()))
- {
- SizedArray<BfIRValue, 1> llvmArgs;
- llvmArgs.push_back(exTypedValue.mValue);
- mBfIRBuilder->CreateCall(moduleMethodInstance.mFunc, llvmArgs);
- }
- }
- }
- if (mCurMethodInstance->mReturnType->IsVoid())
- EmitReturn(BfIRValue());
- else
- EmitReturn(GetDefaultValue(mCurMethodInstance->mReturnType));
- }
- void BfModule::Visit(BfDeleteStatement* deleteStmt)
- {
- UpdateSrcPos(deleteStmt);
- bool isAppendDelete = false;
- BfTypedValue customAllocator;
- if (deleteStmt->mAllocExpr != NULL)
- {
- if (auto expr = BfNodeDynCast<BfExpression>(deleteStmt->mAllocExpr))
- customAllocator = CreateValueFromExpression(expr);
- else if (auto tokenNode = BfNodeDynCast<BfTokenNode>(deleteStmt->mAllocExpr))
- {
- if (tokenNode->mToken == BfToken_Append)
- isAppendDelete = true;
- }
- }
- if (deleteStmt->mExpression == NULL)
- {
- AssertErrorState();
- return;
- }
- auto val = CreateValueFromExpression(deleteStmt->mExpression);
- if (!val)
- return;
- auto checkType = val.mType;
- if (val.mType->IsGenericParam())
- {
- auto genericParamInst = GetGenericParamInstance((BfGenericParamType*)checkType);
- if (genericParamInst->mTypeConstraint != NULL)
- checkType = genericParamInst->mTypeConstraint;
- if (genericParamInst->mGenericParamFlags & (BfGenericParamFlag_Delete | BfGenericParamFlag_Var))
- return;
- }
- if (checkType->IsVar())
- {
- // Mixin or unconstrained generic
- return;
- }
- if ((!checkType->IsPointer()) && (!checkType->IsObject()))
- {
- Fail(StrFormat("Cannot delete a value of type '%s'", TypeToString(val.mType).c_str()), deleteStmt->mDeleteToken);
- return;
- }
- if (val.mType->IsGenericParam())
- return;
- auto bodyBB = mBfIRBuilder->CreateBlock("delete.body");
- auto endBB = mBfIRBuilder->CreateBlock("delete.end");
- bool mayBeSentinel = false;
- if (checkType->IsPointer())
- {
- auto innerType = checkType->GetUnderlyingType();
- if (innerType->IsValuelessType())
- mayBeSentinel = true;
- }
- BfIRValue isNotNull;
- if (mayBeSentinel)
- {
- auto intVal = mBfIRBuilder->CreatePtrToInt(val.mValue, BfTypeCode_IntPtr);
- isNotNull = mBfIRBuilder->CreateCmpGT(intVal, mBfIRBuilder->CreateConst(BfTypeCode_IntPtr, 1), false);
- }
- else
- {
- isNotNull = mBfIRBuilder->CreateIsNotNull(val.mValue);
- }
- mBfIRBuilder->CreateCondBr(isNotNull, bodyBB, endBB);
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- if (val.mType->IsObjectOrInterface())
- {
- EmitObjectAccessCheck(val);
- }
-
- SizedArray<BfIRValue, 4> llvmArgs;
- auto bitAddr = mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->GetPrimitiveType(BfTypeCode_NullPtr));
- llvmArgs.push_back(bitAddr);
- if (val.mType->IsObjectOrInterface())
- {
- auto objectType = mContext->mBfObjectType;
- BfTypeInstance* checkTypeInst = val.mType->ToTypeInstance();
- bool allowPrivate = checkTypeInst == mCurTypeInstance;
- bool allowProtected = allowPrivate || TypeIsSubTypeOf(mCurTypeInstance, checkTypeInst);
- while (checkTypeInst != NULL)
- {
- auto checkTypeDef = checkTypeInst->mTypeDef;
- if (checkTypeDef->mDtorDef != NULL)
- {
- if (!CheckProtection(checkTypeDef->mDtorDef->mProtection, allowProtected, allowPrivate))
- {
- auto error = Fail(StrFormat("'%s.~this()' is inaccessible due to its protection level", TypeToString(checkTypeInst).c_str()), deleteStmt->mExpression); // CS0122
- }
- }
- checkTypeInst = checkTypeInst->mBaseType;
- allowPrivate = false;
- }
- if (mCompiler->mOptions.mObjectHasDebugFlags)
- {
- auto preDelete = GetInternalMethod((deleteStmt->mTargetTypeToken != NULL) ? "Dbg_ObjectPreCustomDelete" : "Dbg_ObjectPreDelete");
- SizedArray<BfIRValue, 4> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(objectType)));
- mBfIRBuilder->CreateCall(preDelete.mFunc, llvmArgs);
- }
- // call dtor
- BfExprEvaluator expressionEvaluator(this);
- PopulateType(val.mType);
- PopulateType(objectType, BfPopulateType_DataAndMethods);
- if (objectType->mVirtualMethodTable.size() == 0)
- {
- if (!mCompiler->IsAutocomplete())
- AssertErrorState();
- }
- else if (!mCompiler->IsSkippingExtraResolveChecks())
- {
- BfMethodInstance* methodInstance = objectType->mVirtualMethodTable[mCompiler->GetVTableMethodOffset() + 0].mImplementingMethod;
- BF_ASSERT(methodInstance->mMethodDef->mName == "~this");
- SizedArray<BfIRValue, 4> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(objectType)));
- expressionEvaluator.CreateCall(methodInstance, methodInstance->mIRFunction, false, llvmArgs);
- }
- if ((deleteStmt->mTargetTypeToken != NULL) && (!isAppendDelete))
- {
- if (deleteStmt->mAllocExpr != NULL)
- {
- if (customAllocator)
- {
- auto customAllocTypeInst = customAllocator.mType->ToTypeInstance();
- if (customAllocTypeInst != NULL)
- {
- if ((customAllocTypeInst != NULL) && (customAllocTypeInst->mTypeDef->GetMethodByName("FreeObject") != NULL))
- {
- BfTypedValueExpression typedValueExpr;
- typedValueExpr.Init(val);
- typedValueExpr.mRefNode = deleteStmt->mAllocExpr;
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfExpression*, 2> argExprs;
- argExprs.push_back(&typedValueExpr);
- BfSizedArray<BfExpression*> sizedArgExprs(argExprs);
- BfResolvedArgs argValues(&sizedArgExprs);
- exprEvaluator.ResolveArgValues(argValues);
- exprEvaluator.mNoBind = true;
- exprEvaluator.MatchMethod(deleteStmt->mAllocExpr, NULL, customAllocator, false, true, "FreeObject", argValues, NULL);
- customAllocator = BfTypedValue();
- }
- }
- }
- }
- }
- else
- {
- if (mCompiler->mOptions.mEnableRealtimeLeakCheck)
- {
- SizedArray<BfIRValue, 4> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(objectType)));
- auto moduleMethodInstance = GetInternalMethod("Dbg_MarkObjectDeleted");
- mBfIRBuilder->CreateCall(moduleMethodInstance.mFunc, llvmArgs);
- }
- else if (!isAppendDelete)
- {
- mBfIRBuilder->CreateCall(GetBuiltInFunc(BfBuiltInFuncType_Free), llvmArgs);
- }
- }
- }
- else
- {
- if ((isAppendDelete) || (customAllocator))
- {
- // Do nothing
- }
- else
- {
- auto func = GetBuiltInFunc(BfBuiltInFuncType_Free);
- if (!func)
- {
- BF_ASSERT(mCompiler->mIsResolveOnly);
- }
- else
- mBfIRBuilder->CreateCall(func, llvmArgs);
- }
- }
- if ((customAllocator) && (customAllocator.mType != GetPrimitiveType(BfTypeCode_NullPtr)))
- {
- auto voidPtrType = GetPrimitiveType(BfTypeCode_NullPtr);
- auto ptrValue = BfTypedValue(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(voidPtrType)), voidPtrType);
- BfTypedValueExpression typedValueExpr;
- typedValueExpr.Init(ptrValue);
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfExpression*, 2> argExprs;
- argExprs.push_back(&typedValueExpr);
- BfSizedArray<BfExpression*> sizedArgExprs(argExprs);
- BfResolvedArgs argValues(&sizedArgExprs);
- exprEvaluator.ResolveArgValues(argValues);
- exprEvaluator.mNoBind = true;
- exprEvaluator.MatchMethod(deleteStmt->mAllocExpr, NULL, customAllocator, false, false, "Free", argValues, NULL);
- }
-
- mBfIRBuilder->CreateBr(endBB);
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- }
- void BfModule::Visit(BfSwitchStatement* switchStmt)
- {
- BfScopeData newScope;
- newScope.mInnerIsConditional = false;
- newScope.mCloseNode = switchStmt;
- if (switchStmt->mCloseBrace != NULL)
- newScope.mCloseNode = switchStmt->mCloseBrace;
- if (switchStmt->mLabelNode != NULL)
- newScope.mLabelNode = switchStmt->mLabelNode->mLabel;
- mCurMethodState->AddScope(&newScope);
- NewScopeState();
- auto valueScopeStartOuter = ValueScopeStart();
- BfTypedValue switchValue;
- if (switchStmt->mSwitchValue == NULL)
- {
- AssertErrorState();
- UpdateSrcPos(switchStmt->mSwitchToken);
- }
- else
- {
- UpdateExprSrcPos(switchStmt->mSwitchValue);
- BfEvalExprFlags flags = BfEvalExprFlags_None;
- flags = BfEvalExprFlags_AllowSplat;
- switchValue = CreateValueFromExpression(switchStmt->mSwitchValue, NULL, flags);
- }
- EmitEnsureInstructionAt();
- if (!switchValue)
- {
- AssertErrorState();
- switchValue = GetDefaultTypedValue(mContext->mBfObjectType);
- }
- // We make the switch value conditional, but all other uses of this scope is conditional since it's conditional on cases
- newScope.mInnerIsConditional = true;
-
- BfTypedValue switchValueAddr = switchValue;
-
- BfLocalVariable* localDef = new BfLocalVariable();
- localDef->mName = "_";
- localDef->mResolvedType = switchValueAddr.mType;
- localDef->mIsReadOnly = true;
- localDef->mIsAssigned = true;
- if (switchValue.IsAddr())
- {
- localDef->mAddr = switchValue.mValue;
- }
- else
- {
- localDef->mValue = switchValue.mValue;
- localDef->mIsSplat = switchValue.IsSplat();
- }
- bool wantsDebugInfo = mHasFullDebugInfo && !mBfIRBuilder->mIgnoreWrites;
- bool tryExtendValue = false;
- bool addDebugInfo = true;
- if ((wantsDebugInfo) && (!switchValue.mType->IsValuelessType()))
- {
- if (IsTargetingBeefBackend())
- {
- // We don't need to make a copy
- if (switchValue.IsSplat())
- {
- localDef->mIsSplat = true;
- if (WantsDebugInfo())
- {
- bool found = false;
- String varName = "_";
- for (auto dbgVar : mCurMethodState->mLocals)
- {
- if (dbgVar->mAddr == switchValue.mValue)
- {
- varName += "$a$" + dbgVar->mName;
- found = true;
- break;
- }
- }
- if (found)
- {
- auto fakeVal = CreateAlloca(GetPrimitiveType(BfTypeCode_Int32), true, "_fake");
- addDebugInfo = false;
- auto diVariable = mBfIRBuilder->DbgCreateAutoVariable(mCurMethodState->mCurScope->mDIScope, varName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, mBfIRBuilder->DbgGetType(localDef->mResolvedType), BfIRInitType_NotNeeded_AliveOnDecl);
- mBfIRBuilder->DbgInsertDeclare(fakeVal, diVariable);
- }
- }
- }
- else
- {
- // if (!localDef->mAddr)
- // {
- // BfIRValue value = localDef->mValue;
- // if (newLocalVar->mConstValue)
- // value = localDef->mConstValue;
- // auto aliasValue = mBfIRBuilder->CreateAliasValue(value);
- // mBfIRBuilder->DbgInsertValueIntrinsic(aliasValue, diVariable);
- // scopeData.mDeferredLifetimeEnds.push_back(aliasValue);
- // }
- tryExtendValue = true;
- }
- }
- else if ((switchValue.mType->IsComposite()) && (switchValue.IsAddr()))
- {
- auto refType = CreateRefType(switchValue.mType);
- auto allocaVal = CreateAlloca(refType);
- mBfIRBuilder->CreateStore(switchValue.mValue, allocaVal);
- auto diType = mBfIRBuilder->DbgGetType(refType);
- auto diVariable = mBfIRBuilder->DbgCreateAutoVariable(mCurMethodState->mCurScope->mDIScope,
- localDef->mName, mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, diType);
- mBfIRBuilder->DbgInsertDeclare(allocaVal, diVariable);
- addDebugInfo = false;
- }
- else
- {
- if (switchValueAddr.IsSplat())
- {
- auto addr = CreateAlloca(switchValue.mType);
- if (switchValue.IsSplat())
- AggregateSplatIntoAddr(switchValue, addr);
- else
- mBfIRBuilder->CreateStore(switchValue.mValue, addr);
- localDef->mAddr = addr;
- localDef->mValue = BfIRValue();
- localDef->mIsSplat = false;
- }
- }
- }
- // bool firstCaseHandled = false;
- // if (!switchStmt->mSwitchCases.IsEmpty())
- // {
- // auto switchCase = switchStmt->mSwitchCases[0];
- // if (switchCase->mCaseExpressions.size() == 0)
- // {
- // auto caseExpr = switchCase->mCaseExpressions[0];
- // if (auto literalExpr = BfNodeDynCast<BfLiteralExpression>(caseExpr))
- // {
- // if (literalExpr->mValue.mTypeCode == BfTypeCode_NullPtr)
- // {
- // }
- // }
- // }
- // }
- BfIRBlock derefBlock;
- BfIRBlock derefContBlock;
-
- BfTypedValue derefSwitchValue;
- if (switchValue.mType->IsPointer())
- {
- auto underlyingType = switchValue.mType->GetUnderlyingType();
- if (underlyingType->IsEnum())
- {
- derefSwitchValue = BfTypedValue(switchValue.mValue, underlyingType, true);
-
- derefBlock = mBfIRBuilder->CreateBlock("switch.deref");
-
- derefContBlock = mBfIRBuilder->CreateBlock("switch.deref");
- mBfIRBuilder->AddBlock(derefContBlock);
- mBfIRBuilder->SetInsertPoint(derefContBlock);
- }
- }
- AddLocalVariableDef(localDef, addDebugInfo, true);
- BfDeferredLocalAssignData deferredLocalAssignData(mCurMethodState->mCurScope);
- deferredLocalAssignData.mVarIdBarrier = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- int numExpressions = 0;
- SizedArray<BfIRBlock, 8> blocks;
- SizedArray<BfWhenExpression*, 8> whenExprs;
- SizedArray<BfIRBlock, 8> whenFailBlocks;
- BfIRBlock defaultBlock;
- auto endBlock = mBfIRBuilder->CreateBlock("switch.end");
- for (BfSwitchCase* switchCase : switchStmt->mSwitchCases)
- {
- auto caseBlock = mBfIRBuilder->CreateBlock(StrFormat("switch.%d", blocks.size()));
- blocks.push_back(caseBlock);
- numExpressions += (int)switchCase->mCaseExpressions.size();
- }
- if (switchStmt->mDefaultCase != NULL)
- {
- defaultBlock = mBfIRBuilder->CreateBlock("default");
- blocks.push_back(defaultBlock);
- }
- else
- defaultBlock = endBlock;
- SizedArray<BfDeferredLocalAssignData, 8> deferredLocalAssignDataVec;
- deferredLocalAssignDataVec.resize(blocks.size());
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- BfTypedValue enumTagVal;
- // Declare cases
- int blockIdx = 0;
- bool hadConstIntVals = false;
- bool hadWhen = false;
- BfIRValue switchStatement;
- auto switchBlock = mBfIRBuilder->GetInsertBlock();
- BfIRBlock noSwitchBlock = mBfIRBuilder->CreateBlock("noSwitch", true);
- BfPrimitiveType* intCoercibleType = GetIntCoercibleType(switchValue.mType);
-
- bool isConstSwitch = false;
- if ((switchValue.mValue.IsConst()) || (switchValue.mType->IsValuelessType()))
- {
- isConstSwitch = true;
- }
- if (switchValue.mValue)
- {
- mBfIRBuilder->PopulateType(switchValue.mType);
- if (intCoercibleType != NULL)
- {
- auto intValue = GetIntCoercible(switchValue);
- switchStatement = mBfIRBuilder->CreateSwitch(intValue.mValue, noSwitchBlock, numExpressions);
- }
- else if (switchValue.mType->IsPayloadEnum())
- {
- enumTagVal = ExtractValue(switchValue, NULL, 2);
- enumTagVal = LoadValue(enumTagVal);
- switchStatement = mBfIRBuilder->CreateSwitch(enumTagVal.mValue, noSwitchBlock, numExpressions);
- }
- else if (!isConstSwitch)
- switchStatement = mBfIRBuilder->CreateSwitch(switchValue.mValue, noSwitchBlock, numExpressions);
- }
- auto valueScopeStartInner = ValueScopeStart();
- mBfIRBuilder->SetInsertPoint(noSwitchBlock);
- bool isPayloadEnum = switchValue.mType->IsPayloadEnum();
- bool isIntegralSwitch = switchValue.mType->IsIntegral() || (intCoercibleType != NULL) || ((switchValue.mType->IsEnum()) && (!isPayloadEnum));
- auto _ShowCaseError = [&] (int64 id, BfAstNode* errNode)
- {
- if (isPayloadEnum)
- {
- auto enumType = switchValue.mType->ToTypeInstance();
- for (auto fieldInstance : enumType->mFieldInstances)
- {
- auto fieldDef = fieldInstance.GetFieldDef();
- if (fieldDef->IsEnumCaseEntry())
- {
- int enumIdx = -fieldInstance.mDataIdx - 1;
- if (enumIdx == id)
- {
- Fail(StrFormat("The switch statement already contains a case for the the value '%s'", fieldDef->mName.c_str()), errNode);
- return;
- }
- }
- }
- }
- Fail(StrFormat("The switch statement already contains a case for the the value '%lld'", id), errNode);
- };
- int caseCount = 0;
- bool allHadReturns = true;
- bool hadCondCase = false;
- BfIRBlock lastDefaultBlock;
- struct _CaseState
- {
- BfIRBlock mCondBlock;
- BfIRBlock mUncondBlock;
- };
- bool hadConstMatch = false;
-
- auto startingLocalVarId = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- Dictionary<int64, _CaseState> handledCases;
- for (BfSwitchCase* switchCase : switchStmt->mSwitchCases)
- {
- deferredLocalAssignDataVec[blockIdx].mScopeData = mCurMethodState->mCurScope;
- deferredLocalAssignDataVec[blockIdx].ExtendFrom(mCurMethodState->mDeferredLocalAssignData);
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignDataVec[blockIdx]);
- mCurMethodState->mDeferredLocalAssignData->mVarIdBarrier = startingLocalVarId;
- SetIllegalSrcPos();
- auto caseBlock = blocks[blockIdx];
- BfScopeData caseScopeData;
- bool openedScope = false;
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- {
- // This does the enum autocomplete popup
- BfAstNode* checkNode = NULL;
- int caseExprIdx = (int)switchCase->mCaseExpressions.size();
- if (caseExprIdx == 0)
- checkNode = switchCase->mCaseToken;
- else if (caseExprIdx - 1 < (int)switchCase->mCaseCommas.size())
- checkNode = switchCase->mCaseCommas[caseExprIdx - 1];
- if (checkNode != NULL)
- mCompiler->mResolvePassData->mAutoComplete->CheckEmptyStart(checkNode, switchValue.mType);
- }
- bool mayHaveMatch = false;
- BfWhenExpression* whenExpr = NULL;
-
- for (BfExpression* caseExpr : switchCase->mCaseExpressions)
- {
- if (auto checkWhenExpr = BfNodeDynCast<BfWhenExpression>(caseExpr))
- {
- hadWhen = true;
- whenExpr = checkWhenExpr;
- }
- }
- BfIRBlock lastNotEqBlock;
- for (BfExpression* caseExpr : switchCase->mCaseExpressions)
- {
- BfConstant* constantInt = NULL;
- if (auto checkWhenExpr = BfNodeDynCast<BfWhenExpression>(caseExpr))
- continue;
- if ((!openedScope) && (isPayloadEnum))
- {
- openedScope = true;
- caseScopeData.mOuterIsConditional = true;
- mCurMethodState->AddScope(&caseScopeData);
- NewScopeState();
- UpdateSrcPos(caseExpr);
- SetIllegalSrcPos();
- }
- BfIRValue eqResult;
- BfIRBlock notEqBB;
- bool handled = false;
- BfTypedValue caseValue;
- BfIRBlock doBlock = caseBlock;
- bool hadConditional = false;
- if (isPayloadEnum)
- {
- auto dscrType = switchValue.mType->ToTypeInstance()->GetDiscriminatorType();
- if (!enumTagVal)
- {
- enumTagVal = ExtractValue(switchValue, NULL, 2);
- enumTagVal = LoadValue(enumTagVal);
- }
- notEqBB = mBfIRBuilder->CreateBlock(StrFormat("switch.notEq.%d", blockIdx), false);
- int tagId = -1;
- BfIRBlock matchBlock;
- BfTypedValue eqTypedResult;
- if (auto bindExpr = BfNodeDynCast<BfEnumCaseBindExpression>(caseExpr))
- {
- eqTypedResult = HandleCaseBind(switchValueAddr, enumTagVal, bindExpr, &caseBlock, ¬EqBB, &matchBlock, &tagId);
- }
- else
- {
- eqTypedResult = TryCaseEnumMatch(switchValueAddr, enumTagVal, caseExpr, &caseBlock, ¬EqBB, &matchBlock, tagId, hadConditional, false);
- if (hadConditional)
- hadCondCase = true;
- }
- if (tagId != -1)
- {
- doBlock = matchBlock; // Jump to binds rather than just the code
- caseValue = BfTypedValue(GetConstValue(tagId, GetPrimitiveType(dscrType->mTypeDef->mTypeCode)), dscrType);
- }
- else
- hadCondCase = true;
- if (eqTypedResult)
- {
- handled = true;
- eqResult = eqTypedResult.mValue;
- }
- }
- if (!eqResult)
- {
- caseValue = CreateValueFromExpression(caseExpr, switchValue.mType, (BfEvalExprFlags)(BfEvalExprFlags_AllowEnumId | BfEvalExprFlags_NoCast));
- if (!caseValue)
- continue;
- }
- BfTypedValue caseIntVal = caseValue;
- if ((isIntegralSwitch) || (isPayloadEnum))
- {
- if ((intCoercibleType != NULL) &&
- (caseValue.mType == switchValue.mType) &&
- (caseValue.mValue.IsConst()))
- {
- caseIntVal = GetIntCoercible(caseValue);
- constantInt = mBfIRBuilder->GetConstant(caseIntVal.mValue);
- }
- else
- {
- // For a non-const case, allow for conversion operators, otherwise cast now
- if ((isIntegralSwitch) && (caseValue.mValue.IsConst()))
- {
- if (caseValue.mType != switchValue.mType)
- {
- caseValue = Cast(caseExpr, caseValue, switchValue.mType);
- if (!caseValue)
- continue;
- caseIntVal = caseValue;
- }
- }
- constantInt = mBfIRBuilder->GetConstant(caseValue.mValue);
- if ((constantInt != NULL) && (!mBfIRBuilder->IsInt(constantInt->mTypeCode)))
- constantInt = NULL;
- }
- }
- if ((!switchStatement) && (!isConstSwitch))
- {
- // Do nothing
- mayHaveMatch = true;
- }
- else if ((constantInt != NULL) && (!hadWhen) && (!isConstSwitch))
- {
- if (!hadConditional)
- {
- _CaseState* caseState = NULL;
- handledCases.TryAdd(constantInt->mInt64, NULL, &caseState);
-
- if (caseState->mUncondBlock)
- {
- _ShowCaseError(constantInt->mInt64, caseExpr);
- }
- else
- {
- caseState->mUncondBlock = doBlock;
- mBfIRBuilder->AddSwitchCase(switchStatement, caseIntVal.mValue, doBlock);
- hadConstIntVals = true;
- }
- }
- mayHaveMatch = true;
- }
- else if (!handled)
- {
- hadCondCase = true;
-
- if (!eqResult)
- {
- BfExprEvaluator exprEvaluator(this);
- BfAstNode* refNode = switchCase->mColonToken;
- if ((caseValue.mType->IsPayloadEnum()) && (caseValue.mValue.IsConst()) && (switchValue.mType == caseValue.mType))
- {
- if (!enumTagVal)
- {
- enumTagVal = ExtractValue(switchValue, NULL, 2);
- enumTagVal = LoadValue(enumTagVal);
- }
- eqResult = mBfIRBuilder->CreateCmpEQ(enumTagVal.mValue, caseValue.mValue);
- }
- else
- {
- exprEvaluator.PerformBinaryOperation(switchStmt->mSwitchValue, caseExpr, BfBinaryOp_Equality, refNode, (BfBinOpFlags)(BfBinOpFlag_ForceLeftType), switchValue, caseValue);
- if (switchStmt->mSwitchValue != NULL)
- UpdateSrcPos(switchStmt->mSwitchValue);
- SetIllegalSrcPos();
- eqResult = exprEvaluator.mResult.mValue;
- if (!eqResult)
- eqResult = GetConstValue(0, boolType);
- }
- }
- ValueScopeEnd(valueScopeStartInner);
- bool isConstResult = false;
- bool constResult = false;
- if (eqResult.IsConst())
- {
- auto constant = mBfIRBuilder->GetConstant(eqResult);
- if (constant->mTypeCode == BfTypeCode_Boolean)
- {
- isConstResult = true;
- constResult = constant->mBool;
- }
- }
- if (isConstResult)
- {
- if (constResult)
- {
- mBfIRBuilder->CreateBr(caseBlock);
- mayHaveMatch = true;
-
- if (whenExpr == NULL)
- {
- hadConstMatch = true;
- }
- else
- {
- notEqBB = mBfIRBuilder->CreateBlock(StrFormat("switch.notEq.%d", blockIdx));
-
- mBfIRBuilder->AddBlock(notEqBB);
- mBfIRBuilder->SetInsertPoint(notEqBB);
- }
- }
- }
- else
- {
- notEqBB = mBfIRBuilder->CreateBlock(StrFormat("switch.notEq.%d", blockIdx));
- mayHaveMatch = true;
- mBfIRBuilder->CreateCondBr(eqResult, caseBlock, notEqBB);
- mBfIRBuilder->AddBlock(notEqBB);
- mBfIRBuilder->SetInsertPoint(notEqBB);
- }
- }
- if (notEqBB)
- lastNotEqBlock = notEqBB;
- if ((!hadCondCase) && (notEqBB))
- lastDefaultBlock = notEqBB;
- }
- if (lastDefaultBlock)
- mBfIRBuilder->SetSwitchDefaultDest(switchStatement, lastDefaultBlock);
- auto prevInsertBlock = mBfIRBuilder->GetInsertBlock();
-
- SetAndRestoreValue<bool> prevIgnoreWrites(mBfIRBuilder->mIgnoreWrites, true, !mayHaveMatch);
-
- mBfIRBuilder->AddBlock(caseBlock);
- mBfIRBuilder->SetInsertPoint(caseBlock);
-
- if (whenExpr != NULL)
- {
- UpdateSrcPos(whenExpr);
- BfTypedValue whenValue;
- if (whenExpr->mExpression != NULL)
- whenValue = CreateValueFromExpression(whenExpr->mExpression, boolType, BfEvalExprFlags_AllowEnumId);
- if (!whenValue)
- {
- AssertErrorState();
- whenValue = GetDefaultTypedValue(boolType);
- }
- bool constResult = false;
- if (mBfIRBuilder->TryGetBool(whenValue.mValue, constResult))
- {
- if (!constResult)
- prevIgnoreWrites.Set();
- }
- else
- {
- BfIRBlock eqBB = mBfIRBuilder->CreateBlock(StrFormat("switch.when.%d", blockIdx));
- mBfIRBuilder->CreateCondBr(whenValue.mValue, eqBB, lastNotEqBlock);
- mBfIRBuilder->AddBlock(eqBB);
- mBfIRBuilder->SetInsertPoint(eqBB);
- }
- }
- BfIRBlock fallthroughBlock;
- if (blockIdx < (int) blocks.size() - 1)
- fallthroughBlock = blocks[blockIdx + 1];
- else
- fallthroughBlock = defaultBlock;
- bool hadReturn = false;
- if ((switchCase->mCodeBlock != NULL) && (!switchCase->mCodeBlock->mChildArr.IsEmpty()))
- {
- UpdateSrcPos(switchCase->mCodeBlock);
- VisitCodeBlock(switchCase->mCodeBlock, BfIRBlock(), endBlock, fallthroughBlock, true, &hadReturn, switchStmt->mLabelNode, openedScope);
- openedScope = false;
- deferredLocalAssignDataVec[blockIdx].mHadReturn = hadReturn;
- caseCount++;
- if ((!hadReturn) && (!mCurMethodState->mDeferredLocalAssignData->mHadFallthrough))
- allHadReturns = false;
- if (auto block = BfNodeDynCast<BfBlock>(switchCase->mCodeBlock))
- {
- //
- }
- else
- {
- if (switchStmt->mCloseBrace != NULL)
- {
- UpdateSrcPos(switchStmt->mCloseBrace);
- }
- EmitEnsureInstructionAt();
- }
- //UpdateSrcPos(switchCase->mCodeBlock);
- //SetIllegalSrcPos();
- mBfIRBuilder->ClearDebugLocation();
- }
- else
- {
- if (openedScope)
- RestoreScopeState();
- mBfIRBuilder->CreateBr(endBlock);
- allHadReturns = false;
- }
- prevIgnoreWrites.Restore();
- mBfIRBuilder->SetInsertPoint(prevInsertBlock);
- blockIdx++;
- }
- // Check for comprehensiveness
- bool isComprehensive = true;
- if ((switchValue) && (switchStmt->mDefaultCase == NULL))
- {
- if (switchValue.mType->IsEnum())
- {
- auto enumType = switchValue.mType->ToTypeInstance();
- if (enumType->IsPayloadEnum())
- {
- int lastTagId = -1;
- for (auto& field : enumType->mFieldInstances)
- {
- auto fieldDef = field.GetFieldDef();
- if (fieldDef == NULL)
- continue;
- if (field.mDataIdx < 0)
- lastTagId = -field.mDataIdx - 1;
- }
- isComprehensive = lastTagId == (int)handledCases.size() - 1;
- }
- else
- {
- for (auto& field : enumType->mFieldInstances)
- {
- auto fieldDef = field.GetFieldDef();
- if ((fieldDef != NULL) && (fieldDef->mFieldDeclaration != NULL) && (fieldDef->mFieldDeclaration->mTypeRef == NULL))
- {
- if (field.mConstIdx != -1)
- {
- auto constant = enumType->mConstHolder->GetConstantById(field.mConstIdx);
- isComprehensive &= handledCases.ContainsKey(constant->mInt64);
- }
- }
- }
- }
- if (!isComprehensive)
- {
- BfAstNode* refNode = switchStmt->mSwitchToken;
- Fail("Switch must be exhaustive, consider adding a default clause", switchStmt->mSwitchToken);
- if ((switchStmt->mCloseBrace) && (mCompiler->IsAutocomplete()) && (mCompiler->mResolvePassData->mAutoComplete->CheckFixit((refNode))))
- {
- BfParserData* parser = refNode->GetSourceData()->ToParserData();
- if (parser != NULL)
- {
- int fileLoc = switchStmt->mCloseBrace->GetSrcStart();
- mCompiler->mResolvePassData->mAutoComplete->AddEntry(AutoCompleteEntry("fixit", StrFormat("default:\tdefault:|%s|%d||default:", parser->mFileName.c_str(), fileLoc).c_str()));
- }
- }
- }
- }
- else
- isComprehensive = false;
- }
- if (!hadConstMatch)
- mBfIRBuilder->CreateBr(defaultBlock);
- mBfIRBuilder->SetInsertPoint(switchBlock);
- if (!hadConstIntVals)
- {
- if (switchStatement)
- mBfIRBuilder->EraseInstFromParent(switchStatement);
- mBfIRBuilder->CreateBr(noSwitchBlock);
- }
- if (switchStmt->mDefaultCase != NULL)
- {
- SetAndRestoreValue<bool> prevIgnoreWrites(mBfIRBuilder->mIgnoreWrites, true, hadConstMatch);
- mBfIRBuilder->AddBlock(defaultBlock);
- mBfIRBuilder->SetInsertPoint(defaultBlock);
- auto switchCase = switchStmt->mDefaultCase;
- if (switchCase->mCodeBlock != NULL)
- {
- isComprehensive = true;
- UpdateSrcPos(switchCase->mCodeBlock);
-
- deferredLocalAssignDataVec[blockIdx].mScopeData = mCurMethodState->mCurScope;
- deferredLocalAssignDataVec[blockIdx].ExtendFrom(mCurMethodState->mDeferredLocalAssignData);
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignDataVec[blockIdx]);
- mCurMethodState->mDeferredLocalAssignData->mVarIdBarrier = startingLocalVarId;
- bool hadReturn = false;
- VisitCodeBlock(switchCase->mCodeBlock, BfIRBlock(), endBlock, BfIRBlock(), true, &hadReturn, switchStmt->mLabelNode);
- deferredLocalAssignDataVec[blockIdx].mHadReturn = hadReturn;
- caseCount++;
- if (!hadReturn)
- allHadReturns = false;
- }
- }
- if (isComprehensive)
- {
- // Merge and apply deferred local assign data
- // We only do this if there's a default case, otherwise we assume we may have missed a case
- // that by definition had no local assigns
- BfDeferredLocalAssignData* mergedDeferredLocalAssignData = NULL;
- for (blockIdx = 0; blockIdx < (int)blocks.size(); blockIdx++)
- {
- auto deferredLocalAssignData = &deferredLocalAssignDataVec[blockIdx];
- if (deferredLocalAssignData->mHadFallthrough)
- continue;
- // if ((deferredLocalAssignData->mHadFallthrough) && (blockIdx < (int)blocks.size() - 1))
- // {
- // auto nextDeferredLocalAssignData = &deferredLocalAssignDataVec[blockIdx + 1];
- // deferredLocalAssignData->SetUnion(*nextDeferredLocalAssignData);
- // }
- if (mergedDeferredLocalAssignData == NULL)
- mergedDeferredLocalAssignData = deferredLocalAssignData;
- else
- mergedDeferredLocalAssignData->SetIntersection(*deferredLocalAssignData);
- }
- if (mergedDeferredLocalAssignData != NULL)
- mCurMethodState->ApplyDeferredLocalAssignData(*mergedDeferredLocalAssignData);
- }
- if ((caseCount > 0) && (allHadReturns) &&
- ((defaultBlock != endBlock) || (isComprehensive)))
- {
- mCurMethodState->SetHadReturn(true);
- mCurMethodState->mLeftBlockUncond = true;
- if (defaultBlock != endBlock)
- mBfIRBuilder->DeleteBlock(endBlock);
- else
- {
- if (switchStmt->mCloseBrace != NULL)
- UpdateSrcPos(switchStmt->mCloseBrace);
- mBfIRBuilder->AddBlock(endBlock);
- mBfIRBuilder->SetInsertPoint(endBlock);
- mBfIRBuilder->CreateUnreachable();
- }
- }
- else
- {
- if (switchStmt->mCloseBrace != NULL)
- UpdateSrcPos(switchStmt->mCloseBrace);
- mBfIRBuilder->AddBlock(endBlock);
- mBfIRBuilder->SetInsertPoint(endBlock);
- }
-
- BfIRValue lifetimeExtendVal;
- if (tryExtendValue)
- {
- if (localDef->mAddr)
- lifetimeExtendVal = localDef->mAddr;
- else
- lifetimeExtendVal = localDef->mValue;
- }
-
- RestoreScopeState();
- if (lifetimeExtendVal)
- mBfIRBuilder->CreateLifetimeExtend(lifetimeExtendVal);
-
- ValueScopeEnd(valueScopeStartOuter);
- }
- static int gRetIdx = 0;
- void BfModule::Visit(BfReturnStatement* returnStmt)
- {
- if (mCurMethodInstance == NULL)
- {
- Fail("Unexpected return", returnStmt);
- return;
- }
-
- UpdateSrcPos(returnStmt);
- EmitEnsureInstructionAt();
- auto retType = mCurMethodInstance->mReturnType;
- if (mCurMethodInstance->IsMixin())
- retType = NULL;
- if (mCurMethodState->mClosureState != NULL)
- retType = mCurMethodState->mClosureState->mReturnType;
- auto checkScope = mCurMethodState->mCurScope;
- while (checkScope != NULL)
- {
- if (checkScope->mIsDeferredBlock)
- {
- Fail("Deferred blocks cannot contain 'return' statements", returnStmt);
- if (returnStmt->mExpression != NULL)
- {
- BfExprEvaluator exprEvaluator(this);
- CreateValueFromExpression(exprEvaluator, returnStmt->mExpression, GetPrimitiveType(BfTypeCode_Var), BfEvalExprFlags_None);
- }
- return;
- }
- checkScope = checkScope->mPrevScope;
- }
- if (retType == NULL)
- {
- if (returnStmt->mExpression != NULL)
- {
- BfExprEvaluator exprEvaluator(this);
- CreateValueFromExpression(exprEvaluator, returnStmt->mExpression, GetPrimitiveType(BfTypeCode_Var), BfEvalExprFlags_None);
- }
- MarkScopeLeft(&mCurMethodState->mHeadScope);
- return;
- }
- if (returnStmt->mExpression == NULL)
- {
- MarkScopeLeft(&mCurMethodState->mHeadScope);
- if (retType->IsVoid())
- {
- EmitReturn(BfIRValue());
- return;
- }
- if (retType != NULL)
- {
- Fail("Expected return value", returnStmt);
- return;
- }
- EmitReturn(GetDefaultValue(retType));
- return;
- }
- BfType* expectingReturnType = retType;
-
- BfType* origType;
- BfExprEvaluator exprEvaluator(this);
- bool alreadyWritten = false;
- if (mCurMethodInstance->HasStructRet())
- exprEvaluator.mReceivingValue = &mCurMethodState->mRetVal;
- if (mCurMethodInstance->mMethodDef->mIsReadOnly)
- exprEvaluator.mAllowReadOnlyReference = true;
- auto retValue = CreateValueFromExpression(exprEvaluator, returnStmt->mExpression, expectingReturnType, BfEvalExprFlags_AllowRefExpr, &origType);
- if (mCurMethodInstance->HasStructRet())
- alreadyWritten = exprEvaluator.mReceivingValue == NULL;
- MarkScopeLeft(&mCurMethodState->mHeadScope);
-
- if (!retValue)
- {
- AssertErrorState();
- if ((expectingReturnType != NULL) && (!expectingReturnType->IsVoid()))
- {
- retValue = GetDefaultTypedValue(expectingReturnType, true);
- }
- else
- {
- EmitReturn(BfIRValue());
- return;
- }
- }
- if (retValue.mType->IsVoid())
- {
- if (retType->IsVoid())
- {
- Warn(0, "Returning void value", returnStmt->mReturnToken);
- EmitReturn(BfIRValue());
- }
- }
- else
- {
- if (retType->IsVoid())
- {
- expectingReturnType = NULL;
- Fail("Attempting to return value from void method", returnStmt->mExpression);
- EmitReturn(BfIRValue());
- return;
- }
- if ((origType != NULL) && (origType->IsStructOrStructPtr()) && (retValue.mType->IsObjectOrInterface()))
- {
- Fail(StrFormat("Stack boxing of type '%s' is not allowed on return statements. Use 'new box' to box on the heap.", TypeToString(origType).c_str()), returnStmt->mExpression);
- }
- if (!alreadyWritten)
- EmitReturn(LoadOrAggregateValue(retValue).mValue);
- else
- EmitReturn(BfIRValue());
- }
- }
- void BfModule::Visit(BfYieldStatement* yieldStmt)
- {
- Fail("Yield not supported", yieldStmt);
- }
- void BfModule::Visit(BfBreakStatement* breakStmt)
- {
- bool inMixinDecl = (mCurMethodInstance != NULL) && (mCurMethodInstance->IsMixin());
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- {
- if (auto identifer = BfNodeDynCast<BfIdentifierNode>(breakStmt->mLabel))
- mCompiler->mResolvePassData->mAutoComplete->CheckLabel(identifer, breakStmt->mBreakNode);
- }
- UpdateSrcPos(breakStmt);
- mBfIRBuilder->CreateEnsureInstructionAt();
- BfBreakData* breakData = mCurMethodState->mBreakData;
- if (breakStmt->mLabel != NULL)
- {
- breakData = FindBreakData(breakStmt->mLabel);
- }
- else
- {
- while (breakData != NULL)
- {
- if (breakData->mIRBreakBlock)
- break;
- breakData = breakData->mPrevBreakData;
- }
- }
- if ((breakData == NULL) || (!breakData->mIRBreakBlock))
- {
- if (inMixinDecl)
- {
- // Our mixin may just require that we're injected into a breakable scope
- }
- else
- Fail("'break' not applicable in this block", breakStmt);
- return;
- }
- if (mCurMethodState->mInDeferredBlock)
- {
- auto checkScope = mCurMethodState->mCurScope;
- while (checkScope != NULL)
- {
- if (checkScope == breakData->mScope)
- break;
- if (checkScope->mIsDeferredBlock)
- {
- Fail("The break target crosses a deferred block boundary", breakStmt);
- return;
- }
- checkScope = checkScope->mPrevScope;
- }
- }
- if (HasDeferredScopeCalls(breakData->mScope))
- {
- EmitDeferredScopeCalls(true, breakData->mScope, breakData->mIRBreakBlock);
- }
- else
- {
- mBfIRBuilder->CreateBr(breakData->mIRBreakBlock);
- }
- mCurMethodState->mLeftBlockUncond = true;
- BfIRValue earliestValueScopeStart;
- auto checkScope = mCurMethodState->mCurScope;
- while (checkScope != NULL)
- {
- if (checkScope->mValueScopeStart)
- earliestValueScopeStart = checkScope->mValueScopeStart;
- if (checkScope == breakData->mScope)
- break;
- checkScope = checkScope->mPrevScope;
- }
- MarkScopeLeft(breakData->mScope);
- ValueScopeEnd(earliestValueScopeStart);
- auto checkBreakData = mCurMethodState->mBreakData;
- while (true)
- {
- checkBreakData->mHadBreak = true;
- if (checkBreakData == breakData)
- break;
- checkBreakData = checkBreakData->mPrevBreakData;
- }
- }
- void BfModule::Visit(BfContinueStatement* continueStmt)
- {
- bool inMixinDecl = (mCurMethodInstance != NULL) && (mCurMethodInstance->IsMixin());
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- {
- if (auto identifer = BfNodeDynCast<BfIdentifierNode>(continueStmt->mLabel))
- mCompiler->mResolvePassData->mAutoComplete->CheckLabel(identifer, continueStmt->mContinueNode);
- }
-
- UpdateSrcPos(continueStmt);
- mBfIRBuilder->CreateEnsureInstructionAt();
- // If we're in a switch, 'break' is valid but we need to continue looking outward for a 'continue' target
- BfBreakData* breakData = mCurMethodState->mBreakData;
- if (continueStmt->mLabel != NULL)
- {
- breakData = FindBreakData(continueStmt->mLabel);
- if ((breakData != NULL) && (!breakData->mIRContinueBlock))
- {
- Fail(StrFormat("'continue' not applicable in '%s", continueStmt->mLabel->ToString().c_str()), continueStmt);
- return;
- }
- }
- else
- {
- while (breakData != NULL)
- {
- if (breakData->mIRContinueBlock)
- break;
- breakData = breakData->mPrevBreakData;
- }
- }
- if ((breakData == NULL) || (!breakData->mIRContinueBlock))
- {
- if (inMixinDecl)
- {
- // Our mixin may just require that we're injected into a breakable scope
- }
- else
- Fail("'Continue' not applicable in this block", continueStmt);
- return;
- }
- BfIRValue earliestValueScopeStart;
- // We don't want to close out our own scope, we want to close out any scopes that were opened after us
- auto nextScope = mCurMethodState->mCurScope;
- while (nextScope != NULL)
- {
- if (nextScope->mValueScopeStart)
- earliestValueScopeStart = nextScope->mValueScopeStart;
- if (nextScope->mPrevScope == breakData->mScope)
- break;
- nextScope = nextScope->mPrevScope;
- }
- if (breakData->mInnerValueScopeStart)
- earliestValueScopeStart = breakData->mInnerValueScopeStart;
- if (mCurMethodState->mInDeferredBlock)
- {
- auto checkScope = mCurMethodState->mCurScope;
- while (checkScope != NULL)
- {
- if (checkScope == breakData->mScope)
- break;
- if (checkScope->mIsDeferredBlock)
- {
- Fail("The continue target crosses a deferred block boundary", continueStmt);
- return;
- }
- checkScope = checkScope->mPrevScope;
- }
- }
- if ((nextScope != NULL) && (HasDeferredScopeCalls(nextScope)))
- {
- EmitDeferredScopeCalls(true, nextScope, breakData->mIRContinueBlock);
- }
- else
- {
- mBfIRBuilder->CreateBr(breakData->mIRContinueBlock);
- }
- MarkScopeLeft(breakData->mScope);
- ValueScopeEnd(earliestValueScopeStart);
- mCurMethodState->mLeftBlockUncond = true;
- if (!mCurMethodState->mInPostReturn)
- mCurMethodState->mHadContinue = true;
- }
- void BfModule::Visit(BfFallthroughStatement* fallthroughStmt)
- {
- UpdateSrcPos(fallthroughStmt);
- if ((mCurMethodState->mBreakData == NULL) || (!mCurMethodState->mBreakData->mIRFallthroughBlock))
- {
- Fail("'Fallthrough' not applicable in this block", fallthroughStmt);
- return;
- }
- EmitDeferredScopeCalls(true, mCurMethodState->mBreakData->mScope, mCurMethodState->mBreakData->mIRFallthroughBlock);
- mCurMethodState->mLeftBlockUncond = true; // Not really a return, but handled the same way
- if (mCurMethodState->mDeferredLocalAssignData != NULL)
- mCurMethodState->mDeferredLocalAssignData->mHadFallthrough = true;
- }
- void BfModule::Visit(BfUsingStatement* usingStmt)
- {
- UpdateSrcPos(usingStmt);
- mCurMethodState->mInHeadScope = false;
- BfScopeData newScope;
- mCurMethodState->AddScope(&newScope);
- NewScopeState();
- if (usingStmt->mVariableDeclaration != NULL)
- UpdateSrcPos(usingStmt->mVariableDeclaration);
- BfTypedValue embeddedValue;
- SizedArray<BfIRValue, 1> llvmArgs;
- BfModuleMethodInstance moduleMethodInstance;
- BfFunctionBindResult functionBindResult;
- BfExprEvaluator exprEvaluator(this);
- bool failed = false;
- if (usingStmt->mVariableDeclaration == NULL)
- {
- AssertErrorState();
- failed = true;
- }
- else if (usingStmt->mVariableDeclaration->mNameNode != NULL)
- {
- BfLocalVariable* localVar = HandleVariableDeclaration(usingStmt->mVariableDeclaration);
- if (localVar == NULL)
- {
- AssertErrorState();
- failed = true;
- }
- else
- {
- embeddedValue = exprEvaluator.LoadLocal(localVar);
- }
- //exprEvaluator.CheckModifyResult(embeddedValue, usingStmt->mVariableDeclaration->mNameNode,);
- }
- else
- {
- embeddedValue = CreateValueFromExpression(usingStmt->mVariableDeclaration->mInitializer);
- if (!embeddedValue)
- failed = true;
- }
- if (!failed)
- {
- exprEvaluator.mFunctionBindResult = &functionBindResult;
- BfResolvedArgs resolvedArgs;
- exprEvaluator.MatchMethod(usingStmt->mVariableDeclaration, NULL, embeddedValue, false, false, "Dispose", resolvedArgs, NULL);
- if (functionBindResult.mMethodInstance != NULL)
- {
- moduleMethodInstance = BfModuleMethodInstance(functionBindResult.mMethodInstance, functionBindResult.mFunc);
- AddDeferredCall(moduleMethodInstance, functionBindResult.mIRArgs, mCurMethodState->mCurScope);
- }
- }
- if (usingStmt->mEmbeddedStatement == NULL)
- {
- AssertErrorState();
- }
- else
- {
- VisitEmbeddedStatement(usingStmt->mEmbeddedStatement);
- }
- RestoreScopeState();
- }
- void BfModule::Visit(BfDoStatement* doStmt)
- {
- UpdateSrcPos(doStmt);
- auto bodyBB = mBfIRBuilder->CreateBlock("do.body", true);
- auto endBB = mBfIRBuilder->CreateBlock("do.end");
- BfScopeData scopeData;
- if (doStmt->mLabelNode != NULL)
- scopeData.mLabelNode = doStmt->mLabelNode->mLabel;
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- BfBreakData breakData;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- mBfIRBuilder->CreateBr(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- VisitEmbeddedStatement(doStmt->mEmbeddedStatement);
- if (!mCurMethodState->mLeftBlockUncond)
- mBfIRBuilder->CreateBr(endBB);
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- RestoreScopeState();
- }
- void BfModule::Visit(BfRepeatStatement* repeatStmt)
- {
- // if (repeatStmt->mCondition != NULL)
- // UpdateSrcPos(repeatStmt->mCondition);
- // else
- UpdateSrcPos(repeatStmt);
- if (repeatStmt->mRepeatToken->mToken == BfToken_Do)
- {
- Fail("Repeat block requires 'repeat' token", repeatStmt->mRepeatToken);
- }
- auto bodyBB = mBfIRBuilder->CreateBlock("repeat.body", true);
- auto condBB = mBfIRBuilder->CreateBlock("repeat.cond");
- auto endBB = mBfIRBuilder->CreateBlock("repeat.end");
- BfScopeData scopeData;
- // We set mIsLoop later
- if (repeatStmt->mLabelNode != NULL)
- scopeData.mLabelNode = repeatStmt->mLabelNode->mLabel;
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- BfBreakData breakData;
- breakData.mIRContinueBlock = condBB;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- mBfIRBuilder->CreateBr(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- scopeData.mIsLoop = true;
- VisitEmbeddedStatement(repeatStmt->mEmbeddedStatement);
- if (!mCurMethodState->mLeftBlockUncond)
- mBfIRBuilder->CreateBr(condBB);
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->mLeftBlockCond = false;
- mBfIRBuilder->AddBlock(condBB);
- mBfIRBuilder->SetInsertPoint(condBB);
- bool isInfiniteLoop = false;
- if (repeatStmt->mCondition != NULL)
- {
- UpdateSrcPos(repeatStmt->mCondition);
- auto checkVal = CreateValueFromExpression(repeatStmt->mCondition, GetPrimitiveType(BfTypeCode_Boolean));
- if (checkVal)
- {
- if ((!breakData.mHadBreak) && (checkVal.mValue.IsConst()))
- {
- auto constVal = mBfIRBuilder->GetConstantById(checkVal.mValue.mId);
- if (constVal->mTypeCode == BfTypeCode_Boolean)
- isInfiniteLoop = constVal->mBool;
- }
- mBfIRBuilder->CreateCondBr(checkVal.mValue, bodyBB, endBB);
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- }
- }
- RestoreScopeState();
- if (isInfiniteLoop)
- EmitDefaultReturn();
- }
- void BfModule::Visit(BfWhileStatement* whileStmt)
- {
- UpdateSrcPos(whileStmt);
- auto condBB = mBfIRBuilder->CreateBlock("while.cond");
- auto bodyBB = mBfIRBuilder->CreateBlock("while.body");
- auto endBB = mBfIRBuilder->CreateBlock("while.end");
- mCurMethodState->mInHeadScope = false;
- BfScopeData scopeData;
- scopeData.mIsLoop = true;
- if (whileStmt->mLabelNode != NULL)
- scopeData.mLabelNode = whileStmt->mLabelNode->mLabel;
- scopeData.mValueScopeStart = ValueScopeStart();
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- BfBreakData breakData;
- breakData.mIRContinueBlock = condBB;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- breakData.mInnerValueScopeStart = scopeData.mValueScopeStart;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- mBfIRBuilder->AddBlock(condBB);
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->SetInsertPoint(condBB);
- BfTypedValue checkVal;
- if (whileStmt->mCondition != NULL)
- {
- UpdateSrcPos(whileStmt->mCondition);
- checkVal = CreateValueFromExpression(whileStmt->mCondition, GetPrimitiveType(BfTypeCode_Boolean));
- }
- if (!checkVal)
- {
- AssertErrorState();
- checkVal = GetDefaultTypedValue(GetPrimitiveType(BfTypeCode_Boolean));
- }
- bool isInfiniteLoop = false;
- if (checkVal.mValue.IsConst())
- {
- auto constVal = mBfIRBuilder->GetConstantById(checkVal.mValue.mId);
- if (constVal->mTypeCode == BfTypeCode_Boolean)
- isInfiniteLoop = constVal->mBool;
- }
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- mBfIRBuilder->CreateCondBr(checkVal.mValue, bodyBB, endBB);
- // Apply deferred local assign to mEmbeddedStatement and itrStmt
- BfDeferredLocalAssignData deferredLocalAssignData(mCurMethodState->mCurScope);
- deferredLocalAssignData.ExtendFrom(mCurMethodState->mDeferredLocalAssignData, false);
- deferredLocalAssignData.mVarIdBarrier = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignData);
- deferredLocalAssignData.mIsUnconditional = isInfiniteLoop;
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- if (whileStmt->mEmbeddedStatement != NULL)
- {
- VisitEmbeddedStatement(whileStmt->mEmbeddedStatement);
- }
- else
- {
- AssertErrorState();
- }
- if (breakData.mHadBreak)
- {
- isInfiniteLoop = false;
- }
- if (!mCurMethodState->mLeftBlockUncond)
- {
- mBfIRBuilder->CreateBr(condBB);
- }
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->mLeftBlockCond = false;
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- RestoreScopeState();
- if ((isInfiniteLoop) && (mCurMethodInstance != NULL))
- EmitDefaultReturn();
- }
- void BfModule::Visit(BfForStatement* forStmt)
- {
- UpdateSrcPos(forStmt);
- auto startBB = mBfIRBuilder->CreateBlock("for.start", true);
- mBfIRBuilder->CreateBr(startBB);
- mBfIRBuilder->SetInsertPoint(startBB);
- BfScopeData scopeData;
- scopeData.mIsLoop = true;
- if (forStmt->mLabelNode != NULL)
- scopeData.mLabelNode = forStmt->mLabelNode->mLabel;
- scopeData.mCloseNode = forStmt;
- scopeData.mValueScopeStart = ValueScopeStart();
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- for (auto initializer : forStmt->mInitializers)
- {
- VisitChild(initializer);
- }
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- auto boolType = GetPrimitiveType(BfTypeCode_Boolean);
- auto condBB = mBfIRBuilder->CreateBlock("for.cond", true);
- auto bodyBB = mBfIRBuilder->CreateBlock("for.body");
- auto incBB = mBfIRBuilder->CreateBlock("for.inc");
- auto endBB = mBfIRBuilder->CreateBlock("for.end");
- BfBreakData breakData;
- breakData.mIRContinueBlock = incBB;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- mBfIRBuilder->CreateBr(condBB);
- bool isInfiniteLoop = false;
- mBfIRBuilder->SetInsertPoint(condBB);
- if (forStmt->mCondition != NULL)
- {
- auto conditionValue = CreateValueFromExpression(forStmt->mCondition, boolType);
- if (!conditionValue)
- conditionValue = GetDefaultTypedValue(boolType);
- auto constant = mBfIRBuilder->GetConstant(conditionValue.mValue);
- if ((constant != NULL) && (constant->mTypeCode == BfTypeCode_Boolean))
- isInfiniteLoop = constant->mBool;
- ValueScopeEnd(scopeData.mValueScopeStart);
- mBfIRBuilder->CreateCondBr(conditionValue.mValue, bodyBB, endBB);
- }
- else
- {
- isInfiniteLoop = true;
- mBfIRBuilder->CreateBr(bodyBB);
- }
- // Apply deferred local assign to mEmbeddedStatement and itrStmt
- BfDeferredLocalAssignData deferredLocalAssignData(mCurMethodState->mCurScope);
- deferredLocalAssignData.ExtendFrom(mCurMethodState->mDeferredLocalAssignData, false);
- deferredLocalAssignData.mVarIdBarrier = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignData);
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- if (forStmt->mEmbeddedStatement != NULL)
- {
- VisitEmbeddedStatement(forStmt->mEmbeddedStatement);
- }
- if (!mCurMethodState->mLeftBlockUncond)
- mBfIRBuilder->CreateBr(incBB);
- mBfIRBuilder->AddBlock(incBB);
- mBfIRBuilder->SetInsertPoint(incBB);
- for (auto itrStmt : forStmt->mIterators)
- {
- VisitChild(itrStmt);
- if ((mCurMethodState->mLeftBlockUncond) && (!mCurMethodState->mHadContinue) && (!mCurMethodState->mInPostReturn))
- Warn(BfWarning_CS0162_UnreachableCode, "Unreachable code", itrStmt);
- }
- ValueScopeEnd(scopeData.mValueScopeStart);
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- // The 'return' may have been inside the block, which may not have been entered if preconditions were not met
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->mLeftBlockCond = false;
- if (breakData.mHadBreak)
- isInfiniteLoop = false;
- RestoreScopeState();
- if (isInfiniteLoop)
- EmitDefaultReturn();
- }
- void BfModule::DoForLess(BfForEachStatement* forEachStmt)
- {
- UpdateSrcPos(forEachStmt);
- BfScopeData scopeData;
- // We set mIsLoop later
- if (forEachStmt->mLabelNode != NULL)
- scopeData.mLabelNode = forEachStmt->mLabelNode->mLabel;
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
-
- auto autoComplete = mCompiler->GetAutoComplete();
- auto isLet = BfNodeDynCast<BfLetTypeReference>(forEachStmt->mVariableTypeRef) != 0;
- auto isVar = BfNodeDynCast<BfVarTypeReference>(forEachStmt->mVariableTypeRef) != 0;
- BfTypedValue target;
- BfType* varType = NULL;
- bool didInference = false;
- if (isLet || isVar)
- {
- if (forEachStmt->mCollectionExpression != NULL)
- target = CreateValueFromExpression(forEachStmt->mCollectionExpression);
- if (target)
- {
- FixIntUnknown(target);
- varType = target.mType;
- }
- if (autoComplete != NULL)
- autoComplete->CheckVarResolution(forEachStmt->mVariableTypeRef, varType);
- didInference = true;
- }
- else
- {
- varType = ResolveTypeRef(forEachStmt->mVariableTypeRef, BfPopulateType_Data, BfResolveTypeRefFlag_AllowRef);
- if (forEachStmt->mCollectionExpression != NULL)
- target = CreateValueFromExpression(forEachStmt->mCollectionExpression, varType);
- }
- if (varType == NULL)
- varType = GetPrimitiveType(BfTypeCode_IntPtr);
- BfType* checkType = varType;
- if (checkType->IsTypedPrimitive())
- checkType = checkType->GetUnderlyingType();
- if (!checkType->IsIntegral())
- {
- Fail(StrFormat("Cannot iterate over '%s' in for-less statements, only integer types are allowed", TypeToString(varType).c_str()), forEachStmt->mVariableTypeRef);
- varType = GetPrimitiveType(BfTypeCode_IntPtr);
- if (didInference)
- target = GetDefaultTypedValue(varType);
- }
- PopulateType(varType, BfPopulateType_Data);
-
- BfLocalVariable* localDef = new BfLocalVariable();
- localDef->mNameNode = BfNodeDynCast<BfIdentifierNode>(forEachStmt->mVariableName);
- localDef->mName = localDef->mNameNode->ToString();
- localDef->mResolvedType = varType;
- BfIRValue varInst;
- if (!varType->IsValuelessType())
- {
- varInst = CreateAlloca(varType);
- }
- localDef->mAddr = varInst;
- localDef->mIsAssigned = true;
- localDef->mReadFromId = 0;
- localDef->mIsReadOnly = isLet || (forEachStmt->mReadOnlyToken != NULL);
- CheckVariableDef(localDef);
- mBfIRBuilder->CreateStore(GetDefaultValue(varType), localDef->mAddr);
- localDef->Init();
- UpdateExprSrcPos(forEachStmt->mVariableName);
- AddLocalVariableDef(localDef, true);
- auto condBB = mBfIRBuilder->CreateBlock("forless.cond", true);
- auto bodyBB = mBfIRBuilder->CreateBlock("forless.body");
- auto incBB = mBfIRBuilder->CreateBlock("forless.inc");
- auto endBB = mBfIRBuilder->CreateBlock("forless.end");
- BfBreakData breakData;
- breakData.mIRContinueBlock = incBB;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->SetInsertPoint(condBB);
- if (forEachStmt->mCollectionExpression != NULL)
- UpdateExprSrcPos(forEachStmt->mCollectionExpression);
- BfIRValue conditionValue;
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- // Cond
- auto valueScopeStart = ValueScopeStart();
- auto localVal = mBfIRBuilder->CreateLoad(localDef->mAddr);
- if ((forEachStmt->mCollectionExpression != NULL) && (!didInference))
- target = CreateValueFromExpression(forEachStmt->mCollectionExpression, varType);
- if (!target)
- {
- // Soldier on
- target = GetDefaultTypedValue(varType);
- }
- if (forEachStmt->mInToken->mToken == BfToken_LessEquals)
- conditionValue = mBfIRBuilder->CreateCmpLTE(localVal, target.mValue, varType->IsSigned());
- else
- conditionValue = mBfIRBuilder->CreateCmpLT(localVal, target.mValue, varType->IsSigned());
- mBfIRBuilder->CreateCondBr(conditionValue, bodyBB, endBB);
- ValueScopeEnd(valueScopeStart);
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- // Body
- scopeData.mIsLoop = true;
- if (forEachStmt->mEmbeddedStatement != NULL)
- {
- VisitEmbeddedStatement(forEachStmt->mEmbeddedStatement);
- }
- // Inc
- if (!mCurMethodState->mLeftBlockUncond)
- {
- mBfIRBuilder->CreateBr(incBB);
- }
- mBfIRBuilder->AddBlock(incBB);
- mBfIRBuilder->SetInsertPoint(incBB);
- if (forEachStmt->mCollectionExpression != NULL)
- UpdateExprSrcPos(forEachStmt->mCollectionExpression);
- auto one = GetConstValue(1, localDef->mResolvedType);
- // We have to reload localVal before the inc, user logic could have changed it
- localVal = mBfIRBuilder->CreateLoad(localDef->mAddr);
- auto result = mBfIRBuilder->CreateAdd(localVal, one);
- mBfIRBuilder->CreateStore(result, localDef->mAddr);
- ValueScopeEnd(valueScopeStart);
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- // The 'return' may have been inside the block, which may not have been entered if preconditions were not met
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->mLeftBlockCond = false;
- RestoreScopeState();
- }
- void BfModule::Visit(BfForEachStatement* forEachStmt)
- {
- if ((forEachStmt->mInToken != NULL) &&
- ((forEachStmt->mInToken->GetToken() == BfToken_LChevron) || (forEachStmt->mInToken->GetToken() == BfToken_LessEquals)))
- {
- DoForLess(forEachStmt);
- return;
- }
- auto autoComplete = mCompiler->GetAutoComplete();
- UpdateSrcPos(forEachStmt);
- BfScopeData scopeData;
- // We set mIsLoop after the non-looped initializations
- if (forEachStmt->mLabelNode != NULL)
- scopeData.mLabelNode = forEachStmt->mLabelNode->mLabel;
- scopeData.mValueScopeStart = ValueScopeStart();
- mCurMethodState->AddScope(&scopeData);
- NewScopeState();
- bool isRefExpression = false;
- BfExpression* collectionExpr = forEachStmt->mCollectionExpression;
- if (auto unaryOpExpr = BfNodeDynCast<BfUnaryOperatorExpression>(collectionExpr))
- {
- if ((unaryOpExpr->mOp == BfUnaryOp_Ref) || (unaryOpExpr->mOp == BfUnaryOp_Mut))
- {
- isRefExpression = true;
- collectionExpr = unaryOpExpr->mExpression;
- }
- }
- BfTypedValue target;
- if (forEachStmt->mCollectionExpression != NULL)
- target = CreateValueFromExpression(collectionExpr);
- if (!target)
- {
- // Soldier on
- target = BfTypedValue(GetDefaultValue(mContext->mBfObjectType), mContext->mBfObjectType);
- }
- bool isLet = (forEachStmt->mVariableTypeRef != NULL) && (forEachStmt->mVariableTypeRef->IsA<BfLetTypeReference>());
- BfType* varType = NULL;
- bool inferVarType = false;
- if ((forEachStmt->mVariableTypeRef == NULL) || (forEachStmt->mVariableTypeRef->IsA<BfVarTypeReference>()) || (isLet))
- {
- if (target.mType->IsSizedArray())
- {
- varType = target.mType->GetUnderlyingType();
- if (isRefExpression)
- varType = CreateRefType(varType);
- }
- else if (target.mType->IsArray())
- {
- varType = target.mType->GetUnderlyingType();
- if (isRefExpression)
- varType = CreateRefType(varType);
- }
- else
- inferVarType = true;
- }
- else
- {
- if (autoComplete != NULL)
- autoComplete->CheckTypeRef(forEachStmt->mVariableTypeRef, false);
- varType = ResolveTypeRef(forEachStmt->mVariableTypeRef, BfPopulateType_Data, BfResolveTypeRefFlag_AllowRef);
- }
- if (varType == NULL)
- varType = mContext->mBfObjectType;
- bool isArray = target.mType->IsArray();
- bool isSizedArray = target.mType->IsSizedArray();
- bool isVarEnumerator = target.mType->IsVar();
- // Array
- BfType* itrType;
- BfTypedValue itr;
- BfTypeInstance* itrInterface = NULL;
- BfTypeInstance* refItrInterface = NULL;
- if (isVarEnumerator)
- varType = GetPrimitiveType(BfTypeCode_Var);
- if (target.mType->IsGenericParam())
- {
- auto genericParamInst = GetGenericParamInstance((BfGenericParamType*)target.mType);
- if ((genericParamInst->mGenericParamFlags & BfGenericParamFlag_Var) != 0)
- {
- varType = GetPrimitiveType(BfTypeCode_Var);
- isVarEnumerator = true;
- }
- if (genericParamInst->mTypeConstraint != NULL)
- {
- if (genericParamInst->mTypeConstraint->IsVar())
- {
- varType = GetPrimitiveType(BfTypeCode_Var);
- isVarEnumerator = true;
- }
- if (genericParamInst->mTypeConstraint->IsGenericTypeInstance())
- {
- auto genericConstraintType = (BfGenericTypeInstance*)genericParamInst->mTypeConstraint;
- if (genericConstraintType->mTypeDef == mCompiler->mSizedArrayTypeDef)
- {
- varType = genericConstraintType->mTypeGenericArguments[0];
- isVarEnumerator = true;
- }
- }
- }
- }
- if (isArray || isSizedArray)
- {
- itrType = GetPrimitiveType(BfTypeCode_IntPtr);
- BfIRValue itrInst = CreateAlloca(itrType);
- itr = BfTypedValue(itrInst, itrType, true);
- }
- else if (isVarEnumerator)
- {
- // Generic method or mixin decl
- }
- else if (!target.mType->IsTypeInstance())
- {
- Fail(StrFormat("Type '%s' cannot be used in enumeration", TypeToString(target.mType).c_str()), forEachStmt->mCollectionExpression);
- }
- else if (forEachStmt->mCollectionExpression != NULL)
- {
- auto targetTypeInstance = target.mType->ToTypeInstance();
- PopulateType(targetTypeInstance, BfPopulateType_DataAndMethods);
-
- itr = target;
- bool hadGetEnumeratorType;
- auto getEnumeratorMethod = GetMethodByName(targetTypeInstance, "GetEnumerator", 0, true);
- if (!getEnumeratorMethod)
- {
- hadGetEnumeratorType = false;
- }
- else if (getEnumeratorMethod.mMethodInstance->mMethodDef->mIsStatic)
- {
- hadGetEnumeratorType = true;
- Fail(StrFormat("Type '%s' does not contain a non-static 'GetEnumerator' method", TypeToString(targetTypeInstance).c_str()), forEachStmt->mCollectionExpression);
- }
- else if (getEnumeratorMethod.mMethodInstance->mMethodDef->mIsConcrete)
- {
- hadGetEnumeratorType = true;
- Fail(StrFormat("Iteration requires a concrete implementation of '%s'", TypeToString(targetTypeInstance).c_str()), forEachStmt->mCollectionExpression);
- }
- else
- {
- hadGetEnumeratorType = true;
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfIRValue, 1> args;
- auto castedTarget = Cast(forEachStmt->mCollectionExpression, target, getEnumeratorMethod.mMethodInstance->GetOwner());
- exprEvaluator.PushThis(forEachStmt->mCollectionExpression, castedTarget, getEnumeratorMethod.mMethodInstance, args);
- itr = exprEvaluator.CreateCall(getEnumeratorMethod.mMethodInstance, mCompiler->IsSkippingExtraResolveChecks() ? BfIRValue() : getEnumeratorMethod.mFunc, false, args);
- }
- if (itr)
- {
- PopulateType(itr.mType, BfPopulateType_DataAndMethods);
- BfGenericTypeInstance* genericItrInterface = NULL;
- auto enumeratorTypeInst = itr.mType->ToTypeInstance();
- if (enumeratorTypeInst != NULL)
- {
- for (auto& interfaceRef : enumeratorTypeInst->mInterfaces)
- {
- BfTypeInstance* interface = interfaceRef.mInterfaceType;
- if (interface->mTypeDef == (isRefExpression ? mCompiler->mGenericIRefEnumeratorTypeDef : mCompiler->mGenericIEnumeratorTypeDef))
- {
- if (genericItrInterface != NULL)
- {
- Fail(StrFormat("Type '%s' implements multiple %s<T> interfaces", TypeToString(itr.mType).c_str(), isRefExpression ? "IRefEnumerator" : "IEnumerator"), forEachStmt->mCollectionExpression);
- return;
- }
- itrInterface = interface;
- genericItrInterface = itrInterface->ToGenericTypeInstance();
- if (inferVarType)
- {
- varType = genericItrInterface->mTypeGenericArguments[0];
- if (isRefExpression)
- varType = CreateRefType(varType);
- }
- }
- }
- if (enumeratorTypeInst->mTypeDef == (isRefExpression ? mCompiler->mGenericIRefEnumeratorTypeDef : mCompiler->mGenericIEnumeratorTypeDef))
- {
- itrInterface = enumeratorTypeInst;
- genericItrInterface = itrInterface->ToGenericTypeInstance();
- if (inferVarType)
- {
- varType = genericItrInterface->mTypeGenericArguments[0];
- if (isRefExpression)
- varType = CreateRefType(varType);
- }
- }
- }
- if (genericItrInterface == NULL)
- {
- if (!hadGetEnumeratorType)
- {
- Fail(StrFormat("Type '%s' must contain a 'GetEnumerator' method or implement an IEnumerator<T> interface", TypeToString(targetTypeInstance).c_str()), forEachStmt->mCollectionExpression);
- }
- else
- Fail(StrFormat("Enumerator type '%s' must implement an %s<T> interface", TypeToString(itr.mType).c_str(), isRefExpression ? "IRefEnumerator" : "IEnumerator"), forEachStmt->mCollectionExpression);
- itrInterface = NULL;
- itr = BfTypedValue();
- }
- else
- {
- itrInterface = genericItrInterface;
- if (isRefExpression)
- {
- refItrInterface = itrInterface;
- PopulateType(refItrInterface);
- // Must IRefEnumeratorf<T> must include only IEnumerator<T>
- BF_ASSERT(refItrInterface->mInterfaces.size() == 1);
- if (refItrInterface->mInterfaces.size() == 1)
- itrInterface = refItrInterface->mInterfaces[0].mInterfaceType;
- }
- itr = MakeAddressable(itr);
- itr = RemoveReadOnly(itr);
- }
- }
- }
- else
- {
- AssertErrorState();
- }
- PopulateType(varType, BfPopulateType_Data);
- // Apply deferred local assign to mEmbeddedStatement and itrStmt
- BfDeferredLocalAssignData deferredLocalAssignData(mCurMethodState->mCurScope);
- deferredLocalAssignData.ExtendFrom(mCurMethodState->mDeferredLocalAssignData, false);
- deferredLocalAssignData.mVarIdBarrier = mCurMethodState->GetRootMethodState()->mCurLocalVarId;
- SetAndRestoreValue<BfDeferredLocalAssignData*> prevDLA(mCurMethodState->mDeferredLocalAssignData, &deferredLocalAssignData);
- if ((target.mType->IsSizedArray()) && (((BfSizedArrayType*)target.mType)->mElementCount == 0))
- {
- EmitEnsureInstructionAt();
- SetAndRestoreValue<bool> ignoreWrites(mBfIRBuilder->mIgnoreWrites, true);
- if (forEachStmt->mEmbeddedStatement != NULL)
- VisitEmbeddedStatement(forEachStmt->mEmbeddedStatement);
- RestoreScopeState();
- return;
- }
- BfIdentifierNode* nameNode = NULL;
- String variableName;
- struct _TupleBind
- {
- BfIdentifierNode* mNameNode;
- String mName;
- BfType* mType;
- BfLocalVariable* mVariable;
- };
- Array<_TupleBind> tupleBinds;
-
- if (forEachStmt->mVariableName != NULL)
- {
- if (auto tupleExpr = BfNodeDynCast<BfTupleExpression>(forEachStmt->mVariableName))
- {
- CheckTupleVariableDeclaration(tupleExpr, varType);
-
- if (varType->IsTuple())
- {
- auto tupleType = (BfTupleType*)varType;
- for (int idx = 0; idx < BF_MIN((int)tupleExpr->mValues.size(), (int)tupleType->mFieldInstances.size()); idx++)
- {
- auto nameNode = tupleExpr->mValues[idx];
-
- _TupleBind tupleBind;
- tupleBind.mNameNode = BfNodeDynCast<BfIdentifierNode>(nameNode);
- if ((tupleBind.mNameNode == NULL) && (nameNode != NULL))
- {
- Fail("Variable name expected", nameNode);
- }
- tupleBind.mName = nameNode->ToString();
- tupleBind.mType = tupleType->mFieldInstances[idx].mResolvedType;
- tupleBind.mVariable = NULL;
-
- tupleBinds.Add(tupleBind);
- if (idx == 0)
- variableName = tupleBind.mName;
- }
- }
- }
- else
- {
- nameNode = BfNodeDynCast<BfIdentifierNode>(forEachStmt->mVariableName);
- if (nameNode != NULL)
- variableName = nameNode->ToString();
- }
- }
-
- if (variableName.IsEmpty())
- variableName = "_";
-
- BfModuleMethodInstance getNextMethodInst;
- BfTypedValue nextResult;
- if ((refItrInterface) || (itrInterface))
- {
- if (isRefExpression)
- {
- PopulateType(refItrInterface, BfPopulateType_Full_Force);
- getNextMethodInst = GetMethodByName(refItrInterface, "GetNextRef");
- }
- else
- {
- PopulateType(itrInterface, BfPopulateType_Full_Force);
- getNextMethodInst = GetMethodByName(itrInterface, "GetNext");
- }
- BF_ASSERT(getNextMethodInst);
- nextResult = BfTypedValue(CreateAlloca(getNextMethodInst.mMethodInstance->mReturnType), getNextMethodInst.mMethodInstance->mReturnType, true);
- }
- BfLocalVariable* itrLocalDef = NULL;
- // Iterator local def
- if (itr)
- {
- BfLocalVariable* localDef = new BfLocalVariable();
- itrLocalDef = localDef;
- localDef->mNameNode = nameNode;
- localDef->mName = variableName;
- localDef->mResolvedType = itr.mType;
- localDef->mAddr = itr.mValue;
- localDef->mIsAssigned = true;
- localDef->mReadFromId = 0;
- localDef->Init();
- UpdateSrcPos(forEachStmt);
- CheckVariableDef(localDef);
- AddLocalVariableDef(localDef, true);
- }
- BfIRValue varInst;
- BfTypedValue varTypedVal;
- bool needsValCopy = true;
-
- // Local variable
- {
- if (!tupleBinds.IsEmpty())
- {
- BF_ASSERT(varType->IsTuple());
- auto tupleType = (BfTupleType*)varType;
- // Tuple binds
- needsValCopy = false;
- if (!nextResult)
- {
- varInst = CreateAlloca(varType);
- varTypedVal = BfTypedValue(varInst, varType, true);
- }
- // Local
- for (int idx = 0; idx < (int)tupleBinds.size(); idx++)
- {
- auto& tupleBind = tupleBinds[idx];
- BfLocalVariable* localDef = new BfLocalVariable();
- localDef->mNameNode = tupleBind.mNameNode;
- localDef->mName = tupleBind.mName;
- localDef->mResolvedType = tupleBind.mType;
- if (!needsValCopy)
- localDef->mResolvedType = CreateRefType(localDef->mResolvedType);
- localDef->mAddr = CreateAlloca(localDef->mResolvedType);
- localDef->mIsAssigned = true;
- localDef->mReadFromId = 0;
- if ((isLet) || (forEachStmt->mReadOnlyToken != NULL))
- localDef->mIsReadOnly = true;
- localDef->Init();
-
- auto fieldInstance = &tupleType->mFieldInstances[idx];
- if (fieldInstance->mDataIdx >= 0)
- {
- auto tuplePtrType = CreatePointerType(varType);
- BfIRValue tuplePtr;
- if (nextResult)
- tuplePtr = mBfIRBuilder->CreateBitCast(nextResult.mValue, mBfIRBuilder->MapType(tuplePtrType));
- else
- tuplePtr = mBfIRBuilder->CreateBitCast(varInst, mBfIRBuilder->MapType(tuplePtrType));
- auto valAddr = mBfIRBuilder->CreateInBoundsGEP(tuplePtr, 0, fieldInstance->mDataIdx);
- mBfIRBuilder->CreateStore(valAddr, localDef->mAddr);
- }
-
- UpdateSrcPos(forEachStmt);
- if ((itrLocalDef != NULL) && (idx == 0))
- {
- localDef->mLocalVarId = itrLocalDef->mLocalVarId;
- localDef->mIsShadow = true;
- }
- else
- {
- CheckVariableDef(localDef);
- }
- AddLocalVariableDef(localDef, true, false, BfIRValue(), BfIRInitType_NotNeeded_AliveOnDecl);
- }
- }
- else
- {
- // Normal case
- if ((nextResult) && (varType->IsComposite()))
- {
- needsValCopy = false;
- varType = CreateRefType(varType);
- }
- // Local
- BfLocalVariable* localDef = new BfLocalVariable();
- localDef->mNameNode = nameNode;
- localDef->mName = variableName;
- localDef->mResolvedType = varType;
- varInst = CreateAlloca(varType);
- localDef->mAddr = varInst;
- localDef->mIsAssigned = true;
- localDef->mReadFromId = 0;
- if ((isLet) || (forEachStmt->mReadOnlyToken != NULL))
- localDef->mIsReadOnly = true;
- localDef->Init();
- if (!needsValCopy)
- {
- auto valAddr = mBfIRBuilder->CreateBitCast(nextResult.mValue, mBfIRBuilder->MapType(varType));
- mBfIRBuilder->CreateStore(valAddr, varInst);
- }
- UpdateSrcPos(forEachStmt);
- if (itrLocalDef != NULL)
- {
- localDef->mLocalVarId = itrLocalDef->mLocalVarId;
- localDef->mIsShadow = true;
- }
- else
- {
- CheckVariableDef(localDef);
- }
- AddLocalVariableDef(localDef, true, false, BfIRValue(), BfIRInitType_NotNeeded_AliveOnDecl);
- varTypedVal = BfTypedValue(varInst, varType, true);
- }
- }
- // Iterator
- if (itr)
- {
- if ((!isArray) && (!isSizedArray))
- {
- BfFunctionBindResult functionBindResult;
- BfExprEvaluator exprEvaluator(this);
- exprEvaluator.mFunctionBindResult = &functionBindResult;
- // Allow for "Dispose" not to exist
- SetAndRestoreValue<bool> prevIgnoreErrors(mIgnoreErrors, true);
- BfResolvedArgs resolvedArgs;
- exprEvaluator.MatchMethod(forEachStmt->mCollectionExpression, NULL, itr, false, false, "Dispose", resolvedArgs, NULL);
- if (functionBindResult.mMethodInstance != NULL)
- {
- BfModuleMethodInstance moduleMethodInstance;
- moduleMethodInstance = BfModuleMethodInstance(functionBindResult.mMethodInstance, functionBindResult.mFunc);
- AddDeferredCall(moduleMethodInstance, functionBindResult.mIRArgs, mCurMethodState->mCurScope);
- }
- }
- }
- scopeData.mIsLoop = true;
- if (isArray || isSizedArray)
- mBfIRBuilder->CreateStore(GetConstValue(0), itr.mValue);
- auto valueScopeStartInner = ValueScopeStart();
- // We may have a call in the loop body
- mCurMethodState->mMayNeedThisAccessCheck = true;
- auto condBB = mBfIRBuilder->CreateBlock("foreach.cond", true);
- auto bodyBB = mBfIRBuilder->CreateBlock("foreach.body");
- auto incBB = mBfIRBuilder->CreateBlock("foreach.inc");
- auto endBB = mBfIRBuilder->CreateBlock("foreach.end");
- BfBreakData breakData;
- breakData.mIRContinueBlock = incBB;
- breakData.mIRBreakBlock = endBB;
- breakData.mScope = &scopeData;
- breakData.mInnerValueScopeStart = valueScopeStartInner;
- breakData.mPrevBreakData = mCurMethodState->mBreakData;
- SetAndRestoreValue<BfBreakData*> prevBreakData(mCurMethodState->mBreakData, &breakData);
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->SetInsertPoint(condBB);
- if (forEachStmt->mCollectionExpression != NULL)
- UpdateExprSrcPos(forEachStmt->mCollectionExpression);
- BfIRValue conditionValue;
- if (isSizedArray) // if (i < lengthof(array)
- {
- auto itrVal = mBfIRBuilder->CreateLoad(itr.mValue);
- auto arrayType = (BfSizedArrayType*)target.mType;
- PopulateType(arrayType, BfPopulateType_DataAndMethods);
- BfIRValue lengthVal = GetConstValue(arrayType->mElementCount);
- conditionValue = mBfIRBuilder->CreateCmpLT(itrVal, lengthVal, true);
- mBfIRBuilder->CreateCondBr(conditionValue, bodyBB, endBB);
- ValueScopeEnd(valueScopeStartInner);
- }
- else if (isArray) // if (i < array.mLength)
- {
- auto itrVal = mBfIRBuilder->CreateLoad(itr.mValue);
- auto arrayType = (BfArrayType*)target.mType;
- PopulateType(arrayType);
- auto arrayBaseValue = mBfIRBuilder->CreateBitCast(target.mValue, mBfIRBuilder->MapType(arrayType->mBaseType, BfIRPopulateType_Full));
- int getLengthBitCount = arrayType->GetLengthBitCount();
- BfIRValue lengthVal;
- if (arrayType->mBaseType->mTypeFailed)
- {
- AssertErrorState();
- if (getLengthBitCount == 64)
- lengthVal = GetConstValue64(0);
- else
- lengthVal = GetConstValue32(0);
- }
- else
- {
- auto lengthValAddr = mBfIRBuilder->CreateInBoundsGEP(arrayBaseValue, 0, 1);
- lengthVal = mBfIRBuilder->CreateLoad(lengthValAddr);
- }
- lengthVal = mBfIRBuilder->CreateNumericCast(lengthVal, true, BfTypeCode_IntPtr);
- conditionValue = mBfIRBuilder->CreateCmpLT(itrVal, lengthVal, true);
- mBfIRBuilder->CreateCondBr(conditionValue, bodyBB, endBB);
- ValueScopeEnd(valueScopeStartInner);
- }
- else // if (itr.MoveNext())
- {
- if (!itr)
- {
- if (!isVarEnumerator)
- AssertErrorState();
- }
- else
- {
- BfExprEvaluator exprEvaluator(this);
- auto itrTypeInstance = itr.mType->ToTypeInstance();
- SizedArray<BfResolvedArg, 0> resolvedArgs;
- BfMethodMatcher methodMatcher(forEachStmt->mCollectionExpression, this, getNextMethodInst.mMethodInstance, resolvedArgs);
- if (isRefExpression)
- methodMatcher.CheckType(refItrInterface, itr, false);
- else
- methodMatcher.CheckType(itrInterface, itr, false);
- methodMatcher.TryDevirtualizeCall(itr);
- exprEvaluator.mReceivingValue = &nextResult;
- auto retVal = exprEvaluator.CreateCall(&methodMatcher, itr);
- if (exprEvaluator.mReceivingValue != NULL)
- {
- AssertErrorState();
- }
- if (retVal)
- {
- auto i8Result = ExtractValue(nextResult, NULL, 2);
- i8Result = LoadValue(i8Result);
- BF_ASSERT(i8Result.mType == GetPrimitiveType(BfTypeCode_Int8));
- conditionValue = mBfIRBuilder->CreateCmpEQ(i8Result.mValue, GetConstValue8(0));
- }
- else
- conditionValue = GetDefaultValue(GetPrimitiveType(BfTypeCode_Boolean));
- mBfIRBuilder->CreateCondBr(conditionValue, bodyBB, endBB);
- ValueScopeEnd(valueScopeStartInner);
- }
- }
- mBfIRBuilder->AddBlock(bodyBB);
- mBfIRBuilder->SetInsertPoint(bodyBB);
- if (!varTypedVal)
- {
- // Nothing to do...
- }
- else if (isSizedArray) // val = array[i]
- {
- auto itrVal = mBfIRBuilder->CreateLoad(itr.mValue);
- auto arrayType = (BfSizedArrayType*)target.mType;
- BfType* ptrType = CreatePointerType(arrayType->mElementType);
- BfTypedValue arrayItem;
-
- if (arrayType->mElementType->IsValuelessType())
- {
- arrayItem = GetDefaultTypedValue(arrayType->mElementType);
- }
- else
- {
- BfIRValue ptrValue = mBfIRBuilder->CreateBitCast(target.mValue, mBfIRBuilder->MapType(ptrType));
- arrayItem = BfTypedValue(CreateIndexedValue(arrayType->mElementType, ptrValue, itrVal), arrayType->mElementType, true);
- if (isRefExpression)
- arrayItem = BfTypedValue(arrayItem.mValue, CreateRefType(arrayItem.mType));
- }
- arrayItem = Cast(forEachStmt->mCollectionExpression, arrayItem, varType, BfCastFlags_Explicit);
- if ((arrayItem) && (!arrayItem.mValue.IsFake()))
- {
- arrayItem = LoadValue(arrayItem);
- if (arrayItem)
- mBfIRBuilder->CreateStore(arrayItem.mValue, varInst);
- }
- }
- else if (isArray) // val = array[i]
- {
- auto itrVal = mBfIRBuilder->CreateLoad(itr.mValue);
- BfTypedValueExpression typedValueExpr;
- typedValueExpr.Init(BfTypedValue(itrVal, itrType));
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfExpression*, 1> indices;
- indices.push_back(&typedValueExpr);
- BfSizedArray<BfExpression*> sizedArgExprs(indices);
- BfResolvedArgs argValues(&sizedArgExprs);
- exprEvaluator.ResolveArgValues(argValues);
- bool boundsCheck = mCompiler->mOptions.mRuntimeChecks;
- auto typeOptions = GetTypeOptions();
- if (typeOptions != NULL)
- boundsCheck = BfTypeOptions::Apply(boundsCheck, typeOptions->mRuntimeChecks);
-
- BfMethodMatcher methodMatcher(forEachStmt->mVariableName, this, "get__", argValues.mResolvedArgs, NULL);
- methodMatcher.mMethodType = BfMethodType_PropertyGetter;
- methodMatcher.CheckType(target.mType->ToTypeInstance(), target, false);
- methodMatcher.mCheckedKind = boundsCheck ? BfCheckedKind_Checked : BfCheckedKind_Unchecked;
- BfTypedValue arrayItem = exprEvaluator.CreateCall(&methodMatcher, target);
- if ((varInst) && (arrayItem))
- {
- if (isRefExpression)
- arrayItem = BfTypedValue(arrayItem.mValue, CreateRefType(arrayItem.mType));
- else if (!arrayItem.mType->IsComposite())
- arrayItem = LoadValue(arrayItem);
- arrayItem = Cast(forEachStmt->mCollectionExpression, arrayItem, varType, BfCastFlags_Explicit);
- arrayItem = LoadValue(arrayItem);
- if (arrayItem)
- mBfIRBuilder->CreateStore(arrayItem.mValue, varInst);
- }
- }
- else
- {
- if (!itr)
- {
- if (!isVarEnumerator)
- AssertErrorState();
- }
- else if (mCompiler->IsAutocomplete())
- {
- // If we don't do this shortcut, we can end up creating temporary "boxed" objects
- }
- else
- {
- if (needsValCopy)
- {
- auto valAddr = mBfIRBuilder->CreateBitCast(nextResult.mValue, mBfIRBuilder->MapType(CreatePointerType(varType)));
- auto val = mBfIRBuilder->CreateLoad(valAddr);
- mBfIRBuilder->CreateStore(val, varInst);
- }
- }
- }
- if (forEachStmt->mEmbeddedStatement != NULL)
- {
- VisitEmbeddedStatement(forEachStmt->mEmbeddedStatement);
- }
- if (!mCurMethodState->mLeftBlockUncond)
- {
- ValueScopeEnd(valueScopeStartInner);
- mBfIRBuilder->CreateBr(incBB);
- }
- mBfIRBuilder->AddBlock(incBB);
- mBfIRBuilder->SetInsertPoint(incBB);
-
- if (isArray || isSizedArray)
- {
- auto val = mBfIRBuilder->CreateLoad(itr.mValue);
- auto result = mBfIRBuilder->CreateAdd(val, GetConstValue(1));
- mBfIRBuilder->CreateStore(result, itr.mValue);
- }
- else
- {
- // Nothing to do
- }
-
- mBfIRBuilder->CreateBr(condBB);
- mBfIRBuilder->AddBlock(endBB);
- mBfIRBuilder->SetInsertPoint(endBB);
- if ((itrLocalDef != NULL) && (itrLocalDef->mDbgVarInst) && (IsTargetingBeefBackend()))
- {
- // If this shadows another enumerator variable then we need to explicitly mark the end of this one
- mBfIRBuilder->DbgLifetimeEnd(itrLocalDef->mDbgVarInst);
- }
- // The 'return' may have been inside the block, which may not have been entered if preconditions were not met
- mCurMethodState->SetHadReturn(false);
- mCurMethodState->mLeftBlockUncond = false;
- mCurMethodState->mLeftBlockCond = false;
- RestoreScopeState();
- }
- void BfModule::Visit(BfDeferStatement* deferStmt)
- {
- if (deferStmt->mTargetNode == NULL)
- {
- AssertErrorState();
- return;
- }
- //TODO: Why in the world didn't we want to be able to step onto a defer statement?
- // We only want the breakpoint to hit on execution of the defer, not on insertion of it
- //SetAndRestoreValue<bool> prevSetIllegalSrcPos(mSetIllegalSrcPosition, true);
- UpdateSrcPos(deferStmt);
- EmitEnsureInstructionAt();
-
- if ((mCompiler->mResolvePassData != NULL) && (mCompiler->mResolvePassData->mAutoComplete != NULL))
- {
- auto targetIdentifier = BfNodeDynCast<BfIdentifierNode>(deferStmt->mScopeName);
- if ((deferStmt->mScopeName == NULL) || (targetIdentifier != NULL))
- mCompiler->mResolvePassData->mAutoComplete->CheckLabel(targetIdentifier, deferStmt->mColonToken);
- }
- BfScopeData* scope = NULL;
- if (deferStmt->mScopeToken != NULL)
- {
- if (deferStmt->mScopeToken->GetToken() == BfToken_Scope)
- scope = mCurMethodState->mCurScope->GetTargetable();
- else
- scope = &mCurMethodState->mHeadScope;
- }
- else if (deferStmt->mScopeName != NULL)
- scope = FindScope(deferStmt->mScopeName, true);
- else
- scope = mCurMethodState->mCurScope;
- if (auto block = BfNodeDynCast<BfBlock>(deferStmt->mTargetNode))
- {
- if (deferStmt->mBind != NULL)
- {
- Array<BfDeferredCapture> captures;
- for (auto identifier : deferStmt->mBind->mParams)
- {
- BfDeferredCapture deferredCapture;
- deferredCapture.mName = identifier->ToString();
- deferredCapture.mValue = CreateValueFromExpression(identifier);
- if (deferredCapture.mValue)
- {
- captures.push_back(deferredCapture);
- }
- }
- AddDeferredBlock(block, scope, &captures);
- }
- else
- AddDeferredBlock(block, scope);
- }
- else if (auto exprStmt = BfNodeDynCast<BfExpressionStatement>(deferStmt->mTargetNode))
- {
- BfExprEvaluator expressionEvaluator(this);
- expressionEvaluator.mDeferCallRef = exprStmt->mExpression;
- expressionEvaluator.mDeferScopeAlloc = scope;
- expressionEvaluator.VisitChild(exprStmt->mExpression);
- }
- else if (auto deleteStmt = BfNodeDynCast<BfDeleteStatement>(deferStmt->mTargetNode))
- {
- if (deleteStmt->mExpression == NULL)
- {
- AssertErrorState();
- return;
- }
- auto val = CreateValueFromExpression(deleteStmt->mExpression);
- if (!val)
- return;
- if (mCompiler->IsAutocomplete())
- return;
- bool isGenericParam = false;
- auto checkType = val.mType;
- if (val.mType->IsGenericParam())
- {
- isGenericParam = true;
- auto genericParamInst = GetGenericParamInstance((BfGenericParamType*)val.mType);
- if (genericParamInst->mGenericParamFlags & BfGenericParamFlag_Delete)
- return;
- if (genericParamInst->mTypeConstraint != NULL)
- checkType = genericParamInst->mTypeConstraint;
- }
- bool isAppendDelete = false;
- BfTypedValue customAllocator;
- if (deleteStmt->mAllocExpr != NULL)
- {
- if (auto expr = BfNodeDynCast<BfExpression>(deleteStmt->mAllocExpr))
- customAllocator = CreateValueFromExpression(expr);
- else if (auto tokenNode = BfNodeDynCast<BfTokenNode>(deleteStmt->mAllocExpr))
- {
- if (tokenNode->mToken == BfToken_Append)
- isAppendDelete = true;
- }
- }
- auto internalType = ResolveTypeDef(mCompiler->mInternalTypeDef);
- PopulateType(checkType);
- if (checkType->IsVar())
- return;
- if ((!checkType->IsObjectOrInterface()) && (!checkType->IsPointer()))
- {
- VisitChild(deferStmt->mTargetNode);
- Fail(StrFormat("Cannot delete a value of type '%s'", TypeToString(val.mType).c_str()), deferStmt->mTargetNode);
- return;
- }
- if (isGenericParam)
- return;
- if (customAllocator)
- {
- BfFunctionBindResult functionBindResult;
- functionBindResult.mWantsArgs = true;
- auto customAllocTypeInst = customAllocator.mType->ToTypeInstance();
- if ((checkType->IsObjectOrInterface()) && (customAllocTypeInst != NULL) && (customAllocTypeInst->mTypeDef->GetMethodByName("FreeObject") != NULL))
- {
- BfTypedValueExpression typedValueExpr;
- typedValueExpr.Init(val);
- typedValueExpr.mRefNode = deleteStmt->mAllocExpr;
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfExpression*, 2> argExprs;
- argExprs.push_back(&typedValueExpr);
- BfSizedArray<BfExpression*> sizedArgExprs(argExprs);
- BfResolvedArgs argValues(&sizedArgExprs);
- exprEvaluator.ResolveArgValues(argValues);
- exprEvaluator.mNoBind = true;
- exprEvaluator.mFunctionBindResult = &functionBindResult;
- exprEvaluator.MatchMethod(deleteStmt->mAllocExpr, NULL, customAllocator, false, false, "FreeObject", argValues, NULL);
- }
- else
- {
- auto voidPtrType = GetPrimitiveType(BfTypeCode_NullPtr);
- auto ptrValue = BfTypedValue(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(voidPtrType)), voidPtrType);
- BfTypedValueExpression typedValueExpr;
- typedValueExpr.Init(ptrValue);
- BfExprEvaluator exprEvaluator(this);
- SizedArray<BfExpression*, 2> argExprs;
- argExprs.push_back(&typedValueExpr);
- BfSizedArray<BfExpression*> sizedArgExprs(argExprs);
- BfResolvedArgs argValues(&sizedArgExprs);
- exprEvaluator.ResolveArgValues(argValues);
- exprEvaluator.mNoBind = true;
- exprEvaluator.mFunctionBindResult = &functionBindResult;
- exprEvaluator.MatchMethod(deleteStmt->mAllocExpr, NULL, customAllocator, false, false, "Free", argValues, NULL);
- }
- if (functionBindResult.mMethodInstance != NULL)
- {
- AddDeferredCall(BfModuleMethodInstance(functionBindResult.mMethodInstance, functionBindResult.mFunc), functionBindResult.mIRArgs, scope, deleteStmt, true);
- }
- }
- if (checkType->IsObjectOrInterface())
- {
- auto objectType = mContext->mBfObjectType;
- PopulateType(objectType);
- BfMethodInstance* methodInstance = objectType->mVirtualMethodTable[mCompiler->GetVTableMethodOffset() + 0].mImplementingMethod;
- BF_ASSERT(methodInstance->mMethodDef->mName == "~this");
- SizedArray<BfIRValue, 1> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->MapType(objectType)));
- if (!customAllocator)
- {
- if (mCompiler->mOptions.mEnableRealtimeLeakCheck)
- {
- auto moduleMethodInstance = GetInternalMethod("Dbg_MarkObjectDeleted");
- AddDeferredCall(moduleMethodInstance, llvmArgs, scope, deleteStmt, false, true);
- }
- else
- {
- auto moduleMethodInstance = GetInternalMethod("Free");
- SizedArray<BfIRValue, 1> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->GetPrimitiveType(BfTypeCode_NullPtr)));
- AddDeferredCall(moduleMethodInstance, llvmArgs, scope, deleteStmt, false, true);
- }
- }
- auto moduleMethodInstance = GetMethodInstance(objectType, methodInstance->mMethodDef, BfTypeVector());
- AddDeferredCall(moduleMethodInstance, llvmArgs, scope, deleteStmt, false, true);
- if (mCompiler->mOptions.mObjectHasDebugFlags)
- {
- auto moduleMethodInstance = GetMethodByName(internalType->ToTypeInstance(), (deleteStmt->mTargetTypeToken != NULL) ? "Dbg_ObjectPreCustomDelete" : "Dbg_ObjectPreDelete");
- AddDeferredCall(moduleMethodInstance, llvmArgs, scope, deleteStmt, false, true);
- }
- }
- else
- {
- if ((!customAllocator) && (!isAppendDelete))
- {
- val = LoadValue(val);
- BfModuleMethodInstance moduleMethodInstance;
- if (mCompiler->mOptions.mEnableRealtimeLeakCheck)
- moduleMethodInstance = GetMethodByName(internalType->ToTypeInstance(), "Dbg_RawFree");
- else
- moduleMethodInstance = GetMethodByName(internalType->ToTypeInstance(), "Free");
- SizedArray<BfIRValue, 1> llvmArgs;
- llvmArgs.push_back(mBfIRBuilder->CreateBitCast(val.mValue, mBfIRBuilder->GetPrimitiveType(BfTypeCode_NullPtr)));
- AddDeferredCall(moduleMethodInstance, llvmArgs, scope, deleteStmt, false, true);
- }
- }
- }
- else
- {
- AssertErrorState();
- VisitChild(deferStmt->mTargetNode);
- }
- }
- void BfModule::Visit(BfBlock* block)
- {
- VisitEmbeddedStatement(block);
- }
- void BfModule::Visit(BfLabeledBlock* labeledBlock)
- {
- VisitEmbeddedStatement(labeledBlock);
- }
- void BfModule::Visit(BfRootNode* rootNode)
- {
- VisitMembers(rootNode);
- }
- void BfModule::Visit(BfInlineAsmStatement* asmStmt)
- {
- #if 0
- enum RegClobberFlags //CDH TODO add support for mmx/xmm/fpst etc (how are these signified in LLVM? check LangRef inline asm docs clobber list info)
- {
- // please keep eax through edx in alphabetical order (grep $BYTEREGS for why)
- REGCLOBBERF_EAX = (1 << 0),
- REGCLOBBERF_EBX = (1 << 1),
- REGCLOBBERF_ECX = (1 << 2),
- REGCLOBBERF_EDX = (1 << 3),
- REGCLOBBERF_ESI = (1 << 4),
- REGCLOBBERF_EDI = (1 << 5),
- REGCLOBBERF_ESP = (1 << 6),
- REGCLOBBERF_EBP = (1 << 7),
- REGCLOBBERF_XMM0 = (1 << 8),
- REGCLOBBERF_XMM1 = (1 << 9),
- REGCLOBBERF_XMM2 = (1 << 10),
- REGCLOBBERF_XMM3 = (1 << 11),
- REGCLOBBERF_XMM4 = (1 << 12),
- REGCLOBBERF_XMM5 = (1 << 13),
- REGCLOBBERF_XMM6 = (1 << 14),
- REGCLOBBERF_XMM7 = (1 << 15),
- REGCLOBBERF_FPST0 = (1 << 16),
- REGCLOBBERF_FPST1 = (1 << 17),
- REGCLOBBERF_FPST2 = (1 << 18),
- REGCLOBBERF_FPST3 = (1 << 19),
- REGCLOBBERF_FPST4 = (1 << 20),
- REGCLOBBERF_FPST5 = (1 << 21),
- REGCLOBBERF_FPST6 = (1 << 22),
- REGCLOBBERF_FPST7 = (1 << 23),
- REGCLOBBERF_MM0 = (1 << 24),
- REGCLOBBERF_MM1 = (1 << 25),
- REGCLOBBERF_MM2 = (1 << 26),
- REGCLOBBERF_MM3 = (1 << 27),
- REGCLOBBERF_MM4 = (1 << 28),
- REGCLOBBERF_MM5 = (1 << 29),
- REGCLOBBERF_MM6 = (1 << 30),
- REGCLOBBERF_MM7 = (1 << 31),
- };
- const char* regClobberNames[] = { "eax", "ebx", "ecx", "edx", "esi", "edi", "esp", "ebp", nullptr }; // must be in same order as flags
- unsigned long regClobberFlags = 0;
- std::function<bool(const StringImpl&, bool)> matchRegFunc = [this, ®ClobberNames, ®ClobberFlags](const StringImpl& name, bool isClobber)
- {
- bool found = false;
- int nameLen = name.length();
- if (nameLen == 3)
- {
- if ((name[0] == 's') && (name[1] == 't') && (name[2] >= '0') && (name[2] <= '7'))
- {
- // st# regs (unparenthesized at this point)
- if (isClobber)
- regClobberFlags |= (REGCLOBBERF_FPST0 << (name[2] - '0'));
- found = true;
- }
- else if ((name[0] == 'm') && (name[1] == 'm') && (name[2] >= '0') && (name[2] <= '7'))
- {
- // mm regs
- if (isClobber)
- regClobberFlags |= (REGCLOBBERF_MM0 << (name[2] - '0'));
- found = true;
- }
- else
- {
- // dword regs
- for (int iRegCheck = 0; regClobberNames[iRegCheck] != nullptr; ++iRegCheck)
- {
- if (!strcmp(name.c_str(), regClobberNames[iRegCheck]))
- {
- if (isClobber)
- regClobberFlags |= (1 << iRegCheck);
- found = true;
- break;
- }
- }
- }
- }
- else if (nameLen == 2)
- {
- // word & byte regs
- for (int iRegCheck = 0; regClobberNames[iRegCheck] != nullptr; ++iRegCheck)
- {
- if (!strcmp(name.c_str(), regClobberNames[iRegCheck] + 1)) // skip leading 'e'
- {
- if (isClobber)
- regClobberFlags |= (1 << iRegCheck);
- found = true;
- break;
- }
- }
- if (!found)
- {
- // check for byte regs for eax through edx (e.g. al, ah, bl, bh....)
- if ((nameLen == 2) && (name[0] >= 'a') && (name[0] <= 'd') && ((name[1] == 'l') || (name[1] == 'h')))
- {
- if (isClobber)
- regClobberFlags |= (1 << (name[0] - 'a'));// $BYTEREGS this is why we want alphabetical order
- found = true;
- }
- }
- }
- else if ((nameLen == 4) && (name[0] == 'x') && (name[1] == 'm') && (name[2] == 'm') && (name[3] >= '0') && (name[3] <= '7'))
- {
- // xmm regs
- if (isClobber)
- regClobberFlags |= (REGCLOBBERF_XMM0 << (name[3] - '0'));
- found = true;
- }
- return found;
- };
- int asmInstCount = (int)asmStmt->mInstructions.size();
- typedef std::map<String, int> StrToVarIndexMap;
- StrToVarIndexMap strToVarIndexMap;
- if (mCompiler->IsAutocomplete())
- {
- // auto-complete "fast-pass" just to eliminate unused/unassigned variable yellow warning flashes
- for (int iInst=0; iInst<asmInstCount; ++iInst)
- {
- auto instNode = asmStmt->mInstructions[iInst];
- BfInlineAsmInstruction::AsmInst& asmInst = instNode->mAsmInst;
- bool hasLabel = !asmInst.mLabel.empty();
- bool hasOpCode = !asmInst.mOpCode.empty();
- if (hasLabel || hasOpCode) // check against blank lines
- {
- if (hasOpCode) // check against label-only lines (which still get written out, but don't do any other processing)
- {
- int argCount = (int)asmInst.mArgs.size();
- for (int iArg=0; iArg<argCount; ++iArg)
- {
- BfInlineAsmInstruction::AsmArg* arg = &asmInst.mArgs[iArg];
- if (arg->mType == BfInlineAsmInstruction::AsmArg::ARGTYPE_IntReg)
- {
- bool found = matchRegFunc(arg->mReg, false);
- if (!found)
- {
- StrToVarIndexMap::iterator it = strToVarIndexMap.find(arg->mReg);
- if (it == strToVarIndexMap.end())
- {
- for (int i = 0; i < (int) mCurMethodState->mLocals.size(); i++)
- {
- auto& checkLocal = mCurMethodState->mLocals[i];
- if (checkLocal.mName == arg->mReg)
- {
- // if you access a variable in asm, we suppress any warnings related to used or assigned, regardless of usage
- checkLocal.mIsReadFrom = true;
- checkLocal.mIsAssigned = true;
- found = true;
- break;
- }
- }
- }
- }
- }
- }
- }
- }
- }
- return;
- }
- int debugLocOffset = 0;
- if (!asmStmt->mInstructions.empty())
- debugLocOffset = asmStmt->mInstructions.front()->GetSrcStart() - asmStmt->GetSrcStart();
- UpdateSrcPos(asmStmt, true, debugLocOffset);
- mCurMethodState->mInHeadScope = false;
- BfScopeData prevScope = mCurMethodState->mCurScope;
- mCurMethodState->mCurScope->mPrevScope = &prevScope;
- NewScopeState();
- bool failed = false;
- if (!mCpu)
- mCpu = new Beefy::X86Cpu();
- //const char* srcAsmText = "nop\nnop\n\n\nmov eax, i\ninc eax\nmov i, eax\n_emit 0x0F\n_emit 0xC7\n_emit 0xF0\n\n\nnop\nnop"; //CDH TODO extract from actual lexical text block
- //const char* srcAsmText = "nop\nnop\n\n\nmov eax, i\ninc eax\nmov i, eax\nrdrand eax\n\n\nnop\nnop"; //CDH TODO extract from actual lexical text block
- //String srcAsmTextStr(&asmStmt->mParser->mSrc[asmStmt->mSrcStart], asmStmt->mSrcEnd - asmStmt->mSrcStart);
- //const char* srcAsmText = srcAsmTextStr.c_str();
- String dstAsmText;
- String constraintStr, clobberStr;
- int constraintCount = 0;
- bool hasMemoryClobber = false;
- Array<Type*> paramTypes;
- SizedArray<Value*, 1> llvmArgs;
- int lastDebugLine = -1;
- int curDebugLineDeltaValue = 0, curDebugLineDeltaRunCount = 0;
- String debugLineSequenceStr;
- int isFirstDebugLine = 1;
- auto maybeEmitDebugLineRun = [&curDebugLineDeltaValue, &curDebugLineDeltaRunCount, &debugLineSequenceStr, &isFirstDebugLine]()
- {
- if (curDebugLineDeltaRunCount > 0)
- {
- for (int i=isFirstDebugLine; i<2; ++i)
- {
- int value = i ? curDebugLineDeltaValue : curDebugLineDeltaRunCount;
- String encodedValue;
- EncodeULEB32(value, encodedValue);
- if (encodedValue.length() > 1)
- debugLineSequenceStr += String("$") + encodedValue + "$";
- else
- debugLineSequenceStr += encodedValue;
- }
- curDebugLineDeltaRunCount = 0;
- isFirstDebugLine = 0;
- }
- };
- auto mangledLabelName = [&asmStmt](const StringImpl& labelName, int numericLabel) -> String
- {
- return StrFormat("%d", numericLabel);
- //return String(".") + labelName;
- //return StrFormat("%s_%p_%d", labelName.c_str(), asmStmt->mSource, asmStmt->mSrcStart); // suffix label name with location information to make it block-specific (since labels get external linkage)
- };
- typedef std::pair<String, int> LabelPair;
- std::unordered_map<String, LabelPair> labelNames;
- // pre-scan instructions for label names
- for (int iInst=0; iInst<asmInstCount; ++iInst)
- {
- auto instNode = asmStmt->mInstructions[iInst];
- BfInlineAsmInstruction::AsmInst& asmInst = instNode->mAsmInst;
- if (!asmInst.mLabel.empty())
- {
- if (labelNames.find(asmInst.mLabel) != labelNames.end())
- {
- Fail(StrFormat("Label \"%s\" already defined in asm block", asmInst.mLabel.c_str()), instNode, true);
- failed = true;
- }
- else
- {
- String mangledLabel(mangledLabelName(asmInst.mLabel, instNode->GetSrcStart()));
- labelNames[asmInst.mLabel] = LabelPair(mangledLabel, instNode->GetSrcStart());
- asmInst.mLabel = mangledLabel;
- }
- }
- }
- for (int iInst=0; iInst<asmInstCount; ++iInst)
- {
- auto instNode = asmStmt->mInstructions[iInst];
- BfInlineAsmInstruction::AsmInst& asmInst = instNode->mAsmInst;
- bool hasLabel = !asmInst.mLabel.empty();
- bool hasOpCode = !asmInst.mOpCode.empty();
- if (hasLabel || hasOpCode) // check against blank lines
- {
- if (hasOpCode) // check against label-only lines (which still get written out, but don't do any other processing)
- {
- int argCount = (int)asmInst.mArgs.size();
- // reasonable defaults for clobber info
- int clobberCount = 1; // destination is usually first arg in Intel syntax
- bool mayClobberMem = true; // we only care about this when it gets turned to false by GetClobbersForMnemonic (no operand form of the instruction clobbers mem)
- // pseudo-ops
- if (asmInst.mOpCode == "_emit")
- {
- asmInst.mOpCode = ".byte";
- }
- else
- {
- Array<int> opcodes;
- if (!mCpu->GetOpcodesForMnemonic(asmInst.mOpCode, opcodes))
- {
- Fail(StrFormat("Unrecognized instruction mnemonic \"%s\"", asmInst.mOpCode.c_str()), instNode, true);
- failed = true;
- }
- else
- {
- Array<int> implicitClobbers;
- mCpu->GetClobbersForMnemonic(asmInst.mOpCode, argCount, implicitClobbers, clobberCount, mayClobberMem);
- for (int iClobberReg : implicitClobbers)
- {
- String regName = CPURegisters::GetRegisterName(iClobberReg);
- std::transform(regName.begin(), regName.end(), regName.begin(), ::tolower);
- matchRegFunc(regName, true);
- }
- }
- }
- String fakeLabel; // used when running label-using instructions through LLVM semantic pre-check
- for (int iArg=0; iArg<argCount; ++iArg)
- {
- BfInlineAsmInstruction::AsmArg* arg = &asmInst.mArgs[iArg];
- if (arg->mType == BfInlineAsmInstruction::AsmArg::ARGTYPE_IntReg)
- {
- bool isClobber = (iArg < clobberCount);
- bool found = matchRegFunc(arg->mReg, isClobber);
- if (!found)
- {
- StrToVarIndexMap::iterator it = strToVarIndexMap.find(arg->mReg);
- if (it != strToVarIndexMap.end())
- {
- arg->mReg = StrFormat("$%d", it->second);
- if (isClobber)
- mayClobberMem = true;
- found = true;
- }
- else
- {
- for (int i = 0; i < (int) mCurMethodState->mLocals.size(); i++)
- {
- auto& checkLocal = mCurMethodState->mLocals[i];
- if (checkLocal.mName == arg->mReg)
- {
- BfIRValue testValue = checkLocal.mAddr;
- llvmArgs.push_back(testValue);
- paramTypes.push_back(testValue->getType());
- arg->mReg = StrFormat("$%d", constraintCount);//CDH TODO does this need size qualifiers for "dword ptr $0" or whatever?
- strToVarIndexMap[checkLocal.mName] = constraintCount;
- constraintStr += "=*m,";
- ++constraintCount;
- if (isClobber)
- mayClobberMem = true;
- // if you access a variable in asm, we suppress any warnings related to used or assigned, regardless of usage
- checkLocal.mIsReadFrom = true;
- checkLocal.mIsAssigned = true;
- found = true;
- break;
- }
- }
- }
- }
- if (!found)
- {
- auto labelIt = labelNames.find(arg->mReg);
- if (labelIt != labelNames.end())
- {
- arg->mReg = labelIt->second.first;
- if (labelIt->second.second <= instNode->GetSrcStart())
- {
- fakeLabel = arg->mReg;
- arg->mReg += "b";
- }
- else
- arg->mReg += "f";
- }
- else
- {
- Fail(StrFormat("Unrecognized variable \"%s\"", arg->mReg.c_str()), instNode, true);
- failed = true;
- }
- }
- }
- else if (arg->mType == BfInlineAsmInstruction::AsmArg::ARGTYPE_FloatReg)
- {
- //CDH individual reg clobber is probably insufficient for fp regs since it's stack-based; without deeper knowledge of how individual instructions
- // manipulate the FP stack, the safest approach is to clobber all FP regs as soon as one of them is involved
- //bool isClobber = (iArg == 0); // destination is first arg in Intel syntax
- //bool found = matchRegFunc(StrFormat("st%d", arg->mInt), isClobber);
- //BF_ASSERT(found);
- for (int iRegCheck=0; iRegCheck<8; ++iRegCheck)
- regClobberFlags |= (REGCLOBBERF_FPST0 << iRegCheck);
- }
- else if (arg->mType == BfInlineAsmInstruction::AsmArg::ARGTYPE_Memory)
- {
- bool isClobber = (iArg < clobberCount);
- // check regs for clobber flags
- /*
- //CDH TODO do we need to set clobber flags for regs that are used *indirectly* like this? Actually I don't think so; commenting out for now
- if (!arg->mReg.empty())
- matchRegFunc(arg->mReg, isClobber);
- if (!arg->mAdjReg.empty())
- matchRegFunc(arg->mAdjReg, isClobber);
- */
- if (isClobber)
- mayClobberMem = true;
- if (!arg->mMemberSuffix.empty())
- {
- //CDH TODO add member support once I know the right way to look up struct member offsets. Once we know the offset,
- // add it to arg->mInt, and set ARGMEMF_ImmediateDisp if it's not set already (member support is just used as an offset)
- Fail("Member suffix syntax is not yet supported", instNode, true);
- failed = true;
- }
- }
- }
- if (mayClobberMem)
- hasMemoryClobber = true;
- //debugLineSequenceStr += StrFormat("%d_", asmInst.mDebugLine);
- int curDebugLine = asmInst.mDebugLine;
- int debugLineDelta = (lastDebugLine > 0) ? curDebugLine - lastDebugLine : curDebugLine;
- lastDebugLine = curDebugLine;
- //String encodedDebugLineDelta;
- //EncodeULEB32(debugLineDelta, encodedDebugLineDelta);
- //debugLineSequenceStr += encodedDebugLineDelta + "_";
- if (curDebugLineDeltaValue != debugLineDelta)
- {
- maybeEmitDebugLineRun();
- curDebugLineDeltaValue = debugLineDelta;
- }
- ++curDebugLineDeltaRunCount;
- // run instruction through LLVM for better semantic errors (can be slow in debug; feel free to comment out this scopeData if it's intolerable; the errors will still be caught at compile time)
- //if (false)
- {
- BfInlineAsmInstruction::AsmInst tempAsmInst(asmInst);
- tempAsmInst.mLabel = fakeLabel;
- for (auto & arg : tempAsmInst.mArgs)
- {
- if ((arg.mType == BfInlineAsmInstruction::AsmArg::ARGTYPE_IntReg) && !arg.mReg.empty() && (arg.mReg[0] == '$'))
- {
- // if we've rewritten a local variable instruction arg to use a $-prefixed input, we can't pass that to LLVM
- // at this stage as it won't recognize it; the actual compilation would have changed all these to use actual
- // memory operand syntax first. However, those changes all work down in LLVM to printIntelMemReference inside
- // of X86AsmPrinter.cpp, and that always results in a [bracketed] memory access string no matter what, which
- // means for our semantic checking purposes here it's sufficient to just use "[eax]" for all such cases,
- // rather than go through an even more expensive setup & teardown process to use the AsmPrinter itself.
- arg.mType = BfInlineAsmInstruction::AsmArg::ARGTYPE_Memory;
- arg.mMemFlags = BfInlineAsmInstruction::AsmArg::ARGMEMF_BaseReg;
- arg.mReg = "eax";
- }
- }
- String llvmError;
- if (!mCpu->ParseInlineAsmInstructionLLVM(tempAsmInst.ToString(), llvmError))
- {
- Fail(StrFormat("Inline asm error: %s", llvmError.c_str()), instNode, true);
- failed = true;
- }
- }
- }
- dstAsmText += asmInst.ToString();
- dstAsmText += "\n";
- }
- }
- maybeEmitDebugLineRun(); // leftovers
- if (failed)
- {
- RestoreScopeState(&prevScope);
- return;
- }
- // prepare constraints/clobbers
- {
- for (int iRegCheck = 0; regClobberNames[iRegCheck] != nullptr; ++iRegCheck)
- {
- if (regClobberFlags & (1 << iRegCheck))
- clobberStr += StrFormat("~{%s},", regClobberNames[iRegCheck]);
- }
- for (int iRegCheck=0; iRegCheck<8; ++iRegCheck)
- {
- if (regClobberFlags & (REGCLOBBERF_XMM0 << iRegCheck))
- clobberStr += StrFormat("~{xmm%d},", iRegCheck);
- if (regClobberFlags & (REGCLOBBERF_FPST0 << iRegCheck))
- clobberStr += StrFormat("~{fp%d},~{st(%d)},", iRegCheck, iRegCheck); // both fp# and st(#) are listed in X86RegisterInfo.td
- if (regClobberFlags & (REGCLOBBERF_MM0 << iRegCheck))
- clobberStr += StrFormat("~{mm%d},", iRegCheck);
- }
- // add wrapping instructions to preserve certain regs (e.g. ESI), due to LLVM bypassing register allocator when choosing a base register
- //CDH TODO currently I'm only shielding against ESI stompage; people generally know not to mess with ESP & EBP, but ESI is still a "general" reg and should be allowed to be clobbered
- //if (regClobberFlags & REGCLOBBERF_ESI)
- //{
- //CDH TODO Bah! This doesn't actually work, because if you do any local variable access after mutating ESI, the variable substitution could have generated a base address dependency
- // on ESI which will not expect it to have changed, e.g. "mov esi, var\ninc esi\nmov var, esi\n" dies if "var" gets internally rewritten to "[esi + displacement]". What to do? Hmm.
- //dstAsmText = String("push esi\n") + dstAsmText + String("pop esi\n");
- //}
- if (hasMemoryClobber)
- clobberStr += "~{memory},";
- clobberStr += "~{dirflag},~{fpsr},~{flags}";
- constraintStr += clobberStr;
- }
- bool wantsDIData = (mBfIRBuilder->DbgHasInfo()) && (!mCurMethodInstance->mIsUnspecialized) && (mHasFullDebugInfo);
- if (wantsDIData)
- {
- static int sVarNum = 0;
- String varName(StrFormat("__asmLines_%d.%s", ++sVarNum, debugLineSequenceStr.c_str()));
- auto varType = GetPrimitiveType(BfTypeCode_Int32);
- auto allocaInst = mBfIRBuilder->CreateAlloca(varType->mLLVMType, 0, varName + ".addr");
- allocaInst->setAlignment(varType->mAlign);
- //paramVar->mAddr = allocaInst;
- auto varValue = GetConstValue(0, varType);
- auto diVariable = mDIBuilder->createAutoVariable(mCurMethodState->mCurScope->mDIScope,
- varName.c_str(), mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, varType->mDIType/*, true*/);
- //auto varValue = llvm::ConstantInt::getTrue(*mLLVMContext);
- //auto varValue = GetDefaultValue(varType);
- //auto varValue = CreateGlobalConstValue(varName, llvm::ConstantInt::getTrue(*mLLVMContext), true);
- //auto varValue = AllocGlobalVariable(*mIRModule, diType->getType(), false, GlobalValue::ExternalLinkage, llvm::ConstantInt::getTrue(*mLLVMContext), varName.c_str());
- //auto diVariable = mDIBuilder->createGlobalVariable(mCurMethodState->mCurScope->mDIScope, varName.c_str(), "", mCurFilePosition.mFileInstance->mDIFile, mCurFilePosition.mCurLine, diType, false, varValue);
- //BasicBlock* block = mBfIRBuilder->GetInsertBlock();
- //auto declareVar = mDIBuilder->insertDeclare(varValue, diVariable, mBfIRBuilder->GetInsertBlock());
- auto declareVar = mDIBuilder->insertDeclare(allocaInst, diVariable, mDIBuilder->createExpression(),
- mIRBuilder->getCurrentDebugLocation(), mBfIRBuilder->GetInsertBlock());
- //auto declareVar = mDIBuilder->insertDbgValueIntrinsic(varValue, 0, diVariable, mBfIRBuilder->GetInsertBlock());
- declareVar->setDebugLoc(mIRBuilder->getCurrentDebugLocation());
- }
- /*
- BfIRValue testValue = NULL;
- for (int i = 0; i < (int) mCurMethodState->mLocals.size(); i++)
- {
- auto& checkLocal = mCurMethodState->mLocals[i];
- if (checkLocal.mName == "i")
- {
- testValue = checkLocal.mAddr;
- break;
- }
- }
- */
- //BF_ASSERT((testValue != NULL) && "Need local variable \"i\"");
- //if (testValue != NULL)
- {
- //Type* voidPtrType = Type::getInt8PtrTy(*mLLVMContext);
- //if (mContext->mAsmObjectCheckFuncType == NULL)
- //{
- //Array<Type*> paramTypes;
- //paramTypes.push_back(voidPtrType);
- //mContext->mAsmObjectCheckFuncType = FunctionType::get(Type::getVoidTy(*mLLVMContext), paramTypes, false);
- //}
- FunctionType* funcType = FunctionType::get(Type::getVoidTy(*mLLVMContext), paramTypes, false);
- //SizedArray<Value*, 1> llvmArgs;
- //llvmArgs.push_back(testValue);
- //CDH REMOVE NOTE
- //generates IR (e.g.):
- // call void asm sideeffect "#4\0Anop\0Anop\0Amovl %eax, %eax\0Anop\0Anop", "~{cc},~{dirflag},~{fpsr},~{flags},~{eax}"() #0, !dbg !492
- static int asmIdx = 0;
- asmIdx++;
- String asmStr = StrFormat("#%d\n", asmIdx) +
- //"nop\nnop\nmovl $0, %eax\nincl %eax\nmovl %eax, $0\nmovl $$0, %ecx\nmovl $$0, %esp\nnop\nnop";
- //"nop\nnop\nmovl ($0), %eax\nincl %eax\nmovl %eax, ($0)\nmovl $$0, %ecx\nmovl $$0, %esp\nnop\nnop";
- //"nop\nnop\nmovl %eax, %eax\nmovl $$0, %ecx\nmovl $$0, %esp\nnop\nnop";
- //"nop\nnop\nmovl %eax, %eax\nnop\nnop";
- //"nop\nnop\n.byte 0x0F\n.byte 0xC7\n.byte 0xF0\nmov eax, 7\nmov ecx, 0\ncpuid\nmov eax, dword ptr $0\ninc eax\nmov dword ptr $0, eax\nmov ecx, 0\nmov esp, 0\nnop\nnop"; // rdrand test
- dstAsmText;
- llvm::InlineAsm* inlineAsm = llvm::InlineAsm::get(funcType,
- //asmStr.c_str(), "~{r},~{cc},~{dirflag},~{fpsr},~{flags},~{eax},~{memory},~{esi},~{esp}", true,
- //asmStr.c_str(), "~{cc},~{dirflag},~{fpsr},~{flags},~{memory},~{ecx},~{esp}", true,
- //DOES NOT WORK (mem not written back to from reg:
- //asmStr.c_str(), "+r,~{cc},~{dirflag},~{fpsr},~{flags},~{memory},~{eax},~{ecx},~{esp}", true,
- //asmStr.c_str(), "+rm,~{cc},~{dirflag},~{fpsr},~{flags},~{memory},~{eax},~{ecx},~{esp}", true,
- asmStr.c_str(), constraintStr.c_str(), true,
- false, /*llvm::InlineAsm::AD_ATT*/llvm::InlineAsm::AD_Intel);
- llvm::CallInst* callInst = mIRBuilder->CreateCall(inlineAsm, llvmArgs);
- //llvm::CallInst* callInst = mIRBuilder->CreateCall(inlineAsm);
- callInst->addAttribute(llvm::AttributeSet::FunctionIndex, llvm::Attribute::NoUnwind);
- }
- RestoreScopeState(&prevScope);
- #endif
- }
|