C/Lzma86Enc.c

   1 /* Lzma86Enc.c -- LZMA + x86 (BCJ) Filter Encoder\r
   2 2009-08-14 : Igor Pavlov : Public domain */\r
   3 \r
   4 #include <string.h>\r
   5 \r
   6 #include "Lzma86.h"\r
   7 \r
   8 #include "Alloc.h"\r
   9 #include "Bra.h"\r
  10 #include "LzmaEnc.h"\r
  11 \r
  12 #define SZE_OUT_OVERFLOW SZE_DATA_ERROR\r
  13 \r
  14 static void *SzAlloc(void *p, size_t size) { p = p; return MyAlloc(size); }\r
  15 static void SzFree(void *p, void *address) { p = p; MyFree(address); }\r
  16 \r
  17 int Lzma86_Encode(Byte *dest, size_t *destLen, const Byte *src, size_t srcLen,\r
  18     int level, UInt32 dictSize, int filterMode)\r
  19 {\r
  20   ISzAlloc g_Alloc = { SzAlloc, SzFree };\r
  21   size_t outSize2 = *destLen;\r
  22   Byte *filteredStream;\r
  23   Bool useFilter;\r
  24   int mainResult = SZ_ERROR_OUTPUT_EOF;\r
  25   CLzmaEncProps props;\r
  26   LzmaEncProps_Init(&props);\r
  27   props.level = level;\r
  28   props.dictSize = dictSize;\r
  29   \r
  30   *destLen = 0;\r
  31   if (outSize2 < LZMA86_HEADER_SIZE)\r
  32     return SZ_ERROR_OUTPUT_EOF;\r
  33 \r
  34   {\r
  35     int i;\r
  36     UInt64 t = srcLen;\r
  37     for (i = 0; i < 8; i++, t >>= 8)\r
  38       dest[LZMA86_SIZE_OFFSET + i] = (Byte)t;\r
  39   }\r
  40 \r
  41   filteredStream = 0;\r
  42   useFilter = (filterMode != SZ_FILTER_NO);\r
  43   if (useFilter)\r
  44   {\r
  45     if (srcLen != 0)\r
  46     {\r
  47       filteredStream = (Byte *)MyAlloc(srcLen);\r
  48       if (filteredStream == 0)\r
  49         return SZ_ERROR_MEM;\r
  50       memcpy(filteredStream, src, srcLen);\r
  51     }\r
  52     {\r
  53       UInt32 x86State;\r
  54       x86_Convert_Init(x86State);\r
  55       x86_Convert(filteredStream, srcLen, 0, &x86State, 1);\r
  56     }\r
  57   }\r
  58 \r
  59   {\r
  60     size_t minSize = 0;\r
  61     Bool bestIsFiltered = False;\r
  62 \r
  63     /* passes for SZ_FILTER_AUTO:\r
  64         0 - BCJ + LZMA\r
  65         1 - LZMA\r
  66         2 - BCJ + LZMA agaian, if pass 0 (BCJ + LZMA) is better.\r
  67     */\r
  68     int numPasses = (filterMode == SZ_FILTER_AUTO) ? 3 : 1;\r
  69 \r
  70     int i;\r
  71     for (i = 0; i < numPasses; i++)\r
  72     {\r
  73       size_t outSizeProcessed = outSize2 - LZMA86_HEADER_SIZE;\r
  74       size_t outPropsSize = 5;\r
  75       SRes curRes;\r
  76       Bool curModeIsFiltered = (numPasses > 1 && i == numPasses - 1);\r
  77       if (curModeIsFiltered && !bestIsFiltered)\r
  78         break;\r
  79       if (useFilter && i == 0)\r
  80         curModeIsFiltered = True;\r
  81       \r
  82       curRes = LzmaEncode(dest + LZMA86_HEADER_SIZE, &outSizeProcessed,\r
  83           curModeIsFiltered ? filteredStream : src, srcLen,\r
  84           &props, dest + 1, &outPropsSize, 0,\r
  85           NULL, &g_Alloc, &g_Alloc);\r
  86       \r
  87       if (curRes != SZ_ERROR_OUTPUT_EOF)\r
  88       {\r
  89         if (curRes != SZ_OK)\r
  90         {\r
  91           mainResult = curRes;\r
  92           break;\r
  93         }\r
  94         if (outSizeProcessed <= minSize || mainResult != SZ_OK)\r
  95         {\r
  96           minSize = outSizeProcessed;\r
  97           bestIsFiltered = curModeIsFiltered;\r
  98           mainResult = SZ_OK;\r
  99         }\r
 100       }\r
 101     }\r
 102     dest[0] = (bestIsFiltered ? 1 : 0);\r
 103     *destLen = LZMA86_HEADER_SIZE + minSize;\r
 104   }\r
 105   if (useFilter)\r
 106     MyFree(filteredStream);\r
 107   return mainResult;\r
 108 }\r