jk.c 162 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359
  1. #include "version.h"
  2. #include <ctype.h>
  3. #include <dlfcn.h>
  4. #include <errno.h>
  5. #include <ffi.h>
  6. #include <gc.h>
  7. #include <limits.h>
  8. #include <math.h>
  9. #include <pcre.h>
  10. #include <setjmp.h>
  11. #include <stdbool.h>
  12. #include <stdint.h>
  13. #include <stdio.h>
  14. #include <stdlib.h>
  15. #include <string.h>
  16. #include <sys/wait.h>
  17. #include <time.h>
  18. #include <unistd.h>
  19. jmp_buf interactive_checkpoint;
  20. bool is_interactive;
  21. void *malloc_checked(size_t size) {
  22. void *p;
  23. if (!(p = GC_MALLOC(size)))
  24. abort();
  25. memset(p, 0, size);
  26. return p;
  27. }
  28. void *malloc_checked_atomic(size_t size) {
  29. void *p;
  30. if (!(p = GC_malloc_atomic(size)))
  31. abort();
  32. memset(p, 0, size);
  33. return p;
  34. }
  35. void *malloc_checked_uncollectable(size_t size) {
  36. void *p;
  37. if (!(p = GC_malloc_uncollectable(size)))
  38. abort();
  39. memset(p, 0, size);
  40. return p;
  41. }
  42. void *realloc_checked(void *p, size_t size) {
  43. if (!(p = GC_REALLOC(p, size)))
  44. abort();
  45. return p;
  46. }
  47. char *strdup_checked(char *s) {
  48. char *p = GC_strdup(s);
  49. if (!p)
  50. abort();
  51. return p;
  52. }
  53. typedef struct {
  54. void **data;
  55. size_t length;
  56. } list_t;
  57. list_t *list_new(void) {
  58. list_t *list = malloc_checked(sizeof(list_t));
  59. list->data = NULL;
  60. list->length = 0;
  61. return list;
  62. }
  63. list_t *list_newk(size_t k) {
  64. list_t *list = malloc_checked(sizeof(list_t));
  65. list->data = malloc_checked(k * sizeof(void *));
  66. list->length = k;
  67. return list;
  68. }
  69. list_t *list_copy(list_t *l) {
  70. list_t *list = list_newk(l->length);
  71. for (size_t i = 0; i < l->length; i++)
  72. list->data[i] = l->data[i];
  73. return list;
  74. }
  75. void list_push(list_t *l, void *v) {
  76. size_t i = l->length++;
  77. l->data = realloc_checked(l->data, l->length * sizeof(void *));
  78. l->data[i] = v;
  79. }
  80. void *list_pop(list_t *l) {
  81. if (!l->data)
  82. return NULL;
  83. size_t i = --l->length;
  84. void *v = l->data[i];
  85. l->data[i] = NULL;
  86. if (!l->length) {
  87. GC_FREE(l->data);
  88. l->data = NULL;
  89. } else
  90. l->data = realloc_checked(l->data, l->length * sizeof(void *));
  91. return v;
  92. }
  93. void *list_index(list_t *l, ssize_t index) {
  94. if (!l->data)
  95. return NULL;
  96. if (index < 0)
  97. index += ((ssize_t)l->length);
  98. if (index < 0 || index >= l->length)
  99. return NULL;
  100. return l->data[index];
  101. }
  102. void list_set(list_t *l, ssize_t index, void *v) {
  103. if (!l->data)
  104. return;
  105. if (index < 0)
  106. index += ((ssize_t)l->length);
  107. if (index < 0 || index >= l->length)
  108. return;
  109. l->data[index] = v;
  110. }
  111. typedef struct {
  112. char *str;
  113. size_t size;
  114. } buffer_t;
  115. buffer_t *buffer_new(void) {
  116. buffer_t *buf = malloc_checked(sizeof(buffer_t));
  117. buf->str = NULL;
  118. buf->size = 0;
  119. return buf;
  120. }
  121. void buffer_append(buffer_t *buf, char c) {
  122. buf->size++;
  123. void *p = malloc_checked_atomic(sizeof(char) * buf->size);
  124. if (buf->str) {
  125. memcpy(p, buf->str, buf->size - 1);
  126. GC_FREE(buf->str);
  127. }
  128. buf->str = p;
  129. buf->str[buf->size - 1] = c;
  130. }
  131. char *buffer_read(buffer_t *buf) {
  132. if (buf->size == 0 || buf->str[buf->size - 1])
  133. buffer_append(buf, 0);
  134. char *str = buf->str;
  135. GC_FREE(buf);
  136. return str;
  137. }
  138. void buffer_append_str(buffer_t *buf, char *s) {
  139. for (size_t i = 0; i < strlen(s); i++)
  140. buffer_append(buf, s[i]);
  141. }
  142. typedef struct {
  143. enum token_tag_t {
  144. T_PUNCT,
  145. T_LPAR,
  146. T_RPAR,
  147. T_NAME,
  148. T_NUMBER,
  149. T_BNUMBER,
  150. T_QUOTE
  151. } tag;
  152. char *text;
  153. } token_t;
  154. typedef struct {
  155. char *source;
  156. size_t len;
  157. size_t pos;
  158. list_t *tokens;
  159. } lexer_t;
  160. lexer_t *lexer_new(void) {
  161. lexer_t *lexer = malloc_checked(sizeof(lexer_t));
  162. return lexer;
  163. }
  164. char lexer_lookahead(lexer_t *lexer, size_t offset) {
  165. size_t pos = lexer->pos + offset;
  166. if (pos >= lexer->len)
  167. return 0;
  168. return lexer->source[pos];
  169. }
  170. char lexer_eat(lexer_t *lexer) {
  171. if (lexer->pos >= lexer->len)
  172. return 0;
  173. return lexer->source[lexer->pos++];
  174. }
  175. void lexer_push_token(lexer_t *lexer, enum token_tag_t tag, char *text) {
  176. token_t *token = malloc_checked(sizeof(token_t));
  177. token->tag = tag;
  178. token->text = text;
  179. list_push(lexer->tokens, token);
  180. }
  181. list_t *guards;
  182. jmp_buf *guard() {
  183. jmp_buf *lb = malloc_checked_atomic(sizeof(jmp_buf));
  184. list_push(guards, lb);
  185. return lb;
  186. }
  187. jmp_buf *guarding() { return list_index(guards, -1); }
  188. void unguard() {
  189. jmp_buf *lb = list_pop(guards);
  190. GC_FREE(lb);
  191. }
  192. void fatal(char *s) {
  193. jmp_buf *lb;
  194. if ((lb = guarding()))
  195. longjmp(*lb, 1);
  196. fprintf(stderr, "|%s error\n", s);
  197. if (is_interactive)
  198. longjmp(interactive_checkpoint, 1);
  199. exit(1);
  200. }
  201. void lexer_error(lexer_t *lexer, char *s) { fatal(s); }
  202. void lexer_lex_number(lexer_t *lexer, bool is_negative) {
  203. buffer_t *buf = buffer_new();
  204. if (is_negative)
  205. buffer_append(buf, '-');
  206. if (lexer_lookahead(lexer, 0) == '.') {
  207. buffer_append(buf, lexer_eat(lexer));
  208. if (!(isdigit(lexer_lookahead(lexer, 0))))
  209. lexer_error(lexer, "trailing-dot");
  210. }
  211. do {
  212. buffer_append(buf, lexer_eat(lexer));
  213. if (lexer_lookahead(lexer, 0) == '`' && isdigit(lexer_lookahead(lexer, 1)))
  214. lexer_eat(lexer);
  215. } while (isdigit(lexer_lookahead(lexer, 0)));
  216. if (lexer_lookahead(lexer, 0) == '.') {
  217. buffer_append(buf, lexer_eat(lexer));
  218. if (!(isdigit(lexer_lookahead(lexer, 0))))
  219. lexer_error(lexer, "trailing-dot");
  220. do {
  221. buffer_append(buf, lexer_eat(lexer));
  222. } while (isdigit(lexer_lookahead(lexer, 0)));
  223. }
  224. lexer_push_token(lexer, T_NUMBER, buffer_read(buf));
  225. }
  226. void lexer_lex(lexer_t *lexer, char *s) {
  227. lexer->source = s;
  228. lexer->len = strlen(s);
  229. lexer->pos = 0;
  230. lexer->tokens = list_new();
  231. while (lexer->pos < lexer->len) {
  232. char c = lexer_lookahead(lexer, 0);
  233. if (c == '/' && !lexer->tokens->data)
  234. break;
  235. if (isspace(c)) {
  236. lexer_eat(lexer);
  237. if (lexer_lookahead(lexer, 0) == '/')
  238. break;
  239. } else if (c == '0' && (lexer_lookahead(lexer, 1) == 'x' ||
  240. lexer_lookahead(lexer, 1) == 'b' ||
  241. lexer_lookahead(lexer, 1) == 'o')) {
  242. lexer_eat(lexer);
  243. buffer_t *buf = buffer_new();
  244. char b = lexer_eat(lexer);
  245. buffer_append(buf, b);
  246. const char *base = b == 'x' ? "0123456789abcdefABCDEF"
  247. : b == 'b' ? "01"
  248. : "01234567";
  249. while (strchr(base, lexer_lookahead(lexer, 0)) != NULL)
  250. buffer_append(buf, lexer_eat(lexer));
  251. lexer_push_token(lexer, T_BNUMBER, buffer_read(buf));
  252. } else if (isdigit(c) || c == '.') {
  253. lexer_lex_number(lexer, false);
  254. } else if (isalpha(c)) {
  255. buffer_t *buf = buffer_new();
  256. do {
  257. buffer_append(buf, lexer_eat(lexer));
  258. } while (isalpha(lexer_lookahead(lexer, 0)));
  259. if (buf->size == 1 && lexer_lookahead(lexer, 0) == '.') {
  260. buffer_append(buf, lexer_eat(lexer));
  261. lexer_push_token(lexer, T_PUNCT, buffer_read(buf));
  262. } else
  263. lexer_push_token(lexer, T_NAME, buffer_read(buf));
  264. } else if (c == '(' || c == ')') {
  265. lexer_eat(lexer);
  266. lexer_push_token(lexer, c == '(' ? T_LPAR : T_RPAR, NULL);
  267. } else if (c == '\'') {
  268. buffer_t *buf = buffer_new();
  269. lexer_eat(lexer);
  270. for (;;) {
  271. if (lexer->pos >= lexer->len)
  272. lexer_error(lexer, "unmatched-quote");
  273. if (lexer_lookahead(lexer, 0) == '\'') {
  274. if (lexer_lookahead(lexer, 1) == '\'') {
  275. buffer_append(buf, lexer_eat(lexer));
  276. lexer_eat(lexer);
  277. continue;
  278. }
  279. lexer_eat(lexer);
  280. break;
  281. }
  282. buffer_append(buf, lexer_eat(lexer));
  283. }
  284. lexer_push_token(lexer, T_QUOTE, buffer_read(buf));
  285. } else if (ispunct(c)) {
  286. char buf[3];
  287. buf[0] = lexer_eat(lexer);
  288. buf[1] = 0;
  289. if (lexer_lookahead(lexer, 0) == '.' ||
  290. lexer_lookahead(lexer, 0) == ':') {
  291. buf[1] = lexer_eat(lexer);
  292. buf[2] = 0;
  293. }
  294. if (strcmp(buf, "-") == 0 && isdigit(lexer_lookahead(lexer, 0))) {
  295. lexer_lex_number(lexer, true);
  296. continue;
  297. }
  298. lexer_push_token(lexer, T_PUNCT, strdup_checked(buf));
  299. } else
  300. lexer_error(lexer, "lex");
  301. }
  302. }
  303. typedef struct _table_t table_t;
  304. typedef struct _table_entry_t table_entry_t;
  305. struct _table_entry_t {
  306. char *key;
  307. void *value;
  308. bool is_deleted;
  309. };
  310. struct _table_t {
  311. table_entry_t *entries;
  312. size_t used;
  313. size_t capacity;
  314. };
  315. #define TABLE_MIN_SIZE 32
  316. table_t *table_new(void) {
  317. table_t *table = malloc_checked(sizeof(table_t));
  318. table->used = 0;
  319. table->capacity = TABLE_MIN_SIZE;
  320. table->entries = malloc_checked(table->capacity * sizeof(table_entry_t));
  321. return table;
  322. }
  323. size_t table_length(table_t *table) { return table->used; }
  324. bool table_empty(table_t *table) { return table->used == 0; }
  325. static uint64_t MM86128(void *key, const int len, uint32_t seed) {
  326. #define ROTL32(x, r) ((x << r) | (x >> (32 - r)))
  327. #define FMIX32(h) \
  328. h ^= h >> 16; \
  329. h *= 0x85ebca6b; \
  330. h ^= h >> 13; \
  331. h *= 0xc2b2ae35; \
  332. h ^= h >> 16;
  333. const uint8_t *data = (const uint8_t *)key;
  334. const int nblocks = len / 16;
  335. uint32_t h1 = seed;
  336. uint32_t h2 = seed;
  337. uint32_t h3 = seed;
  338. uint32_t h4 = seed;
  339. uint32_t c1 = 0x239b961b;
  340. uint32_t c2 = 0xab0e9789;
  341. uint32_t c3 = 0x38b34ae5;
  342. uint32_t c4 = 0xa1e38b93;
  343. const uint32_t *blocks = (const uint32_t *)(data + nblocks * 16);
  344. for (int i = -nblocks; i; i++) {
  345. uint32_t k1 = blocks[i * 4 + 0];
  346. uint32_t k2 = blocks[i * 4 + 1];
  347. uint32_t k3 = blocks[i * 4 + 2];
  348. uint32_t k4 = blocks[i * 4 + 3];
  349. k1 *= c1;
  350. k1 = ROTL32(k1, 15);
  351. k1 *= c2;
  352. h1 ^= k1;
  353. h1 = ROTL32(h1, 19);
  354. h1 += h2;
  355. h1 = h1 * 5 + 0x561ccd1b;
  356. k2 *= c2;
  357. k2 = ROTL32(k2, 16);
  358. k2 *= c3;
  359. h2 ^= k2;
  360. h2 = ROTL32(h2, 17);
  361. h2 += h3;
  362. h2 = h2 * 5 + 0x0bcaa747;
  363. k3 *= c3;
  364. k3 = ROTL32(k3, 17);
  365. k3 *= c4;
  366. h3 ^= k3;
  367. h3 = ROTL32(h3, 15);
  368. h3 += h4;
  369. h3 = h3 * 5 + 0x96cd1c35;
  370. k4 *= c4;
  371. k4 = ROTL32(k4, 18);
  372. k4 *= c1;
  373. h4 ^= k4;
  374. h4 = ROTL32(h4, 13);
  375. h4 += h1;
  376. h4 = h4 * 5 + 0x32ac3b17;
  377. }
  378. const uint8_t *tail = (const uint8_t *)(data + nblocks * 16);
  379. uint32_t k1 = 0;
  380. uint32_t k2 = 0;
  381. uint32_t k3 = 0;
  382. uint32_t k4 = 0;
  383. switch (len & 15) {
  384. case 15:
  385. k4 ^= tail[14] << 16;
  386. case 14:
  387. k4 ^= tail[13] << 8;
  388. case 13:
  389. k4 ^= tail[12] << 0;
  390. k4 *= c4;
  391. k4 = ROTL32(k4, 18);
  392. k4 *= c1;
  393. h4 ^= k4;
  394. case 12:
  395. k3 ^= tail[11] << 24;
  396. case 11:
  397. k3 ^= tail[10] << 16;
  398. case 10:
  399. k3 ^= tail[9] << 8;
  400. case 9:
  401. k3 ^= tail[8] << 0;
  402. k3 *= c3;
  403. k3 = ROTL32(k3, 17);
  404. k3 *= c4;
  405. h3 ^= k3;
  406. case 8:
  407. k2 ^= tail[7] << 24;
  408. case 7:
  409. k2 ^= tail[6] << 16;
  410. case 6:
  411. k2 ^= tail[5] << 8;
  412. case 5:
  413. k2 ^= tail[4] << 0;
  414. k2 *= c2;
  415. k2 = ROTL32(k2, 16);
  416. k2 *= c3;
  417. h2 ^= k2;
  418. case 4:
  419. k1 ^= tail[3] << 24;
  420. case 3:
  421. k1 ^= tail[2] << 16;
  422. case 2:
  423. k1 ^= tail[1] << 8;
  424. case 1:
  425. k1 ^= tail[0] << 0;
  426. k1 *= c1;
  427. k1 = ROTL32(k1, 15);
  428. k1 *= c2;
  429. h1 ^= k1;
  430. }
  431. h1 ^= len;
  432. h2 ^= len;
  433. h3 ^= len;
  434. h4 ^= len;
  435. h1 += h2;
  436. h1 += h3;
  437. h1 += h4;
  438. h2 += h1;
  439. h3 += h1;
  440. h4 += h1;
  441. FMIX32(h1);
  442. FMIX32(h2);
  443. FMIX32(h3);
  444. FMIX32(h4);
  445. h1 += h2;
  446. h1 += h3;
  447. h1 += h4;
  448. h2 += h1;
  449. h3 += h1;
  450. h4 += h1;
  451. return (((uint64_t)h2) << 32) | h1;
  452. }
  453. static uint32_t HASH_SEED = 0;
  454. void *table_get(table_t *table, char *key) {
  455. if (table_empty(table))
  456. return NULL;
  457. uint64_t hash = MM86128(key, strlen(key), HASH_SEED);
  458. size_t index = hash % table->capacity;
  459. size_t i = index;
  460. while (table->entries[i].key) {
  461. if (!table->entries[i].is_deleted &&
  462. strcmp(table->entries[i].key, key) == 0)
  463. return table->entries[i].value;
  464. i++;
  465. if (i >= table->capacity)
  466. i = 0;
  467. if (i == index)
  468. break;
  469. }
  470. return NULL;
  471. }
  472. bool table_has(table_t *table, char *key) {
  473. if (table_empty(table))
  474. return false;
  475. uint64_t hash = MM86128(key, strlen(key), HASH_SEED);
  476. size_t index = hash % table->capacity;
  477. size_t i = index;
  478. while (table->entries[i].key) {
  479. if (!table->entries[i].is_deleted &&
  480. strcmp(table->entries[i].key, key) == 0)
  481. return true;
  482. i++;
  483. if (i >= table->capacity)
  484. i = 0;
  485. if (i == index)
  486. break;
  487. }
  488. return false;
  489. }
  490. static void table_entry_set(table_entry_t *entries, char *key, void *value,
  491. size_t capacity, size_t *used) {
  492. uint64_t hash = MM86128(key, strlen(key), HASH_SEED);
  493. size_t index = hash % capacity;
  494. size_t i = index;
  495. while (entries[i].key) {
  496. if (strcmp(entries[i].key, key) == 0) {
  497. entries[i].value = value;
  498. if (entries[i].is_deleted) {
  499. if (used)
  500. (*used)++;
  501. entries[i].is_deleted = false;
  502. }
  503. return;
  504. } else if (entries[i].is_deleted)
  505. break;
  506. i++;
  507. if (i >= capacity)
  508. i = 0;
  509. if (i == index)
  510. break;
  511. }
  512. if (used)
  513. (*used)++;
  514. entries[i].key = key;
  515. entries[i].value = value;
  516. entries[i].is_deleted = false;
  517. }
  518. table_t *table_set(table_t *table, char *key, void *value) {
  519. if (table->used >= table->capacity) {
  520. size_t capacity = table->capacity + TABLE_MIN_SIZE;
  521. table_entry_t *entries = malloc_checked(capacity * sizeof(table_entry_t));
  522. for (size_t i = 0; i < table->capacity; i++) {
  523. table_entry_t entry = table->entries[i];
  524. if (entry.key && !entry.is_deleted)
  525. table_entry_set(entries, entry.key, entry.value, capacity, NULL);
  526. }
  527. GC_FREE(table->entries);
  528. table->entries = entries;
  529. table->capacity = capacity;
  530. }
  531. table_entry_set(table->entries, key, value, table->capacity, &table->used);
  532. return table;
  533. }
  534. bool table_delete(table_t *table, char *key) {
  535. uint64_t hash = MM86128(key, strlen(key), HASH_SEED);
  536. size_t index = hash % table->capacity;
  537. size_t i = index;
  538. while (table->entries[i].key) {
  539. if (!table->entries[i].is_deleted &&
  540. strcmp(table->entries[i].key, key) == 0) {
  541. table->entries[i].value = NULL;
  542. table->entries[i].is_deleted = true;
  543. table->used--;
  544. if (table->capacity > TABLE_MIN_SIZE &&
  545. table->used <= table->capacity - TABLE_MIN_SIZE) {
  546. size_t capacity = table->capacity - TABLE_MIN_SIZE;
  547. table_entry_t *entries =
  548. malloc_checked(capacity * sizeof(table_entry_t));
  549. for (size_t i = 0; i < table->capacity; i++) {
  550. table_entry_t entry = table->entries[i];
  551. if (entry.key && !entry.is_deleted)
  552. table_entry_set(entries, entry.key, entry.value, capacity, NULL);
  553. }
  554. GC_FREE(table->entries);
  555. table->entries = entries;
  556. table->capacity = capacity;
  557. }
  558. return true;
  559. }
  560. i++;
  561. if (i >= table->capacity)
  562. i = 0;
  563. if (i == index)
  564. break;
  565. }
  566. return false;
  567. }
  568. typedef struct _value_t value_t;
  569. typedef struct _interpreter_t interpreter_t;
  570. typedef struct _verb_t verb_t;
  571. struct _interpreter_t {
  572. table_t *env;
  573. list_t *args;
  574. list_t *selfrefs;
  575. value_t *nil;
  576. value_t *udf;
  577. value_t *unit;
  578. verb_t *at;
  579. bool bn;
  580. };
  581. struct _verb_t {
  582. char *name;
  583. unsigned int rank[3];
  584. list_t *bonds;
  585. bool mark;
  586. bool is_fun;
  587. value_t *(*monad)(interpreter_t *, verb_t *, value_t *);
  588. value_t *(*dyad)(interpreter_t *, verb_t *, value_t *, value_t *);
  589. };
  590. typedef struct {
  591. char *name;
  592. verb_t *(*adverb)(interpreter_t *, value_t *);
  593. verb_t *(*conjunction)(interpreter_t *, value_t *, value_t *);
  594. } adverb_t;
  595. struct _value_t {
  596. enum value_tag_t { ARRAY, VERB, SYMBOL, NUMBER, CHAR, NIL, UDF } tag;
  597. union {
  598. list_t *array;
  599. verb_t *verb;
  600. char *symbol;
  601. double number;
  602. unsigned char _char;
  603. } val;
  604. };
  605. verb_t *verb_new() {
  606. verb_t *verb = malloc_checked(sizeof(verb_t));
  607. return verb;
  608. }
  609. value_t *value_new(enum value_tag_t tag) {
  610. value_t *val;
  611. if (tag > SYMBOL)
  612. val = malloc_checked_atomic(sizeof(value_t));
  613. else
  614. val = malloc_checked(sizeof(value_t));
  615. val->tag = tag;
  616. return val;
  617. }
  618. value_t *value_new_const(enum value_tag_t tag) {
  619. value_t *val = malloc_checked_uncollectable(sizeof(value_t));
  620. val->tag = tag;
  621. return val;
  622. }
  623. value_t *_UNIT;
  624. value_t *value_new_array(list_t *array) {
  625. if (!array->data) {
  626. GC_FREE(array);
  627. return _UNIT;
  628. }
  629. value_t *val = value_new(ARRAY);
  630. val->val.array = array;
  631. return val;
  632. }
  633. table_t *VCACHE;
  634. value_t *value_new_verb(verb_t *verb) {
  635. value_t *val;
  636. if ((val = table_get(VCACHE, verb->name)))
  637. return val;
  638. val = value_new(VERB);
  639. val->val.verb = verb;
  640. return val;
  641. }
  642. table_t *SCACHE;
  643. value_t *value_new_symbol(char *symbol) {
  644. value_t *val;
  645. if ((val = table_get(SCACHE, symbol)))
  646. return val;
  647. val = value_new_const(SYMBOL);
  648. val->val.symbol = symbol;
  649. table_set(SCACHE, symbol, val);
  650. return val;
  651. }
  652. value_t *_NAN, *INF, *NINF;
  653. value_t *NNUMS[8];
  654. value_t *NUMS[256];
  655. value_t *CHARS[256];
  656. value_t *value_new_number(double number) {
  657. if (isnan(number))
  658. return _NAN;
  659. else if (number == INFINITY)
  660. return INF;
  661. else if (number == -INFINITY)
  662. return NINF;
  663. else if (number >= 0 && number < 256 && number == (double)((size_t)number))
  664. return NUMS[(size_t)number];
  665. else if (number < 0 && number >= -8 &&
  666. fabs(number) == (double)((size_t)fabs(number)))
  667. return NNUMS[((size_t)fabs(number)) - 1];
  668. value_t *val = value_new(NUMBER);
  669. val->val.number = number;
  670. return val;
  671. }
  672. value_t *value_new_char(unsigned char _char) { return CHARS[_char]; }
  673. bool value_equals(value_t *x, value_t *y) {
  674. if (x->tag != y->tag)
  675. return false;
  676. switch (x->tag) {
  677. case ARRAY: {
  678. list_t *tx = x->val.array;
  679. list_t *ty = y->val.array;
  680. if (tx->length == 0 && ty->length == 0)
  681. break;
  682. if (tx->length != ty->length)
  683. return false;
  684. for (size_t i = 0; i < tx->length; i++)
  685. if (!value_equals(tx->data[i], ty->data[i]))
  686. return false;
  687. }
  688. case VERB:
  689. return strcmp(x->val.verb->name, x->val.verb->name) == 0;
  690. case SYMBOL:
  691. return strcmp(x->val.symbol, y->val.symbol) == 0;
  692. case NUMBER:
  693. if (isnan(x->val.number) && isnan(y->val.number))
  694. break;
  695. return x->val.number == y->val.number;
  696. case CHAR:
  697. return x == y;
  698. case NIL:
  699. case UDF:
  700. break;
  701. }
  702. return true;
  703. }
  704. bool is_char_array(list_t *a) {
  705. for (size_t i = 0; i < a->length; i++) {
  706. value_t *v = a->data[i];
  707. if (v->tag != CHAR || !isprint(v->val._char))
  708. return false;
  709. }
  710. return true;
  711. }
  712. bool is_bytes_array(list_t *a) {
  713. for (size_t i = 0; i < a->length; i++) {
  714. value_t *v = a->data[i];
  715. if (v->tag != CHAR)
  716. return false;
  717. }
  718. return true;
  719. }
  720. bool is_arrays_array(list_t *a) {
  721. for (size_t i = 0; i < a->length; i++) {
  722. value_t *v = a->data[i];
  723. if (v->tag != ARRAY)
  724. return false;
  725. }
  726. return true;
  727. }
  728. bool nonar(list_t *a) {
  729. if (!a->data)
  730. return true;
  731. for (size_t i = 1; i < a->length; i++) {
  732. value_t *v = a->data[i];
  733. if (v->tag == ARRAY)
  734. return false;
  735. }
  736. return true;
  737. }
  738. bool matp(list_t *a) {
  739. if (a->length < 2)
  740. return false;
  741. size_t rwl = ((value_t *)a->data[0])->val.array->length;
  742. if (rwl < 1)
  743. return false;
  744. for (size_t i = 0; i < a->length; i++) {
  745. value_t *v = a->data[i];
  746. if (v->tag != ARRAY || v->val.array->length != rwl ||
  747. !nonar(v->val.array) || is_char_array(v->val.array))
  748. return false;
  749. }
  750. return true;
  751. }
  752. char *value_show(value_t *v);
  753. char *show_array(value_t *v) {
  754. if (v->tag != ARRAY)
  755. return value_show(v);
  756. list_t *t = v->val.array;
  757. if (!t->data)
  758. return strdup_checked("()");
  759. buffer_t *buf = buffer_new();
  760. if (t->length == 1) {
  761. buffer_append(buf, ',');
  762. char *ts = value_show(t->data[0]);
  763. buffer_append_str(buf, ts);
  764. GC_FREE(ts);
  765. return buffer_read(buf);
  766. }
  767. if (is_char_array(t)) {
  768. for (size_t i = 0; i < t->length; i++)
  769. buffer_append(buf, ((value_t *)t->data[i])->val._char);
  770. return buffer_read(buf);
  771. }
  772. if (!is_arrays_array(t))
  773. for (size_t i = 0; i < t->length; i++) {
  774. char *ts = value_show(t->data[i]);
  775. buffer_append_str(buf, ts);
  776. GC_FREE(ts);
  777. if (i != t->length - 1)
  778. buffer_append(buf, ' ');
  779. }
  780. else if (matp(t)) {
  781. size_t rwl = 0;
  782. size_t pad = 0;
  783. size_t padl = 0;
  784. list_t *ss = list_new();
  785. for (size_t i = 0; i < t->length; i++) {
  786. value_t *rw = t->data[i];
  787. list_t *rwt = rw->val.array;
  788. if (rwl < 1)
  789. rwl = rwt->length;
  790. for (size_t j = 0; j < rwt->length; j++) {
  791. char *s = value_show(rwt->data[j]);
  792. size_t z = strlen(s);
  793. if (z > pad)
  794. pad = z;
  795. if (j == 0 && z > padl)
  796. padl = z;
  797. list_push(ss, s);
  798. }
  799. }
  800. size_t k = 0;
  801. for (size_t i = 0; i < ss->length; i++) {
  802. char *s = ss->data[i];
  803. size_t mp = (k == 0 ? padl : pad) - strlen(s);
  804. while (mp--)
  805. buffer_append(buf, ' ');
  806. buffer_append_str(buf, s);
  807. GC_FREE(s);
  808. if (i != ss->length - 1) {
  809. if (k == rwl - 1) {
  810. k = 0;
  811. buffer_append(buf, '\n');
  812. } else {
  813. buffer_append(buf, ' ');
  814. k++;
  815. }
  816. }
  817. }
  818. GC_FREE(ss->data);
  819. GC_FREE(ss);
  820. } else
  821. for (size_t i = 0; i < t->length; i++) {
  822. value_t *rw = t->data[i];
  823. char *ts = show_array(rw);
  824. buffer_append_str(buf, ts);
  825. GC_FREE(ts);
  826. if (i != t->length - 1)
  827. buffer_append(buf, '\n');
  828. }
  829. return buffer_read(buf);
  830. }
  831. char *value_show(value_t *v) {
  832. switch (v->tag) {
  833. case ARRAY:
  834. return show_array(v);
  835. case VERB:
  836. return strdup_checked(v->val.verb->name);
  837. case SYMBOL:
  838. return strdup_checked(v->val.symbol);
  839. case NUMBER: {
  840. char buf[128];
  841. snprintf(buf, sizeof(buf), "%.15g", v->val.number);
  842. return strdup_checked(buf);
  843. }
  844. case CHAR: {
  845. if (!isprint(v->val._char)) {
  846. char buf[16];
  847. snprintf(buf, sizeof(buf), "4t.%d", v->val._char);
  848. return strdup_checked(buf);
  849. }
  850. char buf[2];
  851. buf[0] = v->val._char;
  852. buf[1] = 0;
  853. return strdup_checked(buf);
  854. }
  855. case NIL:
  856. return strdup_checked("nil");
  857. case UDF:
  858. return strdup_checked("udf");
  859. }
  860. return strdup_checked("<?>");
  861. }
  862. char *value_str(value_t *v) {
  863. if (v->tag == ARRAY && v->val.array->length == 1 &&
  864. ((value_t *)v->val.array->data[0])->tag == CHAR)
  865. return value_show(v->val.array->data[0]);
  866. else if (v->tag == ARRAY && is_bytes_array(v->val.array)) {
  867. buffer_t *buf = buffer_new();
  868. for (size_t i = 0; i < v->val.array->length; i++)
  869. buffer_append(buf, ((value_t *)v->val.array->data[i])->val._char);
  870. return buffer_read(buf);
  871. }
  872. return value_show(v);
  873. }
  874. double get_numeric(value_t *v) {
  875. if (v->tag == CHAR)
  876. return v->val._char;
  877. return v->val.number;
  878. }
  879. bool value_is_truthy(value_t *x) {
  880. switch (x->tag) {
  881. case ARRAY:
  882. return x->val.array->length != 0;
  883. case NUMBER:
  884. case CHAR:
  885. return get_numeric(x) != 0;
  886. case NIL:
  887. case UDF:
  888. return false;
  889. default:
  890. return true;
  891. }
  892. }
  893. verb_t *find_verb(char *s);
  894. interpreter_t *interpreter_new(void) {
  895. interpreter_t *state = malloc_checked(sizeof(interpreter_t));
  896. state->env = table_new();
  897. state->args = list_new();
  898. state->selfrefs = list_new();
  899. state->nil = value_new(NIL);
  900. state->udf = value_new(UDF);
  901. state->unit = _UNIT;
  902. state->at = find_verb("@");
  903. return state;
  904. }
  905. void interpreter_error(interpreter_t *state, char *e) {
  906. fprintf(stderr, "%s error\n", e);
  907. exit(1);
  908. }
  909. value_t *each_rank(interpreter_t *state, verb_t *f, value_t *x, unsigned int d,
  910. unsigned int rm) {
  911. if (!f->monad)
  912. return state->udf;
  913. if (d >= rm || x->tag != ARRAY) {
  914. if (f->mark)
  915. list_push(state->selfrefs, f);
  916. value_t *r = f->monad(state, f, x);
  917. if (f->mark)
  918. list_pop(state->selfrefs);
  919. return r;
  920. }
  921. list_t *t = x->val.array;
  922. if (!t->data)
  923. return x;
  924. list_t *l = list_newk(t->length);
  925. for (size_t i = 0; i < t->length; i++)
  926. l->data[i] = each_rank(state, f, t->data[i], d + 1, rm);
  927. return value_new_array(l);
  928. }
  929. value_t *apply_monad(interpreter_t *state, value_t *f, value_t *x) {
  930. if (f->tag != VERB)
  931. return state->udf;
  932. if (!f->val.verb->monad)
  933. return state->udf;
  934. return each_rank(state, f->val.verb, x, 0, f->val.verb->rank[0]);
  935. }
  936. value_t *together(interpreter_t *state, verb_t *f, value_t *x, value_t *y,
  937. unsigned int dl, unsigned int dr, unsigned int rl,
  938. unsigned int rr) {
  939. if (!f->dyad)
  940. return state->udf;
  941. if (dl >= rl && dr >= rr) {
  942. if (f->mark)
  943. list_push(state->selfrefs, f);
  944. value_t *r = f->dyad(state, f, x, y);
  945. if (f->mark)
  946. list_pop(state->selfrefs);
  947. return r;
  948. }
  949. if (dl < rl && dr < rr && x->tag == ARRAY && y->tag == ARRAY) {
  950. list_t *tx = x->val.array;
  951. list_t *ty = y->val.array;
  952. if (!tx->data || !ty->data)
  953. return !tx->data ? x : y;
  954. list_t *t = list_newk(ty->length < tx->length ? ty->length : tx->length);
  955. for (size_t i = 0; i < tx->length; i++) {
  956. if (i >= ty->length)
  957. break;
  958. t->data[i] =
  959. together(state, f, tx->data[i], ty->data[i], dl + 1, dr + 1, rl, rr);
  960. }
  961. return value_new_array(t);
  962. } else if ((x->tag != ARRAY || dl >= rl) && y->tag == ARRAY && dr < rr) {
  963. list_t *ty = y->val.array;
  964. if (!ty->data)
  965. return y;
  966. list_t *t = list_newk(ty->length);
  967. for (size_t i = 0; i < ty->length; i++)
  968. t->data[i] = together(state, f, x, ty->data[i], dl, dr + 1, rl, rr);
  969. return value_new_array(t);
  970. } else if ((y->tag != ARRAY || dr >= rr) && x->tag == ARRAY && dl < rl) {
  971. list_t *tx = x->val.array;
  972. if (!tx->data)
  973. return x;
  974. list_t *t = list_newk(tx->length);
  975. for (size_t i = 0; i < tx->length; i++)
  976. t->data[i] = together(state, f, tx->data[i], y, dl + 1, dr, rl, rr);
  977. return value_new_array(t);
  978. }
  979. if (f->mark)
  980. list_push(state->selfrefs, f);
  981. value_t *r = f->dyad(state, f, x, y);
  982. if (f->mark)
  983. list_pop(state->selfrefs);
  984. return r;
  985. }
  986. value_t *apply_dyad(interpreter_t *state, value_t *f, value_t *x, value_t *y) {
  987. if (f->tag != VERB)
  988. return state->nil;
  989. return together(state, f->val.verb, x, y, 0, 0, f->val.verb->rank[1],
  990. f->val.verb->rank[2]);
  991. }
  992. typedef struct _node_t node_t;
  993. struct _node_t {
  994. enum node_tag_t {
  995. N_STRAND,
  996. N_LITERAL,
  997. N_INDEX1,
  998. N_INDEX2,
  999. N_FUN,
  1000. N_MONAD,
  1001. N_DYAD,
  1002. N_ADV,
  1003. N_CONJ,
  1004. N_PARTIAL_CONJ,
  1005. N_FORK,
  1006. N_HOOK,
  1007. N_BOND,
  1008. N_OVER,
  1009. N_BIND
  1010. } tag;
  1011. adverb_t *av;
  1012. value_t *v;
  1013. list_t *l;
  1014. node_t *a;
  1015. node_t *b;
  1016. node_t *c;
  1017. size_t dp;
  1018. };
  1019. char *node_show(node_t *n) {
  1020. switch (n->tag) {
  1021. case N_STRAND: {
  1022. buffer_t *buf = buffer_new();
  1023. for (size_t i = 0; i < n->l->length; i++) {
  1024. if (i != 0)
  1025. buffer_append_str(buf, ",:");
  1026. char *s = node_show(n->l->data[i]);
  1027. buffer_append_str(buf, s);
  1028. GC_FREE(s);
  1029. }
  1030. return buffer_read(buf);
  1031. }
  1032. case N_LITERAL:
  1033. return value_show(n->v);
  1034. case N_INDEX1: {
  1035. char *s;
  1036. buffer_t *buf = buffer_new();
  1037. s = node_show(n->a);
  1038. buffer_append_str(buf, s);
  1039. GC_FREE(s);
  1040. buffer_append(buf, ' ');
  1041. s = node_show(n->b);
  1042. buffer_append_str(buf, s);
  1043. GC_FREE(s);
  1044. return buffer_read(buf);
  1045. }
  1046. case N_INDEX2: {
  1047. char *s;
  1048. buffer_t *buf = buffer_new();
  1049. s = node_show(n->a);
  1050. buffer_append_str(buf, s);
  1051. GC_FREE(s);
  1052. buffer_append(buf, ' ');
  1053. s = node_show(n->b);
  1054. buffer_append_str(buf, s);
  1055. GC_FREE(s);
  1056. buffer_append(buf, ' ');
  1057. s = node_show(n->c);
  1058. buffer_append_str(buf, s);
  1059. GC_FREE(s);
  1060. return buffer_read(buf);
  1061. }
  1062. case N_FUN: {
  1063. buffer_t *buf = buffer_new();
  1064. buffer_append(buf, ':');
  1065. char *s = node_show(n->a);
  1066. buffer_append_str(buf, s);
  1067. GC_FREE(s);
  1068. return buffer_read(buf);
  1069. }
  1070. case N_MONAD:
  1071. case N_HOOK:
  1072. case N_BOND:
  1073. case N_OVER: {
  1074. char *s;
  1075. buffer_t *buf = buffer_new();
  1076. s = node_show(n->a);
  1077. buffer_append_str(buf, s);
  1078. GC_FREE(s);
  1079. s = node_show(n->b);
  1080. buffer_append_str(buf, s);
  1081. GC_FREE(s);
  1082. return buffer_read(buf);
  1083. }
  1084. case N_DYAD: {
  1085. char *s;
  1086. buffer_t *buf = buffer_new();
  1087. s = node_show(n->b);
  1088. buffer_append_str(buf, s);
  1089. GC_FREE(s);
  1090. s = node_show(n->a);
  1091. buffer_append_str(buf, s);
  1092. GC_FREE(s);
  1093. s = node_show(n->c);
  1094. buffer_append_str(buf, s);
  1095. GC_FREE(s);
  1096. return buffer_read(buf);
  1097. }
  1098. case N_ADV:
  1099. case N_PARTIAL_CONJ: {
  1100. buffer_t *buf = buffer_new();
  1101. char *s = node_show(n->a);
  1102. buffer_append_str(buf, s);
  1103. GC_FREE(s);
  1104. buffer_append_str(buf, n->av->name);
  1105. return buffer_read(buf);
  1106. }
  1107. case N_CONJ: {
  1108. char *s;
  1109. buffer_t *buf = buffer_new();
  1110. s = node_show(n->a);
  1111. buffer_append_str(buf, s);
  1112. GC_FREE(s);
  1113. buffer_append_str(buf, n->av->name);
  1114. s = node_show(n->b);
  1115. buffer_append_str(buf, s);
  1116. GC_FREE(s);
  1117. return buffer_read(buf);
  1118. }
  1119. case N_FORK: {
  1120. char *s;
  1121. buffer_t *buf = buffer_new();
  1122. s = node_show(n->a);
  1123. buffer_append_str(buf, s);
  1124. GC_FREE(s);
  1125. s = node_show(n->b);
  1126. buffer_append_str(buf, s);
  1127. GC_FREE(s);
  1128. s = node_show(n->c);
  1129. buffer_append_str(buf, s);
  1130. GC_FREE(s);
  1131. return buffer_read(buf);
  1132. }
  1133. case N_BIND: {
  1134. char *s;
  1135. buffer_t *buf = buffer_new();
  1136. s = node_show(n->a);
  1137. buffer_append_str(buf, s);
  1138. GC_FREE(s);
  1139. buffer_append(buf, ':');
  1140. s = node_show(n->b);
  1141. buffer_append_str(buf, s);
  1142. GC_FREE(s);
  1143. return buffer_read(buf);
  1144. }
  1145. }
  1146. return strdup_checked("<?>");
  1147. }
  1148. value_t *_fork_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1149. verb_t *f = list_index(self->bonds, 0);
  1150. verb_t *g = list_index(self->bonds, 1);
  1151. verb_t *h = list_index(self->bonds, 2);
  1152. value_t *l = each_rank(state, f, x, 0, f->rank[0]);
  1153. value_t *r = each_rank(state, h, x, 0, f->rank[0]);
  1154. return together(state, g, l, r, 0, 0, g->rank[1], g->rank[2]);
  1155. }
  1156. value_t *_fork_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1157. value_t *y) {
  1158. verb_t *f = list_index(self->bonds, 0);
  1159. verb_t *g = list_index(self->bonds, 1);
  1160. verb_t *h = list_index(self->bonds, 2);
  1161. value_t *l = each_rank(state, f, x, 0, f->rank[0]);
  1162. value_t *r = each_rank(state, h, y, 0, f->rank[0]);
  1163. return together(state, g, l, r, 0, 0, g->rank[1], g->rank[2]);
  1164. }
  1165. value_t *_hook_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1166. verb_t *f = list_index(self->bonds, 0);
  1167. verb_t *g = list_index(self->bonds, 1);
  1168. value_t *r = each_rank(state, g, x, 0, g->rank[0]);
  1169. return each_rank(state, f, r, 0, f->rank[0]);
  1170. }
  1171. value_t *_hook_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1172. value_t *y) {
  1173. verb_t *f = list_index(self->bonds, 0);
  1174. verb_t *g = list_index(self->bonds, 1);
  1175. value_t *r = together(state, g, x, y, 0, 0, g->rank[1], g->rank[2]);
  1176. return each_rank(state, f, r, 0, f->rank[0]);
  1177. }
  1178. value_t *_bond_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1179. verb_t *f = list_index(self->bonds, 0);
  1180. value_t *g = list_index(self->bonds, 1);
  1181. return together(state, f, g, x, 0, 0, f->rank[1], f->rank[2]);
  1182. }
  1183. value_t *_bond_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1184. value_t *y) {
  1185. verb_t *f = list_index(self->bonds, 0);
  1186. value_t *g = list_index(self->bonds, 1);
  1187. value_t *r = together(state, f, x, y, 0, 0, f->rank[1], f->rank[2]);
  1188. return together(state, f, g, r, 0, 0, f->rank[1], f->rank[2]);
  1189. }
  1190. value_t *_over_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1191. value_t *f = list_index(self->bonds, 0);
  1192. verb_t *g = list_index(self->bonds, 1);
  1193. verb_t *h = list_index(self->bonds, 2);
  1194. value_t *l = each_rank(state, h, x, 0, h->rank[0]);
  1195. return together(state, g, f, l, 0, 0, g->rank[1], g->rank[2]);
  1196. }
  1197. value_t *_over_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1198. value_t *y) {
  1199. value_t *f = list_index(self->bonds, 0);
  1200. verb_t *g = list_index(self->bonds, 1);
  1201. verb_t *h = list_index(self->bonds, 2);
  1202. value_t *l = together(state, h, x, y, 0, 0, h->rank[1], h->rank[2]);
  1203. return together(state, g, f, l, 0, 0, g->rank[1], g->rank[2]);
  1204. }
  1205. bool function_collect_args(node_t *node, unsigned int *argc) {
  1206. if (!node)
  1207. return false;
  1208. if (node->tag == N_LITERAL && node->v->tag == SYMBOL &&
  1209. strcmp(node->v->val.symbol, "y") == 0) {
  1210. *argc = 2;
  1211. return true;
  1212. } else if (node->tag == N_LITERAL && node->v->tag == SYMBOL &&
  1213. strcmp(node->v->val.symbol, "x") == 0) {
  1214. if (*argc < 2)
  1215. *argc = 1;
  1216. } else if (node->tag == N_MONAD || node->tag == N_CONJ ||
  1217. node->tag == N_HOOK || node->tag == N_BOND ||
  1218. node->tag == N_INDEX1) {
  1219. if (function_collect_args(node->a, argc))
  1220. return true;
  1221. if (function_collect_args(node->b, argc))
  1222. return true;
  1223. } else if (node->tag == N_DYAD || node->tag == N_FORK ||
  1224. node->tag == N_OVER || node->tag == N_INDEX2) {
  1225. if (function_collect_args(node->a, argc))
  1226. return true;
  1227. if (function_collect_args(node->b, argc))
  1228. return true;
  1229. if (function_collect_args(node->c, argc))
  1230. return true;
  1231. } else if (node->tag == N_ADV) {
  1232. if (function_collect_args(node->a, argc))
  1233. return true;
  1234. } else if (node->tag == N_STRAND) {
  1235. list_t *t = node->l;
  1236. for (size_t i = 0; i < t->length; i++)
  1237. if (function_collect_args(t->data[i], argc))
  1238. return true;
  1239. } else if (node->tag == N_BIND) {
  1240. if (function_collect_args(node->b, argc))
  1241. return true;
  1242. }
  1243. return false;
  1244. }
  1245. value_t *interpreter_walk(interpreter_t *state, node_t *node);
  1246. value_t *_const_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1247. return interpreter_walk(state, self->bonds->data[0]);
  1248. }
  1249. value_t *_const_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1250. value_t *y) {
  1251. return interpreter_walk(state, self->bonds->data[0]);
  1252. }
  1253. value_t *_constv_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1254. return self->bonds->data[0];
  1255. }
  1256. value_t *_constv_dyad(interpreter_t *state, verb_t *self, value_t *x,
  1257. value_t *y) {
  1258. return self->bonds->data[0];
  1259. }
  1260. value_t *_fun_monad(interpreter_t *state, verb_t *self, value_t *x) {
  1261. list_t *args = list_new();
  1262. list_push(args, x);
  1263. list_push(args, self);
  1264. list_push(state->args, args);
  1265. value_t *r = interpreter_walk(state, self->bonds->data[0]);
  1266. list_pop(state->args);
  1267. GC_FREE(args);
  1268. return r;
  1269. }
  1270. value_t *_fun_dyad(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1271. list_t *args = list_new();
  1272. list_push(args, x);
  1273. list_push(args, y);
  1274. list_push(args, self);
  1275. list_push(state->args, args);
  1276. value_t *r = interpreter_walk(state, self->bonds->data[1]);
  1277. list_pop(state->args);
  1278. GC_FREE(args);
  1279. return r;
  1280. }
  1281. value_t *_partial_conjunction(interpreter_t *state, verb_t *self, value_t *x) {
  1282. adverb_t *av = self->bonds->data[0];
  1283. value_t *a = self->bonds->data[1];
  1284. return value_new_verb(av->conjunction(state, a, x));
  1285. }
  1286. node_t *node_new1(enum node_tag_t tag, node_t *a);
  1287. value_t *interpreter_walk(interpreter_t *state, node_t *node) {
  1288. if (!node)
  1289. return state->nil;
  1290. switch (node->tag) {
  1291. case N_STRAND: {
  1292. list_t *t = list_copy(node->l);
  1293. for (size_t i = 0; i < t->length; i++)
  1294. t->data[i] = interpreter_walk(state, t->data[i]);
  1295. return value_new_array(t);
  1296. }
  1297. case N_LITERAL: {
  1298. value_t *v = node->v;
  1299. value_t *t = NULL;
  1300. if (v->tag == SYMBOL) {
  1301. char *n = v->val.symbol;
  1302. if (state->args->data) {
  1303. list_t *args = list_index(state->args, -1);
  1304. size_t argc = args->length - 1;
  1305. if (argc == 2 && strcmp(n, "y") == 0)
  1306. return args->data[1];
  1307. else if (strcmp(n, "x") == 0)
  1308. return args->data[0];
  1309. }
  1310. if ((t = table_get(state->env, n)))
  1311. return t;
  1312. if (strcmp(n, "T") == 0)
  1313. return value_new_number(time(NULL));
  1314. }
  1315. return v;
  1316. }
  1317. case N_INDEX1:
  1318. return together(state, state->at, interpreter_walk(state, node->a),
  1319. interpreter_walk(state, node->b), 0, 0, state->at->rank[1],
  1320. state->at->rank[2]);
  1321. case N_INDEX2:
  1322. return together(state, state->at,
  1323. together(state, state->at, interpreter_walk(state, node->a),
  1324. interpreter_walk(state, node->b), 0, 0,
  1325. state->at->rank[1], state->at->rank[2]),
  1326. interpreter_walk(state, node->c), 0, 0, state->at->rank[1],
  1327. state->at->rank[2]);
  1328. case N_FUN: {
  1329. unsigned int argc = 0;
  1330. function_collect_args(node->a, &argc);
  1331. verb_t *nv = verb_new();
  1332. if (argc > 0)
  1333. nv->is_fun = true;
  1334. nv->bonds = list_new();
  1335. char *s = node_show(node->a);
  1336. size_t z = strlen(s) + 2;
  1337. nv->name = malloc_checked_atomic(z);
  1338. snprintf(nv->name, z, ":%s", s);
  1339. GC_FREE(s);
  1340. nv->rank[0] = 0;
  1341. nv->rank[1] = 0;
  1342. nv->rank[2] = 0;
  1343. if (argc == 0) {
  1344. list_push(nv->bonds, node->a);
  1345. nv->monad = _const_monad;
  1346. nv->dyad = _const_dyad;
  1347. } else if (argc == 1) {
  1348. list_push(nv->bonds, node->a);
  1349. nv->monad = _fun_monad;
  1350. nv->dyad = NULL;
  1351. } else {
  1352. nv->monad = NULL;
  1353. list_push(nv->bonds, state->udf);
  1354. list_push(nv->bonds, node->a);
  1355. nv->dyad = _fun_dyad;
  1356. }
  1357. return value_new_verb(nv);
  1358. }
  1359. case N_MONAD:
  1360. return apply_monad(state, interpreter_walk(state, node->a),
  1361. interpreter_walk(state, node->b));
  1362. case N_DYAD:
  1363. return apply_dyad(state, interpreter_walk(state, node->a),
  1364. interpreter_walk(state, node->b),
  1365. interpreter_walk(state, node->c));
  1366. case N_ADV: {
  1367. value_t *v = interpreter_walk(state, node->a);
  1368. verb_t *nv = node->av->adverb(state, v);
  1369. if (node->dp < 2)
  1370. nv->mark = true;
  1371. return value_new_verb(nv);
  1372. }
  1373. case N_CONJ: {
  1374. value_t *v1 = interpreter_walk(state, node->a);
  1375. value_t *v2 = interpreter_walk(state, node->b);
  1376. verb_t *nv = node->av->conjunction(state, v1, v2);
  1377. if (node->dp < 2)
  1378. nv->mark = true;
  1379. return value_new_verb(nv);
  1380. }
  1381. case N_PARTIAL_CONJ: {
  1382. verb_t *nv = verb_new();
  1383. value_t *a = interpreter_walk(state, node->a);
  1384. char *r = value_show(a);
  1385. size_t l = strlen(r) + strlen(node->av->name) + 1;
  1386. nv->name = malloc_checked_atomic(l);
  1387. snprintf(nv->name, l, "%s%s", r, node->av->name);
  1388. GC_FREE(r);
  1389. nv->bonds = list_new();
  1390. list_push(nv->bonds, node->av);
  1391. list_push(nv->bonds, a);
  1392. nv->rank[0] = 0;
  1393. nv->rank[1] = 0;
  1394. nv->rank[2] = 0;
  1395. nv->monad = _partial_conjunction;
  1396. nv->dyad = NULL;
  1397. if (node->dp < 2)
  1398. nv->mark = true;
  1399. return value_new_verb(nv);
  1400. }
  1401. case N_FORK: {
  1402. value_t *_f = interpreter_walk(state, node->a);
  1403. if (_f->tag != VERB)
  1404. return state->udf;
  1405. value_t *_g = interpreter_walk(state, node->b);
  1406. if (_g->tag != VERB)
  1407. return state->udf;
  1408. value_t *_h = interpreter_walk(state, node->c);
  1409. if (_h->tag != VERB)
  1410. return state->udf;
  1411. verb_t *f = _f->val.verb;
  1412. verb_t *g = _g->val.verb;
  1413. verb_t *h = _h->val.verb;
  1414. verb_t *nv = verb_new();
  1415. nv->bonds = list_newk(3);
  1416. nv->bonds->data[0] = f;
  1417. nv->bonds->data[1] = g;
  1418. nv->bonds->data[2] = h;
  1419. size_t l = strlen(f->name) + strlen(g->name) + strlen(h->name) + 1;
  1420. nv->name = malloc_checked_atomic(l);
  1421. snprintf(nv->name, l, "%s%s%s", f->name, g->name, h->name);
  1422. nv->rank[0] = 0;
  1423. nv->rank[1] = 0;
  1424. nv->rank[2] = 0;
  1425. nv->monad = _fork_monad;
  1426. nv->dyad = _fork_dyad;
  1427. if (node->dp < 2)
  1428. nv->mark = true;
  1429. return value_new_verb(nv);
  1430. }
  1431. case N_HOOK: {
  1432. value_t *_f = interpreter_walk(state, node->a);
  1433. if (_f->tag != VERB)
  1434. return state->udf;
  1435. value_t *_g = interpreter_walk(state, node->b);
  1436. if (_g->tag != VERB)
  1437. return state->udf;
  1438. verb_t *f = _f->val.verb;
  1439. verb_t *g = _g->val.verb;
  1440. verb_t *nv = verb_new();
  1441. nv->bonds = list_newk(2);
  1442. nv->bonds->data[0] = f;
  1443. nv->bonds->data[1] = g;
  1444. size_t l = strlen(f->name) + strlen(g->name) + 1;
  1445. nv->name = malloc_checked_atomic(l);
  1446. snprintf(nv->name, l, "%s%s", f->name, g->name);
  1447. nv->rank[0] = 0;
  1448. nv->rank[1] = 0;
  1449. nv->rank[2] = 0;
  1450. nv->monad = _hook_monad;
  1451. nv->dyad = _hook_dyad;
  1452. if (node->dp < 2)
  1453. nv->mark = true;
  1454. return value_new_verb(nv);
  1455. }
  1456. case N_BOND: {
  1457. value_t *_f = interpreter_walk(state, node->a);
  1458. if (_f->tag != VERB)
  1459. return state->udf;
  1460. value_t *g = interpreter_walk(state, node->b);
  1461. verb_t *f = _f->val.verb;
  1462. verb_t *nv = verb_new();
  1463. nv->bonds = list_newk(2);
  1464. nv->bonds->data[0] = f;
  1465. nv->bonds->data[1] = g;
  1466. char *r = value_show(g);
  1467. size_t l = strlen(r) + strlen(f->name) + 1;
  1468. nv->name = malloc_checked_atomic(l);
  1469. snprintf(nv->name, l, "%s%s", r, f->name);
  1470. GC_FREE(r);
  1471. nv->rank[0] = 0;
  1472. nv->rank[1] = 0;
  1473. nv->rank[2] = 0;
  1474. nv->monad = _bond_monad;
  1475. nv->dyad = _bond_dyad;
  1476. if (node->dp < 2)
  1477. nv->mark = true;
  1478. return value_new_verb(nv);
  1479. }
  1480. case N_OVER: {
  1481. value_t *f = interpreter_walk(state, node->a);
  1482. value_t *_g = interpreter_walk(state, node->b);
  1483. if (_g->tag != VERB)
  1484. return state->udf;
  1485. value_t *_h = interpreter_walk(state, node->c);
  1486. if (_h->tag != VERB)
  1487. return state->udf;
  1488. verb_t *g = _g->val.verb;
  1489. verb_t *h = _h->val.verb;
  1490. verb_t *nv = verb_new();
  1491. nv->bonds = list_newk(3);
  1492. nv->bonds->data[0] = f;
  1493. nv->bonds->data[1] = g;
  1494. nv->bonds->data[2] = h;
  1495. char *r = value_show(f);
  1496. size_t l = strlen(r) + strlen(g->name) + strlen(h->name) + 1;
  1497. nv->name = malloc_checked_atomic(l);
  1498. snprintf(nv->name, l, "%s%s%s", r, g->name, h->name);
  1499. GC_FREE(r);
  1500. nv->rank[0] = 0;
  1501. nv->rank[1] = 0;
  1502. nv->rank[2] = 0;
  1503. nv->monad = _over_monad;
  1504. nv->dyad = _over_dyad;
  1505. if (node->dp < 2)
  1506. nv->mark = true;
  1507. return value_new_verb(nv);
  1508. }
  1509. case N_BIND: {
  1510. value_t *l = node->a->v;
  1511. node_t *b = node->b;
  1512. if (state->bn || state->args->data || node->dp != 0) {
  1513. table_set(state->env, l->val.symbol, interpreter_walk(state, b));
  1514. break;
  1515. }
  1516. unsigned int argc = 0;
  1517. function_collect_args(b, &argc);
  1518. if (argc != 0)
  1519. b = node_new1(N_FUN, b);
  1520. bool t = state->bn;
  1521. state->bn = true;
  1522. value_t *r = interpreter_walk(state, b);
  1523. state->bn = t;
  1524. if (argc != 0) {
  1525. GC_FREE(r->val.verb->name);
  1526. r->val.verb->name = l->val.symbol;
  1527. }
  1528. if (r->tag == VERB && argc == 0)
  1529. r->val.verb->mark = true;
  1530. value_t *ov = table_get(state->env, l->val.symbol);
  1531. if (ov && ov->tag == VERB && ov->val.verb->is_fun && r->tag == VERB &&
  1532. r->val.verb->is_fun) {
  1533. if (!ov->val.verb->monad && r->val.verb->monad) {
  1534. list_set(ov->val.verb->bonds, 0, r->val.verb->bonds->data[0]);
  1535. ov->val.verb->monad = r->val.verb->monad;
  1536. break;
  1537. }
  1538. if (!ov->val.verb->dyad && r->val.verb->dyad) {
  1539. list_push(ov->val.verb->bonds, r->val.verb->bonds->data[1]);
  1540. ov->val.verb->dyad = r->val.verb->dyad;
  1541. break;
  1542. }
  1543. }
  1544. table_set(state->env, l->val.symbol, r);
  1545. } break;
  1546. }
  1547. return state->nil;
  1548. }
  1549. value_t *verb_const(interpreter_t *state, verb_t *self, value_t *x) {
  1550. verb_t *nv = verb_new();
  1551. nv->bonds = list_newk(1);
  1552. nv->bonds->data[0] = x;
  1553. char *r = value_show(x);
  1554. size_t l = strlen(r) + 2;
  1555. nv->name = malloc_checked_atomic(l);
  1556. snprintf(nv->name, l, ":%s", r);
  1557. nv->rank[0] = 0;
  1558. nv->rank[1] = 0;
  1559. nv->rank[2] = 0;
  1560. nv->monad = _constv_monad;
  1561. nv->dyad = _constv_dyad;
  1562. return value_new_verb(nv);
  1563. }
  1564. value_t *verb_bind(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1565. if (x->tag == SYMBOL) {
  1566. if (y->tag == VERB)
  1567. y->val.verb->mark = true;
  1568. table_set(state->env, x->val.symbol, y);
  1569. }
  1570. return state->udf;
  1571. }
  1572. table_t *Inverses;
  1573. value_t *verb_unbind(interpreter_t *state, verb_t *self, value_t *x) {
  1574. if (x->tag == SYMBOL) {
  1575. table_delete(state->env, x->val.symbol);
  1576. return state->nil;
  1577. }
  1578. return state->udf;
  1579. }
  1580. value_t *verb_obverse(interpreter_t *state, verb_t *self, value_t *x,
  1581. value_t *y) {
  1582. if (x->tag == VERB && y->tag == VERB) {
  1583. verb_t *vx = x->val.verb;
  1584. if (!y->val.verb->monad)
  1585. return state->udf;
  1586. if (vx->is_fun)
  1587. return state->udf;
  1588. if (table_has(Inverses, vx->name))
  1589. return state->udf;
  1590. table_set(Inverses, vx->name, y->val.verb);
  1591. return state->nil;
  1592. }
  1593. return state->udf;
  1594. }
  1595. value_t *verb_flip(interpreter_t *state, verb_t *self, value_t *x) {
  1596. if (x->tag != ARRAY || !x->val.array->data)
  1597. return state->udf;
  1598. if (!is_arrays_array(x->val.array))
  1599. return state->udf;
  1600. list_t *r = list_new();
  1601. value_t *c0 = x->val.array->data[0];
  1602. list_t *c0t = c0->val.array;
  1603. size_t c0l = c0t->length;
  1604. for (size_t i = 0; i < c0l; i++) {
  1605. list_t *nc = list_new();
  1606. for (size_t j = 0; j < x->val.array->length; j++) {
  1607. value_t *rw = x->val.array->data[j];
  1608. list_t *rwt = rw->val.array;
  1609. if (!rwt->data)
  1610. return state->udf;
  1611. value_t *v = list_index(rwt, i);
  1612. if (!v)
  1613. v = rwt->data[0];
  1614. list_push(nc, v);
  1615. }
  1616. list_push(r, value_new_array(nc));
  1617. }
  1618. return value_new_array(r);
  1619. }
  1620. value_t *verb_plus(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1621. if ((x->tag == NUMBER || x->tag == CHAR) &&
  1622. (y->tag == NUMBER || y->tag == CHAR)) {
  1623. if (x->tag == CHAR || y->tag == CHAR)
  1624. return value_new_char(get_numeric(x) + get_numeric(y));
  1625. return value_new_number(get_numeric(x) + get_numeric(y));
  1626. }
  1627. return _NAN;
  1628. }
  1629. value_t *verb_sign(interpreter_t *state, verb_t *self, value_t *x) {
  1630. if (x->tag == NUMBER)
  1631. return x->val.number < 0 ? NNUMS[0] : x->val.number > 0 ? NUMS[1] : NUMS[0];
  1632. return _NAN;
  1633. }
  1634. double gcd(double a, double b) {
  1635. if (b != 0)
  1636. return gcd(b, fmod(a, b));
  1637. else
  1638. return fabs(a);
  1639. }
  1640. value_t *verb_gcd(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1641. if (x->tag == NUMBER && y->tag == NUMBER)
  1642. return value_new_number(gcd(x->val.number, y->val.number));
  1643. return _NAN;
  1644. }
  1645. value_t *verb_sin(interpreter_t *state, verb_t *self, value_t *x) {
  1646. if (x->tag == NUMBER)
  1647. return value_new_number(sin(x->val.number));
  1648. return _NAN;
  1649. }
  1650. value_t *verb_square(interpreter_t *state, verb_t *self, value_t *x) {
  1651. if (x->tag == NUMBER)
  1652. return value_new_number(x->val.number * x->val.number);
  1653. return _NAN;
  1654. }
  1655. value_t *verb_negate(interpreter_t *state, verb_t *self, value_t *x) {
  1656. if (x->tag == NUMBER)
  1657. return value_new_number(-x->val.number);
  1658. return _NAN;
  1659. }
  1660. value_t *verb_minus(interpreter_t *state, verb_t *self, value_t *x,
  1661. value_t *y) {
  1662. if ((x->tag == NUMBER || x->tag == CHAR) &&
  1663. (y->tag == NUMBER || y->tag == CHAR)) {
  1664. if (x->tag == CHAR || y->tag == CHAR)
  1665. return value_new_char(get_numeric(x) - get_numeric(y));
  1666. return value_new_number(get_numeric(x) - get_numeric(y));
  1667. }
  1668. return _NAN;
  1669. }
  1670. value_t *verb_atan(interpreter_t *state, verb_t *self, value_t *x) {
  1671. if (x->tag == NUMBER)
  1672. return value_new_number(atan(x->val.number));
  1673. return _NAN;
  1674. }
  1675. value_t *verb_atan2(interpreter_t *state, verb_t *self, value_t *x,
  1676. value_t *y) {
  1677. if (x->tag == NUMBER && y->tag == NUMBER)
  1678. return value_new_number(atan2(x->val.number, y->val.number));
  1679. return _NAN;
  1680. }
  1681. value_t *verb_first(interpreter_t *state, verb_t *self, value_t *x) {
  1682. if (x->tag != ARRAY)
  1683. return x;
  1684. if (!x->val.array->data)
  1685. return state->udf;
  1686. return x->val.array->data[0];
  1687. }
  1688. value_t *verb_times(interpreter_t *state, verb_t *self, value_t *x,
  1689. value_t *y) {
  1690. if ((x->tag == NUMBER || x->tag == CHAR) &&
  1691. (y->tag == NUMBER || y->tag == CHAR)) {
  1692. if (x->tag == CHAR || y->tag == CHAR)
  1693. return value_new_char(get_numeric(x) * get_numeric(y));
  1694. return value_new_number(get_numeric(x) * get_numeric(y));
  1695. }
  1696. return _NAN;
  1697. }
  1698. double lcm(double a, double b) { return (a * b) / gcd(a, b); }
  1699. uint64_t factorial(uint64_t n) {
  1700. uint64_t r = 1;
  1701. while (n > 0)
  1702. r *= n--;
  1703. return r;
  1704. }
  1705. value_t *verb_factorial(interpreter_t *state, verb_t *self, value_t *x) {
  1706. if (x->tag == NUMBER)
  1707. return value_new_number(factorial((uint64_t)fabs(x->val.number)));
  1708. return _NAN;
  1709. }
  1710. value_t *verb_lcm(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1711. if (x->tag == NUMBER && y->tag == NUMBER)
  1712. return value_new_number(lcm(x->val.number, y->val.number));
  1713. return _NAN;
  1714. }
  1715. value_t *verb_double(interpreter_t *state, verb_t *self, value_t *x) {
  1716. if (x->tag == NUMBER)
  1717. return value_new_number(x->val.number * 2);
  1718. return _NAN;
  1719. }
  1720. value_t *verb_replicate(interpreter_t *state, verb_t *self, value_t *x,
  1721. value_t *y) {
  1722. if (x->tag == NUMBER) {
  1723. size_t k = fabs(x->val.number);
  1724. list_t *r = list_new();
  1725. while (k--)
  1726. list_push(r, y);
  1727. return value_new_array(r);
  1728. }
  1729. return state->udf;
  1730. }
  1731. value_t *verb_reciprocal(interpreter_t *state, verb_t *self, value_t *x) {
  1732. if (x->tag == NUMBER)
  1733. return value_new_number(1 / x->val.number);
  1734. return _NAN;
  1735. }
  1736. value_t *verb_divide(interpreter_t *state, verb_t *self, value_t *x,
  1737. value_t *y) {
  1738. if (x->tag == NUMBER && y->tag == NUMBER) {
  1739. double ny = y->val.number;
  1740. if (ny == 0)
  1741. return INF;
  1742. return value_new_number(x->val.number / ny);
  1743. }
  1744. return _NAN;
  1745. }
  1746. double npower(double base, int n) {
  1747. if (n < 0)
  1748. return npower(1 / base, -n);
  1749. else if (n == 0)
  1750. return 1.0;
  1751. else if (n == 1)
  1752. return base;
  1753. else if (n % 2)
  1754. return base * npower(base * base, n / 2);
  1755. else
  1756. return npower(base * base, n / 2);
  1757. }
  1758. double nroot(double base, int n) {
  1759. if (n == 1)
  1760. return base;
  1761. else if (n <= 0 || base < 0)
  1762. return NAN;
  1763. else {
  1764. double delta, x = base / n;
  1765. do {
  1766. delta = (base / npower(x, n - 1) - x) / n;
  1767. x += delta;
  1768. } while (fabs(delta) >= 1e-8);
  1769. return x;
  1770. }
  1771. }
  1772. value_t *verb_sqrt(interpreter_t *state, verb_t *self, value_t *x) {
  1773. if (x->tag == NUMBER)
  1774. return value_new_number(sqrt(x->val.number));
  1775. return _NAN;
  1776. }
  1777. value_t *verb_root(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1778. if (x->tag == NUMBER && y->tag == NUMBER)
  1779. return value_new_number(nroot(y->val.number, x->val.number));
  1780. return _NAN;
  1781. }
  1782. value_t *verb_halve(interpreter_t *state, verb_t *self, value_t *x) {
  1783. if (x->tag == NUMBER)
  1784. return value_new_number(x->val.number / 2);
  1785. return _NAN;
  1786. }
  1787. value_t *verb_idivide(interpreter_t *state, verb_t *self, value_t *x,
  1788. value_t *y) {
  1789. if ((x->tag == NUMBER || x->tag == CHAR) &&
  1790. (y->tag == NUMBER || y->tag == CHAR)) {
  1791. double a = get_numeric(x);
  1792. double b = get_numeric(y);
  1793. if (x->tag == CHAR || y->tag == CHAR)
  1794. return b == 0 ? state->udf : value_new_char(fabs(trunc(a / b)));
  1795. if (b == 0)
  1796. return INF;
  1797. return value_new_number(trunc(a / b));
  1798. }
  1799. return _NAN;
  1800. }
  1801. value_t *verb_enlist(interpreter_t *state, verb_t *self, value_t *x);
  1802. value_t *verb_pred(interpreter_t *state, verb_t *self, value_t *x);
  1803. value_t *verb_range(interpreter_t *state, verb_t *self, value_t *x, value_t *y);
  1804. value_t *verb_enum(interpreter_t *state, verb_t *self, value_t *x) {
  1805. if (value_equals(x, NUMS[1]))
  1806. return verb_enlist(state, NULL, NUMS[0]);
  1807. else if (value_equals(x, NUMS[0]))
  1808. return state->unit;
  1809. return verb_range(state, self, NUMS[0], verb_pred(state, self, x));
  1810. }
  1811. value_t *verb_mod(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1812. if (x->tag == NUMBER && y->tag == NUMBER) {
  1813. double ny = y->val.number;
  1814. if (ny == 0)
  1815. return _NAN;
  1816. return value_new_number(fmod(x->val.number, ny));
  1817. }
  1818. return _NAN;
  1819. }
  1820. value_t *verb_take(interpreter_t *state, verb_t *self, value_t *x, value_t *y);
  1821. value_t *verb_drop(interpreter_t *state, verb_t *self, value_t *x, value_t *y);
  1822. bool is_bad_num(double v) {
  1823. return isnan(v) || v == INFINITY || v == -INFINITY;
  1824. }
  1825. value_t *verb_odometer(interpreter_t *state, verb_t *self, value_t *x) {
  1826. if (x->tag != ARRAY)
  1827. x = verb_enlist(state, NULL, x);
  1828. else if (x->val.array->length < 2)
  1829. return state->udf;
  1830. size_t p = 1;
  1831. size_t xl = x->val.array->length;
  1832. for (size_t i = 0; i < xl; i++) {
  1833. value_t *it = x->val.array->data[i];
  1834. if (it->tag != NUMBER || is_bad_num(it->val.number))
  1835. return state->udf;
  1836. p *= (size_t)(it->val.number);
  1837. }
  1838. if (p < 1)
  1839. return state->unit;
  1840. uint64_t *lims = malloc_checked_atomic(sizeof(uint64_t) * xl);
  1841. for (size_t i = 0; i < xl; i++)
  1842. lims[i] = (size_t)(((value_t *)x->val.array->data[i])->val.number);
  1843. uint64_t **z = malloc_checked(sizeof(uint64_t *) * p);
  1844. for (size_t i = 0; i < p; i++)
  1845. z[i] = malloc_checked_atomic(sizeof(uint64_t) * xl);
  1846. for (size_t i = 0; i < p - 1; i++) {
  1847. uint64_t *r = z[i];
  1848. uint64_t *s = z[i + 1];
  1849. bool carry = true;
  1850. for (size_t j = 0; j < xl; j++) {
  1851. uint64_t a = xl - 1 - j;
  1852. s[a] = r[a];
  1853. if (carry) {
  1854. s[a]++;
  1855. carry = false;
  1856. }
  1857. if (s[a] >= lims[a]) {
  1858. s[a] = 0;
  1859. carry = true;
  1860. }
  1861. }
  1862. }
  1863. GC_FREE(lims);
  1864. list_t *r = list_newk(p);
  1865. for (size_t i = 0; i < p; i++) {
  1866. list_t *rw = list_newk(xl);
  1867. for (size_t j = 0; j < xl; j++)
  1868. rw->data[j] = value_new_number(z[i][j]);
  1869. r->data[i] = value_new_array(rw);
  1870. GC_FREE(z[i]);
  1871. }
  1872. GC_FREE(z);
  1873. return value_new_array(r);
  1874. }
  1875. value_t *verb_chunks(interpreter_t *state, verb_t *self, value_t *x,
  1876. value_t *y) {
  1877. if (x->tag != NUMBER)
  1878. return state->udf;
  1879. if (y->tag != ARRAY)
  1880. y = verb_enlist(state, NULL, y);
  1881. else if (!y->val.array->data)
  1882. return y;
  1883. list_t *r = list_new();
  1884. size_t cl = fabs(x->val.number);
  1885. for (size_t i = 0; i < y->val.array->length; i += cl)
  1886. list_push(r, verb_take(state, NULL, value_new_number(cl),
  1887. verb_drop(state, NULL, value_new_number(i), y)));
  1888. return value_new_array(r);
  1889. }
  1890. value_t *verb_exp(interpreter_t *state, verb_t *self, value_t *x) {
  1891. if (x->tag == NUMBER)
  1892. return value_new_number(exp(x->val.number));
  1893. return _NAN;
  1894. }
  1895. value_t *verb_power(interpreter_t *state, verb_t *self, value_t *x,
  1896. value_t *y) {
  1897. if (x->tag == NUMBER && y->tag == NUMBER)
  1898. return value_new_number(pow(x->val.number, y->val.number));
  1899. return _NAN;
  1900. }
  1901. value_t *verb_nlog(interpreter_t *state, verb_t *self, value_t *x) {
  1902. if (x->tag == NUMBER)
  1903. return value_new_number(log(x->val.number));
  1904. return _NAN;
  1905. }
  1906. value_t *verb_log(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1907. if (x->tag == NUMBER && y->tag == NUMBER)
  1908. return value_new_number(log(y->val.number) / log(x->val.number));
  1909. return _NAN;
  1910. }
  1911. int bits_needed(uint32_t value) {
  1912. int bits = 0;
  1913. for (int bit_test = 16; bit_test > 0; bit_test >>= 1) {
  1914. if (value >> bit_test != 0) {
  1915. bits += bit_test;
  1916. value >>= bit_test;
  1917. }
  1918. }
  1919. return bits + value;
  1920. }
  1921. value_t *verb_bits(interpreter_t *state, verb_t *self, value_t *x) {
  1922. if (x->tag == NUMBER) {
  1923. int n = x->val.number;
  1924. int bk = bits_needed(n);
  1925. list_t *r = list_newk(bk);
  1926. for (int i = 0; i < bk; i++)
  1927. if ((n & (1 << i)) >> i)
  1928. r->data[i] = NUMS[1];
  1929. else
  1930. r->data[i] = NUMS[0];
  1931. return value_new_array(r);
  1932. }
  1933. return state->udf;
  1934. }
  1935. value_t *verb_reverse(interpreter_t *state, verb_t *self, value_t *x);
  1936. value_t *verb_base(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  1937. if (x->tag == NUMBER && y->tag == NUMBER) {
  1938. size_t v = fabs(y->val.number);
  1939. size_t b = fabs(x->val.number);
  1940. if (b < 2)
  1941. return state->udf;
  1942. list_t *r = list_new();
  1943. while (v > 0) {
  1944. list_push(r, value_new_number(v % b));
  1945. v /= b;
  1946. }
  1947. return verb_reverse(state, NULL, value_new_array(r));
  1948. }
  1949. return state->udf;
  1950. }
  1951. ssize_t indexOf(list_t *l, value_t *x) {
  1952. if (!l->data)
  1953. return -1;
  1954. for (size_t i = 0; i < l->length; i++)
  1955. if (value_equals(l->data[i], x))
  1956. return i;
  1957. return -1;
  1958. }
  1959. value_t *verb_group(interpreter_t *state, verb_t *self, value_t *x) {
  1960. if (x->tag != ARRAY)
  1961. x = verb_enlist(state, NULL, x);
  1962. else if (!x->val.array->data)
  1963. return x;
  1964. list_t *r = list_new();
  1965. list_t *is = list_new();
  1966. for (size_t i = 0; i < x->val.array->length; i++) {
  1967. value_t *v = x->val.array->data[i];
  1968. ssize_t n = indexOf(is, v);
  1969. if (n < 0) {
  1970. list_push(r, verb_enlist(state, NULL, value_new_number(i)));
  1971. list_push(is, v);
  1972. } else {
  1973. value_t *tmp = list_index(r, n);
  1974. list_push(tmp->val.array, value_new_number(i));
  1975. }
  1976. }
  1977. GC_FREE(is->data);
  1978. GC_FREE(is);
  1979. return value_new_array(r);
  1980. }
  1981. value_t *verb_buckets(interpreter_t *state, verb_t *self, value_t *x,
  1982. value_t *y) {
  1983. if (x->tag != ARRAY)
  1984. x = verb_enlist(state, NULL, x);
  1985. else if (!x->val.array->data)
  1986. return y;
  1987. if (y->tag != ARRAY)
  1988. y = verb_enlist(state, NULL, x);
  1989. else if (!y->val.array->data)
  1990. return y;
  1991. list_t *r = list_new();
  1992. size_t mx = 0;
  1993. for (size_t i = 0; i < x->val.array->length; i++) {
  1994. value_t *v = x->val.array->data[i];
  1995. if (v->tag != NUMBER)
  1996. break;
  1997. ssize_t j = v->val.number;
  1998. if (j >= 0 && j > mx)
  1999. mx = j;
  2000. }
  2001. for (size_t i = 0; i < mx + 1; i++)
  2002. list_push(r, list_new());
  2003. if (!r->data) {
  2004. GC_FREE(r);
  2005. return state->unit;
  2006. }
  2007. for (size_t i = 0; i < x->val.array->length; i++) {
  2008. if (i >= y->val.array->length)
  2009. break;
  2010. value_t *v = x->val.array->data[i];
  2011. if (v->tag != NUMBER)
  2012. break;
  2013. ssize_t j = v->val.number;
  2014. if (j >= 0) {
  2015. list_t *b = list_index(r, j);
  2016. if (b)
  2017. list_push(b, y->val.array->data[i]);
  2018. }
  2019. }
  2020. if (x->val.array->length < y->val.array->length) {
  2021. list_t *lb = list_new();
  2022. for (size_t i = x->val.array->length; i < y->val.array->length; i++)
  2023. list_push(lb, y->val.array->data[i]);
  2024. list_push(r, lb);
  2025. }
  2026. for (size_t i = 0; i < r->length; i++)
  2027. r->data[i] = value_new_array(r->data[i]);
  2028. return value_new_array(r);
  2029. }
  2030. value_t *verb_equals(interpreter_t *state, verb_t *self, value_t *x,
  2031. value_t *y) {
  2032. return value_equals(x, y) ? NUMS[1] : NUMS[0];
  2033. }
  2034. value_t *verb_permute(interpreter_t *state, verb_t *self, value_t *x) {
  2035. if (x->tag != ARRAY || x->val.array->length < 2)
  2036. return x;
  2037. list_t *permutation = list_copy(x->val.array);
  2038. size_t length = permutation->length;
  2039. list_t *result = list_new();
  2040. list_push(result, list_copy(permutation));
  2041. list_t *c = list_new();
  2042. for (size_t i = 0; i < length; i++) {
  2043. size_t *n = malloc_checked_atomic(sizeof(size_t));
  2044. list_push(c, n);
  2045. }
  2046. size_t k;
  2047. list_t *p;
  2048. size_t i = 0;
  2049. while (i < length) {
  2050. size_t *n = list_index(c, i);
  2051. if ((*n) < i) {
  2052. k = i % 2 && (*n);
  2053. p = list_index(permutation, i);
  2054. list_set(permutation, i, list_index(permutation, k));
  2055. list_set(permutation, k, p);
  2056. *n = (*n) + 1;
  2057. i = 1;
  2058. list_push(result, list_copy(permutation));
  2059. } else {
  2060. *n = 0;
  2061. i++;
  2062. }
  2063. }
  2064. for (size_t i = 0; i < c->length; i++)
  2065. GC_FREE(c->data[i]);
  2066. GC_FREE(c->data);
  2067. GC_FREE(c);
  2068. GC_FREE(permutation->data);
  2069. GC_FREE(permutation);
  2070. for (size_t i = 0; i < result->length; i++)
  2071. result->data[i] = value_new_array(result->data[i]);
  2072. return value_new_array(result);
  2073. }
  2074. value_t *verb_occurences(interpreter_t *state, verb_t *self, value_t *x) {
  2075. if (x->tag != ARRAY)
  2076. x = verb_enlist(state, NULL, x);
  2077. else if (!x->val.array->data)
  2078. return x;
  2079. list_t *table = list_new();
  2080. list_t *r = list_new();
  2081. for (size_t i = 0; i < x->val.array->length; i++) {
  2082. bool f = false;
  2083. value_t *it = x->val.array->data[i];
  2084. for (size_t j = 0; j < table->length; j++) {
  2085. list_t *p = table->data[j];
  2086. if (value_equals(p->data[0], it)) {
  2087. size_t *n = p->data[1];
  2088. *n = (*n) + 1;
  2089. list_push(r, value_new_number(*n));
  2090. f = true;
  2091. break;
  2092. }
  2093. }
  2094. if (!f) {
  2095. list_t *p = list_newk(2);
  2096. p->data[0] = it;
  2097. size_t *n = malloc_checked_atomic(sizeof(size_t));
  2098. p->data[1] = n;
  2099. list_push(table, p);
  2100. list_push(r, NUMS[0]);
  2101. }
  2102. }
  2103. for (size_t i = 0; i < table->length; i++) {
  2104. list_t *p = table->data[i];
  2105. GC_FREE(p->data[1]);
  2106. GC_FREE(p->data);
  2107. GC_FREE(p);
  2108. }
  2109. GC_FREE(table->data);
  2110. GC_FREE(table);
  2111. return value_new_array(r);
  2112. }
  2113. value_t *verb_mask(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2114. if (x->tag != ARRAY)
  2115. x = verb_enlist(state, NULL, x);
  2116. else if (!x->val.array->data)
  2117. return x;
  2118. if (y->tag != ARRAY)
  2119. y = verb_enlist(state, NULL, y);
  2120. list_t *r = list_new();
  2121. value_t *l = value_new_number(y->val.array->length);
  2122. size_t n = 0;
  2123. size_t k = x->val.array->length;
  2124. for (size_t i = 0; i < k; i++) {
  2125. value_t *s = verb_take(state, NULL, l,
  2126. verb_drop(state, NULL, value_new_number(i), x));
  2127. if (value_equals(s, y)) {
  2128. n++;
  2129. for (size_t j = 0; j < l->val.number; j++, i++)
  2130. list_push(r, value_new_number(n));
  2131. i--;
  2132. } else
  2133. list_push(r, NUMS[0]);
  2134. }
  2135. return value_new_array(r);
  2136. }
  2137. value_t *verb_classify(interpreter_t *state, verb_t *self, value_t *x) {
  2138. if (x->tag != ARRAY)
  2139. x = verb_enlist(state, NULL, x);
  2140. else if (!x->val.array->data)
  2141. return x;
  2142. list_t *table = list_new();
  2143. list_t *r = list_new();
  2144. for (size_t i = 0; i < x->val.array->length; i++) {
  2145. bool f = false;
  2146. value_t *it = x->val.array->data[i];
  2147. for (size_t j = 0; j < table->length; j++) {
  2148. list_t *p = table->data[j];
  2149. if (value_equals(p->data[0], it)) {
  2150. size_t *n = p->data[1];
  2151. list_push(r, value_new_number(*n));
  2152. f = true;
  2153. break;
  2154. }
  2155. }
  2156. if (!f) {
  2157. list_t *p = list_newk(2);
  2158. p->data[0] = it;
  2159. size_t *n = malloc_checked_atomic(sizeof(size_t));
  2160. *n = i++;
  2161. p->data[1] = n;
  2162. list_push(table, p);
  2163. list_push(r, value_new_number(*n));
  2164. }
  2165. }
  2166. for (size_t i = 0; i < table->length; i++) {
  2167. list_t *p = table->data[i];
  2168. GC_FREE(p->data[1]);
  2169. GC_FREE(p->data);
  2170. GC_FREE(p);
  2171. }
  2172. GC_FREE(table->data);
  2173. GC_FREE(table);
  2174. return value_new_array(r);
  2175. }
  2176. value_t *verb_unbits(interpreter_t *state, verb_t *self, value_t *x) {
  2177. if (x->tag != ARRAY)
  2178. x = verb_enlist(state, NULL, x);
  2179. int n = 0;
  2180. for (size_t i = 0; i < x->val.array->length; i++) {
  2181. if (value_is_truthy(x->val.array->data[i]))
  2182. n |= (int)1 << (int)i;
  2183. else
  2184. n &= ~((int)1 << (int)i);
  2185. }
  2186. return value_new_number(n);
  2187. }
  2188. value_t *verb_unbase(interpreter_t *state, verb_t *self, value_t *x,
  2189. value_t *y) {
  2190. if (x->tag == NUMBER) {
  2191. size_t b = fabs(x->val.number);
  2192. if (b < 2)
  2193. return state->udf;
  2194. if (y->tag != ARRAY)
  2195. y = verb_enlist(state, NULL, y);
  2196. size_t n = 0;
  2197. if (!y->val.array->data)
  2198. return state->udf;
  2199. for (size_t i = 0; i < y->val.array->length; i++) {
  2200. value_t *v = y->val.array->data[i];
  2201. if (v->tag != NUMBER)
  2202. break;
  2203. size_t k = fabs(v->val.number);
  2204. n = n * b + k;
  2205. }
  2206. return value_new_number(n);
  2207. }
  2208. return state->udf;
  2209. }
  2210. value_t *verb_not(interpreter_t *state, verb_t *self, value_t *x) {
  2211. return value_is_truthy(x) ? NUMS[0] : NUMS[1];
  2212. }
  2213. value_t *verb_not_equals(interpreter_t *state, verb_t *self, value_t *x,
  2214. value_t *y) {
  2215. return !value_equals(x, y) ? NUMS[1] : NUMS[0];
  2216. }
  2217. value_t *verb_pred(interpreter_t *state, verb_t *self, value_t *x) {
  2218. if (x->tag == NUMBER)
  2219. return value_new_number(x->val.number - 1);
  2220. else if (x->tag == CHAR)
  2221. return value_new_char(x->val._char - 1);
  2222. return _NAN;
  2223. }
  2224. value_t *verb_less(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2225. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2226. (y->tag == NUMBER || y->tag == CHAR)) {
  2227. if (get_numeric(x) < get_numeric(y))
  2228. return NUMS[1];
  2229. return NUMS[0];
  2230. }
  2231. return _NAN;
  2232. }
  2233. value_t *verb_floor(interpreter_t *state, verb_t *self, value_t *x) {
  2234. if (x->tag == NUMBER)
  2235. return value_new_number(floor(x->val.number));
  2236. return _NAN;
  2237. }
  2238. int _compare_up(const void *a, const void *b) {
  2239. value_t *x = (*(list_t **)a)->data[0];
  2240. value_t *y = (*(list_t **)b)->data[0];
  2241. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2242. (y->tag == NUMBER || y->tag == CHAR)) {
  2243. if (get_numeric(x) > get_numeric(y))
  2244. return 1;
  2245. else if (get_numeric(x) < get_numeric(y))
  2246. return -1;
  2247. return 0;
  2248. }
  2249. return 0;
  2250. }
  2251. int _compare_down(const void *a, const void *b) {
  2252. value_t *x = (*(list_t **)a)->data[0];
  2253. value_t *y = (*(list_t **)b)->data[0];
  2254. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2255. (y->tag == NUMBER || y->tag == CHAR)) {
  2256. if (get_numeric(x) > get_numeric(y))
  2257. return -1;
  2258. else if (get_numeric(x) < get_numeric(y))
  2259. return 1;
  2260. return 0;
  2261. }
  2262. return 0;
  2263. }
  2264. value_t *_grade(value_t *x, bool down) {
  2265. if (x->tag != ARRAY || x->val.array->length < 2)
  2266. return x;
  2267. list_t *ps = list_newk(x->val.array->length);
  2268. for (size_t i = 0; i < x->val.array->length; i++) {
  2269. list_t *p = list_newk(2);
  2270. p->data[0] = x->val.array->data[i];
  2271. p->data[1] = value_new_number(i);
  2272. ps->data[i] = p;
  2273. }
  2274. qsort(ps->data, ps->length, sizeof(void *),
  2275. down ? _compare_down : _compare_up);
  2276. for (size_t i = 0; i < ps->length; i++) {
  2277. list_t *p = ps->data[i];
  2278. ps->data[i] = p->data[1];
  2279. GC_FREE(p->data);
  2280. GC_FREE(p);
  2281. }
  2282. return value_new_array(ps);
  2283. }
  2284. value_t *verb_gradedown(interpreter_t *state, verb_t *self, value_t *x) {
  2285. return _grade(x, true);
  2286. }
  2287. value_t *verb_nudge_left(interpreter_t *state, verb_t *self, value_t *x,
  2288. value_t *y) {
  2289. if (y->tag != ARRAY)
  2290. return verb_enlist(state, NULL, x);
  2291. else if (!y->val.array->data)
  2292. return y;
  2293. else if (y->val.array->length < 2)
  2294. return verb_enlist(state, NULL, x);
  2295. list_t *r = list_new();
  2296. for (size_t i = 1; i < y->val.array->length; i++)
  2297. list_push(r, y->val.array->data[i]);
  2298. list_push(r, x);
  2299. return value_new_array(r);
  2300. }
  2301. value_t *verb_lesseq(interpreter_t *state, verb_t *self, value_t *x,
  2302. value_t *y) {
  2303. if (value_equals(x, y))
  2304. return NUMS[1];
  2305. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2306. (y->tag == NUMBER || y->tag == CHAR)) {
  2307. if (get_numeric(x) < get_numeric(y))
  2308. return NUMS[1];
  2309. return NUMS[0];
  2310. }
  2311. return _NAN;
  2312. }
  2313. value_t *verb_succ(interpreter_t *state, verb_t *self, value_t *x) {
  2314. if (x->tag == NUMBER)
  2315. return value_new_number(x->val.number + 1);
  2316. else if (x->tag == CHAR)
  2317. return value_new_char(x->val._char + 1);
  2318. return _NAN;
  2319. }
  2320. value_t *verb_ceil(interpreter_t *state, verb_t *self, value_t *x) {
  2321. if (x->tag == NUMBER)
  2322. return value_new_number(ceil(x->val.number));
  2323. return _NAN;
  2324. }
  2325. value_t *verb_greater(interpreter_t *state, verb_t *self, value_t *x,
  2326. value_t *y) {
  2327. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2328. (y->tag == NUMBER || y->tag == CHAR)) {
  2329. if (get_numeric(x) > get_numeric(y))
  2330. return NUMS[1];
  2331. return NUMS[0];
  2332. }
  2333. return _NAN;
  2334. }
  2335. value_t *verb_greatereq(interpreter_t *state, verb_t *self, value_t *x,
  2336. value_t *y) {
  2337. if (value_equals(x, y))
  2338. return NUMS[1];
  2339. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2340. (y->tag == NUMBER || y->tag == CHAR)) {
  2341. if (get_numeric(x) > get_numeric(y))
  2342. return NUMS[1];
  2343. return NUMS[0];
  2344. }
  2345. return _NAN;
  2346. }
  2347. value_t *verb_gradeup(interpreter_t *state, verb_t *self, value_t *x) {
  2348. return _grade(x, false);
  2349. }
  2350. value_t *verb_nudge_right(interpreter_t *state, verb_t *self, value_t *x,
  2351. value_t *y) {
  2352. if (y->tag != ARRAY)
  2353. return verb_enlist(state, NULL, x);
  2354. else if (!y->val.array->data)
  2355. return y;
  2356. else if (y->val.array->length < 2)
  2357. return verb_enlist(state, NULL, x);
  2358. list_t *r = list_new();
  2359. list_push(r, x);
  2360. for (size_t i = 0; i < y->val.array->length - 1; i++)
  2361. list_push(r, y->val.array->data[i]);
  2362. return value_new_array(r);
  2363. }
  2364. value_t *verb_enlist(interpreter_t *state, verb_t *self, value_t *x) {
  2365. list_t *l = list_new();
  2366. list_push(l, x);
  2367. return value_new_array(l);
  2368. }
  2369. value_t *verb_enpair(interpreter_t *state, verb_t *self, value_t *x,
  2370. value_t *y);
  2371. value_t *verb_join(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2372. list_t *l;
  2373. if (x->tag == ARRAY && y->tag == ARRAY) {
  2374. if (!x->val.array->data && !y->val.array->data)
  2375. return state->unit;
  2376. else if (!x->val.array->data)
  2377. return y;
  2378. else if (!y->val.array->data)
  2379. return x;
  2380. l = list_newk(x->val.array->length + y->val.array->length);
  2381. size_t lp = 0;
  2382. for (size_t i = 0; i < x->val.array->length; i++)
  2383. l->data[lp++] = x->val.array->data[i];
  2384. for (size_t i = 0; i < y->val.array->length; i++)
  2385. l->data[lp++] = y->val.array->data[i];
  2386. } else if (x->tag == ARRAY && y->tag != ARRAY) {
  2387. if (!x->val.array->data)
  2388. return verb_enlist(state, NULL, y);
  2389. l = list_newk(x->val.array->length + 1);
  2390. size_t lp = 0;
  2391. for (size_t i = 0; i < x->val.array->length; i++)
  2392. l->data[lp++] = x->val.array->data[i];
  2393. l->data[lp++] = y;
  2394. } else if (x->tag != ARRAY && y->tag == ARRAY) {
  2395. if (!y->val.array->data)
  2396. return verb_enlist(state, NULL, x);
  2397. l = list_newk(y->val.array->length + 1);
  2398. size_t lp = 0;
  2399. l->data[lp++] = x;
  2400. for (size_t i = 0; i < y->val.array->length; i++)
  2401. l->data[lp++] = y->val.array->data[i];
  2402. } else
  2403. return verb_enpair(state, NULL, x, y);
  2404. return value_new_array(l);
  2405. }
  2406. value_t *verb_enpair(interpreter_t *state, verb_t *self, value_t *x,
  2407. value_t *y) {
  2408. list_t *l = list_newk(2);
  2409. l->data[0] = x;
  2410. l->data[1] = y;
  2411. return value_new_array(l);
  2412. }
  2413. value_t *verb_selfref1(interpreter_t *state, verb_t *self, value_t *x) {
  2414. verb_t *v;
  2415. if (state->args->data)
  2416. v = list_index(list_index(state->args, -1), -1);
  2417. else if (state->selfrefs->data)
  2418. v = list_index(state->selfrefs, -1);
  2419. else
  2420. return state->udf;
  2421. return each_rank(state, v, x, 0, v->rank[0]);
  2422. }
  2423. value_t *verb_selfref2(interpreter_t *state, verb_t *self, value_t *x,
  2424. value_t *y) {
  2425. verb_t *v;
  2426. if (state->args->data)
  2427. v = list_index(list_index(state->args, -1), -1);
  2428. else if (state->selfrefs->data)
  2429. v = list_index(state->selfrefs, -1);
  2430. else
  2431. return state->udf;
  2432. return together(state, v, x, y, 0, 0, v->rank[1], v->rank[2]);
  2433. }
  2434. value_t *verb_take(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2435. if (x->tag == NUMBER) {
  2436. if (y->tag != ARRAY) {
  2437. if (x->val.number == 0)
  2438. return state->unit;
  2439. else
  2440. return x;
  2441. }
  2442. if (x->val.number == 0 || !y->val.array->data)
  2443. return state->unit;
  2444. bool rev = x->val.number < 0;
  2445. size_t k = (size_t)fabs(x->val.number);
  2446. list_t *r = list_newk(y->val.array->length < k ? y->val.array->length : k);
  2447. size_t p = 0;
  2448. if (rev)
  2449. for (ssize_t i = k; i > 0; i--) {
  2450. value_t *v = list_index(y->val.array, -i);
  2451. if (!v)
  2452. continue;
  2453. r->data[p++] = v;
  2454. }
  2455. else
  2456. for (size_t i = 0; i < y->val.array->length && k; i++, k--)
  2457. r->data[p++] = y->val.array->data[i];
  2458. return value_new_array(r);
  2459. }
  2460. return state->udf;
  2461. }
  2462. value_t *verb_where(interpreter_t *state, verb_t *self, value_t *x) {
  2463. if (x->tag != ARRAY)
  2464. x = verb_enlist(state, NULL, x);
  2465. else if (!x->val.array->data)
  2466. return x;
  2467. list_t *r = list_new();
  2468. for (size_t i = 0; i < x->val.array->length; i++) {
  2469. value_t *a = x->val.array->data[i];
  2470. if (a->tag != NUMBER)
  2471. break;
  2472. size_t k = fabs(a->val.number);
  2473. for (size_t j = 0; j < k; j++)
  2474. list_push(r, value_new_number(i));
  2475. }
  2476. return value_new_array(r);
  2477. }
  2478. value_t *verb_copy(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2479. if (x->tag != ARRAY)
  2480. x = verb_enlist(state, NULL, x);
  2481. if (y->tag != ARRAY)
  2482. y = verb_enlist(state, NULL, y);
  2483. list_t *tx = x->val.array;
  2484. list_t *ty = y->val.array;
  2485. if (!tx->data || !ty->data)
  2486. return state->unit;
  2487. list_t *r = list_new();
  2488. for (size_t i = 0; i < tx->length; i++) {
  2489. value_t *a = tx->data[i];
  2490. value_t *b = ty->data[i >= ty->length ? ty->length - 1 : i];
  2491. if (b->tag != NUMBER)
  2492. break;
  2493. size_t k = fabs(b->val.number);
  2494. for (size_t i = 0; i < k; i++)
  2495. list_push(r, a);
  2496. }
  2497. return value_new_array(r);
  2498. }
  2499. value_t *verb_nub(interpreter_t *state, verb_t *self, value_t *x) {
  2500. if (x->tag != ARRAY || !x->val.array->data)
  2501. return x;
  2502. list_t *n = list_newk(x->val.array->length);
  2503. list_t *r = list_new();
  2504. for (size_t i = 0; i < x->val.array->length; i++) {
  2505. bool u = true;
  2506. for (size_t j = 0; j < r->length; j++)
  2507. if (value_equals(x->val.array->data[i], r->data[j])) {
  2508. u = false;
  2509. break;
  2510. }
  2511. if (u)
  2512. list_push(r, x->val.array->data[i]);
  2513. n->data[i] = u ? NUMS[1] : NUMS[0];
  2514. }
  2515. GC_FREE(r->data);
  2516. GC_FREE(r);
  2517. return value_new_array(n);
  2518. }
  2519. value_t *verb_drop(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2520. if (x->tag == NUMBER) {
  2521. if (y->tag != ARRAY) {
  2522. if (x->val.number == 0)
  2523. return y;
  2524. else
  2525. return state->unit;
  2526. }
  2527. if (x->val.number == 0)
  2528. return y;
  2529. if (!y->val.array->data)
  2530. return state->unit;
  2531. bool rev = x->val.number < 0;
  2532. size_t k = (size_t)fabs(x->val.number);
  2533. if (k >= y->val.array->length)
  2534. return state->unit;
  2535. if (rev) {
  2536. size_t l = y->val.array->length;
  2537. if (k >= l)
  2538. return state->unit;
  2539. return verb_take(state, NULL, value_new_number(l - k), y);
  2540. }
  2541. list_t *r = list_newk(y->val.array->length - k);
  2542. size_t rp = 0;
  2543. for (size_t i = k; i < y->val.array->length; i++)
  2544. r->data[rp++] = y->val.array->data[i];
  2545. return value_new_array(r);
  2546. }
  2547. return state->udf;
  2548. }
  2549. value_t *verb_unique(interpreter_t *state, verb_t *self, value_t *x) {
  2550. if (x->tag != ARRAY || !x->val.array->data)
  2551. return x;
  2552. list_t *r = list_new();
  2553. for (size_t i = 0; i < x->val.array->length; i++) {
  2554. bool u = true;
  2555. for (size_t j = 0; j < r->length; j++)
  2556. if (value_equals(x->val.array->data[i], r->data[j])) {
  2557. u = false;
  2558. break;
  2559. }
  2560. if (u)
  2561. list_push(r, x->val.array->data[i]);
  2562. }
  2563. return value_new_array(r);
  2564. }
  2565. value_t *verb_find(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2566. if (y->tag != ARRAY)
  2567. y = verb_enlist(state, self, y);
  2568. else if (!y->val.array->data)
  2569. return state->unit;
  2570. list_t *r = list_new();
  2571. for (size_t i = 0; i < y->val.array->length; i++)
  2572. if (value_equals(y->val.array->data[i], x))
  2573. list_push(r, value_new_number(i));
  2574. return value_new_array(r);
  2575. }
  2576. value_t *verb_indexof(interpreter_t *state, verb_t *self, value_t *x,
  2577. value_t *y) {
  2578. if (y->tag != ARRAY)
  2579. y = verb_enlist(state, self, y);
  2580. else if (!y->val.array->data)
  2581. return state->unit;
  2582. ssize_t n = indexOf(y->val.array, x);
  2583. if (n < 0)
  2584. n = y->val.array->length;
  2585. return value_new_number(n);
  2586. }
  2587. value_t *verb_count(interpreter_t *state, verb_t *self, value_t *x) {
  2588. if (x->tag != ARRAY)
  2589. return NUMS[1];
  2590. return value_new_number(x->val.array->length);
  2591. }
  2592. void flatten(value_t *x, list_t *r) {
  2593. if (x->tag == ARRAY)
  2594. for (size_t i = 0; i < x->val.array->length; i++)
  2595. flatten(x->val.array->data[i], r);
  2596. else
  2597. list_push(r, x);
  2598. }
  2599. value_t *verb_flatten(interpreter_t *state, verb_t *self, value_t *x) {
  2600. if (x->tag != ARRAY || !x->val.array->data)
  2601. return x;
  2602. list_t *r = list_new();
  2603. flatten(x, r);
  2604. return value_new_array(r);
  2605. }
  2606. value_t *verb_minand(interpreter_t *state, verb_t *self, value_t *x,
  2607. value_t *y) {
  2608. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2609. (y->tag == NUMBER || y->tag == CHAR)) {
  2610. if (get_numeric(x) < get_numeric(y))
  2611. return x;
  2612. return y;
  2613. }
  2614. return _NAN;
  2615. }
  2616. value_t *verb_reverse(interpreter_t *state, verb_t *self, value_t *x) {
  2617. if (x->tag != ARRAY)
  2618. return x;
  2619. if (x->val.array->length < 2)
  2620. return x;
  2621. list_t *r = list_newk(x->val.array->length);
  2622. size_t rp = 0;
  2623. for (ssize_t i = x->val.array->length - 1; i >= 0; i--)
  2624. r->data[rp++] = x->val.array->data[i];
  2625. return value_new_array(r);
  2626. }
  2627. value_t *verb_maxor(interpreter_t *state, verb_t *self, value_t *x,
  2628. value_t *y) {
  2629. if ((x->tag == NUMBER || x->tag == CHAR) &&
  2630. (y->tag == NUMBER || y->tag == CHAR)) {
  2631. if (get_numeric(x) > get_numeric(y))
  2632. return x;
  2633. return y;
  2634. }
  2635. return _NAN;
  2636. }
  2637. value_t *verb_rotate(interpreter_t *state, verb_t *self, value_t *x,
  2638. value_t *y) {
  2639. if (y->tag != ARRAY || y->val.array->length < 2)
  2640. return x;
  2641. if (x->tag != NUMBER)
  2642. return state->udf;
  2643. bool rev = x->val.number < 0;
  2644. size_t k = fabs(x->val.number);
  2645. list_t *r = list_copy(y->val.array);
  2646. for (size_t i = 0; i < k; i++) {
  2647. value_t *v;
  2648. if (rev) {
  2649. v = r->data[0];
  2650. for (size_t j = 0; j < r->length - 1; j++)
  2651. r->data[j] = r->data[j + 1];
  2652. r->data[r->length - 1] = v;
  2653. } else {
  2654. v = r->data[r->length - 1];
  2655. for (size_t j = r->length - 1; j > 0; j--)
  2656. r->data[j] = r->data[j - 1];
  2657. r->data[0] = v;
  2658. }
  2659. }
  2660. return value_new_array(r);
  2661. }
  2662. value_t *verb_windows(interpreter_t *state, verb_t *self, value_t *x,
  2663. value_t *y) {
  2664. if (y->tag != ARRAY)
  2665. y = verb_enlist(state, NULL, y);
  2666. else if (!y->val.array->data)
  2667. return y;
  2668. size_t k = fabs(x->val.number);
  2669. size_t l = y->val.array->length;
  2670. list_t *r = list_new();
  2671. for (size_t i = 0; i < l; i++) {
  2672. if (i + k > l)
  2673. break;
  2674. list_push(r, verb_take(state, NULL, value_new_number(k),
  2675. verb_drop(state, NULL, value_new_number(i), y)));
  2676. }
  2677. return value_new_array(r);
  2678. }
  2679. size_t depthOf(value_t *x, size_t d) {
  2680. if (x->tag == ARRAY) {
  2681. if (!x->val.array->data)
  2682. return 0;
  2683. for (size_t i = 0; i < x->val.array->length; i++) {
  2684. size_t d2 = depthOf(x->val.array->data[i], d + 1);
  2685. if (d2 > d)
  2686. d = d2;
  2687. }
  2688. return d;
  2689. }
  2690. return 0;
  2691. }
  2692. value_t *verb_depth(interpreter_t *state, verb_t *self, value_t *x) {
  2693. return value_new_number(depthOf(x, 1));
  2694. }
  2695. value_t *verb_round(interpreter_t *state, verb_t *self, value_t *x) {
  2696. if (x->tag == NUMBER)
  2697. return value_new_number(round(x->val.number));
  2698. return _NAN;
  2699. }
  2700. value_t *verb_abs(interpreter_t *state, verb_t *self, value_t *x) {
  2701. if (x->tag == NUMBER)
  2702. return value_new_number(fabs(x->val.number));
  2703. return _NAN;
  2704. }
  2705. value_t *verb_tail(interpreter_t *state, verb_t *self, value_t *x);
  2706. value_t *verb_at(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2707. if (y->tag != NUMBER)
  2708. return state->udf;
  2709. if (x->tag != ARRAY) {
  2710. if (y->val.number > -1 && y->val.number < 1)
  2711. return x;
  2712. else
  2713. return state->udf;
  2714. }
  2715. if (!x->val.array->data)
  2716. return state->nil;
  2717. value_t *v = list_index(x->val.array, (ssize_t)y->val.number);
  2718. if (!v)
  2719. return state->udf;
  2720. return v;
  2721. }
  2722. value_t *verb_member(interpreter_t *state, verb_t *self, value_t *x,
  2723. value_t *y) {
  2724. if (y->tag != ARRAY)
  2725. y = verb_enlist(state, self, y);
  2726. else if (!y->val.array->data)
  2727. return NUMS[0];
  2728. for (size_t i = 0; i < y->val.array->length; i++)
  2729. if (value_equals(y->val.array->data[i], x))
  2730. return NUMS[1];
  2731. return NUMS[0];
  2732. }
  2733. value_t *verb_shuffle(interpreter_t *state, verb_t *self, value_t *x) {
  2734. if (x->tag != ARRAY)
  2735. x = verb_enlist(state, self, x);
  2736. else if (!x->val.array->data)
  2737. return x;
  2738. list_t *r = list_copy(x->val.array);
  2739. for (size_t i = 0; i < r->length; i++) {
  2740. size_t j = rand() % r->length;
  2741. value_t *tmp = r->data[i];
  2742. r->data[i] = r->data[j];
  2743. r->data[j] = tmp;
  2744. }
  2745. return value_new_array(r);
  2746. }
  2747. value_t *verb_head(interpreter_t *state, verb_t *self, value_t *x) {
  2748. return verb_take(state, NULL, NUMS[2], x);
  2749. }
  2750. value_t *verb_bin(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2751. if (x->tag != ARRAY)
  2752. x = verb_enlist(state, self, x);
  2753. else if (!x->val.array->data)
  2754. return x;
  2755. if (y->tag != ARRAY)
  2756. y = verb_enlist(state, self, x);
  2757. else if (!y->val.array->data)
  2758. return y;
  2759. size_t xl = x->val.array->length;
  2760. list_t *bins = list_new();
  2761. for (size_t i = 0; i < xl; i++) {
  2762. double s;
  2763. double e;
  2764. value_t *vs = x->val.array->data[i];
  2765. if (vs->tag == NUMBER)
  2766. s = vs->val.number;
  2767. else if (vs->tag == CHAR)
  2768. s = vs->val._char;
  2769. else
  2770. return state->udf;
  2771. value_t *ve =
  2772. i == xl - 1 ? value_new_number(s + 1) : x->val.array->data[i + 1];
  2773. if (ve->tag == NUMBER)
  2774. e = fabs(ve->val.number);
  2775. else if (ve->tag == CHAR)
  2776. e = ve->val._char;
  2777. else
  2778. return state->udf;
  2779. if (bins->data) {
  2780. list_t *pp = list_index(bins, -1);
  2781. double *pe = pp->data[0];
  2782. if (s <= (*pe))
  2783. return state->udf;
  2784. }
  2785. double *sn = malloc_checked(sizeof(double));
  2786. *sn = s;
  2787. double *en = malloc_checked(sizeof(double));
  2788. *en = e;
  2789. list_t *p = list_new();
  2790. list_push(p, sn);
  2791. list_push(p, en);
  2792. list_push(bins, p);
  2793. }
  2794. size_t bl = bins->length;
  2795. list_t *r = list_new();
  2796. size_t yl = y->val.array->length;
  2797. for (size_t i = 0; i < yl; i++) {
  2798. value_t *it = y->val.array->data[i];
  2799. double itv;
  2800. if (it->tag == NUMBER)
  2801. itv = it->val.number;
  2802. else if (it->tag == CHAR)
  2803. itv = it->val._char;
  2804. else
  2805. return state->udf;
  2806. list_t *b = bins->data[0];
  2807. double *s = b->data[0];
  2808. if (itv < (*s)) {
  2809. list_push(r, NNUMS[0]);
  2810. continue;
  2811. }
  2812. b = list_index(bins, -1);
  2813. s = b->data[1];
  2814. if (itv >= (*s)) {
  2815. list_push(r, value_new_number(bl - 1));
  2816. continue;
  2817. }
  2818. double v = NAN;
  2819. for (size_t j = 0; j < bl; j++) {
  2820. b = bins->data[j];
  2821. double *s = b->data[0];
  2822. double *e = b->data[1];
  2823. if (itv >= (*s) && itv < (*e)) {
  2824. v = j;
  2825. break;
  2826. }
  2827. }
  2828. if (!isnan(v))
  2829. list_push(r, value_new_number(v));
  2830. }
  2831. for (size_t j = 0; j < bl; j++) {
  2832. list_t *b = bins->data[j];
  2833. GC_FREE(b->data[0]);
  2834. GC_FREE(b->data[1]);
  2835. GC_FREE(b->data);
  2836. GC_FREE(b);
  2837. }
  2838. GC_FREE(bins->data);
  2839. GC_FREE(bins);
  2840. return value_new_array(r);
  2841. }
  2842. value_t *verb_tail(interpreter_t *state, verb_t *self, value_t *x) {
  2843. if (x->tag != ARRAY)
  2844. return x;
  2845. if (!x->val.array->data)
  2846. return state->udf;
  2847. return list_index(x->val.array, -1);
  2848. }
  2849. value_t *verb_cut(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2850. if (x->tag != ARRAY)
  2851. x = verb_enlist(state, self, x);
  2852. else if (!x->val.array->data)
  2853. return x;
  2854. if (y->tag != ARRAY)
  2855. y = verb_enlist(state, self, x);
  2856. else if (!y->val.array->data)
  2857. return x;
  2858. if (x->val.array->length != 2)
  2859. return state->udf;
  2860. value_t *vs = x->val.array->data[0];
  2861. value_t *ve = x->val.array->data[1];
  2862. if (vs->tag != NUMBER || ve->tag != NUMBER)
  2863. return state->udf;
  2864. size_t s = fabs(vs->val.number);
  2865. size_t e = fabs(ve->val.number);
  2866. list_t *r = list_new();
  2867. size_t l = y->val.array->length;
  2868. list_t *pa = list_new();
  2869. for (size_t i = s; i < e && i < l; i++) {
  2870. value_t *v = list_index(y->val.array, i);
  2871. if (!v)
  2872. break;
  2873. list_push(pa, v);
  2874. }
  2875. list_t *pb = list_new();
  2876. for (size_t i = e; i < l; i++) {
  2877. value_t *v = list_index(y->val.array, i);
  2878. if (!v)
  2879. break;
  2880. list_push(pb, v);
  2881. }
  2882. list_push(r, value_new_array(pa));
  2883. list_push(r, value_new_array(pb));
  2884. return value_new_array(r);
  2885. }
  2886. value_t *verb_prefixes(interpreter_t *state, verb_t *self, value_t *x) {
  2887. if (x->tag != ARRAY)
  2888. x = verb_enlist(state, NULL, x);
  2889. list_t *r = list_new();
  2890. for (size_t i = 0; i < x->val.array->length; i++)
  2891. list_push(r, verb_take(state, NULL, value_new_number(i), x));
  2892. list_push(r, x);
  2893. return value_new_array(r);
  2894. }
  2895. value_t *verb_behead(interpreter_t *state, verb_t *self, value_t *x) {
  2896. return verb_drop(state, NULL, NUMS[1], x);
  2897. }
  2898. value_t *verb_curtail(interpreter_t *state, verb_t *self, value_t *x) {
  2899. return verb_drop(state, NULL, NNUMS[0], x);
  2900. }
  2901. value_t *verb_suffixes(interpreter_t *state, verb_t *self, value_t *x) {
  2902. if (x->tag != ARRAY)
  2903. x = verb_enlist(state, NULL, x);
  2904. list_t *r = list_new();
  2905. for (size_t i = 0; i < x->val.array->length; i++)
  2906. list_push(r, verb_drop(state, NULL, value_new_number(i), x));
  2907. list_push(r, state->unit);
  2908. return value_new_array(r);
  2909. }
  2910. value_t *verb_left(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  2911. return x;
  2912. }
  2913. value_t *verb_same(interpreter_t *state, verb_t *self, value_t *x) { return x; }
  2914. value_t *verb_right(interpreter_t *state, verb_t *self, value_t *x,
  2915. value_t *y) {
  2916. return y;
  2917. }
  2918. value_t *verb_symbol(interpreter_t *state, verb_t *self, value_t *x) {
  2919. char *s = value_str(x);
  2920. return value_new_symbol(s);
  2921. }
  2922. value_t *verb_apply1(interpreter_t *state, verb_t *self, value_t *x,
  2923. value_t *y) {
  2924. return apply_monad(state, x, y);
  2925. }
  2926. value_t *verb_apply2(interpreter_t *state, verb_t *self, value_t *x,
  2927. value_t *y) {
  2928. if (y->tag != ARRAY || y->val.array->length < 2)
  2929. return state->udf;
  2930. return apply_dyad(state, x, y->val.array->data[0], y->val.array->data[1]);
  2931. }
  2932. value_t *verb_shape(interpreter_t *state, verb_t *self, value_t *x) {
  2933. if (x->tag != ARRAY || !x->val.array->data)
  2934. return state->unit;
  2935. if (!is_arrays_array(x->val.array))
  2936. return verb_enlist(state, NULL, verb_count(state, NULL, x));
  2937. if (x->val.array->length < 2)
  2938. return verb_enlist(state, NULL,
  2939. verb_shape(state, NULL, x->val.array->data[0]));
  2940. return verb_enpair(state, NULL, verb_count(state, NULL, x),
  2941. verb_count(state, NULL, x->val.array->data[0]));
  2942. }
  2943. value_t *verb_reshape(interpreter_t *state, verb_t *self, value_t *x,
  2944. value_t *y) {
  2945. if (y->tag != ARRAY)
  2946. y = verb_enlist(state, NULL, y);
  2947. else if (!y->val.array->data)
  2948. return y;
  2949. if (x->tag != ARRAY)
  2950. x = verb_enlist(state, NULL, x);
  2951. else if (!x->val.array->data)
  2952. return state->unit;
  2953. list_t *r;
  2954. if (x->val.array->length < 2) {
  2955. value_t *a = x->val.array->data[0];
  2956. if (a->tag != NUMBER)
  2957. return state->udf;
  2958. size_t k = fabs(a->val.number);
  2959. list_t *t = list_new();
  2960. flatten(y, t);
  2961. r = list_newk(k);
  2962. for (size_t i = 0; i < k; i++)
  2963. r->data[i] = t->data[i % t->length];
  2964. } else if (x->val.array->length > 1) {
  2965. value_t *a = x->val.array->data[0];
  2966. if (a->tag != NUMBER)
  2967. return state->udf;
  2968. value_t *b = x->val.array->data[1];
  2969. if (a->tag != NUMBER)
  2970. return state->udf;
  2971. size_t k = fabs(a->val.number);
  2972. size_t l = fabs(b->val.number);
  2973. y = verb_reshape(state, self,
  2974. verb_enlist(state, NULL, value_new_number(k * l)), y);
  2975. r = list_new();
  2976. size_t yp = 0;
  2977. while (k--) {
  2978. list_t *rw = list_new();
  2979. for (size_t i = 0; i < l; i++)
  2980. list_push(rw, y->val.array->data[yp++]);
  2981. list_push(r, value_new_array(rw));
  2982. }
  2983. } else
  2984. return state->udf;
  2985. return value_new_array(r);
  2986. }
  2987. value_t *verb_repr(interpreter_t *state, verb_t *self, value_t *x) {
  2988. char *s = value_show(x);
  2989. list_t *r = list_new();
  2990. for (size_t i = 0; i < strlen(s); i++)
  2991. list_push(r, value_new_char(s[i]));
  2992. GC_FREE(s);
  2993. return value_new_array(r);
  2994. }
  2995. char *format(char *template, list_t *replaces) {
  2996. buffer_t *text = buffer_new();
  2997. bool skip = false;
  2998. size_t ri = 0;
  2999. size_t tl = strlen(template);
  3000. size_t rl = replaces->length;
  3001. for (size_t i = 0; i < tl; i++) {
  3002. char c = template[i];
  3003. if (skip) {
  3004. buffer_append(text, c);
  3005. skip = false;
  3006. continue;
  3007. }
  3008. if (c == '_') {
  3009. char *s = value_show(list_index(replaces, ri));
  3010. buffer_append_str(text, s);
  3011. GC_FREE(s);
  3012. if (ri < rl - 1)
  3013. ri++;
  3014. continue;
  3015. } else if (c == '{') {
  3016. size_t bi = i;
  3017. buffer_t *n = buffer_new();
  3018. i++;
  3019. while (i < tl && template[i] != '}')
  3020. buffer_append(n, template[i++]);
  3021. if (i >= tl || template[i] != '}') {
  3022. GC_FREE(buffer_read(n));
  3023. buffer_append(text, '{');
  3024. i = bi;
  3025. continue;
  3026. }
  3027. char *s = buffer_read(n);
  3028. ssize_t ind = atoi(s);
  3029. GC_FREE(s);
  3030. value_t *v = list_index(replaces, ind);
  3031. if (!v)
  3032. continue;
  3033. s = value_show(v);
  3034. buffer_append_str(text, s);
  3035. GC_FREE(s);
  3036. continue;
  3037. } else if (c == '~') {
  3038. skip = true;
  3039. continue;
  3040. }
  3041. buffer_append(text, c);
  3042. }
  3043. return buffer_read(text);
  3044. }
  3045. value_t *verb_format(interpreter_t *state, verb_t *self, value_t *x,
  3046. value_t *y) {
  3047. if (y->tag != ARRAY)
  3048. y = verb_enlist(state, NULL, x);
  3049. else if (!y->val.array->data)
  3050. return y;
  3051. char *fmt = value_show(x);
  3052. char *s = format(fmt, y->val.array);
  3053. GC_FREE(fmt);
  3054. size_t z = strlen(s);
  3055. list_t *r = list_newk(z);
  3056. for (size_t i = 0; i < z; i++)
  3057. r->data[i] = CHARS[(int)s[i]];
  3058. return value_new_array(r);
  3059. }
  3060. value_t *verb_insert(interpreter_t *state, verb_t *self, value_t *x,
  3061. value_t *y) {
  3062. if (y->tag != ARRAY)
  3063. y = verb_enlist(state, NULL, y);
  3064. else if (!y->val.array->data)
  3065. return y;
  3066. list_t *r = list_newk(y->val.array->length * 2 - 1);
  3067. size_t rp = 0;
  3068. for (size_t i = 0; i < y->val.array->length; i++) {
  3069. r->data[rp++] = y->val.array->data[i];
  3070. if (i != y->val.array->length - 1)
  3071. r->data[rp++] = x;
  3072. }
  3073. return value_new_array(r);
  3074. }
  3075. uint64_t fibonacci(uint64_t n) {
  3076. uint64_t a = 0;
  3077. uint64_t b = 1;
  3078. while (n-- > 1) {
  3079. uint64_t t = a;
  3080. a = b;
  3081. b += t;
  3082. }
  3083. return b;
  3084. }
  3085. value_t *verb_fibonacci(interpreter_t *state, verb_t *self, value_t *x) {
  3086. if (x->tag == NUMBER)
  3087. return value_new_number(fibonacci((uint64_t)fabs(x->val.number)));
  3088. return _NAN;
  3089. }
  3090. value_t *verb_iota(interpreter_t *state, verb_t *self, value_t *x) {
  3091. if (value_equals(x, NUMS[1]))
  3092. return verb_enlist(state, NULL, NUMS[1]);
  3093. else if (value_equals(x, NUMS[0]))
  3094. return state->unit;
  3095. return verb_range(state, self, NUMS[1], x);
  3096. }
  3097. value_t *verb_range(interpreter_t *state, verb_t *self, value_t *x,
  3098. value_t *y) {
  3099. if ((x->tag == NUMBER || x->tag == CHAR) &&
  3100. (y->tag == NUMBER || y->tag == CHAR)) {
  3101. if (x->tag == NUMBER && is_bad_num(x->val.number))
  3102. return state->udf;
  3103. if (y->tag == NUMBER && is_bad_num(y->val.number))
  3104. return state->udf;
  3105. ssize_t s = get_numeric(x);
  3106. ssize_t e = get_numeric(y);
  3107. if (s == e)
  3108. return verb_enlist(state, NULL, x);
  3109. size_t p = 0;
  3110. list_t *r = list_newk((s > e ? s - e : e - s) + 1);
  3111. if (s > e)
  3112. for (ssize_t i = s; i >= e; i--) {
  3113. if (x->tag == CHAR || y->tag == CHAR)
  3114. r->data[p++] = CHARS[i];
  3115. else
  3116. r->data[p++] = value_new_number(i);
  3117. }
  3118. else
  3119. for (ssize_t i = s; i <= e; i++) {
  3120. if (x->tag == CHAR || y->tag == CHAR)
  3121. r->data[p++] = CHARS[i];
  3122. else
  3123. r->data[p++] = value_new_number(i);
  3124. }
  3125. return value_new_array(r);
  3126. }
  3127. return _NAN;
  3128. }
  3129. value_t *verb_deal(interpreter_t *state, verb_t *self, value_t *x) {
  3130. if (x->tag != ARRAY)
  3131. return x;
  3132. if (!x->val.array->data)
  3133. return state->udf;
  3134. return x->val.array->data[rand() % x->val.array->length];
  3135. }
  3136. value_t *verb_roll(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3137. if (x->tag == NUMBER && y->tag == NUMBER) {
  3138. size_t k = fabs(x->val.number);
  3139. size_t d = fabs(y->val.number);
  3140. list_t *r = list_newk(k);
  3141. for (size_t i = 0; i < k; i++)
  3142. r->data[i] = value_new_number(rand() % d);
  3143. return value_new_array(r);
  3144. }
  3145. return state->udf;
  3146. }
  3147. value_t *verb_type(interpreter_t *state, verb_t *self, value_t *x) {
  3148. return NUMS[x->tag];
  3149. }
  3150. value_t *verb_cast(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3151. if (x->tag == NUMBER) {
  3152. int t = fabs(x->val.number);
  3153. if (y->tag == t)
  3154. return y;
  3155. switch (t) {
  3156. case ARRAY:
  3157. if (y->tag == SYMBOL) {
  3158. char *s = y->val.symbol;
  3159. size_t z = strlen(s);
  3160. list_t *r = list_newk(z);
  3161. for (size_t i = 0; i < z; i++)
  3162. r->data[i] = CHARS[(int)s[i]];
  3163. return value_new_array(r);
  3164. }
  3165. break;
  3166. case NUMBER:
  3167. if (y->tag == CHAR)
  3168. return value_new_number(y->val._char);
  3169. else if (y->tag == ARRAY && y->val.array->data &&
  3170. is_char_array(y->val.array)) {
  3171. buffer_t *buf = buffer_new();
  3172. for (size_t i = 0; i < y->val.array->length; i++)
  3173. buffer_append(buf, ((value_t *)y->val.array->data[i])->val._char);
  3174. char *s = buffer_read(buf);
  3175. double r = strtod(s, NULL);
  3176. GC_FREE(s);
  3177. return value_new_number(r);
  3178. }
  3179. break;
  3180. case CHAR:
  3181. if (y->tag == NUMBER)
  3182. return value_new_char(y->val.number);
  3183. break;
  3184. }
  3185. }
  3186. return state->udf;
  3187. }
  3188. value_t *verb_print(interpreter_t *state, verb_t *self, value_t *x) {
  3189. char *s = value_str(x);
  3190. fprintf(stdout, "%s", s);
  3191. GC_FREE(s);
  3192. return state->nil;
  3193. }
  3194. value_t *verb_println(interpreter_t *state, verb_t *self, value_t *x) {
  3195. char *s = value_str(x);
  3196. fprintf(stdout, "%s\n", s);
  3197. GC_FREE(s);
  3198. return state->nil;
  3199. }
  3200. value_t *verb_putch(interpreter_t *state, verb_t *self, value_t *x) {
  3201. if (x->tag != CHAR)
  3202. return state->udf;
  3203. fputc(x->val._char, stdout);
  3204. return state->nil;
  3205. }
  3206. value_t *verb_exit(interpreter_t *state, verb_t *self, value_t *x) {
  3207. if (x->tag != NUMBER)
  3208. return state->udf;
  3209. int code = x->val.number;
  3210. exit(code);
  3211. return state->nil;
  3212. }
  3213. value_t *verb_read(interpreter_t *state, verb_t *self, value_t *x) {
  3214. if (x == NUMS[0]) {
  3215. buffer_t *buf = buffer_new();
  3216. size_t size = 0;
  3217. for (;;) {
  3218. int c = fgetc(stdin);
  3219. if (c < 0)
  3220. break;
  3221. buffer_append(buf, c);
  3222. size++;
  3223. }
  3224. char *s = buffer_read(buf);
  3225. list_t *r = list_newk(size);
  3226. for (size_t i = 0; i < size; i++)
  3227. r->data[i] = CHARS[(int)s[i]];
  3228. GC_FREE(s);
  3229. return value_new_array(r);
  3230. } else if (x == NUMS[1])
  3231. return value_new_char((unsigned char)fgetc(stdin));
  3232. else if (x == NUMS[2]) {
  3233. char line[512];
  3234. if (!fgets(line, sizeof(line), stdin))
  3235. return state->udf;
  3236. size_t z = strlen(line);
  3237. list_t *r = list_newk(z);
  3238. for (size_t i = 0; i < z; i++)
  3239. r->data[i] = CHARS[(int)line[i]];
  3240. return value_new_array(r);
  3241. }
  3242. char *path = value_str(x);
  3243. FILE *fd = fopen(path, "rb");
  3244. if (!fd) {
  3245. GC_FREE(path);
  3246. return state->udf;
  3247. }
  3248. fseek(fd, 0, SEEK_END);
  3249. size_t size = ftell(fd);
  3250. fseek(fd, 0, SEEK_SET);
  3251. unsigned char *buf = malloc_checked(size + 1);
  3252. size = fread(buf, sizeof(unsigned char), size, fd);
  3253. fclose(fd);
  3254. GC_FREE(path);
  3255. list_t *r = list_newk(size);
  3256. for (size_t i = 0; i < size; i++)
  3257. r->data[i] = CHARS[buf[i]];
  3258. GC_FREE(buf);
  3259. return value_new_array(r);
  3260. }
  3261. value_t *verb_write(interpreter_t *state, verb_t *self, value_t *x,
  3262. value_t *y) {
  3263. FILE *fd;
  3264. char *path = NULL;
  3265. if (x->tag != ARRAY)
  3266. x = verb_enlist(state, NULL, x);
  3267. if (y == NUMS[0])
  3268. fd = stderr;
  3269. else {
  3270. path = value_str(y);
  3271. fd = fopen(path, "wb");
  3272. if (!fd) {
  3273. GC_FREE(path);
  3274. return NNUMS[0];
  3275. }
  3276. }
  3277. size_t k = 0;
  3278. for (size_t i = 0; i < x->val.array->length; i++) {
  3279. unsigned char c;
  3280. value_t *v = x->val.array->data[i];
  3281. if (v->tag == NUMBER)
  3282. c = fabs(v->val.number);
  3283. else if (v->tag == CHAR)
  3284. c = v->val._char;
  3285. else
  3286. break;
  3287. fputc(c, fd);
  3288. k++;
  3289. }
  3290. fclose(fd);
  3291. if (path)
  3292. GC_FREE(path);
  3293. return value_new_number(k);
  3294. }
  3295. value_t *verb_system(interpreter_t *state, verb_t *self, value_t *x) {
  3296. char *cmd = value_str(x);
  3297. FILE *pd;
  3298. pd = popen(cmd, "r");
  3299. if (!pd) {
  3300. GC_FREE(cmd);
  3301. return state->udf;
  3302. }
  3303. unsigned char *buffer = NULL;
  3304. size_t buffer_size = 0;
  3305. size_t buffer_allocated = 0;
  3306. size_t bytes_received;
  3307. unsigned char chunk[1024];
  3308. for (;;) {
  3309. bytes_received = fread(chunk, 1, 1024, pd);
  3310. if (bytes_received == 0)
  3311. break;
  3312. size_t head = buffer_size;
  3313. buffer_size += bytes_received;
  3314. if (buffer_size > buffer_allocated) {
  3315. buffer_allocated = buffer_size;
  3316. buffer = realloc_checked(buffer, buffer_allocated);
  3317. }
  3318. for (size_t i = 0; i < bytes_received; i++)
  3319. buffer[head + i] = chunk[i];
  3320. if (feof(pd))
  3321. break;
  3322. }
  3323. pclose(pd);
  3324. GC_FREE(cmd);
  3325. list_t *r = list_newk(buffer_size);
  3326. for (size_t i = 0; i < buffer_size; i++)
  3327. r->data[i] = CHARS[buffer[i]];
  3328. GC_FREE(buffer);
  3329. return value_new_array(r);
  3330. }
  3331. struct files_t {
  3332. FILE *in;
  3333. FILE *out;
  3334. };
  3335. typedef struct files_t files_t;
  3336. struct files_chain_t {
  3337. files_t files;
  3338. pid_t pid;
  3339. struct files_chain_t *next;
  3340. };
  3341. typedef struct files_chain_t files_chain_t;
  3342. static files_chain_t *files_chain;
  3343. void _cleanup_pipe(int *pipe) {
  3344. close(pipe[0]);
  3345. close(pipe[1]);
  3346. }
  3347. static int _do_popen2(files_chain_t *link, const char *command) {
  3348. int child_in[2];
  3349. int child_out[2];
  3350. if (0 != pipe(child_in))
  3351. return -1;
  3352. if (0 != pipe(child_out)) {
  3353. _cleanup_pipe(child_in);
  3354. return -1;
  3355. }
  3356. pid_t cpid = link->pid = fork();
  3357. if (0 > cpid) {
  3358. _cleanup_pipe(child_in);
  3359. _cleanup_pipe(child_out);
  3360. return -1;
  3361. }
  3362. if (0 == cpid) {
  3363. if (0 > dup2(child_in[0], 0) || 0 > dup2(child_out[1], 1))
  3364. _Exit(127);
  3365. _cleanup_pipe(child_in);
  3366. _cleanup_pipe(child_out);
  3367. for (files_chain_t *p = files_chain; p; p = p->next) {
  3368. int fd_in = fileno(p->files.in);
  3369. if (fd_in != 0)
  3370. close(fd_in);
  3371. int fd_out = fileno(p->files.out);
  3372. if (fd_out != 1)
  3373. close(fd_out);
  3374. }
  3375. execl("/bin/sh", "sh", "-c", command, (char *)NULL);
  3376. _Exit(127);
  3377. }
  3378. close(child_in[0]);
  3379. close(child_out[1]);
  3380. link->files.in = fdopen(child_in[1], "w");
  3381. link->files.out = fdopen(child_out[0], "r");
  3382. return 0;
  3383. }
  3384. files_t *popen2(const char *command) {
  3385. files_chain_t *link = (files_chain_t *)malloc(sizeof(files_chain_t));
  3386. if (NULL == link)
  3387. return NULL;
  3388. if (0 > _do_popen2(link, command)) {
  3389. free(link);
  3390. return NULL;
  3391. }
  3392. link->next = files_chain;
  3393. files_chain = link;
  3394. return (files_t *)link;
  3395. }
  3396. int pclose2(files_t *fp) {
  3397. files_chain_t **p = &files_chain;
  3398. int found = 0;
  3399. while (*p) {
  3400. if (*p == (files_chain_t *)fp) {
  3401. *p = (*p)->next;
  3402. found = 1;
  3403. break;
  3404. }
  3405. p = &(*p)->next;
  3406. }
  3407. if (!found)
  3408. return -1;
  3409. if (0 > fclose(fp->out)) {
  3410. free((files_chain_t *)fp);
  3411. return -1;
  3412. }
  3413. int status = -1;
  3414. pid_t wait_pid;
  3415. do {
  3416. wait_pid = waitpid(((files_chain_t *)fp)->pid, &status, 0);
  3417. } while (-1 == wait_pid && EINTR == errno);
  3418. free((files_chain_t *)fp);
  3419. if (wait_pid == -1)
  3420. return -1;
  3421. return status;
  3422. }
  3423. value_t *verb_system2(interpreter_t *state, verb_t *self, value_t *x,
  3424. value_t *y) {
  3425. char *cmd = value_str(y);
  3426. files_t *pd;
  3427. pd = popen2(cmd);
  3428. if (pd == NULL) {
  3429. GC_FREE(cmd);
  3430. return state->udf;
  3431. }
  3432. for (size_t i = 0; i < x->val.array->length; i++) {
  3433. unsigned char c;
  3434. value_t *v = x->val.array->data[i];
  3435. if (v->tag == NUMBER)
  3436. c = fabs(v->val.number);
  3437. else if (v->tag == CHAR)
  3438. c = v->val._char;
  3439. else
  3440. break;
  3441. fputc(c, pd->in);
  3442. }
  3443. fflush(pd->in);
  3444. fclose(pd->in);
  3445. unsigned char *buffer = NULL;
  3446. size_t buffer_size = 0;
  3447. size_t buffer_allocated = 0;
  3448. size_t bytes_received;
  3449. unsigned char chunk[1024];
  3450. for (;;) {
  3451. bytes_received = fread(chunk, 1, 1024, pd->out);
  3452. if (bytes_received == 0)
  3453. break;
  3454. size_t head = buffer_size;
  3455. buffer_size += bytes_received;
  3456. if (buffer_size > buffer_allocated) {
  3457. buffer_allocated = buffer_size;
  3458. buffer = realloc_checked(buffer, buffer_allocated);
  3459. }
  3460. for (size_t i = 0; i < bytes_received; i++)
  3461. buffer[head + i] = chunk[i];
  3462. if (feof(pd->out))
  3463. break;
  3464. }
  3465. pclose2(pd);
  3466. GC_FREE(cmd);
  3467. list_t *r = list_newk(buffer_size);
  3468. for (size_t i = 0; i < buffer_size; i++)
  3469. r->data[i] = CHARS[buffer[i]];
  3470. GC_FREE(buffer);
  3471. return value_new_array(r);
  3472. }
  3473. value_t *verb_shl(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3474. if (x->tag == NUMBER && y->tag == NUMBER)
  3475. return value_new_number(((int)x->val.number) << ((int)y->val.number));
  3476. return _NAN;
  3477. }
  3478. value_t *verb_shr(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3479. if (x->tag == NUMBER && y->tag == NUMBER)
  3480. return value_new_number(((int)x->val.number) >> ((int)y->val.number));
  3481. return _NAN;
  3482. }
  3483. value_t *verb_xor(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3484. if (x->tag == NUMBER && y->tag == NUMBER)
  3485. return value_new_number(((int)x->val.number) ^ ((int)y->val.number));
  3486. return _NAN;
  3487. }
  3488. value_t *verb_band(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3489. if (x->tag == NUMBER && y->tag == NUMBER)
  3490. return value_new_number(((int)x->val.number) & ((int)y->val.number));
  3491. return _NAN;
  3492. }
  3493. list_t *find_primes(uint64_t limit) {
  3494. bool sieve[limit + 1];
  3495. for (uint64_t i = 0; i <= limit; i++)
  3496. sieve[i] = false;
  3497. if (limit > 2)
  3498. sieve[2] = true;
  3499. if (limit > 3)
  3500. sieve[3] = true;
  3501. for (uint64_t x = 1; x * x <= limit; x++)
  3502. for (uint64_t y = 1; y * y <= limit; y++) {
  3503. uint64_t n = (4 * x * x) + (y * y);
  3504. if (n <= limit && (n % 12 == 1 || n % 12 == 5))
  3505. sieve[n] ^= true;
  3506. n = (3 * x * x) + (y * y);
  3507. if (n <= limit && n % 12 == 7)
  3508. sieve[n] ^= true;
  3509. n = (3 * x * x) - (y * y);
  3510. if (x > y && n <= limit && n % 12 == 11)
  3511. sieve[n] ^= true;
  3512. }
  3513. for (uint64_t r = 5; r * r <= limit; r++)
  3514. if (sieve[r])
  3515. for (int i = r * r; i <= limit; i += r * r)
  3516. sieve[i] = false;
  3517. list_t *r = list_new();
  3518. for (uint64_t a = 1; a <= limit; a++)
  3519. if (sieve[a])
  3520. list_push(r, value_new_number(a));
  3521. return r;
  3522. }
  3523. value_t *verb_primes(interpreter_t *state, verb_t *self, value_t *x) {
  3524. if (x->tag == NUMBER && !is_bad_num(x->val.number))
  3525. return value_new_array(find_primes(fabs(x->val.number) + 1));
  3526. return state->udf;
  3527. }
  3528. value_t *verb_parts(interpreter_t *state, verb_t *self, value_t *x,
  3529. value_t *y) {
  3530. if (x->tag != NUMBER)
  3531. return state->udf;
  3532. if (y->tag != ARRAY)
  3533. y = verb_enlist(state, NULL, y);
  3534. else if (!y->val.array->data)
  3535. return y;
  3536. if (is_bad_num(x->val.number) || x->val.number < 1)
  3537. return y;
  3538. size_t np = fabs(x->val.number);
  3539. list_t *r = list_newk(np);
  3540. size_t rp = 0;
  3541. for (ssize_t i = np; i > 0; i--) {
  3542. size_t k = ceil(((double)y->val.array->length) / (double)i);
  3543. r->data[rp++] = verb_take(state, NULL, value_new_number(k), y);
  3544. y = verb_drop(state, NULL, value_new_number(k), y);
  3545. }
  3546. return value_new_array(r);
  3547. }
  3548. value_t *verb_bor(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  3549. if (x->tag == NUMBER && y->tag == NUMBER)
  3550. return value_new_number(((int)x->val.number) | ((int)y->val.number));
  3551. return _NAN;
  3552. }
  3553. value_t *verb_bnot(interpreter_t *state, verb_t *self, value_t *x) {
  3554. if (x->tag == NUMBER)
  3555. return value_new_number(~(int)x->val.number);
  3556. return _NAN;
  3557. }
  3558. list_t *prime_factors(double n) {
  3559. list_t *factors = list_new();
  3560. double divisor = 2;
  3561. while (n >= 2) {
  3562. if (fmod(n, divisor) == 0) {
  3563. list_push(factors, value_new_number(divisor));
  3564. n /= divisor;
  3565. } else
  3566. divisor++;
  3567. }
  3568. return factors;
  3569. }
  3570. value_t *verb_factors(interpreter_t *state, verb_t *self, value_t *x) {
  3571. if (x->tag == NUMBER && !is_bad_num(x->val.number))
  3572. return value_new_array(prime_factors(x->val.number));
  3573. return state->udf;
  3574. }
  3575. value_t *verb_combine(interpreter_t *state, verb_t *self, value_t *x,
  3576. value_t *y) {
  3577. if (x->tag == NUMBER && y->tag == NUMBER && !is_bad_num(x->val.number) &&
  3578. !is_bad_num(y->val.number)) {
  3579. x = verb_base(state, NULL, NUMS[10], x);
  3580. y = verb_base(state, NULL, NUMS[10], y);
  3581. value_t *n = verb_join(state, NULL, x, y);
  3582. return verb_unbase(state, NULL, NUMS[10], n);
  3583. }
  3584. return _NAN;
  3585. }
  3586. value_t *verb_outof(interpreter_t *state, verb_t *self, value_t *x,
  3587. value_t *y) {
  3588. if (x->tag == NUMBER && y->tag == NUMBER && !is_bad_num(x->val.number) &&
  3589. !is_bad_num(y->val.number)) {
  3590. uint64_t a = (uint64_t)fabs(x->val.number);
  3591. uint64_t b = (uint64_t)fabs(y->val.number);
  3592. if (a == 0)
  3593. return NUMS[1];
  3594. if (b == 0)
  3595. return NUMS[0];
  3596. return value_new_number(factorial(b) /
  3597. (factorial(a) * (a >= b ? 1 : factorial(b - a))));
  3598. }
  3599. return _NAN;
  3600. }
  3601. value_t *verb_sort(interpreter_t *state, verb_t *self, value_t *x) {
  3602. value_t *i = verb_gradeup(state, NULL, x);
  3603. return together(state, state->at, x, i, 0, 0, state->at->rank[1],
  3604. state->at->rank[2]);
  3605. }
  3606. value_t *verb_unsort(interpreter_t *state, verb_t *self, value_t *x) {
  3607. value_t *i = verb_gradedown(state, NULL, x);
  3608. return together(state, state->at, x, i, 0, 0, state->at->rank[1],
  3609. state->at->rank[2]);
  3610. }
  3611. value_t *interpreter_run(interpreter_t *state, char *program);
  3612. value_t *verb_eval(interpreter_t *state, verb_t *self, value_t *x) {
  3613. char *s = value_str(x);
  3614. jmp_buf *lb = guard();
  3615. if (setjmp(*lb)) {
  3616. unguard();
  3617. GC_FREE(s);
  3618. return state->udf;
  3619. }
  3620. value_t *v = interpreter_run(state, s);
  3621. GC_FREE(s);
  3622. unguard();
  3623. return v;
  3624. }
  3625. void jkexec(interpreter_t *state, FILE *fd, bool isrepl, char **s);
  3626. value_t *verb_import(interpreter_t *state, verb_t *self, value_t *x) {
  3627. char *path = value_str(x);
  3628. FILE *fd = fopen(path, "rb");
  3629. if (!fd) {
  3630. GC_FREE(path);
  3631. return state->udf;
  3632. }
  3633. char *s = NULL;
  3634. jkexec(state, fd, false, &s);
  3635. if (s)
  3636. GC_FREE(s);
  3637. fclose(fd);
  3638. GC_FREE(path);
  3639. return state->nil;
  3640. }
  3641. value_t *verb_foreign(interpreter_t *state, verb_t *self, value_t *x,
  3642. value_t *y) {
  3643. if (x->tag != ARRAY)
  3644. x = verb_enlist(state, NULL, x);
  3645. char *pth = value_str(y);
  3646. char *lib;
  3647. char *sig;
  3648. char *fun;
  3649. lib = strtok(pth, ":");
  3650. if (!lib)
  3651. return state->udf;
  3652. sig = strtok(NULL, ":");
  3653. if (!sig)
  3654. return state->udf;
  3655. fun = strtok(NULL, ":");
  3656. if (!fun)
  3657. return state->udf;
  3658. size_t argc = strlen(sig);
  3659. if (argc < 1)
  3660. return state->udf;
  3661. argc--;
  3662. if (argc != x->val.array->length)
  3663. return state->udf;
  3664. ffi_cif cif;
  3665. ffi_type *ret;
  3666. ffi_type *args[argc];
  3667. void *values[argc];
  3668. void *pool[argc];
  3669. size_t fc = 0;
  3670. void *retv = NULL;
  3671. char rett;
  3672. size_t retvsz = 0;
  3673. for (int i = 0; i < strlen(sig); i++) {
  3674. ffi_type *t;
  3675. void *v;
  3676. switch (sig[i]) {
  3677. case '$':
  3678. t = &ffi_type_pointer;
  3679. break;
  3680. case 'p':
  3681. t = &ffi_type_pointer;
  3682. break;
  3683. case 'v':
  3684. if (i != 0)
  3685. goto cleanup;
  3686. t = &ffi_type_void;
  3687. break;
  3688. case 'i':
  3689. t = &ffi_type_sint;
  3690. break;
  3691. case 'l':
  3692. t = &ffi_type_slong;
  3693. break;
  3694. case 'f':
  3695. t = &ffi_type_float;
  3696. break;
  3697. case 'd':
  3698. t = &ffi_type_double;
  3699. break;
  3700. case 'c':
  3701. t = &ffi_type_uchar;
  3702. break;
  3703. default:
  3704. goto cleanup;
  3705. }
  3706. if (i == 0) {
  3707. rett = sig[0];
  3708. ret = t;
  3709. switch (rett) {
  3710. case '$':
  3711. case '@':
  3712. retvsz = sizeof(char *);
  3713. break;
  3714. case 'p':
  3715. retvsz = sizeof(void *);
  3716. break;
  3717. case 'v':
  3718. retvsz = 0;
  3719. break;
  3720. case 'i':
  3721. retvsz = sizeof(int);
  3722. break;
  3723. case 'l':
  3724. retvsz = sizeof(long);
  3725. break;
  3726. case 'f':
  3727. retvsz = sizeof(float);
  3728. break;
  3729. case 'd':
  3730. retvsz = sizeof(double);
  3731. break;
  3732. case 'c':
  3733. retvsz = sizeof(unsigned char);
  3734. break;
  3735. }
  3736. } else {
  3737. switch (sig[i]) {
  3738. case '$':
  3739. case '@': {
  3740. value_t *vt = x->val.array->data[i - 1];
  3741. pool[i - 1] = value_str(vt);
  3742. v = pool[i - 1];
  3743. fc++;
  3744. } break;
  3745. case 'p': {
  3746. void *_pv;
  3747. value_t *vt = x->val.array->data[i - 1];
  3748. if (vt->tag != NUMBER)
  3749. goto cleanup;
  3750. _pv = (void *)(size_t)fabs(vt->val.number);
  3751. pool[i - 1] = malloc_checked(sizeof(void *));
  3752. memcpy(pool[i - 1], &_pv, sizeof(void *));
  3753. v = pool[i - 1];
  3754. fc++;
  3755. } break;
  3756. case 'i': {
  3757. int _iv;
  3758. value_t *vt = x->val.array->data[i - 1];
  3759. if (vt->tag != NUMBER)
  3760. goto cleanup;
  3761. _iv = (int)vt->val.number;
  3762. pool[i - 1] = malloc_checked(sizeof(int));
  3763. memcpy(pool[i - 1], &_iv, sizeof(int));
  3764. v = pool[i - 1];
  3765. fc++;
  3766. } break;
  3767. case 'l': {
  3768. long _lv;
  3769. value_t *_vt = x->val.array->data[i - 1];
  3770. if (_vt->tag != NUMBER)
  3771. goto cleanup;
  3772. _lv = (long)_vt->val.number;
  3773. pool[i - 1] = malloc_checked(sizeof(long));
  3774. memcpy(pool[i - 1], &_lv, sizeof(long));
  3775. v = pool[i - 1];
  3776. fc++;
  3777. } break;
  3778. case 'f': {
  3779. float _fv;
  3780. value_t *_vt = x->val.array->data[i - 1];
  3781. if (_vt->tag != NUMBER)
  3782. goto cleanup;
  3783. _fv = (float)_vt->val.number;
  3784. pool[i - 1] = malloc_checked(sizeof(float));
  3785. memcpy(pool[i - 1], &_fv, sizeof(float));
  3786. v = pool[i - 1];
  3787. fc++;
  3788. } break;
  3789. case 'd': {
  3790. double _dv;
  3791. value_t *_vt = x->val.array->data[i - 1];
  3792. if (_vt->tag != NUMBER)
  3793. goto cleanup;
  3794. _dv = (double)_vt->val.number;
  3795. pool[i - 1] = malloc_checked(sizeof(double));
  3796. memcpy(pool[i - 1], &_dv, sizeof(double));
  3797. v = pool[i - 1];
  3798. fc++;
  3799. } break;
  3800. case 'c': {
  3801. unsigned char _cv;
  3802. value_t *_vt = x->val.array->data[i - 1];
  3803. if (_vt->tag != CHAR)
  3804. goto cleanup;
  3805. _cv = (unsigned char)_vt->val._char;
  3806. pool[i - 1] = malloc_checked(sizeof(unsigned char));
  3807. memcpy(pool[i - 1], &_cv, sizeof(unsigned char));
  3808. v = pool[i - 1];
  3809. fc++;
  3810. } break;
  3811. }
  3812. args[i - 1] = t;
  3813. values[i - 1] = v;
  3814. }
  3815. }
  3816. void *dlh = dlopen(lib, RTLD_LAZY);
  3817. if (!dlh)
  3818. goto cleanup;
  3819. void *exfn = dlsym(dlh, fun);
  3820. char *e = dlerror();
  3821. if (!exfn || e)
  3822. goto cleanup;
  3823. if (ffi_prep_cif(&cif, FFI_DEFAULT_ABI, argc, ret, args) != FFI_OK)
  3824. goto cleanup;
  3825. if (retvsz)
  3826. retv = malloc_checked(retvsz);
  3827. ffi_call(&cif, exfn, retv, values);
  3828. dlclose(dlh);
  3829. value_t *rv = state->nil;
  3830. switch (rett) {
  3831. case 'v':
  3832. break;
  3833. case '$': {
  3834. char *s = *(char **)retv;
  3835. size_t z = strlen(s);
  3836. list_t *l = list_newk(z);
  3837. for (size_t i = 0; i < z; i++)
  3838. l->data[i] = CHARS[(int)s[i]];
  3839. rv = value_new_array(l);
  3840. } break;
  3841. case '@': {
  3842. char *s = *(char **)retv;
  3843. size_t z = strlen(s);
  3844. list_t *l = list_newk(z);
  3845. for (size_t i = 0; i < z; i++)
  3846. l->data[i] = CHARS[(int)s[i]];
  3847. rv = value_new_array(l);
  3848. free(s);
  3849. } break;
  3850. case 'p':
  3851. rv = value_new_number((size_t) * (void **)retv);
  3852. break;
  3853. case 'i':
  3854. rv = value_new_number(*(int *)retv);
  3855. break;
  3856. case 'l':
  3857. rv = value_new_number(*(long *)retv);
  3858. break;
  3859. case 'f':
  3860. rv = value_new_number(*(float *)retv);
  3861. break;
  3862. case 'd':
  3863. rv = value_new_number(*(double *)retv);
  3864. break;
  3865. case 'c':
  3866. rv = value_new_char(*(unsigned char *)retv);
  3867. break;
  3868. }
  3869. GC_FREE(retv);
  3870. for (size_t i = 0; i < fc; i++)
  3871. GC_FREE(pool[i]);
  3872. return rv;
  3873. cleanup:
  3874. for (size_t i = 0; i < fc; i++)
  3875. GC_FREE(pool[i]);
  3876. return state->udf;
  3877. }
  3878. value_t *verb_explode(interpreter_t *state, verb_t *self, value_t *x,
  3879. value_t *y) {
  3880. char *del = value_show(x);
  3881. char *s = value_str(y);
  3882. size_t dell = strlen(del);
  3883. size_t sl = strlen(s);
  3884. list_t *r = list_new();
  3885. list_t *t = list_new();
  3886. for (size_t i = 0; i < sl; i++) {
  3887. if (strncmp(&s[i], del, dell) == 0) {
  3888. list_push(r, value_new_array(t));
  3889. t = list_new();
  3890. i += dell - 1;
  3891. continue;
  3892. }
  3893. list_push(t, CHARS[(int)s[i]]);
  3894. }
  3895. GC_FREE(s);
  3896. GC_FREE(del);
  3897. list_push(r, value_new_array(t));
  3898. return value_new_array(r);
  3899. }
  3900. value_t *verb_implode(interpreter_t *state, verb_t *self, value_t *x,
  3901. value_t *y) {
  3902. if (y->tag != ARRAY || !y->val.array->data)
  3903. return y;
  3904. char *del = value_show(x);
  3905. list_t *r = list_new();
  3906. for (size_t i = 0; i < y->val.array->length; i++) {
  3907. char *s = value_show(y->val.array->data[i]);
  3908. char *_s = s;
  3909. while (*_s)
  3910. list_push(r, CHARS[(int)(*_s++)]);
  3911. GC_FREE(s);
  3912. if (i != y->val.array->length - 1) {
  3913. char *s = del;
  3914. while (*s)
  3915. list_push(r, CHARS[(int)(*s++)]);
  3916. }
  3917. }
  3918. GC_FREE(del);
  3919. return value_new_array(r);
  3920. }
  3921. value_t *verb_tackleft(interpreter_t *state, verb_t *self, value_t *x,
  3922. value_t *y) {
  3923. if (y->tag != ARRAY)
  3924. y = verb_enlist(state, NULL, y);
  3925. list_t *r = list_newk(y->val.array->length + 1);
  3926. r->data[0] = x;
  3927. for (size_t i = 0; i < y->val.array->length; i++)
  3928. r->data[i + 1] = y->val.array->data[i];
  3929. return value_new_array(r);
  3930. }
  3931. value_t *verb_tackright(interpreter_t *state, verb_t *self, value_t *x,
  3932. value_t *y) {
  3933. if (y->tag != ARRAY)
  3934. y = verb_enlist(state, NULL, y);
  3935. list_t *r = list_newk(y->val.array->length + 1);
  3936. for (size_t i = 0; i < y->val.array->length; i++)
  3937. r->data[i] = y->val.array->data[i];
  3938. r->data[y->val.array->length] = x;
  3939. return value_new_array(r);
  3940. }
  3941. value_t *verb_eye(interpreter_t *state, verb_t *self, value_t *x) {
  3942. if (x->tag == NUMBER && !is_bad_num(x->val.number)) {
  3943. size_t k = fabs(x->val.number);
  3944. list_t *r = list_newk(k);
  3945. for (size_t i = 0; i < k; i++) {
  3946. list_t *rw = list_newk(k);
  3947. for (size_t j = 0; j < k; j++)
  3948. rw->data[j] = NUMS[i == j];
  3949. r->data[i] = value_new_array(rw);
  3950. }
  3951. return value_new_array(r);
  3952. }
  3953. return state->udf;
  3954. }
  3955. value_t *verb_infix(interpreter_t *state, verb_t *self, value_t *x) {
  3956. return verb_behead(state, NULL, verb_prefixes(state, NULL, x));
  3957. }
  3958. value_t *verb_value(interpreter_t *state, verb_t *self, value_t *x) {
  3959. char *s = value_str(x);
  3960. value_t *r = table_get(state->env, s);
  3961. GC_FREE(s);
  3962. return r ? r : state->udf;
  3963. }
  3964. value_t *verb_lines(interpreter_t *state, verb_t *self, value_t *x) {
  3965. char *s = value_str(x);
  3966. size_t sl = strlen(s);
  3967. list_t *r = list_new();
  3968. list_t *t = list_new();
  3969. for (size_t i = 0; i < sl; i++) {
  3970. if (s[i] == '\n') {
  3971. list_push(r, value_new_array(t));
  3972. t = list_new();
  3973. continue;
  3974. }
  3975. list_push(t, CHARS[(int)s[i]]);
  3976. }
  3977. GC_FREE(s);
  3978. list_push(r, value_new_array(t));
  3979. return value_new_array(r);
  3980. }
  3981. value_t *verb_delete(interpreter_t *state, verb_t *self, value_t *x,
  3982. value_t *y) {
  3983. if (x->tag != NUMBER)
  3984. return state->udf;
  3985. if (y->tag != ARRAY)
  3986. y = verb_enlist(state, NULL, y);
  3987. else if (!y->val.array->data)
  3988. return y;
  3989. size_t z = y->val.array->length;
  3990. ssize_t index = trunc(x->val.number);
  3991. if (index < 0)
  3992. index += ((ssize_t)z);
  3993. if (index < 0 || index >= z)
  3994. return y;
  3995. list_t *r = list_newk(z - 1);
  3996. size_t ri = 0;
  3997. for (size_t i = 0; i < z; i++)
  3998. if (i == index)
  3999. continue;
  4000. else
  4001. r->data[ri++] = y->val.array->data[i];
  4002. return value_new_array(r);
  4003. }
  4004. value_t *verb_rematch(interpreter_t *state, verb_t *self, value_t *x,
  4005. value_t *y) {
  4006. char *pat = value_str(x);
  4007. char *s = value_str(y);
  4008. pcre *re;
  4009. const char *e;
  4010. int eo;
  4011. if (!(re = pcre_compile(pat, 0, &e, &eo, NULL)))
  4012. goto fail;
  4013. int rc = pcre_exec(re, NULL, s, strlen(s), 0, 0, NULL, 0);
  4014. pcre_free(re);
  4015. GC_FREE(pat);
  4016. GC_FREE(s);
  4017. return NUMS[rc >= 0];
  4018. fail:
  4019. GC_FREE(pat);
  4020. GC_FREE(s);
  4021. return state->udf;
  4022. }
  4023. value_t *verb_show(interpreter_t *state, verb_t *self, value_t *x) {
  4024. char *s = value_str(x);
  4025. list_t *r = list_new();
  4026. for (size_t i = 0; i < strlen(s); i++)
  4027. list_push(r, value_new_char(s[i]));
  4028. GC_FREE(s);
  4029. return value_new_array(r);
  4030. }
  4031. value_t *verb_extract(interpreter_t *state, verb_t *self, value_t *x,
  4032. value_t *y) {
  4033. char *pat = value_str(x);
  4034. char *s = value_str(y);
  4035. size_t len = strlen(s);
  4036. pcre *re;
  4037. const char *e;
  4038. int eo;
  4039. if (!(re = pcre_compile(pat, 0, &e, &eo, NULL)))
  4040. goto fail;
  4041. int ov[128 * 3];
  4042. int rc;
  4043. list_t *r = list_new();
  4044. unsigned int of = 0;
  4045. while (of < len &&
  4046. (rc = pcre_exec(re, 0, s, len, of, 0, ov, sizeof(ov))) >= 0) {
  4047. if (rc == 0)
  4048. rc = sizeof(ov) / 3;
  4049. for (int i = 1; i < rc; i++) {
  4050. char *ss = s + ov[2 * i];
  4051. int sl = ov[2 * i + 1] - ov[2 * i];
  4052. if (sl == 0) {
  4053. list_push(r, _UNIT);
  4054. continue;
  4055. }
  4056. list_t *l = list_newk(sl);
  4057. for (int j = 0; j < sl; j++)
  4058. l->data[j] = CHARS[(int)ss[j]];
  4059. list_push(r, value_new_array(l));
  4060. }
  4061. of = ov[1];
  4062. }
  4063. pcre_free(re);
  4064. GC_FREE(pat);
  4065. GC_FREE(s);
  4066. return value_new_array(r);
  4067. fail:
  4068. GC_FREE(pat);
  4069. GC_FREE(s);
  4070. return state->udf;
  4071. }
  4072. value_t *verb_udf1(interpreter_t *state, verb_t *self, value_t *x) {
  4073. return state->udf;
  4074. }
  4075. value_t *verb_udf2(interpreter_t *state, verb_t *self, value_t *x, value_t *y) {
  4076. return state->udf;
  4077. }
  4078. #define X UINT_MAX
  4079. #define DEFVERB(__symb, __rm, __rl, __rr, __monad, __dyad) \
  4080. {__symb, {__rm, __rl, __rr}, NULL, false, \
  4081. false, verb_##__monad, verb_##__dyad}
  4082. #define DEFVERBD(__symb, __rm, __rl, __rr, __monad, __dyad) \
  4083. {__symb ".", {__rm, __rl, __rr}, NULL, false, \
  4084. false, verb_##__monad, verb_##__dyad}
  4085. #define DEFVERBC(__symb, __rm, __rl, __rr, __monad, __dyad) \
  4086. {__symb ":", {__rm, __rl, __rr}, NULL, false, \
  4087. false, verb_##__monad, verb_##__dyad}
  4088. verb_t VERBS[] = {DEFVERB(":", 0, 0, 0, const, bind),
  4089. DEFVERBC(":", X, 0, 0, unbind, obverse),
  4090. DEFVERB("+", 0, X, X, flip, plus),
  4091. DEFVERBD("+", X, X, X, fibonacci, gcd),
  4092. DEFVERBC("+", X, X, X, sin, combine),
  4093. DEFVERB("-", X, X, X, negate, minus),
  4094. DEFVERBD("-", X, X, X, atan, atan2),
  4095. DEFVERB("*", 0, X, X, first, times),
  4096. DEFVERBD("*", X, X, X, factorial, lcm),
  4097. DEFVERBC("*", X, X, 0, double, replicate),
  4098. DEFVERB("%", X, X, X, reciprocal, divide),
  4099. DEFVERBD("%", X, X, X, sqrt, root),
  4100. DEFVERBC("%", X, X, X, halve, idivide),
  4101. DEFVERB("!", X, X, X, enum, mod),
  4102. DEFVERBD("!", X, X, X, iota, range),
  4103. DEFVERBC("!", 0, X, 0, odometer, chunks),
  4104. DEFVERB("^", X, X, X, exp, power),
  4105. DEFVERBD("^", X, X, X, nlog, log),
  4106. DEFVERB("=", 0, X, X, permute, equals),
  4107. DEFVERBD("=", 0, 0, 0, occurences, mask),
  4108. DEFVERBC("=", 0, 0, 0, classify, equals),
  4109. DEFVERB("~", X, X, X, not, not_equals),
  4110. DEFVERBD("~", X, 0, 0, sign, insert),
  4111. DEFVERBC("~", 0, 0, 0, not, not_equals),
  4112. DEFVERB("<", X, X, X, pred, less),
  4113. DEFVERBD("<", X, X, X, floor, lesseq),
  4114. DEFVERBC("<", 0, X, 0, gradedown, nudge_left),
  4115. DEFVERB(">", X, X, X, succ, greater),
  4116. DEFVERBD(">", X, X, X, ceil, greatereq),
  4117. DEFVERBC(">", 0, X, 0, gradeup, nudge_right),
  4118. DEFVERB(",", 0, 0, 0, enlist, join),
  4119. DEFVERBD(",", X, 0, 0, enlist, enpair),
  4120. DEFVERB("#", 0, X, 0, count, take),
  4121. DEFVERBD("#", 0, 0, 0, where, copy),
  4122. DEFVERBC("#", 0, 0, 0, group, buckets),
  4123. DEFVERB("_", 0, X, 0, nub, drop),
  4124. DEFVERBD("_", 0, X, 0, unbits, unbase),
  4125. DEFVERBC("_", X, X, X, bits, base),
  4126. DEFVERB("?", 0, 0, 0, unique, find),
  4127. DEFVERB("&", 0, X, X, flatten, minand),
  4128. DEFVERB("|", 0, X, X, reverse, maxor),
  4129. DEFVERBD("|", X, X, 0, round, rotate),
  4130. DEFVERBC("|", 0, X, 0, depth, windows),
  4131. DEFVERB("@", X, 0, X, abs, at),
  4132. DEFVERBD("@", 0, 0, 0, shuffle, member),
  4133. DEFVERBC("@", 0, 0, 0, infix, indexof),
  4134. DEFVERB("{", 0, 0, 0, head, bin),
  4135. DEFVERBD("{", 0, 0, 0, tail, cut),
  4136. DEFVERBC("{", 0, X, X, prefixes, shl),
  4137. DEFVERB("}", 0, X, X, behead, xor),
  4138. DEFVERBD("}", 0, 0, 0, curtail, band),
  4139. DEFVERBC("}", 0, X, X, suffixes, shr),
  4140. DEFVERB("[", X, 0, 0, factors, left),
  4141. DEFVERBD("[", X, X, X, bnot, bor),
  4142. DEFVERBC("[", X, X, 0, primes, parts),
  4143. DEFVERB("]", 0, 0, 0, same, right),
  4144. DEFVERBD("]", 0, X, X, sort, outof),
  4145. DEFVERBC("]", 0, 0, 0, unsort, explode),
  4146. DEFVERBD("`", 0, 0, 0, symbol, apply1),
  4147. DEFVERBC("`", 0, 0, 0, square, apply2),
  4148. DEFVERB("$", 0, 0, 0, shape, reshape),
  4149. DEFVERBD("$", 0, 0, 0, repr, format),
  4150. DEFVERBC("$", X, 0, 0, eye, implode),
  4151. DEFVERBD("d", 0, X, 0, udf1, delete),
  4152. DEFVERBD("p", 0, 0, 0, print, udf2),
  4153. DEFVERBD("P", 0, 0, 0, println, udf2),
  4154. DEFVERBD("c", X, 0, 0, putch, udf2),
  4155. DEFVERBD("s", 0, 0, 0, selfref1, selfref2),
  4156. DEFVERBD("F", 0, 0, 0, read, write),
  4157. DEFVERBD("r", 0, X, X, deal, roll),
  4158. DEFVERBD("t", 0, 0, 0, type, cast),
  4159. DEFVERBD("E", 0, 0, 0, exit, udf2),
  4160. DEFVERBD("y", 0, 0, 0, system, system2),
  4161. DEFVERBD("e", 0, 0, 0, eval, udf2),
  4162. DEFVERBD("i", 0, 0, 0, import, foreign),
  4163. DEFVERBD("L", 0, 0, 0, lines, tackleft),
  4164. DEFVERBD("R", 0, 0, 0, udf1, tackright),
  4165. DEFVERBD("v", 0, 0, 0, value, udf2),
  4166. DEFVERBD("x", 0, 0, 0, show, rematch),
  4167. DEFVERBD("X", 0, 0, 0, udf1, extract)};
  4168. value_t *_adverb_fold_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4169. if (x->tag != ARRAY || !x->val.array->data)
  4170. return x;
  4171. value_t *_v = self->bonds->data[0];
  4172. if (_v->tag != VERB)
  4173. return state->udf;
  4174. verb_t *v = _v->val.verb;
  4175. value_t *t = x->val.array->data[0];
  4176. list_t *tx = x->val.array;
  4177. for (size_t i = 1; i < tx->length; i++)
  4178. t = together(state, v, t, tx->data[i], 0, 0, v->rank[1], v->rank[2]);
  4179. return t;
  4180. }
  4181. value_t *_adverb_fold_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4182. value_t *y) {
  4183. if (y->tag != ARRAY)
  4184. y = verb_enlist(state, NULL, y);
  4185. else if (!y->val.array->data)
  4186. return x;
  4187. value_t *_v = self->bonds->data[0];
  4188. if (_v->tag != VERB)
  4189. return state->udf;
  4190. verb_t *v = _v->val.verb;
  4191. value_t *t = x;
  4192. list_t *ty = y->val.array;
  4193. for (size_t i = 0; i < ty->length; i++)
  4194. t = together(state, v, t, ty->data[i], 0, 0, v->rank[1], v->rank[2]);
  4195. return t;
  4196. }
  4197. value_t *_adverb_scan_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4198. if (x->tag != ARRAY || !x->val.array->data)
  4199. return x;
  4200. value_t *_v = self->bonds->data[0];
  4201. if (_v->tag != VERB)
  4202. return state->udf;
  4203. verb_t *v = _v->val.verb;
  4204. list_t *r = list_new();
  4205. value_t *t = x->val.array->data[0];
  4206. list_t *tx = x->val.array;
  4207. list_push(r, t);
  4208. for (size_t i = 1; i < tx->length; i++) {
  4209. t = together(state, v, t, tx->data[i], 0, 0, v->rank[1], v->rank[2]);
  4210. list_push(r, t);
  4211. }
  4212. return value_new_array(r);
  4213. }
  4214. value_t *_adverb_scan_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4215. value_t *y) {
  4216. if (y->tag != ARRAY || !y->val.array->data)
  4217. return y;
  4218. value_t *_v = self->bonds->data[0];
  4219. if (_v->tag != VERB)
  4220. return state->udf;
  4221. verb_t *v = _v->val.verb;
  4222. list_t *r = list_new();
  4223. value_t *t = x;
  4224. list_t *ty = y->val.array;
  4225. list_push(r, t);
  4226. for (size_t i = 1; i < ty->length; i++) {
  4227. t = together(state, v, t, ty->data[i], 0, 0, v->rank[1], v->rank[2]);
  4228. list_push(r, t);
  4229. }
  4230. return value_new_array(r);
  4231. }
  4232. value_t *_adverb_each_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4233. value_t *_v = self->bonds->data[0];
  4234. if (_v->tag != VERB)
  4235. return state->udf;
  4236. verb_t *v = _v->val.verb;
  4237. if (x->tag != ARRAY)
  4238. return each_rank(state, v, x, 0, 1);
  4239. if (!x->val.array->data)
  4240. return x;
  4241. return each_rank(state, v, x, 0, 1);
  4242. }
  4243. value_t *_adverb_each_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4244. value_t *y) {
  4245. value_t *_v = self->bonds->data[0];
  4246. if (_v->tag != VERB)
  4247. return state->udf;
  4248. verb_t *v = _v->val.verb;
  4249. if (x->tag != ARRAY)
  4250. x = verb_enlist(state, NULL, x);
  4251. if (y->tag != ARRAY)
  4252. y = verb_enlist(state, NULL, y);
  4253. list_t *r = list_new();
  4254. list_t *tx = x->val.array;
  4255. list_t *ty = y->val.array;
  4256. for (size_t i = 0; i < tx->length && i < ty->length; i++)
  4257. list_push(r, together(state, v, tx->data[i], ty->data[i], 0, 0, v->rank[1],
  4258. v->rank[2]));
  4259. return value_new_array(r);
  4260. }
  4261. value_t *_adverb_converge_monad(interpreter_t *state, verb_t *self,
  4262. value_t *x) {
  4263. value_t *_v = self->bonds->data[0];
  4264. if (_v->tag != VERB)
  4265. return state->udf;
  4266. verb_t *v = _v->val.verb;
  4267. value_t *t;
  4268. for (;;) {
  4269. t = x;
  4270. x = each_rank(state, v, x, 0, v->rank[0]);
  4271. if (value_equals(x, t))
  4272. break;
  4273. }
  4274. return x;
  4275. }
  4276. verb_t *conjunction_bond(interpreter_t *state, value_t *x, value_t *y);
  4277. value_t *_adverb_converge_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4278. value_t *y) {
  4279. value_t *_v = self->bonds->data[0];
  4280. if (_v->tag != VERB)
  4281. return state->udf;
  4282. verb_t *v = _v->val.verb;
  4283. if (y->tag != ARRAY)
  4284. return together(state, v, y, x, 0, 0, v->rank[1], v->rank[2]);
  4285. if (!y->val.array->data)
  4286. return x;
  4287. v = conjunction_bond(state, value_new_verb(v), x);
  4288. return each_rank(state, v, y, 0, 1);
  4289. }
  4290. value_t *_adverb_converges_monad(interpreter_t *state, verb_t *self,
  4291. value_t *x) {
  4292. value_t *_v = self->bonds->data[0];
  4293. if (_v->tag != VERB)
  4294. return state->udf;
  4295. list_t *r = list_new();
  4296. value_t *t;
  4297. list_push(r, x);
  4298. for (;;) {
  4299. t = x;
  4300. x = apply_monad(state, _v, x);
  4301. if (value_equals(x, t))
  4302. break;
  4303. list_push(r, x);
  4304. }
  4305. return value_new_array(r);
  4306. }
  4307. value_t *_adverb_converges_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4308. value_t *y) {
  4309. value_t *_v = self->bonds->data[0];
  4310. if (_v->tag != VERB)
  4311. return state->udf;
  4312. verb_t *v = _v->val.verb;
  4313. if (y->tag != ARRAY)
  4314. return together(state, v, y, x, 0, 0, v->rank[1], v->rank[2]);
  4315. if (!y->val.array->data)
  4316. return x;
  4317. v = conjunction_bond(state, x, value_new_verb(v));
  4318. return each_rank(state, v, y, 0, 1);
  4319. }
  4320. value_t *_adverb_eachprior_monad(interpreter_t *state, verb_t *self,
  4321. value_t *x) {
  4322. if (x->tag != ARRAY || x->val.array->length < 2)
  4323. return x;
  4324. value_t *_v = self->bonds->data[0];
  4325. if (_v->tag != VERB)
  4326. return state->udf;
  4327. verb_t *v = _v->val.verb;
  4328. list_t *r = list_new();
  4329. for (size_t i = 1; i < x->val.array->length; i++)
  4330. list_push(r, together(state, v, x->val.array->data[i],
  4331. x->val.array->data[i - 1], 0, 0, v->rank[1],
  4332. v->rank[2]));
  4333. return value_new_array(r);
  4334. }
  4335. value_t *_adverb_eachprior_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4336. value_t *y) {
  4337. if (y->tag != ARRAY || !y->val.array->data)
  4338. return y;
  4339. value_t *_v = self->bonds->data[0];
  4340. if (_v->tag != VERB)
  4341. return state->udf;
  4342. verb_t *v = _v->val.verb;
  4343. list_t *r = list_new();
  4344. for (size_t i = 0; i < y->val.array->length; i++)
  4345. list_push(r, together(state, v, y->val.array->data[i],
  4346. i == 0 ? x : y->val.array->data[i - 1], 0, 0,
  4347. v->rank[1], v->rank[2]));
  4348. return value_new_array(r);
  4349. }
  4350. value_t *_adverb_reflex_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4351. value_t *_v = self->bonds->data[0];
  4352. if (_v->tag != VERB)
  4353. return state->udf;
  4354. verb_t *v = _v->val.verb;
  4355. return together(state, v, x, x, 0, 0, v->rank[1], v->rank[2]);
  4356. }
  4357. value_t *_adverb_reflex_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4358. value_t *y) {
  4359. value_t *_v = self->bonds->data[0];
  4360. if (_v->tag != VERB)
  4361. return state->udf;
  4362. verb_t *v = _v->val.verb;
  4363. return together(state, v, y, x, 0, 0, v->rank[1], v->rank[2]);
  4364. }
  4365. value_t *_adverb_amend_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4366. return state->udf;
  4367. }
  4368. value_t *_adverb_amend_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4369. value_t *y) {
  4370. if (x->tag != ARRAY)
  4371. x = verb_enlist(state, NULL, x);
  4372. value_t *v = self->bonds->data[0];
  4373. if (v->tag != ARRAY)
  4374. v = verb_enlist(state, NULL, v);
  4375. if (y->tag != ARRAY)
  4376. y = verb_enlist(state, NULL, y);
  4377. list_t *r = list_copy(y->val.array);
  4378. size_t l = x->val.array->length;
  4379. list_t *t = v->val.array;
  4380. for (size_t i = 0; i < t->length; i++) {
  4381. value_t *n = t->data[i];
  4382. if (n->tag != NUMBER)
  4383. break;
  4384. list_set(r, n->val.number, list_index(x->val.array, i < l ? i : l - 1));
  4385. }
  4386. return value_new_array(r);
  4387. }
  4388. value_t *_adverb_filter_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4389. value_t *_v = self->bonds->data[0];
  4390. if (_v->tag != VERB)
  4391. return state->udf;
  4392. if (x->tag != ARRAY)
  4393. x = verb_enlist(state, NULL, x);
  4394. else if (!x->val.array->data)
  4395. return x;
  4396. verb_t *v = _v->val.verb;
  4397. list_t *r = list_new();
  4398. for (size_t i = 0; i < x->val.array->length; i++) {
  4399. value_t *b = each_rank(state, v, x->val.array->data[i], 0, v->rank[0]);
  4400. if (value_is_truthy(b))
  4401. list_push(r, x->val.array->data[i]);
  4402. }
  4403. return value_new_array(r);
  4404. }
  4405. value_t *_adverb_filter_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4406. value_t *y) {
  4407. return state->udf;
  4408. }
  4409. value_t *_adverb_span_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4410. value_t *v = self->bonds->data[0];
  4411. if (v->tag != VERB)
  4412. return state->udf;
  4413. if (x->tag != ARRAY)
  4414. x = verb_enlist(state, NULL, x);
  4415. else if (!x->val.array->data)
  4416. return x;
  4417. list_t *r = list_new();
  4418. list_t *p = list_new();
  4419. for (size_t i = 0; i < x->val.array->length; i++) {
  4420. value_t *b = apply_monad(state, v, x->val.array->data[i]);
  4421. if (value_is_truthy(b)) {
  4422. list_push(r, value_new_array(p));
  4423. p = list_new();
  4424. } else
  4425. list_push(p, x->val.array->data[i]);
  4426. }
  4427. list_push(r, value_new_array(p));
  4428. return value_new_array(r);
  4429. }
  4430. value_t *_adverb_span_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4431. value_t *y) {
  4432. value_t *_v = self->bonds->data[0];
  4433. if (_v->tag != VERB)
  4434. return state->udf;
  4435. verb_t *v = _v->val.verb;
  4436. value_t *r = verb_windows(state, NULL, x, y);
  4437. return each_rank(state, v, r, 0, 1);
  4438. }
  4439. value_t *_adverb_inverse_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4440. value_t *_v = self->bonds->data[0];
  4441. if (_v->tag != VERB)
  4442. return state->udf;
  4443. verb_t *v = _v->val.verb;
  4444. verb_t *iv = table_get(Inverses, v->name);
  4445. if (!iv)
  4446. return state->udf;
  4447. return each_rank(state, iv, x, 0, iv->rank[0]);
  4448. }
  4449. value_t *_adverb_inverse_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4450. value_t *y) {
  4451. value_t *_v = self->bonds->data[0];
  4452. if (_v->tag != VERB)
  4453. return state->udf;
  4454. verb_t *v = _v->val.verb;
  4455. verb_t *iv = table_get(Inverses, v->name);
  4456. if (!iv)
  4457. return state->udf;
  4458. value_t *a = each_rank(state, iv, x, 0, iv->rank[0]);
  4459. value_t *b = each_rank(state, iv, y, 0, iv->rank[0]);
  4460. return apply_dyad(state, _v, a, b);
  4461. }
  4462. #define ADVERB(__name, __symb) \
  4463. verb_t *adverb_##__name(interpreter_t *state, value_t *v) { \
  4464. verb_t *nv = verb_new(); \
  4465. nv->bonds = list_newk(1); \
  4466. nv->bonds->data[0] = v; \
  4467. char *r = value_show(v); \
  4468. size_t l = strlen(r) + strlen(__symb) + 1; \
  4469. nv->name = malloc_checked_atomic(l); \
  4470. snprintf(nv->name, l, "%s" __symb, r); \
  4471. GC_FREE(r); \
  4472. nv->rank[0] = 0; \
  4473. nv->monad = _adverb_##__name##_monad; \
  4474. nv->dyad = _adverb_##__name##_dyad; \
  4475. return nv; \
  4476. }
  4477. ADVERB(fold, "/");
  4478. ADVERB(converge, "/.");
  4479. ADVERB(scan, "\\");
  4480. ADVERB(converges, "\\.");
  4481. ADVERB(each, "\"");
  4482. ADVERB(eachprior, "\".");
  4483. ADVERB(reflex, ";.");
  4484. ADVERB(amend, "`");
  4485. ADVERB(filter, "&.");
  4486. ADVERB(span, "/:");
  4487. ADVERB(inverse, "-:");
  4488. adverb_t ADVERBS[] = {
  4489. {"/", adverb_fold, NULL}, {"/.", adverb_converge, NULL},
  4490. {"\\", adverb_scan, NULL}, {"\\.", adverb_converges, NULL},
  4491. {"\"", adverb_each, NULL}, {"\".", adverb_eachprior, NULL},
  4492. {";.", adverb_reflex, NULL}, {"`", adverb_amend, NULL},
  4493. {"&.", adverb_filter, NULL}, {"/:", adverb_span, NULL},
  4494. {"-:", adverb_inverse, NULL}};
  4495. value_t *_conjunction_bond_monad(interpreter_t *state, verb_t *self,
  4496. value_t *x) {
  4497. value_t *v1 = self->bonds->data[0];
  4498. value_t *v2 = self->bonds->data[1];
  4499. if (v1->tag == VERB && v2->tag == VERB)
  4500. return apply_monad(state, v1, apply_monad(state, v2, x));
  4501. else if (v1->tag == VERB)
  4502. return apply_dyad(state, v1, x, v2);
  4503. else if (v2->tag == VERB)
  4504. return apply_dyad(state, v2, v1, x);
  4505. else
  4506. return state->udf;
  4507. }
  4508. value_t *_conjunction_bond_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4509. value_t *y) {
  4510. value_t *v1 = self->bonds->data[0];
  4511. value_t *v2 = self->bonds->data[1];
  4512. if (v1->tag == VERB && v2->tag == VERB)
  4513. return apply_monad(state, v1, apply_dyad(state, v2, x, y));
  4514. else if (v1->tag == VERB)
  4515. return apply_dyad(state, v1, apply_dyad(state, v1, x, y), v2);
  4516. else if (v2->tag == VERB)
  4517. return apply_dyad(state, v2, v1, apply_dyad(state, v2, x, y));
  4518. else
  4519. return state->udf;
  4520. }
  4521. value_t *_conjunction_pick_monad(interpreter_t *state, verb_t *self,
  4522. value_t *x) {
  4523. value_t *v1 = self->bonds->data[0];
  4524. value_t *v2 = self->bonds->data[1];
  4525. if (v1->tag != VERB || v2->tag != ARRAY)
  4526. return state->udf;
  4527. value_t *n = apply_monad(state, v1, x);
  4528. value_t *f = verb_at(state, NULL, v2, n);
  4529. return apply_monad(state, f, x);
  4530. }
  4531. value_t *_conjunction_pick_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4532. value_t *y) {
  4533. value_t *v1 = self->bonds->data[0];
  4534. value_t *v2 = self->bonds->data[1];
  4535. if (v1->tag != VERB || v2->tag != ARRAY)
  4536. return state->udf;
  4537. value_t *n = apply_dyad(state, v1, x, y);
  4538. value_t *f = verb_at(state, NULL, v2, n);
  4539. return apply_dyad(state, f, x, y);
  4540. }
  4541. value_t *_conjunction_while_monad(interpreter_t *state, verb_t *self,
  4542. value_t *x) {
  4543. value_t *v1 = self->bonds->data[0];
  4544. value_t *v2 = self->bonds->data[1];
  4545. if (v1->tag == VERB) {
  4546. for (;;) {
  4547. if (!value_is_truthy(apply_monad(state, v1, x)))
  4548. break;
  4549. x = apply_monad(state, v2, x);
  4550. }
  4551. } else if (v1->tag == NUMBER) {
  4552. size_t k = (size_t)fabs(v1->val.number);
  4553. for (size_t i = 0; i < k; i++)
  4554. x = apply_monad(state, v2, x);
  4555. }
  4556. return x;
  4557. }
  4558. value_t *_conjunction_while_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4559. value_t *y) {
  4560. value_t *v1 = self->bonds->data[0];
  4561. value_t *v2 = self->bonds->data[1];
  4562. if (v1->tag == VERB) {
  4563. for (;;) {
  4564. if (!value_is_truthy(apply_dyad(state, v1, x, y)))
  4565. break;
  4566. x = apply_dyad(state, v2, x, y);
  4567. }
  4568. } else if (v1->tag == NUMBER) {
  4569. size_t k = (size_t)fabs(v1->val.number);
  4570. for (size_t i = 0; i < k; i++)
  4571. x = apply_dyad(state, v2, x, y);
  4572. }
  4573. return x;
  4574. }
  4575. value_t *_conjunction_rank_monad(interpreter_t *state, verb_t *self,
  4576. value_t *x) {
  4577. value_t *v1 = self->bonds->data[0];
  4578. value_t *v2 = self->bonds->data[1];
  4579. if (v1->tag != VERB || v2->tag != NUMBER)
  4580. return state->udf;
  4581. unsigned int rank =
  4582. v2->val.number == INFINITY ? UINT_MAX : fabs(v2->val.number);
  4583. return each_rank(state, v1->val.verb, x, 0, rank);
  4584. }
  4585. value_t *_conjunction_rank_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4586. value_t *y) {
  4587. value_t *v1 = self->bonds->data[0];
  4588. value_t *v2 = self->bonds->data[1];
  4589. if (v1->tag != VERB)
  4590. return state->udf;
  4591. unsigned int rl;
  4592. unsigned int rr;
  4593. if (v2->tag == NUMBER)
  4594. rl = rr = v2->val.number == INFINITY ? UINT_MAX : fabs(v2->val.number);
  4595. else if (v2->tag == ARRAY && v2->val.array->length == 2) {
  4596. value_t *a = v2->val.array->data[0];
  4597. value_t *b = v2->val.array->data[1];
  4598. if (a->tag != NUMBER)
  4599. return state->udf;
  4600. rl = a->val.number == INFINITY ? UINT_MAX : fabs(a->val.number);
  4601. if (b->tag != NUMBER)
  4602. return state->udf;
  4603. rr = b->val.number == INFINITY ? UINT_MAX : fabs(b->val.number);
  4604. } else
  4605. return state->udf;
  4606. return together(state, v1->val.verb, x, y, 0, 0, rl, rr);
  4607. }
  4608. value_t *_conjunction_monaddyad_monad(interpreter_t *state, verb_t *self,
  4609. value_t *x) {
  4610. value_t *v = self->bonds->data[0];
  4611. if (v->tag != VERB)
  4612. return state->udf;
  4613. return each_rank(state, v->val.verb, x, 0, v->val.verb->rank[0]);
  4614. }
  4615. value_t *_conjunction_monaddyad_dyad(interpreter_t *state, verb_t *self,
  4616. value_t *x, value_t *y) {
  4617. value_t *v = self->bonds->data[1];
  4618. if (v->tag != VERB)
  4619. return state->udf;
  4620. return together(state, v->val.verb, x, y, 0, 0, v->val.verb->rank[1],
  4621. v->val.verb->rank[2]);
  4622. }
  4623. value_t *_conjunction_if_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4624. value_t *v1 = self->bonds->data[0];
  4625. value_t *v2 = self->bonds->data[1];
  4626. if (v1->tag != VERB || v2->tag != VERB)
  4627. return state->udf;
  4628. value_t *b = apply_monad(state, v2, x);
  4629. if (value_is_truthy(b))
  4630. return x;
  4631. return apply_monad(state, v1, x);
  4632. }
  4633. value_t *_conjunction_if_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4634. value_t *y) {
  4635. value_t *v1 = self->bonds->data[0];
  4636. value_t *v2 = self->bonds->data[1];
  4637. if (v1->tag != VERB || v2->tag != VERB)
  4638. return state->udf;
  4639. value_t *b = apply_dyad(state, v2, x, y);
  4640. if (value_is_truthy(b))
  4641. return y;
  4642. return apply_dyad(state, v1, x, y);
  4643. }
  4644. value_t *_conjunction_under_monad(interpreter_t *state, verb_t *self,
  4645. value_t *x) {
  4646. value_t *v1 = self->bonds->data[0];
  4647. value_t *v2 = self->bonds->data[1];
  4648. if (v1->tag != VERB || v2->tag != VERB)
  4649. return state->udf;
  4650. verb_t *iv = table_get(Inverses, v2->val.verb->name);
  4651. if (!iv)
  4652. return state->udf;
  4653. value_t *v = apply_monad(state, v2, x);
  4654. v = apply_monad(state, v1, v);
  4655. return each_rank(state, iv, v, 0, iv->rank[0]);
  4656. }
  4657. value_t *_conjunction_under_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4658. value_t *y) {
  4659. value_t *v1 = self->bonds->data[0];
  4660. value_t *v2 = self->bonds->data[1];
  4661. if (v1->tag != VERB || v2->tag != VERB)
  4662. return state->udf;
  4663. verb_t *iv = table_get(Inverses, v2->val.verb->name);
  4664. if (!iv)
  4665. return state->udf;
  4666. value_t *a = apply_monad(state, v2, x);
  4667. value_t *b = apply_monad(state, v2, y);
  4668. value_t *v = apply_dyad(state, v1, a, b);
  4669. return each_rank(state, iv, v, 0, iv->rank[0]);
  4670. }
  4671. value_t *_conjunction_collect_monad(interpreter_t *state, verb_t *self,
  4672. value_t *x) {
  4673. value_t *v1 = self->bonds->data[0];
  4674. value_t *v2 = self->bonds->data[1];
  4675. list_t *r = list_new();
  4676. if (v1->tag == VERB) {
  4677. for (;;) {
  4678. if (!value_is_truthy(apply_monad(state, v1, x)))
  4679. break;
  4680. list_push(r, x);
  4681. x = apply_monad(state, v2, x);
  4682. }
  4683. } else if (v1->tag == NUMBER) {
  4684. size_t k = (size_t)fabs(v1->val.number);
  4685. for (size_t i = 0; i < k; i++) {
  4686. list_push(r, x);
  4687. x = apply_monad(state, v2, x);
  4688. }
  4689. }
  4690. return value_new_array(r);
  4691. }
  4692. value_t *_conjunction_collect_dyad(interpreter_t *state, verb_t *self,
  4693. value_t *x, value_t *y) {
  4694. value_t *v1 = self->bonds->data[0];
  4695. value_t *v2 = self->bonds->data[1];
  4696. list_t *r = list_new();
  4697. if (v1->tag == VERB) {
  4698. for (;;) {
  4699. if (!value_is_truthy(apply_dyad(state, v1, x, y)))
  4700. break;
  4701. list_push(r, x);
  4702. x = apply_dyad(state, v2, x, y);
  4703. }
  4704. } else if (v1->tag == NUMBER) {
  4705. size_t k = (size_t)fabs(v1->val.number);
  4706. for (size_t i = 0; i < k; i++) {
  4707. list_push(r, x);
  4708. x = apply_dyad(state, v2, x, y);
  4709. }
  4710. }
  4711. return x;
  4712. }
  4713. #define CONJUNCTION(__name, __symb) \
  4714. verb_t *conjunction_##__name(interpreter_t *state, value_t *x, value_t *y) { \
  4715. verb_t *nv = verb_new(); \
  4716. nv->bonds = list_newk(2); \
  4717. nv->bonds->data[0] = x; \
  4718. nv->bonds->data[1] = y; \
  4719. char *rx = value_show(x); \
  4720. char *ry = value_show(y); \
  4721. size_t l = strlen(rx) + strlen(ry) + strlen(__symb) + 1; \
  4722. nv->name = malloc_checked_atomic(l); \
  4723. snprintf(nv->name, l, "%s" __symb "%s", rx, ry); \
  4724. GC_FREE(rx); \
  4725. GC_FREE(ry); \
  4726. nv->rank[0] = 0; \
  4727. nv->rank[1] = 0; \
  4728. nv->rank[1] = 0; \
  4729. nv->monad = _conjunction_##__name##_monad; \
  4730. nv->dyad = _conjunction_##__name##_dyad; \
  4731. return nv; \
  4732. }
  4733. CONJUNCTION(bond, ";");
  4734. CONJUNCTION(pick, "?.");
  4735. CONJUNCTION(while, "?:");
  4736. CONJUNCTION(rank, "\":");
  4737. CONJUNCTION(monaddyad, ";:");
  4738. CONJUNCTION(if, "&:");
  4739. CONJUNCTION(under, "^:");
  4740. CONJUNCTION(collect, "\\:");
  4741. adverb_t CONJUNCTIONS[] = {
  4742. {";", NULL, conjunction_bond}, {"?.", NULL, conjunction_pick},
  4743. {"?:", NULL, conjunction_while}, {"\":", NULL, conjunction_rank},
  4744. {";:", NULL, conjunction_monaddyad}, {"&:", NULL, conjunction_if},
  4745. {"^:", NULL, conjunction_under}, {"\\:", NULL, conjunction_collect}};
  4746. #define countof(x) (sizeof(x) / sizeof((x)[0]))
  4747. #define FINDER(kind, rname, table) \
  4748. kind *find_##rname(char *s) { \
  4749. for (size_t i = 0; i < countof(table); i++) { \
  4750. if (strcmp(table[i].name, s) == 0) \
  4751. return &table[i]; \
  4752. } \
  4753. return NULL; \
  4754. }
  4755. FINDER(verb_t, verb, VERBS);
  4756. FINDER(adverb_t, adverb, ADVERBS);
  4757. FINDER(adverb_t, conjunction, CONJUNCTIONS);
  4758. node_t *node_new(enum node_tag_t tag) {
  4759. node_t *node = malloc_checked(sizeof(node_t));
  4760. node->tag = tag;
  4761. return node;
  4762. }
  4763. node_t *node_new_strand(list_t *l) {
  4764. node_t *node = malloc_checked(sizeof(node_t));
  4765. node->tag = N_STRAND;
  4766. node->l = l;
  4767. return node;
  4768. }
  4769. node_t *node_new_literal(value_t *v) {
  4770. node_t *node = malloc_checked(sizeof(node_t));
  4771. node->tag = N_LITERAL;
  4772. node->v = v;
  4773. return node;
  4774. }
  4775. node_t *node_new1(enum node_tag_t tag, node_t *a) {
  4776. node_t *node = malloc_checked(sizeof(node_t));
  4777. node->tag = tag;
  4778. node->a = a;
  4779. return node;
  4780. }
  4781. node_t *node_new2(enum node_tag_t tag, node_t *a, node_t *b) {
  4782. node_t *node = malloc_checked(sizeof(node_t));
  4783. node->tag = tag;
  4784. node->a = a;
  4785. node->b = b;
  4786. return node;
  4787. }
  4788. node_t *node_new3(enum node_tag_t tag, node_t *a, node_t *b, node_t *c) {
  4789. node_t *node = malloc_checked(sizeof(node_t));
  4790. node->tag = tag;
  4791. node->a = a;
  4792. node->b = b;
  4793. node->c = c;
  4794. return node;
  4795. }
  4796. typedef struct {
  4797. lexer_t *lexer;
  4798. interpreter_t *state;
  4799. size_t pos;
  4800. size_t end;
  4801. size_t dp;
  4802. bool bn;
  4803. } parser_t;
  4804. parser_t *parser_new(interpreter_t *state) {
  4805. parser_t *parser = malloc_checked(sizeof(parser_t));
  4806. parser->state = state;
  4807. return parser;
  4808. }
  4809. void parser_error(parser_t *parser, char *s) { fatal(s); }
  4810. bool parser_done(parser_t *parser) { return parser->pos >= parser->end; }
  4811. token_t *parser_lookahead(parser_t *parser, size_t offset) {
  4812. size_t pos = parser->pos + offset;
  4813. if (pos >= parser->end)
  4814. return NULL;
  4815. return list_index(parser->lexer->tokens, pos);
  4816. }
  4817. bool parser_stop(parser_t *parser) {
  4818. token_t *tok = parser_lookahead(parser, 0);
  4819. if (!tok)
  4820. return true;
  4821. return tok->tag == T_RPAR;
  4822. }
  4823. void parser_eat(parser_t *parser) {
  4824. if (!parser_done(parser))
  4825. parser->pos++;
  4826. }
  4827. node_t *parser_parse_expr(parser_t *parser);
  4828. node_t *parser_parse_verb(parser_t *parser) {
  4829. token_t *tok = parser_lookahead(parser, 0);
  4830. if (!tok || tok->tag != T_PUNCT)
  4831. return NULL;
  4832. verb_t *verb = find_verb(tok->text);
  4833. if (!verb)
  4834. return NULL;
  4835. return node_new_literal(value_new_verb(verb));
  4836. }
  4837. value_t *_adverb_wrapper_monad(interpreter_t *state, verb_t *self, value_t *x) {
  4838. adverb_t *av = self->bonds->data[0];
  4839. if (x->tag != VERB)
  4840. return state->udf;
  4841. return value_new_verb(av->adverb(state, x));
  4842. }
  4843. value_t *_adverb_wrapper_dyad(interpreter_t *state, verb_t *self, value_t *x,
  4844. value_t *y) {
  4845. adverb_t *av = self->bonds->data[0];
  4846. if (x->tag != VERB)
  4847. return state->udf;
  4848. verb_t *v = av->adverb(state, x);
  4849. return each_rank(state, v, y, 0, v->rank[0]);
  4850. }
  4851. node_t *parser_parse_adverb_atom(parser_t *parser) {
  4852. token_t *tok = parser_lookahead(parser, 0);
  4853. if (!tok || tok->tag != T_PUNCT)
  4854. return NULL;
  4855. adverb_t *adverb = find_adverb(tok->text);
  4856. if (!adverb)
  4857. return NULL;
  4858. verb_t *nv = verb_new();
  4859. nv->name = strdup_checked(tok->text);
  4860. nv->bonds = list_newk(1);
  4861. nv->bonds->data[0] = adverb;
  4862. nv->rank[0] = 0;
  4863. nv->rank[1] = 0;
  4864. nv->rank[2] = 0;
  4865. nv->monad = _adverb_wrapper_monad;
  4866. nv->dyad = _adverb_wrapper_dyad;
  4867. return node_new_literal(value_new_verb(nv));
  4868. }
  4869. value_t *_conjunction_wrapper_dyad(interpreter_t *state, verb_t *self,
  4870. value_t *x, value_t *y) {
  4871. adverb_t *av = self->bonds->data[0];
  4872. return value_new_verb(av->conjunction(state, x, y));
  4873. }
  4874. node_t *parser_parse_conjunction_atom(parser_t *parser) {
  4875. token_t *tok = parser_lookahead(parser, 0);
  4876. if (!tok || tok->tag != T_PUNCT)
  4877. return NULL;
  4878. adverb_t *adverb = find_conjunction(tok->text);
  4879. if (!adverb)
  4880. return NULL;
  4881. verb_t *nv = verb_new();
  4882. nv->name = strdup_checked(tok->text);
  4883. nv->bonds = list_newk(1);
  4884. nv->bonds->data[0] = adverb;
  4885. nv->rank[0] = 0;
  4886. nv->rank[1] = 0;
  4887. nv->rank[2] = 0;
  4888. nv->monad = NULL;
  4889. nv->dyad = _conjunction_wrapper_dyad;
  4890. return node_new_literal(value_new_verb(nv));
  4891. }
  4892. node_t *parser_parse_atom(parser_t *parser) {
  4893. token_t *tok = parser_lookahead(parser, 0);
  4894. node_t *node = NULL;
  4895. switch (tok->tag) {
  4896. case T_RPAR:
  4897. parser_error(parser, "unmatched");
  4898. case T_LPAR:
  4899. parser_eat(parser);
  4900. tok = parser_lookahead(parser, 0);
  4901. if (tok && tok->tag == T_RPAR) {
  4902. node = node_new_literal(parser->state->unit);
  4903. break;
  4904. }
  4905. parser->dp++;
  4906. node = parser_parse_expr(parser);
  4907. if (parser->bn)
  4908. node->dp = 2;
  4909. else
  4910. node->dp = parser->dp;
  4911. parser->dp--;
  4912. tok = parser_lookahead(parser, 0);
  4913. if (!tok || tok->tag != T_RPAR)
  4914. parser_error(parser, "unmatched");
  4915. break;
  4916. case T_PUNCT:
  4917. node = parser_parse_verb(parser);
  4918. if (!node)
  4919. node = parser_parse_adverb_atom(parser);
  4920. if (!node)
  4921. node = parser_parse_conjunction_atom(parser);
  4922. if (!node)
  4923. parser_error(parser, "parse");
  4924. break;
  4925. case T_NUMBER:
  4926. node = node_new_literal(value_new_number(strtod(tok->text, NULL)));
  4927. break;
  4928. case T_BNUMBER: {
  4929. if (!tok->text[1])
  4930. parser_error(parser, "trailing-base");
  4931. int base = tok->text[0] == 'x' ? 16 : tok->text[0] == 'b' ? 2 : 8;
  4932. node =
  4933. node_new_literal(value_new_number(strtol(tok->text + 1, NULL, base)));
  4934. } break;
  4935. case T_NAME:
  4936. node = node_new_literal(value_new_symbol(strdup_checked(tok->text)));
  4937. break;
  4938. case T_QUOTE:
  4939. if (!*tok->text)
  4940. node = node_new_literal(parser->state->unit);
  4941. else if (!*(tok->text + 1))
  4942. node = node_new_literal(value_new_char(tok->text[0]));
  4943. else {
  4944. size_t z = strlen(tok->text);
  4945. list_t *r = list_newk(z);
  4946. for (size_t i = 0; i < z; i++)
  4947. r->data[i] = CHARS[(int)tok->text[i]];
  4948. node = node_new_literal(value_new_array(r));
  4949. }
  4950. break;
  4951. }
  4952. if (!node)
  4953. parser_error(parser, "parse");
  4954. parser_eat(parser);
  4955. return node;
  4956. }
  4957. bool isunb(interpreter_t *state, char *s) {
  4958. if (state->args->data) {
  4959. list_t *args = list_index(state->args, -1);
  4960. size_t argc = args->length - 1;
  4961. if (argc == 2 && strcmp(s, "y") == 0)
  4962. return false;
  4963. else if (strcmp(s, "x") == 0)
  4964. return false;
  4965. } else if (table_has(state->env, s))
  4966. return false;
  4967. return true;
  4968. }
  4969. node_t *parser_parse_sequence(parser_t *parser, node_t *a,
  4970. enum token_tag_t tag) {
  4971. token_t *tok;
  4972. if ((tok = parser_lookahead(parser, 0)) &&
  4973. (tok->tag == tag || (tag == T_NUMBER && tok->tag == T_BNUMBER))) {
  4974. if (tag == T_NAME && !isunb(parser->state, tok->text))
  4975. return NULL;
  4976. list_t *as = list_new();
  4977. list_push(as, a->v);
  4978. do {
  4979. if (tag == T_NAME && tok->tag == T_NAME &&
  4980. !isunb(parser->state, tok->text))
  4981. break;
  4982. a = parser_parse_atom(parser);
  4983. list_push(as, a->v);
  4984. } while ((tok = parser_lookahead(parser, 0)) &&
  4985. (tok->tag == tag || (tag == T_NUMBER && tok->tag == T_BNUMBER)));
  4986. return node_new_literal(value_new_array(as));
  4987. }
  4988. return NULL;
  4989. }
  4990. node_t *_parser_parse_noun(parser_t *parser) {
  4991. node_t *n;
  4992. node_t *a = parser_parse_atom(parser);
  4993. if (a->tag == N_LITERAL && a->v->tag == NUMBER &&
  4994. (n = parser_parse_sequence(parser, a, T_NUMBER)))
  4995. return n;
  4996. else if (a->tag == N_LITERAL && a->v->tag == SYMBOL &&
  4997. isunb(parser->state, a->v->val.symbol) &&
  4998. (n = parser_parse_sequence(parser, a, T_NAME)))
  4999. return n;
  5000. else if (a->tag == N_LITERAL &&
  5001. ((a->v->tag == ARRAY && is_char_array(a->v->val.array)) ||
  5002. a->v->tag == CHAR) &&
  5003. (n = parser_parse_sequence(parser, a, T_QUOTE)))
  5004. return n;
  5005. return a;
  5006. }
  5007. node_t *parser_parse_noun(parser_t *parser, bool flat) {
  5008. node_t *a = flat ? parser_parse_atom(parser) : _parser_parse_noun(parser);
  5009. token_t *tok;
  5010. if ((tok = parser_lookahead(parser, 0)) && tok->tag == T_PUNCT &&
  5011. strcmp(tok->text, ",:") == 0) {
  5012. parser_eat(parser);
  5013. list_t *l = list_new();
  5014. list_push(l, a);
  5015. for (;;) {
  5016. if (parser_stop(parser))
  5017. parser_error(parser, "trailing-strand");
  5018. a = flat ? parser_parse_atom(parser) : _parser_parse_noun(parser);
  5019. list_push(l, a);
  5020. if (!((tok = parser_lookahead(parser, 0)) && tok->tag == T_PUNCT &&
  5021. strcmp(tok->text, ",:") == 0))
  5022. break;
  5023. parser_eat(parser);
  5024. }
  5025. return node_new_strand(l);
  5026. }
  5027. return a;
  5028. }
  5029. bool parser_node_is_verbal(parser_t *parser, node_t *n) {
  5030. value_t *v;
  5031. if (n->tag == N_FUN)
  5032. return true;
  5033. else if (n->tag == N_ADV || n->tag == N_CONJ || n->tag == N_PARTIAL_CONJ)
  5034. return true;
  5035. else if (n->tag == N_FORK || n->tag == N_HOOK || n->tag == N_BOND ||
  5036. n->tag == N_OVER)
  5037. return true;
  5038. else if (n->tag == N_LITERAL && n->v->tag == VERB)
  5039. return true;
  5040. else if (n->tag == N_LITERAL && n->v->tag == SYMBOL &&
  5041. (v = table_get(parser->state->env, n->v->val.symbol)) &&
  5042. v->tag == VERB)
  5043. return true;
  5044. return false;
  5045. }
  5046. node_t *parser_parse_adverb(parser_t *parser, node_t *v, bool *flag) {
  5047. token_t *tok;
  5048. adverb_t *adv;
  5049. node_t *t;
  5050. for (;;) {
  5051. tok = parser_lookahead(parser, 0);
  5052. if (!tok || tok->tag != T_PUNCT)
  5053. break;
  5054. if ((adv = find_adverb(tok->text))) {
  5055. if (flag)
  5056. *flag = true;
  5057. parser_eat(parser);
  5058. t = node_new(N_ADV);
  5059. t->av = adv;
  5060. t->a = v;
  5061. v = t;
  5062. } else
  5063. break;
  5064. }
  5065. return v;
  5066. }
  5067. node_t *parser_parse_conjunction(parser_t *parser, node_t *v, bool *flag) {
  5068. token_t *tok;
  5069. adverb_t *adv;
  5070. node_t *t;
  5071. for (;;) {
  5072. tok = parser_lookahead(parser, 0);
  5073. if (!tok || tok->tag != T_PUNCT)
  5074. break;
  5075. if ((adv = find_conjunction(tok->text))) {
  5076. if (flag)
  5077. *flag = true;
  5078. parser_eat(parser);
  5079. if (parser_stop(parser)) {
  5080. t = node_new(N_PARTIAL_CONJ);
  5081. t->av = adv;
  5082. t->a = v;
  5083. } else {
  5084. t = node_new(N_CONJ);
  5085. t->av = adv;
  5086. t->a = v;
  5087. t->b = parser_parse_noun(parser, true);
  5088. }
  5089. v = t;
  5090. } else
  5091. break;
  5092. }
  5093. return v;
  5094. }
  5095. bool is_apply(node_t *n) {
  5096. return n->tag == N_LITERAL && n->v->tag == VERB &&
  5097. (strcmp(n->v->val.verb->name, "`.") == 0 ||
  5098. strcmp(n->v->val.verb->name, "`:") == 0);
  5099. }
  5100. bool is_obverse(node_t *n) {
  5101. return n->tag == N_LITERAL && n->v->tag == VERB &&
  5102. strcmp(n->v->val.verb->name, "::") == 0;
  5103. }
  5104. node_t *parser_parse_expr(parser_t *parser) {
  5105. token_t *tmp;
  5106. list_t *ns = list_new();
  5107. while (!parser_stop(parser)) {
  5108. if (!ns->data && (tmp = parser_lookahead(parser, 0)) &&
  5109. tmp->tag == T_PUNCT && strcmp(tmp->text, ":") == 0) {
  5110. parser_eat(parser);
  5111. node_t *r = parser_parse_expr(parser);
  5112. if (!r)
  5113. r = node_new_literal(parser->state->nil);
  5114. return node_new1(N_FUN, r);
  5115. }
  5116. node_t *n = parser_parse_noun(parser, false);
  5117. if (!ns->data && n->tag == N_LITERAL && n->v->tag == SYMBOL &&
  5118. (tmp = parser_lookahead(parser, 0)) && tmp->tag == T_PUNCT &&
  5119. strcmp(tmp->text, ":") == 0) {
  5120. parser_eat(parser);
  5121. bool t = parser->bn;
  5122. parser->bn = true;
  5123. node_t *r = parser_parse_expr(parser);
  5124. parser->bn = t;
  5125. return node_new2(N_BIND, n, r);
  5126. }
  5127. for (;;) {
  5128. bool flag = false;
  5129. n = parser_parse_adverb(parser, n, &flag);
  5130. n = parser_parse_conjunction(parser, n, &flag);
  5131. if (!flag)
  5132. break;
  5133. }
  5134. list_push(ns, n);
  5135. }
  5136. size_t len;
  5137. node_t *l, *m, *r;
  5138. for (;;) {
  5139. len = ns->length;
  5140. if (len < 2)
  5141. break;
  5142. if (len >= 3 &&
  5143. (is_apply(list_index(ns, -2)) || is_obverse(list_index(ns, -2))) &&
  5144. parser_node_is_verbal(parser, list_index(ns, -1))) {
  5145. r = list_pop(ns);
  5146. m = list_pop(ns);
  5147. l = list_pop(ns);
  5148. list_push(ns, node_new3(N_DYAD, m, l, r));
  5149. } else if (len >= 3 && !parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5150. parser_node_is_verbal(parser, list_index(ns, -2)) &&
  5151. !parser_node_is_verbal(parser, list_index(ns, -3))) {
  5152. r = list_pop(ns);
  5153. m = list_pop(ns);
  5154. l = list_pop(ns);
  5155. list_push(ns, node_new3(N_DYAD, m, l, r));
  5156. } else if (len >= 3 && parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5157. parser_node_is_verbal(parser, list_index(ns, -2)) &&
  5158. parser_node_is_verbal(parser, list_index(ns, -3))) {
  5159. r = list_pop(ns);
  5160. m = list_pop(ns);
  5161. l = list_pop(ns);
  5162. list_push(ns, node_new3(N_FORK, l, m, r));
  5163. } else if (len >= 3 && parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5164. parser_node_is_verbal(parser, list_index(ns, -2)) &&
  5165. !parser_node_is_verbal(parser, list_index(ns, -3))) {
  5166. r = list_pop(ns);
  5167. m = list_pop(ns);
  5168. l = list_pop(ns);
  5169. list_push(ns, node_new3(N_OVER, l, m, r));
  5170. } else if (len >= 2 && is_apply(list_index(ns, -1))) {
  5171. r = list_pop(ns);
  5172. l = list_pop(ns);
  5173. list_push(ns, node_new2(N_BOND, r, l));
  5174. } else if (len >= 2 && !parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5175. parser_node_is_verbal(parser, list_index(ns, -2))) {
  5176. r = list_pop(ns);
  5177. l = list_pop(ns);
  5178. list_push(ns, node_new2(N_MONAD, l, r));
  5179. } else if (len >= 2 && parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5180. parser_node_is_verbal(parser, list_index(ns, -2))) {
  5181. r = list_pop(ns);
  5182. l = list_pop(ns);
  5183. list_push(ns, node_new2(N_HOOK, l, r));
  5184. } else if (len >= 2 && parser_node_is_verbal(parser, list_index(ns, -1)) &&
  5185. !parser_node_is_verbal(parser, list_index(ns, -2))) {
  5186. r = list_pop(ns);
  5187. l = list_pop(ns);
  5188. list_push(ns, node_new2(N_BOND, r, l));
  5189. } else if (len >= 3) {
  5190. r = list_pop(ns);
  5191. m = list_pop(ns);
  5192. l = list_pop(ns);
  5193. list_push(ns, node_new3(N_INDEX2, m, l, r));
  5194. } else if (len >= 2) {
  5195. r = list_pop(ns);
  5196. l = list_pop(ns);
  5197. list_push(ns, node_new2(N_INDEX1, l, r));
  5198. }
  5199. }
  5200. return ns->data ? ns->data[0] : NULL;
  5201. }
  5202. node_t *parser_parse(parser_t *parser, lexer_t *lexer) {
  5203. parser->lexer = lexer;
  5204. parser->pos = 0;
  5205. parser->end = parser->lexer->tokens->length;
  5206. node_t *node = parser_parse_expr(parser);
  5207. if (!parser_done(parser)) {
  5208. token_t *tok = parser_lookahead(parser, 0);
  5209. if (tok && tok->tag == T_RPAR)
  5210. parser_error(parser, "unmatched");
  5211. parser_error(parser, "parse");
  5212. }
  5213. return node;
  5214. }
  5215. value_t *interpreter_run(interpreter_t *state, char *program) {
  5216. lexer_t *lexer = lexer_new();
  5217. lexer_lex(lexer, program);
  5218. parser_t *parser = parser_new(state);
  5219. node_t *node = parser_parse(parser, lexer);
  5220. list_t *t = lexer->tokens;
  5221. for (size_t i = 0; i < t->length; i++) {
  5222. token_t *tok = t->data[i];
  5223. if (tok->text)
  5224. GC_FREE(tok->text);
  5225. GC_FREE(tok);
  5226. }
  5227. GC_FREE(t->data);
  5228. GC_FREE(t);
  5229. value_t *r = interpreter_walk(state, node);
  5230. GC_FREE(parser);
  5231. return r;
  5232. }
  5233. #include "help.h"
  5234. const char *VSTR = VER " " __DATE__;
  5235. void jkexec(interpreter_t *state, FILE *fd, bool isrepl, char **s) {
  5236. value_t *v = NULL;
  5237. list_t *r;
  5238. if (!isrepl)
  5239. r = list_new();
  5240. for (;;) {
  5241. buffer_t *buffer;
  5242. char line[256];
  5243. buffer = buffer_new();
  5244. if (isrepl)
  5245. putc('\t', stdout);
  5246. if (!fgets(line, sizeof(line), fd))
  5247. break;
  5248. if (isrepl) {
  5249. if (strcmp(line, "\\\\\n") == 0)
  5250. break;
  5251. else if (strcmp(line, "\\\n") == 0) {
  5252. printf("%s", HELP);
  5253. continue;
  5254. } else if (strcmp(line, "\\0\n") == 0) {
  5255. printf("%s", SHELP);
  5256. continue;
  5257. } else if (strcmp(line, "\\+\n") == 0) {
  5258. printf("%s", VHELP);
  5259. continue;
  5260. } else if (strcmp(line, "\\a\n") == 0) {
  5261. printf("%s", V2HELP);
  5262. continue;
  5263. } else if (strcmp(line, "\\\"\n") == 0) {
  5264. printf("%s", AHELP);
  5265. continue;
  5266. } else if (strcmp(line, "\\;\n") == 0) {
  5267. printf("%s", CHELP);
  5268. continue;
  5269. } else if (strcmp(line, "\\-:\n") == 0) {
  5270. printf("%s", IHELP);
  5271. continue;
  5272. }
  5273. }
  5274. while (strlen(line) > 2 && strcmp(line + strlen(line) - 3, "..\n") == 0) {
  5275. line[strlen(line) - 3] = 0;
  5276. buffer_append_str(buffer, line);
  5277. if (isrepl)
  5278. putc('\t', stdout);
  5279. if (!fgets(line, sizeof(line), fd))
  5280. return;
  5281. }
  5282. buffer_append_str(buffer, line);
  5283. *s = buffer_read(buffer);
  5284. v = interpreter_run(state, *s);
  5285. GC_FREE(*s);
  5286. *s = NULL;
  5287. if (isrepl && v->tag != NIL) {
  5288. table_set(state->env, "it", v);
  5289. char *s = value_show(v);
  5290. fputs(s, stdout);
  5291. GC_FREE(s);
  5292. if (isrepl)
  5293. putc('\n', stdout);
  5294. } else if (!isrepl && v && v->tag != NIL)
  5295. list_push(r, v);
  5296. }
  5297. if (!isrepl && r->data) {
  5298. char *s = value_show(list_index(r, -1));
  5299. fputs(s, stdout);
  5300. GC_FREE(s);
  5301. }
  5302. }
  5303. int main(int argc, char **argv) {
  5304. GC_INIT();
  5305. GC_enable_incremental();
  5306. guards = list_new();
  5307. is_interactive = isatty(0);
  5308. HASH_SEED = time(NULL);
  5309. srand(HASH_SEED);
  5310. VCACHE = table_new();
  5311. SCACHE = table_new();
  5312. for (size_t i = 0; i < countof(VERBS); i++) {
  5313. value_t *v = value_new_const(VERB);
  5314. v->val.verb = &VERBS[i];
  5315. table_set(VCACHE, VERBS[i].name, v);
  5316. }
  5317. _UNIT = value_new(ARRAY);
  5318. _UNIT->val.array = list_new();
  5319. interpreter_t *state = interpreter_new();
  5320. for (int i = 1; i <= 8; i++) {
  5321. NNUMS[i - 1] = value_new_const(NUMBER);
  5322. NNUMS[i - 1]->val.number = -i;
  5323. }
  5324. for (int i = 0; i < 256; i++) {
  5325. NUMS[i] = value_new_const(NUMBER);
  5326. NUMS[i]->val.number = i;
  5327. }
  5328. list_t *cs = list_newk(256);
  5329. for (int i = 0; i < 256; i++) {
  5330. CHARS[i] = value_new_const(CHAR);
  5331. CHARS[i]->val._char = i;
  5332. cs->data[i] = CHARS[i];
  5333. }
  5334. _NAN = value_new_const(NUMBER);
  5335. _NAN->val.number = NAN;
  5336. INF = value_new_const(NUMBER);
  5337. INF->val.number = INFINITY;
  5338. NINF = value_new_const(NUMBER);
  5339. NINF->val.number = -INFINITY;
  5340. list_t *vs = list_new();
  5341. for (size_t i = 0; i < strlen(VSTR); i++)
  5342. list_push(vs, CHARS[(int)VSTR[i]]);
  5343. table_set(state->env, "A", value_new_array(cs));
  5344. table_set(state->env, "JKV", value_new_array(vs));
  5345. table_set(state->env, "E", value_new_number(exp(1)));
  5346. table_set(state->env, "pi", value_new_number(M_PI));
  5347. table_set(state->env, "tau", value_new_number(M_PI * 2));
  5348. table_set(state->env, "nan", _NAN);
  5349. table_set(state->env, "inf", INF);
  5350. table_set(state->env, "nil", state->nil);
  5351. table_set(state->env, "udf", state->udf);
  5352. Inverses = table_new();
  5353. table_set(Inverses, "+", find_verb("+"));
  5354. table_set(Inverses, "-", find_verb("-"));
  5355. table_set(Inverses, "|", find_verb("|"));
  5356. table_set(Inverses, "~", find_verb("~"));
  5357. table_set(Inverses, "%", find_verb("%"));
  5358. table_set(Inverses, "]", find_verb("]"));
  5359. table_set(Inverses, "*:", find_verb("%:"));
  5360. table_set(Inverses, "%:", find_verb("*:"));
  5361. table_set(Inverses, ">", find_verb("<"));
  5362. table_set(Inverses, "<", find_verb(">"));
  5363. table_set(Inverses, "_.", find_verb("_:"));
  5364. table_set(Inverses, "_:", find_verb("_."));
  5365. table_set(Inverses, "^.", find_verb("^"));
  5366. table_set(Inverses, "^", find_verb("^."));
  5367. table_set(Inverses, "+;.", find_verb("%:"));
  5368. table_set(Inverses, "*/", find_verb("["));
  5369. table_set(Inverses, "[", interpreter_run(state, "*/")->val.verb);
  5370. table_set(Inverses, "!", interpreter_run(state, ">|/")->val.verb);
  5371. table_set(Inverses, "!.", interpreter_run(state, "|/")->val.verb);
  5372. table_set(Inverses, "]@>:", interpreter_run(state, "]@<:")->val.verb);
  5373. table_set(Inverses, "]@<:", interpreter_run(state, "]@>:")->val.verb);
  5374. list_t *args = list_new();
  5375. for (int i = 1; i < argc; i++) {
  5376. list_t *arg = list_new();
  5377. char *s = argv[i];
  5378. while (*s)
  5379. list_push(arg, CHARS[(int)(*s++)]);
  5380. list_push(args, value_new_array(arg));
  5381. }
  5382. table_set(state->env, "args", value_new_array(args));
  5383. if (is_interactive)
  5384. printf("jk\t\\\\ to exit \\ for help\n");
  5385. char *s = NULL;
  5386. if (is_interactive)
  5387. setjmp(interactive_checkpoint);
  5388. if (s) {
  5389. GC_FREE(s);
  5390. s = NULL;
  5391. }
  5392. jkexec(state, stdin, is_interactive, &s);
  5393. }