1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304630563066307630863096310631163126313631463156316631763186319632063216322632363246325632663276328632963306331633263336334633563366337633863396340634163426343634463456346634763486349635063516352635363546355635663576358635963606361636263636364636563666367636863696370637163726373637463756376637763786379638063816382638363846385638663876388638963906391639263936394639563966397639863996400640164026403640464056406640764086409641064116412641364146415641664176418641964206421642264236424642564266427642864296430643164326433643464356436643764386439644064416442644364446445644664476448644964506451645264536454645564566457645864596460646164626463646464656466646764686469647064716472647364746475647664776478647964806481648264836484648564866487648864896490649164926493649464956496649764986499650065016502650365046505650665076508650965106511651265136514651565166517651865196520652165226523652465256526652765286529653065316532653365346535653665376538653965406541654265436544654565466547654865496550655165526553655465556556655765586559656065616562656365646565656665676568656965706571657265736574657565766577657865796580658165826583658465856586658765886589659065916592659365946595659665976598659966006601660266036604660566066607660866096610661166126613661466156616661766186619662066216622662366246625662666276628662966306631663266336634663566366637663866396640664166426643664466456646664766486649665066516652665366546655665666576658665966606661666266636664666566666667666866696670667166726673667466756676667766786679668066816682668366846685668666876688668966906691669266936694669566966697669866996700670167026703670467056706670767086709671067116712671367146715671667176718671967206721672267236724672567266727672867296730673167326733673467356736673767386739674067416742674367446745674667476748674967506751675267536754675567566757675867596760676167626763676467656766676767686769677067716772677367746775677667776778677967806781678267836784678567866787678867896790679167926793679467956796679767986799680068016802680368046805680668076808680968106811681268136814681568166817681868196820682168226823682468256826682768286829683068316832683368346835683668376838683968406841684268436844684568466847684868496850685168526853685468556856685768586859686068616862686368646865686668676868686968706871687268736874687568766877687868796880688168826883688468856886688768886889689068916892689368946895689668976898689969006901690269036904690569066907690869096910691169126913691469156916691769186919692069216922692369246925692669276928692969306931693269336934693569366937693869396940694169426943694469456946694769486949695069516952695369546955695669576958695969606961696269636964696569666967696869696970697169726973697469756976697769786979698069816982698369846985698669876988698969906991699269936994699569966997699869997000700170027003700470057006700770087009701070117012701370147015701670177018701970207021702270237024702570267027702870297030703170327033703470357036703770387039704070417042704370447045704670477048704970507051705270537054705570567057705870597060706170627063706470657066706770687069707070717072707370747075707670777078707970807081708270837084708570867087708870897090709170927093709470957096709770987099710071017102710371047105710671077108710971107111711271137114711571167117711871197120712171227123712471257126712771287129713071317132713371347135713671377138713971407141714271437144714571467147714871497150715171527153715471557156715771587159716071617162716371647165716671677168716971707171717271737174717571767177717871797180718171827183718471857186718771887189719071917192719371947195719671977198719972007201720272037204720572067207720872097210721172127213721472157216721772187219722072217222722372247225722672277228722972307231723272337234723572367237723872397240724172427243724472457246724772487249725072517252725372547255725672577258725972607261726272637264726572667267726872697270727172727273727472757276727772787279728072817282728372847285728672877288728972907291729272937294729572967297729872997300730173027303730473057306730773087309731073117312731373147315731673177318731973207321732273237324732573267327732873297330733173327333733473357336733773387339734073417342734373447345734673477348734973507351735273537354735573567357735873597360736173627363736473657366736773687369737073717372737373747375737673777378737973807381738273837384738573867387738873897390739173927393739473957396739773987399740074017402740374047405740674077408740974107411741274137414741574167417741874197420742174227423742474257426742774287429743074317432743374347435743674377438743974407441744274437444744574467447744874497450745174527453745474557456745774587459746074617462746374647465746674677468746974707471747274737474747574767477747874797480748174827483748474857486748774887489749074917492749374947495749674977498749975007501750275037504750575067507750875097510751175127513751475157516751775187519752075217522752375247525752675277528752975307531753275337534753575367537753875397540754175427543754475457546754775487549755075517552755375547555755675577558755975607561756275637564756575667567756875697570757175727573757475757576757775787579758075817582758375847585758675877588758975907591759275937594759575967597759875997600760176027603760476057606760776087609761076117612761376147615761676177618761976207621762276237624762576267627762876297630763176327633763476357636763776387639764076417642764376447645764676477648764976507651765276537654765576567657765876597660766176627663766476657666766776687669767076717672767376747675767676777678767976807681768276837684768576867687768876897690769176927693769476957696769776987699770077017702770377047705770677077708770977107711771277137714771577167717771877197720772177227723772477257726772777287729773077317732773377347735773677377738773977407741774277437744774577467747774877497750775177527753775477557756775777587759776077617762776377647765776677677768776977707771777277737774777577767777777877797780778177827783778477857786778777887789779077917792779377947795779677977798779978007801780278037804780578067807780878097810781178127813781478157816781778187819782078217822782378247825782678277828782978307831783278337834783578367837783878397840784178427843784478457846784778487849785078517852785378547855785678577858785978607861786278637864786578667867786878697870787178727873787478757876787778787879788078817882788378847885788678877888788978907891789278937894789578967897789878997900790179027903790479057906790779087909791079117912791379147915791679177918791979207921792279237924792579267927792879297930793179327933793479357936793779387939794079417942794379447945794679477948794979507951795279537954795579567957795879597960796179627963796479657966796779687969797079717972797379747975797679777978797979807981798279837984798579867987798879897990799179927993799479957996799779987999800080018002800380048005800680078008800980108011801280138014801580168017801880198020802180228023802480258026802780288029803080318032803380348035803680378038803980408041804280438044804580468047804880498050805180528053805480558056805780588059806080618062806380648065806680678068806980708071807280738074807580768077807880798080808180828083808480858086808780888089809080918092809380948095809680978098809981008101810281038104810581068107810881098110811181128113811481158116811781188119812081218122812381248125812681278128812981308131813281338134813581368137813881398140814181428143814481458146814781488149815081518152815381548155815681578158815981608161816281638164816581668167816881698170817181728173817481758176817781788179818081818182818381848185818681878188818981908191819281938194819581968197819881998200820182028203820482058206820782088209821082118212821382148215821682178218821982208221822282238224822582268227822882298230823182328233823482358236823782388239824082418242824382448245824682478248824982508251825282538254825582568257825882598260826182628263826482658266826782688269827082718272827382748275827682778278827982808281828282838284828582868287828882898290829182928293829482958296829782988299830083018302830383048305830683078308830983108311831283138314831583168317831883198320832183228323832483258326832783288329833083318332833383348335833683378338833983408341834283438344834583468347834883498350835183528353835483558356835783588359836083618362836383648365836683678368836983708371837283738374837583768377837883798380838183828383838483858386838783888389839083918392839383948395839683978398839984008401840284038404840584068407840884098410841184128413841484158416841784188419842084218422842384248425842684278428842984308431843284338434843584368437843884398440844184428443844484458446844784488449845084518452845384548455845684578458845984608461846284638464846584668467846884698470847184728473847484758476847784788479848084818482848384848485848684878488848984908491849284938494849584968497849884998500850185028503850485058506850785088509851085118512851385148515851685178518851985208521852285238524852585268527852885298530853185328533853485358536853785388539854085418542854385448545854685478548854985508551855285538554855585568557855885598560856185628563856485658566856785688569857085718572857385748575857685778578857985808581858285838584858585868587858885898590859185928593859485958596859785988599860086018602860386048605860686078608860986108611861286138614861586168617861886198620862186228623862486258626862786288629863086318632863386348635863686378638863986408641864286438644864586468647864886498650865186528653865486558656865786588659866086618662866386648665866686678668866986708671867286738674867586768677867886798680868186828683868486858686868786888689869086918692869386948695869686978698869987008701870287038704870587068707870887098710871187128713871487158716871787188719872087218722872387248725872687278728872987308731873287338734873587368737873887398740874187428743874487458746874787488749875087518752875387548755875687578758875987608761876287638764876587668767876887698770877187728773877487758776877787788779878087818782878387848785878687878788878987908791879287938794879587968797879887998800880188028803880488058806880788088809881088118812881388148815881688178818881988208821882288238824882588268827882888298830883188328833883488358836883788388839884088418842884388448845884688478848884988508851885288538854885588568857885888598860886188628863886488658866886788688869887088718872887388748875887688778878887988808881888288838884888588868887888888898890889188928893889488958896889788988899890089018902890389048905890689078908890989108911891289138914891589168917891889198920892189228923892489258926892789288929893089318932893389348935893689378938893989408941894289438944894589468947894889498950895189528953895489558956895789588959896089618962896389648965896689678968896989708971897289738974897589768977897889798980898189828983898489858986898789888989899089918992899389948995899689978998899990009001900290039004900590069007900890099010901190129013901490159016901790189019902090219022902390249025902690279028902990309031903290339034903590369037903890399040904190429043904490459046904790489049905090519052905390549055905690579058905990609061906290639064906590669067906890699070907190729073907490759076907790789079908090819082908390849085908690879088908990909091909290939094909590969097909890999100910191029103910491059106910791089109911091119112911391149115911691179118911991209121912291239124912591269127912891299130913191329133913491359136913791389139914091419142914391449145914691479148914991509151915291539154915591569157915891599160916191629163916491659166916791689169917091719172917391749175917691779178917991809181918291839184918591869187918891899190919191929193919491959196919791989199920092019202920392049205920692079208920992109211921292139214921592169217921892199220922192229223922492259226922792289229923092319232923392349235923692379238923992409241924292439244924592469247924892499250925192529253925492559256925792589259926092619262926392649265926692679268926992709271927292739274927592769277927892799280928192829283928492859286928792889289929092919292929392949295929692979298929993009301930293039304930593069307930893099310931193129313931493159316931793189319932093219322932393249325932693279328932993309331933293339334933593369337933893399340934193429343934493459346934793489349935093519352935393549355935693579358935993609361936293639364936593669367936893699370937193729373937493759376937793789379938093819382938393849385938693879388938993909391939293939394939593969397939893999400940194029403940494059406940794089409941094119412941394149415941694179418941994209421942294239424942594269427942894299430943194329433943494359436943794389439944094419442944394449445944694479448944994509451945294539454945594569457945894599460946194629463946494659466946794689469947094719472947394749475947694779478947994809481948294839484948594869487948894899490949194929493949494959496949794989499950095019502950395049505950695079508950995109511951295139514951595169517951895199520952195229523952495259526952795289529953095319532953395349535953695379538953995409541954295439544954595469547954895499550955195529553955495559556955795589559956095619562956395649565956695679568956995709571957295739574957595769577957895799580958195829583958495859586958795889589959095919592959395949595959695979598959996009601960296039604960596069607960896099610961196129613961496159616961796189619962096219622962396249625962696279628962996309631963296339634963596369637963896399640964196429643964496459646964796489649965096519652965396549655965696579658965996609661966296639664966596669667966896699670967196729673967496759676967796789679968096819682968396849685968696879688968996909691969296939694969596969697969896999700970197029703970497059706970797089709971097119712971397149715971697179718971997209721972297239724972597269727972897299730973197329733973497359736973797389739974097419742974397449745974697479748974997509751975297539754975597569757975897599760976197629763976497659766976797689769977097719772977397749775977697779778977997809781978297839784978597869787978897899790979197929793979497959796979797989799980098019802980398049805980698079808980998109811981298139814981598169817981898199820982198229823982498259826982798289829983098319832983398349835983698379838983998409841984298439844984598469847984898499850985198529853985498559856985798589859986098619862986398649865986698679868986998709871987298739874987598769877987898799880988198829883988498859886988798889889989098919892989398949895989698979898989999009901990299039904990599069907990899099910991199129913991499159916991799189919992099219922992399249925992699279928992999309931993299339934993599369937993899399940994199429943994499459946994799489949995099519952995399549955995699579958995999609961996299639964996599669967996899699970997199729973997499759976997799789979998099819982998399849985998699879988998999909991999299939994999599969997999899991000010001100021000310004100051000610007100081000910010100111001210013100141001510016100171001810019100201002110022100231002410025100261002710028100291003010031100321003310034100351003610037100381003910040100411004210043100441004510046100471004810049100501005110052100531005410055100561005710058100591006010061100621006310064100651006610067100681006910070100711007210073100741007510076100771007810079100801008110082100831008410085100861008710088100891009010091100921009310094100951009610097100981009910100101011010210103101041010510106101071010810109101101011110112101131011410115101161011710118101191012010121101221012310124101251012610127101281012910130101311013210133101341013510136101371013810139101401014110142101431014410145101461014710148101491015010151101521015310154101551015610157101581015910160101611016210163101641016510166101671016810169101701017110172101731017410175101761017710178101791018010181101821018310184101851018610187101881018910190101911019210193101941019510196101971019810199102001020110202102031020410205102061020710208102091021010211102121021310214102151021610217102181021910220102211022210223102241022510226102271022810229102301023110232102331023410235102361023710238102391024010241102421024310244102451024610247102481024910250102511025210253102541025510256102571025810259102601026110262102631026410265102661026710268102691027010271102721027310274102751027610277102781027910280102811028210283102841028510286102871028810289102901029110292102931029410295102961029710298102991030010301103021030310304103051030610307103081030910310103111031210313103141031510316103171031810319103201032110322103231032410325103261032710328103291033010331103321033310334103351033610337103381033910340103411034210343103441034510346103471034810349103501035110352103531035410355103561035710358103591036010361103621036310364103651036610367103681036910370103711037210373103741037510376103771037810379103801038110382103831038410385103861038710388103891039010391103921039310394103951039610397103981039910400104011040210403104041040510406104071040810409104101041110412104131041410415104161041710418104191042010421104221042310424104251042610427104281042910430104311043210433104341043510436104371043810439104401044110442104431044410445104461044710448104491045010451104521045310454104551045610457104581045910460104611046210463104641046510466104671046810469104701047110472104731047410475104761047710478104791048010481104821048310484104851048610487104881048910490104911049210493104941049510496104971049810499105001050110502105031050410505105061050710508105091051010511105121051310514105151051610517105181051910520105211052210523105241052510526105271052810529105301053110532105331053410535105361053710538105391054010541105421054310544105451054610547105481054910550105511055210553105541055510556105571055810559105601056110562105631056410565105661056710568105691057010571105721057310574105751057610577105781057910580105811058210583105841058510586105871058810589105901059110592105931059410595105961059710598105991060010601106021060310604106051060610607106081060910610106111061210613106141061510616106171061810619106201062110622106231062410625106261062710628106291063010631106321063310634106351063610637106381063910640106411064210643106441064510646106471064810649106501065110652106531065410655106561065710658106591066010661106621066310664106651066610667106681066910670106711067210673106741067510676106771067810679106801068110682106831068410685106861068710688106891069010691106921069310694106951069610697106981069910700107011070210703107041070510706107071070810709107101071110712107131071410715107161071710718107191072010721107221072310724107251072610727107281072910730107311073210733107341073510736107371073810739107401074110742107431074410745107461074710748107491075010751107521075310754107551075610757107581075910760107611076210763107641076510766107671076810769107701077110772107731077410775107761077710778107791078010781107821078310784107851078610787107881078910790107911079210793107941079510796107971079810799108001080110802108031080410805108061080710808108091081010811108121081310814108151081610817108181081910820108211082210823108241082510826108271082810829108301083110832108331083410835108361083710838108391084010841108421084310844108451084610847108481084910850108511085210853108541085510856108571085810859108601086110862108631086410865108661086710868108691087010871108721087310874108751087610877108781087910880108811088210883108841088510886108871088810889108901089110892108931089410895108961089710898108991090010901109021090310904109051090610907109081090910910109111091210913109141091510916109171091810919109201092110922109231092410925109261092710928109291093010931109321093310934109351093610937109381093910940109411094210943109441094510946109471094810949109501095110952109531095410955109561095710958109591096010961109621096310964109651096610967109681096910970109711097210973109741097510976109771097810979109801098110982109831098410985109861098710988109891099010991109921099310994109951099610997109981099911000110011100211003110041100511006110071100811009110101101111012110131101411015110161101711018110191102011021110221102311024110251102611027110281102911030110311103211033110341103511036110371103811039110401104111042110431104411045110461104711048110491105011051110521105311054110551105611057110581105911060110611106211063110641106511066110671106811069110701107111072110731107411075110761107711078110791108011081110821108311084110851108611087110881108911090110911109211093110941109511096110971109811099111001110111102111031110411105111061110711108111091111011111111121111311114111151111611117111181111911120111211112211123111241112511126111271112811129111301113111132111331113411135111361113711138111391114011141111421114311144111451114611147111481114911150111511115211153111541115511156111571115811159111601116111162111631116411165111661116711168111691117011171111721117311174111751117611177111781117911180111811118211183111841118511186111871118811189111901119111192111931119411195111961119711198111991120011201112021120311204112051120611207112081120911210112111121211213112141121511216112171121811219112201122111222112231122411225112261122711228112291123011231112321123311234112351123611237112381123911240112411124211243112441124511246112471124811249112501125111252112531125411255112561125711258112591126011261112621126311264112651126611267112681126911270112711127211273112741127511276112771127811279112801128111282112831128411285112861128711288112891129011291112921129311294112951129611297112981129911300113011130211303113041130511306113071130811309113101131111312113131131411315113161131711318113191132011321113221132311324113251132611327113281132911330113311133211333113341133511336113371133811339113401134111342113431134411345113461134711348113491135011351113521135311354113551135611357113581135911360113611136211363113641136511366113671136811369113701137111372113731137411375113761137711378113791138011381113821138311384113851138611387113881138911390113911139211393113941139511396113971139811399114001140111402114031140411405114061140711408114091141011411114121141311414114151141611417114181141911420114211142211423114241142511426114271142811429114301143111432114331143411435114361143711438114391144011441114421144311444114451144611447114481144911450114511145211453114541145511456114571145811459114601146111462114631146411465114661146711468114691147011471114721147311474114751147611477114781147911480114811148211483114841148511486114871148811489114901149111492114931149411495114961149711498114991150011501115021150311504115051150611507115081150911510115111151211513115141151511516115171151811519115201152111522115231152411525115261152711528115291153011531115321153311534115351153611537115381153911540115411154211543115441154511546115471154811549115501155111552115531155411555115561155711558115591156011561115621156311564115651156611567115681156911570115711157211573115741157511576115771157811579115801158111582115831158411585115861158711588115891159011591115921159311594115951159611597115981159911600116011160211603116041160511606116071160811609116101161111612116131161411615116161161711618116191162011621116221162311624116251162611627116281162911630116311163211633116341163511636116371163811639116401164111642116431164411645116461164711648116491165011651116521165311654116551165611657116581165911660116611166211663116641166511666116671166811669116701167111672116731167411675116761167711678116791168011681116821168311684116851168611687116881168911690116911169211693116941169511696116971169811699117001170111702117031170411705117061170711708117091171011711117121171311714117151171611717117181171911720117211172211723117241172511726117271172811729117301173111732117331173411735117361173711738117391174011741117421174311744117451174611747117481174911750117511175211753117541175511756117571175811759117601176111762117631176411765117661176711768117691177011771117721177311774117751177611777117781177911780117811178211783117841178511786117871178811789117901179111792117931179411795117961179711798117991180011801118021180311804118051180611807118081180911810118111181211813118141181511816118171181811819118201182111822118231182411825118261182711828118291183011831118321183311834118351183611837118381183911840118411184211843118441184511846118471184811849118501185111852118531185411855118561185711858118591186011861118621186311864118651186611867118681186911870118711187211873118741187511876118771187811879118801188111882118831188411885118861188711888118891189011891118921189311894118951189611897118981189911900119011190211903119041190511906119071190811909119101191111912119131191411915119161191711918119191192011921119221192311924119251192611927119281192911930119311193211933119341193511936119371193811939119401194111942119431194411945119461194711948119491195011951119521195311954119551195611957119581195911960119611196211963119641196511966119671196811969119701197111972119731197411975119761197711978119791198011981119821198311984119851198611987119881198911990119911199211993119941199511996119971199811999120001200112002120031200412005120061200712008120091201012011120121201312014120151201612017120181201912020120211202212023120241202512026120271202812029120301203112032120331203412035120361203712038120391204012041120421204312044120451204612047120481204912050120511205212053120541205512056120571205812059120601206112062120631206412065120661206712068120691207012071120721207312074120751207612077120781207912080120811208212083120841208512086120871208812089120901209112092120931209412095120961209712098120991210012101121021210312104121051210612107121081210912110121111211212113121141211512116121171211812119121201212112122121231212412125121261212712128121291213012131121321213312134121351213612137121381213912140121411214212143121441214512146121471214812149121501215112152121531215412155121561215712158121591216012161121621216312164121651216612167121681216912170121711217212173121741217512176121771217812179121801218112182121831218412185121861218712188121891219012191121921219312194121951219612197121981219912200122011220212203122041220512206122071220812209122101221112212122131221412215122161221712218122191222012221122221222312224122251222612227122281222912230122311223212233122341223512236122371223812239122401224112242122431224412245122461224712248122491225012251122521225312254122551225612257122581225912260122611226212263122641226512266122671226812269122701227112272122731227412275122761227712278122791228012281122821228312284122851228612287122881228912290122911229212293122941229512296122971229812299123001230112302123031230412305123061230712308123091231012311123121231312314123151231612317123181231912320123211232212323123241232512326123271232812329123301233112332123331233412335123361233712338123391234012341123421234312344123451234612347123481234912350123511235212353123541235512356123571235812359123601236112362123631236412365123661236712368123691237012371123721237312374123751237612377123781237912380123811238212383123841238512386123871238812389123901239112392123931239412395123961239712398123991240012401124021240312404124051240612407124081240912410124111241212413124141241512416124171241812419124201242112422124231242412425124261242712428124291243012431124321243312434124351243612437124381243912440124411244212443124441244512446124471244812449124501245112452124531245412455124561245712458124591246012461124621246312464124651246612467124681246912470124711247212473124741247512476124771247812479124801248112482124831248412485124861248712488124891249012491124921249312494124951249612497124981249912500125011250212503125041250512506125071250812509125101251112512125131251412515125161251712518125191252012521125221252312524125251252612527125281252912530125311253212533125341253512536125371253812539125401254112542125431254412545125461254712548125491255012551125521255312554125551255612557125581255912560125611256212563125641256512566125671256812569125701257112572125731257412575125761257712578125791258012581125821258312584125851258612587125881258912590125911259212593125941259512596125971259812599126001260112602126031260412605126061260712608126091261012611126121261312614126151261612617126181261912620126211262212623126241262512626126271262812629126301263112632126331263412635126361263712638126391264012641126421264312644126451264612647126481264912650126511265212653126541265512656126571265812659126601266112662126631266412665126661266712668126691267012671126721267312674126751267612677126781267912680126811268212683126841268512686126871268812689126901269112692126931269412695126961269712698126991270012701127021270312704127051270612707127081270912710127111271212713127141271512716127171271812719127201272112722127231272412725127261272712728127291273012731127321273312734127351273612737127381273912740127411274212743127441274512746127471274812749127501275112752127531275412755127561275712758127591276012761127621276312764127651276612767127681276912770127711277212773127741277512776127771277812779127801278112782127831278412785127861278712788127891279012791127921279312794127951279612797127981279912800128011280212803128041280512806128071280812809128101281112812128131281412815128161281712818128191282012821128221282312824128251282612827128281282912830128311283212833128341283512836128371283812839128401284112842128431284412845128461284712848128491285012851128521285312854128551285612857128581285912860128611286212863128641286512866128671286812869128701287112872128731287412875128761287712878128791288012881128821288312884128851288612887128881288912890128911289212893128941289512896128971289812899129001290112902129031290412905129061290712908129091291012911129121291312914129151291612917129181291912920129211292212923129241292512926129271292812929129301293112932129331293412935129361293712938129391294012941129421294312944129451294612947129481294912950129511295212953129541295512956129571295812959129601296112962129631296412965129661296712968129691297012971129721297312974129751297612977129781297912980129811298212983129841298512986129871298812989129901299112992129931299412995129961299712998129991300013001130021300313004130051300613007130081300913010130111301213013130141301513016130171301813019130201302113022130231302413025130261302713028130291303013031130321303313034130351303613037130381303913040130411304213043130441304513046130471304813049130501305113052130531305413055130561305713058130591306013061130621306313064130651306613067130681306913070130711307213073130741307513076130771307813079130801308113082130831308413085130861308713088130891309013091130921309313094130951309613097130981309913100131011310213103131041310513106131071310813109131101311113112 |
- /* stb.h - v2.37 - Sean's Tool Box -- public domain -- http://nothings.org/stb.h
- no warranty is offered or implied; use this code at your own risk
- This is a single header file with a bunch of useful utilities
- for getting stuff done in C/C++.
- Documentation: http://nothings.org/stb/stb_h.html
- Unit tests: http://nothings.org/stb/stb.c
- ============================================================================
- You MUST
- #define STB_DEFINE
- in EXACTLY _one_ C or C++ file that includes this header, BEFORE the
- include, like this:
- #define STB_DEFINE
- #include "stb.h"
- All other files should just #include "stb.h" without the #define.
- ============================================================================
- Version History
- 2.36 various fixes
- 2.35 fix clang-cl issues with swprintf
- 2.34 fix warnings
- 2.33 more fixes to random numbers
- 2.32 stb_intcmprev, stb_uidict, fix random numbers on Linux
- 2.31 stb_ucharcmp
- 2.30 MinGW fix
- 2.29 attempt to fix use of swprintf()
- 2.28 various new functionality
- 2.27 test _WIN32 not WIN32 in STB_THREADS
- 2.26 various warning & bugfixes
- 2.25 various warning & bugfixes
- 2.24 various warning & bugfixes
- 2.23 fix 2.22
- 2.22 64-bit fixes from '!='; fix stb_sdict_copy() to have preferred name
- 2.21 utf-8 decoder rejects "overlong" encodings; attempted 64-bit improvements
- 2.20 fix to hash "copy" function--reported by someone with handle "!="
- 2.19 ???
- 2.18 stb_readdir_subdirs_mask
- 2.17 stb_cfg_dir
- 2.16 fix stb_bgio_, add stb_bgio_stat(); begin a streaming wrapper
- 2.15 upgraded hash table template to allow:
- - aggregate keys (explicit comparison func for EMPTY and DEL keys)
- - "static" implementations (so they can be culled if unused)
- 2.14 stb_mprintf
- 2.13 reduce identifiable strings in STB_NO_STB_STRINGS
- 2.12 fix STB_ONLY -- lots of uint32s, TRUE/FALSE things had crept in
- 2.11 fix bug in stb_dirtree_get() which caused "c://path" sorts of stuff
- 2.10 STB_F(), STB_I() inline constants (also KI,KU,KF,KD)
- 2.09 stb_box_face_vertex_axis_side
- 2.08 bugfix stb_trimwhite()
- 2.07 colored printing in windows (why are we in 1985?)
- 2.06 comparison functions are now functions-that-return-functions and
- accept a struct-offset as a parameter (not thread-safe)
- 2.05 compile and pass tests under Linux (but no threads); thread cleanup
- 2.04 stb_cubic_bezier_1d, smoothstep, avoid dependency on registry
- 2.03 ?
- 2.02 remove integrated documentation
- 2.01 integrate various fixes; stb_force_uniprocessor
- 2.00 revised stb_dupe to use multiple hashes
- 1.99 stb_charcmp
- 1.98 stb_arr_deleten, stb_arr_insertn
- 1.97 fix stb_newell_normal()
- 1.96 stb_hash_number()
- 1.95 hack stb__rec_max; clean up recursion code to use new functions
- 1.94 stb_dirtree; rename stb_extra to stb_ptrmap
- 1.93 stb_sem_new() API cleanup (no blockflag-starts blocked; use 'extra')
- 1.92 stb_threadqueue--multi reader/writer queue, fixed size or resizeable
- 1.91 stb_bgio_* for reading disk asynchronously
- 1.90 stb_mutex uses CRITICAL_REGION; new stb_sync primitive for thread
- joining; workqueue supports stb_sync instead of stb_semaphore
- 1.89 support ';' in constant-string wildcards; stb_mutex wrapper (can
- implement with EnterCriticalRegion eventually)
- 1.88 portable threading API (only for win32 so far); worker thread queue
- 1.87 fix wildcard handling in stb_readdir_recursive
- 1.86 support ';' in wildcards
- 1.85 make stb_regex work with non-constant strings;
- beginnings of stb_introspect()
- 1.84 (forgot to make notes)
- 1.83 whoops, stb_keep_if_different wasn't deleting the temp file
- 1.82 bring back stb_compress from stb_file.h for cmirror
- 1.81 various bugfixes, STB_FASTMALLOC_INIT inits FASTMALLOC in release
- 1.80 stb_readdir returns utf8; write own utf8-utf16 because lib was wrong
- 1.79 stb_write
- 1.78 calloc() support for malloc wrapper, STB_FASTMALLOC
- 1.77 STB_FASTMALLOC
- 1.76 STB_STUA - Lua-like language; (stb_image, stb_csample, stb_bilinear)
- 1.75 alloc/free array of blocks; stb_hheap bug; a few stb_ps_ funcs;
- hash*getkey, hash*copy; stb_bitset; stb_strnicmp; bugfix stb_bst
- 1.74 stb_replaceinplace; use stdlib C function to convert utf8 to UTF-16
- 1.73 fix performance bug & leak in stb_ischar (C++ port lost a 'static')
- 1.72 remove stb_block, stb_block_manager, stb_decompress (to stb_file.h)
- 1.71 stb_trimwhite, stb_tokens_nested, etc.
- 1.70 back out 1.69 because it might problemize mixed builds; stb_filec()
- 1.69 (stb_file returns 'char *' in C++)
- 1.68 add a special 'tree root' data type for stb_bst; stb_arr_end
- 1.67 full C++ port. (stb_block_manager)
- 1.66 stb_newell_normal
- 1.65 stb_lex_item_wild -- allow wildcard items which MUST match entirely
- 1.64 stb_data
- 1.63 stb_log_name
- 1.62 stb_define_sort; C++ cleanup
- 1.61 stb_hash_fast -- Paul Hsieh's hash function (beats Bob Jenkins'?)
- 1.60 stb_delete_directory_recursive
- 1.59 stb_readdir_recursive
- 1.58 stb_bst variant with parent pointer for O(1) iteration, not O(log N)
- 1.57 replace LCG random with Mersenne Twister (found a public domain one)
- 1.56 stb_perfect_hash, stb_ischar, stb_regex
- 1.55 new stb_bst API allows multiple BSTs per node (e.g. secondary keys)
- 1.54 bugfix: stb_define_hash, stb_wildmatch, regexp
- 1.53 stb_define_hash; recoded stb_extra, stb_sdict use it
- 1.52 stb_rand_define, stb_bst, stb_reverse
- 1.51 fix 'stb_arr_setlen(NULL, 0)'
- 1.50 stb_wordwrap
- 1.49 minor improvements to enable the scripting language
- 1.48 better approach for stb_arr using stb_malloc; more invasive, clearer
- 1.47 stb_lex (lexes stb.h at 1.5ML/s on 3Ghz P4; 60/70% of optimal/flex)
- 1.46 stb_wrapper_*, STB_MALLOC_WRAPPER
- 1.45 lightly tested DFA acceleration of regexp searching
- 1.44 wildcard matching & searching; regexp matching & searching
- 1.43 stb_temp
- 1.42 allow stb_arr to use stb_malloc/realloc; note this is global
- 1.41 make it compile in C++; (disable stb_arr in C++)
- 1.40 stb_dupe tweak; stb_swap; stb_substr
- 1.39 stb_dupe; improve stb_file_max to be less stupid
- 1.38 stb_sha1_file: generate sha1 for file, even > 4GB
- 1.37 stb_file_max; partial support for utf8 filenames in Windows
- 1.36 remove STB__NO_PREFIX - poor interaction with IDE, not worth it
- streamline stb_arr to make it separately publishable
- 1.35 bugfixes for stb_sdict, stb_malloc(0), stristr
- 1.34 (streaming interfaces for stb_compress)
- 1.33 stb_alloc; bug in stb_getopt; remove stb_overflow
- 1.32 (stb_compress returns, smaller&faster; encode window & 64-bit len)
- 1.31 stb_prefix_count
- 1.30 (STB__NO_PREFIX - remove stb_ prefixes for personal projects)
- 1.29 stb_fput_varlen64, etc.
- 1.28 stb_sha1
- 1.27 ?
- 1.26 stb_extra
- 1.25 ?
- 1.24 stb_copyfile
- 1.23 stb_readdir
- 1.22 ?
- 1.21 ?
- 1.20 ?
- 1.19 ?
- 1.18 ?
- 1.17 ?
- 1.16 ?
- 1.15 stb_fixpath, stb_splitpath, stb_strchr2
- 1.14 stb_arr
- 1.13 ?stb, stb_log, stb_fatal
- 1.12 ?stb_hash2
- 1.11 miniML
- 1.10 stb_crc32, stb_adler32
- 1.09 stb_sdict
- 1.08 stb_bitreverse, stb_ispow2, stb_big32
- stb_fopen, stb_fput_varlen, stb_fput_ranged
- stb_fcmp, stb_feq
- 1.07 (stb_encompress)
- 1.06 stb_compress
- 1.05 stb_tokens, (stb_hheap)
- 1.04 stb_rand
- 1.03 ?(s-strings)
- 1.02 ?stb_filelen, stb_tokens
- 1.01 stb_tolower
- 1.00 stb_hash, stb_intcmp
- stb_file, stb_stringfile, stb_fgets
- stb_prefix, stb_strlower, stb_strtok
- stb_image
- (stb_array), (stb_arena)
- Parenthesized items have since been removed.
- LICENSE
- See end of file for license information.
- CREDITS
- Written by Sean Barrett.
- Fixes:
- Philipp Wiesemann
- Robert Nix
- r-lyeh
- blackpawn
- github:Mojofreem
- Ryan Whitworth
- Vincent Isambart
- Mike Sartain
- Eugene Opalev
- Tim Sjostrand
- github:infatum
- Dave Butler (Croepha)
- Ethan Lee (flibitijibibo)
- Brian Collins
- Kyle Langley
- */
- #include <stdarg.h>
- #ifndef STB__INCLUDE_STB_H
- #define STB__INCLUDE_STB_H
- #define STB_VERSION 1
- #ifdef STB_INTROSPECT
- #define STB_DEFINE
- #endif
- #ifdef STB_DEFINE_THREADS
- #ifndef STB_DEFINE
- #define STB_DEFINE
- #endif
- #ifndef STB_THREADS
- #define STB_THREADS
- #endif
- #endif
- #if defined(_WIN32) && !defined(__MINGW32__)
- #ifndef _CRT_SECURE_NO_WARNINGS
- #define _CRT_SECURE_NO_WARNINGS
- #endif
- #ifndef _CRT_NONSTDC_NO_DEPRECATE
- #define _CRT_NONSTDC_NO_DEPRECATE
- #endif
- #ifndef _CRT_NON_CONFORMING_SWPRINTFS
- #define _CRT_NON_CONFORMING_SWPRINTFS
- #endif
- #if !defined(_MSC_VER) || _MSC_VER > 1700
- #include <intrin.h> // _BitScanReverse
- #endif
- #endif
- #include <stdlib.h> // stdlib could have min/max
- #include <stdio.h> // need FILE
- #include <string.h> // stb_define_hash needs memcpy/memset
- #include <time.h> // stb_dirtree
- #ifdef __MINGW32__
- #include <fcntl.h> // O_RDWR
- #endif
- #ifdef STB_PERSONAL
- typedef int Bool;
- #define False 0
- #define True 1
- #endif
- #ifdef STB_MALLOC_WRAPPER_PAGED
- #define STB_MALLOC_WRAPPER_DEBUG
- #endif
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- #define STB_MALLOC_WRAPPER
- #endif
- #ifdef STB_MALLOC_WRAPPER_FASTMALLOC
- #define STB_FASTMALLOC
- #define STB_MALLOC_WRAPPER
- #endif
- #ifdef STB_FASTMALLOC
- #ifndef _WIN32
- #undef STB_FASTMALLOC
- #endif
- #endif
- #ifdef STB_DEFINE
- #include <assert.h>
- #include <stdarg.h>
- #include <stddef.h>
- #include <ctype.h>
- #include <math.h>
- #ifndef _WIN32
- #include <unistd.h>
- #else
- #include <io.h> // _mktemp
- #include <direct.h> // _rmdir
- #endif
- #include <sys/types.h> // stat()/_stat()
- #include <sys/stat.h> // stat()/_stat()
- #endif
- #define stb_min(a,b) ((a) < (b) ? (a) : (b))
- #define stb_max(a,b) ((a) > (b) ? (a) : (b))
- #ifndef STB_ONLY
- #if !defined(__cplusplus) && !defined(min) && !defined(max)
- #define min(x,y) stb_min(x,y)
- #define max(x,y) stb_max(x,y)
- #endif
- #ifndef M_PI
- #define M_PI 3.14159265358979323846f
- #endif
- #ifndef TRUE
- #define TRUE 1
- #define FALSE 0
- #endif
- #ifndef deg2rad
- #define deg2rad(a) ((a)*(M_PI/180))
- #endif
- #ifndef rad2deg
- #define rad2deg(a) ((a)*(180/M_PI))
- #endif
- #ifndef swap
- #ifndef __cplusplus
- #define swap(TYPE,a,b) \
- do { TYPE stb__t; stb__t = (a); (a) = (b); (b) = stb__t; } while (0)
- #endif
- #endif
- typedef unsigned char uint8 ;
- typedef signed char int8 ;
- typedef unsigned short uint16;
- typedef signed short int16;
- #if defined(STB_USE_LONG_FOR_32_BIT_INT) || defined(STB_LONG32)
- typedef unsigned long uint32;
- typedef signed long int32;
- #else
- typedef unsigned int uint32;
- typedef signed int int32;
- #endif
- typedef unsigned char uchar ;
- typedef unsigned short ushort;
- typedef unsigned int uint ;
- typedef unsigned long ulong ;
- // produce compile errors if the sizes aren't right
- typedef char stb__testsize16[sizeof(int16)==2];
- typedef char stb__testsize32[sizeof(int32)==4];
- #endif
- #ifndef STB_TRUE
- #define STB_TRUE 1
- #define STB_FALSE 0
- #endif
- // if we're STB_ONLY, can't rely on uint32 or even uint, so all the
- // variables we'll use herein need typenames prefixed with 'stb':
- typedef unsigned char stb_uchar;
- typedef unsigned char stb_uint8;
- typedef unsigned int stb_uint;
- typedef unsigned short stb_uint16;
- typedef short stb_int16;
- typedef signed char stb_int8;
- #if defined(STB_USE_LONG_FOR_32_BIT_INT) || defined(STB_LONG32)
- typedef unsigned long stb_uint32;
- typedef long stb_int32;
- #else
- typedef unsigned int stb_uint32;
- typedef int stb_int32;
- #endif
- typedef char stb__testsize2_16[sizeof(stb_uint16)==2 ? 1 : -1];
- typedef char stb__testsize2_32[sizeof(stb_uint32)==4 ? 1 : -1];
- #ifdef _MSC_VER
- typedef unsigned __int64 stb_uint64;
- typedef __int64 stb_int64;
- #define STB_IMM_UINT64(literalui64) (literalui64##ui64)
- #define STB_IMM_INT64(literali64) (literali64##i64)
- #else
- // ??
- typedef unsigned long long stb_uint64;
- typedef long long stb_int64;
- #define STB_IMM_UINT64(literalui64) (literalui64##ULL)
- #define STB_IMM_INT64(literali64) (literali64##LL)
- #endif
- typedef char stb__testsize2_64[sizeof(stb_uint64)==8 ? 1 : -1];
- // add platform-specific ways of checking for sizeof(char*) == 8,
- // and make those define STB_PTR64
- #if defined(_WIN64) || defined(__x86_64__) || defined(__ia64__) || defined(__LP64__)
- #define STB_PTR64
- #endif
- #ifdef STB_PTR64
- typedef char stb__testsize2_ptr[sizeof(char *) == 8];
- typedef stb_uint64 stb_uinta;
- typedef stb_int64 stb_inta;
- #else
- typedef char stb__testsize2_ptr[sizeof(char *) == 4];
- typedef stb_uint32 stb_uinta;
- typedef stb_int32 stb_inta;
- #endif
- typedef char stb__testsize2_uinta[sizeof(stb_uinta)==sizeof(char*) ? 1 : -1];
- // if so, we should define an int type that is the pointer size. until then,
- // we'll have to make do with this (which is not the same at all!)
- typedef union
- {
- unsigned int i;
- void * p;
- } stb_uintptr;
- #ifdef __cplusplus
- #define STB_EXTERN extern "C"
- #else
- #define STB_EXTERN extern
- #endif
- // check for well-known debug defines
- #if defined(DEBUG) || defined(_DEBUG) || defined(DBG)
- #ifndef NDEBUG
- #define STB_DEBUG
- #endif
- #endif
- #ifdef STB_DEBUG
- #include <assert.h>
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // C library function platform handling
- //
- #ifdef STB_DEFINE
- #if defined(_WIN32) && defined(__STDC_WANT_SECURE_LIB__)
- static FILE * stb_p_fopen(const char *filename, const char *mode)
- {
- FILE *f;
- if (0 == fopen_s(&f, filename, mode))
- return f;
- else
- return NULL;
- }
- static FILE * stb_p_wfopen(const wchar_t *filename, const wchar_t *mode)
- {
- FILE *f;
- if (0 == _wfopen_s(&f, filename, mode))
- return f;
- else
- return NULL;
- }
- static char *stb_p_strcpy_s(char *a, size_t size, const char *b)
- {
- strcpy_s(a,size,b);
- return a;
- }
- static char *stb_p_strncpy_s(char *a, size_t size, const char *b, size_t count)
- {
- strncpy_s(a,size,b,count);
- return a;
- }
- #define stb_p_mktemp(s) (_mktemp_s(s, strlen(s)+1) == 0)
- #define stb_p_sprintf sprintf_s
- #define stb_p_size(x) ,(x)
- #else
- #define stb_p_fopen fopen
- #define stb_p_wfopen _wfopen
- #define stb_p_strcpy_s(a,s,b) strcpy(a,b)
- #define stb_p_strncpy_s(a,s,b,c) strncpy(a,b,c)
- #define stb_p_mktemp(s) (mktemp(s) != NULL)
- #define stb_p_sprintf sprintf
- #define stb_p_size(x)
- #endif
- #if defined(_WIN32)
- #define stb_p_vsnprintf _vsnprintf
- #else
- #define stb_p_vsnprintf vsnprintf
- #endif
- #endif // STB_DEFINE
- #if defined(_WIN32) && (_MSC_VER >= 1300)
- #define stb_p_stricmp _stricmp
- #define stb_p_strnicmp _strnicmp
- #define stb_p_strdup _strdup
- #else
- #define stb_p_strdup strdup
- #define stb_p_stricmp stricmp
- #define stb_p_strnicmp strnicmp
- #endif
- STB_EXTERN void stb_wrapper_malloc(void *newp, size_t sz, char *file, int line);
- STB_EXTERN void stb_wrapper_free(void *oldp, char *file, int line);
- STB_EXTERN void stb_wrapper_realloc(void *oldp, void *newp, size_t sz, char *file, int line);
- STB_EXTERN void stb_wrapper_calloc(size_t num, size_t sz, char *file, int line);
- STB_EXTERN void stb_wrapper_listall(void (*func)(void *ptr, size_t sz, char *file, int line));
- STB_EXTERN void stb_wrapper_dump(char *filename);
- STB_EXTERN size_t stb_wrapper_allocsize(void *oldp);
- STB_EXTERN void stb_wrapper_check(void *oldp);
- #ifdef STB_DEFINE
- // this is a special function used inside malloc wrapper
- // to do allocations that aren't tracked (to avoid
- // reentrancy). Of course if someone _else_ wraps realloc,
- // this breaks, but if they're doing that AND the malloc
- // wrapper they need to explicitly check for reentrancy.
- //
- // only define realloc_raw() and we do realloc(NULL,sz)
- // for malloc() and realloc(p,0) for free().
- static void * stb__realloc_raw(void *p, int sz)
- {
- if (p == NULL) return malloc(sz);
- if (sz == 0) { free(p); return NULL; }
- return realloc(p,sz);
- }
- #endif
- #ifdef _WIN32
- STB_EXTERN void * stb_smalloc(size_t sz);
- STB_EXTERN void stb_sfree(void *p);
- STB_EXTERN void * stb_srealloc(void *p, size_t sz);
- STB_EXTERN void * stb_scalloc(size_t n, size_t sz);
- STB_EXTERN char * stb_sstrdup(char *s);
- #endif
- #ifdef STB_FASTMALLOC
- #define malloc stb_smalloc
- #define free stb_sfree
- #define realloc stb_srealloc
- #define strdup stb_sstrdup
- #define calloc stb_scalloc
- #endif
- #ifndef STB_MALLOC_ALLCHECK
- #define stb__check(p) 1
- #else
- #ifndef STB_MALLOC_WRAPPER
- #error STB_MALLOC_ALLCHECK requires STB_MALLOC_WRAPPER
- #else
- #define stb__check(p) stb_mcheck(p)
- #endif
- #endif
- #ifdef STB_MALLOC_WRAPPER
- STB_EXTERN void * stb__malloc(size_t, char *, int);
- STB_EXTERN void * stb__realloc(void *, size_t, char *, int);
- STB_EXTERN void * stb__calloc(size_t n, size_t s, char *, int);
- STB_EXTERN void stb__free(void *, char *file, int);
- STB_EXTERN char * stb__strdup(char *s, char *file, int);
- STB_EXTERN void stb_malloc_checkall(void);
- STB_EXTERN void stb_malloc_check_counter(int init_delay, int rep_delay);
- #ifndef STB_MALLOC_WRAPPER_DEBUG
- #define stb_mcheck(p) 1
- #else
- STB_EXTERN int stb_mcheck(void *);
- #endif
- #ifdef STB_DEFINE
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- #define STB__PAD 32
- #define STB__BIAS 16
- #define STB__SIG 0x51b01234
- #define STB__FIXSIZE(sz) (((sz+3) & ~3) + STB__PAD)
- #define STB__ptr(x,y) ((char *) (x) + (y))
- #else
- #define STB__ptr(x,y) (x)
- #define STB__FIXSIZE(sz) (sz)
- #endif
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- int stb_mcheck(void *p)
- {
- unsigned int sz;
- if (p == NULL) return 1;
- p = ((char *) p) - STB__BIAS;
- sz = * (unsigned int *) p;
- assert(* (unsigned int *) STB__ptr(p,4) == STB__SIG);
- assert(* (unsigned int *) STB__ptr(p,8) == STB__SIG);
- assert(* (unsigned int *) STB__ptr(p,12) == STB__SIG);
- assert(* (unsigned int *) STB__ptr(p,sz-4) == STB__SIG+1);
- assert(* (unsigned int *) STB__ptr(p,sz-8) == STB__SIG+1);
- assert(* (unsigned int *) STB__ptr(p,sz-12) == STB__SIG+1);
- assert(* (unsigned int *) STB__ptr(p,sz-16) == STB__SIG+1);
- stb_wrapper_check(STB__ptr(p, STB__BIAS));
- return 1;
- }
- static void stb__check2(void *p, size_t sz, char *file, int line)
- {
- stb_mcheck(p);
- }
- void stb_malloc_checkall(void)
- {
- stb_wrapper_listall(stb__check2);
- }
- #else
- void stb_malloc_checkall(void) { }
- #endif
- static int stb__malloc_wait=(1 << 30), stb__malloc_next_wait = (1 << 30), stb__malloc_iter;
- void stb_malloc_check_counter(int init_delay, int rep_delay)
- {
- stb__malloc_wait = init_delay;
- stb__malloc_next_wait = rep_delay;
- }
- void stb_mcheck_all(void)
- {
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- ++stb__malloc_iter;
- if (--stb__malloc_wait <= 0) {
- stb_malloc_checkall();
- stb__malloc_wait = stb__malloc_next_wait;
- }
- #endif
- }
- #ifdef STB_MALLOC_WRAPPER_PAGED
- #define STB__WINDOWS_PAGE (1 << 12)
- #ifndef _WINDOWS_
- STB_EXTERN __declspec(dllimport) void * __stdcall VirtualAlloc(void *p, unsigned long size, unsigned long type, unsigned long protect);
- STB_EXTERN __declspec(dllimport) int __stdcall VirtualFree(void *p, unsigned long size, unsigned long freetype);
- #endif
- #endif
- static void *stb__malloc_final(size_t sz)
- {
- #ifdef STB_MALLOC_WRAPPER_PAGED
- size_t aligned = (sz + STB__WINDOWS_PAGE - 1) & ~(STB__WINDOWS_PAGE-1);
- char *p = VirtualAlloc(NULL, aligned + STB__WINDOWS_PAGE, 0x2000, 0x04); // RESERVE, READWRITE
- if (p == NULL) return p;
- VirtualAlloc(p, aligned, 0x1000, 0x04); // COMMIT, READWRITE
- return p;
- #else
- return malloc(sz);
- #endif
- }
- static void stb__free_final(void *p)
- {
- #ifdef STB_MALLOC_WRAPPER_PAGED
- VirtualFree(p, 0, 0x8000); // RELEASE
- #else
- free(p);
- #endif
- }
- int stb__malloc_failure;
- #ifdef STB_MALLOC_WRAPPER_PAGED
- static void *stb__realloc_final(void *p, size_t sz, size_t old_sz)
- {
- void *q = stb__malloc_final(sz);
- if (q == NULL)
- return ++stb__malloc_failure, q;
- // @TODO: deal with p being smaller!
- memcpy(q, p, sz < old_sz ? sz : old_sz);
- stb__free_final(p);
- return q;
- }
- #endif
- void stb__free(void *p, char *file, int line)
- {
- stb_mcheck_all();
- if (!p) return;
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- stb_mcheck(p);
- #endif
- stb_wrapper_free(p,file,line);
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- p = STB__ptr(p,-STB__BIAS);
- * (unsigned int *) STB__ptr(p,0) = 0xdeadbeef;
- * (unsigned int *) STB__ptr(p,4) = 0xdeadbeef;
- * (unsigned int *) STB__ptr(p,8) = 0xdeadbeef;
- * (unsigned int *) STB__ptr(p,12) = 0xdeadbeef;
- #endif
- stb__free_final(p);
- }
- void * stb__malloc(size_t sz, char *file, int line)
- {
- void *p;
- stb_mcheck_all();
- if (sz == 0) return NULL;
- p = stb__malloc_final(STB__FIXSIZE(sz));
- if (p == NULL) p = stb__malloc_final(STB__FIXSIZE(sz));
- if (p == NULL) p = stb__malloc_final(STB__FIXSIZE(sz));
- if (p == NULL) {
- ++stb__malloc_failure;
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- stb_malloc_checkall();
- #endif
- return p;
- }
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- * (int *) STB__ptr(p,0) = STB__FIXSIZE(sz);
- * (unsigned int *) STB__ptr(p,4) = STB__SIG;
- * (unsigned int *) STB__ptr(p,8) = STB__SIG;
- * (unsigned int *) STB__ptr(p,12) = STB__SIG;
- * (unsigned int *) STB__ptr(p,STB__FIXSIZE(sz)-4) = STB__SIG+1;
- * (unsigned int *) STB__ptr(p,STB__FIXSIZE(sz)-8) = STB__SIG+1;
- * (unsigned int *) STB__ptr(p,STB__FIXSIZE(sz)-12) = STB__SIG+1;
- * (unsigned int *) STB__ptr(p,STB__FIXSIZE(sz)-16) = STB__SIG+1;
- p = STB__ptr(p, STB__BIAS);
- #endif
- stb_wrapper_malloc(p,sz,file,line);
- return p;
- }
- void * stb__realloc(void *p, size_t sz, char *file, int line)
- {
- void *q;
- stb_mcheck_all();
- if (p == NULL) return stb__malloc(sz,file,line);
- if (sz == 0 ) { stb__free(p,file,line); return NULL; }
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- stb_mcheck(p);
- p = STB__ptr(p,-STB__BIAS);
- #endif
- #ifdef STB_MALLOC_WRAPPER_PAGED
- {
- size_t n = stb_wrapper_allocsize(STB__ptr(p,STB__BIAS));
- if (!n)
- stb_wrapper_check(STB__ptr(p,STB__BIAS));
- q = stb__realloc_final(p, STB__FIXSIZE(sz), STB__FIXSIZE(n));
- }
- #else
- q = realloc(p, STB__FIXSIZE(sz));
- #endif
- if (q == NULL)
- return ++stb__malloc_failure, q;
- #ifdef STB_MALLOC_WRAPPER_DEBUG
- * (int *) STB__ptr(q,0) = STB__FIXSIZE(sz);
- * (unsigned int *) STB__ptr(q,4) = STB__SIG;
- * (unsigned int *) STB__ptr(q,8) = STB__SIG;
- * (unsigned int *) STB__ptr(q,12) = STB__SIG;
- * (unsigned int *) STB__ptr(q,STB__FIXSIZE(sz)-4) = STB__SIG+1;
- * (unsigned int *) STB__ptr(q,STB__FIXSIZE(sz)-8) = STB__SIG+1;
- * (unsigned int *) STB__ptr(q,STB__FIXSIZE(sz)-12) = STB__SIG+1;
- * (unsigned int *) STB__ptr(q,STB__FIXSIZE(sz)-16) = STB__SIG+1;
- q = STB__ptr(q, STB__BIAS);
- p = STB__ptr(p, STB__BIAS);
- #endif
- stb_wrapper_realloc(p,q,sz,file,line);
- return q;
- }
- STB_EXTERN int stb_log2_ceil(size_t);
- static void *stb__calloc(size_t n, size_t sz, char *file, int line)
- {
- void *q;
- stb_mcheck_all();
- if (n == 0 || sz == 0) return NULL;
- if (stb_log2_ceil(n) + stb_log2_ceil(sz) >= 32) return NULL;
- q = stb__malloc(n*sz, file, line);
- if (q) memset(q, 0, n*sz);
- return q;
- }
- char * stb__strdup(char *s, char *file, int line)
- {
- char *p;
- stb_mcheck_all();
- p = stb__malloc(strlen(s)+1, file, line);
- if (!p) return p;
- stb_p_strcpy_s(p, strlen(s)+1, s);
- return p;
- }
- #endif // STB_DEFINE
- #ifdef STB_FASTMALLOC
- #undef malloc
- #undef realloc
- #undef free
- #undef strdup
- #undef calloc
- #endif
- // include everything that might define these, BEFORE making macros
- #include <stdlib.h>
- #include <string.h>
- #include <malloc.h>
- #define malloc(s) stb__malloc ( s, __FILE__, __LINE__)
- #define realloc(p,s) stb__realloc(p,s, __FILE__, __LINE__)
- #define calloc(n,s) stb__calloc (n,s, __FILE__, __LINE__)
- #define free(p) stb__free (p, __FILE__, __LINE__)
- #define strdup(p) stb__strdup (p, __FILE__, __LINE__)
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Windows pretty display
- //
- STB_EXTERN void stbprint(const char *fmt, ...);
- STB_EXTERN char *stb_sprintf(const char *fmt, ...);
- STB_EXTERN char *stb_mprintf(const char *fmt, ...);
- STB_EXTERN int stb_snprintf(char *s, size_t n, const char *fmt, ...);
- STB_EXTERN int stb_vsnprintf(char *s, size_t n, const char *fmt, va_list v);
- #ifdef STB_DEFINE
- int stb_vsnprintf(char *s, size_t n, const char *fmt, va_list v)
- {
- int res;
- #ifdef _WIN32
- #ifdef __STDC_WANT_SECURE_LIB__
- res = _vsnprintf_s(s, n, _TRUNCATE, fmt, v);
- #else
- res = stb_p_vsnprintf(s,n,fmt,v);
- #endif
- #else
- res = vsnprintf(s,n,fmt,v);
- #endif
- if (n) s[n-1] = 0;
- // Unix returns length output would require, Windows returns negative when truncated.
- return (res >= (int) n || res < 0) ? -1 : res;
- }
- int stb_snprintf(char *s, size_t n, const char *fmt, ...)
- {
- int res;
- va_list v;
- va_start(v,fmt);
- res = stb_vsnprintf(s, n, fmt, v);
- va_end(v);
- return res;
- }
- char *stb_sprintf(const char *fmt, ...)
- {
- static char buffer[1024];
- va_list v;
- va_start(v,fmt);
- stb_vsnprintf(buffer,1024,fmt,v);
- va_end(v);
- return buffer;
- }
- char *stb_mprintf(const char *fmt, ...)
- {
- static char buffer[1024];
- va_list v;
- va_start(v,fmt);
- stb_vsnprintf(buffer,1024,fmt,v);
- va_end(v);
- return stb_p_strdup(buffer);
- }
- #ifdef _WIN32
- #ifndef _WINDOWS_
- STB_EXTERN __declspec(dllimport) int __stdcall WriteConsoleA(void *, const void *, unsigned int, unsigned int *, void *);
- STB_EXTERN __declspec(dllimport) void * __stdcall GetStdHandle(unsigned int);
- STB_EXTERN __declspec(dllimport) int __stdcall SetConsoleTextAttribute(void *, unsigned short);
- #endif
- static void stb__print_one(void *handle, char *s, ptrdiff_t len)
- {
- if (len)
- if (0==WriteConsoleA(handle, s, (unsigned) len, NULL,NULL))
- // if it fails, maybe redirected, so output normally...
- // but it's supriously reporting failure now on Win7 and later
- {}//fwrite(s, 1, (unsigned) len, stdout);
- }
- static void stb__print(char *s)
- {
- void *handle = GetStdHandle((unsigned int) -11); // STD_OUTPUT_HANDLE
- int pad=0; // number of padding characters to add
- char *t = s;
- while (*s) {
- int lpad;
- while (*s && *s != '{') {
- if (pad) {
- if (*s == '\r' || *s == '\n')
- pad = 0;
- else if (s[0] == ' ' && s[1] == ' ') {
- stb__print_one(handle, t, s-t);
- t = s;
- while (pad) {
- stb__print_one(handle, t, 1);
- --pad;
- }
- }
- }
- ++s;
- }
- if (!*s) break;
- stb__print_one(handle, t, s-t);
- if (s[1] == '{') {
- ++s;
- continue;
- }
- if (s[1] == '#') {
- t = s+3;
- if (isxdigit(s[2]))
- if (isdigit(s[2]))
- SetConsoleTextAttribute(handle, s[2] - '0');
- else
- SetConsoleTextAttribute(handle, tolower(s[2]) - 'a' + 10);
- else {
- SetConsoleTextAttribute(handle, 0x0f);
- t=s+2;
- }
- } else if (s[1] == '!') {
- SetConsoleTextAttribute(handle, 0x0c);
- t = s+2;
- } else if (s[1] == '@') {
- SetConsoleTextAttribute(handle, 0x09);
- t = s+2;
- } else if (s[1] == '$') {
- SetConsoleTextAttribute(handle, 0x0a);
- t = s+2;
- } else {
- SetConsoleTextAttribute(handle, 0x08); // 0,7,8,15 => shades of grey
- t = s+1;
- }
- lpad = (int) (t-s);
- s = t;
- while (*s && *s != '}') ++s;
- if (!*s) break;
- stb__print_one(handle, t, s-t);
- if (s[1] == '}') {
- t = s+2;
- } else {
- pad += 1+lpad;
- t = s+1;
- }
- s=t;
- SetConsoleTextAttribute(handle, 0x07);
- }
- stb__print_one(handle, t, s-t);
- SetConsoleTextAttribute(handle, 0x07);
- }
- void stbprint(const char *fmt, ...)
- {
- int res;
- char buffer[1024];
- char *tbuf = buffer;
- va_list v;
- va_start(v,fmt);
- res = stb_vsnprintf(buffer, sizeof(buffer), fmt, v);
- va_end(v);
- if (res < 0) {
- tbuf = (char *) malloc(16384);
- va_start(v,fmt);
- res = stb_vsnprintf(tbuf,16384, fmt, v);
- va_end(v);
- tbuf[16383] = 0;
- }
- stb__print(tbuf);
- if (tbuf != buffer)
- free(tbuf);
- }
- #else // _WIN32
- void stbprint(const char *fmt, ...)
- {
- va_list v;
- va_start(v,fmt);
- vprintf(fmt,v);
- va_end(v);
- }
- #endif // _WIN32
- #endif // STB_DEFINE
- //////////////////////////////////////////////////////////////////////////////
- //
- // Windows UTF8 filename handling
- //
- // Windows stupidly treats 8-bit filenames as some dopey code page,
- // rather than utf-8. If we want to use utf8 filenames, we have to
- // convert them to WCHAR explicitly and call WCHAR versions of the
- // file functions. So, ok, we do.
- #ifdef _WIN32
- #define stb__fopen(x,y) stb_p_wfopen((const wchar_t *)stb__from_utf8(x), (const wchar_t *)stb__from_utf8_alt(y))
- #define stb__windows(x,y) x
- #else
- #define stb__fopen(x,y) stb_p_fopen(x,y)
- #define stb__windows(x,y) y
- #endif
- typedef unsigned short stb__wchar;
- STB_EXTERN stb__wchar * stb_from_utf8(stb__wchar *buffer, const char *str, int n);
- STB_EXTERN char * stb_to_utf8 (char *buffer, const stb__wchar *str, int n);
- STB_EXTERN stb__wchar *stb__from_utf8(const char *str);
- STB_EXTERN stb__wchar *stb__from_utf8_alt(const char *str);
- STB_EXTERN char *stb__to_utf8(const stb__wchar *str);
- #ifdef STB_DEFINE
- stb__wchar * stb_from_utf8(stb__wchar *buffer, const char *ostr, int n)
- {
- unsigned char *str = (unsigned char *) ostr;
- stb_uint32 c;
- int i=0;
- --n;
- while (*str) {
- if (i >= n)
- return NULL;
- if (!(*str & 0x80))
- buffer[i++] = *str++;
- else if ((*str & 0xe0) == 0xc0) {
- if (*str < 0xc2) return NULL;
- c = (*str++ & 0x1f) << 6;
- if ((*str & 0xc0) != 0x80) return NULL;
- buffer[i++] = c + (*str++ & 0x3f);
- } else if ((*str & 0xf0) == 0xe0) {
- if (*str == 0xe0 && (str[1] < 0xa0 || str[1] > 0xbf)) return NULL;
- if (*str == 0xed && str[1] > 0x9f) return NULL; // str[1] < 0x80 is checked below
- c = (*str++ & 0x0f) << 12;
- if ((*str & 0xc0) != 0x80) return NULL;
- c += (*str++ & 0x3f) << 6;
- if ((*str & 0xc0) != 0x80) return NULL;
- buffer[i++] = c + (*str++ & 0x3f);
- } else if ((*str & 0xf8) == 0xf0) {
- if (*str > 0xf4) return NULL;
- if (*str == 0xf0 && (str[1] < 0x90 || str[1] > 0xbf)) return NULL;
- if (*str == 0xf4 && str[1] > 0x8f) return NULL; // str[1] < 0x80 is checked below
- c = (*str++ & 0x07) << 18;
- if ((*str & 0xc0) != 0x80) return NULL;
- c += (*str++ & 0x3f) << 12;
- if ((*str & 0xc0) != 0x80) return NULL;
- c += (*str++ & 0x3f) << 6;
- if ((*str & 0xc0) != 0x80) return NULL;
- c += (*str++ & 0x3f);
- // utf-8 encodings of values used in surrogate pairs are invalid
- if ((c & 0xFFFFF800) == 0xD800) return NULL;
- if (c >= 0x10000) {
- c -= 0x10000;
- if (i + 2 > n) return NULL;
- buffer[i++] = 0xD800 | (0x3ff & (c >> 10));
- buffer[i++] = 0xDC00 | (0x3ff & (c ));
- }
- } else
- return NULL;
- }
- buffer[i] = 0;
- return buffer;
- }
- char * stb_to_utf8(char *buffer, const stb__wchar *str, int n)
- {
- int i=0;
- --n;
- while (*str) {
- if (*str < 0x80) {
- if (i+1 > n) return NULL;
- buffer[i++] = (char) *str++;
- } else if (*str < 0x800) {
- if (i+2 > n) return NULL;
- buffer[i++] = 0xc0 + (*str >> 6);
- buffer[i++] = 0x80 + (*str & 0x3f);
- str += 1;
- } else if (*str >= 0xd800 && *str < 0xdc00) {
- stb_uint32 c;
- if (i+4 > n) return NULL;
- c = ((str[0] - 0xd800) << 10) + ((str[1]) - 0xdc00) + 0x10000;
- buffer[i++] = 0xf0 + (c >> 18);
- buffer[i++] = 0x80 + ((c >> 12) & 0x3f);
- buffer[i++] = 0x80 + ((c >> 6) & 0x3f);
- buffer[i++] = 0x80 + ((c ) & 0x3f);
- str += 2;
- } else if (*str >= 0xdc00 && *str < 0xe000) {
- return NULL;
- } else {
- if (i+3 > n) return NULL;
- buffer[i++] = 0xe0 + (*str >> 12);
- buffer[i++] = 0x80 + ((*str >> 6) & 0x3f);
- buffer[i++] = 0x80 + ((*str ) & 0x3f);
- str += 1;
- }
- }
- buffer[i] = 0;
- return buffer;
- }
- stb__wchar *stb__from_utf8(const char *str)
- {
- static stb__wchar buffer[4096];
- return stb_from_utf8(buffer, str, 4096);
- }
- stb__wchar *stb__from_utf8_alt(const char *str)
- {
- static stb__wchar buffer[4096];
- return stb_from_utf8(buffer, str, 4096);
- }
- char *stb__to_utf8(const stb__wchar *str)
- {
- static char buffer[4096];
- return stb_to_utf8(buffer, str, 4096);
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Miscellany
- //
- STB_EXTERN void stb_fatal(const char *fmt, ...);
- STB_EXTERN void stb_(char *fmt, ...);
- STB_EXTERN void stb_append_to_file(char *file, char *fmt, ...);
- STB_EXTERN void stb_log(int active);
- STB_EXTERN void stb_log_fileline(int active);
- STB_EXTERN void stb_log_name(char *filename);
- STB_EXTERN void stb_swap(void *p, void *q, size_t sz);
- STB_EXTERN void *stb_copy(void *p, size_t sz);
- STB_EXTERN void stb_pointer_array_free(void *p, int len);
- STB_EXTERN void **stb_array_block_alloc(int count, int blocksize);
- #define stb_arrcount(x) (sizeof(x)/sizeof((x)[0]))
- STB_EXTERN int stb__record_fileline(const char *f, int n);
- #ifdef STB_DEFINE
- static char *stb__file;
- static int stb__line;
- int stb__record_fileline(const char *f, int n)
- {
- stb__file = (char*) f;
- stb__line = n;
- return 0;
- }
- void stb_fatal(const char *s, ...)
- {
- va_list a;
- if (stb__file)
- fprintf(stderr, "[%s:%d] ", stb__file, stb__line);
- va_start(a,s);
- fputs("Fatal error: ", stderr);
- vfprintf(stderr, s, a);
- va_end(a);
- fputs("\n", stderr);
- #ifdef STB_DEBUG
- #ifdef _MSC_VER
- #ifndef STB_PTR64
- __asm int 3; // trap to debugger!
- #else
- __debugbreak();
- #endif
- #else
- __builtin_trap();
- #endif
- #endif
- exit(1);
- }
- static int stb__log_active=1, stb__log_fileline=1;
- void stb_log(int active)
- {
- stb__log_active = active;
- }
- void stb_log_fileline(int active)
- {
- stb__log_fileline = active;
- }
- #ifdef STB_NO_STB_STRINGS
- const char *stb__log_filename = "temp.log";
- #else
- const char *stb__log_filename = "stb.log";
- #endif
- void stb_log_name(char *s)
- {
- stb__log_filename = s;
- }
- void stb_(char *s, ...)
- {
- if (stb__log_active) {
- FILE *f = stb_p_fopen(stb__log_filename, "a");
- if (f) {
- va_list a;
- if (stb__log_fileline && stb__file)
- fprintf(f, "[%s:%4d] ", stb__file, stb__line);
- va_start(a,s);
- vfprintf(f, s, a);
- va_end(a);
- fputs("\n", f);
- fclose(f);
- }
- }
- }
- void stb_append_to_file(char *filename, char *s, ...)
- {
- FILE *f = stb_p_fopen(filename, "a");
- if (f) {
- va_list a;
- va_start(a,s);
- vfprintf(f, s, a);
- va_end(a);
- fputs("\n", f);
- fclose(f);
- }
- }
- typedef struct { char d[4]; } stb__4;
- typedef struct { char d[8]; } stb__8;
- // optimize the small cases, though you shouldn't be calling this for those!
- void stb_swap(void *p, void *q, size_t sz)
- {
- char buffer[256];
- if (p == q) return;
- if (sz == 4) {
- stb__4 temp = * ( stb__4 *) p;
- * (stb__4 *) p = * ( stb__4 *) q;
- * (stb__4 *) q = temp;
- return;
- } else if (sz == 8) {
- stb__8 temp = * ( stb__8 *) p;
- * (stb__8 *) p = * ( stb__8 *) q;
- * (stb__8 *) q = temp;
- return;
- }
- while (sz > sizeof(buffer)) {
- stb_swap(p, q, sizeof(buffer));
- p = (char *) p + sizeof(buffer);
- q = (char *) q + sizeof(buffer);
- sz -= sizeof(buffer);
- }
- memcpy(buffer, p , sz);
- memcpy(p , q , sz);
- memcpy(q , buffer, sz);
- }
- void *stb_copy(void *p, size_t sz)
- {
- void *q = malloc(sz);
- memcpy(q, p, sz);
- return q;
- }
- void stb_pointer_array_free(void *q, int len)
- {
- void **p = (void **) q;
- int i;
- for (i=0; i < len; ++i)
- free(p[i]);
- }
- void **stb_array_block_alloc(int count, int blocksize)
- {
- int i;
- char *p = (char *) malloc(sizeof(void *) * count + count * blocksize);
- void **q;
- if (p == NULL) return NULL;
- q = (void **) p;
- p += sizeof(void *) * count;
- for (i=0; i < count; ++i)
- q[i] = p + i * blocksize;
- return q;
- }
- #endif
- #ifdef STB_DEBUG
- // tricky hack to allow recording FILE,LINE even in varargs functions
- #define STB__RECORD_FILE(x) (stb__record_fileline(__FILE__, __LINE__),(x))
- #define stb_log STB__RECORD_FILE(stb_log)
- #define stb_ STB__RECORD_FILE(stb_)
- #ifndef STB_FATAL_CLEAN
- #define stb_fatal STB__RECORD_FILE(stb_fatal)
- #endif
- #define STB__DEBUG(x) x
- #else
- #define STB__DEBUG(x)
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_temp
- //
- #define stb_temp(block, sz) stb__temp(block, sizeof(block), (sz))
- STB_EXTERN void * stb__temp(void *b, int b_sz, int want_sz);
- STB_EXTERN void stb_tempfree(void *block, void *ptr);
- #ifdef STB_DEFINE
- void * stb__temp(void *b, int b_sz, int want_sz)
- {
- if (b_sz >= want_sz)
- return b;
- else
- return malloc(want_sz);
- }
- void stb_tempfree(void *b, void *p)
- {
- if (p != b)
- free(p);
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // math/sampling operations
- //
- #define stb_lerp(t,a,b) ( (a) + (t) * (float) ((b)-(a)) )
- #define stb_unlerp(t,a,b) ( ((t) - (a)) / (float) ((b) - (a)) )
- #define stb_clamp(x,xmin,xmax) ((x) < (xmin) ? (xmin) : (x) > (xmax) ? (xmax) : (x))
- STB_EXTERN void stb_newell_normal(float *normal, int num_vert, float **vert, int normalize);
- STB_EXTERN int stb_box_face_vertex_axis_side(int face_number, int vertex_number, int axis);
- STB_EXTERN void stb_linear_controller(float *curpos, float target_pos, float acc, float deacc, float dt);
- STB_EXTERN int stb_float_eq(float x, float y, float delta, int max_ulps);
- STB_EXTERN int stb_is_prime(unsigned int m);
- STB_EXTERN unsigned int stb_power_of_two_nearest_prime(int n);
- STB_EXTERN float stb_smoothstep(float t);
- STB_EXTERN float stb_cubic_bezier_1d(float t, float p0, float p1, float p2, float p3);
- STB_EXTERN double stb_linear_remap(double x, double a, double b,
- double c, double d);
- #ifdef STB_DEFINE
- float stb_smoothstep(float t)
- {
- return (3 - 2*t)*(t*t);
- }
- float stb_cubic_bezier_1d(float t, float p0, float p1, float p2, float p3)
- {
- float it = 1-t;
- return it*it*it*p0 + 3*it*it*t*p1 + 3*it*t*t*p2 + t*t*t*p3;
- }
- void stb_newell_normal(float *normal, int num_vert, float **vert, int normalize)
- {
- int i,j;
- float p;
- normal[0] = normal[1] = normal[2] = 0;
- for (i=num_vert-1,j=0; j < num_vert; i=j++) {
- float *u = vert[i];
- float *v = vert[j];
- normal[0] += (u[1] - v[1]) * (u[2] + v[2]);
- normal[1] += (u[2] - v[2]) * (u[0] + v[0]);
- normal[2] += (u[0] - v[0]) * (u[1] + v[1]);
- }
- if (normalize) {
- p = normal[0]*normal[0] + normal[1]*normal[1] + normal[2]*normal[2];
- p = (float) (1.0 / sqrt(p));
- normal[0] *= p;
- normal[1] *= p;
- normal[2] *= p;
- }
- }
- int stb_box_face_vertex_axis_side(int face_number, int vertex_number, int axis)
- {
- static int box_vertices[6][4][3] =
- {
- { { 1,1,1 }, { 1,0,1 }, { 1,0,0 }, { 1,1,0 } },
- { { 0,0,0 }, { 0,0,1 }, { 0,1,1 }, { 0,1,0 } },
- { { 0,0,0 }, { 0,1,0 }, { 1,1,0 }, { 1,0,0 } },
- { { 0,0,0 }, { 1,0,0 }, { 1,0,1 }, { 0,0,1 } },
- { { 1,1,1 }, { 0,1,1 }, { 0,0,1 }, { 1,0,1 } },
- { { 1,1,1 }, { 1,1,0 }, { 0,1,0 }, { 0,1,1 } },
- };
- assert(face_number >= 0 && face_number < 6);
- assert(vertex_number >= 0 && vertex_number < 4);
- assert(axis >= 0 && axis < 3);
- return box_vertices[face_number][vertex_number][axis];
- }
- void stb_linear_controller(float *curpos, float target_pos, float acc, float deacc, float dt)
- {
- float sign = 1, p, cp = *curpos;
- if (cp == target_pos) return;
- if (target_pos < cp) {
- target_pos = -target_pos;
- cp = -cp;
- sign = -1;
- }
- // first decelerate
- if (cp < 0) {
- p = cp + deacc * dt;
- if (p > 0) {
- p = 0;
- dt = dt - cp / deacc;
- if (dt < 0) dt = 0;
- } else {
- dt = 0;
- }
- cp = p;
- }
- // now accelerate
- p = cp + acc*dt;
- if (p > target_pos) p = target_pos;
- *curpos = p * sign;
- // @TODO: testing
- }
- float stb_quadratic_controller(float target_pos, float curpos, float maxvel, float maxacc, float dt, float *curvel)
- {
- return 0; // @TODO
- }
- int stb_float_eq(float x, float y, float delta, int max_ulps)
- {
- if (fabs(x-y) <= delta) return 1;
- if (abs(*(int *)&x - *(int *)&y) <= max_ulps) return 1;
- return 0;
- }
- int stb_is_prime(unsigned int m)
- {
- unsigned int i,j;
- if (m < 2) return 0;
- if (m == 2) return 1;
- if (!(m & 1)) return 0;
- if (m % 3 == 0) return (m == 3);
- for (i=5; (j=i*i), j <= m && j > i; i += 6) {
- if (m % i == 0) return 0;
- if (m % (i+2) == 0) return 0;
- }
- return 1;
- }
- unsigned int stb_power_of_two_nearest_prime(int n)
- {
- static signed char tab[32] = { 0,0,0,0,1,0,-1,0,1,-1,-1,3,-1,0,-1,2,1,
- 0,2,0,-1,-4,-1,5,-1,18,-2,15,2,-1,2,0 };
- if (!tab[0]) {
- int i;
- for (i=0; i < 32; ++i)
- tab[i] = (1 << i) + 2*tab[i] - 1;
- tab[1] = 2;
- tab[0] = 1;
- }
- if (n >= 32) return 0xfffffffb;
- return tab[n];
- }
- double stb_linear_remap(double x, double x_min, double x_max,
- double out_min, double out_max)
- {
- return stb_lerp(stb_unlerp(x,x_min,x_max),out_min,out_max);
- }
- #endif
- // create a macro so it's faster, but you can get at the function pointer
- #define stb_linear_remap(t,a,b,c,d) stb_lerp(stb_unlerp(t,a,b),c,d)
- //////////////////////////////////////////////////////////////////////////////
- //
- // bit operations
- //
- #define stb_big32(c) (((c)[0]<<24) + (c)[1]*65536 + (c)[2]*256 + (c)[3])
- #define stb_little32(c) (((c)[3]<<24) + (c)[2]*65536 + (c)[1]*256 + (c)[0])
- #define stb_big16(c) ((c)[0]*256 + (c)[1])
- #define stb_little16(c) ((c)[1]*256 + (c)[0])
- STB_EXTERN int stb_bitcount(unsigned int a);
- STB_EXTERN unsigned int stb_bitreverse8(unsigned char n);
- STB_EXTERN unsigned int stb_bitreverse(unsigned int n);
- STB_EXTERN int stb_is_pow2(size_t);
- STB_EXTERN int stb_log2_ceil(size_t);
- STB_EXTERN int stb_log2_floor(size_t);
- STB_EXTERN int stb_lowbit8(unsigned int n);
- STB_EXTERN int stb_highbit8(unsigned int n);
- #ifdef STB_DEFINE
- int stb_bitcount(unsigned int a)
- {
- a = (a & 0x55555555) + ((a >> 1) & 0x55555555); // max 2
- a = (a & 0x33333333) + ((a >> 2) & 0x33333333); // max 4
- a = (a + (a >> 4)) & 0x0f0f0f0f; // max 8 per 4, now 8 bits
- a = (a + (a >> 8)); // max 16 per 8 bits
- a = (a + (a >> 16)); // max 32 per 8 bits
- return a & 0xff;
- }
- unsigned int stb_bitreverse8(unsigned char n)
- {
- n = ((n & 0xAA) >> 1) + ((n & 0x55) << 1);
- n = ((n & 0xCC) >> 2) + ((n & 0x33) << 2);
- return (unsigned char) ((n >> 4) + (n << 4));
- }
- unsigned int stb_bitreverse(unsigned int n)
- {
- n = ((n & 0xAAAAAAAA) >> 1) | ((n & 0x55555555) << 1);
- n = ((n & 0xCCCCCCCC) >> 2) | ((n & 0x33333333) << 2);
- n = ((n & 0xF0F0F0F0) >> 4) | ((n & 0x0F0F0F0F) << 4);
- n = ((n & 0xFF00FF00) >> 8) | ((n & 0x00FF00FF) << 8);
- return (n >> 16) | (n << 16);
- }
- int stb_is_pow2(size_t n)
- {
- return (n & (n-1)) == 0;
- }
- // tricky use of 4-bit table to identify 5 bit positions (note the '-1')
- // 3-bit table would require another tree level; 5-bit table wouldn't save one
- #if defined(_WIN32) && !defined(__MINGW32__)
- #pragma warning(push)
- #pragma warning(disable: 4035) // disable warning about no return value
- int stb_log2_floor(size_t n)
- {
- #if _MSC_VER > 1700
- unsigned long i;
- #ifdef STB_PTR64
- _BitScanReverse64(&i, n);
- #else
- _BitScanReverse(&i, n);
- #endif
- return i != 0 ? i : -1;
- #else
- __asm {
- bsr eax,n
- jnz done
- mov eax,-1
- }
- done:;
- #endif
- }
- #pragma warning(pop)
- #else
- int stb_log2_floor(size_t n)
- {
- static signed char log2_4[16] = { -1,0,1,1,2,2,2,2,3,3,3,3,3,3,3,3 };
- #ifdef STB_PTR64
- if (n >= ((size_t) 1u << 32))
- return stb_log2_floor(n >> 32);
- #endif
- // 2 compares if n < 16, 3 compares otherwise
- if (n < (1U << 14))
- if (n < (1U << 4)) return 0 + log2_4[n ];
- else if (n < (1U << 9)) return 5 + log2_4[n >> 5];
- else return 10 + log2_4[n >> 10];
- else if (n < (1U << 24))
- if (n < (1U << 19)) return 15 + log2_4[n >> 15];
- else return 20 + log2_4[n >> 20];
- else if (n < (1U << 29)) return 25 + log2_4[n >> 25];
- else return 30 + log2_4[n >> 30];
- }
- #endif
- // define ceil from floor
- int stb_log2_ceil(size_t n)
- {
- if (stb_is_pow2(n)) return stb_log2_floor(n);
- else return 1 + stb_log2_floor(n);
- }
- int stb_highbit8(unsigned int n)
- {
- return stb_log2_ceil(n&255);
- }
- int stb_lowbit8(unsigned int n)
- {
- static signed char lowbit4[16] = { -1,0,1,0, 2,0,1,0, 3,0,1,0, 2,0,1,0 };
- int k = lowbit4[n & 15];
- if (k >= 0) return k;
- k = lowbit4[(n >> 4) & 15];
- if (k >= 0) return k+4;
- return k;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // qsort Compare Routines
- //
- #ifdef _WIN32
- #define stb_stricmp(a,b) stb_p_stricmp(a,b)
- #define stb_strnicmp(a,b,n) stb_p_strnicmp(a,b,n)
- #else
- #define stb_stricmp(a,b) strcasecmp(a,b)
- #define stb_strnicmp(a,b,n) strncasecmp(a,b,n)
- #endif
- STB_EXTERN int (*stb_intcmp(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_intcmprev(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_qsort_strcmp(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_qsort_stricmp(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_floatcmp(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_doublecmp(int offset))(const void *a, const void *b);
- STB_EXTERN int (*stb_charcmp(int offset))(const void *a, const void *b);
- #ifdef STB_DEFINE
- static int stb__intcmpoffset, stb__ucharcmpoffset, stb__strcmpoffset;
- static int stb__floatcmpoffset, stb__doublecmpoffset;
- static int stb__memcmpoffset, stb__memcmpsize;
- int stb__intcmp(const void *a, const void *b)
- {
- const int p = *(const int *) ((const char *) a + stb__intcmpoffset);
- const int q = *(const int *) ((const char *) b + stb__intcmpoffset);
- return p < q ? -1 : p > q;
- }
- int stb__intcmprev(const void *a, const void *b)
- {
- const int p = *(const int *) ((const char *) a + stb__intcmpoffset);
- const int q = *(const int *) ((const char *) b + stb__intcmpoffset);
- return q < p ? -1 : q > p;
- }
- int stb__ucharcmp(const void *a, const void *b)
- {
- const int p = *(const unsigned char *) ((const char *) a + stb__ucharcmpoffset);
- const int q = *(const unsigned char *) ((const char *) b + stb__ucharcmpoffset);
- return p < q ? -1 : p > q;
- }
- int stb__floatcmp(const void *a, const void *b)
- {
- const float p = *(const float *) ((const char *) a + stb__floatcmpoffset);
- const float q = *(const float *) ((const char *) b + stb__floatcmpoffset);
- return p < q ? -1 : p > q;
- }
- int stb__doublecmp(const void *a, const void *b)
- {
- const double p = *(const double *) ((const char *) a + stb__doublecmpoffset);
- const double q = *(const double *) ((const char *) b + stb__doublecmpoffset);
- return p < q ? -1 : p > q;
- }
- int stb__qsort_strcmp(const void *a, const void *b)
- {
- const char *p = *(const char **) ((const char *) a + stb__strcmpoffset);
- const char *q = *(const char **) ((const char *) b + stb__strcmpoffset);
- return strcmp(p,q);
- }
- int stb__qsort_stricmp(const void *a, const void *b)
- {
- const char *p = *(const char **) ((const char *) a + stb__strcmpoffset);
- const char *q = *(const char **) ((const char *) b + stb__strcmpoffset);
- return stb_stricmp(p,q);
- }
- int stb__memcmp(const void *a, const void *b)
- {
- return memcmp((char *) a + stb__memcmpoffset, (char *) b + stb__memcmpoffset, stb__memcmpsize);
- }
- int (*stb_intcmp(int offset))(const void *, const void *)
- {
- stb__intcmpoffset = offset;
- return &stb__intcmp;
- }
- int (*stb_intcmprev(int offset))(const void *, const void *)
- {
- stb__intcmpoffset = offset;
- return &stb__intcmprev;
- }
- int (*stb_ucharcmp(int offset))(const void *, const void *)
- {
- stb__ucharcmpoffset = offset;
- return &stb__ucharcmp;
- }
- int (*stb_qsort_strcmp(int offset))(const void *, const void *)
- {
- stb__strcmpoffset = offset;
- return &stb__qsort_strcmp;
- }
- int (*stb_qsort_stricmp(int offset))(const void *, const void *)
- {
- stb__strcmpoffset = offset;
- return &stb__qsort_stricmp;
- }
- int (*stb_floatcmp(int offset))(const void *, const void *)
- {
- stb__floatcmpoffset = offset;
- return &stb__floatcmp;
- }
- int (*stb_doublecmp(int offset))(const void *, const void *)
- {
- stb__doublecmpoffset = offset;
- return &stb__doublecmp;
- }
- int (*stb_memcmp(int offset, int size))(const void *, const void *)
- {
- stb__memcmpoffset = offset;
- stb__memcmpsize = size;
- return &stb__memcmp;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Binary Search Toolkit
- //
- typedef struct
- {
- int minval, maxval, guess;
- int mode, step;
- } stb_search;
- STB_EXTERN int stb_search_binary(stb_search *s, int minv, int maxv, int find_smallest);
- STB_EXTERN int stb_search_open(stb_search *s, int minv, int find_smallest);
- STB_EXTERN int stb_probe(stb_search *s, int compare, int *result); // return 0 when done
- #ifdef STB_DEFINE
- enum
- {
- STB_probe_binary_smallest,
- STB_probe_binary_largest,
- STB_probe_open_smallest,
- STB_probe_open_largest,
- };
- static int stb_probe_guess(stb_search *s, int *result)
- {
- switch(s->mode) {
- case STB_probe_binary_largest:
- if (s->minval == s->maxval) {
- *result = s->minval;
- return 0;
- }
- assert(s->minval < s->maxval);
- // if a < b, then a < p <= b
- s->guess = s->minval + (((unsigned) s->maxval - s->minval + 1) >> 1);
- break;
- case STB_probe_binary_smallest:
- if (s->minval == s->maxval) {
- *result = s->minval;
- return 0;
- }
- assert(s->minval < s->maxval);
- // if a < b, then a <= p < b
- s->guess = s->minval + (((unsigned) s->maxval - s->minval) >> 1);
- break;
- case STB_probe_open_smallest:
- case STB_probe_open_largest:
- s->guess = s->maxval; // guess the current maxval
- break;
- }
- *result = s->guess;
- return 1;
- }
- int stb_probe(stb_search *s, int compare, int *result)
- {
- switch(s->mode) {
- case STB_probe_open_smallest:
- case STB_probe_open_largest: {
- if (compare <= 0) {
- // then it lies within minval & maxval
- if (s->mode == STB_probe_open_smallest)
- s->mode = STB_probe_binary_smallest;
- else
- s->mode = STB_probe_binary_largest;
- } else {
- // otherwise, we need to probe larger
- s->minval = s->maxval + 1;
- s->maxval = s->minval + s->step;
- s->step += s->step;
- }
- break;
- }
- case STB_probe_binary_smallest: {
- // if compare < 0, then s->minval <= a < p
- // if compare = 0, then s->minval <= a <= p
- // if compare > 0, then p < a <= s->maxval
- if (compare <= 0)
- s->maxval = s->guess;
- else
- s->minval = s->guess+1;
- break;
- }
- case STB_probe_binary_largest: {
- // if compare < 0, then s->minval <= a < p
- // if compare = 0, then p <= a <= s->maxval
- // if compare > 0, then p < a <= s->maxval
- if (compare < 0)
- s->maxval = s->guess-1;
- else
- s->minval = s->guess;
- break;
- }
- }
- return stb_probe_guess(s, result);
- }
- int stb_search_binary(stb_search *s, int minv, int maxv, int find_smallest)
- {
- int r;
- if (maxv < minv) return minv-1;
- s->minval = minv;
- s->maxval = maxv;
- s->mode = find_smallest ? STB_probe_binary_smallest : STB_probe_binary_largest;
- stb_probe_guess(s, &r);
- return r;
- }
- int stb_search_open(stb_search *s, int minv, int find_smallest)
- {
- int r;
- s->step = 4;
- s->minval = minv;
- s->maxval = minv+s->step;
- s->mode = find_smallest ? STB_probe_open_smallest : STB_probe_open_largest;
- stb_probe_guess(s, &r);
- return r;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // String Processing
- //
- #define stb_prefixi(s,t) (0==stb_strnicmp((s),(t),strlen(t)))
- enum stb_splitpath_flag
- {
- STB_PATH = 1,
- STB_FILE = 2,
- STB_EXT = 4,
- STB_PATH_FILE = STB_PATH + STB_FILE,
- STB_FILE_EXT = STB_FILE + STB_EXT,
- STB_EXT_NO_PERIOD = 8,
- };
- STB_EXTERN char * stb_skipwhite(char *s);
- STB_EXTERN char * stb_trimwhite(char *s);
- STB_EXTERN char * stb_skipnewline(char *s);
- STB_EXTERN char * stb_strncpy(char *s, char *t, int n);
- STB_EXTERN char * stb_substr(char *t, int n);
- STB_EXTERN char * stb_duplower(char *s);
- STB_EXTERN void stb_tolower (char *s);
- STB_EXTERN char * stb_strchr2 (char *s, char p1, char p2);
- STB_EXTERN char * stb_strrchr2(char *s, char p1, char p2);
- STB_EXTERN char * stb_strtok(char *output, char *src, char *delimit);
- STB_EXTERN char * stb_strtok_keep(char *output, char *src, char *delimit);
- STB_EXTERN char * stb_strtok_invert(char *output, char *src, char *allowed);
- STB_EXTERN char * stb_dupreplace(char *s, char *find, char *replace);
- STB_EXTERN void stb_replaceinplace(char *s, char *find, char *replace);
- STB_EXTERN char * stb_splitpath(char *output, char *src, int flag);
- STB_EXTERN char * stb_splitpathdup(char *src, int flag);
- STB_EXTERN char * stb_replacedir(char *output, char *src, char *dir);
- STB_EXTERN char * stb_replaceext(char *output, char *src, char *ext);
- STB_EXTERN void stb_fixpath(char *path);
- STB_EXTERN char * stb_shorten_path_readable(char *path, int max_len);
- STB_EXTERN int stb_suffix (char *s, char *t);
- STB_EXTERN int stb_suffixi(char *s, char *t);
- STB_EXTERN int stb_prefix (char *s, char *t);
- STB_EXTERN char * stb_strichr(char *s, char t);
- STB_EXTERN char * stb_stristr(char *s, char *t);
- STB_EXTERN int stb_prefix_count(char *s, char *t);
- STB_EXTERN const char * stb_plural(int n); // "s" or ""
- STB_EXTERN size_t stb_strscpy(char *d, const char *s, size_t n);
- STB_EXTERN char **stb_tokens(char *src, char *delimit, int *count);
- STB_EXTERN char **stb_tokens_nested(char *src, char *delimit, int *count, char *nest_in, char *nest_out);
- STB_EXTERN char **stb_tokens_nested_empty(char *src, char *delimit, int *count, char *nest_in, char *nest_out);
- STB_EXTERN char **stb_tokens_allowempty(char *src, char *delimit, int *count);
- STB_EXTERN char **stb_tokens_stripwhite(char *src, char *delimit, int *count);
- STB_EXTERN char **stb_tokens_withdelim(char *src, char *delimit, int *count);
- STB_EXTERN char **stb_tokens_quoted(char *src, char *delimit, int *count);
- // with 'quoted', allow delimiters to appear inside quotation marks, and don't
- // strip whitespace inside them (and we delete the quotation marks unless they
- // appear back to back, in which case they're considered escaped)
- #ifdef STB_DEFINE
- size_t stb_strscpy(char *d, const char *s, size_t n)
- {
- size_t len = strlen(s);
- if (len >= n) {
- if (n) d[0] = 0;
- return 0;
- }
- stb_p_strcpy_s(d,n,s);
- return len;
- }
- const char *stb_plural(int n)
- {
- return n == 1 ? "" : "s";
- }
- int stb_prefix(char *s, char *t)
- {
- while (*t)
- if (*s++ != *t++)
- return STB_FALSE;
- return STB_TRUE;
- }
- int stb_prefix_count(char *s, char *t)
- {
- int c=0;
- while (*t) {
- if (*s++ != *t++)
- break;
- ++c;
- }
- return c;
- }
- int stb_suffix(char *s, char *t)
- {
- size_t n = strlen(s);
- size_t m = strlen(t);
- if (m <= n)
- return 0 == strcmp(s+n-m, t);
- else
- return 0;
- }
- int stb_suffixi(char *s, char *t)
- {
- size_t n = strlen(s);
- size_t m = strlen(t);
- if (m <= n)
- return 0 == stb_stricmp(s+n-m, t);
- else
- return 0;
- }
- // originally I was using this table so that I could create known sentinel
- // values--e.g. change whitetable[0] to be true if I was scanning for whitespace,
- // and false if I was scanning for nonwhite. I don't appear to be using that
- // functionality anymore (I do for tokentable, though), so just replace it
- // with isspace()
- char *stb_skipwhite(char *s)
- {
- while (isspace((unsigned char) *s)) ++s;
- return s;
- }
- char *stb_skipnewline(char *s)
- {
- if (s[0] == '\r' || s[0] == '\n') {
- if (s[0]+s[1] == '\r' + '\n') ++s;
- ++s;
- }
- return s;
- }
- char *stb_trimwhite(char *s)
- {
- int i,n;
- s = stb_skipwhite(s);
- n = (int) strlen(s);
- for (i=n-1; i >= 0; --i)
- if (!isspace(s[i]))
- break;
- s[i+1] = 0;
- return s;
- }
- char *stb_strncpy(char *s, char *t, int n)
- {
- stb_p_strncpy_s(s,n+1,t,n);
- s[n] = 0;
- return s;
- }
- char *stb_substr(char *t, int n)
- {
- char *a;
- int z = (int) strlen(t);
- if (z < n) n = z;
- a = (char *) malloc(n+1);
- stb_p_strncpy_s(a,n+1,t,n);
- a[n] = 0;
- return a;
- }
- char *stb_duplower(char *s)
- {
- char *p = stb_p_strdup(s), *q = p;
- while (*q) {
- *q = tolower(*q);
- ++q;
- }
- return p;
- }
- void stb_tolower(char *s)
- {
- while (*s) {
- *s = tolower(*s);
- ++s;
- }
- }
- char *stb_strchr2(char *s, char x, char y)
- {
- for(; *s; ++s)
- if (*s == x || *s == y)
- return s;
- return NULL;
- }
- char *stb_strrchr2(char *s, char x, char y)
- {
- char *r = NULL;
- for(; *s; ++s)
- if (*s == x || *s == y)
- r = s;
- return r;
- }
- char *stb_strichr(char *s, char t)
- {
- if (tolower(t) == toupper(t))
- return strchr(s,t);
- return stb_strchr2(s, (char) tolower(t), (char) toupper(t));
- }
- char *stb_stristr(char *s, char *t)
- {
- size_t n = strlen(t);
- char *z;
- if (n==0) return s;
- while ((z = stb_strichr(s, *t)) != NULL) {
- if (0==stb_strnicmp(z, t, n))
- return z;
- s = z+1;
- }
- return NULL;
- }
- static char *stb_strtok_raw(char *output, char *src, char *delimit, int keep, int invert)
- {
- if (invert) {
- while (*src && strchr(delimit, *src) != NULL) {
- *output++ = *src++;
- }
- } else {
- while (*src && strchr(delimit, *src) == NULL) {
- *output++ = *src++;
- }
- }
- *output = 0;
- if (keep)
- return src;
- else
- return *src ? src+1 : src;
- }
- char *stb_strtok(char *output, char *src, char *delimit)
- {
- return stb_strtok_raw(output, src, delimit, 0, 0);
- }
- char *stb_strtok_keep(char *output, char *src, char *delimit)
- {
- return stb_strtok_raw(output, src, delimit, 1, 0);
- }
- char *stb_strtok_invert(char *output, char *src, char *delimit)
- {
- return stb_strtok_raw(output, src, delimit, 1,1);
- }
- static char **stb_tokens_raw(char *src_, char *delimit, int *count,
- int stripwhite, int allow_empty, char *start, char *end)
- {
- int nested = 0;
- unsigned char *src = (unsigned char *) src_;
- static char stb_tokentable[256]; // rely on static initializion to 0
- static char stable[256],etable[256];
- char *out;
- char **result;
- int num=0;
- unsigned char *s;
- s = (unsigned char *) delimit; while (*s) stb_tokentable[*s++] = 1;
- if (start) {
- s = (unsigned char *) start; while (*s) stable[*s++] = 1;
- s = (unsigned char *) end; if (s) while (*s) stable[*s++] = 1;
- s = (unsigned char *) end; if (s) while (*s) etable[*s++] = 1;
- }
- stable[0] = 1;
- // two passes through: the first time, counting how many
- s = (unsigned char *) src;
- while (*s) {
- // state: just found delimiter
- // skip further delimiters
- if (!allow_empty) {
- stb_tokentable[0] = 0;
- while (stb_tokentable[*s])
- ++s;
- if (!*s) break;
- }
- ++num;
- // skip further non-delimiters
- stb_tokentable[0] = 1;
- if (stripwhite == 2) { // quoted strings
- while (!stb_tokentable[*s]) {
- if (*s != '"')
- ++s;
- else {
- ++s;
- if (*s == '"')
- ++s; // "" -> ", not start a string
- else {
- // begin a string
- while (*s) {
- if (s[0] == '"') {
- if (s[1] == '"') s += 2; // "" -> "
- else { ++s; break; } // terminating "
- } else
- ++s;
- }
- }
- }
- }
- } else
- while (nested || !stb_tokentable[*s]) {
- if (stable[*s]) {
- if (!*s) break;
- if (end ? etable[*s] : nested)
- --nested;
- else
- ++nested;
- }
- ++s;
- }
- if (allow_empty) {
- if (*s) ++s;
- }
- }
- // now num has the actual count... malloc our output structure
- // need space for all the strings: strings won't be any longer than
- // original input, since for every '\0' there's at least one delimiter
- result = (char **) malloc(sizeof(*result) * (num+1) + (s-src+1));
- if (result == NULL) return result;
- out = (char *) (result + (num+1));
- // second pass: copy out the data
- s = (unsigned char *) src;
- num = 0;
- nested = 0;
- while (*s) {
- char *last_nonwhite;
- // state: just found delimiter
- // skip further delimiters
- if (!allow_empty) {
- stb_tokentable[0] = 0;
- if (stripwhite)
- while (stb_tokentable[*s] || isspace(*s))
- ++s;
- else
- while (stb_tokentable[*s])
- ++s;
- } else if (stripwhite) {
- while (isspace(*s)) ++s;
- }
- if (!*s) break;
- // we're past any leading delimiters and whitespace
- result[num] = out;
- ++num;
- // copy non-delimiters
- stb_tokentable[0] = 1;
- last_nonwhite = out-1;
- if (stripwhite == 2) {
- while (!stb_tokentable[*s]) {
- if (*s != '"') {
- if (!isspace(*s)) last_nonwhite = out;
- *out++ = *s++;
- } else {
- ++s;
- if (*s == '"') {
- if (!isspace(*s)) last_nonwhite = out;
- *out++ = *s++; // "" -> ", not start string
- } else {
- // begin a quoted string
- while (*s) {
- if (s[0] == '"') {
- if (s[1] == '"') { *out++ = *s; s += 2; }
- else { ++s; break; } // terminating "
- } else
- *out++ = *s++;
- }
- last_nonwhite = out-1; // all in quotes counts as non-white
- }
- }
- }
- } else {
- while (nested || !stb_tokentable[*s]) {
- if (!isspace(*s)) last_nonwhite = out;
- if (stable[*s]) {
- if (!*s) break;
- if (end ? etable[*s] : nested)
- --nested;
- else
- ++nested;
- }
- *out++ = *s++;
- }
- }
- if (stripwhite) // rewind to last non-whitespace char
- out = last_nonwhite+1;
- *out++ = '\0';
- if (*s) ++s; // skip delimiter
- }
- s = (unsigned char *) delimit; while (*s) stb_tokentable[*s++] = 0;
- if (start) {
- s = (unsigned char *) start; while (*s) stable[*s++] = 1;
- s = (unsigned char *) end; if (s) while (*s) stable[*s++] = 1;
- s = (unsigned char *) end; if (s) while (*s) etable[*s++] = 1;
- }
- if (count != NULL) *count = num;
- result[num] = 0;
- return result;
- }
- char **stb_tokens(char *src, char *delimit, int *count)
- {
- return stb_tokens_raw(src,delimit,count,0,0,0,0);
- }
- char **stb_tokens_nested(char *src, char *delimit, int *count, char *nest_in, char *nest_out)
- {
- return stb_tokens_raw(src,delimit,count,0,0,nest_in,nest_out);
- }
- char **stb_tokens_nested_empty(char *src, char *delimit, int *count, char *nest_in, char *nest_out)
- {
- return stb_tokens_raw(src,delimit,count,0,1,nest_in,nest_out);
- }
- char **stb_tokens_allowempty(char *src, char *delimit, int *count)
- {
- return stb_tokens_raw(src,delimit,count,0,1,0,0);
- }
- char **stb_tokens_stripwhite(char *src, char *delimit, int *count)
- {
- return stb_tokens_raw(src,delimit,count,1,1,0,0);
- }
- char **stb_tokens_quoted(char *src, char *delimit, int *count)
- {
- return stb_tokens_raw(src,delimit,count,2,1,0,0);
- }
- char *stb_dupreplace(char *src, char *find, char *replace)
- {
- size_t len_find = strlen(find);
- size_t len_replace = strlen(replace);
- int count = 0;
- char *s,*p,*q;
- s = strstr(src, find);
- if (s == NULL) return stb_p_strdup(src);
- do {
- ++count;
- s = strstr(s + len_find, find);
- } while (s != NULL);
- p = (char *) malloc(strlen(src) + count * (len_replace - len_find) + 1);
- if (p == NULL) return p;
- q = p;
- s = src;
- for (;;) {
- char *t = strstr(s, find);
- if (t == NULL) {
- stb_p_strcpy_s(q,strlen(src)+count*(len_replace-len_find)+1,s);
- assert(strlen(p) == strlen(src) + count*(len_replace-len_find));
- return p;
- }
- memcpy(q, s, t-s);
- q += t-s;
- memcpy(q, replace, len_replace);
- q += len_replace;
- s = t + len_find;
- }
- }
- void stb_replaceinplace(char *src, char *find, char *replace)
- {
- size_t len_find = strlen(find);
- size_t len_replace = strlen(replace);
- int delta;
- char *s,*p,*q;
- delta = (int) (len_replace - len_find);
- assert(delta <= 0);
- if (delta > 0) return;
- p = strstr(src, find);
- if (p == NULL) return;
- s = q = p;
- while (*s) {
- memcpy(q, replace, len_replace);
- p += len_find;
- q += len_replace;
- s = strstr(p, find);
- if (s == NULL) s = p + strlen(p);
- memmove(q, p, s-p);
- q += s-p;
- p = s;
- }
- *q = 0;
- }
- void stb_fixpath(char *path)
- {
- for(; *path; ++path)
- if (*path == '\\')
- *path = '/';
- }
- void stb__add_section(char *buffer, char *data, ptrdiff_t curlen, ptrdiff_t newlen)
- {
- if (newlen < curlen) {
- ptrdiff_t z1 = newlen >> 1, z2 = newlen-z1;
- memcpy(buffer, data, z1-1);
- buffer[z1-1] = '.';
- buffer[z1-0] = '.';
- memcpy(buffer+z1+1, data+curlen-z2+1, z2-1);
- } else
- memcpy(buffer, data, curlen);
- }
- char * stb_shorten_path_readable(char *path, int len)
- {
- static char buffer[1024];
- ptrdiff_t n = strlen(path),n1,n2,r1,r2;
- char *s;
- if (n <= len) return path;
- if (len > 1024) return path;
- s = stb_strrchr2(path, '/', '\\');
- if (s) {
- n1 = s - path + 1;
- n2 = n - n1;
- ++s;
- } else {
- n1 = 0;
- n2 = n;
- s = path;
- }
- // now we need to reduce r1 and r2 so that they fit in len
- if (n1 < len>>1) {
- r1 = n1;
- r2 = len - r1;
- } else if (n2 < len >> 1) {
- r2 = n2;
- r1 = len - r2;
- } else {
- r1 = n1 * len / n;
- r2 = n2 * len / n;
- if (r1 < len>>2) r1 = len>>2, r2 = len-r1;
- if (r2 < len>>2) r2 = len>>2, r1 = len-r2;
- }
- assert(r1 <= n1 && r2 <= n2);
- if (n1)
- stb__add_section(buffer, path, n1, r1);
- stb__add_section(buffer+r1, s, n2, r2);
- buffer[len] = 0;
- return buffer;
- }
- static char *stb__splitpath_raw(char *buffer, char *path, int flag)
- {
- ptrdiff_t len=0,x,y, n = (int) strlen(path), f1,f2;
- char *s = stb_strrchr2(path, '/', '\\');
- char *t = strrchr(path, '.');
- if (s && t && t < s) t = NULL;
- if (!s) {
- // check for drive
- if (isalpha(path[0]) && path[1] == ':')
- s = &path[1];
- }
- if (s) ++s;
- if (flag == STB_EXT_NO_PERIOD)
- flag |= STB_EXT;
- if (!(flag & (STB_PATH | STB_FILE | STB_EXT))) return NULL;
- f1 = s == NULL ? 0 : s-path; // start of filename
- f2 = t == NULL ? n : t-path; // just past end of filename
- if (flag & STB_PATH) {
- x = 0; if (f1 == 0 && flag == STB_PATH) len=2;
- } else if (flag & STB_FILE) {
- x = f1;
- } else {
- x = f2;
- if (flag & STB_EXT_NO_PERIOD)
- if (path[x] == '.')
- ++x;
- }
- if (flag & STB_EXT)
- y = n;
- else if (flag & STB_FILE)
- y = f2;
- else
- y = f1;
- if (buffer == NULL) {
- buffer = (char *) malloc(y-x + len + 1);
- if (!buffer) return NULL;
- }
- if (len) { stb_p_strcpy_s(buffer, 3, "./"); return buffer; }
- stb_strncpy(buffer, path+(int)x, (int)(y-x));
- return buffer;
- }
- char *stb_splitpath(char *output, char *src, int flag)
- {
- return stb__splitpath_raw(output, src, flag);
- }
- char *stb_splitpathdup(char *src, int flag)
- {
- return stb__splitpath_raw(NULL, src, flag);
- }
- char *stb_replacedir(char *output, char *src, char *dir)
- {
- char buffer[4096];
- stb_splitpath(buffer, src, STB_FILE | STB_EXT);
- if (dir)
- stb_p_sprintf(output stb_p_size(9999), "%s/%s", dir, buffer);
- else
- stb_p_strcpy_s(output, sizeof(buffer), buffer); // @UNSAFE
- return output;
- }
- char *stb_replaceext(char *output, char *src, char *ext)
- {
- char buffer[4096];
- stb_splitpath(buffer, src, STB_PATH | STB_FILE);
- if (ext)
- stb_p_sprintf(output stb_p_size(9999), "%s.%s", buffer, ext[0] == '.' ? ext+1 : ext);
- else
- stb_p_strcpy_s(output, sizeof(buffer), buffer); // @UNSAFE
- return output;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_alloc - hierarchical allocator
- //
- // inspired by http://swapped.cc/halloc
- //
- //
- // When you alloc a given block through stb_alloc, you have these choices:
- //
- // 1. does it have a parent?
- // 2. can it have children?
- // 3. can it be freed directly?
- // 4. is it transferrable?
- // 5. what is its alignment?
- //
- // Here are interesting combinations of those:
- //
- // children free transfer alignment
- // arena Y Y N n/a
- // no-overhead, chunked N N N normal
- // string pool alloc N N N 1
- // parent-ptr, chunked Y N N normal
- // low-overhead, unchunked N Y Y normal
- // general purpose alloc Y Y Y normal
- //
- // Unchunked allocations will probably return 16-aligned pointers. If
- // we 16-align the results, we have room for 4 pointers. For smaller
- // allocations that allow finer alignment, we can reduce the pointers.
- //
- // The strategy is that given a pointer, assuming it has a header (only
- // the no-overhead allocations have no header), we can determine the
- // type of the header fields, and the number of them, by stepping backwards
- // through memory and looking at the tags in the bottom bits.
- //
- // Implementation strategy:
- // chunked allocations come from the middle of chunks, and can't
- // be freed. thefore they do not need to be on a sibling chain.
- // they may need child pointers if they have children.
- //
- // chunked, with-children
- // void *parent;
- //
- // unchunked, no-children -- reduced storage
- // void *next_sibling;
- // void *prev_sibling_nextp;
- //
- // unchunked, general
- // void *first_child;
- // void *next_sibling;
- // void *prev_sibling_nextp;
- // void *chunks;
- //
- // so, if we code each of these fields with different bit patterns
- // (actually same one for next/prev/child), then we can identify which
- // each one is from the last field.
- STB_EXTERN void stb_free(void *p);
- STB_EXTERN void *stb_malloc_global(size_t size);
- STB_EXTERN void *stb_malloc(void *context, size_t size);
- STB_EXTERN void *stb_malloc_nofree(void *context, size_t size);
- STB_EXTERN void *stb_malloc_leaf(void *context, size_t size);
- STB_EXTERN void *stb_malloc_raw(void *context, size_t size);
- STB_EXTERN void *stb_realloc(void *ptr, size_t newsize);
- STB_EXTERN void stb_reassign(void *new_context, void *ptr);
- STB_EXTERN void stb_malloc_validate(void *p, void *parent);
- extern int stb_alloc_chunk_size ;
- extern int stb_alloc_count_free ;
- extern int stb_alloc_count_alloc;
- extern int stb_alloc_alignment ;
- #ifdef STB_DEFINE
- int stb_alloc_chunk_size = 65536;
- int stb_alloc_count_free = 0;
- int stb_alloc_count_alloc = 0;
- int stb_alloc_alignment = -16;
- typedef struct stb__chunk
- {
- struct stb__chunk *next;
- int data_left;
- int alloc;
- } stb__chunk;
- typedef struct
- {
- void * next;
- void ** prevn;
- } stb__nochildren;
- typedef struct
- {
- void ** prevn;
- void * child;
- void * next;
- stb__chunk *chunks;
- } stb__alloc;
- typedef struct
- {
- stb__alloc *parent;
- } stb__chunked;
- #define STB__PARENT 1
- #define STB__CHUNKS 2
- typedef enum
- {
- STB__nochildren = 0,
- STB__chunked = STB__PARENT,
- STB__alloc = STB__CHUNKS,
- STB__chunk_raw = 4,
- } stb__alloc_type;
- // these functions set the bottom bits of a pointer efficiently
- #define STB__DECODE(x,v) ((void *) ((char *) (x) - (v)))
- #define STB__ENCODE(x,v) ((void *) ((char *) (x) + (v)))
- #define stb__parent(z) (stb__alloc *) STB__DECODE((z)->parent, STB__PARENT)
- #define stb__chunks(z) (stb__chunk *) STB__DECODE((z)->chunks, STB__CHUNKS)
- #define stb__setparent(z,p) (z)->parent = (stb__alloc *) STB__ENCODE((p), STB__PARENT)
- #define stb__setchunks(z,c) (z)->chunks = (stb__chunk *) STB__ENCODE((c), STB__CHUNKS)
- static stb__alloc stb__alloc_global =
- {
- NULL,
- NULL,
- NULL,
- (stb__chunk *) STB__ENCODE(NULL, STB__CHUNKS)
- };
- static stb__alloc_type stb__identify(void *p)
- {
- void **q = (void **) p;
- return (stb__alloc_type) ((stb_uinta) q[-1] & 3);
- }
- static void *** stb__prevn(void *p)
- {
- if (stb__identify(p) == STB__alloc) {
- stb__alloc *s = (stb__alloc *) p - 1;
- return &s->prevn;
- } else {
- stb__nochildren *s = (stb__nochildren *) p - 1;
- return &s->prevn;
- }
- }
- void stb_free(void *p)
- {
- if (p == NULL) return;
- // count frees so that unit tests can see what's happening
- ++stb_alloc_count_free;
- switch(stb__identify(p)) {
- case STB__chunked:
- // freeing a chunked-block with children does nothing;
- // they only get freed when the parent does
- // surely this is wrong, and it should free them immediately?
- // otherwise how are they getting put on the right chain?
- return;
- case STB__nochildren: {
- stb__nochildren *s = (stb__nochildren *) p - 1;
- // unlink from sibling chain
- *(s->prevn) = s->next;
- if (s->next)
- *stb__prevn(s->next) = s->prevn;
- free(s);
- return;
- }
- case STB__alloc: {
- stb__alloc *s = (stb__alloc *) p - 1;
- stb__chunk *c, *n;
- void *q;
- // unlink from sibling chain, if any
- *(s->prevn) = s->next;
- if (s->next)
- *stb__prevn(s->next) = s->prevn;
- // first free chunks
- c = (stb__chunk *) stb__chunks(s);
- while (c != NULL) {
- n = c->next;
- stb_alloc_count_free += c->alloc;
- free(c);
- c = n;
- }
- // validating
- stb__setchunks(s,NULL);
- s->prevn = NULL;
- s->next = NULL;
- // now free children
- while ((q = s->child) != NULL) {
- stb_free(q);
- }
- // now free self
- free(s);
- return;
- }
- default:
- assert(0); /* NOTREACHED */
- }
- }
- void stb_malloc_validate(void *p, void *parent)
- {
- if (p == NULL) return;
- switch(stb__identify(p)) {
- case STB__chunked:
- return;
- case STB__nochildren: {
- stb__nochildren *n = (stb__nochildren *) p - 1;
- if (n->prevn)
- assert(*n->prevn == p);
- if (n->next) {
- assert(*stb__prevn(n->next) == &n->next);
- stb_malloc_validate(n, parent);
- }
- return;
- }
- case STB__alloc: {
- stb__alloc *s = (stb__alloc *) p - 1;
- if (s->prevn)
- assert(*s->prevn == p);
- if (s->child) {
- assert(*stb__prevn(s->child) == &s->child);
- stb_malloc_validate(s->child, p);
- }
- if (s->next) {
- assert(*stb__prevn(s->next) == &s->next);
- stb_malloc_validate(s->next, parent);
- }
- return;
- }
- default:
- assert(0); /* NOTREACHED */
- }
- }
- static void * stb__try_chunk(stb__chunk *c, int size, int align, int pre_align)
- {
- char *memblock = (char *) (c+1), *q;
- stb_inta iq;
- int start_offset;
- // we going to allocate at the end of the chunk, not the start. confusing,
- // but it means we don't need both a 'limit' and a 'cur', just a 'cur'.
- // the block ends at: p + c->data_left
- // then we move back by size
- start_offset = c->data_left - size;
- // now we need to check the alignment of that
- q = memblock + start_offset;
- iq = (stb_inta) q;
- assert(sizeof(q) == sizeof(iq));
- // suppose align = 2
- // then we need to retreat iq far enough that (iq & (2-1)) == 0
- // to get (iq & (align-1)) = 0 requires subtracting (iq & (align-1))
- start_offset -= iq & (align-1);
- assert(((stb_uinta) (memblock+start_offset) & (align-1)) == 0);
- // now, if that + pre_align works, go for it!
- start_offset -= pre_align;
- if (start_offset >= 0) {
- c->data_left = start_offset;
- return memblock + start_offset;
- }
- return NULL;
- }
- static void stb__sort_chunks(stb__alloc *src)
- {
- // of the first two chunks, put the chunk with more data left in it first
- stb__chunk *c = stb__chunks(src), *d;
- if (c == NULL) return;
- d = c->next;
- if (d == NULL) return;
- if (c->data_left > d->data_left) return;
- c->next = d->next;
- d->next = c;
- stb__setchunks(src, d);
- }
- static void * stb__alloc_chunk(stb__alloc *src, int size, int align, int pre_align)
- {
- void *p;
- stb__chunk *c = stb__chunks(src);
- if (c && size <= stb_alloc_chunk_size) {
- p = stb__try_chunk(c, size, align, pre_align);
- if (p) { ++c->alloc; return p; }
- // try a second chunk to reduce wastage
- if (c->next) {
- p = stb__try_chunk(c->next, size, align, pre_align);
- if (p) { ++c->alloc; return p; }
- // put the bigger chunk first, since the second will get buried
- // the upshot of this is that, until it gets allocated from, chunk #2
- // is always the largest remaining chunk. (could formalize
- // this with a heap!)
- stb__sort_chunks(src);
- c = stb__chunks(src);
- }
- }
- // allocate a new chunk
- {
- stb__chunk *n;
- int chunk_size = stb_alloc_chunk_size;
- // we're going to allocate a new chunk to put this in
- if (size > chunk_size)
- chunk_size = size;
- assert(sizeof(*n) + pre_align <= 16);
- // loop trying to allocate a large enough chunk
- // the loop is because the alignment may cause problems if it's big...
- // and we don't know what our chunk alignment is going to be
- while (1) {
- n = (stb__chunk *) malloc(16 + chunk_size);
- if (n == NULL) return NULL;
- n->data_left = chunk_size - sizeof(*n);
- p = stb__try_chunk(n, size, align, pre_align);
- if (p != NULL) {
- n->next = c;
- stb__setchunks(src, n);
- // if we just used up the whole block immediately,
- // move the following chunk up
- n->alloc = 1;
- if (size == chunk_size)
- stb__sort_chunks(src);
- return p;
- }
- free(n);
- chunk_size += 16+align;
- }
- }
- }
- static stb__alloc * stb__get_context(void *context)
- {
- if (context == NULL) {
- return &stb__alloc_global;
- } else {
- int u = stb__identify(context);
- // if context is chunked, grab parent
- if (u == STB__chunked) {
- stb__chunked *s = (stb__chunked *) context - 1;
- return stb__parent(s);
- } else {
- return (stb__alloc *) context - 1;
- }
- }
- }
- static void stb__insert_alloc(stb__alloc *src, stb__alloc *s)
- {
- s->prevn = &src->child;
- s->next = src->child;
- src->child = s+1;
- if (s->next)
- *stb__prevn(s->next) = &s->next;
- }
- static void stb__insert_nochild(stb__alloc *src, stb__nochildren *s)
- {
- s->prevn = &src->child;
- s->next = src->child;
- src->child = s+1;
- if (s->next)
- *stb__prevn(s->next) = &s->next;
- }
- static void * malloc_base(void *context, size_t size, stb__alloc_type t, int align)
- {
- void *p;
- stb__alloc *src = stb__get_context(context);
- if (align <= 0) {
- // compute worst-case C packed alignment
- // e.g. a 24-byte struct is 8-aligned
- int align_proposed = 1 << stb_lowbit8((unsigned int) size);
- if (align_proposed < 0)
- align_proposed = 4;
- if (align_proposed == 0) {
- if (size == 0)
- align_proposed = 1;
- else
- align_proposed = 256;
- }
- // a negative alignment means 'don't align any larger
- // than this'; so -16 means we align 1,2,4,8, or 16
- if (align < 0) {
- if (align_proposed > -align)
- align_proposed = -align;
- }
- align = align_proposed;
- }
- assert(stb_is_pow2(align));
- // don't cause misalignment when allocating nochildren
- if (t == STB__nochildren && align > 8)
- t = STB__alloc;
- switch (t) {
- case STB__alloc: {
- stb__alloc *s = (stb__alloc *) malloc(size + sizeof(*s));
- if (s == NULL) return NULL;
- p = s+1;
- s->child = NULL;
- stb__insert_alloc(src, s);
- stb__setchunks(s,NULL);
- break;
- }
- case STB__nochildren: {
- stb__nochildren *s = (stb__nochildren *) malloc(size + sizeof(*s));
- if (s == NULL) return NULL;
- p = s+1;
- stb__insert_nochild(src, s);
- break;
- }
- case STB__chunk_raw: {
- p = stb__alloc_chunk(src, (int) size, align, 0);
- if (p == NULL) return NULL;
- break;
- }
- case STB__chunked: {
- stb__chunked *s;
- if (align < sizeof(stb_uintptr)) align = sizeof(stb_uintptr);
- s = (stb__chunked *) stb__alloc_chunk(src, (int) size, align, sizeof(*s));
- if (s == NULL) return NULL;
- stb__setparent(s, src);
- p = s+1;
- break;
- }
- default: p = NULL; assert(0); /* NOTREACHED */
- }
- ++stb_alloc_count_alloc;
- return p;
- }
- void *stb_malloc_global(size_t size)
- {
- return malloc_base(NULL, size, STB__alloc, stb_alloc_alignment);
- }
- void *stb_malloc(void *context, size_t size)
- {
- return malloc_base(context, size, STB__alloc, stb_alloc_alignment);
- }
- void *stb_malloc_nofree(void *context, size_t size)
- {
- return malloc_base(context, size, STB__chunked, stb_alloc_alignment);
- }
- void *stb_malloc_leaf(void *context, size_t size)
- {
- return malloc_base(context, size, STB__nochildren, stb_alloc_alignment);
- }
- void *stb_malloc_raw(void *context, size_t size)
- {
- return malloc_base(context, size, STB__chunk_raw, stb_alloc_alignment);
- }
- char *stb_malloc_string(void *context, size_t size)
- {
- return (char *) malloc_base(context, size, STB__chunk_raw, 1);
- }
- void *stb_realloc(void *ptr, size_t newsize)
- {
- stb__alloc_type t;
- if (ptr == NULL) return stb_malloc(NULL, newsize);
- if (newsize == 0) { stb_free(ptr); return NULL; }
- t = stb__identify(ptr);
- assert(t == STB__alloc || t == STB__nochildren);
- if (t == STB__alloc) {
- stb__alloc *s = (stb__alloc *) ptr - 1;
- s = (stb__alloc *) realloc(s, newsize + sizeof(*s));
- if (s == NULL) return NULL;
- ptr = s+1;
- // update pointers
- (*s->prevn) = ptr;
- if (s->next)
- *stb__prevn(s->next) = &s->next;
- if (s->child)
- *stb__prevn(s->child) = &s->child;
- return ptr;
- } else {
- stb__nochildren *s = (stb__nochildren *) ptr - 1;
- s = (stb__nochildren *) realloc(ptr, newsize + sizeof(s));
- if (s == NULL) return NULL;
- // update pointers
- (*s->prevn) = s+1;
- if (s->next)
- *stb__prevn(s->next) = &s->next;
- return s+1;
- }
- }
- void *stb_realloc_c(void *context, void *ptr, size_t newsize)
- {
- if (ptr == NULL) return stb_malloc(context, newsize);
- if (newsize == 0) { stb_free(ptr); return NULL; }
- // @TODO: verify you haven't changed contexts
- return stb_realloc(ptr, newsize);
- }
- void stb_reassign(void *new_context, void *ptr)
- {
- stb__alloc *src = stb__get_context(new_context);
- stb__alloc_type t = stb__identify(ptr);
- assert(t == STB__alloc || t == STB__nochildren);
- if (t == STB__alloc) {
- stb__alloc *s = (stb__alloc *) ptr - 1;
- // unlink from old
- *(s->prevn) = s->next;
- if (s->next)
- *stb__prevn(s->next) = s->prevn;
- stb__insert_alloc(src, s);
- } else {
- stb__nochildren *s = (stb__nochildren *) ptr - 1;
- // unlink from old
- *(s->prevn) = s->next;
- if (s->next)
- *stb__prevn(s->next) = s->prevn;
- stb__insert_nochild(src, s);
- }
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_arr
- //
- // An stb_arr is directly useable as a pointer (use the actual type in your
- // definition), but when it resizes, it returns a new pointer and you can't
- // use the old one, so you have to be careful to copy-in-out as necessary.
- //
- // Use a NULL pointer as a 0-length array.
- //
- // float *my_array = NULL, *temp;
- //
- // // add elements on the end one at a time
- // stb_arr_push(my_array, 0.0f);
- // stb_arr_push(my_array, 1.0f);
- // stb_arr_push(my_array, 2.0f);
- //
- // assert(my_array[1] == 2.0f);
- //
- // // add an uninitialized element at the end, then assign it
- // *stb_arr_add(my_array) = 3.0f;
- //
- // // add three uninitialized elements at the end
- // temp = stb_arr_addn(my_array,3);
- // temp[0] = 4.0f;
- // temp[1] = 5.0f;
- // temp[2] = 6.0f;
- //
- // assert(my_array[5] == 5.0f);
- //
- // // remove the last one
- // stb_arr_pop(my_array);
- //
- // assert(stb_arr_len(my_array) == 6);
- #ifdef STB_MALLOC_WRAPPER
- #define STB__PARAMS , char *file, int line
- #define STB__ARGS , file, line
- #else
- #define STB__PARAMS
- #define STB__ARGS
- #endif
- // calling this function allocates an empty stb_arr attached to p
- // (whereas NULL isn't attached to anything)
- STB_EXTERN void stb_arr_malloc(void **target, void *context);
- // call this function with a non-NULL value to have all successive
- // stbs that are created be attached to the associated parent. Note
- // that once a given stb_arr is non-empty, it stays attached to its
- // current parent, even if you call this function again.
- // it turns the previous value, so you can restore it
- STB_EXTERN void* stb_arr_malloc_parent(void *p);
- // simple functions written on top of other functions
- #define stb_arr_empty(a) ( stb_arr_len(a) == 0 )
- #define stb_arr_add(a) ( stb_arr_addn((a),1) )
- #define stb_arr_push(a,v) ( *stb_arr_add(a)=(v) )
- typedef struct
- {
- int len, limit;
- int stb_malloc;
- unsigned int signature;
- } stb__arr;
- #define stb_arr_signature 0x51bada7b // ends with 0123 in decimal
- // access the header block stored before the data
- #define stb_arrhead(a) /*lint --e(826)*/ (((stb__arr *) (a)) - 1)
- #define stb_arrhead2(a) /*lint --e(826)*/ (((stb__arr *) (a)) - 1)
- #ifdef STB_DEBUG
- #define stb_arr_check(a) assert(!a || stb_arrhead(a)->signature == stb_arr_signature)
- #define stb_arr_check2(a) assert(!a || stb_arrhead2(a)->signature == stb_arr_signature)
- #else
- #define stb_arr_check(a) ((void) 0)
- #define stb_arr_check2(a) ((void) 0)
- #endif
- // ARRAY LENGTH
- // get the array length; special case if pointer is NULL
- #define stb_arr_len(a) (a ? stb_arrhead(a)->len : 0)
- #define stb_arr_len2(a) ((stb__arr *) (a) ? stb_arrhead2(a)->len : 0)
- #define stb_arr_lastn(a) (stb_arr_len(a)-1)
- // check whether a given index is valid -- tests 0 <= i < stb_arr_len(a)
- #define stb_arr_valid(a,i) (a ? (int) (i) < stb_arrhead(a)->len : 0)
- // change the array length so is is exactly N entries long, creating
- // uninitialized entries as needed
- #define stb_arr_setlen(a,n) \
- (stb__arr_setlen((void **) &(a), sizeof(a[0]), (n)))
- // change the array length so that N is a valid index (that is, so
- // it is at least N entries long), creating uninitialized entries as needed
- #define stb_arr_makevalid(a,n) \
- (stb_arr_len(a) < (n)+1 ? stb_arr_setlen(a,(n)+1),(a) : (a))
- // remove the last element of the array, returning it
- #define stb_arr_pop(a) ((stb_arr_check(a), (a))[--stb_arrhead(a)->len])
- // access the last element in the array
- #define stb_arr_last(a) ((stb_arr_check(a), (a))[stb_arr_len(a)-1])
- // is iterator at end of list?
- #define stb_arr_end(a,i) ((i) >= &(a)[stb_arr_len(a)])
- // (internal) change the allocated length of the array
- #define stb_arr__grow(a,n) (stb_arr_check(a), stb_arrhead(a)->len += (n))
- // add N new uninitialized elements to the end of the array
- #define stb_arr__addn(a,n) /*lint --e(826)*/ \
- ((stb_arr_len(a)+(n) > stb_arrcurmax(a)) \
- ? (stb__arr_addlen((void **) &(a),sizeof(*a),(n)),0) \
- : ((stb_arr__grow(a,n), 0)))
- // add N new uninitialized elements to the end of the array, and return
- // a pointer to the first new one
- #define stb_arr_addn(a,n) (stb_arr__addn((a),n),(a)+stb_arr_len(a)-(n))
- // add N new uninitialized elements starting at index 'i'
- #define stb_arr_insertn(a,i,n) (stb__arr_insertn((void **) &(a), sizeof(*a), (i), (n)))
- // insert an element at i
- #define stb_arr_insert(a,i,v) (stb__arr_insertn((void **) &(a), sizeof(*a), (i), (1)), ((a)[i] = v))
- // delete N elements from the middle starting at index 'i'
- #define stb_arr_deleten(a,i,n) (stb__arr_deleten((void **) &(a), sizeof(*a), (i), (n)))
- // delete the i'th element
- #define stb_arr_delete(a,i) stb_arr_deleten(a,i,1)
- // delete the i'th element, swapping down from the end
- #define stb_arr_fastdelete(a,i) \
- (stb_swap(&a[i], &a[stb_arrhead(a)->len-1], sizeof(*a)), stb_arr_pop(a))
- // ARRAY STORAGE
- // get the array maximum storage; special case if NULL
- #define stb_arrcurmax(a) (a ? stb_arrhead(a)->limit : 0)
- #define stb_arrcurmax2(a) (a ? stb_arrhead2(a)->limit : 0)
- // set the maxlength of the array to n in anticipation of further growth
- #define stb_arr_setsize(a,n) (stb_arr_check(a), stb__arr_setsize((void **) &(a),sizeof((a)[0]),n))
- // make sure maxlength is large enough for at least N new allocations
- #define stb_arr_atleast(a,n) (stb_arr_len(a)+(n) > stb_arrcurmax(a) \
- ? stb_arr_setsize((a), (n)) : 0)
- // make a copy of a given array (copies contents via 'memcpy'!)
- #define stb_arr_copy(a) stb__arr_copy(a, sizeof((a)[0]))
- // compute the storage needed to store all the elements of the array
- #define stb_arr_storage(a) (stb_arr_len(a) * sizeof((a)[0]))
- #define stb_arr_for(v,arr) for((v)=(arr); (v) < (arr)+stb_arr_len(arr); ++(v))
- // IMPLEMENTATION
- STB_EXTERN void stb_arr_free_(void **p);
- STB_EXTERN void *stb__arr_copy_(void *p, int elem_size);
- STB_EXTERN void stb__arr_setsize_(void **p, int size, int limit STB__PARAMS);
- STB_EXTERN void stb__arr_setlen_(void **p, int size, int newlen STB__PARAMS);
- STB_EXTERN void stb__arr_addlen_(void **p, int size, int addlen STB__PARAMS);
- STB_EXTERN void stb__arr_deleten_(void **p, int size, int loc, int n STB__PARAMS);
- STB_EXTERN void stb__arr_insertn_(void **p, int size, int loc, int n STB__PARAMS);
- #define stb_arr_free(p) stb_arr_free_((void **) &(p))
- #define stb__arr_copy stb__arr_copy_
- #ifndef STB_MALLOC_WRAPPER
- #define stb__arr_setsize stb__arr_setsize_
- #define stb__arr_setlen stb__arr_setlen_
- #define stb__arr_addlen stb__arr_addlen_
- #define stb__arr_deleten stb__arr_deleten_
- #define stb__arr_insertn stb__arr_insertn_
- #else
- #define stb__arr_addlen(p,s,n) stb__arr_addlen_(p,s,n,__FILE__,__LINE__)
- #define stb__arr_setlen(p,s,n) stb__arr_setlen_(p,s,n,__FILE__,__LINE__)
- #define stb__arr_setsize(p,s,n) stb__arr_setsize_(p,s,n,__FILE__,__LINE__)
- #define stb__arr_deleten(p,s,i,n) stb__arr_deleten_(p,s,i,n,__FILE__,__LINE__)
- #define stb__arr_insertn(p,s,i,n) stb__arr_insertn_(p,s,i,n,__FILE__,__LINE__)
- #endif
- #ifdef STB_DEFINE
- static void *stb__arr_context;
- void *stb_arr_malloc_parent(void *p)
- {
- void *q = stb__arr_context;
- stb__arr_context = p;
- return q;
- }
- void stb_arr_malloc(void **target, void *context)
- {
- stb__arr *q = (stb__arr *) stb_malloc(context, sizeof(*q));
- q->len = q->limit = 0;
- q->stb_malloc = 1;
- q->signature = stb_arr_signature;
- *target = (void *) (q+1);
- }
- static void * stb__arr_malloc(int size)
- {
- if (stb__arr_context)
- return stb_malloc(stb__arr_context, size);
- return malloc(size);
- }
- void * stb__arr_copy_(void *p, int elem_size)
- {
- stb__arr *q;
- if (p == NULL) return p;
- q = (stb__arr *) stb__arr_malloc(sizeof(*q) + elem_size * stb_arrhead2(p)->limit);
- stb_arr_check2(p);
- memcpy(q, stb_arrhead2(p), sizeof(*q) + elem_size * stb_arrhead2(p)->len);
- q->stb_malloc = !!stb__arr_context;
- return q+1;
- }
- void stb_arr_free_(void **pp)
- {
- void *p = *pp;
- stb_arr_check2(p);
- if (p) {
- stb__arr *q = stb_arrhead2(p);
- if (q->stb_malloc)
- stb_free(q);
- else
- free(q);
- }
- *pp = NULL;
- }
- static void stb__arrsize_(void **pp, int size, int limit, int len STB__PARAMS)
- {
- void *p = *pp;
- stb__arr *a;
- stb_arr_check2(p);
- if (p == NULL) {
- if (len == 0 && size == 0) return;
- a = (stb__arr *) stb__arr_malloc(sizeof(*a) + size*limit);
- a->limit = limit;
- a->len = len;
- a->stb_malloc = !!stb__arr_context;
- a->signature = stb_arr_signature;
- } else {
- a = stb_arrhead2(p);
- a->len = len;
- if (a->limit < limit) {
- void *p;
- if (a->limit >= 4 && limit < a->limit * 2)
- limit = a->limit * 2;
- if (a->stb_malloc)
- p = stb_realloc(a, sizeof(*a) + limit*size);
- else
- #ifdef STB_MALLOC_WRAPPER
- p = stb__realloc(a, sizeof(*a) + limit*size, file, line);
- #else
- p = realloc(a, sizeof(*a) + limit*size);
- #endif
- if (p) {
- a = (stb__arr *) p;
- a->limit = limit;
- } else {
- // throw an error!
- }
- }
- }
- a->len = stb_min(a->len, a->limit);
- *pp = a+1;
- }
- void stb__arr_setsize_(void **pp, int size, int limit STB__PARAMS)
- {
- void *p = *pp;
- stb_arr_check2(p);
- stb__arrsize_(pp, size, limit, stb_arr_len2(p) STB__ARGS);
- }
- void stb__arr_setlen_(void **pp, int size, int newlen STB__PARAMS)
- {
- void *p = *pp;
- stb_arr_check2(p);
- if (stb_arrcurmax2(p) < newlen || p == NULL) {
- stb__arrsize_(pp, size, newlen, newlen STB__ARGS);
- } else {
- stb_arrhead2(p)->len = newlen;
- }
- }
- void stb__arr_addlen_(void **p, int size, int addlen STB__PARAMS)
- {
- stb__arr_setlen_(p, size, stb_arr_len2(*p) + addlen STB__ARGS);
- }
- void stb__arr_insertn_(void **pp, int size, int i, int n STB__PARAMS)
- {
- void *p = *pp;
- if (n) {
- int z;
- if (p == NULL) {
- stb__arr_addlen_(pp, size, n STB__ARGS);
- return;
- }
- z = stb_arr_len2(p);
- stb__arr_addlen_(&p, size, n STB__ARGS);
- memmove((char *) p + (i+n)*size, (char *) p + i*size, size * (z-i));
- }
- *pp = p;
- }
- void stb__arr_deleten_(void **pp, int size, int i, int n STB__PARAMS)
- {
- void *p = *pp;
- if (n) {
- memmove((char *) p + i*size, (char *) p + (i+n)*size, size * (stb_arr_len2(p)-(i+n)));
- stb_arrhead2(p)->len -= n;
- }
- *pp = p;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Hashing
- //
- // typical use for this is to make a power-of-two hash table.
- //
- // let N = size of table (2^n)
- // let H = stb_hash(str)
- // let S = stb_rehash(H) | 1
- //
- // then hash probe sequence P(i) for i=0..N-1
- // P(i) = (H + S*i) & (N-1)
- //
- // the idea is that H has 32 bits of hash information, but the
- // table has only, say, 2^20 entries so only uses 20 of the bits.
- // then by rehashing the original H we get 2^12 different probe
- // sequences for a given initial probe location. (So it's optimal
- // for 64K tables and its optimality decreases past that.)
- //
- // ok, so I've added something that generates _two separate_
- // 32-bit hashes simultaneously which should scale better to
- // very large tables.
- STB_EXTERN unsigned int stb_hash(char *str);
- STB_EXTERN unsigned int stb_hashptr(void *p);
- STB_EXTERN unsigned int stb_hashlen(char *str, int len);
- STB_EXTERN unsigned int stb_rehash_improved(unsigned int v);
- STB_EXTERN unsigned int stb_hash_fast(void *p, int len);
- STB_EXTERN unsigned int stb_hash2(char *str, unsigned int *hash2_ptr);
- STB_EXTERN unsigned int stb_hash_number(unsigned int hash);
- #define stb_rehash(x) ((x) + ((x) >> 6) + ((x) >> 19))
- #ifdef STB_DEFINE
- unsigned int stb_hash(char *str)
- {
- unsigned int hash = 0;
- while (*str)
- hash = (hash << 7) + (hash >> 25) + *str++;
- return hash + (hash >> 16);
- }
- unsigned int stb_hashlen(char *str, int len)
- {
- unsigned int hash = 0;
- while (len-- > 0 && *str)
- hash = (hash << 7) + (hash >> 25) + *str++;
- return hash + (hash >> 16);
- }
- unsigned int stb_hashptr(void *p)
- {
- unsigned int x = (unsigned int)(size_t) p;
- // typically lacking in low bits and high bits
- x = stb_rehash(x);
- x += x << 16;
- // pearson's shuffle
- x ^= x << 3;
- x += x >> 5;
- x ^= x << 2;
- x += x >> 15;
- x ^= x << 10;
- return stb_rehash(x);
- }
- unsigned int stb_rehash_improved(unsigned int v)
- {
- return stb_hashptr((void *)(size_t) v);
- }
- unsigned int stb_hash2(char *str, unsigned int *hash2_ptr)
- {
- unsigned int hash1 = 0x3141592c;
- unsigned int hash2 = 0x77f044ed;
- while (*str) {
- hash1 = (hash1 << 7) + (hash1 >> 25) + *str;
- hash2 = (hash2 << 11) + (hash2 >> 21) + *str;
- ++str;
- }
- *hash2_ptr = hash2 + (hash1 >> 16);
- return hash1 + (hash2 >> 16);
- }
- // Paul Hsieh hash
- #define stb__get16(p) ((p)[0] | ((p)[1] << 8))
- unsigned int stb_hash_fast(void *p, int len)
- {
- unsigned char *q = (unsigned char *) p;
- unsigned int hash = len;
- if (len <= 0 || q == NULL) return 0;
- /* Main loop */
- for (;len > 3; len -= 4) {
- unsigned int val;
- hash += stb__get16(q);
- val = (stb__get16(q+2) << 11);
- hash = (hash << 16) ^ hash ^ val;
- q += 4;
- hash += hash >> 11;
- }
- /* Handle end cases */
- switch (len) {
- case 3: hash += stb__get16(q);
- hash ^= hash << 16;
- hash ^= q[2] << 18;
- hash += hash >> 11;
- break;
- case 2: hash += stb__get16(q);
- hash ^= hash << 11;
- hash += hash >> 17;
- break;
- case 1: hash += q[0];
- hash ^= hash << 10;
- hash += hash >> 1;
- break;
- case 0: break;
- }
- /* Force "avalanching" of final 127 bits */
- hash ^= hash << 3;
- hash += hash >> 5;
- hash ^= hash << 4;
- hash += hash >> 17;
- hash ^= hash << 25;
- hash += hash >> 6;
- return hash;
- }
- unsigned int stb_hash_number(unsigned int hash)
- {
- hash ^= hash << 3;
- hash += hash >> 5;
- hash ^= hash << 4;
- hash += hash >> 17;
- hash ^= hash << 25;
- hash += hash >> 6;
- return hash;
- }
- #endif
- #ifdef STB_PERFECT_HASH
- //////////////////////////////////////////////////////////////////////////////
- //
- // Perfect hashing for ints/pointers
- //
- // This is mainly useful for making faster pointer-indexed tables
- // that don't change frequently. E.g. for stb_ischar().
- //
- typedef struct
- {
- stb_uint32 addend;
- stb_uint multiplicand;
- stb_uint b_mask;
- stb_uint8 small_bmap[16];
- stb_uint16 *large_bmap;
- stb_uint table_mask;
- stb_uint32 *table;
- } stb_perfect;
- STB_EXTERN int stb_perfect_create(stb_perfect *,unsigned int*,int n);
- STB_EXTERN void stb_perfect_destroy(stb_perfect *);
- STB_EXTERN int stb_perfect_hash(stb_perfect *, unsigned int x);
- extern int stb_perfect_hash_max_failures;
- #ifdef STB_DEFINE
- int stb_perfect_hash_max_failures;
- int stb_perfect_hash(stb_perfect *p, unsigned int x)
- {
- stb_uint m = x * p->multiplicand;
- stb_uint y = x >> 16;
- stb_uint bv = (m >> 24) + y;
- stb_uint av = (m + y) >> 12;
- if (p->table == NULL) return -1; // uninitialized table fails
- bv &= p->b_mask;
- av &= p->table_mask;
- if (p->large_bmap)
- av ^= p->large_bmap[bv];
- else
- av ^= p->small_bmap[bv];
- return p->table[av] == x ? av : -1;
- }
- static void stb__perfect_prehash(stb_perfect *p, stb_uint x, stb_uint16 *a, stb_uint16 *b)
- {
- stb_uint m = x * p->multiplicand;
- stb_uint y = x >> 16;
- stb_uint bv = (m >> 24) + y;
- stb_uint av = (m + y) >> 12;
- bv &= p->b_mask;
- av &= p->table_mask;
- *b = bv;
- *a = av;
- }
- static unsigned long stb__perfect_rand(void)
- {
- static unsigned long stb__rand;
- stb__rand = stb__rand * 2147001325 + 715136305;
- return 0x31415926 ^ ((stb__rand >> 16) + (stb__rand << 16));
- }
- typedef struct {
- unsigned short count;
- unsigned short b;
- unsigned short map;
- unsigned short *entries;
- } stb__slot;
- static int stb__slot_compare(const void *p, const void *q)
- {
- stb__slot *a = (stb__slot *) p;
- stb__slot *b = (stb__slot *) q;
- return a->count > b->count ? -1 : a->count < b->count; // sort large to small
- }
- int stb_perfect_create(stb_perfect *p, unsigned int *v, int n)
- {
- unsigned int buffer1[64], buffer2[64], buffer3[64], buffer4[64], buffer5[32];
- unsigned short *as = (unsigned short *) stb_temp(buffer1, sizeof(*v)*n);
- unsigned short *bs = (unsigned short *) stb_temp(buffer2, sizeof(*v)*n);
- unsigned short *entries = (unsigned short *) stb_temp(buffer4, sizeof(*entries) * n);
- int size = 1 << stb_log2_ceil(n), bsize=8;
- int failure = 0,i,j,k;
- assert(n <= 32768);
- p->large_bmap = NULL;
- for(;;) {
- stb__slot *bcount = (stb__slot *) stb_temp(buffer3, sizeof(*bcount) * bsize);
- unsigned short *bloc = (unsigned short *) stb_temp(buffer5, sizeof(*bloc) * bsize);
- unsigned short *e;
- int bad=0;
- p->addend = stb__perfect_rand();
- p->multiplicand = stb__perfect_rand() | 1;
- p->table_mask = size-1;
- p->b_mask = bsize-1;
- p->table = (stb_uint32 *) malloc(size * sizeof(*p->table));
- for (i=0; i < bsize; ++i) {
- bcount[i].b = i;
- bcount[i].count = 0;
- bcount[i].map = 0;
- }
- for (i=0; i < n; ++i) {
- stb__perfect_prehash(p, v[i], as+i, bs+i);
- ++bcount[bs[i]].count;
- }
- qsort(bcount, bsize, sizeof(*bcount), stb__slot_compare);
- e = entries; // now setup up their entries index
- for (i=0; i < bsize; ++i) {
- bcount[i].entries = e;
- e += bcount[i].count;
- bcount[i].count = 0;
- bloc[bcount[i].b] = i;
- }
- // now fill them out
- for (i=0; i < n; ++i) {
- int b = bs[i];
- int w = bloc[b];
- bcount[w].entries[bcount[w].count++] = i;
- }
- stb_tempfree(buffer5,bloc);
- // verify
- for (i=0; i < bsize; ++i)
- for (j=0; j < bcount[i].count; ++j)
- assert(bs[bcount[i].entries[j]] == bcount[i].b);
- memset(p->table, 0, size*sizeof(*p->table));
- // check if any b has duplicate a
- for (i=0; i < bsize; ++i) {
- if (bcount[i].count > 1) {
- for (j=0; j < bcount[i].count; ++j) {
- if (p->table[as[bcount[i].entries[j]]])
- bad = 1;
- p->table[as[bcount[i].entries[j]]] = 1;
- }
- for (j=0; j < bcount[i].count; ++j) {
- p->table[as[bcount[i].entries[j]]] = 0;
- }
- if (bad) break;
- }
- }
- if (!bad) {
- // go through the bs and populate the table, first fit
- for (i=0; i < bsize; ++i) {
- if (bcount[i].count) {
- // go through the candidate table[b] values
- for (j=0; j < size; ++j) {
- // go through the a values and see if they fit
- for (k=0; k < bcount[i].count; ++k) {
- int a = as[bcount[i].entries[k]];
- if (p->table[(a^j)&p->table_mask]) {
- break; // fails
- }
- }
- // if succeeded, accept
- if (k == bcount[i].count) {
- bcount[i].map = j;
- for (k=0; k < bcount[i].count; ++k) {
- int a = as[bcount[i].entries[k]];
- p->table[(a^j)&p->table_mask] = 1;
- }
- break;
- }
- }
- if (j == size)
- break; // no match for i'th entry, so break out in failure
- }
- }
- if (i == bsize) {
- // success... fill out map
- if (bsize <= 16 && size <= 256) {
- p->large_bmap = NULL;
- for (i=0; i < bsize; ++i)
- p->small_bmap[bcount[i].b] = (stb_uint8) bcount[i].map;
- } else {
- p->large_bmap = (unsigned short *) malloc(sizeof(*p->large_bmap) * bsize);
- for (i=0; i < bsize; ++i)
- p->large_bmap[bcount[i].b] = bcount[i].map;
- }
- // initialize table to v[0], so empty slots will fail
- for (i=0; i < size; ++i)
- p->table[i] = v[0];
- for (i=0; i < n; ++i)
- if (p->large_bmap)
- p->table[as[i] ^ p->large_bmap[bs[i]]] = v[i];
- else
- p->table[as[i] ^ p->small_bmap[bs[i]]] = v[i];
- // and now validate that none of them collided
- for (i=0; i < n; ++i)
- assert(stb_perfect_hash(p, v[i]) >= 0);
- stb_tempfree(buffer3, bcount);
- break;
- }
- }
- free(p->table);
- p->table = NULL;
- stb_tempfree(buffer3, bcount);
- ++failure;
- if (failure >= 4 && bsize < size) bsize *= 2;
- if (failure >= 8 && (failure & 3) == 0 && size < 4*n) {
- size *= 2;
- bsize *= 2;
- }
- if (failure == 6) {
- // make sure the input data is unique, so we don't infinite loop
- unsigned int *data = (unsigned int *) stb_temp(buffer3, n * sizeof(*data));
- memcpy(data, v, sizeof(*data) * n);
- qsort(data, n, sizeof(*data), stb_intcmp(0));
- for (i=1; i < n; ++i) {
- if (data[i] == data[i-1])
- size = 0; // size is return value, so 0 it
- }
- stb_tempfree(buffer3, data);
- if (!size) break;
- }
- }
- if (failure > stb_perfect_hash_max_failures)
- stb_perfect_hash_max_failures = failure;
- stb_tempfree(buffer1, as);
- stb_tempfree(buffer2, bs);
- stb_tempfree(buffer4, entries);
- return size;
- }
- void stb_perfect_destroy(stb_perfect *p)
- {
- if (p->large_bmap) free(p->large_bmap);
- if (p->table ) free(p->table);
- p->large_bmap = NULL;
- p->table = NULL;
- p->b_mask = 0;
- p->table_mask = 0;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Perfect hash clients
- STB_EXTERN int stb_ischar(char s, char *set);
- #ifdef STB_DEFINE
- int stb_ischar(char c, char *set)
- {
- static unsigned char bit[8] = { 1,2,4,8,16,32,64,128 };
- static stb_perfect p;
- static unsigned char (*tables)[256];
- static char ** sets = NULL;
- int z = stb_perfect_hash(&p, (int)(size_t) set);
- if (z < 0) {
- int i,k,n,j,f;
- // special code that means free all existing data
- if (set == NULL) {
- stb_arr_free(sets);
- free(tables);
- tables = NULL;
- stb_perfect_destroy(&p);
- return 0;
- }
- stb_arr_push(sets, set);
- stb_perfect_destroy(&p);
- n = stb_perfect_create(&p, (unsigned int *) (char **) sets, stb_arr_len(sets));
- assert(n != 0);
- k = (n+7) >> 3;
- tables = (unsigned char (*)[256]) realloc(tables, sizeof(*tables) * k);
- memset(tables, 0, sizeof(*tables) * k);
- for (i=0; i < stb_arr_len(sets); ++i) {
- k = stb_perfect_hash(&p, (int)(size_t) sets[i]);
- assert(k >= 0);
- n = k >> 3;
- f = bit[k&7];
- for (j=0; !j || sets[i][j]; ++j) {
- tables[n][(unsigned char) sets[i][j]] |= f;
- }
- }
- z = stb_perfect_hash(&p, (int)(size_t) set);
- }
- return tables[z >> 3][(unsigned char) c] & bit[z & 7];
- }
- #endif
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Instantiated data structures
- //
- // This is an attempt to implement a templated data structure.
- //
- // Hash table: call stb_define_hash(TYPE,N,KEY,K1,K2,HASH,VALUE)
- // TYPE -- will define a structure type containing the hash table
- // N -- the name, will prefix functions named:
- // N create
- // N destroy
- // N get
- // N set, N add, N update,
- // N remove
- // KEY -- the type of the key. 'x == y' must be valid
- // K1,K2 -- keys never used by the app, used as flags in the hashtable
- // HASH -- a piece of code ending with 'return' that hashes key 'k'
- // VALUE -- the type of the value. 'x = y' must be valid
- //
- // Note that stb_define_hash_base can be used to define more sophisticated
- // hash tables, e.g. those that make copies of the key or use special
- // comparisons (e.g. strcmp).
- #define STB_(prefix,name) stb__##prefix##name
- #define STB__(prefix,name) prefix##name
- #define STB__use(x) x
- #define STB__skip(x)
- #define stb_declare_hash(PREFIX,TYPE,N,KEY,VALUE) \
- typedef struct stb__st_##TYPE TYPE;\
- PREFIX int STB__(N, init)(TYPE *h, int count);\
- PREFIX int STB__(N, memory_usage)(TYPE *h);\
- PREFIX TYPE * STB__(N, create)(void);\
- PREFIX TYPE * STB__(N, copy)(TYPE *h);\
- PREFIX void STB__(N, destroy)(TYPE *h);\
- PREFIX int STB__(N,get_flag)(TYPE *a, KEY k, VALUE *v);\
- PREFIX VALUE STB__(N,get)(TYPE *a, KEY k);\
- PREFIX int STB__(N, set)(TYPE *a, KEY k, VALUE v);\
- PREFIX int STB__(N, add)(TYPE *a, KEY k, VALUE v);\
- PREFIX int STB__(N, update)(TYPE*a,KEY k,VALUE v);\
- PREFIX int STB__(N, remove)(TYPE *a, KEY k, VALUE *v);
- #define STB_nocopy(x) (x)
- #define STB_nodelete(x) 0
- #define STB_nofields
- #define STB_nonullvalue(x)
- #define STB_nullvalue(x) x
- #define STB_safecompare(x) x
- #define STB_nosafe(x)
- #define STB_noprefix
- #ifdef __GNUC__
- #define STB__nogcc(x)
- #else
- #define STB__nogcc(x) x
- #endif
- #define stb_define_hash_base(PREFIX,TYPE,FIELDS,N,NC,LOAD_FACTOR, \
- KEY,EMPTY,DEL,COPY,DISPOSE,SAFE, \
- VCOMPARE,CCOMPARE,HASH, \
- VALUE,HASVNULL,VNULL) \
- \
- typedef struct \
- { \
- KEY k; \
- VALUE v; \
- } STB_(N,_hashpair); \
- \
- STB__nogcc( typedef struct stb__st_##TYPE TYPE; ) \
- struct stb__st_##TYPE { \
- FIELDS \
- STB_(N,_hashpair) *table; \
- unsigned int mask; \
- int count, limit; \
- int deleted; \
- \
- int delete_threshhold; \
- int grow_threshhold; \
- int shrink_threshhold; \
- unsigned char alloced, has_empty, has_del; \
- VALUE ev; VALUE dv; \
- }; \
- \
- static unsigned int STB_(N, hash)(KEY k) \
- { \
- HASH \
- } \
- \
- PREFIX int STB__(N, init)(TYPE *h, int count) \
- { \
- int i; \
- if (count < 4) count = 4; \
- h->limit = count; \
- h->count = 0; \
- h->mask = count-1; \
- h->deleted = 0; \
- h->grow_threshhold = (int) (count * LOAD_FACTOR); \
- h->has_empty = h->has_del = 0; \
- h->alloced = 0; \
- if (count <= 64) \
- h->shrink_threshhold = 0; \
- else \
- h->shrink_threshhold = (int) (count * (LOAD_FACTOR/2.25)); \
- h->delete_threshhold = (int) (count * (1-LOAD_FACTOR)/2); \
- h->table = (STB_(N,_hashpair)*) malloc(sizeof(h->table[0]) * count); \
- if (h->table == NULL) return 0; \
- /* ideally this gets turned into a memset32 automatically */ \
- for (i=0; i < count; ++i) \
- h->table[i].k = EMPTY; \
- return 1; \
- } \
- \
- PREFIX int STB__(N, memory_usage)(TYPE *h) \
- { \
- return sizeof(*h) + h->limit * sizeof(h->table[0]); \
- } \
- \
- PREFIX TYPE * STB__(N, create)(void) \
- { \
- TYPE *h = (TYPE *) malloc(sizeof(*h)); \
- if (h) { \
- if (STB__(N, init)(h, 16)) \
- h->alloced = 1; \
- else { free(h); h=NULL; } \
- } \
- return h; \
- } \
- \
- PREFIX void STB__(N, destroy)(TYPE *a) \
- { \
- int i; \
- for (i=0; i < a->limit; ++i) \
- if (!CCOMPARE(a->table[i].k,EMPTY) && !CCOMPARE(a->table[i].k, DEL)) \
- DISPOSE(a->table[i].k); \
- free(a->table); \
- if (a->alloced) \
- free(a); \
- } \
- \
- static void STB_(N, rehash)(TYPE *a, int count); \
- \
- PREFIX int STB__(N,get_flag)(TYPE *a, KEY k, VALUE *v) \
- { \
- unsigned int h = STB_(N, hash)(k); \
- unsigned int n = h & a->mask, s; \
- if (CCOMPARE(k,EMPTY)){ if (a->has_empty) *v = a->ev; return a->has_empty;}\
- if (CCOMPARE(k,DEL)) { if (a->has_del ) *v = a->dv; return a->has_del; }\
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- SAFE(if (!CCOMPARE(a->table[n].k,DEL))) \
- if (VCOMPARE(a->table[n].k,k)) { *v = a->table[n].v; return 1; } \
- s = stb_rehash(h) | 1; \
- for(;;) { \
- n = (n + s) & a->mask; \
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- SAFE(if (CCOMPARE(a->table[n].k,DEL)) continue;) \
- if (VCOMPARE(a->table[n].k,k)) \
- { *v = a->table[n].v; return 1; } \
- } \
- } \
- \
- HASVNULL( \
- PREFIX VALUE STB__(N,get)(TYPE *a, KEY k) \
- { \
- VALUE v; \
- if (STB__(N,get_flag)(a,k,&v)) return v; \
- else return VNULL; \
- } \
- ) \
- \
- PREFIX int STB__(N,getkey)(TYPE *a, KEY k, KEY *kout) \
- { \
- unsigned int h = STB_(N, hash)(k); \
- unsigned int n = h & a->mask, s; \
- if (CCOMPARE(k,EMPTY)||CCOMPARE(k,DEL)) return 0; \
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- SAFE(if (!CCOMPARE(a->table[n].k,DEL))) \
- if (VCOMPARE(a->table[n].k,k)) { *kout = a->table[n].k; return 1; } \
- s = stb_rehash(h) | 1; \
- for(;;) { \
- n = (n + s) & a->mask; \
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- SAFE(if (CCOMPARE(a->table[n].k,DEL)) continue;) \
- if (VCOMPARE(a->table[n].k,k)) \
- { *kout = a->table[n].k; return 1; } \
- } \
- } \
- \
- static int STB_(N,addset)(TYPE *a, KEY k, VALUE v, \
- int allow_new, int allow_old, int copy) \
- { \
- unsigned int h = STB_(N, hash)(k); \
- unsigned int n = h & a->mask; \
- int b = -1; \
- if (CCOMPARE(k,EMPTY)) { \
- if (a->has_empty ? allow_old : allow_new) { \
- n=a->has_empty; a->ev = v; a->has_empty = 1; return !n; \
- } else return 0; \
- } \
- if (CCOMPARE(k,DEL)) { \
- if (a->has_del ? allow_old : allow_new) { \
- n=a->has_del; a->dv = v; a->has_del = 1; return !n; \
- } else return 0; \
- } \
- if (!CCOMPARE(a->table[n].k, EMPTY)) { \
- unsigned int s; \
- if (CCOMPARE(a->table[n].k, DEL)) \
- b = n; \
- else if (VCOMPARE(a->table[n].k,k)) { \
- if (allow_old) \
- a->table[n].v = v; \
- return !allow_new; \
- } \
- s = stb_rehash(h) | 1; \
- for(;;) { \
- n = (n + s) & a->mask; \
- if (CCOMPARE(a->table[n].k, EMPTY)) break; \
- if (CCOMPARE(a->table[n].k, DEL)) { \
- if (b < 0) b = n; \
- } else if (VCOMPARE(a->table[n].k,k)) { \
- if (allow_old) \
- a->table[n].v = v; \
- return !allow_new; \
- } \
- } \
- } \
- if (!allow_new) return 0; \
- if (b < 0) b = n; else --a->deleted; \
- a->table[b].k = copy ? COPY(k) : k; \
- a->table[b].v = v; \
- ++a->count; \
- if (a->count > a->grow_threshhold) \
- STB_(N,rehash)(a, a->limit*2); \
- return 1; \
- } \
- \
- PREFIX int STB__(N, set)(TYPE *a, KEY k, VALUE v){return STB_(N,addset)(a,k,v,1,1,1);}\
- PREFIX int STB__(N, add)(TYPE *a, KEY k, VALUE v){return STB_(N,addset)(a,k,v,1,0,1);}\
- PREFIX int STB__(N, update)(TYPE*a,KEY k,VALUE v){return STB_(N,addset)(a,k,v,0,1,1);}\
- \
- PREFIX int STB__(N, remove)(TYPE *a, KEY k, VALUE *v) \
- { \
- unsigned int h = STB_(N, hash)(k); \
- unsigned int n = h & a->mask, s; \
- if (CCOMPARE(k,EMPTY)) { if (a->has_empty) { if(v)*v = a->ev; a->has_empty=0; return 1; } return 0; } \
- if (CCOMPARE(k,DEL)) { if (a->has_del ) { if(v)*v = a->dv; a->has_del =0; return 1; } return 0; } \
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- if (SAFE(CCOMPARE(a->table[n].k,DEL) || ) !VCOMPARE(a->table[n].k,k)) { \
- s = stb_rehash(h) | 1; \
- for(;;) { \
- n = (n + s) & a->mask; \
- if (CCOMPARE(a->table[n].k,EMPTY)) return 0; \
- SAFE(if (CCOMPARE(a->table[n].k, DEL)) continue;) \
- if (VCOMPARE(a->table[n].k,k)) break; \
- } \
- } \
- DISPOSE(a->table[n].k); \
- a->table[n].k = DEL; \
- --a->count; \
- ++a->deleted; \
- if (v != NULL) \
- *v = a->table[n].v; \
- if (a->count < a->shrink_threshhold) \
- STB_(N, rehash)(a, a->limit >> 1); \
- else if (a->deleted > a->delete_threshhold) \
- STB_(N, rehash)(a, a->limit); \
- return 1; \
- } \
- \
- PREFIX TYPE * STB__(NC, copy)(TYPE *a) \
- { \
- int i; \
- TYPE *h = (TYPE *) malloc(sizeof(*h)); \
- if (!h) return NULL; \
- if (!STB__(N, init)(h, a->limit)) { free(h); return NULL; } \
- h->count = a->count; \
- h->deleted = a->deleted; \
- h->alloced = 1; \
- h->ev = a->ev; h->dv = a->dv; \
- h->has_empty = a->has_empty; h->has_del = a->has_del; \
- memcpy(h->table, a->table, h->limit * sizeof(h->table[0])); \
- for (i=0; i < a->limit; ++i) \
- if (!CCOMPARE(h->table[i].k,EMPTY) && !CCOMPARE(h->table[i].k,DEL)) \
- h->table[i].k = COPY(h->table[i].k); \
- return h; \
- } \
- \
- static void STB_(N, rehash)(TYPE *a, int count) \
- { \
- int i; \
- TYPE b; \
- STB__(N, init)(&b, count); \
- for (i=0; i < a->limit; ++i) \
- if (!CCOMPARE(a->table[i].k,EMPTY) && !CCOMPARE(a->table[i].k,DEL)) \
- STB_(N,addset)(&b, a->table[i].k, a->table[i].v,1,1,0); \
- free(a->table); \
- a->table = b.table; \
- a->mask = b.mask; \
- a->count = b.count; \
- a->limit = b.limit; \
- a->deleted = b.deleted; \
- a->delete_threshhold = b.delete_threshhold; \
- a->grow_threshhold = b.grow_threshhold; \
- a->shrink_threshhold = b.shrink_threshhold; \
- }
- #define STB_equal(a,b) ((a) == (b))
- #define stb_define_hash(TYPE,N,KEY,EMPTY,DEL,HASH,VALUE) \
- stb_define_hash_base(STB_noprefix, TYPE,STB_nofields,N,NC,0.85f, \
- KEY,EMPTY,DEL,STB_nocopy,STB_nodelete,STB_nosafe, \
- STB_equal,STB_equal,HASH, \
- VALUE,STB_nonullvalue,0)
- #define stb_define_hash_vnull(TYPE,N,KEY,EMPTY,DEL,HASH,VALUE,VNULL) \
- stb_define_hash_base(STB_noprefix, TYPE,STB_nofields,N,NC,0.85f, \
- KEY,EMPTY,DEL,STB_nocopy,STB_nodelete,STB_nosafe, \
- STB_equal,STB_equal,HASH, \
- VALUE,STB_nullvalue,VNULL)
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_ptrmap
- //
- // An stb_ptrmap data structure is an O(1) hash table between pointers. One
- // application is to let you store "extra" data associated with pointers,
- // which is why it was originally called stb_extra.
- stb_declare_hash(STB_EXTERN, stb_ptrmap, stb_ptrmap_, void *, void *)
- stb_declare_hash(STB_EXTERN, stb_idict, stb_idict_, stb_int32, stb_int32)
- stb_declare_hash(STB_EXTERN, stb_uidict, stbi_uidict_, stb_uint32, stb_uint32)
- STB_EXTERN void stb_ptrmap_delete(stb_ptrmap *e, void (*free_func)(void *));
- STB_EXTERN stb_ptrmap *stb_ptrmap_new(void);
- STB_EXTERN stb_idict * stb_idict_new_size(int size);
- STB_EXTERN void stb_idict_remove_all(stb_idict *e);
- STB_EXTERN void stb_uidict_reset(stb_uidict *e);
- #ifdef STB_DEFINE
- #define STB_EMPTY ((void *) 2)
- #define STB_EDEL ((void *) 6)
- stb_define_hash_base(STB_noprefix,stb_ptrmap, STB_nofields, stb_ptrmap_,stb_ptrmap_,0.85f,
- void *,STB_EMPTY,STB_EDEL,STB_nocopy,STB_nodelete,STB_nosafe,
- STB_equal,STB_equal,return stb_hashptr(k);,
- void *,STB_nullvalue,NULL)
- stb_ptrmap *stb_ptrmap_new(void)
- {
- return stb_ptrmap_create();
- }
- void stb_ptrmap_delete(stb_ptrmap *e, void (*free_func)(void *))
- {
- int i;
- if (free_func)
- for (i=0; i < e->limit; ++i)
- if (e->table[i].k != STB_EMPTY && e->table[i].k != STB_EDEL) {
- if (free_func == free)
- free(e->table[i].v); // allow STB_MALLOC_WRAPPER to operate
- else
- free_func(e->table[i].v);
- }
- stb_ptrmap_destroy(e);
- }
- // extra fields needed for stua_dict
- #define STB_IEMPTY ((int) 1)
- #define STB_IDEL ((int) 3)
- stb_define_hash_base(STB_noprefix, stb_idict, short type; short gc; STB_nofields, stb_idict_,stb_idict_,0.95f,
- stb_int32,STB_IEMPTY,STB_IDEL,STB_nocopy,STB_nodelete,STB_nosafe,
- STB_equal,STB_equal,
- return stb_rehash_improved(k);,stb_int32,STB_nonullvalue,0)
- stb_idict * stb_idict_new_size(int size)
- {
- stb_idict *e = (stb_idict *) malloc(sizeof(*e));
- if (e) {
- if (!stb_is_pow2(size))
- size = 1 << stb_log2_ceil(size);
- stb_idict_init(e, size);
- e->alloced = 1;
- }
- return e;
- }
- void stb_idict_remove_all(stb_idict *e)
- {
- int n;
- for (n=0; n < e->limit; ++n)
- e->table[n].k = STB_IEMPTY;
- e->has_empty = e->has_del = 0;
- e->count = 0;
- e->deleted = 0;
- }
- stb_define_hash_base(STB_noprefix, stb_uidict, STB_nofields, stb_uidict_,stb_uidict_,0.85f,
- stb_int32,0xffffffff,0xfffffffe,STB_nocopy,STB_nodelete,STB_nosafe,
- STB_equal,STB_equal,
- return stb_rehash_improved(k);,stb_uint32,STB_nonullvalue,0)
- void stb_uidict_reset(stb_uidict *e)
- {
- int n;
- for (n=0; n < e->limit; ++n)
- e->table[n].k = 0xffffffff;
- e->has_empty = e->has_del = 0;
- e->count = 0;
- e->deleted = 0;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_sparse_ptr_matrix
- //
- // An stb_ptrmap data structure is an O(1) hash table storing an arbitrary
- // block of data for a given pair of pointers.
- //
- // If create=0, returns
- typedef struct stb__st_stb_spmatrix stb_spmatrix;
- STB_EXTERN stb_spmatrix * stb_sparse_ptr_matrix_new(int val_size);
- STB_EXTERN void stb_sparse_ptr_matrix_free(stb_spmatrix *z);
- STB_EXTERN void * stb_sparse_ptr_matrix_get(stb_spmatrix *z, void *a, void *b, int create);
- #ifdef STB_DEFINE
- typedef struct
- {
- void *a;
- void *b;
- } stb__ptrpair;
- static stb__ptrpair stb__ptrpair_empty = { (void *) 1, (void *) 1 };
- static stb__ptrpair stb__ptrpair_del = { (void *) 2, (void *) 2 };
- #define STB__equal_ptrpair(x,y) ((x).a == (y).a && (x).b == (y).b)
- stb_define_hash_base(STB_noprefix, stb_spmatrix, int val_size; void *arena;, stb__spmatrix_,stb__spmatrix_, 0.85,
- stb__ptrpair, stb__ptrpair_empty, stb__ptrpair_del,
- STB_nocopy, STB_nodelete, STB_nosafe,
- STB__equal_ptrpair, STB__equal_ptrpair, return stb_rehash(stb_hashptr(k.a))+stb_hashptr(k.b);,
- void *, STB_nullvalue, 0)
- stb_spmatrix *stb_sparse_ptr_matrix_new(int val_size)
- {
- stb_spmatrix *m = stb__spmatrix_create();
- if (m) m->val_size = val_size;
- if (m) m->arena = stb_malloc_global(1);
- return m;
- }
- void stb_sparse_ptr_matrix_free(stb_spmatrix *z)
- {
- if (z->arena) stb_free(z->arena);
- stb__spmatrix_destroy(z);
- }
- void *stb_sparse_ptr_matrix_get(stb_spmatrix *z, void *a, void *b, int create)
- {
- stb__ptrpair t = { a,b };
- void *data = stb__spmatrix_get(z, t);
- if (!data && create) {
- data = stb_malloc_raw(z->arena, z->val_size);
- if (!data) return NULL;
- memset(data, 0, z->val_size);
- stb__spmatrix_add(z, t, data);
- }
- return data;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // SDICT: Hash Table for Strings (symbol table)
- //
- // if "use_arena=1", then strings will be copied
- // into blocks and never freed until the sdict is freed;
- // otherwise they're malloc()ed and free()d on the fly.
- // (specify use_arena=1 if you never stb_sdict_remove)
- stb_declare_hash(STB_EXTERN, stb_sdict, stb_sdict_, char *, void *)
- STB_EXTERN stb_sdict * stb_sdict_new(int use_arena);
- STB_EXTERN stb_sdict * stb_sdict_copy(stb_sdict*);
- STB_EXTERN void stb_sdict_delete(stb_sdict *);
- STB_EXTERN void * stb_sdict_change(stb_sdict *, char *str, void *p);
- STB_EXTERN int stb_sdict_count(stb_sdict *d);
- STB_EXTERN int stb_sdict_internal_limit(stb_sdict *d);
- STB_EXTERN char * stb_sdict_internal_key(stb_sdict *d, int n);
- STB_EXTERN void * stb_sdict_internal_value(stb_sdict *d, int n);
- #define stb_sdict_for(d,i,q,z) \
- for(i=0; i < stb_sdict_internal_limit(d) ? (q=stb_sdict_internal_key(d,i),z=stb_sdict_internal_value(d,i),1) : 0; ++i) \
- if (q==NULL||q==(void *) 1);else // reversed makes macro friendly
- #ifdef STB_DEFINE
- // if in same translation unit, for speed, don't call accessors
- #undef stb_sdict_for
- #define stb_sdict_for(d,i,q,z) \
- for(i=0; i < (d)->limit ? (q=(d)->table[i].k,z=(d)->table[i].v,1) : 0; ++i) \
- if (q==NULL||q==(void *) 1);else // reversed makes macro friendly
- #define STB_DEL ((void *) 1)
- #define STB_SDEL ((char *) 1)
- #define stb_sdict__copy(x) \
- stb_p_strcpy_s(a->arena ? stb_malloc_string(a->arena, strlen(x)+1) \
- : (char *) malloc(strlen(x)+1), strlen(x)+1, x)
- #define stb_sdict__dispose(x) if (!a->arena) free(x)
- stb_define_hash_base(STB_noprefix, stb_sdict, void*arena;, stb_sdict_,stb_sdictinternal_, 0.85f,
- char *, NULL, STB_SDEL, stb_sdict__copy, stb_sdict__dispose,
- STB_safecompare, !strcmp, STB_equal, return stb_hash(k);,
- void *, STB_nullvalue, NULL)
- int stb_sdict_count(stb_sdict *a)
- {
- return a->count;
- }
- int stb_sdict_internal_limit(stb_sdict *a)
- {
- return a->limit;
- }
- char* stb_sdict_internal_key(stb_sdict *a, int n)
- {
- return a->table[n].k;
- }
- void* stb_sdict_internal_value(stb_sdict *a, int n)
- {
- return a->table[n].v;
- }
- stb_sdict * stb_sdict_new(int use_arena)
- {
- stb_sdict *d = stb_sdict_create();
- if (d == NULL) return NULL;
- d->arena = use_arena ? stb_malloc_global(1) : NULL;
- return d;
- }
- stb_sdict* stb_sdict_copy(stb_sdict *old)
- {
- stb_sdict *n;
- void *old_arena = old->arena;
- void *new_arena = old_arena ? stb_malloc_global(1) : NULL;
- old->arena = new_arena;
- n = stb_sdictinternal_copy(old);
- old->arena = old_arena;
- if (n)
- n->arena = new_arena;
- else if (new_arena)
- stb_free(new_arena);
- return n;
- }
- void stb_sdict_delete(stb_sdict *d)
- {
- if (d->arena)
- stb_free(d->arena);
- stb_sdict_destroy(d);
- }
- void * stb_sdict_change(stb_sdict *d, char *str, void *p)
- {
- void *q = stb_sdict_get(d, str);
- stb_sdict_set(d, str, p);
- return q;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Instantiated data structures
- //
- // This is an attempt to implement a templated data structure.
- // What you do is define a struct foo, and then include several
- // pointer fields to struct foo in your struct. Then you call
- // the instantiator, which creates the functions that implement
- // the data structure. This requires massive undebuggable #defines,
- // so we limit the cases where we do this.
- //
- // AA tree is an encoding of a 2-3 tree whereas RB trees encode a 2-3-4 tree;
- // much simpler code due to fewer cases.
- #define stb__bst_parent(x) x
- #define stb__bst_noparent(x)
- #define stb_bst_fields(N) \
- *STB_(N,left), *STB_(N,right); \
- unsigned char STB_(N,level)
- #define stb_bst_fields_parent(N) \
- *STB_(N,left), *STB_(N,right), *STB_(N,parent); \
- unsigned char STB_(N,level)
- #define STB__level(N,x) ((x) ? (x)->STB_(N,level) : 0)
- #define stb_bst_base(TYPE, N, TREE, M, compare, PAR) \
- \
- static int STB_(N,_compare)(TYPE *p, TYPE *q) \
- { \
- compare \
- } \
- \
- static void STB_(N,setleft)(TYPE *q, TYPE *v) \
- { \
- q->STB_(N,left) = v; \
- PAR(if (v) v->STB_(N,parent) = q;) \
- } \
- \
- static void STB_(N,setright)(TYPE *q, TYPE *v) \
- { \
- q->STB_(N,right) = v; \
- PAR(if (v) v->STB_(N,parent) = q;) \
- } \
- \
- static TYPE *STB_(N,skew)(TYPE *q) \
- { \
- if (q == NULL) return q; \
- if (q->STB_(N,left) \
- && q->STB_(N,left)->STB_(N,level) == q->STB_(N,level)) { \
- TYPE *p = q->STB_(N,left); \
- STB_(N,setleft)(q, p->STB_(N,right)); \
- STB_(N,setright)(p, q); \
- return p; \
- } \
- return q; \
- } \
- \
- static TYPE *STB_(N,split)(TYPE *p) \
- { \
- TYPE *q = p->STB_(N,right); \
- if (q && q->STB_(N,right) \
- && q->STB_(N,right)->STB_(N,level) == p->STB_(N,level)) { \
- STB_(N,setright)(p, q->STB_(N,left)); \
- STB_(N,setleft)(q,p); \
- ++q->STB_(N,level); \
- return q; \
- } \
- return p; \
- } \
- \
- TYPE *STB__(N,insert)(TYPE *tree, TYPE *item) \
- { \
- int c; \
- if (tree == NULL) { \
- item->STB_(N,left) = NULL; \
- item->STB_(N,right) = NULL; \
- item->STB_(N,level) = 1; \
- PAR(item->STB_(N,parent) = NULL;) \
- return item; \
- } \
- c = STB_(N,_compare)(item,tree); \
- if (c == 0) { \
- if (item != tree) { \
- STB_(N,setleft)(item, tree->STB_(N,left)); \
- STB_(N,setright)(item, tree->STB_(N,right)); \
- item->STB_(N,level) = tree->STB_(N,level); \
- PAR(item->STB_(N,parent) = NULL;) \
- } \
- return item; \
- } \
- if (c < 0) \
- STB_(N,setleft )(tree, STB__(N,insert)(tree->STB_(N,left), item)); \
- else \
- STB_(N,setright)(tree, STB__(N,insert)(tree->STB_(N,right), item)); \
- tree = STB_(N,skew)(tree); \
- tree = STB_(N,split)(tree); \
- PAR(tree->STB_(N,parent) = NULL;) \
- return tree; \
- } \
- \
- TYPE *STB__(N,remove)(TYPE *tree, TYPE *item) \
- { \
- static TYPE *delnode, *leaf, *restore; \
- if (tree == NULL) return NULL; \
- leaf = tree; \
- if (STB_(N,_compare)(item, tree) < 0) { \
- STB_(N,setleft)(tree, STB__(N,remove)(tree->STB_(N,left), item)); \
- } else { \
- TYPE *r; \
- delnode = tree; \
- r = STB__(N,remove)(tree->STB_(N,right), item); \
- /* maybe move 'leaf' up to this location */ \
- if (restore == tree) { tree = leaf; leaf = restore = NULL; } \
- STB_(N,setright)(tree,r); \
- assert(tree->STB_(N,right) != tree); \
- } \
- if (tree == leaf) { \
- if (delnode == item) { \
- tree = tree->STB_(N,right); \
- assert(leaf->STB_(N,left) == NULL); \
- /* move leaf (the right sibling) up to delnode */ \
- STB_(N,setleft )(leaf, item->STB_(N,left )); \
- STB_(N,setright)(leaf, item->STB_(N,right)); \
- leaf->STB_(N,level) = item->STB_(N,level); \
- if (leaf != item) \
- restore = delnode; \
- } \
- delnode = NULL; \
- } else { \
- if (STB__level(N,tree->STB_(N,left) ) < tree->STB_(N,level)-1 || \
- STB__level(N,tree->STB_(N,right)) < tree->STB_(N,level)-1) { \
- --tree->STB_(N,level); \
- if (STB__level(N,tree->STB_(N,right)) > tree->STB_(N,level)) \
- tree->STB_(N,right)->STB_(N,level) = tree->STB_(N,level); \
- tree = STB_(N,skew)(tree); \
- STB_(N,setright)(tree, STB_(N,skew)(tree->STB_(N,right))); \
- if (tree->STB_(N,right)) \
- STB_(N,setright)(tree->STB_(N,right), \
- STB_(N,skew)(tree->STB_(N,right)->STB_(N,right))); \
- tree = STB_(N,split)(tree); \
- if (tree->STB_(N,right)) \
- STB_(N,setright)(tree, STB_(N,split)(tree->STB_(N,right))); \
- } \
- } \
- PAR(if (tree) tree->STB_(N,parent) = NULL;) \
- return tree; \
- } \
- \
- TYPE *STB__(N,last)(TYPE *tree) \
- { \
- if (tree) \
- while (tree->STB_(N,right)) tree = tree->STB_(N,right); \
- return tree; \
- } \
- \
- TYPE *STB__(N,first)(TYPE *tree) \
- { \
- if (tree) \
- while (tree->STB_(N,left)) tree = tree->STB_(N,left); \
- return tree; \
- } \
- \
- TYPE *STB__(N,next)(TYPE *tree, TYPE *item) \
- { \
- TYPE *next = NULL; \
- if (item->STB_(N,right)) \
- return STB__(N,first)(item->STB_(N,right)); \
- PAR( \
- while(item->STB_(N,parent)) { \
- TYPE *up = item->STB_(N,parent); \
- if (up->STB_(N,left) == item) return up; \
- item = up; \
- } \
- return NULL; \
- ) \
- while (tree != item) { \
- if (STB_(N,_compare)(item, tree) < 0) { \
- next = tree; \
- tree = tree->STB_(N,left); \
- } else { \
- tree = tree->STB_(N,right); \
- } \
- } \
- return next; \
- } \
- \
- TYPE *STB__(N,prev)(TYPE *tree, TYPE *item) \
- { \
- TYPE *next = NULL; \
- if (item->STB_(N,left)) \
- return STB__(N,last)(item->STB_(N,left)); \
- PAR( \
- while(item->STB_(N,parent)) { \
- TYPE *up = item->STB_(N,parent); \
- if (up->STB_(N,right) == item) return up; \
- item = up; \
- } \
- return NULL; \
- ) \
- while (tree != item) { \
- if (STB_(N,_compare)(item, tree) < 0) { \
- tree = tree->STB_(N,left); \
- } else { \
- next = tree; \
- tree = tree->STB_(N,right); \
- } \
- } \
- return next; \
- } \
- \
- STB__DEBUG( \
- void STB__(N,_validate)(TYPE *tree, int root) \
- { \
- if (tree == NULL) return; \
- PAR(if(root) assert(tree->STB_(N,parent) == NULL);) \
- assert(STB__level(N,tree->STB_(N,left) ) == tree->STB_(N,level)-1); \
- assert(STB__level(N,tree->STB_(N,right)) <= tree->STB_(N,level)); \
- assert(STB__level(N,tree->STB_(N,right)) >= tree->STB_(N,level)-1); \
- if (tree->STB_(N,right)) { \
- assert(STB__level(N,tree->STB_(N,right)->STB_(N,right)) \
- != tree->STB_(N,level)); \
- PAR(assert(tree->STB_(N,right)->STB_(N,parent) == tree);) \
- } \
- PAR(if(tree->STB_(N,left)) assert(tree->STB_(N,left)->STB_(N,parent) == tree);) \
- STB__(N,_validate)(tree->STB_(N,left) ,0); \
- STB__(N,_validate)(tree->STB_(N,right),0); \
- } \
- ) \
- \
- typedef struct \
- { \
- TYPE *root; \
- } TREE; \
- \
- void STB__(M,Insert)(TREE *tree, TYPE *item) \
- { tree->root = STB__(N,insert)(tree->root, item); } \
- void STB__(M,Remove)(TREE *tree, TYPE *item) \
- { tree->root = STB__(N,remove)(tree->root, item); } \
- TYPE *STB__(M,Next)(TREE *tree, TYPE *item) \
- { return STB__(N,next)(tree->root, item); } \
- TYPE *STB__(M,Prev)(TREE *tree, TYPE *item) \
- { return STB__(N,prev)(tree->root, item); } \
- TYPE *STB__(M,First)(TREE *tree) { return STB__(N,first)(tree->root); } \
- TYPE *STB__(M,Last) (TREE *tree) { return STB__(N,last) (tree->root); } \
- void STB__(M,Init)(TREE *tree) { tree->root = NULL; }
- #define stb_bst_find(N,tree,fcompare) \
- { \
- int c; \
- while (tree != NULL) { \
- fcompare \
- if (c == 0) return tree; \
- if (c < 0) tree = tree->STB_(N,left); \
- else tree = tree->STB_(N,right); \
- } \
- return NULL; \
- }
- #define stb_bst_raw(TYPE,N,TREE,M,vfield,VTYPE,compare,PAR) \
- stb_bst_base(TYPE,N,TREE,M, \
- VTYPE a = p->vfield; VTYPE b = q->vfield; return (compare);, PAR ) \
- \
- TYPE *STB__(N,find)(TYPE *tree, VTYPE a) \
- stb_bst_find(N,tree,VTYPE b = tree->vfield; c = (compare);) \
- TYPE *STB__(M,Find)(TREE *tree, VTYPE a) \
- { return STB__(N,find)(tree->root, a); }
- #define stb_bst(TYPE,N,TREE,M,vfield,VTYPE,compare) \
- stb_bst_raw(TYPE,N,TREE,M,vfield,VTYPE,compare,stb__bst_noparent)
- #define stb_bst_parent(TYPE,N,TREE,M,vfield,VTYPE,compare) \
- stb_bst_raw(TYPE,N,TREE,M,vfield,VTYPE,compare,stb__bst_parent)
- //////////////////////////////////////////////////////////////////////////////
- //
- // Pointer Nulling
- //
- // This lets you automatically NULL dangling pointers to "registered"
- // objects. Note that you have to make sure you call the appropriate
- // functions when you free or realloc blocks of memory that contain
- // pointers or pointer targets. stb.h can automatically do this for
- // stb_arr, or for all frees/reallocs if it's wrapping them.
- //
- #ifdef STB_NPTR
- STB_EXTERN void stb_nptr_set(void *address_of_pointer, void *value_to_write);
- STB_EXTERN void stb_nptr_didset(void *address_of_pointer);
- STB_EXTERN void stb_nptr_didfree(void *address_being_freed, int len);
- STB_EXTERN void stb_nptr_free(void *address_being_freed, int len);
- STB_EXTERN void stb_nptr_didrealloc(void *new_address, void *old_address, int len);
- STB_EXTERN void stb_nptr_recache(void); // recache all known pointers
- // do this after pointer sets outside your control, slow
- #ifdef STB_DEFINE
- // for fast updating on free/realloc, we need to be able to find
- // all the objects (pointers and targets) within a given block;
- // this precludes hashing
- // we use a three-level hierarchy of memory to minimize storage:
- // level 1: 65536 pointers to stb__memory_node (always uses 256 KB)
- // level 2: each stb__memory_node represents a 64K block of memory
- // with 256 stb__memory_leafs (worst case 64MB)
- // level 3: each stb__memory_leaf represents 256 bytes of memory
- // using a list of target locations and a list of pointers
- // (which are hopefully fairly short normally!)
- // this approach won't work in 64-bit, which has a much larger address
- // space. need to redesign
- #define STB__NPTR_ROOT_LOG2 16
- #define STB__NPTR_ROOT_NUM (1 << STB__NPTR_ROOT_LOG2)
- #define STB__NPTR_ROOT_SHIFT (32 - STB__NPTR_ROOT_LOG2)
- #define STB__NPTR_NODE_LOG2 5
- #define STB__NPTR_NODE_NUM (1 << STB__NPTR_NODE_LOG2)
- #define STB__NPTR_NODE_MASK (STB__NPTR_NODE_NUM-1)
- #define STB__NPTR_NODE_SHIFT (STB__NPTR_ROOT_SHIFT - STB__NPTR_NODE_LOG2)
- #define STB__NPTR_NODE_OFFSET(x) (((x) >> STB__NPTR_NODE_SHIFT) & STB__NPTR_NODE_MASK)
- typedef struct stb__st_nptr
- {
- void *ptr; // address of actual pointer
- struct stb__st_nptr *next; // next pointer with same target
- struct stb__st_nptr **prev; // prev pointer with same target, address of 'next' field (or first)
- struct stb__st_nptr *next_in_block;
- } stb__nptr;
- typedef struct stb__st_nptr_target
- {
- void *ptr; // address of target
- stb__nptr *first; // address of first nptr pointing to this
- struct stb__st_nptr_target *next_in_block;
- } stb__nptr_target;
- typedef struct
- {
- stb__nptr *pointers;
- stb__nptr_target *targets;
- } stb__memory_leaf;
- typedef struct
- {
- stb__memory_leaf *children[STB__NPTR_NODE_NUM];
- } stb__memory_node;
- stb__memory_node *stb__memtab_root[STB__NPTR_ROOT_NUM];
- static stb__memory_leaf *stb__nptr_find_leaf(void *mem)
- {
- stb_uint32 address = (stb_uint32) mem;
- stb__memory_node *z = stb__memtab_root[address >> STB__NPTR_ROOT_SHIFT];
- if (z)
- return z->children[STB__NPTR_NODE_OFFSET(address)];
- else
- return NULL;
- }
- static void * stb__nptr_alloc(int size)
- {
- return stb__realloc_raw(0,size);
- }
- static void stb__nptr_free(void *p)
- {
- stb__realloc_raw(p,0);
- }
- static stb__memory_leaf *stb__nptr_make_leaf(void *mem)
- {
- stb_uint32 address = (stb_uint32) mem;
- stb__memory_node *z = stb__memtab_root[address >> STB__NPTR_ROOT_SHIFT];
- stb__memory_leaf *f;
- if (!z) {
- int i;
- z = (stb__memory_node *) stb__nptr_alloc(sizeof(*stb__memtab_root[0]));
- stb__memtab_root[address >> STB__NPTR_ROOT_SHIFT] = z;
- for (i=0; i < 256; ++i)
- z->children[i] = 0;
- }
- f = (stb__memory_leaf *) stb__nptr_alloc(sizeof(*f));
- z->children[STB__NPTR_NODE_OFFSET(address)] = f;
- f->pointers = NULL;
- f->targets = NULL;
- return f;
- }
- static stb__nptr_target *stb__nptr_find_target(void *target, int force)
- {
- stb__memory_leaf *p = stb__nptr_find_leaf(target);
- if (p) {
- stb__nptr_target *t = p->targets;
- while (t) {
- if (t->ptr == target)
- return t;
- t = t->next_in_block;
- }
- }
- if (force) {
- stb__nptr_target *t = (stb__nptr_target*) stb__nptr_alloc(sizeof(*t));
- if (!p) p = stb__nptr_make_leaf(target);
- t->ptr = target;
- t->first = NULL;
- t->next_in_block = p->targets;
- p->targets = t;
- return t;
- } else
- return NULL;
- }
- static stb__nptr *stb__nptr_find_pointer(void *ptr, int force)
- {
- stb__memory_leaf *p = stb__nptr_find_leaf(ptr);
- if (p) {
- stb__nptr *t = p->pointers;
- while (t) {
- if (t->ptr == ptr)
- return t;
- t = t->next_in_block;
- }
- }
- if (force) {
- stb__nptr *t = (stb__nptr *) stb__nptr_alloc(sizeof(*t));
- if (!p) p = stb__nptr_make_leaf(ptr);
- t->ptr = ptr;
- t->next = NULL;
- t->prev = NULL;
- t->next_in_block = p->pointers;
- p->pointers = t;
- return t;
- } else
- return NULL;
- }
- void stb_nptr_set(void *address_of_pointer, void *value_to_write)
- {
- if (*(void **)address_of_pointer != value_to_write) {
- *(void **) address_of_pointer = value_to_write;
- stb_nptr_didset(address_of_pointer);
- }
- }
- void stb_nptr_didset(void *address_of_pointer)
- {
- // first unlink from old chain
- void *new_address;
- stb__nptr *p = stb__nptr_find_pointer(address_of_pointer, 1); // force building if doesn't exist
- if (p->prev) { // if p->prev is NULL, we just built it, or it was NULL
- *(p->prev) = p->next;
- if (p->next) p->next->prev = p->prev;
- }
- // now add to new chain
- new_address = *(void **)address_of_pointer;
- if (new_address != NULL) {
- stb__nptr_target *t = stb__nptr_find_target(new_address, 1);
- p->next = t->first;
- if (p->next) p->next->prev = &p->next;
- p->prev = &t->first;
- t->first = p;
- } else {
- p->prev = NULL;
- p->next = NULL;
- }
- }
- void stb__nptr_block(void *address, int len, void (*function)(stb__memory_leaf *f, int datum, void *start, void *end), int datum)
- {
- void *end_address = (void *) ((char *) address + len - 1);
- stb__memory_node *n;
- stb_uint32 start = (stb_uint32) address;
- stb_uint32 end = start + len - 1;
- int b0 = start >> STB__NPTR_ROOT_SHIFT;
- int b1 = end >> STB__NPTR_ROOT_SHIFT;
- int b=b0,i,e0,e1;
- e0 = STB__NPTR_NODE_OFFSET(start);
- if (datum <= 0) {
- // first block
- n = stb__memtab_root[b0];
- if (n) {
- if (b0 != b1)
- e1 = STB__NPTR_NODE_NUM-1;
- else
- e1 = STB__NPTR_NODE_OFFSET(end);
- for (i=e0; i <= e1; ++i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- if (b1 > b0) {
- // blocks other than the first and last block
- for (b=b0+1; b < b1; ++b) {
- n = stb__memtab_root[b];
- if (n)
- for (i=0; i <= STB__NPTR_NODE_NUM-1; ++i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- // last block
- n = stb__memtab_root[b1];
- if (n) {
- e1 = STB__NPTR_NODE_OFFSET(end);
- for (i=0; i <= e1; ++i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- }
- } else {
- if (b1 > b0) {
- // last block
- n = stb__memtab_root[b1];
- if (n) {
- e1 = STB__NPTR_NODE_OFFSET(end);
- for (i=e1; i >= 0; --i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- // blocks other than the first and last block
- for (b=b1-1; b > b0; --b) {
- n = stb__memtab_root[b];
- if (n)
- for (i=STB__NPTR_NODE_NUM-1; i >= 0; --i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- }
- // first block
- n = stb__memtab_root[b0];
- if (n) {
- if (b0 != b1)
- e1 = STB__NPTR_NODE_NUM-1;
- else
- e1 = STB__NPTR_NODE_OFFSET(end);
- for (i=e1; i >= e0; --i)
- if (n->children[i])
- function(n->children[i], datum, address, end_address);
- }
- }
- }
- static void stb__nptr_delete_pointers(stb__memory_leaf *f, int offset, void *start, void *end)
- {
- stb__nptr **p = &f->pointers;
- while (*p) {
- stb__nptr *n = *p;
- if (n->ptr >= start && n->ptr <= end) {
- // unlink
- if (n->prev) {
- *(n->prev) = n->next;
- if (n->next) n->next->prev = n->prev;
- }
- *p = n->next_in_block;
- stb__nptr_free(n);
- } else
- p = &(n->next_in_block);
- }
- }
- static void stb__nptr_delete_targets(stb__memory_leaf *f, int offset, void *start, void *end)
- {
- stb__nptr_target **p = &f->targets;
- while (*p) {
- stb__nptr_target *n = *p;
- if (n->ptr >= start && n->ptr <= end) {
- // null pointers
- stb__nptr *z = n->first;
- while (z) {
- stb__nptr *y = z->next;
- z->prev = NULL;
- z->next = NULL;
- *(void **) z->ptr = NULL;
- z = y;
- }
- // unlink this target
- *p = n->next_in_block;
- stb__nptr_free(n);
- } else
- p = &(n->next_in_block);
- }
- }
- void stb_nptr_didfree(void *address_being_freed, int len)
- {
- // step one: delete all pointers in this block
- stb__nptr_block(address_being_freed, len, stb__nptr_delete_pointers, 0);
- // step two: NULL all pointers to this block; do this second to avoid NULLing deleted pointers
- stb__nptr_block(address_being_freed, len, stb__nptr_delete_targets, 0);
- }
- void stb_nptr_free(void *address_being_freed, int len)
- {
- free(address_being_freed);
- stb_nptr_didfree(address_being_freed, len);
- }
- static void stb__nptr_move_targets(stb__memory_leaf *f, int offset, void *start, void *end)
- {
- stb__nptr_target **t = &f->targets;
- while (*t) {
- stb__nptr_target *n = *t;
- if (n->ptr >= start && n->ptr <= end) {
- stb__nptr *z;
- stb__memory_leaf *f;
- // unlink n
- *t = n->next_in_block;
- // update n to new address
- n->ptr = (void *) ((char *) n->ptr + offset);
- f = stb__nptr_find_leaf(n->ptr);
- if (!f) f = stb__nptr_make_leaf(n->ptr);
- n->next_in_block = f->targets;
- f->targets = n;
- // now go through all pointers and make them point here
- z = n->first;
- while (z) {
- *(void**) z->ptr = n->ptr;
- z = z->next;
- }
- } else
- t = &(n->next_in_block);
- }
- }
- static void stb__nptr_move_pointers(stb__memory_leaf *f, int offset, void *start, void *end)
- {
- stb__nptr **p = &f->pointers;
- while (*p) {
- stb__nptr *n = *p;
- if (n->ptr >= start && n->ptr <= end) {
- // unlink
- *p = n->next_in_block;
- n->ptr = (void *) ((int) n->ptr + offset);
- // move to new block
- f = stb__nptr_find_leaf(n->ptr);
- if (!f) f = stb__nptr_make_leaf(n->ptr);
- n->next_in_block = f->pointers;
- f->pointers = n;
- } else
- p = &(n->next_in_block);
- }
- }
- void stb_nptr_realloc(void *new_address, void *old_address, int len)
- {
- if (new_address == old_address) return;
- // have to move the pointers first, because moving the targets
- // requires writing to the pointers-to-the-targets, and if some of those moved too,
- // we need to make sure we don't write to the old memory
- // step one: move all pointers within the block
- stb__nptr_block(old_address, len, stb__nptr_move_pointers, (char *) new_address - (char *) old_address);
- // step two: move all targets within the block
- stb__nptr_block(old_address, len, stb__nptr_move_targets, (char *) new_address - (char *) old_address);
- }
- void stb_nptr_move(void *new_address, void *old_address)
- {
- stb_nptr_realloc(new_address, old_address, 1);
- }
- void stb_nptr_recache(void)
- {
- int i,j;
- for (i=0; i < STB__NPTR_ROOT_NUM; ++i)
- if (stb__memtab_root[i])
- for (j=0; j < STB__NPTR_NODE_NUM; ++j)
- if (stb__memtab_root[i]->children[j]) {
- stb__nptr *p = stb__memtab_root[i]->children[j]->pointers;
- while (p) {
- stb_nptr_didset(p->ptr);
- p = p->next_in_block;
- }
- }
- }
- #endif // STB_DEFINE
- #endif // STB_NPTR
- //////////////////////////////////////////////////////////////////////////////
- //
- // File Processing
- //
- #ifdef _WIN32
- #define stb_rename(x,y) _wrename((const wchar_t *)stb__from_utf8(x), (const wchar_t *)stb__from_utf8_alt(y))
- #else
- #define stb_rename rename
- #endif
- STB_EXTERN void stb_fput_varlen64(FILE *f, stb_uint64 v);
- STB_EXTERN stb_uint64 stb_fget_varlen64(FILE *f);
- STB_EXTERN int stb_size_varlen64(stb_uint64 v);
- #define stb_filec (char *) stb_file
- #define stb_fileu (unsigned char *) stb_file
- STB_EXTERN void * stb_file(char *filename, size_t *length);
- STB_EXTERN void * stb_file_max(char *filename, size_t *length);
- STB_EXTERN size_t stb_filelen(FILE *f);
- STB_EXTERN int stb_filewrite(char *filename, void *data, size_t length);
- STB_EXTERN int stb_filewritestr(char *filename, char *data);
- STB_EXTERN char ** stb_stringfile(char *filename, int *len);
- STB_EXTERN char ** stb_stringfile_trimmed(char *name, int *len, char comm);
- STB_EXTERN char * stb_fgets(char *buffer, int buflen, FILE *f);
- STB_EXTERN char * stb_fgets_malloc(FILE *f);
- STB_EXTERN int stb_fexists(char *filename);
- STB_EXTERN int stb_fcmp(char *s1, char *s2);
- STB_EXTERN int stb_feq(char *s1, char *s2);
- STB_EXTERN time_t stb_ftimestamp(char *filename);
- STB_EXTERN int stb_fullpath(char *abs, int abs_size, char *rel);
- STB_EXTERN FILE * stb_fopen(char *filename, const char *mode);
- STB_EXTERN int stb_fclose(FILE *f, int keep);
- enum
- {
- stb_keep_no = 0,
- stb_keep_yes = 1,
- stb_keep_if_different = 2,
- };
- STB_EXTERN int stb_copyfile(char *src, char *dest);
- STB_EXTERN void stb_fput_varlen64(FILE *f, stb_uint64 v);
- STB_EXTERN stb_uint64 stb_fget_varlen64(FILE *f);
- STB_EXTERN int stb_size_varlen64(stb_uint64 v);
- STB_EXTERN void stb_fwrite32(FILE *f, stb_uint32 datum);
- STB_EXTERN void stb_fput_varlen (FILE *f, int v);
- STB_EXTERN void stb_fput_varlenu(FILE *f, unsigned int v);
- STB_EXTERN int stb_fget_varlen (FILE *f);
- STB_EXTERN stb_uint stb_fget_varlenu(FILE *f);
- STB_EXTERN void stb_fput_ranged (FILE *f, int v, int b, stb_uint n);
- STB_EXTERN int stb_fget_ranged (FILE *f, int b, stb_uint n);
- STB_EXTERN int stb_size_varlen (int v);
- STB_EXTERN int stb_size_varlenu(unsigned int v);
- STB_EXTERN int stb_size_ranged (int b, stb_uint n);
- STB_EXTERN int stb_fread(void *data, size_t len, size_t count, void *f);
- STB_EXTERN int stb_fwrite(void *data, size_t len, size_t count, void *f);
- #if 0
- typedef struct
- {
- FILE *base_file;
- char *buffer;
- int buffer_size;
- int buffer_off;
- int buffer_left;
- } STBF;
- STB_EXTERN STBF *stb_tfopen(char *filename, char *mode);
- STB_EXTERN int stb_tfread(void *data, size_t len, size_t count, STBF *f);
- STB_EXTERN int stb_tfwrite(void *data, size_t len, size_t count, STBF *f);
- #endif
- #ifdef STB_DEFINE
- #if 0
- STBF *stb_tfopen(char *filename, char *mode)
- {
- STBF *z;
- FILE *f = stb_p_fopen(filename, mode);
- if (!f) return NULL;
- z = (STBF *) malloc(sizeof(*z));
- if (!z) { fclose(f); return NULL; }
- z->base_file = f;
- if (!strcmp(mode, "rb") || !strcmp(mode, "wb")) {
- z->buffer_size = 4096;
- z->buffer_off = z->buffer_size;
- z->buffer_left = 0;
- z->buffer = malloc(z->buffer_size);
- if (!z->buffer) { free(z); fclose(f); return NULL; }
- } else {
- z->buffer = 0;
- z->buffer_size = 0;
- z->buffer_left = 0;
- }
- return z;
- }
- int stb_tfread(void *data, size_t len, size_t count, STBF *f)
- {
- int total = len*count, done=0;
- if (!total) return 0;
- if (total <= z->buffer_left) {
- memcpy(data, z->buffer + z->buffer_off, total);
- z->buffer_off += total;
- z->buffer_left -= total;
- return count;
- } else {
- char *out = (char *) data;
- // consume all buffered data
- memcpy(data, z->buffer + z->buffer_off, z->buffer_left);
- done = z->buffer_left;
- out += z->buffer_left;
- z->buffer_left=0;
- if (total-done > (z->buffer_size >> 1)) {
- done += fread(out
- }
- }
- }
- #endif
- void stb_fwrite32(FILE *f, stb_uint32 x)
- {
- fwrite(&x, 4, 1, f);
- }
- #if defined(_WIN32)
- #define stb__stat _stat
- #else
- #define stb__stat stat
- #endif
- int stb_fexists(char *filename)
- {
- struct stb__stat buf;
- return stb__windows(
- _wstat((const wchar_t *)stb__from_utf8(filename), &buf),
- stat(filename,&buf)
- ) == 0;
- }
- time_t stb_ftimestamp(char *filename)
- {
- struct stb__stat buf;
- if (stb__windows(
- _wstat((const wchar_t *)stb__from_utf8(filename), &buf),
- stat(filename,&buf)
- ) == 0)
- {
- return buf.st_mtime;
- } else {
- return 0;
- }
- }
- size_t stb_filelen(FILE *f)
- {
- long len, pos;
- pos = ftell(f);
- fseek(f, 0, SEEK_END);
- len = ftell(f);
- fseek(f, pos, SEEK_SET);
- return (size_t) len;
- }
- void *stb_file(char *filename, size_t *length)
- {
- FILE *f = stb__fopen(filename, "rb");
- char *buffer;
- size_t len, len2;
- if (!f) return NULL;
- len = stb_filelen(f);
- buffer = (char *) malloc(len+2); // nul + extra
- len2 = fread(buffer, 1, len, f);
- if (len2 == len) {
- if (length) *length = len;
- buffer[len] = 0;
- } else {
- free(buffer);
- buffer = NULL;
- }
- fclose(f);
- return buffer;
- }
- int stb_filewrite(char *filename, void *data, size_t length)
- {
- FILE *f = stb_fopen(filename, "wb");
- if (f) {
- unsigned char *data_ptr = (unsigned char *) data;
- size_t remaining = length;
- while (remaining > 0) {
- size_t len2 = remaining > 65536 ? 65536 : remaining;
- size_t len3 = fwrite(data_ptr, 1, len2, f);
- if (len2 != len3) {
- fprintf(stderr, "Failed while writing %s\n", filename);
- break;
- }
- remaining -= len2;
- data_ptr += len2;
- }
- stb_fclose(f, stb_keep_if_different);
- }
- return f != NULL;
- }
- int stb_filewritestr(char *filename, char *data)
- {
- return stb_filewrite(filename, data, strlen(data));
- }
- void * stb_file_max(char *filename, size_t *length)
- {
- FILE *f = stb__fopen(filename, "rb");
- char *buffer;
- size_t len, maxlen;
- if (!f) return NULL;
- maxlen = *length;
- buffer = (char *) malloc(maxlen+1);
- len = fread(buffer, 1, maxlen, f);
- buffer[len] = 0;
- fclose(f);
- *length = len;
- return buffer;
- }
- char ** stb_stringfile(char *filename, int *plen)
- {
- FILE *f = stb__fopen(filename, "rb");
- char *buffer, **list=NULL, *s;
- size_t len, count, i;
- if (!f) return NULL;
- len = stb_filelen(f);
- buffer = (char *) malloc(len+1);
- len = fread(buffer, 1, len, f);
- buffer[len] = 0;
- fclose(f);
- // two passes through: first time count lines, second time set them
- for (i=0; i < 2; ++i) {
- s = buffer;
- if (i == 1)
- list[0] = s;
- count = 1;
- while (*s) {
- if (*s == '\n' || *s == '\r') {
- // detect if both cr & lf are together
- int crlf = (s[0] + s[1]) == ('\n' + '\r');
- if (i == 1) *s = 0;
- if (crlf) ++s;
- if (s[1]) { // it's not over yet
- if (i == 1) list[count] = s+1;
- ++count;
- }
- }
- ++s;
- }
- if (i == 0) {
- list = (char **) malloc(sizeof(*list) * (count+1) + len+1);
- if (!list) return NULL;
- list[count] = 0;
- // recopy the file so there's just a single allocation to free
- memcpy(&list[count+1], buffer, len+1);
- free(buffer);
- buffer = (char *) &list[count+1];
- if (plen) *plen = (int) count;
- }
- }
- return list;
- }
- char ** stb_stringfile_trimmed(char *name, int *len, char comment)
- {
- int i,n,o=0;
- char **s = stb_stringfile(name, &n);
- if (s == NULL) return NULL;
- for (i=0; i < n; ++i) {
- char *p = stb_skipwhite(s[i]);
- if (*p && *p != comment)
- s[o++] = p;
- }
- s[o] = NULL;
- if (len) *len = o;
- return s;
- }
- char * stb_fgets(char *buffer, int buflen, FILE *f)
- {
- char *p;
- buffer[0] = 0;
- p = fgets(buffer, buflen, f);
- if (p) {
- int n = (int) (strlen(p)-1);
- if (n >= 0)
- if (p[n] == '\n')
- p[n] = 0;
- }
- return p;
- }
- char * stb_fgets_malloc(FILE *f)
- {
- // avoid reallocing for small strings
- char quick_buffer[800];
- quick_buffer[sizeof(quick_buffer)-2] = 0;
- if (!fgets(quick_buffer, sizeof(quick_buffer), f))
- return NULL;
- if (quick_buffer[sizeof(quick_buffer)-2] == 0) {
- size_t n = strlen(quick_buffer);
- if (n > 0 && quick_buffer[n-1] == '\n')
- quick_buffer[n-1] = 0;
- return stb_p_strdup(quick_buffer);
- } else {
- char *p;
- char *a = stb_p_strdup(quick_buffer);
- size_t len = sizeof(quick_buffer)-1;
- while (!feof(f)) {
- if (a[len-1] == '\n') break;
- a = (char *) realloc(a, len*2);
- p = &a[len];
- p[len-2] = 0;
- if (!fgets(p, (int) len, f))
- break;
- if (p[len-2] == 0) {
- len += strlen(p);
- break;
- }
- len = len + (len-1);
- }
- if (a[len-1] == '\n')
- a[len-1] = 0;
- return a;
- }
- }
- int stb_fullpath(char *abs, int abs_size, char *rel)
- {
- #ifdef _WIN32
- return _fullpath(abs, rel, abs_size) != NULL;
- #else
- if (rel[0] == '/' || rel[0] == '~') {
- if ((int) strlen(rel) >= abs_size)
- return 0;
- stb_p_strcpy_s(abs,65536,rel);
- return STB_TRUE;
- } else {
- int n;
- getcwd(abs, abs_size);
- n = strlen(abs);
- if (n+(int) strlen(rel)+2 <= abs_size) {
- abs[n] = '/';
- stb_p_strcpy_s(abs+n+1, 65536,rel);
- return STB_TRUE;
- } else {
- return STB_FALSE;
- }
- }
- #endif
- }
- static int stb_fcmp_core(FILE *f, FILE *g)
- {
- char buf1[1024],buf2[1024];
- int n1,n2, res=0;
- while (1) {
- n1 = (int) fread(buf1, 1, sizeof(buf1), f);
- n2 = (int) fread(buf2, 1, sizeof(buf2), g);
- res = memcmp(buf1,buf2,stb_min(n1,n2));
- if (res)
- break;
- if (n1 != n2) {
- res = n1 < n2 ? -1 : 1;
- break;
- }
- if (n1 == 0)
- break;
- }
- fclose(f);
- fclose(g);
- return res;
- }
- int stb_fcmp(char *s1, char *s2)
- {
- FILE *f = stb__fopen(s1, "rb");
- FILE *g = stb__fopen(s2, "rb");
- if (f == NULL || g == NULL) {
- if (f) fclose(f);
- if (g) {
- fclose(g);
- return STB_TRUE;
- }
- return f != NULL;
- }
- return stb_fcmp_core(f,g);
- }
- int stb_feq(char *s1, char *s2)
- {
- FILE *f = stb__fopen(s1, "rb");
- FILE *g = stb__fopen(s2, "rb");
- if (f == NULL || g == NULL) {
- if (f) fclose(f);
- if (g) fclose(g);
- return f == g;
- }
- // feq is faster because it shortcuts if they're different length
- if (stb_filelen(f) != stb_filelen(g)) {
- fclose(f);
- fclose(g);
- return 0;
- }
- return !stb_fcmp_core(f,g);
- }
- static stb_ptrmap *stb__files;
- typedef struct
- {
- char *temp_name;
- char *name;
- int errors;
- } stb__file_data;
- static FILE *stb__open_temp_file(char *temp_name, char *src_name, const char *mode)
- {
- size_t p;
- #ifdef _MSC_VER
- int j;
- #endif
- FILE *f;
- // try to generate a temporary file in the same directory
- p = strlen(src_name)-1;
- while (p > 0 && src_name[p] != '/' && src_name[p] != '\\'
- && src_name[p] != ':' && src_name[p] != '~')
- --p;
- ++p;
- memcpy(temp_name, src_name, p);
- #ifdef _MSC_VER
- // try multiple times to make a temp file... just in
- // case some other process makes the name first
- for (j=0; j < 32; ++j) {
- stb_p_strcpy_s(temp_name+p, 65536, "stmpXXXXXX");
- if (!stb_p_mktemp(temp_name))
- return 0;
- f = stb_p_fopen(temp_name, mode);
- if (f != NULL)
- break;
- }
- #else
- {
- stb_p_strcpy_s(temp_name+p, 65536, "stmpXXXXXX");
- #ifdef __MINGW32__
- int fd = open(stb_p_mktemp(temp_name), O_RDWR);
- #else
- int fd = mkstemp(temp_name);
- #endif
- if (fd == -1) return NULL;
- f = fdopen(fd, mode);
- if (f == NULL) {
- unlink(temp_name);
- close(fd);
- return NULL;
- }
- }
- #endif
- return f;
- }
- FILE * stb_fopen(char *filename, const char *mode)
- {
- FILE *f;
- char name_full[4096];
- char temp_full[sizeof(name_full) + 12];
- // @TODO: if the file doesn't exist, we can also use the fastpath here
- if (mode[0] != 'w' && !strchr(mode, '+'))
- return stb__fopen(filename, mode);
- // save away the full path to the file so if the program
- // changes the cwd everything still works right! unix has
- // better ways to do this, but we have to work in windows
- name_full[0] = '\0'; // stb_fullpath reads name_full[0]
- if (stb_fullpath(name_full, sizeof(name_full), filename)==0)
- return 0;
- f = stb__open_temp_file(temp_full, name_full, mode);
- if (f != NULL) {
- stb__file_data *d = (stb__file_data *) malloc(sizeof(*d));
- if (!d) { assert(0); /* NOTREACHED */fclose(f); return NULL; }
- if (stb__files == NULL) stb__files = stb_ptrmap_create();
- d->temp_name = stb_p_strdup(temp_full);
- d->name = stb_p_strdup(name_full);
- d->errors = 0;
- stb_ptrmap_add(stb__files, f, d);
- return f;
- }
- return NULL;
- }
- int stb_fclose(FILE *f, int keep)
- {
- stb__file_data *d;
- int ok = STB_FALSE;
- if (f == NULL) return 0;
- if (ferror(f))
- keep = stb_keep_no;
- fclose(f);
- if (stb__files && stb_ptrmap_remove(stb__files, f, (void **) &d)) {
- if (stb__files->count == 0) {
- stb_ptrmap_destroy(stb__files);
- stb__files = NULL;
- }
- } else
- return STB_TRUE; // not special
- if (keep == stb_keep_if_different) {
- // check if the files are identical
- if (stb_feq(d->name, d->temp_name)) {
- keep = stb_keep_no;
- ok = STB_TRUE; // report success if no change
- }
- }
- if (keep == stb_keep_no) {
- remove(d->temp_name);
- } else {
- if (!stb_fexists(d->name)) {
- // old file doesn't exist, so just move the new file over it
- stb_rename(d->temp_name, d->name);
- } else {
- // don't delete the old file yet in case there are troubles! First rename it!
- char preserved_old_file[4096];
- // generate a temp filename in the same directory (also creates it, which we don't need)
- FILE *dummy = stb__open_temp_file(preserved_old_file, d->name, "wb");
- if (dummy != NULL) {
- // we don't actually want the open file
- fclose(dummy);
- // discard what we just created
- remove(preserved_old_file); // if this fails, there's nothing we can do, and following logic handles it as best as possible anyway
- // move the existing file to the preserved name
- if (0 != stb_rename(d->name, preserved_old_file)) { // 0 on success
- // failed, state is:
- // filename -> old file
- // tempname -> new file
- // keep tempname around so we don't lose data
- } else {
- // state is:
- // preserved -> old file
- // tempname -> new file
- // move the new file to the old name
- if (0 == stb_rename(d->temp_name, d->name)) {
- // state is:
- // preserved -> old file
- // filename -> new file
- ok = STB_TRUE;
- // 'filename -> new file' has always been the goal, so clean up
- remove(preserved_old_file); // nothing to be done if it fails
- } else {
- // couldn't rename, so try renaming preserved file back
- // state is:
- // preserved -> old file
- // tempname -> new file
- stb_rename(preserved_old_file, d->name);
- // if the rename failed, there's nothing more we can do
- }
- }
- } else {
- // we couldn't get a temp filename. do this the naive way; the worst case failure here
- // leaves the filename pointing to nothing and the new file as a tempfile
- remove(d->name);
- stb_rename(d->temp_name, d->name);
- }
- }
- }
- free(d->temp_name);
- free(d->name);
- free(d);
- return ok;
- }
- int stb_copyfile(char *src, char *dest)
- {
- char raw_buffer[1024];
- char *buffer;
- int buf_size = 65536;
- FILE *f, *g;
- // if file already exists at destination, do nothing
- if (stb_feq(src, dest)) return STB_TRUE;
- // open file
- f = stb__fopen(src, "rb");
- if (f == NULL) return STB_FALSE;
- // open file for writing
- g = stb__fopen(dest, "wb");
- if (g == NULL) {
- fclose(f);
- return STB_FALSE;
- }
- buffer = (char *) malloc(buf_size);
- if (buffer == NULL) {
- buffer = raw_buffer;
- buf_size = sizeof(raw_buffer);
- }
- while (!feof(f)) {
- size_t n = fread(buffer, 1, buf_size, f);
- if (n != 0)
- fwrite(buffer, 1, n, g);
- }
- fclose(f);
- if (buffer != raw_buffer)
- free(buffer);
- fclose(g);
- return STB_TRUE;
- }
- // varlen:
- // v' = (v >> 31) + (v < 0 ? ~v : v)<<1; // small abs(v) => small v'
- // output v as big endian v'+k for v' <= k:
- // 1 byte : v' <= 0x00000080 ( -64 <= v < 64) 7 bits
- // 2 bytes: v' <= 0x00004000 (-8192 <= v < 8192) 14 bits
- // 3 bytes: v' <= 0x00200000 21 bits
- // 4 bytes: v' <= 0x10000000 28 bits
- // the number of most significant 1-bits in the first byte
- // equals the number of bytes after the first
- #define stb__varlen_xform(v) (v<0 ? (~v << 1)+1 : (v << 1))
- int stb_size_varlen(int v) { return stb_size_varlenu(stb__varlen_xform(v)); }
- int stb_size_varlenu(unsigned int v)
- {
- if (v < 0x00000080) return 1;
- if (v < 0x00004000) return 2;
- if (v < 0x00200000) return 3;
- if (v < 0x10000000) return 4;
- return 5;
- }
- void stb_fput_varlen(FILE *f, int v) { stb_fput_varlenu(f, stb__varlen_xform(v)); }
- void stb_fput_varlenu(FILE *f, unsigned int z)
- {
- if (z >= 0x10000000) fputc(0xF0,f);
- if (z >= 0x00200000) fputc((z < 0x10000000 ? 0xE0 : 0)+(z>>24),f);
- if (z >= 0x00004000) fputc((z < 0x00200000 ? 0xC0 : 0)+(z>>16),f);
- if (z >= 0x00000080) fputc((z < 0x00004000 ? 0x80 : 0)+(z>> 8),f);
- fputc(z,f);
- }
- #define stb_fgetc(f) ((unsigned char) fgetc(f))
- int stb_fget_varlen(FILE *f)
- {
- unsigned int z = stb_fget_varlenu(f);
- return (z & 1) ? ~(z>>1) : (z>>1);
- }
- unsigned int stb_fget_varlenu(FILE *f)
- {
- unsigned int z;
- unsigned char d;
- d = stb_fgetc(f);
- if (d >= 0x80) {
- if (d >= 0xc0) {
- if (d >= 0xe0) {
- if (d == 0xf0) z = stb_fgetc(f) << 24;
- else z = (d - 0xe0) << 24;
- z += stb_fgetc(f) << 16;
- }
- else
- z = (d - 0xc0) << 16;
- z += stb_fgetc(f) << 8;
- } else
- z = (d - 0x80) << 8;
- z += stb_fgetc(f);
- } else
- z = d;
- return z;
- }
- stb_uint64 stb_fget_varlen64(FILE *f)
- {
- stb_uint64 z;
- unsigned char d;
- d = stb_fgetc(f);
- if (d >= 0x80) {
- if (d >= 0xc0) {
- if (d >= 0xe0) {
- if (d >= 0xf0) {
- if (d >= 0xf8) {
- if (d >= 0xfc) {
- if (d >= 0xfe) {
- if (d >= 0xff)
- z = (stb_uint64) stb_fgetc(f) << 56;
- else
- z = (stb_uint64) (d - 0xfe) << 56;
- z |= (stb_uint64) stb_fgetc(f) << 48;
- } else z = (stb_uint64) (d - 0xfc) << 48;
- z |= (stb_uint64) stb_fgetc(f) << 40;
- } else z = (stb_uint64) (d - 0xf8) << 40;
- z |= (stb_uint64) stb_fgetc(f) << 32;
- } else z = (stb_uint64) (d - 0xf0) << 32;
- z |= (stb_uint) stb_fgetc(f) << 24;
- } else z = (stb_uint) (d - 0xe0) << 24;
- z |= (stb_uint) stb_fgetc(f) << 16;
- } else z = (stb_uint) (d - 0xc0) << 16;
- z |= (stb_uint) stb_fgetc(f) << 8;
- } else z = (stb_uint) (d - 0x80) << 8;
- z |= stb_fgetc(f);
- } else
- z = d;
- return (z & 1) ? ~(z >> 1) : (z >> 1);
- }
- int stb_size_varlen64(stb_uint64 v)
- {
- if (v < 0x00000080) return 1;
- if (v < 0x00004000) return 2;
- if (v < 0x00200000) return 3;
- if (v < 0x10000000) return 4;
- if (v < STB_IMM_UINT64(0x0000000800000000)) return 5;
- if (v < STB_IMM_UINT64(0x0000040000000000)) return 6;
- if (v < STB_IMM_UINT64(0x0002000000000000)) return 7;
- if (v < STB_IMM_UINT64(0x0100000000000000)) return 8;
- return 9;
- }
- void stb_fput_varlen64(FILE *f, stb_uint64 v)
- {
- stb_uint64 z = stb__varlen_xform(v);
- int first=1;
- if (z >= STB_IMM_UINT64(0x100000000000000)) {
- fputc(0xff,f);
- first=0;
- }
- if (z >= STB_IMM_UINT64(0x02000000000000)) fputc((first ? 0xFE : 0)+(char)(z>>56),f), first=0;
- if (z >= STB_IMM_UINT64(0x00040000000000)) fputc((first ? 0xFC : 0)+(char)(z>>48),f), first=0;
- if (z >= STB_IMM_UINT64(0x00000800000000)) fputc((first ? 0xF8 : 0)+(char)(z>>40),f), first=0;
- if (z >= STB_IMM_UINT64(0x00000010000000)) fputc((first ? 0xF0 : 0)+(char)(z>>32),f), first=0;
- if (z >= STB_IMM_UINT64(0x00000000200000)) fputc((first ? 0xE0 : 0)+(char)(z>>24),f), first=0;
- if (z >= STB_IMM_UINT64(0x00000000004000)) fputc((first ? 0xC0 : 0)+(char)(z>>16),f), first=0;
- if (z >= STB_IMM_UINT64(0x00000000000080)) fputc((first ? 0x80 : 0)+(char)(z>> 8),f), first=0;
- fputc((char)z,f);
- }
- void stb_fput_ranged(FILE *f, int v, int b, stb_uint n)
- {
- v -= b;
- if (n <= (1 << 31))
- assert((stb_uint) v < n);
- if (n > (1 << 24)) fputc(v >> 24, f);
- if (n > (1 << 16)) fputc(v >> 16, f);
- if (n > (1 << 8)) fputc(v >> 8, f);
- fputc(v,f);
- }
- int stb_fget_ranged(FILE *f, int b, stb_uint n)
- {
- unsigned int v=0;
- if (n > (1 << 24)) v += stb_fgetc(f) << 24;
- if (n > (1 << 16)) v += stb_fgetc(f) << 16;
- if (n > (1 << 8)) v += stb_fgetc(f) << 8;
- v += stb_fgetc(f);
- return b+v;
- }
- int stb_size_ranged(int b, stb_uint n)
- {
- if (n > (1 << 24)) return 4;
- if (n > (1 << 16)) return 3;
- if (n > (1 << 8)) return 2;
- return 1;
- }
- void stb_fput_string(FILE *f, char *s)
- {
- size_t len = strlen(s);
- stb_fput_varlenu(f, (unsigned int) len);
- fwrite(s, 1, len, f);
- }
- // inverse of the above algorithm
- char *stb_fget_string(FILE *f, void *p)
- {
- char *s;
- int len = stb_fget_varlenu(f);
- if (len > 4096) return NULL;
- s = p ? stb_malloc_string(p, len+1) : (char *) malloc(len+1);
- fread(s, 1, len, f);
- s[len] = 0;
- return s;
- }
- char *stb_strdup(char *str, void *pool)
- {
- size_t len = strlen(str);
- char *p = stb_malloc_string(pool, len+1);
- stb_p_strcpy_s(p, len+1, str);
- return p;
- }
- // strip the trailing '/' or '\\' from a directory so we can refer to it
- // as a file for _stat()
- char *stb_strip_final_slash(char *t)
- {
- if (t[0]) {
- char *z = t + strlen(t) - 1;
- // *z is the last character
- if (*z == '\\' || *z == '/')
- if (z != t+2 || t[1] != ':') // but don't strip it if it's e.g. "c:/"
- *z = 0;
- if (*z == '\\')
- *z = '/'; // canonicalize to make sure it matches db
- }
- return t;
- }
- char *stb_strip_final_slash_regardless(char *t)
- {
- if (t[0]) {
- char *z = t + strlen(t) - 1;
- // *z is the last character
- if (*z == '\\' || *z == '/')
- *z = 0;
- if (*z == '\\')
- *z = '/'; // canonicalize to make sure it matches db
- }
- return t;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Options parsing
- //
- STB_EXTERN char **stb_getopt_param(int *argc, char **argv, char *param);
- STB_EXTERN char **stb_getopt(int *argc, char **argv);
- STB_EXTERN void stb_getopt_free(char **opts);
- #ifdef STB_DEFINE
- void stb_getopt_free(char **opts)
- {
- int i;
- char ** o2 = opts;
- for (i=0; i < stb_arr_len(o2); ++i)
- free(o2[i]);
- stb_arr_free(o2);
- }
- char **stb_getopt(int *argc, char **argv)
- {
- return stb_getopt_param(argc, argv, (char*) "");
- }
- char **stb_getopt_param(int *argc, char **argv, char *param)
- {
- char ** opts=NULL;
- int i,j=1;
- for (i=1; i < *argc; ++i) {
- if (argv[i][0] != '-') {
- argv[j++] = argv[i];
- } else {
- if (argv[i][1] == 0) { // plain - == don't parse further options
- ++i;
- while (i < *argc)
- argv[j++] = argv[i++];
- break;
- } else if (argv[i][1] == '-') {
- // copy argument through including initial '-' for clarity
- stb_arr_push(opts, stb_p_strdup(argv[i]));
- } else {
- int k;
- char *q = argv[i]; // traverse options list
- for (k=1; q[k]; ++k) {
- char *s;
- if (strchr(param, q[k])) { // does it take a parameter?
- char *t = &q[k+1], z = q[k];
- size_t len=0;
- if (*t == 0) {
- if (i == *argc-1) { // takes a parameter, but none found
- *argc = 0;
- stb_getopt_free(opts);
- return NULL;
- }
- t = argv[++i];
- } else
- k += (int) strlen(t);
- len = strlen(t);
- s = (char *) malloc(len+2);
- if (!s) return NULL;
- s[0] = z;
- stb_p_strcpy_s(s+1, len+2, t);
- } else {
- // no parameter
- s = (char *) malloc(2);
- if (!s) return NULL;
- s[0] = q[k];
- s[1] = 0;
- }
- stb_arr_push(opts, s);
- }
- }
- }
- }
- stb_arr_push(opts, NULL);
- *argc = j;
- return opts;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Portable directory reading
- //
- STB_EXTERN char **stb_readdir_files (char *dir);
- STB_EXTERN char **stb_readdir_files_mask(char *dir, char *wild);
- STB_EXTERN char **stb_readdir_subdirs(char *dir);
- STB_EXTERN char **stb_readdir_subdirs_mask(char *dir, char *wild);
- STB_EXTERN void stb_readdir_free (char **files);
- STB_EXTERN char **stb_readdir_recursive(char *dir, char *filespec);
- STB_EXTERN void stb_delete_directory_recursive(char *dir);
- #ifdef STB_DEFINE
- #ifdef _MSC_VER
- #include <io.h>
- #else
- #include <unistd.h>
- #include <dirent.h>
- #endif
- void stb_readdir_free(char **files)
- {
- char **f2 = files;
- int i;
- for (i=0; i < stb_arr_len(f2); ++i)
- free(f2[i]);
- stb_arr_free(f2);
- }
- static int isdotdirname(char *name)
- {
- if (name[0] == '.')
- return (name[1] == '.') ? !name[2] : !name[1];
- return 0;
- }
- STB_EXTERN int stb_wildmatchi(char *expr, char *candidate);
- static char **readdir_raw(char *dir, int return_subdirs, char *mask)
- {
- char **results = NULL;
- char buffer[4096], with_slash[4096];
- size_t n;
- #ifdef WIN32
- stb__wchar *ws;
- struct _wfinddata_t data;
- #ifdef _WIN64
- const intptr_t none = -1;
- intptr_t z;
- #else
- const long none = -1;
- long z;
- #endif
- #else // !WIN32
- const DIR *none = NULL;
- DIR *z;
- #endif
- n = stb_strscpy(buffer,dir,sizeof(buffer));
- if (!n || n >= sizeof(buffer))
- return NULL;
- stb_fixpath(buffer);
- if (n > 0 && (buffer[n-1] != '/')) {
- buffer[n++] = '/';
- }
- buffer[n] = 0;
- if (!stb_strscpy(with_slash,buffer,sizeof(with_slash)))
- return NULL;
- #ifdef WIN32
- if (!stb_strscpy(buffer+n,"*.*",sizeof(buffer)-n))
- return NULL;
- ws = stb__from_utf8(buffer);
- z = _wfindfirst((wchar_t *)ws, &data);
- #else
- z = opendir(dir);
- #endif
- if (z != none) {
- int nonempty = STB_TRUE;
- #ifndef WIN32
- struct dirent *data = readdir(z);
- nonempty = (data != NULL);
- #endif
- if (nonempty) {
- do {
- int is_subdir;
- #ifdef WIN32
- char *name = stb__to_utf8((stb__wchar *)data.name);
- if (name == NULL) {
- fprintf(stderr, "%s to convert '%S' to %s!\n", "Unable", data.name, "utf8");
- continue;
- }
- is_subdir = !!(data.attrib & _A_SUBDIR);
- #else
- char *name = data->d_name;
- if (!stb_strscpy(buffer+n,name,sizeof(buffer)-n))
- break;
- // Could follow DT_LNK, but would need to check for recursive links.
- is_subdir = !!(data->d_type & DT_DIR);
- #endif
- if (is_subdir == return_subdirs) {
- if (!is_subdir || !isdotdirname(name)) {
- if (!mask || stb_wildmatchi(mask, name)) {
- char buffer[4096],*p=buffer;
- if ( stb_snprintf(buffer, sizeof(buffer), "%s%s", with_slash, name) < 0 )
- break;
- if (buffer[0] == '.' && buffer[1] == '/')
- p = buffer+2;
- stb_arr_push(results, stb_p_strdup(p));
- }
- }
- }
- }
- #ifdef WIN32
- while (0 == _wfindnext(z, &data));
- #else
- while ((data = readdir(z)) != NULL);
- #endif
- }
- #ifdef WIN32
- _findclose(z);
- #else
- closedir(z);
- #endif
- }
- return results;
- }
- char **stb_readdir_files (char *dir) { return readdir_raw(dir, 0, NULL); }
- char **stb_readdir_subdirs(char *dir) { return readdir_raw(dir, 1, NULL); }
- char **stb_readdir_files_mask(char *dir, char *wild) { return readdir_raw(dir, 0, wild); }
- char **stb_readdir_subdirs_mask(char *dir, char *wild) { return readdir_raw(dir, 1, wild); }
- int stb__rec_max=0x7fffffff;
- static char **stb_readdir_rec(char **sofar, char *dir, char *filespec)
- {
- char **files;
- char ** dirs;
- char **p;
- if (stb_arr_len(sofar) >= stb__rec_max) return sofar;
- files = stb_readdir_files_mask(dir, filespec);
- stb_arr_for(p, files) {
- stb_arr_push(sofar, stb_p_strdup(*p));
- if (stb_arr_len(sofar) >= stb__rec_max) break;
- }
- stb_readdir_free(files);
- if (stb_arr_len(sofar) >= stb__rec_max) return sofar;
- dirs = stb_readdir_subdirs(dir);
- stb_arr_for(p, dirs)
- sofar = stb_readdir_rec(sofar, *p, filespec);
- stb_readdir_free(dirs);
- return sofar;
- }
- char **stb_readdir_recursive(char *dir, char *filespec)
- {
- return stb_readdir_rec(NULL, dir, filespec);
- }
- void stb_delete_directory_recursive(char *dir)
- {
- char **list = stb_readdir_subdirs(dir);
- int i;
- for (i=0; i < stb_arr_len(list); ++i)
- stb_delete_directory_recursive(list[i]);
- stb_arr_free(list);
- list = stb_readdir_files(dir);
- for (i=0; i < stb_arr_len(list); ++i)
- if (!remove(list[i])) {
- // on windows, try again after making it writeable; don't ALWAYS
- // do this first since that would be slow in the normal case
- #ifdef _MSC_VER
- _chmod(list[i], _S_IWRITE);
- remove(list[i]);
- #endif
- }
- stb_arr_free(list);
- stb__windows(_rmdir,rmdir)(dir);
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // construct trees from filenames; useful for cmirror summaries
- typedef struct stb_dirtree2 stb_dirtree2;
- struct stb_dirtree2
- {
- stb_dirtree2 **subdirs;
- // make convenient for stb_summarize_tree
- int num_subdir;
- float weight;
- // actual data
- char *fullpath;
- char *relpath;
- char **files;
- };
- STB_EXTERN stb_dirtree2 *stb_dirtree2_from_files_relative(char *src, char **filelist, int count);
- STB_EXTERN stb_dirtree2 *stb_dirtree2_from_files(char **filelist, int count);
- STB_EXTERN int stb_dir_is_prefix(char *dir, int dirlen, char *file);
- #ifdef STB_DEFINE
- int stb_dir_is_prefix(char *dir, int dirlen, char *file)
- {
- if (dirlen == 0) return STB_TRUE;
- if (stb_strnicmp(dir, file, dirlen)) return STB_FALSE;
- if (file[dirlen] == '/' || file[dirlen] == '\\') return STB_TRUE;
- return STB_FALSE;
- }
- stb_dirtree2 *stb_dirtree2_from_files_relative(char *src, char **filelist, int count)
- {
- char buffer1[1024];
- int i;
- int dlen = (int) strlen(src), elen;
- stb_dirtree2 *d;
- char ** descendents = NULL;
- char ** files = NULL;
- char *s;
- if (!count) return NULL;
- // first find all the ones that belong here... note this is will take O(NM) with N files and M subdirs
- for (i=0; i < count; ++i) {
- if (stb_dir_is_prefix(src, dlen, filelist[i])) {
- stb_arr_push(descendents, filelist[i]);
- }
- }
- if (descendents == NULL)
- return NULL;
- elen = dlen;
- // skip a leading slash
- if (elen == 0 && (descendents[0][0] == '/' || descendents[0][0] == '\\'))
- ++elen;
- else if (elen)
- ++elen;
- // now extract all the ones that have their root here
- for (i=0; i < stb_arr_len(descendents);) {
- if (!stb_strchr2(descendents[i]+elen, '/', '\\')) {
- stb_arr_push(files, descendents[i]);
- descendents[i] = descendents[stb_arr_len(descendents)-1];
- stb_arr_pop(descendents);
- } else
- ++i;
- }
- // now create a record
- d = (stb_dirtree2 *) malloc(sizeof(*d));
- d->files = files;
- d->subdirs = NULL;
- d->fullpath = stb_p_strdup(src);
- s = stb_strrchr2(d->fullpath, '/', '\\');
- if (s)
- ++s;
- else
- s = d->fullpath;
- d->relpath = s;
- // now create the children
- qsort(descendents, stb_arr_len(descendents), sizeof(char *), stb_qsort_stricmp(0));
- buffer1[0] = 0;
- for (i=0; i < stb_arr_len(descendents); ++i) {
- char buffer2[1024];
- char *s = descendents[i] + elen, *t;
- t = stb_strchr2(s, '/', '\\');
- assert(t);
- stb_strncpy(buffer2, descendents[i], (int) (t-descendents[i]+1));
- if (stb_stricmp(buffer1, buffer2)) {
- stb_dirtree2 *t = stb_dirtree2_from_files_relative(buffer2, descendents, stb_arr_len(descendents));
- assert(t != NULL);
- stb_p_strcpy_s(buffer1, sizeof(buffer1), buffer2);
- stb_arr_push(d->subdirs, t);
- }
- }
- d->num_subdir = stb_arr_len(d->subdirs);
- d->weight = 0;
- return d;
- }
- stb_dirtree2 *stb_dirtree2_from_files(char **filelist, int count)
- {
- return stb_dirtree2_from_files_relative((char*) "", filelist, count);
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Checksums: CRC-32, ADLER32, SHA-1
- //
- // CRC-32 and ADLER32 allow streaming blocks
- // SHA-1 requires either a complete buffer, max size 2^32 - 73
- // or it can checksum directly from a file, max 2^61
- #define STB_ADLER32_SEED 1
- #define STB_CRC32_SEED 0 // note that we logical NOT this in the code
- STB_EXTERN stb_uint
- stb_adler32(stb_uint adler32, stb_uchar *buffer, stb_uint buflen);
- STB_EXTERN stb_uint
- stb_crc32_block(stb_uint crc32, stb_uchar *buffer, stb_uint len);
- STB_EXTERN stb_uint stb_crc32(unsigned char *buffer, stb_uint len);
- STB_EXTERN void stb_sha1(
- unsigned char output[20], unsigned char *buffer, unsigned int len);
- STB_EXTERN int stb_sha1_file(unsigned char output[20], char *file);
- STB_EXTERN void stb_sha1_readable(char display[27], unsigned char sha[20]);
- #ifdef STB_DEFINE
- stb_uint stb_crc32_block(stb_uint crc, unsigned char *buffer, stb_uint len)
- {
- static stb_uint crc_table[256];
- stb_uint i,j,s;
- crc = ~crc;
- if (crc_table[1] == 0)
- for(i=0; i < 256; i++) {
- for (s=i, j=0; j < 8; ++j)
- s = (s >> 1) ^ (s & 1 ? 0xedb88320 : 0);
- crc_table[i] = s;
- }
- for (i=0; i < len; ++i)
- crc = (crc >> 8) ^ crc_table[buffer[i] ^ (crc & 0xff)];
- return ~crc;
- }
- stb_uint stb_crc32(unsigned char *buffer, stb_uint len)
- {
- return stb_crc32_block(0, buffer, len);
- }
- stb_uint stb_adler32(stb_uint adler32, stb_uchar *buffer, stb_uint buflen)
- {
- const unsigned long ADLER_MOD = 65521;
- unsigned long s1 = adler32 & 0xffff, s2 = adler32 >> 16;
- unsigned long blocklen, i;
- blocklen = buflen % 5552;
- while (buflen) {
- for (i=0; i + 7 < blocklen; i += 8) {
- s1 += buffer[0], s2 += s1;
- s1 += buffer[1], s2 += s1;
- s1 += buffer[2], s2 += s1;
- s1 += buffer[3], s2 += s1;
- s1 += buffer[4], s2 += s1;
- s1 += buffer[5], s2 += s1;
- s1 += buffer[6], s2 += s1;
- s1 += buffer[7], s2 += s1;
- buffer += 8;
- }
- for (; i < blocklen; ++i)
- s1 += *buffer++, s2 += s1;
- s1 %= ADLER_MOD, s2 %= ADLER_MOD;
- buflen -= blocklen;
- blocklen = 5552;
- }
- return (s2 << 16) + s1;
- }
- static void stb__sha1(stb_uchar *chunk, stb_uint h[5])
- {
- int i;
- stb_uint a,b,c,d,e;
- stb_uint w[80];
- for (i=0; i < 16; ++i)
- w[i] = stb_big32(&chunk[i*4]);
- for (i=16; i < 80; ++i) {
- stb_uint t;
- t = w[i-3] ^ w[i-8] ^ w[i-14] ^ w[i-16];
- w[i] = (t + t) | (t >> 31);
- }
- a = h[0];
- b = h[1];
- c = h[2];
- d = h[3];
- e = h[4];
- #define STB__SHA1(k,f) \
- { \
- stb_uint temp = (a << 5) + (a >> 27) + (f) + e + (k) + w[i]; \
- e = d; \
- d = c; \
- c = (b << 30) + (b >> 2); \
- b = a; \
- a = temp; \
- }
- i=0;
- for (; i < 20; ++i) STB__SHA1(0x5a827999, d ^ (b & (c ^ d)) );
- for (; i < 40; ++i) STB__SHA1(0x6ed9eba1, b ^ c ^ d );
- for (; i < 60; ++i) STB__SHA1(0x8f1bbcdc, (b & c) + (d & (b ^ c)) );
- for (; i < 80; ++i) STB__SHA1(0xca62c1d6, b ^ c ^ d );
- #undef STB__SHA1
- h[0] += a;
- h[1] += b;
- h[2] += c;
- h[3] += d;
- h[4] += e;
- }
- void stb_sha1(stb_uchar output[20], stb_uchar *buffer, stb_uint len)
- {
- unsigned char final_block[128];
- stb_uint end_start, final_len, j;
- int i;
- stb_uint h[5];
- h[0] = 0x67452301;
- h[1] = 0xefcdab89;
- h[2] = 0x98badcfe;
- h[3] = 0x10325476;
- h[4] = 0xc3d2e1f0;
- // we need to write padding to the last one or two
- // blocks, so build those first into 'final_block'
- // we have to write one special byte, plus the 8-byte length
- // compute the block where the data runs out
- end_start = len & ~63;
- // compute the earliest we can encode the length
- if (((len+9) & ~63) == end_start) {
- // it all fits in one block, so fill a second-to-last block
- end_start -= 64;
- }
- final_len = end_start + 128;
- // now we need to copy the data in
- assert(end_start + 128 >= len+9);
- assert(end_start < len || len < 64-9);
- j = 0;
- if (end_start > len)
- j = (stb_uint) - (int) end_start;
- for (; end_start + j < len; ++j)
- final_block[j] = buffer[end_start + j];
- final_block[j++] = 0x80;
- while (j < 128-5) // 5 byte length, so write 4 extra padding bytes
- final_block[j++] = 0;
- // big-endian size
- final_block[j++] = len >> 29;
- final_block[j++] = len >> 21;
- final_block[j++] = len >> 13;
- final_block[j++] = len >> 5;
- final_block[j++] = len << 3;
- assert(j == 128 && end_start + j == final_len);
- for (j=0; j < final_len; j += 64) { // 512-bit chunks
- if (j+64 >= end_start+64)
- stb__sha1(&final_block[j - end_start], h);
- else
- stb__sha1(&buffer[j], h);
- }
- for (i=0; i < 5; ++i) {
- output[i*4 + 0] = h[i] >> 24;
- output[i*4 + 1] = h[i] >> 16;
- output[i*4 + 2] = h[i] >> 8;
- output[i*4 + 3] = h[i] >> 0;
- }
- }
- #ifdef _MSC_VER
- int stb_sha1_file(stb_uchar output[20], char *file)
- {
- int i;
- stb_uint64 length=0;
- unsigned char buffer[128];
- FILE *f = stb__fopen(file, "rb");
- stb_uint h[5];
- if (f == NULL) return 0; // file not found
- h[0] = 0x67452301;
- h[1] = 0xefcdab89;
- h[2] = 0x98badcfe;
- h[3] = 0x10325476;
- h[4] = 0xc3d2e1f0;
- for(;;) {
- size_t n = fread(buffer, 1, 64, f);
- if (n == 64) {
- stb__sha1(buffer, h);
- length += n;
- } else {
- int block = 64;
- length += n;
- buffer[n++] = 0x80;
- // if there isn't enough room for the length, double the block
- if (n + 8 > 64)
- block = 128;
- // pad to end
- memset(buffer+n, 0, block-8-n);
- i = block - 8;
- buffer[i++] = (stb_uchar) (length >> 53);
- buffer[i++] = (stb_uchar) (length >> 45);
- buffer[i++] = (stb_uchar) (length >> 37);
- buffer[i++] = (stb_uchar) (length >> 29);
- buffer[i++] = (stb_uchar) (length >> 21);
- buffer[i++] = (stb_uchar) (length >> 13);
- buffer[i++] = (stb_uchar) (length >> 5);
- buffer[i++] = (stb_uchar) (length << 3);
- assert(i == block);
- stb__sha1(buffer, h);
- if (block == 128)
- stb__sha1(buffer+64, h);
- else
- assert(block == 64);
- break;
- }
- }
- fclose(f);
- for (i=0; i < 5; ++i) {
- output[i*4 + 0] = h[i] >> 24;
- output[i*4 + 1] = h[i] >> 16;
- output[i*4 + 2] = h[i] >> 8;
- output[i*4 + 3] = h[i] >> 0;
- }
- return 1;
- }
- #endif // _MSC_VER
- // client can truncate this wherever they like
- void stb_sha1_readable(char display[27], unsigned char sha[20])
- {
- char encoding[65] = "0123456789abcdefghijklmnopqrstuv"
- "wxyzABCDEFGHIJKLMNOPQRSTUVWXYZ%$";
- int num_bits = 0, acc=0;
- int i=0,o=0;
- while (o < 26) {
- int v;
- // expand the accumulator
- if (num_bits < 6) {
- assert(i != 20);
- acc += sha[i++] << num_bits;
- num_bits += 8;
- }
- v = acc & ((1 << 6) - 1);
- display[o++] = encoding[v];
- acc >>= 6;
- num_bits -= 6;
- }
- assert(num_bits == 20*8 - 26*6);
- display[o++] = encoding[acc];
- }
- #endif // STB_DEFINE
- ///////////////////////////////////////////////////////////
- //
- // simplified WINDOWS registry interface... hopefully
- // we'll never actually use this?
- #if defined(_WIN32)
- STB_EXTERN void * stb_reg_open(const char *mode, const char *where); // mode: "rHKLM" or "rHKCU" or "w.."
- STB_EXTERN void stb_reg_close(void *reg);
- STB_EXTERN int stb_reg_read(void *zreg, const char *str, void *data, unsigned long len);
- STB_EXTERN int stb_reg_read_string(void *zreg, const char *str, char *data, int len);
- STB_EXTERN void stb_reg_write(void *zreg, const char *str, const void *data, unsigned long len);
- STB_EXTERN void stb_reg_write_string(void *zreg, const char *str, const char *data);
- #if defined(STB_DEFINE) && !defined(STB_NO_REGISTRY)
- #define STB_HAS_REGISTRY
- #ifndef _WINDOWS_
- #define HKEY void *
- STB_EXTERN __declspec(dllimport) long __stdcall RegCloseKey ( HKEY hKey );
- STB_EXTERN __declspec(dllimport) long __stdcall RegCreateKeyExA ( HKEY hKey, const char * lpSubKey,
- int Reserved, char * lpClass, int dwOptions,
- int samDesired, void *lpSecurityAttributes, HKEY * phkResult, int * lpdwDisposition );
- STB_EXTERN __declspec(dllimport) long __stdcall RegDeleteKeyA ( HKEY hKey, const char * lpSubKey );
- STB_EXTERN __declspec(dllimport) long __stdcall RegQueryValueExA ( HKEY hKey, const char * lpValueName,
- int * lpReserved, unsigned long * lpType, unsigned char * lpData, unsigned long * lpcbData );
- STB_EXTERN __declspec(dllimport) long __stdcall RegSetValueExA ( HKEY hKey, const char * lpValueName,
- int Reserved, int dwType, const unsigned char* lpData, int cbData );
- STB_EXTERN __declspec(dllimport) long __stdcall RegOpenKeyExA ( HKEY hKey, const char * lpSubKey,
- int ulOptions, int samDesired, HKEY * phkResult );
- #endif // _WINDOWS_
- #define STB__REG_OPTION_NON_VOLATILE 0
- #define STB__REG_KEY_ALL_ACCESS 0x000f003f
- #define STB__REG_KEY_READ 0x00020019
- #ifdef _M_AMD64
- #define STB__HKEY_CURRENT_USER 0x80000001ull
- #define STB__HKEY_LOCAL_MACHINE 0x80000002ull
- #else
- #define STB__HKEY_CURRENT_USER 0x80000001
- #define STB__HKEY_LOCAL_MACHINE 0x80000002
- #endif
- void *stb_reg_open(const char *mode, const char *where)
- {
- long res;
- HKEY base;
- HKEY zreg;
- if (!stb_stricmp(mode+1, "cu") || !stb_stricmp(mode+1, "hkcu"))
- base = (HKEY) STB__HKEY_CURRENT_USER;
- else if (!stb_stricmp(mode+1, "lm") || !stb_stricmp(mode+1, "hklm"))
- base = (HKEY) STB__HKEY_LOCAL_MACHINE;
- else
- return NULL;
- if (mode[0] == 'r')
- res = RegOpenKeyExA(base, where, 0, STB__REG_KEY_READ, &zreg);
- else if (mode[0] == 'w')
- res = RegCreateKeyExA(base, where, 0, NULL, STB__REG_OPTION_NON_VOLATILE, STB__REG_KEY_ALL_ACCESS, NULL, &zreg, NULL);
- else
- return NULL;
- return res ? NULL : zreg;
- }
- void stb_reg_close(void *reg)
- {
- RegCloseKey((HKEY) reg);
- }
- #define STB__REG_SZ 1
- #define STB__REG_BINARY 3
- #define STB__REG_DWORD 4
- int stb_reg_read(void *zreg, const char *str, void *data, unsigned long len)
- {
- unsigned long type;
- unsigned long alen = len;
- if (0 == RegQueryValueExA((HKEY) zreg, str, 0, &type, (unsigned char *) data, &len))
- if (type == STB__REG_BINARY || type == STB__REG_SZ || type == STB__REG_DWORD) {
- if (len < alen)
- *((char *) data + len) = 0;
- return 1;
- }
- return 0;
- }
- void stb_reg_write(void *zreg, const char *str, const void *data, unsigned long len)
- {
- if (zreg)
- RegSetValueExA((HKEY) zreg, str, 0, STB__REG_BINARY, (const unsigned char *) data, len);
- }
- int stb_reg_read_string(void *zreg, const char *str, char *data, int len)
- {
- if (!stb_reg_read(zreg, str, data, len)) return 0;
- data[len-1] = 0; // force a 0 at the end of the string no matter what
- return 1;
- }
- void stb_reg_write_string(void *zreg, const char *str, const char *data)
- {
- if (zreg)
- RegSetValueExA((HKEY) zreg, str, 0, STB__REG_SZ, (const unsigned char *) data, (int) strlen(data)+1);
- }
- #endif // STB_DEFINE
- #endif // _WIN32
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_cfg - This is like the registry, but the config info
- // is all stored in plain old files where we can
- // backup and restore them easily. The LOCATION of
- // the config files is gotten from... the registry!
- #ifndef STB_NO_STB_STRINGS
- typedef struct stb_cfg_st stb_cfg;
- STB_EXTERN stb_cfg * stb_cfg_open(char *config, const char *mode); // mode = "r", "w"
- STB_EXTERN void stb_cfg_close(stb_cfg *cfg);
- STB_EXTERN int stb_cfg_read(stb_cfg *cfg, char *key, void *value, int len);
- STB_EXTERN void stb_cfg_write(stb_cfg *cfg, char *key, void *value, int len);
- STB_EXTERN int stb_cfg_read_string(stb_cfg *cfg, char *key, char *value, int len);
- STB_EXTERN void stb_cfg_write_string(stb_cfg *cfg, char *key, char *value);
- STB_EXTERN int stb_cfg_delete(stb_cfg *cfg, char *key);
- STB_EXTERN void stb_cfg_set_directory(char *dir);
- #ifdef STB_DEFINE
- typedef struct
- {
- char *key;
- void *value;
- int value_len;
- } stb__cfg_item;
- struct stb_cfg_st
- {
- stb__cfg_item *data;
- char *loaded_file; // this needs to be freed
- FILE *f; // write the data to this file on close
- };
- static const char *stb__cfg_sig = "sTbCoNfIg!\0\0";
- static char stb__cfg_dir[512];
- STB_EXTERN void stb_cfg_set_directory(char *dir)
- {
- stb_p_strcpy_s(stb__cfg_dir, sizeof(stb__cfg_dir), dir);
- }
- STB_EXTERN stb_cfg * stb_cfg_open(char *config, const char *mode)
- {
- size_t len;
- stb_cfg *z;
- char file[512];
- if (mode[0] != 'r' && mode[0] != 'w') return NULL;
- if (!stb__cfg_dir[0]) {
- #ifdef _WIN32
- stb_p_strcpy_s(stb__cfg_dir, sizeof(stb__cfg_dir), "c:/stb");
- #else
- strcpy(stb__cfg_dir, "~/.stbconfig");
- #endif
- #ifdef STB_HAS_REGISTRY
- {
- void *reg = stb_reg_open("rHKLM", "Software\\SilverSpaceship\\stb");
- if (reg) {
- stb_reg_read_string(reg, "config_dir", stb__cfg_dir, sizeof(stb__cfg_dir));
- stb_reg_close(reg);
- }
- }
- #endif
- }
- stb_p_sprintf(file stb_p_size(sizeof(file)), "%s/%s.cfg", stb__cfg_dir, config);
- z = (stb_cfg *) stb_malloc(0, sizeof(*z));
- z->data = NULL;
- z->loaded_file = stb_filec(file, &len);
- if (z->loaded_file) {
- char *s = z->loaded_file;
- if (!memcmp(s, stb__cfg_sig, 12)) {
- char *s = z->loaded_file + 12;
- while (s < z->loaded_file + len) {
- stb__cfg_item a;
- int n = *(stb_int16 *) s;
- a.key = s+2;
- s = s+2 + n;
- a.value_len = *(int *) s;
- s += 4;
- a.value = s;
- s += a.value_len;
- stb_arr_push(z->data, a);
- }
- assert(s == z->loaded_file + len);
- }
- }
- if (mode[0] == 'w')
- z->f = stb_p_fopen(file, "wb");
- else
- z->f = NULL;
- return z;
- }
- void stb_cfg_close(stb_cfg *z)
- {
- if (z->f) {
- int i;
- // write the file out
- fwrite(stb__cfg_sig, 12, 1, z->f);
- for (i=0; i < stb_arr_len(z->data); ++i) {
- stb_int16 n = (stb_int16) strlen(z->data[i].key)+1;
- fwrite(&n, 2, 1, z->f);
- fwrite(z->data[i].key, n, 1, z->f);
- fwrite(&z->data[i].value_len, 4, 1, z->f);
- fwrite(z->data[i].value, z->data[i].value_len, 1, z->f);
- }
- fclose(z->f);
- }
- stb_arr_free(z->data);
- stb_free(z);
- }
- int stb_cfg_read(stb_cfg *z, char *key, void *value, int len)
- {
- int i;
- for (i=0; i < stb_arr_len(z->data); ++i) {
- if (!stb_stricmp(z->data[i].key, key)) {
- int n = stb_min(len, z->data[i].value_len);
- memcpy(value, z->data[i].value, n);
- if (n < len)
- *((char *) value + n) = 0;
- return 1;
- }
- }
- return 0;
- }
- void stb_cfg_write(stb_cfg *z, char *key, void *value, int len)
- {
- int i;
- for (i=0; i < stb_arr_len(z->data); ++i)
- if (!stb_stricmp(z->data[i].key, key))
- break;
- if (i == stb_arr_len(z->data)) {
- stb__cfg_item p;
- p.key = stb_strdup(key, z);
- p.value = NULL;
- p.value_len = 0;
- stb_arr_push(z->data, p);
- }
- z->data[i].value = stb_malloc(z, len);
- z->data[i].value_len = len;
- memcpy(z->data[i].value, value, len);
- }
- int stb_cfg_delete(stb_cfg *z, char *key)
- {
- int i;
- for (i=0; i < stb_arr_len(z->data); ++i)
- if (!stb_stricmp(z->data[i].key, key)) {
- stb_arr_fastdelete(z->data, i);
- return 1;
- }
- return 0;
- }
- int stb_cfg_read_string(stb_cfg *z, char *key, char *value, int len)
- {
- if (!stb_cfg_read(z, key, value, len)) return 0;
- value[len-1] = 0;
- return 1;
- }
- void stb_cfg_write_string(stb_cfg *z, char *key, char *value)
- {
- stb_cfg_write(z, key, value, (int) strlen(value)+1);
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_dirtree - load a description of a directory tree
- // uses a cache and stat()s the directories for changes
- // MUCH faster on NTFS, _wrong_ on FAT32, so should
- // ignore the db on FAT32
- #ifdef _WIN32
- typedef struct
- {
- char * path; // full path from passed-in root
- time_t last_modified;
- int num_files;
- int flag;
- } stb_dirtree_dir;
- typedef struct
- {
- char *name; // name relative to path
- int dir; // index into dirs[] array
- stb_int64 size; // size, max 4GB
- time_t last_modified;
- int flag;
- } stb_dirtree_file;
- typedef struct
- {
- stb_dirtree_dir *dirs;
- stb_dirtree_file *files;
- // internal use
- void * string_pool; // used to free data en masse
- } stb_dirtree;
- extern void stb_dirtree_free ( stb_dirtree *d );
- extern stb_dirtree *stb_dirtree_get ( char *dir);
- extern stb_dirtree *stb_dirtree_get_dir ( char *dir, char *cache_dir);
- extern stb_dirtree *stb_dirtree_get_with_file ( char *dir, char *cache_file);
- // get a list of all the files recursively underneath 'dir'
- //
- // cache_file is used to store a copy of the directory tree to speed up
- // later calls. It must be unique to 'dir' and the current working
- // directory! Otherwise who knows what will happen (a good solution
- // is to put it _in_ dir, but this API doesn't force that).
- //
- // Also, it might be possible to break this if you have two different processes
- // do a call to stb_dirtree_get() with the same cache file at about the same
- // time, but I _think_ it might just work.
- // i needed to build an identical data structure representing the state of
- // a mirrored copy WITHOUT bothering to rescan it (i.e. we're mirroring to
- // it WITHOUT scanning it, e.g. it's over the net), so this requires access
- // to all of the innards.
- extern void stb_dirtree_db_add_dir(stb_dirtree *active, char *path, time_t last);
- extern void stb_dirtree_db_add_file(stb_dirtree *active, char *name, int dir, stb_int64 size, time_t last);
- extern void stb_dirtree_db_read(stb_dirtree *target, char *filename, char *dir);
- extern void stb_dirtree_db_write(stb_dirtree *target, char *filename, char *dir);
- #ifdef STB_DEFINE
- static void stb__dirtree_add_dir(char *path, time_t last, stb_dirtree *active)
- {
- stb_dirtree_dir d;
- d.last_modified = last;
- d.num_files = 0;
- d.path = stb_strdup(path, active->string_pool);
- stb_arr_push(active->dirs, d);
- }
- static void stb__dirtree_add_file(char *name, int dir, stb_int64 size, time_t last, stb_dirtree *active)
- {
- stb_dirtree_file f;
- f.dir = dir;
- f.size = size;
- f.last_modified = last;
- f.name = stb_strdup(name, active->string_pool);
- ++active->dirs[dir].num_files;
- stb_arr_push(active->files, f);
- }
- // version 02 supports > 4GB files
- static char stb__signature[12] = { 's', 'T', 'b', 'D', 'i', 'R', 't', 'R', 'e', 'E', '0', '2' };
- static void stb__dirtree_save_db(char *filename, stb_dirtree *data, char *root)
- {
- int i, num_dirs_final=0, num_files_final;
- char *info = root ? root : (char*)"";
- int *remap;
- FILE *f = stb_p_fopen(filename, "wb");
- if (!f) return;
- fwrite(stb__signature, sizeof(stb__signature), 1, f);
- fwrite(info, strlen(info)+1, 1, f);
- // need to be slightly tricky and not write out NULLed directories, nor the root
- // build remapping table of all dirs we'll be writing out
- remap = (int *) malloc(sizeof(remap[0]) * stb_arr_len(data->dirs));
- for (i=0; i < stb_arr_len(data->dirs); ++i) {
- if (data->dirs[i].path == NULL || (root && 0==stb_stricmp(data->dirs[i].path, root))) {
- remap[i] = -1;
- } else {
- remap[i] = num_dirs_final++;
- }
- }
- fwrite(&num_dirs_final, 4, 1, f);
- for (i=0; i < stb_arr_len(data->dirs); ++i) {
- if (remap[i] >= 0) {
- fwrite(&data->dirs[i].last_modified, 4, 1, f);
- stb_fput_string(f, data->dirs[i].path);
- }
- }
- num_files_final = 0;
- for (i=0; i < stb_arr_len(data->files); ++i)
- if (remap[data->files[i].dir] >= 0 && data->files[i].name)
- ++num_files_final;
- fwrite(&num_files_final, 4, 1, f);
- for (i=0; i < stb_arr_len(data->files); ++i) {
- if (remap[data->files[i].dir] >= 0 && data->files[i].name) {
- stb_fput_ranged(f, remap[data->files[i].dir], 0, num_dirs_final);
- stb_fput_varlen64(f, data->files[i].size);
- fwrite(&data->files[i].last_modified, 4, 1, f);
- stb_fput_string(f, data->files[i].name);
- }
- }
- fclose(f);
- }
- // note: stomps any existing data, rather than appending
- static void stb__dirtree_load_db(char *filename, stb_dirtree *data, char *dir)
- {
- char sig[2048];
- int i,n;
- FILE *f = stb_p_fopen(filename, "rb");
- if (!f) return;
- data->string_pool = stb_malloc(0,1);
- fread(sig, sizeof(stb__signature), 1, f);
- if (memcmp(stb__signature, sig, sizeof(stb__signature))) { fclose(f); return; }
- if (!fread(sig, strlen(dir)+1, 1, f)) { fclose(f); return; }
- if (stb_stricmp(sig,dir)) { fclose(f); return; }
- // we can just read them straight in, because they're guaranteed to be valid
- fread(&n, 4, 1, f);
- stb_arr_setlen(data->dirs, n);
- for(i=0; i < stb_arr_len(data->dirs); ++i) {
- fread(&data->dirs[i].last_modified, 4, 1, f);
- data->dirs[i].path = stb_fget_string(f, data->string_pool);
- if (data->dirs[i].path == NULL) goto bail;
- }
- fread(&n, 4, 1, f);
- stb_arr_setlen(data->files, n);
- for (i=0; i < stb_arr_len(data->files); ++i) {
- data->files[i].dir = stb_fget_ranged(f, 0, stb_arr_len(data->dirs));
- data->files[i].size = stb_fget_varlen64(f);
- fread(&data->files[i].last_modified, 4, 1, f);
- data->files[i].name = stb_fget_string(f, data->string_pool);
- if (data->files[i].name == NULL) goto bail;
- }
- if (0) {
- bail:
- stb_arr_free(data->dirs);
- stb_arr_free(data->files);
- }
- fclose(f);
- }
- FILE *hlog;
- static int stb__dircount, stb__dircount_mask, stb__showfile;
- static void stb__dirtree_scandir(char *path, time_t last_time, stb_dirtree *active)
- {
- // this is dumb depth first; theoretically it might be faster
- // to fully traverse each directory before visiting its children,
- // but it's complicated and didn't seem like a gain in the test app
- int n;
- struct _wfinddatai64_t c_file;
- long hFile;
- stb__wchar full_path[1024];
- int has_slash;
- if (stb__showfile) printf("<");
- has_slash = (path[0] && path[strlen(path)-1] == '/');
- // @TODO: do this concatenation without using swprintf to avoid this mess:
- #if (defined(_MSC_VER) && _MSC_VER < 1400) // || (defined(__clang__))
- // confusingly, Windows Kits\10 needs to go down this path?!?
- // except now it doesn't, I don't know what changed
- if (has_slash)
- swprintf(full_path, L"%s*", stb__from_utf8(path));
- else
- swprintf(full_path, L"%s/*", stb__from_utf8(path));
- #else
- if (has_slash)
- swprintf((wchar_t *) full_path, (size_t) 1024, L"%s*", (wchar_t *) stb__from_utf8(path));
- else
- swprintf((wchar_t *) full_path, (size_t) 1024, L"%s/*", (wchar_t *) stb__from_utf8(path));
- #endif
- // it's possible this directory is already present: that means it was in the
- // cache, but its parent wasn't... in that case, we're done with it
- if (stb__showfile) printf("C[%d]", stb_arr_len(active->dirs));
- for (n=0; n < stb_arr_len(active->dirs); ++n)
- if (0 == stb_stricmp(active->dirs[n].path, path)) {
- if (stb__showfile) printf("D");
- return;
- }
- if (stb__showfile) printf("E");
- // otherwise, we need to add it
- stb__dirtree_add_dir(path, last_time, active);
- n = stb_arr_lastn(active->dirs);
- if (stb__showfile) printf("[");
- if( (hFile = (long) _wfindfirsti64( (wchar_t *) full_path, &c_file )) != -1L ) {
- do {
- if (stb__showfile) printf(")");
- if (c_file.attrib & _A_SUBDIR) {
- // ignore subdirectories starting with '.', e.g. "." and ".."
- if (c_file.name[0] != '.') {
- char *new_path = (char *) full_path;
- char *temp = stb__to_utf8((stb__wchar *) c_file.name);
- if (has_slash)
- stb_p_sprintf(new_path stb_p_size(sizeof(full_path)), "%s%s", path, temp);
- else
- stb_p_sprintf(new_path stb_p_size(sizeof(full_path)), "%s/%s", path, temp);
- if (stb__dircount_mask) {
- ++stb__dircount;
- if (!(stb__dircount & stb__dircount_mask)) {
- char dummy_path[128], *pad;
- stb_strncpy(dummy_path, new_path, sizeof(dummy_path)-1);
- if (strlen(dummy_path) > 96) {
- stb_p_strcpy_s(dummy_path+96/2-1,128, "...");
- stb_p_strcpy_s(dummy_path+96/2+2,128, new_path + strlen(new_path)-96/2+2);
- }
- pad = dummy_path + strlen(dummy_path);
- while (pad < dummy_path+98)
- *pad++ = ' ';
- *pad = 0;
- printf("%s\r", dummy_path);
- #if 0
- if (hlog == 0) {
- hlog = stb_p_fopen("c:/x/temp.log", "w");
- fprintf(hlog, "%s\n", dummy_path);
- }
- #endif
- }
- }
- stb__dirtree_scandir(new_path, c_file.time_write, active);
- }
- } else {
- char *temp = stb__to_utf8((stb__wchar *) c_file.name);
- stb__dirtree_add_file(temp, n, c_file.size, c_file.time_write, active);
- }
- if (stb__showfile) printf("(");
- } while( _wfindnexti64( hFile, &c_file ) == 0 );
- if (stb__showfile) printf("]");
- _findclose( hFile );
- }
- if (stb__showfile) printf(">\n");
- }
- // scan the database and see if it's all valid
- static int stb__dirtree_update_db(stb_dirtree *db, stb_dirtree *active)
- {
- int changes_detected = STB_FALSE;
- int i;
- int *remap;
- int *rescan=NULL;
- remap = (int *) malloc(sizeof(remap[0]) * stb_arr_len(db->dirs));
- memset(remap, 0, sizeof(remap[0]) * stb_arr_len(db->dirs));
- rescan = NULL;
- for (i=0; i < stb_arr_len(db->dirs); ++i) {
- struct _stat info;
- if (stb__dircount_mask) {
- ++stb__dircount;
- if (!(stb__dircount & stb__dircount_mask)) {
- printf(".");
- }
- }
- if (0 == _stat(db->dirs[i].path, &info)) {
- if (info.st_mode & _S_IFDIR) {
- // it's still a directory, as expected
- int n = abs((int) (info.st_mtime - db->dirs[i].last_modified));
- if (n > 1 && n != 3600) { // the 3600 is a hack because sometimes this jumps for no apparent reason, even when no time zone or DST issues are at play
- // it's changed! force a rescan
- // we don't want to scan it until we've stat()d its
- // subdirs, though, so we queue it
- if (stb__showfile) printf("Changed: %s - %08x:%08x\n", db->dirs[i].path, (unsigned int) db->dirs[i].last_modified, (unsigned int) info.st_mtime);
- stb_arr_push(rescan, i);
- // update the last_mod time
- db->dirs[i].last_modified = info.st_mtime;
- // ignore existing files in this dir
- remap[i] = -1;
- changes_detected = STB_TRUE;
- } else {
- // it hasn't changed, just copy it through unchanged
- stb__dirtree_add_dir(db->dirs[i].path, db->dirs[i].last_modified, active);
- remap[i] = stb_arr_lastn(active->dirs);
- }
- } else {
- // this path used to refer to a directory, but now it's a file!
- // assume that the parent directory is going to be forced to rescan anyway
- goto delete_entry;
- }
- } else {
- delete_entry:
- // directory no longer exists, so don't copy it
- // we don't free it because it's in the string pool now
- db->dirs[i].path = NULL;
- remap[i] = -1;
- changes_detected = STB_TRUE;
- }
- }
- // at this point, we have:
- //
- // <rescan> holds a list of directory indices that need to be scanned due to being out of date
- // <remap> holds the directory index in <active> for each dir in <db>, if it exists; -1 if not
- // directories in <rescan> are not in <active> yet
- // so we can go ahead and remap all the known files right now
- for (i=0; i < stb_arr_len(db->files); ++i) {
- int dir = db->files[i].dir;
- if (remap[dir] >= 0) {
- stb__dirtree_add_file(db->files[i].name, remap[dir], db->files[i].size, db->files[i].last_modified, active);
- }
- }
- // at this point we're done with db->files, and done with remap
- free(remap);
- // now scan those directories using the standard scan
- for (i=0; i < stb_arr_len(rescan); ++i) {
- int z = rescan[i];
- stb__dirtree_scandir(db->dirs[z].path, db->dirs[z].last_modified, active);
- }
- stb_arr_free(rescan);
- return changes_detected;
- }
- static void stb__dirtree_free_raw(stb_dirtree *d)
- {
- stb_free(d->string_pool);
- stb_arr_free(d->dirs);
- stb_arr_free(d->files);
- }
- stb_dirtree *stb_dirtree_get_with_file(char *dir, char *cache_file)
- {
- stb_dirtree *output = (stb_dirtree *) malloc(sizeof(*output));
- stb_dirtree db,active;
- int prev_dir_count, cache_mismatch;
- char *stripped_dir; // store the directory name without a trailing '/' or '\\'
- // load the database of last-known state on disk
- db.string_pool = NULL;
- db.files = NULL;
- db.dirs = NULL;
- stripped_dir = stb_strip_final_slash(stb_p_strdup(dir));
- if (cache_file != NULL)
- stb__dirtree_load_db(cache_file, &db, stripped_dir);
- else if (stb__showfile)
- printf("No cache file\n");
- active.files = NULL;
- active.dirs = NULL;
- active.string_pool = stb_malloc(0,1); // @TODO: share string pools between both?
- // check all the directories in the database; make note if
- // anything we scanned had changed, and rescan those things
- cache_mismatch = stb__dirtree_update_db(&db, &active);
- // check the root tree
- prev_dir_count = stb_arr_len(active.dirs); // record how many directories we've seen
- stb__dirtree_scandir(stripped_dir, 0, &active); // no last_modified time available for root
- if (stb__dircount_mask)
- printf(" \r");
- // done with the DB; write it back out if any changes, i.e. either
- // 1. any inconsistency found between cached information and actual disk
- // or 2. if scanning the root found any new directories--which we detect because
- // more than one directory got added to the active db during that scan
- if (cache_mismatch || stb_arr_len(active.dirs) > prev_dir_count+1)
- stb__dirtree_save_db(cache_file, &active, stripped_dir);
- free(stripped_dir);
- stb__dirtree_free_raw(&db);
- *output = active;
- return output;
- }
- stb_dirtree *stb_dirtree_get_dir(char *dir, char *cache_dir)
- {
- int i;
- stb_uint8 sha[20];
- char dir_lower[1024];
- char cache_file[1024],*s;
- if (cache_dir == NULL)
- return stb_dirtree_get_with_file(dir, NULL);
- stb_p_strcpy_s(dir_lower, sizeof(dir_lower), dir);
- stb_tolower(dir_lower);
- stb_sha1(sha, (unsigned char *) dir_lower, (unsigned int) strlen(dir_lower));
- stb_p_strcpy_s(cache_file, sizeof(cache_file), cache_dir);
- s = cache_file + strlen(cache_file);
- if (s[-1] != '/' && s[-1] != '\\') *s++ = '/';
- stb_p_strcpy_s(s, sizeof(cache_file), "dirtree_");
- s += strlen(s);
- for (i=0; i < 8; ++i) {
- char *hex = (char*)"0123456789abcdef";
- stb_uint z = sha[i];
- *s++ = hex[z >> 4];
- *s++ = hex[z & 15];
- }
- stb_p_strcpy_s(s, sizeof(cache_file), ".bin");
- return stb_dirtree_get_with_file(dir, cache_file);
- }
- stb_dirtree *stb_dirtree_get(char *dir)
- {
- char cache_dir[256];
- stb_p_strcpy_s(cache_dir, sizeof(cache_dir), "c:/bindata");
- #ifdef STB_HAS_REGISTRY
- {
- void *reg = stb_reg_open("rHKLM", "Software\\SilverSpaceship\\stb");
- if (reg) {
- stb_reg_read(reg, "dirtree", cache_dir, sizeof(cache_dir));
- stb_reg_close(reg);
- }
- }
- #endif
- return stb_dirtree_get_dir(dir, cache_dir);
- }
- void stb_dirtree_free(stb_dirtree *d)
- {
- stb__dirtree_free_raw(d);
- free(d);
- }
- void stb_dirtree_db_add_dir(stb_dirtree *active, char *path, time_t last)
- {
- stb__dirtree_add_dir(path, last, active);
- }
- void stb_dirtree_db_add_file(stb_dirtree *active, char *name, int dir, stb_int64 size, time_t last)
- {
- stb__dirtree_add_file(name, dir, size, last, active);
- }
- void stb_dirtree_db_read(stb_dirtree *target, char *filename, char *dir)
- {
- char *s = stb_strip_final_slash(stb_p_strdup(dir));
- target->dirs = 0;
- target->files = 0;
- target->string_pool = 0;
- stb__dirtree_load_db(filename, target, s);
- free(s);
- }
- void stb_dirtree_db_write(stb_dirtree *target, char *filename, char *dir)
- {
- stb__dirtree_save_db(filename, target, 0); // don't strip out any directories
- }
- #endif // STB_DEFINE
- #endif // _WIN32
- #endif // STB_NO_STB_STRINGS
- //////////////////////////////////////////////////////////////////////////////
- //
- // STB_MALLOC_WRAPPER
- //
- // you can use the wrapper functions with your own malloc wrapper,
- // or define STB_MALLOC_WRAPPER project-wide to have
- // malloc/free/realloc/strdup all get vectored to it
- // this has too many very specific error messages you could google for and find in stb.h,
- // so don't use it if they don't want any stb.h-identifiable strings
- #if defined(STB_DEFINE) && !defined(STB_NO_STB_STRINGS)
- typedef struct
- {
- void *p;
- char *file;
- int line;
- size_t size;
- } stb_malloc_record;
- #ifndef STB_MALLOC_HISTORY_COUNT
- #define STB_MALLOC_HISTORY_COUNT 50 // 800 bytes
- #endif
- stb_malloc_record *stb__allocations;
- static int stb__alloc_size, stb__alloc_limit, stb__alloc_mask;
- int stb__alloc_count;
- stb_malloc_record stb__alloc_history[STB_MALLOC_HISTORY_COUNT];
- int stb__history_pos;
- static int stb__hashfind(void *p)
- {
- stb_uint32 h = stb_hashptr(p);
- int s,n = h & stb__alloc_mask;
- if (stb__allocations[n].p == p)
- return n;
- s = stb_rehash(h)|1;
- for(;;) {
- if (stb__allocations[n].p == NULL)
- return -1;
- n = (n+s) & stb__alloc_mask;
- if (stb__allocations[n].p == p)
- return n;
- }
- }
- size_t stb_wrapper_allocsize(void *p)
- {
- int n = stb__hashfind(p);
- if (n < 0) return 0;
- return stb__allocations[n].size;
- }
- static int stb__historyfind(void *p)
- {
- int n = stb__history_pos;
- int i;
- for (i=0; i < STB_MALLOC_HISTORY_COUNT; ++i) {
- if (--n < 0) n = STB_MALLOC_HISTORY_COUNT-1;
- if (stb__alloc_history[n].p == p)
- return n;
- }
- return -1;
- }
- static void stb__add_alloc(void *p, size_t sz, char *file, int line);
- static void stb__grow_alloc(void)
- {
- int i,old_num = stb__alloc_size;
- stb_malloc_record *old = stb__allocations;
- if (stb__alloc_size == 0)
- stb__alloc_size = 64;
- else
- stb__alloc_size *= 2;
- stb__allocations = (stb_malloc_record *) stb__realloc_raw(NULL, stb__alloc_size * sizeof(stb__allocations[0]));
- if (stb__allocations == NULL)
- stb_fatal("Internal error: couldn't grow malloc wrapper table");
- memset(stb__allocations, 0, stb__alloc_size * sizeof(stb__allocations[0]));
- stb__alloc_limit = (stb__alloc_size*3)>>2;
- stb__alloc_mask = stb__alloc_size-1;
- stb__alloc_count = 0;
- for (i=0; i < old_num; ++i)
- if (old[i].p > STB_DEL) {
- stb__add_alloc(old[i].p, old[i].size, old[i].file, old[i].line);
- assert(stb__hashfind(old[i].p) >= 0);
- }
- for (i=0; i < old_num; ++i)
- if (old[i].p > STB_DEL)
- assert(stb__hashfind(old[i].p) >= 0);
- stb__realloc_raw(old, 0);
- }
- static void stb__add_alloc(void *p, size_t sz, char *file, int line)
- {
- stb_uint32 h;
- int n;
- if (stb__alloc_count >= stb__alloc_limit)
- stb__grow_alloc();
- h = stb_hashptr(p);
- n = h & stb__alloc_mask;
- if (stb__allocations[n].p > STB_DEL) {
- int s = stb_rehash(h)|1;
- do {
- n = (n+s) & stb__alloc_mask;
- } while (stb__allocations[n].p > STB_DEL);
- }
- assert(stb__allocations[n].p == NULL || stb__allocations[n].p == STB_DEL);
- stb__allocations[n].p = p;
- stb__allocations[n].size = sz;
- stb__allocations[n].line = line;
- stb__allocations[n].file = file;
- ++stb__alloc_count;
- }
- static void stb__remove_alloc(int n, char *file, int line)
- {
- stb__alloc_history[stb__history_pos] = stb__allocations[n];
- stb__alloc_history[stb__history_pos].file = file;
- stb__alloc_history[stb__history_pos].line = line;
- if (++stb__history_pos == STB_MALLOC_HISTORY_COUNT)
- stb__history_pos = 0;
- stb__allocations[n].p = STB_DEL;
- --stb__alloc_count;
- }
- void stb_wrapper_malloc(void *p, size_t sz, char *file, int line)
- {
- if (!p) return;
- stb__add_alloc(p,sz,file,line);
- }
- void stb_wrapper_free(void *p, char *file, int line)
- {
- int n;
- if (p == NULL) return;
- n = stb__hashfind(p);
- if (n >= 0)
- stb__remove_alloc(n, file, line);
- else {
- // tried to free something we hadn't allocated!
- n = stb__historyfind(p);
- assert(0); /* NOTREACHED */
- if (n >= 0)
- stb_fatal("Attempted to free %d-byte block %p at %s:%d previously freed/realloced at %s:%d",
- stb__alloc_history[n].size, p,
- file, line,
- stb__alloc_history[n].file, stb__alloc_history[n].line);
- else
- stb_fatal("Attempted to free unknown block %p at %s:%d", p, file,line);
- }
- }
- void stb_wrapper_check(void *p)
- {
- int n;
- if (p == NULL) return;
- n = stb__hashfind(p);
- if (n >= 0) return;
- for (n=0; n < stb__alloc_size; ++n)
- if (stb__allocations[n].p == p)
- stb_fatal("Internal error: pointer %p was allocated, but hash search failed", p);
- // tried to free something that wasn't allocated!
- n = stb__historyfind(p);
- if (n >= 0)
- stb_fatal("Checked %d-byte block %p previously freed/realloced at %s:%d",
- stb__alloc_history[n].size, p,
- stb__alloc_history[n].file, stb__alloc_history[n].line);
- stb_fatal("Checked unknown block %p");
- }
- void stb_wrapper_realloc(void *p, void *q, size_t sz, char *file, int line)
- {
- int n;
- if (p == NULL) { stb_wrapper_malloc(q, sz, file, line); return; }
- if (q == NULL) return; // nothing happened
- n = stb__hashfind(p);
- if (n == -1) {
- // tried to free something we hadn't allocated!
- // this is weird, though, because we got past the realloc!
- n = stb__historyfind(p);
- assert(0); /* NOTREACHED */
- if (n >= 0)
- stb_fatal("Attempted to realloc %d-byte block %p at %s:%d previously freed/realloced at %s:%d",
- stb__alloc_history[n].size, p,
- file, line,
- stb__alloc_history[n].file, stb__alloc_history[n].line);
- else
- stb_fatal("Attempted to realloc unknown block %p at %s:%d", p, file,line);
- } else {
- if (q == p) {
- stb__allocations[n].size = sz;
- stb__allocations[n].file = file;
- stb__allocations[n].line = line;
- } else {
- stb__remove_alloc(n, file, line);
- stb__add_alloc(q,sz,file,line);
- }
- }
- }
- void stb_wrapper_listall(void (*func)(void *ptr, size_t sz, char *file, int line))
- {
- int i;
- for (i=0; i < stb__alloc_size; ++i)
- if (stb__allocations[i].p > STB_DEL)
- func(stb__allocations[i].p , stb__allocations[i].size,
- stb__allocations[i].file, stb__allocations[i].line);
- }
- void stb_wrapper_dump(char *filename)
- {
- int i;
- FILE *f = stb_p_fopen(filename, "w");
- if (!f) return;
- for (i=0; i < stb__alloc_size; ++i)
- if (stb__allocations[i].p > STB_DEL)
- fprintf(f, "%p %7d - %4d %s\n",
- stb__allocations[i].p , (int) stb__allocations[i].size,
- stb__allocations[i].line, stb__allocations[i].file);
- }
- #endif // STB_DEFINE
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_pointer_set
- //
- //
- // For data structures that support querying by key, data structure
- // classes always hand-wave away the issue of what to do if two entries
- // have the same key: basically, store a linked list of all the nodes
- // which have the same key (a LISP-style list).
- //
- // The thing is, it's not that trivial. If you have an O(log n)
- // lookup data structure, but then n/4 items have the same value,
- // you don't want to spend O(n) time scanning that list when
- // deleting an item if you already have a pointer to the item.
- // (You have to spend O(n) time enumerating all the items with
- // a given key, sure, and you can't accelerate deleting a particular
- // item if you only have the key, not a pointer to the item.)
- //
- // I'm going to call this data structure, whatever it turns out to
- // be, a "pointer set", because we don't store any associated data for
- // items in this data structure, we just answer the question of
- // whether an item is in it or not (it's effectively one bit per pointer).
- // Technically they don't have to be pointers; you could cast ints
- // to (void *) if you want, but you can't store 0 or 1 because of the
- // hash table.
- //
- // Since the fastest data structure we might want to add support for
- // identical-keys to is a hash table with O(1)-ish lookup time,
- // that means that the conceptual "linked list of all items with
- // the same indexed value" that we build needs to have the same
- // performance; that way when we index a table we think is arbitrary
- // ints, but in fact half of them are 0, we don't get screwed.
- //
- // Therefore, it needs to be a hash table, at least when it gets
- // large. On the other hand, when the data has totally arbitrary ints
- // or floats, there won't be many collisions, and we'll have tons of
- // 1-item bitmaps. That will be grossly inefficient as hash tables;
- // trade-off; the hash table is reasonably efficient per-item when
- // it's large, but not when it's small. So we need to do something
- // Judy-like and use different strategies depending on the size.
- //
- // Like Judy, we'll use the bottom bit to encode the strategy:
- //
- // bottom bits:
- // 00 - direct pointer
- // 01 - 4-item bucket (16 bytes, no length, NULLs)
- // 10 - N-item array
- // 11 - hash table
- typedef struct stb_ps stb_ps;
- STB_EXTERN int stb_ps_find (stb_ps *ps, void *value);
- STB_EXTERN stb_ps * stb_ps_add (stb_ps *ps, void *value);
- STB_EXTERN stb_ps * stb_ps_remove(stb_ps *ps, void *value);
- STB_EXTERN stb_ps * stb_ps_remove_any(stb_ps *ps, void **value);
- STB_EXTERN void stb_ps_delete(stb_ps *ps);
- STB_EXTERN int stb_ps_count (stb_ps *ps);
- STB_EXTERN stb_ps * stb_ps_copy (stb_ps *ps);
- STB_EXTERN int stb_ps_subset(stb_ps *bigger, stb_ps *smaller);
- STB_EXTERN int stb_ps_eq (stb_ps *p0, stb_ps *p1);
- STB_EXTERN void ** stb_ps_getlist (stb_ps *ps, int *count);
- STB_EXTERN int stb_ps_writelist(stb_ps *ps, void **list, int size );
- // enum and fastlist don't allocate storage, but you must consume the
- // list before there's any chance the data structure gets screwed up;
- STB_EXTERN int stb_ps_enum (stb_ps *ps, void *data,
- int (*func)(void *value, void*data) );
- STB_EXTERN void ** stb_ps_fastlist(stb_ps *ps, int *count);
- // result:
- // returns a list, *count is the length of that list,
- // but some entries of the list may be invalid;
- // test with 'stb_ps_fastlist_valid(x)'
- #define stb_ps_fastlist_valid(x) ((stb_uinta) (x) > 1)
- #ifdef STB_DEFINE
- enum
- {
- STB_ps_direct = 0,
- STB_ps_bucket = 1,
- STB_ps_array = 2,
- STB_ps_hash = 3,
- };
- #define STB_BUCKET_SIZE 4
- typedef struct
- {
- void *p[STB_BUCKET_SIZE];
- } stb_ps_bucket;
- #define GetBucket(p) ((stb_ps_bucket *) ((char *) (p) - STB_ps_bucket))
- #define EncodeBucket(p) ((stb_ps *) ((char *) (p) + STB_ps_bucket))
- static void stb_bucket_free(stb_ps_bucket *b)
- {
- free(b);
- }
- static stb_ps_bucket *stb_bucket_create2(void *v0, void *v1)
- {
- stb_ps_bucket *b = (stb_ps_bucket*) malloc(sizeof(*b));
- b->p[0] = v0;
- b->p[1] = v1;
- b->p[2] = NULL;
- b->p[3] = NULL;
- return b;
- }
- static stb_ps_bucket * stb_bucket_create3(void **v)
- {
- stb_ps_bucket *b = (stb_ps_bucket*) malloc(sizeof(*b));
- b->p[0] = v[0];
- b->p[1] = v[1];
- b->p[2] = v[2];
- b->p[3] = NULL;
- return b;
- }
- // could use stb_arr, but this will save us memory
- typedef struct
- {
- int count;
- void *p[1];
- } stb_ps_array;
- #define GetArray(p) ((stb_ps_array *) ((char *) (p) - STB_ps_array))
- #define EncodeArray(p) ((stb_ps *) ((char *) (p) + STB_ps_array))
- static int stb_ps_array_max = 13;
- typedef struct
- {
- int size, mask;
- int count, count_deletes;
- int grow_threshhold;
- int shrink_threshhold;
- int rehash_threshhold;
- int any_offset;
- void *table[1];
- } stb_ps_hash;
- #define GetHash(p) ((stb_ps_hash *) ((char *) (p) - STB_ps_hash))
- #define EncodeHash(p) ((stb_ps *) ((char *) (p) + STB_ps_hash))
- #define stb_ps_empty(v) (((stb_uint32) v) <= 1)
- static stb_ps_hash *stb_ps_makehash(int size, int old_size, void **old_data)
- {
- int i;
- stb_ps_hash *h = (stb_ps_hash *) malloc(sizeof(*h) + (size-1) * sizeof(h->table[0]));
- assert(stb_is_pow2(size));
- h->size = size;
- h->mask = size-1;
- h->shrink_threshhold = (int) (0.3f * size);
- h-> grow_threshhold = (int) (0.8f * size);
- h->rehash_threshhold = (int) (0.9f * size);
- h->count = 0;
- h->count_deletes = 0;
- h->any_offset = 0;
- memset(h->table, 0, size * sizeof(h->table[0]));
- for (i=0; i < old_size; ++i)
- if (!stb_ps_empty((size_t)old_data[i]))
- stb_ps_add(EncodeHash(h), old_data[i]);
- return h;
- }
- void stb_ps_delete(stb_ps *ps)
- {
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct: break;
- case STB_ps_bucket: stb_bucket_free(GetBucket(ps)); break;
- case STB_ps_array : free(GetArray(ps)); break;
- case STB_ps_hash : free(GetHash(ps)); break;
- }
- }
- stb_ps *stb_ps_copy(stb_ps *ps)
- {
- int i;
- // not a switch: order based on expected performance/power-law distribution
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct: return ps;
- case STB_ps_bucket: {
- stb_ps_bucket *n = (stb_ps_bucket *) malloc(sizeof(*n));
- *n = *GetBucket(ps);
- return EncodeBucket(n);
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- stb_ps_array *n = (stb_ps_array *) malloc(sizeof(*n) + stb_ps_array_max * sizeof(n->p[0]));
- n->count = a->count;
- for (i=0; i < a->count; ++i)
- n->p[i] = a->p[i];
- return EncodeArray(n);
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- stb_ps_hash *n = stb_ps_makehash(h->size, h->size, h->table);
- return EncodeHash(n);
- }
- }
- assert(0); /* NOTREACHED */
- return NULL;
- }
- int stb_ps_find(stb_ps *ps, void *value)
- {
- int i, code = 3 & (int)(size_t) ps;
- assert((3 & (int)(size_t) value) == STB_ps_direct);
- assert(stb_ps_fastlist_valid(value));
- // not a switch: order based on expected performance/power-law distribution
- if (code == STB_ps_direct)
- return value == ps;
- if (code == STB_ps_bucket) {
- stb_ps_bucket *b = GetBucket(ps);
- assert(STB_BUCKET_SIZE == 4);
- if (b->p[0] == value || b->p[1] == value ||
- b->p[2] == value || b->p[3] == value)
- return STB_TRUE;
- return STB_FALSE;
- }
- if (code == STB_ps_array) {
- stb_ps_array *a = GetArray(ps);
- for (i=0; i < a->count; ++i)
- if (a->p[i] == value)
- return STB_TRUE;
- return STB_FALSE;
- } else {
- stb_ps_hash *h = GetHash(ps);
- stb_uint32 hash = stb_hashptr(value);
- stb_uint32 s, n = hash & h->mask;
- void **t = h->table;
- if (t[n] == value) return STB_TRUE;
- if (t[n] == NULL) return STB_FALSE;
- s = stb_rehash(hash) | 1;
- do {
- n = (n + s) & h->mask;
- if (t[n] == value) return STB_TRUE;
- } while (t[n] != NULL);
- return STB_FALSE;
- }
- }
- stb_ps * stb_ps_add (stb_ps *ps, void *value)
- {
- #ifdef STB_DEBUG
- assert(!stb_ps_find(ps,value));
- #endif
- if (value == NULL) return ps; // ignore NULL adds to avoid bad breakage
- assert((3 & (int)(size_t) value) == STB_ps_direct);
- assert(stb_ps_fastlist_valid(value));
- assert(value != STB_DEL); // STB_DEL is less likely
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- if (ps == NULL) return (stb_ps *) value;
- return EncodeBucket(stb_bucket_create2(ps,value));
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- stb_ps_array *a;
- assert(STB_BUCKET_SIZE == 4);
- if (b->p[0] == NULL) { b->p[0] = value; return ps; }
- if (b->p[1] == NULL) { b->p[1] = value; return ps; }
- if (b->p[2] == NULL) { b->p[2] = value; return ps; }
- if (b->p[3] == NULL) { b->p[3] = value; return ps; }
- a = (stb_ps_array *) malloc(sizeof(*a) + 7 * sizeof(a->p[0])); // 8 slots, must be 2^k
- memcpy(a->p, b, sizeof(*b));
- a->p[4] = value;
- a->count = 5;
- stb_bucket_free(b);
- return EncodeArray(a);
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- if (a->count == stb_ps_array_max) {
- // promote from array to hash
- stb_ps_hash *h = stb_ps_makehash(2 << stb_log2_ceil(a->count), a->count, a->p);
- free(a);
- return stb_ps_add(EncodeHash(h), value);
- }
- // do we need to resize the array? the array doubles in size when it
- // crosses a power-of-two
- if ((a->count & (a->count-1))==0) {
- int newsize = a->count*2;
- // clamp newsize to max if:
- // 1. it's larger than max
- // 2. newsize*1.5 is larger than max (to avoid extra resizing)
- if (newsize + a->count > stb_ps_array_max)
- newsize = stb_ps_array_max;
- a = (stb_ps_array *) realloc(a, sizeof(*a) + (newsize-1) * sizeof(a->p[0]));
- }
- a->p[a->count++] = value;
- return EncodeArray(a);
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- stb_uint32 hash = stb_hashptr(value);
- stb_uint32 n = hash & h->mask;
- void **t = h->table;
- // find first NULL or STB_DEL entry
- if (!stb_ps_empty((size_t)t[n])) {
- stb_uint32 s = stb_rehash(hash) | 1;
- do {
- n = (n + s) & h->mask;
- } while (!stb_ps_empty((size_t)t[n]));
- }
- if (t[n] == STB_DEL)
- -- h->count_deletes;
- t[n] = value;
- ++ h->count;
- if (h->count == h->grow_threshhold) {
- stb_ps_hash *h2 = stb_ps_makehash(h->size*2, h->size, t);
- free(h);
- return EncodeHash(h2);
- }
- if (h->count + h->count_deletes == h->rehash_threshhold) {
- stb_ps_hash *h2 = stb_ps_makehash(h->size, h->size, t);
- free(h);
- return EncodeHash(h2);
- }
- return ps;
- }
- }
- return NULL; /* NOTREACHED */
- }
- stb_ps *stb_ps_remove(stb_ps *ps, void *value)
- {
- #ifdef STB_DEBUG
- assert(stb_ps_find(ps, value));
- #endif
- assert((3 & (int)(size_t) value) == STB_ps_direct);
- if (value == NULL) return ps; // ignore NULL removes to avoid bad breakage
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- return ps == value ? NULL : ps;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- int count=0;
- assert(STB_BUCKET_SIZE == 4);
- if (b->p[0] == value) b->p[0] = NULL; else count += (b->p[0] != NULL);
- if (b->p[1] == value) b->p[1] = NULL; else count += (b->p[1] != NULL);
- if (b->p[2] == value) b->p[2] = NULL; else count += (b->p[2] != NULL);
- if (b->p[3] == value) b->p[3] = NULL; else count += (b->p[3] != NULL);
- if (count == 1) { // shrink bucket at size 1
- value = b->p[0];
- if (value == NULL) value = b->p[1];
- if (value == NULL) value = b->p[2];
- if (value == NULL) value = b->p[3];
- assert(value != NULL);
- stb_bucket_free(b);
- return (stb_ps *) value; // return STB_ps_direct of value
- }
- return ps;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- int i;
- for (i=0; i < a->count; ++i) {
- if (a->p[i] == value) {
- a->p[i] = a->p[--a->count];
- if (a->count == 3) { // shrink to bucket!
- stb_ps_bucket *b = stb_bucket_create3(a->p);
- free(a);
- return EncodeBucket(b);
- }
- return ps;
- }
- }
- return ps;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- stb_uint32 hash = stb_hashptr(value);
- stb_uint32 s, n = hash & h->mask;
- void **t = h->table;
- if (t[n] != value) {
- s = stb_rehash(hash) | 1;
- do {
- n = (n + s) & h->mask;
- } while (t[n] != value);
- }
- t[n] = STB_DEL;
- -- h->count;
- ++ h->count_deletes;
- // should we shrink down to an array?
- if (h->count < stb_ps_array_max) {
- int n = 1 << stb_log2_floor(stb_ps_array_max);
- if (h->count < n) {
- stb_ps_array *a = (stb_ps_array *) malloc(sizeof(*a) + (n-1) * sizeof(a->p[0]));
- int i,j=0;
- for (i=0; i < h->size; ++i)
- if (!stb_ps_empty((size_t)t[i]))
- a->p[j++] = t[i];
- assert(j == h->count);
- a->count = j;
- free(h);
- return EncodeArray(a);
- }
- }
- if (h->count == h->shrink_threshhold) {
- stb_ps_hash *h2 = stb_ps_makehash(h->size >> 1, h->size, t);
- free(h);
- return EncodeHash(h2);
- }
- return ps;
- }
- }
- return ps; /* NOTREACHED */
- }
- stb_ps *stb_ps_remove_any(stb_ps *ps, void **value)
- {
- assert(ps != NULL);
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- *value = ps;
- return NULL;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- int count=0, slast=0, last=0;
- assert(STB_BUCKET_SIZE == 4);
- if (b->p[0]) { ++count; last = 0; }
- if (b->p[1]) { ++count; slast = last; last = 1; }
- if (b->p[2]) { ++count; slast = last; last = 2; }
- if (b->p[3]) { ++count; slast = last; last = 3; }
- *value = b->p[last];
- b->p[last] = 0;
- if (count == 2) {
- void *leftover = b->p[slast]; // second to last
- stb_bucket_free(b);
- return (stb_ps *) leftover;
- }
- return ps;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- *value = a->p[a->count-1];
- if (a->count == 4)
- return stb_ps_remove(ps, *value);
- --a->count;
- return ps;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- void **t = h->table;
- stb_uint32 n = h->any_offset;
- while (stb_ps_empty((size_t)t[n]))
- n = (n + 1) & h->mask;
- *value = t[n];
- h->any_offset = (n+1) & h->mask;
- // check if we need to skip down to the previous type
- if (h->count-1 < stb_ps_array_max || h->count-1 == h->shrink_threshhold)
- return stb_ps_remove(ps, *value);
- t[n] = STB_DEL;
- -- h->count;
- ++ h->count_deletes;
- return ps;
- }
- }
- return ps; /* NOTREACHED */
- }
- void ** stb_ps_getlist(stb_ps *ps, int *count)
- {
- int i,n=0;
- void **p = NULL;
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- if (ps == NULL) { *count = 0; return NULL; }
- p = (void **) malloc(sizeof(*p) * 1);
- p[0] = ps;
- *count = 1;
- return p;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- p = (void **) malloc(sizeof(*p) * STB_BUCKET_SIZE);
- for (i=0; i < STB_BUCKET_SIZE; ++i)
- if (b->p[i] != NULL)
- p[n++] = b->p[i];
- break;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- p = (void **) malloc(sizeof(*p) * a->count);
- memcpy(p, a->p, sizeof(*p) * a->count);
- *count = a->count;
- return p;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- p = (void **) malloc(sizeof(*p) * h->count);
- for (i=0; i < h->size; ++i)
- if (!stb_ps_empty((size_t)h->table[i]))
- p[n++] = h->table[i];
- break;
- }
- }
- *count = n;
- return p;
- }
- int stb_ps_writelist(stb_ps *ps, void **list, int size )
- {
- int i,n=0;
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- if (ps == NULL || size <= 0) return 0;
- list[0] = ps;
- return 1;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- for (i=0; i < STB_BUCKET_SIZE; ++i)
- if (b->p[i] != NULL && n < size)
- list[n++] = b->p[i];
- return n;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- n = stb_min(size, a->count);
- memcpy(list, a->p, sizeof(*list) * n);
- return n;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- if (size <= 0) return 0;
- for (i=0; i < h->count; ++i) {
- if (!stb_ps_empty((size_t)h->table[i])) {
- list[n++] = h->table[i];
- if (n == size) break;
- }
- }
- return n;
- }
- }
- return 0; /* NOTREACHED */
- }
- int stb_ps_enum(stb_ps *ps, void *data, int (*func)(void *value, void *data))
- {
- int i;
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- if (ps == NULL) return STB_TRUE;
- return func(ps, data);
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- for (i=0; i < STB_BUCKET_SIZE; ++i)
- if (b->p[i] != NULL)
- if (!func(b->p[i], data))
- return STB_FALSE;
- return STB_TRUE;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- for (i=0; i < a->count; ++i)
- if (!func(a->p[i], data))
- return STB_FALSE;
- return STB_TRUE;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- for (i=0; i < h->count; ++i)
- if (!stb_ps_empty((size_t)h->table[i]))
- if (!func(h->table[i], data))
- return STB_FALSE;
- return STB_TRUE;
- }
- }
- return STB_TRUE; /* NOTREACHED */
- }
- int stb_ps_count (stb_ps *ps)
- {
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- return ps != NULL;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- return (b->p[0] != NULL) + (b->p[1] != NULL) +
- (b->p[2] != NULL) + (b->p[3] != NULL);
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- return a->count;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- return h->count;
- }
- }
- return 0;
- }
- void ** stb_ps_fastlist(stb_ps *ps, int *count)
- {
- static void *storage;
- switch (3 & (int)(size_t) ps) {
- case STB_ps_direct:
- if (ps == NULL) { *count = 0; return NULL; }
- storage = ps;
- *count = 1;
- return &storage;
- case STB_ps_bucket: {
- stb_ps_bucket *b = GetBucket(ps);
- *count = STB_BUCKET_SIZE;
- return b->p;
- }
- case STB_ps_array: {
- stb_ps_array *a = GetArray(ps);
- *count = a->count;
- return a->p;
- }
- case STB_ps_hash: {
- stb_ps_hash *h = GetHash(ps);
- *count = h->size;
- return h->table;
- }
- }
- return NULL; /* NOTREACHED */
- }
- int stb_ps_subset(stb_ps *bigger, stb_ps *smaller)
- {
- int i, listlen;
- void **list = stb_ps_fastlist(smaller, &listlen);
- for(i=0; i < listlen; ++i)
- if (stb_ps_fastlist_valid(list[i]))
- if (!stb_ps_find(bigger, list[i]))
- return 0;
- return 1;
- }
- int stb_ps_eq(stb_ps *p0, stb_ps *p1)
- {
- if (stb_ps_count(p0) != stb_ps_count(p1))
- return 0;
- return stb_ps_subset(p0, p1);
- }
- #undef GetBucket
- #undef GetArray
- #undef GetHash
- #undef EncodeBucket
- #undef EncodeArray
- #undef EncodeHash
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Random Numbers via Meresenne Twister or LCG
- //
- STB_EXTERN unsigned int stb_srandLCG(unsigned int seed);
- STB_EXTERN unsigned int stb_randLCG(void);
- STB_EXTERN double stb_frandLCG(void);
- STB_EXTERN void stb_srand(unsigned int seed);
- STB_EXTERN unsigned int stb_rand(void);
- STB_EXTERN double stb_frand(void);
- STB_EXTERN void stb_shuffle(void *p, size_t n, size_t sz,
- unsigned int seed);
- STB_EXTERN void stb_reverse(void *p, size_t n, size_t sz);
- STB_EXTERN unsigned int stb_randLCG_explicit(unsigned int seed);
- #define stb_rand_define(x,y) \
- \
- unsigned int x(void) \
- { \
- static unsigned int stb__rand = y; \
- stb__rand = stb__rand * 2147001325 + 715136305; /* BCPL */ \
- return 0x31415926 ^ ((stb__rand >> 16) + (stb__rand << 16)); \
- }
- #ifdef STB_DEFINE
- unsigned int stb_randLCG_explicit(unsigned int seed)
- {
- return seed * 2147001325 + 715136305;
- }
- static unsigned int stb__rand_seed=0;
- unsigned int stb_srandLCG(unsigned int seed)
- {
- unsigned int previous = stb__rand_seed;
- stb__rand_seed = seed;
- return previous;
- }
- unsigned int stb_randLCG(void)
- {
- stb__rand_seed = stb__rand_seed * 2147001325 + 715136305; // BCPL generator
- // shuffle non-random bits to the middle, and xor to decorrelate with seed
- return 0x31415926 ^ ((stb__rand_seed >> 16) + (stb__rand_seed << 16));
- }
- double stb_frandLCG(void)
- {
- return stb_randLCG() / ((double) (1 << 16) * (1 << 16));
- }
- void stb_shuffle(void *p, size_t n, size_t sz, unsigned int seed)
- {
- char *a;
- unsigned int old_seed;
- int i;
- if (seed)
- old_seed = stb_srandLCG(seed);
- a = (char *) p + (n-1) * sz;
- for (i=(int) n; i > 1; --i) {
- int j = stb_randLCG() % i;
- stb_swap(a, (char *) p + j * sz, sz);
- a -= sz;
- }
- if (seed)
- stb_srandLCG(old_seed);
- }
- void stb_reverse(void *p, size_t n, size_t sz)
- {
- size_t i,j = n-1;
- for (i=0; i < j; ++i,--j) {
- stb_swap((char *) p + i * sz, (char *) p + j * sz, sz);
- }
- }
- // public domain Mersenne Twister by Michael Brundage
- #define STB__MT_LEN 624
- int stb__mt_index = STB__MT_LEN*sizeof(int)+1;
- unsigned int stb__mt_buffer[STB__MT_LEN];
- void stb_srand(unsigned int seed)
- {
- int i;
- stb__mt_buffer[0]= seed & 0xffffffffUL;
- for (i=1 ; i < STB__MT_LEN; ++i)
- stb__mt_buffer[i] = (1812433253UL * (stb__mt_buffer[i-1] ^ (stb__mt_buffer[i-1] >> 30)) + i);
- stb__mt_index = STB__MT_LEN*sizeof(unsigned int);
- }
- #define STB__MT_IA 397
- #define STB__MT_IB (STB__MT_LEN - STB__MT_IA)
- #define STB__UPPER_MASK 0x80000000
- #define STB__LOWER_MASK 0x7FFFFFFF
- #define STB__MATRIX_A 0x9908B0DF
- #define STB__TWIST(b,i,j) ((b)[i] & STB__UPPER_MASK) | ((b)[j] & STB__LOWER_MASK)
- #define STB__MAGIC(s) (((s)&1)*STB__MATRIX_A)
- unsigned int stb_rand()
- {
- unsigned int * b = stb__mt_buffer;
- int idx = stb__mt_index;
- unsigned int s,r;
- int i;
- if (idx >= STB__MT_LEN*sizeof(unsigned int)) {
- if (idx > STB__MT_LEN*sizeof(unsigned int))
- stb_srand(0);
- idx = 0;
- i = 0;
- for (; i < STB__MT_IB; i++) {
- s = STB__TWIST(b, i, i+1);
- b[i] = b[i + STB__MT_IA] ^ (s >> 1) ^ STB__MAGIC(s);
- }
- for (; i < STB__MT_LEN-1; i++) {
- s = STB__TWIST(b, i, i+1);
- b[i] = b[i - STB__MT_IB] ^ (s >> 1) ^ STB__MAGIC(s);
- }
- s = STB__TWIST(b, STB__MT_LEN-1, 0);
- b[STB__MT_LEN-1] = b[STB__MT_IA-1] ^ (s >> 1) ^ STB__MAGIC(s);
- }
- stb__mt_index = idx + sizeof(unsigned int);
- r = *(unsigned int *)((unsigned char *)b + idx);
- r ^= (r >> 11);
- r ^= (r << 7) & 0x9D2C5680;
- r ^= (r << 15) & 0xEFC60000;
- r ^= (r >> 18);
- return r;
- }
- double stb_frand(void)
- {
- return stb_rand() / ((double) (1 << 16) * (1 << 16));
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_dupe
- //
- // stb_dupe is a duplicate-finding system for very, very large data
- // structures--large enough that sorting is too slow, but not so large
- // that we can't keep all the data in memory. using it works as follows:
- //
- // 1. create an stb_dupe:
- // provide a hash function
- // provide an equality function
- // provide an estimate for the size
- // optionally provide a comparison function
- //
- // 2. traverse your data, 'adding' pointers to the stb_dupe
- //
- // 3. finish and ask for duplicates
- //
- // the stb_dupe will discard its intermediate data and build
- // a collection of sorted lists of duplicates, with non-duplicate
- // entries omitted entirely
- //
- //
- // Implementation strategy:
- //
- // while collecting the N items, we keep a hash table of approximate
- // size sqrt(N). (if you tell use the N up front, the hash table is
- // just that size exactly)
- //
- // each entry in the hash table is just an stb__arr of pointers (no need
- // to use stb_ps, because we don't need to delete from these)
- //
- // for step 3, for each entry in the hash table, we apply stb_dupe to it
- // recursively. once the size gets small enough (or doesn't decrease
- // significantly), we switch to either using qsort() on the comparison
- // function, or else we just do the icky N^2 gather
- typedef struct stb_dupe stb_dupe;
- typedef int (*stb_compare_func)(void *a, void *b);
- typedef int (*stb_hash_func)(void *a, unsigned int seed);
- STB_EXTERN void stb_dupe_free(stb_dupe *sd);
- STB_EXTERN stb_dupe *stb_dupe_create(stb_hash_func hash,
- stb_compare_func eq, int size, stb_compare_func ineq);
- STB_EXTERN void stb_dupe_add(stb_dupe *sd, void *item);
- STB_EXTERN void stb_dupe_finish(stb_dupe *sd);
- STB_EXTERN int stb_dupe_numsets(stb_dupe *sd);
- STB_EXTERN void **stb_dupe_set(stb_dupe *sd, int num);
- STB_EXTERN int stb_dupe_set_count(stb_dupe *sd, int num);
- struct stb_dupe
- {
- void ***hash_table;
- int hash_size;
- int size_log2;
- int population;
- int hash_shift;
- stb_hash_func hash;
- stb_compare_func eq;
- stb_compare_func ineq;
- void ***dupes;
- };
- #ifdef STB_DEFINE
- int stb_dupe_numsets(stb_dupe *sd)
- {
- assert(sd->hash_table == NULL);
- return stb_arr_len(sd->dupes);
- }
- void **stb_dupe_set(stb_dupe *sd, int num)
- {
- assert(sd->hash_table == NULL);
- return sd->dupes[num];
- }
- int stb_dupe_set_count(stb_dupe *sd, int num)
- {
- assert(sd->hash_table == NULL);
- return stb_arr_len(sd->dupes[num]);
- }
- stb_dupe *stb_dupe_create(stb_hash_func hash, stb_compare_func eq, int size,
- stb_compare_func ineq)
- {
- int i, hsize;
- stb_dupe *sd = (stb_dupe *) malloc(sizeof(*sd));
- sd->size_log2 = 4;
- hsize = 1 << sd->size_log2;
- while (hsize * hsize < size) {
- ++sd->size_log2;
- hsize *= 2;
- }
- sd->hash = hash;
- sd->eq = eq;
- sd->ineq = ineq;
- sd->hash_shift = 0;
- sd->population = 0;
- sd->hash_size = hsize;
- sd->hash_table = (void ***) malloc(sizeof(*sd->hash_table) * hsize);
- for (i=0; i < hsize; ++i)
- sd->hash_table[i] = NULL;
- sd->dupes = NULL;
- return sd;
- }
- void stb_dupe_add(stb_dupe *sd, void *item)
- {
- stb_uint32 hash = sd->hash(item, sd->hash_shift);
- int z = hash & (sd->hash_size-1);
- stb_arr_push(sd->hash_table[z], item);
- ++sd->population;
- }
- void stb_dupe_free(stb_dupe *sd)
- {
- int i;
- for (i=0; i < stb_arr_len(sd->dupes); ++i)
- if (sd->dupes[i])
- stb_arr_free(sd->dupes[i]);
- stb_arr_free(sd->dupes);
- free(sd);
- }
- static stb_compare_func stb__compare;
- static int stb__dupe_compare(const void *a, const void *b)
- {
- void *p = *(void **) a;
- void *q = *(void **) b;
- return stb__compare(p,q);
- }
- void stb_dupe_finish(stb_dupe *sd)
- {
- int i,j,k;
- assert(sd->dupes == NULL);
- for (i=0; i < sd->hash_size; ++i) {
- void ** list = sd->hash_table[i];
- if (list != NULL) {
- int n = stb_arr_len(list);
- // @TODO: measure to find good numbers instead of just making them up!
- int thresh = (sd->ineq ? 200 : 20);
- // if n is large enough to be worth it, and n is smaller than
- // before (so we can guarantee we'll use a smaller hash table);
- // and there are enough hash bits left, assuming full 32-bit hash
- if (n > thresh && n < (sd->population >> 3) && sd->hash_shift + sd->size_log2*2 < 32) {
- // recursively process this row using stb_dupe, O(N log log N)
- stb_dupe *d = stb_dupe_create(sd->hash, sd->eq, n, sd->ineq);
- d->hash_shift = stb_randLCG_explicit(sd->hash_shift);
- for (j=0; j < n; ++j)
- stb_dupe_add(d, list[j]);
- stb_arr_free(sd->hash_table[i]);
- stb_dupe_finish(d);
- for (j=0; j < stb_arr_len(d->dupes); ++j) {
- stb_arr_push(sd->dupes, d->dupes[j]);
- d->dupes[j] = NULL; // take over ownership
- }
- stb_dupe_free(d);
- } else if (sd->ineq) {
- // process this row using qsort(), O(N log N)
- stb__compare = sd->ineq;
- qsort(list, n, sizeof(list[0]), stb__dupe_compare);
- // find equal subsequences of the list
- for (j=0; j < n-1; ) {
- // find a subsequence from j..k
- for (k=j; k < n; ++k)
- // only use ineq so eq can be left undefined
- if (sd->ineq(list[j], list[k]))
- break;
- // k is the first one not in the subsequence
- if (k-j > 1) {
- void **mylist = NULL;
- stb_arr_setlen(mylist, k-j);
- memcpy(mylist, list+j, sizeof(list[j]) * (k-j));
- stb_arr_push(sd->dupes, mylist);
- }
- j = k;
- }
- stb_arr_free(sd->hash_table[i]);
- } else {
- // process this row using eq(), O(N^2)
- for (j=0; j < n; ++j) {
- if (list[j] != NULL) {
- void **output = NULL;
- for (k=j+1; k < n; ++k) {
- if (sd->eq(list[j], list[k])) {
- if (output == NULL)
- stb_arr_push(output, list[j]);
- stb_arr_push(output, list[k]);
- list[k] = NULL;
- }
- }
- list[j] = NULL;
- if (output)
- stb_arr_push(sd->dupes, output);
- }
- }
- stb_arr_free(sd->hash_table[i]);
- }
- }
- }
- free(sd->hash_table);
- sd->hash_table = NULL;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // templatized Sort routine
- //
- // This is an attempt to implement a templated sorting algorithm.
- // To use it, you have to explicitly instantiate it as a _function_,
- // then you call that function. This allows the comparison to be inlined,
- // giving the sort similar performance to C++ sorts.
- //
- // It implements quicksort with three-way-median partitioning (generally
- // well-behaved), with a final insertion sort pass.
- //
- // When you define the compare expression, you should assume you have
- // elements of your array pointed to by 'a' and 'b', and perform the comparison
- // on those. OR you can use one or more statements; first say '0;', then
- // write whatever code you want, and compute the result into a variable 'c'.
- #define stb_declare_sort(FUNCNAME, TYPE) \
- void FUNCNAME(TYPE *p, int n)
- #define stb_define_sort(FUNCNAME,TYPE,COMPARE) \
- stb__define_sort( void, FUNCNAME,TYPE,COMPARE)
- #define stb_define_sort_static(FUNCNAME,TYPE,COMPARE) \
- stb__define_sort(static void, FUNCNAME,TYPE,COMPARE)
- #define stb__define_sort(MODE, FUNCNAME, TYPE, COMPARE) \
- \
- static void STB_(FUNCNAME,_ins_sort)(TYPE *p, int n) \
- { \
- int i,j; \
- for (i=1; i < n; ++i) { \
- TYPE t = p[i], *a = &t; \
- j = i; \
- while (j > 0) { \
- TYPE *b = &p[j-1]; \
- int c = COMPARE; \
- if (!c) break; \
- p[j] = p[j-1]; \
- --j; \
- } \
- if (i != j) \
- p[j] = t; \
- } \
- } \
- \
- static void STB_(FUNCNAME,_quicksort)(TYPE *p, int n) \
- { \
- /* threshold for transitioning to insertion sort */ \
- while (n > 12) { \
- TYPE *a,*b,t; \
- int c01,c12,c,m,i,j; \
- \
- /* compute median of three */ \
- m = n >> 1; \
- a = &p[0]; \
- b = &p[m]; \
- c = COMPARE; \
- c01 = c; \
- a = &p[m]; \
- b = &p[n-1]; \
- c = COMPARE; \
- c12 = c; \
- /* if 0 >= mid >= end, or 0 < mid < end, then use mid */ \
- if (c01 != c12) { \
- /* otherwise, we'll need to swap something else to middle */ \
- int z; \
- a = &p[0]; \
- b = &p[n-1]; \
- c = COMPARE; \
- /* 0>mid && mid<n: 0>n => n; 0<n => 0 */ \
- /* 0<mid && mid>n: 0>n => 0; 0<n => n */ \
- z = (c == c12) ? 0 : n-1; \
- t = p[z]; \
- p[z] = p[m]; \
- p[m] = t; \
- } \
- /* now p[m] is the median-of-three */ \
- /* swap it to the beginning so it won't move around */ \
- t = p[0]; \
- p[0] = p[m]; \
- p[m] = t; \
- \
- /* partition loop */ \
- i=1; \
- j=n-1; \
- for(;;) { \
- /* handling of equality is crucial here */ \
- /* for sentinels & efficiency with duplicates */ \
- b = &p[0]; \
- for (;;++i) { \
- a=&p[i]; \
- c = COMPARE; \
- if (!c) break; \
- } \
- a = &p[0]; \
- for (;;--j) { \
- b=&p[j]; \
- c = COMPARE; \
- if (!c) break; \
- } \
- /* make sure we haven't crossed */ \
- if (i >= j) break; \
- t = p[i]; \
- p[i] = p[j]; \
- p[j] = t; \
- \
- ++i; \
- --j; \
- } \
- /* recurse on smaller side, iterate on larger */ \
- if (j < (n-i)) { \
- STB_(FUNCNAME,_quicksort)(p,j); \
- p = p+i; \
- n = n-i; \
- } else { \
- STB_(FUNCNAME,_quicksort)(p+i, n-i); \
- n = j; \
- } \
- } \
- } \
- \
- MODE FUNCNAME(TYPE *p, int n) \
- { \
- STB_(FUNCNAME, _quicksort)(p, n); \
- STB_(FUNCNAME, _ins_sort)(p, n); \
- } \
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_bitset an array of booleans indexed by integers
- //
- typedef stb_uint32 stb_bitset;
- STB_EXTERN stb_bitset *stb_bitset_new(int value, int len);
- #define stb_bitset_clearall(arr,len) (memset(arr, 0, 4 * (len)))
- #define stb_bitset_setall(arr,len) (memset(arr, 255, 4 * (len)))
- #define stb_bitset_setbit(arr,n) ((arr)[(n) >> 5] |= (1 << (n & 31)))
- #define stb_bitset_clearbit(arr,n) ((arr)[(n) >> 5] &= ~(1 << (n & 31)))
- #define stb_bitset_testbit(arr,n) ((arr)[(n) >> 5] & (1 << (n & 31)))
- STB_EXTERN stb_bitset *stb_bitset_union(stb_bitset *p0, stb_bitset *p1, int len);
- STB_EXTERN int *stb_bitset_getlist(stb_bitset *out, int start, int end);
- STB_EXTERN int stb_bitset_eq(stb_bitset *p0, stb_bitset *p1, int len);
- STB_EXTERN int stb_bitset_disjoint(stb_bitset *p0, stb_bitset *p1, int len);
- STB_EXTERN int stb_bitset_disjoint_0(stb_bitset *p0, stb_bitset *p1, int len);
- STB_EXTERN int stb_bitset_subset(stb_bitset *bigger, stb_bitset *smaller, int len);
- STB_EXTERN int stb_bitset_unioneq_changed(stb_bitset *p0, stb_bitset *p1, int len);
- #ifdef STB_DEFINE
- int stb_bitset_eq(stb_bitset *p0, stb_bitset *p1, int len)
- {
- int i;
- for (i=0; i < len; ++i)
- if (p0[i] != p1[i]) return 0;
- return 1;
- }
- int stb_bitset_disjoint(stb_bitset *p0, stb_bitset *p1, int len)
- {
- int i;
- for (i=0; i < len; ++i)
- if (p0[i] & p1[i]) return 0;
- return 1;
- }
- int stb_bitset_disjoint_0(stb_bitset *p0, stb_bitset *p1, int len)
- {
- int i;
- for (i=0; i < len; ++i)
- if ((p0[i] | p1[i]) != 0xffffffff) return 0;
- return 1;
- }
- int stb_bitset_subset(stb_bitset *bigger, stb_bitset *smaller, int len)
- {
- int i;
- for (i=0; i < len; ++i)
- if ((bigger[i] & smaller[i]) != smaller[i]) return 0;
- return 1;
- }
- stb_bitset *stb_bitset_union(stb_bitset *p0, stb_bitset *p1, int len)
- {
- int i;
- stb_bitset *d = (stb_bitset *) malloc(sizeof(*d) * len);
- for (i=0; i < len; ++i) d[i] = p0[i] | p1[i];
- return d;
- }
- int stb_bitset_unioneq_changed(stb_bitset *p0, stb_bitset *p1, int len)
- {
- int i, changed=0;
- for (i=0; i < len; ++i) {
- stb_bitset d = p0[i] | p1[i];
- if (d != p0[i]) {
- p0[i] = d;
- changed = 1;
- }
- }
- return changed;
- }
- stb_bitset *stb_bitset_new(int value, int len)
- {
- int i;
- stb_bitset *d = (stb_bitset *) malloc(sizeof(*d) * len);
- if (value) value = 0xffffffff;
- for (i=0; i < len; ++i) d[i] = value;
- return d;
- }
- int *stb_bitset_getlist(stb_bitset *out, int start, int end)
- {
- int *list = NULL;
- int i;
- for (i=start; i < end; ++i)
- if (stb_bitset_testbit(out, i))
- stb_arr_push(list, i);
- return list;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_wordwrap quality word-wrapping for fixed-width fonts
- //
- STB_EXTERN int stb_wordwrap(int *pairs, int pair_max, int count, char *str);
- STB_EXTERN int *stb_wordwrapalloc(int count, char *str);
- #ifdef STB_DEFINE
- int stb_wordwrap(int *pairs, int pair_max, int count, char *str)
- {
- int n=0,i=0, start=0,nonwhite=0;
- if (pairs == NULL) pair_max = 0x7ffffff0;
- else pair_max *= 2;
- // parse
- for(;;) {
- int s=i; // first whitespace char; last nonwhite+1
- int w; // word start
- // accept whitespace
- while (isspace(str[i])) {
- if (str[i] == '\n' || str[i] == '\r') {
- if (str[i] + str[i+1] == '\n' + '\r') ++i;
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = s-start;
- n += 2;
- nonwhite=0;
- start = i+1;
- s = start;
- }
- ++i;
- }
- if (i >= start+count) {
- // we've gone off the end using whitespace
- if (nonwhite) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = s-start;
- n += 2;
- start = s = i;
- nonwhite=0;
- } else {
- // output all the whitespace
- while (i >= start+count) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = count;
- n += 2;
- start += count;
- }
- s = start;
- }
- }
- if (str[i] == 0) break;
- // now scan out a word and see if it fits
- w = i;
- while (str[i] && !isspace(str[i])) {
- ++i;
- }
- // wrapped?
- if (i > start + count) {
- // huge?
- if (i-s <= count) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = s-start;
- n += 2;
- start = w;
- } else {
- // This word is longer than one line. If we wrap it onto N lines
- // there are leftover chars. do those chars fit on the cur line?
- // But if we have leading whitespace, we force it to start here.
- if ((w-start) + ((i-w) % count) <= count || !nonwhite) {
- // output a full line
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = count;
- n += 2;
- start += count;
- w = start;
- } else {
- // output a partial line, trimming trailing whitespace
- if (s != start) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = s-start;
- n += 2;
- start = w;
- }
- }
- // now output full lines as needed
- while (start + count <= i) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = count;
- n += 2;
- start += count;
- }
- }
- }
- nonwhite=1;
- }
- if (start < i) {
- if (n >= pair_max) return -1;
- if (pairs) pairs[n] = start, pairs[n+1] = i-start;
- n += 2;
- }
- return n>>1;
- }
- int *stb_wordwrapalloc(int count, char *str)
- {
- int n = stb_wordwrap(NULL,0,count,str);
- int *z = NULL;
- stb_arr_setlen(z, n*2);
- stb_wordwrap(z, n, count, str);
- return z;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // stb_match: wildcards and regexping
- //
- STB_EXTERN int stb_wildmatch (char *expr, char *candidate);
- STB_EXTERN int stb_wildmatchi(char *expr, char *candidate);
- STB_EXTERN int stb_wildfind (char *expr, char *candidate);
- STB_EXTERN int stb_wildfindi (char *expr, char *candidate);
- STB_EXTERN int stb_regex(char *regex, char *candidate);
- typedef struct stb_matcher stb_matcher;
- STB_EXTERN stb_matcher *stb_regex_matcher(char *regex);
- STB_EXTERN int stb_matcher_match(stb_matcher *m, char *str);
- STB_EXTERN int stb_matcher_find(stb_matcher *m, char *str);
- STB_EXTERN void stb_matcher_free(stb_matcher *f);
- STB_EXTERN stb_matcher *stb_lex_matcher(void);
- STB_EXTERN int stb_lex_item(stb_matcher *m, const char *str, int result);
- STB_EXTERN int stb_lex_item_wild(stb_matcher *matcher, const char *regex, int result);
- STB_EXTERN int stb_lex(stb_matcher *m, char *str, int *len);
- #ifdef STB_DEFINE
- static int stb__match_qstring(char *candidate, char *qstring, int qlen, int insensitive)
- {
- int i;
- if (insensitive) {
- for (i=0; i < qlen; ++i)
- if (qstring[i] == '?') {
- if (!candidate[i]) return 0;
- } else
- if (tolower(qstring[i]) != tolower(candidate[i]))
- return 0;
- } else {
- for (i=0; i < qlen; ++i)
- if (qstring[i] == '?') {
- if (!candidate[i]) return 0;
- } else
- if (qstring[i] != candidate[i])
- return 0;
- }
- return 1;
- }
- static int stb__find_qstring(char *candidate, char *qstring, int qlen, int insensitive)
- {
- char c;
- int offset=0;
- while (*qstring == '?') {
- ++qstring;
- --qlen;
- ++candidate;
- if (qlen == 0) return 0;
- if (*candidate == 0) return -1;
- }
- c = *qstring++;
- --qlen;
- if (insensitive) c = tolower(c);
- while (candidate[offset]) {
- if (c == (insensitive ? tolower(candidate[offset]) : candidate[offset]))
- if (stb__match_qstring(candidate+offset+1, qstring, qlen, insensitive))
- return offset;
- ++offset;
- }
- return -1;
- }
- int stb__wildmatch_raw2(char *expr, char *candidate, int search, int insensitive)
- {
- int where=0;
- int start = -1;
- if (!search) {
- // parse to first '*'
- if (*expr != '*')
- start = 0;
- while (*expr != '*') {
- if (!*expr)
- return *candidate == 0 ? 0 : -1;
- if (*expr == '?') {
- if (!*candidate) return -1;
- } else {
- if (insensitive) {
- if (tolower(*candidate) != tolower(*expr))
- return -1;
- } else
- if (*candidate != *expr)
- return -1;
- }
- ++candidate, ++expr, ++where;
- }
- } else {
- // 0-length search string
- if (!*expr)
- return 0;
- }
- assert(search || *expr == '*');
- if (!search)
- ++expr;
- // implicit '*' at this point
- while (*expr) {
- int o=0;
- // combine redundant * characters
- while (expr[0] == '*') ++expr;
- // ok, at this point, expr[-1] == '*',
- // and expr[0] != '*'
- if (!expr[0]) return start >= 0 ? start : 0;
- // now find next '*'
- o = 0;
- while (expr[o] != '*') {
- if (expr[o] == 0)
- break;
- ++o;
- }
- // if no '*', scan to end, then match at end
- if (expr[o] == 0 && !search) {
- int z;
- for (z=0; z < o; ++z)
- if (candidate[z] == 0)
- return -1;
- while (candidate[z])
- ++z;
- // ok, now check if they match
- if (stb__match_qstring(candidate+z-o, expr, o, insensitive))
- return start >= 0 ? start : 0;
- return -1;
- } else {
- // if yes '*', then do stb__find_qmatch on the intervening chars
- int n = stb__find_qstring(candidate, expr, o, insensitive);
- if (n < 0)
- return -1;
- if (start < 0)
- start = where + n;
- expr += o;
- candidate += n+o;
- }
- if (*expr == 0) {
- assert(search);
- return start;
- }
- assert(*expr == '*');
- ++expr;
- }
- return start >= 0 ? start : 0;
- }
- int stb__wildmatch_raw(char *expr, char *candidate, int search, int insensitive)
- {
- char buffer[256];
- // handle multiple search strings
- char *s = strchr(expr, ';');
- char *last = expr;
- while (s) {
- int z;
- // need to allow for non-writeable strings... assume they're small
- if (s - last < 256) {
- stb_strncpy(buffer, last, (int) (s-last+1));
- buffer[s-last] = 0;
- z = stb__wildmatch_raw2(buffer, candidate, search, insensitive);
- } else {
- *s = 0;
- z = stb__wildmatch_raw2(last, candidate, search, insensitive);
- *s = ';';
- }
- if (z >= 0) return z;
- last = s+1;
- s = strchr(last, ';');
- }
- return stb__wildmatch_raw2(last, candidate, search, insensitive);
- }
- int stb_wildmatch(char *expr, char *candidate)
- {
- return stb__wildmatch_raw(expr, candidate, 0,0) >= 0;
- }
- int stb_wildmatchi(char *expr, char *candidate)
- {
- return stb__wildmatch_raw(expr, candidate, 0,1) >= 0;
- }
- int stb_wildfind(char *expr, char *candidate)
- {
- return stb__wildmatch_raw(expr, candidate, 1,0);
- }
- int stb_wildfindi(char *expr, char *candidate)
- {
- return stb__wildmatch_raw(expr, candidate, 1,1);
- }
- typedef struct
- {
- stb_int16 transition[256];
- } stb_dfa;
- // an NFA node represents a state you're in; it then has
- // an arbitrary number of edges dangling off of it
- // note this isn't utf8-y
- typedef struct
- {
- stb_int16 match; // character/set to match
- stb_uint16 node; // output node to go to
- } stb_nfa_edge;
- typedef struct
- {
- stb_int16 goal; // does reaching this win the prize?
- stb_uint8 active; // is this in the active list
- stb_nfa_edge *out;
- stb_uint16 *eps; // list of epsilon closures
- } stb_nfa_node;
- #define STB__DFA_UNDEF -1
- #define STB__DFA_GOAL -2
- #define STB__DFA_END -3
- #define STB__DFA_MGOAL -4
- #define STB__DFA_VALID 0
- #define STB__NFA_STOP_GOAL -1
- // compiled regexp
- struct stb_matcher
- {
- stb_uint16 start_node;
- stb_int16 dfa_start;
- stb_uint32 *charset;
- int num_charset;
- int match_start;
- stb_nfa_node *nodes;
- int does_lex;
- // dfa matcher
- stb_dfa * dfa;
- stb_uint32 * dfa_mapping;
- stb_int16 * dfa_result;
- int num_words_per_dfa;
- };
- static int stb__add_node(stb_matcher *matcher)
- {
- stb_nfa_node z;
- z.active = 0;
- z.eps = 0;
- z.goal = 0;
- z.out = 0;
- stb_arr_push(matcher->nodes, z);
- return stb_arr_len(matcher->nodes)-1;
- }
- static void stb__add_epsilon(stb_matcher *matcher, int from, int to)
- {
- assert(from != to);
- if (matcher->nodes[from].eps == NULL)
- stb_arr_malloc((void **) &matcher->nodes[from].eps, matcher);
- stb_arr_push(matcher->nodes[from].eps, to);
- }
- static void stb__add_edge(stb_matcher *matcher, int from, int to, int type)
- {
- stb_nfa_edge z = { (stb_int16)type, (stb_uint16)to };
- if (matcher->nodes[from].out == NULL)
- stb_arr_malloc((void **) &matcher->nodes[from].out, matcher);
- stb_arr_push(matcher->nodes[from].out, z);
- }
- static char *stb__reg_parse_alt(stb_matcher *m, int s, char *r, stb_uint16 *e);
- static char *stb__reg_parse(stb_matcher *matcher, int start, char *regex, stb_uint16 *end)
- {
- int n;
- int last_start = -1;
- stb_uint16 last_end = start;
- while (*regex) {
- switch (*regex) {
- case '(':
- last_start = last_end;
- regex = stb__reg_parse_alt(matcher, last_end, regex+1, &last_end);
- if (regex == NULL || *regex != ')')
- return NULL;
- ++regex;
- break;
- case '|':
- case ')':
- *end = last_end;
- return regex;
- case '?':
- if (last_start < 0) return NULL;
- stb__add_epsilon(matcher, last_start, last_end);
- ++regex;
- break;
- case '*':
- if (last_start < 0) return NULL;
- stb__add_epsilon(matcher, last_start, last_end);
- // fall through
- case '+':
- if (last_start < 0) return NULL;
- stb__add_epsilon(matcher, last_end, last_start);
- // prevent links back to last_end from chaining to last_start
- n = stb__add_node(matcher);
- stb__add_epsilon(matcher, last_end, n);
- last_end = n;
- ++regex;
- break;
- case '{': // not supported!
- // @TODO: given {n,m}, clone last_start to last_end m times,
- // and include epsilons from start to first m-n blocks
- return NULL;
- case '\\':
- ++regex;
- if (!*regex) return NULL;
- // fallthrough
- default: // match exactly this character
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, *regex);
- last_start = last_end;
- last_end = n;
- ++regex;
- break;
- case '$':
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, '\n');
- last_start = last_end;
- last_end = n;
- ++regex;
- break;
- case '.':
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, -1);
- last_start = last_end;
- last_end = n;
- ++regex;
- break;
- case '[': {
- stb_uint8 flags[256];
- int invert = 0,z;
- ++regex;
- if (matcher->num_charset == 0) {
- matcher->charset = (stb_uint *) stb_malloc(matcher, sizeof(*matcher->charset) * 256);
- memset(matcher->charset, 0, sizeof(*matcher->charset) * 256);
- }
- memset(flags,0,sizeof(flags));
- // leading ^ is special
- if (*regex == '^')
- ++regex, invert = 1;
- // leading ] is special
- if (*regex == ']') {
- flags[(int) ']'] = 1;
- ++regex;
- }
- while (*regex != ']') {
- stb_uint a;
- if (!*regex) return NULL;
- a = *regex++;
- if (regex[0] == '-' && regex[1] != ']') {
- stb_uint i,b = regex[1];
- regex += 2;
- if (b == 0) return NULL;
- if (a > b) return NULL;
- for (i=a; i <= b; ++i)
- flags[i] = 1;
- } else
- flags[a] = 1;
- }
- ++regex;
- if (invert) {
- int i;
- for (i=0; i < 256; ++i)
- flags[i] = 1-flags[i];
- }
- // now check if any existing charset matches
- for (z=0; z < matcher->num_charset; ++z) {
- int i, k[2] = { 0, 1 << z};
- for (i=0; i < 256; ++i) {
- unsigned int f = k[flags[i]];
- if ((matcher->charset[i] & k[1]) != f)
- break;
- }
- if (i == 256) break;
- }
- if (z == matcher->num_charset) {
- int i;
- ++matcher->num_charset;
- if (matcher->num_charset > 32) {
- assert(0); /* NOTREACHED */
- return NULL; // too many charsets, oops
- }
- for (i=0; i < 256; ++i)
- if (flags[i])
- matcher->charset[i] |= (1 << z);
- }
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, -2 - z);
- last_start = last_end;
- last_end = n;
- break;
- }
- }
- }
- *end = last_end;
- return regex;
- }
- static char *stb__reg_parse_alt(stb_matcher *matcher, int start, char *regex, stb_uint16 *end)
- {
- stb_uint16 last_end = start;
- stb_uint16 main_end;
- int head, tail;
- head = stb__add_node(matcher);
- stb__add_epsilon(matcher, start, head);
- regex = stb__reg_parse(matcher, head, regex, &last_end);
- if (regex == NULL) return NULL;
- if (*regex == 0 || *regex == ')') {
- *end = last_end;
- return regex;
- }
- main_end = last_end;
- tail = stb__add_node(matcher);
- stb__add_epsilon(matcher, last_end, tail);
- // start alternatives from the same starting node; use epsilon
- // transitions to combine their endings
- while(*regex && *regex != ')') {
- assert(*regex == '|');
- head = stb__add_node(matcher);
- stb__add_epsilon(matcher, start, head);
- regex = stb__reg_parse(matcher, head, regex+1, &last_end);
- if (regex == NULL)
- return NULL;
- stb__add_epsilon(matcher, last_end, tail);
- }
- *end = tail;
- return regex;
- }
- static char *stb__wild_parse(stb_matcher *matcher, int start, char *str, stb_uint16 *end)
- {
- int n;
- stb_uint16 last_end;
- last_end = stb__add_node(matcher);
- stb__add_epsilon(matcher, start, last_end);
- while (*str) {
- switch (*str) {
- // fallthrough
- default: // match exactly this character
- n = stb__add_node(matcher);
- if (toupper(*str) == tolower(*str)) {
- stb__add_edge(matcher, last_end, n, *str);
- } else {
- stb__add_edge(matcher, last_end, n, tolower(*str));
- stb__add_edge(matcher, last_end, n, toupper(*str));
- }
- last_end = n;
- ++str;
- break;
- case '?':
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, -1);
- last_end = n;
- ++str;
- break;
- case '*':
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, -1);
- stb__add_epsilon(matcher, last_end, n);
- stb__add_epsilon(matcher, n, last_end);
- last_end = n;
- ++str;
- break;
- }
- }
- // now require end of string to match
- n = stb__add_node(matcher);
- stb__add_edge(matcher, last_end, n, 0);
- last_end = n;
- *end = last_end;
- return str;
- }
- static int stb__opt(stb_matcher *m, int n)
- {
- for(;;) {
- stb_nfa_node *p = &m->nodes[n];
- if (p->goal) return n;
- if (stb_arr_len(p->out)) return n;
- if (stb_arr_len(p->eps) != 1) return n;
- n = p->eps[0];
- }
- }
- static void stb__optimize(stb_matcher *m)
- {
- // if the target of any edge is a node with exactly
- // one out-epsilon, shorten it
- int i,j;
- for (i=0; i < stb_arr_len(m->nodes); ++i) {
- stb_nfa_node *p = &m->nodes[i];
- for (j=0; j < stb_arr_len(p->out); ++j)
- p->out[j].node = stb__opt(m,p->out[j].node);
- for (j=0; j < stb_arr_len(p->eps); ++j)
- p->eps[j] = stb__opt(m,p->eps[j] );
- }
- m->start_node = stb__opt(m,m->start_node);
- }
- void stb_matcher_free(stb_matcher *f)
- {
- stb_free(f);
- }
- static stb_matcher *stb__alloc_matcher(void)
- {
- stb_matcher *matcher = (stb_matcher *) stb_malloc(0,sizeof(*matcher));
- matcher->start_node = 0;
- stb_arr_malloc((void **) &matcher->nodes, matcher);
- matcher->num_charset = 0;
- matcher->match_start = 0;
- matcher->does_lex = 0;
- matcher->dfa_start = STB__DFA_UNDEF;
- stb_arr_malloc((void **) &matcher->dfa, matcher);
- stb_arr_malloc((void **) &matcher->dfa_mapping, matcher);
- stb_arr_malloc((void **) &matcher->dfa_result, matcher);
- stb__add_node(matcher);
- return matcher;
- }
- static void stb__lex_reset(stb_matcher *matcher)
- {
- // flush cached dfa data
- stb_arr_setlen(matcher->dfa, 0);
- stb_arr_setlen(matcher->dfa_mapping, 0);
- stb_arr_setlen(matcher->dfa_result, 0);
- matcher->dfa_start = STB__DFA_UNDEF;
- }
- stb_matcher *stb_regex_matcher(char *regex)
- {
- char *z;
- stb_uint16 end;
- stb_matcher *matcher = stb__alloc_matcher();
- if (*regex == '^') {
- matcher->match_start = 1;
- ++regex;
- }
- z = stb__reg_parse_alt(matcher, matcher->start_node, regex, &end);
- if (!z || *z) {
- stb_free(matcher);
- return NULL;
- }
- ((matcher->nodes)[(int) end]).goal = STB__NFA_STOP_GOAL;
- return matcher;
- }
- stb_matcher *stb_lex_matcher(void)
- {
- stb_matcher *matcher = stb__alloc_matcher();
- matcher->match_start = 1;
- matcher->does_lex = 1;
- return matcher;
- }
- int stb_lex_item(stb_matcher *matcher, const char *regex, int result)
- {
- char *z;
- stb_uint16 end;
- z = stb__reg_parse_alt(matcher, matcher->start_node, (char*) regex, &end);
- if (z == NULL)
- return 0;
- stb__lex_reset(matcher);
- matcher->nodes[(int) end].goal = result;
- return 1;
- }
- int stb_lex_item_wild(stb_matcher *matcher, const char *regex, int result)
- {
- char *z;
- stb_uint16 end;
- z = stb__wild_parse(matcher, matcher->start_node, (char*) regex, &end);
- if (z == NULL)
- return 0;
- stb__lex_reset(matcher);
- matcher->nodes[(int) end].goal = result;
- return 1;
- }
- static void stb__clear(stb_matcher *m, stb_uint16 *list)
- {
- int i;
- for (i=0; i < stb_arr_len(list); ++i)
- m->nodes[(int) list[i]].active = 0;
- }
- static int stb__clear_goalcheck(stb_matcher *m, stb_uint16 *list)
- {
- int i, t=0;
- for (i=0; i < stb_arr_len(list); ++i) {
- t += m->nodes[(int) list[i]].goal;
- m->nodes[(int) list[i]].active = 0;
- }
- return t;
- }
- static stb_uint16 * stb__add_if_inactive(stb_matcher *m, stb_uint16 *list, int n)
- {
- if (!m->nodes[n].active) {
- stb_arr_push(list, n);
- m->nodes[n].active = 1;
- }
- return list;
- }
- static stb_uint16 * stb__eps_closure(stb_matcher *m, stb_uint16 *list)
- {
- int i,n = stb_arr_len(list);
- for(i=0; i < n; ++i) {
- stb_uint16 *e = m->nodes[(int) list[i]].eps;
- if (e) {
- int j,k = stb_arr_len(e);
- for (j=0; j < k; ++j)
- list = stb__add_if_inactive(m, list, e[j]);
- n = stb_arr_len(list);
- }
- }
- return list;
- }
- int stb_matcher_match(stb_matcher *m, char *str)
- {
- int result = 0;
- int i,j,y,z;
- stb_uint16 *previous = NULL;
- stb_uint16 *current = NULL;
- stb_uint16 *temp;
- stb_arr_setsize(previous, 4);
- stb_arr_setsize(current, 4);
- previous = stb__add_if_inactive(m, previous, m->start_node);
- previous = stb__eps_closure(m,previous);
- stb__clear(m, previous);
- while (*str && stb_arr_len(previous)) {
- y = stb_arr_len(previous);
- for (i=0; i < y; ++i) {
- stb_nfa_node *n = &m->nodes[(int) previous[i]];
- z = stb_arr_len(n->out);
- for (j=0; j < z; ++j) {
- if (n->out[j].match >= 0) {
- if (n->out[j].match == *str)
- current = stb__add_if_inactive(m, current, n->out[j].node);
- } else if (n->out[j].match == -1) {
- if (*str != '\n')
- current = stb__add_if_inactive(m, current, n->out[j].node);
- } else if (n->out[j].match < -1) {
- int z = -n->out[j].match - 2;
- if (m->charset[(stb_uint8) *str] & (1 << z))
- current = stb__add_if_inactive(m, current, n->out[j].node);
- }
- }
- }
- stb_arr_setlen(previous, 0);
- temp = previous;
- previous = current;
- current = temp;
- previous = stb__eps_closure(m,previous);
- stb__clear(m, previous);
- ++str;
- }
- // transition to pick up a '$' at the end
- y = stb_arr_len(previous);
- for (i=0; i < y; ++i)
- m->nodes[(int) previous[i]].active = 1;
- for (i=0; i < y; ++i) {
- stb_nfa_node *n = &m->nodes[(int) previous[i]];
- z = stb_arr_len(n->out);
- for (j=0; j < z; ++j) {
- if (n->out[j].match == '\n')
- current = stb__add_if_inactive(m, current, n->out[j].node);
- }
- }
- previous = stb__eps_closure(m,previous);
- stb__clear(m, previous);
- y = stb_arr_len(previous);
- for (i=0; i < y; ++i)
- if (m->nodes[(int) previous[i]].goal)
- result = 1;
- stb_arr_free(previous);
- stb_arr_free(current);
- return result && *str == 0;
- }
- stb_int16 stb__get_dfa_node(stb_matcher *m, stb_uint16 *list)
- {
- stb_uint16 node;
- stb_uint32 data[8], *state, *newstate;
- int i,j,n;
- state = (stb_uint32 *) stb_temp(data, m->num_words_per_dfa * 4);
- memset(state, 0, m->num_words_per_dfa*4);
- n = stb_arr_len(list);
- for (i=0; i < n; ++i) {
- int x = list[i];
- state[x >> 5] |= 1 << (x & 31);
- }
- // @TODO use a hash table
- n = stb_arr_len(m->dfa_mapping);
- i=j=0;
- for(; j < n; ++i, j += m->num_words_per_dfa) {
- // @TODO special case for <= 32
- if (!memcmp(state, m->dfa_mapping + j, m->num_words_per_dfa*4)) {
- node = i;
- goto done;
- }
- }
- assert(stb_arr_len(m->dfa) == i);
- node = i;
- newstate = stb_arr_addn(m->dfa_mapping, m->num_words_per_dfa);
- memcpy(newstate, state, m->num_words_per_dfa*4);
- // set all transitions to 'unknown'
- stb_arr_add(m->dfa);
- memset(m->dfa[i].transition, -1, sizeof(m->dfa[i].transition));
- if (m->does_lex) {
- int result = -1;
- n = stb_arr_len(list);
- for (i=0; i < n; ++i) {
- if (m->nodes[(int) list[i]].goal > result)
- result = m->nodes[(int) list[i]].goal;
- }
- stb_arr_push(m->dfa_result, result);
- }
- done:
- stb_tempfree(data, state);
- return node;
- }
- static int stb__matcher_dfa(stb_matcher *m, char *str_c, int *len)
- {
- stb_uint8 *str = (stb_uint8 *) str_c;
- stb_int16 node,prevnode;
- stb_dfa *trans;
- int match_length = 0;
- stb_int16 match_result=0;
- if (m->dfa_start == STB__DFA_UNDEF) {
- stb_uint16 *list;
- m->num_words_per_dfa = (stb_arr_len(m->nodes)+31) >> 5;
- stb__optimize(m);
- list = stb__add_if_inactive(m, NULL, m->start_node);
- list = stb__eps_closure(m,list);
- if (m->does_lex) {
- m->dfa_start = stb__get_dfa_node(m,list);
- stb__clear(m, list);
- // DON'T allow start state to be a goal state!
- // this allows people to specify regexes that can match 0
- // characters without them actually matching (also we don't
- // check _before_ advancing anyway
- if (m->dfa_start <= STB__DFA_MGOAL)
- m->dfa_start = -(m->dfa_start - STB__DFA_MGOAL);
- } else {
- if (stb__clear_goalcheck(m, list))
- m->dfa_start = STB__DFA_GOAL;
- else
- m->dfa_start = stb__get_dfa_node(m,list);
- }
- stb_arr_free(list);
- }
- prevnode = STB__DFA_UNDEF;
- node = m->dfa_start;
- trans = m->dfa;
- if (m->dfa_start == STB__DFA_GOAL)
- return 1;
- for(;;) {
- assert(node >= STB__DFA_VALID);
- // fast inner DFA loop; especially if STB__DFA_VALID is 0
- do {
- prevnode = node;
- node = trans[node].transition[*str++];
- } while (node >= STB__DFA_VALID);
- assert(node >= STB__DFA_MGOAL - stb_arr_len(m->dfa));
- assert(node < stb_arr_len(m->dfa));
- // special case for lex: need _longest_ match, so notice goal
- // state without stopping
- if (node <= STB__DFA_MGOAL) {
- match_length = (int) (str - (stb_uint8 *) str_c);
- node = -(node - STB__DFA_MGOAL);
- match_result = node;
- continue;
- }
- // slow NFA->DFA conversion
- // or we hit the goal or the end of the string, but those
- // can only happen once per search...
- if (node == STB__DFA_UNDEF) {
- // build a list -- @TODO special case <= 32 states
- // heck, use a more compact data structure for <= 16 and <= 8 ?!
- // @TODO keep states/newstates around instead of reallocating them
- stb_uint16 *states = NULL;
- stb_uint16 *newstates = NULL;
- int i,j,y,z;
- stb_uint32 *flags = &m->dfa_mapping[prevnode * m->num_words_per_dfa];
- assert(prevnode != STB__DFA_UNDEF);
- stb_arr_setsize(states, 4);
- stb_arr_setsize(newstates,4);
- for (j=0; j < m->num_words_per_dfa; ++j) {
- for (i=0; i < 32; ++i) {
- if (*flags & (1 << i))
- stb_arr_push(states, j*32+i);
- }
- ++flags;
- }
- // states is now the states we were in in the previous node;
- // so now we can compute what node it transitions to on str[-1]
- y = stb_arr_len(states);
- for (i=0; i < y; ++i) {
- stb_nfa_node *n = &m->nodes[(int) states[i]];
- z = stb_arr_len(n->out);
- for (j=0; j < z; ++j) {
- if (n->out[j].match >= 0) {
- if (n->out[j].match == str[-1] || (str[-1] == 0 && n->out[j].match == '\n'))
- newstates = stb__add_if_inactive(m, newstates, n->out[j].node);
- } else if (n->out[j].match == -1) {
- if (str[-1] != '\n' && str[-1])
- newstates = stb__add_if_inactive(m, newstates, n->out[j].node);
- } else if (n->out[j].match < -1) {
- int z = -n->out[j].match - 2;
- if (m->charset[str[-1]] & (1 << z))
- newstates = stb__add_if_inactive(m, newstates, n->out[j].node);
- }
- }
- }
- // AND add in the start state!
- if (!m->match_start || (str[-1] == '\n' && !m->does_lex))
- newstates = stb__add_if_inactive(m, newstates, m->start_node);
- // AND epsilon close it
- newstates = stb__eps_closure(m, newstates);
- // if it's a goal state, then that's all there is to it
- if (stb__clear_goalcheck(m, newstates)) {
- if (m->does_lex) {
- match_length = (int) (str - (stb_uint8 *) str_c);
- node = stb__get_dfa_node(m,newstates);
- match_result = node;
- node = -node + STB__DFA_MGOAL;
- trans = m->dfa; // could have gotten realloc()ed
- } else
- node = STB__DFA_GOAL;
- } else if (str[-1] == 0 || stb_arr_len(newstates) == 0) {
- node = STB__DFA_END;
- } else {
- node = stb__get_dfa_node(m,newstates);
- trans = m->dfa; // could have gotten realloc()ed
- }
- trans[prevnode].transition[str[-1]] = node;
- if (node <= STB__DFA_MGOAL)
- node = -(node - STB__DFA_MGOAL);
- stb_arr_free(newstates);
- stb_arr_free(states);
- }
- if (node == STB__DFA_GOAL) {
- return 1;
- }
- if (node == STB__DFA_END) {
- if (m->does_lex) {
- if (match_result) {
- if (len) *len = match_length;
- return m->dfa_result[(int) match_result];
- }
- }
- return 0;
- }
- assert(node != STB__DFA_UNDEF);
- }
- }
- int stb_matcher_find(stb_matcher *m, char *str)
- {
- assert(m->does_lex == 0);
- return stb__matcher_dfa(m, str, NULL);
- }
- int stb_lex(stb_matcher *m, char *str, int *len)
- {
- assert(m->does_lex);
- return stb__matcher_dfa(m, str, len);
- }
- #ifdef STB_PERFECT_HASH
- int stb_regex(char *regex, char *str)
- {
- static stb_perfect p;
- static stb_matcher ** matchers;
- static char ** regexps;
- static char ** regexp_cache;
- static unsigned short *mapping;
- int z = stb_perfect_hash(&p, (int)(size_t) regex);
- if (z >= 0) {
- if (strcmp(regex, regexp_cache[(int) mapping[z]])) {
- int i = mapping[z];
- stb_matcher_free(matchers[i]);
- free(regexp_cache[i]);
- regexps[i] = regex;
- regexp_cache[i] = stb_p_strdup(regex);
- matchers[i] = stb_regex_matcher(regex);
- }
- } else {
- int i,n;
- if (regex == NULL) {
- for (i=0; i < stb_arr_len(matchers); ++i) {
- stb_matcher_free(matchers[i]);
- free(regexp_cache[i]);
- }
- stb_arr_free(matchers);
- stb_arr_free(regexps);
- stb_arr_free(regexp_cache);
- stb_perfect_destroy(&p);
- free(mapping); mapping = NULL;
- return -1;
- }
- stb_arr_push(regexps, regex);
- stb_arr_push(regexp_cache, stb_p_strdup(regex));
- stb_arr_push(matchers, stb_regex_matcher(regex));
- stb_perfect_destroy(&p);
- n = stb_perfect_create(&p, (unsigned int *) (char **) regexps, stb_arr_len(regexps));
- mapping = (unsigned short *) realloc(mapping, n * sizeof(*mapping));
- for (i=0; i < stb_arr_len(regexps); ++i)
- mapping[stb_perfect_hash(&p, (int)(size_t) regexps[i])] = i;
- z = stb_perfect_hash(&p, (int)(size_t) regex);
- }
- return stb_matcher_find(matchers[(int) mapping[z]], str);
- }
- #endif
- #endif // STB_DEFINE
- #if 0
- //////////////////////////////////////////////////////////////////////////////
- //
- // C source-code introspection
- //
- // runtime structure
- typedef struct
- {
- char *name;
- char *type; // base type
- char *comment; // content of comment field
- int size; // size of base type
- int offset; // field offset
- int arrcount[8]; // array sizes; -1 = pointer indirection; 0 = end of list
- } stb_info_field;
- typedef struct
- {
- char *structname;
- int size;
- int num_fields;
- stb_info_field *fields;
- } stb_info_struct;
- extern stb_info_struct stb_introspect_output[];
- //
- STB_EXTERN void stb_introspect_precompiled(stb_info_struct *compiled);
- STB_EXTERN void stb__introspect(char *path, char *file);
- #define stb_introspect_ship() stb__introspect(NULL, NULL, stb__introspect_output)
- #ifdef STB_SHIP
- #define stb_introspect() stb_introspect_ship()
- #define stb_introspect_path(p) stb_introspect_ship()
- #else
- // bootstrapping: define stb_introspect() (or 'path') the first time
- #define stb_introspect() stb__introspect(NULL, __FILE__, NULL)
- #define stb_introspect_auto() stb__introspect(NULL, __FILE__, stb__introspect_output)
- #define stb_introspect_path(p) stb__introspect(p, __FILE__, NULL)
- #define stb_introspect_path(p) stb__introspect(p, __FILE__, NULL)
- #endif
- #ifdef STB_DEFINE
- #ifndef STB_INTROSPECT_CPP
- #ifdef __cplusplus
- #define STB_INTROSPECT_CPP 1
- #else
- #define STB_INTROSPECT_CPP 0
- #endif
- #endif
- void stb_introspect_precompiled(stb_info_struct *compiled)
- {
- }
- static void stb__introspect_filename(char *buffer, char *path)
- {
- #if STB_INTROSPECT_CPP
- stb_p_sprintf(buffer stb_p_size(9999), "%s/stb_introspect.cpp", path);
- #else
- stb_p_sprintf(buffer stb_p_size(9999), "%s/stb_introspect.c", path);
- #endif
- }
- static void stb__introspect_compute(char *path, char *file)
- {
- int i;
- char ** include_list = NULL;
- char ** introspect_list = NULL;
- FILE *f;
- f = stb_p_fopen(file, "w");
- if (!f) return;
- fputs("// if you get compiler errors, change the following 0 to a 1:\n", f);
- fputs("#define STB_INTROSPECT_INVALID 0\n\n", f);
- fputs("// this will force the code to compile, and force the introspector\n", f);
- fputs("// to run and then exit, allowing you to recompile\n\n\n", f);
- fputs("#include \"stb.h\"\n\n",f );
- fputs("#if STB_INTROSPECT_INVALID\n", f);
- fputs(" stb_info_struct stb__introspect_output[] = { (void *) 1 }\n", f);
- fputs("#else\n\n", f);
- for (i=0; i < stb_arr_len(include_list); ++i)
- fprintf(f, " #include \"%s\"\n", include_list[i]);
- fputs(" stb_info_struct stb__introspect_output[] =\n{\n", f);
- for (i=0; i < stb_arr_len(introspect_list); ++i)
- fprintf(f, " stb_introspect_%s,\n", introspect_list[i]);
- fputs(" };\n", f);
- fputs("#endif\n", f);
- fclose(f);
- }
- static stb_info_struct *stb__introspect_info;
- #ifndef STB_SHIP
- #endif
- void stb__introspect(char *path, char *file, stb_info_struct *compiled)
- {
- static int first=1;
- if (!first) return;
- first=0;
- stb__introspect_info = compiled;
- #ifndef STB_SHIP
- if (path || file) {
- int bail_flag = compiled && compiled[0].structname == (void *) 1;
- int needs_building = bail_flag;
- struct stb__stat st;
- char buffer[1024], buffer2[1024];
- if (!path) {
- stb_splitpath(buffer, file, STB_PATH);
- path = buffer;
- }
- // bail if the source path doesn't exist
- if (!stb_fexists(path)) return;
- stb__introspect_filename(buffer2, path);
- // get source/include files timestamps, compare to output-file timestamp;
- // if mismatched, regenerate
- if (stb__stat(buffer2, &st))
- needs_building = STB_TRUE;
- {
- // find any file that contains an introspection command and is newer
- // if needs_building is already true, we don't need to do this test,
- // but we still need these arrays, so go ahead and get them
- char **all[3];
- all[0] = stb_readdir_files_mask(path, "*.h");
- all[1] = stb_readdir_files_mask(path, "*.c");
- all[2] = stb_readdir_files_mask(path, "*.cpp");
- int i,j;
- if (needs_building) {
- for (j=0; j < 3; ++j) {
- for (i=0; i < stb_arr_len(all[j]); ++i) {
- struct stb__stat st2;
- if (!stb__stat(all[j][i], &st2)) {
- if (st.st_mtime < st2.st_mtime) {
- char *z = stb_filec(all[j][i], NULL);
- int found=STB_FALSE;
- while (y) {
- y = strstr(y, "//si");
- if (y && isspace(y[4])) {
- found = STB_TRUE;
- break;
- }
- }
- needs_building = STB_TRUE;
- goto done;
- }
- }
- }
- }
- done:;
- }
- char *z = stb_filec(all[i], NULL), *y = z;
- int found=STB_FALSE;
- while (y) {
- y = strstr(y, "//si");
- if (y && isspace(y[4])) {
- found = STB_TRUE;
- break;
- }
- }
- if (found)
- stb_arr_push(introspect_h, stb_p_strdup(all[i]));
- free(z);
- }
- }
- stb_readdir_free(all);
- if (!needs_building) {
- for (i=0; i < stb_arr_len(introspect_h); ++i) {
- struct stb__stat st2;
- if (!stb__stat(introspect_h[i], &st2))
- if (st.st_mtime < st2.st_mtime)
- needs_building = STB_TRUE;
- }
- }
- if (needs_building) {
- stb__introspect_compute(path, buffer2);
- }
- }
- }
- #endif
- }
- #endif
- #endif
- #ifdef STB_INTROSPECT
- // compile-time code-generator
- #define INTROSPECT(x) int main(int argc, char **argv) { stb__introspect(__FILE__); return 0; }
- #define FILE(x)
- void stb__introspect(char *filename)
- {
- char *file = stb_file(filename, NULL);
- char *s = file, *t, **p;
- char *out_name = "stb_introspect.c";
- char *out_path;
- STB_ARR(char) filelist = NULL;
- int i,n;
- if (!file) stb_fatal("Couldn't open %s", filename);
- out_path = stb_splitpathdup(filename, STB_PATH);
- // search for the macros
- while (*s) {
- char buffer[256];
- while (*s && !isupper(*s)) ++s;
- s = stb_strtok_invert(buffer, s, "ABCDEFGHIJKLMNOPQRSTUVWXYZ");
- s = stb_skipwhite(s);
- if (*s == '(') {
- ++s;
- t = strchr(s, ')');
- if (t == NULL) stb_fatal("Error parsing %s", filename);
- }
- }
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // STB-C sliding-window dictionary compression
- //
- // This uses a DEFLATE-style sliding window, but no bitwise entropy.
- // Everything is on byte boundaries, so you could then apply a byte-wise
- // entropy code, though that's nowhere near as effective.
- //
- // An STB-C stream begins with a 16-byte header:
- // 4 bytes: 0x57 0xBC 0x00 0x00
- // 8 bytes: big-endian size of decompressed data, 64-bits
- // 4 bytes: big-endian size of window (how far back decompressor may need)
- //
- // The following symbols appear in the stream (these were determined ad hoc,
- // not by analysis):
- //
- // [dict] 00000100 yyyyyyyy yyyyyyyy yyyyyyyy xxxxxxxx xxxxxxxx
- // [END] 00000101 11111010 cccccccc cccccccc cccccccc cccccccc
- // [dict] 00000110 yyyyyyyy yyyyyyyy yyyyyyyy xxxxxxxx
- // [literals] 00000111 zzzzzzzz zzzzzzzz
- // [literals] 00001zzz zzzzzzzz
- // [dict] 00010yyy yyyyyyyy yyyyyyyy xxxxxxxx xxxxxxxx
- // [dict] 00011yyy yyyyyyyy yyyyyyyy xxxxxxxx
- // [literals] 001zzzzz
- // [dict] 01yyyyyy yyyyyyyy xxxxxxxx
- // [dict] 1xxxxxxx yyyyyyyy
- //
- // xxxxxxxx: match length - 1
- // yyyyyyyy: backwards distance - 1
- // zzzzzzzz: num literals - 1
- // cccccccc: adler32 checksum of decompressed data
- // (all big-endian)
- STB_EXTERN stb_uint stb_decompress_length(stb_uchar *input);
- STB_EXTERN stb_uint stb_decompress(stb_uchar *out,stb_uchar *in,stb_uint len);
- STB_EXTERN stb_uint stb_compress (stb_uchar *out,stb_uchar *in,stb_uint len);
- STB_EXTERN void stb_compress_window(int z);
- STB_EXTERN void stb_compress_hashsize(unsigned int z);
- STB_EXTERN int stb_compress_tofile(char *filename, char *in, stb_uint len);
- STB_EXTERN int stb_compress_intofile(FILE *f, char *input, stb_uint len);
- STB_EXTERN char *stb_decompress_fromfile(char *filename, stb_uint *len);
- STB_EXTERN int stb_compress_stream_start(FILE *f);
- STB_EXTERN void stb_compress_stream_end(int close);
- STB_EXTERN void stb_write(char *data, int data_len);
- #ifdef STB_DEFINE
- stb_uint stb_decompress_length(stb_uchar *input)
- {
- return (input[8] << 24) + (input[9] << 16) + (input[10] << 8) + input[11];
- }
- //////////////////// decompressor ///////////////////////
- // simple implementation that just writes whole thing into big block
- static unsigned char *stb__barrier;
- static unsigned char *stb__barrier2;
- static unsigned char *stb__barrier3;
- static unsigned char *stb__barrier4;
- static stb_uchar *stb__dout;
- static void stb__match(stb_uchar *data, stb_uint length)
- {
- // INVERSE of memmove... write each byte before copying the next...
- assert (stb__dout + length <= stb__barrier);
- if (stb__dout + length > stb__barrier) { stb__dout += length; return; }
- if (data < stb__barrier4) { stb__dout = stb__barrier+1; return; }
- while (length--) *stb__dout++ = *data++;
- }
- static void stb__lit(stb_uchar *data, stb_uint length)
- {
- assert (stb__dout + length <= stb__barrier);
- if (stb__dout + length > stb__barrier) { stb__dout += length; return; }
- if (data < stb__barrier2) { stb__dout = stb__barrier+1; return; }
- memcpy(stb__dout, data, length);
- stb__dout += length;
- }
- #define stb__in2(x) ((i[x] << 8) + i[(x)+1])
- #define stb__in3(x) ((i[x] << 16) + stb__in2((x)+1))
- #define stb__in4(x) ((i[x] << 24) + stb__in3((x)+1))
- static stb_uchar *stb_decompress_token(stb_uchar *i)
- {
- if (*i >= 0x20) { // use fewer if's for cases that expand small
- if (*i >= 0x80) stb__match(stb__dout-i[1]-1, i[0] - 0x80 + 1), i += 2;
- else if (*i >= 0x40) stb__match(stb__dout-(stb__in2(0) - 0x4000 + 1), i[2]+1), i += 3;
- else /* *i >= 0x20 */ stb__lit(i+1, i[0] - 0x20 + 1), i += 1 + (i[0] - 0x20 + 1);
- } else { // more ifs for cases that expand large, since overhead is amortized
- if (*i >= 0x18) stb__match(stb__dout-(stb__in3(0) - 0x180000 + 1), i[3]+1), i += 4;
- else if (*i >= 0x10) stb__match(stb__dout-(stb__in3(0) - 0x100000 + 1), stb__in2(3)+1), i += 5;
- else if (*i >= 0x08) stb__lit(i+2, stb__in2(0) - 0x0800 + 1), i += 2 + (stb__in2(0) - 0x0800 + 1);
- else if (*i == 0x07) stb__lit(i+3, stb__in2(1) + 1), i += 3 + (stb__in2(1) + 1);
- else if (*i == 0x06) stb__match(stb__dout-(stb__in3(1)+1), i[4]+1), i += 5;
- else if (*i == 0x04) stb__match(stb__dout-(stb__in3(1)+1), stb__in2(4)+1), i += 6;
- }
- return i;
- }
- stb_uint stb_decompress(stb_uchar *output, stb_uchar *i, stb_uint length)
- {
- stb_uint olen;
- if (stb__in4(0) != 0x57bC0000) return 0;
- if (stb__in4(4) != 0) return 0; // error! stream is > 4GB
- olen = stb_decompress_length(i);
- stb__barrier2 = i;
- stb__barrier3 = i+length;
- stb__barrier = output + olen;
- stb__barrier4 = output;
- i += 16;
- stb__dout = output;
- while (1) {
- stb_uchar *old_i = i;
- i = stb_decompress_token(i);
- if (i == old_i) {
- if (*i == 0x05 && i[1] == 0xfa) {
- assert(stb__dout == output + olen);
- if (stb__dout != output + olen) return 0;
- if (stb_adler32(1, output, olen) != (stb_uint) stb__in4(2))
- return 0;
- return olen;
- } else {
- assert(0); /* NOTREACHED */
- return 0;
- }
- }
- assert(stb__dout <= output + olen);
- if (stb__dout > output + olen)
- return 0;
- }
- }
- char *stb_decompress_fromfile(char *filename, unsigned int *len)
- {
- unsigned int n;
- char *q;
- unsigned char *p;
- FILE *f = stb_p_fopen(filename, "rb"); if (f == NULL) return NULL;
- fseek(f, 0, SEEK_END);
- n = ftell(f);
- fseek(f, 0, SEEK_SET);
- p = (unsigned char * ) malloc(n); if (p == NULL) return NULL;
- fread(p, 1, n, f);
- fclose(f);
- if (p == NULL) return NULL;
- if (p[0] != 0x57 || p[1] != 0xBc || p[2] || p[3]) { free(p); return NULL; }
- q = (char *) malloc(stb_decompress_length(p)+1);
- if (!q) { free(p); return NULL; }
- *len = stb_decompress((unsigned char *) q, p, n);
- if (*len) q[*len] = 0;
- free(p);
- return q;
- }
- #if 0
- // streaming decompressor
- static struct
- {
- stb__uchar *in_buffer;
- stb__uchar *match;
- stb__uint pending_literals;
- stb__uint pending_match;
- } xx;
- static void stb__match(stb_uchar *data, stb_uint length)
- {
- // INVERSE of memmove... write each byte before copying the next...
- assert (stb__dout + length <= stb__barrier);
- if (stb__dout + length > stb__barrier) { stb__dout += length; return; }
- if (data < stb__barrier2) { stb__dout = stb__barrier+1; return; }
- while (length--) *stb__dout++ = *data++;
- }
- static void stb__lit(stb_uchar *data, stb_uint length)
- {
- assert (stb__dout + length <= stb__barrier);
- if (stb__dout + length > stb__barrier) { stb__dout += length; return; }
- if (data < stb__barrier2) { stb__dout = stb__barrier+1; return; }
- memcpy(stb__dout, data, length);
- stb__dout += length;
- }
- static void sx_match(stb_uchar *data, stb_uint length)
- {
- xx.match = data;
- xx.pending_match = length;
- }
- static void sx_lit(stb_uchar *data, stb_uint length)
- {
- xx.pending_lit = length;
- }
- static int stb_decompress_token_state(void)
- {
- stb__uchar *i = xx.in_buffer;
- if (*i >= 0x20) { // use fewer if's for cases that expand small
- if (*i >= 0x80) sx_match(stb__dout-i[1]-1, i[0] - 0x80 + 1), i += 2;
- else if (*i >= 0x40) sx_match(stb__dout-(stb__in2(0) - 0x4000 + 1), i[2]+1), i += 3;
- else /* *i >= 0x20 */ sx_lit(i+1, i[0] - 0x20 + 1), i += 1;
- } else { // more ifs for cases that expand large, since overhead is amortized
- if (*i >= 0x18) sx_match(stb__dout-(stb__in3(0) - 0x180000 + 1), i[3]+1), i += 4;
- else if (*i >= 0x10) sx_match(stb__dout-(stb__in3(0) - 0x100000 + 1), stb__in2(3)+1), i += 5;
- else if (*i >= 0x08) sx_lit(i+2, stb__in2(0) - 0x0800 + 1), i += 2;
- else if (*i == 0x07) sx_lit(i+3, stb__in2(1) + 1), i += 3;
- else if (*i == 0x06) sx_match(stb__dout-(stb__in3(1)+1), i[4]+1), i += 5;
- else if (*i == 0x04) sx_match(stb__dout-(stb__in3(1)+1), stb__in2(4)+1), i += 6;
- else return 0;
- }
- xx.in_buffer = i;
- return 1;
- }
- #endif
- //////////////////// compressor ///////////////////////
- static unsigned int stb_matchlen(stb_uchar *m1, stb_uchar *m2, stb_uint maxlen)
- {
- stb_uint i;
- for (i=0; i < maxlen; ++i)
- if (m1[i] != m2[i]) return i;
- return i;
- }
- // simple implementation that just takes the source data in a big block
- static stb_uchar *stb__out;
- static FILE *stb__outfile;
- static stb_uint stb__outbytes;
- static void stb__write(unsigned char v)
- {
- fputc(v, stb__outfile);
- ++stb__outbytes;
- }
- #define stb_out(v) (stb__out ? (void)(*stb__out++ = (stb_uchar) (v)) : stb__write((stb_uchar) (v)))
- static void stb_out2(stb_uint v)
- {
- stb_out(v >> 8);
- stb_out(v);
- }
- static void stb_out3(stb_uint v) { stb_out(v >> 16); stb_out(v >> 8); stb_out(v); }
- static void stb_out4(stb_uint v) { stb_out(v >> 24); stb_out(v >> 16);
- stb_out(v >> 8 ); stb_out(v); }
- static void outliterals(stb_uchar *in, ptrdiff_t numlit)
- {
- while (numlit > 65536) {
- outliterals(in,65536);
- in += 65536;
- numlit -= 65536;
- }
- if (numlit == 0) ;
- else if (numlit <= 32) stb_out (0x000020 + (stb_uint) numlit-1);
- else if (numlit <= 2048) stb_out2(0x000800 + (stb_uint) numlit-1);
- else /* numlit <= 65536) */ stb_out3(0x070000 + (stb_uint) numlit-1);
- if (stb__out) {
- memcpy(stb__out,in,numlit);
- stb__out += numlit;
- } else
- fwrite(in, 1, numlit, stb__outfile);
- }
- static int stb__window = 0x40000; // 256K
- void stb_compress_window(int z)
- {
- if (z >= 0x1000000) z = 0x1000000; // limit of implementation
- if (z < 0x100) z = 0x100; // insanely small
- stb__window = z;
- }
- static int stb_not_crap(int best, int dist)
- {
- return ((best > 2 && dist <= 0x00100)
- || (best > 5 && dist <= 0x04000)
- || (best > 7 && dist <= 0x80000));
- }
- static stb_uint stb__hashsize = 32768;
- void stb_compress_hashsize(unsigned int y)
- {
- unsigned int z = 1024;
- while (z < y) z <<= 1;
- stb__hashsize = z >> 2; // pass in bytes, store #pointers
- }
- // note that you can play with the hashing functions all you
- // want without needing to change the decompressor
- #define stb__hc(q,h,c) (((h) << 7) + ((h) >> 25) + q[c])
- #define stb__hc2(q,h,c,d) (((h) << 14) + ((h) >> 18) + (q[c] << 7) + q[d])
- #define stb__hc3(q,c,d,e) ((q[c] << 14) + (q[d] << 7) + q[e])
- static stb_uint32 stb__running_adler;
- static int stb_compress_chunk(stb_uchar *history,
- stb_uchar *start,
- stb_uchar *end,
- int length,
- int *pending_literals,
- stb_uchar **chash,
- stb_uint mask)
- {
- int window = stb__window;
- stb_uint match_max;
- stb_uchar *lit_start = start - *pending_literals;
- stb_uchar *q = start;
- #define STB__SCRAMBLE(h) (((h) + ((h) >> 16)) & mask)
- // stop short of the end so we don't scan off the end doing
- // the hashing; this means we won't compress the last few bytes
- // unless they were part of something longer
- while (q < start+length && q+12 < end) {
- int m;
- stb_uint h1,h2,h3,h4, h;
- stb_uchar *t;
- int best = 2, dist=0;
- if (q+65536 > end)
- match_max = (stb_uint) (end-q);
- else
- match_max = 65536u;
- #define stb__nc(b,d) ((d) <= window && ((b) > 9 || stb_not_crap(b,d)))
- #define STB__TRY(t,p) /* avoid retrying a match we already tried */ \
- if (p ? dist != (int) (q-t) : 1) \
- if ((m = (int) stb_matchlen(t, q, match_max)) > best)\
- if (stb__nc(m,(int) (q-(t)))) \
- best = m, dist = (int) (q - (t))
- // rather than search for all matches, only try 4 candidate locations,
- // chosen based on 4 different hash functions of different lengths.
- // this strategy is inspired by LZO; hashing is unrolled here using the
- // 'hc' macro
- h = stb__hc3(q,0, 1, 2); h1 = STB__SCRAMBLE(h);
- t = chash[h1]; if (t) STB__TRY(t,0);
- h = stb__hc2(q,h, 3, 4); h2 = STB__SCRAMBLE(h);
- h = stb__hc2(q,h, 5, 6); t = chash[h2]; if (t) STB__TRY(t,1);
- h = stb__hc2(q,h, 7, 8); h3 = STB__SCRAMBLE(h);
- h = stb__hc2(q,h, 9,10); t = chash[h3]; if (t) STB__TRY(t,1);
- h = stb__hc2(q,h,11,12); h4 = STB__SCRAMBLE(h);
- t = chash[h4]; if (t) STB__TRY(t,1);
- // because we use a shared hash table, can only update it
- // _after_ we've probed all of them
- chash[h1] = chash[h2] = chash[h3] = chash[h4] = q;
- if (best > 2)
- assert(dist > 0);
- // see if our best match qualifies
- if (best < 3) { // fast path literals
- ++q;
- } else if (best > 2 && best <= 0x80 && dist <= 0x100) {
- outliterals(lit_start, q-lit_start); lit_start = (q += best);
- stb_out(0x80 + best-1);
- stb_out(dist-1);
- } else if (best > 5 && best <= 0x100 && dist <= 0x4000) {
- outliterals(lit_start, q-lit_start); lit_start = (q += best);
- stb_out2(0x4000 + dist-1);
- stb_out(best-1);
- } else if (best > 7 && best <= 0x100 && dist <= 0x80000) {
- outliterals(lit_start, q-lit_start); lit_start = (q += best);
- stb_out3(0x180000 + dist-1);
- stb_out(best-1);
- } else if (best > 8 && best <= 0x10000 && dist <= 0x80000) {
- outliterals(lit_start, q-lit_start); lit_start = (q += best);
- stb_out3(0x100000 + dist-1);
- stb_out2(best-1);
- } else if (best > 9 && dist <= 0x1000000) {
- if (best > 65536) best = 65536;
- outliterals(lit_start, q-lit_start); lit_start = (q += best);
- if (best <= 0x100) {
- stb_out(0x06);
- stb_out3(dist-1);
- stb_out(best-1);
- } else {
- stb_out(0x04);
- stb_out3(dist-1);
- stb_out2(best-1);
- }
- } else { // fallback literals if no match was a balanced tradeoff
- ++q;
- }
- }
- // if we didn't get all the way, add the rest to literals
- if (q-start < length)
- q = start+length;
- // the literals are everything from lit_start to q
- *pending_literals = (int) (q - lit_start);
- stb__running_adler = stb_adler32(stb__running_adler, start, (int) (q - start));
- return (int) (q - start);
- }
- static int stb_compress_inner(stb_uchar *input, stb_uint length)
- {
- int literals = 0;
- stb_uint len,i;
- stb_uchar **chash;
- chash = (stb_uchar**) malloc(stb__hashsize * sizeof(stb_uchar*));
- if (chash == NULL) return 0; // failure
- for (i=0; i < stb__hashsize; ++i)
- chash[i] = NULL;
- // stream signature
- stb_out(0x57); stb_out(0xbc);
- stb_out2(0);
- stb_out4(0); // 64-bit length requires 32-bit leading 0
- stb_out4(length);
- stb_out4(stb__window);
- stb__running_adler = 1;
- len = stb_compress_chunk(input, input, input+length, length, &literals, chash, stb__hashsize-1);
- assert(len == length);
- outliterals(input+length - literals, literals);
- free(chash);
- stb_out2(0x05fa); // end opcode
- stb_out4(stb__running_adler);
- return 1; // success
- }
- stb_uint stb_compress(stb_uchar *out, stb_uchar *input, stb_uint length)
- {
- stb__out = out;
- stb__outfile = NULL;
- stb_compress_inner(input, length);
- return (stb_uint) (stb__out - out);
- }
- int stb_compress_tofile(char *filename, char *input, unsigned int length)
- {
- //int maxlen = length + 512 + (length >> 2); // total guess
- //char *buffer = (char *) malloc(maxlen);
- //int blen = stb_compress((stb_uchar*)buffer, (stb_uchar*)input, length);
- stb__out = NULL;
- stb__outfile = stb_p_fopen(filename, "wb");
- if (!stb__outfile) return 0;
- stb__outbytes = 0;
- if (!stb_compress_inner((stb_uchar*)input, length))
- return 0;
- fclose(stb__outfile);
- return stb__outbytes;
- }
- int stb_compress_intofile(FILE *f, char *input, unsigned int length)
- {
- //int maxlen = length + 512 + (length >> 2); // total guess
- //char *buffer = (char*)malloc(maxlen);
- //int blen = stb_compress((stb_uchar*)buffer, (stb_uchar*)input, length);
- stb__out = NULL;
- stb__outfile = f;
- if (!stb__outfile) return 0;
- stb__outbytes = 0;
- if (!stb_compress_inner((stb_uchar*)input, length))
- return 0;
- return stb__outbytes;
- }
- ////////////////////// streaming I/O version /////////////////////
- static size_t stb_out_backpatch_id(void)
- {
- if (stb__out)
- return (size_t) stb__out;
- else
- return ftell(stb__outfile);
- }
- static void stb_out_backpatch(size_t id, stb_uint value)
- {
- stb_uchar data[4] = { (stb_uchar)(value >> 24), (stb_uchar)(value >> 16), (stb_uchar)(value >> 8), (stb_uchar)(value) };
- if (stb__out) {
- memcpy((void *) id, data, 4);
- } else {
- stb_uint where = ftell(stb__outfile);
- fseek(stb__outfile, (long) id, SEEK_SET);
- fwrite(data, 4, 1, stb__outfile);
- fseek(stb__outfile, where, SEEK_SET);
- }
- }
- // ok, the wraparound buffer was a total failure. let's instead
- // use a copying-in-place buffer, which lets us share the code.
- // This is way less efficient but it'll do for now.
- static struct
- {
- stb_uchar *buffer;
- int size; // physical size of buffer in bytes
- int valid; // amount of valid data in bytes
- int start; // bytes of data already output
- int window;
- int fsize;
- int pending_literals; // bytes not-quite output but counted in start
- int length_id;
- stb_uint total_bytes;
- stb_uchar **chash;
- stb_uint hashmask;
- } xtb;
- static int stb_compress_streaming_start(void)
- {
- stb_uint i;
- xtb.size = stb__window * 3;
- xtb.buffer = (stb_uchar*)malloc(xtb.size);
- if (!xtb.buffer) return 0;
- xtb.chash = (stb_uchar**)malloc(sizeof(*xtb.chash) * stb__hashsize);
- if (!xtb.chash) {
- free(xtb.buffer);
- return 0;
- }
- for (i=0; i < stb__hashsize; ++i)
- xtb.chash[i] = NULL;
- xtb.hashmask = stb__hashsize-1;
- xtb.valid = 0;
- xtb.start = 0;
- xtb.window = stb__window;
- xtb.fsize = stb__window;
- xtb.pending_literals = 0;
- xtb.total_bytes = 0;
- // stream signature
- stb_out(0x57); stb_out(0xbc); stb_out2(0);
- stb_out4(0); // 64-bit length requires 32-bit leading 0
- xtb.length_id = (int) stb_out_backpatch_id();
- stb_out4(0); // we don't know the output length yet
- stb_out4(stb__window);
- stb__running_adler = 1;
- return 1;
- }
- static int stb_compress_streaming_end(void)
- {
- // flush out any remaining data
- stb_compress_chunk(xtb.buffer, xtb.buffer+xtb.start, xtb.buffer+xtb.valid,
- xtb.valid-xtb.start, &xtb.pending_literals, xtb.chash, xtb.hashmask);
- // write out pending literals
- outliterals(xtb.buffer + xtb.valid - xtb.pending_literals, xtb.pending_literals);
- stb_out2(0x05fa); // end opcode
- stb_out4(stb__running_adler);
- stb_out_backpatch(xtb.length_id, xtb.total_bytes);
- free(xtb.buffer);
- free(xtb.chash);
- return 1;
- }
- void stb_write(char *data, int data_len)
- {
- stb_uint i;
- // @TODO: fast path for filling the buffer and doing nothing else
- // if (xtb.valid + data_len < xtb.size)
- xtb.total_bytes += data_len;
- while (data_len) {
- // fill buffer
- if (xtb.valid < xtb.size) {
- int amt = xtb.size - xtb.valid;
- if (data_len < amt) amt = data_len;
- memcpy(xtb.buffer + xtb.valid, data, amt);
- data_len -= amt;
- data += amt;
- xtb.valid += amt;
- }
- if (xtb.valid < xtb.size)
- return;
- // at this point, the buffer is full
- // if we can process some data, go for it; make sure
- // we leave an 'fsize's worth of data, though
- if (xtb.start + xtb.fsize < xtb.valid) {
- int amount = (xtb.valid - xtb.fsize) - xtb.start;
- int n;
- assert(amount > 0);
- n = stb_compress_chunk(xtb.buffer, xtb.buffer + xtb.start, xtb.buffer + xtb.valid,
- amount, &xtb.pending_literals, xtb.chash, xtb.hashmask);
- xtb.start += n;
- }
- assert(xtb.start + xtb.fsize >= xtb.valid);
- // at this point, our future size is too small, so we
- // need to flush some history. we, in fact, flush exactly
- // one window's worth of history
- {
- int flush = xtb.window;
- assert(xtb.start >= flush);
- assert(xtb.valid >= flush);
- // if 'pending literals' extends back into the shift region,
- // write them out
- if (xtb.start - xtb.pending_literals < flush) {
- outliterals(xtb.buffer + xtb.start - xtb.pending_literals, xtb.pending_literals);
- xtb.pending_literals = 0;
- }
- // now shift the window
- memmove(xtb.buffer, xtb.buffer + flush, xtb.valid - flush);
- xtb.start -= flush;
- xtb.valid -= flush;
- for (i=0; i <= xtb.hashmask; ++i)
- if (xtb.chash[i] < xtb.buffer + flush)
- xtb.chash[i] = NULL;
- else
- xtb.chash[i] -= flush;
- }
- // and now that we've made room for more data, go back to the top
- }
- }
- int stb_compress_stream_start(FILE *f)
- {
- stb__out = NULL;
- stb__outfile = f;
- if (f == NULL)
- return 0;
- if (!stb_compress_streaming_start())
- return 0;
- return 1;
- }
- void stb_compress_stream_end(int close)
- {
- stb_compress_streaming_end();
- if (close && stb__outfile) {
- fclose(stb__outfile);
- }
- }
- #endif // STB_DEFINE
- //////////////////////////////////////////////////////////////////////////////
- //
- // File abstraction... tired of not having this... we can write
- // compressors to be layers over these that auto-close their children.
- typedef struct stbfile
- {
- int (*getbyte)(struct stbfile *); // -1 on EOF
- unsigned int (*getdata)(struct stbfile *, void *block, unsigned int len);
- int (*putbyte)(struct stbfile *, int byte);
- unsigned int (*putdata)(struct stbfile *, void *block, unsigned int len);
- unsigned int (*size)(struct stbfile *);
- unsigned int (*tell)(struct stbfile *);
- void (*backpatch)(struct stbfile *, unsigned int tell, void *block, unsigned int len);
- void (*close)(struct stbfile *);
- FILE *f; // file to fread/fwrite
- unsigned char *buffer; // input/output buffer
- unsigned char *indata, *inend; // input buffer
- union {
- int various;
- void *ptr;
- };
- } stbfile;
- STB_EXTERN unsigned int stb_getc(stbfile *f); // read
- STB_EXTERN int stb_putc(stbfile *f, int ch); // write
- STB_EXTERN unsigned int stb_getdata(stbfile *f, void *buffer, unsigned int len); // read
- STB_EXTERN unsigned int stb_putdata(stbfile *f, void *buffer, unsigned int len); // write
- STB_EXTERN unsigned int stb_tell(stbfile *f); // read
- STB_EXTERN unsigned int stb_size(stbfile *f); // read/write
- STB_EXTERN void stb_backpatch(stbfile *f, unsigned int tell, void *buffer, unsigned int len); // write
- #ifdef STB_DEFINE
- unsigned int stb_getc(stbfile *f) { return f->getbyte(f); }
- int stb_putc(stbfile *f, int ch) { return f->putbyte(f, ch); }
- unsigned int stb_getdata(stbfile *f, void *buffer, unsigned int len)
- {
- return f->getdata(f, buffer, len);
- }
- unsigned int stb_putdata(stbfile *f, void *buffer, unsigned int len)
- {
- return f->putdata(f, buffer, len);
- }
- void stb_close(stbfile *f)
- {
- f->close(f);
- free(f);
- }
- unsigned int stb_tell(stbfile *f) { return f->tell(f); }
- unsigned int stb_size(stbfile *f) { return f->size(f); }
- void stb_backpatch(stbfile *f, unsigned int tell, void *buffer, unsigned int len)
- {
- f->backpatch(f,tell,buffer,len);
- }
- // FILE * implementation
- static int stb__fgetbyte(stbfile *f) { return fgetc(f->f); }
- static int stb__fputbyte(stbfile *f, int ch) { return fputc(ch, f->f)==0; }
- static unsigned int stb__fgetdata(stbfile *f, void *buffer, unsigned int len) { return (unsigned int) fread(buffer,1,len,f->f); }
- static unsigned int stb__fputdata(stbfile *f, void *buffer, unsigned int len) { return (unsigned int) fwrite(buffer,1,len,f->f); }
- static unsigned int stb__fsize(stbfile *f) { return (unsigned int) stb_filelen(f->f); }
- static unsigned int stb__ftell(stbfile *f) { return (unsigned int) ftell(f->f); }
- static void stb__fbackpatch(stbfile *f, unsigned int where, void *buffer, unsigned int len)
- {
- fseek(f->f, where, SEEK_SET);
- fwrite(buffer, 1, len, f->f);
- fseek(f->f, 0, SEEK_END);
- }
- static void stb__fclose(stbfile *f) { fclose(f->f); }
- stbfile *stb_openf(FILE *f)
- {
- stbfile m = { stb__fgetbyte, stb__fgetdata,
- stb__fputbyte, stb__fputdata,
- stb__fsize, stb__ftell, stb__fbackpatch, stb__fclose,
- 0,0,0, };
- stbfile *z = (stbfile *) malloc(sizeof(*z));
- if (z) {
- *z = m;
- z->f = f;
- }
- return z;
- }
- static int stb__nogetbyte(stbfile *f) { assert(0); return -1; }
- static unsigned int stb__nogetdata(stbfile *f, void *buffer, unsigned int len) { assert(0); return 0; }
- static int stb__noputbyte(stbfile *f, int ch) { assert(0); return 0; }
- static unsigned int stb__noputdata(stbfile *f, void *buffer, unsigned int len) { assert(0); return 0; }
- static void stb__nobackpatch(stbfile *f, unsigned int where, void *buffer, unsigned int len) { assert(0); }
- static int stb__bgetbyte(stbfile *s)
- {
- if (s->indata < s->inend)
- return *s->indata++;
- else
- return -1;
- }
- static unsigned int stb__bgetdata(stbfile *s, void *buffer, unsigned int len)
- {
- if (s->indata + len > s->inend)
- len = (unsigned int) (s->inend - s->indata);
- memcpy(buffer, s->indata, len);
- s->indata += len;
- return len;
- }
- static unsigned int stb__bsize(stbfile *s) { return (unsigned int) (s->inend - s->buffer); }
- static unsigned int stb__btell(stbfile *s) { return (unsigned int) (s->indata - s->buffer); }
- static void stb__bclose(stbfile *s)
- {
- if (s->various)
- free(s->buffer);
- }
- stbfile *stb_open_inbuffer(void *buffer, unsigned int len)
- {
- stbfile m = { stb__bgetbyte, stb__bgetdata,
- stb__noputbyte, stb__noputdata,
- stb__bsize, stb__btell, stb__nobackpatch, stb__bclose };
- stbfile *z = (stbfile *) malloc(sizeof(*z));
- if (z) {
- *z = m;
- z->buffer = (unsigned char *) buffer;
- z->indata = z->buffer;
- z->inend = z->indata + len;
- }
- return z;
- }
- stbfile *stb_open_inbuffer_free(void *buffer, unsigned int len)
- {
- stbfile *z = stb_open_inbuffer(buffer, len);
- if (z)
- z->various = 1; // free
- return z;
- }
- #ifndef STB_VERSION
- // if we've been cut-and-pasted elsewhere, you get a limited
- // version of stb_open, without the 'k' flag and utf8 support
- static void stb__fclose2(stbfile *f)
- {
- fclose(f->f);
- }
- stbfile *stb_open(char *filename, char *mode)
- {
- FILE *f = stb_p_fopen(filename, mode);
- stbfile *s;
- if (f == NULL) return NULL;
- s = stb_openf(f);
- if (s)
- s->close = stb__fclose2;
- return s;
- }
- #else
- // the full version depends on some code in stb.h; this
- // also includes the memory buffer output format implemented with stb_arr
- static void stb__fclose2(stbfile *f)
- {
- stb_fclose(f->f, f->various);
- }
- stbfile *stb_open(char *filename, char *mode)
- {
- FILE *f = stb_fopen(filename, mode[0] == 'k' ? mode+1 : mode);
- stbfile *s;
- if (f == NULL) return NULL;
- s = stb_openf(f);
- if (s) {
- s->close = stb__fclose2;
- s->various = mode[0] == 'k' ? stb_keep_if_different : stb_keep_yes;
- }
- return s;
- }
- static int stb__aputbyte(stbfile *f, int ch)
- {
- stb_arr_push(f->buffer, ch);
- return 1;
- }
- static unsigned int stb__aputdata(stbfile *f, void *data, unsigned int len)
- {
- memcpy(stb_arr_addn(f->buffer, (int) len), data, len);
- return len;
- }
- static unsigned int stb__asize(stbfile *f) { return stb_arr_len(f->buffer); }
- static void stb__abackpatch(stbfile *f, unsigned int where, void *data, unsigned int len)
- {
- memcpy(f->buffer+where, data, len);
- }
- static void stb__aclose(stbfile *f)
- {
- *(unsigned char **) f->ptr = f->buffer;
- }
- stbfile *stb_open_outbuffer(unsigned char **update_on_close)
- {
- stbfile m = { stb__nogetbyte, stb__nogetdata,
- stb__aputbyte, stb__aputdata,
- stb__asize, stb__asize, stb__abackpatch, stb__aclose };
- stbfile *z = (stbfile *) malloc(sizeof(*z));
- if (z) {
- z->ptr = update_on_close;
- *z = m;
- }
- return z;
- }
- #endif
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Arithmetic coder... based on cbloom's notes on the subject, should be
- // less code than a huffman code.
- typedef struct
- {
- unsigned int range_low;
- unsigned int range_high;
- unsigned int code, range; // decode
- int buffered_u8;
- int pending_ffs;
- stbfile *output;
- } stb_arith;
- STB_EXTERN void stb_arith_init_encode(stb_arith *a, stbfile *out);
- STB_EXTERN void stb_arith_init_decode(stb_arith *a, stbfile *in);
- STB_EXTERN stbfile *stb_arith_encode_close(stb_arith *a);
- STB_EXTERN stbfile *stb_arith_decode_close(stb_arith *a);
- STB_EXTERN void stb_arith_encode(stb_arith *a, unsigned int totalfreq, unsigned int freq, unsigned int cumfreq);
- STB_EXTERN void stb_arith_encode_log2(stb_arith *a, unsigned int totalfreq2, unsigned int freq, unsigned int cumfreq);
- STB_EXTERN unsigned int stb_arith_decode_value(stb_arith *a, unsigned int totalfreq);
- STB_EXTERN void stb_arith_decode_advance(stb_arith *a, unsigned int totalfreq, unsigned int freq, unsigned int cumfreq);
- STB_EXTERN unsigned int stb_arith_decode_value_log2(stb_arith *a, unsigned int totalfreq2);
- STB_EXTERN void stb_arith_decode_advance_log2(stb_arith *a, unsigned int totalfreq2, unsigned int freq, unsigned int cumfreq);
- STB_EXTERN void stb_arith_encode_byte(stb_arith *a, int byte);
- STB_EXTERN int stb_arith_decode_byte(stb_arith *a);
- // this is a memory-inefficient way of doing things, but it's
- // fast(?) and simple
- typedef struct
- {
- unsigned short cumfreq;
- unsigned short samples;
- } stb_arith_symstate_item;
- typedef struct
- {
- int num_sym;
- unsigned int pow2;
- int countdown;
- stb_arith_symstate_item data[1];
- } stb_arith_symstate;
- #ifdef STB_DEFINE
- void stb_arith_init_encode(stb_arith *a, stbfile *out)
- {
- a->range_low = 0;
- a->range_high = 0xffffffff;
- a->pending_ffs = -1; // means no buffered character currently, to speed up normal case
- a->output = out;
- }
- static void stb__arith_carry(stb_arith *a)
- {
- int i;
- assert(a->pending_ffs != -1); // can't carry with no data
- stb_putc(a->output, a->buffered_u8);
- for (i=0; i < a->pending_ffs; ++i)
- stb_putc(a->output, 0);
- }
- static void stb__arith_putbyte(stb_arith *a, int byte)
- {
- if (a->pending_ffs) {
- if (a->pending_ffs == -1) { // means no buffered data; encoded for fast path efficiency
- if (byte == 0xff)
- stb_putc(a->output, byte); // just write it immediately
- else {
- a->buffered_u8 = byte;
- a->pending_ffs = 0;
- }
- } else if (byte == 0xff) {
- ++a->pending_ffs;
- } else {
- int i;
- stb_putc(a->output, a->buffered_u8);
- for (i=0; i < a->pending_ffs; ++i)
- stb_putc(a->output, 0xff);
- }
- } else if (byte == 0xff) {
- ++a->pending_ffs;
- } else {
- // fast path
- stb_putc(a->output, a->buffered_u8);
- a->buffered_u8 = byte;
- }
- }
- static void stb__arith_flush(stb_arith *a)
- {
- if (a->pending_ffs >= 0) {
- int i;
- stb_putc(a->output, a->buffered_u8);
- for (i=0; i < a->pending_ffs; ++i)
- stb_putc(a->output, 0xff);
- }
- }
- static void stb__renorm_encoder(stb_arith *a)
- {
- stb__arith_putbyte(a, a->range_low >> 24);
- a->range_low <<= 8;
- a->range_high = (a->range_high << 8) | 0xff;
- }
- static void stb__renorm_decoder(stb_arith *a)
- {
- int c = stb_getc(a->output);
- a->code = (a->code << 8) + (c >= 0 ? c : 0); // if EOF, insert 0
- }
- void stb_arith_encode(stb_arith *a, unsigned int totalfreq, unsigned int freq, unsigned int cumfreq)
- {
- unsigned int range = a->range_high - a->range_low;
- unsigned int old = a->range_low;
- range /= totalfreq;
- a->range_low += range * cumfreq;
- a->range_high = a->range_low + range*freq;
- if (a->range_low < old)
- stb__arith_carry(a);
- while (a->range_high - a->range_low < 0x1000000)
- stb__renorm_encoder(a);
- }
- void stb_arith_encode_log2(stb_arith *a, unsigned int totalfreq2, unsigned int freq, unsigned int cumfreq)
- {
- unsigned int range = a->range_high - a->range_low;
- unsigned int old = a->range_low;
- range >>= totalfreq2;
- a->range_low += range * cumfreq;
- a->range_high = a->range_low + range*freq;
- if (a->range_low < old)
- stb__arith_carry(a);
- while (a->range_high - a->range_low < 0x1000000)
- stb__renorm_encoder(a);
- }
- unsigned int stb_arith_decode_value(stb_arith *a, unsigned int totalfreq)
- {
- unsigned int freqsize = a->range / totalfreq;
- unsigned int z = a->code / freqsize;
- return z >= totalfreq ? totalfreq-1 : z;
- }
- void stb_arith_decode_advance(stb_arith *a, unsigned int totalfreq, unsigned int freq, unsigned int cumfreq)
- {
- unsigned int freqsize = a->range / totalfreq; // @OPTIMIZE, share with above divide somehow?
- a->code -= freqsize * cumfreq;
- a->range = freqsize * freq;
- while (a->range < 0x1000000)
- stb__renorm_decoder(a);
- }
- unsigned int stb_arith_decode_value_log2(stb_arith *a, unsigned int totalfreq2)
- {
- unsigned int freqsize = a->range >> totalfreq2;
- unsigned int z = a->code / freqsize;
- return z >= (1U<<totalfreq2) ? (1U<<totalfreq2)-1 : z;
- }
- void stb_arith_decode_advance_log2(stb_arith *a, unsigned int totalfreq2, unsigned int freq, unsigned int cumfreq)
- {
- unsigned int freqsize = a->range >> totalfreq2;
- a->code -= freqsize * cumfreq;
- a->range = freqsize * freq;
- while (a->range < 0x1000000)
- stb__renorm_decoder(a);
- }
- stbfile *stb_arith_encode_close(stb_arith *a)
- {
- // put exactly as many bytes as we'll read, so we can turn on/off arithmetic coding in a stream
- stb__arith_putbyte(a, a->range_low >> 24);
- stb__arith_putbyte(a, a->range_low >> 16);
- stb__arith_putbyte(a, a->range_low >> 8);
- stb__arith_putbyte(a, a->range_low >> 0);
- stb__arith_flush(a);
- return a->output;
- }
- stbfile *stb_arith_decode_close(stb_arith *a)
- {
- return a->output;
- }
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Threads
- //
- #ifndef _WIN32
- #ifdef STB_THREADS
- #error "threads not implemented except for Windows"
- #endif
- #endif
- // call this function to free any global variables for memory testing
- STB_EXTERN void stb_thread_cleanup(void);
- typedef void * (*stb_thread_func)(void *);
- // do not rely on these types, this is an implementation detail.
- // compare against STB_THREAD_NULL and ST_SEMAPHORE_NULL
- typedef void *stb_thread;
- typedef void *stb_semaphore;
- typedef void *stb_mutex;
- typedef struct stb__sync *stb_sync;
- #define STB_SEMAPHORE_NULL NULL
- #define STB_THREAD_NULL NULL
- #define STB_MUTEX_NULL NULL
- #define STB_SYNC_NULL NULL
- // get the number of processors (limited to those in the affinity mask for this process).
- STB_EXTERN int stb_processor_count(void);
- // force to run on a single core -- needed for RDTSC to work, e.g. for iprof
- STB_EXTERN void stb_force_uniprocessor(void);
- // stb_work functions: queue up work to be done by some worker threads
- // set number of threads to serve the queue; you can change this on the fly,
- // but if you decrease it, it won't decrease until things currently on the
- // queue are finished
- STB_EXTERN void stb_work_numthreads(int n);
- // set maximum number of units in the queue; you can only set this BEFORE running any work functions
- STB_EXTERN int stb_work_maxunits(int n);
- // enqueue some work to be done (can do this from any thread, or even from a piece of work);
- // return value of f is stored in *return_code if non-NULL
- STB_EXTERN int stb_work(stb_thread_func f, void *d, volatile void **return_code);
- // as above, but stb_sync_reach is called on 'rel' after work is complete
- STB_EXTERN int stb_work_reach(stb_thread_func f, void *d, volatile void **return_code, stb_sync rel);
- // necessary to call this when using volatile to order writes/reads
- STB_EXTERN void stb_barrier(void);
- // support for independent queues with their own threads
- typedef struct stb__workqueue stb_workqueue;
- STB_EXTERN stb_workqueue*stb_workq_new(int numthreads, int max_units);
- STB_EXTERN stb_workqueue*stb_workq_new_flags(int numthreads, int max_units, int no_add_mutex, int no_remove_mutex);
- STB_EXTERN void stb_workq_delete(stb_workqueue *q);
- STB_EXTERN void stb_workq_numthreads(stb_workqueue *q, int n);
- STB_EXTERN int stb_workq(stb_workqueue *q, stb_thread_func f, void *d, volatile void **return_code);
- STB_EXTERN int stb_workq_reach(stb_workqueue *q, stb_thread_func f, void *d, volatile void **return_code, stb_sync rel);
- STB_EXTERN int stb_workq_length(stb_workqueue *q);
- STB_EXTERN stb_thread stb_create_thread (stb_thread_func f, void *d);
- STB_EXTERN stb_thread stb_create_thread2(stb_thread_func f, void *d, volatile void **return_code, stb_semaphore rel);
- STB_EXTERN void stb_destroy_thread(stb_thread t);
- STB_EXTERN stb_semaphore stb_sem_new(int max_val);
- STB_EXTERN stb_semaphore stb_sem_new_extra(int max_val, int start_val);
- STB_EXTERN void stb_sem_delete (stb_semaphore s);
- STB_EXTERN void stb_sem_waitfor(stb_semaphore s);
- STB_EXTERN void stb_sem_release(stb_semaphore s);
- STB_EXTERN stb_mutex stb_mutex_new(void);
- STB_EXTERN void stb_mutex_delete(stb_mutex m);
- STB_EXTERN void stb_mutex_begin(stb_mutex m);
- STB_EXTERN void stb_mutex_end(stb_mutex m);
- STB_EXTERN stb_sync stb_sync_new(void);
- STB_EXTERN void stb_sync_delete(stb_sync s);
- STB_EXTERN int stb_sync_set_target(stb_sync s, int count);
- STB_EXTERN void stb_sync_reach_and_wait(stb_sync s); // wait for 'target' reachers
- STB_EXTERN int stb_sync_reach(stb_sync s);
- typedef struct stb__threadqueue stb_threadqueue;
- #define STB_THREADQ_DYNAMIC 0
- STB_EXTERN stb_threadqueue *stb_threadq_new(int item_size, int num_items, int many_add, int many_remove);
- STB_EXTERN void stb_threadq_delete(stb_threadqueue *tq);
- STB_EXTERN int stb_threadq_get(stb_threadqueue *tq, void *output);
- STB_EXTERN void stb_threadq_get_block(stb_threadqueue *tq, void *output);
- STB_EXTERN int stb_threadq_add(stb_threadqueue *tq, void *input);
- // can return FALSE if STB_THREADQ_DYNAMIC and attempt to grow fails
- STB_EXTERN int stb_threadq_add_block(stb_threadqueue *tq, void *input);
- #ifdef STB_THREADS
- #ifdef STB_DEFINE
- typedef struct
- {
- stb_thread_func f;
- void *d;
- volatile void **return_val;
- stb_semaphore sem;
- } stb__thread;
- // this is initialized along all possible paths to create threads, therefore
- // it's always initialized before any other threads are create, therefore
- // it's free of races AS LONG AS you only create threads through stb_*
- static stb_mutex stb__threadmutex, stb__workmutex;
- static void stb__threadmutex_init(void)
- {
- if (stb__threadmutex == STB_SEMAPHORE_NULL) {
- stb__threadmutex = stb_mutex_new();
- stb__workmutex = stb_mutex_new();
- }
- }
- #ifdef STB_THREAD_TEST
- volatile float stb__t1=1, stb__t2;
- static void stb__wait(int n)
- {
- float z = 0;
- int i;
- for (i=0; i < n; ++i)
- z += 1 / (stb__t1+i);
- stb__t2 = z;
- }
- #else
- #define stb__wait(x)
- #endif
- #ifdef _WIN32
- // avoid including windows.h -- note that our definitions aren't
- // exactly the same (we don't define the security descriptor struct)
- // so if you want to include windows.h, make sure you do it first.
- #include <process.h>
- #ifndef _WINDOWS_ // check windows.h guard
- #define STB__IMPORT STB_EXTERN __declspec(dllimport)
- #define STB__DW unsigned long
- STB__IMPORT int __stdcall TerminateThread(void *, STB__DW);
- STB__IMPORT void * __stdcall CreateSemaphoreA(void *sec, long,long,char*);
- STB__IMPORT int __stdcall CloseHandle(void *);
- STB__IMPORT STB__DW __stdcall WaitForSingleObject(void *, STB__DW);
- STB__IMPORT int __stdcall ReleaseSemaphore(void *, long, long *);
- STB__IMPORT void __stdcall Sleep(STB__DW);
- #endif
- // necessary to call this when using volatile to order writes/reads
- void stb_barrier(void)
- {
- #ifdef MemoryBarrier
- MemoryBarrier();
- #else
- long temp;
- __asm xchg temp,eax;
- #endif
- }
- static void stb__thread_run(void *t)
- {
- void *res;
- stb__thread info = * (stb__thread *) t;
- free(t);
- res = info.f(info.d);
- if (info.return_val)
- *info.return_val = res;
- if (info.sem != STB_SEMAPHORE_NULL)
- stb_sem_release(info.sem);
- }
- static stb_thread stb_create_thread_raw(stb_thread_func f, void *d, volatile void **return_code, stb_semaphore rel)
- {
- #ifdef _MT
- #if defined(STB_FASTMALLOC) && !defined(STB_FASTMALLOC_ITS_OKAY_I_ONLY_MALLOC_IN_ONE_THREAD)
- stb_fatal("Error! Cannot use STB_FASTMALLOC with threads.\n");
- return STB_THREAD_NULL;
- #else
- unsigned long id;
- stb__thread *data = (stb__thread *) malloc(sizeof(*data));
- if (!data) return NULL;
- stb__threadmutex_init();
- data->f = f;
- data->d = d;
- data->return_val = return_code;
- data->sem = rel;
- id = _beginthread(stb__thread_run, 0, data);
- if (id == -1) return NULL;
- return (void *) id;
- #endif
- #else
- #ifdef STB_NO_STB_STRINGS
- stb_fatal("Invalid compilation");
- #else
- stb_fatal("Must compile mult-threaded to use stb_thread/stb_work.");
- #endif
- return NULL;
- #endif
- }
- // trivial win32 wrappers
- void stb_destroy_thread(stb_thread t) { TerminateThread(t,0); }
- stb_semaphore stb_sem_new(int maxv) {return CreateSemaphoreA(NULL,0,maxv,NULL); }
- stb_semaphore stb_sem_new_extra(int maxv,int start){return CreateSemaphoreA(NULL,start,maxv,NULL); }
- void stb_sem_delete(stb_semaphore s) { if (s != NULL) CloseHandle(s); }
- void stb_sem_waitfor(stb_semaphore s) { WaitForSingleObject(s, 0xffffffff); } // INFINITE
- void stb_sem_release(stb_semaphore s) { ReleaseSemaphore(s,1,NULL); }
- static void stb__thread_sleep(int ms) { Sleep(ms); }
- #ifndef _WINDOWS_
- STB__IMPORT int __stdcall GetProcessAffinityMask(void *, STB__DW *, STB__DW *);
- STB__IMPORT void * __stdcall GetCurrentProcess(void);
- STB__IMPORT int __stdcall SetProcessAffinityMask(void *, STB__DW);
- #endif
- int stb_processor_count(void)
- {
- unsigned long proc,sys;
- GetProcessAffinityMask(GetCurrentProcess(), &proc, &sys);
- return stb_bitcount(proc);
- }
- void stb_force_uniprocessor(void)
- {
- unsigned long proc,sys;
- GetProcessAffinityMask(GetCurrentProcess(), &proc, &sys);
- if (stb_bitcount(proc) > 1) {
- int z;
- for (z=0; z < 32; ++z)
- if (proc & (1 << z))
- break;
- if (z < 32) {
- proc = 1 << z;
- SetProcessAffinityMask(GetCurrentProcess(), proc);
- }
- }
- }
- #ifdef _WINDOWS_
- #define STB_MUTEX_NATIVE
- void *stb_mutex_new(void)
- {
- CRITICAL_SECTION *p = (CRITICAL_SECTION *) malloc(sizeof(*p));
- if (p)
- #if _WIN32_WINNT >= 0x0500
- InitializeCriticalSectionAndSpinCount(p, 500);
- #else
- InitializeCriticalSection(p);
- #endif
- return p;
- }
- void stb_mutex_delete(void *p)
- {
- if (p) {
- DeleteCriticalSection((CRITICAL_SECTION *) p);
- free(p);
- }
- }
- void stb_mutex_begin(void *p)
- {
- stb__wait(500);
- if (p)
- EnterCriticalSection((CRITICAL_SECTION *) p);
- }
- void stb_mutex_end(void *p)
- {
- if (p)
- LeaveCriticalSection((CRITICAL_SECTION *) p);
- stb__wait(500);
- }
- #endif // _WINDOWS_
- #if 0
- // for future reference,
- // InterlockedCompareExchange for x86:
- int cas64_mp(void * dest, void * xcmp, void * xxchg) {
- __asm
- {
- mov esi, [xxchg] ; exchange
- mov ebx, [esi + 0]
- mov ecx, [esi + 4]
- mov esi, [xcmp] ; comparand
- mov eax, [esi + 0]
- mov edx, [esi + 4]
- mov edi, [dest] ; destination
- lock cmpxchg8b [edi]
- jz yyyy;
- mov [esi + 0], eax;
- mov [esi + 4], edx;
- yyyy:
- xor eax, eax;
- setz al;
- };
- inline unsigned __int64 _InterlockedCompareExchange64(volatile unsigned __int64 *dest
- ,unsigned __int64 exchange
- ,unsigned __int64 comperand)
- {
- //value returned in eax::edx
- __asm {
- lea esi,comperand;
- lea edi,exchange;
- mov eax,[esi];
- mov edx,4[esi];
- mov ebx,[edi];
- mov ecx,4[edi];
- mov esi,dest;
- lock CMPXCHG8B [esi];
- }
- #endif // #if 0
- #endif // _WIN32
- stb_thread stb_create_thread2(stb_thread_func f, void *d, volatile void **return_code, stb_semaphore rel)
- {
- return stb_create_thread_raw(f,d,return_code,rel);
- }
- stb_thread stb_create_thread(stb_thread_func f, void *d)
- {
- return stb_create_thread2(f,d,NULL,STB_SEMAPHORE_NULL);
- }
- // mutex implemented by wrapping semaphore
- #ifndef STB_MUTEX_NATIVE
- stb_mutex stb_mutex_new(void) { return stb_sem_new_extra(1,1); }
- void stb_mutex_delete(stb_mutex m) { stb_sem_delete (m); }
- void stb_mutex_begin(stb_mutex m) { stb__wait(500); if (m) stb_sem_waitfor(m); }
- void stb_mutex_end(stb_mutex m) { if (m) stb_sem_release(m); stb__wait(500); }
- #endif
- // thread merge operation
- struct stb__sync
- {
- int target; // target number of threads to hit it
- int sofar; // total threads that hit it
- int waiting; // total threads waiting
- stb_mutex start; // mutex to prevent starting again before finishing previous
- stb_mutex mutex; // mutex while tweaking state
- stb_semaphore release; // semaphore wake up waiting threads
- // we have to wake them up one at a time, rather than using a single release
- // call, because win32 semaphores don't let you dynamically change the max count!
- };
- stb_sync stb_sync_new(void)
- {
- stb_sync s = (stb_sync) malloc(sizeof(*s));
- if (!s) return s;
- s->target = s->sofar = s->waiting = 0;
- s->mutex = stb_mutex_new();
- s->start = stb_mutex_new();
- s->release = stb_sem_new(1);
- if (s->mutex == STB_MUTEX_NULL || s->release == STB_SEMAPHORE_NULL || s->start == STB_MUTEX_NULL) {
- stb_mutex_delete(s->mutex);
- stb_mutex_delete(s->mutex);
- stb_sem_delete(s->release);
- free(s);
- return NULL;
- }
- return s;
- }
- void stb_sync_delete(stb_sync s)
- {
- if (s->waiting) {
- // it's bad to delete while there are threads waiting!
- // shall we wait for them to reach, or just bail? just bail
- assert(0);
- }
- stb_mutex_delete(s->mutex);
- stb_mutex_delete(s->release);
- free(s);
- }
- int stb_sync_set_target(stb_sync s, int count)
- {
- // don't allow setting a target until the last one is fully released;
- // note that this can lead to inefficient pipelining, and maybe we'd
- // be better off ping-ponging between two internal syncs?
- // I tried seeing how often this happened using TryEnterCriticalSection
- // and could _never_ get it to happen in imv(stb), even with more threads
- // than processors. So who knows!
- stb_mutex_begin(s->start);
- // this mutex is pointless, since it's not valid for threads
- // to call reach() before anyone calls set_target() anyway
- stb_mutex_begin(s->mutex);
- assert(s->target == 0); // enforced by start mutex
- s->target = count;
- s->sofar = 0;
- s->waiting = 0;
- stb_mutex_end(s->mutex);
- return STB_TRUE;
- }
- void stb__sync_release(stb_sync s)
- {
- if (s->waiting)
- stb_sem_release(s->release);
- else {
- s->target = 0;
- stb_mutex_end(s->start);
- }
- }
- int stb_sync_reach(stb_sync s)
- {
- int n;
- stb_mutex_begin(s->mutex);
- assert(s->sofar < s->target);
- n = ++s->sofar; // record this value to avoid possible race if we did 'return s->sofar';
- if (s->sofar == s->target)
- stb__sync_release(s);
- stb_mutex_end(s->mutex);
- return n;
- }
- void stb_sync_reach_and_wait(stb_sync s)
- {
- stb_mutex_begin(s->mutex);
- assert(s->sofar < s->target);
- ++s->sofar;
- if (s->sofar == s->target) {
- stb__sync_release(s);
- stb_mutex_end(s->mutex);
- } else {
- ++s->waiting; // we're waiting, so one more waiter
- stb_mutex_end(s->mutex); // release the mutex to other threads
- stb_sem_waitfor(s->release); // wait for merge completion
- stb_mutex_begin(s->mutex); // on merge completion, grab the mutex
- --s->waiting; // we're done waiting
- stb__sync_release(s); // restart the next waiter
- stb_mutex_end(s->mutex); // and now we're done
- // this ends the same as the first case, but it's a lot
- // clearer to understand without sharing the code
- }
- }
- struct stb__threadqueue
- {
- stb_mutex add, remove;
- stb_semaphore nonempty, nonfull;
- int head_blockers; // number of threads blocking--used to know whether to release(avail)
- int tail_blockers;
- int head, tail, array_size, growable;
- int item_size;
- char *data;
- };
- static int stb__tq_wrap(volatile stb_threadqueue *z, int p)
- {
- if (p == z->array_size)
- return p - z->array_size;
- else
- return p;
- }
- int stb__threadq_get_raw(stb_threadqueue *tq2, void *output, int block)
- {
- volatile stb_threadqueue *tq = (volatile stb_threadqueue *) tq2;
- if (tq->head == tq->tail && !block) return 0;
- stb_mutex_begin(tq->remove);
- while (tq->head == tq->tail) {
- if (!block) {
- stb_mutex_end(tq->remove);
- return 0;
- }
- ++tq->head_blockers;
- stb_mutex_end(tq->remove);
- stb_sem_waitfor(tq->nonempty);
- stb_mutex_begin(tq->remove);
- --tq->head_blockers;
- }
- memcpy(output, tq->data + tq->head*tq->item_size, tq->item_size);
- stb_barrier();
- tq->head = stb__tq_wrap(tq, tq->head+1);
- stb_sem_release(tq->nonfull);
- if (tq->head_blockers) // can't check if actually non-empty due to race?
- stb_sem_release(tq->nonempty); // if there are other blockers, wake one
- stb_mutex_end(tq->remove);
- return STB_TRUE;
- }
- int stb__threadq_grow(volatile stb_threadqueue *tq)
- {
- int n;
- char *p;
- assert(tq->remove != STB_MUTEX_NULL); // must have this to allow growth!
- stb_mutex_begin(tq->remove);
- n = tq->array_size * 2;
- p = (char *) realloc(tq->data, n * tq->item_size);
- if (p == NULL) {
- stb_mutex_end(tq->remove);
- stb_mutex_end(tq->add);
- return STB_FALSE;
- }
- if (tq->tail < tq->head) {
- memcpy(p + tq->array_size * tq->item_size, p, tq->tail * tq->item_size);
- tq->tail += tq->array_size;
- }
- tq->data = p;
- tq->array_size = n;
- stb_mutex_end(tq->remove);
- return STB_TRUE;
- }
- int stb__threadq_add_raw(stb_threadqueue *tq2, void *input, int block)
- {
- int tail,pos;
- volatile stb_threadqueue *tq = (volatile stb_threadqueue *) tq2;
- stb_mutex_begin(tq->add);
- for(;;) {
- pos = tq->tail;
- tail = stb__tq_wrap(tq, pos+1);
- if (tail != tq->head) break;
- // full
- if (tq->growable) {
- if (!stb__threadq_grow(tq)) {
- stb_mutex_end(tq->add);
- return STB_FALSE; // out of memory
- }
- } else if (!block) {
- stb_mutex_end(tq->add);
- return STB_FALSE;
- } else {
- ++tq->tail_blockers;
- stb_mutex_end(tq->add);
- stb_sem_waitfor(tq->nonfull);
- stb_mutex_begin(tq->add);
- --tq->tail_blockers;
- }
- }
- memcpy(tq->data + tq->item_size * pos, input, tq->item_size);
- stb_barrier();
- tq->tail = tail;
- stb_sem_release(tq->nonempty);
- if (tq->tail_blockers) // can't check if actually non-full due to race?
- stb_sem_release(tq->nonfull);
- stb_mutex_end(tq->add);
- return STB_TRUE;
- }
- int stb_threadq_length(stb_threadqueue *tq2)
- {
- int a,b,n;
- volatile stb_threadqueue *tq = (volatile stb_threadqueue *) tq2;
- stb_mutex_begin(tq->add);
- a = tq->head;
- b = tq->tail;
- n = tq->array_size;
- stb_mutex_end(tq->add);
- if (a > b) b += n;
- return b-a;
- }
- int stb_threadq_get(stb_threadqueue *tq, void *output)
- {
- return stb__threadq_get_raw(tq, output, STB_FALSE);
- }
- void stb_threadq_get_block(stb_threadqueue *tq, void *output)
- {
- stb__threadq_get_raw(tq, output, STB_TRUE);
- }
- int stb_threadq_add(stb_threadqueue *tq, void *input)
- {
- return stb__threadq_add_raw(tq, input, STB_FALSE);
- }
- int stb_threadq_add_block(stb_threadqueue *tq, void *input)
- {
- return stb__threadq_add_raw(tq, input, STB_TRUE);
- }
- void stb_threadq_delete(stb_threadqueue *tq)
- {
- if (tq) {
- free(tq->data);
- stb_mutex_delete(tq->add);
- stb_mutex_delete(tq->remove);
- stb_sem_delete(tq->nonempty);
- stb_sem_delete(tq->nonfull);
- free(tq);
- }
- }
- #define STB_THREADQUEUE_DYNAMIC 0
- stb_threadqueue *stb_threadq_new(int item_size, int num_items, int many_add, int many_remove)
- {
- int error=0;
- stb_threadqueue *tq = (stb_threadqueue *) malloc(sizeof(*tq));
- if (tq == NULL) return NULL;
- if (num_items == STB_THREADQUEUE_DYNAMIC) {
- tq->growable = STB_TRUE;
- num_items = 32;
- } else
- tq->growable = STB_FALSE;
- tq->item_size = item_size;
- tq->array_size = num_items+1;
- tq->add = tq->remove = STB_MUTEX_NULL;
- tq->nonempty = tq->nonfull = STB_SEMAPHORE_NULL;
- tq->data = NULL;
- if (many_add)
- { tq->add = stb_mutex_new(); if (tq->add == STB_MUTEX_NULL) goto error; }
- if (many_remove || tq->growable)
- { tq->remove = stb_mutex_new(); if (tq->remove == STB_MUTEX_NULL) goto error; }
- tq->nonempty = stb_sem_new(1); if (tq->nonempty == STB_SEMAPHORE_NULL) goto error;
- tq->nonfull = stb_sem_new(1); if (tq->nonfull == STB_SEMAPHORE_NULL) goto error;
- tq->data = (char *) malloc(tq->item_size * tq->array_size);
- if (tq->data == NULL) goto error;
- tq->head = tq->tail = 0;
- tq->head_blockers = tq->tail_blockers = 0;
- return tq;
- error:
- stb_threadq_delete(tq);
- return NULL;
- }
- typedef struct
- {
- stb_thread_func f;
- void *d;
- volatile void **retval;
- stb_sync sync;
- } stb__workinfo;
- //static volatile stb__workinfo *stb__work;
- struct stb__workqueue
- {
- int numthreads;
- stb_threadqueue *tq;
- };
- static stb_workqueue *stb__work_global;
- static void *stb__thread_workloop(void *p)
- {
- volatile stb_workqueue *q = (volatile stb_workqueue *) p;
- for(;;) {
- void *z;
- stb__workinfo w;
- stb_threadq_get_block(q->tq, &w);
- if (w.f == NULL) // null work is a signal to end the thread
- return NULL;
- z = w.f(w.d);
- if (w.retval) { stb_barrier(); *w.retval = z; }
- if (w.sync != STB_SYNC_NULL) stb_sync_reach(w.sync);
- }
- }
- stb_workqueue *stb_workq_new(int num_threads, int max_units)
- {
- return stb_workq_new_flags(num_threads, max_units, 0,0);
- }
- stb_workqueue *stb_workq_new_flags(int numthreads, int max_units, int no_add_mutex, int no_remove_mutex)
- {
- stb_workqueue *q = (stb_workqueue *) malloc(sizeof(*q));
- if (q == NULL) return NULL;
- q->tq = stb_threadq_new(sizeof(stb__workinfo), max_units, !no_add_mutex, !no_remove_mutex);
- if (q->tq == NULL) { free(q); return NULL; }
- q->numthreads = 0;
- stb_workq_numthreads(q, numthreads);
- return q;
- }
- void stb_workq_delete(stb_workqueue *q)
- {
- while (stb_workq_length(q) != 0)
- stb__thread_sleep(1);
- stb_threadq_delete(q->tq);
- free(q);
- }
- static int stb__work_maxitems = STB_THREADQUEUE_DYNAMIC;
- static void stb_work_init(int num_threads)
- {
- if (stb__work_global == NULL) {
- stb__threadmutex_init();
- stb_mutex_begin(stb__workmutex);
- stb_barrier();
- if (*(stb_workqueue * volatile *) &stb__work_global == NULL)
- stb__work_global = stb_workq_new(num_threads, stb__work_maxitems);
- stb_mutex_end(stb__workmutex);
- }
- }
- static int stb__work_raw(stb_workqueue *q, stb_thread_func f, void *d, volatile void **return_code, stb_sync rel)
- {
- stb__workinfo w;
- if (q == NULL) {
- stb_work_init(1);
- q = stb__work_global;
- }
- w.f = f;
- w.d = d;
- w.retval = return_code;
- w.sync = rel;
- return stb_threadq_add(q->tq, &w);
- }
- int stb_workq_length(stb_workqueue *q)
- {
- return stb_threadq_length(q->tq);
- }
- int stb_workq(stb_workqueue *q, stb_thread_func f, void *d, volatile void **return_code)
- {
- if (f == NULL) return 0;
- return stb_workq_reach(q, f, d, return_code, NULL);
- }
- int stb_workq_reach(stb_workqueue *q, stb_thread_func f, void *d, volatile void **return_code, stb_sync rel)
- {
- if (f == NULL) return 0;
- return stb__work_raw(q, f, d, return_code, rel);
- }
- static void stb__workq_numthreads(stb_workqueue *q, int n)
- {
- while (q->numthreads < n) {
- stb_create_thread(stb__thread_workloop, q);
- ++q->numthreads;
- }
- while (q->numthreads > n) {
- stb__work_raw(q, NULL, NULL, NULL, NULL);
- --q->numthreads;
- }
- }
- void stb_workq_numthreads(stb_workqueue *q, int n)
- {
- stb_mutex_begin(stb__threadmutex);
- stb__workq_numthreads(q,n);
- stb_mutex_end(stb__threadmutex);
- }
- int stb_work_maxunits(int n)
- {
- if (stb__work_global == NULL) {
- stb__work_maxitems = n;
- stb_work_init(1);
- }
- return stb__work_maxitems;
- }
- int stb_work(stb_thread_func f, void *d, volatile void **return_code)
- {
- return stb_workq(stb__work_global, f,d,return_code);
- }
- int stb_work_reach(stb_thread_func f, void *d, volatile void **return_code, stb_sync rel)
- {
- return stb_workq_reach(stb__work_global, f,d,return_code,rel);
- }
- void stb_work_numthreads(int n)
- {
- if (stb__work_global == NULL)
- stb_work_init(n);
- else
- stb_workq_numthreads(stb__work_global, n);
- }
- #endif // STB_DEFINE
- //////////////////////////////////////////////////////////////////////////////
- //
- // Background disk I/O
- //
- //
- #define STB_BGIO_READ_ALL (-1)
- STB_EXTERN int stb_bgio_read (char *filename, int offset, int len, stb_uchar **result, int *olen);
- STB_EXTERN int stb_bgio_readf (FILE *f , int offset, int len, stb_uchar **result, int *olen);
- STB_EXTERN int stb_bgio_read_to (char *filename, int offset, int len, stb_uchar *buffer, int *olen);
- STB_EXTERN int stb_bgio_readf_to(FILE *f , int offset, int len, stb_uchar *buffer, int *olen);
- typedef struct
- {
- int have_data;
- int is_valid;
- int is_dir;
- time_t filetime;
- stb_int64 filesize;
- } stb_bgstat;
- STB_EXTERN int stb_bgio_stat (char *filename, stb_bgstat *result);
- #ifdef STB_DEFINE
- static stb_workqueue *stb__diskio;
- static stb_mutex stb__diskio_mutex;
- void stb_thread_cleanup(void)
- {
- if (stb__work_global) stb_workq_delete(stb__work_global); stb__work_global = NULL;
- if (stb__threadmutex) stb_mutex_delete(stb__threadmutex); stb__threadmutex = NULL;
- if (stb__workmutex) stb_mutex_delete(stb__workmutex); stb__workmutex = NULL;
- if (stb__diskio) stb_workq_delete(stb__diskio); stb__diskio = NULL;
- if (stb__diskio_mutex)stb_mutex_delete(stb__diskio_mutex);stb__diskio_mutex= NULL;
- }
- typedef struct
- {
- char *filename;
- FILE *f;
- int offset;
- int len;
- stb_bgstat *stat_out;
- stb_uchar *output;
- stb_uchar **result;
- int *len_output;
- int *flag;
- } stb__disk_command;
- #define STB__MAX_DISK_COMMAND 100
- static stb__disk_command stb__dc_queue[STB__MAX_DISK_COMMAND];
- static int stb__dc_offset;
- void stb__io_init(void)
- {
- if (!stb__diskio) {
- stb__threadmutex_init();
- stb_mutex_begin(stb__threadmutex);
- stb_barrier();
- if (*(stb_thread * volatile *) &stb__diskio == NULL) {
- stb__diskio_mutex = stb_mutex_new();
- // use many threads so OS can try to schedule seeks
- stb__diskio = stb_workq_new_flags(16,STB__MAX_DISK_COMMAND,STB_FALSE,STB_FALSE);
- }
- stb_mutex_end(stb__threadmutex);
- }
- }
- static void * stb__io_error(stb__disk_command *dc)
- {
- if (dc->len_output) *dc->len_output = 0;
- if (dc->result) *dc->result = NULL;
- if (dc->flag) *dc->flag = -1;
- return NULL;
- }
- static void * stb__io_task(void *p)
- {
- stb__disk_command *dc = (stb__disk_command *) p;
- int len;
- FILE *f;
- stb_uchar *buf;
- if (dc->stat_out) {
- struct _stati64 s;
- if (!_stati64(dc->filename, &s)) {
- dc->stat_out->filesize = s.st_size;
- dc->stat_out->filetime = s.st_mtime;
- dc->stat_out->is_dir = s.st_mode & _S_IFDIR;
- dc->stat_out->is_valid = (s.st_mode & _S_IFREG) || dc->stat_out->is_dir;
- } else
- dc->stat_out->is_valid = 0;
- stb_barrier();
- dc->stat_out->have_data = 1;
- free(dc->filename);
- return 0;
- }
- if (dc->f) {
- #ifdef WIN32
- f = _fdopen(_dup(_fileno(dc->f)), "rb");
- #else
- f = fdopen(dup(fileno(dc->f)), "rb");
- #endif
- if (!f)
- return stb__io_error(dc);
- } else {
- f = fopen(dc->filename, "rb");
- free(dc->filename);
- if (!f)
- return stb__io_error(dc);
- }
- len = dc->len;
- if (len < 0) {
- fseek(f, 0, SEEK_END);
- len = ftell(f) - dc->offset;
- }
- if (fseek(f, dc->offset, SEEK_SET)) {
- fclose(f);
- return stb__io_error(dc);
- }
- if (dc->output)
- buf = dc->output;
- else {
- buf = (stb_uchar *) malloc(len);
- if (buf == NULL) {
- fclose(f);
- return stb__io_error(dc);
- }
- }
- len = fread(buf, 1, len, f);
- fclose(f);
- if (dc->len_output) *dc->len_output = len;
- if (dc->result) *dc->result = buf;
- if (dc->flag) *dc->flag = 1;
- return NULL;
- }
- int stb__io_add(char *fname, FILE *f, int off, int len, stb_uchar *out, stb_uchar **result, int *olen, int *flag, stb_bgstat *stat)
- {
- int res;
- stb__io_init();
- // do memory allocation outside of mutex
- if (fname) fname = stb_p_strdup(fname);
- stb_mutex_begin(stb__diskio_mutex);
- {
- stb__disk_command *dc = &stb__dc_queue[stb__dc_offset];
- dc->filename = fname;
- dc->f = f;
- dc->offset = off;
- dc->len = len;
- dc->output = out;
- dc->result = result;
- dc->len_output = olen;
- dc->flag = flag;
- dc->stat_out = stat;
- res = stb_workq(stb__diskio, stb__io_task, dc, NULL);
- if (res)
- stb__dc_offset = (stb__dc_offset + 1 == STB__MAX_DISK_COMMAND ? 0 : stb__dc_offset+1);
- }
- stb_mutex_end(stb__diskio_mutex);
- return res;
- }
- int stb_bgio_read(char *filename, int offset, int len, stb_uchar **result, int *olen)
- {
- return stb__io_add(filename,NULL,offset,len,NULL,result,olen,NULL,NULL);
- }
- int stb_bgio_readf(FILE *f, int offset, int len, stb_uchar **result, int *olen)
- {
- return stb__io_add(NULL,f,offset,len,NULL,result,olen,NULL,NULL);
- }
- int stb_bgio_read_to(char *filename, int offset, int len, stb_uchar *buffer, int *olen)
- {
- return stb__io_add(filename,NULL,offset,len,buffer,NULL,olen,NULL,NULL);
- }
- int stb_bgio_readf_to(FILE *f, int offset, int len, stb_uchar *buffer, int *olen)
- {
- return stb__io_add(NULL,f,offset,len,buffer,NULL,olen,NULL,NULL);
- }
- STB_EXTERN int stb_bgio_stat (char *filename, stb_bgstat *result)
- {
- result->have_data = 0;
- return stb__io_add(filename,NULL,0,0,0,NULL,0,NULL, result);
- }
- #endif
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //
- // Fast malloc implementation
- //
- // This is a clone of TCMalloc, but without the thread support.
- // 1. large objects are allocated directly, page-aligned
- // 2. small objects are allocated in homogeonous heaps, 0 overhead
- //
- // We keep an allocation table for pages a la TCMalloc. This would
- // require 4MB for the entire address space, but we only allocate
- // the parts that are in use. The overhead from using homogenous heaps
- // everywhere is 3MB. (That is, if you allocate 1 object of each size,
- // you'll use 3MB.)
- #if defined(STB_DEFINE) && ((defined(_WIN32) && !defined(_M_AMD64)) || defined(STB_FASTMALLOC))
- #ifdef _WIN32
- #ifndef _WINDOWS_
- #ifndef STB__IMPORT
- #define STB__IMPORT STB_EXTERN __declspec(dllimport)
- #define STB__DW unsigned long
- #endif
- STB__IMPORT void * __stdcall VirtualAlloc(void *p, unsigned long size, unsigned long type, unsigned long protect);
- STB__IMPORT int __stdcall VirtualFree(void *p, unsigned long size, unsigned long freetype);
- #endif
- #define stb__alloc_pages_raw(x) (stb_uint32) VirtualAlloc(NULL, (x), 0x3000, 0x04)
- #define stb__dealloc_pages_raw(p) VirtualFree((void *) p, 0, 0x8000)
- #else
- #error "Platform not currently supported"
- #endif
- typedef struct stb__span
- {
- int start, len;
- struct stb__span *next, *prev;
- void *first_free;
- unsigned short list; // 1..256 free; 257..511 sizeclass; 0=large block
- short allocations; // # outstanding allocations for sizeclass
- } stb__span; // 24
- static stb__span **stb__span_for_page;
- static int stb__firstpage, stb__lastpage;
- static void stb__update_page_range(int first, int last)
- {
- stb__span **sfp;
- int i, f,l;
- if (first >= stb__firstpage && last <= stb__lastpage) return;
- if (stb__span_for_page == NULL) {
- f = first;
- l = f+stb_max(last-f, 16384);
- l = stb_min(l, 1<<20);
- } else if (last > stb__lastpage) {
- f = stb__firstpage;
- l = f + (stb__lastpage - f) * 2;
- l = stb_clamp(last, l,1<<20);
- } else {
- l = stb__lastpage;
- f = l - (l - stb__firstpage) * 2;
- f = stb_clamp(f, 0,first);
- }
- sfp = (stb__span **) stb__alloc_pages_raw(sizeof(void *) * (l-f));
- for (i=f; i < stb__firstpage; ++i) sfp[i - f] = NULL;
- for ( ; i < stb__lastpage ; ++i) sfp[i - f] = stb__span_for_page[i - stb__firstpage];
- for ( ; i < l ; ++i) sfp[i - f] = NULL;
- if (stb__span_for_page) stb__dealloc_pages_raw(stb__span_for_page);
- stb__firstpage = f;
- stb__lastpage = l;
- stb__span_for_page = sfp;
- }
- static stb__span *stb__span_free=NULL;
- static stb__span *stb__span_first, *stb__span_end;
- static stb__span *stb__span_alloc(void)
- {
- stb__span *s = stb__span_free;
- if (s)
- stb__span_free = s->next;
- else {
- if (!stb__span_first) {
- stb__span_first = (stb__span *) stb__alloc_pages_raw(65536);
- if (stb__span_first == NULL) return NULL;
- stb__span_end = stb__span_first + (65536 / sizeof(stb__span));
- }
- s = stb__span_first++;
- if (stb__span_first == stb__span_end) stb__span_first = NULL;
- }
- return s;
- }
- static stb__span *stb__spanlist[512];
- static void stb__spanlist_unlink(stb__span *s)
- {
- if (s->prev)
- s->prev->next = s->next;
- else {
- int n = s->list;
- assert(stb__spanlist[n] == s);
- stb__spanlist[n] = s->next;
- }
- if (s->next)
- s->next->prev = s->prev;
- s->next = s->prev = NULL;
- s->list = 0;
- }
- static void stb__spanlist_add(int n, stb__span *s)
- {
- s->list = n;
- s->next = stb__spanlist[n];
- s->prev = NULL;
- stb__spanlist[n] = s;
- if (s->next) s->next->prev = s;
- }
- #define stb__page_shift 12
- #define stb__page_size (1 << stb__page_shift)
- #define stb__page_number(x) ((x) >> stb__page_shift)
- #define stb__page_address(x) ((x) << stb__page_shift)
- static void stb__set_span_for_page(stb__span *s)
- {
- int i;
- for (i=0; i < s->len; ++i)
- stb__span_for_page[s->start + i - stb__firstpage] = s;
- }
- static stb__span *stb__coalesce(stb__span *a, stb__span *b)
- {
- assert(a->start + a->len == b->start);
- if (a->list) stb__spanlist_unlink(a);
- if (b->list) stb__spanlist_unlink(b);
- a->len += b->len;
- b->len = 0;
- b->next = stb__span_free;
- stb__span_free = b;
- stb__set_span_for_page(a);
- return a;
- }
- static void stb__free_span(stb__span *s)
- {
- stb__span *n = NULL;
- if (s->start > stb__firstpage) {
- n = stb__span_for_page[s->start-1 - stb__firstpage];
- if (n && n->allocations == -2 && n->start + n->len == s->start) s = stb__coalesce(n,s);
- }
- if (s->start + s->len < stb__lastpage) {
- n = stb__span_for_page[s->start + s->len - stb__firstpage];
- if (n && n->allocations == -2 && s->start + s->len == n->start) s = stb__coalesce(s,n);
- }
- s->allocations = -2;
- stb__spanlist_add(s->len > 256 ? 256 : s->len, s);
- }
- static stb__span *stb__alloc_pages(int num)
- {
- stb__span *s = stb__span_alloc();
- int p;
- if (!s) return NULL;
- p = stb__alloc_pages_raw(num << stb__page_shift);
- if (p == 0) { s->next = stb__span_free; stb__span_free = s; return 0; }
- assert(stb__page_address(stb__page_number(p)) == p);
- p = stb__page_number(p);
- stb__update_page_range(p, p+num);
- s->start = p;
- s->len = num;
- s->next = NULL;
- s->prev = NULL;
- stb__set_span_for_page(s);
- return s;
- }
- static stb__span *stb__alloc_span(int pagecount)
- {
- int i;
- stb__span *p = NULL;
- for(i=pagecount; i < 256; ++i)
- if (stb__spanlist[i]) {
- p = stb__spanlist[i];
- break;
- }
- if (!p) {
- p = stb__spanlist[256];
- while (p && p->len < pagecount)
- p = p->next;
- }
- if (!p) {
- p = stb__alloc_pages(pagecount < 16 ? 16 : pagecount);
- if (p == NULL) return 0;
- } else
- stb__spanlist_unlink(p);
- if (p->len > pagecount) {
- stb__span *q = stb__span_alloc();
- if (q) {
- q->start = p->start + pagecount;
- q->len = p->len - pagecount;
- p->len = pagecount;
- for (i=0; i < q->len; ++i)
- stb__span_for_page[q->start+i - stb__firstpage] = q;
- stb__spanlist_add(q->len > 256 ? 256 : q->len, q);
- }
- }
- return p;
- }
- #define STB__MAX_SMALL_SIZE 32768
- #define STB__MAX_SIZE_CLASSES 256
- static unsigned char stb__class_base[32];
- static unsigned char stb__class_shift[32];
- static unsigned char stb__pages_for_class[STB__MAX_SIZE_CLASSES];
- static int stb__size_for_class[STB__MAX_SIZE_CLASSES];
- stb__span *stb__get_nonempty_sizeclass(int c)
- {
- int s = c + 256, i, size, tsize; // remap to span-list index
- char *z;
- void *q;
- stb__span *p = stb__spanlist[s];
- if (p) {
- if (p->first_free) return p; // fast path: it's in the first one in list
- for (p=p->next; p; p=p->next)
- if (p->first_free) {
- // move to front for future queries
- stb__spanlist_unlink(p);
- stb__spanlist_add(s, p);
- return p;
- }
- }
- // no non-empty ones, so allocate a new one
- p = stb__alloc_span(stb__pages_for_class[c]);
- if (!p) return NULL;
- // create the free list up front
- size = stb__size_for_class[c];
- tsize = stb__pages_for_class[c] << stb__page_shift;
- i = 0;
- z = (char *) stb__page_address(p->start);
- q = NULL;
- while (i + size <= tsize) {
- * (void **) z = q; q = z;
- z += size;
- i += size;
- }
- p->first_free = q;
- p->allocations = 0;
- stb__spanlist_add(s,p);
- return p;
- }
- static int stb__sizeclass(size_t sz)
- {
- int z = stb_log2_floor(sz); // -1 below to group e.g. 13,14,15,16 correctly
- return stb__class_base[z] + ((sz-1) >> stb__class_shift[z]);
- }
- static void stb__init_sizeclass(void)
- {
- int i, size, overhead;
- int align_shift = 2; // allow 4-byte and 12-byte blocks as well, vs. TCMalloc
- int next_class = 1;
- int last_log = 0;
- for (i = 0; i < align_shift; i++) {
- stb__class_base [i] = next_class;
- stb__class_shift[i] = align_shift;
- }
- for (size = 1 << align_shift; size <= STB__MAX_SMALL_SIZE; size += 1 << align_shift) {
- i = stb_log2_floor(size);
- if (i > last_log) {
- if (size == 16) ++align_shift; // switch from 4-byte to 8-byte alignment
- else if (size >= 128 && align_shift < 8) ++align_shift;
- stb__class_base[i] = next_class - ((size-1) >> align_shift);
- stb__class_shift[i] = align_shift;
- last_log = i;
- }
- stb__size_for_class[next_class++] = size;
- }
- for (i=1; i <= STB__MAX_SMALL_SIZE; ++i)
- assert(i <= stb__size_for_class[stb__sizeclass(i)]);
- overhead = 0;
- for (i = 1; i < next_class; i++) {
- int s = stb__size_for_class[i];
- size = stb__page_size;
- while (size % s > size >> 3)
- size += stb__page_size;
- stb__pages_for_class[i] = (unsigned char) (size >> stb__page_shift);
- overhead += size;
- }
- assert(overhead < (4 << 20)); // make sure it's under 4MB of overhead
- }
- #ifdef STB_DEBUG
- #define stb__smemset(a,b,c) memset((void *) a, b, c)
- #elif defined(STB_FASTMALLOC_INIT)
- #define stb__smemset(a,b,c) memset((void *) a, b, c)
- #else
- #define stb__smemset(a,b,c)
- #endif
- void *stb_smalloc(size_t sz)
- {
- stb__span *s;
- if (sz == 0) return NULL;
- if (stb__size_for_class[1] == 0) stb__init_sizeclass();
- if (sz > STB__MAX_SMALL_SIZE) {
- s = stb__alloc_span((sz + stb__page_size - 1) >> stb__page_shift);
- if (s == NULL) return NULL;
- s->list = 0;
- s->next = s->prev = NULL;
- s->allocations = -32767;
- stb__smemset(stb__page_address(s->start), 0xcd, (sz+3)&~3);
- return (void *) stb__page_address(s->start);
- } else {
- void *p;
- int c = stb__sizeclass(sz);
- s = stb__spanlist[256+c];
- if (!s || !s->first_free)
- s = stb__get_nonempty_sizeclass(c);
- if (s == NULL) return NULL;
- p = s->first_free;
- s->first_free = * (void **) p;
- ++s->allocations;
- stb__smemset(p,0xcd, sz);
- return p;
- }
- }
- int stb_ssize(void *p)
- {
- stb__span *s;
- if (p == NULL) return 0;
- s = stb__span_for_page[stb__page_number((stb_uint) p) - stb__firstpage];
- if (s->list >= 256) {
- return stb__size_for_class[s->list - 256];
- } else {
- assert(s->list == 0);
- return s->len << stb__page_shift;
- }
- }
- void stb_sfree(void *p)
- {
- stb__span *s;
- if (p == NULL) return;
- s = stb__span_for_page[stb__page_number((stb_uint) p) - stb__firstpage];
- if (s->list >= 256) {
- stb__smemset(p, 0xfe, stb__size_for_class[s->list-256]);
- * (void **) p = s->first_free;
- s->first_free = p;
- if (--s->allocations == 0) {
- stb__spanlist_unlink(s);
- stb__free_span(s);
- }
- } else {
- assert(s->list == 0);
- stb__smemset(p, 0xfe, stb_ssize(p));
- stb__free_span(s);
- }
- }
- void *stb_srealloc(void *p, size_t sz)
- {
- size_t cur_size;
- if (p == NULL) return stb_smalloc(sz);
- if (sz == 0) { stb_sfree(p); return NULL; }
- cur_size = stb_ssize(p);
- if (sz > cur_size || sz <= (cur_size >> 1)) {
- void *q;
- if (sz > cur_size && sz < (cur_size << 1)) sz = cur_size << 1;
- q = stb_smalloc(sz); if (q == NULL) return NULL;
- memcpy(q, p, sz < cur_size ? sz : cur_size);
- stb_sfree(p);
- return q;
- }
- return p;
- }
- void *stb_scalloc(size_t n, size_t sz)
- {
- void *p;
- if (n == 0 || sz == 0) return NULL;
- if (stb_log2_ceil(n) + stb_log2_ceil(n) >= 32) return NULL;
- p = stb_smalloc(n*sz);
- if (p) memset(p, 0, n*sz);
- return p;
- }
- char *stb_sstrdup(char *s)
- {
- int n = strlen(s);
- char *p = (char *) stb_smalloc(n+1);
- if (p) stb_p_strcpy_s(p,n+1,s);
- return p;
- }
- #endif // STB_DEFINE
- //////////////////////////////////////////////////////////////////////////////
- //
- // Source code constants
- //
- // This is a trivial system to let you specify constants in source code,
- // then while running you can change the constants.
- //
- // Note that you can't wrap the #defines, because we need to know their
- // names. So we provide a pre-wrapped version without 'STB_' for convenience;
- // to request it, #define STB_CONVENIENT_H, yielding:
- // KI -- integer
- // KU -- unsigned integer
- // KF -- float
- // KD -- double
- // KS -- string constant
- //
- // Defaults to functioning in debug build, not in release builds.
- // To force on, define STB_ALWAYS_H
- #ifdef STB_CONVENIENT_H
- #define KI(x) STB_I(x)
- #define KU(x) STB_UI(x)
- #define KF(x) STB_F(x)
- #define KD(x) STB_D(x)
- #define KS(x) STB_S(x)
- #endif
- STB_EXTERN void stb_source_path(char *str);
- #ifdef STB_DEFINE
- char *stb__source_path;
- void stb_source_path(char *path)
- {
- stb__source_path = path;
- }
- char *stb__get_sourcefile_path(char *file)
- {
- static char filebuf[512];
- if (stb__source_path) {
- stb_p_sprintf(filebuf stb_p_size(sizeof(filebuf)), "%s/%s", stb__source_path, file);
- if (stb_fexists(filebuf)) return filebuf;
- }
- if (stb_fexists(file)) return file;
- stb_p_sprintf(filebuf stb_p_size(sizeof(filebuf)), "../%s", file);
- if (!stb_fexists(filebuf)) return filebuf;
- return file;
- }
- #endif
- #define STB_F(x) ((float) STB_H(x))
- #define STB_UI(x) ((unsigned int) STB_I(x))
- #if !defined(STB_DEBUG) && !defined(STB_ALWAYS_H)
- #define STB_D(x) ((double) (x))
- #define STB_I(x) ((int) (x))
- #define STB_S(x) ((char *) (x))
- #else
- #define STB_D(x) stb__double_constant(__FILE__, __LINE__-1, (x))
- #define STB_I(x) stb__int_constant(__FILE__, __LINE__-1, (x))
- #define STB_S(x) stb__string_constant(__FILE__, __LINE__-1, (x))
- STB_EXTERN double stb__double_constant(char *file, int line, double x);
- STB_EXTERN int stb__int_constant(char *file, int line, int x);
- STB_EXTERN char * stb__string_constant(char *file, int line, char *str);
- #ifdef STB_DEFINE
- enum
- {
- STB__CTYPE_int,
- STB__CTYPE_uint,
- STB__CTYPE_float,
- STB__CTYPE_double,
- STB__CTYPE_string,
- };
- typedef struct
- {
- int line;
- int type;
- union {
- int ival;
- double dval;
- char *sval;
- };
- } stb__Entry;
- typedef struct
- {
- stb__Entry *entries;
- char *filename;
- time_t timestamp;
- char **file_data;
- int file_len;
- unsigned short *line_index;
- } stb__FileEntry;
- static void stb__constant_parse(stb__FileEntry *f, int i)
- {
- char *s;
- int n;
- if (!stb_arr_valid(f->entries, i)) return;
- n = f->entries[i].line;
- if (n >= f->file_len) return;
- s = f->file_data[n];
- switch (f->entries[i].type) {
- case STB__CTYPE_float:
- while (*s) {
- if (!strncmp(s, "STB_D(", 6)) { s+=6; goto matched_float; }
- if (!strncmp(s, "STB_F(", 6)) { s+=6; goto matched_float; }
- if (!strncmp(s, "KD(", 3)) { s+=3; goto matched_float; }
- if (!strncmp(s, "KF(", 3)) { s+=3; goto matched_float; }
- ++s;
- }
- break;
- matched_float:
- f->entries[i].dval = strtod(s, NULL);
- break;
- case STB__CTYPE_int:
- while (*s) {
- if (!strncmp(s, "STB_I(", 6)) { s+=6; goto matched_int; }
- if (!strncmp(s, "STB_UI(", 7)) { s+=7; goto matched_int; }
- if (!strncmp(s, "KI(", 3)) { s+=3; goto matched_int; }
- if (!strncmp(s, "KU(", 3)) { s+=3; goto matched_int; }
- ++s;
- }
- break;
- matched_int: {
- int neg=0;
- s = stb_skipwhite(s);
- while (*s == '-') { neg = !neg; s = stb_skipwhite(s+1); } // handle '- - 5', pointlessly
- if (s[0] == '0' && tolower(s[1]) == 'x')
- f->entries[i].ival = strtol(s, NULL, 16);
- else if (s[0] == '0')
- f->entries[i].ival = strtol(s, NULL, 8);
- else
- f->entries[i].ival = strtol(s, NULL, 10);
- if (neg) f->entries[i].ival = -f->entries[i].ival;
- break;
- }
- case STB__CTYPE_string:
- // @TODO
- break;
- }
- }
- static stb_sdict *stb__constant_file_hash;
- stb__Entry *stb__constant_get_entry(char *filename, int line, int type)
- {
- int i;
- stb__FileEntry *f;
- if (stb__constant_file_hash == NULL)
- stb__constant_file_hash = stb_sdict_new(STB_TRUE);
- f = (stb__FileEntry*) stb_sdict_get(stb__constant_file_hash, filename);
- if (f == NULL) {
- char *s = stb__get_sourcefile_path(filename);
- if (s == NULL || !stb_fexists(s)) return 0;
- f = (stb__FileEntry *) malloc(sizeof(*f));
- f->timestamp = stb_ftimestamp(s);
- f->file_data = stb_stringfile(s, &f->file_len);
- f->filename = stb_p_strdup(s); // cache the full path
- f->entries = NULL;
- f->line_index = 0;
- stb_arr_setlen(f->line_index, f->file_len);
- memset(f->line_index, 0xff, stb_arr_storage(f->line_index));
- } else {
- time_t t = stb_ftimestamp(f->filename);
- if (f->timestamp != t) {
- f->timestamp = t;
- free(f->file_data);
- f->file_data = stb_stringfile(f->filename, &f->file_len);
- stb_arr_setlen(f->line_index, f->file_len);
- for (i=0; i < stb_arr_len(f->entries); ++i)
- stb__constant_parse(f, i);
- }
- }
- if (line >= f->file_len) return 0;
- if (f->line_index[line] >= stb_arr_len(f->entries)) {
- // need a new entry
- int n = stb_arr_len(f->entries);
- stb__Entry e;
- e.line = line;
- if (line < f->file_len)
- f->line_index[line] = n;
- e.type = type;
- stb_arr_push(f->entries, e);
- stb__constant_parse(f, n);
- }
- return f->entries + f->line_index[line];
- }
- double stb__double_constant(char *file, int line, double x)
- {
- stb__Entry *e = stb__constant_get_entry(file, line, STB__CTYPE_float);
- if (!e) return x;
- return e->dval;
- }
- int stb__int_constant(char *file, int line, int x)
- {
- stb__Entry *e = stb__constant_get_entry(file, line, STB__CTYPE_int);
- if (!e) return x;
- return e->ival;
- }
- char * stb__string_constant(char *file, int line, char *x)
- {
- stb__Entry *e = stb__constant_get_entry(file, line, STB__CTYPE_string);
- if (!e) return x;
- return e->sval;
- }
- #endif // STB_DEFINE
- #endif // !STB_DEBUG && !STB_ALWAYS_H
- #undef STB_EXTERN
- #endif // STB_INCLUDE_STB_H
- /*
- ------------------------------------------------------------------------------
- This software is available under 2 licenses -- choose whichever you prefer.
- ------------------------------------------------------------------------------
- ALTERNATIVE A - MIT License
- Copyright (c) 2017 Sean Barrett
- Permission is hereby granted, free of charge, to any person obtaining a copy of
- this software and associated documentation files (the "Software"), to deal in
- the Software without restriction, including without limitation the rights to
- use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
- of the Software, and to permit persons to whom the Software is furnished to do
- so, subject to the following conditions:
- The above copyright notice and this permission notice shall be included in all
- copies or substantial portions of the Software.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- SOFTWARE.
- ------------------------------------------------------------------------------
- ALTERNATIVE B - Public Domain (www.unlicense.org)
- This is free and unencumbered software released into the public domain.
- Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
- software, either in source code form or as a compiled binary, for any purpose,
- commercial or non-commercial, and by any means.
- In jurisdictions that recognize copyright laws, the author or authors of this
- software dedicate any and all copyright interest in the software to the public
- domain. We make this dedication for the benefit of the public at large and to
- the detriment of our heirs and successors. We intend this dedication to be an
- overt act of relinquishment in perpetuity of all present and future rights to
- this software under copyright law.
- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
- ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
- WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- ------------------------------------------------------------------------------
- */
|