jk.c 163 KB

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