| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105 |
- // basisu_enc.cpp
- // Copyright (C) 2019-2021 Binomial LLC. All Rights Reserved.
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- #include "basisu_enc.h"
- #include "basisu_resampler.h"
- #include "basisu_resampler_filters.h"
- #include "basisu_etc.h"
- #include "../transcoder/basisu_transcoder.h"
- #include "basisu_bc7enc.h"
- #include "jpgd.h"
- #include "pvpngreader.h"
- #include "basisu_opencl.h"
- #include <vector>
- #define MINIZ_HEADER_FILE_ONLY
- #define MINIZ_NO_ZLIB_COMPATIBLE_NAMES
- #include "basisu_miniz.h"
- #if defined(_WIN32)
- // For QueryPerformanceCounter/QueryPerformanceFrequency
- #define WIN32_LEAN_AND_MEAN
- #include <windows.h>
- #endif
- namespace basisu
- {
- uint64_t interval_timer::g_init_ticks, interval_timer::g_freq;
- double interval_timer::g_timer_freq;
- #if BASISU_SUPPORT_SSE
- bool g_cpu_supports_sse41;
- #endif
- uint8_t g_hamming_dist[256] =
- {
- 0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4,
- 1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
- 1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
- 1, 2, 2, 3, 2, 3, 3, 4, 2, 3, 3, 4, 3, 4, 4, 5,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
- 2, 3, 3, 4, 3, 4, 4, 5, 3, 4, 4, 5, 4, 5, 5, 6,
- 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
- 3, 4, 4, 5, 4, 5, 5, 6, 4, 5, 5, 6, 5, 6, 6, 7,
- 4, 5, 5, 6, 5, 6, 6, 7, 5, 6, 6, 7, 6, 7, 7, 8
- };
- // This is a Public Domain 8x8 font from here:
- // https://github.com/dhepper/font8x8/blob/master/font8x8_basic.h
- const uint8_t g_debug_font8x8_basic[127 - 32 + 1][8] =
- {
- { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}, // U+0020 ( )
- { 0x18, 0x3C, 0x3C, 0x18, 0x18, 0x00, 0x18, 0x00}, // U+0021 (!)
- { 0x36, 0x36, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}, // U+0022 (")
- { 0x36, 0x36, 0x7F, 0x36, 0x7F, 0x36, 0x36, 0x00}, // U+0023 (#)
- { 0x0C, 0x3E, 0x03, 0x1E, 0x30, 0x1F, 0x0C, 0x00}, // U+0024 ($)
- { 0x00, 0x63, 0x33, 0x18, 0x0C, 0x66, 0x63, 0x00}, // U+0025 (%)
- { 0x1C, 0x36, 0x1C, 0x6E, 0x3B, 0x33, 0x6E, 0x00}, // U+0026 (&)
- { 0x06, 0x06, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00}, // U+0027 (')
- { 0x18, 0x0C, 0x06, 0x06, 0x06, 0x0C, 0x18, 0x00}, // U+0028 (()
- { 0x06, 0x0C, 0x18, 0x18, 0x18, 0x0C, 0x06, 0x00}, // U+0029 ())
- { 0x00, 0x66, 0x3C, 0xFF, 0x3C, 0x66, 0x00, 0x00}, // U+002A (*)
- { 0x00, 0x0C, 0x0C, 0x3F, 0x0C, 0x0C, 0x00, 0x00}, // U+002B (+)
- { 0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x0C, 0x06}, // U+002C (,)
- { 0x00, 0x00, 0x00, 0x3F, 0x00, 0x00, 0x00, 0x00}, // U+002D (-)
- { 0x00, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x0C, 0x00}, // U+002E (.)
- { 0x60, 0x30, 0x18, 0x0C, 0x06, 0x03, 0x01, 0x00}, // U+002F (/)
- { 0x3E, 0x63, 0x73, 0x7B, 0x6F, 0x67, 0x3E, 0x00}, // U+0030 (0)
- { 0x0C, 0x0E, 0x0C, 0x0C, 0x0C, 0x0C, 0x3F, 0x00}, // U+0031 (1)
- { 0x1E, 0x33, 0x30, 0x1C, 0x06, 0x33, 0x3F, 0x00}, // U+0032 (2)
- { 0x1E, 0x33, 0x30, 0x1C, 0x30, 0x33, 0x1E, 0x00}, // U+0033 (3)
- { 0x38, 0x3C, 0x36, 0x33, 0x7F, 0x30, 0x78, 0x00}, // U+0034 (4)
- { 0x3F, 0x03, 0x1F, 0x30, 0x30, 0x33, 0x1E, 0x00}, // U+0035 (5)
- { 0x1C, 0x06, 0x03, 0x1F, 0x33, 0x33, 0x1E, 0x00}, // U+0036 (6)
- { 0x3F, 0x33, 0x30, 0x18, 0x0C, 0x0C, 0x0C, 0x00}, // U+0037 (7)
- { 0x1E, 0x33, 0x33, 0x1E, 0x33, 0x33, 0x1E, 0x00}, // U+0038 (8)
- { 0x1E, 0x33, 0x33, 0x3E, 0x30, 0x18, 0x0E, 0x00}, // U+0039 (9)
- { 0x00, 0x0C, 0x0C, 0x00, 0x00, 0x0C, 0x0C, 0x00}, // U+003A (:)
- { 0x00, 0x0C, 0x0C, 0x00, 0x00, 0x0C, 0x0C, 0x06}, // U+003B (;)
- { 0x18, 0x0C, 0x06, 0x03, 0x06, 0x0C, 0x18, 0x00}, // U+003C (<)
- { 0x00, 0x00, 0x3F, 0x00, 0x00, 0x3F, 0x00, 0x00}, // U+003D (=)
- { 0x06, 0x0C, 0x18, 0x30, 0x18, 0x0C, 0x06, 0x00}, // U+003E (>)
- { 0x1E, 0x33, 0x30, 0x18, 0x0C, 0x00, 0x0C, 0x00}, // U+003F (?)
- { 0x3E, 0x63, 0x7B, 0x7B, 0x7B, 0x03, 0x1E, 0x00}, // U+0040 (@)
- { 0x0C, 0x1E, 0x33, 0x33, 0x3F, 0x33, 0x33, 0x00}, // U+0041 (A)
- { 0x3F, 0x66, 0x66, 0x3E, 0x66, 0x66, 0x3F, 0x00}, // U+0042 (B)
- { 0x3C, 0x66, 0x03, 0x03, 0x03, 0x66, 0x3C, 0x00}, // U+0043 (C)
- { 0x1F, 0x36, 0x66, 0x66, 0x66, 0x36, 0x1F, 0x00}, // U+0044 (D)
- { 0x7F, 0x46, 0x16, 0x1E, 0x16, 0x46, 0x7F, 0x00}, // U+0045 (E)
- { 0x7F, 0x46, 0x16, 0x1E, 0x16, 0x06, 0x0F, 0x00}, // U+0046 (F)
- { 0x3C, 0x66, 0x03, 0x03, 0x73, 0x66, 0x7C, 0x00}, // U+0047 (G)
- { 0x33, 0x33, 0x33, 0x3F, 0x33, 0x33, 0x33, 0x00}, // U+0048 (H)
- { 0x1E, 0x0C, 0x0C, 0x0C, 0x0C, 0x0C, 0x1E, 0x00}, // U+0049 (I)
- { 0x78, 0x30, 0x30, 0x30, 0x33, 0x33, 0x1E, 0x00}, // U+004A (J)
- { 0x67, 0x66, 0x36, 0x1E, 0x36, 0x66, 0x67, 0x00}, // U+004B (K)
- { 0x0F, 0x06, 0x06, 0x06, 0x46, 0x66, 0x7F, 0x00}, // U+004C (L)
- { 0x63, 0x77, 0x7F, 0x7F, 0x6B, 0x63, 0x63, 0x00}, // U+004D (M)
- { 0x63, 0x67, 0x6F, 0x7B, 0x73, 0x63, 0x63, 0x00}, // U+004E (N)
- { 0x1C, 0x36, 0x63, 0x63, 0x63, 0x36, 0x1C, 0x00}, // U+004F (O)
- { 0x3F, 0x66, 0x66, 0x3E, 0x06, 0x06, 0x0F, 0x00}, // U+0050 (P)
- { 0x1E, 0x33, 0x33, 0x33, 0x3B, 0x1E, 0x38, 0x00}, // U+0051 (Q)
- { 0x3F, 0x66, 0x66, 0x3E, 0x36, 0x66, 0x67, 0x00}, // U+0052 (R)
- { 0x1E, 0x33, 0x07, 0x0E, 0x38, 0x33, 0x1E, 0x00}, // U+0053 (S)
- { 0x3F, 0x2D, 0x0C, 0x0C, 0x0C, 0x0C, 0x1E, 0x00}, // U+0054 (T)
- { 0x33, 0x33, 0x33, 0x33, 0x33, 0x33, 0x3F, 0x00}, // U+0055 (U)
- { 0x33, 0x33, 0x33, 0x33, 0x33, 0x1E, 0x0C, 0x00}, // U+0056 (V)
- { 0x63, 0x63, 0x63, 0x6B, 0x7F, 0x77, 0x63, 0x00}, // U+0057 (W)
- { 0x63, 0x63, 0x36, 0x1C, 0x1C, 0x36, 0x63, 0x00}, // U+0058 (X)
- { 0x33, 0x33, 0x33, 0x1E, 0x0C, 0x0C, 0x1E, 0x00}, // U+0059 (Y)
- { 0x7F, 0x63, 0x31, 0x18, 0x4C, 0x66, 0x7F, 0x00}, // U+005A (Z)
- { 0x1E, 0x06, 0x06, 0x06, 0x06, 0x06, 0x1E, 0x00}, // U+005B ([)
- { 0x03, 0x06, 0x0C, 0x18, 0x30, 0x60, 0x40, 0x00}, // U+005C (\)
- { 0x1E, 0x18, 0x18, 0x18, 0x18, 0x18, 0x1E, 0x00}, // U+005D (])
- { 0x08, 0x1C, 0x36, 0x63, 0x00, 0x00, 0x00, 0x00}, // U+005E (^)
- { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xFF}, // U+005F (_)
- { 0x0C, 0x0C, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00}, // U+0060 (`)
- { 0x00, 0x00, 0x1E, 0x30, 0x3E, 0x33, 0x6E, 0x00}, // U+0061 (a)
- { 0x07, 0x06, 0x06, 0x3E, 0x66, 0x66, 0x3B, 0x00}, // U+0062 (b)
- { 0x00, 0x00, 0x1E, 0x33, 0x03, 0x33, 0x1E, 0x00}, // U+0063 (c)
- { 0x38, 0x30, 0x30, 0x3e, 0x33, 0x33, 0x6E, 0x00}, // U+0064 (d)
- { 0x00, 0x00, 0x1E, 0x33, 0x3f, 0x03, 0x1E, 0x00}, // U+0065 (e)
- { 0x1C, 0x36, 0x06, 0x0f, 0x06, 0x06, 0x0F, 0x00}, // U+0066 (f)
- { 0x00, 0x00, 0x6E, 0x33, 0x33, 0x3E, 0x30, 0x1F}, // U+0067 (g)
- { 0x07, 0x06, 0x36, 0x6E, 0x66, 0x66, 0x67, 0x00}, // U+0068 (h)
- { 0x0C, 0x00, 0x0E, 0x0C, 0x0C, 0x0C, 0x1E, 0x00}, // U+0069 (i)
- { 0x30, 0x00, 0x30, 0x30, 0x30, 0x33, 0x33, 0x1E}, // U+006A (j)
- { 0x07, 0x06, 0x66, 0x36, 0x1E, 0x36, 0x67, 0x00}, // U+006B (k)
- { 0x0E, 0x0C, 0x0C, 0x0C, 0x0C, 0x0C, 0x1E, 0x00}, // U+006C (l)
- { 0x00, 0x00, 0x33, 0x7F, 0x7F, 0x6B, 0x63, 0x00}, // U+006D (m)
- { 0x00, 0x00, 0x1F, 0x33, 0x33, 0x33, 0x33, 0x00}, // U+006E (n)
- { 0x00, 0x00, 0x1E, 0x33, 0x33, 0x33, 0x1E, 0x00}, // U+006F (o)
- { 0x00, 0x00, 0x3B, 0x66, 0x66, 0x3E, 0x06, 0x0F}, // U+0070 (p)
- { 0x00, 0x00, 0x6E, 0x33, 0x33, 0x3E, 0x30, 0x78}, // U+0071 (q)
- { 0x00, 0x00, 0x3B, 0x6E, 0x66, 0x06, 0x0F, 0x00}, // U+0072 (r)
- { 0x00, 0x00, 0x3E, 0x03, 0x1E, 0x30, 0x1F, 0x00}, // U+0073 (s)
- { 0x08, 0x0C, 0x3E, 0x0C, 0x0C, 0x2C, 0x18, 0x00}, // U+0074 (t)
- { 0x00, 0x00, 0x33, 0x33, 0x33, 0x33, 0x6E, 0x00}, // U+0075 (u)
- { 0x00, 0x00, 0x33, 0x33, 0x33, 0x1E, 0x0C, 0x00}, // U+0076 (v)
- { 0x00, 0x00, 0x63, 0x6B, 0x7F, 0x7F, 0x36, 0x00}, // U+0077 (w)
- { 0x00, 0x00, 0x63, 0x36, 0x1C, 0x36, 0x63, 0x00}, // U+0078 (x)
- { 0x00, 0x00, 0x33, 0x33, 0x33, 0x3E, 0x30, 0x1F}, // U+0079 (y)
- { 0x00, 0x00, 0x3F, 0x19, 0x0C, 0x26, 0x3F, 0x00}, // U+007A (z)
- { 0x38, 0x0C, 0x0C, 0x07, 0x0C, 0x0C, 0x38, 0x00}, // U+007B ({)
- { 0x18, 0x18, 0x18, 0x00, 0x18, 0x18, 0x18, 0x00}, // U+007C (|)
- { 0x07, 0x0C, 0x0C, 0x38, 0x0C, 0x0C, 0x07, 0x00}, // U+007D (})
- { 0x6E, 0x3B, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}, // U+007E (~)
- { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00} // U+007F
- };
- bool g_library_initialized;
- std::mutex g_encoder_init_mutex;
- // Encoder library initialization (just call once at startup)
- void basisu_encoder_init(bool use_opencl, bool opencl_force_serialization)
- {
- std::lock_guard<std::mutex> lock(g_encoder_init_mutex);
- if (g_library_initialized)
- return;
- detect_sse41();
- basist::basisu_transcoder_init();
- pack_etc1_solid_color_init();
- //uastc_init();
- bc7enc_compress_block_init(); // must be after uastc_init()
- // Don't bother initializing the OpenCL module at all if it's been completely disabled.
- if (use_opencl)
- {
- opencl_init(opencl_force_serialization);
- }
- g_library_initialized = true;
- }
- void basisu_encoder_deinit()
- {
- opencl_deinit();
- g_library_initialized = false;
- }
- void error_vprintf(const char* pFmt, va_list args)
- {
- char buf[8192];
- #ifdef _WIN32
- vsprintf_s(buf, sizeof(buf), pFmt, args);
- #else
- vsnprintf(buf, sizeof(buf), pFmt, args);
- #endif
- fprintf(stderr, "ERROR: %s", buf);
- }
- void error_printf(const char *pFmt, ...)
- {
- va_list args;
- va_start(args, pFmt);
- error_vprintf(pFmt, args);
- va_end(args);
- }
- #if defined(_WIN32)
- inline void query_counter(timer_ticks* pTicks)
- {
- QueryPerformanceCounter(reinterpret_cast<LARGE_INTEGER*>(pTicks));
- }
- inline void query_counter_frequency(timer_ticks* pTicks)
- {
- QueryPerformanceFrequency(reinterpret_cast<LARGE_INTEGER*>(pTicks));
- }
- #elif defined(__APPLE__) || defined(__FreeBSD__) || defined(__OpenBSD__)
- #include <sys/time.h>
- inline void query_counter(timer_ticks* pTicks)
- {
- struct timeval cur_time;
- gettimeofday(&cur_time, NULL);
- *pTicks = static_cast<unsigned long long>(cur_time.tv_sec) * 1000000ULL + static_cast<unsigned long long>(cur_time.tv_usec);
- }
- inline void query_counter_frequency(timer_ticks* pTicks)
- {
- *pTicks = 1000000;
- }
- #elif defined(__GNUC__)
- #include <sys/timex.h>
- inline void query_counter(timer_ticks* pTicks)
- {
- struct timeval cur_time;
- gettimeofday(&cur_time, NULL);
- *pTicks = static_cast<unsigned long long>(cur_time.tv_sec) * 1000000ULL + static_cast<unsigned long long>(cur_time.tv_usec);
- }
- inline void query_counter_frequency(timer_ticks* pTicks)
- {
- *pTicks = 1000000;
- }
- #else
- #error TODO
- #endif
-
- interval_timer::interval_timer() : m_start_time(0), m_stop_time(0), m_started(false), m_stopped(false)
- {
- if (!g_timer_freq)
- init();
- }
- void interval_timer::start()
- {
- query_counter(&m_start_time);
- m_started = true;
- m_stopped = false;
- }
- void interval_timer::stop()
- {
- assert(m_started);
- query_counter(&m_stop_time);
- m_stopped = true;
- }
- double interval_timer::get_elapsed_secs() const
- {
- assert(m_started);
- if (!m_started)
- return 0;
- timer_ticks stop_time = m_stop_time;
- if (!m_stopped)
- query_counter(&stop_time);
- timer_ticks delta = stop_time - m_start_time;
- return delta * g_timer_freq;
- }
-
- void interval_timer::init()
- {
- if (!g_timer_freq)
- {
- query_counter_frequency(&g_freq);
- g_timer_freq = 1.0f / g_freq;
- query_counter(&g_init_ticks);
- }
- }
- timer_ticks interval_timer::get_ticks()
- {
- if (!g_timer_freq)
- init();
- timer_ticks ticks;
- query_counter(&ticks);
- return ticks - g_init_ticks;
- }
- double interval_timer::ticks_to_secs(timer_ticks ticks)
- {
- if (!g_timer_freq)
- init();
- return ticks * g_timer_freq;
- }
-
- const uint32_t MAX_32BIT_ALLOC_SIZE = 250000000;
-
- bool load_tga(const char* pFilename, image& img)
- {
- int w = 0, h = 0, n_chans = 0;
- uint8_t* pImage_data = read_tga(pFilename, w, h, n_chans);
-
- if ((!pImage_data) || (!w) || (!h) || ((n_chans != 3) && (n_chans != 4)))
- {
- error_printf("Failed loading .TGA image \"%s\"!\n", pFilename);
- if (pImage_data)
- free(pImage_data);
-
- return false;
- }
- if (sizeof(void *) == sizeof(uint32_t))
- {
- if ((w * h * n_chans) > MAX_32BIT_ALLOC_SIZE)
- {
- error_printf("Image \"%s\" is too large (%ux%u) to process in a 32-bit build!\n", pFilename, w, h);
- if (pImage_data)
- free(pImage_data);
- return false;
- }
- }
-
- img.resize(w, h);
- const uint8_t *pSrc = pImage_data;
- for (int y = 0; y < h; y++)
- {
- color_rgba *pDst = &img(0, y);
- for (int x = 0; x < w; x++)
- {
- pDst->r = pSrc[0];
- pDst->g = pSrc[1];
- pDst->b = pSrc[2];
- pDst->a = (n_chans == 3) ? 255 : pSrc[3];
- pSrc += n_chans;
- ++pDst;
- }
- }
- free(pImage_data);
- return true;
- }
- bool load_png(const uint8_t *pBuf, size_t buf_size, image &img, const char *pFilename)
- {
- interval_timer tm;
- tm.start();
-
- if (!buf_size)
- return false;
- uint32_t width = 0, height = 0, num_chans = 0;
- void* pImage = pv_png::load_png(pBuf, buf_size, 4, width, height, num_chans);
- if (!pBuf)
- {
- error_printf("pv_png::load_png failed while loading image \"%s\"\n", pFilename);
- return false;
- }
- img.grant_ownership(reinterpret_cast<color_rgba*>(pImage), width, height);
- //debug_printf("Total load_png() time: %3.3f secs\n", tm.get_elapsed_secs());
- return true;
- }
-
- bool load_png(const char* pFilename, image& img)
- {
- uint8_vec buffer;
- if (!read_file_to_vec(pFilename, buffer))
- {
- error_printf("load_png: Failed reading file \"%s\"!\n", pFilename);
- return false;
- }
- return load_png(buffer.data(), buffer.size(), img, pFilename);
- }
- bool load_jpg(const char *pFilename, image& img)
- {
- int width = 0, height = 0, actual_comps = 0;
- uint8_t *pImage_data = jpgd::decompress_jpeg_image_from_file(pFilename, &width, &height, &actual_comps, 4, jpgd::jpeg_decoder::cFlagLinearChromaFiltering);
- if (!pImage_data)
- return false;
-
- img.init(pImage_data, width, height, 4);
-
- free(pImage_data);
- return true;
- }
- bool load_image(const char* pFilename, image& img)
- {
- std::string ext(string_get_extension(std::string(pFilename)));
- if (ext.length() == 0)
- return false;
- const char *pExt = ext.c_str();
- if (strcasecmp(pExt, "png") == 0)
- return load_png(pFilename, img);
- if (strcasecmp(pExt, "tga") == 0)
- return load_tga(pFilename, img);
- if ( (strcasecmp(pExt, "jpg") == 0) || (strcasecmp(pExt, "jfif") == 0) || (strcasecmp(pExt, "jpeg") == 0) )
- return load_jpg(pFilename, img);
- return false;
- }
-
- bool save_png(const char* pFilename, const image &img, uint32_t image_save_flags, uint32_t grayscale_comp)
- {
- if (!img.get_total_pixels())
- return false;
-
- void* pPNG_data = nullptr;
- size_t PNG_data_size = 0;
- if (image_save_flags & cImageSaveGrayscale)
- {
- uint8_vec g_pixels(img.get_total_pixels());
- uint8_t* pDst = &g_pixels[0];
- for (uint32_t y = 0; y < img.get_height(); y++)
- for (uint32_t x = 0; x < img.get_width(); x++)
- *pDst++ = img(x, y)[grayscale_comp];
- pPNG_data = buminiz::tdefl_write_image_to_png_file_in_memory_ex(g_pixels.data(), img.get_width(), img.get_height(), 1, &PNG_data_size, 1, false);
- }
- else
- {
- bool has_alpha = false;
-
- if ((image_save_flags & cImageSaveIgnoreAlpha) == 0)
- has_alpha = img.has_alpha();
- if (!has_alpha)
- {
- uint8_vec rgb_pixels(img.get_total_pixels() * 3);
- uint8_t* pDst = &rgb_pixels[0];
- for (uint32_t y = 0; y < img.get_height(); y++)
- {
- const color_rgba* pSrc = &img(0, y);
- for (uint32_t x = 0; x < img.get_width(); x++)
- {
- pDst[0] = pSrc->r;
- pDst[1] = pSrc->g;
- pDst[2] = pSrc->b;
-
- pSrc++;
- pDst += 3;
- }
- }
- pPNG_data = buminiz::tdefl_write_image_to_png_file_in_memory_ex(rgb_pixels.data(), img.get_width(), img.get_height(), 3, &PNG_data_size, 1, false);
- }
- else
- {
- pPNG_data = buminiz::tdefl_write_image_to_png_file_in_memory_ex(img.get_ptr(), img.get_width(), img.get_height(), 4, &PNG_data_size, 1, false);
- }
- }
- if (!pPNG_data)
- return false;
- bool status = write_data_to_file(pFilename, pPNG_data, PNG_data_size);
- if (!status)
- {
- error_printf("save_png: Failed writing to filename \"%s\"!\n", pFilename);
- }
- free(pPNG_data);
-
- return status;
- }
-
- bool read_file_to_vec(const char* pFilename, uint8_vec& data)
- {
- FILE* pFile = nullptr;
- #ifdef _WIN32
- fopen_s(&pFile, pFilename, "rb");
- #else
- pFile = fopen(pFilename, "rb");
- #endif
- if (!pFile)
- return false;
-
- fseek(pFile, 0, SEEK_END);
- #ifdef _WIN32
- int64_t filesize = _ftelli64(pFile);
- #else
- int64_t filesize = ftello(pFile);
- #endif
- if (filesize < 0)
- {
- fclose(pFile);
- return false;
- }
- fseek(pFile, 0, SEEK_SET);
- if (sizeof(size_t) == sizeof(uint32_t))
- {
- if (filesize > 0x70000000)
- {
- // File might be too big to load safely in one alloc
- fclose(pFile);
- return false;
- }
- }
- if (!data.try_resize((size_t)filesize))
- {
- fclose(pFile);
- return false;
- }
- if (filesize)
- {
- if (fread(&data[0], 1, (size_t)filesize, pFile) != (size_t)filesize)
- {
- fclose(pFile);
- return false;
- }
- }
- fclose(pFile);
- return true;
- }
- bool write_data_to_file(const char* pFilename, const void* pData, size_t len)
- {
- FILE* pFile = nullptr;
- #ifdef _WIN32
- fopen_s(&pFile, pFilename, "wb");
- #else
- pFile = fopen(pFilename, "wb");
- #endif
- if (!pFile)
- return false;
- if (len)
- {
- if (fwrite(pData, 1, len, pFile) != len)
- {
- fclose(pFile);
- return false;
- }
- }
- return fclose(pFile) != EOF;
- }
- float linear_to_srgb(float l)
- {
- assert(l >= 0.0f && l <= 1.0f);
- if (l < .0031308f)
- return saturate(l * 12.92f);
- else
- return saturate(1.055f * powf(l, 1.0f/2.4f) - .055f);
- }
- float srgb_to_linear(float s)
- {
- assert(s >= 0.0f && s <= 1.0f);
- if (s < .04045f)
- return saturate(s * (1.0f/12.92f));
- else
- return saturate(powf((s + .055f) * (1.0f/1.055f), 2.4f));
- }
- bool image_resample(const image &src, image &dst, bool srgb,
- const char *pFilter, float filter_scale,
- bool wrapping,
- uint32_t first_comp, uint32_t num_comps)
- {
- assert((first_comp + num_comps) <= 4);
- const int cMaxComps = 4;
-
- const uint32_t src_w = src.get_width(), src_h = src.get_height();
- const uint32_t dst_w = dst.get_width(), dst_h = dst.get_height();
-
- if (maximum(src_w, src_h) > BASISU_RESAMPLER_MAX_DIMENSION)
- {
- printf("Image is too large!\n");
- return false;
- }
- if (!src_w || !src_h || !dst_w || !dst_h)
- return false;
-
- if ((num_comps < 1) || (num_comps > cMaxComps))
- return false;
-
- if ((minimum(dst_w, dst_h) < 1) || (maximum(dst_w, dst_h) > BASISU_RESAMPLER_MAX_DIMENSION))
- {
- printf("Image is too large!\n");
- return false;
- }
- if ((src_w == dst_w) && (src_h == dst_h))
- {
- dst = src;
- return true;
- }
- float srgb_to_linear_table[256];
- if (srgb)
- {
- for (int i = 0; i < 256; ++i)
- srgb_to_linear_table[i] = srgb_to_linear((float)i * (1.0f/255.0f));
- }
- const int LINEAR_TO_SRGB_TABLE_SIZE = 8192;
- uint8_t linear_to_srgb_table[LINEAR_TO_SRGB_TABLE_SIZE];
- if (srgb)
- {
- for (int i = 0; i < LINEAR_TO_SRGB_TABLE_SIZE; ++i)
- linear_to_srgb_table[i] = (uint8_t)clamp<int>((int)(255.0f * linear_to_srgb((float)i * (1.0f / (LINEAR_TO_SRGB_TABLE_SIZE - 1))) + .5f), 0, 255);
- }
- std::vector<float> samples[cMaxComps];
- Resampler *resamplers[cMaxComps];
-
- resamplers[0] = new Resampler(src_w, src_h, dst_w, dst_h,
- wrapping ? Resampler::BOUNDARY_WRAP : Resampler::BOUNDARY_CLAMP, 0.0f, 1.0f,
- pFilter, nullptr, nullptr, filter_scale, filter_scale, 0, 0);
- samples[0].resize(src_w);
- for (uint32_t i = 1; i < num_comps; ++i)
- {
- resamplers[i] = new Resampler(src_w, src_h, dst_w, dst_h,
- wrapping ? Resampler::BOUNDARY_WRAP : Resampler::BOUNDARY_CLAMP, 0.0f, 1.0f,
- pFilter, resamplers[0]->get_clist_x(), resamplers[0]->get_clist_y(), filter_scale, filter_scale, 0, 0);
- samples[i].resize(src_w);
- }
- uint32_t dst_y = 0;
- for (uint32_t src_y = 0; src_y < src_h; ++src_y)
- {
- const color_rgba *pSrc = &src(0, src_y);
- // Put source lines into resampler(s)
- for (uint32_t x = 0; x < src_w; ++x)
- {
- for (uint32_t c = 0; c < num_comps; ++c)
- {
- const uint32_t comp_index = first_comp + c;
- const uint32_t v = (*pSrc)[comp_index];
- if (!srgb || (comp_index == 3))
- samples[c][x] = v * (1.0f / 255.0f);
- else
- samples[c][x] = srgb_to_linear_table[v];
- }
- pSrc++;
- }
- for (uint32_t c = 0; c < num_comps; ++c)
- {
- if (!resamplers[c]->put_line(&samples[c][0]))
- {
- for (uint32_t i = 0; i < num_comps; i++)
- delete resamplers[i];
- return false;
- }
- }
- // Now retrieve any output lines
- for (;;)
- {
- uint32_t c;
- for (c = 0; c < num_comps; ++c)
- {
- const uint32_t comp_index = first_comp + c;
- const float *pOutput_samples = resamplers[c]->get_line();
- if (!pOutput_samples)
- break;
- const bool linear_flag = !srgb || (comp_index == 3);
-
- color_rgba *pDst = &dst(0, dst_y);
- for (uint32_t x = 0; x < dst_w; x++)
- {
- // TODO: Add dithering
- if (linear_flag)
- {
- int j = (int)(255.0f * pOutput_samples[x] + .5f);
- (*pDst)[comp_index] = (uint8_t)clamp<int>(j, 0, 255);
- }
- else
- {
- int j = (int)((LINEAR_TO_SRGB_TABLE_SIZE - 1) * pOutput_samples[x] + .5f);
- (*pDst)[comp_index] = linear_to_srgb_table[clamp<int>(j, 0, LINEAR_TO_SRGB_TABLE_SIZE - 1)];
- }
- pDst++;
- }
- }
- if (c < num_comps)
- break;
- ++dst_y;
- }
- }
- for (uint32_t i = 0; i < num_comps; ++i)
- delete resamplers[i];
- return true;
- }
- void canonical_huffman_calculate_minimum_redundancy(sym_freq *A, int num_syms)
- {
- // See the paper "In-Place Calculation of Minimum Redundancy Codes" by Moffat and Katajainen
- if (!num_syms)
- return;
- if (1 == num_syms)
- {
- A[0].m_key = 1;
- return;
- }
-
- A[0].m_key += A[1].m_key;
-
- int s = 2, r = 0, next;
- for (next = 1; next < (num_syms - 1); ++next)
- {
- if ((s >= num_syms) || (A[r].m_key < A[s].m_key))
- {
- A[next].m_key = A[r].m_key;
- A[r].m_key = next;
- ++r;
- }
- else
- {
- A[next].m_key = A[s].m_key;
- ++s;
- }
- if ((s >= num_syms) || ((r < next) && A[r].m_key < A[s].m_key))
- {
- A[next].m_key = A[next].m_key + A[r].m_key;
- A[r].m_key = next;
- ++r;
- }
- else
- {
- A[next].m_key = A[next].m_key + A[s].m_key;
- ++s;
- }
- }
- A[num_syms - 2].m_key = 0;
- for (next = num_syms - 3; next >= 0; --next)
- {
- A[next].m_key = 1 + A[A[next].m_key].m_key;
- }
- int num_avail = 1, num_used = 0, depth = 0;
- r = num_syms - 2;
- next = num_syms - 1;
- while (num_avail > 0)
- {
- for ( ; (r >= 0) && ((int)A[r].m_key == depth); ++num_used, --r )
- ;
- for ( ; num_avail > num_used; --next, --num_avail)
- A[next].m_key = depth;
- num_avail = 2 * num_used;
- num_used = 0;
- ++depth;
- }
- }
- void canonical_huffman_enforce_max_code_size(int *pNum_codes, int code_list_len, int max_code_size)
- {
- int i;
- uint32_t total = 0;
- if (code_list_len <= 1)
- return;
- for (i = max_code_size + 1; i <= cHuffmanMaxSupportedInternalCodeSize; i++)
- pNum_codes[max_code_size] += pNum_codes[i];
- for (i = max_code_size; i > 0; i--)
- total += (((uint32_t)pNum_codes[i]) << (max_code_size - i));
- while (total != (1UL << max_code_size))
- {
- pNum_codes[max_code_size]--;
- for (i = max_code_size - 1; i > 0; i--)
- {
- if (pNum_codes[i])
- {
- pNum_codes[i]--;
- pNum_codes[i + 1] += 2;
- break;
- }
- }
- total--;
- }
- }
- sym_freq *canonical_huffman_radix_sort_syms(uint32_t num_syms, sym_freq *pSyms0, sym_freq *pSyms1)
- {
- uint32_t total_passes = 2, pass_shift, pass, i, hist[256 * 2];
- sym_freq *pCur_syms = pSyms0, *pNew_syms = pSyms1;
- clear_obj(hist);
- for (i = 0; i < num_syms; i++)
- {
- uint32_t freq = pSyms0[i].m_key;
-
- // We scale all input frequencies to 16-bits.
- assert(freq <= UINT16_MAX);
- hist[freq & 0xFF]++;
- hist[256 + ((freq >> 8) & 0xFF)]++;
- }
- while ((total_passes > 1) && (num_syms == hist[(total_passes - 1) * 256]))
- total_passes--;
- for (pass_shift = 0, pass = 0; pass < total_passes; pass++, pass_shift += 8)
- {
- const uint32_t *pHist = &hist[pass << 8];
- uint32_t offsets[256], cur_ofs = 0;
- for (i = 0; i < 256; i++)
- {
- offsets[i] = cur_ofs;
- cur_ofs += pHist[i];
- }
- for (i = 0; i < num_syms; i++)
- pNew_syms[offsets[(pCur_syms[i].m_key >> pass_shift) & 0xFF]++] = pCur_syms[i];
- sym_freq *t = pCur_syms;
- pCur_syms = pNew_syms;
- pNew_syms = t;
- }
- return pCur_syms;
- }
- bool huffman_encoding_table::init(uint32_t num_syms, const uint16_t *pFreq, uint32_t max_code_size)
- {
- if (max_code_size > cHuffmanMaxSupportedCodeSize)
- return false;
- if ((!num_syms) || (num_syms > cHuffmanMaxSyms))
- return false;
- uint32_t total_used_syms = 0;
- for (uint32_t i = 0; i < num_syms; i++)
- if (pFreq[i])
- total_used_syms++;
- if (!total_used_syms)
- return false;
- std::vector<sym_freq> sym_freq0(total_used_syms), sym_freq1(total_used_syms);
- for (uint32_t i = 0, j = 0; i < num_syms; i++)
- {
- if (pFreq[i])
- {
- sym_freq0[j].m_key = pFreq[i];
- sym_freq0[j++].m_sym_index = static_cast<uint16_t>(i);
- }
- }
- sym_freq *pSym_freq = canonical_huffman_radix_sort_syms(total_used_syms, &sym_freq0[0], &sym_freq1[0]);
- canonical_huffman_calculate_minimum_redundancy(pSym_freq, total_used_syms);
- int num_codes[cHuffmanMaxSupportedInternalCodeSize + 1];
- clear_obj(num_codes);
- for (uint32_t i = 0; i < total_used_syms; i++)
- {
- if (pSym_freq[i].m_key > cHuffmanMaxSupportedInternalCodeSize)
- return false;
- num_codes[pSym_freq[i].m_key]++;
- }
- canonical_huffman_enforce_max_code_size(num_codes, total_used_syms, max_code_size);
- m_code_sizes.resize(0);
- m_code_sizes.resize(num_syms);
- m_codes.resize(0);
- m_codes.resize(num_syms);
- for (uint32_t i = 1, j = total_used_syms; i <= max_code_size; i++)
- for (uint32_t l = num_codes[i]; l > 0; l--)
- m_code_sizes[pSym_freq[--j].m_sym_index] = static_cast<uint8_t>(i);
- uint32_t next_code[cHuffmanMaxSupportedInternalCodeSize + 1];
- next_code[1] = 0;
- for (uint32_t j = 0, i = 2; i <= max_code_size; i++)
- next_code[i] = j = ((j + num_codes[i - 1]) << 1);
- for (uint32_t i = 0; i < num_syms; i++)
- {
- uint32_t rev_code = 0, code, code_size;
- if ((code_size = m_code_sizes[i]) == 0)
- continue;
- if (code_size > cHuffmanMaxSupportedInternalCodeSize)
- return false;
- code = next_code[code_size]++;
- for (uint32_t l = code_size; l > 0; l--, code >>= 1)
- rev_code = (rev_code << 1) | (code & 1);
- m_codes[i] = static_cast<uint16_t>(rev_code);
- }
- return true;
- }
- bool huffman_encoding_table::init(uint32_t num_syms, const uint32_t *pSym_freq, uint32_t max_code_size)
- {
- if ((!num_syms) || (num_syms > cHuffmanMaxSyms))
- return false;
- uint16_vec sym_freq(num_syms);
- uint32_t max_freq = 0;
- for (uint32_t i = 0; i < num_syms; i++)
- max_freq = maximum(max_freq, pSym_freq[i]);
- if (max_freq < UINT16_MAX)
- {
- for (uint32_t i = 0; i < num_syms; i++)
- sym_freq[i] = static_cast<uint16_t>(pSym_freq[i]);
- }
- else
- {
- for (uint32_t i = 0; i < num_syms; i++)
- {
- if (pSym_freq[i])
- {
- uint32_t f = static_cast<uint32_t>((static_cast<uint64_t>(pSym_freq[i]) * 65534U + (max_freq >> 1)) / max_freq);
- sym_freq[i] = static_cast<uint16_t>(clamp<uint32_t>(f, 1, 65534));
- }
- }
- }
- return init(num_syms, &sym_freq[0], max_code_size);
- }
- void bitwise_coder::end_nonzero_run(uint16_vec &syms, uint32_t &run_size, uint32_t len)
- {
- if (run_size)
- {
- if (run_size < cHuffmanSmallRepeatSizeMin)
- {
- while (run_size--)
- syms.push_back(static_cast<uint16_t>(len));
- }
- else if (run_size <= cHuffmanSmallRepeatSizeMax)
- {
- syms.push_back(static_cast<uint16_t>(cHuffmanSmallRepeatCode | ((run_size - cHuffmanSmallRepeatSizeMin) << 6)));
- }
- else
- {
- assert((run_size >= cHuffmanBigRepeatSizeMin) && (run_size <= cHuffmanBigRepeatSizeMax));
- syms.push_back(static_cast<uint16_t>(cHuffmanBigRepeatCode | ((run_size - cHuffmanBigRepeatSizeMin) << 6)));
- }
- }
- run_size = 0;
- }
- void bitwise_coder::end_zero_run(uint16_vec &syms, uint32_t &run_size)
- {
- if (run_size)
- {
- if (run_size < cHuffmanSmallZeroRunSizeMin)
- {
- while (run_size--)
- syms.push_back(0);
- }
- else if (run_size <= cHuffmanSmallZeroRunSizeMax)
- {
- syms.push_back(static_cast<uint16_t>(cHuffmanSmallZeroRunCode | ((run_size - cHuffmanSmallZeroRunSizeMin) << 6)));
- }
- else
- {
- assert((run_size >= cHuffmanBigZeroRunSizeMin) && (run_size <= cHuffmanBigZeroRunSizeMax));
- syms.push_back(static_cast<uint16_t>(cHuffmanBigZeroRunCode | ((run_size - cHuffmanBigZeroRunSizeMin) << 6)));
- }
- }
- run_size = 0;
- }
- uint32_t bitwise_coder::emit_huffman_table(const huffman_encoding_table &tab)
- {
- const uint64_t start_bits = m_total_bits;
- const uint8_vec &code_sizes = tab.get_code_sizes();
- uint32_t total_used = tab.get_total_used_codes();
- put_bits(total_used, cHuffmanMaxSymsLog2);
-
- if (!total_used)
- return 0;
- uint16_vec syms;
- syms.reserve(total_used + 16);
- uint32_t prev_code_len = UINT_MAX, zero_run_size = 0, nonzero_run_size = 0;
- for (uint32_t i = 0; i <= total_used; ++i)
- {
- const uint32_t code_len = (i == total_used) ? 0xFF : code_sizes[i];
- assert((code_len == 0xFF) || (code_len <= 16));
- if (code_len)
- {
- end_zero_run(syms, zero_run_size);
- if (code_len != prev_code_len)
- {
- end_nonzero_run(syms, nonzero_run_size, prev_code_len);
- if (code_len != 0xFF)
- syms.push_back(static_cast<uint16_t>(code_len));
- }
- else if (++nonzero_run_size == cHuffmanBigRepeatSizeMax)
- end_nonzero_run(syms, nonzero_run_size, prev_code_len);
- }
- else
- {
- end_nonzero_run(syms, nonzero_run_size, prev_code_len);
- if (++zero_run_size == cHuffmanBigZeroRunSizeMax)
- end_zero_run(syms, zero_run_size);
- }
- prev_code_len = code_len;
- }
- histogram h(cHuffmanTotalCodelengthCodes);
- for (uint32_t i = 0; i < syms.size(); i++)
- h.inc(syms[i] & 63);
- huffman_encoding_table ct;
- if (!ct.init(h, 7))
- return 0;
- assert(cHuffmanTotalSortedCodelengthCodes == cHuffmanTotalCodelengthCodes);
- uint32_t total_codelength_codes;
- for (total_codelength_codes = cHuffmanTotalSortedCodelengthCodes; total_codelength_codes > 0; total_codelength_codes--)
- if (ct.get_code_sizes()[g_huffman_sorted_codelength_codes[total_codelength_codes - 1]])
- break;
- assert(total_codelength_codes);
- put_bits(total_codelength_codes, 5);
- for (uint32_t i = 0; i < total_codelength_codes; i++)
- put_bits(ct.get_code_sizes()[g_huffman_sorted_codelength_codes[i]], 3);
- for (uint32_t i = 0; i < syms.size(); ++i)
- {
- const uint32_t l = syms[i] & 63, e = syms[i] >> 6;
- put_code(l, ct);
-
- if (l == cHuffmanSmallZeroRunCode)
- put_bits(e, cHuffmanSmallZeroRunExtraBits);
- else if (l == cHuffmanBigZeroRunCode)
- put_bits(e, cHuffmanBigZeroRunExtraBits);
- else if (l == cHuffmanSmallRepeatCode)
- put_bits(e, cHuffmanSmallRepeatExtraBits);
- else if (l == cHuffmanBigRepeatCode)
- put_bits(e, cHuffmanBigRepeatExtraBits);
- }
- return (uint32_t)(m_total_bits - start_bits);
- }
- bool huffman_test(int rand_seed)
- {
- histogram h(19);
- // Feed in a fibonacci sequence to force large codesizes
- h[0] += 1; h[1] += 1; h[2] += 2; h[3] += 3;
- h[4] += 5; h[5] += 8; h[6] += 13; h[7] += 21;
- h[8] += 34; h[9] += 55; h[10] += 89; h[11] += 144;
- h[12] += 233; h[13] += 377; h[14] += 610; h[15] += 987;
- h[16] += 1597; h[17] += 2584; h[18] += 4181;
- huffman_encoding_table etab;
- etab.init(h, 16);
-
- {
- bitwise_coder c;
- c.init(1024);
- c.emit_huffman_table(etab);
- for (int i = 0; i < 19; i++)
- c.put_code(i, etab);
- c.flush();
- basist::bitwise_decoder d;
- d.init(&c.get_bytes()[0], static_cast<uint32_t>(c.get_bytes().size()));
- basist::huffman_decoding_table dtab;
- bool success = d.read_huffman_table(dtab);
- if (!success)
- {
- assert(0);
- printf("Failure 5\n");
- return false;
- }
- for (uint32_t i = 0; i < 19; i++)
- {
- uint32_t s = d.decode_huffman(dtab);
- if (s != i)
- {
- assert(0);
- printf("Failure 5\n");
- return false;
- }
- }
- }
- basisu::rand r;
- r.seed(rand_seed);
- for (int iter = 0; iter < 500000; iter++)
- {
- printf("%u\n", iter);
- uint32_t max_sym = r.irand(0, 8193);
- uint32_t num_codes = r.irand(1, 10000);
- uint_vec syms(num_codes);
- for (uint32_t i = 0; i < num_codes; i++)
- {
- if (r.bit())
- syms[i] = r.irand(0, max_sym);
- else
- {
- int s = (int)(r.gaussian((float)max_sym / 2, (float)maximum<int>(1, max_sym / 2)) + .5f);
- s = basisu::clamp<int>(s, 0, max_sym);
- syms[i] = s;
- }
- }
- histogram h1(max_sym + 1);
- for (uint32_t i = 0; i < num_codes; i++)
- h1[syms[i]]++;
- huffman_encoding_table etab2;
- if (!etab2.init(h1, 16))
- {
- assert(0);
- printf("Failed 0\n");
- return false;
- }
- bitwise_coder c;
- c.init(1024);
- c.emit_huffman_table(etab2);
- for (uint32_t i = 0; i < num_codes; i++)
- c.put_code(syms[i], etab2);
- c.flush();
- basist::bitwise_decoder d;
- d.init(&c.get_bytes()[0], (uint32_t)c.get_bytes().size());
- basist::huffman_decoding_table dtab;
- bool success = d.read_huffman_table(dtab);
- if (!success)
- {
- assert(0);
- printf("Failed 2\n");
- return false;
- }
- for (uint32_t i = 0; i < num_codes; i++)
- {
- uint32_t s = d.decode_huffman(dtab);
- if (s != syms[i])
- {
- assert(0);
- printf("Failed 4\n");
- return false;
- }
- }
- }
- return true;
- }
- void palette_index_reorderer::init(uint32_t num_indices, const uint32_t *pIndices, uint32_t num_syms, pEntry_dist_func pDist_func, void *pCtx, float dist_func_weight)
- {
- assert((num_syms > 0) && (num_indices > 0));
- assert((dist_func_weight >= 0.0f) && (dist_func_weight <= 1.0f));
- clear();
- m_remap_table.resize(num_syms);
- m_entries_picked.reserve(num_syms);
- m_total_count_to_picked.resize(num_syms);
- if (num_indices <= 1)
- return;
- prepare_hist(num_syms, num_indices, pIndices);
- find_initial(num_syms);
- while (m_entries_to_do.size())
- {
- // Find the best entry to move into the picked list.
- uint32_t best_entry;
- double best_count;
- find_next_entry(best_entry, best_count, pDist_func, pCtx, dist_func_weight);
- // We now have chosen an entry to place in the picked list, now determine which side it goes on.
- const uint32_t entry_to_move = m_entries_to_do[best_entry];
-
- float side = pick_side(num_syms, entry_to_move, pDist_func, pCtx, dist_func_weight);
-
- // Put entry_to_move either on the "left" or "right" side of the picked entries
- if (side <= 0)
- m_entries_picked.push_back(entry_to_move);
- else
- m_entries_picked.insert(m_entries_picked.begin(), entry_to_move);
- // Erase best_entry from the todo list
- m_entries_to_do.erase(m_entries_to_do.begin() + best_entry);
- // We've just moved best_entry to the picked list, so now we need to update m_total_count_to_picked[] to factor the additional count to best_entry
- for (uint32_t i = 0; i < m_entries_to_do.size(); i++)
- m_total_count_to_picked[m_entries_to_do[i]] += get_hist(m_entries_to_do[i], entry_to_move, num_syms);
- }
- for (uint32_t i = 0; i < num_syms; i++)
- m_remap_table[m_entries_picked[i]] = i;
- }
- void palette_index_reorderer::prepare_hist(uint32_t num_syms, uint32_t num_indices, const uint32_t *pIndices)
- {
- m_hist.resize(0);
- m_hist.resize(num_syms * num_syms);
- for (uint32_t i = 0; i < num_indices; i++)
- {
- const uint32_t idx = pIndices[i];
- inc_hist(idx, (i < (num_indices - 1)) ? pIndices[i + 1] : -1, num_syms);
- inc_hist(idx, (i > 0) ? pIndices[i - 1] : -1, num_syms);
- }
- }
- void palette_index_reorderer::find_initial(uint32_t num_syms)
- {
- uint32_t max_count = 0, max_index = 0;
- for (uint32_t i = 0; i < num_syms * num_syms; i++)
- if (m_hist[i] > max_count)
- max_count = m_hist[i], max_index = i;
- uint32_t a = max_index / num_syms, b = max_index % num_syms;
- m_entries_picked.push_back(a);
- m_entries_picked.push_back(b);
- for (uint32_t i = 0; i < num_syms; i++)
- if ((i != b) && (i != a))
- m_entries_to_do.push_back(i);
- for (uint32_t i = 0; i < m_entries_to_do.size(); i++)
- for (uint32_t j = 0; j < m_entries_picked.size(); j++)
- m_total_count_to_picked[m_entries_to_do[i]] += get_hist(m_entries_to_do[i], m_entries_picked[j], num_syms);
- }
- void palette_index_reorderer::find_next_entry(uint32_t &best_entry, double &best_count, pEntry_dist_func pDist_func, void *pCtx, float dist_func_weight)
- {
- best_entry = 0;
- best_count = 0;
- for (uint32_t i = 0; i < m_entries_to_do.size(); i++)
- {
- const uint32_t u = m_entries_to_do[i];
- double total_count = m_total_count_to_picked[u];
- if (pDist_func)
- {
- float w = maximum<float>((*pDist_func)(u, m_entries_picked.front(), pCtx), (*pDist_func)(u, m_entries_picked.back(), pCtx));
- assert((w >= 0.0f) && (w <= 1.0f));
- total_count = (total_count + 1.0f) * lerp(1.0f - dist_func_weight, 1.0f + dist_func_weight, w);
- }
- if (total_count <= best_count)
- continue;
- best_entry = i;
- best_count = total_count;
- }
- }
- float palette_index_reorderer::pick_side(uint32_t num_syms, uint32_t entry_to_move, pEntry_dist_func pDist_func, void *pCtx, float dist_func_weight)
- {
- float which_side = 0;
- int l_count = 0, r_count = 0;
- for (uint32_t j = 0; j < m_entries_picked.size(); j++)
- {
- const int count = get_hist(entry_to_move, m_entries_picked[j], num_syms), r = ((int)m_entries_picked.size() + 1 - 2 * (j + 1));
- which_side += static_cast<float>(r * count);
- if (r >= 0)
- l_count += r * count;
- else
- r_count += -r * count;
- }
- if (pDist_func)
- {
- float w_left = lerp(1.0f - dist_func_weight, 1.0f + dist_func_weight, (*pDist_func)(entry_to_move, m_entries_picked.front(), pCtx));
- float w_right = lerp(1.0f - dist_func_weight, 1.0f + dist_func_weight, (*pDist_func)(entry_to_move, m_entries_picked.back(), pCtx));
- which_side = w_left * l_count - w_right * r_count;
- }
- return which_side;
- }
- void image_metrics::calc(const image &a, const image &b, uint32_t first_chan, uint32_t total_chans, bool avg_comp_error, bool use_601_luma)
- {
- assert((first_chan < 4U) && (first_chan + total_chans <= 4U));
- const uint32_t width = basisu::minimum(a.get_width(), b.get_width());
- const uint32_t height = basisu::minimum(a.get_height(), b.get_height());
- double hist[256];
- clear_obj(hist);
- for (uint32_t y = 0; y < height; y++)
- {
- for (uint32_t x = 0; x < width; x++)
- {
- const color_rgba &ca = a(x, y), &cb = b(x, y);
- if (total_chans)
- {
- for (uint32_t c = 0; c < total_chans; c++)
- hist[iabs(ca[first_chan + c] - cb[first_chan + c])]++;
- }
- else
- {
- if (use_601_luma)
- hist[iabs(ca.get_601_luma() - cb.get_601_luma())]++;
- else
- hist[iabs(ca.get_709_luma() - cb.get_709_luma())]++;
- }
- }
- }
- m_max = 0;
- double sum = 0.0f, sum2 = 0.0f;
- for (uint32_t i = 0; i < 256; i++)
- {
- if (hist[i])
- {
- m_max = basisu::maximum<float>(m_max, (float)i);
- double v = i * hist[i];
- sum += v;
- sum2 += i * v;
- }
- }
- double total_values = (double)width * (double)height;
- if (avg_comp_error)
- total_values *= (double)clamp<uint32_t>(total_chans, 1, 4);
- m_mean = (float)clamp<double>(sum / total_values, 0.0f, 255.0);
- m_mean_squared = (float)clamp<double>(sum2 / total_values, 0.0f, 255.0f * 255.0f);
- m_rms = (float)sqrt(m_mean_squared);
- m_psnr = m_rms ? (float)clamp<double>(log10(255.0 / m_rms) * 20.0f, 0.0f, 100.0f) : 100.0f;
- }
- void fill_buffer_with_random_bytes(void *pBuf, size_t size, uint32_t seed)
- {
- rand r(seed);
- uint8_t *pDst = static_cast<uint8_t *>(pBuf);
- while (size >= sizeof(uint32_t))
- {
- *(uint32_t *)pDst = r.urand32();
- pDst += sizeof(uint32_t);
- size -= sizeof(uint32_t);
- }
- while (size)
- {
- *pDst++ = r.byte();
- size--;
- }
- }
- uint32_t hash_hsieh(const uint8_t *pBuf, size_t len)
- {
- if (!pBuf || !len)
- return 0;
- uint32_t h = static_cast<uint32_t>(len);
- const uint32_t bytes_left = len & 3;
- len >>= 2;
- while (len--)
- {
- const uint16_t *pWords = reinterpret_cast<const uint16_t *>(pBuf);
- h += pWords[0];
-
- const uint32_t t = (pWords[1] << 11) ^ h;
- h = (h << 16) ^ t;
-
- pBuf += sizeof(uint32_t);
-
- h += h >> 11;
- }
- switch (bytes_left)
- {
- case 1:
- h += *reinterpret_cast<const signed char*>(pBuf);
- h ^= h << 10;
- h += h >> 1;
- break;
- case 2:
- h += *reinterpret_cast<const uint16_t *>(pBuf);
- h ^= h << 11;
- h += h >> 17;
- break;
- case 3:
- h += *reinterpret_cast<const uint16_t *>(pBuf);
- h ^= h << 16;
- h ^= (static_cast<signed char>(pBuf[sizeof(uint16_t)])) << 18;
- h += h >> 11;
- break;
- default:
- break;
- }
-
- h ^= h << 3;
- h += h >> 5;
- h ^= h << 4;
- h += h >> 17;
- h ^= h << 25;
- h += h >> 6;
- return h;
- }
- job_pool::job_pool(uint32_t num_threads) :
- m_num_active_jobs(0),
- m_kill_flag(false)
- {
- assert(num_threads >= 1U);
- debug_printf("job_pool::job_pool: %u total threads\n", num_threads);
- if (num_threads > 1)
- {
- m_threads.resize(num_threads - 1);
- for (int i = 0; i < ((int)num_threads - 1); i++)
- m_threads[i] = std::thread([this, i] { job_thread(i); });
- }
- }
- job_pool::~job_pool()
- {
- debug_printf("job_pool::~job_pool\n");
-
- // Notify all workers that they need to die right now.
- m_kill_flag = true;
-
- m_has_work.notify_all();
- // Wait for all workers to die.
- for (uint32_t i = 0; i < m_threads.size(); i++)
- m_threads[i].join();
- }
-
- void job_pool::add_job(const std::function<void()>& job)
- {
- std::unique_lock<std::mutex> lock(m_mutex);
- m_queue.emplace_back(job);
- const size_t queue_size = m_queue.size();
- lock.unlock();
- if (queue_size > 1)
- m_has_work.notify_one();
- }
- void job_pool::add_job(std::function<void()>&& job)
- {
- std::unique_lock<std::mutex> lock(m_mutex);
- m_queue.emplace_back(std::move(job));
-
- const size_t queue_size = m_queue.size();
- lock.unlock();
- if (queue_size > 1)
- {
- m_has_work.notify_one();
- }
- }
- void job_pool::wait_for_all()
- {
- std::unique_lock<std::mutex> lock(m_mutex);
- // Drain the job queue on the calling thread.
- while (!m_queue.empty())
- {
- std::function<void()> job(m_queue.back());
- m_queue.pop_back();
- lock.unlock();
- job();
- lock.lock();
- }
- // The queue is empty, now wait for all active jobs to finish up.
- m_no_more_jobs.wait(lock, [this]{ return !m_num_active_jobs; } );
- }
- void job_pool::job_thread(uint32_t index)
- {
- BASISU_NOTE_UNUSED(index);
- //debug_printf("job_pool::job_thread: starting %u\n", index);
-
- while (true)
- {
- std::unique_lock<std::mutex> lock(m_mutex);
- // Wait for any jobs to be issued.
- m_has_work.wait(lock, [this] { return m_kill_flag || m_queue.size(); } );
- // Check to see if we're supposed to exit.
- if (m_kill_flag)
- break;
- // Get the job and execute it.
- std::function<void()> job(m_queue.back());
- m_queue.pop_back();
- ++m_num_active_jobs;
- lock.unlock();
- job();
- lock.lock();
- --m_num_active_jobs;
- // Now check if there are no more jobs remaining.
- const bool all_done = m_queue.empty() && !m_num_active_jobs;
-
- lock.unlock();
- if (all_done)
- m_no_more_jobs.notify_all();
- }
- //debug_printf("job_pool::job_thread: exiting\n");
- }
- // .TGA image loading
- #pragma pack(push)
- #pragma pack(1)
- struct tga_header
- {
- uint8_t m_id_len;
- uint8_t m_cmap;
- uint8_t m_type;
- packed_uint<2> m_cmap_first;
- packed_uint<2> m_cmap_len;
- uint8_t m_cmap_bpp;
- packed_uint<2> m_x_org;
- packed_uint<2> m_y_org;
- packed_uint<2> m_width;
- packed_uint<2> m_height;
- uint8_t m_depth;
- uint8_t m_desc;
- };
- #pragma pack(pop)
- const uint32_t MAX_TGA_IMAGE_SIZE = 16384;
- enum tga_image_type
- {
- cITPalettized = 1,
- cITRGB = 2,
- cITGrayscale = 3
- };
- uint8_t *read_tga(const uint8_t *pBuf, uint32_t buf_size, int &width, int &height, int &n_chans)
- {
- width = 0;
- height = 0;
- n_chans = 0;
- if (buf_size <= sizeof(tga_header))
- return nullptr;
- const tga_header &hdr = *reinterpret_cast<const tga_header *>(pBuf);
- if ((!hdr.m_width) || (!hdr.m_height) || (hdr.m_width > MAX_TGA_IMAGE_SIZE) || (hdr.m_height > MAX_TGA_IMAGE_SIZE))
- return nullptr;
- if (hdr.m_desc >> 6)
- return nullptr;
- // Simple validation
- if ((hdr.m_cmap != 0) && (hdr.m_cmap != 1))
- return nullptr;
-
- if (hdr.m_cmap)
- {
- if ((hdr.m_cmap_bpp == 0) || (hdr.m_cmap_bpp > 32))
- return nullptr;
- // Nobody implements CMapFirst correctly, so we're not supporting it. Never seen it used, either.
- if (hdr.m_cmap_first != 0)
- return nullptr;
- }
- const bool x_flipped = (hdr.m_desc & 0x10) != 0;
- const bool y_flipped = (hdr.m_desc & 0x20) == 0;
- bool rle_flag = false;
- int file_image_type = hdr.m_type;
- if (file_image_type > 8)
- {
- file_image_type -= 8;
- rle_flag = true;
- }
- const tga_image_type image_type = static_cast<tga_image_type>(file_image_type);
- switch (file_image_type)
- {
- case cITRGB:
- if (hdr.m_depth == 8)
- return nullptr;
- break;
- case cITPalettized:
- if ((hdr.m_depth != 8) || (hdr.m_cmap != 1) || (hdr.m_cmap_len == 0))
- return nullptr;
- break;
- case cITGrayscale:
- if ((hdr.m_cmap != 0) || (hdr.m_cmap_len != 0))
- return nullptr;
- if ((hdr.m_depth != 8) && (hdr.m_depth != 16))
- return nullptr;
- break;
- default:
- return nullptr;
- }
- uint32_t tga_bytes_per_pixel = 0;
- switch (hdr.m_depth)
- {
- case 32:
- tga_bytes_per_pixel = 4;
- n_chans = 4;
- break;
- case 24:
- tga_bytes_per_pixel = 3;
- n_chans = 3;
- break;
- case 16:
- case 15:
- tga_bytes_per_pixel = 2;
- // For compatibility with stb_image_write.h
- n_chans = ((file_image_type == cITGrayscale) && (hdr.m_depth == 16)) ? 4 : 3;
- break;
- case 8:
- tga_bytes_per_pixel = 1;
- // For palettized RGBA support, which both FreeImage and stb_image support.
- n_chans = ((file_image_type == cITPalettized) && (hdr.m_cmap_bpp == 32)) ? 4 : 3;
- break;
- default:
- return nullptr;
- }
- //const uint32_t bytes_per_line = hdr.m_width * tga_bytes_per_pixel;
- const uint8_t *pSrc = pBuf + sizeof(tga_header);
- uint32_t bytes_remaining = buf_size - sizeof(tga_header);
- if (hdr.m_id_len)
- {
- if (bytes_remaining < hdr.m_id_len)
- return nullptr;
- pSrc += hdr.m_id_len;
- bytes_remaining += hdr.m_id_len;
- }
- color_rgba pal[256];
- for (uint32_t i = 0; i < 256; i++)
- pal[i].set(0, 0, 0, 255);
- if ((hdr.m_cmap) && (hdr.m_cmap_len))
- {
- if (image_type == cITPalettized)
- {
- // Note I cannot find any files using 32bpp palettes in the wild (never seen any in ~30 years).
- if ( ((hdr.m_cmap_bpp != 32) && (hdr.m_cmap_bpp != 24) && (hdr.m_cmap_bpp != 15) && (hdr.m_cmap_bpp != 16)) || (hdr.m_cmap_len > 256) )
- return nullptr;
- if (hdr.m_cmap_bpp == 32)
- {
- const uint32_t pal_size = hdr.m_cmap_len * 4;
- if (bytes_remaining < pal_size)
- return nullptr;
- for (uint32_t i = 0; i < hdr.m_cmap_len; i++)
- {
- pal[i].r = pSrc[i * 4 + 2];
- pal[i].g = pSrc[i * 4 + 1];
- pal[i].b = pSrc[i * 4 + 0];
- pal[i].a = pSrc[i * 4 + 3];
- }
- bytes_remaining -= pal_size;
- pSrc += pal_size;
- }
- else if (hdr.m_cmap_bpp == 24)
- {
- const uint32_t pal_size = hdr.m_cmap_len * 3;
- if (bytes_remaining < pal_size)
- return nullptr;
- for (uint32_t i = 0; i < hdr.m_cmap_len; i++)
- {
- pal[i].r = pSrc[i * 3 + 2];
- pal[i].g = pSrc[i * 3 + 1];
- pal[i].b = pSrc[i * 3 + 0];
- pal[i].a = 255;
- }
- bytes_remaining -= pal_size;
- pSrc += pal_size;
- }
- else
- {
- const uint32_t pal_size = hdr.m_cmap_len * 2;
- if (bytes_remaining < pal_size)
- return nullptr;
- for (uint32_t i = 0; i < hdr.m_cmap_len; i++)
- {
- const uint32_t v = pSrc[i * 2 + 0] | (pSrc[i * 2 + 1] << 8);
- pal[i].r = (((v >> 10) & 31) * 255 + 15) / 31;
- pal[i].g = (((v >> 5) & 31) * 255 + 15) / 31;
- pal[i].b = ((v & 31) * 255 + 15) / 31;
- pal[i].a = 255;
- }
- bytes_remaining -= pal_size;
- pSrc += pal_size;
- }
- }
- else
- {
- const uint32_t bytes_to_skip = (hdr.m_cmap_bpp >> 3) * hdr.m_cmap_len;
- if (bytes_remaining < bytes_to_skip)
- return nullptr;
- pSrc += bytes_to_skip;
- bytes_remaining += bytes_to_skip;
- }
- }
-
- width = hdr.m_width;
- height = hdr.m_height;
- const uint32_t source_pitch = width * tga_bytes_per_pixel;
- const uint32_t dest_pitch = width * n_chans;
-
- uint8_t *pImage = (uint8_t *)malloc(dest_pitch * height);
- if (!pImage)
- return nullptr;
- std::vector<uint8_t> input_line_buf;
- if (rle_flag)
- input_line_buf.resize(source_pitch);
- int run_type = 0, run_remaining = 0;
- uint8_t run_pixel[4];
- memset(run_pixel, 0, sizeof(run_pixel));
- for (int y = 0; y < height; y++)
- {
- const uint8_t *pLine_data;
- if (rle_flag)
- {
- int pixels_remaining = width;
- uint8_t *pDst = &input_line_buf[0];
- do
- {
- if (!run_remaining)
- {
- if (bytes_remaining < 1)
- {
- free(pImage);
- return nullptr;
- }
- int v = *pSrc++;
- bytes_remaining--;
- run_type = v & 0x80;
- run_remaining = (v & 0x7F) + 1;
- if (run_type)
- {
- if (bytes_remaining < tga_bytes_per_pixel)
- {
- free(pImage);
- return nullptr;
- }
- memcpy(run_pixel, pSrc, tga_bytes_per_pixel);
- pSrc += tga_bytes_per_pixel;
- bytes_remaining -= tga_bytes_per_pixel;
- }
- }
- const uint32_t n = basisu::minimum<uint32_t>(pixels_remaining, run_remaining);
- pixels_remaining -= n;
- run_remaining -= n;
- if (run_type)
- {
- for (uint32_t i = 0; i < n; i++)
- for (uint32_t j = 0; j < tga_bytes_per_pixel; j++)
- *pDst++ = run_pixel[j];
- }
- else
- {
- const uint32_t bytes_wanted = n * tga_bytes_per_pixel;
- if (bytes_remaining < bytes_wanted)
- {
- free(pImage);
- return nullptr;
- }
- memcpy(pDst, pSrc, bytes_wanted);
- pDst += bytes_wanted;
- pSrc += bytes_wanted;
- bytes_remaining -= bytes_wanted;
- }
- } while (pixels_remaining);
- assert((pDst - &input_line_buf[0]) == width * tga_bytes_per_pixel);
- pLine_data = &input_line_buf[0];
- }
- else
- {
- if (bytes_remaining < source_pitch)
- {
- free(pImage);
- return nullptr;
- }
- pLine_data = pSrc;
- bytes_remaining -= source_pitch;
- pSrc += source_pitch;
- }
- // Convert to 24bpp RGB or 32bpp RGBA.
- uint8_t *pDst = pImage + (y_flipped ? (height - 1 - y) : y) * dest_pitch + (x_flipped ? (width - 1) * n_chans : 0);
- const int dst_stride = x_flipped ? -((int)n_chans) : n_chans;
- switch (hdr.m_depth)
- {
- case 32:
- assert(tga_bytes_per_pixel == 4 && n_chans == 4);
- for (int i = 0; i < width; i++, pLine_data += 4, pDst += dst_stride)
- {
- pDst[0] = pLine_data[2];
- pDst[1] = pLine_data[1];
- pDst[2] = pLine_data[0];
- pDst[3] = pLine_data[3];
- }
- break;
- case 24:
- assert(tga_bytes_per_pixel == 3 && n_chans == 3);
- for (int i = 0; i < width; i++, pLine_data += 3, pDst += dst_stride)
- {
- pDst[0] = pLine_data[2];
- pDst[1] = pLine_data[1];
- pDst[2] = pLine_data[0];
- }
- break;
- case 16:
- case 15:
- if (image_type == cITRGB)
- {
- assert(tga_bytes_per_pixel == 2 && n_chans == 3);
- for (int i = 0; i < width; i++, pLine_data += 2, pDst += dst_stride)
- {
- const uint32_t v = pLine_data[0] | (pLine_data[1] << 8);
- pDst[0] = (((v >> 10) & 31) * 255 + 15) / 31;
- pDst[1] = (((v >> 5) & 31) * 255 + 15) / 31;
- pDst[2] = ((v & 31) * 255 + 15) / 31;
- }
- }
- else
- {
- assert(image_type == cITGrayscale && tga_bytes_per_pixel == 2 && n_chans == 4);
- for (int i = 0; i < width; i++, pLine_data += 2, pDst += dst_stride)
- {
- pDst[0] = pLine_data[0];
- pDst[1] = pLine_data[0];
- pDst[2] = pLine_data[0];
- pDst[3] = pLine_data[1];
- }
- }
- break;
- case 8:
- assert(tga_bytes_per_pixel == 1);
- if (image_type == cITPalettized)
- {
- if (hdr.m_cmap_bpp == 32)
- {
- assert(n_chans == 4);
- for (int i = 0; i < width; i++, pLine_data++, pDst += dst_stride)
- {
- const uint32_t c = *pLine_data;
- pDst[0] = pal[c].r;
- pDst[1] = pal[c].g;
- pDst[2] = pal[c].b;
- pDst[3] = pal[c].a;
- }
- }
- else
- {
- assert(n_chans == 3);
- for (int i = 0; i < width; i++, pLine_data++, pDst += dst_stride)
- {
- const uint32_t c = *pLine_data;
- pDst[0] = pal[c].r;
- pDst[1] = pal[c].g;
- pDst[2] = pal[c].b;
- }
- }
- }
- else
- {
- assert(n_chans == 3);
- for (int i = 0; i < width; i++, pLine_data++, pDst += dst_stride)
- {
- const uint8_t c = *pLine_data;
- pDst[0] = c;
- pDst[1] = c;
- pDst[2] = c;
- }
- }
- break;
- default:
- assert(0);
- break;
- }
- } // y
- return pImage;
- }
- uint8_t *read_tga(const char *pFilename, int &width, int &height, int &n_chans)
- {
- width = height = n_chans = 0;
- uint8_vec filedata;
- if (!read_file_to_vec(pFilename, filedata))
- return nullptr;
- if (!filedata.size() || (filedata.size() > UINT32_MAX))
- return nullptr;
-
- return read_tga(&filedata[0], (uint32_t)filedata.size(), width, height, n_chans);
- }
- void image::debug_text(uint32_t x_ofs, uint32_t y_ofs, uint32_t scale_x, uint32_t scale_y, const color_rgba& fg, const color_rgba* pBG, bool alpha_only, const char* pFmt, ...)
- {
- char buf[2048];
- va_list args;
- va_start(args, pFmt);
- #ifdef _WIN32
- vsprintf_s(buf, sizeof(buf), pFmt, args);
- #else
- vsnprintf(buf, sizeof(buf), pFmt, args);
- #endif
- va_end(args);
- const char* p = buf;
- const uint32_t orig_x_ofs = x_ofs;
- while (*p)
- {
- uint8_t c = *p++;
- if ((c < 32) || (c > 127))
- c = '.';
- const uint8_t* pGlpyh = &g_debug_font8x8_basic[c - 32][0];
- for (uint32_t y = 0; y < 8; y++)
- {
- uint32_t row_bits = pGlpyh[y];
- for (uint32_t x = 0; x < 8; x++)
- {
- const uint32_t q = row_bits & (1 << x);
-
- const color_rgba* pColor = q ? &fg : pBG;
- if (!pColor)
- continue;
- if (alpha_only)
- fill_box_alpha(x_ofs + x * scale_x, y_ofs + y * scale_y, scale_x, scale_y, *pColor);
- else
- fill_box(x_ofs + x * scale_x, y_ofs + y * scale_y, scale_x, scale_y, *pColor);
- }
- }
- x_ofs += 8 * scale_x;
- if ((x_ofs + 8 * scale_x) > m_width)
- {
- x_ofs = orig_x_ofs;
- y_ofs += 8 * scale_y;
- }
- }
- }
-
- } // namespace basisu
|