lzham_symbol_codec.cpp 54 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484
  1. // File: lzham_symbol_codec.cpp
  2. // See Copyright Notice and license at the end of include/lzham.h
  3. #include "lzham_core.h"
  4. #include "lzham_symbol_codec.h"
  5. #include "lzham_huffman_codes.h"
  6. // Was 16 in previous versions.
  7. #define LZHAM_DEFAULT_MAX_UPDATE_INTERVAL 64
  8. // Was 40 in previous versions
  9. // Keep in sync with default setting in g_table_update_settings[]
  10. #define LZHAM_DEFAULT_ADAPT_RATE 64U
  11. namespace lzham
  12. {
  13. // Using a fixed table to convert from scaled probability to scaled bits for determinism across compilers/run-time libs/platforms.
  14. uint32 g_prob_cost[cSymbolCodecArithProbScale] =
  15. {
  16. 0x0,0xB000000,0xA000000,0x96A3FE6,0x9000000,0x8AD961F,0x86A3FE6,0x8315130,0x8000000,0x7D47FCC,0x7AD961F,
  17. 0x78A62B0,0x76A3FE6,0x74CAFFC,0x7315130,0x717D605,0x7000000,0x6E99C09,0x6D47FCC,0x6C087D3,0x6AD961F,0x69B9116,
  18. 0x68A62B0,0x679F7D8,0x66A3FE6,0x65B2C3E,0x64CAFFC,0x63EBFB1,0x6315130,0x6245B5C,0x617D605,0x60BB9CA,0x6000000,
  19. 0x5F4A296,0x5E99C09,0x5DEE74F,0x5D47FCC,0x5CA6144,0x5C087D3,0x5B6EFE1,0x5AD961F,0x5A47779,0x59B9116,0x592E050,
  20. 0x58A62B0,0x58215EA,0x579F7D8,0x5720677,0x56A3FE6,0x562A260,0x55B2C3E,0x553DBEF,0x54CAFFC,0x545A701,0x53EBFB1,
  21. 0x537F8CF,0x5315130,0x52AC7B8,0x5245B5C,0x51E0B1B,0x517D605,0x511BB33,0x50BB9CA,0x505D0FC,0x5000000,0x4FA461A,
  22. 0x4F4A296,0x4EF14C7,0x4E99C09,0x4E437BE,0x4DEE74F,0x4D9AA2C,0x4D47FCC,0x4CF67A8,0x4CA6144,0x4C56C23,0x4C087D3,
  23. 0x4BBB3E1,0x4B6EFE1,0x4B23B6D,0x4AD961F,0x4A8FF97,0x4A47779,0x49FFD6A,0x49B9116,0x4973228,0x492E050,0x48E9B41,
  24. 0x48A62B0,0x4863655,0x48215EA,0x47E012C,0x479F7D8,0x475F9B0,0x4720677,0x46E1DF1,0x46A3FE6,0x4666C1D,0x462A260,
  25. 0x45EE27C,0x45B2C3E,0x4577F74,0x453DBEF,0x4504180,0x44CAFFC,0x4492735,0x445A701,0x4422F38,0x43EBFB1,0x43B5846,
  26. 0x437F8CF,0x434A129,0x4315130,0x42E08C0,0x42AC7B8,0x4278DF7,0x4245B5C,0x4212FC7,0x41E0B1B,0x41AED39,0x417D605,
  27. 0x414C561,0x411BB33,0x40EB75F,0x40BB9CA,0x408C25C,0x405D0FC,0x402E58F,0x4000000,0x3FD2036,0x3FA461A,0x3F77197,
  28. 0x3F4A296,0x3F1D903,0x3EF14C7,0x3EC55D0,0x3E99C09,0x3E6E75F,0x3E437BE,0x3E18D14,0x3DEE74F,0x3DC465D,0x3D9AA2C,
  29. 0x3D712AC,0x3D47FCC,0x3D1F17A,0x3CF67A8,0x3CCE246,0x3CA6144,0x3C7E492,0x3C56C23,0x3C2F7E8,0x3C087D3,0x3BE1BD5,
  30. 0x3BBB3E1,0x3B94FE9,0x3B6EFE1,0x3B493BC,0x3B23B6D,0x3AFE6E7,0x3AD961F,0x3AB4908,0x3A8FF97,0x3A6B9C0,0x3A47779,
  31. 0x3A238B5,0x39FFD6A,0x39DC58E,0x39B9116,0x3995FF7,0x3973228,0x395079E,0x392E050,0x390BC34,0x38E9B41,0x38C7D6E,
  32. 0x38A62B0,0x3884B01,0x3863655,0x38424A6,0x38215EA,0x3800A1A,0x37E012C,0x37BFB18,0x379F7D8,0x377F762,0x375F9B0,
  33. 0x373FEBA,0x3720677,0x37010E1,0x36E1DF1,0x36C2DA0,0x36A3FE6,0x36854BC,0x3666C1D,0x3648600,0x362A260,0x360C136,
  34. 0x35EE27C,0x35D062B,0x35B2C3E,0x35954AD,0x3577F74,0x355AC8C,0x353DBEF,0x3520D98,0x3504180,0x34E77A4,0x34CAFFC,
  35. 0x34AEA83,0x3492735,0x347660B,0x345A701,0x343EA12,0x3422F38,0x340766F,0x33EBFB1,0x33D0AFA,0x33B5846,0x339A78E,
  36. 0x337F8CF,0x3364C05,0x334A129,0x332F839,0x3315130,0x32FAC09,0x32E08C0,0x32C6751,0x32AC7B8,0x32929F1,0x3278DF7,
  37. 0x325F3C6,0x3245B5C,0x322C4B2,0x3212FC7,0x31F9C96,0x31E0B1B,0x31C7B53,0x31AED39,0x31960CB,0x317D605,0x3164CE2,
  38. 0x314C561,0x3133F7D,0x311BB33,0x310387F,0x30EB75F,0x30D37CE,0x30BB9CA,0x30A3D50,0x308C25C,0x30748EC,0x305D0FC,
  39. 0x3045A88,0x302E58F,0x301720E,0x3000000,0x2FE8F64,0x2FD2036,0x2FBB274,0x2FA461A,0x2F8DB27,0x2F77197,0x2F60968,
  40. 0x2F4A296,0x2F33D20,0x2F1D903,0x2F0763B,0x2EF14C7,0x2EDB4A5,0x2EC55D0,0x2EAF848,0x2E99C09,0x2E84111,0x2E6E75F,
  41. 0x2E58EEE,0x2E437BE,0x2E2E1CB,0x2E18D14,0x2E03996,0x2DEE74F,0x2DD963D,0x2DC465D,0x2DAF7AD,0x2D9AA2C,0x2D85DD7,
  42. 0x2D712AC,0x2D5C8A9,0x2D47FCC,0x2D33812,0x2D1F17A,0x2D0AC02,0x2CF67A8,0x2CE246A,0x2CCE246,0x2CBA13A,0x2CA6144,
  43. 0x2C92262,0x2C7E492,0x2C6A7D4,0x2C56C23,0x2C43180,0x2C2F7E8,0x2C1BF5A,0x2C087D3,0x2BF5151,0x2BE1BD5,0x2BCE75A,
  44. 0x2BBB3E1,0x2BA8166,0x2B94FE9,0x2B81F68,0x2B6EFE1,0x2B5C153,0x2B493BC,0x2B3671A,0x2B23B6D,0x2B110B1,0x2AFE6E7,
  45. 0x2AEBE0C,0x2AD961F,0x2AC6F1E,0x2AB4908,0x2AA23DC,0x2A8FF97,0x2A7DC39,0x2A6B9C0,0x2A5982B,0x2A47779,0x2A357A7,
  46. 0x2A238B5,0x2A11AA1,0x29FFD6A,0x29EE10F,0x29DC58E,0x29CAAE6,0x29B9116,0x29A781C,0x2995FF7,0x29848A6,0x2973228,
  47. 0x2961C7B,0x295079E,0x293F390,0x292E050,0x291CDDD,0x290BC34,0x28FAB56,0x28E9B41,0x28D8BF4,0x28C7D6E,0x28B6FAD,
  48. 0x28A62B0,0x2895677,0x2884B01,0x287404B,0x2863655,0x2852D1F,0x28424A6,0x2831CEA,0x28215EA,0x2810FA5,0x2800A1A,
  49. 0x27F0547,0x27E012C,0x27CFDC7,0x27BFB18,0x27AF91E,0x279F7D8,0x278F744,0x277F762,0x276F831,0x275F9B0,0x274FBDE,
  50. 0x273FEBA,0x2730242,0x2720677,0x2710B57,0x27010E1,0x26F1715,0x26E1DF1,0x26D2575,0x26C2DA0,0x26B3670,0x26A3FE6,
  51. 0x26949FF,0x26854BC,0x267601C,0x2666C1D,0x26578BE,0x2648600,0x26393E1,0x262A260,0x261B17D,0x260C136,0x25FD18C,
  52. 0x25EE27C,0x25DF407,0x25D062B,0x25C18E8,0x25B2C3E,0x25A402A,0x25954AD,0x25869C6,0x2577F74,0x25695B6,0x255AC8C,
  53. 0x254C3F4,0x253DBEF,0x252F47B,0x2520D98,0x2512744,0x2504180,0x24F5C4B,0x24E77A4,0x24D9389,0x24CAFFC,0x24BCCFA,
  54. 0x24AEA83,0x24A0897,0x2492735,0x248465C,0x247660B,0x2468643,0x245A701,0x244C847,0x243EA12,0x2430C63,0x2422F38,
  55. 0x2415292,0x240766F,0x23F9ACF,0x23EBFB1,0x23DE515,0x23D0AFA,0x23C3160,0x23B5846,0x23A7FAB,0x239A78E,0x238CFF0,
  56. 0x237F8CF,0x237222C,0x2364C05,0x2357659,0x234A129,0x233CC74,0x232F839,0x2322478,0x2315130,0x2307E61,0x22FAC09,
  57. 0x22EDA29,0x22E08C0,0x22D37CE,0x22C6751,0x22B974A,0x22AC7B8,0x229F89B,0x22929F1,0x2285BBA,0x2278DF7,0x226C0A6,
  58. 0x225F3C6,0x2252758,0x2245B5C,0x2238FCF,0x222C4B2,0x221FA05,0x2212FC7,0x22065F7,0x21F9C96,0x21ED3A2,0x21E0B1B,
  59. 0x21D4301,0x21C7B53,0x21BB410,0x21AED39,0x21A26CD,0x21960CB,0x2189B33,0x217D605,0x217113F,0x2164CE2,0x21588EE,
  60. 0x214C561,0x214023B,0x2133F7D,0x2127D25,0x211BB33,0x210F9A6,0x210387F,0x20F77BD,0x20EB75F,0x20DF765,0x20D37CE,
  61. 0x20C789B,0x20BB9CA,0x20AFB5C,0x20A3D50,0x2097FA6,0x208C25C,0x2080574,0x20748EC,0x2068CC4,0x205D0FC,0x2051593,
  62. 0x2045A88,0x2039FDD,0x202E58F,0x2022BA0,0x201720E,0x200B8D8,0x2000000,0x1FF4784,0x1FE8F64,0x1FDD79F,0x1FD2036,
  63. 0x1FC6928,0x1FBB274,0x1FAFC1A,0x1FA461A,0x1F99074,0x1F8DB27,0x1F82633,0x1F77197,0x1F6BD53,0x1F60968,0x1F555D3,
  64. 0x1F4A296,0x1F3EFB0,0x1F33D20,0x1F28AE6,0x1F1D903,0x1F12774,0x1F0763B,0x1EFC557,0x1EF14C7,0x1EE648C,0x1EDB4A5,
  65. 0x1ED0511,0x1EC55D0,0x1EBA6E3,0x1EAF848,0x1EA49FF,0x1E99C09,0x1E8EE64,0x1E84111,0x1E79410,0x1E6E75F,0x1E63AFE,
  66. 0x1E58EEE,0x1E4E32E,0x1E437BE,0x1E38C9D,0x1E2E1CB,0x1E23748,0x1E18D14,0x1E0E32E,0x1E03996,0x1DF904C,0x1DEE74F,
  67. 0x1DE3E9F,0x1DD963D,0x1DCEE27,0x1DC465D,0x1DB9EDF,0x1DAF7AD,0x1DA50C7,0x1D9AA2C,0x1D903DC,0x1D85DD7,0x1D7B81C,
  68. 0x1D712AC,0x1D66D86,0x1D5C8A9,0x1D52416,0x1D47FCC,0x1D3DBCA,0x1D33812,0x1D294A2,0x1D1F17A,0x1D14E9B,0x1D0AC02,
  69. 0x1D009B2,0x1CF67A8,0x1CEC5E6,0x1CE246A,0x1CD8335,0x1CCE246,0x1CC419D,0x1CBA13A,0x1CB011C,0x1CA6144,0x1C9C1B0,
  70. 0x1C92262,0x1C88358,0x1C7E492,0x1C74611,0x1C6A7D4,0x1C609DA,0x1C56C23,0x1C4CEB0,0x1C43180,0x1C39493,0x1C2F7E8,
  71. 0x1C25B80,0x1C1BF5A,0x1C12375,0x1C087D3,0x1BFEC71,0x1BF5151,0x1BEB673,0x1BE1BD5,0x1BD8177,0x1BCE75A,0x1BC4D7D,
  72. 0x1BBB3E1,0x1BB1A84,0x1BA8166,0x1B9E888,0x1B94FE9,0x1B8B789,0x1B81F68,0x1B78786,0x1B6EFE1,0x1B6587B,0x1B5C153,
  73. 0x1B52A69,0x1B493BC,0x1B3FD4D,0x1B3671A,0x1B2D125,0x1B23B6D,0x1B1A5F1,0x1B110B1,0x1B07BAE,0x1AFE6E7,0x1AF525C,
  74. 0x1AEBE0C,0x1AE29F8,0x1AD961F,0x1AD0281,0x1AC6F1E,0x1ABDBF6,0x1AB4908,0x1AAB655,0x1AA23DC,0x1A9919C,0x1A8FF97,
  75. 0x1A86DCB,0x1A7DC39,0x1A74AE0,0x1A6B9C0,0x1A628DA,0x1A5982B,0x1A507B6,0x1A47779,0x1A3E774,0x1A357A7,0x1A2C812,
  76. 0x1A238B5,0x1A1A98F,0x1A11AA1,0x1A08BEA,0x19FFD6A,0x19F6F21,0x19EE10F,0x19E5333,0x19DC58E,0x19D381F,0x19CAAE6,
  77. 0x19C1DE3,0x19B9116,0x19B047E,0x19A781C,0x199EBEF,0x1995FF7,0x198D434,0x19848A6,0x197BD4D,0x1973228,0x196A737,
  78. 0x1961C7B,0x19591F3,0x195079E,0x1947D7D,0x193F390,0x19369D7,0x192E050,0x19256FD,0x191CDDD,0x19144EF,0x190BC34,
  79. 0x19033AC,0x18FAB56,0x18F2333,0x18E9B41,0x18E1382,0x18D8BF4,0x18D0498,0x18C7D6E,0x18BF675,0x18B6FAD,0x18AE916,
  80. 0x18A62B0,0x189DC7C,0x1895677,0x188D0A4,0x1884B01,0x187C58E,0x187404B,0x186BB38,0x1863655,0x185B1A2,0x1852D1F,
  81. 0x184A8CB,0x18424A6,0x183A0B1,0x1831CEA,0x1829953,0x18215EA,0x18192B0,0x1810FA5,0x1808CC8,0x1800A1A,0x17F8799,
  82. 0x17F0547,0x17E8322,0x17E012C,0x17D7F63,0x17CFDC7,0x17C7C59,0x17BFB18,0x17B7A05,0x17AF91E,0x17A7865,0x179F7D8,
  83. 0x1797778,0x178F744,0x178773D,0x177F762,0x17777B4,0x176F831,0x17678DB,0x175F9B0,0x1757AB1,0x174FBDE,0x1747D36,
  84. 0x173FEBA,0x1738068,0x1730242,0x1728447,0x1720677,0x17188D2,0x1710B57,0x1708E07,0x17010E1,0x16F93E6,0x16F1715,
  85. 0x16E9A6E,0x16E1DF1,0x16DA19E,0x16D2575,0x16CA976,0x16C2DA0,0x16BB1F3,0x16B3670,0x16ABB16,0x16A3FE6,0x169C4DE,
  86. 0x16949FF,0x168CF49,0x16854BC,0x167DA58,0x167601C,0x166E608,0x1666C1D,0x165F25A,0x16578BE,0x164FF4B,0x1648600,
  87. 0x1640CDD,0x16393E1,0x1631B0D,0x162A260,0x16229DB,0x161B17D,0x1613946,0x160C136,0x160494D,0x15FD18C,0x15F59F0,
  88. 0x15EE27C,0x15E6B2E,0x15DF407,0x15D7D06,0x15D062B,0x15C8F77,0x15C18E8,0x15BA280,0x15B2C3E,0x15AB621,0x15A402A,
  89. 0x159CA59,0x15954AD,0x158DF27,0x15869C6,0x157F48A,0x1577F74,0x1570A82,0x15695B6,0x156210E,0x155AC8C,0x155382E,
  90. 0x154C3F4,0x1544FDF,0x153DBEF,0x1536823,0x152F47B,0x15280F7,0x1520D98,0x1519A5C,0x1512744,0x150B450,0x1504180,
  91. 0x14FCED4,0x14F5C4B,0x14EE9E6,0x14E77A4,0x14E0585,0x14D9389,0x14D21B1,0x14CAFFC,0x14C3E69,0x14BCCFA,0x14B5BAD,
  92. 0x14AEA83,0x14A797C,0x14A0897,0x14997D5,0x1492735,0x148B6B7,0x148465C,0x147D622,0x147660B,0x146F616,0x1468643,
  93. 0x1461691,0x145A701,0x1453793,0x144C847,0x144591C,0x143EA12,0x1437B2A,0x1430C63,0x1429DBD,0x1422F38,0x141C0D5,
  94. 0x1415292,0x140E470,0x140766F,0x140088F,0x13F9ACF,0x13F2D30,0x13EBFB1,0x13E5253,0x13DE515,0x13D77F8,0x13D0AFA,
  95. 0x13C9E1D,0x13C3160,0x13BC4C3,0x13B5846,0x13AEBE8,0x13A7FAB,0x13A138D,0x139A78E,0x1393BAF,0x138CFF0,0x1386450,
  96. 0x137F8CF,0x1378D6E,0x137222C,0x136B709,0x1364C05,0x135E11F,0x1357659,0x1350BB2,0x134A129,0x13436C0,0x133CC74,
  97. 0x1336248,0x132F839,0x1328E4A,0x1322478,0x131BAC5,0x1315130,0x130E7B9,0x1307E61,0x1301526,0x12FAC09,0x12F430A,
  98. 0x12EDA29,0x12E7166,0x12E08C0,0x12DA038,0x12D37CE,0x12CCF81,0x12C6751,0x12BFF3F,0x12B974A,0x12B2F73,0x12AC7B8,
  99. 0x12A601B,0x129F89B,0x1299137,0x12929F1,0x128C2C7,0x1285BBA,0x127F4CA,0x1278DF7,0x1272740,0x126C0A6,0x1265A28,
  100. 0x125F3C6,0x1258D81,0x1252758,0x124C14C,0x1245B5C,0x123F587,0x1238FCF,0x1232A33,0x122C4B2,0x1225F4E,0x121FA05,
  101. 0x12194D8,0x1212FC7,0x120CAD1,0x12065F7,0x1200139,0x11F9C96,0x11F380E,0x11ED3A2,0x11E6F51,0x11E0B1B,0x11DA700,
  102. 0x11D4301,0x11CDF1C,0x11C7B53,0x11C17A4,0x11BB410,0x11B5097,0x11AED39,0x11A89F6,0x11A26CD,0x119C3BF,0x11960CB,
  103. 0x118FDF2,0x1189B33,0x118388F,0x117D605,0x1177395,0x117113F,0x116AF04,0x1164CE2,0x115EADB,0x11588EE,0x115271A,
  104. 0x114C561,0x11463C1,0x114023B,0x113A0CF,0x1133F7D,0x112DE44,0x1127D25,0x1121C1F,0x111BB33,0x1115A60,0x110F9A6,
  105. 0x1109906,0x110387F,0x10FD811,0x10F77BD,0x10F1781,0x10EB75F,0x10E5755,0x10DF765,0x10D978D,0x10D37CE,0x10CD828,
  106. 0x10C789B,0x10C1926,0x10BB9CA,0x10B5A87,0x10AFB5C,0x10A9C4A,0x10A3D50,0x109DE6F,0x1097FA6,0x10920F5,0x108C25C,
  107. 0x10863DC,0x1080574,0x107A724,0x10748EC,0x106EACC,0x1068CC4,0x1062ED4,0x105D0FC,0x105733B,0x1051593,0x104B802,
  108. 0x1045A88,0x103FD27,0x1039FDD,0x10342AA,0x102E58F,0x102888C,0x1022BA0,0x101CECB,0x101720E,0x1011567,0x100B8D8,
  109. 0x1005C61,0x1000000,0xFFA3B6,0xFF4784,0xFEEB68,0xFE8F64,0xFE3376,0xFDD79F,0xFD7BDF,0xFD2036,0xFCC4A3,
  110. 0xFC6928,0xFC0DC2,0xFBB274,0xFB573C,0xFAFC1A,0xFAA10F,0xFA461A,0xF9EB3C,0xF99074,0xF935C2,0xF8DB27,
  111. 0xF880A2,0xF82633,0xF7CBDA,0xF77197,0xF7176A,0xF6BD53,0xF66353,0xF60968,0xF5AF93,0xF555D3,0xF4FC2A,
  112. 0xF4A296,0xF44918,0xF3EFB0,0xF3965D,0xF33D20,0xF2E3F9,0xF28AE6,0xF231EA,0xF1D903,0xF18031,0xF12774,
  113. 0xF0CECD,0xF0763B,0xF01DBF,0xEFC557,0xEF6D05,0xEF14C7,0xEEBC9F,0xEE648C,0xEE0C8E,0xEDB4A5,0xED5CD0,
  114. 0xED0511,0xECAD66,0xEC55D0,0xEBFE4F,0xEBA6E3,0xEB4F8B,0xEAF848,0xEAA119,0xEA49FF,0xE9F2FA,0xE99C09,
  115. 0xE9452D,0xE8EE64,0xE897B1,0xE84111,0xE7EA86,0xE79410,0xE73DAD,0xE6E75F,0xE69124,0xE63AFE,0xE5E4EC,
  116. 0xE58EEE,0xE53904,0xE4E32E,0xE48D6C,0xE437BE,0xE3E223,0xE38C9D,0xE3372A,0xE2E1CB,0xE28C80,0xE23748,
  117. 0xE1E224,0xE18D14,0xE13817,0xE0E32E,0xE08E58,0xE03996,0xDFE4E7,0xDF904C,0xDF3BC4,0xDEE74F,0xDE92ED,
  118. 0xDE3E9F,0xDDEA64,0xDD963D,0xDD4228,0xDCEE27,0xDC9A38,0xDC465D,0xDBF295,0xDB9EDF,0xDB4B3D,0xDAF7AD,
  119. 0xDAA431,0xDA50C7,0xD9FD70,0xD9AA2C,0xD956FB,0xD903DC,0xD8B0D0,0xD85DD7,0xD80AF1,0xD7B81C,0xD7655B,
  120. 0xD712AC,0xD6C010,0xD66D86,0xD61B0E,0xD5C8A9,0xD57656,0xD52416,0xD4D1E7,0xD47FCC,0xD42DC2,0xD3DBCA,
  121. 0xD389E5,0xD33812,0xD2E651,0xD294A2,0xD24305,0xD1F17A,0xD1A001,0xD14E9B,0xD0FD46,0xD0AC02,0xD05AD1,
  122. 0xD009B2,0xCFB8A4,0xCF67A8,0xCF16BE,0xCEC5E6,0xCE751F,0xCE246A,0xCDD3C7,0xCD8335,0xCD32B5,0xCCE246,
  123. 0xCC91E9,0xCC419D,0xCBF163,0xCBA13A,0xCB5122,0xCB011C,0xCAB127,0xCA6144,0xCA1171,0xC9C1B0,0xC97200,
  124. 0xC92262,0xC8D2D4,0xC88358,0xC833ED,0xC7E492,0xC79549,0xC74611,0xC6F6EA,0xC6A7D4,0xC658CE,0xC609DA,
  125. 0xC5BAF6,0xC56C23,0xC51D61,0xC4CEB0,0xC48010,0xC43180,0xC3E301,0xC39493,0xC34635,0xC2F7E8,0xC2A9AC,
  126. 0xC25B80,0xC20D64,0xC1BF5A,0xC1715F,0xC12375,0xC0D59C,0xC087D3,0xC03A1A,0xBFEC71,0xBF9ED9,0xBF5151,
  127. 0xBF03DA,0xBEB673,0xBE691B,0xBE1BD5,0xBDCE9E,0xBD8177,0xBD3461,0xBCE75A,0xBC9A64,0xBC4D7D,0xBC00A7,
  128. 0xBBB3E1,0xBB672A,0xBB1A84,0xBACDED,0xBA8166,0xBA34EF,0xB9E888,0xB99C31,0xB94FE9,0xB903B1,0xB8B789,
  129. 0xB86B71,0xB81F68,0xB7D36F,0xB78786,0xB73BAC,0xB6EFE1,0xB6A427,0xB6587B,0xB60CDF,0xB5C153,0xB575D6,
  130. 0xB52A69,0xB4DF0B,0xB493BC,0xB4487D,0xB3FD4D,0xB3B22C,0xB3671A,0xB31C18,0xB2D125,0xB28641,0xB23B6D,
  131. 0xB1F0A7,0xB1A5F1,0xB15B4A,0xB110B1,0xB0C628,0xB07BAE,0xB03143,0xAFE6E7,0xAF9C9A,0xAF525C,0xAF082C,
  132. 0xAEBE0C,0xAE73FA,0xAE29F8,0xADE004,0xAD961F,0xAD4C49,0xAD0281,0xACB8C8,0xAC6F1E,0xAC2583,0xABDBF6,
  133. 0xAB9278,0xAB4908,0xAAFFA7,0xAAB655,0xAA6D11,0xAA23DC,0xA9DAB5,0xA9919C,0xA94893,0xA8FF97,0xA8B6AA,
  134. 0xA86DCB,0xA824FB,0xA7DC39,0xA79386,0xA74AE0,0xA70249,0xA6B9C0,0xA67146,0xA628DA,0xA5E07B,0xA5982B,
  135. 0xA54FEA,0xA507B6,0xA4BF90,0xA47779,0xA42F6F,0xA3E774,0xA39F87,0xA357A7,0xA30FD6,0xA2C812,0xA2805D,
  136. 0xA238B5,0xA1F11B,0xA1A98F,0xA16211,0xA11AA1,0xA0D33F,0xA08BEA,0xA044A4,0x9FFD6A,0x9FB63F,0x9F6F21,
  137. 0x9F2811,0x9EE10F,0x9E9A1B,0x9E5333,0x9E0C5A,0x9DC58E,0x9D7ED0,0x9D381F,0x9CF17C,0x9CAAE6,0x9C645E,
  138. 0x9C1DE3,0x9BD776,0x9B9116,0x9B4AC3,0x9B047E,0x9ABE46,0x9A781C,0x9A31FF,0x99EBEF,0x99A5EC,0x995FF7,
  139. 0x991A0F,0x98D434,0x988E67,0x9848A6,0x9802F3,0x97BD4D,0x9777B4,0x973228,0x96ECA9,0x96A737,0x9661D3,
  140. 0x961C7B,0x95D730,0x9591F3,0x954CC2,0x95079E,0x94C287,0x947D7D,0x943880,0x93F390,0x93AEAD,0x9369D7,
  141. 0x93250D,0x92E050,0x929BA0,0x9256FD,0x921266,0x91CDDD,0x91895F,0x9144EF,0x91008B,0x90BC34,0x9077EA,
  142. 0x9033AC,0x8FEF7B,0x8FAB56,0x8F673E,0x8F2333,0x8EDF34,0x8E9B41,0x8E575B,0x8E1382,0x8DCFB5,0x8D8BF4,
  143. 0x8D4840,0x8D0498,0x8CC0FD,0x8C7D6E,0x8C39EB,0x8BF675,0x8BB30B,0x8B6FAD,0x8B2C5B,0x8AE916,0x8AA5DD,
  144. 0x8A62B0,0x8A1F90,0x89DC7C,0x899973,0x895677,0x891388,0x88D0A4,0x888DCC,0x884B01,0x880841,0x87C58E,
  145. 0x8782E6,0x87404B,0x86FDBC,0x86BB38,0x8678C1,0x863655,0x85F3F6,0x85B1A2,0x856F5B,0x852D1F,0x84EAEF,
  146. 0x84A8CB,0x8466B3,0x8424A6,0x83E2A6,0x83A0B1,0x835EC8,0x831CEA,0x82DB19,0x829953,0x825799,0x8215EA,
  147. 0x81D448,0x8192B0,0x815125,0x810FA5,0x80CE31,0x808CC8,0x804B6B,0x800A1A,0x7FC8D4,0x7F8799,0x7F466A,
  148. 0x7F0547,0x7EC42F,0x7E8322,0x7E4221,0x7E012C,0x7DC041,0x7D7F63,0x7D3E8F,0x7CFDC7,0x7CBD0B,0x7C7C59,
  149. 0x7C3BB3,0x7BFB18,0x7BBA89,0x7B7A05,0x7B398C,0x7AF91E,0x7AB8BC,0x7A7865,0x7A3819,0x79F7D8,0x79B7A2,
  150. 0x797778,0x793759,0x78F744,0x78B73B,0x78773D,0x78374A,0x77F762,0x77B786,0x7777B4,0x7737ED,0x76F831,
  151. 0x76B881,0x7678DB,0x763940,0x75F9B0,0x75BA2B,0x757AB1,0x753B42,0x74FBDE,0x74BC84,0x747D36,0x743DF2,
  152. 0x73FEBA,0x73BF8C,0x738068,0x734150,0x730242,0x72C33F,0x728447,0x72455A,0x720677,0x71C79F,0x7188D2,
  153. 0x714A0F,0x710B57,0x70CCAA,0x708E07,0x704F6F,0x7010E1,0x6FD25E,0x6F93E6,0x6F5578,0x6F1715,0x6ED8BC,
  154. 0x6E9A6E,0x6E5C2B,0x6E1DF1,0x6DDFC3,0x6DA19E,0x6D6385,0x6D2575,0x6CE770,0x6CA976,0x6C6B86,0x6C2DA0,
  155. 0x6BEFC4,0x6BB1F3,0x6B742D,0x6B3670,0x6AF8BE,0x6ABB16,0x6A7D79,0x6A3FE6,0x6A025D,0x69C4DE,0x69876A,
  156. 0x6949FF,0x690C9F,0x68CF49,0x6891FE,0x6854BC,0x681785,0x67DA58,0x679D35,0x67601C,0x67230D,0x66E608,
  157. 0x66A90D,0x666C1D,0x662F36,0x65F25A,0x65B587,0x6578BE,0x653C00,0x64FF4B,0x64C2A1,0x648600,0x644969,
  158. 0x640CDD,0x63D05A,0x6393E1,0x635772,0x631B0D,0x62DEB2,0x62A260,0x626619,0x6229DB,0x61EDA7,0x61B17D,
  159. 0x61755D,0x613946,0x60FD39,0x60C136,0x60853D,0x60494D,0x600D68,0x5FD18C,0x5F95B9,0x5F59F0,0x5F1E31,
  160. 0x5EE27C,0x5EA6D0,0x5E6B2E,0x5E2F96,0x5DF407,0x5DB882,0x5D7D06,0x5D4194,0x5D062B,0x5CCACC,0x5C8F77,
  161. 0x5C542B,0x5C18E8,0x5BDDAF,0x5BA280,0x5B675A,0x5B2C3E,0x5AF12B,0x5AB621,0x5A7B21,0x5A402A,0x5A053D,
  162. 0x59CA59,0x598F7E,0x5954AD,0x5919E5,0x58DF27,0x58A472,0x5869C6,0x582F23,0x57F48A,0x57B9FA,0x577F74,
  163. 0x5744F6,0x570A82,0x56D018,0x5695B6,0x565B5E,0x56210E,0x55E6C8,0x55AC8C,0x557258,0x55382E,0x54FE0C,
  164. 0x54C3F4,0x5489E5,0x544FDF,0x5415E2,0x53DBEF,0x53A204,0x536823,0x532E4A,0x52F47B,0x52BAB5,0x5280F7,
  165. 0x524743,0x520D98,0x51D3F5,0x519A5C,0x5160CC,0x512744,0x50EDC6,0x50B450,0x507AE4,0x504180,0x500826,
  166. 0x4FCED4,0x4F958B,0x4F5C4B,0x4F2314,0x4EE9E6,0x4EB0C0,0x4E77A4,0x4E3E90,0x4E0585,0x4DCC83,0x4D9389,
  167. 0x4D5A99,0x4D21B1,0x4CE8D2,0x4CAFFC,0x4C772E,0x4C3E69,0x4C05AD,0x4BCCFA,0x4B944F,0x4B5BAD,0x4B2314,
  168. 0x4AEA83,0x4AB1FB,0x4A797C,0x4A4105,0x4A0897,0x49D031,0x4997D5,0x495F80,0x492735,0x48EEF2,0x48B6B7,
  169. 0x487E85,0x48465C,0x480E3B,0x47D622,0x479E13,0x47660B,0x472E0C,0x46F616,0x46BE28,0x468643,0x464E66,
  170. 0x461691,0x45DEC5,0x45A701,0x456F46,0x453793,0x44FFE9,0x44C847,0x4490AD,0x44591C,0x442193,0x43EA12,
  171. 0x43B29A,0x437B2A,0x4343C2,0x430C63,0x42D50C,0x429DBD,0x426676,0x422F38,0x41F802,0x41C0D5,0x4189AF,
  172. 0x415292,0x411B7D,0x40E470,0x40AD6B,0x40766F,0x403F7B,0x40088F,0x3FD1AB,0x3F9ACF,0x3F63FB,0x3F2D30,
  173. 0x3EF66D,0x3EBFB1,0x3E88FE,0x3E5253,0x3E1BB0,0x3DE515,0x3DAE83,0x3D77F8,0x3D4175,0x3D0AFA,0x3CD488,
  174. 0x3C9E1D,0x3C67BB,0x3C3160,0x3BFB0E,0x3BC4C3,0x3B8E80,0x3B5846,0x3B2213,0x3AEBE8,0x3AB5C5,0x3A7FAB,
  175. 0x3A4998,0x3A138D,0x39DD89,0x39A78E,0x39719B,0x393BAF,0x3905CC,0x38CFF0,0x389A1C,0x386450,0x382E8C,
  176. 0x37F8CF,0x37C31B,0x378D6E,0x3757C9,0x37222C,0x36EC96,0x36B709,0x368183,0x364C05,0x36168E,0x35E11F,
  177. 0x35ABB9,0x357659,0x354102,0x350BB2,0x34D66A,0x34A129,0x346BF1,0x3436C0,0x340196,0x33CC74,0x33975A,
  178. 0x336248,0x332D3D,0x32F839,0x32C33E,0x328E4A,0x32595D,0x322478,0x31EF9B,0x31BAC5,0x3185F7,0x315130,
  179. 0x311C71,0x30E7B9,0x30B309,0x307E61,0x3049C0,0x301526,0x2FE094,0x2FAC09,0x2F7786,0x2F430A,0x2F0E96,
  180. 0x2EDA29,0x2EA5C4,0x2E7166,0x2E3D0F,0x2E08C0,0x2DD479,0x2DA038,0x2D6BFF,0x2D37CE,0x2D03A4,0x2CCF81,
  181. 0x2C9B66,0x2C6751,0x2C3345,0x2BFF3F,0x2BCB41,0x2B974A,0x2B635B,0x2B2F73,0x2AFB92,0x2AC7B8,0x2A93E6,
  182. 0x2A601B,0x2A2C57,0x29F89B,0x29C4E5,0x299137,0x295D90,0x2929F1,0x28F658,0x28C2C7,0x288F3D,0x285BBA,
  183. 0x28283F,0x27F4CA,0x27C15D,0x278DF7,0x275A98,0x272740,0x26F3EF,0x26C0A6,0x268D63,0x265A28,0x2626F4,
  184. 0x25F3C6,0x25C0A0,0x258D81,0x255A69,0x252758,0x24F44F,0x24C14C,0x248E50,0x245B5C,0x24286E,0x23F587,
  185. 0x23C2A8,0x238FCF,0x235CFD,0x232A33,0x22F76F,0x22C4B2,0x2291FD,0x225F4E,0x222CA6,0x21FA05,0x21C76B,
  186. 0x2194D8,0x21624C,0x212FC7,0x20FD49,0x20CAD1,0x209861,0x2065F7,0x203395,0x200139,0x1FCEE4,0x1F9C96,
  187. 0x1F6A4F,0x1F380E,0x1F05D5,0x1ED3A2,0x1EA176,0x1E6F51,0x1E3D32,0x1E0B1B,0x1DD90A,0x1DA700,0x1D74FD,
  188. 0x1D4301,0x1D110B,0x1CDF1C,0x1CAD34,0x1C7B53,0x1C4978,0x1C17A4,0x1BE5D7,0x1BB410,0x1B8250,0x1B5097,
  189. 0x1B1EE5,0x1AED39,0x1ABB94,0x1A89F6,0x1A585E,0x1A26CD,0x19F542,0x19C3BF,0x199241,0x1960CB,0x192F5B,
  190. 0x18FDF2,0x18CC8F,0x189B33,0x1869DE,0x18388F,0x180746,0x17D605,0x17A4C9,0x177395,0x174267,0x17113F,
  191. 0x16E01E,0x16AF04,0x167DF0,0x164CE2,0x161BDC,0x15EADB,0x15B9E1,0x1588EE,0x155801,0x15271A,0x14F63A,
  192. 0x14C561,0x14948E,0x1463C1,0x1432FB,0x14023B,0x13D182,0x13A0CF,0x137023,0x133F7D,0x130EDD,0x12DE44,
  193. 0x12ADB1,0x127D25,0x124C9F,0x121C1F,0x11EBA6,0x11BB33,0x118AC6,0x115A60,0x112A00,0x10F9A6,0x10C953,
  194. 0x109906,0x1068BF,0x10387F,0x100845,0xFD811,0xFA7E4,0xF77BD,0xF479C,0xF1781,0xEE76D,0xEB75F,
  195. 0xE8757,0xE5755,0xE275A,0xDF765,0xDC776,0xD978D,0xD67AA,0xD37CE,0xD07F8,0xCD828,0xCA85E,
  196. 0xC789B,0xC48DD,0xC1926,0xBE975,0xBB9CA,0xB8A26,0xB5A87,0xB2AEF,0xAFB5C,0xACBD0,0xA9C4A,
  197. 0xA6CCA,0xA3D50,0xA0DDC,0x9DE6F,0x9AF07,0x97FA6,0x9504A,0x920F5,0x8F1A6,0x8C25C,0x89319,
  198. 0x863DC,0x834A5,0x80574,0x7D649,0x7A724,0x77805,0x748EC,0x719D9,0x6EACC,0x6BBC5,0x68CC4,
  199. 0x65DC9,0x62ED4,0x5FFE5,0x5D0FC,0x5A218,0x5733B,0x54464,0x51593,0x4E6C7,0x4B802,0x48942,
  200. 0x45A88,0x42BD5,0x3FD27,0x3CE7F,0x39FDD,0x37141,0x342AA,0x3141A,0x2E58F,0x2B70B,0x2888C,
  201. 0x25A13,0x22BA0,0x1FD33,0x1CECB,0x1A069,0x1720E,0x143B8,0x11567,0xE71D,0xB8D8,0x8A9A,
  202. 0x5C61,0x2E2D
  203. };
  204. #define LZHAM_CREATE_PROB_COST_TABLE 0
  205. #if LZHAM_CREATE_PROB_COST_TABLE
  206. class arith_prob_cost_initializer
  207. {
  208. public:
  209. arith_prob_cost_initializer()
  210. {
  211. const double cInvLn2 = 1.4426950408889634073599246810019; // 1.0/ln(2)
  212. for (uint i = 0; i < cSymbolCodecArithProbScale; i++)
  213. {
  214. double flBits = i ? (-log(i * (1.0 / cSymbolCodecArithProbScale)) * cInvLn2) : 0;
  215. g_prob_cost[i] = static_cast<uint32>(floor(.5f + flBits * cBitCostScale));
  216. printf("0x%X,", g_prob_cost[i]);
  217. if ((i % 11) == 10) printf("\n");
  218. }
  219. printf("\n");
  220. }
  221. };
  222. static arith_prob_cost_initializer g_prob_cost_initializer;
  223. #endif
  224. raw_quasi_adaptive_huffman_data_model::raw_quasi_adaptive_huffman_data_model(bool encoding, uint total_syms, uint max_update_interval, uint adapt_rate) :
  225. m_pDecode_tables(NULL),
  226. m_total_syms(0),
  227. m_max_cycle(0),
  228. m_update_cycle(0),
  229. m_symbols_until_update(0),
  230. m_total_count(0),
  231. m_decoder_table_bits(0),
  232. m_max_update_interval(static_cast<uint16>(max_update_interval)),
  233. m_adapt_rate(static_cast<uint16>(adapt_rate)),
  234. m_encoding(encoding)
  235. {
  236. if (total_syms)
  237. {
  238. init2(encoding, total_syms, max_update_interval, adapt_rate, NULL);
  239. }
  240. }
  241. raw_quasi_adaptive_huffman_data_model::raw_quasi_adaptive_huffman_data_model(const raw_quasi_adaptive_huffman_data_model& other) :
  242. m_pDecode_tables(NULL),
  243. m_total_syms(0),
  244. m_max_cycle(0),
  245. m_update_cycle(0),
  246. m_symbols_until_update(0),
  247. m_total_count(0),
  248. m_decoder_table_bits(0),
  249. m_max_update_interval(0),
  250. m_adapt_rate(0),
  251. m_encoding(false)
  252. {
  253. *this = other;
  254. }
  255. raw_quasi_adaptive_huffman_data_model::~raw_quasi_adaptive_huffman_data_model()
  256. {
  257. if (m_pDecode_tables)
  258. lzham_delete(m_pDecode_tables);
  259. }
  260. bool raw_quasi_adaptive_huffman_data_model::assign(const raw_quasi_adaptive_huffman_data_model& rhs)
  261. {
  262. if (this == &rhs)
  263. return true;
  264. m_total_syms = rhs.m_total_syms;
  265. m_max_cycle = rhs.m_max_cycle;
  266. m_update_cycle = rhs.m_update_cycle;
  267. m_symbols_until_update = rhs.m_symbols_until_update;
  268. m_total_count = rhs.m_total_count;
  269. m_sym_freq = rhs.m_sym_freq;
  270. m_initial_sym_freq = rhs.m_initial_sym_freq;
  271. m_codes = rhs.m_codes;
  272. m_code_sizes = rhs.m_code_sizes;
  273. if (rhs.m_pDecode_tables)
  274. {
  275. if (m_pDecode_tables)
  276. {
  277. if (!m_pDecode_tables->assign(*rhs.m_pDecode_tables))
  278. {
  279. clear();
  280. return false;
  281. }
  282. }
  283. else
  284. {
  285. m_pDecode_tables = lzham_new<prefix_coding::decoder_tables>(*rhs.m_pDecode_tables);
  286. if (!m_pDecode_tables)
  287. {
  288. clear();
  289. return false;
  290. }
  291. }
  292. }
  293. else if (m_pDecode_tables)
  294. {
  295. lzham_delete(m_pDecode_tables);
  296. m_pDecode_tables = NULL;
  297. }
  298. m_decoder_table_bits = rhs.m_decoder_table_bits;
  299. m_encoding = rhs.m_encoding;
  300. m_max_update_interval = rhs.m_max_update_interval;
  301. m_adapt_rate = rhs.m_adapt_rate;
  302. return true;
  303. }
  304. raw_quasi_adaptive_huffman_data_model& raw_quasi_adaptive_huffman_data_model::operator= (const raw_quasi_adaptive_huffman_data_model& rhs)
  305. {
  306. assign(rhs);
  307. return *this;
  308. }
  309. void raw_quasi_adaptive_huffman_data_model::clear()
  310. {
  311. m_sym_freq.clear();
  312. m_initial_sym_freq.clear();
  313. m_codes.clear();
  314. m_code_sizes.clear();
  315. m_max_cycle = 0;
  316. m_total_syms = 0;
  317. m_update_cycle = 0;
  318. m_symbols_until_update = 0;
  319. m_decoder_table_bits = 0;
  320. m_total_count = 0;
  321. if (m_pDecode_tables)
  322. {
  323. lzham_delete(m_pDecode_tables);
  324. m_pDecode_tables = NULL;
  325. }
  326. m_max_update_interval = 0;
  327. m_adapt_rate = 0;
  328. }
  329. bool raw_quasi_adaptive_huffman_data_model::init2(bool encoding, uint total_syms, uint max_update_interval, uint adapt_rate, const uint16 *pInitial_sym_freq)
  330. {
  331. LZHAM_ASSERT(max_update_interval <= 0xFFFF);
  332. LZHAM_ASSERT(adapt_rate <= 0xFFFF);
  333. m_encoding = encoding;
  334. m_max_update_interval = static_cast<uint16>(max_update_interval);
  335. m_adapt_rate = static_cast<uint16>(adapt_rate);
  336. m_symbols_until_update = 0;
  337. if (!m_sym_freq.try_resize(total_syms))
  338. {
  339. clear();
  340. return false;
  341. }
  342. if (pInitial_sym_freq)
  343. {
  344. if (!m_initial_sym_freq.try_resize(total_syms))
  345. {
  346. clear();
  347. return false;
  348. }
  349. memcpy(m_initial_sym_freq.begin(), pInitial_sym_freq, total_syms * m_initial_sym_freq.size_in_bytes());
  350. }
  351. if (!m_code_sizes.try_resize(total_syms))
  352. {
  353. clear();
  354. return false;
  355. }
  356. m_total_syms = total_syms;
  357. uint max_table_bits = 0;
  358. if (m_total_syms <= 8)
  359. max_table_bits = 4;
  360. else
  361. max_table_bits = 1 + math::ceil_log2i(m_total_syms);
  362. m_decoder_table_bits = static_cast<uint8>(math::minimum(max_table_bits, prefix_coding::cMaxTableBits));
  363. if (m_encoding)
  364. {
  365. lzham_delete(m_pDecode_tables);
  366. m_pDecode_tables = NULL;
  367. if (!m_codes.try_resize(total_syms))
  368. {
  369. clear();
  370. return false;
  371. }
  372. }
  373. else if (!m_pDecode_tables)
  374. {
  375. m_pDecode_tables = lzham_new<prefix_coding::decoder_tables>();
  376. if (!m_pDecode_tables)
  377. {
  378. clear();
  379. return false;
  380. }
  381. }
  382. m_max_cycle = (LZHAM_MAX(24, m_total_syms) + 6) * (m_max_update_interval ? m_max_update_interval : LZHAM_DEFAULT_MAX_UPDATE_INTERVAL);
  383. m_max_cycle = LZHAM_MIN(m_max_cycle, 32767);
  384. reset();
  385. return true;
  386. }
  387. bool raw_quasi_adaptive_huffman_data_model::reset()
  388. {
  389. if (!m_total_syms)
  390. return true;
  391. bool sym_freq_all_ones = false;
  392. if (m_initial_sym_freq.size())
  393. {
  394. m_update_cycle = 0;
  395. for (uint i = 0; i < m_total_syms; i++)
  396. {
  397. uint sym_freq = m_initial_sym_freq[i];
  398. m_sym_freq[i] = static_cast<uint16>(sym_freq);
  399. // Slam m_update_cycle to a specific value so update_tables() sets m_total_count to the proper value
  400. m_update_cycle += sym_freq;
  401. }
  402. }
  403. else
  404. {
  405. for (uint i = 0; i < m_total_syms; i++)
  406. m_sym_freq[i] = 1;
  407. // Slam m_update_cycle to a specific value so update_tables() sets m_total_count to the proper value
  408. m_update_cycle = m_total_syms;
  409. sym_freq_all_ones = true;
  410. }
  411. m_total_count = 0;
  412. m_symbols_until_update = 0;
  413. if (!update_tables(LZHAM_MIN(m_max_cycle, 16), sym_freq_all_ones)) // this was 8 in the alphas
  414. return false;
  415. return true;
  416. }
  417. void raw_quasi_adaptive_huffman_data_model::rescale()
  418. {
  419. uint total_freq = 0;
  420. for (uint i = 0; i < m_total_syms; i++)
  421. {
  422. uint freq = (m_sym_freq[i] + 1) >> 1;
  423. total_freq += freq;
  424. m_sym_freq[i] = static_cast<uint16>(freq);
  425. }
  426. m_total_count = total_freq;
  427. }
  428. void raw_quasi_adaptive_huffman_data_model::reset_update_rate()
  429. {
  430. m_total_count += (m_update_cycle - m_symbols_until_update);
  431. #ifdef _DEBUG
  432. uint actual_total = 0;
  433. for (uint i = 0; i < m_sym_freq.size(); i++)
  434. actual_total += m_sym_freq[i];
  435. LZHAM_ASSERT(actual_total == m_total_count);
  436. #endif
  437. if (m_total_count > m_total_syms)
  438. rescale();
  439. m_symbols_until_update = m_update_cycle = LZHAM_MIN(8, m_update_cycle);
  440. }
  441. bool raw_quasi_adaptive_huffman_data_model::update_tables(int force_update_cycle, bool sym_freq_all_ones)
  442. {
  443. LZHAM_ASSERT(!m_symbols_until_update);
  444. m_total_count += m_update_cycle;
  445. LZHAM_ASSERT(m_total_count <= 65535);
  446. while (m_total_count >= 32768)
  447. rescale();
  448. uint max_code_size = 0;
  449. if ((sym_freq_all_ones) && (m_total_syms >= 2))
  450. {
  451. // Shortcut building the Huffman codes if we know all the sym freqs are 1.
  452. uint base_code_size = math::floor_log2i(m_total_syms);
  453. uint num_left = m_total_syms - (1 << base_code_size);
  454. num_left *= 2;
  455. if (num_left > m_total_syms)
  456. num_left = m_total_syms;
  457. memset(&m_code_sizes[0], base_code_size + 1, num_left);
  458. memset(&m_code_sizes[num_left], base_code_size, m_total_syms - num_left);
  459. max_code_size = base_code_size + (num_left ? 1 : 0);
  460. }
  461. bool status = false;
  462. if (!max_code_size)
  463. {
  464. uint table_size = get_generate_huffman_codes_table_size();
  465. void *pTables = alloca(table_size);
  466. uint total_freq = 0;
  467. status = generate_huffman_codes(pTables, m_total_syms, &m_sym_freq[0], &m_code_sizes[0], max_code_size, total_freq);
  468. LZHAM_ASSERT(status);
  469. LZHAM_ASSERT(total_freq == m_total_count);
  470. if ((!status) || (total_freq != m_total_count))
  471. return false;
  472. if (max_code_size > prefix_coding::cMaxExpectedCodeSize)
  473. {
  474. status = prefix_coding::limit_max_code_size(m_total_syms, &m_code_sizes[0], prefix_coding::cMaxExpectedCodeSize);
  475. LZHAM_ASSERT(status);
  476. if (!status)
  477. return false;
  478. }
  479. }
  480. if (force_update_cycle >= 0)
  481. m_symbols_until_update = m_update_cycle = force_update_cycle;
  482. else
  483. {
  484. m_update_cycle = (31U + m_update_cycle * LZHAM_MAX(32U, (m_adapt_rate ? m_adapt_rate : LZHAM_DEFAULT_ADAPT_RATE))) >> 5U;
  485. if (m_update_cycle > m_max_cycle)
  486. m_update_cycle = m_max_cycle;
  487. m_symbols_until_update = m_update_cycle;
  488. }
  489. if (m_encoding)
  490. status = prefix_coding::generate_codes(m_total_syms, &m_code_sizes[0], &m_codes[0]);
  491. else
  492. {
  493. uint actual_table_bits = m_decoder_table_bits;
  494. // Try to see if using the accel table is actually worth the trouble of constructing it.
  495. uint cost_to_use_table = (1 << actual_table_bits) + 64;
  496. uint cost_to_not_use_table = m_symbols_until_update * math::floor_log2i(m_total_syms);
  497. if (cost_to_not_use_table <= cost_to_use_table)
  498. actual_table_bits = 0;
  499. status = prefix_coding::generate_decoder_tables(m_total_syms, &m_code_sizes[0], m_pDecode_tables, actual_table_bits);
  500. }
  501. LZHAM_ASSERT(status);
  502. if (!status)
  503. return false;
  504. return true;
  505. }
  506. bool raw_quasi_adaptive_huffman_data_model::update_sym(uint sym)
  507. {
  508. uint freq = m_sym_freq[sym];
  509. freq++;
  510. m_sym_freq[sym] = static_cast<uint16>(freq);
  511. LZHAM_ASSERT(freq <= UINT16_MAX);
  512. if (--m_symbols_until_update == 0)
  513. {
  514. if (!update_tables())
  515. return false;
  516. }
  517. return true;
  518. }
  519. adaptive_bit_model::adaptive_bit_model(float prob0)
  520. {
  521. set_probability_0(prob0);
  522. }
  523. adaptive_bit_model::adaptive_bit_model(const adaptive_bit_model& other) :
  524. m_bit_0_prob(other.m_bit_0_prob)
  525. {
  526. }
  527. void adaptive_bit_model::set_probability_0(float prob0)
  528. {
  529. m_bit_0_prob = static_cast<uint16>(math::clamp<uint>((uint)(prob0 * cSymbolCodecArithProbScale), 1, cSymbolCodecArithProbScale - 1));
  530. }
  531. adaptive_arith_data_model::adaptive_arith_data_model(bool encoding, uint total_syms)
  532. {
  533. init(encoding, total_syms);
  534. }
  535. adaptive_arith_data_model::adaptive_arith_data_model(const adaptive_arith_data_model& other)
  536. {
  537. m_total_syms = other.m_total_syms;
  538. m_probs = other.m_probs;
  539. }
  540. adaptive_arith_data_model::~adaptive_arith_data_model()
  541. {
  542. }
  543. adaptive_arith_data_model& adaptive_arith_data_model::operator= (const adaptive_arith_data_model& rhs)
  544. {
  545. m_total_syms = rhs.m_total_syms;
  546. m_probs = rhs.m_probs;
  547. return *this;
  548. }
  549. void adaptive_arith_data_model::clear()
  550. {
  551. m_total_syms = 0;
  552. m_probs.clear();
  553. }
  554. bool adaptive_arith_data_model::init(bool encoding, uint total_syms)
  555. {
  556. LZHAM_NOTE_UNUSED(encoding);
  557. if (!total_syms)
  558. {
  559. clear();
  560. return true;
  561. }
  562. if ((total_syms < 2) || (!math::is_power_of_2(total_syms)))
  563. total_syms = math::next_pow2(total_syms);
  564. m_total_syms = total_syms;
  565. if (!m_probs.try_resize(m_total_syms))
  566. return false;
  567. return true;
  568. }
  569. void adaptive_arith_data_model::reset()
  570. {
  571. for (uint i = 0; i < m_probs.size(); i++)
  572. m_probs[i].clear();
  573. }
  574. void adaptive_arith_data_model::reset_update_rate()
  575. {
  576. }
  577. bool adaptive_arith_data_model::update(uint sym)
  578. {
  579. uint node = 1;
  580. uint bitmask = m_total_syms;
  581. do
  582. {
  583. bitmask >>= 1;
  584. uint bit = (sym & bitmask) ? 1 : 0;
  585. m_probs[node].update(bit);
  586. node = (node << 1) + bit;
  587. } while (bitmask > 1);
  588. return true;
  589. }
  590. bit_cost_t adaptive_arith_data_model::get_cost(uint sym) const
  591. {
  592. uint node = 1;
  593. uint bitmask = m_total_syms;
  594. bit_cost_t cost = 0;
  595. do
  596. {
  597. bitmask >>= 1;
  598. uint bit = (sym & bitmask) ? 1 : 0;
  599. cost += m_probs[node].get_cost(bit);
  600. node = (node << 1) + bit;
  601. } while (bitmask > 1);
  602. return cost;
  603. }
  604. symbol_codec::symbol_codec()
  605. {
  606. clear();
  607. }
  608. void symbol_codec::reset()
  609. {
  610. m_pDecode_buf = NULL;
  611. m_pDecode_buf_next = NULL;
  612. m_pDecode_buf_end = NULL;
  613. m_decode_buf_size = 0;
  614. m_bit_buf = 0;
  615. m_bit_count = 0;
  616. m_total_model_updates = 0;
  617. m_mode = cNull;
  618. m_total_bits_written = 0;
  619. m_arith_base = 0;
  620. m_arith_value = 0;
  621. m_arith_length = 0;
  622. m_arith_total_bits = 0;
  623. m_output_buf.try_resize(0);
  624. m_arith_output_buf.try_resize(0);
  625. m_output_syms.try_resize(0);
  626. m_pDecode_need_bytes_func = NULL;
  627. m_pDecode_private_data = NULL;
  628. m_pSaved_huff_model = NULL;
  629. m_pSaved_model = NULL;
  630. m_saved_node_index = 0;
  631. }
  632. void symbol_codec::clear()
  633. {
  634. reset();
  635. m_output_buf.clear();
  636. m_arith_output_buf.clear();
  637. m_output_syms.clear();
  638. }
  639. bool symbol_codec::start_encoding(uint expected_file_size)
  640. {
  641. m_mode = cEncoding;
  642. m_total_model_updates = 0;
  643. m_total_bits_written = 0;
  644. if (!put_bits_init(expected_file_size))
  645. return false;
  646. m_output_syms.try_resize(0);
  647. arith_start_encoding();
  648. return true;
  649. }
  650. bool symbol_codec::encode_bits(uint bits, uint num_bits)
  651. {
  652. LZHAM_ASSERT(m_mode == cEncoding);
  653. if (!num_bits)
  654. return true;
  655. LZHAM_ASSERT((num_bits == 32) || (bits <= ((1U << num_bits) - 1)));
  656. if (num_bits > 16)
  657. {
  658. if (!record_put_bits(bits >> 16, num_bits - 16))
  659. return false;
  660. if (!record_put_bits(bits & 0xFFFF, 16))
  661. return false;
  662. }
  663. else
  664. {
  665. if (!record_put_bits(bits, num_bits))
  666. return false;
  667. }
  668. return true;
  669. }
  670. bool symbol_codec::encode_arith_init()
  671. {
  672. LZHAM_ASSERT(m_mode == cEncoding);
  673. output_symbol sym;
  674. sym.m_bits = 0;
  675. sym.m_num_bits = output_symbol::cArithInit;
  676. sym.m_arith_prob0 = 0;
  677. if (!m_output_syms.try_push_back(sym))
  678. return false;
  679. return true;
  680. }
  681. bool symbol_codec::encode_align_to_byte()
  682. {
  683. LZHAM_ASSERT(m_mode == cEncoding);
  684. output_symbol sym;
  685. sym.m_bits = 0;
  686. sym.m_num_bits = output_symbol::cAlignToByteSym;
  687. sym.m_arith_prob0 = 0;
  688. if (!m_output_syms.try_push_back(sym))
  689. return false;
  690. return true;
  691. }
  692. bool symbol_codec::encode(uint sym, quasi_adaptive_huffman_data_model& model)
  693. {
  694. LZHAM_ASSERT(m_mode == cEncoding);
  695. LZHAM_ASSERT(model.m_encoding);
  696. if (!record_put_bits(model.m_codes[sym], model.m_code_sizes[sym]))
  697. return false;
  698. uint freq = model.m_sym_freq[sym];
  699. freq++;
  700. model.m_sym_freq[sym] = static_cast<uint16>(freq);
  701. LZHAM_ASSERT(freq <= UINT16_MAX);
  702. if (--model.m_symbols_until_update == 0)
  703. {
  704. m_total_model_updates++;
  705. if (!model.update_tables())
  706. return false;
  707. }
  708. return true;
  709. }
  710. void symbol_codec::arith_propagate_carry()
  711. {
  712. int index = m_arith_output_buf.size() - 1;
  713. while (index >= 0)
  714. {
  715. uint c = m_arith_output_buf[index];
  716. if (c == 0xFF)
  717. m_arith_output_buf[index] = 0;
  718. else
  719. {
  720. m_arith_output_buf[index]++;
  721. break;
  722. }
  723. index--;
  724. }
  725. }
  726. bool symbol_codec::arith_renorm_enc_interval()
  727. {
  728. do
  729. {
  730. if (!m_arith_output_buf.try_push_back((m_arith_base >> 24) & 0xFF))
  731. return false;
  732. m_total_bits_written += 8;
  733. m_arith_base <<= 8;
  734. } while ((m_arith_length <<= 8) < cSymbolCodecArithMinLen);
  735. return true;
  736. }
  737. void symbol_codec::arith_start_encoding()
  738. {
  739. m_arith_output_buf.try_resize(0);
  740. m_arith_base = 0;
  741. m_arith_value = 0;
  742. m_arith_length = cSymbolCodecArithMaxLen;
  743. m_arith_total_bits = 0;
  744. }
  745. bool symbol_codec::encode(uint bit, adaptive_bit_model& model, bool update_model)
  746. {
  747. LZHAM_ASSERT(m_mode == cEncoding);
  748. m_arith_total_bits++;
  749. output_symbol sym;
  750. sym.m_bits = bit;
  751. sym.m_num_bits = -1;
  752. sym.m_arith_prob0 = model.m_bit_0_prob;
  753. if (!m_output_syms.try_push_back(sym))
  754. return false;
  755. uint x = model.m_bit_0_prob * (m_arith_length >> cSymbolCodecArithProbBits);
  756. if (!bit)
  757. {
  758. if (update_model)
  759. model.m_bit_0_prob += ((cSymbolCodecArithProbScale - model.m_bit_0_prob) >> cSymbolCodecArithProbMoveBits);
  760. m_arith_length = x;
  761. }
  762. else
  763. {
  764. if (update_model)
  765. model.m_bit_0_prob -= (model.m_bit_0_prob >> cSymbolCodecArithProbMoveBits);
  766. uint orig_base = m_arith_base;
  767. m_arith_base += x;
  768. m_arith_length -= x;
  769. if (orig_base > m_arith_base)
  770. arith_propagate_carry();
  771. }
  772. if (m_arith_length < cSymbolCodecArithMinLen)
  773. {
  774. if (!arith_renorm_enc_interval())
  775. return false;
  776. }
  777. return true;
  778. }
  779. bool symbol_codec::encode(uint sym, adaptive_arith_data_model& model)
  780. {
  781. uint node = 1;
  782. uint bitmask = model.m_total_syms;
  783. do
  784. {
  785. bitmask >>= 1;
  786. uint bit = (sym & bitmask) ? 1 : 0;
  787. if (!encode(bit, model.m_probs[node]))
  788. return false;
  789. node = (node << 1) + bit;
  790. } while (bitmask > 1);
  791. return true;
  792. }
  793. bool symbol_codec::arith_stop_encoding()
  794. {
  795. uint orig_base = m_arith_base;
  796. if (m_arith_length > 2 * cSymbolCodecArithMinLen)
  797. {
  798. m_arith_base += cSymbolCodecArithMinLen;
  799. m_arith_length = (cSymbolCodecArithMinLen >> 1);
  800. }
  801. else
  802. {
  803. m_arith_base += (cSymbolCodecArithMinLen >> 1);
  804. m_arith_length = (cSymbolCodecArithMinLen >> 9);
  805. }
  806. if (orig_base > m_arith_base)
  807. arith_propagate_carry();
  808. if (!arith_renorm_enc_interval())
  809. return false;
  810. while (m_arith_output_buf.size() < 4)
  811. {
  812. if (!m_arith_output_buf.try_push_back(0))
  813. return false;
  814. m_total_bits_written += 8;
  815. }
  816. return true;
  817. }
  818. bool symbol_codec::stop_encoding(bool support_arith)
  819. {
  820. LZHAM_ASSERT(m_mode == cEncoding);
  821. if (support_arith)
  822. {
  823. if (!arith_stop_encoding())
  824. return false;
  825. }
  826. if (!assemble_output_buf())
  827. return false;
  828. m_mode = cNull;
  829. return true;
  830. }
  831. bool symbol_codec::record_put_bits(uint bits, uint num_bits)
  832. {
  833. LZHAM_ASSERT(m_mode == cEncoding);
  834. LZHAM_ASSERT(num_bits <= 25);
  835. LZHAM_ASSERT(m_bit_count >= 25);
  836. if (!num_bits)
  837. return true;
  838. m_total_bits_written += num_bits;
  839. output_symbol sym;
  840. sym.m_bits = bits;
  841. sym.m_num_bits = (uint16)num_bits;
  842. sym.m_arith_prob0 = 0;
  843. if (!m_output_syms.try_push_back(sym))
  844. return false;
  845. return true;
  846. }
  847. bool symbol_codec::put_bits_init(uint expected_size)
  848. {
  849. m_bit_buf = 0;
  850. m_bit_count = cBitBufSize;
  851. m_output_buf.try_resize(0);
  852. if (!m_output_buf.try_reserve(expected_size))
  853. return false;
  854. return true;
  855. }
  856. bool symbol_codec::put_bits(uint bits, uint num_bits)
  857. {
  858. LZHAM_ASSERT(num_bits <= 25);
  859. LZHAM_ASSERT(m_bit_count >= 25);
  860. if (!num_bits)
  861. return true;
  862. m_bit_count -= num_bits;
  863. m_bit_buf |= (static_cast<bit_buf_t>(bits) << m_bit_count);
  864. m_total_bits_written += num_bits;
  865. while (m_bit_count <= (cBitBufSize - 8))
  866. {
  867. if (!m_output_buf.try_push_back(static_cast<uint8>(m_bit_buf >> (cBitBufSize - 8))))
  868. return false;
  869. m_bit_buf <<= 8;
  870. m_bit_count += 8;
  871. }
  872. return true;
  873. }
  874. bool symbol_codec::put_bits_align_to_byte()
  875. {
  876. uint num_bits_in = cBitBufSize - m_bit_count;
  877. if (num_bits_in & 7)
  878. {
  879. if (!put_bits(0, 8 - (num_bits_in & 7)))
  880. return false;
  881. }
  882. return true;
  883. }
  884. bool symbol_codec::flush_bits()
  885. {
  886. return put_bits(0, 7); // to ensure the last bits are flushed
  887. }
  888. bool symbol_codec::assemble_output_buf()
  889. {
  890. m_total_bits_written = 0;
  891. uint arith_buf_ofs = 0;
  892. // Intermix the final Arithmetic, Huffman, or plain bits to a single combined bitstream.
  893. // All bits from each source must be output in exactly the same order that the decompressor will read them.
  894. for (uint sym_index = 0; sym_index < m_output_syms.size(); sym_index++)
  895. {
  896. const output_symbol& sym = m_output_syms[sym_index];
  897. if (sym.m_num_bits == output_symbol::cAlignToByteSym)
  898. {
  899. if (!put_bits_align_to_byte())
  900. return false;
  901. }
  902. else if (sym.m_num_bits == output_symbol::cArithInit)
  903. {
  904. LZHAM_ASSERT(m_arith_output_buf.size());
  905. if (m_arith_output_buf.size())
  906. {
  907. m_arith_length = cSymbolCodecArithMaxLen;
  908. m_arith_value = 0;
  909. for (uint i = 0; i < 4; i++)
  910. {
  911. const uint c = m_arith_output_buf[arith_buf_ofs++];
  912. m_arith_value = (m_arith_value << 8) | c;
  913. if (!put_bits(c, 8))
  914. return false;
  915. }
  916. }
  917. }
  918. else if (sym.m_num_bits == output_symbol::cArithSym)
  919. {
  920. // This renorm logic must match the logic used in the arithmetic decoder.
  921. if (m_arith_length < cSymbolCodecArithMinLen)
  922. {
  923. do
  924. {
  925. const uint c = (arith_buf_ofs < m_arith_output_buf.size()) ? m_arith_output_buf[arith_buf_ofs++] : 0;
  926. if (!put_bits(c, 8))
  927. return false;
  928. m_arith_value = (m_arith_value << 8) | c;
  929. } while ((m_arith_length <<= 8) < cSymbolCodecArithMinLen);
  930. }
  931. uint x = sym.m_arith_prob0 * (m_arith_length >> cSymbolCodecArithProbBits);
  932. uint bit = (m_arith_value >= x);
  933. if (bit == 0)
  934. {
  935. m_arith_length = x;
  936. }
  937. else
  938. {
  939. m_arith_value -= x;
  940. m_arith_length -= x;
  941. }
  942. LZHAM_VERIFY(bit == sym.m_bits);
  943. }
  944. else
  945. {
  946. // Huffman or plain bits
  947. if (!put_bits(sym.m_bits, sym.m_num_bits))
  948. return false;
  949. }
  950. }
  951. return flush_bits();
  952. }
  953. //------------------------------------------------------------------------------------------------------------------
  954. // Decoding
  955. //------------------------------------------------------------------------------------------------------------------
  956. bool symbol_codec::start_decoding(const uint8* pBuf, size_t buf_size, bool eof_flag, need_bytes_func_ptr pNeed_bytes_func, void *pPrivate_data)
  957. {
  958. if (!buf_size)
  959. return false;
  960. m_total_model_updates = 0;
  961. m_pDecode_buf = pBuf;
  962. m_pDecode_buf_next = pBuf;
  963. m_decode_buf_size = buf_size;
  964. m_pDecode_buf_end = pBuf + buf_size;
  965. m_pDecode_need_bytes_func = pNeed_bytes_func;
  966. m_pDecode_private_data = pPrivate_data;
  967. m_decode_buf_eof = eof_flag;
  968. m_bit_buf = 0;
  969. m_bit_count = 0;
  970. m_mode = cDecoding;
  971. return true;
  972. }
  973. uint symbol_codec::decode_bits(uint num_bits)
  974. {
  975. LZHAM_ASSERT(m_mode == cDecoding);
  976. if (!num_bits)
  977. return 0;
  978. if (num_bits > 16)
  979. {
  980. uint a = get_bits(num_bits - 16);
  981. uint b = get_bits(16);
  982. return (a << 16) | b;
  983. }
  984. else
  985. return get_bits(num_bits);
  986. }
  987. void symbol_codec::decode_remove_bits(uint num_bits)
  988. {
  989. LZHAM_ASSERT(m_mode == cDecoding);
  990. while (num_bits > 16)
  991. {
  992. remove_bits(16);
  993. num_bits -= 16;
  994. }
  995. remove_bits(num_bits);
  996. }
  997. uint symbol_codec::decode_peek_bits(uint num_bits)
  998. {
  999. LZHAM_ASSERT(m_mode == cDecoding);
  1000. LZHAM_ASSERT(num_bits <= 25);
  1001. if (!num_bits)
  1002. return 0;
  1003. while (m_bit_count < (int)num_bits)
  1004. {
  1005. uint c = 0;
  1006. if (m_pDecode_buf_next == m_pDecode_buf_end)
  1007. {
  1008. if (!m_decode_buf_eof)
  1009. {
  1010. m_pDecode_need_bytes_func(m_pDecode_buf_next - m_pDecode_buf, m_pDecode_private_data, m_pDecode_buf, m_decode_buf_size, m_decode_buf_eof);
  1011. m_pDecode_buf_end = m_pDecode_buf + m_decode_buf_size;
  1012. m_pDecode_buf_next = m_pDecode_buf;
  1013. if (m_pDecode_buf_next < m_pDecode_buf_end) c = *m_pDecode_buf_next++;
  1014. }
  1015. }
  1016. else
  1017. c = *m_pDecode_buf_next++;
  1018. m_bit_count += 8;
  1019. LZHAM_ASSERT(m_bit_count <= cBitBufSize);
  1020. m_bit_buf |= (static_cast<bit_buf_t>(c) << (cBitBufSize - m_bit_count));
  1021. }
  1022. return static_cast<uint>(m_bit_buf >> (cBitBufSize - num_bits));
  1023. }
  1024. uint symbol_codec::decode(quasi_adaptive_huffman_data_model& model)
  1025. {
  1026. LZHAM_ASSERT(m_mode == cDecoding);
  1027. LZHAM_ASSERT(!model.m_encoding);
  1028. const prefix_coding::decoder_tables* pTables = model.m_pDecode_tables;
  1029. while (m_bit_count < (cBitBufSize - 8))
  1030. {
  1031. uint c = 0;
  1032. if (m_pDecode_buf_next == m_pDecode_buf_end)
  1033. {
  1034. if (!m_decode_buf_eof)
  1035. {
  1036. m_pDecode_need_bytes_func(m_pDecode_buf_next - m_pDecode_buf, m_pDecode_private_data, m_pDecode_buf, m_decode_buf_size, m_decode_buf_eof);
  1037. m_pDecode_buf_end = m_pDecode_buf + m_decode_buf_size;
  1038. m_pDecode_buf_next = m_pDecode_buf;
  1039. if (m_pDecode_buf_next < m_pDecode_buf_end) c = *m_pDecode_buf_next++;
  1040. }
  1041. }
  1042. else
  1043. c = *m_pDecode_buf_next++;
  1044. m_bit_count += 8;
  1045. m_bit_buf |= (static_cast<bit_buf_t>(c) << (cBitBufSize - m_bit_count));
  1046. }
  1047. uint k = static_cast<uint>((m_bit_buf >> (cBitBufSize - 16)) + 1);
  1048. uint sym, len;
  1049. if (k <= pTables->m_table_max_code)
  1050. {
  1051. uint32 t = pTables->m_lookup[m_bit_buf >> (cBitBufSize - pTables->m_table_bits)];
  1052. LZHAM_ASSERT(t != UINT32_MAX);
  1053. sym = t & UINT16_MAX;
  1054. len = t >> 16;
  1055. LZHAM_ASSERT(model.m_code_sizes[sym] == len);
  1056. }
  1057. else
  1058. {
  1059. len = pTables->m_decode_start_code_size;
  1060. for ( ; ; )
  1061. {
  1062. if (k <= pTables->m_max_codes[len - 1])
  1063. break;
  1064. len++;
  1065. }
  1066. int val_ptr = pTables->m_val_ptrs[len - 1] + static_cast<int>((m_bit_buf >> (cBitBufSize - len)));
  1067. if (((uint)val_ptr >= model.m_total_syms))
  1068. {
  1069. // corrupted stream, or a bug
  1070. LZHAM_ASSERT(0);
  1071. return 0;
  1072. }
  1073. sym = pTables->m_sorted_symbol_order[val_ptr];
  1074. }
  1075. m_bit_buf <<= len;
  1076. m_bit_count -= len;
  1077. uint freq = model.m_sym_freq[sym];
  1078. freq++;
  1079. model.m_sym_freq[sym] = static_cast<uint16>(freq);
  1080. LZHAM_ASSERT(freq <= UINT16_MAX);
  1081. if (--model.m_symbols_until_update == 0)
  1082. {
  1083. m_total_model_updates++;
  1084. model.update_tables();
  1085. }
  1086. return sym;
  1087. }
  1088. uint64 symbol_codec::stop_decoding()
  1089. {
  1090. LZHAM_ASSERT(m_mode == cDecoding);
  1091. uint64 n = m_pDecode_buf_next - m_pDecode_buf;
  1092. m_mode = cNull;
  1093. return n;
  1094. }
  1095. uint symbol_codec::get_bits(uint num_bits)
  1096. {
  1097. LZHAM_ASSERT(num_bits <= 25);
  1098. if (!num_bits)
  1099. return 0;
  1100. while (m_bit_count < (int)num_bits)
  1101. {
  1102. uint c = 0;
  1103. if (m_pDecode_buf_next == m_pDecode_buf_end)
  1104. {
  1105. if (!m_decode_buf_eof)
  1106. {
  1107. m_pDecode_need_bytes_func(m_pDecode_buf_next - m_pDecode_buf, m_pDecode_private_data, m_pDecode_buf, m_decode_buf_size, m_decode_buf_eof);
  1108. m_pDecode_buf_end = m_pDecode_buf + m_decode_buf_size;
  1109. m_pDecode_buf_next = m_pDecode_buf;
  1110. if (m_pDecode_buf_next < m_pDecode_buf_end) c = *m_pDecode_buf_next++;
  1111. }
  1112. }
  1113. else
  1114. c = *m_pDecode_buf_next++;
  1115. m_bit_count += 8;
  1116. LZHAM_ASSERT(m_bit_count <= cBitBufSize);
  1117. m_bit_buf |= (static_cast<bit_buf_t>(c) << (cBitBufSize - m_bit_count));
  1118. }
  1119. uint result = static_cast<uint>(m_bit_buf >> (cBitBufSize - num_bits));
  1120. m_bit_buf <<= num_bits;
  1121. m_bit_count -= num_bits;
  1122. return result;
  1123. }
  1124. void symbol_codec::remove_bits(uint num_bits)
  1125. {
  1126. LZHAM_ASSERT(num_bits <= 25);
  1127. if (!num_bits)
  1128. return;
  1129. while (m_bit_count < (int)num_bits)
  1130. {
  1131. uint c = 0;
  1132. if (m_pDecode_buf_next == m_pDecode_buf_end)
  1133. {
  1134. if (!m_decode_buf_eof)
  1135. {
  1136. m_pDecode_need_bytes_func(m_pDecode_buf_next - m_pDecode_buf, m_pDecode_private_data, m_pDecode_buf, m_decode_buf_size, m_decode_buf_eof);
  1137. m_pDecode_buf_end = m_pDecode_buf + m_decode_buf_size;
  1138. m_pDecode_buf_next = m_pDecode_buf;
  1139. if (m_pDecode_buf_next < m_pDecode_buf_end) c = *m_pDecode_buf_next++;
  1140. }
  1141. }
  1142. else
  1143. c = *m_pDecode_buf_next++;
  1144. m_bit_count += 8;
  1145. LZHAM_ASSERT(m_bit_count <= cBitBufSize);
  1146. m_bit_buf |= (static_cast<bit_buf_t>(c) << (cBitBufSize - m_bit_count));
  1147. }
  1148. m_bit_buf <<= num_bits;
  1149. m_bit_count -= num_bits;
  1150. }
  1151. void symbol_codec::decode_align_to_byte()
  1152. {
  1153. LZHAM_ASSERT(m_mode == cDecoding);
  1154. if (m_bit_count & 7)
  1155. {
  1156. remove_bits(m_bit_count & 7);
  1157. }
  1158. }
  1159. int symbol_codec::decode_remove_byte_from_bit_buf()
  1160. {
  1161. if (m_bit_count < 8)
  1162. return -1;
  1163. int result = static_cast<int>(m_bit_buf >> (cBitBufSize - 8));
  1164. m_bit_buf <<= 8;
  1165. m_bit_count -= 8;
  1166. return result;
  1167. }
  1168. uint symbol_codec::decode(adaptive_bit_model& model, bool update_model)
  1169. {
  1170. while (m_arith_length < cSymbolCodecArithMinLen)
  1171. {
  1172. uint c = get_bits(8);
  1173. m_arith_value = (m_arith_value << 8) | c;
  1174. m_arith_length <<= 8;
  1175. }
  1176. uint x = model.m_bit_0_prob * (m_arith_length >> cSymbolCodecArithProbBits);
  1177. uint bit = (m_arith_value >= x);
  1178. if (!bit)
  1179. {
  1180. if (update_model)
  1181. model.m_bit_0_prob += ((cSymbolCodecArithProbScale - model.m_bit_0_prob) >> cSymbolCodecArithProbMoveBits);
  1182. m_arith_length = x;
  1183. }
  1184. else
  1185. {
  1186. if (update_model)
  1187. model.m_bit_0_prob -= (model.m_bit_0_prob >> cSymbolCodecArithProbMoveBits);
  1188. m_arith_value -= x;
  1189. m_arith_length -= x;
  1190. }
  1191. return bit;
  1192. }
  1193. uint symbol_codec::decode(adaptive_arith_data_model& model)
  1194. {
  1195. uint node = 1;
  1196. do
  1197. {
  1198. uint bit = decode(model.m_probs[node]);
  1199. node = (node << 1) + bit;
  1200. } while (node < model.m_total_syms);
  1201. return node - model.m_total_syms;
  1202. }
  1203. void symbol_codec::start_arith_decoding()
  1204. {
  1205. LZHAM_ASSERT(m_mode == cDecoding);
  1206. m_arith_length = cSymbolCodecArithMaxLen;
  1207. m_arith_value = 0;
  1208. m_arith_value = (get_bits(8) << 24);
  1209. m_arith_value |= (get_bits(8) << 16);
  1210. m_arith_value |= (get_bits(8) << 8);
  1211. m_arith_value |= get_bits(8);
  1212. }
  1213. void symbol_codec::decode_need_bytes()
  1214. {
  1215. if (!m_decode_buf_eof)
  1216. {
  1217. m_pDecode_need_bytes_func(m_pDecode_buf_next - m_pDecode_buf, m_pDecode_private_data, m_pDecode_buf, m_decode_buf_size, m_decode_buf_eof);
  1218. m_pDecode_buf_end = m_pDecode_buf + m_decode_buf_size;
  1219. m_pDecode_buf_next = m_pDecode_buf;
  1220. }
  1221. }
  1222. } // namespace lzham