tcompression.c 77.8 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

/* README.md   TAOS compression
 *
 * INTEGER Compression Algorithm:
S
Shengliang Guan 已提交
19
 *   To compress integers (including char, short, int32_t, int64_t), the difference
H
hzcheng 已提交
20 21
 *   between two integers is calculated at first. Then the difference is
 *   transformed to positive by zig-zag encoding method
22
 *   (https://gist.github.com/mfuerstenau/ba870a29e16536fdbaba). Then the value is
H
hzcheng 已提交
23 24 25
 *   encoded using simple 8B method. For more information about simple 8B,
 *   refer to https://en.wikipedia.org/wiki/8b/10b_encoding.
 *
26
 *   NOTE : For bigint, only 59 bits can be used, which means data from -(2**59) to (2**59)-1
H
hzcheng 已提交
27 28 29
 *   are allowed.
 *
 * BOOLEAN Compression Algorithm:
30
 *   We provide two methods for compress boolean types. Because boolean types in C
31
 *   code are char bytes with 0 and 1 values only, only one bit can used to discriminate
H
hzcheng 已提交
32
 *   the values.
33
 *   1. The first method is using only 1 bit to represent the boolean value with 1 for
H
hzcheng 已提交
34
 *   true and 0 for false. Then the compression rate is 1/8.
35
 *   2. The second method is using run length encoding (RLE) methods. This method works
H
hzcheng 已提交
36 37 38 39 40 41
 *   better when there are a lot of consecutive true values or false values.
 *
 * STRING Compression Algorithm:
 *   We us LZ4 method to compress the string type.
 *
 * FLOAT Compression Algorithm:
42 43 44 45 46
 *   We use the same method with Akumuli to compress float and double types. The compression
 *   algorithm assumes the float/double values change slightly. So we take the XOR between two
 *   adjacent values. Then compare the number of leading zeros and trailing zeros. If the number
 *   of leading zeros are larger than the trailing zeros, then record the last serveral bytes
 *   of the XORed value with informations. If not, record the first corresponding bytes.
H
hzcheng 已提交
47 48 49
 *
 */

S
Shengliang Guan 已提交
50
#define _DEFAULT_SOURCE
H
Haojun Liao 已提交
51
#include "tcompression.h"
S
Shengliang Guan 已提交
52
#include "lz4.h"
H
Hongze Cheng 已提交
53
#include "tRealloc.h"
S
log  
Shengliang Guan 已提交
54
#include "tlog.h"
H
hzcheng 已提交
55

S
Shengliang Guan 已提交
56 57 58 59 60 61
#ifdef TD_TSZ
#include "td_sz.h"
#endif

static const int32_t TEST_NUMBER = 1;
#define is_bigendian()     ((*(char *)&TEST_NUMBER) == 0)
H
Hongze Cheng 已提交
62
#define SIMPLE8B_MAX_INT64 ((uint64_t)1152921504606846974LL)
H
hzcheng 已提交
63

S
Shengliang Guan 已提交
64
#define safeInt64Add(a, b)  (((a >= 0) && (b <= INT64_MAX - a)) || ((a < 0) && (b >= INT64_MIN - a)))
H
Hongze Cheng 已提交
65 66
#define ZIGZAG_ENCODE(T, v) (((u##T)((v) >> (sizeof(T) * 8 - 1))) ^ (((u##T)(v)) << 1))  // zigzag encode
#define ZIGZAG_DECODE(T, v) (((v) >> 1) ^ -((T)((v)&1)))                                 // zigzag decode
H
hzcheng 已提交
67

T
tickduan 已提交
68
#ifdef TD_TSZ
S
Shengliang Guan 已提交
69
bool lossyFloat = false;
T
tickduan 已提交
70 71 72
bool lossyDouble = false;

// init call
S
Shengliang Guan 已提交
73 74 75
int32_t tsCompressInit() {
  // config
  if (lossyColumns[0] == 0) {
T
tickduan 已提交
76 77 78 79 80
    lossyFloat = false;
    lossyDouble = false;
    return 0;
  }

S
Shengliang Guan 已提交
81
  lossyFloat = strstr(lossyColumns, "float") != NULL;
T
tickduan 已提交
82 83
  lossyDouble = strstr(lossyColumns, "double") != NULL;

S
Shengliang Guan 已提交
84 85
  if (lossyFloat == false && lossyDouble == false) return 0;

T
tickduan 已提交
86
  tdszInit(fPrecision, dPrecision, maxRange, curRange, Compressor);
87 88
  if (lossyFloat) uTrace("lossy compression float  is opened. ");
  if (lossyDouble) uTrace("lossy compression double is opened. ");
T
tickduan 已提交
89 90 91
  return 1;
}
// exit call
S
Shengliang Guan 已提交
92
void tsCompressExit() { tdszExit(); }
T
tickduan 已提交
93

94
#endif
T
tickduan 已提交
95

H
hzcheng 已提交
96 97 98
/*
 * Compress Integer (Simple8B).
 */
S
Shengliang Guan 已提交
99
int32_t tsCompressINTImp(const char *const input, const int32_t nelements, char *const output, const char type) {
H
hzcheng 已提交
100 101
  // Selector value:              0    1   2   3   4   5   6   7   8  9  10  11
  // 12  13  14  15
S
Shengliang Guan 已提交
102 103 104
  char    bit_per_integer[] = {0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 15, 20, 30, 60};
  int32_t selector_to_elems[] = {240, 120, 60, 30, 20, 15, 12, 10, 8, 7, 6, 5, 4, 3, 2, 1};
  char    bit_to_selector[] = {0,  2,  3,  4,  5,  6,  7,  8,  9,  10, 10, 11, 11, 12, 12, 12, 13, 13, 13, 13, 13,
H
Hongze Cheng 已提交
105 106
                               14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15,
                               15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15};
H
hzcheng 已提交
107 108

  // get the byte limit.
S
Shengliang Guan 已提交
109
  int32_t word_length = 0;
H
hzcheng 已提交
110 111 112 113 114 115 116 117 118 119 120 121 122 123
  switch (type) {
    case TSDB_DATA_TYPE_BIGINT:
      word_length = LONG_BYTES;
      break;
    case TSDB_DATA_TYPE_INT:
      word_length = INT_BYTES;
      break;
    case TSDB_DATA_TYPE_SMALLINT:
      word_length = SHORT_BYTES;
      break;
    case TSDB_DATA_TYPE_TINYINT:
      word_length = CHAR_BYTES;
      break;
    default:
124
      uError("Invalid compress integer type:%d", type);
H
Hongze Cheng 已提交
125
      return -1;
H
hzcheng 已提交
126 127
  }

S
Shengliang Guan 已提交
128 129
  int32_t byte_limit = nelements * word_length + 1;
  int32_t opos = 1;
H
hzcheng 已提交
130 131
  int64_t prev_value = 0;

S
Shengliang Guan 已提交
132
  for (int32_t i = 0; i < nelements;) {
H
hzcheng 已提交
133 134
    char    selector = 0;
    char    bit = 0;
S
Shengliang Guan 已提交
135
    int32_t elems = 0;
H
hzcheng 已提交
136 137
    int64_t prev_value_tmp = prev_value;

S
Shengliang Guan 已提交
138
    for (int32_t j = i; j < nelements; j++) {
H
hzcheng 已提交
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
      // Read data from the input stream and convert it to INT64 type.
      int64_t curr_value = 0;
      switch (type) {
        case TSDB_DATA_TYPE_TINYINT:
          curr_value = (int64_t)(*((int8_t *)input + j));
          break;
        case TSDB_DATA_TYPE_SMALLINT:
          curr_value = (int64_t)(*((int16_t *)input + j));
          break;
        case TSDB_DATA_TYPE_INT:
          curr_value = (int64_t)(*((int32_t *)input + j));
          break;
        case TSDB_DATA_TYPE_BIGINT:
          curr_value = (int64_t)(*((int64_t *)input + j));
          break;
      }
      // Get difference.
Y
yihaoDeng 已提交
156
      if (!safeInt64Add(curr_value, -prev_value_tmp)) goto _copy_and_exit;
H
hzcheng 已提交
157 158 159

      int64_t diff = curr_value - prev_value_tmp;
      // Zigzag encode the value.
H
Hongze Cheng 已提交
160
      uint64_t zigzag_value = ZIGZAG_ENCODE(int64_t, diff);
H
hzcheng 已提交
161 162 163

      if (zigzag_value >= SIMPLE8B_MAX_INT64) goto _copy_and_exit;

S
TD-1037  
Shengliang Guan 已提交
164
      int64_t tmp_bit;
H
hzcheng 已提交
165 166 167 168
      if (zigzag_value == 0) {
        // Take care here, __builtin_clzl give wrong anser for value 0;
        tmp_bit = 0;
      } else {
169
        tmp_bit = (LONG_BYTES * BITS_PER_BYTE) - BUILDIN_CLZL(zigzag_value);
H
hzcheng 已提交
170 171
      }

S
Shengliang Guan 已提交
172 173
      if (elems + 1 <= selector_to_elems[(int32_t)selector] &&
          elems + 1 <= selector_to_elems[(int32_t)(bit_to_selector[(int32_t)tmp_bit])]) {
H
hzcheng 已提交
174
        // If can hold another one.
S
Shengliang Guan 已提交
175
        selector = selector > bit_to_selector[(int32_t)tmp_bit] ? selector : bit_to_selector[(int32_t)tmp_bit];
H
hzcheng 已提交
176
        elems++;
S
Shengliang Guan 已提交
177
        bit = bit_per_integer[(int32_t)selector];
H
hzcheng 已提交
178 179
      } else {
        // if cannot hold another one.
S
Shengliang Guan 已提交
180 181 182
        while (elems < selector_to_elems[(int32_t)selector]) selector++;
        elems = selector_to_elems[(int32_t)selector];
        bit = bit_per_integer[(int32_t)selector];
H
hzcheng 已提交
183 184 185 186 187 188 189
        break;
      }
      prev_value_tmp = curr_value;
    }

    uint64_t buffer = 0;
    buffer |= (uint64_t)selector;
S
Shengliang Guan 已提交
190
    for (int32_t k = 0; k < elems; k++) {
H
hzcheng 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206
      int64_t curr_value = 0; /* get current values */
      switch (type) {
        case TSDB_DATA_TYPE_TINYINT:
          curr_value = (int64_t)(*((int8_t *)input + i));
          break;
        case TSDB_DATA_TYPE_SMALLINT:
          curr_value = (int64_t)(*((int16_t *)input + i));
          break;
        case TSDB_DATA_TYPE_INT:
          curr_value = (int64_t)(*((int32_t *)input + i));
          break;
        case TSDB_DATA_TYPE_BIGINT:
          curr_value = (int64_t)(*((int64_t *)input + i));
          break;
      }
      int64_t  diff = curr_value - prev_value;
H
Hongze Cheng 已提交
207
      uint64_t zigzag_value = ZIGZAG_ENCODE(int64_t, diff);
H
hzcheng 已提交
208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229
      buffer |= ((zigzag_value & INT64MASK(bit)) << (bit * k + 4));
      i++;
      prev_value = curr_value;
    }

    // Output the encoded value to the output.
    if (opos + sizeof(buffer) <= byte_limit) {
      memcpy(output + opos, &buffer, sizeof(buffer));
      opos += sizeof(buffer);
    } else {
    _copy_and_exit:
      output[0] = 1;
      memcpy(output + 1, input, byte_limit - 1);
      return byte_limit;
    }
  }

  // set the indicator.
  output[0] = 0;
  return opos;
}

S
Shengliang Guan 已提交
230
int32_t tsDecompressINTImp(const char *const input, const int32_t nelements, char *const output, const char type) {
231

S
Shengliang Guan 已提交
232
  int32_t word_length = 0;
H
hzcheng 已提交
233 234 235 236 237 238 239 240 241 242 243 244 245 246
  switch (type) {
    case TSDB_DATA_TYPE_BIGINT:
      word_length = LONG_BYTES;
      break;
    case TSDB_DATA_TYPE_INT:
      word_length = INT_BYTES;
      break;
    case TSDB_DATA_TYPE_SMALLINT:
      word_length = SHORT_BYTES;
      break;
    case TSDB_DATA_TYPE_TINYINT:
      word_length = CHAR_BYTES;
      break;
    default:
247
      uError("Invalid decompress integer type:%d", type);
H
Hongze Cheng 已提交
248
      return -1;
H
hzcheng 已提交
249 250 251 252 253 254 255 256 257 258
  }

  // If not compressed.
  if (input[0] == 1) {
    memcpy(output, input + 1, nelements * word_length);
    return nelements * word_length;
  }

  // Selector value:              0    1   2   3   4   5   6   7   8  9  10  11
  // 12  13  14  15
S
Shengliang Guan 已提交
259 260
  char    bit_per_integer[] = {0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 15, 20, 30, 60};
  int32_t selector_to_elems[] = {240, 120, 60, 30, 20, 15, 12, 10, 8, 7, 6, 5, 4, 3, 2, 1};
H
hzcheng 已提交
261 262

  const char *ip = input + 1;
S
Shengliang Guan 已提交
263 264
  int32_t     count = 0;
  int32_t     _pos = 0;
H
hzcheng 已提交
265 266
  int64_t     prev_value = 0;

267
#if __AVX2__
H
hzcheng 已提交
268
  while (1) {
269
    if (_pos == nelements) break;
H
hzcheng 已提交
270 271 272 273

    uint64_t w = 0;
    memcpy(&w, ip, LONG_BYTES);

S
Shengliang Guan 已提交
274 275 276
    char    selector = (char)(w & INT64MASK(4));       // selector = 4
    char    bit = bit_per_integer[(int32_t)selector];  // bit = 3
    int32_t elems = selector_to_elems[(int32_t)selector];
H
hzcheng 已提交
277

H
Haojun Liao 已提交
278
    // Optimize the performance, by remove the constantly switch operation.
279
    int32_t  v = 4;
280 281
    uint64_t zigzag_value = 0;
    uint64_t mask = INT64MASK(bit);
H
Haojun Liao 已提交
282 283 284

    switch (type) {
      case TSDB_DATA_TYPE_BIGINT: {
285 286
        int64_t* p = (int64_t*) output;

287 288
        int32_t gRemainder = (nelements - _pos);
        int32_t num = (gRemainder > elems)? elems:gRemainder;
289

290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307
        int32_t batch = num >> 2;
        int32_t remain = num & 0x03;
        if (selector == 0 || selector == 1) {
          if (tsAVX2Enable && tsSIMDBuiltins) {
            for (int32_t i = 0; i < batch; ++i) {
              __m256i prev = _mm256_set1_epi64x(prev_value);
              _mm256_storeu_si256((__m256i *)&p[_pos], prev);
              _pos += 4;
            }

            for (int32_t i = 0; i < remain; ++i) {
              p[_pos++] = prev_value;
            }
          } else {
            for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
              p[_pos++] = prev_value;
              v += bit;
            }
H
Haojun Liao 已提交
308
          }
309
        } else {
310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
          if (tsAVX2Enable && tsSIMDBuiltins) {
            __m256i base = _mm256_set1_epi64x(w);
            __m256i maskVal = _mm256_set1_epi64x(mask);

            __m256i shiftBits = _mm256_set_epi64x(bit * 3 + 4, bit * 2 + 4, bit + 4, 4);
            __m256i inc = _mm256_set1_epi64x(bit << 2);

            for (int32_t i = 0; i < batch; ++i) {
              __m256i after = _mm256_srlv_epi64(base, shiftBits);
              __m256i zigzagVal = _mm256_and_si256(after, maskVal);

              // ZIGZAG_DECODE(T, v) (((v) >> 1) ^ -((T)((v)&1)))
              __m256i signmask = _mm256_and_si256(_mm256_set1_epi64x(1), zigzagVal);
              signmask = _mm256_sub_epi64(_mm256_setzero_si256(), signmask);
              // get the four zigzag values here
              __m256i delta = _mm256_xor_si256(_mm256_srli_epi64(zigzagVal, 1), signmask);

              // calculate the cumulative sum (prefix sum) for each number
              // decode[0] = prev_value + final[0]
              // decode[1] = decode[0] + final[1]   -----> prev_value + final[0] + final[1]
              // decode[2] = decode[1] + final[1]   -----> prev_value + final[0] + final[1] + final[2]
              // decode[3] = decode[2] + final[1]   -----> prev_value + final[0] + final[1] + final[2] + final[3]

              //  1, 2, 3, 4
              //+ 0, 1, 2, 3
              //  1, 3, 5, 7
              // shift and add for the first round
              __m128i prev = _mm_set1_epi64x(prev_value);
              delta = _mm256_add_epi64(delta, _mm256_slli_si256(delta, 8));
              _mm256_storeu_si256((__m256i *)&p[_pos], delta);

              //  1, 3, 5, 7
              //+ 0, 0, 1, 3
              //  1, 3, 6, 10
              // shift and add operation for the second round
              __m128i firstPart = _mm_loadu_si128((__m128i *)&p[_pos]);
              __m128i secPart = _mm_add_epi64(_mm_loadu_si128((__m128i *)&p[_pos + 2]), firstPart);
              firstPart = _mm_add_epi64(firstPart, prev);
              secPart = _mm_add_epi64(secPart, prev);

              // save it in the memory
              _mm_storeu_si128((__m128i *)&p[_pos], firstPart);
              _mm_storeu_si128((__m128i *)&p[_pos + 2], secPart);

              shiftBits = _mm256_add_epi64(shiftBits, inc);
              prev_value = p[_pos + 3];
              _pos += 4;
            }

            // handle the remain value
            for (int32_t i = 0; i < remain; i++) {
              zigzag_value = ((w >> (v + (batch * bit))) & mask);
              prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);

              p[_pos++] = prev_value;
              v += bit;
            }
          } else {
            for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
              zigzag_value = ((w >> v) & mask);
              prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);

              p[_pos++] = prev_value;
              v += bit;
            }
375
          }
H
Haojun Liao 已提交
376 377 378
        }
      } break;
      case TSDB_DATA_TYPE_INT: {
379 380
        int32_t* p = (int32_t*) output;

381
        if (selector == 0 || selector == 1) {
382 383
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
            p[_pos++] = (int32_t)prev_value;
H
Haojun Liao 已提交
384
          }
385
        } else {
386
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
387 388
            zigzag_value = ((w >> v) & mask);
            prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);
H
Haojun Liao 已提交
389

390
            p[_pos++] = (int32_t)prev_value;
391 392
            v += bit;
          }
H
Haojun Liao 已提交
393 394 395
        }
      } break;
      case TSDB_DATA_TYPE_SMALLINT: {
396 397 398 399 400 401 402 403
        int16_t* p = (int16_t*) output;

        if (selector == 0 || selector == 1) {
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
            p[_pos++] = (int16_t)prev_value;
          }
        } else {
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
404 405
            zigzag_value = ((w >> v) & mask);
            prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);
H
Haojun Liao 已提交
406

407 408 409
            p[_pos++] = (int16_t)prev_value;
            v += bit;
          }
H
Haojun Liao 已提交
410 411 412 413
        }
      } break;

      case TSDB_DATA_TYPE_TINYINT: {
414 415 416 417 418 419 420 421
        int8_t *p = (int8_t *)output;

        if (selector == 0 || selector == 1) {
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
            p[_pos++] = (int8_t)prev_value;
          }
        } else {
          for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
422 423
            zigzag_value = ((w >> v) & mask);
            prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);
H
Haojun Liao 已提交
424

425 426 427
            p[_pos++] = (int8_t)prev_value;
            v += bit;
          }
H
Haojun Liao 已提交
428 429 430 431
        }
      } break;
    }

H
hzcheng 已提交
432 433 434 435
    ip += LONG_BYTES;
  }

  return nelements * word_length;
436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488
#else

  while (1) {
    if (count == nelements) break;

    uint64_t w = 0;
    memcpy(&w, ip, LONG_BYTES);

    char    selector = (char)(w & INT64MASK(4));       // selector = 4
    char    bit = bit_per_integer[(int32_t)selector];  // bit = 3
    int32_t elems = selector_to_elems[(int32_t)selector];

    for (int32_t i = 0; i < elems; i++) {
      uint64_t zigzag_value;

      if (selector == 0 || selector == 1) {
        zigzag_value = 0;
      } else {
        zigzag_value = ((w >> (4 + bit * i)) & INT64MASK(bit));
      }
      int64_t diff = ZIGZAG_DECODE(int64_t, zigzag_value);
      int64_t curr_value = diff + prev_value;
      prev_value = curr_value;

      switch (type) {
        case TSDB_DATA_TYPE_BIGINT:
          *((int64_t *)output + _pos) = (int64_t)curr_value;
          _pos++;
          break;
        case TSDB_DATA_TYPE_INT:
          *((int32_t *)output + _pos) = (int32_t)curr_value;
          _pos++;
          break;
        case TSDB_DATA_TYPE_SMALLINT:
          *((int16_t *)output + _pos) = (int16_t)curr_value;
          _pos++;
          break;
        case TSDB_DATA_TYPE_TINYINT:
          *((int8_t *)output + _pos) = (int8_t)curr_value;
          _pos++;
          break;
        default:
          perror("Wrong integer types.\n");
          return -1;
      }
      count++;
      if (count == nelements) break;
    }
    ip += LONG_BYTES;
  }

  return nelements * word_length;
#endif
H
hzcheng 已提交
489 490 491 492 493
}

/* ----------------------------------------------Bool Compression
 * ---------------------------------------------- */
// TODO: You can also implement it using RLE method.
S
Shengliang Guan 已提交
494 495 496
int32_t tsCompressBoolImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t pos = -1;
  int32_t ele_per_byte = BITS_PER_BYTE / 2;
H
hzcheng 已提交
497

S
Shengliang Guan 已提交
498
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516
    if (i % ele_per_byte == 0) {
      pos++;
      output[pos] = 0;
    }

    uint8_t t = 0;
    if (input[i] == 1) {
      t = (((uint8_t)1) << (2 * (i % ele_per_byte)));
      output[pos] |= t;
    } else if (input[i] == 0) {
      t = ((uint8_t)1 << (2 * (i % ele_per_byte))) - 1;
      /* t = (~((( uint8_t)1) << (7-i%BITS_PER_BYTE))); */
      output[pos] &= t;
    } else if (input[i] == TSDB_DATA_BOOL_NULL) {
      t = ((uint8_t)2 << (2 * (i % ele_per_byte)));
      /* t = (~((( uint8_t)1) << (7-i%BITS_PER_BYTE))); */
      output[pos] |= t;
    } else {
517
      uError("Invalid compress bool value:%d", output[pos]);
H
Hongze Cheng 已提交
518
      return -1;
H
hzcheng 已提交
519 520 521 522 523 524
    }
  }

  return pos + 1;
}

S
Shengliang Guan 已提交
525 526 527
int32_t tsDecompressBoolImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t ipos = -1, opos = 0;
  int32_t ele_per_byte = BITS_PER_BYTE / 2;
H
hzcheng 已提交
528

S
Shengliang Guan 已提交
529
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546
    if (i % ele_per_byte == 0) {
      ipos++;
    }

    uint8_t ele = (input[ipos] >> (2 * (i % ele_per_byte))) & INT8MASK(2);
    if (ele == 1) {
      output[opos++] = 1;
    } else if (ele == 2) {
      output[opos++] = TSDB_DATA_BOOL_NULL;
    } else {
      output[opos++] = 0;
    }
  }

  return nelements;
}

547
#if 0
H
hzcheng 已提交
548
/* Run Length Encoding(RLE) Method */
S
Shengliang Guan 已提交
549 550
int32_t tsCompressBoolRLEImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t _pos = 0;
H
hzcheng 已提交
551

S
Shengliang Guan 已提交
552
  for (int32_t i = 0; i < nelements;) {
H
hzcheng 已提交
553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573
    unsigned char counter = 1;
    char          num = input[i];

    for (++i; i < nelements; i++) {
      if (input[i] == num) {
        counter++;
        if (counter == INT8MASK(7)) {
          i++;
          break;
        }
      } else {
        break;
      }
    }

    // Encode the data.
    if (num == 1) {
      output[_pos++] = INT8MASK(1) | (counter << 1);
    } else if (num == 0) {
      output[_pos++] = (counter << 1) | INT8MASK(0);
    } else {
574
      uError("Invalid compress bool value:%d", output[_pos]);
H
Hongze Cheng 已提交
575
      return -1;
H
hzcheng 已提交
576 577 578 579 580 581
    }
  }

  return _pos;
}

S
Shengliang Guan 已提交
582 583
int32_t tsDecompressBoolRLEImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t ipos = 0, opos = 0;
H
hzcheng 已提交
584 585 586 587 588 589 590 591 592 593 594 595
  while (1) {
    char     encode = input[ipos++];
    unsigned counter = (encode >> 1) & INT8MASK(7);
    char     value = encode & INT8MASK(1);

    memset(output + opos, value, counter);
    opos += counter;
    if (opos >= nelements) {
      return nelements;
    }
  }
}
596
#endif
H
hzcheng 已提交
597 598 599 600 601 602

/* ----------------------------------------------String Compression
 * ---------------------------------------------- */
// Note: the size of the output must be larger than input_size + 1 and
// LZ4_compressBound(size) + 1;
// >= max(input_size, LZ4_compressBound(input_size)) + 1;
S
Shengliang Guan 已提交
603
int32_t tsCompressStringImp(const char *const input, int32_t inputSize, char *const output, int32_t outputSize) {
H
hzcheng 已提交
604
  // Try to compress using LZ4 algorithm.
S
Shengliang Guan 已提交
605
  const int32_t compressed_data_size = LZ4_compress_default(input, output + 1, inputSize, outputSize - 1);
H
hzcheng 已提交
606 607 608 609 610 611 612 613 614 615 616 617 618

  // If cannot compress or after compression, data becomes larger.
  if (compressed_data_size <= 0 || compressed_data_size > inputSize) {
    /* First byte is for indicator */
    output[0] = 0;
    memcpy(output + 1, input, inputSize);
    return inputSize + 1;
  }

  output[0] = 1;
  return compressed_data_size + 1;
}

S
Shengliang Guan 已提交
619
int32_t tsDecompressStringImp(const char *const input, int32_t compressedSize, char *const output, int32_t outputSize) {
H
hzcheng 已提交
620
  // compressedSize is the size of data after compression.
S
Shengliang Guan 已提交
621

H
hzcheng 已提交
622 623
  if (input[0] == 1) {
    /* It is compressed by LZ4 algorithm */
S
Shengliang Guan 已提交
624
    const int32_t decompressed_size = LZ4_decompress_safe(input + 1, output, compressedSize - 1, outputSize);
H
hzcheng 已提交
625
    if (decompressed_size < 0) {
626
      uError("Failed to decompress string with LZ4 algorithm, decompressed size:%d", decompressed_size);
H
Hongze Cheng 已提交
627
      return -1;
H
hzcheng 已提交
628 629 630 631 632 633 634 635
    }

    return decompressed_size;
  } else if (input[0] == 0) {
    /* It is not compressed by LZ4 algorithm */
    memcpy(output, input + 1, compressedSize - 1);
    return compressedSize - 1;
  } else {
636
    uError("Invalid decompress string indicator:%d", input[0]);
H
Hongze Cheng 已提交
637
    return -1;
H
hzcheng 已提交
638 639 640 641 642 643
  }
}

/* --------------------------------------------Timestamp Compression
 * ---------------------------------------------- */
// TODO: Take care here, we assumes little endian encoding.
S
Shengliang Guan 已提交
644 645
int32_t tsCompressTimestampImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t _pos = 1;
646
  ASSERTS(nelements >= 0, "nelements is negative");
H
hzcheng 已提交
647 648 649 650 651

  if (nelements == 0) return 0;

  int64_t *istream = (int64_t *)input;

S
Shengliang Guan 已提交
652 653 654 655
  int64_t prev_value = istream[0];
  if (prev_value >= 0x8000000000000000) {
    uWarn("compression timestamp is over signed long long range. ts = 0x%" PRIx64 " \n", prev_value);
    goto _exit_over;
T
tickduan 已提交
656
  }
H
hzcheng 已提交
657 658 659 660
  int64_t  prev_delta = -prev_value;
  uint8_t  flags = 0, flag1 = 0, flag2 = 0;
  uint64_t dd1 = 0, dd2 = 0;

S
Shengliang Guan 已提交
661
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
662 663 664 665 666 667
    int64_t curr_value = istream[i];
    if (!safeInt64Add(curr_value, -prev_value)) goto _exit_over;
    int64_t curr_delta = curr_value - prev_value;
    if (!safeInt64Add(curr_delta, -prev_delta)) goto _exit_over;
    int64_t delta_of_delta = curr_delta - prev_delta;
    // zigzag encode the value.
H
Hongze Cheng 已提交
668
    uint64_t zigzag_value = ZIGZAG_ENCODE(int64_t, delta_of_delta);
H
hzcheng 已提交
669 670 671 672 673 674
    if (i % 2 == 0) {
      flags = 0;
      dd1 = zigzag_value;
      if (dd1 == 0) {
        flag1 = 0;
      } else {
H
Hongze Cheng 已提交
675
        flag1 = (uint8_t)(LONG_BYTES - BUILDIN_CLZL(dd1) / BITS_PER_BYTE);
H
hzcheng 已提交
676 677 678 679 680 681
      }
    } else {
      dd2 = zigzag_value;
      if (dd2 == 0) {
        flag2 = 0;
      } else {
H
Hongze Cheng 已提交
682
        flag2 = (uint8_t)(LONG_BYTES - BUILDIN_CLZL(dd2) / BITS_PER_BYTE);
H
hzcheng 已提交
683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739
      }
      flags = flag1 | (flag2 << 4);
      // Encode the flag.
      if ((_pos + CHAR_BYTES - 1) >= nelements * LONG_BYTES) goto _exit_over;
      memcpy(output + _pos, &flags, CHAR_BYTES);
      _pos += CHAR_BYTES;
      /* Here, we assume it is little endian encoding method. */
      // Encode dd1
      if (is_bigendian()) {
        if ((_pos + flag1 - 1) >= nelements * LONG_BYTES) goto _exit_over;
        memcpy(output + _pos, (char *)(&dd1) + LONG_BYTES - flag1, flag1);
      } else {
        if ((_pos + flag1 - 1) >= nelements * LONG_BYTES) goto _exit_over;
        memcpy(output + _pos, (char *)(&dd1), flag1);
      }
      _pos += flag1;
      // Encode dd2;
      if (is_bigendian()) {
        if ((_pos + flag2 - 1) >= nelements * LONG_BYTES) goto _exit_over;
        memcpy(output + _pos, (char *)(&dd2) + LONG_BYTES - flag2, flag2);
      } else {
        if ((_pos + flag2 - 1) >= nelements * LONG_BYTES) goto _exit_over;
        memcpy(output + _pos, (char *)(&dd2), flag2);
      }
      _pos += flag2;
    }
    prev_value = curr_value;
    prev_delta = curr_delta;
  }

  if (nelements % 2 == 1) {
    flag2 = 0;
    flags = flag1 | (flag2 << 4);
    // Encode the flag.
    if ((_pos + CHAR_BYTES - 1) >= nelements * LONG_BYTES) goto _exit_over;
    memcpy(output + _pos, &flags, CHAR_BYTES);
    _pos += CHAR_BYTES;
    // Encode dd1;
    if (is_bigendian()) {
      if ((_pos + flag1 - 1) >= nelements * LONG_BYTES) goto _exit_over;
      memcpy(output + _pos, (char *)(&dd1) + LONG_BYTES - flag1, flag1);
    } else {
      if ((_pos + flag1 - 1) >= nelements * LONG_BYTES) goto _exit_over;
      memcpy(output + _pos, (char *)(&dd1), flag1);
    }
    _pos += flag1;
  }

  output[0] = 1;  // Means the string is compressed
  return _pos;

_exit_over:
  output[0] = 0;  // Means the string is not compressed
  memcpy(output + 1, input, nelements * LONG_BYTES);
  return nelements * LONG_BYTES + 1;
}

S
Shengliang Guan 已提交
740
int32_t tsDecompressTimestampImp(const char *const input, const int32_t nelements, char *const output) {
741
  ASSERTS(nelements >= 0, "nelements is negative");
H
hzcheng 已提交
742 743 744 745 746 747 748 749
  if (nelements == 0) return 0;

  if (input[0] == 0) {
    memcpy(output, input + 1, nelements * LONG_BYTES);
    return nelements * LONG_BYTES;
  } else if (input[0] == 1) {  // Decompress
    int64_t *ostream = (int64_t *)output;

S
Shengliang Guan 已提交
750
    int32_t ipos = 1, opos = 0;
H
hzcheng 已提交
751 752 753 754 755 756 757 758 759 760 761 762 763 764
    int8_t  nbytes = 0;
    int64_t prev_value = 0;
    int64_t prev_delta = 0;
    int64_t delta_of_delta = 0;

    while (1) {
      uint8_t flags = input[ipos++];
      // Decode dd1
      uint64_t dd1 = 0;
      nbytes = flags & INT8MASK(4);
      if (nbytes == 0) {
        delta_of_delta = 0;
      } else {
        if (is_bigendian()) {
H
Hui Li 已提交
765
          memcpy(((char *)(&dd1)) + LONG_BYTES - nbytes, input + ipos, nbytes);
H
hzcheng 已提交
766 767 768
        } else {
          memcpy(&dd1, input + ipos, nbytes);
        }
H
Hongze Cheng 已提交
769
        delta_of_delta = ZIGZAG_DECODE(int64_t, dd1);
H
hzcheng 已提交
770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789
      }
      ipos += nbytes;
      if (opos == 0) {
        prev_value = delta_of_delta;
        prev_delta = 0;
        ostream[opos++] = delta_of_delta;
      } else {
        prev_delta = delta_of_delta + prev_delta;
        prev_value = prev_value + prev_delta;
        ostream[opos++] = prev_value;
      }
      if (opos == nelements) return nelements * LONG_BYTES;

      // Decode dd2
      uint64_t dd2 = 0;
      nbytes = (flags >> 4) & INT8MASK(4);
      if (nbytes == 0) {
        delta_of_delta = 0;
      } else {
        if (is_bigendian()) {
H
Hui Li 已提交
790
          memcpy(((char *)(&dd2)) + LONG_BYTES - nbytes, input + ipos, nbytes);
H
hzcheng 已提交
791 792 793 794
        } else {
          memcpy(&dd2, input + ipos, nbytes);
        }
        // zigzag_decoding
H
Hongze Cheng 已提交
795
        delta_of_delta = ZIGZAG_DECODE(int64_t, dd2);
H
hzcheng 已提交
796 797 798 799 800 801 802 803 804
      }
      ipos += nbytes;
      prev_delta = delta_of_delta + prev_delta;
      prev_value = prev_value + prev_delta;
      ostream[opos++] = prev_value;
      if (opos == nelements) return nelements * LONG_BYTES;
    }

  } else {
805
    ASSERT(0);
S
TD-1057  
Shengliang Guan 已提交
806
    return -1;
H
hzcheng 已提交
807 808 809 810
  }
}
/* --------------------------------------------Double Compression
 * ---------------------------------------------- */
S
Shengliang Guan 已提交
811
void encodeDoubleValue(uint64_t diff, uint8_t flag, char *const output, int32_t *const pos) {
H
hzcheng 已提交
812
  uint8_t nbytes = (flag & INT8MASK(3)) + 1;
S
Shengliang Guan 已提交
813
  int32_t nshift = (LONG_BYTES * BITS_PER_BYTE - nbytes * BITS_PER_BYTE) * (flag >> 3);
H
hzcheng 已提交
814 815 816 817 818 819 820 821 822
  diff >>= nshift;

  while (nbytes) {
    output[(*pos)++] = (int8_t)(diff & INT64MASK(8));
    diff >>= BITS_PER_BYTE;
    nbytes--;
  }
}

S
Shengliang Guan 已提交
823 824 825
int32_t tsCompressDoubleImp(const char *const input, const int32_t nelements, char *const output) {
  int32_t byte_limit = nelements * DOUBLE_BYTES + 1;
  int32_t opos = 1;
H
hzcheng 已提交
826 827 828 829 830 831 832 833

  uint64_t prev_value = 0;
  uint64_t prev_diff = 0;
  uint8_t  prev_flag = 0;

  double *istream = (double *)input;

  // Main loop
S
Shengliang Guan 已提交
834
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
835 836 837 838 839 840 841 842 843 844 845
    union {
      double   real;
      uint64_t bits;
    } curr;

    curr.real = istream[i];

    // Here we assume the next value is the same as previous one.
    uint64_t predicted = prev_value;
    uint64_t diff = curr.bits ^ predicted;

S
Shengliang Guan 已提交
846 847
    int32_t leading_zeros = LONG_BYTES * BITS_PER_BYTE;
    int32_t trailing_zeros = leading_zeros;
H
hzcheng 已提交
848 849

    if (diff) {
850 851
      trailing_zeros = BUILDIN_CTZL(diff);
      leading_zeros = BUILDIN_CLZL(diff);
H
hzcheng 已提交
852 853 854 855 856 857
    }

    uint8_t nbytes = 0;
    uint8_t flag;

    if (trailing_zeros > leading_zeros) {
H
Hongze Cheng 已提交
858
      nbytes = (uint8_t)(LONG_BYTES - trailing_zeros / BITS_PER_BYTE);
H
hzcheng 已提交
859 860 861 862

      if (nbytes > 0) nbytes--;
      flag = ((uint8_t)1 << 3) | nbytes;
    } else {
H
Hongze Cheng 已提交
863
      nbytes = (uint8_t)(LONG_BYTES - leading_zeros / BITS_PER_BYTE);
H
hzcheng 已提交
864 865 866 867 868 869 870 871
      if (nbytes > 0) nbytes--;
      flag = nbytes;
    }

    if (i % 2 == 0) {
      prev_diff = diff;
      prev_flag = flag;
    } else {
S
Shengliang Guan 已提交
872 873
      int32_t nbyte1 = (prev_flag & INT8MASK(3)) + 1;
      int32_t nbyte2 = (flag & INT8MASK(3)) + 1;
H
hzcheng 已提交
874 875 876 877 878 879 880 881 882 883 884 885 886 887 888
      if (opos + 1 + nbyte1 + nbyte2 <= byte_limit) {
        uint8_t flags = prev_flag | (flag << 4);
        output[opos++] = flags;
        encodeDoubleValue(prev_diff, prev_flag, output, &opos);
        encodeDoubleValue(diff, flag, output, &opos);
      } else {
        output[0] = 1;
        memcpy(output + 1, input, byte_limit - 1);
        return byte_limit;
      }
    }
    prev_value = curr.bits;
  }

  if (nelements % 2) {
S
Shengliang Guan 已提交
889 890
    int32_t nbyte1 = (prev_flag & INT8MASK(3)) + 1;
    int32_t nbyte2 = 1;
H
hzcheng 已提交
891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906
    if (opos + 1 + nbyte1 + nbyte2 <= byte_limit) {
      uint8_t flags = prev_flag;
      output[opos++] = flags;
      encodeDoubleValue(prev_diff, prev_flag, output, &opos);
      encodeDoubleValue(0ul, 0, output, &opos);
    } else {
      output[0] = 1;
      memcpy(output + 1, input, byte_limit - 1);
      return byte_limit;
    }
  }

  output[0] = 0;
  return opos;
}

S
Shengliang Guan 已提交
907
uint64_t decodeDoubleValue(const char *const input, int32_t *const ipos, uint8_t flag) {
H
hzcheng 已提交
908
  uint64_t diff = 0ul;
S
Shengliang Guan 已提交
909 910
  int32_t  nbytes = (flag & INT8MASK(3)) + 1;
  for (int32_t i = 0; i < nbytes; i++) {
H
hzcheng 已提交
911 912
    diff = diff | ((INT64MASK(8) & input[(*ipos)++]) << BITS_PER_BYTE * i);
  }
S
Shengliang Guan 已提交
913
  int32_t shift_width = (LONG_BYTES * BITS_PER_BYTE - nbytes * BITS_PER_BYTE) * (flag >> 3);
H
hzcheng 已提交
914 915 916 917 918
  diff <<= shift_width;

  return diff;
}

S
Shengliang Guan 已提交
919
int32_t tsDecompressDoubleImp(const char *const input, const int32_t nelements, char *const output) {
H
hzcheng 已提交
920 921 922 923 924 925 926 927 928
  // output stream
  double *ostream = (double *)output;

  if (input[0] == 1) {
    memcpy(output, input + 1, nelements * DOUBLE_BYTES);
    return nelements * DOUBLE_BYTES;
  }

  uint8_t  flags = 0;
S
Shengliang Guan 已提交
929 930
  int32_t  ipos = 1;
  int32_t  opos = 0;
H
hzcheng 已提交
931 932
  uint64_t prev_value = 0;

S
Shengliang Guan 已提交
933
  for (int32_t i = 0; i < nelements; i++) {
H
Haojun Liao 已提交
934
    if ((i & 0x01) == 0) {
H
hzcheng 已提交
935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958
      flags = input[ipos++];
    }

    uint8_t flag = flags & INT8MASK(4);
    flags >>= 4;

    uint64_t diff = decodeDoubleValue(input, &ipos, flag);
    union {
      uint64_t bits;
      double   real;
    } curr;

    uint64_t predicted = prev_value;
    curr.bits = predicted ^ diff;
    prev_value = curr.bits;

    ostream[opos++] = curr.real;
  }

  return nelements * DOUBLE_BYTES;
}

/* --------------------------------------------Float Compression
 * ---------------------------------------------- */
S
Shengliang Guan 已提交
959
void encodeFloatValue(uint32_t diff, uint8_t flag, char *const output, int32_t *const pos) {
H
hzcheng 已提交
960
  uint8_t nbytes = (flag & INT8MASK(3)) + 1;
S
Shengliang Guan 已提交
961
  int32_t nshift = (FLOAT_BYTES * BITS_PER_BYTE - nbytes * BITS_PER_BYTE) * (flag >> 3);
H
hzcheng 已提交
962 963 964 965 966 967 968 969 970
  diff >>= nshift;

  while (nbytes) {
    output[(*pos)++] = (int8_t)(diff & INT32MASK(8));
    diff >>= BITS_PER_BYTE;
    nbytes--;
  }
}

S
Shengliang Guan 已提交
971 972 973 974
int32_t tsCompressFloatImp(const char *const input, const int32_t nelements, char *const output) {
  float  *istream = (float *)input;
  int32_t byte_limit = nelements * FLOAT_BYTES + 1;
  int32_t opos = 1;
H
hzcheng 已提交
975 976 977 978 979 980

  uint32_t prev_value = 0;
  uint32_t prev_diff = 0;
  uint8_t  prev_flag = 0;

  // Main loop
S
Shengliang Guan 已提交
981
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
982 983 984 985 986 987 988 989 990 991 992
    union {
      float    real;
      uint32_t bits;
    } curr;

    curr.real = istream[i];

    // Here we assume the next value is the same as previous one.
    uint32_t predicted = prev_value;
    uint32_t diff = curr.bits ^ predicted;

H
Hongze Cheng 已提交
993 994
    int32_t clz = FLOAT_BYTES * BITS_PER_BYTE;
    int32_t ctz = clz;
H
hzcheng 已提交
995 996

    if (diff) {
H
Hongze Cheng 已提交
997 998
      ctz = BUILDIN_CTZ(diff);
      clz = BUILDIN_CLZ(diff);
H
hzcheng 已提交
999 1000 1001 1002 1003
    }

    uint8_t nbytes = 0;
    uint8_t flag;

H
Hongze Cheng 已提交
1004 1005
    if (ctz > clz) {
      nbytes = (uint8_t)(FLOAT_BYTES - ctz / BITS_PER_BYTE);
H
hzcheng 已提交
1006 1007 1008 1009

      if (nbytes > 0) nbytes--;
      flag = ((uint8_t)1 << 3) | nbytes;
    } else {
H
Hongze Cheng 已提交
1010
      nbytes = (uint8_t)(FLOAT_BYTES - clz / BITS_PER_BYTE);
H
hzcheng 已提交
1011 1012 1013 1014 1015 1016 1017 1018
      if (nbytes > 0) nbytes--;
      flag = nbytes;
    }

    if (i % 2 == 0) {
      prev_diff = diff;
      prev_flag = flag;
    } else {
S
Shengliang Guan 已提交
1019 1020
      int32_t nbyte1 = (prev_flag & INT8MASK(3)) + 1;
      int32_t nbyte2 = (flag & INT8MASK(3)) + 1;
H
hzcheng 已提交
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035
      if (opos + 1 + nbyte1 + nbyte2 <= byte_limit) {
        uint8_t flags = prev_flag | (flag << 4);
        output[opos++] = flags;
        encodeFloatValue(prev_diff, prev_flag, output, &opos);
        encodeFloatValue(diff, flag, output, &opos);
      } else {
        output[0] = 1;
        memcpy(output + 1, input, byte_limit - 1);
        return byte_limit;
      }
    }
    prev_value = curr.bits;
  }

  if (nelements % 2) {
S
Shengliang Guan 已提交
1036 1037
    int32_t nbyte1 = (prev_flag & INT8MASK(3)) + 1;
    int32_t nbyte2 = 1;
H
hzcheng 已提交
1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053
    if (opos + 1 + nbyte1 + nbyte2 <= byte_limit) {
      uint8_t flags = prev_flag;
      output[opos++] = flags;
      encodeFloatValue(prev_diff, prev_flag, output, &opos);
      encodeFloatValue(0, 0, output, &opos);
    } else {
      output[0] = 1;
      memcpy(output + 1, input, byte_limit - 1);
      return byte_limit;
    }
  }

  output[0] = 0;
  return opos;
}

S
Shengliang Guan 已提交
1054
uint32_t decodeFloatValue(const char *const input, int32_t *const ipos, uint8_t flag) {
H
hzcheng 已提交
1055
  uint32_t diff = 0ul;
S
Shengliang Guan 已提交
1056 1057
  int32_t  nbytes = (flag & INT8MASK(3)) + 1;
  for (int32_t i = 0; i < nbytes; i++) {
H
hzcheng 已提交
1058 1059
    diff = diff | ((INT32MASK(8) & input[(*ipos)++]) << BITS_PER_BYTE * i);
  }
S
Shengliang Guan 已提交
1060
  int32_t shift_width = (FLOAT_BYTES * BITS_PER_BYTE - nbytes * BITS_PER_BYTE) * (flag >> 3);
H
hzcheng 已提交
1061 1062 1063 1064 1065
  diff <<= shift_width;

  return diff;
}

S
Shengliang Guan 已提交
1066
int32_t tsDecompressFloatImp(const char *const input, const int32_t nelements, char *const output) {
H
hzcheng 已提交
1067 1068 1069 1070 1071 1072 1073 1074
  float *ostream = (float *)output;

  if (input[0] == 1) {
    memcpy(output, input + 1, nelements * FLOAT_BYTES);
    return nelements * FLOAT_BYTES;
  }

  uint8_t  flags = 0;
S
Shengliang Guan 已提交
1075 1076
  int32_t  ipos = 1;
  int32_t  opos = 0;
H
hzcheng 已提交
1077 1078
  uint32_t prev_value = 0;

S
Shengliang Guan 已提交
1079
  for (int32_t i = 0; i < nelements; i++) {
H
hzcheng 已提交
1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
    if (i % 2 == 0) {
      flags = input[ipos++];
    }

    uint8_t flag = flags & INT8MASK(4);
    flags >>= 4;

    uint32_t diff = decodeFloatValue(input, &ipos, flag);
    union {
      uint32_t bits;
      float    real;
    } curr;

    uint32_t predicted = prev_value;
    curr.bits = predicted ^ diff;
    prev_value = curr.bits;

    ostream[opos++] = curr.real;
  }

  return nelements * FLOAT_BYTES;
}
T
tickduan 已提交
1102

S
Shengliang Guan 已提交
1103
#ifdef TD_TSZ
T
tickduan 已提交
1104 1105 1106
//
//   ----------  float double lossy  -----------
//
S
Shengliang Guan 已提交
1107 1108 1109
int32_t tsCompressFloatLossyImp(const char *input, const int32_t nelements, char *const output) {
  // compress with sz
  int32_t       compressedSize = tdszCompress(SZ_FLOAT, input, nelements, output + 1);
T
tickduan 已提交
1110
  unsigned char algo = ALGO_SZ_LOSSY << 1;
S
Shengliang Guan 已提交
1111
  if (compressedSize == 0 || compressedSize >= nelements * sizeof(float)) {
T
tickduan 已提交
1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122
    // compressed error or large than original
    output[0] = MODE_NOCOMPRESS | algo;
    memcpy(output + 1, input, nelements * sizeof(float));
    compressedSize = 1 + nelements * sizeof(float);
  } else {
    // compressed successfully
    output[0] = MODE_COMPRESS | algo;
    compressedSize += 1;
  }

  return compressedSize;
T
tickduan 已提交
1123 1124
}

S
Shengliang Guan 已提交
1125 1126 1127 1128
int32_t tsDecompressFloatLossyImp(const char *input, int32_t compressedSize, const int32_t nelements,
                                  char *const output) {
  int32_t decompressedSize = 0;
  if (HEAD_MODE(input[0]) == MODE_NOCOMPRESS) {
T
tickduan 已提交
1129 1130 1131 1132 1133
    // orginal so memcpy directly
    decompressedSize = nelements * sizeof(float);
    memcpy(output, input + 1, decompressedSize);

    return decompressedSize;
S
Shengliang Guan 已提交
1134
  }
T
tickduan 已提交
1135 1136 1137

  // decompressed with sz
  return tdszDecompress(SZ_FLOAT, input + 1, compressedSize - 1, nelements, output);
T
tickduan 已提交
1138 1139
}

S
Shengliang Guan 已提交
1140 1141 1142
int32_t tsCompressDoubleLossyImp(const char *input, const int32_t nelements, char *const output) {
  // compress with sz
  int32_t       compressedSize = tdszCompress(SZ_DOUBLE, input, nelements, output + 1);
T
tickduan 已提交
1143
  unsigned char algo = ALGO_SZ_LOSSY << 1;
S
Shengliang Guan 已提交
1144
  if (compressedSize == 0 || compressedSize >= nelements * sizeof(double)) {
T
tickduan 已提交
1145 1146 1147 1148 1149 1150 1151 1152 1153 1154
    // compressed error or large than original
    output[0] = MODE_NOCOMPRESS | algo;
    memcpy(output + 1, input, nelements * sizeof(double));
    compressedSize = 1 + nelements * sizeof(double);
  } else {
    // compressed successfully
    output[0] = MODE_COMPRESS | algo;
    compressedSize += 1;
  }

S
Shengliang Guan 已提交
1155
  return compressedSize;
T
tickduan 已提交
1156 1157
}

S
Shengliang Guan 已提交
1158 1159 1160 1161
int32_t tsDecompressDoubleLossyImp(const char *input, int32_t compressedSize, const int32_t nelements,
                                   char *const output) {
  int32_t decompressedSize = 0;
  if (HEAD_MODE(input[0]) == MODE_NOCOMPRESS) {
T
tickduan 已提交
1162 1163 1164 1165 1166
    // orginal so memcpy directly
    decompressedSize = nelements * sizeof(double);
    memcpy(output, input + 1, decompressedSize);

    return decompressedSize;
S
Shengliang Guan 已提交
1167
  }
T
tickduan 已提交
1168 1169 1170

  // decompressed with sz
  return tdszDecompress(SZ_DOUBLE, input + 1, compressedSize - 1, nelements, output);
1171
}
Y
yihaoDeng 已提交
1172
#endif
H
Hongze Cheng 已提交
1173 1174 1175 1176

/*************************************************************************
 *                  STREAM COMPRESSION
 *************************************************************************/
H
Hongze Cheng 已提交
1177
#define I64_SAFE_ADD(a, b) (((a) >= 0 && (b) <= INT64_MAX - (a)) || ((a) < 0 && (b) >= INT64_MIN - (a)))
H
Hongze Cheng 已提交
1178

H
Hongze Cheng 已提交
1179
static int32_t tCompBoolStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1180
static int32_t tCompBool(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1181 1182 1183
static int32_t tCompBoolEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);

static int32_t tCompIntStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1184
static int32_t tCompInt(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1185 1186 1187
static int32_t tCompIntEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);

static int32_t tCompFloatStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1188
static int32_t tCompFloat(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1189 1190 1191
static int32_t tCompFloatEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);

static int32_t tCompDoubleStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1192
static int32_t tCompDouble(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1193 1194 1195
static int32_t tCompDoubleEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);

static int32_t tCompTimestampStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1196
static int32_t tCompTimestamp(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1197 1198 1199
static int32_t tCompTimestampEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);

static int32_t tCompBinaryStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1200
static int32_t tCompBinary(SCompressor *pCmprsor, const void *pData, int32_t nData);
H
Hongze Cheng 已提交
1201
static int32_t tCompBinaryEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData);
H
Hongze Cheng 已提交
1202

H
Hongze Cheng 已提交
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
static FORCE_INLINE int64_t tGetI64OfI8(const void *pData) { return *(int8_t *)pData; }
static FORCE_INLINE int64_t tGetI64OfI16(const void *pData) { return *(int16_t *)pData; }
static FORCE_INLINE int64_t tGetI64OfI32(const void *pData) { return *(int32_t *)pData; }
static FORCE_INLINE int64_t tGetI64OfI64(const void *pData) { return *(int64_t *)pData; }

static FORCE_INLINE void tPutI64OfI8(int64_t v, void *pData) { *(int8_t *)pData = v; }
static FORCE_INLINE void tPutI64OfI16(int64_t v, void *pData) { *(int16_t *)pData = v; }
static FORCE_INLINE void tPutI64OfI32(int64_t v, void *pData) { *(int32_t *)pData = v; }
static FORCE_INLINE void tPutI64OfI64(int64_t v, void *pData) { *(int64_t *)pData = v; }

H
Hongze Cheng 已提交
1213 1214 1215 1216
static struct {
  int8_t  type;
  int32_t bytes;
  int8_t  isVarLen;
H
Hongze Cheng 已提交
1217
  int32_t (*startFn)(SCompressor *, int8_t type, int8_t cmprAlg);
H
Hongze Cheng 已提交
1218
  int32_t (*cmprFn)(SCompressor *, const void *, int32_t nData);
H
Hongze Cheng 已提交
1219
  int32_t (*endFn)(SCompressor *, const uint8_t **, int32_t *);
H
Hongze Cheng 已提交
1220
  int64_t (*getI64)(const void *pData);
H
Hongze Cheng 已提交
1221
  void (*putI64)(int64_t v, void *pData);
H
Hongze Cheng 已提交
1222
} DATA_TYPE_INFO[] = {
H
Hongze Cheng 已提交
1223 1224 1225 1226 1227 1228
    {.type = TSDB_DATA_TYPE_NULL,
     .bytes = 0,
     .isVarLen = 0,
     .startFn = NULL,
     .cmprFn = NULL,
     .endFn = NULL,
H
Hongze Cheng 已提交
1229 1230
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1231 1232 1233 1234 1235
    {.type = TSDB_DATA_TYPE_BOOL,
     .bytes = 1,
     .isVarLen = 0,
     .startFn = tCompBoolStart,
     .cmprFn = tCompBool,
H
Hongze Cheng 已提交
1236
     .endFn = tCompBoolEnd,
H
Hongze Cheng 已提交
1237 1238
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1239 1240 1241 1242 1243
    {.type = TSDB_DATA_TYPE_TINYINT,
     .bytes = 1,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1244
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1245 1246
     .getI64 = tGetI64OfI8,
     .putI64 = tPutI64OfI8},
H
Hongze Cheng 已提交
1247 1248 1249 1250 1251
    {.type = TSDB_DATA_TYPE_SMALLINT,
     .bytes = 2,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1252
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1253 1254
     .getI64 = tGetI64OfI16,
     .putI64 = tPutI64OfI16},
H
Hongze Cheng 已提交
1255 1256 1257 1258 1259
    {.type = TSDB_DATA_TYPE_INT,
     .bytes = 4,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1260
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1261 1262
     .getI64 = tGetI64OfI32,
     .putI64 = tPutI64OfI32},
H
Hongze Cheng 已提交
1263 1264 1265 1266 1267
    {.type = TSDB_DATA_TYPE_BIGINT,
     .bytes = 8,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1268
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1269 1270
     .getI64 = tGetI64OfI64,
     .putI64 = tPutI64OfI64},
H
Hongze Cheng 已提交
1271 1272 1273 1274 1275
    {.type = TSDB_DATA_TYPE_FLOAT,
     .bytes = 4,
     .isVarLen = 0,
     .startFn = tCompFloatStart,
     .cmprFn = tCompFloat,
H
Hongze Cheng 已提交
1276
     .endFn = tCompFloatEnd,
H
Hongze Cheng 已提交
1277 1278
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1279 1280 1281 1282 1283
    {.type = TSDB_DATA_TYPE_DOUBLE,
     .bytes = 8,
     .isVarLen = 0,
     .startFn = tCompDoubleStart,
     .cmprFn = tCompDouble,
H
Hongze Cheng 已提交
1284
     .endFn = tCompDoubleEnd,
H
Hongze Cheng 已提交
1285 1286
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1287 1288 1289 1290 1291
    {.type = TSDB_DATA_TYPE_VARCHAR,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1292
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1293 1294
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1295 1296 1297
    {.type = TSDB_DATA_TYPE_TIMESTAMP,
     .bytes = 8,
     .isVarLen = 0,
H
Hongze Cheng 已提交
1298 1299
     .startFn = tCompTimestampStart,
     .cmprFn = tCompTimestamp,
H
Hongze Cheng 已提交
1300
     .endFn = tCompTimestampEnd,
H
Hongze Cheng 已提交
1301 1302
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1303 1304 1305 1306 1307
    {.type = TSDB_DATA_TYPE_NCHAR,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1308
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1309 1310
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1311 1312 1313 1314 1315
    {.type = TSDB_DATA_TYPE_UTINYINT,
     .bytes = 1,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1316
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1317 1318
     .getI64 = tGetI64OfI8,
     .putI64 = tPutI64OfI8},
H
Hongze Cheng 已提交
1319 1320 1321 1322 1323
    {.type = TSDB_DATA_TYPE_USMALLINT,
     .bytes = 2,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1324
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1325 1326
     .getI64 = tGetI64OfI16,
     .putI64 = tPutI64OfI16},
H
Hongze Cheng 已提交
1327 1328 1329 1330 1331
    {.type = TSDB_DATA_TYPE_UINT,
     .bytes = 4,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1332
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1333 1334
     .getI64 = tGetI64OfI32,
     .putI64 = tPutI64OfI32},
H
Hongze Cheng 已提交
1335 1336 1337 1338 1339
    {.type = TSDB_DATA_TYPE_UBIGINT,
     .bytes = 8,
     .isVarLen = 0,
     .startFn = tCompIntStart,
     .cmprFn = tCompInt,
H
Hongze Cheng 已提交
1340
     .endFn = tCompIntEnd,
H
Hongze Cheng 已提交
1341 1342
     .getI64 = tGetI64OfI64,
     .putI64 = tPutI64OfI64},
H
Hongze Cheng 已提交
1343 1344 1345 1346 1347
    {.type = TSDB_DATA_TYPE_JSON,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1348
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1349 1350
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1351 1352 1353 1354 1355
    {.type = TSDB_DATA_TYPE_VARBINARY,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1356
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1357 1358
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1359 1360 1361 1362 1363
    {.type = TSDB_DATA_TYPE_DECIMAL,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1364
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1365 1366
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1367 1368 1369 1370 1371
    {.type = TSDB_DATA_TYPE_BLOB,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1372
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1373 1374
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1375 1376 1377 1378 1379
    {.type = TSDB_DATA_TYPE_MEDIUMBLOB,
     .bytes = 1,
     .isVarLen = 1,
     .startFn = tCompBinaryStart,
     .cmprFn = tCompBinary,
H
Hongze Cheng 已提交
1380
     .endFn = tCompBinaryEnd,
H
Hongze Cheng 已提交
1381 1382
     .getI64 = NULL,
     .putI64 = NULL},
H
Hongze Cheng 已提交
1383 1384 1385
};

struct SCompressor {
H
Hongze Cheng 已提交
1386 1387
  int8_t   type;
  int8_t   cmprAlg;
H
Hongze Cheng 已提交
1388
  int8_t   autoAlloc;
H
Hongze Cheng 已提交
1389
  int32_t  nVal;
H
Hongze Cheng 已提交
1390 1391 1392
  uint8_t *pBuf;
  int32_t  nBuf;
  uint8_t *aBuf[1];
H
Hongze Cheng 已提交
1393 1394 1395 1396 1397 1398 1399 1400 1401
  union {
    // Timestamp ----
    struct {
      int64_t  ts_prev_val;
      int64_t  ts_prev_delta;
      uint8_t *ts_flag_p;
    };
    // Integer ----
    struct {
H
Hongze Cheng 已提交
1402 1403 1404 1405
      int64_t  i_prev;
      int32_t  i_selector;
      int32_t  i_start;
      int32_t  i_end;
H
Hongze Cheng 已提交
1406
      int32_t  i_nEle;
H
Hongze Cheng 已提交
1407 1408
      uint64_t i_aZigzag[241];
      int8_t   i_aBitN[241];
H
Hongze Cheng 已提交
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420
    };
    // Float ----
    struct {
      uint32_t f_prev;
      uint8_t *f_flag_p;
    };
    // Double ----
    struct {
      uint64_t d_prev;
      uint8_t *d_flag_p;
    };
  };
H
Hongze Cheng 已提交
1421
};
H
Hongze Cheng 已提交
1422

H
Hongze Cheng 已提交
1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442
static int32_t tTwoStageComp(SCompressor *pCmprsor, int32_t *szComp) {
  int32_t code = 0;

  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], pCmprsor->nBuf + 1))) {
    return code;
  }

  *szComp = LZ4_compress_default(pCmprsor->pBuf, pCmprsor->aBuf[0] + 1, pCmprsor->nBuf, pCmprsor->nBuf);
  if (*szComp && *szComp < pCmprsor->nBuf) {
    pCmprsor->aBuf[0][0] = 1;
    *szComp += 1;
  } else {
    pCmprsor->aBuf[0][0] = 0;
    memcpy(pCmprsor->aBuf[0] + 1, pCmprsor->pBuf, pCmprsor->nBuf);
    *szComp = pCmprsor->nBuf + 1;
  }

  return code;
}

H
Hongze Cheng 已提交
1443
// Timestamp =====================================================
H
Hongze Cheng 已提交
1444
static int32_t tCompTimestampStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
H
Hongze Cheng 已提交
1445
  int32_t code = 0;
H
Hongze Cheng 已提交
1446 1447 1448 1449 1450 1451 1452 1453

  pCmprsor->nBuf = 1;

  code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf);
  if (code) return code;

  pCmprsor->pBuf[0] = 1;

H
Hongze Cheng 已提交
1454 1455 1456
  return code;
}

H
Hongze Cheng 已提交
1457
static int32_t tCompTSSwitchToCopy(SCompressor *pCmprsor) {
H
Hongze Cheng 已提交
1458 1459
  int32_t code = 0;

H
Hongze Cheng 已提交
1460
  if (pCmprsor->nVal == 0) goto _exit;
H
Hongze Cheng 已提交
1461

H
Hongze Cheng 已提交
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471
  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], sizeof(int64_t) * pCmprsor->nVal + 1))) {
    return code;
  }

  int32_t n = 1;
  int32_t nBuf = 1;
  int64_t value;
  int64_t delta;
  for (int32_t iVal = 0; iVal < pCmprsor->nVal;) {
    uint8_t aN[2] = {(pCmprsor->pBuf[n] & 0xf), (pCmprsor->pBuf[n] >> 4)};
H
Hongze Cheng 已提交
1472

H
Hongze Cheng 已提交
1473
    n++;
H
Hongze Cheng 已提交
1474

H
Hongze Cheng 已提交
1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488
    for (int32_t i = 0; i < 2; i++) {
      uint64_t vZigzag = 0;
      for (uint8_t j = 0; j < aN[i]; j++) {
        vZigzag |= (((uint64_t)pCmprsor->pBuf[n]) << (8 * j));
        n++;
      }

      int64_t delta_of_delta = ZIGZAG_DECODE(int64_t, vZigzag);
      if (iVal) {
        delta = delta_of_delta + delta;
        value = delta + value;
      } else {
        delta = 0;
        value = delta_of_delta;
H
Hongze Cheng 已提交
1489 1490
      }

H
Hongze Cheng 已提交
1491 1492
      memcpy(pCmprsor->aBuf[0] + nBuf, &value, sizeof(value));
      nBuf += sizeof(int64_t);
H
Hongze Cheng 已提交
1493

H
Hongze Cheng 已提交
1494 1495
      iVal++;
      if (iVal >= pCmprsor->nVal) break;
H
Hongze Cheng 已提交
1496 1497
    }
  }
H
Hongze Cheng 已提交
1498

H
Hongze Cheng 已提交
1499
  ASSERT(n == pCmprsor->nBuf && nBuf == sizeof(int64_t) * pCmprsor->nVal + 1);
H
Hongze Cheng 已提交
1500 1501 1502 1503 1504 1505 1506 1507

  uint8_t *pBuf = pCmprsor->pBuf;
  pCmprsor->pBuf = pCmprsor->aBuf[0];
  pCmprsor->aBuf[0] = pBuf;
  pCmprsor->nBuf = nBuf;

_exit:
  pCmprsor->pBuf[0] = 0;
H
Hongze Cheng 已提交
1508 1509
  return code;
}
H
Hongze Cheng 已提交
1510
static int32_t tCompTimestamp(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
1511 1512
  int32_t code = 0;

H
Hongze Cheng 已提交
1513 1514
  int64_t ts = *(int64_t *)pData;
  ASSERT(nData == 8);
H
Hongze Cheng 已提交
1515

H
Hongze Cheng 已提交
1516
  if (pCmprsor->pBuf[0] == 1) {
H
Hongze Cheng 已提交
1517
    if (pCmprsor->nVal == 0) {
H
Hongze Cheng 已提交
1518 1519 1520
      pCmprsor->ts_prev_val = ts;
      pCmprsor->ts_prev_delta = -ts;
    }
H
Hongze Cheng 已提交
1521

H
Hongze Cheng 已提交
1522
    if (!I64_SAFE_ADD(ts, -pCmprsor->ts_prev_val)) {
H
Hongze Cheng 已提交
1523
      code = tCompTSSwitchToCopy(pCmprsor);
H
Hongze Cheng 已提交
1524 1525 1526 1527
      if (code) return code;
      goto _copy_cmpr;
    }
    int64_t delta = ts - pCmprsor->ts_prev_val;
H
Hongze Cheng 已提交
1528

H
Hongze Cheng 已提交
1529
    if (!I64_SAFE_ADD(delta, -pCmprsor->ts_prev_delta)) {
H
Hongze Cheng 已提交
1530
      code = tCompTSSwitchToCopy(pCmprsor);
H
Hongze Cheng 已提交
1531 1532 1533 1534 1535
      if (code) return code;
      goto _copy_cmpr;
    }
    int64_t  delta_of_delta = delta - pCmprsor->ts_prev_delta;
    uint64_t vZigzag = ZIGZAG_ENCODE(int64_t, delta_of_delta);
H
Hongze Cheng 已提交
1536

H
Hongze Cheng 已提交
1537 1538
    pCmprsor->ts_prev_val = ts;
    pCmprsor->ts_prev_delta = delta;
H
Hongze Cheng 已提交
1539

H
Hongze Cheng 已提交
1540
    if ((pCmprsor->nVal & 0x1) == 0) {
H
Hongze Cheng 已提交
1541 1542
      if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + 17))) {
        return code;
H
Hongze Cheng 已提交
1543
      }
H
Hongze Cheng 已提交
1544

H
Hongze Cheng 已提交
1545 1546
      pCmprsor->ts_flag_p = pCmprsor->pBuf + pCmprsor->nBuf;
      pCmprsor->nBuf++;
H
Hongze Cheng 已提交
1547 1548
      pCmprsor->ts_flag_p[0] = 0;
      while (vZigzag) {
H
Hongze Cheng 已提交
1549 1550
        pCmprsor->pBuf[pCmprsor->nBuf] = (vZigzag & 0xff);
        pCmprsor->nBuf++;
H
Hongze Cheng 已提交
1551 1552 1553 1554 1555
        pCmprsor->ts_flag_p[0]++;
        vZigzag >>= 8;
      }
    } else {
      while (vZigzag) {
H
Hongze Cheng 已提交
1556 1557
        pCmprsor->pBuf[pCmprsor->nBuf] = (vZigzag & 0xff);
        pCmprsor->nBuf++;
H
Hongze Cheng 已提交
1558 1559 1560
        pCmprsor->ts_flag_p[0] += 0x10;
        vZigzag >>= 8;
      }
H
Hongze Cheng 已提交
1561 1562
    }
  } else {
H
Hongze Cheng 已提交
1563
  _copy_cmpr:
H
Hongze Cheng 已提交
1564 1565
    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + sizeof(ts)))) {
      return code;
H
Hongze Cheng 已提交
1566 1567
    }

H
Hongze Cheng 已提交
1568 1569
    memcpy(pCmprsor->pBuf + pCmprsor->nBuf, &ts, sizeof(ts));
    pCmprsor->nBuf += sizeof(ts);
H
Hongze Cheng 已提交
1570
  }
H
Hongze Cheng 已提交
1571
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
1572 1573 1574 1575

  return code;
}

H
Hongze Cheng 已提交
1576 1577
static int32_t tCompTimestampEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1578 1579

  if (pCmprsor->nBuf >= sizeof(int64_t) * pCmprsor->nVal + 1 && pCmprsor->pBuf[0] == 1) {
H
Hongze Cheng 已提交
1580
    code = tCompTSSwitchToCopy(pCmprsor);
H
Hongze Cheng 已提交
1581 1582 1583 1584
    if (code) return code;
  }

  if (pCmprsor->cmprAlg == TWO_STAGE_COMP) {
H
Hongze Cheng 已提交
1585 1586
    code = tTwoStageComp(pCmprsor, nData);
    if (code) return code;
H
Hongze Cheng 已提交
1587 1588 1589 1590 1591 1592 1593 1594
    *ppData = pCmprsor->aBuf[0];
  } else if (pCmprsor->cmprAlg == ONE_STAGE_COMP) {
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  } else {
    ASSERT(0);
  }

H
Hongze Cheng 已提交
1595 1596 1597
  return code;
}

H
Hongze Cheng 已提交
1598 1599
// Integer =====================================================
#define SIMPLE8B_MAX ((uint64_t)1152921504606846974LL)
H
Hongze Cheng 已提交
1600
static const uint8_t BIT_PER_INTEGER[] = {0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 15, 20, 30, 60};
H
Hongze Cheng 已提交
1601
static const int32_t SELECTOR_TO_ELEMS[] = {240, 120, 60, 30, 20, 15, 12, 10, 8, 7, 6, 5, 4, 3, 2, 1};
H
Hongze Cheng 已提交
1602
static const uint8_t BIT_TO_SELECTOR[] = {0,  2,  3,  4,  5,  6,  7,  8,  9,  10, 10, 11, 11, 12, 12, 12,
H
Hongze Cheng 已提交
1603 1604 1605
                                          13, 13, 13, 13, 13, 14, 14, 14, 14, 14, 14, 14, 14, 14, 14, 15,
                                          15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15,
                                          15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15, 15};
H
Hongze Cheng 已提交
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617
static const int32_t NEXT_IDX[] = {
    1,   2,   3,   4,   5,   6,   7,   8,   9,   10,  11,  12,  13,  14,  15,  16,  17,  18,  19,  20,  21,  22,
    23,  24,  25,  26,  27,  28,  29,  30,  31,  32,  33,  34,  35,  36,  37,  38,  39,  40,  41,  42,  43,  44,
    45,  46,  47,  48,  49,  50,  51,  52,  53,  54,  55,  56,  57,  58,  59,  60,  61,  62,  63,  64,  65,  66,
    67,  68,  69,  70,  71,  72,  73,  74,  75,  76,  77,  78,  79,  80,  81,  82,  83,  84,  85,  86,  87,  88,
    89,  90,  91,  92,  93,  94,  95,  96,  97,  98,  99,  100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110,
    111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132,
    133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154,
    155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176,
    177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198,
    199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220,
    221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 0};
H
Hongze Cheng 已提交
1618

H
Hongze Cheng 已提交
1619
static int32_t tCompIntStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
H
Hongze Cheng 已提交
1620
  int32_t code = 0;
H
Hongze Cheng 已提交
1621

H
Hongze Cheng 已提交
1622 1623 1624 1625
  pCmprsor->i_prev = 0;
  pCmprsor->i_selector = 0;
  pCmprsor->i_start = 0;
  pCmprsor->i_end = 0;
H
Hongze Cheng 已提交
1626
  pCmprsor->i_nEle = 0;
H
Hongze Cheng 已提交
1627 1628
  pCmprsor->nBuf = 1;

H
Hongze Cheng 已提交
1629
  code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf);
H
Hongze Cheng 已提交
1630 1631 1632 1633 1634 1635 1636 1637 1638
  if (code) return code;

  pCmprsor->pBuf[0] = 0;

  return code;
}

static int32_t tCompIntSwitchToCopy(SCompressor *pCmprsor) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1639 1640 1641

  if (pCmprsor->nVal == 0) goto _exit;

H
Hongze Cheng 已提交
1642 1643 1644 1645 1646 1647 1648 1649
  int32_t size = DATA_TYPE_INFO[pCmprsor->type].bytes * pCmprsor->nVal + 1;
  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], size))) {
    return code;
  }

  int32_t n = 1;
  int32_t nBuf = 1;
  int64_t vPrev = 0;
H
Hongze Cheng 已提交
1650
  while (n < pCmprsor->nBuf) {
H
Hongze Cheng 已提交
1651 1652 1653 1654
    uint64_t b;
    memcpy(&b, pCmprsor->pBuf + n, sizeof(b));
    n += sizeof(b);

H
Hongze Cheng 已提交
1655 1656 1657 1658
    int32_t  i_selector = (b & 0xf);
    int32_t  nEle = SELECTOR_TO_ELEMS[i_selector];
    uint8_t  bits = BIT_PER_INTEGER[i_selector];
    uint64_t mask = (((uint64_t)1) << bits) - 1;
H
Hongze Cheng 已提交
1659
    for (int32_t iEle = 0; iEle < nEle; iEle++) {
H
Hongze Cheng 已提交
1660
      uint64_t vZigzag = (b >> (bits * iEle + 4)) & mask;
H
Hongze Cheng 已提交
1661
      vPrev = ZIGZAG_DECODE(int64_t, vZigzag) + vPrev;
H
Hongze Cheng 已提交
1662

H
Hongze Cheng 已提交
1663
      DATA_TYPE_INFO[pCmprsor->type].putI64(vPrev, pCmprsor->aBuf[0] + nBuf);
H
Hongze Cheng 已提交
1664 1665
      nBuf += DATA_TYPE_INFO[pCmprsor->type].bytes;
    }
H
Hongze Cheng 已提交
1666 1667
  }

H
Hongze Cheng 已提交
1668
  while (pCmprsor->i_nEle) {
H
Hongze Cheng 已提交
1669 1670 1671 1672
    vPrev = ZIGZAG_DECODE(int64_t, pCmprsor->i_aZigzag[pCmprsor->i_start]) + vPrev;

    memcpy(pCmprsor->aBuf[0] + nBuf, &vPrev, DATA_TYPE_INFO[pCmprsor->type].bytes);
    nBuf += DATA_TYPE_INFO[pCmprsor->type].bytes;
H
Hongze Cheng 已提交
1673 1674 1675

    pCmprsor->i_start = NEXT_IDX[pCmprsor->i_start];
    pCmprsor->i_nEle--;
H
Hongze Cheng 已提交
1676 1677 1678 1679 1680 1681 1682 1683
  }

  ASSERT(n == pCmprsor->nBuf && nBuf == size);

  uint8_t *pBuf = pCmprsor->pBuf;
  pCmprsor->pBuf = pCmprsor->aBuf[0];
  pCmprsor->aBuf[0] = pBuf;
  pCmprsor->nBuf = size;
H
Hongze Cheng 已提交
1684 1685 1686

_exit:
  pCmprsor->pBuf[0] = 1;
H
Hongze Cheng 已提交
1687 1688 1689
  return code;
}

H
Hongze Cheng 已提交
1690
static int32_t tCompInt(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
1691 1692
  int32_t code = 0;

H
Hongze Cheng 已提交
1693
  ASSERT(nData == DATA_TYPE_INFO[pCmprsor->type].bytes);
H
Hongze Cheng 已提交
1694

H
Hongze Cheng 已提交
1695
  if (pCmprsor->pBuf[0] == 0) {
H
Hongze Cheng 已提交
1696
    int64_t val = DATA_TYPE_INFO[pCmprsor->type].getI64(pData);
H
Hongze Cheng 已提交
1697

H
Hongze Cheng 已提交
1698
    if (!I64_SAFE_ADD(val, -pCmprsor->i_prev)) {
H
Hongze Cheng 已提交
1699 1700
      code = tCompIntSwitchToCopy(pCmprsor);
      if (code) return code;
H
Hongze Cheng 已提交
1701 1702
      goto _copy_cmpr;
    }
H
Hongze Cheng 已提交
1703

H
Hongze Cheng 已提交
1704 1705 1706
    int64_t  diff = val - pCmprsor->i_prev;
    uint64_t vZigzag = ZIGZAG_ENCODE(int64_t, diff);
    if (vZigzag >= SIMPLE8B_MAX) {
H
Hongze Cheng 已提交
1707 1708
      code = tCompIntSwitchToCopy(pCmprsor);
      if (code) return code;
H
Hongze Cheng 已提交
1709
      goto _copy_cmpr;
H
Hongze Cheng 已提交
1710 1711
    }

H
Hongze Cheng 已提交
1712 1713
    int8_t nBit = (vZigzag) ? (64 - BUILDIN_CLZL(vZigzag)) : 0;
    pCmprsor->i_prev = val;
H
Hongze Cheng 已提交
1714

H
Hongze Cheng 已提交
1715
    for (;;) {
H
Hongze Cheng 已提交
1716 1717
      if (pCmprsor->i_nEle + 1 <= SELECTOR_TO_ELEMS[pCmprsor->i_selector] &&
          pCmprsor->i_nEle + 1 <= SELECTOR_TO_ELEMS[BIT_TO_SELECTOR[nBit]]) {
H
Hongze Cheng 已提交
1718 1719 1720 1721 1722
        if (pCmprsor->i_selector < BIT_TO_SELECTOR[nBit]) {
          pCmprsor->i_selector = BIT_TO_SELECTOR[nBit];
        }
        pCmprsor->i_aZigzag[pCmprsor->i_end] = vZigzag;
        pCmprsor->i_aBitN[pCmprsor->i_end] = nBit;
H
Hongze Cheng 已提交
1723 1724
        pCmprsor->i_end = NEXT_IDX[pCmprsor->i_end];
        pCmprsor->i_nEle++;
H
Hongze Cheng 已提交
1725 1726
        break;
      } else {
H
Hongze Cheng 已提交
1727
        if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1728 1729 1730 1731 1732
          int32_t lidx = pCmprsor->i_selector + 1;
          int32_t ridx = 15;
          while (lidx <= ridx) {
            pCmprsor->i_selector = (lidx + ridx) >> 1;

H
Hongze Cheng 已提交
1733
            if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1734
              lidx = pCmprsor->i_selector + 1;
H
Hongze Cheng 已提交
1735
            } else if (pCmprsor->i_nEle > SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1736 1737 1738 1739 1740 1741
              ridx = pCmprsor->i_selector - 1;
            } else {
              break;
            }
          }

H
Hongze Cheng 已提交
1742
          if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) pCmprsor->i_selector++;
H
Hongze Cheng 已提交
1743
        }
H
Hongze Cheng 已提交
1744
        int32_t nEle = SELECTOR_TO_ELEMS[pCmprsor->i_selector];
H
Hongze Cheng 已提交
1745

H
Hongze Cheng 已提交
1746 1747
        if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + sizeof(uint64_t)))) {
          return code;
H
Hongze Cheng 已提交
1748
        }
H
Hongze Cheng 已提交
1749

H
Hongze Cheng 已提交
1750 1751
        uint64_t *bp = (uint64_t *)(pCmprsor->pBuf + pCmprsor->nBuf);
        pCmprsor->nBuf += sizeof(uint64_t);
H
Hongze Cheng 已提交
1752 1753 1754
        bp[0] = pCmprsor->i_selector;
        uint8_t bits = BIT_PER_INTEGER[pCmprsor->i_selector];
        for (int32_t iVal = 0; iVal < nEle; iVal++) {
H
Hongze Cheng 已提交
1755 1756 1757
          bp[0] |= (pCmprsor->i_aZigzag[pCmprsor->i_start] << (bits * iVal + 4));
          pCmprsor->i_start = NEXT_IDX[pCmprsor->i_start];
          pCmprsor->i_nEle--;
H
Hongze Cheng 已提交
1758
        }
H
Hongze Cheng 已提交
1759 1760 1761

        // reset and continue
        pCmprsor->i_selector = 0;
H
Hongze Cheng 已提交
1762
        for (int32_t iVal = pCmprsor->i_start; iVal < pCmprsor->i_end; iVal = NEXT_IDX[iVal]) {
H
Hongze Cheng 已提交
1763 1764 1765 1766 1767
          if (pCmprsor->i_selector < BIT_TO_SELECTOR[pCmprsor->i_aBitN[iVal]]) {
            pCmprsor->i_selector = BIT_TO_SELECTOR[pCmprsor->i_aBitN[iVal]];
          }
        }
      }
H
Hongze Cheng 已提交
1768 1769 1770
    }
  } else {
  _copy_cmpr:
H
Hongze Cheng 已提交
1771
    code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + nData);
H
Hongze Cheng 已提交
1772 1773
    if (code) return code;

H
Hongze Cheng 已提交
1774 1775
    memcpy(pCmprsor->pBuf + pCmprsor->nBuf, pData, nData);
    pCmprsor->nBuf += nData;
H
Hongze Cheng 已提交
1776 1777
  }
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
1778 1779 1780 1781

  return code;
}

H
Hongze Cheng 已提交
1782
static int32_t tCompIntEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
H
Hongze Cheng 已提交
1783
  int32_t code = 0;
H
Hongze Cheng 已提交
1784

H
Hongze Cheng 已提交
1785 1786
  for (; pCmprsor->i_nEle;) {
    if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1787 1788 1789 1790 1791
      int32_t lidx = pCmprsor->i_selector + 1;
      int32_t ridx = 15;
      while (lidx <= ridx) {
        pCmprsor->i_selector = (lidx + ridx) >> 1;

H
Hongze Cheng 已提交
1792
        if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1793
          lidx = pCmprsor->i_selector + 1;
H
Hongze Cheng 已提交
1794
        } else if (pCmprsor->i_nEle > SELECTOR_TO_ELEMS[pCmprsor->i_selector]) {
H
Hongze Cheng 已提交
1795 1796 1797 1798 1799 1800
          ridx = pCmprsor->i_selector - 1;
        } else {
          break;
        }
      }

H
Hongze Cheng 已提交
1801
      if (pCmprsor->i_nEle < SELECTOR_TO_ELEMS[pCmprsor->i_selector]) pCmprsor->i_selector++;
H
Hongze Cheng 已提交
1802
    }
H
Hongze Cheng 已提交
1803
    int32_t nEle = SELECTOR_TO_ELEMS[pCmprsor->i_selector];
H
Hongze Cheng 已提交
1804 1805 1806 1807 1808 1809 1810 1811 1812 1813

    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + sizeof(uint64_t)))) {
      return code;
    }

    uint64_t *bp = (uint64_t *)(pCmprsor->pBuf + pCmprsor->nBuf);
    pCmprsor->nBuf += sizeof(uint64_t);
    bp[0] = pCmprsor->i_selector;
    uint8_t bits = BIT_PER_INTEGER[pCmprsor->i_selector];
    for (int32_t iVal = 0; iVal < nEle; iVal++) {
H
Hongze Cheng 已提交
1814 1815 1816
      bp[0] |= (pCmprsor->i_aZigzag[pCmprsor->i_start] << (bits * iVal + 4));
      pCmprsor->i_start = NEXT_IDX[pCmprsor->i_start];
      pCmprsor->i_nEle--;
H
Hongze Cheng 已提交
1817 1818 1819 1820 1821
    }

    pCmprsor->i_selector = 0;
  }

H
Hongze Cheng 已提交
1822 1823 1824 1825 1826 1827
  if (pCmprsor->nBuf >= DATA_TYPE_INFO[pCmprsor->type].bytes * pCmprsor->nVal + 1 && pCmprsor->pBuf[0] == 0) {
    code = tCompIntSwitchToCopy(pCmprsor);
    if (code) return code;
  }

  if (pCmprsor->cmprAlg == TWO_STAGE_COMP) {
H
Hongze Cheng 已提交
1828 1829
    code = tTwoStageComp(pCmprsor, nData);
    if (code) return code;
H
Hongze Cheng 已提交
1830 1831 1832 1833 1834 1835 1836 1837
    *ppData = pCmprsor->aBuf[0];
  } else if (pCmprsor->cmprAlg == ONE_STAGE_COMP) {
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  } else {
    ASSERT(0);
  }

H
Hongze Cheng 已提交
1838 1839 1840
  return code;
}

H
Hongze Cheng 已提交
1841 1842 1843
// Float =====================================================
static int32_t tCompFloatStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1844

H
Hongze Cheng 已提交
1845 1846
  pCmprsor->f_prev = 0;
  pCmprsor->f_flag_p = NULL;
H
Hongze Cheng 已提交
1847 1848 1849 1850 1851 1852 1853 1854

  pCmprsor->nBuf = 1;

  code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf);
  if (code) return code;

  pCmprsor->pBuf[0] = 0;

H
Hongze Cheng 已提交
1855 1856 1857
  return code;
}

H
Hongze Cheng 已提交
1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910
static int32_t tCompFloatSwitchToCopy(SCompressor *pCmprsor) {
  int32_t code = 0;

  if (pCmprsor->nVal == 0) goto _exit;

  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], sizeof(float) * pCmprsor->nVal + 1))) {
    return code;
  }

  int32_t n = 1;
  int32_t nBuf = 1;
  union {
    float    f;
    uint32_t u;
  } val = {.u = 0};

  for (int32_t iVal = 0; iVal < pCmprsor->nVal;) {
    uint8_t flags[2] = {(pCmprsor->pBuf[n] & 0xf), (pCmprsor->pBuf[n] >> 4)};

    n++;

    for (int8_t i = 0; i < 2; i++) {
      uint8_t flag = flags[i];

      uint32_t diff = 0;
      int8_t   nBytes = (flag & 0x7) + 1;
      for (int j = 0; j < nBytes; j++) {
        diff |= (((uint32_t)pCmprsor->pBuf[n]) << (8 * j));
        n++;
      }

      if (flag & 0x8) {
        diff <<= (32 - nBytes * 8);
      }

      val.u ^= diff;

      memcpy(pCmprsor->aBuf[0] + nBuf, &val.f, sizeof(val));
      nBuf += sizeof(val);

      iVal++;
      if (iVal >= pCmprsor->nVal) break;
    }
  }
  uint8_t *pBuf = pCmprsor->pBuf;
  pCmprsor->pBuf = pCmprsor->aBuf[0];
  pCmprsor->aBuf[0] = pBuf;
  pCmprsor->nBuf = nBuf;

_exit:
  pCmprsor->pBuf[0] = 1;
  return code;
}
H
Hongze Cheng 已提交
1911
static int32_t tCompFloat(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
1912 1913
  int32_t code = 0;

H
Hongze Cheng 已提交
1914 1915
  ASSERT(nData == sizeof(float));

H
Hongze Cheng 已提交
1916 1917 1918
  union {
    float    f;
    uint32_t u;
H
Hongze Cheng 已提交
1919
  } val = {.f = *(float *)pData};
H
Hongze Cheng 已提交
1920 1921 1922 1923 1924 1925 1926 1927 1928

  uint32_t diff = val.u ^ pCmprsor->f_prev;
  pCmprsor->f_prev = val.u;

  int32_t clz, ctz;
  if (diff) {
    clz = BUILDIN_CLZ(diff);
    ctz = BUILDIN_CTZ(diff);
  } else {
H
Hongze Cheng 已提交
1929 1930
    clz = 32;
    ctz = 32;
H
Hongze Cheng 已提交
1931 1932
  }

H
Hongze Cheng 已提交
1933 1934 1935 1936 1937 1938 1939 1940 1941
  uint8_t nBytes;
  if (clz < ctz) {
    nBytes = sizeof(uint32_t) - ctz / BITS_PER_BYTE;
    if (nBytes) diff >>= (32 - nBytes * BITS_PER_BYTE);
  } else {
    nBytes = sizeof(uint32_t) - clz / BITS_PER_BYTE;
  }
  if (nBytes == 0) nBytes++;

H
Hongze Cheng 已提交
1942
  if ((pCmprsor->nVal & 0x1) == 0) {
H
Hongze Cheng 已提交
1943 1944
    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + 9))) {
      return code;
H
Hongze Cheng 已提交
1945
    }
H
Hongze Cheng 已提交
1946

H
Hongze Cheng 已提交
1947 1948
    pCmprsor->f_flag_p = &pCmprsor->pBuf[pCmprsor->nBuf];
    pCmprsor->nBuf++;
H
Hongze Cheng 已提交
1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961

    if (clz < ctz) {
      pCmprsor->f_flag_p[0] = (0x08 | (nBytes - 1));
    } else {
      pCmprsor->f_flag_p[0] = nBytes - 1;
    }
  } else {
    if (clz < ctz) {
      pCmprsor->f_flag_p[0] |= ((0x08 | (nBytes - 1)) << 4);
    } else {
      pCmprsor->f_flag_p[0] |= ((nBytes - 1) << 4);
    }
  }
H
Hongze Cheng 已提交
1962
  for (; nBytes; nBytes--) {
H
Hongze Cheng 已提交
1963 1964
    pCmprsor->pBuf[pCmprsor->nBuf] = (diff & 0xff);
    pCmprsor->nBuf++;
H
Hongze Cheng 已提交
1965 1966
    diff >>= BITS_PER_BYTE;
  }
H
Hongze Cheng 已提交
1967
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
1968

H
Hongze Cheng 已提交
1969 1970 1971
  return code;
}

H
Hongze Cheng 已提交
1972
static int32_t tCompFloatEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
H
Hongze Cheng 已提交
1973
  int32_t code = 0;
H
Hongze Cheng 已提交
1974 1975 1976 1977 1978 1979 1980

  if (pCmprsor->nBuf >= sizeof(float) * pCmprsor->nVal + 1) {
    code = tCompFloatSwitchToCopy(pCmprsor);
    if (code) return code;
  }

  if (pCmprsor->cmprAlg == TWO_STAGE_COMP) {
H
Hongze Cheng 已提交
1981 1982
    code = tTwoStageComp(pCmprsor, nData);
    if (code) return code;
H
Hongze Cheng 已提交
1983 1984 1985 1986 1987 1988 1989 1990
    *ppData = pCmprsor->aBuf[0];
  } else if (pCmprsor->cmprAlg == ONE_STAGE_COMP) {
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  } else {
    ASSERT(0);
  }

H
Hongze Cheng 已提交
1991 1992 1993
  return code;
}

H
Hongze Cheng 已提交
1994 1995 1996
// Double =====================================================
static int32_t tCompDoubleStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
  int32_t code = 0;
H
Hongze Cheng 已提交
1997

H
Hongze Cheng 已提交
1998 1999
  pCmprsor->d_prev = 0;
  pCmprsor->d_flag_p = NULL;
H
Hongze Cheng 已提交
2000 2001 2002 2003 2004 2005 2006 2007

  pCmprsor->nBuf = 1;

  code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf);
  if (code) return code;

  pCmprsor->pBuf[0] = 0;

H
Hongze Cheng 已提交
2008 2009 2010
  return code;
}

H
Hongze Cheng 已提交
2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063
static int32_t tCompDoubleSwitchToCopy(SCompressor *pCmprsor) {
  int32_t code = 0;

  if (pCmprsor->nVal == 0) goto _exit;

  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], sizeof(double) * pCmprsor->nVal + 1))) {
    return code;
  }

  int32_t n = 1;
  int32_t nBuf = 1;
  union {
    double   f;
    uint64_t u;
  } val = {.u = 0};

  for (int32_t iVal = 0; iVal < pCmprsor->nVal;) {
    uint8_t flags[2] = {(pCmprsor->pBuf[n] & 0xf), (pCmprsor->pBuf[n] >> 4)};

    n++;

    for (int8_t i = 0; i < 2; i++) {
      uint8_t flag = flags[i];

      uint64_t diff = 0;
      int8_t   nBytes = (flag & 0x7) + 1;
      for (int j = 0; j < nBytes; j++) {
        diff |= (((uint64_t)pCmprsor->pBuf[n]) << (8 * j));
        n++;
      }

      if (flag & 0x8) {
        diff <<= (64 - nBytes * 8);
      }

      val.u ^= diff;

      memcpy(pCmprsor->aBuf[0] + nBuf, &val.f, sizeof(val));
      nBuf += sizeof(val);

      iVal++;
      if (iVal >= pCmprsor->nVal) break;
    }
  }
  uint8_t *pBuf = pCmprsor->pBuf;
  pCmprsor->pBuf = pCmprsor->aBuf[0];
  pCmprsor->aBuf[0] = pBuf;
  pCmprsor->nBuf = nBuf;

_exit:
  pCmprsor->pBuf[0] = 1;
  return code;
}
H
Hongze Cheng 已提交
2064
static int32_t tCompDouble(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
2065 2066
  int32_t code = 0;

H
Hongze Cheng 已提交
2067 2068
  ASSERT(nData == sizeof(double));

H
Hongze Cheng 已提交
2069 2070 2071
  union {
    double   d;
    uint64_t u;
H
Hongze Cheng 已提交
2072
  } val = {.d = *(double *)pData};
H
Hongze Cheng 已提交
2073 2074 2075 2076 2077 2078 2079 2080 2081

  uint64_t diff = val.u ^ pCmprsor->d_prev;
  pCmprsor->d_prev = val.u;

  int32_t clz, ctz;
  if (diff) {
    clz = BUILDIN_CLZL(diff);
    ctz = BUILDIN_CTZL(diff);
  } else {
H
Hongze Cheng 已提交
2082 2083
    clz = 64;
    ctz = 64;
H
Hongze Cheng 已提交
2084 2085
  }

H
Hongze Cheng 已提交
2086 2087 2088 2089 2090 2091 2092 2093 2094
  uint8_t nBytes;
  if (clz < ctz) {
    nBytes = sizeof(uint64_t) - ctz / BITS_PER_BYTE;
    if (nBytes) diff >>= (64 - nBytes * BITS_PER_BYTE);
  } else {
    nBytes = sizeof(uint64_t) - clz / BITS_PER_BYTE;
  }
  if (nBytes == 0) nBytes++;

H
Hongze Cheng 已提交
2095
  if ((pCmprsor->nVal & 0x1) == 0) {
H
Hongze Cheng 已提交
2096 2097
    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + 17))) {
      return code;
H
Hongze Cheng 已提交
2098
    }
H
Hongze Cheng 已提交
2099

H
Hongze Cheng 已提交
2100 2101
    pCmprsor->d_flag_p = &pCmprsor->pBuf[pCmprsor->nBuf];
    pCmprsor->nBuf++;
H
Hongze Cheng 已提交
2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114

    if (clz < ctz) {
      pCmprsor->d_flag_p[0] = (0x08 | (nBytes - 1));
    } else {
      pCmprsor->d_flag_p[0] = nBytes - 1;
    }
  } else {
    if (clz < ctz) {
      pCmprsor->d_flag_p[0] |= ((0x08 | (nBytes - 1)) << 4);
    } else {
      pCmprsor->d_flag_p[0] |= ((nBytes - 1) << 4);
    }
  }
H
Hongze Cheng 已提交
2115
  for (; nBytes; nBytes--) {
H
Hongze Cheng 已提交
2116 2117
    pCmprsor->pBuf[pCmprsor->nBuf] = (diff & 0xff);
    pCmprsor->nBuf++;
H
Hongze Cheng 已提交
2118 2119
    diff >>= BITS_PER_BYTE;
  }
H
Hongze Cheng 已提交
2120
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
2121

H
Hongze Cheng 已提交
2122 2123 2124
  return code;
}

H
Hongze Cheng 已提交
2125
static int32_t tCompDoubleEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
H
Hongze Cheng 已提交
2126
  int32_t code = 0;
H
Hongze Cheng 已提交
2127 2128 2129 2130 2131 2132 2133

  if (pCmprsor->nBuf >= sizeof(double) * pCmprsor->nVal + 1) {
    code = tCompDoubleSwitchToCopy(pCmprsor);
    if (code) return code;
  }

  if (pCmprsor->cmprAlg == TWO_STAGE_COMP) {
H
Hongze Cheng 已提交
2134 2135
    code = tTwoStageComp(pCmprsor, nData);
    if (code) return code;
H
Hongze Cheng 已提交
2136 2137 2138 2139 2140 2141 2142 2143
    *ppData = pCmprsor->aBuf[0];
  } else if (pCmprsor->cmprAlg == ONE_STAGE_COMP) {
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  } else {
    ASSERT(0);
  }

H
Hongze Cheng 已提交
2144 2145 2146
  return code;
}

H
Hongze Cheng 已提交
2147 2148
// Binary =====================================================
static int32_t tCompBinaryStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
H
Hongze Cheng 已提交
2149 2150
  pCmprsor->nBuf = 1;
  return 0;
H
Hongze Cheng 已提交
2151 2152
}

H
Hongze Cheng 已提交
2153
static int32_t tCompBinary(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
2154 2155 2156
  int32_t code = 0;

  if (nData) {
H
Hongze Cheng 已提交
2157 2158
    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf + nData))) {
      return code;
H
Hongze Cheng 已提交
2159
    }
H
Hongze Cheng 已提交
2160

H
Hongze Cheng 已提交
2161 2162
    memcpy(pCmprsor->pBuf + pCmprsor->nBuf, pData, nData);
    pCmprsor->nBuf += nData;
H
Hongze Cheng 已提交
2163
  }
H
Hongze Cheng 已提交
2164
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
2165 2166 2167 2168

  return code;
}

H
Hongze Cheng 已提交
2169 2170
static int32_t tCompBinaryEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
  int32_t code = 0;
H
Hongze Cheng 已提交
2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189

  if (pCmprsor->nBuf == 1) return code;

  if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->aBuf[0], pCmprsor->nBuf))) {
    return code;
  }

  int32_t szComp =
      LZ4_compress_default(pCmprsor->pBuf + 1, pCmprsor->aBuf[0] + 1, pCmprsor->nBuf - 1, pCmprsor->nBuf - 1);
  if (szComp && szComp < pCmprsor->nBuf - 1) {
    pCmprsor->aBuf[0][0] = 1;
    *ppData = pCmprsor->aBuf[0];
    *nData = szComp + 1;
  } else {
    pCmprsor->pBuf[0] = 0;
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  }

H
Hongze Cheng 已提交
2190 2191 2192
  return code;
}

H
Hongze Cheng 已提交
2193 2194 2195
// Bool =====================================================
static const uint8_t BOOL_CMPR_TABLE[] = {0b01, 0b0100, 0b010000, 0b01000000};

H
Hongze Cheng 已提交
2196
static int32_t tCompBoolStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
H
Hongze Cheng 已提交
2197 2198
  pCmprsor->nBuf = 0;
  return 0;
H
Hongze Cheng 已提交
2199 2200
}

H
Hongze Cheng 已提交
2201
static int32_t tCompBool(SCompressor *pCmprsor, const void *pData, int32_t nData) {
H
Hongze Cheng 已提交
2202 2203
  int32_t code = 0;

H
Hongze Cheng 已提交
2204 2205
  bool vBool = *(int8_t *)pData;

H
Hongze Cheng 已提交
2206
  int32_t mod4 = (pCmprsor->nVal & 3);
H
Hongze Cheng 已提交
2207
  if (mod4 == 0) {
H
Hongze Cheng 已提交
2208
    pCmprsor->nBuf++;
H
Hongze Cheng 已提交
2209

H
Hongze Cheng 已提交
2210 2211
    if (pCmprsor->autoAlloc && (code = tRealloc(&pCmprsor->pBuf, pCmprsor->nBuf))) {
      return code;
H
Hongze Cheng 已提交
2212 2213
    }

H
Hongze Cheng 已提交
2214
    pCmprsor->pBuf[pCmprsor->nBuf - 1] = 0;
H
Hongze Cheng 已提交
2215
  }
H
Hongze Cheng 已提交
2216
  if (vBool) {
H
Hongze Cheng 已提交
2217
    pCmprsor->pBuf[pCmprsor->nBuf - 1] |= BOOL_CMPR_TABLE[mod4];
H
Hongze Cheng 已提交
2218
  }
H
Hongze Cheng 已提交
2219
  pCmprsor->nVal++;
H
Hongze Cheng 已提交
2220 2221 2222 2223

  return code;
}

H
Hongze Cheng 已提交
2224 2225
static int32_t tCompBoolEnd(SCompressor *pCmprsor, const uint8_t **ppData, int32_t *nData) {
  int32_t code = 0;
H
Hongze Cheng 已提交
2226 2227

  if (pCmprsor->cmprAlg == TWO_STAGE_COMP) {
H
Hongze Cheng 已提交
2228 2229
    code = tTwoStageComp(pCmprsor, nData);
    if (code) return code;
H
Hongze Cheng 已提交
2230 2231 2232 2233 2234 2235 2236 2237
    *ppData = pCmprsor->aBuf[0];
  } else if (pCmprsor->cmprAlg == ONE_STAGE_COMP) {
    *ppData = pCmprsor->pBuf;
    *nData = pCmprsor->nBuf;
  } else {
    ASSERT(0);
  }

H
Hongze Cheng 已提交
2238 2239 2240
  return code;
}

H
Hongze Cheng 已提交
2241 2242 2243 2244 2245 2246 2247
// SCompressor =====================================================
int32_t tCompressorCreate(SCompressor **ppCmprsor) {
  int32_t code = 0;

  *ppCmprsor = (SCompressor *)taosMemoryCalloc(1, sizeof(SCompressor));
  if ((*ppCmprsor) == NULL) {
    code = TSDB_CODE_OUT_OF_MEMORY;
H
Hongze Cheng 已提交
2248
    return code;
H
Hongze Cheng 已提交
2249 2250 2251 2252 2253 2254 2255 2256
  }

  return code;
}

int32_t tCompressorDestroy(SCompressor *pCmprsor) {
  int32_t code = 0;

H
Hongze Cheng 已提交
2257 2258
  tFree(pCmprsor->pBuf);

H
Hongze Cheng 已提交
2259 2260 2261
  int32_t nBuf = sizeof(pCmprsor->aBuf) / sizeof(pCmprsor->aBuf[0]);
  for (int32_t iBuf = 0; iBuf < nBuf; iBuf++) {
    tFree(pCmprsor->aBuf[iBuf]);
H
Hongze Cheng 已提交
2262 2263
  }

H
Hongze Cheng 已提交
2264 2265
  taosMemoryFree(pCmprsor);

H
Hongze Cheng 已提交
2266 2267 2268
  return code;
}

H
Hongze Cheng 已提交
2269
int32_t tCompressStart(SCompressor *pCmprsor, int8_t type, int8_t cmprAlg) {
H
Hongze Cheng 已提交
2270 2271 2272 2273
  int32_t code = 0;

  pCmprsor->type = type;
  pCmprsor->cmprAlg = cmprAlg;
H
Hongze Cheng 已提交
2274
  pCmprsor->autoAlloc = 1;
H
Hongze Cheng 已提交
2275
  pCmprsor->nVal = 0;
H
Hongze Cheng 已提交
2276

H
Hongze Cheng 已提交
2277 2278
  if (DATA_TYPE_INFO[type].startFn) {
    DATA_TYPE_INFO[type].startFn(pCmprsor, type, cmprAlg);
H
Hongze Cheng 已提交
2279 2280 2281 2282 2283
  }

  return code;
}

H
Hongze Cheng 已提交
2284
int32_t tCompressEnd(SCompressor *pCmprsor, const uint8_t **ppOut, int32_t *nOut, int32_t *nOrigin) {
H
Hongze Cheng 已提交
2285 2286
  int32_t code = 0;

H
Hongze Cheng 已提交
2287 2288 2289 2290 2291 2292 2293 2294 2295
  *ppOut = NULL;
  *nOut = 0;
  if (nOrigin) {
    if (DATA_TYPE_INFO[pCmprsor->type].isVarLen) {
      *nOrigin = pCmprsor->nBuf - 1;
    } else {
      *nOrigin = pCmprsor->nVal * DATA_TYPE_INFO[pCmprsor->type].bytes;
    }
  }
H
Hongze Cheng 已提交
2296

H
Hongze Cheng 已提交
2297 2298
  if (pCmprsor->nVal == 0) return code;

H
Hongze Cheng 已提交
2299
  if (DATA_TYPE_INFO[pCmprsor->type].endFn) {
H
Hongze Cheng 已提交
2300
    return DATA_TYPE_INFO[pCmprsor->type].endFn(pCmprsor, ppOut, nOut);
H
Hongze Cheng 已提交
2301 2302 2303 2304 2305
  }

  return code;
}

H
Hongze Cheng 已提交
2306 2307
int32_t tCompress(SCompressor *pCmprsor, const void *pData, int64_t nData) {
  return DATA_TYPE_INFO[pCmprsor->type].cmprFn(pCmprsor, pData, nData);
H
Hongze Cheng 已提交
2308 2309 2310 2311 2312 2313
}

/*************************************************************************
 *                  REGULAR COMPRESSION
 *************************************************************************/
// Timestamp =====================================================
H
Hongze Cheng 已提交
2314 2315 2316 2317 2318 2319 2320
int32_t tsCompressTimestamp(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                            int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressTimestampImp(pIn, nEle, pOut);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressTimestampImp(pIn, nEle, pBuf);
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2321
  } else {
2322
    ASSERTS(0, "compress algo not one or two stage");
H
Hongze Cheng 已提交
2323 2324 2325 2326
    return -1;
  }
}

H
Hongze Cheng 已提交
2327 2328 2329 2330 2331 2332 2333
int32_t tsDecompressTimestamp(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg,
                              void *pBuf, int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressTimestampImp(pIn, nEle, pOut);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressTimestampImp(pBuf, nEle, pOut);
H
Hongze Cheng 已提交
2334
  } else {
2335
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2336 2337 2338 2339 2340
    return -1;
  }
}

// Float =====================================================
H
Hongze Cheng 已提交
2341 2342
int32_t tsCompressFloat(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                        int32_t nBuf) {
H
Hongze Cheng 已提交
2343 2344 2345
#ifdef TD_TSZ
  // lossy mode
  if (lossyFloat) {
H
Hongze Cheng 已提交
2346
    return tsCompressFloatLossyImp(pIn, nEle, pOut);
H
Hongze Cheng 已提交
2347 2348 2349
    // lossless mode
  } else {
#endif
H
Hongze Cheng 已提交
2350 2351 2352 2353 2354
    if (cmprAlg == ONE_STAGE_COMP) {
      return tsCompressFloatImp(pIn, nEle, pOut);
    } else if (cmprAlg == TWO_STAGE_COMP) {
      int32_t len = tsCompressFloatImp(pIn, nEle, pBuf);
      return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2355
    } else {
2356
      ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2357 2358 2359 2360 2361 2362 2363
      return -1;
    }
#ifdef TD_TSZ
  }
#endif
}

H
Hongze Cheng 已提交
2364 2365
int32_t tsDecompressFloat(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                          int32_t nBuf) {
H
Hongze Cheng 已提交
2366
#ifdef TD_TSZ
H
Hongze Cheng 已提交
2367
  if (HEAD_ALGO(pIn[0]) == ALGO_SZ_LOSSY) {
H
Hongze Cheng 已提交
2368
    // decompress lossy
H
Hongze Cheng 已提交
2369
    return tsDecompressFloatLossyImp(pIn, nIn, nEle, pOut);
H
Hongze Cheng 已提交
2370 2371 2372
  } else {
#endif
    // decompress lossless
H
Hongze Cheng 已提交
2373 2374 2375 2376 2377
    if (cmprAlg == ONE_STAGE_COMP) {
      return tsDecompressFloatImp(pIn, nEle, pOut);
    } else if (cmprAlg == TWO_STAGE_COMP) {
      if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
      return tsDecompressFloatImp(pBuf, nEle, pOut);
H
Hongze Cheng 已提交
2378
    } else {
2379
      ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2380 2381 2382 2383 2384 2385 2386 2387
      return -1;
    }
#ifdef TD_TSZ
  }
#endif
}

// Double =====================================================
H
Hongze Cheng 已提交
2388 2389
int32_t tsCompressDouble(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                         int32_t nBuf) {
H
Hongze Cheng 已提交
2390 2391 2392
#ifdef TD_TSZ
  if (lossyDouble) {
    // lossy mode
H
Hongze Cheng 已提交
2393
    return tsCompressDoubleLossyImp(pIn, nEle, pOut);
H
Hongze Cheng 已提交
2394 2395 2396
  } else {
#endif
    // lossless mode
H
Hongze Cheng 已提交
2397 2398 2399 2400 2401
    if (cmprAlg == ONE_STAGE_COMP) {
      return tsCompressDoubleImp(pIn, nEle, pOut);
    } else if (cmprAlg == TWO_STAGE_COMP) {
      int32_t len = tsCompressDoubleImp(pIn, nEle, pBuf);
      return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2402
    } else {
2403
      ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2404 2405 2406 2407 2408 2409 2410
      return -1;
    }
#ifdef TD_TSZ
  }
#endif
}

H
Hongze Cheng 已提交
2411 2412
int32_t tsDecompressDouble(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                           int32_t nBuf) {
H
Hongze Cheng 已提交
2413 2414 2415
#ifdef TD_TSZ
  if (HEAD_ALGO(input[0]) == ALGO_SZ_LOSSY) {
    // decompress lossy
H
Hongze Cheng 已提交
2416
    return tsDecompressDoubleLossyImp(pIn, nIn, nEle, pOut);
H
Hongze Cheng 已提交
2417 2418 2419
  } else {
#endif
    // decompress lossless
H
Hongze Cheng 已提交
2420 2421 2422 2423 2424
    if (cmprAlg == ONE_STAGE_COMP) {
      return tsDecompressDoubleImp(pIn, nEle, pOut);
    } else if (cmprAlg == TWO_STAGE_COMP) {
      if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
      return tsDecompressDoubleImp(pBuf, nEle, pOut);
H
Hongze Cheng 已提交
2425
    } else {
2426
      ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2427 2428 2429 2430 2431 2432 2433 2434
      return -1;
    }
#ifdef TD_TSZ
  }
#endif
}

// Binary =====================================================
H
Hongze Cheng 已提交
2435 2436 2437
int32_t tsCompressString(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                         int32_t nBuf) {
  return tsCompressStringImp(pIn, nIn, pOut, nOut);
H
Hongze Cheng 已提交
2438 2439
}

H
Hongze Cheng 已提交
2440 2441 2442
int32_t tsDecompressString(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                           int32_t nBuf) {
  return tsDecompressStringImp(pIn, nIn, pOut, nOut);
H
Hongze Cheng 已提交
2443 2444 2445
}

// Bool =====================================================
H
Hongze Cheng 已提交
2446 2447 2448 2449 2450 2451
int32_t tsCompressBool(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                       int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressBoolImp(pIn, nEle, pOut);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressBoolImp(pIn, nEle, pBuf);
2452 2453 2454
    if (len < 0) {
      return -1;
    }
H
Hongze Cheng 已提交
2455
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2456
  } else {
2457
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2458 2459 2460 2461
    return -1;
  }
}

H
Hongze Cheng 已提交
2462 2463 2464 2465 2466 2467 2468
int32_t tsDecompressBool(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                         int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressBoolImp(pIn, nEle, pOut);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressBoolImp(pBuf, nEle, pOut);
H
Hongze Cheng 已提交
2469
  } else {
2470
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2471 2472 2473 2474 2475
    return -1;
  }
}

// Tinyint =====================================================
H
Hongze Cheng 已提交
2476 2477 2478 2479 2480 2481 2482
int32_t tsCompressTinyint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                          int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_TINYINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressINTImp(pIn, nEle, pBuf, TSDB_DATA_TYPE_TINYINT);
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2483
  } else {
2484
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2485 2486 2487 2488
    return -1;
  }
}

H
Hongze Cheng 已提交
2489 2490 2491 2492 2493 2494 2495
int32_t tsDecompressTinyint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                            int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_TINYINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressINTImp(pBuf, nEle, pOut, TSDB_DATA_TYPE_TINYINT);
H
Hongze Cheng 已提交
2496
  } else {
2497
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2498 2499 2500 2501 2502
    return -1;
  }
}

// Smallint =====================================================
H
Hongze Cheng 已提交
2503 2504 2505 2506 2507 2508 2509
int32_t tsCompressSmallint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                           int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_SMALLINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressINTImp(pIn, nEle, pBuf, TSDB_DATA_TYPE_SMALLINT);
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2510
  } else {
2511
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2512 2513 2514 2515
    return -1;
  }
}

H
Hongze Cheng 已提交
2516 2517 2518 2519 2520 2521 2522
int32_t tsDecompressSmallint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg,
                             void *pBuf, int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_SMALLINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressINTImp(pBuf, nEle, pOut, TSDB_DATA_TYPE_SMALLINT);
H
Hongze Cheng 已提交
2523
  } else {
2524
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2525 2526 2527 2528 2529
    return -1;
  }
}

// Int =====================================================
H
Hongze Cheng 已提交
2530 2531 2532 2533 2534 2535 2536
int32_t tsCompressInt(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                      int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_INT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressINTImp(pIn, nEle, pBuf, TSDB_DATA_TYPE_INT);
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2537
  } else {
2538
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2539 2540 2541 2542
    return -1;
  }
}

H
Hongze Cheng 已提交
2543 2544 2545 2546 2547 2548 2549
int32_t tsDecompressInt(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                        int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_INT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressINTImp(pBuf, nEle, pOut, TSDB_DATA_TYPE_INT);
H
Hongze Cheng 已提交
2550
  } else {
2551
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2552 2553 2554 2555 2556
    return -1;
  }
}

// Bigint =====================================================
H
Hongze Cheng 已提交
2557 2558 2559 2560 2561 2562 2563
int32_t tsCompressBigint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                         int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsCompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_BIGINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    int32_t len = tsCompressINTImp(pIn, nEle, pBuf, TSDB_DATA_TYPE_BIGINT);
    return tsCompressStringImp(pBuf, len, pOut, nOut);
H
Hongze Cheng 已提交
2564
  } else {
2565
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2566 2567 2568 2569
    return -1;
  }
}

H
Hongze Cheng 已提交
2570 2571 2572 2573 2574 2575 2576
int32_t tsDecompressBigint(void *pIn, int32_t nIn, int32_t nEle, void *pOut, int32_t nOut, uint8_t cmprAlg, void *pBuf,
                           int32_t nBuf) {
  if (cmprAlg == ONE_STAGE_COMP) {
    return tsDecompressINTImp(pIn, nEle, pOut, TSDB_DATA_TYPE_BIGINT);
  } else if (cmprAlg == TWO_STAGE_COMP) {
    if (tsDecompressStringImp(pIn, nIn, pBuf, nBuf) < 0) return -1;
    return tsDecompressINTImp(pBuf, nEle, pOut, TSDB_DATA_TYPE_BIGINT);
H
Hongze Cheng 已提交
2577
  } else {
2578
    ASSERTS(0, "compress algo invalid");
H
Hongze Cheng 已提交
2579 2580 2581
    return -1;
  }
}