aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/streams/zstd/zstd.h
blob: 4f1583d067faf74a6eb7a37b5664bc31b16eee96 (plain) (blame)
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
#pragma once 
 
#include <util/generic/ptr.h> 
#include <util/stream/input.h> 
#include <util/stream/output.h> 
 
/** 
 * @addtogroup Streams_Archs 
 * @{ 
 */ 
 
// @brief Stream to compress into zstd archive 
class TZstdCompress: public IOutputStream { 
public: 
    /** 
      @param slave stream to write compressed data to 
      @param quality, higher quality - slower but better compression. 
             0 is default compression (see constant ZSTD_CLEVEL_DEFAULT(3))
             max compression is  ZSTD_MAX_CLEVEL (22) 
    */ 
    explicit TZstdCompress(IOutputStream* slave, int quality = 0);
    ~TZstdCompress() override; 
private: 
    void DoWrite(const void* buffer, size_t size) override; 
    void DoFlush() override; 
    void DoFinish() override; 
 
public: 
    class TImpl; 
    THolder<TImpl> Impl_; 
}; 
 
//////////////////////////////////////////////////////////////////////////////// 
 
// @brief Buffered stream to decompress from zstd archive 
class TZstdDecompress: public IInputStream { 
public: 
    /** 
      @param slave stream to read compressed data from 
      @param bufferSize approximate size of buffer compressed data is read in 
    */ 
    explicit TZstdDecompress(IInputStream* slave, size_t bufferSize = 8 * 1024); 
    ~TZstdDecompress() override; 
 
private: 
    size_t DoRead(void* buffer, size_t size) override; 
 
private: 
    class TImpl; 
    THolder<TImpl> Impl_; 
}; 
 
/** @} */