jk.c 142 KB

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