| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927592859295930593159325933593459355936593759385939594059415942594359445945594659475948594959505951595259535954595559565957595859595960596159625963596459655966596759685969597059715972597359745975597659775978597959805981598259835984598559865987598859895990599159925993599459955996599759985999600060016002600360046005600660076008600960106011601260136014601560166017601860196020602160226023602460256026602760286029603060316032603360346035603660376038603960406041604260436044604560466047604860496050605160526053605460556056605760586059606060616062606360646065606660676068606960706071607260736074607560766077607860796080608160826083608460856086608760886089609060916092609360946095609660976098609961006101610261036104610561066107610861096110611161126113611461156116611761186119612061216122612361246125612661276128612961306131613261336134613561366137613861396140614161426143614461456146614761486149615061516152615361546155615661576158615961606161616261636164616561666167616861696170617161726173617461756176617761786179618061816182618361846185618661876188618961906191619261936194619561966197619861996200620162026203620462056206620762086209621062116212621362146215621662176218621962206221622262236224622562266227622862296230623162326233623462356236623762386239624062416242624362446245624662476248624962506251625262536254625562566257625862596260626162626263626462656266626762686269627062716272627362746275627662776278627962806281628262836284628562866287628862896290629162926293629462956296629762986299630063016302630363046305630663076308630963106311631263136314631563166317631863196320632163226323632463256326632763286329633063316332633363346335633663376338633963406341634263436344634563466347634863496350635163526353635463556356635763586359636063616362636363646365636663676368636963706371637263736374637563766377637863796380638163826383638463856386638763886389639063916392639363946395639663976398639964006401640264036404640564066407640864096410641164126413641464156416641764186419642064216422642364246425642664276428642964306431643264336434643564366437643864396440644164426443644464456446644764486449645064516452645364546455645664576458645964606461646264636464646564666467646864696470647164726473647464756476647764786479648064816482648364846485648664876488648964906491649264936494649564966497649864996500650165026503650465056506650765086509651065116512651365146515651665176518651965206521652265236524652565266527652865296530653165326533653465356536653765386539654065416542654365446545654665476548654965506551655265536554655565566557655865596560656165626563656465656566656765686569657065716572657365746575657665776578657965806581658265836584658565866587658865896590659165926593659465956596659765986599660066016602660366046605660666076608660966106611661266136614661566166617661866196620662166226623662466256626662766286629663066316632663366346635663666376638663966406641664266436644664566466647664866496650665166526653665466556656665766586659666066616662666366646665666666676668666966706671667266736674667566766677667866796680668166826683668466856686668766886689669066916692669366946695669666976698669967006701670267036704670567066707670867096710671167126713671467156716671767186719672067216722672367246725672667276728672967306731673267336734673567366737673867396740674167426743674467456746674767486749675067516752675367546755675667576758675967606761676267636764676567666767676867696770677167726773677467756776677767786779678067816782678367846785678667876788678967906791679267936794679567966797679867996800680168026803680468056806680768086809681068116812681368146815681668176818681968206821682268236824682568266827682868296830683168326833683468356836683768386839684068416842684368446845684668476848684968506851685268536854685568566857685868596860686168626863686468656866686768686869687068716872687368746875687668776878687968806881688268836884688568866887688868896890689168926893689468956896689768986899690069016902690369046905690669076908690969106911691269136914691569166917691869196920692169226923692469256926692769286929693069316932693369346935693669376938693969406941694269436944694569466947694869496950695169526953695469556956695769586959696069616962696369646965696669676968696969706971697269736974697569766977697869796980698169826983698469856986698769886989699069916992699369946995699669976998699970007001700270037004700570067007700870097010701170127013701470157016701770187019702070217022702370247025702670277028702970307031703270337034703570367037703870397040704170427043704470457046704770487049705070517052705370547055705670577058705970607061706270637064706570667067706870697070707170727073707470757076707770787079708070817082708370847085708670877088708970907091709270937094709570967097709870997100710171027103710471057106710771087109711071117112711371147115711671177118711971207121712271237124712571267127712871297130713171327133713471357136713771387139714071417142714371447145714671477148714971507151715271537154715571567157715871597160716171627163716471657166716771687169717071717172717371747175717671777178717971807181718271837184718571867187718871897190719171927193719471957196719771987199720072017202720372047205720672077208720972107211721272137214721572167217721872197220722172227223722472257226722772287229723072317232723372347235723672377238723972407241724272437244724572467247724872497250725172527253725472557256725772587259726072617262726372647265726672677268726972707271727272737274727572767277727872797280728172827283728472857286728772887289729072917292729372947295729672977298729973007301730273037304730573067307730873097310731173127313731473157316731773187319732073217322732373247325732673277328732973307331733273337334733573367337733873397340734173427343734473457346734773487349735073517352735373547355735673577358735973607361736273637364736573667367736873697370737173727373737473757376737773787379738073817382738373847385738673877388738973907391739273937394739573967397739873997400740174027403740474057406740774087409741074117412741374147415741674177418741974207421742274237424742574267427742874297430743174327433743474357436743774387439744074417442744374447445744674477448744974507451745274537454745574567457745874597460746174627463746474657466746774687469747074717472747374747475747674777478747974807481748274837484748574867487748874897490749174927493749474957496749774987499750075017502750375047505750675077508750975107511751275137514751575167517751875197520752175227523752475257526752775287529753075317532753375347535753675377538753975407541754275437544754575467547754875497550755175527553755475557556755775587559756075617562756375647565756675677568756975707571757275737574757575767577757875797580758175827583758475857586758775887589759075917592759375947595759675977598759976007601760276037604760576067607760876097610761176127613761476157616761776187619762076217622762376247625762676277628762976307631763276337634763576367637763876397640764176427643764476457646764776487649765076517652765376547655765676577658765976607661766276637664766576667667766876697670767176727673767476757676767776787679768076817682768376847685768676877688768976907691769276937694769576967697769876997700770177027703770477057706770777087709771077117712771377147715771677177718771977207721772277237724772577267727772877297730773177327733773477357736773777387739774077417742774377447745774677477748774977507751775277537754775577567757775877597760776177627763776477657766776777687769777077717772777377747775777677777778777977807781778277837784778577867787778877897790779177927793779477957796779777987799780078017802780378047805780678077808780978107811781278137814781578167817781878197820782178227823782478257826782778287829783078317832783378347835783678377838783978407841784278437844784578467847784878497850785178527853785478557856785778587859786078617862786378647865786678677868786978707871787278737874787578767877787878797880788178827883788478857886788778887889789078917892789378947895789678977898789979007901790279037904790579067907790879097910791179127913791479157916791779187919792079217922792379247925792679277928792979307931793279337934793579367937793879397940794179427943794479457946794779487949795079517952795379547955795679577958795979607961796279637964796579667967796879697970797179727973797479757976797779787979798079817982798379847985798679877988798979907991799279937994799579967997799879998000800180028003800480058006800780088009801080118012801380148015801680178018801980208021802280238024802580268027802880298030803180328033803480358036803780388039804080418042804380448045804680478048804980508051805280538054805580568057805880598060806180628063806480658066806780688069807080718072807380748075807680778078807980808081808280838084808580868087808880898090809180928093809480958096809780988099810081018102810381048105810681078108810981108111811281138114811581168117811881198120812181228123812481258126812781288129813081318132813381348135813681378138813981408141814281438144814581468147814881498150815181528153815481558156815781588159816081618162816381648165816681678168816981708171817281738174817581768177817881798180818181828183818481858186818781888189819081918192819381948195819681978198819982008201820282038204820582068207820882098210821182128213821482158216821782188219822082218222822382248225822682278228822982308231823282338234823582368237823882398240824182428243824482458246824782488249825082518252825382548255825682578258825982608261826282638264826582668267826882698270827182728273827482758276827782788279828082818282828382848285828682878288828982908291829282938294829582968297829882998300830183028303830483058306830783088309831083118312831383148315831683178318831983208321832283238324832583268327832883298330833183328333833483358336833783388339834083418342834383448345834683478348834983508351835283538354835583568357835883598360836183628363836483658366836783688369837083718372837383748375837683778378837983808381838283838384838583868387838883898390839183928393839483958396839783988399840084018402840384048405840684078408840984108411841284138414841584168417841884198420842184228423842484258426842784288429843084318432843384348435843684378438843984408441844284438444844584468447844884498450845184528453845484558456845784588459846084618462846384648465846684678468846984708471847284738474847584768477847884798480848184828483848484858486848784888489849084918492849384948495849684978498849985008501850285038504850585068507850885098510851185128513851485158516851785188519852085218522852385248525852685278528852985308531853285338534853585368537853885398540854185428543854485458546854785488549855085518552855385548555855685578558855985608561856285638564856585668567856885698570857185728573857485758576857785788579858085818582858385848585858685878588858985908591859285938594859585968597859885998600860186028603860486058606860786088609861086118612861386148615861686178618861986208621862286238624862586268627862886298630863186328633863486358636863786388639864086418642864386448645864686478648864986508651865286538654865586568657865886598660866186628663866486658666866786688669867086718672867386748675867686778678867986808681868286838684868586868687868886898690869186928693869486958696869786988699870087018702870387048705870687078708870987108711871287138714871587168717871887198720872187228723872487258726872787288729873087318732873387348735873687378738873987408741874287438744874587468747874887498750875187528753875487558756875787588759876087618762876387648765876687678768876987708771877287738774877587768777877887798780878187828783878487858786878787888789879087918792879387948795879687978798879988008801880288038804880588068807880888098810881188128813881488158816881788188819882088218822882388248825882688278828882988308831883288338834883588368837883888398840884188428843884488458846884788488849885088518852885388548855885688578858885988608861886288638864886588668867886888698870887188728873887488758876887788788879888088818882888388848885888688878888888988908891889288938894889588968897889888998900890189028903890489058906890789088909891089118912891389148915891689178918891989208921892289238924892589268927892889298930893189328933893489358936893789388939894089418942894389448945894689478948894989508951895289538954895589568957895889598960896189628963896489658966896789688969897089718972897389748975897689778978897989808981898289838984898589868987898889898990899189928993899489958996899789988999900090019002900390049005900690079008900990109011901290139014901590169017901890199020902190229023902490259026902790289029903090319032903390349035903690379038903990409041904290439044904590469047904890499050905190529053905490559056905790589059906090619062906390649065906690679068906990709071907290739074907590769077907890799080908190829083908490859086908790889089909090919092909390949095909690979098909991009101910291039104910591069107910891099110911191129113911491159116911791189119912091219122912391249125912691279128912991309131913291339134913591369137913891399140914191429143914491459146914791489149915091519152915391549155915691579158915991609161916291639164916591669167916891699170917191729173917491759176917791789179918091819182918391849185918691879188918991909191919291939194919591969197919891999200920192029203920492059206920792089209921092119212921392149215921692179218921992209221922292239224922592269227922892299230923192329233923492359236923792389239924092419242924392449245924692479248924992509251925292539254925592569257925892599260926192629263926492659266926792689269927092719272927392749275927692779278927992809281928292839284928592869287928892899290929192929293929492959296929792989299930093019302930393049305930693079308930993109311931293139314931593169317931893199320932193229323932493259326932793289329933093319332933393349335933693379338933993409341934293439344934593469347934893499350935193529353935493559356935793589359936093619362936393649365936693679368936993709371937293739374937593769377937893799380938193829383938493859386938793889389939093919392939393949395939693979398939994009401940294039404940594069407940894099410941194129413941494159416941794189419942094219422942394249425942694279428942994309431943294339434943594369437943894399440944194429443944494459446944794489449945094519452945394549455945694579458945994609461946294639464946594669467946894699470947194729473947494759476947794789479948094819482948394849485948694879488948994909491949294939494949594969497949894999500950195029503950495059506950795089509951095119512951395149515951695179518951995209521952295239524952595269527952895299530953195329533953495359536953795389539954095419542954395449545954695479548954995509551955295539554955595569557955895599560956195629563956495659566956795689569957095719572957395749575957695779578957995809581958295839584958595869587958895899590959195929593959495959596959795989599960096019602960396049605960696079608960996109611961296139614961596169617961896199620962196229623962496259626962796289629963096319632963396349635963696379638963996409641964296439644964596469647964896499650965196529653965496559656965796589659966096619662966396649665966696679668966996709671967296739674967596769677967896799680968196829683968496859686968796889689969096919692969396949695969696979698969997009701970297039704970597069707970897099710971197129713971497159716971797189719972097219722972397249725972697279728972997309731973297339734973597369737973897399740974197429743974497459746974797489749975097519752975397549755975697579758975997609761976297639764976597669767976897699770977197729773977497759776977797789779978097819782978397849785978697879788978997909791979297939794979597969797979897999800980198029803980498059806980798089809981098119812981398149815981698179818981998209821982298239824982598269827982898299830983198329833983498359836983798389839984098419842984398449845984698479848984998509851985298539854985598569857985898599860986198629863986498659866986798689869987098719872987398749875987698779878987998809881988298839884988598869887988898899890989198929893989498959896989798989899990099019902990399049905990699079908990999109911991299139914991599169917991899199920992199229923992499259926992799289929993099319932993399349935993699379938993999409941994299439944994599469947994899499950995199529953995499559956995799589959996099619962996399649965996699679968996999709971997299739974997599769977997899799980998199829983998499859986998799889989999099919992999399949995999699979998999910000100011000210003100041000510006100071000810009100101001110012100131001410015100161001710018100191002010021100221002310024100251002610027100281002910030100311003210033100341003510036100371003810039100401004110042100431004410045100461004710048100491005010051100521005310054100551005610057100581005910060100611006210063100641006510066100671006810069100701007110072100731007410075100761007710078100791008010081100821008310084100851008610087100881008910090100911009210093100941009510096100971009810099101001010110102101031010410105101061010710108101091011010111101121011310114101151011610117101181011910120101211012210123101241012510126101271012810129101301013110132101331013410135101361013710138101391014010141101421014310144101451014610147101481014910150101511015210153101541015510156101571015810159101601016110162101631016410165101661016710168101691017010171101721017310174101751017610177101781017910180101811018210183101841018510186101871018810189101901019110192101931019410195101961019710198101991020010201102021020310204102051020610207102081020910210102111021210213102141021510216102171021810219102201022110222102231022410225102261022710228102291023010231102321023310234102351023610237102381023910240102411024210243102441024510246102471024810249102501025110252102531025410255102561025710258102591026010261102621026310264102651026610267102681026910270102711027210273102741027510276102771027810279102801028110282102831028410285102861028710288102891029010291102921029310294102951029610297102981029910300103011030210303103041030510306103071030810309103101031110312103131031410315103161031710318103191032010321103221032310324103251032610327103281032910330103311033210333103341033510336103371033810339103401034110342103431034410345103461034710348103491035010351103521035310354103551035610357103581035910360103611036210363103641036510366103671036810369103701037110372103731037410375103761037710378103791038010381103821038310384103851038610387103881038910390103911039210393103941039510396103971039810399104001040110402104031040410405104061040710408104091041010411104121041310414104151041610417104181041910420104211042210423104241042510426104271042810429104301043110432104331043410435104361043710438104391044010441104421044310444104451044610447104481044910450104511045210453104541045510456104571045810459104601046110462104631046410465104661046710468104691047010471104721047310474104751047610477104781047910480104811048210483104841048510486104871048810489104901049110492104931049410495104961049710498104991050010501105021050310504105051050610507105081050910510105111051210513105141051510516105171051810519105201052110522105231052410525105261052710528105291053010531105321053310534105351053610537105381053910540105411054210543105441054510546105471054810549105501055110552105531055410555105561055710558105591056010561105621056310564105651056610567105681056910570105711057210573105741057510576105771057810579105801058110582105831058410585105861058710588105891059010591105921059310594105951059610597105981059910600106011060210603106041060510606106071060810609106101061110612106131061410615106161061710618106191062010621106221062310624106251062610627106281062910630106311063210633106341063510636106371063810639106401064110642106431064410645106461064710648106491065010651 |
- /* stb_image_resize2 - v2.17 - public domain image resizing
- by Jeff Roberts (v2) and Jorge L Rodriguez
- http://github.com/nothings/stb
- Can be threaded with the extended API. SSE2, AVX, Neon and WASM SIMD support. Only
- scaling and translation is supported, no rotations or shears.
- COMPILING & LINKING
- In one C/C++ file that #includes this file, do this:
- #define STB_IMAGE_RESIZE_IMPLEMENTATION
- before the #include. That will create the implementation in that file.
- EASY API CALLS:
- Easy API downsamples w/Mitchell filter, upsamples w/cubic interpolation, clamps to edge.
- stbir_resize_uint8_srgb( input_pixels, input_w, input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout_enum )
- stbir_resize_uint8_linear( input_pixels, input_w, input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout_enum )
- stbir_resize_float_linear( input_pixels, input_w, input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout_enum )
- If you pass NULL or zero for the output_pixels, we will allocate the output buffer
- for you and return it from the function (free with free() or STBIR_FREE).
- As a special case, XX_stride_in_bytes of 0 means packed continuously in memory.
- API LEVELS
- There are three levels of API - easy-to-use, medium-complexity and extended-complexity.
- See the "header file" section of the source for API documentation.
- ADDITIONAL DOCUMENTATION
- MEMORY ALLOCATION
- By default, we use malloc and free for memory allocation. To override the
- memory allocation, before the implementation #include, add a:
- #define STBIR_MALLOC(size,user_data) ...
- #define STBIR_FREE(ptr,user_data) ...
- Each resize makes exactly one call to malloc/free (unless you use the
- extended API where you can do one allocation for many resizes). Under
- address sanitizer, we do separate allocations to find overread/writes.
- PERFORMANCE
- This library was written with an emphasis on performance. When testing
- stb_image_resize with RGBA, the fastest mode is STBIR_4CHANNEL with
- STBIR_TYPE_UINT8 pixels and CLAMPed edges (which is what many other resize
- libs do by default). Also, make sure SIMD is turned on of course (default
- for 64-bit targets). Avoid WRAP edge mode if you want the fastest speed.
- This library also comes with profiling built-in. If you define STBIR_PROFILE,
- you can use the advanced API and get low-level profiling information by
- calling stbir_resize_extended_profile_info() or stbir_resize_split_profile_info()
- after a resize.
- SIMD
- Most of the routines have optimized SSE2, AVX, NEON and WASM versions.
- On Microsoft compilers, we automatically turn on SIMD for 64-bit x64 and
- ARM; for 32-bit x86 and ARM, you select SIMD mode by defining STBIR_SSE2 or
- STBIR_NEON. For AVX and AVX2, we auto-select it by detecting the /arch:AVX
- or /arch:AVX2 switches. You can also always manually turn SSE2, AVX or AVX2
- support on by defining STBIR_SSE2, STBIR_AVX or STBIR_AVX2.
- On Linux, SSE2 and Neon is on by default for 64-bit x64 or ARM64. For 32-bit,
- we select x86 SIMD mode by whether you have -msse2, -mavx or -mavx2 enabled
- on the command line. For 32-bit ARM, you must pass -mfpu=neon-vfpv4 for both
- clang and GCC, but GCC also requires an additional -mfp16-format=ieee to
- automatically enable NEON.
- On x86 platforms, you can also define STBIR_FP16C to turn on FP16C instructions
- for converting back and forth to half-floats. This is autoselected when we
- are using AVX2. Clang and GCC also require the -mf16c switch. ARM always uses
- the built-in half float hardware NEON instructions.
- You can also tell us to use multiply-add instructions with STBIR_USE_FMA.
- Because x86 doesn't always have fma, we turn it off by default to maintain
- determinism across all platforms. If you don't care about non-FMA determinism
- and are willing to restrict yourself to more recent x86 CPUs (around the AVX
- timeframe), then fma will give you around a 15% speedup.
- You can force off SIMD in all cases by defining STBIR_NO_SIMD. You can turn
- off AVX or AVX2 specifically with STBIR_NO_AVX or STBIR_NO_AVX2. AVX is 10%
- to 40% faster, and AVX2 is generally another 12%.
- ALPHA CHANNEL
- Most of the resizing functions provide the ability to control how the alpha
- channel of an image is processed.
- When alpha represents transparency, it is important that when combining
- colors with filtering, the pixels should not be treated equally; they
- should use a weighted average based on their alpha values. For example,
- if a pixel is 1% opaque bright green and another pixel is 99% opaque
- black and you average them, the average will be 50% opaque, but the
- unweighted average and will be a middling green color, while the weighted
- average will be nearly black. This means the unweighted version introduced
- green energy that didn't exist in the source image.
- (If you want to know why this makes sense, you can work out the math for
- the following: consider what happens if you alpha composite a source image
- over a fixed color and then average the output, vs. if you average the
- source image pixels and then composite that over the same fixed color.
- Only the weighted average produces the same result as the ground truth
- composite-then-average result.)
- Therefore, it is in general best to "alpha weight" the pixels when applying
- filters to them. This essentially means multiplying the colors by the alpha
- values before combining them, and then dividing by the alpha value at the
- end.
- The computer graphics industry introduced a technique called "premultiplied
- alpha" or "associated alpha" in which image colors are stored in image files
- already multiplied by their alpha. This saves some math when compositing,
- and also avoids the need to divide by the alpha at the end (which is quite
- inefficient). However, while premultiplied alpha is common in the movie CGI
- industry, it is not commonplace in other industries like videogames, and most
- consumer file formats are generally expected to contain not-premultiplied
- colors. For example, Photoshop saves PNG files "unpremultiplied", and web
- browsers like Chrome and Firefox expect PNG images to be unpremultiplied.
- Note that there are three possibilities that might describe your image
- and resize expectation:
- 1. images are not premultiplied, alpha weighting is desired
- 2. images are not premultiplied, alpha weighting is not desired
- 3. images are premultiplied
- Both case #2 and case #3 require the exact same math: no alpha weighting
- should be applied or removed. Only case 1 requires extra math operations;
- the other two cases can be handled identically.
- stb_image_resize expects case #1 by default, applying alpha weighting to
- images, expecting the input images to be unpremultiplied. This is what the
- COLOR+ALPHA buffer types tell the resizer to do.
- When you use the pixel layouts STBIR_RGBA, STBIR_BGRA, STBIR_ARGB,
- STBIR_ABGR, STBIR_RX, or STBIR_XR you are telling us that the pixels are
- non-premultiplied. In these cases, the resizer will alpha weight the colors
- (effectively creating the premultiplied image), do the filtering, and then
- convert back to non-premult on exit.
- When you use the pixel layouts STBIR_RGBA_PM, STBIR_RGBA_PM, STBIR_RGBA_PM,
- STBIR_RGBA_PM, STBIR_RX_PM or STBIR_XR_PM, you are telling that the pixels
- ARE premultiplied. In this case, the resizer doesn't have to do the
- premultipling - it can filter directly on the input. This about twice as
- fast as the non-premultiplied case, so it's the right option if your data is
- already setup correctly.
- When you use the pixel layout STBIR_4CHANNEL or STBIR_2CHANNEL, you are
- telling us that there is no channel that represents transparency; it may be
- RGB and some unrelated fourth channel that has been stored in the alpha
- channel, but it is actually not alpha. No special processing will be
- performed.
- The difference between the generic 4 or 2 channel layouts, and the
- specialized _PM versions is with the _PM versions you are telling us that
- the data *is* alpha, just don't premultiply it. That's important when
- using SRGB pixel formats, we need to know where the alpha is, because
- it is converted linearly (rather than with the SRGB converters).
- Because alpha weighting produces the same effect as premultiplying, you
- even have the option with non-premultiplied inputs to let the resizer
- produce a premultiplied output. Because the intially computed alpha-weighted
- output image is effectively premultiplied, this is actually more performant
- than the normal path which un-premultiplies the output image as a final step.
- Finally, when converting both in and out of non-premulitplied space (for
- example, when using STBIR_RGBA), we go to somewhat heroic measures to
- ensure that areas with zero alpha value pixels get something reasonable
- in the RGB values. If you don't care about the RGB values of zero alpha
- pixels, you can call the stbir_set_non_pm_alpha_speed_over_quality()
- function - this runs a premultiplied resize about 25% faster. That said,
- when you really care about speed, using premultiplied pixels for both in
- and out (STBIR_RGBA_PM, etc) much faster than both of these premultiplied
- options.
- PIXEL LAYOUT CONVERSION
- The resizer can convert from some pixel layouts to others. When using the
- stbir_set_pixel_layouts(), you can, for example, specify STBIR_RGBA
- on input, and STBIR_ARGB on output, and it will re-organize the channels
- during the resize. Currently, you can only convert between two pixel
- layouts with the same number of channels.
- DETERMINISM
- We commit to being deterministic (from x64 to ARM to scalar to SIMD, etc).
- This requires compiling with fast-math off (using at least /fp:precise).
- Also, you must turn off fp-contracting (which turns mult+adds into fmas)!
- We attempt to do this with pragmas, but with Clang, you usually want to add
- -ffp-contract=off to the command line as well.
- For 32-bit x86, you must use SSE and SSE2 codegen for determinism. That is,
- if the scalar x87 unit gets used at all, we immediately lose determinism.
- On Microsoft Visual Studio 2008 and earlier, from what we can tell there is
- no way to be deterministic in 32-bit x86 (some x87 always leaks in, even
- with fp:strict). On 32-bit x86 GCC, determinism requires both -msse2 and
- -fpmath=sse.
- Note that we will not be deterministic with float data containing NaNs -
- the NaNs will propagate differently on different SIMD and platforms.
- If you turn on STBIR_USE_FMA, then we will be deterministic with other
- fma targets, but we will differ from non-fma targets (this is unavoidable,
- because a fma isn't simply an add with a mult - it also introduces a
- rounding difference compared to non-fma instruction sequences.
- FLOAT PIXEL FORMAT RANGE
- Any range of values can be used for the non-alpha float data that you pass
- in (0 to 1, -1 to 1, whatever). However, if you are inputting float values
- but *outputting* bytes or shorts, you must use a range of 0 to 1 so that we
- scale back properly. The alpha channel must also be 0 to 1 for any format
- that does premultiplication prior to resizing.
- Note also that with float output, using filters with negative lobes, the
- output filtered values might go slightly out of range. You can define
- STBIR_FLOAT_LOW_CLAMP and/or STBIR_FLOAT_HIGH_CLAMP to specify the range
- to clamp to on output, if that's important.
- MAX/MIN SCALE FACTORS
- The input pixel resolutions are in integers, and we do the internal pointer
- resolution in size_t sized integers. However, the scale ratio from input
- resolution to output resolution is calculated in float form. This means
- the effective possible scale ratio is limited to 24 bits (or 16 million
- to 1). As you get close to the size of the float resolution (again, 16
- million pixels wide or high), you might start seeing float inaccuracy
- issues in general in the pipeline. If you have to do extreme resizes,
- you can usually do this is multiple stages (using float intermediate
- buffers).
- FLIPPED IMAGES
- Stride is just the delta from one scanline to the next. This means you can
- use a negative stride to handle inverted images (point to the final
- scanline and use a negative stride). You can invert the input or output,
- using negative strides.
- DEFAULT FILTERS
- For functions which don't provide explicit control over what filters to
- use, you can change the compile-time defaults with:
- #define STBIR_DEFAULT_FILTER_UPSAMPLE STBIR_FILTER_something
- #define STBIR_DEFAULT_FILTER_DOWNSAMPLE STBIR_FILTER_something
- See stbir_filter in the header-file section for the list of filters.
- NEW FILTERS
- A number of 1D filter kernels are supplied. For a list of supported
- filters, see the stbir_filter enum. You can install your own filters by
- using the stbir_set_filter_callbacks function.
- PROGRESS
- For interactive use with slow resize operations, you can use the
- scanline callbacks in the extended API. It would have to be a *very* large
- image resample to need progress though - we're very fast.
- CEIL and FLOOR
- In scalar mode, the only functions we use from math.h are ceilf and floorf,
- but if you have your own versions, you can define the STBIR_CEILF(v) and
- STBIR_FLOORF(v) macros and we'll use them instead. In SIMD, we just use
- our own versions.
- ASSERT
- Define STBIR_ASSERT(boolval) to override assert() and not use assert.h
- PORTING FROM VERSION 1
- The API has changed. You can continue to use the old version of stb_image_resize.h,
- which is available in the "deprecated/" directory.
- If you're using the old simple-to-use API, porting is straightforward.
- (For more advanced APIs, read the documentation.)
- stbir_resize_uint8():
- - call `stbir_resize_uint8_linear`, cast channel count to `stbir_pixel_layout`
- stbir_resize_float():
- - call `stbir_resize_float_linear`, cast channel count to `stbir_pixel_layout`
- stbir_resize_uint8_srgb():
- - function name is unchanged
- - cast channel count to `stbir_pixel_layout`
- - above is sufficient unless your image has alpha and it's not RGBA/BGRA
- - in that case, follow the below instructions for stbir_resize_uint8_srgb_edgemode
- stbir_resize_uint8_srgb_edgemode()
- - switch to the "medium complexity" API
- - stbir_resize(), very similar API but a few more parameters:
- - pixel_layout: cast channel count to `stbir_pixel_layout`
- - data_type: STBIR_TYPE_UINT8_SRGB
- - edge: unchanged (STBIR_EDGE_WRAP, etc.)
- - filter: STBIR_FILTER_DEFAULT
- - which channel is alpha is specified in stbir_pixel_layout, see enum for details
- FUTURE TODOS
- * For polyphase integral filters, we just memcpy the coeffs to dupe
- them, but we should indirect and use the same coeff memory.
- * Add pixel layout conversions for sensible different channel counts
- (maybe, 1->3/4, 3->4, 4->1, 3->1).
- * For SIMD encode and decode scanline routines, do any pre-aligning
- for bad input/output buffer alignments and pitch?
- * For very wide scanlines, we should we do vertical strips to stay within
- L2 cache. Maybe do chunks of 1K pixels at a time. There would be
- some pixel reconversion, but probably dwarfed by things falling out
- of cache. Probably also something possible with alternating between
- scattering and gathering at high resize scales?
- * Should we have a multiple MIPs at the same time function (could keep
- more memory in cache during multiple resizes)?
- * Rewrite the coefficient generator to do many at once.
- * AVX-512 vertical kernels - worried about downclocking here.
- * Convert the reincludes to macros when we know they aren't changing.
- * Experiment with pivoting the horizontal and always using the
- vertical filters (which are faster, but perhaps not enough to overcome
- the pivot cost and the extra memory touches). Need to buffer the whole
- image so have to balance memory use.
- * Most of our code is internally function pointers, should we compile
- all the SIMD stuff always and dynamically dispatch?
- CONTRIBUTORS
- Jeff Roberts: 2.0 implementation, optimizations, SIMD
- Martins Mozeiko: NEON simd, WASM simd, clang and GCC whisperer
- Fabian Giesen: half float and srgb converters
- Sean Barrett: API design, optimizations
- Jorge L Rodriguez: Original 1.0 implementation
- Aras Pranckevicius: bugfixes
- Nathan Reed: warning fixes for 1.0
- REVISIONS
- 2.17 (2025-10-25) silly format bug in easy-to-use APIs.
- 2.16 (2025-10-21) fixed the easy-to-use APIs to allow inverted bitmaps (negative
- strides), fix vertical filter kernel callback, fix threaded
- gather buffer priming (and assert).
- (thanks adipose, TainZerL, and Harrison Green)
- 2.15 (2025-07-17) fixed an assert in debug mode when using floats with input
- callbacks, work around GCC warning when adding to null ptr
- (thanks Johannes Spohr and Pyry Kovanen).
- 2.14 (2025-05-09) fixed a bug using downsampling gather horizontal first, and
- scatter with vertical first.
- 2.13 (2025-02-27) fixed a bug when using input callbacks, turned off simd for
- tiny-c, fixed some variables that should have been static,
- fixes a bug when calculating temp memory with resizes that
- exceed 2GB of temp memory (very large resizes).
- 2.12 (2024-10-18) fix incorrect use of user_data with STBIR_FREE
- 2.11 (2024-09-08) fix harmless asan warnings in 2-channel and 3-channel mode
- with AVX-2, fix some weird scaling edge conditions with
- point sample mode.
- 2.10 (2024-07-27) fix the defines GCC and mingw for loop unroll control,
- fix MSVC 32-bit arm half float routines.
- 2.09 (2024-06-19) fix the defines for 32-bit ARM GCC builds (was selecting
- hardware half floats).
- 2.08 (2024-06-10) fix for RGB->BGR three channel flips and add SIMD (thanks
- to Ryan Salsbury), fix for sub-rect resizes, use the
- pragmas to control unrolling when they are available.
- 2.07 (2024-05-24) fix for slow final split during threaded conversions of very
- wide scanlines when downsampling (caused by extra input
- converting), fix for wide scanline resamples with many
- splits (int overflow), fix GCC warning.
- 2.06 (2024-02-10) fix for identical width/height 3x or more down-scaling
- undersampling a single row on rare resize ratios (about 1%).
- 2.05 (2024-02-07) fix for 2 pixel to 1 pixel resizes with wrap (thanks Aras),
- fix for output callback (thanks Julien Koenen).
- 2.04 (2023-11-17) fix for rare AVX bug, shadowed symbol (thanks Nikola Smiljanic).
- 2.03 (2023-11-01) ASAN and TSAN warnings fixed, minor tweaks.
- 2.00 (2023-10-10) mostly new source: new api, optimizations, simd, vertical-first, etc
- 2x-5x faster without simd, 4x-12x faster with simd,
- in some cases, 20x to 40x faster esp resizing large to very small.
- 0.96 (2019-03-04) fixed warnings
- 0.95 (2017-07-23) fixed warnings
- 0.94 (2017-03-18) fixed warnings
- 0.93 (2017-03-03) fixed bug with certain combinations of heights
- 0.92 (2017-01-02) fix integer overflow on large (>2GB) images
- 0.91 (2016-04-02) fix warnings; fix handling of subpixel regions
- 0.90 (2014-09-17) first released version
- LICENSE
- See end of file for license information.
- */
- #if !defined(STB_IMAGE_RESIZE_DO_HORIZONTALS) && !defined(STB_IMAGE_RESIZE_DO_VERTICALS) && !defined(STB_IMAGE_RESIZE_DO_CODERS) // for internal re-includes
- #ifndef STBIR_INCLUDE_STB_IMAGE_RESIZE2_H
- #define STBIR_INCLUDE_STB_IMAGE_RESIZE2_H
- #include <stddef.h>
- #ifdef _MSC_VER
- typedef unsigned char stbir_uint8;
- typedef unsigned short stbir_uint16;
- typedef unsigned int stbir_uint32;
- typedef unsigned __int64 stbir_uint64;
- #else
- #include <stdint.h>
- typedef uint8_t stbir_uint8;
- typedef uint16_t stbir_uint16;
- typedef uint32_t stbir_uint32;
- typedef uint64_t stbir_uint64;
- #endif
- #ifndef STBIRDEF
- #ifdef STB_IMAGE_RESIZE_STATIC
- #define STBIRDEF static
- #else
- #ifdef __cplusplus
- #define STBIRDEF extern "C"
- #else
- #define STBIRDEF extern
- #endif
- #endif
- #endif
- //////////////////////////////////////////////////////////////////////////////
- //// start "header file" ///////////////////////////////////////////////////
- //
- // Easy-to-use API:
- //
- // * stride is the offset between successive rows of image data
- // in memory, in bytes. specify 0 for packed continuously in memory
- // * colorspace is linear or sRGB as specified by function name
- // * Uses the default filters
- // * Uses edge mode clamped
- // * returned result is 1 for success or 0 in case of an error.
- // stbir_pixel_layout specifies:
- // number of channels
- // order of channels
- // whether color is premultiplied by alpha
- // for back compatibility, you can cast the old channel count to an stbir_pixel_layout
- typedef enum
- {
- STBIR_1CHANNEL = 1,
- STBIR_2CHANNEL = 2,
- STBIR_RGB = 3, // 3-chan, with order specified (for channel flipping)
- STBIR_BGR = 0, // 3-chan, with order specified (for channel flipping)
- STBIR_4CHANNEL = 5,
- STBIR_RGBA = 4, // alpha formats, where alpha is NOT premultiplied into color channels
- STBIR_BGRA = 6,
- STBIR_ARGB = 7,
- STBIR_ABGR = 8,
- STBIR_RA = 9,
- STBIR_AR = 10,
- STBIR_RGBA_PM = 11, // alpha formats, where alpha is premultiplied into color channels
- STBIR_BGRA_PM = 12,
- STBIR_ARGB_PM = 13,
- STBIR_ABGR_PM = 14,
- STBIR_RA_PM = 15,
- STBIR_AR_PM = 16,
- STBIR_RGBA_NO_AW = 11, // alpha formats, where NO alpha weighting is applied at all!
- STBIR_BGRA_NO_AW = 12, // these are just synonyms for the _PM flags (which also do
- STBIR_ARGB_NO_AW = 13, // no alpha weighting). These names just make it more clear
- STBIR_ABGR_NO_AW = 14, // for some folks).
- STBIR_RA_NO_AW = 15,
- STBIR_AR_NO_AW = 16,
- } stbir_pixel_layout;
- //===============================================================
- // Simple-complexity API
- //
- // If output_pixels is NULL (0), then we will allocate the buffer and return it to you.
- //--------------------------------
- STBIRDEF unsigned char * stbir_resize_uint8_srgb( const unsigned char *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_type );
- STBIRDEF unsigned char * stbir_resize_uint8_linear( const unsigned char *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_type );
- STBIRDEF float * stbir_resize_float_linear( const float *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- float *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_type );
- //===============================================================
- //===============================================================
- // Medium-complexity API
- //
- // This extends the easy-to-use API as follows:
- //
- // * Can specify the datatype - U8, U8_SRGB, U16, FLOAT, HALF_FLOAT
- // * Edge wrap can selected explicitly
- // * Filter can be selected explicitly
- //--------------------------------
- typedef enum
- {
- STBIR_EDGE_CLAMP = 0,
- STBIR_EDGE_REFLECT = 1,
- STBIR_EDGE_WRAP = 2, // this edge mode is slower and uses more memory
- STBIR_EDGE_ZERO = 3,
- } stbir_edge;
- typedef enum
- {
- STBIR_FILTER_DEFAULT = 0, // use same filter type that easy-to-use API chooses
- STBIR_FILTER_BOX = 1, // A trapezoid w/1-pixel wide ramps, same result as box for integer scale ratios
- STBIR_FILTER_TRIANGLE = 2, // On upsampling, produces same results as bilinear texture filtering
- STBIR_FILTER_CUBICBSPLINE = 3, // The cubic b-spline (aka Mitchell-Netrevalli with B=1,C=0), gaussian-esque
- STBIR_FILTER_CATMULLROM = 4, // An interpolating cubic spline
- STBIR_FILTER_MITCHELL = 5, // Mitchell-Netrevalli filter with B=1/3, C=1/3
- STBIR_FILTER_POINT_SAMPLE = 6, // Simple point sampling
- STBIR_FILTER_OTHER = 7, // User callback specified
- } stbir_filter;
- typedef enum
- {
- STBIR_TYPE_UINT8 = 0,
- STBIR_TYPE_UINT8_SRGB = 1,
- STBIR_TYPE_UINT8_SRGB_ALPHA = 2, // alpha channel, when present, should also be SRGB (this is very unusual)
- STBIR_TYPE_UINT16 = 3,
- STBIR_TYPE_FLOAT = 4,
- STBIR_TYPE_HALF_FLOAT = 5
- } stbir_datatype;
- // medium api
- STBIRDEF void * stbir_resize( const void *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- void *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout, stbir_datatype data_type,
- stbir_edge edge, stbir_filter filter );
- //===============================================================
- //===============================================================
- // Extended-complexity API
- //
- // This API exposes all resize functionality.
- //
- // * Separate filter types for each axis
- // * Separate edge modes for each axis
- // * Separate input and output data types
- // * Can specify regions with subpixel correctness
- // * Can specify alpha flags
- // * Can specify a memory callback
- // * Can specify a callback data type for pixel input and output
- // * Can be threaded for a single resize
- // * Can be used to resize many frames without recalculating the sampler info
- //
- // Use this API as follows:
- // 1) Call the stbir_resize_init function on a local STBIR_RESIZE structure
- // 2) Call any of the stbir_set functions
- // 3) Optionally call stbir_build_samplers() if you are going to resample multiple times
- // with the same input and output dimensions (like resizing video frames)
- // 4) Resample by calling stbir_resize_extended().
- // 5) Call stbir_free_samplers() if you called stbir_build_samplers()
- //--------------------------------
- // Types:
- // INPUT CALLBACK: this callback is used for input scanlines
- typedef void const * stbir_input_callback( void * optional_output, void const * input_ptr, int num_pixels, int x, int y, void * context );
- // OUTPUT CALLBACK: this callback is used for output scanlines
- typedef void stbir_output_callback( void const * output_ptr, int num_pixels, int y, void * context );
- // callbacks for user installed filters
- typedef float stbir__kernel_callback( float x, float scale, void * user_data ); // centered at zero
- typedef float stbir__support_callback( float scale, void * user_data );
- // internal structure with precomputed scaling
- typedef struct stbir__info stbir__info;
- typedef struct STBIR_RESIZE // use the stbir_resize_init and stbir_override functions to set these values for future compatibility
- {
- void * user_data;
- void const * input_pixels;
- int input_w, input_h;
- double input_s0, input_t0, input_s1, input_t1;
- stbir_input_callback * input_cb;
- void * output_pixels;
- int output_w, output_h;
- int output_subx, output_suby, output_subw, output_subh;
- stbir_output_callback * output_cb;
- int input_stride_in_bytes;
- int output_stride_in_bytes;
- int splits;
- int fast_alpha;
- int needs_rebuild;
- int called_alloc;
- stbir_pixel_layout input_pixel_layout_public;
- stbir_pixel_layout output_pixel_layout_public;
- stbir_datatype input_data_type;
- stbir_datatype output_data_type;
- stbir_filter horizontal_filter, vertical_filter;
- stbir_edge horizontal_edge, vertical_edge;
- stbir__kernel_callback * horizontal_filter_kernel; stbir__support_callback * horizontal_filter_support;
- stbir__kernel_callback * vertical_filter_kernel; stbir__support_callback * vertical_filter_support;
- stbir__info * samplers;
- } STBIR_RESIZE;
- // extended complexity api
- // First off, you must ALWAYS call stbir_resize_init on your resize structure before any of the other calls!
- STBIRDEF void stbir_resize_init( STBIR_RESIZE * resize,
- const void *input_pixels, int input_w, int input_h, int input_stride_in_bytes, // stride can be zero
- void *output_pixels, int output_w, int output_h, int output_stride_in_bytes, // stride can be zero
- stbir_pixel_layout pixel_layout, stbir_datatype data_type );
- //===============================================================
- // You can update these parameters any time after resize_init and there is no cost
- //--------------------------------
- STBIRDEF void stbir_set_datatypes( STBIR_RESIZE * resize, stbir_datatype input_type, stbir_datatype output_type );
- STBIRDEF void stbir_set_pixel_callbacks( STBIR_RESIZE * resize, stbir_input_callback * input_cb, stbir_output_callback * output_cb ); // no callbacks by default
- STBIRDEF void stbir_set_user_data( STBIR_RESIZE * resize, void * user_data ); // pass back STBIR_RESIZE* by default
- STBIRDEF void stbir_set_buffer_ptrs( STBIR_RESIZE * resize, const void * input_pixels, int input_stride_in_bytes, void * output_pixels, int output_stride_in_bytes );
- //===============================================================
- //===============================================================
- // If you call any of these functions, you will trigger a sampler rebuild!
- //--------------------------------
- STBIRDEF int stbir_set_pixel_layouts( STBIR_RESIZE * resize, stbir_pixel_layout input_pixel_layout, stbir_pixel_layout output_pixel_layout ); // sets new buffer layouts
- STBIRDEF int stbir_set_edgemodes( STBIR_RESIZE * resize, stbir_edge horizontal_edge, stbir_edge vertical_edge ); // CLAMP by default
- STBIRDEF int stbir_set_filters( STBIR_RESIZE * resize, stbir_filter horizontal_filter, stbir_filter vertical_filter ); // STBIR_DEFAULT_FILTER_UPSAMPLE/DOWNSAMPLE by default
- STBIRDEF int stbir_set_filter_callbacks( STBIR_RESIZE * resize, stbir__kernel_callback * horizontal_filter, stbir__support_callback * horizontal_support, stbir__kernel_callback * vertical_filter, stbir__support_callback * vertical_support );
- STBIRDEF int stbir_set_pixel_subrect( STBIR_RESIZE * resize, int subx, int suby, int subw, int subh ); // sets both sub-regions (full regions by default)
- STBIRDEF int stbir_set_input_subrect( STBIR_RESIZE * resize, double s0, double t0, double s1, double t1 ); // sets input sub-region (full region by default)
- STBIRDEF int stbir_set_output_pixel_subrect( STBIR_RESIZE * resize, int subx, int suby, int subw, int subh ); // sets output sub-region (full region by default)
- // when inputting AND outputting non-premultiplied alpha pixels, we use a slower but higher quality technique
- // that fills the zero alpha pixel's RGB values with something plausible. If you don't care about areas of
- // zero alpha, you can call this function to get about a 25% speed improvement for STBIR_RGBA to STBIR_RGBA
- // types of resizes.
- STBIRDEF int stbir_set_non_pm_alpha_speed_over_quality( STBIR_RESIZE * resize, int non_pma_alpha_speed_over_quality );
- //===============================================================
- //===============================================================
- // You can call build_samplers to prebuild all the internal data we need to resample.
- // Then, if you call resize_extended many times with the same resize, you only pay the
- // cost once.
- // If you do call build_samplers, you MUST call free_samplers eventually.
- //--------------------------------
- // This builds the samplers and does one allocation
- STBIRDEF int stbir_build_samplers( STBIR_RESIZE * resize );
- // You MUST call this, if you call stbir_build_samplers or stbir_build_samplers_with_splits
- STBIRDEF void stbir_free_samplers( STBIR_RESIZE * resize );
- //===============================================================
- // And this is the main function to perform the resize synchronously on one thread.
- STBIRDEF int stbir_resize_extended( STBIR_RESIZE * resize );
- //===============================================================
- // Use these functions for multithreading.
- // 1) You call stbir_build_samplers_with_splits first on the main thread
- // 2) Then stbir_resize_with_split on each thread
- // 3) stbir_free_samplers when done on the main thread
- //--------------------------------
- // This will build samplers for threading.
- // You can pass in the number of threads you'd like to use (try_splits).
- // It returns the number of splits (threads) that you can call it with.
- /// It might be less if the image resize can't be split up that many ways.
- STBIRDEF int stbir_build_samplers_with_splits( STBIR_RESIZE * resize, int try_splits );
- // This function does a split of the resizing (you call this fuction for each
- // split, on multiple threads). A split is a piece of the output resize pixel space.
- // Note that you MUST call stbir_build_samplers_with_splits before stbir_resize_extended_split!
- // Usually, you will always call stbir_resize_split with split_start as the thread_index
- // and "1" for the split_count.
- // But, if you have a weird situation where you MIGHT want 8 threads, but sometimes
- // only 4 threads, you can use 0,2,4,6 for the split_start's and use "2" for the
- // split_count each time to turn in into a 4 thread resize. (This is unusual).
- STBIRDEF int stbir_resize_extended_split( STBIR_RESIZE * resize, int split_start, int split_count );
- //===============================================================
- //===============================================================
- // Pixel Callbacks info:
- //--------------------------------
- // The input callback is super flexible - it calls you with the input address
- // (based on the stride and base pointer), it gives you an optional_output
- // pointer that you can fill, or you can just return your own pointer into
- // your own data.
- //
- // You can also do conversion from non-supported data types if necessary - in
- // this case, you ignore the input_ptr and just use the x and y parameters to
- // calculate your own input_ptr based on the size of each non-supported pixel.
- // (Something like the third example below.)
- //
- // You can also install just an input or just an output callback by setting the
- // callback that you don't want to zero.
- //
- // First example, progress: (getting a callback that you can monitor the progress):
- // void const * my_callback( void * optional_output, void const * input_ptr, int num_pixels, int x, int y, void * context )
- // {
- // percentage_done = y / input_height;
- // return input_ptr; // use buffer from call
- // }
- //
- // Next example, copying: (copy from some other buffer or stream):
- // void const * my_callback( void * optional_output, void const * input_ptr, int num_pixels, int x, int y, void * context )
- // {
- // CopyOrStreamData( optional_output, other_data_src, num_pixels * pixel_width_in_bytes );
- // return optional_output; // return the optional buffer that we filled
- // }
- //
- // Third example, input another buffer without copying: (zero-copy from other buffer):
- // void const * my_callback( void * optional_output, void const * input_ptr, int num_pixels, int x, int y, void * context )
- // {
- // void * pixels = ( (char*) other_image_base ) + ( y * other_image_stride ) + ( x * other_pixel_width_in_bytes );
- // return pixels; // return pointer to your data without copying
- // }
- //
- //
- // The output callback is considerably simpler - it just calls you so that you can dump
- // out each scanline. You could even directly copy out to disk if you have a simple format
- // like TGA or BMP. You can also convert to other output types here if you want.
- //
- // Simple example:
- // void const * my_output( void * output_ptr, int num_pixels, int y, void * context )
- // {
- // percentage_done = y / output_height;
- // fwrite( output_ptr, pixel_width_in_bytes, num_pixels, output_file );
- // }
- //===============================================================
- //===============================================================
- // optional built-in profiling API
- //--------------------------------
- #ifdef STBIR_PROFILE
- typedef struct STBIR_PROFILE_INFO
- {
- stbir_uint64 total_clocks;
- // how many clocks spent (of total_clocks) in the various resize routines, along with a string description
- // there are "resize_count" number of zones
- stbir_uint64 clocks[ 8 ];
- char const ** descriptions;
- // count of clocks and descriptions
- stbir_uint32 count;
- } STBIR_PROFILE_INFO;
- // use after calling stbir_resize_extended (or stbir_build_samplers or stbir_build_samplers_with_splits)
- STBIRDEF void stbir_resize_build_profile_info( STBIR_PROFILE_INFO * out_info, STBIR_RESIZE const * resize );
- // use after calling stbir_resize_extended
- STBIRDEF void stbir_resize_extended_profile_info( STBIR_PROFILE_INFO * out_info, STBIR_RESIZE const * resize );
- // use after calling stbir_resize_extended_split
- STBIRDEF void stbir_resize_split_profile_info( STBIR_PROFILE_INFO * out_info, STBIR_RESIZE const * resize, int split_start, int split_num );
- //===============================================================
- #endif
- //// end header file /////////////////////////////////////////////////////
- #endif // STBIR_INCLUDE_STB_IMAGE_RESIZE2_H
- #if defined(STB_IMAGE_RESIZE_IMPLEMENTATION) || defined(STB_IMAGE_RESIZE2_IMPLEMENTATION)
- #ifndef STBIR_ASSERT
- #include <assert.h>
- #define STBIR_ASSERT(x) assert(x)
- #endif
- #ifndef STBIR_MALLOC
- #include <stdlib.h>
- #define STBIR_MALLOC(size,user_data) ((void)(user_data), malloc(size))
- #define STBIR_FREE(ptr,user_data) ((void)(user_data), free(ptr))
- // (we used the comma operator to evaluate user_data, to avoid "unused parameter" warnings)
- #endif
- #ifdef _MSC_VER
- #define stbir__inline __forceinline
- #else
- #define stbir__inline __inline__
- // Clang address sanitizer
- #if defined(__has_feature)
- #if __has_feature(address_sanitizer) || __has_feature(memory_sanitizer)
- #ifndef STBIR__SEPARATE_ALLOCATIONS
- #define STBIR__SEPARATE_ALLOCATIONS
- #endif
- #endif
- #endif
- #endif
- // GCC and MSVC
- #if defined(__SANITIZE_ADDRESS__)
- #ifndef STBIR__SEPARATE_ALLOCATIONS
- #define STBIR__SEPARATE_ALLOCATIONS
- #endif
- #endif
- // Always turn off automatic FMA use - use STBIR_USE_FMA if you want.
- // Otherwise, this is a determinism disaster.
- #ifndef STBIR_DONT_CHANGE_FP_CONTRACT // override in case you don't want this behavior
- #if defined(_MSC_VER) && !defined(__clang__)
- #if _MSC_VER > 1200
- #pragma fp_contract(off)
- #endif
- #elif defined(__GNUC__) && !defined(__clang__)
- #pragma GCC optimize("fp-contract=off")
- #else
- #pragma STDC FP_CONTRACT OFF
- #endif
- #endif
- #ifdef _MSC_VER
- #define STBIR__UNUSED(v) (void)(v)
- #else
- #define STBIR__UNUSED(v) (void)sizeof(v)
- #endif
- #define STBIR__ARRAY_SIZE(a) (sizeof((a))/sizeof((a)[0]))
- #ifndef STBIR_DEFAULT_FILTER_UPSAMPLE
- #define STBIR_DEFAULT_FILTER_UPSAMPLE STBIR_FILTER_CATMULLROM
- #endif
- #ifndef STBIR_DEFAULT_FILTER_DOWNSAMPLE
- #define STBIR_DEFAULT_FILTER_DOWNSAMPLE STBIR_FILTER_MITCHELL
- #endif
- #ifndef STBIR__HEADER_FILENAME
- #define STBIR__HEADER_FILENAME "stb_image_resize2.h"
- #endif
- // the internal pixel layout enums are in a different order, so we can easily do range comparisons of types
- // the public pixel layout is ordered in a way that if you cast num_channels (1-4) to the enum, you get something sensible
- typedef enum
- {
- STBIRI_1CHANNEL = 0,
- STBIRI_2CHANNEL = 1,
- STBIRI_RGB = 2,
- STBIRI_BGR = 3,
- STBIRI_4CHANNEL = 4,
- STBIRI_RGBA = 5,
- STBIRI_BGRA = 6,
- STBIRI_ARGB = 7,
- STBIRI_ABGR = 8,
- STBIRI_RA = 9,
- STBIRI_AR = 10,
- STBIRI_RGBA_PM = 11,
- STBIRI_BGRA_PM = 12,
- STBIRI_ARGB_PM = 13,
- STBIRI_ABGR_PM = 14,
- STBIRI_RA_PM = 15,
- STBIRI_AR_PM = 16,
- } stbir_internal_pixel_layout;
- // define the public pixel layouts to not compile inside the implementation (to avoid accidental use)
- #define STBIR_BGR bad_dont_use_in_implementation
- #define STBIR_1CHANNEL STBIR_BGR
- #define STBIR_2CHANNEL STBIR_BGR
- #define STBIR_RGB STBIR_BGR
- #define STBIR_RGBA STBIR_BGR
- #define STBIR_4CHANNEL STBIR_BGR
- #define STBIR_BGRA STBIR_BGR
- #define STBIR_ARGB STBIR_BGR
- #define STBIR_ABGR STBIR_BGR
- #define STBIR_RA STBIR_BGR
- #define STBIR_AR STBIR_BGR
- #define STBIR_RGBA_PM STBIR_BGR
- #define STBIR_BGRA_PM STBIR_BGR
- #define STBIR_ARGB_PM STBIR_BGR
- #define STBIR_ABGR_PM STBIR_BGR
- #define STBIR_RA_PM STBIR_BGR
- #define STBIR_AR_PM STBIR_BGR
- // must match stbir_datatype
- static unsigned char stbir__type_size[] = {
- 1,1,1,2,4,2 // STBIR_TYPE_UINT8,STBIR_TYPE_UINT8_SRGB,STBIR_TYPE_UINT8_SRGB_ALPHA,STBIR_TYPE_UINT16,STBIR_TYPE_FLOAT,STBIR_TYPE_HALF_FLOAT
- };
- // When gathering, the contributors are which source pixels contribute.
- // When scattering, the contributors are which destination pixels are contributed to.
- typedef struct
- {
- int n0; // First contributing pixel
- int n1; // Last contributing pixel
- } stbir__contributors;
- typedef struct
- {
- int lowest; // First sample index for whole filter
- int highest; // Last sample index for whole filter
- int widest; // widest single set of samples for an output
- } stbir__filter_extent_info;
- typedef struct
- {
- int n0; // First pixel of decode buffer to write to
- int n1; // Last pixel of decode that will be written to
- int pixel_offset_for_input; // Pixel offset into input_scanline
- } stbir__span;
- typedef struct stbir__scale_info
- {
- int input_full_size;
- int output_sub_size;
- float scale;
- float inv_scale;
- float pixel_shift; // starting shift in output pixel space (in pixels)
- int scale_is_rational;
- stbir_uint32 scale_numerator, scale_denominator;
- } stbir__scale_info;
- typedef struct
- {
- stbir__contributors * contributors;
- float* coefficients;
- stbir__contributors * gather_prescatter_contributors;
- float * gather_prescatter_coefficients;
- stbir__scale_info scale_info;
- float support;
- stbir_filter filter_enum;
- stbir__kernel_callback * filter_kernel;
- stbir__support_callback * filter_support;
- stbir_edge edge;
- int coefficient_width;
- int filter_pixel_width;
- int filter_pixel_margin;
- int num_contributors;
- int contributors_size;
- int coefficients_size;
- stbir__filter_extent_info extent_info;
- int is_gather; // 0 = scatter, 1 = gather with scale >= 1, 2 = gather with scale < 1
- int gather_prescatter_num_contributors;
- int gather_prescatter_coefficient_width;
- int gather_prescatter_contributors_size;
- int gather_prescatter_coefficients_size;
- } stbir__sampler;
- typedef struct
- {
- stbir__contributors conservative;
- int edge_sizes[2]; // this can be less than filter_pixel_margin, if the filter and scaling falls off
- stbir__span spans[2]; // can be two spans, if doing input subrect with clamp mode WRAP
- } stbir__extents;
- typedef struct
- {
- #ifdef STBIR_PROFILE
- union
- {
- struct { stbir_uint64 total, looping, vertical, horizontal, decode, encode, alpha, unalpha; } named;
- stbir_uint64 array[8];
- } profile;
- stbir_uint64 * current_zone_excluded_ptr;
- #endif
- float* decode_buffer;
- int ring_buffer_first_scanline;
- int ring_buffer_last_scanline;
- int ring_buffer_begin_index; // first_scanline is at this index in the ring buffer
- int start_output_y, end_output_y;
- int start_input_y, end_input_y; // used in scatter only
- #ifdef STBIR__SEPARATE_ALLOCATIONS
- float** ring_buffers; // one pointer for each ring buffer
- #else
- float* ring_buffer; // one big buffer that we index into
- #endif
- float* vertical_buffer;
- char no_cache_straddle[64];
- } stbir__per_split_info;
- typedef float * stbir__decode_pixels_func( float * decode, int width_times_channels, void const * input );
- typedef void stbir__alpha_weight_func( float * decode_buffer, int width_times_channels );
- typedef void stbir__horizontal_gather_channels_func( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer,
- stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width );
- typedef void stbir__alpha_unweight_func(float * encode_buffer, int width_times_channels );
- typedef void stbir__encode_pixels_func( void * output, int width_times_channels, float const * encode );
- struct stbir__info
- {
- #ifdef STBIR_PROFILE
- union
- {
- struct { stbir_uint64 total, build, alloc, horizontal, vertical, cleanup, pivot; } named;
- stbir_uint64 array[7];
- } profile;
- stbir_uint64 * current_zone_excluded_ptr;
- #endif
- stbir__sampler horizontal;
- stbir__sampler vertical;
- void const * input_data;
- void * output_data;
- int input_stride_bytes;
- int output_stride_bytes;
- int ring_buffer_length_bytes; // The length of an individual entry in the ring buffer. The total number of ring buffers is stbir__get_filter_pixel_width(filter)
- int ring_buffer_num_entries; // Total number of entries in the ring buffer.
- stbir_datatype input_type;
- stbir_datatype output_type;
- stbir_input_callback * in_pixels_cb;
- void * user_data;
- stbir_output_callback * out_pixels_cb;
- stbir__extents scanline_extents;
- void * alloced_mem;
- stbir__per_split_info * split_info; // by default 1, but there will be N of these allocated based on the thread init you did
- stbir__decode_pixels_func * decode_pixels;
- stbir__alpha_weight_func * alpha_weight;
- stbir__horizontal_gather_channels_func * horizontal_gather_channels;
- stbir__alpha_unweight_func * alpha_unweight;
- stbir__encode_pixels_func * encode_pixels;
- int alloc_ring_buffer_num_entries; // Number of entries in the ring buffer that will be allocated
- int splits; // count of splits
- stbir_internal_pixel_layout input_pixel_layout_internal;
- stbir_internal_pixel_layout output_pixel_layout_internal;
- int input_color_and_type;
- int offset_x, offset_y; // offset within output_data
- int vertical_first;
- int channels;
- int effective_channels; // same as channels, except on RGBA/ARGB (7), or XA/AX (3)
- size_t alloced_total;
- };
- #define stbir__max_uint8_as_float 255.0f
- #define stbir__max_uint16_as_float 65535.0f
- #define stbir__max_uint8_as_float_inverted 3.9215689e-03f // (1.0f/255.0f)
- #define stbir__max_uint16_as_float_inverted 1.5259022e-05f // (1.0f/65535.0f)
- #define stbir__small_float ((float)1 / (1 << 20) / (1 << 20) / (1 << 20) / (1 << 20) / (1 << 20) / (1 << 20))
- // min/max friendly
- #define STBIR_CLAMP(x, xmin, xmax) for(;;) { \
- if ( (x) < (xmin) ) (x) = (xmin); \
- if ( (x) > (xmax) ) (x) = (xmax); \
- break; \
- }
- static stbir__inline int stbir__min(int a, int b)
- {
- return a < b ? a : b;
- }
- static stbir__inline int stbir__max(int a, int b)
- {
- return a > b ? a : b;
- }
- static float stbir__srgb_uchar_to_linear_float[256] = {
- 0.000000f, 0.000304f, 0.000607f, 0.000911f, 0.001214f, 0.001518f, 0.001821f, 0.002125f, 0.002428f, 0.002732f, 0.003035f,
- 0.003347f, 0.003677f, 0.004025f, 0.004391f, 0.004777f, 0.005182f, 0.005605f, 0.006049f, 0.006512f, 0.006995f, 0.007499f,
- 0.008023f, 0.008568f, 0.009134f, 0.009721f, 0.010330f, 0.010960f, 0.011612f, 0.012286f, 0.012983f, 0.013702f, 0.014444f,
- 0.015209f, 0.015996f, 0.016807f, 0.017642f, 0.018500f, 0.019382f, 0.020289f, 0.021219f, 0.022174f, 0.023153f, 0.024158f,
- 0.025187f, 0.026241f, 0.027321f, 0.028426f, 0.029557f, 0.030713f, 0.031896f, 0.033105f, 0.034340f, 0.035601f, 0.036889f,
- 0.038204f, 0.039546f, 0.040915f, 0.042311f, 0.043735f, 0.045186f, 0.046665f, 0.048172f, 0.049707f, 0.051269f, 0.052861f,
- 0.054480f, 0.056128f, 0.057805f, 0.059511f, 0.061246f, 0.063010f, 0.064803f, 0.066626f, 0.068478f, 0.070360f, 0.072272f,
- 0.074214f, 0.076185f, 0.078187f, 0.080220f, 0.082283f, 0.084376f, 0.086500f, 0.088656f, 0.090842f, 0.093059f, 0.095307f,
- 0.097587f, 0.099899f, 0.102242f, 0.104616f, 0.107023f, 0.109462f, 0.111932f, 0.114435f, 0.116971f, 0.119538f, 0.122139f,
- 0.124772f, 0.127438f, 0.130136f, 0.132868f, 0.135633f, 0.138432f, 0.141263f, 0.144128f, 0.147027f, 0.149960f, 0.152926f,
- 0.155926f, 0.158961f, 0.162029f, 0.165132f, 0.168269f, 0.171441f, 0.174647f, 0.177888f, 0.181164f, 0.184475f, 0.187821f,
- 0.191202f, 0.194618f, 0.198069f, 0.201556f, 0.205079f, 0.208637f, 0.212231f, 0.215861f, 0.219526f, 0.223228f, 0.226966f,
- 0.230740f, 0.234551f, 0.238398f, 0.242281f, 0.246201f, 0.250158f, 0.254152f, 0.258183f, 0.262251f, 0.266356f, 0.270498f,
- 0.274677f, 0.278894f, 0.283149f, 0.287441f, 0.291771f, 0.296138f, 0.300544f, 0.304987f, 0.309469f, 0.313989f, 0.318547f,
- 0.323143f, 0.327778f, 0.332452f, 0.337164f, 0.341914f, 0.346704f, 0.351533f, 0.356400f, 0.361307f, 0.366253f, 0.371238f,
- 0.376262f, 0.381326f, 0.386430f, 0.391573f, 0.396755f, 0.401978f, 0.407240f, 0.412543f, 0.417885f, 0.423268f, 0.428691f,
- 0.434154f, 0.439657f, 0.445201f, 0.450786f, 0.456411f, 0.462077f, 0.467784f, 0.473532f, 0.479320f, 0.485150f, 0.491021f,
- 0.496933f, 0.502887f, 0.508881f, 0.514918f, 0.520996f, 0.527115f, 0.533276f, 0.539480f, 0.545725f, 0.552011f, 0.558340f,
- 0.564712f, 0.571125f, 0.577581f, 0.584078f, 0.590619f, 0.597202f, 0.603827f, 0.610496f, 0.617207f, 0.623960f, 0.630757f,
- 0.637597f, 0.644480f, 0.651406f, 0.658375f, 0.665387f, 0.672443f, 0.679543f, 0.686685f, 0.693872f, 0.701102f, 0.708376f,
- 0.715694f, 0.723055f, 0.730461f, 0.737911f, 0.745404f, 0.752942f, 0.760525f, 0.768151f, 0.775822f, 0.783538f, 0.791298f,
- 0.799103f, 0.806952f, 0.814847f, 0.822786f, 0.830770f, 0.838799f, 0.846873f, 0.854993f, 0.863157f, 0.871367f, 0.879622f,
- 0.887923f, 0.896269f, 0.904661f, 0.913099f, 0.921582f, 0.930111f, 0.938686f, 0.947307f, 0.955974f, 0.964686f, 0.973445f,
- 0.982251f, 0.991102f, 1.0f
- };
- typedef union
- {
- unsigned int u;
- float f;
- } stbir__FP32;
- // From https://gist.github.com/rygorous/2203834
- static const stbir_uint32 fp32_to_srgb8_tab4[104] = {
- 0x0073000d, 0x007a000d, 0x0080000d, 0x0087000d, 0x008d000d, 0x0094000d, 0x009a000d, 0x00a1000d,
- 0x00a7001a, 0x00b4001a, 0x00c1001a, 0x00ce001a, 0x00da001a, 0x00e7001a, 0x00f4001a, 0x0101001a,
- 0x010e0033, 0x01280033, 0x01410033, 0x015b0033, 0x01750033, 0x018f0033, 0x01a80033, 0x01c20033,
- 0x01dc0067, 0x020f0067, 0x02430067, 0x02760067, 0x02aa0067, 0x02dd0067, 0x03110067, 0x03440067,
- 0x037800ce, 0x03df00ce, 0x044600ce, 0x04ad00ce, 0x051400ce, 0x057b00c5, 0x05dd00bc, 0x063b00b5,
- 0x06970158, 0x07420142, 0x07e30130, 0x087b0120, 0x090b0112, 0x09940106, 0x0a1700fc, 0x0a9500f2,
- 0x0b0f01cb, 0x0bf401ae, 0x0ccb0195, 0x0d950180, 0x0e56016e, 0x0f0d015e, 0x0fbc0150, 0x10630143,
- 0x11070264, 0x1238023e, 0x1357021d, 0x14660201, 0x156601e9, 0x165a01d3, 0x174401c0, 0x182401af,
- 0x18fe0331, 0x1a9602fe, 0x1c1502d2, 0x1d7e02ad, 0x1ed4028d, 0x201a0270, 0x21520256, 0x227d0240,
- 0x239f0443, 0x25c003fe, 0x27bf03c4, 0x29a10392, 0x2b6a0367, 0x2d1d0341, 0x2ebe031f, 0x304d0300,
- 0x31d105b0, 0x34a80555, 0x37520507, 0x39d504c5, 0x3c37048b, 0x3e7c0458, 0x40a8042a, 0x42bd0401,
- 0x44c20798, 0x488e071e, 0x4c1c06b6, 0x4f76065d, 0x52a50610, 0x55ac05cc, 0x5892058f, 0x5b590559,
- 0x5e0c0a23, 0x631c0980, 0x67db08f6, 0x6c55087f, 0x70940818, 0x74a007bd, 0x787d076c, 0x7c330723,
- };
- static stbir__inline stbir_uint8 stbir__linear_to_srgb_uchar(float in)
- {
- static const stbir__FP32 almostone = { 0x3f7fffff }; // 1-eps
- static const stbir__FP32 minval = { (127-13) << 23 };
- stbir_uint32 tab,bias,scale,t;
- stbir__FP32 f;
- // Clamp to [2^(-13), 1-eps]; these two values map to 0 and 1, respectively.
- // The tests are carefully written so that NaNs map to 0, same as in the reference
- // implementation.
- if (!(in > minval.f)) // written this way to catch NaNs
- return 0;
- if (in > almostone.f)
- return 255;
- // Do the table lookup and unpack bias, scale
- f.f = in;
- tab = fp32_to_srgb8_tab4[(f.u - minval.u) >> 20];
- bias = (tab >> 16) << 9;
- scale = tab & 0xffff;
- // Grab next-highest mantissa bits and perform linear interpolation
- t = (f.u >> 12) & 0xff;
- return (unsigned char) ((bias + scale*t) >> 16);
- }
- #ifndef STBIR_FORCE_GATHER_FILTER_SCANLINES_AMOUNT
- #define STBIR_FORCE_GATHER_FILTER_SCANLINES_AMOUNT 32 // when downsampling and <= 32 scanlines of buffering, use gather. gather used down to 1/8th scaling for 25% win.
- #endif
- #ifndef STBIR_FORCE_MINIMUM_SCANLINES_FOR_SPLITS
- #define STBIR_FORCE_MINIMUM_SCANLINES_FOR_SPLITS 4 // when threading, what is the minimum number of scanlines for a split?
- #endif
- #define STBIR_INPUT_CALLBACK_PADDING 3
- #ifdef _M_IX86_FP
- #if ( _M_IX86_FP >= 1 )
- #ifndef STBIR_SSE
- #define STBIR_SSE
- #endif
- #endif
- #endif
- #ifdef __TINYC__
- // tiny c has no intrinsics yet - this can become a version check if they add them
- #define STBIR_NO_SIMD
- #endif
- #if defined(_x86_64) || defined( __x86_64__ ) || defined( _M_X64 ) || defined(__x86_64) || defined(_M_AMD64) || defined(__SSE2__) || defined(STBIR_SSE) || defined(STBIR_SSE2)
- #ifndef STBIR_SSE2
- #define STBIR_SSE2
- #endif
- #if defined(__AVX__) || defined(STBIR_AVX2)
- #ifndef STBIR_AVX
- #ifndef STBIR_NO_AVX
- #define STBIR_AVX
- #endif
- #endif
- #endif
- #if defined(__AVX2__) || defined(STBIR_AVX2)
- #ifndef STBIR_NO_AVX2
- #ifndef STBIR_AVX2
- #define STBIR_AVX2
- #endif
- #if defined( _MSC_VER ) && !defined(__clang__)
- #ifndef STBIR_FP16C // FP16C instructions are on all AVX2 cpus, so we can autoselect it here on microsoft - clang needs -m16c
- #define STBIR_FP16C
- #endif
- #endif
- #endif
- #endif
- #ifdef __F16C__
- #ifndef STBIR_FP16C // turn on FP16C instructions if the define is set (for clang and gcc)
- #define STBIR_FP16C
- #endif
- #endif
- #endif
- #if defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ ) || ((__ARM_NEON_FP & 4) != 0) || defined(__ARM_NEON__)
- #ifndef STBIR_NEON
- #define STBIR_NEON
- #endif
- #endif
- #if defined(_M_ARM) || defined(__arm__)
- #ifdef STBIR_USE_FMA
- #undef STBIR_USE_FMA // no FMA for 32-bit arm on MSVC
- #endif
- #endif
- #if defined(__wasm__) && defined(__wasm_simd128__)
- #ifndef STBIR_WASM
- #define STBIR_WASM
- #endif
- #endif
- // restrict pointers for the output pointers, other loop and unroll control
- #if defined( _MSC_VER ) && !defined(__clang__)
- #define STBIR_STREAMOUT_PTR( star ) star __restrict
- #define STBIR_NO_UNROLL( ptr ) __assume(ptr) // this oddly keeps msvc from unrolling a loop
- #if _MSC_VER >= 1900
- #define STBIR_NO_UNROLL_LOOP_START __pragma(loop( no_vector ))
- #else
- #define STBIR_NO_UNROLL_LOOP_START
- #endif
- #elif defined( __clang__ )
- #define STBIR_STREAMOUT_PTR( star ) star __restrict__
- #define STBIR_NO_UNROLL( ptr ) __asm__ (""::"r"(ptr))
- #if ( __clang_major__ >= 4 ) || ( ( __clang_major__ >= 3 ) && ( __clang_minor__ >= 5 ) )
- #define STBIR_NO_UNROLL_LOOP_START _Pragma("clang loop unroll(disable)") _Pragma("clang loop vectorize(disable)")
- #else
- #define STBIR_NO_UNROLL_LOOP_START
- #endif
- #elif defined( __GNUC__ )
- #define STBIR_STREAMOUT_PTR( star ) star __restrict__
- #define STBIR_NO_UNROLL( ptr ) __asm__ (""::"r"(ptr))
- #if __GNUC__ >= 14
- #define STBIR_NO_UNROLL_LOOP_START _Pragma("GCC unroll 0") _Pragma("GCC novector")
- #else
- #define STBIR_NO_UNROLL_LOOP_START
- #endif
- #define STBIR_NO_UNROLL_LOOP_START_INF_FOR
- #else
- #define STBIR_STREAMOUT_PTR( star ) star
- #define STBIR_NO_UNROLL( ptr )
- #define STBIR_NO_UNROLL_LOOP_START
- #endif
- #ifndef STBIR_NO_UNROLL_LOOP_START_INF_FOR
- #define STBIR_NO_UNROLL_LOOP_START_INF_FOR STBIR_NO_UNROLL_LOOP_START
- #endif
- #ifdef STBIR_NO_SIMD // force simd off for whatever reason
- // force simd off overrides everything else, so clear it all
- #ifdef STBIR_SSE2
- #undef STBIR_SSE2
- #endif
- #ifdef STBIR_AVX
- #undef STBIR_AVX
- #endif
- #ifdef STBIR_NEON
- #undef STBIR_NEON
- #endif
- #ifdef STBIR_AVX2
- #undef STBIR_AVX2
- #endif
- #ifdef STBIR_FP16C
- #undef STBIR_FP16C
- #endif
- #ifdef STBIR_WASM
- #undef STBIR_WASM
- #endif
- #ifdef STBIR_SIMD
- #undef STBIR_SIMD
- #endif
- #else // STBIR_SIMD
- #ifdef STBIR_SSE2
- #include <emmintrin.h>
- #define stbir__simdf __m128
- #define stbir__simdi __m128i
- #define stbir_simdi_castf( reg ) _mm_castps_si128(reg)
- #define stbir_simdf_casti( reg ) _mm_castsi128_ps(reg)
- #define stbir__simdf_load( reg, ptr ) (reg) = _mm_loadu_ps( (float const*)(ptr) )
- #define stbir__simdi_load( reg, ptr ) (reg) = _mm_loadu_si128 ( (stbir__simdi const*)(ptr) )
- #define stbir__simdf_load1( out, ptr ) (out) = _mm_load_ss( (float const*)(ptr) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdi_load1( out, ptr ) (out) = _mm_castps_si128( _mm_load_ss( (float const*)(ptr) ))
- #define stbir__simdf_load1z( out, ptr ) (out) = _mm_load_ss( (float const*)(ptr) ) // top values must be zero
- #define stbir__simdf_frep4( fvar ) _mm_set_ps1( fvar )
- #define stbir__simdf_load1frep4( out, fvar ) (out) = _mm_set_ps1( fvar )
- #define stbir__simdf_load2( out, ptr ) (out) = _mm_castsi128_ps( _mm_loadl_epi64( (__m128i*)(ptr)) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdf_load2z( out, ptr ) (out) = _mm_castsi128_ps( _mm_loadl_epi64( (__m128i*)(ptr)) ) // top values must be zero
- #define stbir__simdf_load2hmerge( out, reg, ptr ) (out) = _mm_castpd_ps(_mm_loadh_pd( _mm_castps_pd(reg), (double*)(ptr) ))
- #define stbir__simdf_zeroP() _mm_setzero_ps()
- #define stbir__simdf_zero( reg ) (reg) = _mm_setzero_ps()
- #define stbir__simdf_store( ptr, reg ) _mm_storeu_ps( (float*)(ptr), reg )
- #define stbir__simdf_store1( ptr, reg ) _mm_store_ss( (float*)(ptr), reg )
- #define stbir__simdf_store2( ptr, reg ) _mm_storel_epi64( (__m128i*)(ptr), _mm_castps_si128(reg) )
- #define stbir__simdf_store2h( ptr, reg ) _mm_storeh_pd( (double*)(ptr), _mm_castps_pd(reg) )
- #define stbir__simdi_store( ptr, reg ) _mm_storeu_si128( (__m128i*)(ptr), reg )
- #define stbir__simdi_store1( ptr, reg ) _mm_store_ss( (float*)(ptr), _mm_castsi128_ps(reg) )
- #define stbir__simdi_store2( ptr, reg ) _mm_storel_epi64( (__m128i*)(ptr), (reg) )
- #define stbir__prefetch( ptr ) _mm_prefetch((char*)(ptr), _MM_HINT_T0 )
- #define stbir__simdi_expand_u8_to_u32(out0,out1,out2,out3,ireg) \
- { \
- stbir__simdi zero = _mm_setzero_si128(); \
- out2 = _mm_unpacklo_epi8( ireg, zero ); \
- out3 = _mm_unpackhi_epi8( ireg, zero ); \
- out0 = _mm_unpacklo_epi16( out2, zero ); \
- out1 = _mm_unpackhi_epi16( out2, zero ); \
- out2 = _mm_unpacklo_epi16( out3, zero ); \
- out3 = _mm_unpackhi_epi16( out3, zero ); \
- }
- #define stbir__simdi_expand_u8_to_1u32(out,ireg) \
- { \
- stbir__simdi zero = _mm_setzero_si128(); \
- out = _mm_unpacklo_epi8( ireg, zero ); \
- out = _mm_unpacklo_epi16( out, zero ); \
- }
- #define stbir__simdi_expand_u16_to_u32(out0,out1,ireg) \
- { \
- stbir__simdi zero = _mm_setzero_si128(); \
- out0 = _mm_unpacklo_epi16( ireg, zero ); \
- out1 = _mm_unpackhi_epi16( ireg, zero ); \
- }
- #define stbir__simdf_convert_float_to_i32( i, f ) (i) = _mm_cvttps_epi32(f)
- #define stbir__simdf_convert_float_to_int( f ) _mm_cvtt_ss2si(f)
- #define stbir__simdf_convert_float_to_uint8( f ) ((unsigned char)_mm_cvtsi128_si32(_mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(f,STBIR__CONSTF(STBIR_max_uint8_as_float)),_mm_setzero_ps()))))
- #define stbir__simdf_convert_float_to_short( f ) ((unsigned short)_mm_cvtsi128_si32(_mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(f,STBIR__CONSTF(STBIR_max_uint16_as_float)),_mm_setzero_ps()))))
- #define stbir__simdi_to_int( i ) _mm_cvtsi128_si32(i)
- #define stbir__simdi_convert_i32_to_float(out, ireg) (out) = _mm_cvtepi32_ps( ireg )
- #define stbir__simdf_add( out, reg0, reg1 ) (out) = _mm_add_ps( reg0, reg1 )
- #define stbir__simdf_mult( out, reg0, reg1 ) (out) = _mm_mul_ps( reg0, reg1 )
- #define stbir__simdf_mult_mem( out, reg, ptr ) (out) = _mm_mul_ps( reg, _mm_loadu_ps( (float const*)(ptr) ) )
- #define stbir__simdf_mult1_mem( out, reg, ptr ) (out) = _mm_mul_ss( reg, _mm_load_ss( (float const*)(ptr) ) )
- #define stbir__simdf_add_mem( out, reg, ptr ) (out) = _mm_add_ps( reg, _mm_loadu_ps( (float const*)(ptr) ) )
- #define stbir__simdf_add1_mem( out, reg, ptr ) (out) = _mm_add_ss( reg, _mm_load_ss( (float const*)(ptr) ) )
- #ifdef STBIR_USE_FMA // not on by default to maintain bit identical simd to non-simd
- #include <immintrin.h>
- #define stbir__simdf_madd( out, add, mul1, mul2 ) (out) = _mm_fmadd_ps( mul1, mul2, add )
- #define stbir__simdf_madd1( out, add, mul1, mul2 ) (out) = _mm_fmadd_ss( mul1, mul2, add )
- #define stbir__simdf_madd_mem( out, add, mul, ptr ) (out) = _mm_fmadd_ps( mul, _mm_loadu_ps( (float const*)(ptr) ), add )
- #define stbir__simdf_madd1_mem( out, add, mul, ptr ) (out) = _mm_fmadd_ss( mul, _mm_load_ss( (float const*)(ptr) ), add )
- #else
- #define stbir__simdf_madd( out, add, mul1, mul2 ) (out) = _mm_add_ps( add, _mm_mul_ps( mul1, mul2 ) )
- #define stbir__simdf_madd1( out, add, mul1, mul2 ) (out) = _mm_add_ss( add, _mm_mul_ss( mul1, mul2 ) )
- #define stbir__simdf_madd_mem( out, add, mul, ptr ) (out) = _mm_add_ps( add, _mm_mul_ps( mul, _mm_loadu_ps( (float const*)(ptr) ) ) )
- #define stbir__simdf_madd1_mem( out, add, mul, ptr ) (out) = _mm_add_ss( add, _mm_mul_ss( mul, _mm_load_ss( (float const*)(ptr) ) ) )
- #endif
- #define stbir__simdf_add1( out, reg0, reg1 ) (out) = _mm_add_ss( reg0, reg1 )
- #define stbir__simdf_mult1( out, reg0, reg1 ) (out) = _mm_mul_ss( reg0, reg1 )
- #define stbir__simdf_and( out, reg0, reg1 ) (out) = _mm_and_ps( reg0, reg1 )
- #define stbir__simdf_or( out, reg0, reg1 ) (out) = _mm_or_ps( reg0, reg1 )
- #define stbir__simdf_min( out, reg0, reg1 ) (out) = _mm_min_ps( reg0, reg1 )
- #define stbir__simdf_max( out, reg0, reg1 ) (out) = _mm_max_ps( reg0, reg1 )
- #define stbir__simdf_min1( out, reg0, reg1 ) (out) = _mm_min_ss( reg0, reg1 )
- #define stbir__simdf_max1( out, reg0, reg1 ) (out) = _mm_max_ss( reg0, reg1 )
- #define stbir__simdf_0123ABCDto3ABx( out, reg0, reg1 ) (out)=_mm_castsi128_ps( _mm_shuffle_epi32( _mm_castps_si128( _mm_shuffle_ps( reg1,reg0, (0<<0) + (1<<2) + (2<<4) + (3<<6) )), (3<<0) + (0<<2) + (1<<4) + (2<<6) ) )
- #define stbir__simdf_0123ABCDto23Ax( out, reg0, reg1 ) (out)=_mm_castsi128_ps( _mm_shuffle_epi32( _mm_castps_si128( _mm_shuffle_ps( reg1,reg0, (0<<0) + (1<<2) + (2<<4) + (3<<6) )), (2<<0) + (3<<2) + (0<<4) + (1<<6) ) )
- static const stbir__simdf STBIR_zeroones = { 0.0f,1.0f,0.0f,1.0f };
- static const stbir__simdf STBIR_onezeros = { 1.0f,0.0f,1.0f,0.0f };
- #define stbir__simdf_aaa1( out, alp, ones ) (out)=_mm_castsi128_ps( _mm_shuffle_epi32( _mm_castps_si128( _mm_movehl_ps( ones, alp ) ), (1<<0) + (1<<2) + (1<<4) + (2<<6) ) )
- #define stbir__simdf_1aaa( out, alp, ones ) (out)=_mm_castsi128_ps( _mm_shuffle_epi32( _mm_castps_si128( _mm_movelh_ps( ones, alp ) ), (0<<0) + (2<<2) + (2<<4) + (2<<6) ) )
- #define stbir__simdf_a1a1( out, alp, ones) (out) = _mm_or_ps( _mm_castsi128_ps( _mm_srli_epi64( _mm_castps_si128(alp), 32 ) ), STBIR_zeroones )
- #define stbir__simdf_1a1a( out, alp, ones) (out) = _mm_or_ps( _mm_castsi128_ps( _mm_slli_epi64( _mm_castps_si128(alp), 32 ) ), STBIR_onezeros )
- #define stbir__simdf_swiz( reg, one, two, three, four ) _mm_castsi128_ps( _mm_shuffle_epi32( _mm_castps_si128( reg ), (one<<0) + (two<<2) + (three<<4) + (four<<6) ) )
- #define stbir__simdi_and( out, reg0, reg1 ) (out) = _mm_and_si128( reg0, reg1 )
- #define stbir__simdi_or( out, reg0, reg1 ) (out) = _mm_or_si128( reg0, reg1 )
- #define stbir__simdi_16madd( out, reg0, reg1 ) (out) = _mm_madd_epi16( reg0, reg1 )
- #define stbir__simdf_pack_to_8bytes(out,aa,bb) \
- { \
- stbir__simdf af,bf; \
- stbir__simdi a,b; \
- af = _mm_min_ps( aa, STBIR_max_uint8_as_float ); \
- bf = _mm_min_ps( bb, STBIR_max_uint8_as_float ); \
- af = _mm_max_ps( af, _mm_setzero_ps() ); \
- bf = _mm_max_ps( bf, _mm_setzero_ps() ); \
- a = _mm_cvttps_epi32( af ); \
- b = _mm_cvttps_epi32( bf ); \
- a = _mm_packs_epi32( a, b ); \
- out = _mm_packus_epi16( a, a ); \
- }
- #define stbir__simdf_load4_transposed( o0, o1, o2, o3, ptr ) \
- stbir__simdf_load( o0, (ptr) ); \
- stbir__simdf_load( o1, (ptr)+4 ); \
- stbir__simdf_load( o2, (ptr)+8 ); \
- stbir__simdf_load( o3, (ptr)+12 ); \
- { \
- __m128 tmp0, tmp1, tmp2, tmp3; \
- tmp0 = _mm_unpacklo_ps(o0, o1); \
- tmp2 = _mm_unpacklo_ps(o2, o3); \
- tmp1 = _mm_unpackhi_ps(o0, o1); \
- tmp3 = _mm_unpackhi_ps(o2, o3); \
- o0 = _mm_movelh_ps(tmp0, tmp2); \
- o1 = _mm_movehl_ps(tmp2, tmp0); \
- o2 = _mm_movelh_ps(tmp1, tmp3); \
- o3 = _mm_movehl_ps(tmp3, tmp1); \
- }
- #define stbir__interleave_pack_and_store_16_u8( ptr, r0, r1, r2, r3 ) \
- r0 = _mm_packs_epi32( r0, r1 ); \
- r2 = _mm_packs_epi32( r2, r3 ); \
- r1 = _mm_unpacklo_epi16( r0, r2 ); \
- r3 = _mm_unpackhi_epi16( r0, r2 ); \
- r0 = _mm_unpacklo_epi16( r1, r3 ); \
- r2 = _mm_unpackhi_epi16( r1, r3 ); \
- r0 = _mm_packus_epi16( r0, r2 ); \
- stbir__simdi_store( ptr, r0 ); \
- #define stbir__simdi_32shr( out, reg, imm ) out = _mm_srli_epi32( reg, imm )
- #if defined(_MSC_VER) && !defined(__clang__)
- // msvc inits with 8 bytes
- #define STBIR__CONST_32_TO_8( v ) (char)(unsigned char)((v)&255),(char)(unsigned char)(((v)>>8)&255),(char)(unsigned char)(((v)>>16)&255),(char)(unsigned char)(((v)>>24)&255)
- #define STBIR__CONST_4_32i( v ) STBIR__CONST_32_TO_8( v ), STBIR__CONST_32_TO_8( v ), STBIR__CONST_32_TO_8( v ), STBIR__CONST_32_TO_8( v )
- #define STBIR__CONST_4d_32i( v0, v1, v2, v3 ) STBIR__CONST_32_TO_8( v0 ), STBIR__CONST_32_TO_8( v1 ), STBIR__CONST_32_TO_8( v2 ), STBIR__CONST_32_TO_8( v3 )
- #else
- // everything else inits with long long's
- #define STBIR__CONST_4_32i( v ) (long long)((((stbir_uint64)(stbir_uint32)(v))<<32)|((stbir_uint64)(stbir_uint32)(v))),(long long)((((stbir_uint64)(stbir_uint32)(v))<<32)|((stbir_uint64)(stbir_uint32)(v)))
- #define STBIR__CONST_4d_32i( v0, v1, v2, v3 ) (long long)((((stbir_uint64)(stbir_uint32)(v1))<<32)|((stbir_uint64)(stbir_uint32)(v0))),(long long)((((stbir_uint64)(stbir_uint32)(v3))<<32)|((stbir_uint64)(stbir_uint32)(v2)))
- #endif
- #define STBIR__SIMDF_CONST(var, x) stbir__simdf var = { x, x, x, x }
- #define STBIR__SIMDI_CONST(var, x) stbir__simdi var = { STBIR__CONST_4_32i(x) }
- #define STBIR__CONSTF(var) (var)
- #define STBIR__CONSTI(var) (var)
- #if defined(STBIR_AVX) || defined(__SSE4_1__)
- #include <smmintrin.h>
- #define stbir__simdf_pack_to_8words(out,reg0,reg1) out = _mm_packus_epi32(_mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(reg0,STBIR__CONSTF(STBIR_max_uint16_as_float)),_mm_setzero_ps())), _mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(reg1,STBIR__CONSTF(STBIR_max_uint16_as_float)),_mm_setzero_ps())))
- #else
- static STBIR__SIMDI_CONST(stbir__s32_32768, 32768);
- static STBIR__SIMDI_CONST(stbir__s16_32768, ((32768<<16)|32768));
- #define stbir__simdf_pack_to_8words(out,reg0,reg1) \
- { \
- stbir__simdi tmp0,tmp1; \
- tmp0 = _mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(reg0,STBIR__CONSTF(STBIR_max_uint16_as_float)),_mm_setzero_ps())); \
- tmp1 = _mm_cvttps_epi32(_mm_max_ps(_mm_min_ps(reg1,STBIR__CONSTF(STBIR_max_uint16_as_float)),_mm_setzero_ps())); \
- tmp0 = _mm_sub_epi32( tmp0, stbir__s32_32768 ); \
- tmp1 = _mm_sub_epi32( tmp1, stbir__s32_32768 ); \
- out = _mm_packs_epi32( tmp0, tmp1 ); \
- out = _mm_sub_epi16( out, stbir__s16_32768 ); \
- }
- #endif
- #define STBIR_SIMD
- // if we detect AVX, set the simd8 defines
- #ifdef STBIR_AVX
- #include <immintrin.h>
- #define STBIR_SIMD8
- #define stbir__simdf8 __m256
- #define stbir__simdi8 __m256i
- #define stbir__simdf8_load( out, ptr ) (out) = _mm256_loadu_ps( (float const *)(ptr) )
- #define stbir__simdi8_load( out, ptr ) (out) = _mm256_loadu_si256( (__m256i const *)(ptr) )
- #define stbir__simdf8_mult( out, a, b ) (out) = _mm256_mul_ps( (a), (b) )
- #define stbir__simdf8_store( ptr, out ) _mm256_storeu_ps( (float*)(ptr), out )
- #define stbir__simdi8_store( ptr, reg ) _mm256_storeu_si256( (__m256i*)(ptr), reg )
- #define stbir__simdf8_frep8( fval ) _mm256_set1_ps( fval )
- #define stbir__simdf8_min( out, reg0, reg1 ) (out) = _mm256_min_ps( reg0, reg1 )
- #define stbir__simdf8_max( out, reg0, reg1 ) (out) = _mm256_max_ps( reg0, reg1 )
- #define stbir__simdf8_add4halves( out, bot4, top8 ) (out) = _mm_add_ps( bot4, _mm256_extractf128_ps( top8, 1 ) )
- #define stbir__simdf8_mult_mem( out, reg, ptr ) (out) = _mm256_mul_ps( reg, _mm256_loadu_ps( (float const*)(ptr) ) )
- #define stbir__simdf8_add_mem( out, reg, ptr ) (out) = _mm256_add_ps( reg, _mm256_loadu_ps( (float const*)(ptr) ) )
- #define stbir__simdf8_add( out, a, b ) (out) = _mm256_add_ps( a, b )
- #define stbir__simdf8_load1b( out, ptr ) (out) = _mm256_broadcast_ss( ptr )
- #define stbir__simdf_load1rep4( out, ptr ) (out) = _mm_broadcast_ss( ptr ) // avx load instruction
- #define stbir__simdi8_convert_i32_to_float(out, ireg) (out) = _mm256_cvtepi32_ps( ireg )
- #define stbir__simdf8_convert_float_to_i32( i, f ) (i) = _mm256_cvttps_epi32(f)
- #define stbir__simdf8_bot4s( out, a, b ) (out) = _mm256_permute2f128_ps(a,b, (0<<0)+(2<<4) )
- #define stbir__simdf8_top4s( out, a, b ) (out) = _mm256_permute2f128_ps(a,b, (1<<0)+(3<<4) )
- #define stbir__simdf8_gettop4( reg ) _mm256_extractf128_ps(reg,1)
- #ifdef STBIR_AVX2
- #define stbir__simdi8_expand_u8_to_u32(out0,out1,ireg) \
- { \
- stbir__simdi8 a, zero =_mm256_setzero_si256();\
- a = _mm256_permute4x64_epi64( _mm256_unpacklo_epi8( _mm256_permute4x64_epi64(_mm256_castsi128_si256(ireg),(0<<0)+(2<<2)+(1<<4)+(3<<6)), zero ),(0<<0)+(2<<2)+(1<<4)+(3<<6)); \
- out0 = _mm256_unpacklo_epi16( a, zero ); \
- out1 = _mm256_unpackhi_epi16( a, zero ); \
- }
- #define stbir__simdf8_pack_to_16bytes(out,aa,bb) \
- { \
- stbir__simdi8 t; \
- stbir__simdf8 af,bf; \
- stbir__simdi8 a,b; \
- af = _mm256_min_ps( aa, STBIR_max_uint8_as_floatX ); \
- bf = _mm256_min_ps( bb, STBIR_max_uint8_as_floatX ); \
- af = _mm256_max_ps( af, _mm256_setzero_ps() ); \
- bf = _mm256_max_ps( bf, _mm256_setzero_ps() ); \
- a = _mm256_cvttps_epi32( af ); \
- b = _mm256_cvttps_epi32( bf ); \
- t = _mm256_permute4x64_epi64( _mm256_packs_epi32( a, b ), (0<<0)+(2<<2)+(1<<4)+(3<<6) ); \
- out = _mm256_castsi256_si128( _mm256_permute4x64_epi64( _mm256_packus_epi16( t, t ), (0<<0)+(2<<2)+(1<<4)+(3<<6) ) ); \
- }
- #define stbir__simdi8_expand_u16_to_u32(out,ireg) out = _mm256_unpacklo_epi16( _mm256_permute4x64_epi64(_mm256_castsi128_si256(ireg),(0<<0)+(2<<2)+(1<<4)+(3<<6)), _mm256_setzero_si256() );
- #define stbir__simdf8_pack_to_16words(out,aa,bb) \
- { \
- stbir__simdf8 af,bf; \
- stbir__simdi8 a,b; \
- af = _mm256_min_ps( aa, STBIR_max_uint16_as_floatX ); \
- bf = _mm256_min_ps( bb, STBIR_max_uint16_as_floatX ); \
- af = _mm256_max_ps( af, _mm256_setzero_ps() ); \
- bf = _mm256_max_ps( bf, _mm256_setzero_ps() ); \
- a = _mm256_cvttps_epi32( af ); \
- b = _mm256_cvttps_epi32( bf ); \
- (out) = _mm256_permute4x64_epi64( _mm256_packus_epi32(a, b), (0<<0)+(2<<2)+(1<<4)+(3<<6) ); \
- }
- #else
- #define stbir__simdi8_expand_u8_to_u32(out0,out1,ireg) \
- { \
- stbir__simdi a,zero = _mm_setzero_si128(); \
- a = _mm_unpacklo_epi8( ireg, zero ); \
- out0 = _mm256_setr_m128i( _mm_unpacklo_epi16( a, zero ), _mm_unpackhi_epi16( a, zero ) ); \
- a = _mm_unpackhi_epi8( ireg, zero ); \
- out1 = _mm256_setr_m128i( _mm_unpacklo_epi16( a, zero ), _mm_unpackhi_epi16( a, zero ) ); \
- }
- #define stbir__simdf8_pack_to_16bytes(out,aa,bb) \
- { \
- stbir__simdi t; \
- stbir__simdf8 af,bf; \
- stbir__simdi8 a,b; \
- af = _mm256_min_ps( aa, STBIR_max_uint8_as_floatX ); \
- bf = _mm256_min_ps( bb, STBIR_max_uint8_as_floatX ); \
- af = _mm256_max_ps( af, _mm256_setzero_ps() ); \
- bf = _mm256_max_ps( bf, _mm256_setzero_ps() ); \
- a = _mm256_cvttps_epi32( af ); \
- b = _mm256_cvttps_epi32( bf ); \
- out = _mm_packs_epi32( _mm256_castsi256_si128(a), _mm256_extractf128_si256( a, 1 ) ); \
- out = _mm_packus_epi16( out, out ); \
- t = _mm_packs_epi32( _mm256_castsi256_si128(b), _mm256_extractf128_si256( b, 1 ) ); \
- t = _mm_packus_epi16( t, t ); \
- out = _mm_castps_si128( _mm_shuffle_ps( _mm_castsi128_ps(out), _mm_castsi128_ps(t), (0<<0)+(1<<2)+(0<<4)+(1<<6) ) ); \
- }
- #define stbir__simdi8_expand_u16_to_u32(out,ireg) \
- { \
- stbir__simdi a,b,zero = _mm_setzero_si128(); \
- a = _mm_unpacklo_epi16( ireg, zero ); \
- b = _mm_unpackhi_epi16( ireg, zero ); \
- out = _mm256_insertf128_si256( _mm256_castsi128_si256( a ), b, 1 ); \
- }
- #define stbir__simdf8_pack_to_16words(out,aa,bb) \
- { \
- stbir__simdi t0,t1; \
- stbir__simdf8 af,bf; \
- stbir__simdi8 a,b; \
- af = _mm256_min_ps( aa, STBIR_max_uint16_as_floatX ); \
- bf = _mm256_min_ps( bb, STBIR_max_uint16_as_floatX ); \
- af = _mm256_max_ps( af, _mm256_setzero_ps() ); \
- bf = _mm256_max_ps( bf, _mm256_setzero_ps() ); \
- a = _mm256_cvttps_epi32( af ); \
- b = _mm256_cvttps_epi32( bf ); \
- t0 = _mm_packus_epi32( _mm256_castsi256_si128(a), _mm256_extractf128_si256( a, 1 ) ); \
- t1 = _mm_packus_epi32( _mm256_castsi256_si128(b), _mm256_extractf128_si256( b, 1 ) ); \
- out = _mm256_setr_m128i( t0, t1 ); \
- }
- #endif
- static __m256i stbir_00001111 = { STBIR__CONST_4d_32i( 0, 0, 0, 0 ), STBIR__CONST_4d_32i( 1, 1, 1, 1 ) };
- #define stbir__simdf8_0123to00001111( out, in ) (out) = _mm256_permutevar_ps ( in, stbir_00001111 )
- static __m256i stbir_22223333 = { STBIR__CONST_4d_32i( 2, 2, 2, 2 ), STBIR__CONST_4d_32i( 3, 3, 3, 3 ) };
- #define stbir__simdf8_0123to22223333( out, in ) (out) = _mm256_permutevar_ps ( in, stbir_22223333 )
- #define stbir__simdf8_0123to2222( out, in ) (out) = stbir__simdf_swiz(_mm256_castps256_ps128(in), 2,2,2,2 )
- #define stbir__simdf8_load4b( out, ptr ) (out) = _mm256_broadcast_ps( (__m128 const *)(ptr) )
- static __m256i stbir_00112233 = { STBIR__CONST_4d_32i( 0, 0, 1, 1 ), STBIR__CONST_4d_32i( 2, 2, 3, 3 ) };
- #define stbir__simdf8_0123to00112233( out, in ) (out) = _mm256_permutevar_ps ( in, stbir_00112233 )
- #define stbir__simdf8_add4( out, a8, b ) (out) = _mm256_add_ps( a8, _mm256_castps128_ps256( b ) )
- static __m256i stbir_load6 = { STBIR__CONST_4_32i( 0x80000000 ), STBIR__CONST_4d_32i( 0x80000000, 0x80000000, 0, 0 ) };
- #define stbir__simdf8_load6z( out, ptr ) (out) = _mm256_maskload_ps( ptr, stbir_load6 )
- #define stbir__simdf8_0123to00000000( out, in ) (out) = _mm256_shuffle_ps ( in, in, (0<<0)+(0<<2)+(0<<4)+(0<<6) )
- #define stbir__simdf8_0123to11111111( out, in ) (out) = _mm256_shuffle_ps ( in, in, (1<<0)+(1<<2)+(1<<4)+(1<<6) )
- #define stbir__simdf8_0123to22222222( out, in ) (out) = _mm256_shuffle_ps ( in, in, (2<<0)+(2<<2)+(2<<4)+(2<<6) )
- #define stbir__simdf8_0123to33333333( out, in ) (out) = _mm256_shuffle_ps ( in, in, (3<<0)+(3<<2)+(3<<4)+(3<<6) )
- #define stbir__simdf8_0123to21032103( out, in ) (out) = _mm256_shuffle_ps ( in, in, (2<<0)+(1<<2)+(0<<4)+(3<<6) )
- #define stbir__simdf8_0123to32103210( out, in ) (out) = _mm256_shuffle_ps ( in, in, (3<<0)+(2<<2)+(1<<4)+(0<<6) )
- #define stbir__simdf8_0123to12301230( out, in ) (out) = _mm256_shuffle_ps ( in, in, (1<<0)+(2<<2)+(3<<4)+(0<<6) )
- #define stbir__simdf8_0123to10321032( out, in ) (out) = _mm256_shuffle_ps ( in, in, (1<<0)+(0<<2)+(3<<4)+(2<<6) )
- #define stbir__simdf8_0123to30123012( out, in ) (out) = _mm256_shuffle_ps ( in, in, (3<<0)+(0<<2)+(1<<4)+(2<<6) )
- #define stbir__simdf8_0123to11331133( out, in ) (out) = _mm256_shuffle_ps ( in, in, (1<<0)+(1<<2)+(3<<4)+(3<<6) )
- #define stbir__simdf8_0123to00220022( out, in ) (out) = _mm256_shuffle_ps ( in, in, (0<<0)+(0<<2)+(2<<4)+(2<<6) )
- #define stbir__simdf8_aaa1( out, alp, ones ) (out) = _mm256_blend_ps( alp, ones, (1<<0)+(1<<1)+(1<<2)+(0<<3)+(1<<4)+(1<<5)+(1<<6)+(0<<7)); (out)=_mm256_shuffle_ps( out,out, (3<<0) + (3<<2) + (3<<4) + (0<<6) )
- #define stbir__simdf8_1aaa( out, alp, ones ) (out) = _mm256_blend_ps( alp, ones, (0<<0)+(1<<1)+(1<<2)+(1<<3)+(0<<4)+(1<<5)+(1<<6)+(1<<7)); (out)=_mm256_shuffle_ps( out,out, (1<<0) + (0<<2) + (0<<4) + (0<<6) )
- #define stbir__simdf8_a1a1( out, alp, ones) (out) = _mm256_blend_ps( alp, ones, (1<<0)+(0<<1)+(1<<2)+(0<<3)+(1<<4)+(0<<5)+(1<<6)+(0<<7)); (out)=_mm256_shuffle_ps( out,out, (1<<0) + (0<<2) + (3<<4) + (2<<6) )
- #define stbir__simdf8_1a1a( out, alp, ones) (out) = _mm256_blend_ps( alp, ones, (0<<0)+(1<<1)+(0<<2)+(1<<3)+(0<<4)+(1<<5)+(0<<6)+(1<<7)); (out)=_mm256_shuffle_ps( out,out, (1<<0) + (0<<2) + (3<<4) + (2<<6) )
- #define stbir__simdf8_zero( reg ) (reg) = _mm256_setzero_ps()
- #ifdef STBIR_USE_FMA // not on by default to maintain bit identical simd to non-simd
- #define stbir__simdf8_madd( out, add, mul1, mul2 ) (out) = _mm256_fmadd_ps( mul1, mul2, add )
- #define stbir__simdf8_madd_mem( out, add, mul, ptr ) (out) = _mm256_fmadd_ps( mul, _mm256_loadu_ps( (float const*)(ptr) ), add )
- #define stbir__simdf8_madd_mem4( out, add, mul, ptr )(out) = _mm256_fmadd_ps( _mm256_setr_m128( mul, _mm_setzero_ps() ), _mm256_setr_m128( _mm_loadu_ps( (float const*)(ptr) ), _mm_setzero_ps() ), add )
- #else
- #define stbir__simdf8_madd( out, add, mul1, mul2 ) (out) = _mm256_add_ps( add, _mm256_mul_ps( mul1, mul2 ) )
- #define stbir__simdf8_madd_mem( out, add, mul, ptr ) (out) = _mm256_add_ps( add, _mm256_mul_ps( mul, _mm256_loadu_ps( (float const*)(ptr) ) ) )
- #define stbir__simdf8_madd_mem4( out, add, mul, ptr ) (out) = _mm256_add_ps( add, _mm256_setr_m128( _mm_mul_ps( mul, _mm_loadu_ps( (float const*)(ptr) ) ), _mm_setzero_ps() ) )
- #endif
- #define stbir__if_simdf8_cast_to_simdf4( val ) _mm256_castps256_ps128( val )
- #endif
- #ifdef STBIR_FLOORF
- #undef STBIR_FLOORF
- #endif
- #define STBIR_FLOORF stbir_simd_floorf
- static stbir__inline float stbir_simd_floorf(float x) // martins floorf
- {
- #if defined(STBIR_AVX) || defined(__SSE4_1__) || defined(STBIR_SSE41)
- __m128 t = _mm_set_ss(x);
- return _mm_cvtss_f32( _mm_floor_ss(t, t) );
- #else
- __m128 f = _mm_set_ss(x);
- __m128 t = _mm_cvtepi32_ps(_mm_cvttps_epi32(f));
- __m128 r = _mm_add_ss(t, _mm_and_ps(_mm_cmplt_ss(f, t), _mm_set_ss(-1.0f)));
- return _mm_cvtss_f32(r);
- #endif
- }
- #ifdef STBIR_CEILF
- #undef STBIR_CEILF
- #endif
- #define STBIR_CEILF stbir_simd_ceilf
- static stbir__inline float stbir_simd_ceilf(float x) // martins ceilf
- {
- #if defined(STBIR_AVX) || defined(__SSE4_1__) || defined(STBIR_SSE41)
- __m128 t = _mm_set_ss(x);
- return _mm_cvtss_f32( _mm_ceil_ss(t, t) );
- #else
- __m128 f = _mm_set_ss(x);
- __m128 t = _mm_cvtepi32_ps(_mm_cvttps_epi32(f));
- __m128 r = _mm_add_ss(t, _mm_and_ps(_mm_cmplt_ss(t, f), _mm_set_ss(1.0f)));
- return _mm_cvtss_f32(r);
- #endif
- }
- #elif defined(STBIR_NEON)
- #include <arm_neon.h>
- #define stbir__simdf float32x4_t
- #define stbir__simdi uint32x4_t
- #define stbir_simdi_castf( reg ) vreinterpretq_u32_f32(reg)
- #define stbir_simdf_casti( reg ) vreinterpretq_f32_u32(reg)
- #define stbir__simdf_load( reg, ptr ) (reg) = vld1q_f32( (float const*)(ptr) )
- #define stbir__simdi_load( reg, ptr ) (reg) = vld1q_u32( (uint32_t const*)(ptr) )
- #define stbir__simdf_load1( out, ptr ) (out) = vld1q_dup_f32( (float const*)(ptr) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdi_load1( out, ptr ) (out) = vld1q_dup_u32( (uint32_t const*)(ptr) )
- #define stbir__simdf_load1z( out, ptr ) (out) = vld1q_lane_f32( (float const*)(ptr), vdupq_n_f32(0), 0 ) // top values must be zero
- #define stbir__simdf_frep4( fvar ) vdupq_n_f32( fvar )
- #define stbir__simdf_load1frep4( out, fvar ) (out) = vdupq_n_f32( fvar )
- #define stbir__simdf_load2( out, ptr ) (out) = vcombine_f32( vld1_f32( (float const*)(ptr) ), vcreate_f32(0) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdf_load2z( out, ptr ) (out) = vcombine_f32( vld1_f32( (float const*)(ptr) ), vcreate_f32(0) ) // top values must be zero
- #define stbir__simdf_load2hmerge( out, reg, ptr ) (out) = vcombine_f32( vget_low_f32(reg), vld1_f32( (float const*)(ptr) ) )
- #define stbir__simdf_zeroP() vdupq_n_f32(0)
- #define stbir__simdf_zero( reg ) (reg) = vdupq_n_f32(0)
- #define stbir__simdf_store( ptr, reg ) vst1q_f32( (float*)(ptr), reg )
- #define stbir__simdf_store1( ptr, reg ) vst1q_lane_f32( (float*)(ptr), reg, 0)
- #define stbir__simdf_store2( ptr, reg ) vst1_f32( (float*)(ptr), vget_low_f32(reg) )
- #define stbir__simdf_store2h( ptr, reg ) vst1_f32( (float*)(ptr), vget_high_f32(reg) )
- #define stbir__simdi_store( ptr, reg ) vst1q_u32( (uint32_t*)(ptr), reg )
- #define stbir__simdi_store1( ptr, reg ) vst1q_lane_u32( (uint32_t*)(ptr), reg, 0 )
- #define stbir__simdi_store2( ptr, reg ) vst1_u32( (uint32_t*)(ptr), vget_low_u32(reg) )
- #define stbir__prefetch( ptr )
- #define stbir__simdi_expand_u8_to_u32(out0,out1,out2,out3,ireg) \
- { \
- uint16x8_t l = vmovl_u8( vget_low_u8 ( vreinterpretq_u8_u32(ireg) ) ); \
- uint16x8_t h = vmovl_u8( vget_high_u8( vreinterpretq_u8_u32(ireg) ) ); \
- out0 = vmovl_u16( vget_low_u16 ( l ) ); \
- out1 = vmovl_u16( vget_high_u16( l ) ); \
- out2 = vmovl_u16( vget_low_u16 ( h ) ); \
- out3 = vmovl_u16( vget_high_u16( h ) ); \
- }
- #define stbir__simdi_expand_u8_to_1u32(out,ireg) \
- { \
- uint16x8_t tmp = vmovl_u8( vget_low_u8( vreinterpretq_u8_u32(ireg) ) ); \
- out = vmovl_u16( vget_low_u16( tmp ) ); \
- }
- #define stbir__simdi_expand_u16_to_u32(out0,out1,ireg) \
- { \
- uint16x8_t tmp = vreinterpretq_u16_u32(ireg); \
- out0 = vmovl_u16( vget_low_u16 ( tmp ) ); \
- out1 = vmovl_u16( vget_high_u16( tmp ) ); \
- }
- #define stbir__simdf_convert_float_to_i32( i, f ) (i) = vreinterpretq_u32_s32( vcvtq_s32_f32(f) )
- #define stbir__simdf_convert_float_to_int( f ) vgetq_lane_s32(vcvtq_s32_f32(f), 0)
- #define stbir__simdi_to_int( i ) (int)vgetq_lane_u32(i, 0)
- #define stbir__simdf_convert_float_to_uint8( f ) ((unsigned char)vgetq_lane_s32(vcvtq_s32_f32(vmaxq_f32(vminq_f32(f,STBIR__CONSTF(STBIR_max_uint8_as_float)),vdupq_n_f32(0))), 0))
- #define stbir__simdf_convert_float_to_short( f ) ((unsigned short)vgetq_lane_s32(vcvtq_s32_f32(vmaxq_f32(vminq_f32(f,STBIR__CONSTF(STBIR_max_uint16_as_float)),vdupq_n_f32(0))), 0))
- #define stbir__simdi_convert_i32_to_float(out, ireg) (out) = vcvtq_f32_s32( vreinterpretq_s32_u32(ireg) )
- #define stbir__simdf_add( out, reg0, reg1 ) (out) = vaddq_f32( reg0, reg1 )
- #define stbir__simdf_mult( out, reg0, reg1 ) (out) = vmulq_f32( reg0, reg1 )
- #define stbir__simdf_mult_mem( out, reg, ptr ) (out) = vmulq_f32( reg, vld1q_f32( (float const*)(ptr) ) )
- #define stbir__simdf_mult1_mem( out, reg, ptr ) (out) = vmulq_f32( reg, vld1q_dup_f32( (float const*)(ptr) ) )
- #define stbir__simdf_add_mem( out, reg, ptr ) (out) = vaddq_f32( reg, vld1q_f32( (float const*)(ptr) ) )
- #define stbir__simdf_add1_mem( out, reg, ptr ) (out) = vaddq_f32( reg, vld1q_dup_f32( (float const*)(ptr) ) )
- #ifdef STBIR_USE_FMA // not on by default to maintain bit identical simd to non-simd (and also x64 no madd to arm madd)
- #define stbir__simdf_madd( out, add, mul1, mul2 ) (out) = vfmaq_f32( add, mul1, mul2 )
- #define stbir__simdf_madd1( out, add, mul1, mul2 ) (out) = vfmaq_f32( add, mul1, mul2 )
- #define stbir__simdf_madd_mem( out, add, mul, ptr ) (out) = vfmaq_f32( add, mul, vld1q_f32( (float const*)(ptr) ) )
- #define stbir__simdf_madd1_mem( out, add, mul, ptr ) (out) = vfmaq_f32( add, mul, vld1q_dup_f32( (float const*)(ptr) ) )
- #else
- #define stbir__simdf_madd( out, add, mul1, mul2 ) (out) = vaddq_f32( add, vmulq_f32( mul1, mul2 ) )
- #define stbir__simdf_madd1( out, add, mul1, mul2 ) (out) = vaddq_f32( add, vmulq_f32( mul1, mul2 ) )
- #define stbir__simdf_madd_mem( out, add, mul, ptr ) (out) = vaddq_f32( add, vmulq_f32( mul, vld1q_f32( (float const*)(ptr) ) ) )
- #define stbir__simdf_madd1_mem( out, add, mul, ptr ) (out) = vaddq_f32( add, vmulq_f32( mul, vld1q_dup_f32( (float const*)(ptr) ) ) )
- #endif
- #define stbir__simdf_add1( out, reg0, reg1 ) (out) = vaddq_f32( reg0, reg1 )
- #define stbir__simdf_mult1( out, reg0, reg1 ) (out) = vmulq_f32( reg0, reg1 )
- #define stbir__simdf_and( out, reg0, reg1 ) (out) = vreinterpretq_f32_u32( vandq_u32( vreinterpretq_u32_f32(reg0), vreinterpretq_u32_f32(reg1) ) )
- #define stbir__simdf_or( out, reg0, reg1 ) (out) = vreinterpretq_f32_u32( vorrq_u32( vreinterpretq_u32_f32(reg0), vreinterpretq_u32_f32(reg1) ) )
- #define stbir__simdf_min( out, reg0, reg1 ) (out) = vminq_f32( reg0, reg1 )
- #define stbir__simdf_max( out, reg0, reg1 ) (out) = vmaxq_f32( reg0, reg1 )
- #define stbir__simdf_min1( out, reg0, reg1 ) (out) = vminq_f32( reg0, reg1 )
- #define stbir__simdf_max1( out, reg0, reg1 ) (out) = vmaxq_f32( reg0, reg1 )
- #define stbir__simdf_0123ABCDto3ABx( out, reg0, reg1 ) (out) = vextq_f32( reg0, reg1, 3 )
- #define stbir__simdf_0123ABCDto23Ax( out, reg0, reg1 ) (out) = vextq_f32( reg0, reg1, 2 )
- #define stbir__simdf_a1a1( out, alp, ones ) (out) = vzipq_f32(vuzpq_f32(alp, alp).val[1], ones).val[0]
- #define stbir__simdf_1a1a( out, alp, ones ) (out) = vzipq_f32(ones, vuzpq_f32(alp, alp).val[0]).val[0]
- #if defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ )
- #define stbir__simdf_aaa1( out, alp, ones ) (out) = vcopyq_laneq_f32(vdupq_n_f32(vgetq_lane_f32(alp, 3)), 3, ones, 3)
- #define stbir__simdf_1aaa( out, alp, ones ) (out) = vcopyq_laneq_f32(vdupq_n_f32(vgetq_lane_f32(alp, 0)), 0, ones, 0)
- #if defined( _MSC_VER ) && !defined(__clang__)
- #define stbir_make16(a,b,c,d) vcombine_u8( \
- vcreate_u8( (4*a+0) | ((4*a+1)<<8) | ((4*a+2)<<16) | ((4*a+3)<<24) | \
- ((stbir_uint64)(4*b+0)<<32) | ((stbir_uint64)(4*b+1)<<40) | ((stbir_uint64)(4*b+2)<<48) | ((stbir_uint64)(4*b+3)<<56)), \
- vcreate_u8( (4*c+0) | ((4*c+1)<<8) | ((4*c+2)<<16) | ((4*c+3)<<24) | \
- ((stbir_uint64)(4*d+0)<<32) | ((stbir_uint64)(4*d+1)<<40) | ((stbir_uint64)(4*d+2)<<48) | ((stbir_uint64)(4*d+3)<<56) ) )
- static stbir__inline uint8x16x2_t stbir_make16x2(float32x4_t rega,float32x4_t regb)
- {
- uint8x16x2_t r = { vreinterpretq_u8_f32(rega), vreinterpretq_u8_f32(regb) };
- return r;
- }
- #else
- #define stbir_make16(a,b,c,d) (uint8x16_t){4*a+0,4*a+1,4*a+2,4*a+3,4*b+0,4*b+1,4*b+2,4*b+3,4*c+0,4*c+1,4*c+2,4*c+3,4*d+0,4*d+1,4*d+2,4*d+3}
- #define stbir_make16x2(a,b) (uint8x16x2_t){{vreinterpretq_u8_f32(a),vreinterpretq_u8_f32(b)}}
- #endif
- #define stbir__simdf_swiz( reg, one, two, three, four ) vreinterpretq_f32_u8( vqtbl1q_u8( vreinterpretq_u8_f32(reg), stbir_make16(one, two, three, four) ) )
- #define stbir__simdf_swiz2( rega, regb, one, two, three, four ) vreinterpretq_f32_u8( vqtbl2q_u8( stbir_make16x2(rega,regb), stbir_make16(one, two, three, four) ) )
- #define stbir__simdi_16madd( out, reg0, reg1 ) \
- { \
- int16x8_t r0 = vreinterpretq_s16_u32(reg0); \
- int16x8_t r1 = vreinterpretq_s16_u32(reg1); \
- int32x4_t tmp0 = vmull_s16( vget_low_s16(r0), vget_low_s16(r1) ); \
- int32x4_t tmp1 = vmull_s16( vget_high_s16(r0), vget_high_s16(r1) ); \
- (out) = vreinterpretq_u32_s32( vpaddq_s32(tmp0, tmp1) ); \
- }
- #else
- #define stbir__simdf_aaa1( out, alp, ones ) (out) = vsetq_lane_f32(1.0f, vdupq_n_f32(vgetq_lane_f32(alp, 3)), 3)
- #define stbir__simdf_1aaa( out, alp, ones ) (out) = vsetq_lane_f32(1.0f, vdupq_n_f32(vgetq_lane_f32(alp, 0)), 0)
- #if defined( _MSC_VER ) && !defined(__clang__)
- static stbir__inline uint8x8x2_t stbir_make8x2(float32x4_t reg)
- {
- uint8x8x2_t r = { { vget_low_u8(vreinterpretq_u8_f32(reg)), vget_high_u8(vreinterpretq_u8_f32(reg)) } };
- return r;
- }
- #define stbir_make8(a,b) vcreate_u8( \
- (4*a+0) | ((4*a+1)<<8) | ((4*a+2)<<16) | ((4*a+3)<<24) | \
- ((stbir_uint64)(4*b+0)<<32) | ((stbir_uint64)(4*b+1)<<40) | ((stbir_uint64)(4*b+2)<<48) | ((stbir_uint64)(4*b+3)<<56) )
- #else
- #define stbir_make8x2(reg) (uint8x8x2_t){ { vget_low_u8(vreinterpretq_u8_f32(reg)), vget_high_u8(vreinterpretq_u8_f32(reg)) } }
- #define stbir_make8(a,b) (uint8x8_t){4*a+0,4*a+1,4*a+2,4*a+3,4*b+0,4*b+1,4*b+2,4*b+3}
- #endif
- #define stbir__simdf_swiz( reg, one, two, three, four ) vreinterpretq_f32_u8( vcombine_u8( \
- vtbl2_u8( stbir_make8x2( reg ), stbir_make8( one, two ) ), \
- vtbl2_u8( stbir_make8x2( reg ), stbir_make8( three, four ) ) ) )
- #define stbir__simdi_16madd( out, reg0, reg1 ) \
- { \
- int16x8_t r0 = vreinterpretq_s16_u32(reg0); \
- int16x8_t r1 = vreinterpretq_s16_u32(reg1); \
- int32x4_t tmp0 = vmull_s16( vget_low_s16(r0), vget_low_s16(r1) ); \
- int32x4_t tmp1 = vmull_s16( vget_high_s16(r0), vget_high_s16(r1) ); \
- int32x2_t out0 = vpadd_s32( vget_low_s32(tmp0), vget_high_s32(tmp0) ); \
- int32x2_t out1 = vpadd_s32( vget_low_s32(tmp1), vget_high_s32(tmp1) ); \
- (out) = vreinterpretq_u32_s32( vcombine_s32(out0, out1) ); \
- }
- #endif
- #define stbir__simdi_and( out, reg0, reg1 ) (out) = vandq_u32( reg0, reg1 )
- #define stbir__simdi_or( out, reg0, reg1 ) (out) = vorrq_u32( reg0, reg1 )
- #define stbir__simdf_pack_to_8bytes(out,aa,bb) \
- { \
- float32x4_t af = vmaxq_f32( vminq_f32(aa,STBIR__CONSTF(STBIR_max_uint8_as_float) ), vdupq_n_f32(0) ); \
- float32x4_t bf = vmaxq_f32( vminq_f32(bb,STBIR__CONSTF(STBIR_max_uint8_as_float) ), vdupq_n_f32(0) ); \
- int16x4_t ai = vqmovn_s32( vcvtq_s32_f32( af ) ); \
- int16x4_t bi = vqmovn_s32( vcvtq_s32_f32( bf ) ); \
- uint8x8_t out8 = vqmovun_s16( vcombine_s16(ai, bi) ); \
- out = vreinterpretq_u32_u8( vcombine_u8(out8, out8) ); \
- }
- #define stbir__simdf_pack_to_8words(out,aa,bb) \
- { \
- float32x4_t af = vmaxq_f32( vminq_f32(aa,STBIR__CONSTF(STBIR_max_uint16_as_float) ), vdupq_n_f32(0) ); \
- float32x4_t bf = vmaxq_f32( vminq_f32(bb,STBIR__CONSTF(STBIR_max_uint16_as_float) ), vdupq_n_f32(0) ); \
- int32x4_t ai = vcvtq_s32_f32( af ); \
- int32x4_t bi = vcvtq_s32_f32( bf ); \
- out = vreinterpretq_u32_u16( vcombine_u16(vqmovun_s32(ai), vqmovun_s32(bi)) ); \
- }
- #define stbir__interleave_pack_and_store_16_u8( ptr, r0, r1, r2, r3 ) \
- { \
- int16x4x2_t tmp0 = vzip_s16( vqmovn_s32(vreinterpretq_s32_u32(r0)), vqmovn_s32(vreinterpretq_s32_u32(r2)) ); \
- int16x4x2_t tmp1 = vzip_s16( vqmovn_s32(vreinterpretq_s32_u32(r1)), vqmovn_s32(vreinterpretq_s32_u32(r3)) ); \
- uint8x8x2_t out = \
- { { \
- vqmovun_s16( vcombine_s16(tmp0.val[0], tmp0.val[1]) ), \
- vqmovun_s16( vcombine_s16(tmp1.val[0], tmp1.val[1]) ), \
- } }; \
- vst2_u8(ptr, out); \
- }
- #define stbir__simdf_load4_transposed( o0, o1, o2, o3, ptr ) \
- { \
- float32x4x4_t tmp = vld4q_f32(ptr); \
- o0 = tmp.val[0]; \
- o1 = tmp.val[1]; \
- o2 = tmp.val[2]; \
- o3 = tmp.val[3]; \
- }
- #define stbir__simdi_32shr( out, reg, imm ) out = vshrq_n_u32( reg, imm )
- #if defined( _MSC_VER ) && !defined(__clang__)
- #define STBIR__SIMDF_CONST(var, x) __declspec(align(8)) float var[] = { x, x, x, x }
- #define STBIR__SIMDI_CONST(var, x) __declspec(align(8)) uint32_t var[] = { x, x, x, x }
- #define STBIR__CONSTF(var) (*(const float32x4_t*)var)
- #define STBIR__CONSTI(var) (*(const uint32x4_t*)var)
- #else
- #define STBIR__SIMDF_CONST(var, x) stbir__simdf var = { x, x, x, x }
- #define STBIR__SIMDI_CONST(var, x) stbir__simdi var = { x, x, x, x }
- #define STBIR__CONSTF(var) (var)
- #define STBIR__CONSTI(var) (var)
- #endif
- #ifdef STBIR_FLOORF
- #undef STBIR_FLOORF
- #endif
- #define STBIR_FLOORF stbir_simd_floorf
- static stbir__inline float stbir_simd_floorf(float x)
- {
- #if defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ )
- return vget_lane_f32( vrndm_f32( vdup_n_f32(x) ), 0);
- #else
- float32x2_t f = vdup_n_f32(x);
- float32x2_t t = vcvt_f32_s32(vcvt_s32_f32(f));
- uint32x2_t a = vclt_f32(f, t);
- uint32x2_t b = vreinterpret_u32_f32(vdup_n_f32(-1.0f));
- float32x2_t r = vadd_f32(t, vreinterpret_f32_u32(vand_u32(a, b)));
- return vget_lane_f32(r, 0);
- #endif
- }
- #ifdef STBIR_CEILF
- #undef STBIR_CEILF
- #endif
- #define STBIR_CEILF stbir_simd_ceilf
- static stbir__inline float stbir_simd_ceilf(float x)
- {
- #if defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ )
- return vget_lane_f32( vrndp_f32( vdup_n_f32(x) ), 0);
- #else
- float32x2_t f = vdup_n_f32(x);
- float32x2_t t = vcvt_f32_s32(vcvt_s32_f32(f));
- uint32x2_t a = vclt_f32(t, f);
- uint32x2_t b = vreinterpret_u32_f32(vdup_n_f32(1.0f));
- float32x2_t r = vadd_f32(t, vreinterpret_f32_u32(vand_u32(a, b)));
- return vget_lane_f32(r, 0);
- #endif
- }
- #define STBIR_SIMD
- #elif defined(STBIR_WASM)
- #include <wasm_simd128.h>
- #define stbir__simdf v128_t
- #define stbir__simdi v128_t
- #define stbir_simdi_castf( reg ) (reg)
- #define stbir_simdf_casti( reg ) (reg)
- #define stbir__simdf_load( reg, ptr ) (reg) = wasm_v128_load( (void const*)(ptr) )
- #define stbir__simdi_load( reg, ptr ) (reg) = wasm_v128_load( (void const*)(ptr) )
- #define stbir__simdf_load1( out, ptr ) (out) = wasm_v128_load32_splat( (void const*)(ptr) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdi_load1( out, ptr ) (out) = wasm_v128_load32_splat( (void const*)(ptr) )
- #define stbir__simdf_load1z( out, ptr ) (out) = wasm_v128_load32_zero( (void const*)(ptr) ) // top values must be zero
- #define stbir__simdf_frep4( fvar ) wasm_f32x4_splat( fvar )
- #define stbir__simdf_load1frep4( out, fvar ) (out) = wasm_f32x4_splat( fvar )
- #define stbir__simdf_load2( out, ptr ) (out) = wasm_v128_load64_splat( (void const*)(ptr) ) // top values can be random (not denormal or nan for perf)
- #define stbir__simdf_load2z( out, ptr ) (out) = wasm_v128_load64_zero( (void const*)(ptr) ) // top values must be zero
- #define stbir__simdf_load2hmerge( out, reg, ptr ) (out) = wasm_v128_load64_lane( (void const*)(ptr), reg, 1 )
- #define stbir__simdf_zeroP() wasm_f32x4_const_splat(0)
- #define stbir__simdf_zero( reg ) (reg) = wasm_f32x4_const_splat(0)
- #define stbir__simdf_store( ptr, reg ) wasm_v128_store( (void*)(ptr), reg )
- #define stbir__simdf_store1( ptr, reg ) wasm_v128_store32_lane( (void*)(ptr), reg, 0 )
- #define stbir__simdf_store2( ptr, reg ) wasm_v128_store64_lane( (void*)(ptr), reg, 0 )
- #define stbir__simdf_store2h( ptr, reg ) wasm_v128_store64_lane( (void*)(ptr), reg, 1 )
- #define stbir__simdi_store( ptr, reg ) wasm_v128_store( (void*)(ptr), reg )
- #define stbir__simdi_store1( ptr, reg ) wasm_v128_store32_lane( (void*)(ptr), reg, 0 )
- #define stbir__simdi_store2( ptr, reg ) wasm_v128_store64_lane( (void*)(ptr), reg, 0 )
- #define stbir__prefetch( ptr )
- #define stbir__simdi_expand_u8_to_u32(out0,out1,out2,out3,ireg) \
- { \
- v128_t l = wasm_u16x8_extend_low_u8x16 ( ireg ); \
- v128_t h = wasm_u16x8_extend_high_u8x16( ireg ); \
- out0 = wasm_u32x4_extend_low_u16x8 ( l ); \
- out1 = wasm_u32x4_extend_high_u16x8( l ); \
- out2 = wasm_u32x4_extend_low_u16x8 ( h ); \
- out3 = wasm_u32x4_extend_high_u16x8( h ); \
- }
- #define stbir__simdi_expand_u8_to_1u32(out,ireg) \
- { \
- v128_t tmp = wasm_u16x8_extend_low_u8x16(ireg); \
- out = wasm_u32x4_extend_low_u16x8(tmp); \
- }
- #define stbir__simdi_expand_u16_to_u32(out0,out1,ireg) \
- { \
- out0 = wasm_u32x4_extend_low_u16x8 ( ireg ); \
- out1 = wasm_u32x4_extend_high_u16x8( ireg ); \
- }
- #define stbir__simdf_convert_float_to_i32( i, f ) (i) = wasm_i32x4_trunc_sat_f32x4(f)
- #define stbir__simdf_convert_float_to_int( f ) wasm_i32x4_extract_lane(wasm_i32x4_trunc_sat_f32x4(f), 0)
- #define stbir__simdi_to_int( i ) wasm_i32x4_extract_lane(i, 0)
- #define stbir__simdf_convert_float_to_uint8( f ) ((unsigned char)wasm_i32x4_extract_lane(wasm_i32x4_trunc_sat_f32x4(wasm_f32x4_max(wasm_f32x4_min(f,STBIR_max_uint8_as_float),wasm_f32x4_const_splat(0))), 0))
- #define stbir__simdf_convert_float_to_short( f ) ((unsigned short)wasm_i32x4_extract_lane(wasm_i32x4_trunc_sat_f32x4(wasm_f32x4_max(wasm_f32x4_min(f,STBIR_max_uint16_as_float),wasm_f32x4_const_splat(0))), 0))
- #define stbir__simdi_convert_i32_to_float(out, ireg) (out) = wasm_f32x4_convert_i32x4(ireg)
- #define stbir__simdf_add( out, reg0, reg1 ) (out) = wasm_f32x4_add( reg0, reg1 )
- #define stbir__simdf_mult( out, reg0, reg1 ) (out) = wasm_f32x4_mul( reg0, reg1 )
- #define stbir__simdf_mult_mem( out, reg, ptr ) (out) = wasm_f32x4_mul( reg, wasm_v128_load( (void const*)(ptr) ) )
- #define stbir__simdf_mult1_mem( out, reg, ptr ) (out) = wasm_f32x4_mul( reg, wasm_v128_load32_splat( (void const*)(ptr) ) )
- #define stbir__simdf_add_mem( out, reg, ptr ) (out) = wasm_f32x4_add( reg, wasm_v128_load( (void const*)(ptr) ) )
- #define stbir__simdf_add1_mem( out, reg, ptr ) (out) = wasm_f32x4_add( reg, wasm_v128_load32_splat( (void const*)(ptr) ) )
- #define stbir__simdf_madd( out, add, mul1, mul2 ) (out) = wasm_f32x4_add( add, wasm_f32x4_mul( mul1, mul2 ) )
- #define stbir__simdf_madd1( out, add, mul1, mul2 ) (out) = wasm_f32x4_add( add, wasm_f32x4_mul( mul1, mul2 ) )
- #define stbir__simdf_madd_mem( out, add, mul, ptr ) (out) = wasm_f32x4_add( add, wasm_f32x4_mul( mul, wasm_v128_load( (void const*)(ptr) ) ) )
- #define stbir__simdf_madd1_mem( out, add, mul, ptr ) (out) = wasm_f32x4_add( add, wasm_f32x4_mul( mul, wasm_v128_load32_splat( (void const*)(ptr) ) ) )
- #define stbir__simdf_add1( out, reg0, reg1 ) (out) = wasm_f32x4_add( reg0, reg1 )
- #define stbir__simdf_mult1( out, reg0, reg1 ) (out) = wasm_f32x4_mul( reg0, reg1 )
- #define stbir__simdf_and( out, reg0, reg1 ) (out) = wasm_v128_and( reg0, reg1 )
- #define stbir__simdf_or( out, reg0, reg1 ) (out) = wasm_v128_or( reg0, reg1 )
- #define stbir__simdf_min( out, reg0, reg1 ) (out) = wasm_f32x4_min( reg0, reg1 )
- #define stbir__simdf_max( out, reg0, reg1 ) (out) = wasm_f32x4_max( reg0, reg1 )
- #define stbir__simdf_min1( out, reg0, reg1 ) (out) = wasm_f32x4_min( reg0, reg1 )
- #define stbir__simdf_max1( out, reg0, reg1 ) (out) = wasm_f32x4_max( reg0, reg1 )
- #define stbir__simdf_0123ABCDto3ABx( out, reg0, reg1 ) (out) = wasm_i32x4_shuffle( reg0, reg1, 3, 4, 5, -1 )
- #define stbir__simdf_0123ABCDto23Ax( out, reg0, reg1 ) (out) = wasm_i32x4_shuffle( reg0, reg1, 2, 3, 4, -1 )
- #define stbir__simdf_aaa1(out,alp,ones) (out) = wasm_i32x4_shuffle(alp, ones, 3, 3, 3, 4)
- #define stbir__simdf_1aaa(out,alp,ones) (out) = wasm_i32x4_shuffle(alp, ones, 4, 0, 0, 0)
- #define stbir__simdf_a1a1(out,alp,ones) (out) = wasm_i32x4_shuffle(alp, ones, 1, 4, 3, 4)
- #define stbir__simdf_1a1a(out,alp,ones) (out) = wasm_i32x4_shuffle(alp, ones, 4, 0, 4, 2)
- #define stbir__simdf_swiz( reg, one, two, three, four ) wasm_i32x4_shuffle(reg, reg, one, two, three, four)
- #define stbir__simdi_and( out, reg0, reg1 ) (out) = wasm_v128_and( reg0, reg1 )
- #define stbir__simdi_or( out, reg0, reg1 ) (out) = wasm_v128_or( reg0, reg1 )
- #define stbir__simdi_16madd( out, reg0, reg1 ) (out) = wasm_i32x4_dot_i16x8( reg0, reg1 )
- #define stbir__simdf_pack_to_8bytes(out,aa,bb) \
- { \
- v128_t af = wasm_f32x4_max( wasm_f32x4_min(aa, STBIR_max_uint8_as_float), wasm_f32x4_const_splat(0) ); \
- v128_t bf = wasm_f32x4_max( wasm_f32x4_min(bb, STBIR_max_uint8_as_float), wasm_f32x4_const_splat(0) ); \
- v128_t ai = wasm_i32x4_trunc_sat_f32x4( af ); \
- v128_t bi = wasm_i32x4_trunc_sat_f32x4( bf ); \
- v128_t out16 = wasm_i16x8_narrow_i32x4( ai, bi ); \
- out = wasm_u8x16_narrow_i16x8( out16, out16 ); \
- }
- #define stbir__simdf_pack_to_8words(out,aa,bb) \
- { \
- v128_t af = wasm_f32x4_max( wasm_f32x4_min(aa, STBIR_max_uint16_as_float), wasm_f32x4_const_splat(0)); \
- v128_t bf = wasm_f32x4_max( wasm_f32x4_min(bb, STBIR_max_uint16_as_float), wasm_f32x4_const_splat(0)); \
- v128_t ai = wasm_i32x4_trunc_sat_f32x4( af ); \
- v128_t bi = wasm_i32x4_trunc_sat_f32x4( bf ); \
- out = wasm_u16x8_narrow_i32x4( ai, bi ); \
- }
- #define stbir__interleave_pack_and_store_16_u8( ptr, r0, r1, r2, r3 ) \
- { \
- v128_t tmp0 = wasm_i16x8_narrow_i32x4(r0, r1); \
- v128_t tmp1 = wasm_i16x8_narrow_i32x4(r2, r3); \
- v128_t tmp = wasm_u8x16_narrow_i16x8(tmp0, tmp1); \
- tmp = wasm_i8x16_shuffle(tmp, tmp, 0, 4, 8, 12, 1, 5, 9, 13, 2, 6, 10, 14, 3, 7, 11, 15); \
- wasm_v128_store( (void*)(ptr), tmp); \
- }
- #define stbir__simdf_load4_transposed( o0, o1, o2, o3, ptr ) \
- { \
- v128_t t0 = wasm_v128_load( ptr ); \
- v128_t t1 = wasm_v128_load( ptr+4 ); \
- v128_t t2 = wasm_v128_load( ptr+8 ); \
- v128_t t3 = wasm_v128_load( ptr+12 ); \
- v128_t s0 = wasm_i32x4_shuffle(t0, t1, 0, 4, 2, 6); \
- v128_t s1 = wasm_i32x4_shuffle(t0, t1, 1, 5, 3, 7); \
- v128_t s2 = wasm_i32x4_shuffle(t2, t3, 0, 4, 2, 6); \
- v128_t s3 = wasm_i32x4_shuffle(t2, t3, 1, 5, 3, 7); \
- o0 = wasm_i32x4_shuffle(s0, s2, 0, 1, 4, 5); \
- o1 = wasm_i32x4_shuffle(s1, s3, 0, 1, 4, 5); \
- o2 = wasm_i32x4_shuffle(s0, s2, 2, 3, 6, 7); \
- o3 = wasm_i32x4_shuffle(s1, s3, 2, 3, 6, 7); \
- }
- #define stbir__simdi_32shr( out, reg, imm ) out = wasm_u32x4_shr( reg, imm )
- typedef float stbir__f32x4 __attribute__((__vector_size__(16), __aligned__(16)));
- #define STBIR__SIMDF_CONST(var, x) stbir__simdf var = (v128_t)(stbir__f32x4){ x, x, x, x }
- #define STBIR__SIMDI_CONST(var, x) stbir__simdi var = { x, x, x, x }
- #define STBIR__CONSTF(var) (var)
- #define STBIR__CONSTI(var) (var)
- #ifdef STBIR_FLOORF
- #undef STBIR_FLOORF
- #endif
- #define STBIR_FLOORF stbir_simd_floorf
- static stbir__inline float stbir_simd_floorf(float x)
- {
- return wasm_f32x4_extract_lane( wasm_f32x4_floor( wasm_f32x4_splat(x) ), 0);
- }
- #ifdef STBIR_CEILF
- #undef STBIR_CEILF
- #endif
- #define STBIR_CEILF stbir_simd_ceilf
- static stbir__inline float stbir_simd_ceilf(float x)
- {
- return wasm_f32x4_extract_lane( wasm_f32x4_ceil( wasm_f32x4_splat(x) ), 0);
- }
- #define STBIR_SIMD
- #endif // SSE2/NEON/WASM
- #endif // NO SIMD
- #ifdef STBIR_SIMD8
- #define stbir__simdfX stbir__simdf8
- #define stbir__simdiX stbir__simdi8
- #define stbir__simdfX_load stbir__simdf8_load
- #define stbir__simdiX_load stbir__simdi8_load
- #define stbir__simdfX_mult stbir__simdf8_mult
- #define stbir__simdfX_add_mem stbir__simdf8_add_mem
- #define stbir__simdfX_madd_mem stbir__simdf8_madd_mem
- #define stbir__simdfX_store stbir__simdf8_store
- #define stbir__simdiX_store stbir__simdi8_store
- #define stbir__simdf_frepX stbir__simdf8_frep8
- #define stbir__simdfX_madd stbir__simdf8_madd
- #define stbir__simdfX_min stbir__simdf8_min
- #define stbir__simdfX_max stbir__simdf8_max
- #define stbir__simdfX_aaa1 stbir__simdf8_aaa1
- #define stbir__simdfX_1aaa stbir__simdf8_1aaa
- #define stbir__simdfX_a1a1 stbir__simdf8_a1a1
- #define stbir__simdfX_1a1a stbir__simdf8_1a1a
- #define stbir__simdfX_convert_float_to_i32 stbir__simdf8_convert_float_to_i32
- #define stbir__simdfX_pack_to_words stbir__simdf8_pack_to_16words
- #define stbir__simdfX_zero stbir__simdf8_zero
- #define STBIR_onesX STBIR_ones8
- #define STBIR_max_uint8_as_floatX STBIR_max_uint8_as_float8
- #define STBIR_max_uint16_as_floatX STBIR_max_uint16_as_float8
- #define STBIR_simd_point5X STBIR_simd_point58
- #define stbir__simdfX_float_count 8
- #define stbir__simdfX_0123to1230 stbir__simdf8_0123to12301230
- #define stbir__simdfX_0123to2103 stbir__simdf8_0123to21032103
- static const stbir__simdf8 STBIR_max_uint16_as_float_inverted8 = { stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted,stbir__max_uint16_as_float_inverted };
- static const stbir__simdf8 STBIR_max_uint8_as_float_inverted8 = { stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted,stbir__max_uint8_as_float_inverted };
- static const stbir__simdf8 STBIR_ones8 = { 1.0,1.0,1.0,1.0,1.0,1.0,1.0,1.0 };
- static const stbir__simdf8 STBIR_simd_point58 = { 0.5,0.5,0.5,0.5,0.5,0.5,0.5,0.5 };
- static const stbir__simdf8 STBIR_max_uint8_as_float8 = { stbir__max_uint8_as_float,stbir__max_uint8_as_float,stbir__max_uint8_as_float,stbir__max_uint8_as_float, stbir__max_uint8_as_float,stbir__max_uint8_as_float,stbir__max_uint8_as_float,stbir__max_uint8_as_float };
- static const stbir__simdf8 STBIR_max_uint16_as_float8 = { stbir__max_uint16_as_float,stbir__max_uint16_as_float,stbir__max_uint16_as_float,stbir__max_uint16_as_float, stbir__max_uint16_as_float,stbir__max_uint16_as_float,stbir__max_uint16_as_float,stbir__max_uint16_as_float };
- #else
- #define stbir__simdfX stbir__simdf
- #define stbir__simdiX stbir__simdi
- #define stbir__simdfX_load stbir__simdf_load
- #define stbir__simdiX_load stbir__simdi_load
- #define stbir__simdfX_mult stbir__simdf_mult
- #define stbir__simdfX_add_mem stbir__simdf_add_mem
- #define stbir__simdfX_madd_mem stbir__simdf_madd_mem
- #define stbir__simdfX_store stbir__simdf_store
- #define stbir__simdiX_store stbir__simdi_store
- #define stbir__simdf_frepX stbir__simdf_frep4
- #define stbir__simdfX_madd stbir__simdf_madd
- #define stbir__simdfX_min stbir__simdf_min
- #define stbir__simdfX_max stbir__simdf_max
- #define stbir__simdfX_aaa1 stbir__simdf_aaa1
- #define stbir__simdfX_1aaa stbir__simdf_1aaa
- #define stbir__simdfX_a1a1 stbir__simdf_a1a1
- #define stbir__simdfX_1a1a stbir__simdf_1a1a
- #define stbir__simdfX_convert_float_to_i32 stbir__simdf_convert_float_to_i32
- #define stbir__simdfX_pack_to_words stbir__simdf_pack_to_8words
- #define stbir__simdfX_zero stbir__simdf_zero
- #define STBIR_onesX STBIR__CONSTF(STBIR_ones)
- #define STBIR_simd_point5X STBIR__CONSTF(STBIR_simd_point5)
- #define STBIR_max_uint8_as_floatX STBIR__CONSTF(STBIR_max_uint8_as_float)
- #define STBIR_max_uint16_as_floatX STBIR__CONSTF(STBIR_max_uint16_as_float)
- #define stbir__simdfX_float_count 4
- #define stbir__if_simdf8_cast_to_simdf4( val ) ( val )
- #define stbir__simdfX_0123to1230 stbir__simdf_0123to1230
- #define stbir__simdfX_0123to2103 stbir__simdf_0123to2103
- #endif
- #if defined(STBIR_NEON) && !defined(_M_ARM) && !defined(__arm__)
- #if defined( _MSC_VER ) && !defined(__clang__)
- typedef __int16 stbir__FP16;
- #else
- typedef float16_t stbir__FP16;
- #endif
- #else // no NEON, or 32-bit ARM for MSVC
- typedef union stbir__FP16
- {
- unsigned short u;
- } stbir__FP16;
- #endif
- #if (!defined(STBIR_NEON) && !defined(STBIR_FP16C)) || (defined(STBIR_NEON) && defined(_M_ARM)) || (defined(STBIR_NEON) && defined(__arm__))
- // Fabian's half float routines, see: https://gist.github.com/rygorous/2156668
- static stbir__inline float stbir__half_to_float( stbir__FP16 h )
- {
- static const stbir__FP32 magic = { (254 - 15) << 23 };
- static const stbir__FP32 was_infnan = { (127 + 16) << 23 };
- stbir__FP32 o;
- o.u = (h.u & 0x7fff) << 13; // exponent/mantissa bits
- o.f *= magic.f; // exponent adjust
- if (o.f >= was_infnan.f) // make sure Inf/NaN survive
- o.u |= 255 << 23;
- o.u |= (h.u & 0x8000) << 16; // sign bit
- return o.f;
- }
- static stbir__inline stbir__FP16 stbir__float_to_half(float val)
- {
- stbir__FP32 f32infty = { 255 << 23 };
- stbir__FP32 f16max = { (127 + 16) << 23 };
- stbir__FP32 denorm_magic = { ((127 - 15) + (23 - 10) + 1) << 23 };
- unsigned int sign_mask = 0x80000000u;
- stbir__FP16 o = { 0 };
- stbir__FP32 f;
- unsigned int sign;
- f.f = val;
- sign = f.u & sign_mask;
- f.u ^= sign;
- if (f.u >= f16max.u) // result is Inf or NaN (all exponent bits set)
- o.u = (f.u > f32infty.u) ? 0x7e00 : 0x7c00; // NaN->qNaN and Inf->Inf
- else // (De)normalized number or zero
- {
- if (f.u < (113 << 23)) // resulting FP16 is subnormal or zero
- {
- // use a magic value to align our 10 mantissa bits at the bottom of
- // the float. as long as FP addition is round-to-nearest-even this
- // just works.
- f.f += denorm_magic.f;
- // and one integer subtract of the bias later, we have our final float!
- o.u = (unsigned short) ( f.u - denorm_magic.u );
- }
- else
- {
- unsigned int mant_odd = (f.u >> 13) & 1; // resulting mantissa is odd
- // update exponent, rounding bias part 1
- f.u = f.u + ((15u - 127) << 23) + 0xfff;
- // rounding bias part 2
- f.u += mant_odd;
- // take the bits!
- o.u = (unsigned short) ( f.u >> 13 );
- }
- }
- o.u |= sign >> 16;
- return o;
- }
- #endif
- #if defined(STBIR_FP16C)
- #include <immintrin.h>
- static stbir__inline void stbir__half_to_float_SIMD(float * output, stbir__FP16 const * input)
- {
- _mm256_storeu_ps( (float*)output, _mm256_cvtph_ps( _mm_loadu_si128( (__m128i const* )input ) ) );
- }
- static stbir__inline void stbir__float_to_half_SIMD(stbir__FP16 * output, float const * input)
- {
- _mm_storeu_si128( (__m128i*)output, _mm256_cvtps_ph( _mm256_loadu_ps( input ), 0 ) );
- }
- static stbir__inline float stbir__half_to_float( stbir__FP16 h )
- {
- return _mm_cvtss_f32( _mm_cvtph_ps( _mm_cvtsi32_si128( (int)h.u ) ) );
- }
- static stbir__inline stbir__FP16 stbir__float_to_half( float f )
- {
- stbir__FP16 h;
- h.u = (unsigned short) _mm_cvtsi128_si32( _mm_cvtps_ph( _mm_set_ss( f ), 0 ) );
- return h;
- }
- #elif defined(STBIR_SSE2)
- // Fabian's half float routines, see: https://gist.github.com/rygorous/2156668
- stbir__inline static void stbir__half_to_float_SIMD(float * output, void const * input)
- {
- static const STBIR__SIMDI_CONST(mask_nosign, 0x7fff);
- static const STBIR__SIMDI_CONST(smallest_normal, 0x0400);
- static const STBIR__SIMDI_CONST(infinity, 0x7c00);
- static const STBIR__SIMDI_CONST(expadjust_normal, (127 - 15) << 23);
- static const STBIR__SIMDI_CONST(magic_denorm, 113 << 23);
- __m128i i = _mm_loadu_si128 ( (__m128i const*)(input) );
- __m128i h = _mm_unpacklo_epi16 ( i, _mm_setzero_si128() );
- __m128i mnosign = STBIR__CONSTI(mask_nosign);
- __m128i eadjust = STBIR__CONSTI(expadjust_normal);
- __m128i smallest = STBIR__CONSTI(smallest_normal);
- __m128i infty = STBIR__CONSTI(infinity);
- __m128i expmant = _mm_and_si128(mnosign, h);
- __m128i justsign = _mm_xor_si128(h, expmant);
- __m128i b_notinfnan = _mm_cmpgt_epi32(infty, expmant);
- __m128i b_isdenorm = _mm_cmpgt_epi32(smallest, expmant);
- __m128i shifted = _mm_slli_epi32(expmant, 13);
- __m128i adj_infnan = _mm_andnot_si128(b_notinfnan, eadjust);
- __m128i adjusted = _mm_add_epi32(eadjust, shifted);
- __m128i den1 = _mm_add_epi32(shifted, STBIR__CONSTI(magic_denorm));
- __m128i adjusted2 = _mm_add_epi32(adjusted, adj_infnan);
- __m128 den2 = _mm_sub_ps(_mm_castsi128_ps(den1), *(const __m128 *)&magic_denorm);
- __m128 adjusted3 = _mm_and_ps(den2, _mm_castsi128_ps(b_isdenorm));
- __m128 adjusted4 = _mm_andnot_ps(_mm_castsi128_ps(b_isdenorm), _mm_castsi128_ps(adjusted2));
- __m128 adjusted5 = _mm_or_ps(adjusted3, adjusted4);
- __m128i sign = _mm_slli_epi32(justsign, 16);
- __m128 final = _mm_or_ps(adjusted5, _mm_castsi128_ps(sign));
- stbir__simdf_store( output + 0, final );
- h = _mm_unpackhi_epi16 ( i, _mm_setzero_si128() );
- expmant = _mm_and_si128(mnosign, h);
- justsign = _mm_xor_si128(h, expmant);
- b_notinfnan = _mm_cmpgt_epi32(infty, expmant);
- b_isdenorm = _mm_cmpgt_epi32(smallest, expmant);
- shifted = _mm_slli_epi32(expmant, 13);
- adj_infnan = _mm_andnot_si128(b_notinfnan, eadjust);
- adjusted = _mm_add_epi32(eadjust, shifted);
- den1 = _mm_add_epi32(shifted, STBIR__CONSTI(magic_denorm));
- adjusted2 = _mm_add_epi32(adjusted, adj_infnan);
- den2 = _mm_sub_ps(_mm_castsi128_ps(den1), *(const __m128 *)&magic_denorm);
- adjusted3 = _mm_and_ps(den2, _mm_castsi128_ps(b_isdenorm));
- adjusted4 = _mm_andnot_ps(_mm_castsi128_ps(b_isdenorm), _mm_castsi128_ps(adjusted2));
- adjusted5 = _mm_or_ps(adjusted3, adjusted4);
- sign = _mm_slli_epi32(justsign, 16);
- final = _mm_or_ps(adjusted5, _mm_castsi128_ps(sign));
- stbir__simdf_store( output + 4, final );
- // ~38 SSE2 ops for 8 values
- }
- // Fabian's round-to-nearest-even float to half
- // ~48 SSE2 ops for 8 output
- stbir__inline static void stbir__float_to_half_SIMD(void * output, float const * input)
- {
- static const STBIR__SIMDI_CONST(mask_sign, 0x80000000u);
- static const STBIR__SIMDI_CONST(c_f16max, (127 + 16) << 23); // all FP32 values >=this round to +inf
- static const STBIR__SIMDI_CONST(c_nanbit, 0x200);
- static const STBIR__SIMDI_CONST(c_infty_as_fp16, 0x7c00);
- static const STBIR__SIMDI_CONST(c_min_normal, (127 - 14) << 23); // smallest FP32 that yields a normalized FP16
- static const STBIR__SIMDI_CONST(c_subnorm_magic, ((127 - 15) + (23 - 10) + 1) << 23);
- static const STBIR__SIMDI_CONST(c_normal_bias, 0xfff - ((127 - 15) << 23)); // adjust exponent and add mantissa rounding
- __m128 f = _mm_loadu_ps(input);
- __m128 msign = _mm_castsi128_ps(STBIR__CONSTI(mask_sign));
- __m128 justsign = _mm_and_ps(msign, f);
- __m128 absf = _mm_xor_ps(f, justsign);
- __m128i absf_int = _mm_castps_si128(absf); // the cast is "free" (extra bypass latency, but no thruput hit)
- __m128i f16max = STBIR__CONSTI(c_f16max);
- __m128 b_isnan = _mm_cmpunord_ps(absf, absf); // is this a NaN?
- __m128i b_isregular = _mm_cmpgt_epi32(f16max, absf_int); // (sub)normalized or special?
- __m128i nanbit = _mm_and_si128(_mm_castps_si128(b_isnan), STBIR__CONSTI(c_nanbit));
- __m128i inf_or_nan = _mm_or_si128(nanbit, STBIR__CONSTI(c_infty_as_fp16)); // output for specials
- __m128i min_normal = STBIR__CONSTI(c_min_normal);
- __m128i b_issub = _mm_cmpgt_epi32(min_normal, absf_int);
- // "result is subnormal" path
- __m128 subnorm1 = _mm_add_ps(absf, _mm_castsi128_ps(STBIR__CONSTI(c_subnorm_magic))); // magic value to round output mantissa
- __m128i subnorm2 = _mm_sub_epi32(_mm_castps_si128(subnorm1), STBIR__CONSTI(c_subnorm_magic)); // subtract out bias
- // "result is normal" path
- __m128i mantoddbit = _mm_slli_epi32(absf_int, 31 - 13); // shift bit 13 (mantissa LSB) to sign
- __m128i mantodd = _mm_srai_epi32(mantoddbit, 31); // -1 if FP16 mantissa odd, else 0
- __m128i round1 = _mm_add_epi32(absf_int, STBIR__CONSTI(c_normal_bias));
- __m128i round2 = _mm_sub_epi32(round1, mantodd); // if mantissa LSB odd, bias towards rounding up (RTNE)
- __m128i normal = _mm_srli_epi32(round2, 13); // rounded result
- // combine the two non-specials
- __m128i nonspecial = _mm_or_si128(_mm_and_si128(subnorm2, b_issub), _mm_andnot_si128(b_issub, normal));
- // merge in specials as well
- __m128i joined = _mm_or_si128(_mm_and_si128(nonspecial, b_isregular), _mm_andnot_si128(b_isregular, inf_or_nan));
- __m128i sign_shift = _mm_srai_epi32(_mm_castps_si128(justsign), 16);
- __m128i final2, final= _mm_or_si128(joined, sign_shift);
- f = _mm_loadu_ps(input+4);
- justsign = _mm_and_ps(msign, f);
- absf = _mm_xor_ps(f, justsign);
- absf_int = _mm_castps_si128(absf); // the cast is "free" (extra bypass latency, but no thruput hit)
- b_isnan = _mm_cmpunord_ps(absf, absf); // is this a NaN?
- b_isregular = _mm_cmpgt_epi32(f16max, absf_int); // (sub)normalized or special?
- nanbit = _mm_and_si128(_mm_castps_si128(b_isnan), c_nanbit);
- inf_or_nan = _mm_or_si128(nanbit, STBIR__CONSTI(c_infty_as_fp16)); // output for specials
- b_issub = _mm_cmpgt_epi32(min_normal, absf_int);
- // "result is subnormal" path
- subnorm1 = _mm_add_ps(absf, _mm_castsi128_ps(STBIR__CONSTI(c_subnorm_magic))); // magic value to round output mantissa
- subnorm2 = _mm_sub_epi32(_mm_castps_si128(subnorm1), STBIR__CONSTI(c_subnorm_magic)); // subtract out bias
- // "result is normal" path
- mantoddbit = _mm_slli_epi32(absf_int, 31 - 13); // shift bit 13 (mantissa LSB) to sign
- mantodd = _mm_srai_epi32(mantoddbit, 31); // -1 if FP16 mantissa odd, else 0
- round1 = _mm_add_epi32(absf_int, STBIR__CONSTI(c_normal_bias));
- round2 = _mm_sub_epi32(round1, mantodd); // if mantissa LSB odd, bias towards rounding up (RTNE)
- normal = _mm_srli_epi32(round2, 13); // rounded result
- // combine the two non-specials
- nonspecial = _mm_or_si128(_mm_and_si128(subnorm2, b_issub), _mm_andnot_si128(b_issub, normal));
- // merge in specials as well
- joined = _mm_or_si128(_mm_and_si128(nonspecial, b_isregular), _mm_andnot_si128(b_isregular, inf_or_nan));
- sign_shift = _mm_srai_epi32(_mm_castps_si128(justsign), 16);
- final2 = _mm_or_si128(joined, sign_shift);
- final = _mm_packs_epi32(final, final2);
- stbir__simdi_store( output,final );
- }
- #elif defined(STBIR_NEON) && defined(_MSC_VER) && defined(_M_ARM64) && !defined(__clang__) // 64-bit ARM on MSVC (not clang)
- static stbir__inline void stbir__half_to_float_SIMD(float * output, stbir__FP16 const * input)
- {
- float16x4_t in0 = vld1_f16(input + 0);
- float16x4_t in1 = vld1_f16(input + 4);
- vst1q_f32(output + 0, vcvt_f32_f16(in0));
- vst1q_f32(output + 4, vcvt_f32_f16(in1));
- }
- static stbir__inline void stbir__float_to_half_SIMD(stbir__FP16 * output, float const * input)
- {
- float16x4_t out0 = vcvt_f16_f32(vld1q_f32(input + 0));
- float16x4_t out1 = vcvt_f16_f32(vld1q_f32(input + 4));
- vst1_f16(output+0, out0);
- vst1_f16(output+4, out1);
- }
- static stbir__inline float stbir__half_to_float( stbir__FP16 h )
- {
- return vgetq_lane_f32(vcvt_f32_f16(vld1_dup_f16(&h)), 0);
- }
- static stbir__inline stbir__FP16 stbir__float_to_half( float f )
- {
- return vget_lane_f16(vcvt_f16_f32(vdupq_n_f32(f)), 0).n16_u16[0];
- }
- #elif defined(STBIR_NEON) && ( defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ ) ) // 64-bit ARM
- static stbir__inline void stbir__half_to_float_SIMD(float * output, stbir__FP16 const * input)
- {
- float16x8_t in = vld1q_f16(input);
- vst1q_f32(output + 0, vcvt_f32_f16(vget_low_f16(in)));
- vst1q_f32(output + 4, vcvt_f32_f16(vget_high_f16(in)));
- }
- static stbir__inline void stbir__float_to_half_SIMD(stbir__FP16 * output, float const * input)
- {
- float16x4_t out0 = vcvt_f16_f32(vld1q_f32(input + 0));
- float16x4_t out1 = vcvt_f16_f32(vld1q_f32(input + 4));
- vst1q_f16(output, vcombine_f16(out0, out1));
- }
- static stbir__inline float stbir__half_to_float( stbir__FP16 h )
- {
- return vgetq_lane_f32(vcvt_f32_f16(vdup_n_f16(h)), 0);
- }
- static stbir__inline stbir__FP16 stbir__float_to_half( float f )
- {
- return vget_lane_f16(vcvt_f16_f32(vdupq_n_f32(f)), 0);
- }
- #elif defined(STBIR_WASM) || (defined(STBIR_NEON) && (defined(_MSC_VER) || defined(_M_ARM) || defined(__arm__))) // WASM or 32-bit ARM on MSVC/clang
- static stbir__inline void stbir__half_to_float_SIMD(float * output, stbir__FP16 const * input)
- {
- for (int i=0; i<8; i++)
- {
- output[i] = stbir__half_to_float(input[i]);
- }
- }
- static stbir__inline void stbir__float_to_half_SIMD(stbir__FP16 * output, float const * input)
- {
- for (int i=0; i<8; i++)
- {
- output[i] = stbir__float_to_half(input[i]);
- }
- }
- #endif
- #ifdef STBIR_SIMD
- #define stbir__simdf_0123to3333( out, reg ) (out) = stbir__simdf_swiz( reg, 3,3,3,3 )
- #define stbir__simdf_0123to2222( out, reg ) (out) = stbir__simdf_swiz( reg, 2,2,2,2 )
- #define stbir__simdf_0123to1111( out, reg ) (out) = stbir__simdf_swiz( reg, 1,1,1,1 )
- #define stbir__simdf_0123to0000( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,0,0 )
- #define stbir__simdf_0123to0003( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,0,3 )
- #define stbir__simdf_0123to0001( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,0,1 )
- #define stbir__simdf_0123to1122( out, reg ) (out) = stbir__simdf_swiz( reg, 1,1,2,2 )
- #define stbir__simdf_0123to2333( out, reg ) (out) = stbir__simdf_swiz( reg, 2,3,3,3 )
- #define stbir__simdf_0123to0023( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,2,3 )
- #define stbir__simdf_0123to1230( out, reg ) (out) = stbir__simdf_swiz( reg, 1,2,3,0 )
- #define stbir__simdf_0123to2103( out, reg ) (out) = stbir__simdf_swiz( reg, 2,1,0,3 )
- #define stbir__simdf_0123to3210( out, reg ) (out) = stbir__simdf_swiz( reg, 3,2,1,0 )
- #define stbir__simdf_0123to2301( out, reg ) (out) = stbir__simdf_swiz( reg, 2,3,0,1 )
- #define stbir__simdf_0123to3012( out, reg ) (out) = stbir__simdf_swiz( reg, 3,0,1,2 )
- #define stbir__simdf_0123to0011( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,1,1 )
- #define stbir__simdf_0123to1100( out, reg ) (out) = stbir__simdf_swiz( reg, 1,1,0,0 )
- #define stbir__simdf_0123to2233( out, reg ) (out) = stbir__simdf_swiz( reg, 2,2,3,3 )
- #define stbir__simdf_0123to1133( out, reg ) (out) = stbir__simdf_swiz( reg, 1,1,3,3 )
- #define stbir__simdf_0123to0022( out, reg ) (out) = stbir__simdf_swiz( reg, 0,0,2,2 )
- #define stbir__simdf_0123to1032( out, reg ) (out) = stbir__simdf_swiz( reg, 1,0,3,2 )
- typedef union stbir__simdi_u32
- {
- stbir_uint32 m128i_u32[4];
- int m128i_i32[4];
- stbir__simdi m128i_i128;
- } stbir__simdi_u32;
- static const int STBIR_mask[9] = { 0,0,0,-1,-1,-1,0,0,0 };
- static const STBIR__SIMDF_CONST(STBIR_max_uint8_as_float, stbir__max_uint8_as_float);
- static const STBIR__SIMDF_CONST(STBIR_max_uint16_as_float, stbir__max_uint16_as_float);
- static const STBIR__SIMDF_CONST(STBIR_max_uint8_as_float_inverted, stbir__max_uint8_as_float_inverted);
- static const STBIR__SIMDF_CONST(STBIR_max_uint16_as_float_inverted, stbir__max_uint16_as_float_inverted);
- static const STBIR__SIMDF_CONST(STBIR_simd_point5, 0.5f);
- static const STBIR__SIMDF_CONST(STBIR_ones, 1.0f);
- static const STBIR__SIMDI_CONST(STBIR_almost_zero, (127 - 13) << 23);
- static const STBIR__SIMDI_CONST(STBIR_almost_one, 0x3f7fffff);
- static const STBIR__SIMDI_CONST(STBIR_mastissa_mask, 0xff);
- static const STBIR__SIMDI_CONST(STBIR_topscale, 0x02000000);
- // Basically, in simd mode, we unroll the proper amount, and we don't want
- // the non-simd remnant loops to be unroll because they only run a few times
- // Adding this switch saves about 5K on clang which is Captain Unroll the 3rd.
- #define STBIR_SIMD_STREAMOUT_PTR( star ) STBIR_STREAMOUT_PTR( star )
- #define STBIR_SIMD_NO_UNROLL(ptr) STBIR_NO_UNROLL(ptr)
- #define STBIR_SIMD_NO_UNROLL_LOOP_START STBIR_NO_UNROLL_LOOP_START
- #define STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR STBIR_NO_UNROLL_LOOP_START_INF_FOR
- #ifdef STBIR_MEMCPY
- #undef STBIR_MEMCPY
- #endif
- #define STBIR_MEMCPY stbir_simd_memcpy
- // override normal use of memcpy with much simpler copy (faster and smaller with our sized copies)
- static void stbir_simd_memcpy( void * dest, void const * src, size_t bytes )
- {
- char STBIR_SIMD_STREAMOUT_PTR (*) d = (char*) dest;
- char STBIR_SIMD_STREAMOUT_PTR( * ) d_end = ((char*) dest) + bytes;
- ptrdiff_t ofs_to_src = (char*)src - (char*)dest;
- // check overlaps
- STBIR_ASSERT( ( ( d >= ( (char*)src) + bytes ) ) || ( ( d + bytes ) <= (char*)src ) );
- if ( bytes < (16*stbir__simdfX_float_count) )
- {
- if ( bytes < 16 )
- {
- if ( bytes )
- {
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do
- {
- STBIR_SIMD_NO_UNROLL(d);
- d[ 0 ] = d[ ofs_to_src ];
- ++d;
- } while ( d < d_end );
- }
- }
- else
- {
- stbir__simdf x;
- // do one unaligned to get us aligned for the stream out below
- stbir__simdf_load( x, ( d + ofs_to_src ) );
- stbir__simdf_store( d, x );
- d = (char*)( ( ( (size_t)d ) + 16 ) & ~15 );
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- STBIR_SIMD_NO_UNROLL(d);
- if ( d > ( d_end - 16 ) )
- {
- if ( d == d_end )
- return;
- d = d_end - 16;
- }
- stbir__simdf_load( x, ( d + ofs_to_src ) );
- stbir__simdf_store( d, x );
- d += 16;
- }
- }
- }
- else
- {
- stbir__simdfX x0,x1,x2,x3;
- // do one unaligned to get us aligned for the stream out below
- stbir__simdfX_load( x0, ( d + ofs_to_src ) + 0*stbir__simdfX_float_count );
- stbir__simdfX_load( x1, ( d + ofs_to_src ) + 4*stbir__simdfX_float_count );
- stbir__simdfX_load( x2, ( d + ofs_to_src ) + 8*stbir__simdfX_float_count );
- stbir__simdfX_load( x3, ( d + ofs_to_src ) + 12*stbir__simdfX_float_count );
- stbir__simdfX_store( d + 0*stbir__simdfX_float_count, x0 );
- stbir__simdfX_store( d + 4*stbir__simdfX_float_count, x1 );
- stbir__simdfX_store( d + 8*stbir__simdfX_float_count, x2 );
- stbir__simdfX_store( d + 12*stbir__simdfX_float_count, x3 );
- d = (char*)( ( ( (size_t)d ) + (16*stbir__simdfX_float_count) ) & ~((16*stbir__simdfX_float_count)-1) );
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- STBIR_SIMD_NO_UNROLL(d);
- if ( d > ( d_end - (16*stbir__simdfX_float_count) ) )
- {
- if ( d == d_end )
- return;
- d = d_end - (16*stbir__simdfX_float_count);
- }
- stbir__simdfX_load( x0, ( d + ofs_to_src ) + 0*stbir__simdfX_float_count );
- stbir__simdfX_load( x1, ( d + ofs_to_src ) + 4*stbir__simdfX_float_count );
- stbir__simdfX_load( x2, ( d + ofs_to_src ) + 8*stbir__simdfX_float_count );
- stbir__simdfX_load( x3, ( d + ofs_to_src ) + 12*stbir__simdfX_float_count );
- stbir__simdfX_store( d + 0*stbir__simdfX_float_count, x0 );
- stbir__simdfX_store( d + 4*stbir__simdfX_float_count, x1 );
- stbir__simdfX_store( d + 8*stbir__simdfX_float_count, x2 );
- stbir__simdfX_store( d + 12*stbir__simdfX_float_count, x3 );
- d += (16*stbir__simdfX_float_count);
- }
- }
- }
- // memcpy that is specically intentionally overlapping (src is smaller then dest, so can be
- // a normal forward copy, bytes is divisible by 4 and bytes is greater than or equal to
- // the diff between dest and src)
- static void stbir_overlapping_memcpy( void * dest, void const * src, size_t bytes )
- {
- char STBIR_SIMD_STREAMOUT_PTR (*) sd = (char*) src;
- char STBIR_SIMD_STREAMOUT_PTR( * ) s_end = ((char*) src) + bytes;
- ptrdiff_t ofs_to_dest = (char*)dest - (char*)src;
- if ( ofs_to_dest >= 16 ) // is the overlap more than 16 away?
- {
- char STBIR_SIMD_STREAMOUT_PTR( * ) s_end16 = ((char*) src) + (bytes&~15);
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do
- {
- stbir__simdf x;
- STBIR_SIMD_NO_UNROLL(sd);
- stbir__simdf_load( x, sd );
- stbir__simdf_store( ( sd + ofs_to_dest ), x );
- sd += 16;
- } while ( sd < s_end16 );
- if ( sd == s_end )
- return;
- }
- do
- {
- STBIR_SIMD_NO_UNROLL(sd);
- *(int*)( sd + ofs_to_dest ) = *(int*) sd;
- sd += 4;
- } while ( sd < s_end );
- }
- #else // no SSE2
- // when in scalar mode, we let unrolling happen, so this macro just does the __restrict
- #define STBIR_SIMD_STREAMOUT_PTR( star ) STBIR_STREAMOUT_PTR( star )
- #define STBIR_SIMD_NO_UNROLL(ptr)
- #define STBIR_SIMD_NO_UNROLL_LOOP_START
- #define STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- #endif // SSE2
- #ifdef STBIR_PROFILE
- #ifndef STBIR_PROFILE_FUNC
- #if defined(_x86_64) || defined( __x86_64__ ) || defined( _M_X64 ) || defined(__x86_64) || defined(__SSE2__) || defined(STBIR_SSE) || defined( _M_IX86_FP ) || defined(__i386) || defined( __i386__ ) || defined( _M_IX86 ) || defined( _X86_ )
- #ifdef _MSC_VER
- STBIRDEF stbir_uint64 __rdtsc();
- #define STBIR_PROFILE_FUNC() __rdtsc()
- #else // non msvc
- static stbir__inline stbir_uint64 STBIR_PROFILE_FUNC()
- {
- stbir_uint32 lo, hi;
- asm volatile ("rdtsc" : "=a" (lo), "=d" (hi) );
- return ( ( (stbir_uint64) hi ) << 32 ) | ( (stbir_uint64) lo );
- }
- #endif // msvc
- #elif defined( _M_ARM64 ) || defined( __aarch64__ ) || defined( __arm64__ ) || defined(__ARM_NEON__)
- #if defined( _MSC_VER ) && !defined(__clang__)
- #define STBIR_PROFILE_FUNC() _ReadStatusReg(ARM64_CNTVCT)
- #else
- static stbir__inline stbir_uint64 STBIR_PROFILE_FUNC()
- {
- stbir_uint64 tsc;
- asm volatile("mrs %0, cntvct_el0" : "=r" (tsc));
- return tsc;
- }
- #endif
- #else // x64, arm
- #error Unknown platform for profiling.
- #endif // x64, arm
- #endif // STBIR_PROFILE_FUNC
- #define STBIR_ONLY_PROFILE_GET_SPLIT_INFO ,stbir__per_split_info * split_info
- #define STBIR_ONLY_PROFILE_SET_SPLIT_INFO ,split_info
- #define STBIR_ONLY_PROFILE_BUILD_GET_INFO ,stbir__info * profile_info
- #define STBIR_ONLY_PROFILE_BUILD_SET_INFO ,profile_info
- // super light-weight micro profiler
- #define STBIR_PROFILE_START_ll( info, wh ) { stbir_uint64 wh##thiszonetime = STBIR_PROFILE_FUNC(); stbir_uint64 * wh##save_parent_excluded_ptr = info->current_zone_excluded_ptr; stbir_uint64 wh##current_zone_excluded = 0; info->current_zone_excluded_ptr = &wh##current_zone_excluded;
- #define STBIR_PROFILE_END_ll( info, wh ) wh##thiszonetime = STBIR_PROFILE_FUNC() - wh##thiszonetime; info->profile.named.wh += wh##thiszonetime - wh##current_zone_excluded; *wh##save_parent_excluded_ptr += wh##thiszonetime; info->current_zone_excluded_ptr = wh##save_parent_excluded_ptr; }
- #define STBIR_PROFILE_FIRST_START_ll( info, wh ) { int i; info->current_zone_excluded_ptr = &info->profile.named.total; for(i=0;i<STBIR__ARRAY_SIZE(info->profile.array);i++) info->profile.array[i]=0; } STBIR_PROFILE_START_ll( info, wh );
- #define STBIR_PROFILE_CLEAR_EXTRAS_ll( info, num ) { int extra; for(extra=1;extra<(num);extra++) { int i; for(i=0;i<STBIR__ARRAY_SIZE((info)->profile.array);i++) (info)[extra].profile.array[i]=0; } }
- // for thread data
- #define STBIR_PROFILE_START( wh ) STBIR_PROFILE_START_ll( split_info, wh )
- #define STBIR_PROFILE_END( wh ) STBIR_PROFILE_END_ll( split_info, wh )
- #define STBIR_PROFILE_FIRST_START( wh ) STBIR_PROFILE_FIRST_START_ll( split_info, wh )
- #define STBIR_PROFILE_CLEAR_EXTRAS() STBIR_PROFILE_CLEAR_EXTRAS_ll( split_info, split_count )
- // for build data
- #define STBIR_PROFILE_BUILD_START( wh ) STBIR_PROFILE_START_ll( profile_info, wh )
- #define STBIR_PROFILE_BUILD_END( wh ) STBIR_PROFILE_END_ll( profile_info, wh )
- #define STBIR_PROFILE_BUILD_FIRST_START( wh ) STBIR_PROFILE_FIRST_START_ll( profile_info, wh )
- #define STBIR_PROFILE_BUILD_CLEAR( info ) { int i; for(i=0;i<STBIR__ARRAY_SIZE(info->profile.array);i++) info->profile.array[i]=0; }
- #else // no profile
- #define STBIR_ONLY_PROFILE_GET_SPLIT_INFO
- #define STBIR_ONLY_PROFILE_SET_SPLIT_INFO
- #define STBIR_ONLY_PROFILE_BUILD_GET_INFO
- #define STBIR_ONLY_PROFILE_BUILD_SET_INFO
- #define STBIR_PROFILE_START( wh )
- #define STBIR_PROFILE_END( wh )
- #define STBIR_PROFILE_FIRST_START( wh )
- #define STBIR_PROFILE_CLEAR_EXTRAS( )
- #define STBIR_PROFILE_BUILD_START( wh )
- #define STBIR_PROFILE_BUILD_END( wh )
- #define STBIR_PROFILE_BUILD_FIRST_START( wh )
- #define STBIR_PROFILE_BUILD_CLEAR( info )
- #endif // stbir_profile
- #ifndef STBIR_CEILF
- #include <math.h>
- #if _MSC_VER <= 1200 // support VC6 for Sean
- #define STBIR_CEILF(x) ((float)ceil((float)(x)))
- #define STBIR_FLOORF(x) ((float)floor((float)(x)))
- #else
- #define STBIR_CEILF(x) ceilf(x)
- #define STBIR_FLOORF(x) floorf(x)
- #endif
- #endif
- #ifndef STBIR_MEMCPY
- // For memcpy
- #include <string.h>
- #define STBIR_MEMCPY( dest, src, len ) memcpy( dest, src, len )
- #endif
- #ifndef STBIR_SIMD
- // memcpy that is specifically intentionally overlapping (src is smaller then dest, so can be
- // a normal forward copy, bytes is divisible by 4 and bytes is greater than or equal to
- // the diff between dest and src)
- static void stbir_overlapping_memcpy( void * dest, void const * src, size_t bytes )
- {
- char STBIR_SIMD_STREAMOUT_PTR (*) sd = (char*) src;
- char STBIR_SIMD_STREAMOUT_PTR( * ) s_end = ((char*) src) + bytes;
- ptrdiff_t ofs_to_dest = (char*)dest - (char*)src;
- if ( ofs_to_dest >= 8 ) // is the overlap more than 8 away?
- {
- char STBIR_SIMD_STREAMOUT_PTR( * ) s_end8 = ((char*) src) + (bytes&~7);
- STBIR_NO_UNROLL_LOOP_START
- do
- {
- STBIR_NO_UNROLL(sd);
- *(stbir_uint64*)( sd + ofs_to_dest ) = *(stbir_uint64*) sd;
- sd += 8;
- } while ( sd < s_end8 );
- if ( sd == s_end )
- return;
- }
- STBIR_NO_UNROLL_LOOP_START
- do
- {
- STBIR_NO_UNROLL(sd);
- *(int*)( sd + ofs_to_dest ) = *(int*) sd;
- sd += 4;
- } while ( sd < s_end );
- }
- #endif
- static float stbir__filter_trapezoid(float x, float scale, void * user_data)
- {
- float halfscale = scale / 2;
- float t = 0.5f + halfscale;
- STBIR_ASSERT(scale <= 1);
- STBIR__UNUSED(user_data);
- if ( x < 0.0f ) x = -x;
- if (x >= t)
- return 0.0f;
- else
- {
- float r = 0.5f - halfscale;
- if (x <= r)
- return 1.0f;
- else
- return (t - x) / scale;
- }
- }
- static float stbir__support_trapezoid(float scale, void * user_data)
- {
- STBIR__UNUSED(user_data);
- return 0.5f + scale / 2.0f;
- }
- static float stbir__filter_triangle(float x, float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- if ( x < 0.0f ) x = -x;
- if (x <= 1.0f)
- return 1.0f - x;
- else
- return 0.0f;
- }
- static float stbir__filter_point(float x, float s, void * user_data)
- {
- STBIR__UNUSED(x);
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- return 1.0f;
- }
- static float stbir__filter_cubic(float x, float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- if ( x < 0.0f ) x = -x;
- if (x < 1.0f)
- return (4.0f + x*x*(3.0f*x - 6.0f))/6.0f;
- else if (x < 2.0f)
- return (8.0f + x*(-12.0f + x*(6.0f - x)))/6.0f;
- return (0.0f);
- }
- static float stbir__filter_catmullrom(float x, float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- if ( x < 0.0f ) x = -x;
- if (x < 1.0f)
- return 1.0f - x*x*(2.5f - 1.5f*x);
- else if (x < 2.0f)
- return 2.0f - x*(4.0f + x*(0.5f*x - 2.5f));
- return (0.0f);
- }
- static float stbir__filter_mitchell(float x, float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- if ( x < 0.0f ) x = -x;
- if (x < 1.0f)
- return (16.0f + x*x*(21.0f * x - 36.0f))/18.0f;
- else if (x < 2.0f)
- return (32.0f + x*(-60.0f + x*(36.0f - 7.0f*x)))/18.0f;
- return (0.0f);
- }
- static float stbir__support_zeropoint5(float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- return 0.5f;
- }
- static float stbir__support_one(float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- return 1;
- }
- static float stbir__support_two(float s, void * user_data)
- {
- STBIR__UNUSED(s);
- STBIR__UNUSED(user_data);
- return 2;
- }
- // This is the maximum number of input samples that can affect an output sample
- // with the given filter from the output pixel's perspective
- static int stbir__get_filter_pixel_width(stbir__support_callback * support, float scale, void * user_data)
- {
- STBIR_ASSERT(support != 0);
- if ( scale >= ( 1.0f-stbir__small_float ) ) // upscale
- return (int)STBIR_CEILF(support(1.0f/scale,user_data) * 2.0f);
- else
- return (int)STBIR_CEILF(support(scale,user_data) * 2.0f / scale);
- }
- // this is how many coefficents per run of the filter (which is different
- // from the filter_pixel_width depending on if we are scattering or gathering)
- static int stbir__get_coefficient_width(stbir__sampler * samp, int is_gather, void * user_data)
- {
- float scale = samp->scale_info.scale;
- stbir__support_callback * support = samp->filter_support;
- switch( is_gather )
- {
- case 1:
- return (int)STBIR_CEILF(support(1.0f / scale, user_data) * 2.0f);
- case 2:
- return (int)STBIR_CEILF(support(scale, user_data) * 2.0f / scale);
- case 0:
- return (int)STBIR_CEILF(support(scale, user_data) * 2.0f);
- default:
- STBIR_ASSERT( (is_gather >= 0 ) && (is_gather <= 2 ) );
- return 0;
- }
- }
- static int stbir__get_contributors(stbir__sampler * samp, int is_gather)
- {
- if (is_gather)
- return samp->scale_info.output_sub_size;
- else
- return (samp->scale_info.input_full_size + samp->filter_pixel_margin * 2);
- }
- static int stbir__edge_zero_full( int n, int max )
- {
- STBIR__UNUSED(n);
- STBIR__UNUSED(max);
- return 0; // NOTREACHED
- }
- static int stbir__edge_clamp_full( int n, int max )
- {
- if (n < 0)
- return 0;
- if (n >= max)
- return max - 1;
- return n; // NOTREACHED
- }
- static int stbir__edge_reflect_full( int n, int max )
- {
- if (n < 0)
- {
- if (n > -max)
- return -n;
- else
- return max - 1;
- }
- if (n >= max)
- {
- int max2 = max * 2;
- if (n >= max2)
- return 0;
- else
- return max2 - n - 1;
- }
- return n; // NOTREACHED
- }
- static int stbir__edge_wrap_full( int n, int max )
- {
- if (n >= 0)
- return (n % max);
- else
- {
- int m = (-n) % max;
- if (m != 0)
- m = max - m;
- return (m);
- }
- }
- typedef int stbir__edge_wrap_func( int n, int max );
- static stbir__edge_wrap_func * stbir__edge_wrap_slow[] =
- {
- stbir__edge_clamp_full, // STBIR_EDGE_CLAMP
- stbir__edge_reflect_full, // STBIR_EDGE_REFLECT
- stbir__edge_wrap_full, // STBIR_EDGE_WRAP
- stbir__edge_zero_full, // STBIR_EDGE_ZERO
- };
- stbir__inline static int stbir__edge_wrap(stbir_edge edge, int n, int max)
- {
- // avoid per-pixel switch
- if (n >= 0 && n < max)
- return n;
- return stbir__edge_wrap_slow[edge]( n, max );
- }
- #define STBIR__MERGE_RUNS_PIXEL_THRESHOLD 16
- // get information on the extents of a sampler
- static void stbir__get_extents( stbir__sampler * samp, stbir__extents * scanline_extents )
- {
- int j, stop;
- int left_margin, right_margin;
- int min_n = 0x7fffffff, max_n = -0x7fffffff;
- int min_left = 0x7fffffff, max_left = -0x7fffffff;
- int min_right = 0x7fffffff, max_right = -0x7fffffff;
- stbir_edge edge = samp->edge;
- stbir__contributors* contributors = samp->contributors;
- int output_sub_size = samp->scale_info.output_sub_size;
- int input_full_size = samp->scale_info.input_full_size;
- int filter_pixel_margin = samp->filter_pixel_margin;
- STBIR_ASSERT( samp->is_gather );
- stop = output_sub_size;
- for (j = 0; j < stop; j++ )
- {
- STBIR_ASSERT( contributors[j].n1 >= contributors[j].n0 );
- if ( contributors[j].n0 < min_n )
- {
- min_n = contributors[j].n0;
- stop = j + filter_pixel_margin; // if we find a new min, only scan another filter width
- if ( stop > output_sub_size ) stop = output_sub_size;
- }
- }
- stop = 0;
- for (j = output_sub_size - 1; j >= stop; j-- )
- {
- STBIR_ASSERT( contributors[j].n1 >= contributors[j].n0 );
- if ( contributors[j].n1 > max_n )
- {
- max_n = contributors[j].n1;
- stop = j - filter_pixel_margin; // if we find a new max, only scan another filter width
- if (stop<0) stop = 0;
- }
- }
- STBIR_ASSERT( scanline_extents->conservative.n0 <= min_n );
- STBIR_ASSERT( scanline_extents->conservative.n1 >= max_n );
- // now calculate how much into the margins we really read
- left_margin = 0;
- if ( min_n < 0 )
- {
- left_margin = -min_n;
- min_n = 0;
- }
- right_margin = 0;
- if ( max_n >= input_full_size )
- {
- right_margin = max_n - input_full_size + 1;
- max_n = input_full_size - 1;
- }
- // index 1 is margin pixel extents (how many pixels we hang over the edge)
- scanline_extents->edge_sizes[0] = left_margin;
- scanline_extents->edge_sizes[1] = right_margin;
- // index 2 is pixels read from the input
- scanline_extents->spans[0].n0 = min_n;
- scanline_extents->spans[0].n1 = max_n;
- scanline_extents->spans[0].pixel_offset_for_input = min_n;
- // default to no other input range
- scanline_extents->spans[1].n0 = 0;
- scanline_extents->spans[1].n1 = -1;
- scanline_extents->spans[1].pixel_offset_for_input = 0;
- // don't have to do edge calc for zero clamp
- if ( edge == STBIR_EDGE_ZERO )
- return;
- // convert margin pixels to the pixels within the input (min and max)
- for( j = -left_margin ; j < 0 ; j++ )
- {
- int p = stbir__edge_wrap( edge, j, input_full_size );
- if ( p < min_left )
- min_left = p;
- if ( p > max_left )
- max_left = p;
- }
- for( j = input_full_size ; j < (input_full_size + right_margin) ; j++ )
- {
- int p = stbir__edge_wrap( edge, j, input_full_size );
- if ( p < min_right )
- min_right = p;
- if ( p > max_right )
- max_right = p;
- }
- // merge the left margin pixel region if it connects within 4 pixels of main pixel region
- if ( min_left != 0x7fffffff )
- {
- if ( ( ( min_left <= min_n ) && ( ( max_left + STBIR__MERGE_RUNS_PIXEL_THRESHOLD ) >= min_n ) ) ||
- ( ( min_n <= min_left ) && ( ( max_n + STBIR__MERGE_RUNS_PIXEL_THRESHOLD ) >= max_left ) ) )
- {
- scanline_extents->spans[0].n0 = min_n = stbir__min( min_n, min_left );
- scanline_extents->spans[0].n1 = max_n = stbir__max( max_n, max_left );
- scanline_extents->spans[0].pixel_offset_for_input = min_n;
- left_margin = 0;
- }
- }
- // merge the right margin pixel region if it connects within 4 pixels of main pixel region
- if ( min_right != 0x7fffffff )
- {
- if ( ( ( min_right <= min_n ) && ( ( max_right + STBIR__MERGE_RUNS_PIXEL_THRESHOLD ) >= min_n ) ) ||
- ( ( min_n <= min_right ) && ( ( max_n + STBIR__MERGE_RUNS_PIXEL_THRESHOLD ) >= max_right ) ) )
- {
- scanline_extents->spans[0].n0 = min_n = stbir__min( min_n, min_right );
- scanline_extents->spans[0].n1 = max_n = stbir__max( max_n, max_right );
- scanline_extents->spans[0].pixel_offset_for_input = min_n;
- right_margin = 0;
- }
- }
- STBIR_ASSERT( scanline_extents->conservative.n0 <= min_n );
- STBIR_ASSERT( scanline_extents->conservative.n1 >= max_n );
- // you get two ranges when you have the WRAP edge mode and you are doing just the a piece of the resize
- // so you need to get a second run of pixels from the opposite side of the scanline (which you
- // wouldn't need except for WRAP)
- // if we can't merge the min_left range, add it as a second range
- if ( ( left_margin ) && ( min_left != 0x7fffffff ) )
- {
- stbir__span * newspan = scanline_extents->spans + 1;
- STBIR_ASSERT( right_margin == 0 );
- if ( min_left < scanline_extents->spans[0].n0 )
- {
- scanline_extents->spans[1].pixel_offset_for_input = scanline_extents->spans[0].n0;
- scanline_extents->spans[1].n0 = scanline_extents->spans[0].n0;
- scanline_extents->spans[1].n1 = scanline_extents->spans[0].n1;
- --newspan;
- }
- newspan->pixel_offset_for_input = min_left;
- newspan->n0 = -left_margin;
- newspan->n1 = ( max_left - min_left ) - left_margin;
- scanline_extents->edge_sizes[0] = 0; // don't need to copy the left margin, since we are directly decoding into the margin
- }
- // if we can't merge the min_left range, add it as a second range
- else
- if ( ( right_margin ) && ( min_right != 0x7fffffff ) )
- {
- stbir__span * newspan = scanline_extents->spans + 1;
- if ( min_right < scanline_extents->spans[0].n0 )
- {
- scanline_extents->spans[1].pixel_offset_for_input = scanline_extents->spans[0].n0;
- scanline_extents->spans[1].n0 = scanline_extents->spans[0].n0;
- scanline_extents->spans[1].n1 = scanline_extents->spans[0].n1;
- --newspan;
- }
- newspan->pixel_offset_for_input = min_right;
- newspan->n0 = scanline_extents->spans[1].n1 + 1;
- newspan->n1 = scanline_extents->spans[1].n1 + 1 + ( max_right - min_right );
- scanline_extents->edge_sizes[1] = 0; // don't need to copy the right margin, since we are directly decoding into the margin
- }
- // sort the spans into write output order
- if ( ( scanline_extents->spans[1].n1 > scanline_extents->spans[1].n0 ) && ( scanline_extents->spans[0].n0 > scanline_extents->spans[1].n0 ) )
- {
- stbir__span tspan = scanline_extents->spans[0];
- scanline_extents->spans[0] = scanline_extents->spans[1];
- scanline_extents->spans[1] = tspan;
- }
- }
- static void stbir__calculate_in_pixel_range( int * first_pixel, int * last_pixel, float out_pixel_center, float out_filter_radius, float inv_scale, float out_shift, int input_size, stbir_edge edge )
- {
- int first, last;
- float out_pixel_influence_lowerbound = out_pixel_center - out_filter_radius;
- float out_pixel_influence_upperbound = out_pixel_center + out_filter_radius;
- float in_pixel_influence_lowerbound = (out_pixel_influence_lowerbound + out_shift) * inv_scale;
- float in_pixel_influence_upperbound = (out_pixel_influence_upperbound + out_shift) * inv_scale;
- first = (int)(STBIR_FLOORF(in_pixel_influence_lowerbound + 0.5f));
- last = (int)(STBIR_FLOORF(in_pixel_influence_upperbound - 0.5f));
- if ( last < first ) last = first; // point sample mode can span a value *right* at 0.5, and cause these to cross
- if ( edge == STBIR_EDGE_WRAP )
- {
- if ( first < -input_size )
- first = -input_size;
- if ( last >= (input_size*2))
- last = (input_size*2) - 1;
- }
- *first_pixel = first;
- *last_pixel = last;
- }
- static void stbir__calculate_coefficients_for_gather_upsample( float out_filter_radius, stbir__kernel_callback * kernel, stbir__scale_info * scale_info, int num_contributors, stbir__contributors* contributors, float* coefficient_group, int coefficient_width, stbir_edge edge, void * user_data )
- {
- int n, end;
- float inv_scale = scale_info->inv_scale;
- float out_shift = scale_info->pixel_shift;
- int input_size = scale_info->input_full_size;
- int numerator = scale_info->scale_numerator;
- int polyphase = ( ( scale_info->scale_is_rational ) && ( numerator < num_contributors ) );
- // Looping through out pixels
- end = num_contributors; if ( polyphase ) end = numerator;
- for (n = 0; n < end; n++)
- {
- int i;
- int last_non_zero;
- float out_pixel_center = (float)n + 0.5f;
- float in_center_of_out = (out_pixel_center + out_shift) * inv_scale;
- int in_first_pixel, in_last_pixel;
- stbir__calculate_in_pixel_range( &in_first_pixel, &in_last_pixel, out_pixel_center, out_filter_radius, inv_scale, out_shift, input_size, edge );
- // make sure we never generate a range larger than our precalculated coeff width
- // this only happens in point sample mode, but it's a good safe thing to do anyway
- if ( ( in_last_pixel - in_first_pixel + 1 ) > coefficient_width )
- in_last_pixel = in_first_pixel + coefficient_width - 1;
- last_non_zero = -1;
- for (i = 0; i <= in_last_pixel - in_first_pixel; i++)
- {
- float in_pixel_center = (float)(i + in_first_pixel) + 0.5f;
- float coeff = kernel(in_center_of_out - in_pixel_center, inv_scale, user_data);
- // kill denormals
- if ( ( ( coeff < stbir__small_float ) && ( coeff > -stbir__small_float ) ) )
- {
- if ( i == 0 ) // if we're at the front, just eat zero contributors
- {
- STBIR_ASSERT ( ( in_last_pixel - in_first_pixel ) != 0 ); // there should be at least one contrib
- ++in_first_pixel;
- i--;
- continue;
- }
- coeff = 0; // make sure is fully zero (should keep denormals away)
- }
- else
- last_non_zero = i;
- coefficient_group[i] = coeff;
- }
- in_last_pixel = last_non_zero+in_first_pixel; // kills trailing zeros
- contributors->n0 = in_first_pixel;
- contributors->n1 = in_last_pixel;
- STBIR_ASSERT(contributors->n1 >= contributors->n0);
- ++contributors;
- coefficient_group += coefficient_width;
- }
- }
- static void stbir__insert_coeff( stbir__contributors * contribs, float * coeffs, int new_pixel, float new_coeff, int max_width )
- {
- if ( new_pixel <= contribs->n1 ) // before the end
- {
- if ( new_pixel < contribs->n0 ) // before the front?
- {
- if ( ( contribs->n1 - new_pixel + 1 ) <= max_width )
- {
- int j, o = contribs->n0 - new_pixel;
- for ( j = contribs->n1 - contribs->n0 ; j <= 0 ; j-- )
- coeffs[ j + o ] = coeffs[ j ];
- for ( j = 1 ; j < o ; j-- )
- coeffs[ j ] = coeffs[ 0 ];
- coeffs[ 0 ] = new_coeff;
- contribs->n0 = new_pixel;
- }
- }
- else
- {
- coeffs[ new_pixel - contribs->n0 ] += new_coeff;
- }
- }
- else
- {
- if ( ( new_pixel - contribs->n0 + 1 ) <= max_width )
- {
- int j, e = new_pixel - contribs->n0;
- for( j = ( contribs->n1 - contribs->n0 ) + 1 ; j < e ; j++ ) // clear in-betweens coeffs if there are any
- coeffs[j] = 0;
- coeffs[ e ] = new_coeff;
- contribs->n1 = new_pixel;
- }
- }
- }
- static void stbir__calculate_out_pixel_range( int * first_pixel, int * last_pixel, float in_pixel_center, float in_pixels_radius, float scale, float out_shift, int out_size )
- {
- float in_pixel_influence_lowerbound = in_pixel_center - in_pixels_radius;
- float in_pixel_influence_upperbound = in_pixel_center + in_pixels_radius;
- float out_pixel_influence_lowerbound = in_pixel_influence_lowerbound * scale - out_shift;
- float out_pixel_influence_upperbound = in_pixel_influence_upperbound * scale - out_shift;
- int out_first_pixel = (int)(STBIR_FLOORF(out_pixel_influence_lowerbound + 0.5f));
- int out_last_pixel = (int)(STBIR_FLOORF(out_pixel_influence_upperbound - 0.5f));
- if ( out_first_pixel < 0 )
- out_first_pixel = 0;
- if ( out_last_pixel >= out_size )
- out_last_pixel = out_size - 1;
- *first_pixel = out_first_pixel;
- *last_pixel = out_last_pixel;
- }
- static void stbir__calculate_coefficients_for_gather_downsample( int start, int end, float in_pixels_radius, stbir__kernel_callback * kernel, stbir__scale_info * scale_info, int coefficient_width, int num_contributors, stbir__contributors * contributors, float * coefficient_group, void * user_data )
- {
- int in_pixel;
- int i;
- int first_out_inited = -1;
- float scale = scale_info->scale;
- float out_shift = scale_info->pixel_shift;
- int out_size = scale_info->output_sub_size;
- int numerator = scale_info->scale_numerator;
- int polyphase = ( ( scale_info->scale_is_rational ) && ( numerator < out_size ) );
- STBIR__UNUSED(num_contributors);
- // Loop through the input pixels
- for (in_pixel = start; in_pixel < end; in_pixel++)
- {
- float in_pixel_center = (float)in_pixel + 0.5f;
- float out_center_of_in = in_pixel_center * scale - out_shift;
- int out_first_pixel, out_last_pixel;
- stbir__calculate_out_pixel_range( &out_first_pixel, &out_last_pixel, in_pixel_center, in_pixels_radius, scale, out_shift, out_size );
- if ( out_first_pixel > out_last_pixel )
- continue;
- // clamp or exit if we are using polyphase filtering, and the limit is up
- if ( polyphase )
- {
- // when polyphase, you only have to do coeffs up to the numerator count
- if ( out_first_pixel == numerator )
- break;
- // don't do any extra work, clamp last pixel at numerator too
- if ( out_last_pixel >= numerator )
- out_last_pixel = numerator - 1;
- }
- for (i = 0; i <= out_last_pixel - out_first_pixel; i++)
- {
- float out_pixel_center = (float)(i + out_first_pixel) + 0.5f;
- float x = out_pixel_center - out_center_of_in;
- float coeff = kernel(x, scale, user_data) * scale;
- // kill the coeff if it's too small (avoid denormals)
- if ( ( ( coeff < stbir__small_float ) && ( coeff > -stbir__small_float ) ) )
- coeff = 0.0f;
- {
- int out = i + out_first_pixel;
- float * coeffs = coefficient_group + out * coefficient_width;
- stbir__contributors * contribs = contributors + out;
- // is this the first time this output pixel has been seen? Init it.
- if ( out > first_out_inited )
- {
- STBIR_ASSERT( out == ( first_out_inited + 1 ) ); // ensure we have only advanced one at time
- first_out_inited = out;
- contribs->n0 = in_pixel;
- contribs->n1 = in_pixel;
- coeffs[0] = coeff;
- }
- else
- {
- // insert on end (always in order)
- if ( coeffs[0] == 0.0f ) // if the first coefficent is zero, then zap it for this coeffs
- {
- STBIR_ASSERT( ( in_pixel - contribs->n0 ) == 1 ); // ensure that when we zap, we're at the 2nd pos
- contribs->n0 = in_pixel;
- }
- contribs->n1 = in_pixel;
- STBIR_ASSERT( ( in_pixel - contribs->n0 ) < coefficient_width );
- coeffs[in_pixel - contribs->n0] = coeff;
- }
- }
- }
- }
- }
- #ifdef STBIR_RENORMALIZE_IN_FLOAT
- #define STBIR_RENORM_TYPE float
- #else
- #define STBIR_RENORM_TYPE double
- #endif
- static void stbir__cleanup_gathered_coefficients( stbir_edge edge, stbir__filter_extent_info* filter_info, stbir__scale_info * scale_info, int num_contributors, stbir__contributors* contributors, float * coefficient_group, int coefficient_width )
- {
- int input_size = scale_info->input_full_size;
- int input_last_n1 = input_size - 1;
- int n, end;
- int lowest = 0x7fffffff;
- int highest = -0x7fffffff;
- int widest = -1;
- int numerator = scale_info->scale_numerator;
- int denominator = scale_info->scale_denominator;
- int polyphase = ( ( scale_info->scale_is_rational ) && ( numerator < num_contributors ) );
- float * coeffs;
- stbir__contributors * contribs;
- // weight all the coeffs for each sample
- coeffs = coefficient_group;
- contribs = contributors;
- end = num_contributors; if ( polyphase ) end = numerator;
- for (n = 0; n < end; n++)
- {
- int i;
- STBIR_RENORM_TYPE filter_scale, total_filter = 0;
- int e;
- // add all contribs
- e = contribs->n1 - contribs->n0;
- for( i = 0 ; i <= e ; i++ )
- {
- total_filter += (STBIR_RENORM_TYPE) coeffs[i];
- STBIR_ASSERT( ( coeffs[i] >= -2.0f ) && ( coeffs[i] <= 2.0f ) ); // check for wonky weights
- }
- // rescale
- if ( ( total_filter < stbir__small_float ) && ( total_filter > -stbir__small_float ) )
- {
- // all coeffs are extremely small, just zero it
- contribs->n1 = contribs->n0;
- coeffs[0] = 0.0f;
- }
- else
- {
- // if the total isn't 1.0, rescale everything
- if ( ( total_filter < (1.0f-stbir__small_float) ) || ( total_filter > (1.0f+stbir__small_float) ) )
- {
- filter_scale = ((STBIR_RENORM_TYPE)1.0) / total_filter;
- // scale them all
- for (i = 0; i <= e; i++)
- coeffs[i] = (float) ( coeffs[i] * filter_scale );
- }
- }
- ++contribs;
- coeffs += coefficient_width;
- }
- // if we have a rational for the scale, we can exploit the polyphaseness to not calculate
- // most of the coefficients, so we copy them here
- if ( polyphase )
- {
- stbir__contributors * prev_contribs = contributors;
- stbir__contributors * cur_contribs = contributors + numerator;
- for( n = numerator ; n < num_contributors ; n++ )
- {
- cur_contribs->n0 = prev_contribs->n0 + denominator;
- cur_contribs->n1 = prev_contribs->n1 + denominator;
- ++cur_contribs;
- ++prev_contribs;
- }
- stbir_overlapping_memcpy( coefficient_group + numerator * coefficient_width, coefficient_group, ( num_contributors - numerator ) * coefficient_width * sizeof( coeffs[ 0 ] ) );
- }
- coeffs = coefficient_group;
- contribs = contributors;
- for (n = 0; n < num_contributors; n++)
- {
- int i;
- // in zero edge mode, just remove out of bounds contribs completely (since their weights are accounted for now)
- if ( edge == STBIR_EDGE_ZERO )
- {
- // shrink the right side if necessary
- if ( contribs->n1 > input_last_n1 )
- contribs->n1 = input_last_n1;
- // shrink the left side
- if ( contribs->n0 < 0 )
- {
- int j, left, skips = 0;
- skips = -contribs->n0;
- contribs->n0 = 0;
- // now move down the weights
- left = contribs->n1 - contribs->n0 + 1;
- if ( left > 0 )
- {
- for( j = 0 ; j < left ; j++ )
- coeffs[ j ] = coeffs[ j + skips ];
- }
- }
- }
- else if ( ( edge == STBIR_EDGE_CLAMP ) || ( edge == STBIR_EDGE_REFLECT ) )
- {
- // for clamp and reflect, calculate the true inbounds position (based on edge type) and just add that to the existing weight
- // right hand side first
- if ( contribs->n1 > input_last_n1 )
- {
- int start = contribs->n0;
- int endi = contribs->n1;
- contribs->n1 = input_last_n1;
- for( i = input_size; i <= endi; i++ )
- stbir__insert_coeff( contribs, coeffs, stbir__edge_wrap_slow[edge]( i, input_size ), coeffs[i-start], coefficient_width );
- }
- // now check left hand edge
- if ( contribs->n0 < 0 )
- {
- int save_n0;
- float save_n0_coeff;
- float * c = coeffs - ( contribs->n0 + 1 );
- // reinsert the coeffs with it reflected or clamped (insert accumulates, if the coeffs exist)
- for( i = -1 ; i > contribs->n0 ; i-- )
- stbir__insert_coeff( contribs, coeffs, stbir__edge_wrap_slow[edge]( i, input_size ), *c--, coefficient_width );
- save_n0 = contribs->n0;
- save_n0_coeff = c[0]; // save it, since we didn't do the final one (i==n0), because there might be too many coeffs to hold (before we resize)!
- // now slide all the coeffs down (since we have accumulated them in the positive contribs) and reset the first contrib
- contribs->n0 = 0;
- for(i = 0 ; i <= contribs->n1 ; i++ )
- coeffs[i] = coeffs[i-save_n0];
- // now that we have shrunk down the contribs, we insert the first one safely
- stbir__insert_coeff( contribs, coeffs, stbir__edge_wrap_slow[edge]( save_n0, input_size ), save_n0_coeff, coefficient_width );
- }
- }
- if ( contribs->n0 <= contribs->n1 )
- {
- int diff = contribs->n1 - contribs->n0 + 1;
- while ( diff && ( coeffs[ diff-1 ] == 0.0f ) )
- --diff;
- contribs->n1 = contribs->n0 + diff - 1;
- if ( contribs->n0 <= contribs->n1 )
- {
- if ( contribs->n0 < lowest )
- lowest = contribs->n0;
- if ( contribs->n1 > highest )
- highest = contribs->n1;
- if ( diff > widest )
- widest = diff;
- }
- // re-zero out unused coefficients (if any)
- for( i = diff ; i < coefficient_width ; i++ )
- coeffs[i] = 0.0f;
- }
- ++contribs;
- coeffs += coefficient_width;
- }
- filter_info->lowest = lowest;
- filter_info->highest = highest;
- filter_info->widest = widest;
- }
- #undef STBIR_RENORM_TYPE
- static int stbir__pack_coefficients( int num_contributors, stbir__contributors* contributors, float * coefficents, int coefficient_width, int widest, int row0, int row1 )
- {
- #define STBIR_MOVE_1( dest, src ) { STBIR_NO_UNROLL(dest); ((stbir_uint32*)(dest))[0] = ((stbir_uint32*)(src))[0]; }
- #define STBIR_MOVE_2( dest, src ) { STBIR_NO_UNROLL(dest); ((stbir_uint64*)(dest))[0] = ((stbir_uint64*)(src))[0]; }
- #ifdef STBIR_SIMD
- #define STBIR_MOVE_4( dest, src ) { stbir__simdf t; STBIR_NO_UNROLL(dest); stbir__simdf_load( t, src ); stbir__simdf_store( dest, t ); }
- #else
- #define STBIR_MOVE_4( dest, src ) { STBIR_NO_UNROLL(dest); ((stbir_uint64*)(dest))[0] = ((stbir_uint64*)(src))[0]; ((stbir_uint64*)(dest))[1] = ((stbir_uint64*)(src))[1]; }
- #endif
- int row_end = row1 + 1;
- STBIR__UNUSED( row0 ); // only used in an assert
- if ( coefficient_width != widest )
- {
- float * pc = coefficents;
- float * coeffs = coefficents;
- float * pc_end = coefficents + num_contributors * widest;
- switch( widest )
- {
- case 1:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_1( pc, coeffs );
- ++pc;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 2:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_2( pc, coeffs );
- pc += 2;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 3:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_2( pc, coeffs );
- STBIR_MOVE_1( pc+2, coeffs+2 );
- pc += 3;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 4:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- pc += 4;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 5:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_1( pc+4, coeffs+4 );
- pc += 5;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 6:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_2( pc+4, coeffs+4 );
- pc += 6;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 7:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_2( pc+4, coeffs+4 );
- STBIR_MOVE_1( pc+6, coeffs+6 );
- pc += 7;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 8:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_4( pc+4, coeffs+4 );
- pc += 8;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 9:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_4( pc+4, coeffs+4 );
- STBIR_MOVE_1( pc+8, coeffs+8 );
- pc += 9;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 10:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_4( pc+4, coeffs+4 );
- STBIR_MOVE_2( pc+8, coeffs+8 );
- pc += 10;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 11:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_4( pc+4, coeffs+4 );
- STBIR_MOVE_2( pc+8, coeffs+8 );
- STBIR_MOVE_1( pc+10, coeffs+10 );
- pc += 11;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- case 12:
- STBIR_NO_UNROLL_LOOP_START
- do {
- STBIR_MOVE_4( pc, coeffs );
- STBIR_MOVE_4( pc+4, coeffs+4 );
- STBIR_MOVE_4( pc+8, coeffs+8 );
- pc += 12;
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- default:
- STBIR_NO_UNROLL_LOOP_START
- do {
- float * copy_end = pc + widest - 4;
- float * c = coeffs;
- do {
- STBIR_NO_UNROLL( pc );
- STBIR_MOVE_4( pc, c );
- pc += 4;
- c += 4;
- } while ( pc <= copy_end );
- copy_end += 4;
- STBIR_NO_UNROLL_LOOP_START
- while ( pc < copy_end )
- {
- STBIR_MOVE_1( pc, c );
- ++pc; ++c;
- }
- coeffs += coefficient_width;
- } while ( pc < pc_end );
- break;
- }
- }
- // some horizontal routines read one float off the end (which is then masked off), so put in a sentinal so we don't read an snan or denormal
- coefficents[ widest * num_contributors ] = 8888.0f;
- // the minimum we might read for unrolled filters widths is 12. So, we need to
- // make sure we never read outside the decode buffer, by possibly moving
- // the sample area back into the scanline, and putting zeros weights first.
- // we start on the right edge and check until we're well past the possible
- // clip area (2*widest).
- {
- stbir__contributors * contribs = contributors + num_contributors - 1;
- float * coeffs = coefficents + widest * ( num_contributors - 1 );
- // go until no chance of clipping (this is usually less than 8 lops)
- while ( ( contribs >= contributors ) && ( ( contribs->n0 + widest*2 ) >= row_end ) )
- {
- // might we clip??
- if ( ( contribs->n0 + widest ) > row_end )
- {
- int stop_range = widest;
- // if range is larger than 12, it will be handled by generic loops that can terminate on the exact length
- // of this contrib n1, instead of a fixed widest amount - so calculate this
- if ( widest > 12 )
- {
- int mod;
- // how far will be read in the n_coeff loop (which depends on the widest count mod4);
- mod = widest & 3;
- stop_range = ( ( ( contribs->n1 - contribs->n0 + 1 ) - mod + 3 ) & ~3 ) + mod;
- // the n_coeff loops do a minimum amount of coeffs, so factor that in!
- if ( stop_range < ( 8 + mod ) ) stop_range = 8 + mod;
- }
- // now see if we still clip with the refined range
- if ( ( contribs->n0 + stop_range ) > row_end )
- {
- int new_n0 = row_end - stop_range;
- int num = contribs->n1 - contribs->n0 + 1;
- int backup = contribs->n0 - new_n0;
- float * from_co = coeffs + num - 1;
- float * to_co = from_co + backup;
- STBIR_ASSERT( ( new_n0 >= row0 ) && ( new_n0 < contribs->n0 ) );
- // move the coeffs over
- while( num )
- {
- *to_co-- = *from_co--;
- --num;
- }
- // zero new positions
- while ( to_co >= coeffs )
- *to_co-- = 0;
- // set new start point
- contribs->n0 = new_n0;
- if ( widest > 12 )
- {
- int mod;
- // how far will be read in the n_coeff loop (which depends on the widest count mod4);
- mod = widest & 3;
- stop_range = ( ( ( contribs->n1 - contribs->n0 + 1 ) - mod + 3 ) & ~3 ) + mod;
- // the n_coeff loops do a minimum amount of coeffs, so factor that in!
- if ( stop_range < ( 8 + mod ) ) stop_range = 8 + mod;
- }
- }
- }
- --contribs;
- coeffs -= widest;
- }
- }
- return widest;
- #undef STBIR_MOVE_1
- #undef STBIR_MOVE_2
- #undef STBIR_MOVE_4
- }
- static void stbir__calculate_filters( stbir__sampler * samp, stbir__sampler * other_axis_for_pivot, void * user_data STBIR_ONLY_PROFILE_BUILD_GET_INFO )
- {
- int n;
- float scale = samp->scale_info.scale;
- stbir__kernel_callback * kernel = samp->filter_kernel;
- stbir__support_callback * support = samp->filter_support;
- float inv_scale = samp->scale_info.inv_scale;
- int input_full_size = samp->scale_info.input_full_size;
- int gather_num_contributors = samp->num_contributors;
- stbir__contributors* gather_contributors = samp->contributors;
- float * gather_coeffs = samp->coefficients;
- int gather_coefficient_width = samp->coefficient_width;
- switch ( samp->is_gather )
- {
- case 1: // gather upsample
- {
- float out_pixels_radius = support(inv_scale,user_data) * scale;
- stbir__calculate_coefficients_for_gather_upsample( out_pixels_radius, kernel, &samp->scale_info, gather_num_contributors, gather_contributors, gather_coeffs, gather_coefficient_width, samp->edge, user_data );
- STBIR_PROFILE_BUILD_START( cleanup );
- stbir__cleanup_gathered_coefficients( samp->edge, &samp->extent_info, &samp->scale_info, gather_num_contributors, gather_contributors, gather_coeffs, gather_coefficient_width );
- STBIR_PROFILE_BUILD_END( cleanup );
- }
- break;
- case 0: // scatter downsample (only on vertical)
- case 2: // gather downsample
- {
- float in_pixels_radius = support(scale,user_data) * inv_scale;
- int filter_pixel_margin = samp->filter_pixel_margin;
- int input_end = input_full_size + filter_pixel_margin;
- // if this is a scatter, we do a downsample gather to get the coeffs, and then pivot after
- if ( !samp->is_gather )
- {
- // check if we are using the same gather downsample on the horizontal as this vertical,
- // if so, then we don't have to generate them, we can just pivot from the horizontal.
- if ( other_axis_for_pivot )
- {
- gather_contributors = other_axis_for_pivot->contributors;
- gather_coeffs = other_axis_for_pivot->coefficients;
- gather_coefficient_width = other_axis_for_pivot->coefficient_width;
- gather_num_contributors = other_axis_for_pivot->num_contributors;
- samp->extent_info.lowest = other_axis_for_pivot->extent_info.lowest;
- samp->extent_info.highest = other_axis_for_pivot->extent_info.highest;
- samp->extent_info.widest = other_axis_for_pivot->extent_info.widest;
- goto jump_right_to_pivot;
- }
- gather_contributors = samp->gather_prescatter_contributors;
- gather_coeffs = samp->gather_prescatter_coefficients;
- gather_coefficient_width = samp->gather_prescatter_coefficient_width;
- gather_num_contributors = samp->gather_prescatter_num_contributors;
- }
- stbir__calculate_coefficients_for_gather_downsample( -filter_pixel_margin, input_end, in_pixels_radius, kernel, &samp->scale_info, gather_coefficient_width, gather_num_contributors, gather_contributors, gather_coeffs, user_data );
- STBIR_PROFILE_BUILD_START( cleanup );
- stbir__cleanup_gathered_coefficients( samp->edge, &samp->extent_info, &samp->scale_info, gather_num_contributors, gather_contributors, gather_coeffs, gather_coefficient_width );
- STBIR_PROFILE_BUILD_END( cleanup );
- if ( !samp->is_gather )
- {
- // if this is a scatter (vertical only), then we need to pivot the coeffs
- stbir__contributors * scatter_contributors;
- int highest_set;
- jump_right_to_pivot:
- STBIR_PROFILE_BUILD_START( pivot );
- highest_set = (-filter_pixel_margin) - 1;
- for (n = 0; n < gather_num_contributors; n++)
- {
- int k;
- int gn0 = gather_contributors->n0, gn1 = gather_contributors->n1;
- int scatter_coefficient_width = samp->coefficient_width;
- float * scatter_coeffs = samp->coefficients + ( gn0 + filter_pixel_margin ) * scatter_coefficient_width;
- float * g_coeffs = gather_coeffs;
- scatter_contributors = samp->contributors + ( gn0 + filter_pixel_margin );
- for (k = gn0 ; k <= gn1 ; k++ )
- {
- float gc = *g_coeffs++;
-
- // skip zero and denormals - must skip zeros to avoid adding coeffs beyond scatter_coefficient_width
- // (which happens when pivoting from horizontal, which might have dummy zeros)
- if ( ( ( gc >= stbir__small_float ) || ( gc <= -stbir__small_float ) ) )
- {
- if ( ( k > highest_set ) || ( scatter_contributors->n0 > scatter_contributors->n1 ) )
- {
- {
- // if we are skipping over several contributors, we need to clear the skipped ones
- stbir__contributors * clear_contributors = samp->contributors + ( highest_set + filter_pixel_margin + 1);
- while ( clear_contributors < scatter_contributors )
- {
- clear_contributors->n0 = 0;
- clear_contributors->n1 = -1;
- ++clear_contributors;
- }
- }
- scatter_contributors->n0 = n;
- scatter_contributors->n1 = n;
- scatter_coeffs[0] = gc;
- highest_set = k;
- }
- else
- {
- stbir__insert_coeff( scatter_contributors, scatter_coeffs, n, gc, scatter_coefficient_width );
- }
- STBIR_ASSERT( ( scatter_contributors->n1 - scatter_contributors->n0 + 1 ) <= scatter_coefficient_width );
- }
- ++scatter_contributors;
- scatter_coeffs += scatter_coefficient_width;
- }
- ++gather_contributors;
- gather_coeffs += gather_coefficient_width;
- }
- // now clear any unset contribs
- {
- stbir__contributors * clear_contributors = samp->contributors + ( highest_set + filter_pixel_margin + 1);
- stbir__contributors * end_contributors = samp->contributors + samp->num_contributors;
- while ( clear_contributors < end_contributors )
- {
- clear_contributors->n0 = 0;
- clear_contributors->n1 = -1;
- ++clear_contributors;
- }
- }
- STBIR_PROFILE_BUILD_END( pivot );
- }
- }
- break;
- }
- }
- //========================================================================================================
- // scanline decoders and encoders
- #define stbir__coder_min_num 1
- #define STB_IMAGE_RESIZE_DO_CODERS
- #include STBIR__HEADER_FILENAME
- #define stbir__decode_suffix BGRA
- #define stbir__decode_swizzle
- #define stbir__decode_order0 2
- #define stbir__decode_order1 1
- #define stbir__decode_order2 0
- #define stbir__decode_order3 3
- #define stbir__encode_order0 2
- #define stbir__encode_order1 1
- #define stbir__encode_order2 0
- #define stbir__encode_order3 3
- #define stbir__coder_min_num 4
- #define STB_IMAGE_RESIZE_DO_CODERS
- #include STBIR__HEADER_FILENAME
- #define stbir__decode_suffix ARGB
- #define stbir__decode_swizzle
- #define stbir__decode_order0 1
- #define stbir__decode_order1 2
- #define stbir__decode_order2 3
- #define stbir__decode_order3 0
- #define stbir__encode_order0 3
- #define stbir__encode_order1 0
- #define stbir__encode_order2 1
- #define stbir__encode_order3 2
- #define stbir__coder_min_num 4
- #define STB_IMAGE_RESIZE_DO_CODERS
- #include STBIR__HEADER_FILENAME
- #define stbir__decode_suffix ABGR
- #define stbir__decode_swizzle
- #define stbir__decode_order0 3
- #define stbir__decode_order1 2
- #define stbir__decode_order2 1
- #define stbir__decode_order3 0
- #define stbir__encode_order0 3
- #define stbir__encode_order1 2
- #define stbir__encode_order2 1
- #define stbir__encode_order3 0
- #define stbir__coder_min_num 4
- #define STB_IMAGE_RESIZE_DO_CODERS
- #include STBIR__HEADER_FILENAME
- #define stbir__decode_suffix AR
- #define stbir__decode_swizzle
- #define stbir__decode_order0 1
- #define stbir__decode_order1 0
- #define stbir__decode_order2 3
- #define stbir__decode_order3 2
- #define stbir__encode_order0 1
- #define stbir__encode_order1 0
- #define stbir__encode_order2 3
- #define stbir__encode_order3 2
- #define stbir__coder_min_num 2
- #define STB_IMAGE_RESIZE_DO_CODERS
- #include STBIR__HEADER_FILENAME
- // fancy alpha means we expand to keep both premultipied and non-premultiplied color channels
- static void stbir__fancy_alpha_weight_4ch( float * out_buffer, int width_times_channels )
- {
- float STBIR_STREAMOUT_PTR(*) out = out_buffer;
- float const * end_decode = out_buffer + ( width_times_channels / 4 ) * 7; // decode buffer aligned to end of out_buffer
- float STBIR_STREAMOUT_PTR(*) decode = (float*)end_decode - width_times_channels;
- // fancy alpha is stored internally as R G B A Rpm Gpm Bpm
- #ifdef STBIR_SIMD
- #ifdef STBIR_SIMD8
- decode += 16;
- STBIR_NO_UNROLL_LOOP_START
- while ( decode <= end_decode )
- {
- stbir__simdf8 d0,d1,a0,a1,p0,p1;
- STBIR_NO_UNROLL(decode);
- stbir__simdf8_load( d0, decode-16 );
- stbir__simdf8_load( d1, decode-16+8 );
- stbir__simdf8_0123to33333333( a0, d0 );
- stbir__simdf8_0123to33333333( a1, d1 );
- stbir__simdf8_mult( p0, a0, d0 );
- stbir__simdf8_mult( p1, a1, d1 );
- stbir__simdf8_bot4s( a0, d0, p0 );
- stbir__simdf8_bot4s( a1, d1, p1 );
- stbir__simdf8_top4s( d0, d0, p0 );
- stbir__simdf8_top4s( d1, d1, p1 );
- stbir__simdf8_store ( out, a0 );
- stbir__simdf8_store ( out+7, d0 );
- stbir__simdf8_store ( out+14, a1 );
- stbir__simdf8_store ( out+21, d1 );
- decode += 16;
- out += 28;
- }
- decode -= 16;
- #else
- decode += 8;
- STBIR_NO_UNROLL_LOOP_START
- while ( decode <= end_decode )
- {
- stbir__simdf d0,a0,d1,a1,p0,p1;
- STBIR_NO_UNROLL(decode);
- stbir__simdf_load( d0, decode-8 );
- stbir__simdf_load( d1, decode-8+4 );
- stbir__simdf_0123to3333( a0, d0 );
- stbir__simdf_0123to3333( a1, d1 );
- stbir__simdf_mult( p0, a0, d0 );
- stbir__simdf_mult( p1, a1, d1 );
- stbir__simdf_store ( out, d0 );
- stbir__simdf_store ( out+4, p0 );
- stbir__simdf_store ( out+7, d1 );
- stbir__simdf_store ( out+7+4, p1 );
- decode += 8;
- out += 14;
- }
- decode -= 8;
- #endif
- // might be one last odd pixel
- #ifdef STBIR_SIMD8
- STBIR_NO_UNROLL_LOOP_START
- while ( decode < end_decode )
- #else
- if ( decode < end_decode )
- #endif
- {
- stbir__simdf d,a,p;
- STBIR_NO_UNROLL(decode);
- stbir__simdf_load( d, decode );
- stbir__simdf_0123to3333( a, d );
- stbir__simdf_mult( p, a, d );
- stbir__simdf_store ( out, d );
- stbir__simdf_store ( out+4, p );
- decode += 4;
- out += 7;
- }
- #else
- while( decode < end_decode )
- {
- float r = decode[0], g = decode[1], b = decode[2], alpha = decode[3];
- out[0] = r;
- out[1] = g;
- out[2] = b;
- out[3] = alpha;
- out[4] = r * alpha;
- out[5] = g * alpha;
- out[6] = b * alpha;
- out += 7;
- decode += 4;
- }
- #endif
- }
- static void stbir__fancy_alpha_weight_2ch( float * out_buffer, int width_times_channels )
- {
- float STBIR_STREAMOUT_PTR(*) out = out_buffer;
- float const * end_decode = out_buffer + ( width_times_channels / 2 ) * 3;
- float STBIR_STREAMOUT_PTR(*) decode = (float*)end_decode - width_times_channels;
- // for fancy alpha, turns into: [X A Xpm][X A Xpm],etc
- #ifdef STBIR_SIMD
- decode += 8;
- if ( decode <= end_decode )
- {
- STBIR_NO_UNROLL_LOOP_START
- do {
- #ifdef STBIR_SIMD8
- stbir__simdf8 d0,a0,p0;
- STBIR_NO_UNROLL(decode);
- stbir__simdf8_load( d0, decode-8 );
- stbir__simdf8_0123to11331133( p0, d0 );
- stbir__simdf8_0123to00220022( a0, d0 );
- stbir__simdf8_mult( p0, p0, a0 );
- stbir__simdf_store2( out, stbir__if_simdf8_cast_to_simdf4( d0 ) );
- stbir__simdf_store( out+2, stbir__if_simdf8_cast_to_simdf4( p0 ) );
- stbir__simdf_store2h( out+3, stbir__if_simdf8_cast_to_simdf4( d0 ) );
- stbir__simdf_store2( out+6, stbir__simdf8_gettop4( d0 ) );
- stbir__simdf_store( out+8, stbir__simdf8_gettop4( p0 ) );
- stbir__simdf_store2h( out+9, stbir__simdf8_gettop4( d0 ) );
- #else
- stbir__simdf d0,a0,d1,a1,p0,p1;
- STBIR_NO_UNROLL(decode);
- stbir__simdf_load( d0, decode-8 );
- stbir__simdf_load( d1, decode-8+4 );
- stbir__simdf_0123to1133( p0, d0 );
- stbir__simdf_0123to1133( p1, d1 );
- stbir__simdf_0123to0022( a0, d0 );
- stbir__simdf_0123to0022( a1, d1 );
- stbir__simdf_mult( p0, p0, a0 );
- stbir__simdf_mult( p1, p1, a1 );
- stbir__simdf_store2( out, d0 );
- stbir__simdf_store( out+2, p0 );
- stbir__simdf_store2h( out+3, d0 );
- stbir__simdf_store2( out+6, d1 );
- stbir__simdf_store( out+8, p1 );
- stbir__simdf_store2h( out+9, d1 );
- #endif
- decode += 8;
- out += 12;
- } while ( decode <= end_decode );
- }
- decode -= 8;
- #endif
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode < end_decode )
- {
- float x = decode[0], y = decode[1];
- STBIR_SIMD_NO_UNROLL(decode);
- out[0] = x;
- out[1] = y;
- out[2] = x * y;
- out += 3;
- decode += 2;
- }
- }
- static void stbir__fancy_alpha_unweight_4ch( float * encode_buffer, int width_times_channels )
- {
- float STBIR_SIMD_STREAMOUT_PTR(*) encode = encode_buffer;
- float STBIR_SIMD_STREAMOUT_PTR(*) input = encode_buffer;
- float const * end_output = encode_buffer + width_times_channels;
- // fancy RGBA is stored internally as R G B A Rpm Gpm Bpm
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float alpha = input[3];
- #ifdef STBIR_SIMD
- stbir__simdf i,ia;
- STBIR_SIMD_NO_UNROLL(encode);
- if ( alpha < stbir__small_float )
- {
- stbir__simdf_load( i, input );
- stbir__simdf_store( encode, i );
- }
- else
- {
- stbir__simdf_load1frep4( ia, 1.0f / alpha );
- stbir__simdf_load( i, input+4 );
- stbir__simdf_mult( i, i, ia );
- stbir__simdf_store( encode, i );
- encode[3] = alpha;
- }
- #else
- if ( alpha < stbir__small_float )
- {
- encode[0] = input[0];
- encode[1] = input[1];
- encode[2] = input[2];
- }
- else
- {
- float ialpha = 1.0f / alpha;
- encode[0] = input[4] * ialpha;
- encode[1] = input[5] * ialpha;
- encode[2] = input[6] * ialpha;
- }
- encode[3] = alpha;
- #endif
- input += 7;
- encode += 4;
- } while ( encode < end_output );
- }
- // format: [X A Xpm][X A Xpm] etc
- static void stbir__fancy_alpha_unweight_2ch( float * encode_buffer, int width_times_channels )
- {
- float STBIR_SIMD_STREAMOUT_PTR(*) encode = encode_buffer;
- float STBIR_SIMD_STREAMOUT_PTR(*) input = encode_buffer;
- float const * end_output = encode_buffer + width_times_channels;
- do {
- float alpha = input[1];
- encode[0] = input[0];
- if ( alpha >= stbir__small_float )
- encode[0] = input[2] / alpha;
- encode[1] = alpha;
- input += 3;
- encode += 2;
- } while ( encode < end_output );
- }
- static void stbir__simple_alpha_weight_4ch( float * decode_buffer, int width_times_channels )
- {
- float STBIR_STREAMOUT_PTR(*) decode = decode_buffer;
- float const * end_decode = decode_buffer + width_times_channels;
- #ifdef STBIR_SIMD
- {
- decode += 2 * stbir__simdfX_float_count;
- STBIR_NO_UNROLL_LOOP_START
- while ( decode <= end_decode )
- {
- stbir__simdfX d0,a0,d1,a1;
- STBIR_NO_UNROLL(decode);
- stbir__simdfX_load( d0, decode-2*stbir__simdfX_float_count );
- stbir__simdfX_load( d1, decode-2*stbir__simdfX_float_count+stbir__simdfX_float_count );
- stbir__simdfX_aaa1( a0, d0, STBIR_onesX );
- stbir__simdfX_aaa1( a1, d1, STBIR_onesX );
- stbir__simdfX_mult( d0, d0, a0 );
- stbir__simdfX_mult( d1, d1, a1 );
- stbir__simdfX_store ( decode-2*stbir__simdfX_float_count, d0 );
- stbir__simdfX_store ( decode-2*stbir__simdfX_float_count+stbir__simdfX_float_count, d1 );
- decode += 2 * stbir__simdfX_float_count;
- }
- decode -= 2 * stbir__simdfX_float_count;
- // few last pixels remnants
- #ifdef STBIR_SIMD8
- STBIR_NO_UNROLL_LOOP_START
- while ( decode < end_decode )
- #else
- if ( decode < end_decode )
- #endif
- {
- stbir__simdf d,a;
- stbir__simdf_load( d, decode );
- stbir__simdf_aaa1( a, d, STBIR__CONSTF(STBIR_ones) );
- stbir__simdf_mult( d, d, a );
- stbir__simdf_store ( decode, d );
- decode += 4;
- }
- }
- #else
- while( decode < end_decode )
- {
- float alpha = decode[3];
- decode[0] *= alpha;
- decode[1] *= alpha;
- decode[2] *= alpha;
- decode += 4;
- }
- #endif
- }
- static void stbir__simple_alpha_weight_2ch( float * decode_buffer, int width_times_channels )
- {
- float STBIR_STREAMOUT_PTR(*) decode = decode_buffer;
- float const * end_decode = decode_buffer + width_times_channels;
- #ifdef STBIR_SIMD
- decode += 2 * stbir__simdfX_float_count;
- STBIR_NO_UNROLL_LOOP_START
- while ( decode <= end_decode )
- {
- stbir__simdfX d0,a0,d1,a1;
- STBIR_NO_UNROLL(decode);
- stbir__simdfX_load( d0, decode-2*stbir__simdfX_float_count );
- stbir__simdfX_load( d1, decode-2*stbir__simdfX_float_count+stbir__simdfX_float_count );
- stbir__simdfX_a1a1( a0, d0, STBIR_onesX );
- stbir__simdfX_a1a1( a1, d1, STBIR_onesX );
- stbir__simdfX_mult( d0, d0, a0 );
- stbir__simdfX_mult( d1, d1, a1 );
- stbir__simdfX_store ( decode-2*stbir__simdfX_float_count, d0 );
- stbir__simdfX_store ( decode-2*stbir__simdfX_float_count+stbir__simdfX_float_count, d1 );
- decode += 2 * stbir__simdfX_float_count;
- }
- decode -= 2 * stbir__simdfX_float_count;
- #endif
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode < end_decode )
- {
- float alpha = decode[1];
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0] *= alpha;
- decode += 2;
- }
- }
- static void stbir__simple_alpha_unweight_4ch( float * encode_buffer, int width_times_channels )
- {
- float STBIR_SIMD_STREAMOUT_PTR(*) encode = encode_buffer;
- float const * end_output = encode_buffer + width_times_channels;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float alpha = encode[3];
- #ifdef STBIR_SIMD
- stbir__simdf i,ia;
- STBIR_SIMD_NO_UNROLL(encode);
- if ( alpha >= stbir__small_float )
- {
- stbir__simdf_load1frep4( ia, 1.0f / alpha );
- stbir__simdf_load( i, encode );
- stbir__simdf_mult( i, i, ia );
- stbir__simdf_store( encode, i );
- encode[3] = alpha;
- }
- #else
- if ( alpha >= stbir__small_float )
- {
- float ialpha = 1.0f / alpha;
- encode[0] *= ialpha;
- encode[1] *= ialpha;
- encode[2] *= ialpha;
- }
- #endif
- encode += 4;
- } while ( encode < end_output );
- }
- static void stbir__simple_alpha_unweight_2ch( float * encode_buffer, int width_times_channels )
- {
- float STBIR_SIMD_STREAMOUT_PTR(*) encode = encode_buffer;
- float const * end_output = encode_buffer + width_times_channels;
- do {
- float alpha = encode[1];
- if ( alpha >= stbir__small_float )
- encode[0] /= alpha;
- encode += 2;
- } while ( encode < end_output );
- }
- // only used in RGB->BGR or BGR->RGB
- static void stbir__simple_flip_3ch( float * decode_buffer, int width_times_channels )
- {
- float STBIR_STREAMOUT_PTR(*) decode = decode_buffer;
- float const * end_decode = decode_buffer + width_times_channels;
- #ifdef STBIR_SIMD
- #ifdef stbir__simdf_swiz2 // do we have two argument swizzles?
- end_decode -= 12;
- STBIR_NO_UNROLL_LOOP_START
- while( decode <= end_decode )
- {
- // on arm64 8 instructions, no overlapping stores
- stbir__simdf a,b,c,na,nb;
- STBIR_SIMD_NO_UNROLL(decode);
- stbir__simdf_load( a, decode );
- stbir__simdf_load( b, decode+4 );
- stbir__simdf_load( c, decode+8 );
- na = stbir__simdf_swiz2( a, b, 2, 1, 0, 5 );
- b = stbir__simdf_swiz2( a, b, 4, 3, 6, 7 );
- nb = stbir__simdf_swiz2( b, c, 0, 1, 4, 3 );
- c = stbir__simdf_swiz2( b, c, 2, 7, 6, 5 );
- stbir__simdf_store( decode, na );
- stbir__simdf_store( decode+4, nb );
- stbir__simdf_store( decode+8, c );
- decode += 12;
- }
- end_decode += 12;
- #else
- end_decode -= 24;
- STBIR_NO_UNROLL_LOOP_START
- while( decode <= end_decode )
- {
- // 26 instructions on x64
- stbir__simdf a,b,c,d,e,f,g;
- float i21, i23;
- STBIR_SIMD_NO_UNROLL(decode);
- stbir__simdf_load( a, decode );
- stbir__simdf_load( b, decode+3 );
- stbir__simdf_load( c, decode+6 );
- stbir__simdf_load( d, decode+9 );
- stbir__simdf_load( e, decode+12 );
- stbir__simdf_load( f, decode+15 );
- stbir__simdf_load( g, decode+18 );
- a = stbir__simdf_swiz( a, 2, 1, 0, 3 );
- b = stbir__simdf_swiz( b, 2, 1, 0, 3 );
- c = stbir__simdf_swiz( c, 2, 1, 0, 3 );
- d = stbir__simdf_swiz( d, 2, 1, 0, 3 );
- e = stbir__simdf_swiz( e, 2, 1, 0, 3 );
- f = stbir__simdf_swiz( f, 2, 1, 0, 3 );
- g = stbir__simdf_swiz( g, 2, 1, 0, 3 );
- // stores overlap, need to be in order,
- stbir__simdf_store( decode, a );
- i21 = decode[21];
- stbir__simdf_store( decode+3, b );
- i23 = decode[23];
- stbir__simdf_store( decode+6, c );
- stbir__simdf_store( decode+9, d );
- stbir__simdf_store( decode+12, e );
- stbir__simdf_store( decode+15, f );
- stbir__simdf_store( decode+18, g );
- decode[21] = i23;
- decode[23] = i21;
- decode += 24;
- }
- end_decode += 24;
- #endif
- #else
- end_decode -= 12;
- STBIR_NO_UNROLL_LOOP_START
- while( decode <= end_decode )
- {
- // 16 instructions
- float t0,t1,t2,t3;
- STBIR_NO_UNROLL(decode);
- t0 = decode[0]; t1 = decode[3]; t2 = decode[6]; t3 = decode[9];
- decode[0] = decode[2]; decode[3] = decode[5]; decode[6] = decode[8]; decode[9] = decode[11];
- decode[2] = t0; decode[5] = t1; decode[8] = t2; decode[11] = t3;
- decode += 12;
- }
- end_decode += 12;
- #endif
- STBIR_NO_UNROLL_LOOP_START
- while( decode < end_decode )
- {
- float t = decode[0];
- STBIR_NO_UNROLL(decode);
- decode[0] = decode[2];
- decode[2] = t;
- decode += 3;
- }
- }
- static void stbir__decode_scanline(stbir__info const * stbir_info, int n, float * output_buffer STBIR_ONLY_PROFILE_GET_SPLIT_INFO )
- {
- int channels = stbir_info->channels;
- int effective_channels = stbir_info->effective_channels;
- int input_sample_in_bytes = stbir__type_size[stbir_info->input_type] * channels;
- stbir_edge edge_horizontal = stbir_info->horizontal.edge;
- stbir_edge edge_vertical = stbir_info->vertical.edge;
- int row = stbir__edge_wrap(edge_vertical, n, stbir_info->vertical.scale_info.input_full_size);
- const void* input_plane_data = ( (char *) stbir_info->input_data ) + (size_t)row * (size_t) stbir_info->input_stride_bytes;
- stbir__span const * spans = stbir_info->scanline_extents.spans;
- float * full_decode_buffer = output_buffer - stbir_info->scanline_extents.conservative.n0 * effective_channels;
- float * last_decoded = 0;
- // if we are on edge_zero, and we get in here with an out of bounds n, then the calculate filters has failed
- STBIR_ASSERT( !(edge_vertical == STBIR_EDGE_ZERO && (n < 0 || n >= stbir_info->vertical.scale_info.input_full_size)) );
- do
- {
- float * decode_buffer;
- void const * input_data;
- float * end_decode;
- int width_times_channels;
- int width;
- if ( spans->n1 < spans->n0 )
- break;
- width = spans->n1 + 1 - spans->n0;
- decode_buffer = full_decode_buffer + spans->n0 * effective_channels;
- end_decode = full_decode_buffer + ( spans->n1 + 1 ) * effective_channels;
- width_times_channels = width * channels;
- // read directly out of input plane by default
- input_data = ( (char*)input_plane_data ) + spans->pixel_offset_for_input * input_sample_in_bytes;
- // if we have an input callback, call it to get the input data
- if ( stbir_info->in_pixels_cb )
- {
- // call the callback with a temp buffer (that they can choose to use or not). the temp is just right aligned memory in the decode_buffer itself
- input_data = stbir_info->in_pixels_cb( ( (char*) end_decode ) - ( width * input_sample_in_bytes ) + ( ( stbir_info->input_type != STBIR_TYPE_FLOAT ) ? ( sizeof(float)*STBIR_INPUT_CALLBACK_PADDING ) : 0 ), input_plane_data, width, spans->pixel_offset_for_input, row, stbir_info->user_data );
- }
- STBIR_PROFILE_START( decode );
- // convert the pixels info the float decode_buffer, (we index from end_decode, so that when channels<effective_channels, we are right justified in the buffer)
- last_decoded = stbir_info->decode_pixels( (float*)end_decode - width_times_channels, width_times_channels, input_data );
- STBIR_PROFILE_END( decode );
- if (stbir_info->alpha_weight)
- {
- STBIR_PROFILE_START( alpha );
- stbir_info->alpha_weight( decode_buffer, width_times_channels );
- STBIR_PROFILE_END( alpha );
- }
- ++spans;
- } while ( spans <= ( &stbir_info->scanline_extents.spans[1] ) );
- // handle the edge_wrap filter (all other types are handled back out at the calculate_filter stage)
- // basically the idea here is that if we have the whole scanline in memory, we don't redecode the
- // wrapped edge pixels, and instead just memcpy them from the scanline into the edge positions
- if ( ( edge_horizontal == STBIR_EDGE_WRAP ) && ( stbir_info->scanline_extents.edge_sizes[0] | stbir_info->scanline_extents.edge_sizes[1] ) )
- {
- // this code only runs if we're in edge_wrap, and we're doing the entire scanline
- int e, start_x[2];
- int input_full_size = stbir_info->horizontal.scale_info.input_full_size;
- start_x[0] = -stbir_info->scanline_extents.edge_sizes[0]; // left edge start x
- start_x[1] = input_full_size; // right edge
- for( e = 0; e < 2 ; e++ )
- {
- // do each margin
- int margin = stbir_info->scanline_extents.edge_sizes[e];
- if ( margin )
- {
- int x = start_x[e];
- float * marg = full_decode_buffer + x * effective_channels;
- float const * src = full_decode_buffer + stbir__edge_wrap(edge_horizontal, x, input_full_size) * effective_channels;
- STBIR_MEMCPY( marg, src, margin * effective_channels * sizeof(float) );
- if ( e == 1 ) last_decoded = marg + margin * effective_channels;
- }
- }
- }
-
- // some of the horizontal gathers read one float off the edge (which is masked out), but we force a zero here to make sure no NaNs leak in
- // (we can't pre-zero it, because the input callback can use that area as padding)
- last_decoded[0] = 0.0f;
- // we clear this extra float, because the final output pixel filter kernel might have used one less coeff than the max filter width
- // when this happens, we do read that pixel from the input, so it too could be Nan, so just zero an extra one.
- // this fits because each scanline is padded by three floats (STBIR_INPUT_CALLBACK_PADDING)
- last_decoded[1] = 0.0f;
- }
- //=================
- // Do 1 channel horizontal routines
- #ifdef STBIR_SIMD
- #define stbir__1_coeff_only() \
- stbir__simdf tot,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1( c, hc ); \
- stbir__simdf_mult1_mem( tot, c, decode );
- #define stbir__2_coeff_only() \
- stbir__simdf tot,c,d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2z( c, hc ); \
- stbir__simdf_load2( d, decode ); \
- stbir__simdf_mult( tot, c, d ); \
- stbir__simdf_0123to1230( c, tot ); \
- stbir__simdf_add1( tot, tot, c );
- #define stbir__3_coeff_only() \
- stbir__simdf tot,c,t; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( c, hc ); \
- stbir__simdf_mult_mem( tot, c, decode ); \
- stbir__simdf_0123to1230( c, tot ); \
- stbir__simdf_0123to2301( t, tot ); \
- stbir__simdf_add1( tot, tot, c ); \
- stbir__simdf_add1( tot, tot, t );
- #define stbir__store_output_tiny() \
- stbir__simdf_store1( output, tot ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 1;
- #define stbir__4_coeff_start() \
- stbir__simdf tot,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( c, hc ); \
- stbir__simdf_mult_mem( tot, c, decode ); \
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( c, hc + (ofs) ); \
- stbir__simdf_madd_mem( tot, tot, c, decode+(ofs) );
- #define stbir__1_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- stbir__simdf_load1z( c, hc + (ofs) ); \
- stbir__simdf_load1( d, decode + (ofs) ); \
- stbir__simdf_madd( tot, tot, d, c ); }
- #define stbir__2_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- stbir__simdf_load2z( c, hc+(ofs) ); \
- stbir__simdf_load2( d, decode+(ofs) ); \
- stbir__simdf_madd( tot, tot, d, c ); }
- #define stbir__3_coeff_setup() \
- stbir__simdf mask; \
- stbir__simdf_load( mask, STBIR_mask + 3 );
- #define stbir__3_coeff_remnant( ofs ) \
- stbir__simdf_load( c, hc+(ofs) ); \
- stbir__simdf_and( c, c, mask ); \
- stbir__simdf_madd_mem( tot, tot, c, decode+(ofs) );
- #define stbir__store_output() \
- stbir__simdf_0123to2301( c, tot ); \
- stbir__simdf_add( tot, tot, c ); \
- stbir__simdf_0123to1230( c, tot ); \
- stbir__simdf_add1( tot, tot, c ); \
- stbir__simdf_store1( output, tot ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 1;
- #else
- #define stbir__1_coeff_only() \
- float tot; \
- tot = decode[0]*hc[0];
- #define stbir__2_coeff_only() \
- float tot; \
- tot = decode[0] * hc[0]; \
- tot += decode[1] * hc[1];
- #define stbir__3_coeff_only() \
- float tot; \
- tot = decode[0] * hc[0]; \
- tot += decode[1] * hc[1]; \
- tot += decode[2] * hc[2];
- #define stbir__store_output_tiny() \
- output[0] = tot; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 1;
- #define stbir__4_coeff_start() \
- float tot0,tot1,tot2,tot3; \
- tot0 = decode[0] * hc[0]; \
- tot1 = decode[1] * hc[1]; \
- tot2 = decode[2] * hc[2]; \
- tot3 = decode[3] * hc[3];
- #define stbir__4_coeff_continue_from_4( ofs ) \
- tot0 += decode[0+(ofs)] * hc[0+(ofs)]; \
- tot1 += decode[1+(ofs)] * hc[1+(ofs)]; \
- tot2 += decode[2+(ofs)] * hc[2+(ofs)]; \
- tot3 += decode[3+(ofs)] * hc[3+(ofs)];
- #define stbir__1_coeff_remnant( ofs ) \
- tot0 += decode[0+(ofs)] * hc[0+(ofs)];
- #define stbir__2_coeff_remnant( ofs ) \
- tot0 += decode[0+(ofs)] * hc[0+(ofs)]; \
- tot1 += decode[1+(ofs)] * hc[1+(ofs)]; \
- #define stbir__3_coeff_remnant( ofs ) \
- tot0 += decode[0+(ofs)] * hc[0+(ofs)]; \
- tot1 += decode[1+(ofs)] * hc[1+(ofs)]; \
- tot2 += decode[2+(ofs)] * hc[2+(ofs)];
- #define stbir__store_output() \
- output[0] = (tot0+tot2)+(tot1+tot3); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 1;
- #endif
- #define STBIR__horizontal_channels 1
- #define STB_IMAGE_RESIZE_DO_HORIZONTALS
- #include STBIR__HEADER_FILENAME
- //=================
- // Do 2 channel horizontal routines
- #ifdef STBIR_SIMD
- #define stbir__1_coeff_only() \
- stbir__simdf tot,c,d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1z( c, hc ); \
- stbir__simdf_0123to0011( c, c ); \
- stbir__simdf_load2( d, decode ); \
- stbir__simdf_mult( tot, d, c );
- #define stbir__2_coeff_only() \
- stbir__simdf tot,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( c, hc ); \
- stbir__simdf_0123to0011( c, c ); \
- stbir__simdf_mult_mem( tot, c, decode );
- #define stbir__3_coeff_only() \
- stbir__simdf tot,c,cs,d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_mult_mem( tot, c, decode ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_load2z( d, decode+4 ); \
- stbir__simdf_madd( tot, tot, d, c );
- #define stbir__store_output_tiny() \
- stbir__simdf_0123to2301( c, tot ); \
- stbir__simdf_add( tot, tot, c ); \
- stbir__simdf_store2( output, tot ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 2;
- #ifdef STBIR_SIMD8
- #define stbir__4_coeff_start() \
- stbir__simdf8 tot0,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc ); \
- stbir__simdf8_0123to00112233( c, cs ); \
- stbir__simdf8_mult_mem( tot0, c, decode );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00112233( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*2 );
- #define stbir__1_coeff_remnant( ofs ) \
- { stbir__simdf t,d; \
- stbir__simdf_load1z( t, hc + (ofs) ); \
- stbir__simdf_load2( d, decode + (ofs) * 2 ); \
- stbir__simdf_0123to0011( t, t ); \
- stbir__simdf_mult( t, t, d ); \
- stbir__simdf8_add4( tot0, tot0, t ); }
-
- #define stbir__2_coeff_remnant( ofs ) \
- { stbir__simdf t; \
- stbir__simdf_load2( t, hc + (ofs) ); \
- stbir__simdf_0123to0011( t, t ); \
- stbir__simdf_mult_mem( t, t, decode+(ofs)*2 ); \
- stbir__simdf8_add4( tot0, tot0, t ); }
- #define stbir__3_coeff_remnant( ofs ) \
- { stbir__simdf8 d; \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00112233( c, cs ); \
- stbir__simdf8_load6z( d, decode+(ofs)*2 ); \
- stbir__simdf8_madd( tot0, tot0, c, d ); }
- #define stbir__store_output() \
- { stbir__simdf t,d; \
- stbir__simdf8_add4halves( t, stbir__if_simdf8_cast_to_simdf4(tot0), tot0 ); \
- stbir__simdf_0123to2301( d, t ); \
- stbir__simdf_add( t, t, d ); \
- stbir__simdf_store2( output, t ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 2; }
- #else
- #define stbir__4_coeff_start() \
- stbir__simdf tot0,tot1,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_0123to2233( c, cs ); \
- stbir__simdf_mult_mem( tot1, c, decode+4 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*2 ); \
- stbir__simdf_0123to2233( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*2+4 );
- #define stbir__1_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- stbir__simdf_load1z( cs, hc + (ofs) ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_load2( d, decode + (ofs) * 2 ); \
- stbir__simdf_madd( tot0, tot0, d, c ); }
- #define stbir__2_coeff_remnant( ofs ) \
- stbir__simdf_load2( cs, hc + (ofs) ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*2 );
- #define stbir__3_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0011( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*2 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_load2z( d, decode + (ofs) * 2 + 4 ); \
- stbir__simdf_madd( tot1, tot1, d, c ); }
- #define stbir__store_output() \
- stbir__simdf_add( tot0, tot0, tot1 ); \
- stbir__simdf_0123to2301( c, tot0 ); \
- stbir__simdf_add( tot0, tot0, c ); \
- stbir__simdf_store2( output, tot0 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 2;
- #endif
- #else
- #define stbir__1_coeff_only() \
- float tota,totb,c; \
- c = hc[0]; \
- tota = decode[0]*c; \
- totb = decode[1]*c;
- #define stbir__2_coeff_only() \
- float tota,totb,c; \
- c = hc[0]; \
- tota = decode[0]*c; \
- totb = decode[1]*c; \
- c = hc[1]; \
- tota += decode[2]*c; \
- totb += decode[3]*c;
- // this weird order of add matches the simd
- #define stbir__3_coeff_only() \
- float tota,totb,c; \
- c = hc[0]; \
- tota = decode[0]*c; \
- totb = decode[1]*c; \
- c = hc[2]; \
- tota += decode[4]*c; \
- totb += decode[5]*c; \
- c = hc[1]; \
- tota += decode[2]*c; \
- totb += decode[3]*c;
- #define stbir__store_output_tiny() \
- output[0] = tota; \
- output[1] = totb; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 2;
- #define stbir__4_coeff_start() \
- float tota0,tota1,tota2,tota3,totb0,totb1,totb2,totb3,c; \
- c = hc[0]; \
- tota0 = decode[0]*c; \
- totb0 = decode[1]*c; \
- c = hc[1]; \
- tota1 = decode[2]*c; \
- totb1 = decode[3]*c; \
- c = hc[2]; \
- tota2 = decode[4]*c; \
- totb2 = decode[5]*c; \
- c = hc[3]; \
- tota3 = decode[6]*c; \
- totb3 = decode[7]*c;
- #define stbir__4_coeff_continue_from_4( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*2]*c; \
- totb0 += decode[1+(ofs)*2]*c; \
- c = hc[1+(ofs)]; \
- tota1 += decode[2+(ofs)*2]*c; \
- totb1 += decode[3+(ofs)*2]*c; \
- c = hc[2+(ofs)]; \
- tota2 += decode[4+(ofs)*2]*c; \
- totb2 += decode[5+(ofs)*2]*c; \
- c = hc[3+(ofs)]; \
- tota3 += decode[6+(ofs)*2]*c; \
- totb3 += decode[7+(ofs)*2]*c;
- #define stbir__1_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*2] * c; \
- totb0 += decode[1+(ofs)*2] * c;
- #define stbir__2_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*2] * c; \
- totb0 += decode[1+(ofs)*2] * c; \
- c = hc[1+(ofs)]; \
- tota1 += decode[2+(ofs)*2] * c; \
- totb1 += decode[3+(ofs)*2] * c;
- #define stbir__3_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*2] * c; \
- totb0 += decode[1+(ofs)*2] * c; \
- c = hc[1+(ofs)]; \
- tota1 += decode[2+(ofs)*2] * c; \
- totb1 += decode[3+(ofs)*2] * c; \
- c = hc[2+(ofs)]; \
- tota2 += decode[4+(ofs)*2] * c; \
- totb2 += decode[5+(ofs)*2] * c;
- #define stbir__store_output() \
- output[0] = (tota0+tota2)+(tota1+tota3); \
- output[1] = (totb0+totb2)+(totb1+totb3); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 2;
- #endif
- #define STBIR__horizontal_channels 2
- #define STB_IMAGE_RESIZE_DO_HORIZONTALS
- #include STBIR__HEADER_FILENAME
- //=================
- // Do 3 channel horizontal routines
- #ifdef STBIR_SIMD
- #define stbir__1_coeff_only() \
- stbir__simdf tot,c,d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1z( c, hc ); \
- stbir__simdf_0123to0001( c, c ); \
- stbir__simdf_load( d, decode ); \
- stbir__simdf_mult( tot, d, c );
- #define stbir__2_coeff_only() \
- stbir__simdf tot,c,cs,d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_load( d, decode ); \
- stbir__simdf_mult( tot, d, c ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_load( d, decode+3 ); \
- stbir__simdf_madd( tot, tot, d, c );
- #define stbir__3_coeff_only() \
- stbir__simdf tot,c,d,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_load( d, decode ); \
- stbir__simdf_mult( tot, d, c ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_load( d, decode+3 ); \
- stbir__simdf_madd( tot, tot, d, c ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_load( d, decode+6 ); \
- stbir__simdf_madd( tot, tot, d, c );
- #define stbir__store_output_tiny() \
- stbir__simdf_store2( output, tot ); \
- stbir__simdf_0123to2301( tot, tot ); \
- stbir__simdf_store1( output+2, tot ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 3;
- #ifdef STBIR_SIMD8
- // we're loading from the XXXYYY decode by -1 to get the XXXYYY into different halves of the AVX reg fyi
- #define stbir__4_coeff_start() \
- stbir__simdf8 tot0,tot1,c,cs; stbir__simdf t; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_mult_mem( tot0, c, decode - 1 ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_mult_mem( tot1, c, decode+6 - 1 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*3 - 1 ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+(ofs)*3 + 6 - 1 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1rep4( t, hc + (ofs) ); \
- stbir__simdf8_madd_mem4( tot0, tot0, t, decode+(ofs)*3 - 1 );
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) - 2 ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*3 - 1 );
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*3 - 1 ); \
- stbir__simdf8_0123to2222( t, cs ); \
- stbir__simdf8_madd_mem4( tot1, tot1, t, decode+(ofs)*3 + 6 - 1 );
- #define stbir__store_output() \
- stbir__simdf8_add( tot0, tot0, tot1 ); \
- stbir__simdf_0123to1230( t, stbir__if_simdf8_cast_to_simdf4( tot0 ) ); \
- stbir__simdf8_add4halves( t, t, tot0 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 3; \
- if ( output < output_end ) \
- { \
- stbir__simdf_store( output-3, t ); \
- continue; \
- } \
- { stbir__simdf tt; stbir__simdf_0123to2301( tt, t ); \
- stbir__simdf_store2( output-3, t ); \
- stbir__simdf_store1( output+2-3, tt ); } \
- break;
- #else
- #define stbir__4_coeff_start() \
- stbir__simdf tot0,tot1,tot2,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0001( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_0123to1122( c, cs ); \
- stbir__simdf_mult_mem( tot1, c, decode+4 ); \
- stbir__simdf_0123to2333( c, cs ); \
- stbir__simdf_mult_mem( tot2, c, decode+8 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0001( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*3 ); \
- stbir__simdf_0123to1122( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*3+4 ); \
- stbir__simdf_0123to2333( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+(ofs)*3+8 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1z( c, hc + (ofs) ); \
- stbir__simdf_0123to0001( c, c ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*3 );
- #define stbir__2_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2z( cs, hc + (ofs) ); \
- stbir__simdf_0123to0001( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*3 ); \
- stbir__simdf_0123to1122( c, cs ); \
- stbir__simdf_load2z( d, decode+(ofs)*3+4 ); \
- stbir__simdf_madd( tot1, tot1, c, d ); }
- #define stbir__3_coeff_remnant( ofs ) \
- { stbir__simdf d; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0001( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*3 ); \
- stbir__simdf_0123to1122( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*3+4 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_load1z( d, decode+(ofs)*3+8 ); \
- stbir__simdf_madd( tot2, tot2, c, d ); }
- #define stbir__store_output() \
- stbir__simdf_0123ABCDto3ABx( c, tot0, tot1 ); \
- stbir__simdf_0123ABCDto23Ax( cs, tot1, tot2 ); \
- stbir__simdf_0123to1230( tot2, tot2 ); \
- stbir__simdf_add( tot0, tot0, cs ); \
- stbir__simdf_add( c, c, tot2 ); \
- stbir__simdf_add( tot0, tot0, c ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 3; \
- if ( output < output_end ) \
- { \
- stbir__simdf_store( output-3, tot0 ); \
- continue; \
- } \
- stbir__simdf_0123to2301( tot1, tot0 ); \
- stbir__simdf_store2( output-3, tot0 ); \
- stbir__simdf_store1( output+2-3, tot1 ); \
- break;
- #endif
- #else
- #define stbir__1_coeff_only() \
- float tot0, tot1, tot2, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c;
- #define stbir__2_coeff_only() \
- float tot0, tot1, tot2, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c; \
- c = hc[1]; \
- tot0 += decode[3]*c; \
- tot1 += decode[4]*c; \
- tot2 += decode[5]*c;
- #define stbir__3_coeff_only() \
- float tot0, tot1, tot2, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c; \
- c = hc[1]; \
- tot0 += decode[3]*c; \
- tot1 += decode[4]*c; \
- tot2 += decode[5]*c; \
- c = hc[2]; \
- tot0 += decode[6]*c; \
- tot1 += decode[7]*c; \
- tot2 += decode[8]*c;
- #define stbir__store_output_tiny() \
- output[0] = tot0; \
- output[1] = tot1; \
- output[2] = tot2; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 3;
- #define stbir__4_coeff_start() \
- float tota0,tota1,tota2,totb0,totb1,totb2,totc0,totc1,totc2,totd0,totd1,totd2,c; \
- c = hc[0]; \
- tota0 = decode[0]*c; \
- tota1 = decode[1]*c; \
- tota2 = decode[2]*c; \
- c = hc[1]; \
- totb0 = decode[3]*c; \
- totb1 = decode[4]*c; \
- totb2 = decode[5]*c; \
- c = hc[2]; \
- totc0 = decode[6]*c; \
- totc1 = decode[7]*c; \
- totc2 = decode[8]*c; \
- c = hc[3]; \
- totd0 = decode[9]*c; \
- totd1 = decode[10]*c; \
- totd2 = decode[11]*c;
- #define stbir__4_coeff_continue_from_4( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*3]*c; \
- tota1 += decode[1+(ofs)*3]*c; \
- tota2 += decode[2+(ofs)*3]*c; \
- c = hc[1+(ofs)]; \
- totb0 += decode[3+(ofs)*3]*c; \
- totb1 += decode[4+(ofs)*3]*c; \
- totb2 += decode[5+(ofs)*3]*c; \
- c = hc[2+(ofs)]; \
- totc0 += decode[6+(ofs)*3]*c; \
- totc1 += decode[7+(ofs)*3]*c; \
- totc2 += decode[8+(ofs)*3]*c; \
- c = hc[3+(ofs)]; \
- totd0 += decode[9+(ofs)*3]*c; \
- totd1 += decode[10+(ofs)*3]*c; \
- totd2 += decode[11+(ofs)*3]*c;
- #define stbir__1_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*3]*c; \
- tota1 += decode[1+(ofs)*3]*c; \
- tota2 += decode[2+(ofs)*3]*c;
- #define stbir__2_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*3]*c; \
- tota1 += decode[1+(ofs)*3]*c; \
- tota2 += decode[2+(ofs)*3]*c; \
- c = hc[1+(ofs)]; \
- totb0 += decode[3+(ofs)*3]*c; \
- totb1 += decode[4+(ofs)*3]*c; \
- totb2 += decode[5+(ofs)*3]*c; \
- #define stbir__3_coeff_remnant( ofs ) \
- c = hc[0+(ofs)]; \
- tota0 += decode[0+(ofs)*3]*c; \
- tota1 += decode[1+(ofs)*3]*c; \
- tota2 += decode[2+(ofs)*3]*c; \
- c = hc[1+(ofs)]; \
- totb0 += decode[3+(ofs)*3]*c; \
- totb1 += decode[4+(ofs)*3]*c; \
- totb2 += decode[5+(ofs)*3]*c; \
- c = hc[2+(ofs)]; \
- totc0 += decode[6+(ofs)*3]*c; \
- totc1 += decode[7+(ofs)*3]*c; \
- totc2 += decode[8+(ofs)*3]*c;
- #define stbir__store_output() \
- output[0] = (tota0+totc0)+(totb0+totd0); \
- output[1] = (tota1+totc1)+(totb1+totd1); \
- output[2] = (tota2+totc2)+(totb2+totd2); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 3;
- #endif
- #define STBIR__horizontal_channels 3
- #define STB_IMAGE_RESIZE_DO_HORIZONTALS
- #include STBIR__HEADER_FILENAME
- //=================
- // Do 4 channel horizontal routines
- #ifdef STBIR_SIMD
- #define stbir__1_coeff_only() \
- stbir__simdf tot,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1( c, hc ); \
- stbir__simdf_0123to0000( c, c ); \
- stbir__simdf_mult_mem( tot, c, decode );
- #define stbir__2_coeff_only() \
- stbir__simdf tot,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot, c, decode ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot, tot, c, decode+4 );
- #define stbir__3_coeff_only() \
- stbir__simdf tot,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot, c, decode ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot, tot, c, decode+4 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot, tot, c, decode+8 );
- #define stbir__store_output_tiny() \
- stbir__simdf_store( output, tot ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 4;
- #ifdef STBIR_SIMD8
- #define stbir__4_coeff_start() \
- stbir__simdf8 tot0,c,cs; stbir__simdf t; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_mult_mem( tot0, c, decode ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+8 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*4 ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*4+8 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1rep4( t, hc + (ofs) ); \
- stbir__simdf8_madd_mem4( tot0, tot0, t, decode+(ofs)*4 );
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) - 2 ); \
- stbir__simdf8_0123to22223333( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*4 );
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00001111( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*4 ); \
- stbir__simdf8_0123to2222( t, cs ); \
- stbir__simdf8_madd_mem4( tot0, tot0, t, decode+(ofs)*4+8 );
- #define stbir__store_output() \
- stbir__simdf8_add4halves( t, stbir__if_simdf8_cast_to_simdf4(tot0), tot0 ); \
- stbir__simdf_store( output, t ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 4;
- #else
- #define stbir__4_coeff_start() \
- stbir__simdf tot0,tot1,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_mult_mem( tot1, c, decode+4 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+8 ); \
- stbir__simdf_0123to3333( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+12 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*4+4 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4+8 ); \
- stbir__simdf_0123to3333( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*4+12 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1( c, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, c ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4 );
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*4+4 );
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*4+4 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*4+8 );
- #define stbir__store_output() \
- stbir__simdf_add( tot0, tot0, tot1 ); \
- stbir__simdf_store( output, tot0 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 4;
- #endif
- #else
- #define stbir__1_coeff_only() \
- float p0,p1,p2,p3,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0]; \
- p0 = decode[0] * c; \
- p1 = decode[1] * c; \
- p2 = decode[2] * c; \
- p3 = decode[3] * c;
- #define stbir__2_coeff_only() \
- float p0,p1,p2,p3,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0]; \
- p0 = decode[0] * c; \
- p1 = decode[1] * c; \
- p2 = decode[2] * c; \
- p3 = decode[3] * c; \
- c = hc[1]; \
- p0 += decode[4] * c; \
- p1 += decode[5] * c; \
- p2 += decode[6] * c; \
- p3 += decode[7] * c;
- #define stbir__3_coeff_only() \
- float p0,p1,p2,p3,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0]; \
- p0 = decode[0] * c; \
- p1 = decode[1] * c; \
- p2 = decode[2] * c; \
- p3 = decode[3] * c; \
- c = hc[1]; \
- p0 += decode[4] * c; \
- p1 += decode[5] * c; \
- p2 += decode[6] * c; \
- p3 += decode[7] * c; \
- c = hc[2]; \
- p0 += decode[8] * c; \
- p1 += decode[9] * c; \
- p2 += decode[10] * c; \
- p3 += decode[11] * c;
- #define stbir__store_output_tiny() \
- output[0] = p0; \
- output[1] = p1; \
- output[2] = p2; \
- output[3] = p3; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 4;
- #define stbir__4_coeff_start() \
- float x0,x1,x2,x3,y0,y1,y2,y3,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0]; \
- x0 = decode[0] * c; \
- x1 = decode[1] * c; \
- x2 = decode[2] * c; \
- x3 = decode[3] * c; \
- c = hc[1]; \
- y0 = decode[4] * c; \
- y1 = decode[5] * c; \
- y2 = decode[6] * c; \
- y3 = decode[7] * c; \
- c = hc[2]; \
- x0 += decode[8] * c; \
- x1 += decode[9] * c; \
- x2 += decode[10] * c; \
- x3 += decode[11] * c; \
- c = hc[3]; \
- y0 += decode[12] * c; \
- y1 += decode[13] * c; \
- y2 += decode[14] * c; \
- y3 += decode[15] * c;
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*4] * c; \
- x1 += decode[1+(ofs)*4] * c; \
- x2 += decode[2+(ofs)*4] * c; \
- x3 += decode[3+(ofs)*4] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[4+(ofs)*4] * c; \
- y1 += decode[5+(ofs)*4] * c; \
- y2 += decode[6+(ofs)*4] * c; \
- y3 += decode[7+(ofs)*4] * c; \
- c = hc[2+(ofs)]; \
- x0 += decode[8+(ofs)*4] * c; \
- x1 += decode[9+(ofs)*4] * c; \
- x2 += decode[10+(ofs)*4] * c; \
- x3 += decode[11+(ofs)*4] * c; \
- c = hc[3+(ofs)]; \
- y0 += decode[12+(ofs)*4] * c; \
- y1 += decode[13+(ofs)*4] * c; \
- y2 += decode[14+(ofs)*4] * c; \
- y3 += decode[15+(ofs)*4] * c;
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*4] * c; \
- x1 += decode[1+(ofs)*4] * c; \
- x2 += decode[2+(ofs)*4] * c; \
- x3 += decode[3+(ofs)*4] * c;
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*4] * c; \
- x1 += decode[1+(ofs)*4] * c; \
- x2 += decode[2+(ofs)*4] * c; \
- x3 += decode[3+(ofs)*4] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[4+(ofs)*4] * c; \
- y1 += decode[5+(ofs)*4] * c; \
- y2 += decode[6+(ofs)*4] * c; \
- y3 += decode[7+(ofs)*4] * c;
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*4] * c; \
- x1 += decode[1+(ofs)*4] * c; \
- x2 += decode[2+(ofs)*4] * c; \
- x3 += decode[3+(ofs)*4] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[4+(ofs)*4] * c; \
- y1 += decode[5+(ofs)*4] * c; \
- y2 += decode[6+(ofs)*4] * c; \
- y3 += decode[7+(ofs)*4] * c; \
- c = hc[2+(ofs)]; \
- x0 += decode[8+(ofs)*4] * c; \
- x1 += decode[9+(ofs)*4] * c; \
- x2 += decode[10+(ofs)*4] * c; \
- x3 += decode[11+(ofs)*4] * c;
- #define stbir__store_output() \
- output[0] = x0 + y0; \
- output[1] = x1 + y1; \
- output[2] = x2 + y2; \
- output[3] = x3 + y3; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 4;
- #endif
- #define STBIR__horizontal_channels 4
- #define STB_IMAGE_RESIZE_DO_HORIZONTALS
- #include STBIR__HEADER_FILENAME
- //=================
- // Do 7 channel horizontal routines
- #ifdef STBIR_SIMD
- #define stbir__1_coeff_only() \
- stbir__simdf tot0,tot1,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1( c, hc ); \
- stbir__simdf_0123to0000( c, c ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_mult_mem( tot1, c, decode+3 );
- #define stbir__2_coeff_only() \
- stbir__simdf tot0,tot1,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_mult_mem( tot1, c, decode+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c,decode+10 );
- #define stbir__3_coeff_only() \
- stbir__simdf tot0,tot1,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_mult_mem( tot1, c, decode+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+10 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+14 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+17 );
- #define stbir__store_output_tiny() \
- stbir__simdf_store( output+3, tot1 ); \
- stbir__simdf_store( output, tot0 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 7;
- #ifdef STBIR_SIMD8
- #define stbir__4_coeff_start() \
- stbir__simdf8 tot0,tot1,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc ); \
- stbir__simdf8_0123to00000000( c, cs ); \
- stbir__simdf8_mult_mem( tot0, c, decode ); \
- stbir__simdf8_0123to11111111( c, cs ); \
- stbir__simdf8_mult_mem( tot1, c, decode+7 ); \
- stbir__simdf8_0123to22222222( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+14 ); \
- stbir__simdf8_0123to33333333( c, cs ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+21 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00000000( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf8_0123to11111111( c, cs ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+(ofs)*7+7 ); \
- stbir__simdf8_0123to22222222( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7+14 ); \
- stbir__simdf8_0123to33333333( c, cs ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+(ofs)*7+21 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load1b( c, hc + (ofs) ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7 );
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load1b( c, hc + (ofs) ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf8_load1b( c, hc + (ofs)+1 ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+(ofs)*7+7 );
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf8_load4b( cs, hc + (ofs) ); \
- stbir__simdf8_0123to00000000( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf8_0123to11111111( c, cs ); \
- stbir__simdf8_madd_mem( tot1, tot1, c, decode+(ofs)*7+7 ); \
- stbir__simdf8_0123to22222222( c, cs ); \
- stbir__simdf8_madd_mem( tot0, tot0, c, decode+(ofs)*7+14 );
- #define stbir__store_output() \
- stbir__simdf8_add( tot0, tot0, tot1 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 7; \
- if ( output < output_end ) \
- { \
- stbir__simdf8_store( output-7, tot0 ); \
- continue; \
- } \
- stbir__simdf_store( output-7+3, stbir__simdf_swiz(stbir__simdf8_gettop4(tot0),0,0,1,2) ); \
- stbir__simdf_store( output-7, stbir__if_simdf8_cast_to_simdf4(tot0) ); \
- break;
- #else
- #define stbir__4_coeff_start() \
- stbir__simdf tot0,tot1,tot2,tot3,c,cs; \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_mult_mem( tot0, c, decode ); \
- stbir__simdf_mult_mem( tot1, c, decode+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_mult_mem( tot2, c, decode+7 ); \
- stbir__simdf_mult_mem( tot3, c, decode+10 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+14 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+17 ); \
- stbir__simdf_0123to3333( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+21 ); \
- stbir__simdf_madd_mem( tot3, tot3, c, decode+24 );
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+(ofs)*7+7 ); \
- stbir__simdf_madd_mem( tot3, tot3, c, decode+(ofs)*7+10 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7+14 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+17 ); \
- stbir__simdf_0123to3333( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+(ofs)*7+21 ); \
- stbir__simdf_madd_mem( tot3, tot3, c, decode+(ofs)*7+24 );
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load1( c, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, c ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+3 ); \
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load2( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+(ofs)*7+7 ); \
- stbir__simdf_madd_mem( tot3, tot3, c, decode+(ofs)*7+10 );
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- stbir__simdf_load( cs, hc + (ofs) ); \
- stbir__simdf_0123to0000( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+3 ); \
- stbir__simdf_0123to1111( c, cs ); \
- stbir__simdf_madd_mem( tot2, tot2, c, decode+(ofs)*7+7 ); \
- stbir__simdf_madd_mem( tot3, tot3, c, decode+(ofs)*7+10 ); \
- stbir__simdf_0123to2222( c, cs ); \
- stbir__simdf_madd_mem( tot0, tot0, c, decode+(ofs)*7+14 ); \
- stbir__simdf_madd_mem( tot1, tot1, c, decode+(ofs)*7+17 );
- #define stbir__store_output() \
- stbir__simdf_add( tot0, tot0, tot2 ); \
- stbir__simdf_add( tot1, tot1, tot3 ); \
- stbir__simdf_store( output+3, tot1 ); \
- stbir__simdf_store( output, tot0 ); \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 7;
- #endif
- #else
- #define stbir__1_coeff_only() \
- float tot0, tot1, tot2, tot3, tot4, tot5, tot6, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c; \
- tot3 = decode[3]*c; \
- tot4 = decode[4]*c; \
- tot5 = decode[5]*c; \
- tot6 = decode[6]*c;
- #define stbir__2_coeff_only() \
- float tot0, tot1, tot2, tot3, tot4, tot5, tot6, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c; \
- tot3 = decode[3]*c; \
- tot4 = decode[4]*c; \
- tot5 = decode[5]*c; \
- tot6 = decode[6]*c; \
- c = hc[1]; \
- tot0 += decode[7]*c; \
- tot1 += decode[8]*c; \
- tot2 += decode[9]*c; \
- tot3 += decode[10]*c; \
- tot4 += decode[11]*c; \
- tot5 += decode[12]*c; \
- tot6 += decode[13]*c; \
- #define stbir__3_coeff_only() \
- float tot0, tot1, tot2, tot3, tot4, tot5, tot6, c; \
- c = hc[0]; \
- tot0 = decode[0]*c; \
- tot1 = decode[1]*c; \
- tot2 = decode[2]*c; \
- tot3 = decode[3]*c; \
- tot4 = decode[4]*c; \
- tot5 = decode[5]*c; \
- tot6 = decode[6]*c; \
- c = hc[1]; \
- tot0 += decode[7]*c; \
- tot1 += decode[8]*c; \
- tot2 += decode[9]*c; \
- tot3 += decode[10]*c; \
- tot4 += decode[11]*c; \
- tot5 += decode[12]*c; \
- tot6 += decode[13]*c; \
- c = hc[2]; \
- tot0 += decode[14]*c; \
- tot1 += decode[15]*c; \
- tot2 += decode[16]*c; \
- tot3 += decode[17]*c; \
- tot4 += decode[18]*c; \
- tot5 += decode[19]*c; \
- tot6 += decode[20]*c; \
- #define stbir__store_output_tiny() \
- output[0] = tot0; \
- output[1] = tot1; \
- output[2] = tot2; \
- output[3] = tot3; \
- output[4] = tot4; \
- output[5] = tot5; \
- output[6] = tot6; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 7;
- #define stbir__4_coeff_start() \
- float x0,x1,x2,x3,x4,x5,x6,y0,y1,y2,y3,y4,y5,y6,c; \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0]; \
- x0 = decode[0] * c; \
- x1 = decode[1] * c; \
- x2 = decode[2] * c; \
- x3 = decode[3] * c; \
- x4 = decode[4] * c; \
- x5 = decode[5] * c; \
- x6 = decode[6] * c; \
- c = hc[1]; \
- y0 = decode[7] * c; \
- y1 = decode[8] * c; \
- y2 = decode[9] * c; \
- y3 = decode[10] * c; \
- y4 = decode[11] * c; \
- y5 = decode[12] * c; \
- y6 = decode[13] * c; \
- c = hc[2]; \
- x0 += decode[14] * c; \
- x1 += decode[15] * c; \
- x2 += decode[16] * c; \
- x3 += decode[17] * c; \
- x4 += decode[18] * c; \
- x5 += decode[19] * c; \
- x6 += decode[20] * c; \
- c = hc[3]; \
- y0 += decode[21] * c; \
- y1 += decode[22] * c; \
- y2 += decode[23] * c; \
- y3 += decode[24] * c; \
- y4 += decode[25] * c; \
- y5 += decode[26] * c; \
- y6 += decode[27] * c;
- #define stbir__4_coeff_continue_from_4( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*7] * c; \
- x1 += decode[1+(ofs)*7] * c; \
- x2 += decode[2+(ofs)*7] * c; \
- x3 += decode[3+(ofs)*7] * c; \
- x4 += decode[4+(ofs)*7] * c; \
- x5 += decode[5+(ofs)*7] * c; \
- x6 += decode[6+(ofs)*7] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[7+(ofs)*7] * c; \
- y1 += decode[8+(ofs)*7] * c; \
- y2 += decode[9+(ofs)*7] * c; \
- y3 += decode[10+(ofs)*7] * c; \
- y4 += decode[11+(ofs)*7] * c; \
- y5 += decode[12+(ofs)*7] * c; \
- y6 += decode[13+(ofs)*7] * c; \
- c = hc[2+(ofs)]; \
- x0 += decode[14+(ofs)*7] * c; \
- x1 += decode[15+(ofs)*7] * c; \
- x2 += decode[16+(ofs)*7] * c; \
- x3 += decode[17+(ofs)*7] * c; \
- x4 += decode[18+(ofs)*7] * c; \
- x5 += decode[19+(ofs)*7] * c; \
- x6 += decode[20+(ofs)*7] * c; \
- c = hc[3+(ofs)]; \
- y0 += decode[21+(ofs)*7] * c; \
- y1 += decode[22+(ofs)*7] * c; \
- y2 += decode[23+(ofs)*7] * c; \
- y3 += decode[24+(ofs)*7] * c; \
- y4 += decode[25+(ofs)*7] * c; \
- y5 += decode[26+(ofs)*7] * c; \
- y6 += decode[27+(ofs)*7] * c;
- #define stbir__1_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*7] * c; \
- x1 += decode[1+(ofs)*7] * c; \
- x2 += decode[2+(ofs)*7] * c; \
- x3 += decode[3+(ofs)*7] * c; \
- x4 += decode[4+(ofs)*7] * c; \
- x5 += decode[5+(ofs)*7] * c; \
- x6 += decode[6+(ofs)*7] * c; \
- #define stbir__2_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*7] * c; \
- x1 += decode[1+(ofs)*7] * c; \
- x2 += decode[2+(ofs)*7] * c; \
- x3 += decode[3+(ofs)*7] * c; \
- x4 += decode[4+(ofs)*7] * c; \
- x5 += decode[5+(ofs)*7] * c; \
- x6 += decode[6+(ofs)*7] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[7+(ofs)*7] * c; \
- y1 += decode[8+(ofs)*7] * c; \
- y2 += decode[9+(ofs)*7] * c; \
- y3 += decode[10+(ofs)*7] * c; \
- y4 += decode[11+(ofs)*7] * c; \
- y5 += decode[12+(ofs)*7] * c; \
- y6 += decode[13+(ofs)*7] * c; \
- #define stbir__3_coeff_remnant( ofs ) \
- STBIR_SIMD_NO_UNROLL(decode); \
- c = hc[0+(ofs)]; \
- x0 += decode[0+(ofs)*7] * c; \
- x1 += decode[1+(ofs)*7] * c; \
- x2 += decode[2+(ofs)*7] * c; \
- x3 += decode[3+(ofs)*7] * c; \
- x4 += decode[4+(ofs)*7] * c; \
- x5 += decode[5+(ofs)*7] * c; \
- x6 += decode[6+(ofs)*7] * c; \
- c = hc[1+(ofs)]; \
- y0 += decode[7+(ofs)*7] * c; \
- y1 += decode[8+(ofs)*7] * c; \
- y2 += decode[9+(ofs)*7] * c; \
- y3 += decode[10+(ofs)*7] * c; \
- y4 += decode[11+(ofs)*7] * c; \
- y5 += decode[12+(ofs)*7] * c; \
- y6 += decode[13+(ofs)*7] * c; \
- c = hc[2+(ofs)]; \
- x0 += decode[14+(ofs)*7] * c; \
- x1 += decode[15+(ofs)*7] * c; \
- x2 += decode[16+(ofs)*7] * c; \
- x3 += decode[17+(ofs)*7] * c; \
- x4 += decode[18+(ofs)*7] * c; \
- x5 += decode[19+(ofs)*7] * c; \
- x6 += decode[20+(ofs)*7] * c; \
- #define stbir__store_output() \
- output[0] = x0 + y0; \
- output[1] = x1 + y1; \
- output[2] = x2 + y2; \
- output[3] = x3 + y3; \
- output[4] = x4 + y4; \
- output[5] = x5 + y5; \
- output[6] = x6 + y6; \
- horizontal_coefficients += coefficient_width; \
- ++horizontal_contributors; \
- output += 7;
- #endif
- #define STBIR__horizontal_channels 7
- #define STB_IMAGE_RESIZE_DO_HORIZONTALS
- #include STBIR__HEADER_FILENAME
- // include all of the vertical resamplers (both scatter and gather versions)
- #define STBIR__vertical_channels 1
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 1
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 2
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 2
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 3
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 3
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 4
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 4
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 5
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 5
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 6
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 6
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 7
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 7
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 8
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #include STBIR__HEADER_FILENAME
- #define STBIR__vertical_channels 8
- #define STB_IMAGE_RESIZE_DO_VERTICALS
- #define STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #include STBIR__HEADER_FILENAME
- typedef void STBIR_VERTICAL_GATHERFUNC( float * output, float const * coeffs, float const ** inputs, float const * input0_end );
- static STBIR_VERTICAL_GATHERFUNC * stbir__vertical_gathers[ 8 ] =
- {
- stbir__vertical_gather_with_1_coeffs,stbir__vertical_gather_with_2_coeffs,stbir__vertical_gather_with_3_coeffs,stbir__vertical_gather_with_4_coeffs,stbir__vertical_gather_with_5_coeffs,stbir__vertical_gather_with_6_coeffs,stbir__vertical_gather_with_7_coeffs,stbir__vertical_gather_with_8_coeffs
- };
- static STBIR_VERTICAL_GATHERFUNC * stbir__vertical_gathers_continues[ 8 ] =
- {
- stbir__vertical_gather_with_1_coeffs_cont,stbir__vertical_gather_with_2_coeffs_cont,stbir__vertical_gather_with_3_coeffs_cont,stbir__vertical_gather_with_4_coeffs_cont,stbir__vertical_gather_with_5_coeffs_cont,stbir__vertical_gather_with_6_coeffs_cont,stbir__vertical_gather_with_7_coeffs_cont,stbir__vertical_gather_with_8_coeffs_cont
- };
- typedef void STBIR_VERTICAL_SCATTERFUNC( float ** outputs, float const * coeffs, float const * input, float const * input_end );
- static STBIR_VERTICAL_SCATTERFUNC * stbir__vertical_scatter_sets[ 8 ] =
- {
- stbir__vertical_scatter_with_1_coeffs,stbir__vertical_scatter_with_2_coeffs,stbir__vertical_scatter_with_3_coeffs,stbir__vertical_scatter_with_4_coeffs,stbir__vertical_scatter_with_5_coeffs,stbir__vertical_scatter_with_6_coeffs,stbir__vertical_scatter_with_7_coeffs,stbir__vertical_scatter_with_8_coeffs
- };
- static STBIR_VERTICAL_SCATTERFUNC * stbir__vertical_scatter_blends[ 8 ] =
- {
- stbir__vertical_scatter_with_1_coeffs_cont,stbir__vertical_scatter_with_2_coeffs_cont,stbir__vertical_scatter_with_3_coeffs_cont,stbir__vertical_scatter_with_4_coeffs_cont,stbir__vertical_scatter_with_5_coeffs_cont,stbir__vertical_scatter_with_6_coeffs_cont,stbir__vertical_scatter_with_7_coeffs_cont,stbir__vertical_scatter_with_8_coeffs_cont
- };
- static void stbir__encode_scanline( stbir__info const * stbir_info, void *output_buffer_data, float * encode_buffer, int row STBIR_ONLY_PROFILE_GET_SPLIT_INFO )
- {
- int num_pixels = stbir_info->horizontal.scale_info.output_sub_size;
- int channels = stbir_info->channels;
- int width_times_channels = num_pixels * channels;
- void * output_buffer;
- // un-alpha weight if we need to
- if ( stbir_info->alpha_unweight )
- {
- STBIR_PROFILE_START( unalpha );
- stbir_info->alpha_unweight( encode_buffer, width_times_channels );
- STBIR_PROFILE_END( unalpha );
- }
- // write directly into output by default
- output_buffer = output_buffer_data;
- // if we have an output callback, we first convert the decode buffer in place (and then hand that to the callback)
- if ( stbir_info->out_pixels_cb )
- output_buffer = encode_buffer;
- STBIR_PROFILE_START( encode );
- // convert into the output buffer
- stbir_info->encode_pixels( output_buffer, width_times_channels, encode_buffer );
- STBIR_PROFILE_END( encode );
- // if we have an output callback, call it to send the data
- if ( stbir_info->out_pixels_cb )
- stbir_info->out_pixels_cb( output_buffer, num_pixels, row, stbir_info->user_data );
- }
- // Get the ring buffer pointer for an index
- static float* stbir__get_ring_buffer_entry(stbir__info const * stbir_info, stbir__per_split_info const * split_info, int index )
- {
- STBIR_ASSERT( index < stbir_info->ring_buffer_num_entries );
- #ifdef STBIR__SEPARATE_ALLOCATIONS
- return split_info->ring_buffers[ index ];
- #else
- return (float*) ( ( (char*) split_info->ring_buffer ) + ( index * stbir_info->ring_buffer_length_bytes ) );
- #endif
- }
- // Get the specified scan line from the ring buffer
- static float* stbir__get_ring_buffer_scanline(stbir__info const * stbir_info, stbir__per_split_info const * split_info, int get_scanline)
- {
- int ring_buffer_index = (split_info->ring_buffer_begin_index + (get_scanline - split_info->ring_buffer_first_scanline)) % stbir_info->ring_buffer_num_entries;
- return stbir__get_ring_buffer_entry( stbir_info, split_info, ring_buffer_index );
- }
- static void stbir__resample_horizontal_gather(stbir__info const * stbir_info, float* output_buffer, float const * input_buffer STBIR_ONLY_PROFILE_GET_SPLIT_INFO )
- {
- float const * decode_buffer = input_buffer - ( stbir_info->scanline_extents.conservative.n0 * stbir_info->effective_channels );
- STBIR_PROFILE_START( horizontal );
- if ( ( stbir_info->horizontal.filter_enum == STBIR_FILTER_POINT_SAMPLE ) && ( stbir_info->horizontal.scale_info.scale == 1.0f ) )
- STBIR_MEMCPY( output_buffer, input_buffer, stbir_info->horizontal.scale_info.output_sub_size * sizeof( float ) * stbir_info->effective_channels );
- else
- stbir_info->horizontal_gather_channels( output_buffer, stbir_info->horizontal.scale_info.output_sub_size, decode_buffer, stbir_info->horizontal.contributors, stbir_info->horizontal.coefficients, stbir_info->horizontal.coefficient_width );
- STBIR_PROFILE_END( horizontal );
- }
- static void stbir__resample_vertical_gather(stbir__info const * stbir_info, stbir__per_split_info* split_info, int n, int contrib_n0, int contrib_n1, float const * vertical_coefficients )
- {
- float* encode_buffer = split_info->vertical_buffer;
- float* decode_buffer = split_info->decode_buffer;
- int vertical_first = stbir_info->vertical_first;
- int width = (vertical_first) ? ( stbir_info->scanline_extents.conservative.n1-stbir_info->scanline_extents.conservative.n0+1 ) : stbir_info->horizontal.scale_info.output_sub_size;
- int width_times_channels = stbir_info->effective_channels * width;
- STBIR_ASSERT( stbir_info->vertical.is_gather );
- // loop over the contributing scanlines and scale into the buffer
- STBIR_PROFILE_START( vertical );
- {
- int k = 0, total = contrib_n1 - contrib_n0 + 1;
- STBIR_ASSERT( total > 0 );
- do {
- float const * inputs[8];
- int i, cnt = total; if ( cnt > 8 ) cnt = 8;
- for( i = 0 ; i < cnt ; i++ )
- inputs[ i ] = stbir__get_ring_buffer_scanline(stbir_info, split_info, k+i+contrib_n0 );
- // call the N scanlines at a time function (up to 8 scanlines of blending at once)
- ((k==0)?stbir__vertical_gathers:stbir__vertical_gathers_continues)[cnt-1]( (vertical_first) ? decode_buffer : encode_buffer, vertical_coefficients + k, inputs, inputs[0] + width_times_channels );
- k += cnt;
- total -= cnt;
- } while ( total );
- }
- STBIR_PROFILE_END( vertical );
- if ( vertical_first )
- {
- // Now resample the gathered vertical data in the horizontal axis into the encode buffer
- decode_buffer[ width_times_channels ] = 0.0f; // clear two over for horizontals with a remnant of 3
- decode_buffer[ width_times_channels+1 ] = 0.0f;
- stbir__resample_horizontal_gather(stbir_info, encode_buffer, decode_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- }
- stbir__encode_scanline( stbir_info, ( (char *) stbir_info->output_data ) + ((size_t)n * (size_t)stbir_info->output_stride_bytes),
- encode_buffer, n STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- }
- static void stbir__decode_and_resample_for_vertical_gather_loop(stbir__info const * stbir_info, stbir__per_split_info* split_info, int n)
- {
- int ring_buffer_index;
- float* ring_buffer;
- // Decode the nth scanline from the source image into the decode buffer.
- stbir__decode_scanline( stbir_info, n, split_info->decode_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // update new end scanline
- split_info->ring_buffer_last_scanline = n;
- // get ring buffer
- ring_buffer_index = (split_info->ring_buffer_begin_index + (split_info->ring_buffer_last_scanline - split_info->ring_buffer_first_scanline)) % stbir_info->ring_buffer_num_entries;
- ring_buffer = stbir__get_ring_buffer_entry(stbir_info, split_info, ring_buffer_index);
- // Now resample it into the ring buffer.
- stbir__resample_horizontal_gather( stbir_info, ring_buffer, split_info->decode_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // Now it's sitting in the ring buffer ready to be used as source for the vertical sampling.
- }
- static void stbir__vertical_gather_loop( stbir__info const * stbir_info, stbir__per_split_info* split_info, int split_count )
- {
- int y, start_output_y, end_output_y;
- stbir__contributors* vertical_contributors = stbir_info->vertical.contributors;
- float const * vertical_coefficients = stbir_info->vertical.coefficients;
- STBIR_ASSERT( stbir_info->vertical.is_gather );
- start_output_y = split_info->start_output_y;
- end_output_y = split_info[split_count-1].end_output_y;
- vertical_contributors += start_output_y;
- vertical_coefficients += start_output_y * stbir_info->vertical.coefficient_width;
- // initialize the ring buffer for gathering
- split_info->ring_buffer_begin_index = 0;
- split_info->ring_buffer_first_scanline = vertical_contributors->n0;
- split_info->ring_buffer_last_scanline = split_info->ring_buffer_first_scanline - 1; // means "empty"
- for (y = start_output_y; y < end_output_y; y++)
- {
- int in_first_scanline, in_last_scanline;
- in_first_scanline = vertical_contributors->n0;
- in_last_scanline = vertical_contributors->n1;
- // make sure the indexing hasn't broken
- STBIR_ASSERT( in_first_scanline >= split_info->ring_buffer_first_scanline );
- // Load in new scanlines
- while (in_last_scanline > split_info->ring_buffer_last_scanline)
- {
- STBIR_ASSERT( ( split_info->ring_buffer_last_scanline - split_info->ring_buffer_first_scanline + 1 ) <= stbir_info->ring_buffer_num_entries );
- // make sure there was room in the ring buffer when we add new scanlines
- if ( ( split_info->ring_buffer_last_scanline - split_info->ring_buffer_first_scanline + 1 ) == stbir_info->ring_buffer_num_entries )
- {
- split_info->ring_buffer_first_scanline++;
- split_info->ring_buffer_begin_index++;
- }
- if ( stbir_info->vertical_first )
- {
- float * ring_buffer = stbir__get_ring_buffer_scanline( stbir_info, split_info, ++split_info->ring_buffer_last_scanline );
- // Decode the nth scanline from the source image into the decode buffer.
- stbir__decode_scanline( stbir_info, split_info->ring_buffer_last_scanline, ring_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- }
- else
- {
- stbir__decode_and_resample_for_vertical_gather_loop(stbir_info, split_info, split_info->ring_buffer_last_scanline + 1);
- }
- }
- // Now all buffers should be ready to write a row of vertical sampling, so do it.
- stbir__resample_vertical_gather(stbir_info, split_info, y, in_first_scanline, in_last_scanline, vertical_coefficients );
- ++vertical_contributors;
- vertical_coefficients += stbir_info->vertical.coefficient_width;
- }
- }
- #define STBIR__FLOAT_EMPTY_MARKER 3.0e+38F
- #define STBIR__FLOAT_BUFFER_IS_EMPTY(ptr) ((ptr)[0]==STBIR__FLOAT_EMPTY_MARKER)
- static void stbir__encode_first_scanline_from_scatter(stbir__info const * stbir_info, stbir__per_split_info* split_info)
- {
- // evict a scanline out into the output buffer
- float* ring_buffer_entry = stbir__get_ring_buffer_entry(stbir_info, split_info, split_info->ring_buffer_begin_index );
- // dump the scanline out
- stbir__encode_scanline( stbir_info, ( (char *)stbir_info->output_data ) + ( (size_t)split_info->ring_buffer_first_scanline * (size_t)stbir_info->output_stride_bytes ), ring_buffer_entry, split_info->ring_buffer_first_scanline STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // mark it as empty
- ring_buffer_entry[ 0 ] = STBIR__FLOAT_EMPTY_MARKER;
- // advance the first scanline
- split_info->ring_buffer_first_scanline++;
- if ( ++split_info->ring_buffer_begin_index == stbir_info->ring_buffer_num_entries )
- split_info->ring_buffer_begin_index = 0;
- }
- static void stbir__horizontal_resample_and_encode_first_scanline_from_scatter(stbir__info const * stbir_info, stbir__per_split_info* split_info)
- {
- // evict a scanline out into the output buffer
- float* ring_buffer_entry = stbir__get_ring_buffer_entry(stbir_info, split_info, split_info->ring_buffer_begin_index );
- // Now resample it into the buffer.
- stbir__resample_horizontal_gather( stbir_info, split_info->vertical_buffer, ring_buffer_entry STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // dump the scanline out
- stbir__encode_scanline( stbir_info, ( (char *)stbir_info->output_data ) + ( (size_t)split_info->ring_buffer_first_scanline * (size_t)stbir_info->output_stride_bytes ), split_info->vertical_buffer, split_info->ring_buffer_first_scanline STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // mark it as empty
- ring_buffer_entry[ 0 ] = STBIR__FLOAT_EMPTY_MARKER;
- // advance the first scanline
- split_info->ring_buffer_first_scanline++;
- if ( ++split_info->ring_buffer_begin_index == stbir_info->ring_buffer_num_entries )
- split_info->ring_buffer_begin_index = 0;
- }
- static void stbir__resample_vertical_scatter(stbir__info const * stbir_info, stbir__per_split_info* split_info, int n0, int n1, float const * vertical_coefficients, float const * vertical_buffer, float const * vertical_buffer_end )
- {
- STBIR_ASSERT( !stbir_info->vertical.is_gather );
- STBIR_PROFILE_START( vertical );
- {
- int k = 0, total = n1 - n0 + 1;
- STBIR_ASSERT( total > 0 );
- do {
- float * outputs[8];
- int i, n = total; if ( n > 8 ) n = 8;
- for( i = 0 ; i < n ; i++ )
- {
- outputs[ i ] = stbir__get_ring_buffer_scanline(stbir_info, split_info, k+i+n0 );
- if ( ( i ) && ( STBIR__FLOAT_BUFFER_IS_EMPTY( outputs[i] ) != STBIR__FLOAT_BUFFER_IS_EMPTY( outputs[0] ) ) ) // make sure runs are of the same type
- {
- n = i;
- break;
- }
- }
- // call the scatter to N scanlines at a time function (up to 8 scanlines of scattering at once)
- ((STBIR__FLOAT_BUFFER_IS_EMPTY( outputs[0] ))?stbir__vertical_scatter_sets:stbir__vertical_scatter_blends)[n-1]( outputs, vertical_coefficients + k, vertical_buffer, vertical_buffer_end );
- k += n;
- total -= n;
- } while ( total );
- }
- STBIR_PROFILE_END( vertical );
- }
- typedef void stbir__handle_scanline_for_scatter_func(stbir__info const * stbir_info, stbir__per_split_info* split_info);
- static void stbir__vertical_scatter_loop( stbir__info const * stbir_info, stbir__per_split_info* split_info, int split_count )
- {
- int y, start_output_y, end_output_y, start_input_y, end_input_y;
- stbir__contributors* vertical_contributors = stbir_info->vertical.contributors;
- float const * vertical_coefficients = stbir_info->vertical.coefficients;
- stbir__handle_scanline_for_scatter_func * handle_scanline_for_scatter;
- void * scanline_scatter_buffer;
- void * scanline_scatter_buffer_end;
- int on_first_input_y, last_input_y;
- int width = (stbir_info->vertical_first) ? ( stbir_info->scanline_extents.conservative.n1-stbir_info->scanline_extents.conservative.n0+1 ) : stbir_info->horizontal.scale_info.output_sub_size;
- int width_times_channels = stbir_info->effective_channels * width;
- STBIR_ASSERT( !stbir_info->vertical.is_gather );
- start_output_y = split_info->start_output_y;
- end_output_y = split_info[split_count-1].end_output_y; // may do multiple split counts
- start_input_y = split_info->start_input_y;
- end_input_y = split_info[split_count-1].end_input_y;
- // adjust for starting offset start_input_y
- y = start_input_y + stbir_info->vertical.filter_pixel_margin;
- vertical_contributors += y ;
- vertical_coefficients += stbir_info->vertical.coefficient_width * y;
- if ( stbir_info->vertical_first )
- {
- handle_scanline_for_scatter = stbir__horizontal_resample_and_encode_first_scanline_from_scatter;
- scanline_scatter_buffer = split_info->decode_buffer;
- scanline_scatter_buffer_end = ( (char*) scanline_scatter_buffer ) + sizeof( float ) * stbir_info->effective_channels * (stbir_info->scanline_extents.conservative.n1-stbir_info->scanline_extents.conservative.n0+1);
- }
- else
- {
- handle_scanline_for_scatter = stbir__encode_first_scanline_from_scatter;
- scanline_scatter_buffer = split_info->vertical_buffer;
- scanline_scatter_buffer_end = ( (char*) scanline_scatter_buffer ) + sizeof( float ) * stbir_info->effective_channels * stbir_info->horizontal.scale_info.output_sub_size;
- }
- // initialize the ring buffer for scattering
- split_info->ring_buffer_first_scanline = start_output_y;
- split_info->ring_buffer_last_scanline = -1;
- split_info->ring_buffer_begin_index = -1;
- // mark all the buffers as empty to start
- for( y = 0 ; y < stbir_info->ring_buffer_num_entries ; y++ )
- {
- float * decode_buffer = stbir__get_ring_buffer_entry( stbir_info, split_info, y );
- decode_buffer[ width_times_channels ] = 0.0f; // clear two over for horizontals with a remnant of 3
- decode_buffer[ width_times_channels+1 ] = 0.0f;
- decode_buffer[0] = STBIR__FLOAT_EMPTY_MARKER; // only used on scatter
- }
- // do the loop in input space
- on_first_input_y = 1; last_input_y = start_input_y;
- for (y = start_input_y ; y < end_input_y; y++)
- {
- int out_first_scanline, out_last_scanline;
- out_first_scanline = vertical_contributors->n0;
- out_last_scanline = vertical_contributors->n1;
- STBIR_ASSERT(out_last_scanline - out_first_scanline + 1 <= stbir_info->ring_buffer_num_entries);
- if ( ( out_last_scanline >= out_first_scanline ) && ( ( ( out_first_scanline >= start_output_y ) && ( out_first_scanline < end_output_y ) ) || ( ( out_last_scanline >= start_output_y ) && ( out_last_scanline < end_output_y ) ) ) )
- {
- float const * vc = vertical_coefficients;
- // keep track of the range actually seen for the next resize
- last_input_y = y;
- if ( ( on_first_input_y ) && ( y > start_input_y ) )
- split_info->start_input_y = y;
- on_first_input_y = 0;
- // clip the region
- if ( out_first_scanline < start_output_y )
- {
- vc += start_output_y - out_first_scanline;
- out_first_scanline = start_output_y;
- }
- if ( out_last_scanline >= end_output_y )
- out_last_scanline = end_output_y - 1;
- // if very first scanline, init the index
- if (split_info->ring_buffer_begin_index < 0)
- split_info->ring_buffer_begin_index = out_first_scanline - start_output_y;
- STBIR_ASSERT( split_info->ring_buffer_begin_index <= out_first_scanline );
- // Decode the nth scanline from the source image into the decode buffer.
- stbir__decode_scanline( stbir_info, y, split_info->decode_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // When horizontal first, we resample horizontally into the vertical buffer before we scatter it out
- if ( !stbir_info->vertical_first )
- stbir__resample_horizontal_gather( stbir_info, split_info->vertical_buffer, split_info->decode_buffer STBIR_ONLY_PROFILE_SET_SPLIT_INFO );
- // Now it's sitting in the buffer ready to be distributed into the ring buffers.
- // evict from the ringbuffer, if we need are full
- if ( ( ( split_info->ring_buffer_last_scanline - split_info->ring_buffer_first_scanline + 1 ) == stbir_info->ring_buffer_num_entries ) &&
- ( out_last_scanline > split_info->ring_buffer_last_scanline ) )
- handle_scanline_for_scatter( stbir_info, split_info );
- // Now the horizontal buffer is ready to write to all ring buffer rows, so do it.
- stbir__resample_vertical_scatter(stbir_info, split_info, out_first_scanline, out_last_scanline, vc, (float*)scanline_scatter_buffer, (float*)scanline_scatter_buffer_end );
- // update the end of the buffer
- if ( out_last_scanline > split_info->ring_buffer_last_scanline )
- split_info->ring_buffer_last_scanline = out_last_scanline;
- }
- ++vertical_contributors;
- vertical_coefficients += stbir_info->vertical.coefficient_width;
- }
- // now evict the scanlines that are left over in the ring buffer
- while ( split_info->ring_buffer_first_scanline < end_output_y )
- handle_scanline_for_scatter(stbir_info, split_info);
- // update the end_input_y if we do multiple resizes with the same data
- ++last_input_y;
- for( y = 0 ; y < split_count; y++ )
- if ( split_info[y].end_input_y > last_input_y )
- split_info[y].end_input_y = last_input_y;
- }
- static stbir__kernel_callback * stbir__builtin_kernels[] = { 0, stbir__filter_trapezoid, stbir__filter_triangle, stbir__filter_cubic, stbir__filter_catmullrom, stbir__filter_mitchell, stbir__filter_point };
- static stbir__support_callback * stbir__builtin_supports[] = { 0, stbir__support_trapezoid, stbir__support_one, stbir__support_two, stbir__support_two, stbir__support_two, stbir__support_zeropoint5 };
- static void stbir__set_sampler(stbir__sampler * samp, stbir_filter filter, stbir__kernel_callback * kernel, stbir__support_callback * support, stbir_edge edge, stbir__scale_info * scale_info, int always_gather, void * user_data )
- {
- // set filter
- if (filter == 0)
- {
- filter = STBIR_DEFAULT_FILTER_DOWNSAMPLE; // default to downsample
- if (scale_info->scale >= ( 1.0f - stbir__small_float ) )
- {
- if ( (scale_info->scale <= ( 1.0f + stbir__small_float ) ) && ( STBIR_CEILF(scale_info->pixel_shift) == scale_info->pixel_shift ) )
- filter = STBIR_FILTER_POINT_SAMPLE;
- else
- filter = STBIR_DEFAULT_FILTER_UPSAMPLE;
- }
- }
- samp->filter_enum = filter;
- STBIR_ASSERT(samp->filter_enum != 0);
- STBIR_ASSERT((unsigned)samp->filter_enum < STBIR_FILTER_OTHER);
- samp->filter_kernel = stbir__builtin_kernels[ filter ];
- samp->filter_support = stbir__builtin_supports[ filter ];
- if ( kernel && support )
- {
- samp->filter_kernel = kernel;
- samp->filter_support = support;
- samp->filter_enum = STBIR_FILTER_OTHER;
- }
- samp->edge = edge;
- samp->filter_pixel_width = stbir__get_filter_pixel_width (samp->filter_support, scale_info->scale, user_data );
- // Gather is always better, but in extreme downsamples, you have to most or all of the data in memory
- // For horizontal, we always have all the pixels, so we always use gather here (always_gather==1).
- // For vertical, we use gather if scaling up (which means we will have samp->filter_pixel_width
- // scanlines in memory at once).
- samp->is_gather = 0;
- if ( scale_info->scale >= ( 1.0f - stbir__small_float ) )
- samp->is_gather = 1;
- else if ( ( always_gather ) || ( samp->filter_pixel_width <= STBIR_FORCE_GATHER_FILTER_SCANLINES_AMOUNT ) )
- samp->is_gather = 2;
- // pre calculate stuff based on the above
- samp->coefficient_width = stbir__get_coefficient_width(samp, samp->is_gather, user_data);
- // filter_pixel_width is the conservative size in pixels of input that affect an output pixel.
- // In rare cases (only with 2 pix to 1 pix with the default filters), it's possible that the
- // filter will extend before or after the scanline beyond just one extra entire copy of the
- // scanline (we would hit the edge twice). We don't let you do that, so we clamp the total
- // width to 3x the total of input pixel (once for the scanline, once for the left side
- // overhang, and once for the right side). We only do this for edge mode, since the other
- // modes can just re-edge clamp back in again.
- if ( edge == STBIR_EDGE_WRAP )
- if ( samp->filter_pixel_width > ( scale_info->input_full_size * 3 ) )
- samp->filter_pixel_width = scale_info->input_full_size * 3;
- // This is how much to expand buffers to account for filters seeking outside
- // the image boundaries.
- samp->filter_pixel_margin = samp->filter_pixel_width / 2;
-
- // filter_pixel_margin is the amount that this filter can overhang on just one side of either
- // end of the scanline (left or the right). Since we only allow you to overhang 1 scanline's
- // worth of pixels, we clamp this one side of overhang to the input scanline size. Again,
- // this clamping only happens in rare cases with the default filters (2 pix to 1 pix).
- if ( edge == STBIR_EDGE_WRAP )
- if ( samp->filter_pixel_margin > scale_info->input_full_size )
- samp->filter_pixel_margin = scale_info->input_full_size;
- samp->num_contributors = stbir__get_contributors(samp, samp->is_gather);
- samp->contributors_size = samp->num_contributors * sizeof(stbir__contributors);
- samp->coefficients_size = samp->num_contributors * samp->coefficient_width * sizeof(float) + sizeof(float)*STBIR_INPUT_CALLBACK_PADDING; // extra sizeof(float) is padding
- samp->gather_prescatter_contributors = 0;
- samp->gather_prescatter_coefficients = 0;
- if ( samp->is_gather == 0 )
- {
- samp->gather_prescatter_coefficient_width = samp->filter_pixel_width;
- samp->gather_prescatter_num_contributors = stbir__get_contributors(samp, 2);
- samp->gather_prescatter_contributors_size = samp->gather_prescatter_num_contributors * sizeof(stbir__contributors);
- samp->gather_prescatter_coefficients_size = samp->gather_prescatter_num_contributors * samp->gather_prescatter_coefficient_width * sizeof(float);
- }
- }
- static void stbir__get_conservative_extents( stbir__sampler * samp, stbir__contributors * range, void * user_data )
- {
- float scale = samp->scale_info.scale;
- float out_shift = samp->scale_info.pixel_shift;
- stbir__support_callback * support = samp->filter_support;
- int input_full_size = samp->scale_info.input_full_size;
- stbir_edge edge = samp->edge;
- float inv_scale = samp->scale_info.inv_scale;
- STBIR_ASSERT( samp->is_gather != 0 );
- if ( samp->is_gather == 1 )
- {
- int in_first_pixel, in_last_pixel;
- float out_filter_radius = support(inv_scale, user_data) * scale;
- stbir__calculate_in_pixel_range( &in_first_pixel, &in_last_pixel, 0.5, out_filter_radius, inv_scale, out_shift, input_full_size, edge );
- range->n0 = in_first_pixel;
- stbir__calculate_in_pixel_range( &in_first_pixel, &in_last_pixel, ( (float)(samp->scale_info.output_sub_size-1) ) + 0.5f, out_filter_radius, inv_scale, out_shift, input_full_size, edge );
- range->n1 = in_last_pixel;
- }
- else if ( samp->is_gather == 2 ) // downsample gather, refine
- {
- float in_pixels_radius = support(scale, user_data) * inv_scale;
- int filter_pixel_margin = samp->filter_pixel_margin;
- int output_sub_size = samp->scale_info.output_sub_size;
- int input_end;
- int n;
- int in_first_pixel, in_last_pixel;
- // get a conservative area of the input range
- stbir__calculate_in_pixel_range( &in_first_pixel, &in_last_pixel, 0, 0, inv_scale, out_shift, input_full_size, edge );
- range->n0 = in_first_pixel;
- stbir__calculate_in_pixel_range( &in_first_pixel, &in_last_pixel, (float)output_sub_size, 0, inv_scale, out_shift, input_full_size, edge );
- range->n1 = in_last_pixel;
- // now go through the margin to the start of area to find bottom
- n = range->n0 + 1;
- input_end = -filter_pixel_margin;
- while( n >= input_end )
- {
- int out_first_pixel, out_last_pixel;
- stbir__calculate_out_pixel_range( &out_first_pixel, &out_last_pixel, ((float)n)+0.5f, in_pixels_radius, scale, out_shift, output_sub_size );
- if ( out_first_pixel > out_last_pixel )
- break;
- if ( ( out_first_pixel < output_sub_size ) || ( out_last_pixel >= 0 ) )
- range->n0 = n;
- --n;
- }
- // now go through the end of the area through the margin to find top
- n = range->n1 - 1;
- input_end = n + 1 + filter_pixel_margin;
- while( n <= input_end )
- {
- int out_first_pixel, out_last_pixel;
- stbir__calculate_out_pixel_range( &out_first_pixel, &out_last_pixel, ((float)n)+0.5f, in_pixels_radius, scale, out_shift, output_sub_size );
- if ( out_first_pixel > out_last_pixel )
- break;
- if ( ( out_first_pixel < output_sub_size ) || ( out_last_pixel >= 0 ) )
- range->n1 = n;
- ++n;
- }
- }
- if ( samp->edge == STBIR_EDGE_WRAP )
- {
- // if we are wrapping, and we are very close to the image size (so the edges might merge), just use the scanline up to the edge
- if ( ( range->n0 > 0 ) && ( range->n1 >= input_full_size ) )
- {
- int marg = range->n1 - input_full_size + 1;
- if ( ( marg + STBIR__MERGE_RUNS_PIXEL_THRESHOLD ) >= range->n0 )
- range->n0 = 0;
- }
- if ( ( range->n0 < 0 ) && ( range->n1 < (input_full_size-1) ) )
- {
- int marg = -range->n0;
- if ( ( input_full_size - marg - STBIR__MERGE_RUNS_PIXEL_THRESHOLD - 1 ) <= range->n1 )
- range->n1 = input_full_size - 1;
- }
- }
- else
- {
- // for non-edge-wrap modes, we never read over the edge, so clamp
- if ( range->n0 < 0 )
- range->n0 = 0;
- if ( range->n1 >= input_full_size )
- range->n1 = input_full_size - 1;
- }
- }
- static void stbir__get_split_info( stbir__per_split_info* split_info, int splits, int output_height, int vertical_pixel_margin, int input_full_height, int is_gather, stbir__contributors * contribs )
- {
- int i, cur;
- int left = output_height;
- cur = 0;
- for( i = 0 ; i < splits ; i++ )
- {
- int each;
- split_info[i].start_output_y = cur;
- each = left / ( splits - i );
- split_info[i].end_output_y = cur + each;
- // ok, when we are gathering, we need to make sure we are starting on a y offset that doesn't have
- // a "special" set of coefficients. Basically, with exactly the right filter at exactly the right
- // resize at exactly the right phase, some of the coefficents can be zero. When they are zero, we
- // don't process them at all. But this leads to a tricky thing with the thread splits, where we
- // might have a set of two coeffs like this for example: (4,4) and (3,6). The 4,4 means there was
- // just one single coeff because things worked out perfectly (normally, they all have 4 coeffs
- // like the range 3,6. The problem is that if we start right on the (4,4) on a brand new thread,
- // then when we get to (3,6), we don't have the "3" sample in memory (because we didn't load
- // it on the initial (4,4) range because it didn't have a 3 (we only add new samples that are
- // larger than our existing samples - it's just how the eviction works). So, our solution here
- // is pretty simple, if we start right on a range that has samples that start earlier, then we
- // simply bump up our previous thread split range to include it, and then start this threads
- // range with the smaller sample. It just moves one scanline from one thread split to another,
- // so that we end with the unusual one, instead of start with it. To do this, we check 2-4
- // sample at each thread split start and then occassionally move them.
-
- if ( ( is_gather ) && ( i ) )
- {
- stbir__contributors * small_contribs;
- int j, smallest, stop, start_n0;
- stbir__contributors * split_contribs = contribs + cur;
- // scan for a max of 3x the filter width or until the next thread split
- stop = vertical_pixel_margin * 3;
- if ( each < stop )
- stop = each;
- // loops a few times before early out
- smallest = 0;
- small_contribs = split_contribs;
- start_n0 = small_contribs->n0;
- for( j = 1 ; j <= stop ; j++ )
- {
- ++split_contribs;
- if ( split_contribs->n0 > start_n0 )
- break;
- if ( split_contribs->n0 < small_contribs->n0 )
- {
- small_contribs = split_contribs;
- smallest = j;
- }
- }
- split_info[i-1].end_output_y += smallest;
- split_info[i].start_output_y += smallest;
- }
- cur += each;
- left -= each;
- // scatter range (updated to minimum as you run it)
- split_info[i].start_input_y = -vertical_pixel_margin;
- split_info[i].end_input_y = input_full_height + vertical_pixel_margin;
- }
- }
- static void stbir__free_internal_mem( stbir__info *info )
- {
- #define STBIR__FREE_AND_CLEAR( ptr ) { if ( ptr ) { void * p = (ptr); (ptr) = 0; STBIR_FREE( p, info->user_data); } }
- if ( info )
- {
- #ifndef STBIR__SEPARATE_ALLOCATIONS
- STBIR__FREE_AND_CLEAR( info->alloced_mem );
- #else
- int i,j;
- if ( ( info->vertical.gather_prescatter_contributors ) && ( (void*)info->vertical.gather_prescatter_contributors != (void*)info->split_info[0].decode_buffer ) )
- {
- STBIR__FREE_AND_CLEAR( info->vertical.gather_prescatter_coefficients );
- STBIR__FREE_AND_CLEAR( info->vertical.gather_prescatter_contributors );
- }
- for( i = 0 ; i < info->splits ; i++ )
- {
- for( j = 0 ; j < info->alloc_ring_buffer_num_entries ; j++ )
- {
- #ifdef STBIR_SIMD8
- if ( info->effective_channels == 3 )
- --info->split_info[i].ring_buffers[j]; // avx in 3 channel mode needs one float at the start of the buffer
- #endif
- STBIR__FREE_AND_CLEAR( info->split_info[i].ring_buffers[j] );
- }
- #ifdef STBIR_SIMD8
- if ( info->effective_channels == 3 )
- --info->split_info[i].decode_buffer; // avx in 3 channel mode needs one float at the start of the buffer
- #endif
- STBIR__FREE_AND_CLEAR( info->split_info[i].decode_buffer );
- STBIR__FREE_AND_CLEAR( info->split_info[i].ring_buffers );
- STBIR__FREE_AND_CLEAR( info->split_info[i].vertical_buffer );
- }
- STBIR__FREE_AND_CLEAR( info->split_info );
- if ( info->vertical.coefficients != info->horizontal.coefficients )
- {
- STBIR__FREE_AND_CLEAR( info->vertical.coefficients );
- STBIR__FREE_AND_CLEAR( info->vertical.contributors );
- }
- STBIR__FREE_AND_CLEAR( info->horizontal.coefficients );
- STBIR__FREE_AND_CLEAR( info->horizontal.contributors );
- STBIR__FREE_AND_CLEAR( info->alloced_mem );
- STBIR_FREE( info, info->user_data );
- #endif
- }
- #undef STBIR__FREE_AND_CLEAR
- }
- static int stbir__get_max_split( int splits, int height )
- {
- int i;
- int max = 0;
- for( i = 0 ; i < splits ; i++ )
- {
- int each = height / ( splits - i );
- if ( each > max )
- max = each;
- height -= each;
- }
- return max;
- }
- static stbir__horizontal_gather_channels_func ** stbir__horizontal_gather_n_coeffs_funcs[8] =
- {
- 0, stbir__horizontal_gather_1_channels_with_n_coeffs_funcs, stbir__horizontal_gather_2_channels_with_n_coeffs_funcs, stbir__horizontal_gather_3_channels_with_n_coeffs_funcs, stbir__horizontal_gather_4_channels_with_n_coeffs_funcs, 0,0, stbir__horizontal_gather_7_channels_with_n_coeffs_funcs
- };
- static stbir__horizontal_gather_channels_func ** stbir__horizontal_gather_channels_funcs[8] =
- {
- 0, stbir__horizontal_gather_1_channels_funcs, stbir__horizontal_gather_2_channels_funcs, stbir__horizontal_gather_3_channels_funcs, stbir__horizontal_gather_4_channels_funcs, 0,0, stbir__horizontal_gather_7_channels_funcs
- };
- // there are six resize classifications: 0 == vertical scatter, 1 == vertical gather < 1x scale, 2 == vertical gather 1x-2x scale, 4 == vertical gather < 3x scale, 4 == vertical gather > 3x scale, 5 == <=4 pixel height, 6 == <=4 pixel wide column
- #define STBIR_RESIZE_CLASSIFICATIONS 8
- static float stbir__compute_weights[5][STBIR_RESIZE_CLASSIFICATIONS][4]= // 5 = 0=1chan, 1=2chan, 2=3chan, 3=4chan, 4=7chan
- {
- {
- { 1.00000f, 1.00000f, 0.31250f, 1.00000f },
- { 0.56250f, 0.59375f, 0.00000f, 0.96875f },
- { 1.00000f, 0.06250f, 0.00000f, 1.00000f },
- { 0.00000f, 0.09375f, 1.00000f, 1.00000f },
- { 1.00000f, 1.00000f, 1.00000f, 1.00000f },
- { 0.03125f, 0.12500f, 1.00000f, 1.00000f },
- { 0.06250f, 0.12500f, 0.00000f, 1.00000f },
- { 0.00000f, 1.00000f, 0.00000f, 0.03125f },
- }, {
- { 0.00000f, 0.84375f, 0.00000f, 0.03125f },
- { 0.09375f, 0.93750f, 0.00000f, 0.78125f },
- { 0.87500f, 0.21875f, 0.00000f, 0.96875f },
- { 0.09375f, 0.09375f, 1.00000f, 1.00000f },
- { 1.00000f, 1.00000f, 1.00000f, 1.00000f },
- { 0.03125f, 0.12500f, 1.00000f, 1.00000f },
- { 0.06250f, 0.12500f, 0.00000f, 1.00000f },
- { 0.00000f, 1.00000f, 0.00000f, 0.53125f },
- }, {
- { 0.00000f, 0.53125f, 0.00000f, 0.03125f },
- { 0.06250f, 0.96875f, 0.00000f, 0.53125f },
- { 0.87500f, 0.18750f, 0.00000f, 0.93750f },
- { 0.00000f, 0.09375f, 1.00000f, 1.00000f },
- { 1.00000f, 1.00000f, 1.00000f, 1.00000f },
- { 0.03125f, 0.12500f, 1.00000f, 1.00000f },
- { 0.06250f, 0.12500f, 0.00000f, 1.00000f },
- { 0.00000f, 1.00000f, 0.00000f, 0.56250f },
- }, {
- { 0.00000f, 0.50000f, 0.00000f, 0.71875f },
- { 0.06250f, 0.84375f, 0.00000f, 0.87500f },
- { 1.00000f, 0.50000f, 0.50000f, 0.96875f },
- { 1.00000f, 0.09375f, 0.31250f, 0.50000f },
- { 1.00000f, 1.00000f, 1.00000f, 1.00000f },
- { 1.00000f, 0.03125f, 0.03125f, 0.53125f },
- { 0.18750f, 0.12500f, 0.00000f, 1.00000f },
- { 0.00000f, 1.00000f, 0.03125f, 0.18750f },
- }, {
- { 0.00000f, 0.59375f, 0.00000f, 0.96875f },
- { 0.06250f, 0.81250f, 0.06250f, 0.59375f },
- { 0.75000f, 0.43750f, 0.12500f, 0.96875f },
- { 0.87500f, 0.06250f, 0.18750f, 0.43750f },
- { 1.00000f, 1.00000f, 1.00000f, 1.00000f },
- { 0.15625f, 0.12500f, 1.00000f, 1.00000f },
- { 0.06250f, 0.12500f, 0.00000f, 1.00000f },
- { 0.00000f, 1.00000f, 0.03125f, 0.34375f },
- }
- };
- // structure that allow us to query and override info for training the costs
- typedef struct STBIR__V_FIRST_INFO
- {
- double v_cost, h_cost;
- int control_v_first; // 0 = no control, 1 = force hori, 2 = force vert
- int v_first;
- int v_resize_classification;
- int is_gather;
- } STBIR__V_FIRST_INFO;
- #ifdef STBIR__V_FIRST_INFO_BUFFER
- static STBIR__V_FIRST_INFO STBIR__V_FIRST_INFO_BUFFER = {0};
- #define STBIR__V_FIRST_INFO_POINTER &STBIR__V_FIRST_INFO_BUFFER
- #else
- #define STBIR__V_FIRST_INFO_POINTER 0
- #endif
- // Figure out whether to scale along the horizontal or vertical first.
- // This only *super* important when you are scaling by a massively
- // different amount in the vertical vs the horizontal (for example, if
- // you are scaling by 2x in the width, and 0.5x in the height, then you
- // want to do the vertical scale first, because it's around 3x faster
- // in that order.
- //
- // In more normal circumstances, this makes a 20-40% differences, so
- // it's good to get right, but not critical. The normal way that you
- // decide which direction goes first is just figuring out which
- // direction does more multiplies. But with modern CPUs with their
- // fancy caches and SIMD and high IPC abilities, so there's just a lot
- // more that goes into it.
- //
- // My handwavy sort of solution is to have an app that does a whole
- // bunch of timing for both vertical and horizontal first modes,
- // and then another app that can read lots of these timing files
- // and try to search for the best weights to use. Dotimings.c
- // is the app that does a bunch of timings, and vf_train.c is the
- // app that solves for the best weights (and shows how well it
- // does currently).
- static int stbir__should_do_vertical_first( float weights_table[STBIR_RESIZE_CLASSIFICATIONS][4], int horizontal_filter_pixel_width, float horizontal_scale, int horizontal_output_size, int vertical_filter_pixel_width, float vertical_scale, int vertical_output_size, int is_gather, STBIR__V_FIRST_INFO * info )
- {
- double v_cost, h_cost;
- float * weights;
- int vertical_first;
- int v_classification;
- // categorize the resize into buckets
- if ( ( vertical_output_size <= 4 ) || ( horizontal_output_size <= 4 ) )
- v_classification = ( vertical_output_size < horizontal_output_size ) ? 6 : 7;
- else if ( vertical_scale <= 1.0f )
- v_classification = ( is_gather ) ? 1 : 0;
- else if ( vertical_scale <= 2.0f)
- v_classification = 2;
- else if ( vertical_scale <= 3.0f)
- v_classification = 3;
- else if ( vertical_scale <= 4.0f)
- v_classification = 5;
- else
- v_classification = 6;
- // use the right weights
- weights = weights_table[ v_classification ];
- // this is the costs when you don't take into account modern CPUs with high ipc and simd and caches - wish we had a better estimate
- h_cost = (float)horizontal_filter_pixel_width * weights[0] + horizontal_scale * (float)vertical_filter_pixel_width * weights[1];
- v_cost = (float)vertical_filter_pixel_width * weights[2] + vertical_scale * (float)horizontal_filter_pixel_width * weights[3];
- // use computation estimate to decide vertical first or not
- vertical_first = ( v_cost <= h_cost ) ? 1 : 0;
- // save these, if requested
- if ( info )
- {
- info->h_cost = h_cost;
- info->v_cost = v_cost;
- info->v_resize_classification = v_classification;
- info->v_first = vertical_first;
- info->is_gather = is_gather;
- }
- // and this allows us to override everything for testing (see dotiming.c)
- if ( ( info ) && ( info->control_v_first ) )
- vertical_first = ( info->control_v_first == 2 ) ? 1 : 0;
- return vertical_first;
- }
- // layout lookups - must match stbir_internal_pixel_layout
- static unsigned char stbir__pixel_channels[] = {
- 1,2,3,3,4, // 1ch, 2ch, rgb, bgr, 4ch
- 4,4,4,4,2,2, // RGBA,BGRA,ARGB,ABGR,RA,AR
- 4,4,4,4,2,2, // RGBA_PM,BGRA_PM,ARGB_PM,ABGR_PM,RA_PM,AR_PM
- };
- // the internal pixel layout enums are in a different order, so we can easily do range comparisons of types
- // the public pixel layout is ordered in a way that if you cast num_channels (1-4) to the enum, you get something sensible
- static stbir_internal_pixel_layout stbir__pixel_layout_convert_public_to_internal[] = {
- STBIRI_BGR, STBIRI_1CHANNEL, STBIRI_2CHANNEL, STBIRI_RGB, STBIRI_RGBA,
- STBIRI_4CHANNEL, STBIRI_BGRA, STBIRI_ARGB, STBIRI_ABGR, STBIRI_RA, STBIRI_AR,
- STBIRI_RGBA_PM, STBIRI_BGRA_PM, STBIRI_ARGB_PM, STBIRI_ABGR_PM, STBIRI_RA_PM, STBIRI_AR_PM,
- };
- static stbir__info * stbir__alloc_internal_mem_and_build_samplers( stbir__sampler * horizontal, stbir__sampler * vertical, stbir__contributors * conservative, stbir_pixel_layout input_pixel_layout_public, stbir_pixel_layout output_pixel_layout_public, int splits, int new_x, int new_y, int fast_alpha, void * user_data STBIR_ONLY_PROFILE_BUILD_GET_INFO )
- {
- static char stbir_channel_count_index[8]={ 9,0,1,2, 3,9,9,4 };
- stbir__info * info = 0;
- void * alloced = 0;
- size_t alloced_total = 0;
- int vertical_first;
- size_t decode_buffer_size, ring_buffer_length_bytes, ring_buffer_size, vertical_buffer_size;
- int alloc_ring_buffer_num_entries;
- int alpha_weighting_type = 0; // 0=none, 1=simple, 2=fancy
- int conservative_split_output_size = stbir__get_max_split( splits, vertical->scale_info.output_sub_size );
- stbir_internal_pixel_layout input_pixel_layout = stbir__pixel_layout_convert_public_to_internal[ input_pixel_layout_public ];
- stbir_internal_pixel_layout output_pixel_layout = stbir__pixel_layout_convert_public_to_internal[ output_pixel_layout_public ];
- int channels = stbir__pixel_channels[ input_pixel_layout ];
- int effective_channels = channels;
- // first figure out what type of alpha weighting to use (if any)
- if ( ( horizontal->filter_enum != STBIR_FILTER_POINT_SAMPLE ) || ( vertical->filter_enum != STBIR_FILTER_POINT_SAMPLE ) ) // no alpha weighting on point sampling
- {
- if ( ( input_pixel_layout >= STBIRI_RGBA ) && ( input_pixel_layout <= STBIRI_AR ) && ( output_pixel_layout >= STBIRI_RGBA ) && ( output_pixel_layout <= STBIRI_AR ) )
- {
- if ( fast_alpha )
- {
- alpha_weighting_type = 4;
- }
- else
- {
- static int fancy_alpha_effective_cnts[6] = { 7, 7, 7, 7, 3, 3 };
- alpha_weighting_type = 2;
- effective_channels = fancy_alpha_effective_cnts[ input_pixel_layout - STBIRI_RGBA ];
- }
- }
- else if ( ( input_pixel_layout >= STBIRI_RGBA_PM ) && ( input_pixel_layout <= STBIRI_AR_PM ) && ( output_pixel_layout >= STBIRI_RGBA ) && ( output_pixel_layout <= STBIRI_AR ) )
- {
- // input premult, output non-premult
- alpha_weighting_type = 3;
- }
- else if ( ( input_pixel_layout >= STBIRI_RGBA ) && ( input_pixel_layout <= STBIRI_AR ) && ( output_pixel_layout >= STBIRI_RGBA_PM ) && ( output_pixel_layout <= STBIRI_AR_PM ) )
- {
- // input non-premult, output premult
- alpha_weighting_type = 1;
- }
- }
- // channel in and out count must match currently
- if ( channels != stbir__pixel_channels[ output_pixel_layout ] )
- return 0;
- // get vertical first
- vertical_first = stbir__should_do_vertical_first( stbir__compute_weights[ (int)stbir_channel_count_index[ effective_channels ] ], horizontal->filter_pixel_width, horizontal->scale_info.scale, horizontal->scale_info.output_sub_size, vertical->filter_pixel_width, vertical->scale_info.scale, vertical->scale_info.output_sub_size, vertical->is_gather, STBIR__V_FIRST_INFO_POINTER );
- // sometimes read one float off in some of the unrolled loops (with a weight of zero coeff, so it doesn't have an effect)
- // we use a few extra floats instead of just 1, so that input callback buffer can overlap with the decode buffer without
- // the conversion routines overwriting the callback input data.
- decode_buffer_size = ( conservative->n1 - conservative->n0 + 1 ) * effective_channels * sizeof(float) + sizeof(float)*STBIR_INPUT_CALLBACK_PADDING; // extra floats for input callback stagger
- #if defined( STBIR__SEPARATE_ALLOCATIONS ) && defined(STBIR_SIMD8)
- if ( effective_channels == 3 )
- decode_buffer_size += sizeof(float); // avx in 3 channel mode needs one float at the start of the buffer (only with separate allocations)
- #endif
- ring_buffer_length_bytes = (size_t)horizontal->scale_info.output_sub_size * (size_t)effective_channels * sizeof(float) + sizeof(float)*STBIR_INPUT_CALLBACK_PADDING; // extra floats for padding
- // if we do vertical first, the ring buffer holds a whole decoded line
- if ( vertical_first )
- ring_buffer_length_bytes = ( decode_buffer_size + 15 ) & ~15;
- if ( ( ring_buffer_length_bytes & 4095 ) == 0 ) ring_buffer_length_bytes += 64*3; // avoid 4k alias
- // One extra entry because floating point precision problems sometimes cause an extra to be necessary.
- alloc_ring_buffer_num_entries = vertical->filter_pixel_width + 1;
- // we never need more ring buffer entries than the scanlines we're outputting when in scatter mode
- if ( ( !vertical->is_gather ) && ( alloc_ring_buffer_num_entries > conservative_split_output_size ) )
- alloc_ring_buffer_num_entries = conservative_split_output_size;
- ring_buffer_size = (size_t)alloc_ring_buffer_num_entries * (size_t)ring_buffer_length_bytes;
- // The vertical buffer is used differently, depending on whether we are scattering
- // the vertical scanlines, or gathering them.
- // If scattering, it's used at the temp buffer to accumulate each output.
- // If gathering, it's just the output buffer.
- vertical_buffer_size = (size_t)horizontal->scale_info.output_sub_size * (size_t)effective_channels * sizeof(float) + sizeof(float); // extra float for padding
- // we make two passes through this loop, 1st to add everything up, 2nd to allocate and init
- for(;;)
- {
- int i;
- void * advance_mem = alloced;
- int copy_horizontal = 0;
- stbir__sampler * possibly_use_horizontal_for_pivot = 0;
- #ifdef STBIR__SEPARATE_ALLOCATIONS
- #define STBIR__NEXT_PTR( ptr, size, ntype ) if ( alloced ) { void * p = STBIR_MALLOC( size, user_data); if ( p == 0 ) { stbir__free_internal_mem( info ); return 0; } (ptr) = (ntype*)p; }
- #else
- #define STBIR__NEXT_PTR( ptr, size, ntype ) advance_mem = (void*) ( ( ((size_t)advance_mem) + 15 ) & ~15 ); if ( alloced ) ptr = (ntype*)advance_mem; advance_mem = (char*)(((size_t)advance_mem) + (size));
- #endif
- STBIR__NEXT_PTR( info, sizeof( stbir__info ), stbir__info );
- STBIR__NEXT_PTR( info->split_info, sizeof( stbir__per_split_info ) * splits, stbir__per_split_info );
- if ( info )
- {
- static stbir__alpha_weight_func * fancy_alpha_weights[6] = { stbir__fancy_alpha_weight_4ch, stbir__fancy_alpha_weight_4ch, stbir__fancy_alpha_weight_4ch, stbir__fancy_alpha_weight_4ch, stbir__fancy_alpha_weight_2ch, stbir__fancy_alpha_weight_2ch };
- static stbir__alpha_unweight_func * fancy_alpha_unweights[6] = { stbir__fancy_alpha_unweight_4ch, stbir__fancy_alpha_unweight_4ch, stbir__fancy_alpha_unweight_4ch, stbir__fancy_alpha_unweight_4ch, stbir__fancy_alpha_unweight_2ch, stbir__fancy_alpha_unweight_2ch };
- static stbir__alpha_weight_func * simple_alpha_weights[6] = { stbir__simple_alpha_weight_4ch, stbir__simple_alpha_weight_4ch, stbir__simple_alpha_weight_4ch, stbir__simple_alpha_weight_4ch, stbir__simple_alpha_weight_2ch, stbir__simple_alpha_weight_2ch };
- static stbir__alpha_unweight_func * simple_alpha_unweights[6] = { stbir__simple_alpha_unweight_4ch, stbir__simple_alpha_unweight_4ch, stbir__simple_alpha_unweight_4ch, stbir__simple_alpha_unweight_4ch, stbir__simple_alpha_unweight_2ch, stbir__simple_alpha_unweight_2ch };
- // initialize info fields
- info->alloced_mem = alloced;
- info->alloced_total = alloced_total;
- info->channels = channels;
- info->effective_channels = effective_channels;
- info->offset_x = new_x;
- info->offset_y = new_y;
- info->alloc_ring_buffer_num_entries = (int)alloc_ring_buffer_num_entries;
- info->ring_buffer_num_entries = 0;
- info->ring_buffer_length_bytes = (int)ring_buffer_length_bytes;
- info->splits = splits;
- info->vertical_first = vertical_first;
- info->input_pixel_layout_internal = input_pixel_layout;
- info->output_pixel_layout_internal = output_pixel_layout;
- // setup alpha weight functions
- info->alpha_weight = 0;
- info->alpha_unweight = 0;
- // handle alpha weighting functions and overrides
- if ( alpha_weighting_type == 2 )
- {
- // high quality alpha multiplying on the way in, dividing on the way out
- info->alpha_weight = fancy_alpha_weights[ input_pixel_layout - STBIRI_RGBA ];
- info->alpha_unweight = fancy_alpha_unweights[ output_pixel_layout - STBIRI_RGBA ];
- }
- else if ( alpha_weighting_type == 4 )
- {
- // fast alpha multiplying on the way in, dividing on the way out
- info->alpha_weight = simple_alpha_weights[ input_pixel_layout - STBIRI_RGBA ];
- info->alpha_unweight = simple_alpha_unweights[ output_pixel_layout - STBIRI_RGBA ];
- }
- else if ( alpha_weighting_type == 1 )
- {
- // fast alpha on the way in, leave in premultiplied form on way out
- info->alpha_weight = simple_alpha_weights[ input_pixel_layout - STBIRI_RGBA ];
- }
- else if ( alpha_weighting_type == 3 )
- {
- // incoming is premultiplied, fast alpha dividing on the way out - non-premultiplied output
- info->alpha_unweight = simple_alpha_unweights[ output_pixel_layout - STBIRI_RGBA ];
- }
- // handle 3-chan color flipping, using the alpha weight path
- if ( ( ( input_pixel_layout == STBIRI_RGB ) && ( output_pixel_layout == STBIRI_BGR ) ) ||
- ( ( input_pixel_layout == STBIRI_BGR ) && ( output_pixel_layout == STBIRI_RGB ) ) )
- {
- // do the flipping on the smaller of the two ends
- if ( horizontal->scale_info.scale < 1.0f )
- info->alpha_unweight = stbir__simple_flip_3ch;
- else
- info->alpha_weight = stbir__simple_flip_3ch;
- }
- }
- // get all the per-split buffers
- for( i = 0 ; i < splits ; i++ )
- {
- STBIR__NEXT_PTR( info->split_info[i].decode_buffer, decode_buffer_size, float );
- #ifdef STBIR__SEPARATE_ALLOCATIONS
- #ifdef STBIR_SIMD8
- if ( ( info ) && ( effective_channels == 3 ) )
- ++info->split_info[i].decode_buffer; // avx in 3 channel mode needs one float at the start of the buffer
- #endif
- STBIR__NEXT_PTR( info->split_info[i].ring_buffers, alloc_ring_buffer_num_entries * sizeof(float*), float* );
- {
- int j;
- for( j = 0 ; j < alloc_ring_buffer_num_entries ; j++ )
- {
- STBIR__NEXT_PTR( info->split_info[i].ring_buffers[j], ring_buffer_length_bytes, float );
- #ifdef STBIR_SIMD8
- if ( ( info ) && ( effective_channels == 3 ) )
- ++info->split_info[i].ring_buffers[j]; // avx in 3 channel mode needs one float at the start of the buffer
- #endif
- }
- }
- #else
- STBIR__NEXT_PTR( info->split_info[i].ring_buffer, ring_buffer_size, float );
- #endif
- STBIR__NEXT_PTR( info->split_info[i].vertical_buffer, vertical_buffer_size, float );
- }
- // alloc memory for to-be-pivoted coeffs (if necessary)
- if ( vertical->is_gather == 0 )
- {
- size_t both;
- size_t temp_mem_amt;
- // when in vertical scatter mode, we first build the coefficients in gather mode, and then pivot after,
- // that means we need two buffers, so we try to use the decode buffer and ring buffer for this. if that
- // is too small, we just allocate extra memory to use as this temp.
- both = (size_t)vertical->gather_prescatter_contributors_size + (size_t)vertical->gather_prescatter_coefficients_size;
- #ifdef STBIR__SEPARATE_ALLOCATIONS
- temp_mem_amt = decode_buffer_size;
- #ifdef STBIR_SIMD8
- if ( effective_channels == 3 )
- --temp_mem_amt; // avx in 3 channel mode needs one float at the start of the buffer
- #endif
- #else
- temp_mem_amt = (size_t)( decode_buffer_size + ring_buffer_size + vertical_buffer_size ) * (size_t)splits;
- #endif
- if ( temp_mem_amt >= both )
- {
- if ( info )
- {
- vertical->gather_prescatter_contributors = (stbir__contributors*)info->split_info[0].decode_buffer;
- vertical->gather_prescatter_coefficients = (float*) ( ( (char*)info->split_info[0].decode_buffer ) + vertical->gather_prescatter_contributors_size );
- }
- }
- else
- {
- // ring+decode memory is too small, so allocate temp memory
- STBIR__NEXT_PTR( vertical->gather_prescatter_contributors, vertical->gather_prescatter_contributors_size, stbir__contributors );
- STBIR__NEXT_PTR( vertical->gather_prescatter_coefficients, vertical->gather_prescatter_coefficients_size, float );
- }
- }
- STBIR__NEXT_PTR( horizontal->contributors, horizontal->contributors_size, stbir__contributors );
- STBIR__NEXT_PTR( horizontal->coefficients, horizontal->coefficients_size, float );
- // are the two filters identical?? (happens a lot with mipmap generation)
- if ( ( horizontal->filter_kernel == vertical->filter_kernel ) && ( horizontal->filter_support == vertical->filter_support ) && ( horizontal->edge == vertical->edge ) && ( horizontal->scale_info.output_sub_size == vertical->scale_info.output_sub_size ) )
- {
- float diff_scale = horizontal->scale_info.scale - vertical->scale_info.scale;
- float diff_shift = horizontal->scale_info.pixel_shift - vertical->scale_info.pixel_shift;
- if ( diff_scale < 0.0f ) diff_scale = -diff_scale;
- if ( diff_shift < 0.0f ) diff_shift = -diff_shift;
- if ( ( diff_scale <= stbir__small_float ) && ( diff_shift <= stbir__small_float ) )
- {
- if ( horizontal->is_gather == vertical->is_gather )
- {
- copy_horizontal = 1;
- goto no_vert_alloc;
- }
- // everything matches, but vertical is scatter, horizontal is gather, use horizontal coeffs for vertical pivot coeffs
- possibly_use_horizontal_for_pivot = horizontal;
- }
- }
- STBIR__NEXT_PTR( vertical->contributors, vertical->contributors_size, stbir__contributors );
- STBIR__NEXT_PTR( vertical->coefficients, vertical->coefficients_size, float );
- no_vert_alloc:
- if ( info )
- {
- STBIR_PROFILE_BUILD_START( horizontal );
- stbir__calculate_filters( horizontal, 0, user_data STBIR_ONLY_PROFILE_BUILD_SET_INFO );
- // setup the horizontal gather functions
- // start with defaulting to the n_coeffs functions (specialized on channels and remnant leftover)
- info->horizontal_gather_channels = stbir__horizontal_gather_n_coeffs_funcs[ effective_channels ][ horizontal->extent_info.widest & 3 ];
- // but if the number of coeffs <= 12, use another set of special cases. <=12 coeffs is any enlarging resize, or shrinking resize down to about 1/3 size
- if ( horizontal->extent_info.widest <= 12 )
- info->horizontal_gather_channels = stbir__horizontal_gather_channels_funcs[ effective_channels ][ horizontal->extent_info.widest - 1 ];
- info->scanline_extents.conservative.n0 = conservative->n0;
- info->scanline_extents.conservative.n1 = conservative->n1;
- // get exact extents
- stbir__get_extents( horizontal, &info->scanline_extents );
- // pack the horizontal coeffs
- horizontal->coefficient_width = stbir__pack_coefficients(horizontal->num_contributors, horizontal->contributors, horizontal->coefficients, horizontal->coefficient_width, horizontal->extent_info.widest, info->scanline_extents.conservative.n0, info->scanline_extents.conservative.n1 );
- STBIR_MEMCPY( &info->horizontal, horizontal, sizeof( stbir__sampler ) );
- STBIR_PROFILE_BUILD_END( horizontal );
- if ( copy_horizontal )
- {
- STBIR_MEMCPY( &info->vertical, horizontal, sizeof( stbir__sampler ) );
- }
- else
- {
- STBIR_PROFILE_BUILD_START( vertical );
- stbir__calculate_filters( vertical, possibly_use_horizontal_for_pivot, user_data STBIR_ONLY_PROFILE_BUILD_SET_INFO );
- STBIR_MEMCPY( &info->vertical, vertical, sizeof( stbir__sampler ) );
- STBIR_PROFILE_BUILD_END( vertical );
- }
- // setup the vertical split ranges
- stbir__get_split_info( info->split_info, info->splits, info->vertical.scale_info.output_sub_size, info->vertical.filter_pixel_margin, info->vertical.scale_info.input_full_size, info->vertical.is_gather, info->vertical.contributors );
- // now we know precisely how many entries we need
- info->ring_buffer_num_entries = info->vertical.extent_info.widest;
- // we never need more ring buffer entries than the scanlines we're outputting
- if ( ( !info->vertical.is_gather ) && ( info->ring_buffer_num_entries > conservative_split_output_size ) )
- info->ring_buffer_num_entries = conservative_split_output_size;
- STBIR_ASSERT( info->ring_buffer_num_entries <= info->alloc_ring_buffer_num_entries );
- }
- #undef STBIR__NEXT_PTR
- // is this the first time through loop?
- if ( info == 0 )
- {
- alloced_total = ( 15 + (size_t)advance_mem );
- alloced = STBIR_MALLOC( alloced_total, user_data );
- if ( alloced == 0 )
- return 0;
- }
- else
- return info; // success
- }
- }
- static int stbir__perform_resize( stbir__info const * info, int split_start, int split_count )
- {
- stbir__per_split_info * split_info = info->split_info + split_start;
- STBIR_PROFILE_CLEAR_EXTRAS();
- STBIR_PROFILE_FIRST_START( looping );
- if (info->vertical.is_gather)
- stbir__vertical_gather_loop( info, split_info, split_count );
- else
- stbir__vertical_scatter_loop( info, split_info, split_count );
- STBIR_PROFILE_END( looping );
- return 1;
- }
- static void stbir__update_info_from_resize( stbir__info * info, STBIR_RESIZE * resize )
- {
- static stbir__decode_pixels_func * decode_simple[STBIR_TYPE_HALF_FLOAT-STBIR_TYPE_UINT8_SRGB+1]=
- {
- /* 1ch-4ch */ stbir__decode_uint8_srgb, stbir__decode_uint8_srgb, 0, stbir__decode_float_linear, stbir__decode_half_float_linear,
- };
- static stbir__decode_pixels_func * decode_alphas[STBIRI_AR-STBIRI_RGBA+1][STBIR_TYPE_HALF_FLOAT-STBIR_TYPE_UINT8_SRGB+1]=
- {
- { /* RGBA */ stbir__decode_uint8_srgb4_linearalpha, stbir__decode_uint8_srgb, 0, stbir__decode_float_linear, stbir__decode_half_float_linear },
- { /* BGRA */ stbir__decode_uint8_srgb4_linearalpha_BGRA, stbir__decode_uint8_srgb_BGRA, 0, stbir__decode_float_linear_BGRA, stbir__decode_half_float_linear_BGRA },
- { /* ARGB */ stbir__decode_uint8_srgb4_linearalpha_ARGB, stbir__decode_uint8_srgb_ARGB, 0, stbir__decode_float_linear_ARGB, stbir__decode_half_float_linear_ARGB },
- { /* ABGR */ stbir__decode_uint8_srgb4_linearalpha_ABGR, stbir__decode_uint8_srgb_ABGR, 0, stbir__decode_float_linear_ABGR, stbir__decode_half_float_linear_ABGR },
- { /* RA */ stbir__decode_uint8_srgb2_linearalpha, stbir__decode_uint8_srgb, 0, stbir__decode_float_linear, stbir__decode_half_float_linear },
- { /* AR */ stbir__decode_uint8_srgb2_linearalpha_AR, stbir__decode_uint8_srgb_AR, 0, stbir__decode_float_linear_AR, stbir__decode_half_float_linear_AR },
- };
- static stbir__decode_pixels_func * decode_simple_scaled_or_not[2][2]=
- {
- { stbir__decode_uint8_linear_scaled, stbir__decode_uint8_linear }, { stbir__decode_uint16_linear_scaled, stbir__decode_uint16_linear },
- };
- static stbir__decode_pixels_func * decode_alphas_scaled_or_not[STBIRI_AR-STBIRI_RGBA+1][2][2]=
- {
- { /* RGBA */ { stbir__decode_uint8_linear_scaled, stbir__decode_uint8_linear }, { stbir__decode_uint16_linear_scaled, stbir__decode_uint16_linear } },
- { /* BGRA */ { stbir__decode_uint8_linear_scaled_BGRA, stbir__decode_uint8_linear_BGRA }, { stbir__decode_uint16_linear_scaled_BGRA, stbir__decode_uint16_linear_BGRA } },
- { /* ARGB */ { stbir__decode_uint8_linear_scaled_ARGB, stbir__decode_uint8_linear_ARGB }, { stbir__decode_uint16_linear_scaled_ARGB, stbir__decode_uint16_linear_ARGB } },
- { /* ABGR */ { stbir__decode_uint8_linear_scaled_ABGR, stbir__decode_uint8_linear_ABGR }, { stbir__decode_uint16_linear_scaled_ABGR, stbir__decode_uint16_linear_ABGR } },
- { /* RA */ { stbir__decode_uint8_linear_scaled, stbir__decode_uint8_linear }, { stbir__decode_uint16_linear_scaled, stbir__decode_uint16_linear } },
- { /* AR */ { stbir__decode_uint8_linear_scaled_AR, stbir__decode_uint8_linear_AR }, { stbir__decode_uint16_linear_scaled_AR, stbir__decode_uint16_linear_AR } }
- };
- static stbir__encode_pixels_func * encode_simple[STBIR_TYPE_HALF_FLOAT-STBIR_TYPE_UINT8_SRGB+1]=
- {
- /* 1ch-4ch */ stbir__encode_uint8_srgb, stbir__encode_uint8_srgb, 0, stbir__encode_float_linear, stbir__encode_half_float_linear,
- };
- static stbir__encode_pixels_func * encode_alphas[STBIRI_AR-STBIRI_RGBA+1][STBIR_TYPE_HALF_FLOAT-STBIR_TYPE_UINT8_SRGB+1]=
- {
- { /* RGBA */ stbir__encode_uint8_srgb4_linearalpha, stbir__encode_uint8_srgb, 0, stbir__encode_float_linear, stbir__encode_half_float_linear },
- { /* BGRA */ stbir__encode_uint8_srgb4_linearalpha_BGRA, stbir__encode_uint8_srgb_BGRA, 0, stbir__encode_float_linear_BGRA, stbir__encode_half_float_linear_BGRA },
- { /* ARGB */ stbir__encode_uint8_srgb4_linearalpha_ARGB, stbir__encode_uint8_srgb_ARGB, 0, stbir__encode_float_linear_ARGB, stbir__encode_half_float_linear_ARGB },
- { /* ABGR */ stbir__encode_uint8_srgb4_linearalpha_ABGR, stbir__encode_uint8_srgb_ABGR, 0, stbir__encode_float_linear_ABGR, stbir__encode_half_float_linear_ABGR },
- { /* RA */ stbir__encode_uint8_srgb2_linearalpha, stbir__encode_uint8_srgb, 0, stbir__encode_float_linear, stbir__encode_half_float_linear },
- { /* AR */ stbir__encode_uint8_srgb2_linearalpha_AR, stbir__encode_uint8_srgb_AR, 0, stbir__encode_float_linear_AR, stbir__encode_half_float_linear_AR }
- };
- static stbir__encode_pixels_func * encode_simple_scaled_or_not[2][2]=
- {
- { stbir__encode_uint8_linear_scaled, stbir__encode_uint8_linear }, { stbir__encode_uint16_linear_scaled, stbir__encode_uint16_linear },
- };
- static stbir__encode_pixels_func * encode_alphas_scaled_or_not[STBIRI_AR-STBIRI_RGBA+1][2][2]=
- {
- { /* RGBA */ { stbir__encode_uint8_linear_scaled, stbir__encode_uint8_linear }, { stbir__encode_uint16_linear_scaled, stbir__encode_uint16_linear } },
- { /* BGRA */ { stbir__encode_uint8_linear_scaled_BGRA, stbir__encode_uint8_linear_BGRA }, { stbir__encode_uint16_linear_scaled_BGRA, stbir__encode_uint16_linear_BGRA } },
- { /* ARGB */ { stbir__encode_uint8_linear_scaled_ARGB, stbir__encode_uint8_linear_ARGB }, { stbir__encode_uint16_linear_scaled_ARGB, stbir__encode_uint16_linear_ARGB } },
- { /* ABGR */ { stbir__encode_uint8_linear_scaled_ABGR, stbir__encode_uint8_linear_ABGR }, { stbir__encode_uint16_linear_scaled_ABGR, stbir__encode_uint16_linear_ABGR } },
- { /* RA */ { stbir__encode_uint8_linear_scaled, stbir__encode_uint8_linear }, { stbir__encode_uint16_linear_scaled, stbir__encode_uint16_linear } },
- { /* AR */ { stbir__encode_uint8_linear_scaled_AR, stbir__encode_uint8_linear_AR }, { stbir__encode_uint16_linear_scaled_AR, stbir__encode_uint16_linear_AR } }
- };
- stbir__decode_pixels_func * decode_pixels = 0;
- stbir__encode_pixels_func * encode_pixels = 0;
- stbir_datatype input_type, output_type;
- input_type = resize->input_data_type;
- output_type = resize->output_data_type;
- info->input_data = resize->input_pixels;
- info->input_stride_bytes = resize->input_stride_in_bytes;
- info->output_stride_bytes = resize->output_stride_in_bytes;
- // if we're completely point sampling, then we can turn off SRGB
- if ( ( info->horizontal.filter_enum == STBIR_FILTER_POINT_SAMPLE ) && ( info->vertical.filter_enum == STBIR_FILTER_POINT_SAMPLE ) )
- {
- if ( ( ( input_type == STBIR_TYPE_UINT8_SRGB ) || ( input_type == STBIR_TYPE_UINT8_SRGB_ALPHA ) ) &&
- ( ( output_type == STBIR_TYPE_UINT8_SRGB ) || ( output_type == STBIR_TYPE_UINT8_SRGB_ALPHA ) ) )
- {
- input_type = STBIR_TYPE_UINT8;
- output_type = STBIR_TYPE_UINT8;
- }
- }
- // recalc the output and input strides
- if ( info->input_stride_bytes == 0 )
- info->input_stride_bytes = info->channels * info->horizontal.scale_info.input_full_size * stbir__type_size[input_type];
- if ( info->output_stride_bytes == 0 )
- info->output_stride_bytes = info->channels * info->horizontal.scale_info.output_sub_size * stbir__type_size[output_type];
- // calc offset
- info->output_data = ( (char*) resize->output_pixels ) + ( (size_t) info->offset_y * (size_t) resize->output_stride_in_bytes ) + ( info->offset_x * info->channels * stbir__type_size[output_type] );
- info->in_pixels_cb = resize->input_cb;
- info->user_data = resize->user_data;
- info->out_pixels_cb = resize->output_cb;
- // setup the input format converters
- if ( ( input_type == STBIR_TYPE_UINT8 ) || ( input_type == STBIR_TYPE_UINT16 ) )
- {
- int non_scaled = 0;
- // check if we can run unscaled - 0-255.0/0-65535.0 instead of 0-1.0 (which is a tiny bit faster when doing linear 8->8 or 16->16)
- if ( ( !info->alpha_weight ) && ( !info->alpha_unweight ) ) // don't short circuit when alpha weighting (get everything to 0-1.0 as usual)
- if ( ( ( input_type == STBIR_TYPE_UINT8 ) && ( output_type == STBIR_TYPE_UINT8 ) ) || ( ( input_type == STBIR_TYPE_UINT16 ) && ( output_type == STBIR_TYPE_UINT16 ) ) )
- non_scaled = 1;
- if ( info->input_pixel_layout_internal <= STBIRI_4CHANNEL )
- decode_pixels = decode_simple_scaled_or_not[ input_type == STBIR_TYPE_UINT16 ][ non_scaled ];
- else
- decode_pixels = decode_alphas_scaled_or_not[ ( info->input_pixel_layout_internal - STBIRI_RGBA ) % ( STBIRI_AR-STBIRI_RGBA+1 ) ][ input_type == STBIR_TYPE_UINT16 ][ non_scaled ];
- }
- else
- {
- if ( info->input_pixel_layout_internal <= STBIRI_4CHANNEL )
- decode_pixels = decode_simple[ input_type - STBIR_TYPE_UINT8_SRGB ];
- else
- decode_pixels = decode_alphas[ ( info->input_pixel_layout_internal - STBIRI_RGBA ) % ( STBIRI_AR-STBIRI_RGBA+1 ) ][ input_type - STBIR_TYPE_UINT8_SRGB ];
- }
- // setup the output format converters
- if ( ( output_type == STBIR_TYPE_UINT8 ) || ( output_type == STBIR_TYPE_UINT16 ) )
- {
- int non_scaled = 0;
- // check if we can run unscaled - 0-255.0/0-65535.0 instead of 0-1.0 (which is a tiny bit faster when doing linear 8->8 or 16->16)
- if ( ( !info->alpha_weight ) && ( !info->alpha_unweight ) ) // don't short circuit when alpha weighting (get everything to 0-1.0 as usual)
- if ( ( ( input_type == STBIR_TYPE_UINT8 ) && ( output_type == STBIR_TYPE_UINT8 ) ) || ( ( input_type == STBIR_TYPE_UINT16 ) && ( output_type == STBIR_TYPE_UINT16 ) ) )
- non_scaled = 1;
- if ( info->output_pixel_layout_internal <= STBIRI_4CHANNEL )
- encode_pixels = encode_simple_scaled_or_not[ output_type == STBIR_TYPE_UINT16 ][ non_scaled ];
- else
- encode_pixels = encode_alphas_scaled_or_not[ ( info->output_pixel_layout_internal - STBIRI_RGBA ) % ( STBIRI_AR-STBIRI_RGBA+1 ) ][ output_type == STBIR_TYPE_UINT16 ][ non_scaled ];
- }
- else
- {
- if ( info->output_pixel_layout_internal <= STBIRI_4CHANNEL )
- encode_pixels = encode_simple[ output_type - STBIR_TYPE_UINT8_SRGB ];
- else
- encode_pixels = encode_alphas[ ( info->output_pixel_layout_internal - STBIRI_RGBA ) % ( STBIRI_AR-STBIRI_RGBA+1 ) ][ output_type - STBIR_TYPE_UINT8_SRGB ];
- }
- info->input_type = input_type;
- info->output_type = output_type;
- info->decode_pixels = decode_pixels;
- info->encode_pixels = encode_pixels;
- }
- static void stbir__clip( int * outx, int * outsubw, int outw, double * u0, double * u1 )
- {
- double per, adj;
- int over;
- // do left/top edge
- if ( *outx < 0 )
- {
- per = ( (double)*outx ) / ( (double)*outsubw ); // is negative
- adj = per * ( *u1 - *u0 );
- *u0 -= adj; // increases u0
- *outx = 0;
- }
- // do right/bot edge
- over = outw - ( *outx + *outsubw );
- if ( over < 0 )
- {
- per = ( (double)over ) / ( (double)*outsubw ); // is negative
- adj = per * ( *u1 - *u0 );
- *u1 += adj; // decrease u1
- *outsubw = outw - *outx;
- }
- }
- // converts a double to a rational that has less than one float bit of error (returns 0 if unable to do so)
- static int stbir__double_to_rational(double f, stbir_uint32 limit, stbir_uint32 *numer, stbir_uint32 *denom, int limit_denom ) // limit_denom (1) or limit numer (0)
- {
- double err;
- stbir_uint64 top, bot;
- stbir_uint64 numer_last = 0;
- stbir_uint64 denom_last = 1;
- stbir_uint64 numer_estimate = 1;
- stbir_uint64 denom_estimate = 0;
- // scale to past float error range
- top = (stbir_uint64)( f * (double)(1 << 25) );
- bot = 1 << 25;
- // keep refining, but usually stops in a few loops - usually 5 for bad cases
- for(;;)
- {
- stbir_uint64 est, temp;
- // hit limit, break out and do best full range estimate
- if ( ( ( limit_denom ) ? denom_estimate : numer_estimate ) >= limit )
- break;
- // is the current error less than 1 bit of a float? if so, we're done
- if ( denom_estimate )
- {
- err = ( (double)numer_estimate / (double)denom_estimate ) - f;
- if ( err < 0.0 ) err = -err;
- if ( err < ( 1.0 / (double)(1<<24) ) )
- {
- // yup, found it
- *numer = (stbir_uint32) numer_estimate;
- *denom = (stbir_uint32) denom_estimate;
- return 1;
- }
- }
- // no more refinement bits left? break out and do full range estimate
- if ( bot == 0 )
- break;
- // gcd the estimate bits
- est = top / bot;
- temp = top % bot;
- top = bot;
- bot = temp;
- // move remainders
- temp = est * denom_estimate + denom_last;
- denom_last = denom_estimate;
- denom_estimate = temp;
- // move remainders
- temp = est * numer_estimate + numer_last;
- numer_last = numer_estimate;
- numer_estimate = temp;
- }
- // we didn't fine anything good enough for float, use a full range estimate
- if ( limit_denom )
- {
- numer_estimate= (stbir_uint64)( f * (double)limit + 0.5 );
- denom_estimate = limit;
- }
- else
- {
- numer_estimate = limit;
- denom_estimate = (stbir_uint64)( ( (double)limit / f ) + 0.5 );
- }
- *numer = (stbir_uint32) numer_estimate;
- *denom = (stbir_uint32) denom_estimate;
- err = ( denom_estimate ) ? ( ( (double)(stbir_uint32)numer_estimate / (double)(stbir_uint32)denom_estimate ) - f ) : 1.0;
- if ( err < 0.0 ) err = -err;
- return ( err < ( 1.0 / (double)(1<<24) ) ) ? 1 : 0;
- }
- static int stbir__calculate_region_transform( stbir__scale_info * scale_info, int output_full_range, int * output_offset, int output_sub_range, int input_full_range, double input_s0, double input_s1 )
- {
- double output_range, input_range, output_s, input_s, ratio, scale;
- input_s = input_s1 - input_s0;
- // null area
- if ( ( output_full_range == 0 ) || ( input_full_range == 0 ) ||
- ( output_sub_range == 0 ) || ( input_s <= stbir__small_float ) )
- return 0;
- // are either of the ranges completely out of bounds?
- if ( ( *output_offset >= output_full_range ) || ( ( *output_offset + output_sub_range ) <= 0 ) || ( input_s0 >= (1.0f-stbir__small_float) ) || ( input_s1 <= stbir__small_float ) )
- return 0;
- output_range = (double)output_full_range;
- input_range = (double)input_full_range;
- output_s = ( (double)output_sub_range) / output_range;
- // figure out the scaling to use
- ratio = output_s / input_s;
- // save scale before clipping
- scale = ( output_range / input_range ) * ratio;
- scale_info->scale = (float)scale;
- scale_info->inv_scale = (float)( 1.0 / scale );
- // clip output area to left/right output edges (and adjust input area)
- stbir__clip( output_offset, &output_sub_range, output_full_range, &input_s0, &input_s1 );
- // recalc input area
- input_s = input_s1 - input_s0;
- // after clipping do we have zero input area?
- if ( input_s <= stbir__small_float )
- return 0;
- // calculate and store the starting source offsets in output pixel space
- scale_info->pixel_shift = (float) ( input_s0 * ratio * output_range );
- scale_info->scale_is_rational = stbir__double_to_rational( scale, ( scale <= 1.0 ) ? output_full_range : input_full_range, &scale_info->scale_numerator, &scale_info->scale_denominator, ( scale >= 1.0 ) );
- scale_info->input_full_size = input_full_range;
- scale_info->output_sub_size = output_sub_range;
- return 1;
- }
- static void stbir__init_and_set_layout( STBIR_RESIZE * resize, stbir_pixel_layout pixel_layout, stbir_datatype data_type )
- {
- resize->input_cb = 0;
- resize->output_cb = 0;
- resize->user_data = resize;
- resize->samplers = 0;
- resize->called_alloc = 0;
- resize->horizontal_filter = STBIR_FILTER_DEFAULT;
- resize->horizontal_filter_kernel = 0; resize->horizontal_filter_support = 0;
- resize->vertical_filter = STBIR_FILTER_DEFAULT;
- resize->vertical_filter_kernel = 0; resize->vertical_filter_support = 0;
- resize->horizontal_edge = STBIR_EDGE_CLAMP;
- resize->vertical_edge = STBIR_EDGE_CLAMP;
- resize->input_s0 = 0; resize->input_t0 = 0; resize->input_s1 = 1; resize->input_t1 = 1;
- resize->output_subx = 0; resize->output_suby = 0; resize->output_subw = resize->output_w; resize->output_subh = resize->output_h;
- resize->input_data_type = data_type;
- resize->output_data_type = data_type;
- resize->input_pixel_layout_public = pixel_layout;
- resize->output_pixel_layout_public = pixel_layout;
- resize->needs_rebuild = 1;
- }
- STBIRDEF void stbir_resize_init( STBIR_RESIZE * resize,
- const void *input_pixels, int input_w, int input_h, int input_stride_in_bytes, // stride can be zero
- void *output_pixels, int output_w, int output_h, int output_stride_in_bytes, // stride can be zero
- stbir_pixel_layout pixel_layout, stbir_datatype data_type )
- {
- resize->input_pixels = input_pixels;
- resize->input_w = input_w;
- resize->input_h = input_h;
- resize->input_stride_in_bytes = input_stride_in_bytes;
- resize->output_pixels = output_pixels;
- resize->output_w = output_w;
- resize->output_h = output_h;
- resize->output_stride_in_bytes = output_stride_in_bytes;
- resize->fast_alpha = 0;
- stbir__init_and_set_layout( resize, pixel_layout, data_type );
- }
- // You can update parameters any time after resize_init
- STBIRDEF void stbir_set_datatypes( STBIR_RESIZE * resize, stbir_datatype input_type, stbir_datatype output_type ) // by default, datatype from resize_init
- {
- resize->input_data_type = input_type;
- resize->output_data_type = output_type;
- if ( ( resize->samplers ) && ( !resize->needs_rebuild ) )
- stbir__update_info_from_resize( resize->samplers, resize );
- }
- STBIRDEF void stbir_set_pixel_callbacks( STBIR_RESIZE * resize, stbir_input_callback * input_cb, stbir_output_callback * output_cb ) // no callbacks by default
- {
- resize->input_cb = input_cb;
- resize->output_cb = output_cb;
- if ( ( resize->samplers ) && ( !resize->needs_rebuild ) )
- {
- resize->samplers->in_pixels_cb = input_cb;
- resize->samplers->out_pixels_cb = output_cb;
- }
- }
- STBIRDEF void stbir_set_user_data( STBIR_RESIZE * resize, void * user_data ) // pass back STBIR_RESIZE* by default
- {
- resize->user_data = user_data;
- if ( ( resize->samplers ) && ( !resize->needs_rebuild ) )
- resize->samplers->user_data = user_data;
- }
- STBIRDEF void stbir_set_buffer_ptrs( STBIR_RESIZE * resize, const void * input_pixels, int input_stride_in_bytes, void * output_pixels, int output_stride_in_bytes )
- {
- resize->input_pixels = input_pixels;
- resize->input_stride_in_bytes = input_stride_in_bytes;
- resize->output_pixels = output_pixels;
- resize->output_stride_in_bytes = output_stride_in_bytes;
- if ( ( resize->samplers ) && ( !resize->needs_rebuild ) )
- stbir__update_info_from_resize( resize->samplers, resize );
- }
- STBIRDEF int stbir_set_edgemodes( STBIR_RESIZE * resize, stbir_edge horizontal_edge, stbir_edge vertical_edge ) // CLAMP by default
- {
- resize->horizontal_edge = horizontal_edge;
- resize->vertical_edge = vertical_edge;
- resize->needs_rebuild = 1;
- return 1;
- }
- STBIRDEF int stbir_set_filters( STBIR_RESIZE * resize, stbir_filter horizontal_filter, stbir_filter vertical_filter ) // STBIR_DEFAULT_FILTER_UPSAMPLE/DOWNSAMPLE by default
- {
- resize->horizontal_filter = horizontal_filter;
- resize->vertical_filter = vertical_filter;
- resize->needs_rebuild = 1;
- return 1;
- }
- STBIRDEF int stbir_set_filter_callbacks( STBIR_RESIZE * resize, stbir__kernel_callback * horizontal_filter, stbir__support_callback * horizontal_support, stbir__kernel_callback * vertical_filter, stbir__support_callback * vertical_support )
- {
- resize->horizontal_filter_kernel = horizontal_filter; resize->horizontal_filter_support = horizontal_support;
- resize->vertical_filter_kernel = vertical_filter; resize->vertical_filter_support = vertical_support;
- resize->needs_rebuild = 1;
- return 1;
- }
- STBIRDEF int stbir_set_pixel_layouts( STBIR_RESIZE * resize, stbir_pixel_layout input_pixel_layout, stbir_pixel_layout output_pixel_layout ) // sets new pixel layouts
- {
- resize->input_pixel_layout_public = input_pixel_layout;
- resize->output_pixel_layout_public = output_pixel_layout;
- resize->needs_rebuild = 1;
- return 1;
- }
- STBIRDEF int stbir_set_non_pm_alpha_speed_over_quality( STBIR_RESIZE * resize, int non_pma_alpha_speed_over_quality ) // sets alpha speed
- {
- resize->fast_alpha = non_pma_alpha_speed_over_quality;
- resize->needs_rebuild = 1;
- return 1;
- }
- STBIRDEF int stbir_set_input_subrect( STBIR_RESIZE * resize, double s0, double t0, double s1, double t1 ) // sets input region (full region by default)
- {
- resize->input_s0 = s0;
- resize->input_t0 = t0;
- resize->input_s1 = s1;
- resize->input_t1 = t1;
- resize->needs_rebuild = 1;
- // are we inbounds?
- if ( ( s1 < stbir__small_float ) || ( (s1-s0) < stbir__small_float ) ||
- ( t1 < stbir__small_float ) || ( (t1-t0) < stbir__small_float ) ||
- ( s0 > (1.0f-stbir__small_float) ) ||
- ( t0 > (1.0f-stbir__small_float) ) )
- return 0;
- return 1;
- }
- STBIRDEF int stbir_set_output_pixel_subrect( STBIR_RESIZE * resize, int subx, int suby, int subw, int subh ) // sets input region (full region by default)
- {
- resize->output_subx = subx;
- resize->output_suby = suby;
- resize->output_subw = subw;
- resize->output_subh = subh;
- resize->needs_rebuild = 1;
- // are we inbounds?
- if ( ( subx >= resize->output_w ) || ( ( subx + subw ) <= 0 ) || ( suby >= resize->output_h ) || ( ( suby + subh ) <= 0 ) || ( subw == 0 ) || ( subh == 0 ) )
- return 0;
- return 1;
- }
- STBIRDEF int stbir_set_pixel_subrect( STBIR_RESIZE * resize, int subx, int suby, int subw, int subh ) // sets both regions (full regions by default)
- {
- double s0, t0, s1, t1;
- s0 = ( (double)subx ) / ( (double)resize->output_w );
- t0 = ( (double)suby ) / ( (double)resize->output_h );
- s1 = ( (double)(subx+subw) ) / ( (double)resize->output_w );
- t1 = ( (double)(suby+subh) ) / ( (double)resize->output_h );
- resize->input_s0 = s0;
- resize->input_t0 = t0;
- resize->input_s1 = s1;
- resize->input_t1 = t1;
- resize->output_subx = subx;
- resize->output_suby = suby;
- resize->output_subw = subw;
- resize->output_subh = subh;
- resize->needs_rebuild = 1;
- // are we inbounds?
- if ( ( subx >= resize->output_w ) || ( ( subx + subw ) <= 0 ) || ( suby >= resize->output_h ) || ( ( suby + subh ) <= 0 ) || ( subw == 0 ) || ( subh == 0 ) )
- return 0;
- return 1;
- }
- static int stbir__perform_build( STBIR_RESIZE * resize, int splits )
- {
- stbir__contributors conservative = { 0, 0 };
- stbir__sampler horizontal, vertical;
- int new_output_subx, new_output_suby;
- stbir__info * out_info;
- #ifdef STBIR_PROFILE
- stbir__info profile_infod; // used to contain building profile info before everything is allocated
- stbir__info * profile_info = &profile_infod;
- #endif
- // have we already built the samplers?
- if ( resize->samplers )
- return 0;
- #define STBIR_RETURN_ERROR_AND_ASSERT( exp ) STBIR_ASSERT( !(exp) ); if (exp) return 0;
- STBIR_RETURN_ERROR_AND_ASSERT( (unsigned)resize->horizontal_filter >= STBIR_FILTER_OTHER)
- STBIR_RETURN_ERROR_AND_ASSERT( (unsigned)resize->vertical_filter >= STBIR_FILTER_OTHER)
- #undef STBIR_RETURN_ERROR_AND_ASSERT
- if ( splits <= 0 )
- return 0;
- STBIR_PROFILE_BUILD_FIRST_START( build );
- new_output_subx = resize->output_subx;
- new_output_suby = resize->output_suby;
- // do horizontal clip and scale calcs
- if ( !stbir__calculate_region_transform( &horizontal.scale_info, resize->output_w, &new_output_subx, resize->output_subw, resize->input_w, resize->input_s0, resize->input_s1 ) )
- return 0;
- // do vertical clip and scale calcs
- if ( !stbir__calculate_region_transform( &vertical.scale_info, resize->output_h, &new_output_suby, resize->output_subh, resize->input_h, resize->input_t0, resize->input_t1 ) )
- return 0;
- // if nothing to do, just return
- if ( ( horizontal.scale_info.output_sub_size == 0 ) || ( vertical.scale_info.output_sub_size == 0 ) )
- return 0;
- stbir__set_sampler(&horizontal, resize->horizontal_filter, resize->horizontal_filter_kernel, resize->horizontal_filter_support, resize->horizontal_edge, &horizontal.scale_info, 1, resize->user_data );
- stbir__get_conservative_extents( &horizontal, &conservative, resize->user_data );
- stbir__set_sampler(&vertical, resize->vertical_filter, resize->vertical_filter_kernel, resize->vertical_filter_support, resize->vertical_edge, &vertical.scale_info, 0, resize->user_data );
- if ( ( vertical.scale_info.output_sub_size / splits ) < STBIR_FORCE_MINIMUM_SCANLINES_FOR_SPLITS ) // each split should be a minimum of 4 scanlines (handwavey choice)
- {
- splits = vertical.scale_info.output_sub_size / STBIR_FORCE_MINIMUM_SCANLINES_FOR_SPLITS;
- if ( splits == 0 ) splits = 1;
- }
- STBIR_PROFILE_BUILD_START( alloc );
- out_info = stbir__alloc_internal_mem_and_build_samplers( &horizontal, &vertical, &conservative, resize->input_pixel_layout_public, resize->output_pixel_layout_public, splits, new_output_subx, new_output_suby, resize->fast_alpha, resize->user_data STBIR_ONLY_PROFILE_BUILD_SET_INFO );
- STBIR_PROFILE_BUILD_END( alloc );
- STBIR_PROFILE_BUILD_END( build );
- if ( out_info )
- {
- resize->splits = splits;
- resize->samplers = out_info;
- resize->needs_rebuild = 0;
- #ifdef STBIR_PROFILE
- STBIR_MEMCPY( &out_info->profile, &profile_infod.profile, sizeof( out_info->profile ) );
- #endif
- // update anything that can be changed without recalcing samplers
- stbir__update_info_from_resize( out_info, resize );
- return splits;
- }
- return 0;
- }
- void stbir_free_samplers( STBIR_RESIZE * resize )
- {
- if ( resize->samplers )
- {
- stbir__free_internal_mem( resize->samplers );
- resize->samplers = 0;
- resize->called_alloc = 0;
- }
- }
- STBIRDEF int stbir_build_samplers_with_splits( STBIR_RESIZE * resize, int splits )
- {
- if ( ( resize->samplers == 0 ) || ( resize->needs_rebuild ) )
- {
- if ( resize->samplers )
- stbir_free_samplers( resize );
- resize->called_alloc = 1;
- return stbir__perform_build( resize, splits );
- }
- STBIR_PROFILE_BUILD_CLEAR( resize->samplers );
- return 1;
- }
- STBIRDEF int stbir_build_samplers( STBIR_RESIZE * resize )
- {
- return stbir_build_samplers_with_splits( resize, 1 );
- }
- STBIRDEF int stbir_resize_extended( STBIR_RESIZE * resize )
- {
- int result;
- if ( ( resize->samplers == 0 ) || ( resize->needs_rebuild ) )
- {
- int alloc_state = resize->called_alloc; // remember allocated state
- if ( resize->samplers )
- {
- stbir__free_internal_mem( resize->samplers );
- resize->samplers = 0;
- }
- if ( !stbir_build_samplers( resize ) )
- return 0;
- resize->called_alloc = alloc_state;
- // if build_samplers succeeded (above), but there are no samplers set, then
- // the area to stretch into was zero pixels, so don't do anything and return
- // success
- if ( resize->samplers == 0 )
- return 1;
- }
- else
- {
- // didn't build anything - clear it
- STBIR_PROFILE_BUILD_CLEAR( resize->samplers );
- }
- // do resize
- result = stbir__perform_resize( resize->samplers, 0, resize->splits );
- // if we alloced, then free
- if ( !resize->called_alloc )
- {
- stbir_free_samplers( resize );
- resize->samplers = 0;
- }
- return result;
- }
- STBIRDEF int stbir_resize_extended_split( STBIR_RESIZE * resize, int split_start, int split_count )
- {
- STBIR_ASSERT( resize->samplers );
- // if we're just doing the whole thing, call full
- if ( ( split_start == -1 ) || ( ( split_start == 0 ) && ( split_count == resize->splits ) ) )
- return stbir_resize_extended( resize );
- // you **must** build samplers first when using split resize
- if ( ( resize->samplers == 0 ) || ( resize->needs_rebuild ) )
- return 0;
- if ( ( split_start >= resize->splits ) || ( split_start < 0 ) || ( ( split_start + split_count ) > resize->splits ) || ( split_count <= 0 ) )
- return 0;
- // do resize
- return stbir__perform_resize( resize->samplers, split_start, split_count );
- }
- static void * stbir_quick_resize_helper( const void *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- void *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout, stbir_datatype data_type, stbir_edge edge, stbir_filter filter )
- {
- STBIR_RESIZE resize;
- int scanline_output_in_bytes;
- int positive_output_stride_in_bytes;
- void * start_ptr;
- void * free_ptr;
- scanline_output_in_bytes = output_w * stbir__type_size[ data_type ] * stbir__pixel_channels[ stbir__pixel_layout_convert_public_to_internal[ pixel_layout ] ];
- if ( scanline_output_in_bytes == 0 )
- return 0;
- // if zero stride, use scanline output
- if ( output_stride_in_bytes == 0 )
- output_stride_in_bytes = scanline_output_in_bytes;
- // abs value for inverted images (negative pitches)
- positive_output_stride_in_bytes = output_stride_in_bytes;
- if ( positive_output_stride_in_bytes < 0 )
- positive_output_stride_in_bytes = -positive_output_stride_in_bytes;
- // is the requested stride smaller than the scanline output? if so, just fail
- if ( positive_output_stride_in_bytes < scanline_output_in_bytes )
- return 0;
- start_ptr = output_pixels;
- free_ptr = 0; // no free pointer, since they passed buffer to use
- // did they pass a zero for the dest? if so, allocate the buffer
- if ( output_pixels == 0 )
- {
- size_t size;
- char * ptr;
-
- size = (size_t)positive_output_stride_in_bytes * (size_t)output_h;
- if ( size == 0 )
- return 0;
- ptr = (char*) STBIR_MALLOC( size, 0 );
- if ( ptr == 0 )
- return 0;
- free_ptr = ptr;
- // point at the last scanline, if they requested a flipped image
- if ( output_stride_in_bytes < 0 )
- start_ptr = ptr + ( (size_t)positive_output_stride_in_bytes * (size_t)( output_h - 1 ) );
- else
- start_ptr = ptr;
- }
- // ok, now do the resize
- stbir_resize_init( &resize,
- input_pixels, input_w, input_h, input_stride_in_bytes,
- start_ptr, output_w, output_h, output_stride_in_bytes,
- pixel_layout, data_type );
- resize.horizontal_edge = edge;
- resize.vertical_edge = edge;
- resize.horizontal_filter = filter;
- resize.vertical_filter = filter;
- if ( !stbir_resize_extended( &resize ) )
- {
- if ( free_ptr )
- STBIR_FREE( free_ptr, 0 );
- return 0;
- }
- return (free_ptr) ? free_ptr : start_ptr;
- }
- STBIRDEF unsigned char * stbir_resize_uint8_linear( const unsigned char *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout )
- {
- return (unsigned char *) stbir_quick_resize_helper( input_pixels , input_w , input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout, STBIR_TYPE_UINT8, STBIR_EDGE_CLAMP, STBIR_FILTER_DEFAULT );
- }
- STBIRDEF unsigned char * stbir_resize_uint8_srgb( const unsigned char *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- unsigned char *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout )
- {
- return (unsigned char *) stbir_quick_resize_helper( input_pixels , input_w , input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout, STBIR_TYPE_UINT8_SRGB, STBIR_EDGE_CLAMP, STBIR_FILTER_DEFAULT );
- }
- STBIRDEF float * stbir_resize_float_linear( const float *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- float *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout )
- {
- return (float *) stbir_quick_resize_helper( input_pixels , input_w , input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout, STBIR_TYPE_FLOAT, STBIR_EDGE_CLAMP, STBIR_FILTER_DEFAULT );
- }
- STBIRDEF void * stbir_resize( const void *input_pixels , int input_w , int input_h, int input_stride_in_bytes,
- void *output_pixels, int output_w, int output_h, int output_stride_in_bytes,
- stbir_pixel_layout pixel_layout, stbir_datatype data_type,
- stbir_edge edge, stbir_filter filter )
- {
- return (void *) stbir_quick_resize_helper( input_pixels , input_w , input_h, input_stride_in_bytes,
- output_pixels, output_w, output_h, output_stride_in_bytes,
- pixel_layout, data_type, edge, filter );
- }
- #ifdef STBIR_PROFILE
- STBIRDEF void stbir_resize_build_profile_info( STBIR_PROFILE_INFO * info, STBIR_RESIZE const * resize )
- {
- static char const * bdescriptions[6] = { "Building", "Allocating", "Horizontal sampler", "Vertical sampler", "Coefficient cleanup", "Coefficient piovot" } ;
- stbir__info* samp = resize->samplers;
- int i;
- typedef int testa[ (STBIR__ARRAY_SIZE( bdescriptions ) == (STBIR__ARRAY_SIZE( samp->profile.array )-1) )?1:-1];
- typedef int testb[ (sizeof( samp->profile.array ) == (sizeof(samp->profile.named)) )?1:-1];
- typedef int testc[ (sizeof( info->clocks ) >= (sizeof(samp->profile.named)) )?1:-1];
- for( i = 0 ; i < STBIR__ARRAY_SIZE( bdescriptions ) ; i++)
- info->clocks[i] = samp->profile.array[i+1];
- info->total_clocks = samp->profile.named.total;
- info->descriptions = bdescriptions;
- info->count = STBIR__ARRAY_SIZE( bdescriptions );
- }
- STBIRDEF void stbir_resize_split_profile_info( STBIR_PROFILE_INFO * info, STBIR_RESIZE const * resize, int split_start, int split_count )
- {
- static char const * descriptions[7] = { "Looping", "Vertical sampling", "Horizontal sampling", "Scanline input", "Scanline output", "Alpha weighting", "Alpha unweighting" };
- stbir__per_split_info * split_info;
- int s, i;
- typedef int testa[ (STBIR__ARRAY_SIZE( descriptions ) == (STBIR__ARRAY_SIZE( split_info->profile.array )-1) )?1:-1];
- typedef int testb[ (sizeof( split_info->profile.array ) == (sizeof(split_info->profile.named)) )?1:-1];
- typedef int testc[ (sizeof( info->clocks ) >= (sizeof(split_info->profile.named)) )?1:-1];
- if ( split_start == -1 )
- {
- split_start = 0;
- split_count = resize->samplers->splits;
- }
- if ( ( split_start >= resize->splits ) || ( split_start < 0 ) || ( ( split_start + split_count ) > resize->splits ) || ( split_count <= 0 ) )
- {
- info->total_clocks = 0;
- info->descriptions = 0;
- info->count = 0;
- return;
- }
- split_info = resize->samplers->split_info + split_start;
- // sum up the profile from all the splits
- for( i = 0 ; i < STBIR__ARRAY_SIZE( descriptions ) ; i++ )
- {
- stbir_uint64 sum = 0;
- for( s = 0 ; s < split_count ; s++ )
- sum += split_info[s].profile.array[i+1];
- info->clocks[i] = sum;
- }
- info->total_clocks = split_info->profile.named.total;
- info->descriptions = descriptions;
- info->count = STBIR__ARRAY_SIZE( descriptions );
- }
- STBIRDEF void stbir_resize_extended_profile_info( STBIR_PROFILE_INFO * info, STBIR_RESIZE const * resize )
- {
- stbir_resize_split_profile_info( info, resize, -1, 0 );
- }
- #endif // STBIR_PROFILE
- #undef STBIR_BGR
- #undef STBIR_1CHANNEL
- #undef STBIR_2CHANNEL
- #undef STBIR_RGB
- #undef STBIR_RGBA
- #undef STBIR_4CHANNEL
- #undef STBIR_BGRA
- #undef STBIR_ARGB
- #undef STBIR_ABGR
- #undef STBIR_RA
- #undef STBIR_AR
- #undef STBIR_RGBA_PM
- #undef STBIR_BGRA_PM
- #undef STBIR_ARGB_PM
- #undef STBIR_ABGR_PM
- #undef STBIR_RA_PM
- #undef STBIR_AR_PM
- #endif // STB_IMAGE_RESIZE_IMPLEMENTATION
- #else // STB_IMAGE_RESIZE_HORIZONTALS&STB_IMAGE_RESIZE_DO_VERTICALS
- // we reinclude the header file to define all the horizontal functions
- // specializing each function for the number of coeffs is 20-40% faster *OVERALL*
- // by including the header file again this way, we can still debug the functions
- #define STBIR_strs_join2( start, mid, end ) start##mid##end
- #define STBIR_strs_join1( start, mid, end ) STBIR_strs_join2( start, mid, end )
- #define STBIR_strs_join24( start, mid1, mid2, end ) start##mid1##mid2##end
- #define STBIR_strs_join14( start, mid1, mid2, end ) STBIR_strs_join24( start, mid1, mid2, end )
- #ifdef STB_IMAGE_RESIZE_DO_CODERS
- #ifdef stbir__decode_suffix
- #define STBIR__CODER_NAME( name ) STBIR_strs_join1( name, _, stbir__decode_suffix )
- #else
- #define STBIR__CODER_NAME( name ) name
- #endif
- #ifdef stbir__decode_swizzle
- #define stbir__decode_simdf8_flip(reg) STBIR_strs_join1( STBIR_strs_join1( STBIR_strs_join1( STBIR_strs_join1( stbir__simdf8_0123to,stbir__decode_order0,stbir__decode_order1),stbir__decode_order2,stbir__decode_order3),stbir__decode_order0,stbir__decode_order1),stbir__decode_order2,stbir__decode_order3)(reg, reg)
- #define stbir__decode_simdf4_flip(reg) STBIR_strs_join1( STBIR_strs_join1( stbir__simdf_0123to,stbir__decode_order0,stbir__decode_order1),stbir__decode_order2,stbir__decode_order3)(reg, reg)
- #define stbir__encode_simdf8_unflip(reg) STBIR_strs_join1( STBIR_strs_join1( STBIR_strs_join1( STBIR_strs_join1( stbir__simdf8_0123to,stbir__encode_order0,stbir__encode_order1),stbir__encode_order2,stbir__encode_order3),stbir__encode_order0,stbir__encode_order1),stbir__encode_order2,stbir__encode_order3)(reg, reg)
- #define stbir__encode_simdf4_unflip(reg) STBIR_strs_join1( STBIR_strs_join1( stbir__simdf_0123to,stbir__encode_order0,stbir__encode_order1),stbir__encode_order2,stbir__encode_order3)(reg, reg)
- #else
- #define stbir__decode_order0 0
- #define stbir__decode_order1 1
- #define stbir__decode_order2 2
- #define stbir__decode_order3 3
- #define stbir__encode_order0 0
- #define stbir__encode_order1 1
- #define stbir__encode_order2 2
- #define stbir__encode_order3 3
- #define stbir__decode_simdf8_flip(reg)
- #define stbir__decode_simdf4_flip(reg)
- #define stbir__encode_simdf8_unflip(reg)
- #define stbir__encode_simdf4_unflip(reg)
- #endif
- #ifdef STBIR_SIMD8
- #define stbir__encode_simdfX_unflip stbir__encode_simdf8_unflip
- #else
- #define stbir__encode_simdfX_unflip stbir__encode_simdf4_unflip
- #endif
- static float * STBIR__CODER_NAME( stbir__decode_uint8_linear_scaled )( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned char const * input = (unsigned char const*)inputp;
- #ifdef STBIR_SIMD
- unsigned char const * end_input_m16 = input + width_times_channels - 16;
- if ( width_times_channels >= 16 )
- {
- decode_end -= 16;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- #ifdef STBIR_SIMD8
- stbir__simdi i; stbir__simdi8 o0,o1;
- stbir__simdf8 of0, of1;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi8_expand_u8_to_u32( o0, o1, i );
- stbir__simdi8_convert_i32_to_float( of0, o0 );
- stbir__simdi8_convert_i32_to_float( of1, o1 );
- stbir__simdf8_mult( of0, of0, STBIR_max_uint8_as_float_inverted8);
- stbir__simdf8_mult( of1, of1, STBIR_max_uint8_as_float_inverted8);
- stbir__decode_simdf8_flip( of0 );
- stbir__decode_simdf8_flip( of1 );
- stbir__simdf8_store( decode + 0, of0 );
- stbir__simdf8_store( decode + 8, of1 );
- #else
- stbir__simdi i, o0, o1, o2, o3;
- stbir__simdf of0, of1, of2, of3;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi_expand_u8_to_u32( o0,o1,o2,o3,i);
- stbir__simdi_convert_i32_to_float( of0, o0 );
- stbir__simdi_convert_i32_to_float( of1, o1 );
- stbir__simdi_convert_i32_to_float( of2, o2 );
- stbir__simdi_convert_i32_to_float( of3, o3 );
- stbir__simdf_mult( of0, of0, STBIR__CONSTF(STBIR_max_uint8_as_float_inverted) );
- stbir__simdf_mult( of1, of1, STBIR__CONSTF(STBIR_max_uint8_as_float_inverted) );
- stbir__simdf_mult( of2, of2, STBIR__CONSTF(STBIR_max_uint8_as_float_inverted) );
- stbir__simdf_mult( of3, of3, STBIR__CONSTF(STBIR_max_uint8_as_float_inverted) );
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__decode_simdf4_flip( of2 );
- stbir__decode_simdf4_flip( of3 );
- stbir__simdf_store( decode + 0, of0 );
- stbir__simdf_store( decode + 4, of1 );
- stbir__simdf_store( decode + 8, of2 );
- stbir__simdf_store( decode + 12, of3 );
- #endif
- decode += 16;
- input += 16;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 16 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m16;
- }
- return decode_end + 16;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = ((float)(input[stbir__decode_order0])) * stbir__max_uint8_as_float_inverted;
- decode[1-4] = ((float)(input[stbir__decode_order1])) * stbir__max_uint8_as_float_inverted;
- decode[2-4] = ((float)(input[stbir__decode_order2])) * stbir__max_uint8_as_float_inverted;
- decode[3-4] = ((float)(input[stbir__decode_order3])) * stbir__max_uint8_as_float_inverted;
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = ((float)(input[stbir__decode_order0])) * stbir__max_uint8_as_float_inverted;
- #if stbir__coder_min_num >= 2
- decode[1] = ((float)(input[stbir__decode_order1])) * stbir__max_uint8_as_float_inverted;
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = ((float)(input[stbir__decode_order2])) * stbir__max_uint8_as_float_inverted;
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- static void STBIR__CODER_NAME( stbir__encode_uint8_linear_scaled )( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned char STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned char *) outputp;
- unsigned char * end_output = ( (unsigned char *) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= stbir__simdfX_float_count*2 )
- {
- float const * end_encode_m8 = encode + width_times_channels - stbir__simdfX_float_count*2;
- end_output -= stbir__simdfX_float_count*2;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdfX e0, e1;
- stbir__simdi i;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdfX_madd_mem( e0, STBIR_simd_point5X, STBIR_max_uint8_as_floatX, encode );
- stbir__simdfX_madd_mem( e1, STBIR_simd_point5X, STBIR_max_uint8_as_floatX, encode+stbir__simdfX_float_count );
- stbir__encode_simdfX_unflip( e0 );
- stbir__encode_simdfX_unflip( e1 );
- #ifdef STBIR_SIMD8
- stbir__simdf8_pack_to_16bytes( i, e0, e1 );
- stbir__simdi_store( output, i );
- #else
- stbir__simdf_pack_to_8bytes( i, e0, e1 );
- stbir__simdi_store2( output, i );
- #endif
- encode += stbir__simdfX_float_count*2;
- output += stbir__simdfX_float_count*2;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + stbir__simdfX_float_count*2 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- stbir__simdf e0;
- stbir__simdi i0;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_load( e0, encode );
- stbir__simdf_madd( e0, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint8_as_float), e0 );
- stbir__encode_simdf4_unflip( e0 );
- stbir__simdf_pack_to_8bytes( i0, e0, e0 ); // only use first 4
- *(int*)(output-4) = stbir__simdi_to_int( i0 );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- stbir__simdf e0;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_madd1_mem( e0, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint8_as_float), encode+stbir__encode_order0 ); output[0] = stbir__simdf_convert_float_to_uint8( e0 );
- #if stbir__coder_min_num >= 2
- stbir__simdf_madd1_mem( e0, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint8_as_float), encode+stbir__encode_order1 ); output[1] = stbir__simdf_convert_float_to_uint8( e0 );
- #endif
- #if stbir__coder_min_num >= 3
- stbir__simdf_madd1_mem( e0, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint8_as_float), encode+stbir__encode_order2 ); output[2] = stbir__simdf_convert_float_to_uint8( e0 );
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- #else
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- while( output <= end_output )
- {
- float f;
- f = encode[stbir__encode_order0] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[0-4] = (unsigned char)f;
- f = encode[stbir__encode_order1] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[1-4] = (unsigned char)f;
- f = encode[stbir__encode_order2] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[2-4] = (unsigned char)f;
- f = encode[stbir__encode_order3] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[3-4] = (unsigned char)f;
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- float f;
- STBIR_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[0] = (unsigned char)f;
- #if stbir__coder_min_num >= 2
- f = encode[stbir__encode_order1] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[1] = (unsigned char)f;
- #endif
- #if stbir__coder_min_num >= 3
- f = encode[stbir__encode_order2] * stbir__max_uint8_as_float + 0.5f; STBIR_CLAMP(f, 0, 255); output[2] = (unsigned char)f;
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- #endif
- }
- static float * STBIR__CODER_NAME(stbir__decode_uint8_linear)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned char const * input = (unsigned char const*)inputp;
- #ifdef STBIR_SIMD
- unsigned char const * end_input_m16 = input + width_times_channels - 16;
- if ( width_times_channels >= 16 )
- {
- decode_end -= 16;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- #ifdef STBIR_SIMD8
- stbir__simdi i; stbir__simdi8 o0,o1;
- stbir__simdf8 of0, of1;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi8_expand_u8_to_u32( o0, o1, i );
- stbir__simdi8_convert_i32_to_float( of0, o0 );
- stbir__simdi8_convert_i32_to_float( of1, o1 );
- stbir__decode_simdf8_flip( of0 );
- stbir__decode_simdf8_flip( of1 );
- stbir__simdf8_store( decode + 0, of0 );
- stbir__simdf8_store( decode + 8, of1 );
- #else
- stbir__simdi i, o0, o1, o2, o3;
- stbir__simdf of0, of1, of2, of3;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi_expand_u8_to_u32( o0,o1,o2,o3,i);
- stbir__simdi_convert_i32_to_float( of0, o0 );
- stbir__simdi_convert_i32_to_float( of1, o1 );
- stbir__simdi_convert_i32_to_float( of2, o2 );
- stbir__simdi_convert_i32_to_float( of3, o3 );
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__decode_simdf4_flip( of2 );
- stbir__decode_simdf4_flip( of3 );
- stbir__simdf_store( decode + 0, of0 );
- stbir__simdf_store( decode + 4, of1 );
- stbir__simdf_store( decode + 8, of2 );
- stbir__simdf_store( decode + 12, of3 );
- #endif
- decode += 16;
- input += 16;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 16 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m16;
- }
- return decode_end + 16;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = ((float)(input[stbir__decode_order0]));
- decode[1-4] = ((float)(input[stbir__decode_order1]));
- decode[2-4] = ((float)(input[stbir__decode_order2]));
- decode[3-4] = ((float)(input[stbir__decode_order3]));
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = ((float)(input[stbir__decode_order0]));
- #if stbir__coder_min_num >= 2
- decode[1] = ((float)(input[stbir__decode_order1]));
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = ((float)(input[stbir__decode_order2]));
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- static void STBIR__CODER_NAME( stbir__encode_uint8_linear )( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned char STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned char *) outputp;
- unsigned char * end_output = ( (unsigned char *) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= stbir__simdfX_float_count*2 )
- {
- float const * end_encode_m8 = encode + width_times_channels - stbir__simdfX_float_count*2;
- end_output -= stbir__simdfX_float_count*2;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdfX e0, e1;
- stbir__simdi i;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdfX_add_mem( e0, STBIR_simd_point5X, encode );
- stbir__simdfX_add_mem( e1, STBIR_simd_point5X, encode+stbir__simdfX_float_count );
- stbir__encode_simdfX_unflip( e0 );
- stbir__encode_simdfX_unflip( e1 );
- #ifdef STBIR_SIMD8
- stbir__simdf8_pack_to_16bytes( i, e0, e1 );
- stbir__simdi_store( output, i );
- #else
- stbir__simdf_pack_to_8bytes( i, e0, e1 );
- stbir__simdi_store2( output, i );
- #endif
- encode += stbir__simdfX_float_count*2;
- output += stbir__simdfX_float_count*2;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + stbir__simdfX_float_count*2 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- stbir__simdf e0;
- stbir__simdi i0;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_load( e0, encode );
- stbir__simdf_add( e0, STBIR__CONSTF(STBIR_simd_point5), e0 );
- stbir__encode_simdf4_unflip( e0 );
- stbir__simdf_pack_to_8bytes( i0, e0, e0 ); // only use first 4
- *(int*)(output-4) = stbir__simdi_to_int( i0 );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #else
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- while( output <= end_output )
- {
- float f;
- f = encode[stbir__encode_order0] + 0.5f; STBIR_CLAMP(f, 0, 255); output[0-4] = (unsigned char)f;
- f = encode[stbir__encode_order1] + 0.5f; STBIR_CLAMP(f, 0, 255); output[1-4] = (unsigned char)f;
- f = encode[stbir__encode_order2] + 0.5f; STBIR_CLAMP(f, 0, 255); output[2-4] = (unsigned char)f;
- f = encode[stbir__encode_order3] + 0.5f; STBIR_CLAMP(f, 0, 255); output[3-4] = (unsigned char)f;
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- float f;
- STBIR_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] + 0.5f; STBIR_CLAMP(f, 0, 255); output[0] = (unsigned char)f;
- #if stbir__coder_min_num >= 2
- f = encode[stbir__encode_order1] + 0.5f; STBIR_CLAMP(f, 0, 255); output[1] = (unsigned char)f;
- #endif
- #if stbir__coder_min_num >= 3
- f = encode[stbir__encode_order2] + 0.5f; STBIR_CLAMP(f, 0, 255); output[2] = (unsigned char)f;
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- }
- static float * STBIR__CODER_NAME(stbir__decode_uint8_srgb)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned char const * input = (unsigned char const *)inputp;
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- while( decode <= decode_end )
- {
- decode[0-4] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order0 ] ];
- decode[1-4] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order1 ] ];
- decode[2-4] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order2 ] ];
- decode[3-4] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order3 ] ];
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order0 ] ];
- #if stbir__coder_min_num >= 2
- decode[1] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order1 ] ];
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = stbir__srgb_uchar_to_linear_float[ input[ stbir__decode_order2 ] ];
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- #define stbir__min_max_shift20( i, f ) \
- stbir__simdf_max( f, f, stbir_simdf_casti(STBIR__CONSTI( STBIR_almost_zero )) ); \
- stbir__simdf_min( f, f, stbir_simdf_casti(STBIR__CONSTI( STBIR_almost_one )) ); \
- stbir__simdi_32shr( i, stbir_simdi_castf( f ), 20 );
- #define stbir__scale_and_convert( i, f ) \
- stbir__simdf_madd( f, STBIR__CONSTF( STBIR_simd_point5 ), STBIR__CONSTF( STBIR_max_uint8_as_float ), f ); \
- stbir__simdf_max( f, f, stbir__simdf_zeroP() ); \
- stbir__simdf_min( f, f, STBIR__CONSTF( STBIR_max_uint8_as_float ) ); \
- stbir__simdf_convert_float_to_i32( i, f );
- #define stbir__linear_to_srgb_finish( i, f ) \
- { \
- stbir__simdi temp; \
- stbir__simdi_32shr( temp, stbir_simdi_castf( f ), 12 ) ; \
- stbir__simdi_and( temp, temp, STBIR__CONSTI(STBIR_mastissa_mask) ); \
- stbir__simdi_or( temp, temp, STBIR__CONSTI(STBIR_topscale) ); \
- stbir__simdi_16madd( i, i, temp ); \
- stbir__simdi_32shr( i, i, 16 ); \
- }
- #define stbir__simdi_table_lookup2( v0,v1, table ) \
- { \
- stbir__simdi_u32 temp0,temp1; \
- temp0.m128i_i128 = v0; \
- temp1.m128i_i128 = v1; \
- temp0.m128i_u32[0] = table[temp0.m128i_i32[0]]; temp0.m128i_u32[1] = table[temp0.m128i_i32[1]]; temp0.m128i_u32[2] = table[temp0.m128i_i32[2]]; temp0.m128i_u32[3] = table[temp0.m128i_i32[3]]; \
- temp1.m128i_u32[0] = table[temp1.m128i_i32[0]]; temp1.m128i_u32[1] = table[temp1.m128i_i32[1]]; temp1.m128i_u32[2] = table[temp1.m128i_i32[2]]; temp1.m128i_u32[3] = table[temp1.m128i_i32[3]]; \
- v0 = temp0.m128i_i128; \
- v1 = temp1.m128i_i128; \
- }
- #define stbir__simdi_table_lookup3( v0,v1,v2, table ) \
- { \
- stbir__simdi_u32 temp0,temp1,temp2; \
- temp0.m128i_i128 = v0; \
- temp1.m128i_i128 = v1; \
- temp2.m128i_i128 = v2; \
- temp0.m128i_u32[0] = table[temp0.m128i_i32[0]]; temp0.m128i_u32[1] = table[temp0.m128i_i32[1]]; temp0.m128i_u32[2] = table[temp0.m128i_i32[2]]; temp0.m128i_u32[3] = table[temp0.m128i_i32[3]]; \
- temp1.m128i_u32[0] = table[temp1.m128i_i32[0]]; temp1.m128i_u32[1] = table[temp1.m128i_i32[1]]; temp1.m128i_u32[2] = table[temp1.m128i_i32[2]]; temp1.m128i_u32[3] = table[temp1.m128i_i32[3]]; \
- temp2.m128i_u32[0] = table[temp2.m128i_i32[0]]; temp2.m128i_u32[1] = table[temp2.m128i_i32[1]]; temp2.m128i_u32[2] = table[temp2.m128i_i32[2]]; temp2.m128i_u32[3] = table[temp2.m128i_i32[3]]; \
- v0 = temp0.m128i_i128; \
- v1 = temp1.m128i_i128; \
- v2 = temp2.m128i_i128; \
- }
- #define stbir__simdi_table_lookup4( v0,v1,v2,v3, table ) \
- { \
- stbir__simdi_u32 temp0,temp1,temp2,temp3; \
- temp0.m128i_i128 = v0; \
- temp1.m128i_i128 = v1; \
- temp2.m128i_i128 = v2; \
- temp3.m128i_i128 = v3; \
- temp0.m128i_u32[0] = table[temp0.m128i_i32[0]]; temp0.m128i_u32[1] = table[temp0.m128i_i32[1]]; temp0.m128i_u32[2] = table[temp0.m128i_i32[2]]; temp0.m128i_u32[3] = table[temp0.m128i_i32[3]]; \
- temp1.m128i_u32[0] = table[temp1.m128i_i32[0]]; temp1.m128i_u32[1] = table[temp1.m128i_i32[1]]; temp1.m128i_u32[2] = table[temp1.m128i_i32[2]]; temp1.m128i_u32[3] = table[temp1.m128i_i32[3]]; \
- temp2.m128i_u32[0] = table[temp2.m128i_i32[0]]; temp2.m128i_u32[1] = table[temp2.m128i_i32[1]]; temp2.m128i_u32[2] = table[temp2.m128i_i32[2]]; temp2.m128i_u32[3] = table[temp2.m128i_i32[3]]; \
- temp3.m128i_u32[0] = table[temp3.m128i_i32[0]]; temp3.m128i_u32[1] = table[temp3.m128i_i32[1]]; temp3.m128i_u32[2] = table[temp3.m128i_i32[2]]; temp3.m128i_u32[3] = table[temp3.m128i_i32[3]]; \
- v0 = temp0.m128i_i128; \
- v1 = temp1.m128i_i128; \
- v2 = temp2.m128i_i128; \
- v3 = temp3.m128i_i128; \
- }
- static void STBIR__CODER_NAME( stbir__encode_uint8_srgb )( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned char STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned char*) outputp;
- unsigned char * end_output = ( (unsigned char*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 16 )
- {
- float const * end_encode_m16 = encode + width_times_channels - 16;
- end_output -= 16;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdf f0, f1, f2, f3;
- stbir__simdi i0, i1, i2, i3;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdf_load4_transposed( f0, f1, f2, f3, encode );
- stbir__min_max_shift20( i0, f0 );
- stbir__min_max_shift20( i1, f1 );
- stbir__min_max_shift20( i2, f2 );
- stbir__min_max_shift20( i3, f3 );
- stbir__simdi_table_lookup4( i0, i1, i2, i3, ( fp32_to_srgb8_tab4 - (127-13)*8 ) );
- stbir__linear_to_srgb_finish( i0, f0 );
- stbir__linear_to_srgb_finish( i1, f1 );
- stbir__linear_to_srgb_finish( i2, f2 );
- stbir__linear_to_srgb_finish( i3, f3 );
- stbir__interleave_pack_and_store_16_u8( output, STBIR_strs_join1(i, ,stbir__encode_order0), STBIR_strs_join1(i, ,stbir__encode_order1), STBIR_strs_join1(i, ,stbir__encode_order2), STBIR_strs_join1(i, ,stbir__encode_order3) );
- encode += 16;
- output += 16;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + 16 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m16;
- }
- return;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while ( output <= end_output )
- {
- STBIR_SIMD_NO_UNROLL(encode);
- output[0-4] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order0] );
- output[1-4] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order1] );
- output[2-4] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order2] );
- output[3-4] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order3] );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- STBIR_NO_UNROLL(encode);
- output[0] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order0] );
- #if stbir__coder_min_num >= 2
- output[1] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order1] );
- #endif
- #if stbir__coder_min_num >= 3
- output[2] = stbir__linear_to_srgb_uchar( encode[stbir__encode_order2] );
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- }
- #if ( stbir__coder_min_num == 4 ) || ( ( stbir__coder_min_num == 1 ) && ( !defined(stbir__decode_swizzle) ) )
- static float * STBIR__CODER_NAME(stbir__decode_uint8_srgb4_linearalpha)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned char const * input = (unsigned char const *)inputp;
- do {
- decode[0] = stbir__srgb_uchar_to_linear_float[ input[stbir__decode_order0] ];
- decode[1] = stbir__srgb_uchar_to_linear_float[ input[stbir__decode_order1] ];
- decode[2] = stbir__srgb_uchar_to_linear_float[ input[stbir__decode_order2] ];
- decode[3] = ( (float) input[stbir__decode_order3] ) * stbir__max_uint8_as_float_inverted;
- input += 4;
- decode += 4;
- } while( decode < decode_end );
- return decode_end;
- }
- static void STBIR__CODER_NAME( stbir__encode_uint8_srgb4_linearalpha )( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned char STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned char*) outputp;
- unsigned char * end_output = ( (unsigned char*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 16 )
- {
- float const * end_encode_m16 = encode + width_times_channels - 16;
- end_output -= 16;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdf f0, f1, f2, f3;
- stbir__simdi i0, i1, i2, i3;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdf_load4_transposed( f0, f1, f2, f3, encode );
- stbir__min_max_shift20( i0, f0 );
- stbir__min_max_shift20( i1, f1 );
- stbir__min_max_shift20( i2, f2 );
- stbir__scale_and_convert( i3, f3 );
- stbir__simdi_table_lookup3( i0, i1, i2, ( fp32_to_srgb8_tab4 - (127-13)*8 ) );
- stbir__linear_to_srgb_finish( i0, f0 );
- stbir__linear_to_srgb_finish( i1, f1 );
- stbir__linear_to_srgb_finish( i2, f2 );
- stbir__interleave_pack_and_store_16_u8( output, STBIR_strs_join1(i, ,stbir__encode_order0), STBIR_strs_join1(i, ,stbir__encode_order1), STBIR_strs_join1(i, ,stbir__encode_order2), STBIR_strs_join1(i, ,stbir__encode_order3) );
- output += 16;
- encode += 16;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + 16 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m16;
- }
- return;
- }
- #endif
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float f;
- STBIR_SIMD_NO_UNROLL(encode);
- output[stbir__decode_order0] = stbir__linear_to_srgb_uchar( encode[0] );
- output[stbir__decode_order1] = stbir__linear_to_srgb_uchar( encode[1] );
- output[stbir__decode_order2] = stbir__linear_to_srgb_uchar( encode[2] );
- f = encode[3] * stbir__max_uint8_as_float + 0.5f;
- STBIR_CLAMP(f, 0, 255);
- output[stbir__decode_order3] = (unsigned char) f;
- output += 4;
- encode += 4;
- } while( output < end_output );
- }
- #endif
- #if ( stbir__coder_min_num == 2 ) || ( ( stbir__coder_min_num == 1 ) && ( !defined(stbir__decode_swizzle) ) )
- static float * STBIR__CODER_NAME(stbir__decode_uint8_srgb2_linearalpha)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned char const * input = (unsigned char const *)inputp;
- decode += 4;
- while( decode <= decode_end )
- {
- decode[0-4] = stbir__srgb_uchar_to_linear_float[ input[stbir__decode_order0] ];
- decode[1-4] = ( (float) input[stbir__decode_order1] ) * stbir__max_uint8_as_float_inverted;
- decode[2-4] = stbir__srgb_uchar_to_linear_float[ input[stbir__decode_order0+2] ];
- decode[3-4] = ( (float) input[stbir__decode_order1+2] ) * stbir__max_uint8_as_float_inverted;
- input += 4;
- decode += 4;
- }
- decode -= 4;
- if( decode < decode_end )
- {
- decode[0] = stbir__srgb_uchar_to_linear_float[ stbir__decode_order0 ];
- decode[1] = ( (float) input[stbir__decode_order1] ) * stbir__max_uint8_as_float_inverted;
- }
- return decode_end;
- }
- static void STBIR__CODER_NAME( stbir__encode_uint8_srgb2_linearalpha )( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned char STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned char*) outputp;
- unsigned char * end_output = ( (unsigned char*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 16 )
- {
- float const * end_encode_m16 = encode + width_times_channels - 16;
- end_output -= 16;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdf f0, f1, f2, f3;
- stbir__simdi i0, i1, i2, i3;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdf_load4_transposed( f0, f1, f2, f3, encode );
- stbir__min_max_shift20( i0, f0 );
- stbir__scale_and_convert( i1, f1 );
- stbir__min_max_shift20( i2, f2 );
- stbir__scale_and_convert( i3, f3 );
- stbir__simdi_table_lookup2( i0, i2, ( fp32_to_srgb8_tab4 - (127-13)*8 ) );
- stbir__linear_to_srgb_finish( i0, f0 );
- stbir__linear_to_srgb_finish( i2, f2 );
- stbir__interleave_pack_and_store_16_u8( output, STBIR_strs_join1(i, ,stbir__encode_order0), STBIR_strs_join1(i, ,stbir__encode_order1), STBIR_strs_join1(i, ,stbir__encode_order2), STBIR_strs_join1(i, ,stbir__encode_order3) );
- output += 16;
- encode += 16;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + 16 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m16;
- }
- return;
- }
- #endif
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float f;
- STBIR_SIMD_NO_UNROLL(encode);
- output[stbir__decode_order0] = stbir__linear_to_srgb_uchar( encode[0] );
- f = encode[1] * stbir__max_uint8_as_float + 0.5f;
- STBIR_CLAMP(f, 0, 255);
- output[stbir__decode_order1] = (unsigned char) f;
- output += 2;
- encode += 2;
- } while( output < end_output );
- }
- #endif
- static float * STBIR__CODER_NAME(stbir__decode_uint16_linear_scaled)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned short const * input = (unsigned short const *)inputp;
- #ifdef STBIR_SIMD
- unsigned short const * end_input_m8 = input + width_times_channels - 8;
- if ( width_times_channels >= 8 )
- {
- decode_end -= 8;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- #ifdef STBIR_SIMD8
- stbir__simdi i; stbir__simdi8 o;
- stbir__simdf8 of;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi8_expand_u16_to_u32( o, i );
- stbir__simdi8_convert_i32_to_float( of, o );
- stbir__simdf8_mult( of, of, STBIR_max_uint16_as_float_inverted8);
- stbir__decode_simdf8_flip( of );
- stbir__simdf8_store( decode + 0, of );
- #else
- stbir__simdi i, o0, o1;
- stbir__simdf of0, of1;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi_expand_u16_to_u32( o0,o1,i );
- stbir__simdi_convert_i32_to_float( of0, o0 );
- stbir__simdi_convert_i32_to_float( of1, o1 );
- stbir__simdf_mult( of0, of0, STBIR__CONSTF(STBIR_max_uint16_as_float_inverted) );
- stbir__simdf_mult( of1, of1, STBIR__CONSTF(STBIR_max_uint16_as_float_inverted));
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__simdf_store( decode + 0, of0 );
- stbir__simdf_store( decode + 4, of1 );
- #endif
- decode += 8;
- input += 8;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 8 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m8;
- }
- return decode_end + 8;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = ((float)(input[stbir__decode_order0])) * stbir__max_uint16_as_float_inverted;
- decode[1-4] = ((float)(input[stbir__decode_order1])) * stbir__max_uint16_as_float_inverted;
- decode[2-4] = ((float)(input[stbir__decode_order2])) * stbir__max_uint16_as_float_inverted;
- decode[3-4] = ((float)(input[stbir__decode_order3])) * stbir__max_uint16_as_float_inverted;
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = ((float)(input[stbir__decode_order0])) * stbir__max_uint16_as_float_inverted;
- #if stbir__coder_min_num >= 2
- decode[1] = ((float)(input[stbir__decode_order1])) * stbir__max_uint16_as_float_inverted;
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = ((float)(input[stbir__decode_order2])) * stbir__max_uint16_as_float_inverted;
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- static void STBIR__CODER_NAME(stbir__encode_uint16_linear_scaled)( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned short STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned short*) outputp;
- unsigned short * end_output = ( (unsigned short*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- {
- if ( width_times_channels >= stbir__simdfX_float_count*2 )
- {
- float const * end_encode_m8 = encode + width_times_channels - stbir__simdfX_float_count*2;
- end_output -= stbir__simdfX_float_count*2;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdfX e0, e1;
- stbir__simdiX i;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdfX_madd_mem( e0, STBIR_simd_point5X, STBIR_max_uint16_as_floatX, encode );
- stbir__simdfX_madd_mem( e1, STBIR_simd_point5X, STBIR_max_uint16_as_floatX, encode+stbir__simdfX_float_count );
- stbir__encode_simdfX_unflip( e0 );
- stbir__encode_simdfX_unflip( e1 );
- stbir__simdfX_pack_to_words( i, e0, e1 );
- stbir__simdiX_store( output, i );
- encode += stbir__simdfX_float_count*2;
- output += stbir__simdfX_float_count*2;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + stbir__simdfX_float_count*2 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- }
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- stbir__simdf e;
- stbir__simdi i;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_load( e, encode );
- stbir__simdf_madd( e, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint16_as_float), e );
- stbir__encode_simdf4_unflip( e );
- stbir__simdf_pack_to_8words( i, e, e ); // only use first 4
- stbir__simdi_store2( output-4, i );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- stbir__simdf e;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_madd1_mem( e, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint16_as_float), encode+stbir__encode_order0 ); output[0] = stbir__simdf_convert_float_to_short( e );
- #if stbir__coder_min_num >= 2
- stbir__simdf_madd1_mem( e, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint16_as_float), encode+stbir__encode_order1 ); output[1] = stbir__simdf_convert_float_to_short( e );
- #endif
- #if stbir__coder_min_num >= 3
- stbir__simdf_madd1_mem( e, STBIR__CONSTF(STBIR_simd_point5), STBIR__CONSTF(STBIR_max_uint16_as_float), encode+stbir__encode_order2 ); output[2] = stbir__simdf_convert_float_to_short( e );
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- #else
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- float f;
- STBIR_SIMD_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[0-4] = (unsigned short)f;
- f = encode[stbir__encode_order1] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[1-4] = (unsigned short)f;
- f = encode[stbir__encode_order2] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[2-4] = (unsigned short)f;
- f = encode[stbir__encode_order3] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[3-4] = (unsigned short)f;
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- float f;
- STBIR_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[0] = (unsigned short)f;
- #if stbir__coder_min_num >= 2
- f = encode[stbir__encode_order1] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[1] = (unsigned short)f;
- #endif
- #if stbir__coder_min_num >= 3
- f = encode[stbir__encode_order2] * stbir__max_uint16_as_float + 0.5f; STBIR_CLAMP(f, 0, 65535); output[2] = (unsigned short)f;
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- #endif
- }
- static float * STBIR__CODER_NAME(stbir__decode_uint16_linear)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- unsigned short const * input = (unsigned short const *)inputp;
- #ifdef STBIR_SIMD
- unsigned short const * end_input_m8 = input + width_times_channels - 8;
- if ( width_times_channels >= 8 )
- {
- decode_end -= 8;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- #ifdef STBIR_SIMD8
- stbir__simdi i; stbir__simdi8 o;
- stbir__simdf8 of;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi8_expand_u16_to_u32( o, i );
- stbir__simdi8_convert_i32_to_float( of, o );
- stbir__decode_simdf8_flip( of );
- stbir__simdf8_store( decode + 0, of );
- #else
- stbir__simdi i, o0, o1;
- stbir__simdf of0, of1;
- STBIR_NO_UNROLL(decode);
- stbir__simdi_load( i, input );
- stbir__simdi_expand_u16_to_u32( o0, o1, i );
- stbir__simdi_convert_i32_to_float( of0, o0 );
- stbir__simdi_convert_i32_to_float( of1, o1 );
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__simdf_store( decode + 0, of0 );
- stbir__simdf_store( decode + 4, of1 );
- #endif
- decode += 8;
- input += 8;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 8 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m8;
- }
- return decode_end + 8;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = ((float)(input[stbir__decode_order0]));
- decode[1-4] = ((float)(input[stbir__decode_order1]));
- decode[2-4] = ((float)(input[stbir__decode_order2]));
- decode[3-4] = ((float)(input[stbir__decode_order3]));
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = ((float)(input[stbir__decode_order0]));
- #if stbir__coder_min_num >= 2
- decode[1] = ((float)(input[stbir__decode_order1]));
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = ((float)(input[stbir__decode_order2]));
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- static void STBIR__CODER_NAME(stbir__encode_uint16_linear)( void * outputp, int width_times_channels, float const * encode )
- {
- unsigned short STBIR_SIMD_STREAMOUT_PTR( * ) output = (unsigned short*) outputp;
- unsigned short * end_output = ( (unsigned short*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- {
- if ( width_times_channels >= stbir__simdfX_float_count*2 )
- {
- float const * end_encode_m8 = encode + width_times_channels - stbir__simdfX_float_count*2;
- end_output -= stbir__simdfX_float_count*2;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdfX e0, e1;
- stbir__simdiX i;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdfX_add_mem( e0, STBIR_simd_point5X, encode );
- stbir__simdfX_add_mem( e1, STBIR_simd_point5X, encode+stbir__simdfX_float_count );
- stbir__encode_simdfX_unflip( e0 );
- stbir__encode_simdfX_unflip( e1 );
- stbir__simdfX_pack_to_words( i, e0, e1 );
- stbir__simdiX_store( output, i );
- encode += stbir__simdfX_float_count*2;
- output += stbir__simdfX_float_count*2;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + stbir__simdfX_float_count*2 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- }
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- stbir__simdf e;
- stbir__simdi i;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_load( e, encode );
- stbir__simdf_add( e, STBIR__CONSTF(STBIR_simd_point5), e );
- stbir__encode_simdf4_unflip( e );
- stbir__simdf_pack_to_8words( i, e, e ); // only use first 4
- stbir__simdi_store2( output-4, i );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #else
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- float f;
- STBIR_SIMD_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[0-4] = (unsigned short)f;
- f = encode[stbir__encode_order1] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[1-4] = (unsigned short)f;
- f = encode[stbir__encode_order2] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[2-4] = (unsigned short)f;
- f = encode[stbir__encode_order3] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[3-4] = (unsigned short)f;
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- float f;
- STBIR_NO_UNROLL(encode);
- f = encode[stbir__encode_order0] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[0] = (unsigned short)f;
- #if stbir__coder_min_num >= 2
- f = encode[stbir__encode_order1] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[1] = (unsigned short)f;
- #endif
- #if stbir__coder_min_num >= 3
- f = encode[stbir__encode_order2] + 0.5f; STBIR_CLAMP(f, 0, 65535); output[2] = (unsigned short)f;
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- }
- static float * STBIR__CODER_NAME(stbir__decode_half_float_linear)( float * decodep, int width_times_channels, void const * inputp )
- {
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- stbir__FP16 const * input = (stbir__FP16 const *)inputp;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 8 )
- {
- stbir__FP16 const * end_input_m8 = input + width_times_channels - 8;
- decode_end -= 8;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- STBIR_NO_UNROLL(decode);
- stbir__half_to_float_SIMD( decode, input );
- #ifdef stbir__decode_swizzle
- #ifdef STBIR_SIMD8
- {
- stbir__simdf8 of;
- stbir__simdf8_load( of, decode );
- stbir__decode_simdf8_flip( of );
- stbir__simdf8_store( decode, of );
- }
- #else
- {
- stbir__simdf of0,of1;
- stbir__simdf_load( of0, decode );
- stbir__simdf_load( of1, decode+4 );
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__simdf_store( decode, of0 );
- stbir__simdf_store( decode+4, of1 );
- }
- #endif
- #endif
- decode += 8;
- input += 8;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 8 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m8;
- }
- return decode_end + 8;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = stbir__half_to_float(input[stbir__decode_order0]);
- decode[1-4] = stbir__half_to_float(input[stbir__decode_order1]);
- decode[2-4] = stbir__half_to_float(input[stbir__decode_order2]);
- decode[3-4] = stbir__half_to_float(input[stbir__decode_order3]);
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = stbir__half_to_float(input[stbir__decode_order0]);
- #if stbir__coder_min_num >= 2
- decode[1] = stbir__half_to_float(input[stbir__decode_order1]);
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = stbir__half_to_float(input[stbir__decode_order2]);
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- }
- static void STBIR__CODER_NAME( stbir__encode_half_float_linear )( void * outputp, int width_times_channels, float const * encode )
- {
- stbir__FP16 STBIR_SIMD_STREAMOUT_PTR( * ) output = (stbir__FP16*) outputp;
- stbir__FP16 * end_output = ( (stbir__FP16*) output ) + width_times_channels;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 8 )
- {
- float const * end_encode_m8 = encode + width_times_channels - 8;
- end_output -= 8;
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- STBIR_SIMD_NO_UNROLL(encode);
- #ifdef stbir__decode_swizzle
- #ifdef STBIR_SIMD8
- {
- stbir__simdf8 of;
- stbir__simdf8_load( of, encode );
- stbir__encode_simdf8_unflip( of );
- stbir__float_to_half_SIMD( output, (float*)&of );
- }
- #else
- {
- stbir__simdf of[2];
- stbir__simdf_load( of[0], encode );
- stbir__simdf_load( of[1], encode+4 );
- stbir__encode_simdf4_unflip( of[0] );
- stbir__encode_simdf4_unflip( of[1] );
- stbir__float_to_half_SIMD( output, (float*)of );
- }
- #endif
- #else
- stbir__float_to_half_SIMD( output, encode );
- #endif
- encode += 8;
- output += 8;
- if ( output <= end_output )
- continue;
- if ( output == ( end_output + 8 ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- STBIR_SIMD_NO_UNROLL(output);
- output[0-4] = stbir__float_to_half(encode[stbir__encode_order0]);
- output[1-4] = stbir__float_to_half(encode[stbir__encode_order1]);
- output[2-4] = stbir__float_to_half(encode[stbir__encode_order2]);
- output[3-4] = stbir__float_to_half(encode[stbir__encode_order3]);
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- STBIR_NO_UNROLL(output);
- output[0] = stbir__float_to_half(encode[stbir__encode_order0]);
- #if stbir__coder_min_num >= 2
- output[1] = stbir__float_to_half(encode[stbir__encode_order1]);
- #endif
- #if stbir__coder_min_num >= 3
- output[2] = stbir__float_to_half(encode[stbir__encode_order2]);
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- }
- static float * STBIR__CODER_NAME(stbir__decode_float_linear)( float * decodep, int width_times_channels, void const * inputp )
- {
- #ifdef stbir__decode_swizzle
- float STBIR_STREAMOUT_PTR( * ) decode = decodep;
- float * decode_end = (float*) decode + width_times_channels;
- float const * input = (float const *)inputp;
- #ifdef STBIR_SIMD
- if ( width_times_channels >= 16 )
- {
- float const * end_input_m16 = input + width_times_channels - 16;
- decode_end -= 16;
- STBIR_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- STBIR_NO_UNROLL(decode);
- #ifdef stbir__decode_swizzle
- #ifdef STBIR_SIMD8
- {
- stbir__simdf8 of0,of1;
- stbir__simdf8_load( of0, input );
- stbir__simdf8_load( of1, input+8 );
- stbir__decode_simdf8_flip( of0 );
- stbir__decode_simdf8_flip( of1 );
- stbir__simdf8_store( decode, of0 );
- stbir__simdf8_store( decode+8, of1 );
- }
- #else
- {
- stbir__simdf of0,of1,of2,of3;
- stbir__simdf_load( of0, input );
- stbir__simdf_load( of1, input+4 );
- stbir__simdf_load( of2, input+8 );
- stbir__simdf_load( of3, input+12 );
- stbir__decode_simdf4_flip( of0 );
- stbir__decode_simdf4_flip( of1 );
- stbir__decode_simdf4_flip( of2 );
- stbir__decode_simdf4_flip( of3 );
- stbir__simdf_store( decode, of0 );
- stbir__simdf_store( decode+4, of1 );
- stbir__simdf_store( decode+8, of2 );
- stbir__simdf_store( decode+12, of3 );
- }
- #endif
- #endif
- decode += 16;
- input += 16;
- if ( decode <= decode_end )
- continue;
- if ( decode == ( decode_end + 16 ) )
- break;
- decode = decode_end; // backup and do last couple
- input = end_input_m16;
- }
- return decode_end + 16;
- }
- #endif
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- decode += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( decode <= decode_end )
- {
- STBIR_SIMD_NO_UNROLL(decode);
- decode[0-4] = input[stbir__decode_order0];
- decode[1-4] = input[stbir__decode_order1];
- decode[2-4] = input[stbir__decode_order2];
- decode[3-4] = input[stbir__decode_order3];
- decode += 4;
- input += 4;
- }
- decode -= 4;
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( decode < decode_end )
- {
- STBIR_NO_UNROLL(decode);
- decode[0] = input[stbir__decode_order0];
- #if stbir__coder_min_num >= 2
- decode[1] = input[stbir__decode_order1];
- #endif
- #if stbir__coder_min_num >= 3
- decode[2] = input[stbir__decode_order2];
- #endif
- decode += stbir__coder_min_num;
- input += stbir__coder_min_num;
- }
- #endif
- return decode_end;
- #else
- if ( (void*)decodep != inputp )
- STBIR_MEMCPY( decodep, inputp, width_times_channels * sizeof( float ) );
- return decodep + width_times_channels;
- #endif
- }
- static void STBIR__CODER_NAME( stbir__encode_float_linear )( void * outputp, int width_times_channels, float const * encode )
- {
- #if !defined( STBIR_FLOAT_HIGH_CLAMP ) && !defined(STBIR_FLOAT_LO_CLAMP) && !defined(stbir__decode_swizzle)
- if ( (void*)outputp != (void*) encode )
- STBIR_MEMCPY( outputp, encode, width_times_channels * sizeof( float ) );
- #else
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = (float*) outputp;
- float * end_output = ( (float*) output ) + width_times_channels;
- #ifdef STBIR_FLOAT_HIGH_CLAMP
- #define stbir_scalar_hi_clamp( v ) if ( v > STBIR_FLOAT_HIGH_CLAMP ) v = STBIR_FLOAT_HIGH_CLAMP;
- #else
- #define stbir_scalar_hi_clamp( v )
- #endif
- #ifdef STBIR_FLOAT_LOW_CLAMP
- #define stbir_scalar_lo_clamp( v ) if ( v < STBIR_FLOAT_LOW_CLAMP ) v = STBIR_FLOAT_LOW_CLAMP;
- #else
- #define stbir_scalar_lo_clamp( v )
- #endif
- #ifdef STBIR_SIMD
- #ifdef STBIR_FLOAT_HIGH_CLAMP
- const stbir__simdfX high_clamp = stbir__simdf_frepX(STBIR_FLOAT_HIGH_CLAMP);
- #endif
- #ifdef STBIR_FLOAT_LOW_CLAMP
- const stbir__simdfX low_clamp = stbir__simdf_frepX(STBIR_FLOAT_LOW_CLAMP);
- #endif
- if ( width_times_channels >= ( stbir__simdfX_float_count * 2 ) )
- {
- float const * end_encode_m8 = encode + width_times_channels - ( stbir__simdfX_float_count * 2 );
- end_output -= ( stbir__simdfX_float_count * 2 );
- STBIR_SIMD_NO_UNROLL_LOOP_START_INF_FOR
- for(;;)
- {
- stbir__simdfX e0, e1;
- STBIR_SIMD_NO_UNROLL(encode);
- stbir__simdfX_load( e0, encode );
- stbir__simdfX_load( e1, encode+stbir__simdfX_float_count );
- #ifdef STBIR_FLOAT_HIGH_CLAMP
- stbir__simdfX_min( e0, e0, high_clamp );
- stbir__simdfX_min( e1, e1, high_clamp );
- #endif
- #ifdef STBIR_FLOAT_LOW_CLAMP
- stbir__simdfX_max( e0, e0, low_clamp );
- stbir__simdfX_max( e1, e1, low_clamp );
- #endif
- stbir__encode_simdfX_unflip( e0 );
- stbir__encode_simdfX_unflip( e1 );
- stbir__simdfX_store( output, e0 );
- stbir__simdfX_store( output+stbir__simdfX_float_count, e1 );
- encode += stbir__simdfX_float_count * 2;
- output += stbir__simdfX_float_count * 2;
- if ( output < end_output )
- continue;
- if ( output == ( end_output + ( stbir__simdfX_float_count * 2 ) ) )
- break;
- output = end_output; // backup and do last couple
- encode = end_encode_m8;
- }
- return;
- }
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- stbir__simdf e0;
- STBIR_NO_UNROLL(encode);
- stbir__simdf_load( e0, encode );
- #ifdef STBIR_FLOAT_HIGH_CLAMP
- stbir__simdf_min( e0, e0, high_clamp );
- #endif
- #ifdef STBIR_FLOAT_LOW_CLAMP
- stbir__simdf_max( e0, e0, low_clamp );
- #endif
- stbir__encode_simdf4_unflip( e0 );
- stbir__simdf_store( output-4, e0 );
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #else
- // try to do blocks of 4 when you can
- #if stbir__coder_min_num != 3 // doesn't divide cleanly by four
- output += 4;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while( output <= end_output )
- {
- float e;
- STBIR_SIMD_NO_UNROLL(encode);
- e = encode[ stbir__encode_order0 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[0-4] = e;
- e = encode[ stbir__encode_order1 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[1-4] = e;
- e = encode[ stbir__encode_order2 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[2-4] = e;
- e = encode[ stbir__encode_order3 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[3-4] = e;
- output += 4;
- encode += 4;
- }
- output -= 4;
- #endif
- #endif
- // do the remnants
- #if stbir__coder_min_num < 4
- STBIR_NO_UNROLL_LOOP_START
- while( output < end_output )
- {
- float e;
- STBIR_NO_UNROLL(encode);
- e = encode[ stbir__encode_order0 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[0] = e;
- #if stbir__coder_min_num >= 2
- e = encode[ stbir__encode_order1 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[1] = e;
- #endif
- #if stbir__coder_min_num >= 3
- e = encode[ stbir__encode_order2 ]; stbir_scalar_hi_clamp( e ); stbir_scalar_lo_clamp( e ); output[2] = e;
- #endif
- output += stbir__coder_min_num;
- encode += stbir__coder_min_num;
- }
- #endif
- #endif
- }
- #undef stbir__decode_suffix
- #undef stbir__decode_simdf8_flip
- #undef stbir__decode_simdf4_flip
- #undef stbir__decode_order0
- #undef stbir__decode_order1
- #undef stbir__decode_order2
- #undef stbir__decode_order3
- #undef stbir__encode_order0
- #undef stbir__encode_order1
- #undef stbir__encode_order2
- #undef stbir__encode_order3
- #undef stbir__encode_simdf8_unflip
- #undef stbir__encode_simdf4_unflip
- #undef stbir__encode_simdfX_unflip
- #undef STBIR__CODER_NAME
- #undef stbir__coder_min_num
- #undef stbir__decode_swizzle
- #undef stbir_scalar_hi_clamp
- #undef stbir_scalar_lo_clamp
- #undef STB_IMAGE_RESIZE_DO_CODERS
- #elif defined( STB_IMAGE_RESIZE_DO_VERTICALS)
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #define STBIR_chans( start, end ) STBIR_strs_join14(start,STBIR__vertical_channels,end,_cont)
- #else
- #define STBIR_chans( start, end ) STBIR_strs_join1(start,STBIR__vertical_channels,end)
- #endif
- #if STBIR__vertical_channels >= 1
- #define stbIF0( code ) code
- #else
- #define stbIF0( code )
- #endif
- #if STBIR__vertical_channels >= 2
- #define stbIF1( code ) code
- #else
- #define stbIF1( code )
- #endif
- #if STBIR__vertical_channels >= 3
- #define stbIF2( code ) code
- #else
- #define stbIF2( code )
- #endif
- #if STBIR__vertical_channels >= 4
- #define stbIF3( code ) code
- #else
- #define stbIF3( code )
- #endif
- #if STBIR__vertical_channels >= 5
- #define stbIF4( code ) code
- #else
- #define stbIF4( code )
- #endif
- #if STBIR__vertical_channels >= 6
- #define stbIF5( code ) code
- #else
- #define stbIF5( code )
- #endif
- #if STBIR__vertical_channels >= 7
- #define stbIF6( code ) code
- #else
- #define stbIF6( code )
- #endif
- #if STBIR__vertical_channels >= 8
- #define stbIF7( code ) code
- #else
- #define stbIF7( code )
- #endif
- static void STBIR_chans( stbir__vertical_scatter_with_,_coeffs)( float ** outputs, float const * vertical_coefficients, float const * input, float const * input_end )
- {
- stbIF0( float STBIR_SIMD_STREAMOUT_PTR( * ) output0 = outputs[0]; float c0s = vertical_coefficients[0]; )
- stbIF1( float STBIR_SIMD_STREAMOUT_PTR( * ) output1 = outputs[1]; float c1s = vertical_coefficients[1]; )
- stbIF2( float STBIR_SIMD_STREAMOUT_PTR( * ) output2 = outputs[2]; float c2s = vertical_coefficients[2]; )
- stbIF3( float STBIR_SIMD_STREAMOUT_PTR( * ) output3 = outputs[3]; float c3s = vertical_coefficients[3]; )
- stbIF4( float STBIR_SIMD_STREAMOUT_PTR( * ) output4 = outputs[4]; float c4s = vertical_coefficients[4]; )
- stbIF5( float STBIR_SIMD_STREAMOUT_PTR( * ) output5 = outputs[5]; float c5s = vertical_coefficients[5]; )
- stbIF6( float STBIR_SIMD_STREAMOUT_PTR( * ) output6 = outputs[6]; float c6s = vertical_coefficients[6]; )
- stbIF7( float STBIR_SIMD_STREAMOUT_PTR( * ) output7 = outputs[7]; float c7s = vertical_coefficients[7]; )
- #ifdef STBIR_SIMD
- {
- stbIF0(stbir__simdfX c0 = stbir__simdf_frepX( c0s ); )
- stbIF1(stbir__simdfX c1 = stbir__simdf_frepX( c1s ); )
- stbIF2(stbir__simdfX c2 = stbir__simdf_frepX( c2s ); )
- stbIF3(stbir__simdfX c3 = stbir__simdf_frepX( c3s ); )
- stbIF4(stbir__simdfX c4 = stbir__simdf_frepX( c4s ); )
- stbIF5(stbir__simdfX c5 = stbir__simdf_frepX( c5s ); )
- stbIF6(stbir__simdfX c6 = stbir__simdf_frepX( c6s ); )
- stbIF7(stbir__simdfX c7 = stbir__simdf_frepX( c7s ); )
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while ( ( (char*)input_end - (char*) input ) >= (16*stbir__simdfX_float_count) )
- {
- stbir__simdfX o0, o1, o2, o3, r0, r1, r2, r3;
- STBIR_SIMD_NO_UNROLL(output0);
- stbir__simdfX_load( r0, input ); stbir__simdfX_load( r1, input+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input+(3*stbir__simdfX_float_count) );
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( stbir__simdfX_load( o0, output0 ); stbir__simdfX_load( o1, output0+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output0+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output0+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c0 ); stbir__simdfX_madd( o1, o1, r1, c0 ); stbir__simdfX_madd( o2, o2, r2, c0 ); stbir__simdfX_madd( o3, o3, r3, c0 );
- stbir__simdfX_store( output0, o0 ); stbir__simdfX_store( output0+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output0+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output0+(3*stbir__simdfX_float_count), o3 ); )
- stbIF1( stbir__simdfX_load( o0, output1 ); stbir__simdfX_load( o1, output1+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output1+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output1+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c1 ); stbir__simdfX_madd( o1, o1, r1, c1 ); stbir__simdfX_madd( o2, o2, r2, c1 ); stbir__simdfX_madd( o3, o3, r3, c1 );
- stbir__simdfX_store( output1, o0 ); stbir__simdfX_store( output1+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output1+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output1+(3*stbir__simdfX_float_count), o3 ); )
- stbIF2( stbir__simdfX_load( o0, output2 ); stbir__simdfX_load( o1, output2+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output2+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output2+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c2 ); stbir__simdfX_madd( o1, o1, r1, c2 ); stbir__simdfX_madd( o2, o2, r2, c2 ); stbir__simdfX_madd( o3, o3, r3, c2 );
- stbir__simdfX_store( output2, o0 ); stbir__simdfX_store( output2+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output2+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output2+(3*stbir__simdfX_float_count), o3 ); )
- stbIF3( stbir__simdfX_load( o0, output3 ); stbir__simdfX_load( o1, output3+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output3+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output3+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c3 ); stbir__simdfX_madd( o1, o1, r1, c3 ); stbir__simdfX_madd( o2, o2, r2, c3 ); stbir__simdfX_madd( o3, o3, r3, c3 );
- stbir__simdfX_store( output3, o0 ); stbir__simdfX_store( output3+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output3+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output3+(3*stbir__simdfX_float_count), o3 ); )
- stbIF4( stbir__simdfX_load( o0, output4 ); stbir__simdfX_load( o1, output4+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output4+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output4+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c4 ); stbir__simdfX_madd( o1, o1, r1, c4 ); stbir__simdfX_madd( o2, o2, r2, c4 ); stbir__simdfX_madd( o3, o3, r3, c4 );
- stbir__simdfX_store( output4, o0 ); stbir__simdfX_store( output4+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output4+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output4+(3*stbir__simdfX_float_count), o3 ); )
- stbIF5( stbir__simdfX_load( o0, output5 ); stbir__simdfX_load( o1, output5+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output5+(2*stbir__simdfX_float_count)); stbir__simdfX_load( o3, output5+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c5 ); stbir__simdfX_madd( o1, o1, r1, c5 ); stbir__simdfX_madd( o2, o2, r2, c5 ); stbir__simdfX_madd( o3, o3, r3, c5 );
- stbir__simdfX_store( output5, o0 ); stbir__simdfX_store( output5+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output5+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output5+(3*stbir__simdfX_float_count), o3 ); )
- stbIF6( stbir__simdfX_load( o0, output6 ); stbir__simdfX_load( o1, output6+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output6+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output6+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c6 ); stbir__simdfX_madd( o1, o1, r1, c6 ); stbir__simdfX_madd( o2, o2, r2, c6 ); stbir__simdfX_madd( o3, o3, r3, c6 );
- stbir__simdfX_store( output6, o0 ); stbir__simdfX_store( output6+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output6+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output6+(3*stbir__simdfX_float_count), o3 ); )
- stbIF7( stbir__simdfX_load( o0, output7 ); stbir__simdfX_load( o1, output7+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output7+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output7+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c7 ); stbir__simdfX_madd( o1, o1, r1, c7 ); stbir__simdfX_madd( o2, o2, r2, c7 ); stbir__simdfX_madd( o3, o3, r3, c7 );
- stbir__simdfX_store( output7, o0 ); stbir__simdfX_store( output7+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output7+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output7+(3*stbir__simdfX_float_count), o3 ); )
- #else
- stbIF0( stbir__simdfX_mult( o0, r0, c0 ); stbir__simdfX_mult( o1, r1, c0 ); stbir__simdfX_mult( o2, r2, c0 ); stbir__simdfX_mult( o3, r3, c0 );
- stbir__simdfX_store( output0, o0 ); stbir__simdfX_store( output0+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output0+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output0+(3*stbir__simdfX_float_count), o3 ); )
- stbIF1( stbir__simdfX_mult( o0, r0, c1 ); stbir__simdfX_mult( o1, r1, c1 ); stbir__simdfX_mult( o2, r2, c1 ); stbir__simdfX_mult( o3, r3, c1 );
- stbir__simdfX_store( output1, o0 ); stbir__simdfX_store( output1+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output1+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output1+(3*stbir__simdfX_float_count), o3 ); )
- stbIF2( stbir__simdfX_mult( o0, r0, c2 ); stbir__simdfX_mult( o1, r1, c2 ); stbir__simdfX_mult( o2, r2, c2 ); stbir__simdfX_mult( o3, r3, c2 );
- stbir__simdfX_store( output2, o0 ); stbir__simdfX_store( output2+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output2+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output2+(3*stbir__simdfX_float_count), o3 ); )
- stbIF3( stbir__simdfX_mult( o0, r0, c3 ); stbir__simdfX_mult( o1, r1, c3 ); stbir__simdfX_mult( o2, r2, c3 ); stbir__simdfX_mult( o3, r3, c3 );
- stbir__simdfX_store( output3, o0 ); stbir__simdfX_store( output3+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output3+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output3+(3*stbir__simdfX_float_count), o3 ); )
- stbIF4( stbir__simdfX_mult( o0, r0, c4 ); stbir__simdfX_mult( o1, r1, c4 ); stbir__simdfX_mult( o2, r2, c4 ); stbir__simdfX_mult( o3, r3, c4 );
- stbir__simdfX_store( output4, o0 ); stbir__simdfX_store( output4+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output4+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output4+(3*stbir__simdfX_float_count), o3 ); )
- stbIF5( stbir__simdfX_mult( o0, r0, c5 ); stbir__simdfX_mult( o1, r1, c5 ); stbir__simdfX_mult( o2, r2, c5 ); stbir__simdfX_mult( o3, r3, c5 );
- stbir__simdfX_store( output5, o0 ); stbir__simdfX_store( output5+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output5+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output5+(3*stbir__simdfX_float_count), o3 ); )
- stbIF6( stbir__simdfX_mult( o0, r0, c6 ); stbir__simdfX_mult( o1, r1, c6 ); stbir__simdfX_mult( o2, r2, c6 ); stbir__simdfX_mult( o3, r3, c6 );
- stbir__simdfX_store( output6, o0 ); stbir__simdfX_store( output6+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output6+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output6+(3*stbir__simdfX_float_count), o3 ); )
- stbIF7( stbir__simdfX_mult( o0, r0, c7 ); stbir__simdfX_mult( o1, r1, c7 ); stbir__simdfX_mult( o2, r2, c7 ); stbir__simdfX_mult( o3, r3, c7 );
- stbir__simdfX_store( output7, o0 ); stbir__simdfX_store( output7+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output7+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output7+(3*stbir__simdfX_float_count), o3 ); )
- #endif
- input += (4*stbir__simdfX_float_count);
- stbIF0( output0 += (4*stbir__simdfX_float_count); ) stbIF1( output1 += (4*stbir__simdfX_float_count); ) stbIF2( output2 += (4*stbir__simdfX_float_count); ) stbIF3( output3 += (4*stbir__simdfX_float_count); ) stbIF4( output4 += (4*stbir__simdfX_float_count); ) stbIF5( output5 += (4*stbir__simdfX_float_count); ) stbIF6( output6 += (4*stbir__simdfX_float_count); ) stbIF7( output7 += (4*stbir__simdfX_float_count); )
- }
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while ( ( (char*)input_end - (char*) input ) >= 16 )
- {
- stbir__simdf o0, r0;
- STBIR_SIMD_NO_UNROLL(output0);
- stbir__simdf_load( r0, input );
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( stbir__simdf_load( o0, output0 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c0 ) ); stbir__simdf_store( output0, o0 ); )
- stbIF1( stbir__simdf_load( o0, output1 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c1 ) ); stbir__simdf_store( output1, o0 ); )
- stbIF2( stbir__simdf_load( o0, output2 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c2 ) ); stbir__simdf_store( output2, o0 ); )
- stbIF3( stbir__simdf_load( o0, output3 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c3 ) ); stbir__simdf_store( output3, o0 ); )
- stbIF4( stbir__simdf_load( o0, output4 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c4 ) ); stbir__simdf_store( output4, o0 ); )
- stbIF5( stbir__simdf_load( o0, output5 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c5 ) ); stbir__simdf_store( output5, o0 ); )
- stbIF6( stbir__simdf_load( o0, output6 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c6 ) ); stbir__simdf_store( output6, o0 ); )
- stbIF7( stbir__simdf_load( o0, output7 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c7 ) ); stbir__simdf_store( output7, o0 ); )
- #else
- stbIF0( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c0 ) ); stbir__simdf_store( output0, o0 ); )
- stbIF1( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c1 ) ); stbir__simdf_store( output1, o0 ); )
- stbIF2( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c2 ) ); stbir__simdf_store( output2, o0 ); )
- stbIF3( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c3 ) ); stbir__simdf_store( output3, o0 ); )
- stbIF4( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c4 ) ); stbir__simdf_store( output4, o0 ); )
- stbIF5( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c5 ) ); stbir__simdf_store( output5, o0 ); )
- stbIF6( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c6 ) ); stbir__simdf_store( output6, o0 ); )
- stbIF7( stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c7 ) ); stbir__simdf_store( output7, o0 ); )
- #endif
- input += 4;
- stbIF0( output0 += 4; ) stbIF1( output1 += 4; ) stbIF2( output2 += 4; ) stbIF3( output3 += 4; ) stbIF4( output4 += 4; ) stbIF5( output5 += 4; ) stbIF6( output6 += 4; ) stbIF7( output7 += 4; )
- }
- }
- #else
- STBIR_NO_UNROLL_LOOP_START
- while ( ( (char*)input_end - (char*) input ) >= 16 )
- {
- float r0, r1, r2, r3;
- STBIR_NO_UNROLL(input);
- r0 = input[0], r1 = input[1], r2 = input[2], r3 = input[3];
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( output0[0] += ( r0 * c0s ); output0[1] += ( r1 * c0s ); output0[2] += ( r2 * c0s ); output0[3] += ( r3 * c0s ); )
- stbIF1( output1[0] += ( r0 * c1s ); output1[1] += ( r1 * c1s ); output1[2] += ( r2 * c1s ); output1[3] += ( r3 * c1s ); )
- stbIF2( output2[0] += ( r0 * c2s ); output2[1] += ( r1 * c2s ); output2[2] += ( r2 * c2s ); output2[3] += ( r3 * c2s ); )
- stbIF3( output3[0] += ( r0 * c3s ); output3[1] += ( r1 * c3s ); output3[2] += ( r2 * c3s ); output3[3] += ( r3 * c3s ); )
- stbIF4( output4[0] += ( r0 * c4s ); output4[1] += ( r1 * c4s ); output4[2] += ( r2 * c4s ); output4[3] += ( r3 * c4s ); )
- stbIF5( output5[0] += ( r0 * c5s ); output5[1] += ( r1 * c5s ); output5[2] += ( r2 * c5s ); output5[3] += ( r3 * c5s ); )
- stbIF6( output6[0] += ( r0 * c6s ); output6[1] += ( r1 * c6s ); output6[2] += ( r2 * c6s ); output6[3] += ( r3 * c6s ); )
- stbIF7( output7[0] += ( r0 * c7s ); output7[1] += ( r1 * c7s ); output7[2] += ( r2 * c7s ); output7[3] += ( r3 * c7s ); )
- #else
- stbIF0( output0[0] = ( r0 * c0s ); output0[1] = ( r1 * c0s ); output0[2] = ( r2 * c0s ); output0[3] = ( r3 * c0s ); )
- stbIF1( output1[0] = ( r0 * c1s ); output1[1] = ( r1 * c1s ); output1[2] = ( r2 * c1s ); output1[3] = ( r3 * c1s ); )
- stbIF2( output2[0] = ( r0 * c2s ); output2[1] = ( r1 * c2s ); output2[2] = ( r2 * c2s ); output2[3] = ( r3 * c2s ); )
- stbIF3( output3[0] = ( r0 * c3s ); output3[1] = ( r1 * c3s ); output3[2] = ( r2 * c3s ); output3[3] = ( r3 * c3s ); )
- stbIF4( output4[0] = ( r0 * c4s ); output4[1] = ( r1 * c4s ); output4[2] = ( r2 * c4s ); output4[3] = ( r3 * c4s ); )
- stbIF5( output5[0] = ( r0 * c5s ); output5[1] = ( r1 * c5s ); output5[2] = ( r2 * c5s ); output5[3] = ( r3 * c5s ); )
- stbIF6( output6[0] = ( r0 * c6s ); output6[1] = ( r1 * c6s ); output6[2] = ( r2 * c6s ); output6[3] = ( r3 * c6s ); )
- stbIF7( output7[0] = ( r0 * c7s ); output7[1] = ( r1 * c7s ); output7[2] = ( r2 * c7s ); output7[3] = ( r3 * c7s ); )
- #endif
- input += 4;
- stbIF0( output0 += 4; ) stbIF1( output1 += 4; ) stbIF2( output2 += 4; ) stbIF3( output3 += 4; ) stbIF4( output4 += 4; ) stbIF5( output5 += 4; ) stbIF6( output6 += 4; ) stbIF7( output7 += 4; )
- }
- #endif
- STBIR_NO_UNROLL_LOOP_START
- while ( input < input_end )
- {
- float r = input[0];
- STBIR_NO_UNROLL(output0);
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( output0[0] += ( r * c0s ); )
- stbIF1( output1[0] += ( r * c1s ); )
- stbIF2( output2[0] += ( r * c2s ); )
- stbIF3( output3[0] += ( r * c3s ); )
- stbIF4( output4[0] += ( r * c4s ); )
- stbIF5( output5[0] += ( r * c5s ); )
- stbIF6( output6[0] += ( r * c6s ); )
- stbIF7( output7[0] += ( r * c7s ); )
- #else
- stbIF0( output0[0] = ( r * c0s ); )
- stbIF1( output1[0] = ( r * c1s ); )
- stbIF2( output2[0] = ( r * c2s ); )
- stbIF3( output3[0] = ( r * c3s ); )
- stbIF4( output4[0] = ( r * c4s ); )
- stbIF5( output5[0] = ( r * c5s ); )
- stbIF6( output6[0] = ( r * c6s ); )
- stbIF7( output7[0] = ( r * c7s ); )
- #endif
- ++input;
- stbIF0( ++output0; ) stbIF1( ++output1; ) stbIF2( ++output2; ) stbIF3( ++output3; ) stbIF4( ++output4; ) stbIF5( ++output5; ) stbIF6( ++output6; ) stbIF7( ++output7; )
- }
- }
- static void STBIR_chans( stbir__vertical_gather_with_,_coeffs)( float * outputp, float const * vertical_coefficients, float const ** inputs, float const * input0_end )
- {
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = outputp;
- stbIF0( float const * input0 = inputs[0]; float c0s = vertical_coefficients[0]; )
- stbIF1( float const * input1 = inputs[1]; float c1s = vertical_coefficients[1]; )
- stbIF2( float const * input2 = inputs[2]; float c2s = vertical_coefficients[2]; )
- stbIF3( float const * input3 = inputs[3]; float c3s = vertical_coefficients[3]; )
- stbIF4( float const * input4 = inputs[4]; float c4s = vertical_coefficients[4]; )
- stbIF5( float const * input5 = inputs[5]; float c5s = vertical_coefficients[5]; )
- stbIF6( float const * input6 = inputs[6]; float c6s = vertical_coefficients[6]; )
- stbIF7( float const * input7 = inputs[7]; float c7s = vertical_coefficients[7]; )
- #if ( STBIR__vertical_channels == 1 ) && !defined(STB_IMAGE_RESIZE_VERTICAL_CONTINUE)
- // check single channel one weight
- if ( ( c0s >= (1.0f-0.000001f) ) && ( c0s <= (1.0f+0.000001f) ) )
- {
- STBIR_MEMCPY( output, input0, (char*)input0_end - (char*)input0 );
- return;
- }
- #endif
- #ifdef STBIR_SIMD
- {
- stbIF0(stbir__simdfX c0 = stbir__simdf_frepX( c0s ); )
- stbIF1(stbir__simdfX c1 = stbir__simdf_frepX( c1s ); )
- stbIF2(stbir__simdfX c2 = stbir__simdf_frepX( c2s ); )
- stbIF3(stbir__simdfX c3 = stbir__simdf_frepX( c3s ); )
- stbIF4(stbir__simdfX c4 = stbir__simdf_frepX( c4s ); )
- stbIF5(stbir__simdfX c5 = stbir__simdf_frepX( c5s ); )
- stbIF6(stbir__simdfX c6 = stbir__simdf_frepX( c6s ); )
- stbIF7(stbir__simdfX c7 = stbir__simdf_frepX( c7s ); )
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while ( ( (char*)input0_end - (char*) input0 ) >= (16*stbir__simdfX_float_count) )
- {
- stbir__simdfX o0, o1, o2, o3, r0, r1, r2, r3;
- STBIR_SIMD_NO_UNROLL(output);
- // prefetch four loop iterations ahead (doesn't affect much for small resizes, but helps with big ones)
- stbIF0( stbir__prefetch( input0 + (16*stbir__simdfX_float_count) ); )
- stbIF1( stbir__prefetch( input1 + (16*stbir__simdfX_float_count) ); )
- stbIF2( stbir__prefetch( input2 + (16*stbir__simdfX_float_count) ); )
- stbIF3( stbir__prefetch( input3 + (16*stbir__simdfX_float_count) ); )
- stbIF4( stbir__prefetch( input4 + (16*stbir__simdfX_float_count) ); )
- stbIF5( stbir__prefetch( input5 + (16*stbir__simdfX_float_count) ); )
- stbIF6( stbir__prefetch( input6 + (16*stbir__simdfX_float_count) ); )
- stbIF7( stbir__prefetch( input7 + (16*stbir__simdfX_float_count) ); )
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( stbir__simdfX_load( o0, output ); stbir__simdfX_load( o1, output+stbir__simdfX_float_count ); stbir__simdfX_load( o2, output+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( o3, output+(3*stbir__simdfX_float_count) );
- stbir__simdfX_load( r0, input0 ); stbir__simdfX_load( r1, input0+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input0+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input0+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c0 ); stbir__simdfX_madd( o1, o1, r1, c0 ); stbir__simdfX_madd( o2, o2, r2, c0 ); stbir__simdfX_madd( o3, o3, r3, c0 ); )
- #else
- stbIF0( stbir__simdfX_load( r0, input0 ); stbir__simdfX_load( r1, input0+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input0+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input0+(3*stbir__simdfX_float_count) );
- stbir__simdfX_mult( o0, r0, c0 ); stbir__simdfX_mult( o1, r1, c0 ); stbir__simdfX_mult( o2, r2, c0 ); stbir__simdfX_mult( o3, r3, c0 ); )
- #endif
- stbIF1( stbir__simdfX_load( r0, input1 ); stbir__simdfX_load( r1, input1+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input1+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input1+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c1 ); stbir__simdfX_madd( o1, o1, r1, c1 ); stbir__simdfX_madd( o2, o2, r2, c1 ); stbir__simdfX_madd( o3, o3, r3, c1 ); )
- stbIF2( stbir__simdfX_load( r0, input2 ); stbir__simdfX_load( r1, input2+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input2+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input2+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c2 ); stbir__simdfX_madd( o1, o1, r1, c2 ); stbir__simdfX_madd( o2, o2, r2, c2 ); stbir__simdfX_madd( o3, o3, r3, c2 ); )
- stbIF3( stbir__simdfX_load( r0, input3 ); stbir__simdfX_load( r1, input3+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input3+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input3+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c3 ); stbir__simdfX_madd( o1, o1, r1, c3 ); stbir__simdfX_madd( o2, o2, r2, c3 ); stbir__simdfX_madd( o3, o3, r3, c3 ); )
- stbIF4( stbir__simdfX_load( r0, input4 ); stbir__simdfX_load( r1, input4+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input4+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input4+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c4 ); stbir__simdfX_madd( o1, o1, r1, c4 ); stbir__simdfX_madd( o2, o2, r2, c4 ); stbir__simdfX_madd( o3, o3, r3, c4 ); )
- stbIF5( stbir__simdfX_load( r0, input5 ); stbir__simdfX_load( r1, input5+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input5+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input5+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c5 ); stbir__simdfX_madd( o1, o1, r1, c5 ); stbir__simdfX_madd( o2, o2, r2, c5 ); stbir__simdfX_madd( o3, o3, r3, c5 ); )
- stbIF6( stbir__simdfX_load( r0, input6 ); stbir__simdfX_load( r1, input6+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input6+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input6+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c6 ); stbir__simdfX_madd( o1, o1, r1, c6 ); stbir__simdfX_madd( o2, o2, r2, c6 ); stbir__simdfX_madd( o3, o3, r3, c6 ); )
- stbIF7( stbir__simdfX_load( r0, input7 ); stbir__simdfX_load( r1, input7+stbir__simdfX_float_count ); stbir__simdfX_load( r2, input7+(2*stbir__simdfX_float_count) ); stbir__simdfX_load( r3, input7+(3*stbir__simdfX_float_count) );
- stbir__simdfX_madd( o0, o0, r0, c7 ); stbir__simdfX_madd( o1, o1, r1, c7 ); stbir__simdfX_madd( o2, o2, r2, c7 ); stbir__simdfX_madd( o3, o3, r3, c7 ); )
- stbir__simdfX_store( output, o0 ); stbir__simdfX_store( output+stbir__simdfX_float_count, o1 ); stbir__simdfX_store( output+(2*stbir__simdfX_float_count), o2 ); stbir__simdfX_store( output+(3*stbir__simdfX_float_count), o3 );
- output += (4*stbir__simdfX_float_count);
- stbIF0( input0 += (4*stbir__simdfX_float_count); ) stbIF1( input1 += (4*stbir__simdfX_float_count); ) stbIF2( input2 += (4*stbir__simdfX_float_count); ) stbIF3( input3 += (4*stbir__simdfX_float_count); ) stbIF4( input4 += (4*stbir__simdfX_float_count); ) stbIF5( input5 += (4*stbir__simdfX_float_count); ) stbIF6( input6 += (4*stbir__simdfX_float_count); ) stbIF7( input7 += (4*stbir__simdfX_float_count); )
- }
- STBIR_SIMD_NO_UNROLL_LOOP_START
- while ( ( (char*)input0_end - (char*) input0 ) >= 16 )
- {
- stbir__simdf o0, r0;
- STBIR_SIMD_NO_UNROLL(output);
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( stbir__simdf_load( o0, output ); stbir__simdf_load( r0, input0 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c0 ) ); )
- #else
- stbIF0( stbir__simdf_load( r0, input0 ); stbir__simdf_mult( o0, r0, stbir__if_simdf8_cast_to_simdf4( c0 ) ); )
- #endif
- stbIF1( stbir__simdf_load( r0, input1 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c1 ) ); )
- stbIF2( stbir__simdf_load( r0, input2 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c2 ) ); )
- stbIF3( stbir__simdf_load( r0, input3 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c3 ) ); )
- stbIF4( stbir__simdf_load( r0, input4 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c4 ) ); )
- stbIF5( stbir__simdf_load( r0, input5 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c5 ) ); )
- stbIF6( stbir__simdf_load( r0, input6 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c6 ) ); )
- stbIF7( stbir__simdf_load( r0, input7 ); stbir__simdf_madd( o0, o0, r0, stbir__if_simdf8_cast_to_simdf4( c7 ) ); )
- stbir__simdf_store( output, o0 );
- output += 4;
- stbIF0( input0 += 4; ) stbIF1( input1 += 4; ) stbIF2( input2 += 4; ) stbIF3( input3 += 4; ) stbIF4( input4 += 4; ) stbIF5( input5 += 4; ) stbIF6( input6 += 4; ) stbIF7( input7 += 4; )
- }
- }
- #else
- STBIR_NO_UNROLL_LOOP_START
- while ( ( (char*)input0_end - (char*) input0 ) >= 16 )
- {
- float o0, o1, o2, o3;
- STBIR_NO_UNROLL(output);
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( o0 = output[0] + input0[0] * c0s; o1 = output[1] + input0[1] * c0s; o2 = output[2] + input0[2] * c0s; o3 = output[3] + input0[3] * c0s; )
- #else
- stbIF0( o0 = input0[0] * c0s; o1 = input0[1] * c0s; o2 = input0[2] * c0s; o3 = input0[3] * c0s; )
- #endif
- stbIF1( o0 += input1[0] * c1s; o1 += input1[1] * c1s; o2 += input1[2] * c1s; o3 += input1[3] * c1s; )
- stbIF2( o0 += input2[0] * c2s; o1 += input2[1] * c2s; o2 += input2[2] * c2s; o3 += input2[3] * c2s; )
- stbIF3( o0 += input3[0] * c3s; o1 += input3[1] * c3s; o2 += input3[2] * c3s; o3 += input3[3] * c3s; )
- stbIF4( o0 += input4[0] * c4s; o1 += input4[1] * c4s; o2 += input4[2] * c4s; o3 += input4[3] * c4s; )
- stbIF5( o0 += input5[0] * c5s; o1 += input5[1] * c5s; o2 += input5[2] * c5s; o3 += input5[3] * c5s; )
- stbIF6( o0 += input6[0] * c6s; o1 += input6[1] * c6s; o2 += input6[2] * c6s; o3 += input6[3] * c6s; )
- stbIF7( o0 += input7[0] * c7s; o1 += input7[1] * c7s; o2 += input7[2] * c7s; o3 += input7[3] * c7s; )
- output[0] = o0; output[1] = o1; output[2] = o2; output[3] = o3;
- output += 4;
- stbIF0( input0 += 4; ) stbIF1( input1 += 4; ) stbIF2( input2 += 4; ) stbIF3( input3 += 4; ) stbIF4( input4 += 4; ) stbIF5( input5 += 4; ) stbIF6( input6 += 4; ) stbIF7( input7 += 4; )
- }
- #endif
- STBIR_NO_UNROLL_LOOP_START
- while ( input0 < input0_end )
- {
- float o0;
- STBIR_NO_UNROLL(output);
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- stbIF0( o0 = output[0] + input0[0] * c0s; )
- #else
- stbIF0( o0 = input0[0] * c0s; )
- #endif
- stbIF1( o0 += input1[0] * c1s; )
- stbIF2( o0 += input2[0] * c2s; )
- stbIF3( o0 += input3[0] * c3s; )
- stbIF4( o0 += input4[0] * c4s; )
- stbIF5( o0 += input5[0] * c5s; )
- stbIF6( o0 += input6[0] * c6s; )
- stbIF7( o0 += input7[0] * c7s; )
- output[0] = o0;
- ++output;
- stbIF0( ++input0; ) stbIF1( ++input1; ) stbIF2( ++input2; ) stbIF3( ++input3; ) stbIF4( ++input4; ) stbIF5( ++input5; ) stbIF6( ++input6; ) stbIF7( ++input7; )
- }
- }
- #undef stbIF0
- #undef stbIF1
- #undef stbIF2
- #undef stbIF3
- #undef stbIF4
- #undef stbIF5
- #undef stbIF6
- #undef stbIF7
- #undef STB_IMAGE_RESIZE_DO_VERTICALS
- #undef STBIR__vertical_channels
- #undef STB_IMAGE_RESIZE_DO_HORIZONTALS
- #undef STBIR_strs_join24
- #undef STBIR_strs_join14
- #undef STBIR_chans
- #ifdef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #undef STB_IMAGE_RESIZE_VERTICAL_CONTINUE
- #endif
- #else // !STB_IMAGE_RESIZE_DO_VERTICALS
- #define STBIR_chans( start, end ) STBIR_strs_join1(start,STBIR__horizontal_channels,end)
- #ifndef stbir__2_coeff_only
- #define stbir__2_coeff_only() \
- stbir__1_coeff_only(); \
- stbir__1_coeff_remnant(1);
- #endif
- #ifndef stbir__2_coeff_remnant
- #define stbir__2_coeff_remnant( ofs ) \
- stbir__1_coeff_remnant(ofs); \
- stbir__1_coeff_remnant((ofs)+1);
- #endif
- #ifndef stbir__3_coeff_only
- #define stbir__3_coeff_only() \
- stbir__2_coeff_only(); \
- stbir__1_coeff_remnant(2);
- #endif
- #ifndef stbir__3_coeff_remnant
- #define stbir__3_coeff_remnant( ofs ) \
- stbir__2_coeff_remnant(ofs); \
- stbir__1_coeff_remnant((ofs)+2);
- #endif
- #ifndef stbir__3_coeff_setup
- #define stbir__3_coeff_setup()
- #endif
- #ifndef stbir__4_coeff_start
- #define stbir__4_coeff_start() \
- stbir__2_coeff_only(); \
- stbir__2_coeff_remnant(2);
- #endif
- #ifndef stbir__4_coeff_continue_from_4
- #define stbir__4_coeff_continue_from_4( ofs ) \
- stbir__2_coeff_remnant(ofs); \
- stbir__2_coeff_remnant((ofs)+2);
- #endif
- #ifndef stbir__store_output_tiny
- #define stbir__store_output_tiny stbir__store_output
- #endif
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_1_coeff)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__1_coeff_only();
- stbir__store_output_tiny();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_2_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__2_coeff_only();
- stbir__store_output_tiny();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_3_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__3_coeff_only();
- stbir__store_output_tiny();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_4_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_5_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__1_coeff_remnant(4);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_6_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__2_coeff_remnant(4);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_7_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- stbir__3_coeff_setup();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__3_coeff_remnant(4);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_8_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__4_coeff_continue_from_4(4);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_9_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__4_coeff_continue_from_4(4);
- stbir__1_coeff_remnant(8);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_10_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__4_coeff_continue_from_4(4);
- stbir__2_coeff_remnant(8);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_11_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- stbir__3_coeff_setup();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__4_coeff_continue_from_4(4);
- stbir__3_coeff_remnant(8);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_12_coeffs)( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- stbir__4_coeff_continue_from_4(4);
- stbir__4_coeff_continue_from_4(8);
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_n_coeffs_mod0 )( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- int n = ( ( horizontal_contributors->n1 - horizontal_contributors->n0 + 1 ) - 4 + 3 ) >> 2;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- hc += 4;
- decode += STBIR__horizontal_channels * 4;
- stbir__4_coeff_continue_from_4( 0 );
- --n;
- } while ( n > 0 );
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_n_coeffs_mod1 )( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- int n = ( ( horizontal_contributors->n1 - horizontal_contributors->n0 + 1 ) - 5 + 3 ) >> 2;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- hc += 4;
- decode += STBIR__horizontal_channels * 4;
- stbir__4_coeff_continue_from_4( 0 );
- --n;
- } while ( n > 0 );
- stbir__1_coeff_remnant( 4 );
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_n_coeffs_mod2 )( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- int n = ( ( horizontal_contributors->n1 - horizontal_contributors->n0 + 1 ) - 6 + 3 ) >> 2;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- hc += 4;
- decode += STBIR__horizontal_channels * 4;
- stbir__4_coeff_continue_from_4( 0 );
- --n;
- } while ( n > 0 );
- stbir__2_coeff_remnant( 4 );
- stbir__store_output();
- } while ( output < output_end );
- }
- static void STBIR_chans( stbir__horizontal_gather_,_channels_with_n_coeffs_mod3 )( float * output_buffer, unsigned int output_sub_size, float const * decode_buffer, stbir__contributors const * horizontal_contributors, float const * horizontal_coefficients, int coefficient_width )
- {
- float const * output_end = output_buffer + output_sub_size * STBIR__horizontal_channels;
- float STBIR_SIMD_STREAMOUT_PTR( * ) output = output_buffer;
- stbir__3_coeff_setup();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- float const * decode = decode_buffer + horizontal_contributors->n0 * STBIR__horizontal_channels;
- int n = ( ( horizontal_contributors->n1 - horizontal_contributors->n0 + 1 ) - 7 + 3 ) >> 2;
- float const * hc = horizontal_coefficients;
- stbir__4_coeff_start();
- STBIR_SIMD_NO_UNROLL_LOOP_START
- do {
- hc += 4;
- decode += STBIR__horizontal_channels * 4;
- stbir__4_coeff_continue_from_4( 0 );
- --n;
- } while ( n > 0 );
- stbir__3_coeff_remnant( 4 );
- stbir__store_output();
- } while ( output < output_end );
- }
- static stbir__horizontal_gather_channels_func * STBIR_chans(stbir__horizontal_gather_,_channels_with_n_coeffs_funcs)[4]=
- {
- STBIR_chans(stbir__horizontal_gather_,_channels_with_n_coeffs_mod0),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_n_coeffs_mod1),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_n_coeffs_mod2),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_n_coeffs_mod3),
- };
- static stbir__horizontal_gather_channels_func * STBIR_chans(stbir__horizontal_gather_,_channels_funcs)[12]=
- {
- STBIR_chans(stbir__horizontal_gather_,_channels_with_1_coeff),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_2_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_3_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_4_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_5_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_6_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_7_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_8_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_9_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_10_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_11_coeffs),
- STBIR_chans(stbir__horizontal_gather_,_channels_with_12_coeffs),
- };
- #undef STBIR__horizontal_channels
- #undef STB_IMAGE_RESIZE_DO_HORIZONTALS
- #undef stbir__1_coeff_only
- #undef stbir__1_coeff_remnant
- #undef stbir__2_coeff_only
- #undef stbir__2_coeff_remnant
- #undef stbir__3_coeff_only
- #undef stbir__3_coeff_remnant
- #undef stbir__3_coeff_setup
- #undef stbir__4_coeff_start
- #undef stbir__4_coeff_continue_from_4
- #undef stbir__store_output
- #undef stbir__store_output_tiny
- #undef STBIR_chans
- #endif // HORIZONALS
- #undef STBIR_strs_join2
- #undef STBIR_strs_join1
- #endif // STB_IMAGE_RESIZE_DO_HORIZONTALS/VERTICALS/CODERS
- /*
- ------------------------------------------------------------------------------
- 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.
- ------------------------------------------------------------------------------
- */
|