jk.c 181 KB

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