deps/libchdr/deps/zstd-1.5.5/contrib/seqBench/seqBench.c

   1 #define ZSTD_STATIC_LINKING_ONLY
   2 #include <zstd.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <assert.h>
   6 #include <string.h>
   7
   8 int main(int argc, char *argv[]) {
   9     ZSTD_CCtx* zc = ZSTD_createCCtx();
  10
  11     if (argc != 2) {
  12         printf("Usage: seqBench <file>\n"); // TODO provide the block delim option here
  13         return 1;
  14     }
  15
  16     FILE *f = fopen(argv[1], "rb");
  17     fseek(f, 0, SEEK_END);
  18     long inBufSize = ftell(f);
  19     fseek(f, 0, SEEK_SET);
  20
  21     char *inBuf = malloc(inBufSize + 1);
  22     fread(inBuf, inBufSize, 1, f);
  23     fclose(f);
  24
  25     size_t seqsSize = ZSTD_sequenceBound(inBufSize);
  26     ZSTD_Sequence *seqs = (ZSTD_Sequence*)malloc(seqsSize * sizeof(ZSTD_Sequence));
  27     char *outBuf = malloc(ZSTD_compressBound(inBufSize));
  28
  29     ZSTD_generateSequences(zc, seqs, seqsSize, inBuf, inBufSize);
  30     ZSTD_CCtx_setParameter(zc, ZSTD_c_blockDelimiters, ZSTD_sf_explicitBlockDelimiters);
  31     size_t outBufSize = ZSTD_compressSequences(zc, outBuf, inBufSize, seqs, seqsSize, inBuf, inBufSize);
  32     if (ZSTD_isError(outBufSize)) {
  33         printf("ERROR: %lu\n", outBufSize);
  34         return 1;
  35     }
  36
  37     char *validationBuf = malloc(inBufSize);
  38     ZSTD_decompress(validationBuf, inBufSize, outBuf, outBufSize);
  39
  40     if (memcmp(inBuf, validationBuf, inBufSize) == 0) {
  41         printf("Compression and decompression were successful!\n");
  42     } else {
  43         printf("ERROR: input and validation buffers don't match!\n");
  44         for (int i = 0; i < inBufSize; i++) {
  45             if (inBuf[i] != validationBuf[i]) {
  46                 printf("First bad index: %d\n", i);
  47                 break;
  48             }
  49         }
  50     }
  51
  52     return 0;
  53 }