zstdmt_compress.h 8.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. /*
  2. * Copyright (c) 2016-present, Yann Collet, Facebook, Inc.
  3. * All rights reserved.
  4. *
  5. * This source code is licensed under both the BSD-style license (found in the
  6. * LICENSE file in the root directory of this source tree) and the GPLv2 (found
  7. * in the COPYING file in the root directory of this source tree).
  8. * You may select, at your option, one of the above-listed licenses.
  9. */
  10. #ifndef ZSTDMT_COMPRESS_H
  11. #define ZSTDMT_COMPRESS_H
  12. #if defined (__cplusplus)
  13. extern "C" {
  14. #endif
  15. /* Note : This is an internal API.
  16. * Some methods are still exposed (ZSTDLIB_API),
  17. * because it used to be the only way to invoke MT compression.
  18. * Now, it's recommended to use ZSTD_compress_generic() instead.
  19. * These methods will stop being exposed in a future version */
  20. /* === Dependencies === */
  21. #include <stddef.h> /* size_t */
  22. #define ZSTD_STATIC_LINKING_ONLY /* ZSTD_parameters */
  23. #include "zstd.h" /* ZSTD_inBuffer, ZSTD_outBuffer, ZSTDLIB_API */
  24. /* === Constants === */
  25. #ifndef ZSTDMT_NBWORKERS_MAX
  26. # define ZSTDMT_NBWORKERS_MAX 200
  27. #endif
  28. #ifndef ZSTDMT_JOBSIZE_MIN
  29. # define ZSTDMT_JOBSIZE_MIN (1 MB)
  30. #endif
  31. #define ZSTDMT_JOBSIZE_MAX (MEM_32bits() ? (512 MB) : (1024 MB))
  32. /* === Memory management === */
  33. typedef struct ZSTDMT_CCtx_s ZSTDMT_CCtx;
  34. ZSTDLIB_API ZSTDMT_CCtx* ZSTDMT_createCCtx(unsigned nbWorkers);
  35. ZSTDLIB_API ZSTDMT_CCtx* ZSTDMT_createCCtx_advanced(unsigned nbWorkers,
  36. ZSTD_customMem cMem);
  37. ZSTDLIB_API size_t ZSTDMT_freeCCtx(ZSTDMT_CCtx* mtctx);
  38. ZSTDLIB_API size_t ZSTDMT_sizeof_CCtx(ZSTDMT_CCtx* mtctx);
  39. /* === Simple one-pass compression function === */
  40. ZSTDLIB_API size_t ZSTDMT_compressCCtx(ZSTDMT_CCtx* mtctx,
  41. void* dst, size_t dstCapacity,
  42. const void* src, size_t srcSize,
  43. int compressionLevel);
  44. /* === Streaming functions === */
  45. ZSTDLIB_API size_t ZSTDMT_initCStream(ZSTDMT_CCtx* mtctx, int compressionLevel);
  46. ZSTDLIB_API size_t ZSTDMT_resetCStream(ZSTDMT_CCtx* mtctx, unsigned long long pledgedSrcSize); /**< if srcSize is not known at reset time, use ZSTD_CONTENTSIZE_UNKNOWN. Note: for compatibility with older programs, 0 means the same as ZSTD_CONTENTSIZE_UNKNOWN, but it will change in the future to mean "empty" */
  47. ZSTDLIB_API size_t ZSTDMT_nextInputSizeHint(const ZSTDMT_CCtx* mtctx);
  48. ZSTDLIB_API size_t ZSTDMT_compressStream(ZSTDMT_CCtx* mtctx, ZSTD_outBuffer* output, ZSTD_inBuffer* input);
  49. ZSTDLIB_API size_t ZSTDMT_flushStream(ZSTDMT_CCtx* mtctx, ZSTD_outBuffer* output); /**< @return : 0 == all flushed; >0 : still some data to be flushed; or an error code (ZSTD_isError()) */
  50. ZSTDLIB_API size_t ZSTDMT_endStream(ZSTDMT_CCtx* mtctx, ZSTD_outBuffer* output); /**< @return : 0 == all flushed; >0 : still some data to be flushed; or an error code (ZSTD_isError()) */
  51. /* === Advanced functions and parameters === */
  52. ZSTDLIB_API size_t ZSTDMT_compress_advanced(ZSTDMT_CCtx* mtctx,
  53. void* dst, size_t dstCapacity,
  54. const void* src, size_t srcSize,
  55. const ZSTD_CDict* cdict,
  56. ZSTD_parameters params,
  57. int overlapLog);
  58. ZSTDLIB_API size_t ZSTDMT_initCStream_advanced(ZSTDMT_CCtx* mtctx,
  59. const void* dict, size_t dictSize, /* dict can be released after init, a local copy is preserved within zcs */
  60. ZSTD_parameters params,
  61. unsigned long long pledgedSrcSize); /* pledgedSrcSize is optional and can be zero == unknown */
  62. ZSTDLIB_API size_t ZSTDMT_initCStream_usingCDict(ZSTDMT_CCtx* mtctx,
  63. const ZSTD_CDict* cdict,
  64. ZSTD_frameParameters fparams,
  65. unsigned long long pledgedSrcSize); /* note : zero means empty */
  66. /* ZSTDMT_parameter :
  67. * List of parameters that can be set using ZSTDMT_setMTCtxParameter() */
  68. typedef enum {
  69. ZSTDMT_p_jobSize, /* Each job is compressed in parallel. By default, this value is dynamically determined depending on compression parameters. Can be set explicitly here. */
  70. ZSTDMT_p_overlapLog, /* Each job may reload a part of previous job to enhance compressionr ratio; 0 == no overlap, 6(default) == use 1/8th of window, >=9 == use full window. This is a "sticky" parameter : its value will be re-used on next compression job */
  71. ZSTDMT_p_rsyncable /* Enables rsyncable mode. */
  72. } ZSTDMT_parameter;
  73. /* ZSTDMT_setMTCtxParameter() :
  74. * allow setting individual parameters, one at a time, among a list of enums defined in ZSTDMT_parameter.
  75. * The function must be called typically after ZSTD_createCCtx() but __before ZSTDMT_init*() !__
  76. * Parameters not explicitly reset by ZSTDMT_init*() remain the same in consecutive compression sessions.
  77. * @return : 0, or an error code (which can be tested using ZSTD_isError()) */
  78. ZSTDLIB_API size_t ZSTDMT_setMTCtxParameter(ZSTDMT_CCtx* mtctx, ZSTDMT_parameter parameter, int value);
  79. /* ZSTDMT_getMTCtxParameter() :
  80. * Query the ZSTDMT_CCtx for a parameter value.
  81. * @return : 0, or an error code (which can be tested using ZSTD_isError()) */
  82. ZSTDLIB_API size_t ZSTDMT_getMTCtxParameter(ZSTDMT_CCtx* mtctx, ZSTDMT_parameter parameter, int* value);
  83. /*! ZSTDMT_compressStream_generic() :
  84. * Combines ZSTDMT_compressStream() with optional ZSTDMT_flushStream() or ZSTDMT_endStream()
  85. * depending on flush directive.
  86. * @return : minimum amount of data still to be flushed
  87. * 0 if fully flushed
  88. * or an error code
  89. * note : needs to be init using any ZSTD_initCStream*() variant */
  90. ZSTDLIB_API size_t ZSTDMT_compressStream_generic(ZSTDMT_CCtx* mtctx,
  91. ZSTD_outBuffer* output,
  92. ZSTD_inBuffer* input,
  93. ZSTD_EndDirective endOp);
  94. /* ========================================================
  95. * === Private interface, for use by ZSTD_compress.c ===
  96. * === Not exposed in libzstd. Never invoke directly ===
  97. * ======================================================== */
  98. /*! ZSTDMT_toFlushNow()
  99. * Tell how many bytes are ready to be flushed immediately.
  100. * Probe the oldest active job (not yet entirely flushed) and check its output buffer.
  101. * If return 0, it means there is no active job,
  102. * or, it means oldest job is still active, but everything produced has been flushed so far,
  103. * therefore flushing is limited by speed of oldest job. */
  104. size_t ZSTDMT_toFlushNow(ZSTDMT_CCtx* mtctx);
  105. /*! ZSTDMT_CCtxParam_setMTCtxParameter()
  106. * like ZSTDMT_setMTCtxParameter(), but into a ZSTD_CCtx_Params */
  107. size_t ZSTDMT_CCtxParam_setMTCtxParameter(ZSTD_CCtx_params* params, ZSTDMT_parameter parameter, int value);
  108. /*! ZSTDMT_CCtxParam_setNbWorkers()
  109. * Set nbWorkers, and clamp it.
  110. * Also reset jobSize and overlapLog */
  111. size_t ZSTDMT_CCtxParam_setNbWorkers(ZSTD_CCtx_params* params, unsigned nbWorkers);
  112. /*! ZSTDMT_updateCParams_whileCompressing() :
  113. * Updates only a selected set of compression parameters, to remain compatible with current frame.
  114. * New parameters will be applied to next compression job. */
  115. void ZSTDMT_updateCParams_whileCompressing(ZSTDMT_CCtx* mtctx, const ZSTD_CCtx_params* cctxParams);
  116. /*! ZSTDMT_getFrameProgression():
  117. * tells how much data has been consumed (input) and produced (output) for current frame.
  118. * able to count progression inside worker threads.
  119. */
  120. ZSTD_frameProgression ZSTDMT_getFrameProgression(ZSTDMT_CCtx* mtctx);
  121. /*! ZSTDMT_initCStream_internal() :
  122. * Private use only. Init streaming operation.
  123. * expects params to be valid.
  124. * must receive dict, or cdict, or none, but not both.
  125. * @return : 0, or an error code */
  126. size_t ZSTDMT_initCStream_internal(ZSTDMT_CCtx* zcs,
  127. const void* dict, size_t dictSize, ZSTD_dictContentType_e dictContentType,
  128. const ZSTD_CDict* cdict,
  129. ZSTD_CCtx_params params, unsigned long long pledgedSrcSize);
  130. #if defined (__cplusplus)
  131. }
  132. #endif
  133. #endif /* ZSTDMT_COMPRESS_H */