2
* GRUB -- GRand Unified Bootloader
3
* Copyright (c) 1999-2008 Igor Pavlov
4
* Copyright (C) 2008 Free Software Foundation, Inc.
6
* GRUB is free software: you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation, either version 3 of the License, or
9
* (at your option) any later version.
11
* GRUB is distributed in the hope that it will be useful,
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
* GNU General Public License for more details.
16
* You should have received a copy of the GNU General Public License
17
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
21
* This code was taken from LZMA SDK 4.58 beta, and was slightly modified
22
* to adapt it to GRUB's requirement.
24
* See <http://www.7-zip.org>, for more information about LZMA.
30
#include <grub/lib/LzmaEnc.h>
32
#include <grub/lib/LzFind.h>
34
#include <grub/lib/LzFindMt.h>
37
/* #define SHOW_STAT */
38
/* #define SHOW_STAT2 */
44
#define kBlockSizeMax ((1 << LZMA_NUM_BLOCK_SIZE_BITS) - 1)
46
#define kBlockSize (9 << 10)
47
#define kUnpackBlockSize (1 << 18)
48
#define kMatchArraySize (1 << 21)
49
#define kMatchRecordMaxSize ((LZMA_MATCH_LEN_MAX * 2 + 3) * LZMA_MATCH_LEN_MAX)
51
#define kNumMaxDirectBits (31)
53
#define kNumTopBits 24
54
#define kTopValue ((UInt32)1 << kNumTopBits)
56
#define kNumBitModelTotalBits 11
57
#define kBitModelTotal (1 << kNumBitModelTotalBits)
58
#define kNumMoveBits 5
59
#define kProbInitValue (kBitModelTotal >> 1)
61
#define kNumMoveReducingBits 4
62
#define kNumBitPriceShiftBits 4
63
#define kBitPrice (1 << kNumBitPriceShiftBits)
65
void LzmaEncProps_Init(CLzmaEncProps *p)
68
p->dictSize = p->mc = 0;
69
p->lc = p->lp = p->pb = p->algo = p->fb = p->btMode = p->numHashBytes = p->numThreads = -1;
73
void LzmaEncProps_Normalize(CLzmaEncProps *p)
76
if (level < 0) level = 5;
78
if (p->dictSize == 0) p->dictSize = (level <= 5 ? (1 << (level * 2 + 14)) : (level == 6 ? (1 << 25) : (1 << 26)));
79
if (p->lc < 0) p->lc = 3;
80
if (p->lp < 0) p->lp = 0;
81
if (p->pb < 0) p->pb = 2;
82
if (p->algo < 0) p->algo = (level < 5 ? 0 : 1);
83
if (p->fb < 0) p->fb = (level < 7 ? 32 : 64);
84
if (p->btMode < 0) p->btMode = (p->algo == 0 ? 0 : 1);
85
if (p->numHashBytes < 0) p->numHashBytes = 4;
86
if (p->mc == 0) p->mc = (16 + (p->fb >> 1)) >> (p->btMode ? 0 : 1);
87
if (p->numThreads < 0) p->numThreads = ((p->btMode && p->algo) ? 2 : 1);
90
UInt32 LzmaEncProps_GetDictSize(const CLzmaEncProps *props2)
92
CLzmaEncProps props = *props2;
93
LzmaEncProps_Normalize(&props);
94
return props.dictSize;
97
/* #define LZMA_LOG_BSR */
98
/* Define it for Intel's CPU */
103
#define kDicLogSizeMaxCompress 30
105
#define BSR2_RET(pos, res) { unsigned long i; _BitScanReverse(&i, (pos)); res = (i + i) + ((pos >> (i - 1)) & 1); }
107
UInt32 GetPosSlot1(UInt32 pos)
113
#define GetPosSlot2(pos, res) { BSR2_RET(pos, res); }
114
#define GetPosSlot(pos, res) { if (pos < 2) res = pos; else BSR2_RET(pos, res); }
118
#define kNumLogBits (9 + (int)sizeof(size_t) / 2)
119
#define kDicLogSizeMaxCompress ((kNumLogBits - 1) * 2 + 7)
121
void LzmaEnc_FastPosInit(Byte *g_FastPos)
127
for (slotFast = 2; slotFast < kNumLogBits * 2; slotFast++)
129
UInt32 k = (1 << ((slotFast >> 1) - 1));
131
for (j = 0; j < k; j++, c++)
132
g_FastPos[c] = (Byte)slotFast;
136
#define BSR2_RET(pos, res) { UInt32 i = 6 + ((kNumLogBits - 1) & \
137
(0 - (((((UInt32)1 << (kNumLogBits + 6)) - 1) - pos) >> 31))); \
138
res = p->g_FastPos[pos >> i] + (i * 2); }
140
#define BSR2_RET(pos, res) { res = (pos < (1 << (kNumLogBits + 6))) ? \
141
p->g_FastPos[pos >> 6] + 12 : \
142
p->g_FastPos[pos >> (6 + kNumLogBits - 1)] + (6 + (kNumLogBits - 1)) * 2; }
145
#define GetPosSlot1(pos) p->g_FastPos[pos]
146
#define GetPosSlot2(pos, res) { BSR2_RET(pos, res); }
147
#define GetPosSlot(pos, res) { if (pos < kNumFullDistances) res = p->g_FastPos[pos]; else BSR2_RET(pos, res); }
152
#define LZMA_NUM_REPS 4
154
typedef unsigned CState;
156
typedef struct _COptimal
169
UInt32 backs[LZMA_NUM_REPS];
172
#define kNumOpts (1 << 12)
174
#define kNumLenToPosStates 4
175
#define kNumPosSlotBits 6
176
#define kDicLogSizeMin 0
177
#define kDicLogSizeMax 32
178
#define kDistTableSizeMax (kDicLogSizeMax * 2)
181
#define kNumAlignBits 4
182
#define kAlignTableSize (1 << kNumAlignBits)
183
#define kAlignMask (kAlignTableSize - 1)
185
#define kStartPosModelIndex 4
186
#define kEndPosModelIndex 14
187
#define kNumPosModels (kEndPosModelIndex - kStartPosModelIndex)
189
#define kNumFullDistances (1 << (kEndPosModelIndex / 2))
192
#define CLzmaProb UInt32
194
#define CLzmaProb UInt16
197
#define LZMA_PB_MAX 4
198
#define LZMA_LC_MAX 8
199
#define LZMA_LP_MAX 4
201
#define LZMA_NUM_PB_STATES_MAX (1 << LZMA_PB_MAX)
204
#define kLenNumLowBits 3
205
#define kLenNumLowSymbols (1 << kLenNumLowBits)
206
#define kLenNumMidBits 3
207
#define kLenNumMidSymbols (1 << kLenNumMidBits)
208
#define kLenNumHighBits 8
209
#define kLenNumHighSymbols (1 << kLenNumHighBits)
211
#define kLenNumSymbolsTotal (kLenNumLowSymbols + kLenNumMidSymbols + kLenNumHighSymbols)
213
#define LZMA_MATCH_LEN_MIN 2
214
#define LZMA_MATCH_LEN_MAX (LZMA_MATCH_LEN_MIN + kLenNumSymbolsTotal - 1)
216
#define kNumStates 12
222
CLzmaProb low[LZMA_NUM_PB_STATES_MAX << kLenNumLowBits];
223
CLzmaProb mid[LZMA_NUM_PB_STATES_MAX << kLenNumMidBits];
224
CLzmaProb high[kLenNumHighSymbols];
230
UInt32 prices[LZMA_NUM_PB_STATES_MAX][kLenNumSymbolsTotal];
232
UInt32 counters[LZMA_NUM_PB_STATES_MAX];
235
typedef struct _CRangeEnc
244
ISeqOutStream *outStream;
249
typedef struct _CSeqInStreamBuf
251
ISeqInStream funcTable;
256
static SRes MyRead(void *pp, void *data, size_t *size)
258
size_t curSize = *size;
259
CSeqInStreamBuf *p = (CSeqInStreamBuf *)pp;
260
if (p->rem < curSize)
262
memcpy(data, p->data, curSize);
273
CLzmaProb isMatch[kNumStates][LZMA_NUM_PB_STATES_MAX];
274
CLzmaProb isRep[kNumStates];
275
CLzmaProb isRepG0[kNumStates];
276
CLzmaProb isRepG1[kNumStates];
277
CLzmaProb isRepG2[kNumStates];
278
CLzmaProb isRep0Long[kNumStates][LZMA_NUM_PB_STATES_MAX];
280
CLzmaProb posSlotEncoder[kNumLenToPosStates][1 << kNumPosSlotBits];
281
CLzmaProb posEncoders[kNumFullDistances - kEndPosModelIndex];
282
CLzmaProb posAlignEncoder[1 << kNumAlignBits];
285
CLenPriceEnc repLenEnc;
287
UInt32 reps[LZMA_NUM_REPS];
291
typedef struct _CLzmaEnc
293
IMatchFinder matchFinder;
294
void *matchFinderObj;
296
#ifdef COMPRESS_MF_MT
298
CMatchFinderMt matchFinderMt;
301
CMatchFinder matchFinderBase;
303
#ifdef COMPRESS_MF_MT
307
UInt32 optimumEndIndex;
308
UInt32 optimumCurrentIndex;
310
Bool longestMatchWasFound;
311
UInt32 longestMatchLength;
312
UInt32 numDistancePairs;
314
COptimal opt[kNumOpts];
317
Byte g_FastPos[1 << kNumLogBits];
320
UInt32 ProbPrices[kBitModelTotal >> kNumMoveReducingBits];
321
UInt32 matchDistances[LZMA_MATCH_LEN_MAX * 2 + 2 + 1];
323
UInt32 additionalOffset;
324
UInt32 reps[LZMA_NUM_REPS];
327
UInt32 posSlotPrices[kNumLenToPosStates][kDistTableSizeMax];
328
UInt32 distancesPrices[kNumLenToPosStates][kNumFullDistances];
329
UInt32 alignPrices[kAlignTableSize];
330
UInt32 alignPriceCount;
332
UInt32 distTableSize;
335
unsigned lpMask, pbMask;
339
CLzmaProb isMatch[kNumStates][LZMA_NUM_PB_STATES_MAX];
340
CLzmaProb isRep[kNumStates];
341
CLzmaProb isRepG0[kNumStates];
342
CLzmaProb isRepG1[kNumStates];
343
CLzmaProb isRepG2[kNumStates];
344
CLzmaProb isRep0Long[kNumStates][LZMA_NUM_PB_STATES_MAX];
346
CLzmaProb posSlotEncoder[kNumLenToPosStates][1 << kNumPosSlotBits];
347
CLzmaProb posEncoders[kNumFullDistances - kEndPosModelIndex];
348
CLzmaProb posAlignEncoder[1 << kNumAlignBits];
351
CLenPriceEnc repLenEnc;
361
UInt32 matchPriceCount;
367
UInt32 matchFinderCycles;
369
ISeqInStream *inStream;
370
CSeqInStreamBuf seqBufInStream;
372
CSaveState saveState;
375
void LzmaEnc_SaveState(CLzmaEncHandle pp)
377
CLzmaEnc *p = (CLzmaEnc *)pp;
378
CSaveState *dest = &p->saveState;
380
dest->lenEnc = p->lenEnc;
381
dest->repLenEnc = p->repLenEnc;
382
dest->state = p->state;
384
for (i = 0; i < kNumStates; i++)
386
memcpy(dest->isMatch[i], p->isMatch[i], sizeof(p->isMatch[i]));
387
memcpy(dest->isRep0Long[i], p->isRep0Long[i], sizeof(p->isRep0Long[i]));
389
for (i = 0; i < kNumLenToPosStates; i++)
390
memcpy(dest->posSlotEncoder[i], p->posSlotEncoder[i], sizeof(p->posSlotEncoder[i]));
391
memcpy(dest->isRep, p->isRep, sizeof(p->isRep));
392
memcpy(dest->isRepG0, p->isRepG0, sizeof(p->isRepG0));
393
memcpy(dest->isRepG1, p->isRepG1, sizeof(p->isRepG1));
394
memcpy(dest->isRepG2, p->isRepG2, sizeof(p->isRepG2));
395
memcpy(dest->posEncoders, p->posEncoders, sizeof(p->posEncoders));
396
memcpy(dest->posAlignEncoder, p->posAlignEncoder, sizeof(p->posAlignEncoder));
397
memcpy(dest->reps, p->reps, sizeof(p->reps));
398
memcpy(dest->litProbs, p->litProbs, (0x300 << p->lclp) * sizeof(CLzmaProb));
401
void LzmaEnc_RestoreState(CLzmaEncHandle pp)
403
CLzmaEnc *dest = (CLzmaEnc *)pp;
404
const CSaveState *p = &dest->saveState;
406
dest->lenEnc = p->lenEnc;
407
dest->repLenEnc = p->repLenEnc;
408
dest->state = p->state;
410
for (i = 0; i < kNumStates; i++)
412
memcpy(dest->isMatch[i], p->isMatch[i], sizeof(p->isMatch[i]));
413
memcpy(dest->isRep0Long[i], p->isRep0Long[i], sizeof(p->isRep0Long[i]));
415
for (i = 0; i < kNumLenToPosStates; i++)
416
memcpy(dest->posSlotEncoder[i], p->posSlotEncoder[i], sizeof(p->posSlotEncoder[i]));
417
memcpy(dest->isRep, p->isRep, sizeof(p->isRep));
418
memcpy(dest->isRepG0, p->isRepG0, sizeof(p->isRepG0));
419
memcpy(dest->isRepG1, p->isRepG1, sizeof(p->isRepG1));
420
memcpy(dest->isRepG2, p->isRepG2, sizeof(p->isRepG2));
421
memcpy(dest->posEncoders, p->posEncoders, sizeof(p->posEncoders));
422
memcpy(dest->posAlignEncoder, p->posAlignEncoder, sizeof(p->posAlignEncoder));
423
memcpy(dest->reps, p->reps, sizeof(p->reps));
424
memcpy(dest->litProbs, p->litProbs, (0x300 << dest->lclp) * sizeof(CLzmaProb));
427
SRes LzmaEnc_SetProps(CLzmaEncHandle pp, const CLzmaEncProps *props2)
429
CLzmaEnc *p = (CLzmaEnc *)pp;
430
CLzmaEncProps props = *props2;
431
LzmaEncProps_Normalize(&props);
433
if (props.lc > LZMA_LC_MAX || props.lp > LZMA_LP_MAX || props.pb > LZMA_PB_MAX ||
434
props.dictSize > (1U << kDicLogSizeMaxCompress) || props.dictSize > (1 << 30))
435
return SZ_ERROR_PARAM;
436
p->dictSize = props.dictSize;
437
p->matchFinderCycles = props.mc;
439
unsigned fb = props.fb;
442
if (fb > LZMA_MATCH_LEN_MAX)
443
fb = LZMA_MATCH_LEN_MAX;
444
p->numFastBytes = fb;
449
p->fastMode = (props.algo == 0);
450
p->matchFinderBase.btMode = props.btMode;
452
UInt32 numHashBytes = 4;
455
if (props.numHashBytes < 2)
457
else if (props.numHashBytes < 4)
458
numHashBytes = props.numHashBytes;
460
p->matchFinderBase.numHashBytes = numHashBytes;
463
p->matchFinderBase.cutValue = props.mc;
465
p->writeEndMark = props.writeEndMark;
467
#ifdef COMPRESS_MF_MT
469
if (newMultiThread != _multiThread)
471
ReleaseMatchFinder();
472
_multiThread = newMultiThread;
475
p->multiThread = (props.numThreads > 1);
481
static const int kLiteralNextStates[kNumStates] = {0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 4, 5};
482
static const int kMatchNextStates[kNumStates] = {7, 7, 7, 7, 7, 7, 7, 10, 10, 10, 10, 10};
483
static const int kRepNextStates[kNumStates] = {8, 8, 8, 8, 8, 8, 8, 11, 11, 11, 11, 11};
484
static const int kShortRepNextStates[kNumStates]= {9, 9, 9, 9, 9, 9, 9, 11, 11, 11, 11, 11};
487
void UpdateChar() { Index = kLiteralNextStates[Index]; }
488
void UpdateMatch() { Index = kMatchNextStates[Index]; }
489
void UpdateRep() { Index = kRepNextStates[Index]; }
490
void UpdateShortRep() { Index = kShortRepNextStates[Index]; }
493
#define IsCharState(s) ((s) < 7)
496
#define GetLenToPosState(len) (((len) < kNumLenToPosStates + 1) ? (len) - 2 : kNumLenToPosStates - 1)
498
#define kInfinityPrice (1 << 30)
500
static void RangeEnc_Construct(CRangeEnc *p)
506
#define RangeEnc_GetProcessed(p) ((p)->processed + ((p)->buf - (p)->bufBase) + (p)->cacheSize)
508
#define RC_BUF_SIZE (1 << 16)
509
static int RangeEnc_Alloc(CRangeEnc *p, ISzAlloc *alloc)
513
p->bufBase = (Byte *)alloc->Alloc(alloc, RC_BUF_SIZE);
516
p->bufLim = p->bufBase + RC_BUF_SIZE;
521
static void RangeEnc_Free(CRangeEnc *p, ISzAlloc *alloc)
523
alloc->Free(alloc, p->bufBase);
527
static void RangeEnc_Init(CRangeEnc *p)
531
p->range = 0xFFFFFFFF;
541
static void RangeEnc_FlushStream(CRangeEnc *p)
546
num = p->buf - p->bufBase;
547
if (num != p->outStream->Write(p->outStream, p->bufBase, num))
548
p->res = SZ_ERROR_WRITE;
553
static void MY_FAST_CALL RangeEnc_ShiftLow(CRangeEnc *p)
555
if ((UInt32)p->low < (UInt32)0xFF000000 || (int)(p->low >> 32) != 0)
557
Byte temp = p->cache;
561
*buf++ = (Byte)(temp + (Byte)(p->low >> 32));
563
if (buf == p->bufLim)
564
RangeEnc_FlushStream(p);
567
while (--p->cacheSize != 0);
568
p->cache = (Byte)((UInt32)p->low >> 24);
571
p->low = (UInt32)p->low << 8;
574
static void RangeEnc_FlushData(CRangeEnc *p)
577
for (i = 0; i < 5; i++)
578
RangeEnc_ShiftLow(p);
581
static void RangeEnc_EncodeDirectBits(CRangeEnc *p, UInt32 value, int numBits)
586
p->low += p->range & (0 - ((value >> --numBits) & 1));
587
if (p->range < kTopValue)
590
RangeEnc_ShiftLow(p);
593
while (numBits != 0);
596
static void RangeEnc_EncodeBit(CRangeEnc *p, CLzmaProb *prob, UInt32 symbol)
599
UInt32 newBound = (p->range >> kNumBitModelTotalBits) * ttt;
603
ttt += (kBitModelTotal - ttt) >> kNumMoveBits;
608
p->range -= newBound;
609
ttt -= ttt >> kNumMoveBits;
611
*prob = (CLzmaProb)ttt;
612
if (p->range < kTopValue)
615
RangeEnc_ShiftLow(p);
619
static void LitEnc_Encode(CRangeEnc *p, CLzmaProb *probs, UInt32 symbol)
624
RangeEnc_EncodeBit(p, probs + (symbol >> 8), (symbol >> 7) & 1);
627
while (symbol < 0x10000);
630
static void LitEnc_EncodeMatched(CRangeEnc *p, CLzmaProb *probs, UInt32 symbol, UInt32 matchByte)
637
RangeEnc_EncodeBit(p, probs + (offs + (matchByte & offs) + (symbol >> 8)), (symbol >> 7) & 1);
639
offs &= ~(matchByte ^ symbol);
641
while (symbol < 0x10000);
644
void LzmaEnc_InitPriceTables(UInt32 *ProbPrices)
647
for (i = (1 << kNumMoveReducingBits) / 2; i < kBitModelTotal; i += (1 << kNumMoveReducingBits))
649
const int kCyclesBits = kNumBitPriceShiftBits;
653
for (j = 0; j < kCyclesBits; j++)
657
while (w >= ((UInt32)1 << 16))
663
ProbPrices[i >> kNumMoveReducingBits] = ((kNumBitModelTotalBits << kCyclesBits) - 15 - bitCount);
668
#define GET_PRICE(prob, symbol) \
669
p->ProbPrices[((prob) ^ (((-(int)(symbol))) & (kBitModelTotal - 1))) >> kNumMoveReducingBits];
671
#define GET_PRICEa(prob, symbol) \
672
ProbPrices[((prob) ^ ((-((int)(symbol))) & (kBitModelTotal - 1))) >> kNumMoveReducingBits];
674
#define GET_PRICE_0(prob) p->ProbPrices[(prob) >> kNumMoveReducingBits]
675
#define GET_PRICE_1(prob) p->ProbPrices[((prob) ^ (kBitModelTotal - 1)) >> kNumMoveReducingBits]
677
#define GET_PRICE_0a(prob) ProbPrices[(prob) >> kNumMoveReducingBits]
678
#define GET_PRICE_1a(prob) ProbPrices[((prob) ^ (kBitModelTotal - 1)) >> kNumMoveReducingBits]
680
static UInt32 LitEnc_GetPrice(const CLzmaProb *probs, UInt32 symbol, UInt32 *ProbPrices)
686
price += GET_PRICEa(probs[symbol >> 8], (symbol >> 7) & 1);
689
while (symbol < 0x10000);
693
static UInt32 LitEnc_GetPriceMatched(const CLzmaProb *probs, UInt32 symbol, UInt32 matchByte, UInt32 *ProbPrices)
701
price += GET_PRICEa(probs[offs + (matchByte & offs) + (symbol >> 8)], (symbol >> 7) & 1);
703
offs &= ~(matchByte ^ symbol);
705
while (symbol < 0x10000);
710
static void RcTree_Encode(CRangeEnc *rc, CLzmaProb *probs, int numBitLevels, UInt32 symbol)
714
for (i = numBitLevels; i != 0 ;)
718
bit = (symbol >> i) & 1;
719
RangeEnc_EncodeBit(rc, probs + m, bit);
724
static void RcTree_ReverseEncode(CRangeEnc *rc, CLzmaProb *probs, int numBitLevels, UInt32 symbol)
728
for (i = 0; i < numBitLevels; i++)
730
UInt32 bit = symbol & 1;
731
RangeEnc_EncodeBit(rc, probs + m, bit);
737
static UInt32 RcTree_GetPrice(const CLzmaProb *probs, int numBitLevels, UInt32 symbol, UInt32 *ProbPrices)
740
symbol |= (1 << numBitLevels);
743
price += GET_PRICEa(probs[symbol >> 1], symbol & 1);
749
static UInt32 RcTree_ReverseGetPrice(const CLzmaProb *probs, int numBitLevels, UInt32 symbol, UInt32 *ProbPrices)
754
for (i = numBitLevels; i != 0; i--)
756
UInt32 bit = symbol & 1;
758
price += GET_PRICEa(probs[m], bit);
765
static void LenEnc_Init(CLenEnc *p)
768
p->choice = p->choice2 = kProbInitValue;
769
for (i = 0; i < (LZMA_NUM_PB_STATES_MAX << kLenNumLowBits); i++)
770
p->low[i] = kProbInitValue;
771
for (i = 0; i < (LZMA_NUM_PB_STATES_MAX << kLenNumMidBits); i++)
772
p->mid[i] = kProbInitValue;
773
for (i = 0; i < kLenNumHighSymbols; i++)
774
p->high[i] = kProbInitValue;
777
static void LenEnc_Encode(CLenEnc *p, CRangeEnc *rc, UInt32 symbol, UInt32 posState)
779
if (symbol < kLenNumLowSymbols)
781
RangeEnc_EncodeBit(rc, &p->choice, 0);
782
RcTree_Encode(rc, p->low + (posState << kLenNumLowBits), kLenNumLowBits, symbol);
786
RangeEnc_EncodeBit(rc, &p->choice, 1);
787
if (symbol < kLenNumLowSymbols + kLenNumMidSymbols)
789
RangeEnc_EncodeBit(rc, &p->choice2, 0);
790
RcTree_Encode(rc, p->mid + (posState << kLenNumMidBits), kLenNumMidBits, symbol - kLenNumLowSymbols);
794
RangeEnc_EncodeBit(rc, &p->choice2, 1);
795
RcTree_Encode(rc, p->high, kLenNumHighBits, symbol - kLenNumLowSymbols - kLenNumMidSymbols);
800
static void LenEnc_SetPrices(CLenEnc *p, UInt32 posState, UInt32 numSymbols, UInt32 *prices, UInt32 *ProbPrices)
802
UInt32 a0 = GET_PRICE_0a(p->choice);
803
UInt32 a1 = GET_PRICE_1a(p->choice);
804
UInt32 b0 = a1 + GET_PRICE_0a(p->choice2);
805
UInt32 b1 = a1 + GET_PRICE_1a(p->choice2);
807
for (i = 0; i < kLenNumLowSymbols; i++)
811
prices[i] = a0 + RcTree_GetPrice(p->low + (posState << kLenNumLowBits), kLenNumLowBits, i, ProbPrices);
813
for (; i < kLenNumLowSymbols + kLenNumMidSymbols; i++)
817
prices[i] = b0 + RcTree_GetPrice(p->mid + (posState << kLenNumMidBits), kLenNumMidBits, i - kLenNumLowSymbols, ProbPrices);
819
for (; i < numSymbols; i++)
820
prices[i] = b1 + RcTree_GetPrice(p->high, kLenNumHighBits, i - kLenNumLowSymbols - kLenNumMidSymbols, ProbPrices);
823
static void MY_FAST_CALL LenPriceEnc_UpdateTable(CLenPriceEnc *p, UInt32 posState, UInt32 *ProbPrices)
825
LenEnc_SetPrices(&p->p, posState, p->tableSize, p->prices[posState], ProbPrices);
826
p->counters[posState] = p->tableSize;
829
static void LenPriceEnc_UpdateTables(CLenPriceEnc *p, UInt32 numPosStates, UInt32 *ProbPrices)
832
for (posState = 0; posState < numPosStates; posState++)
833
LenPriceEnc_UpdateTable(p, posState, ProbPrices);
836
static void LenEnc_Encode2(CLenPriceEnc *p, CRangeEnc *rc, UInt32 symbol, UInt32 posState, Bool updatePrice, UInt32 *ProbPrices)
838
LenEnc_Encode(&p->p, rc, symbol, posState);
840
if (--p->counters[posState] == 0)
841
LenPriceEnc_UpdateTable(p, posState, ProbPrices);
847
static void MovePos(CLzmaEnc *p, UInt32 num)
851
printf("\n MovePos %d", num);
855
p->additionalOffset += num;
856
p->matchFinder.Skip(p->matchFinderObj, num);
860
static UInt32 ReadMatchDistances(CLzmaEnc *p, UInt32 *numDistancePairsRes)
862
UInt32 lenRes = 0, numDistancePairs;
863
numDistancePairs = p->matchFinder.GetMatches(p->matchFinderObj, p->matchDistances);
865
printf("\n i = %d numPairs = %d ", ttt, numDistancePairs / 2);
872
for (i = 0; i < numDistancePairs; i += 2)
873
printf("%2d %6d | ", p->matchDistances[i], p->matchDistances[i + 1]);
876
if (numDistancePairs > 0)
878
lenRes = p->matchDistances[numDistancePairs - 2];
879
if (lenRes == p->numFastBytes)
881
UInt32 numAvail = p->matchFinder.GetNumAvailableBytes(p->matchFinderObj) + 1;
882
const Byte *pby = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - 1;
883
UInt32 distance = p->matchDistances[numDistancePairs - 1] + 1;
884
if (numAvail > LZMA_MATCH_LEN_MAX)
885
numAvail = LZMA_MATCH_LEN_MAX;
888
const Byte *pby2 = pby - distance;
889
for (; lenRes < numAvail && pby[lenRes] == pby2[lenRes]; lenRes++);
893
p->additionalOffset++;
894
*numDistancePairsRes = numDistancePairs;
899
#define MakeAsChar(p) (p)->backPrev = (UInt32)(-1); (p)->prev1IsChar = False;
900
#define MakeAsShortRep(p) (p)->backPrev = 0; (p)->prev1IsChar = False;
901
#define IsShortRep(p) ((p)->backPrev == 0)
903
static UInt32 GetRepLen1Price(CLzmaEnc *p, UInt32 state, UInt32 posState)
906
GET_PRICE_0(p->isRepG0[state]) +
907
GET_PRICE_0(p->isRep0Long[state][posState]);
910
static UInt32 GetPureRepPrice(CLzmaEnc *p, UInt32 repIndex, UInt32 state, UInt32 posState)
915
price = GET_PRICE_0(p->isRepG0[state]);
916
price += GET_PRICE_1(p->isRep0Long[state][posState]);
920
price = GET_PRICE_1(p->isRepG0[state]);
922
price += GET_PRICE_0(p->isRepG1[state]);
925
price += GET_PRICE_1(p->isRepG1[state]);
926
price += GET_PRICE(p->isRepG2[state], repIndex - 2);
932
static UInt32 GetRepPrice(CLzmaEnc *p, UInt32 repIndex, UInt32 len, UInt32 state, UInt32 posState)
934
return p->repLenEnc.prices[posState][len - LZMA_MATCH_LEN_MIN] +
935
GetPureRepPrice(p, repIndex, state, posState);
938
static UInt32 Backward(CLzmaEnc *p, UInt32 *backRes, UInt32 cur)
940
UInt32 posMem = p->opt[cur].posPrev;
941
UInt32 backMem = p->opt[cur].backPrev;
942
p->optimumEndIndex = cur;
945
if (p->opt[cur].prev1IsChar)
947
MakeAsChar(&p->opt[posMem])
948
p->opt[posMem].posPrev = posMem - 1;
949
if (p->opt[cur].prev2)
951
p->opt[posMem - 1].prev1IsChar = False;
952
p->opt[posMem - 1].posPrev = p->opt[cur].posPrev2;
953
p->opt[posMem - 1].backPrev = p->opt[cur].backPrev2;
957
UInt32 posPrev = posMem;
958
UInt32 backCur = backMem;
960
backMem = p->opt[posPrev].backPrev;
961
posMem = p->opt[posPrev].posPrev;
963
p->opt[posPrev].backPrev = backCur;
964
p->opt[posPrev].posPrev = cur;
969
*backRes = p->opt[0].backPrev;
970
p->optimumCurrentIndex = p->opt[0].posPrev;
971
return p->optimumCurrentIndex;
974
#define LIT_PROBS(pos, prevByte) (p->litProbs + ((((pos) & p->lpMask) << p->lc) + ((prevByte) >> (8 - p->lc))) * 0x300)
976
static UInt32 GetOptimum(CLzmaEnc *p, UInt32 position, UInt32 *backRes)
978
UInt32 numAvailableBytes, lenMain, numDistancePairs;
980
UInt32 reps[LZMA_NUM_REPS];
981
UInt32 repLens[LZMA_NUM_REPS];
982
UInt32 repMaxIndex, i;
983
UInt32 *matchDistances;
984
Byte currentByte, matchByte;
986
UInt32 matchPrice, repMatchPrice;
989
UInt32 normalMatchPrice;
991
if (p->optimumEndIndex != p->optimumCurrentIndex)
993
const COptimal *opt = &p->opt[p->optimumCurrentIndex];
994
UInt32 lenRes = opt->posPrev - p->optimumCurrentIndex;
995
*backRes = opt->backPrev;
996
p->optimumCurrentIndex = opt->posPrev;
999
p->optimumCurrentIndex = p->optimumEndIndex = 0;
1001
numAvailableBytes = p->matchFinder.GetNumAvailableBytes(p->matchFinderObj);
1003
if (!p->longestMatchWasFound)
1005
lenMain = ReadMatchDistances(p, &numDistancePairs);
1009
lenMain = p->longestMatchLength;
1010
numDistancePairs = p->numDistancePairs;
1011
p->longestMatchWasFound = False;
1014
data = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - 1;
1015
if (numAvailableBytes < 2)
1017
*backRes = (UInt32)(-1);
1020
if (numAvailableBytes > LZMA_MATCH_LEN_MAX)
1021
numAvailableBytes = LZMA_MATCH_LEN_MAX;
1024
for (i = 0; i < LZMA_NUM_REPS; i++)
1028
reps[i] = p->reps[i];
1029
data2 = data - (reps[i] + 1);
1030
if (data[0] != data2[0] || data[1] != data2[1])
1035
for (lenTest = 2; lenTest < numAvailableBytes && data[lenTest] == data2[lenTest]; lenTest++);
1036
repLens[i] = lenTest;
1037
if (lenTest > repLens[repMaxIndex])
1040
if (repLens[repMaxIndex] >= p->numFastBytes)
1043
*backRes = repMaxIndex;
1044
lenRes = repLens[repMaxIndex];
1045
MovePos(p, lenRes - 1);
1049
matchDistances = p->matchDistances;
1050
if (lenMain >= p->numFastBytes)
1052
*backRes = matchDistances[numDistancePairs - 1] + LZMA_NUM_REPS;
1053
MovePos(p, lenMain - 1);
1056
currentByte = *data;
1057
matchByte = *(data - (reps[0] + 1));
1059
if (lenMain < 2 && currentByte != matchByte && repLens[repMaxIndex] < 2)
1061
*backRes = (UInt32)-1;
1065
p->opt[0].state = (CState)p->state;
1067
posState = (position & p->pbMask);
1070
const CLzmaProb *probs = LIT_PROBS(position, *(data - 1));
1071
p->opt[1].price = GET_PRICE_0(p->isMatch[p->state][posState]) +
1072
(!IsCharState(p->state) ?
1073
LitEnc_GetPriceMatched(probs, currentByte, matchByte, p->ProbPrices) :
1074
LitEnc_GetPrice(probs, currentByte, p->ProbPrices));
1077
MakeAsChar(&p->opt[1]);
1079
matchPrice = GET_PRICE_1(p->isMatch[p->state][posState]);
1080
repMatchPrice = matchPrice + GET_PRICE_1(p->isRep[p->state]);
1082
if (matchByte == currentByte)
1084
UInt32 shortRepPrice = repMatchPrice + GetRepLen1Price(p, p->state, posState);
1085
if (shortRepPrice < p->opt[1].price)
1087
p->opt[1].price = shortRepPrice;
1088
MakeAsShortRep(&p->opt[1]);
1091
lenEnd = ((lenMain >= repLens[repMaxIndex]) ? lenMain : repLens[repMaxIndex]);
1095
*backRes = p->opt[1].backPrev;
1099
p->opt[1].posPrev = 0;
1100
for (i = 0; i < LZMA_NUM_REPS; i++)
1101
p->opt[0].backs[i] = reps[i];
1105
p->opt[len--].price = kInfinityPrice;
1108
for (i = 0; i < LZMA_NUM_REPS; i++)
1110
UInt32 repLen = repLens[i];
1114
price = repMatchPrice + GetPureRepPrice(p, i, p->state, posState);
1117
UInt32 curAndLenPrice = price + p->repLenEnc.prices[posState][repLen - 2];
1118
COptimal *opt = &p->opt[repLen];
1119
if (curAndLenPrice < opt->price)
1121
opt->price = curAndLenPrice;
1124
opt->prev1IsChar = False;
1127
while (--repLen >= 2);
1130
normalMatchPrice = matchPrice + GET_PRICE_0(p->isRep[p->state]);
1132
len = ((repLens[0] >= 2) ? repLens[0] + 1 : 2);
1136
while (len > matchDistances[offs])
1141
UInt32 distance = matchDistances[offs + 1];
1143
UInt32 curAndLenPrice = normalMatchPrice + p->lenEnc.prices[posState][len - LZMA_MATCH_LEN_MIN];
1144
UInt32 lenToPosState = GetLenToPosState(len);
1145
if (distance < kNumFullDistances)
1146
curAndLenPrice += p->distancesPrices[lenToPosState][distance];
1150
GetPosSlot2(distance, slot);
1151
curAndLenPrice += p->alignPrices[distance & kAlignMask] + p->posSlotPrices[lenToPosState][slot];
1154
if (curAndLenPrice < opt->price)
1156
opt->price = curAndLenPrice;
1158
opt->backPrev = distance + LZMA_NUM_REPS;
1159
opt->prev1IsChar = False;
1161
if (len == matchDistances[offs])
1164
if (offs == numDistancePairs)
1176
printf("\n pos = %4X", position);
1177
for (i = cur; i <= lenEnd; i++)
1178
printf("\nprice[%4X] = %d", position - cur + i, p->opt[i].price);
1184
UInt32 numAvailableBytesFull, newLen, numDistancePairs;
1191
Byte currentByte, matchByte;
1193
UInt32 curAnd1Price;
1195
UInt32 matchPrice, repMatchPrice;
1196
UInt32 numAvailableBytes;
1201
return Backward(p, backRes, cur);
1203
numAvailableBytesFull = p->matchFinder.GetNumAvailableBytes(p->matchFinderObj);
1204
newLen = ReadMatchDistances(p, &numDistancePairs);
1205
if (newLen >= p->numFastBytes)
1207
p->numDistancePairs = numDistancePairs;
1208
p->longestMatchLength = newLen;
1209
p->longestMatchWasFound = True;
1210
return Backward(p, backRes, cur);
1213
curOpt = &p->opt[cur];
1214
posPrev = curOpt->posPrev;
1215
if (curOpt->prev1IsChar)
1220
state = p->opt[curOpt->posPrev2].state;
1221
if (curOpt->backPrev2 < LZMA_NUM_REPS)
1222
state = kRepNextStates[state];
1224
state = kMatchNextStates[state];
1227
state = p->opt[posPrev].state;
1228
state = kLiteralNextStates[state];
1231
state = p->opt[posPrev].state;
1232
if (posPrev == cur - 1)
1234
if (IsShortRep(curOpt))
1235
state = kShortRepNextStates[state];
1237
state = kLiteralNextStates[state];
1242
const COptimal *prevOpt;
1243
if (curOpt->prev1IsChar && curOpt->prev2)
1245
posPrev = curOpt->posPrev2;
1246
pos = curOpt->backPrev2;
1247
state = kRepNextStates[state];
1251
pos = curOpt->backPrev;
1252
if (pos < LZMA_NUM_REPS)
1253
state = kRepNextStates[state];
1255
state = kMatchNextStates[state];
1257
prevOpt = &p->opt[posPrev];
1258
if (pos < LZMA_NUM_REPS)
1261
reps[0] = prevOpt->backs[pos];
1262
for (i = 1; i <= pos; i++)
1263
reps[i] = prevOpt->backs[i - 1];
1264
for (; i < LZMA_NUM_REPS; i++)
1265
reps[i] = prevOpt->backs[i];
1270
reps[0] = (pos - LZMA_NUM_REPS);
1271
for (i = 1; i < LZMA_NUM_REPS; i++)
1272
reps[i] = prevOpt->backs[i - 1];
1275
curOpt->state = (CState)state;
1277
curOpt->backs[0] = reps[0];
1278
curOpt->backs[1] = reps[1];
1279
curOpt->backs[2] = reps[2];
1280
curOpt->backs[3] = reps[3];
1282
curPrice = curOpt->price;
1284
data = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - 1;
1285
currentByte = *data;
1286
matchByte = *(data - (reps[0] + 1));
1288
posState = (position & p->pbMask);
1290
curAnd1Price = curPrice + GET_PRICE_0(p->isMatch[state][posState]);
1292
const CLzmaProb *probs = LIT_PROBS(position, *(data - 1));
1294
(!IsCharState(state) ?
1295
LitEnc_GetPriceMatched(probs, currentByte, matchByte, p->ProbPrices) :
1296
LitEnc_GetPrice(probs, currentByte, p->ProbPrices));
1299
nextOpt = &p->opt[cur + 1];
1301
if (curAnd1Price < nextOpt->price)
1303
nextOpt->price = curAnd1Price;
1304
nextOpt->posPrev = cur;
1305
MakeAsChar(nextOpt);
1309
matchPrice = curPrice + GET_PRICE_1(p->isMatch[state][posState]);
1310
repMatchPrice = matchPrice + GET_PRICE_1(p->isRep[state]);
1312
if (matchByte == currentByte && !(nextOpt->posPrev < cur && nextOpt->backPrev == 0))
1314
UInt32 shortRepPrice = repMatchPrice + GetRepLen1Price(p, state, posState);
1315
if (shortRepPrice <= nextOpt->price)
1317
nextOpt->price = shortRepPrice;
1318
nextOpt->posPrev = cur;
1319
MakeAsShortRep(nextOpt);
1325
UInt32 temp = kNumOpts - 1 - cur;
1326
if (temp < numAvailableBytesFull)
1327
numAvailableBytesFull = temp;
1329
numAvailableBytes = numAvailableBytesFull;
1331
if (numAvailableBytes < 2)
1333
if (numAvailableBytes > p->numFastBytes)
1334
numAvailableBytes = p->numFastBytes;
1335
if (!nextIsChar && matchByte != currentByte) /* speed optimization */
1337
/* try Literal + rep0 */
1340
const Byte *data2 = data - (reps[0] + 1);
1341
UInt32 limit = p->numFastBytes + 1;
1342
if (limit > numAvailableBytesFull)
1343
limit = numAvailableBytesFull;
1345
for (temp = 1; temp < limit && data[temp] == data2[temp]; temp++);
1346
lenTest2 = temp - 1;
1349
UInt32 state2 = kLiteralNextStates[state];
1350
UInt32 posStateNext = (position + 1) & p->pbMask;
1351
UInt32 nextRepMatchPrice = curAnd1Price +
1352
GET_PRICE_1(p->isMatch[state2][posStateNext]) +
1353
GET_PRICE_1(p->isRep[state2]);
1354
/* for (; lenTest2 >= 2; lenTest2--) */
1356
UInt32 curAndLenPrice;
1358
UInt32 offset = cur + 1 + lenTest2;
1359
while (lenEnd < offset)
1360
p->opt[++lenEnd].price = kInfinityPrice;
1361
curAndLenPrice = nextRepMatchPrice + GetRepPrice(p, 0, lenTest2, state2, posStateNext);
1362
opt = &p->opt[offset];
1363
if (curAndLenPrice < opt->price)
1365
opt->price = curAndLenPrice;
1366
opt->posPrev = cur + 1;
1368
opt->prev1IsChar = True;
1375
startLen = 2; /* speed optimization */
1378
for (repIndex = 0; repIndex < LZMA_NUM_REPS; repIndex++)
1383
const Byte *data2 = data - (reps[repIndex] + 1);
1384
if (data[0] != data2[0] || data[1] != data2[1])
1386
for (lenTest = 2; lenTest < numAvailableBytes && data[lenTest] == data2[lenTest]; lenTest++);
1387
while (lenEnd < cur + lenTest)
1388
p->opt[++lenEnd].price = kInfinityPrice;
1389
lenTestTemp = lenTest;
1390
price = repMatchPrice + GetPureRepPrice(p, repIndex, state, posState);
1393
UInt32 curAndLenPrice = price + p->repLenEnc.prices[posState][lenTest - 2];
1394
COptimal *opt = &p->opt[cur + lenTest];
1395
if (curAndLenPrice < opt->price)
1397
opt->price = curAndLenPrice;
1399
opt->backPrev = repIndex;
1400
opt->prev1IsChar = False;
1403
while (--lenTest >= 2);
1404
lenTest = lenTestTemp;
1407
startLen = lenTest + 1;
1411
UInt32 lenTest2 = lenTest + 1;
1412
UInt32 limit = lenTest2 + p->numFastBytes;
1413
UInt32 nextRepMatchPrice;
1414
if (limit > numAvailableBytesFull)
1415
limit = numAvailableBytesFull;
1416
for (; lenTest2 < limit && data[lenTest2] == data2[lenTest2]; lenTest2++);
1417
lenTest2 -= lenTest + 1;
1420
UInt32 state2 = kRepNextStates[state];
1421
UInt32 posStateNext = (position + lenTest) & p->pbMask;
1422
UInt32 curAndLenCharPrice =
1423
price + p->repLenEnc.prices[posState][lenTest - 2] +
1424
GET_PRICE_0(p->isMatch[state2][posStateNext]) +
1425
LitEnc_GetPriceMatched(LIT_PROBS(position + lenTest, data[lenTest - 1]),
1426
data[lenTest], data2[lenTest], p->ProbPrices);
1427
state2 = kLiteralNextStates[state2];
1428
posStateNext = (position + lenTest + 1) & p->pbMask;
1429
nextRepMatchPrice = curAndLenCharPrice +
1430
GET_PRICE_1(p->isMatch[state2][posStateNext]) +
1431
GET_PRICE_1(p->isRep[state2]);
1433
/* for (; lenTest2 >= 2; lenTest2--) */
1435
UInt32 curAndLenPrice;
1437
UInt32 offset = cur + lenTest + 1 + lenTest2;
1438
while (lenEnd < offset)
1439
p->opt[++lenEnd].price = kInfinityPrice;
1440
curAndLenPrice = nextRepMatchPrice + GetRepPrice(p, 0, lenTest2, state2, posStateNext);
1441
opt = &p->opt[offset];
1442
if (curAndLenPrice < opt->price)
1444
opt->price = curAndLenPrice;
1445
opt->posPrev = cur + lenTest + 1;
1447
opt->prev1IsChar = True;
1449
opt->posPrev2 = cur;
1450
opt->backPrev2 = repIndex;
1457
/* for (UInt32 lenTest = 2; lenTest <= newLen; lenTest++) */
1458
if (newLen > numAvailableBytes)
1460
newLen = numAvailableBytes;
1461
for (numDistancePairs = 0; newLen > matchDistances[numDistancePairs]; numDistancePairs += 2);
1462
matchDistances[numDistancePairs] = newLen;
1463
numDistancePairs += 2;
1465
if (newLen >= startLen)
1467
UInt32 normalMatchPrice = matchPrice + GET_PRICE_0(p->isRep[state]);
1468
UInt32 offs, curBack, posSlot;
1470
while (lenEnd < cur + newLen)
1471
p->opt[++lenEnd].price = kInfinityPrice;
1474
while (startLen > matchDistances[offs])
1476
curBack = matchDistances[offs + 1];
1477
GetPosSlot2(curBack, posSlot);
1478
for (lenTest = /*2*/ startLen; ; lenTest++)
1480
UInt32 curAndLenPrice = normalMatchPrice + p->lenEnc.prices[posState][lenTest - LZMA_MATCH_LEN_MIN];
1481
UInt32 lenToPosState = GetLenToPosState(lenTest);
1483
if (curBack < kNumFullDistances)
1484
curAndLenPrice += p->distancesPrices[lenToPosState][curBack];
1486
curAndLenPrice += p->posSlotPrices[lenToPosState][posSlot] + p->alignPrices[curBack & kAlignMask];
1488
opt = &p->opt[cur + lenTest];
1489
if (curAndLenPrice < opt->price)
1491
opt->price = curAndLenPrice;
1493
opt->backPrev = curBack + LZMA_NUM_REPS;
1494
opt->prev1IsChar = False;
1497
if (/*_maxMode && */lenTest == matchDistances[offs])
1499
/* Try Match + Literal + Rep0 */
1500
const Byte *data2 = data - (curBack + 1);
1501
UInt32 lenTest2 = lenTest + 1;
1502
UInt32 limit = lenTest2 + p->numFastBytes;
1503
UInt32 nextRepMatchPrice;
1504
if (limit > numAvailableBytesFull)
1505
limit = numAvailableBytesFull;
1506
for (; lenTest2 < limit && data[lenTest2] == data2[lenTest2]; lenTest2++);
1507
lenTest2 -= lenTest + 1;
1510
UInt32 state2 = kMatchNextStates[state];
1511
UInt32 posStateNext = (position + lenTest) & p->pbMask;
1512
UInt32 curAndLenCharPrice = curAndLenPrice +
1513
GET_PRICE_0(p->isMatch[state2][posStateNext]) +
1514
LitEnc_GetPriceMatched(LIT_PROBS(position + lenTest, data[lenTest - 1]),
1515
data[lenTest], data2[lenTest], p->ProbPrices);
1516
state2 = kLiteralNextStates[state2];
1517
posStateNext = (posStateNext + 1) & p->pbMask;
1518
nextRepMatchPrice = curAndLenCharPrice +
1519
GET_PRICE_1(p->isMatch[state2][posStateNext]) +
1520
GET_PRICE_1(p->isRep[state2]);
1522
/* for (; lenTest2 >= 2; lenTest2--) */
1524
UInt32 offset = cur + lenTest + 1 + lenTest2;
1525
UInt32 curAndLenPrice;
1527
while (lenEnd < offset)
1528
p->opt[++lenEnd].price = kInfinityPrice;
1529
curAndLenPrice = nextRepMatchPrice + GetRepPrice(p, 0, lenTest2, state2, posStateNext);
1530
opt = &p->opt[offset];
1531
if (curAndLenPrice < opt->price)
1533
opt->price = curAndLenPrice;
1534
opt->posPrev = cur + lenTest + 1;
1536
opt->prev1IsChar = True;
1538
opt->posPrev2 = cur;
1539
opt->backPrev2 = curBack + LZMA_NUM_REPS;
1544
if (offs == numDistancePairs)
1546
curBack = matchDistances[offs + 1];
1547
if (curBack >= kNumFullDistances)
1548
GetPosSlot2(curBack, posSlot);
1555
#define ChangePair(smallDist, bigDist) (((bigDist) >> 7) > (smallDist))
1557
static UInt32 GetOptimumFast(CLzmaEnc *p, UInt32 *backRes)
1559
UInt32 numAvailableBytes = p->matchFinder.GetNumAvailableBytes(p->matchFinderObj);
1560
UInt32 lenMain, numDistancePairs;
1562
UInt32 repLens[LZMA_NUM_REPS];
1563
UInt32 repMaxIndex, i;
1564
UInt32 *matchDistances;
1567
if (!p->longestMatchWasFound)
1569
lenMain = ReadMatchDistances(p, &numDistancePairs);
1573
lenMain = p->longestMatchLength;
1574
numDistancePairs = p->numDistancePairs;
1575
p->longestMatchWasFound = False;
1578
data = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - 1;
1579
if (numAvailableBytes > LZMA_MATCH_LEN_MAX)
1580
numAvailableBytes = LZMA_MATCH_LEN_MAX;
1581
if (numAvailableBytes < 2)
1583
*backRes = (UInt32)(-1);
1589
for (i = 0; i < LZMA_NUM_REPS; i++)
1591
const Byte *data2 = data - (p->reps[i] + 1);
1593
if (data[0] != data2[0] || data[1] != data2[1])
1598
for (len = 2; len < numAvailableBytes && data[len] == data2[len]; len++);
1599
if (len >= p->numFastBytes)
1602
MovePos(p, len - 1);
1606
if (len > repLens[repMaxIndex])
1609
matchDistances = p->matchDistances;
1610
if (lenMain >= p->numFastBytes)
1612
*backRes = matchDistances[numDistancePairs - 1] + LZMA_NUM_REPS;
1613
MovePos(p, lenMain - 1);
1617
backMain = 0; /* for GCC */
1620
backMain = matchDistances[numDistancePairs - 1];
1621
while (numDistancePairs > 2 && lenMain == matchDistances[numDistancePairs - 4] + 1)
1623
if (!ChangePair(matchDistances[numDistancePairs - 3], backMain))
1625
numDistancePairs -= 2;
1626
lenMain = matchDistances[numDistancePairs - 2];
1627
backMain = matchDistances[numDistancePairs - 1];
1629
if (lenMain == 2 && backMain >= 0x80)
1633
if (repLens[repMaxIndex] >= 2)
1635
if (repLens[repMaxIndex] + 1 >= lenMain ||
1636
(repLens[repMaxIndex] + 2 >= lenMain && (backMain > (1 << 9))) ||
1637
(repLens[repMaxIndex] + 3 >= lenMain && (backMain > (1 << 15))))
1640
*backRes = repMaxIndex;
1641
lenRes = repLens[repMaxIndex];
1642
MovePos(p, lenRes - 1);
1647
if (lenMain >= 2 && numAvailableBytes > 2)
1650
numAvailableBytes = p->matchFinder.GetNumAvailableBytes(p->matchFinderObj);
1651
p->longestMatchLength = ReadMatchDistances(p, &p->numDistancePairs);
1652
if (p->longestMatchLength >= 2)
1654
UInt32 newDistance = matchDistances[p->numDistancePairs - 1];
1655
if ((p->longestMatchLength >= lenMain && newDistance < backMain) ||
1656
(p->longestMatchLength == lenMain + 1 && !ChangePair(backMain, newDistance)) ||
1657
(p->longestMatchLength > lenMain + 1) ||
1658
(p->longestMatchLength + 1 >= lenMain && lenMain >= 3 && ChangePair(newDistance, backMain)))
1660
p->longestMatchWasFound = True;
1661
*backRes = (UInt32)(-1);
1665
data = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - 1;
1666
for (i = 0; i < LZMA_NUM_REPS; i++)
1669
const Byte *data2 = data - (p->reps[i] + 1);
1670
if (data[1] != data2[1] || data[2] != data2[2])
1675
for (len = 2; len < numAvailableBytes && data[len] == data2[len]; len++);
1676
if (len + 1 >= lenMain)
1678
p->longestMatchWasFound = True;
1679
*backRes = (UInt32)(-1);
1683
*backRes = backMain + LZMA_NUM_REPS;
1684
MovePos(p, lenMain - 2);
1687
*backRes = (UInt32)(-1);
1691
static void WriteEndMarker(CLzmaEnc *p, UInt32 posState)
1694
RangeEnc_EncodeBit(&p->rc, &p->isMatch[p->state][posState], 1);
1695
RangeEnc_EncodeBit(&p->rc, &p->isRep[p->state], 0);
1696
p->state = kMatchNextStates[p->state];
1697
len = LZMA_MATCH_LEN_MIN;
1698
LenEnc_Encode2(&p->lenEnc, &p->rc, len - LZMA_MATCH_LEN_MIN, posState, !p->fastMode, p->ProbPrices);
1699
RcTree_Encode(&p->rc, p->posSlotEncoder[GetLenToPosState(len)], kNumPosSlotBits, (1 << kNumPosSlotBits) - 1);
1700
RangeEnc_EncodeDirectBits(&p->rc, (((UInt32)1 << 30) - 1) >> kNumAlignBits, 30 - kNumAlignBits);
1701
RcTree_ReverseEncode(&p->rc, p->posAlignEncoder, kNumAlignBits, kAlignMask);
1704
static SRes CheckErrors(CLzmaEnc *p)
1706
if (p->result != SZ_OK)
1708
if (p->rc.res != SZ_OK)
1709
p->result = SZ_ERROR_WRITE;
1710
if (p->matchFinderBase.result != SZ_OK)
1711
p->result = SZ_ERROR_READ;
1712
if (p->result != SZ_OK)
1717
static SRes Flush(CLzmaEnc *p, UInt32 nowPos)
1719
/* ReleaseMFStream(); */
1721
if (p->writeEndMark)
1722
WriteEndMarker(p, nowPos & p->pbMask);
1723
RangeEnc_FlushData(&p->rc);
1724
RangeEnc_FlushStream(&p->rc);
1725
return CheckErrors(p);
1728
static void FillAlignPrices(CLzmaEnc *p)
1731
for (i = 0; i < kAlignTableSize; i++)
1732
p->alignPrices[i] = RcTree_ReverseGetPrice(p->posAlignEncoder, kNumAlignBits, i, p->ProbPrices);
1733
p->alignPriceCount = 0;
1736
static void FillDistancesPrices(CLzmaEnc *p)
1738
UInt32 tempPrices[kNumFullDistances];
1739
UInt32 i, lenToPosState;
1740
for (i = kStartPosModelIndex; i < kNumFullDistances; i++)
1742
UInt32 posSlot = GetPosSlot1(i);
1743
UInt32 footerBits = ((posSlot >> 1) - 1);
1744
UInt32 base = ((2 | (posSlot & 1)) << footerBits);
1745
tempPrices[i] = RcTree_ReverseGetPrice(p->posEncoders + base - posSlot - 1, footerBits, i - base, p->ProbPrices);
1748
for (lenToPosState = 0; lenToPosState < kNumLenToPosStates; lenToPosState++)
1751
const CLzmaProb *encoder = p->posSlotEncoder[lenToPosState];
1752
UInt32 *posSlotPrices = p->posSlotPrices[lenToPosState];
1753
for (posSlot = 0; posSlot < p->distTableSize; posSlot++)
1754
posSlotPrices[posSlot] = RcTree_GetPrice(encoder, kNumPosSlotBits, posSlot, p->ProbPrices);
1755
for (posSlot = kEndPosModelIndex; posSlot < p->distTableSize; posSlot++)
1756
posSlotPrices[posSlot] += ((((posSlot >> 1) - 1) - kNumAlignBits) << kNumBitPriceShiftBits);
1759
UInt32 *distancesPrices = p->distancesPrices[lenToPosState];
1761
for (i = 0; i < kStartPosModelIndex; i++)
1762
distancesPrices[i] = posSlotPrices[i];
1763
for (; i < kNumFullDistances; i++)
1764
distancesPrices[i] = posSlotPrices[GetPosSlot1(i)] + tempPrices[i];
1767
p->matchPriceCount = 0;
1770
void LzmaEnc_Construct(CLzmaEnc *p)
1772
RangeEnc_Construct(&p->rc);
1773
MatchFinder_Construct(&p->matchFinderBase);
1774
#ifdef COMPRESS_MF_MT
1775
MatchFinderMt_Construct(&p->matchFinderMt);
1776
p->matchFinderMt.MatchFinder = &p->matchFinderBase;
1780
CLzmaEncProps props;
1781
LzmaEncProps_Init(&props);
1782
LzmaEnc_SetProps(p, &props);
1785
#ifndef LZMA_LOG_BSR
1786
LzmaEnc_FastPosInit(p->g_FastPos);
1789
LzmaEnc_InitPriceTables(p->ProbPrices);
1791
p->saveState.litProbs = 0;
1794
CLzmaEncHandle LzmaEnc_Create(ISzAlloc *alloc)
1797
p = alloc->Alloc(alloc, sizeof(CLzmaEnc));
1799
LzmaEnc_Construct((CLzmaEnc *)p);
1803
void LzmaEnc_FreeLits(CLzmaEnc *p, ISzAlloc *alloc)
1805
alloc->Free(alloc, p->litProbs);
1806
alloc->Free(alloc, p->saveState.litProbs);
1808
p->saveState.litProbs = 0;
1811
void LzmaEnc_Destruct(CLzmaEnc *p, ISzAlloc *alloc, ISzAlloc *allocBig)
1813
#ifdef COMPRESS_MF_MT
1814
MatchFinderMt_Destruct(&p->matchFinderMt, allocBig);
1816
MatchFinder_Free(&p->matchFinderBase, allocBig);
1817
LzmaEnc_FreeLits(p, alloc);
1818
RangeEnc_Free(&p->rc, alloc);
1821
void LzmaEnc_Destroy(CLzmaEncHandle p, ISzAlloc *alloc, ISzAlloc *allocBig)
1823
LzmaEnc_Destruct((CLzmaEnc *)p, alloc, allocBig);
1824
alloc->Free(alloc, p);
1827
static SRes LzmaEnc_CodeOneBlock(CLzmaEnc *p, Bool useLimits, UInt32 maxPackSize, UInt32 maxUnpackSize)
1829
UInt32 nowPos32, startPos32;
1830
if (p->inStream != 0)
1832
p->matchFinderBase.stream = p->inStream;
1833
p->matchFinder.Init(p->matchFinderObj);
1839
RINOK(CheckErrors(p));
1841
nowPos32 = (UInt32)p->nowPos64;
1842
startPos32 = nowPos32;
1844
if (p->nowPos64 == 0)
1846
UInt32 numDistancePairs;
1848
if (p->matchFinder.GetNumAvailableBytes(p->matchFinderObj) == 0)
1849
return Flush(p, nowPos32);
1850
ReadMatchDistances(p, &numDistancePairs);
1851
RangeEnc_EncodeBit(&p->rc, &p->isMatch[p->state][0], 0);
1852
p->state = kLiteralNextStates[p->state];
1853
curByte = p->matchFinder.GetIndexByte(p->matchFinderObj, 0 - p->additionalOffset);
1854
LitEnc_Encode(&p->rc, p->litProbs, curByte);
1855
p->additionalOffset--;
1859
if (p->matchFinder.GetNumAvailableBytes(p->matchFinderObj) != 0)
1862
UInt32 pos, len, posState;
1865
len = GetOptimumFast(p, &pos);
1867
len = GetOptimum(p, nowPos32, &pos);
1870
printf("\n pos = %4X, len = %d pos = %d", nowPos32, len, pos);
1873
posState = nowPos32 & p->pbMask;
1874
if (len == 1 && pos == 0xFFFFFFFF)
1880
RangeEnc_EncodeBit(&p->rc, &p->isMatch[p->state][posState], 0);
1881
data = p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - p->additionalOffset;
1883
probs = LIT_PROBS(nowPos32, *(data - 1));
1884
if (IsCharState(p->state))
1885
LitEnc_Encode(&p->rc, probs, curByte);
1887
LitEnc_EncodeMatched(&p->rc, probs, curByte, *(data - p->reps[0] - 1));
1888
p->state = kLiteralNextStates[p->state];
1892
RangeEnc_EncodeBit(&p->rc, &p->isMatch[p->state][posState], 1);
1893
if (pos < LZMA_NUM_REPS)
1895
RangeEnc_EncodeBit(&p->rc, &p->isRep[p->state], 1);
1898
RangeEnc_EncodeBit(&p->rc, &p->isRepG0[p->state], 0);
1899
RangeEnc_EncodeBit(&p->rc, &p->isRep0Long[p->state][posState], ((len == 1) ? 0 : 1));
1903
UInt32 distance = p->reps[pos];
1904
RangeEnc_EncodeBit(&p->rc, &p->isRepG0[p->state], 1);
1906
RangeEnc_EncodeBit(&p->rc, &p->isRepG1[p->state], 0);
1909
RangeEnc_EncodeBit(&p->rc, &p->isRepG1[p->state], 1);
1910
RangeEnc_EncodeBit(&p->rc, &p->isRepG2[p->state], pos - 2);
1912
p->reps[3] = p->reps[2];
1913
p->reps[2] = p->reps[1];
1915
p->reps[1] = p->reps[0];
1916
p->reps[0] = distance;
1919
p->state = kShortRepNextStates[p->state];
1922
LenEnc_Encode2(&p->repLenEnc, &p->rc, len - LZMA_MATCH_LEN_MIN, posState, !p->fastMode, p->ProbPrices);
1923
p->state = kRepNextStates[p->state];
1929
RangeEnc_EncodeBit(&p->rc, &p->isRep[p->state], 0);
1930
p->state = kMatchNextStates[p->state];
1931
LenEnc_Encode2(&p->lenEnc, &p->rc, len - LZMA_MATCH_LEN_MIN, posState, !p->fastMode, p->ProbPrices);
1932
pos -= LZMA_NUM_REPS;
1933
GetPosSlot(pos, posSlot);
1934
RcTree_Encode(&p->rc, p->posSlotEncoder[GetLenToPosState(len)], kNumPosSlotBits, posSlot);
1936
if (posSlot >= kStartPosModelIndex)
1938
UInt32 footerBits = ((posSlot >> 1) - 1);
1939
UInt32 base = ((2 | (posSlot & 1)) << footerBits);
1940
UInt32 posReduced = pos - base;
1942
if (posSlot < kEndPosModelIndex)
1943
RcTree_ReverseEncode(&p->rc, p->posEncoders + base - posSlot - 1, footerBits, posReduced);
1946
RangeEnc_EncodeDirectBits(&p->rc, posReduced >> kNumAlignBits, footerBits - kNumAlignBits);
1947
RcTree_ReverseEncode(&p->rc, p->posAlignEncoder, kNumAlignBits, posReduced & kAlignMask);
1948
p->alignPriceCount++;
1951
p->reps[3] = p->reps[2];
1952
p->reps[2] = p->reps[1];
1953
p->reps[1] = p->reps[0];
1955
p->matchPriceCount++;
1958
p->additionalOffset -= len;
1960
if (p->additionalOffset == 0)
1965
if (p->matchPriceCount >= (1 << 7))
1966
FillDistancesPrices(p);
1967
if (p->alignPriceCount >= kAlignTableSize)
1970
if (p->matchFinder.GetNumAvailableBytes(p->matchFinderObj) == 0)
1972
processed = nowPos32 - startPos32;
1975
if (processed + kNumOpts + 300 >= maxUnpackSize ||
1976
RangeEnc_GetProcessed(&p->rc) + kNumOpts * 2 >= maxPackSize)
1979
else if (processed >= (1 << 15))
1981
p->nowPos64 += nowPos32 - startPos32;
1982
return CheckErrors(p);
1986
p->nowPos64 += nowPos32 - startPos32;
1987
return Flush(p, nowPos32);
1990
#define kBigHashDicLimit ((UInt32)1 << 24)
1992
static SRes LzmaEnc_Alloc(CLzmaEnc *p, UInt32 keepWindowSize, ISzAlloc *alloc, ISzAlloc *allocBig)
1994
UInt32 beforeSize = kNumOpts;
1996
if (!RangeEnc_Alloc(&p->rc, alloc))
1997
return SZ_ERROR_MEM;
1998
btMode = (p->matchFinderBase.btMode != 0);
1999
#ifdef COMPRESS_MF_MT
2000
p->mtMode = (p->multiThread && !p->fastMode && btMode);
2004
unsigned lclp = p->lc + p->lp;
2005
if (p->litProbs == 0 || p->saveState.litProbs == 0 || p->lclp != lclp)
2007
LzmaEnc_FreeLits(p, alloc);
2008
p->litProbs = (CLzmaProb *)alloc->Alloc(alloc, (0x300 << lclp) * sizeof(CLzmaProb));
2009
p->saveState.litProbs = (CLzmaProb *)alloc->Alloc(alloc, (0x300 << lclp) * sizeof(CLzmaProb));
2010
if (p->litProbs == 0 || p->saveState.litProbs == 0)
2012
LzmaEnc_FreeLits(p, alloc);
2013
return SZ_ERROR_MEM;
2019
p->matchFinderBase.bigHash = (p->dictSize > kBigHashDicLimit);
2021
if (beforeSize + p->dictSize < keepWindowSize)
2022
beforeSize = keepWindowSize - p->dictSize;
2024
#ifdef COMPRESS_MF_MT
2027
RINOK(MatchFinderMt_Create(&p->matchFinderMt, p->dictSize, beforeSize, p->numFastBytes, LZMA_MATCH_LEN_MAX, allocBig));
2028
p->matchFinderObj = &p->matchFinderMt;
2029
MatchFinderMt_CreateVTable(&p->matchFinderMt, &p->matchFinder);
2034
if (!MatchFinder_Create(&p->matchFinderBase, p->dictSize, beforeSize, p->numFastBytes, LZMA_MATCH_LEN_MAX, allocBig))
2035
return SZ_ERROR_MEM;
2036
p->matchFinderObj = &p->matchFinderBase;
2037
MatchFinder_CreateVTable(&p->matchFinderBase, &p->matchFinder);
2042
void LzmaEnc_Init(CLzmaEnc *p)
2046
for(i = 0 ; i < LZMA_NUM_REPS; i++)
2049
RangeEnc_Init(&p->rc);
2052
for (i = 0; i < kNumStates; i++)
2055
for (j = 0; j < LZMA_NUM_PB_STATES_MAX; j++)
2057
p->isMatch[i][j] = kProbInitValue;
2058
p->isRep0Long[i][j] = kProbInitValue;
2060
p->isRep[i] = kProbInitValue;
2061
p->isRepG0[i] = kProbInitValue;
2062
p->isRepG1[i] = kProbInitValue;
2063
p->isRepG2[i] = kProbInitValue;
2067
UInt32 num = 0x300 << (p->lp + p->lc);
2068
for (i = 0; i < num; i++)
2069
p->litProbs[i] = kProbInitValue;
2073
for (i = 0; i < kNumLenToPosStates; i++)
2075
CLzmaProb *probs = p->posSlotEncoder[i];
2077
for (j = 0; j < (1 << kNumPosSlotBits); j++)
2078
probs[j] = kProbInitValue;
2082
for(i = 0; i < kNumFullDistances - kEndPosModelIndex; i++)
2083
p->posEncoders[i] = kProbInitValue;
2086
LenEnc_Init(&p->lenEnc.p);
2087
LenEnc_Init(&p->repLenEnc.p);
2089
for (i = 0; i < (1 << kNumAlignBits); i++)
2090
p->posAlignEncoder[i] = kProbInitValue;
2092
p->longestMatchWasFound = False;
2093
p->optimumEndIndex = 0;
2094
p->optimumCurrentIndex = 0;
2095
p->additionalOffset = 0;
2097
p->pbMask = (1 << p->pb) - 1;
2098
p->lpMask = (1 << p->lp) - 1;
2101
void LzmaEnc_InitPrices(CLzmaEnc *p)
2105
FillDistancesPrices(p);
2109
p->lenEnc.tableSize =
2110
p->repLenEnc.tableSize =
2111
p->numFastBytes + 1 - LZMA_MATCH_LEN_MIN;
2112
LenPriceEnc_UpdateTables(&p->lenEnc, 1 << p->pb, p->ProbPrices);
2113
LenPriceEnc_UpdateTables(&p->repLenEnc, 1 << p->pb, p->ProbPrices);
2116
static SRes LzmaEnc_AllocAndInit(CLzmaEnc *p, UInt32 keepWindowSize, ISzAlloc *alloc, ISzAlloc *allocBig)
2119
for (i = 0; i < (UInt32)kDicLogSizeMaxCompress; i++)
2120
if (p->dictSize <= ((UInt32)1 << i))
2122
p->distTableSize = i * 2;
2124
p->finished = False;
2126
RINOK(LzmaEnc_Alloc(p, keepWindowSize, alloc, allocBig));
2128
LzmaEnc_InitPrices(p);
2133
static SRes LzmaEnc_Prepare(CLzmaEncHandle pp, ISeqInStream *inStream, ISeqOutStream *outStream,
2134
ISzAlloc *alloc, ISzAlloc *allocBig)
2136
CLzmaEnc *p = (CLzmaEnc *)pp;
2137
p->inStream = inStream;
2138
p->rc.outStream = outStream;
2139
return LzmaEnc_AllocAndInit(p, 0, alloc, allocBig);
2142
SRes LzmaEnc_PrepareForLzma2(CLzmaEncHandle pp,
2143
ISeqInStream *inStream, UInt32 keepWindowSize,
2144
ISzAlloc *alloc, ISzAlloc *allocBig)
2146
CLzmaEnc *p = (CLzmaEnc *)pp;
2147
p->inStream = inStream;
2148
return LzmaEnc_AllocAndInit(p, keepWindowSize, alloc, allocBig);
2151
static void LzmaEnc_SetInputBuf(CLzmaEnc *p, const Byte *src, SizeT srcLen)
2153
p->seqBufInStream.funcTable.Read = MyRead;
2154
p->seqBufInStream.data = src;
2155
p->seqBufInStream.rem = srcLen;
2158
SRes LzmaEnc_MemPrepare(CLzmaEncHandle pp, const Byte *src, SizeT srcLen,
2159
UInt32 keepWindowSize, ISzAlloc *alloc, ISzAlloc *allocBig)
2161
CLzmaEnc *p = (CLzmaEnc *)pp;
2162
LzmaEnc_SetInputBuf(p, src, srcLen);
2163
p->inStream = &p->seqBufInStream.funcTable;
2164
return LzmaEnc_AllocAndInit(p, keepWindowSize, alloc, allocBig);
2167
void LzmaEnc_Finish(CLzmaEncHandle pp)
2169
#ifdef COMPRESS_MF_MT
2170
CLzmaEnc *p = (CLzmaEnc *)pp;
2172
MatchFinderMt_ReleaseStream(&p->matchFinderMt);
2178
typedef struct _CSeqOutStreamBuf
2180
ISeqOutStream funcTable;
2186
static size_t MyWrite(void *pp, const void *data, size_t size)
2188
CSeqOutStreamBuf *p = (CSeqOutStreamBuf *)pp;
2194
memcpy(p->data, data, size);
2201
UInt32 LzmaEnc_GetNumAvailableBytes(CLzmaEncHandle pp)
2203
const CLzmaEnc *p = (CLzmaEnc *)pp;
2204
return p->matchFinder.GetNumAvailableBytes(p->matchFinderObj);
2207
const Byte *LzmaEnc_GetCurBuf(CLzmaEncHandle pp)
2209
const CLzmaEnc *p = (CLzmaEnc *)pp;
2210
return p->matchFinder.GetPointerToCurrentPos(p->matchFinderObj) - p->additionalOffset;
2213
SRes LzmaEnc_CodeOneMemBlock(CLzmaEncHandle pp, Bool reInit,
2214
Byte *dest, size_t *destLen, UInt32 desiredPackSize, UInt32 *unpackSize)
2216
CLzmaEnc *p = (CLzmaEnc *)pp;
2219
CSeqOutStreamBuf outStream;
2221
outStream.funcTable.Write = MyWrite;
2222
outStream.data = dest;
2223
outStream.rem = *destLen;
2224
outStream.overflow = False;
2226
p->writeEndMark = False;
2227
p->finished = False;
2232
LzmaEnc_InitPrices(p);
2233
nowPos64 = p->nowPos64;
2234
RangeEnc_Init(&p->rc);
2235
p->rc.outStream = &outStream.funcTable;
2237
res = LzmaEnc_CodeOneBlock(pp, True, desiredPackSize, *unpackSize);
2239
*unpackSize = (UInt32)(p->nowPos64 - nowPos64);
2240
*destLen -= outStream.rem;
2241
if (outStream.overflow)
2242
return SZ_ERROR_OUTPUT_EOF;
2247
SRes LzmaEnc_Encode(CLzmaEncHandle pp, ISeqOutStream *outStream, ISeqInStream *inStream, ICompressProgress *progress,
2248
ISzAlloc *alloc, ISzAlloc *allocBig)
2250
CLzmaEnc *p = (CLzmaEnc *)pp;
2253
#ifdef COMPRESS_MF_MT
2254
Byte allocaDummy[0x300];
2256
for (i = 0; i < 16; i++)
2257
allocaDummy[i] = (Byte)i;
2260
RINOK(LzmaEnc_Prepare(pp, inStream, outStream, alloc, allocBig));
2264
res = LzmaEnc_CodeOneBlock(pp, False, 0, 0);
2265
if (res != SZ_OK || p->finished != 0)
2269
res = progress->Progress(progress, p->nowPos64, RangeEnc_GetProcessed(&p->rc));
2272
res = SZ_ERROR_PROGRESS;
2281
SRes LzmaEnc_WriteProperties(CLzmaEncHandle pp, Byte *props, SizeT *size)
2283
CLzmaEnc *p = (CLzmaEnc *)pp;
2285
UInt32 dictSize = p->dictSize;
2286
if (*size < LZMA_PROPS_SIZE)
2287
return SZ_ERROR_PARAM;
2288
*size = LZMA_PROPS_SIZE;
2289
props[0] = (Byte)((p->pb * 5 + p->lp) * 9 + p->lc);
2291
for (i = 11; i <= 30; i++)
2293
if (dictSize <= ((UInt32)2 << i))
2295
dictSize = (2 << i);
2298
if (dictSize <= ((UInt32)3 << i))
2300
dictSize = (3 << i);
2305
for (i = 0; i < 4; i++)
2306
props[1 + i] = (Byte)(dictSize >> (8 * i));
2310
SRes LzmaEnc_MemEncode(CLzmaEncHandle pp, Byte *dest, SizeT *destLen, const Byte *src, SizeT srcLen,
2311
int writeEndMark, ICompressProgress *progress, ISzAlloc *alloc, ISzAlloc *allocBig)
2314
CLzmaEnc *p = (CLzmaEnc *)pp;
2316
CSeqOutStreamBuf outStream;
2318
LzmaEnc_SetInputBuf(p, src, srcLen);
2320
outStream.funcTable.Write = MyWrite;
2321
outStream.data = dest;
2322
outStream.rem = *destLen;
2323
outStream.overflow = False;
2325
p->writeEndMark = writeEndMark;
2326
res = LzmaEnc_Encode(pp, &outStream.funcTable, &p->seqBufInStream.funcTable,
2327
progress, alloc, allocBig);
2329
*destLen -= outStream.rem;
2330
if (outStream.overflow)
2331
return SZ_ERROR_OUTPUT_EOF;
2335
SRes LzmaEncode(Byte *dest, SizeT *destLen, const Byte *src, SizeT srcLen,
2336
const CLzmaEncProps *props, Byte *propsEncoded, SizeT *propsSize, int writeEndMark,
2337
ICompressProgress *progress, ISzAlloc *alloc, ISzAlloc *allocBig)
2339
CLzmaEnc *p = (CLzmaEnc *)LzmaEnc_Create(alloc);
2342
return SZ_ERROR_MEM;
2344
res = LzmaEnc_SetProps(p, props);
2347
res = LzmaEnc_WriteProperties(p, propsEncoded, propsSize);
2349
res = LzmaEnc_MemEncode(p, dest, destLen, src, srcLen,
2350
writeEndMark, progress, alloc, allocBig);
2353
LzmaEnc_Destroy(p, alloc, allocBig);