Allow Zstandard compression level to be configured.

PDAL · Mar 8, 2019 · 055044b · 055044b
1 parent c9dbb39
commit 055044b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pdal/compression/ZstdCompression.cpp b/pdal/compression/ZstdCompression.cpp
@@ -47,10 +47,10 @@ class ZstdCompressorImpl
     char m_tmpbuf[CHUNKSIZE];
     BlockCb m_cb;
 
-    ZstdCompressorImpl(BlockCb cb) : m_cb(cb)
+    ZstdCompressorImpl(BlockCb cb, int compressionLevel) : m_cb(cb)
     {
         m_strm = ZSTD_createCStream();
-        ZSTD_initCStream(m_strm, 15);
+        ZSTD_initCStream(m_strm, compressionLevel);
     }
 
     ~ZstdCompressorImpl()
@@ -92,7 +92,12 @@ class ZstdCompressorImpl
 };
 
 ZstdCompressor::ZstdCompressor(BlockCb cb) :
-    m_impl(new ZstdCompressorImpl(cb))
+    ZstdCompressor(cb, 15)
+{}
+
+
+ZstdCompressor::ZstdCompressor(BlockCb cb, int compressionLevel) :
+    m_impl(new ZstdCompressorImpl(cb, compressionLevel))
 {}
 
 

diff --git a/pdal/compression/ZstdCompression.hpp b/pdal/compression/ZstdCompression.hpp
@@ -44,6 +44,7 @@ class ZstdCompressor : public Compressor
 {
 public:
     PDAL_DLL ZstdCompressor(BlockCb cb);
+    PDAL_DLL ZstdCompressor(BlockCb cb, int compressionLevel);
     PDAL_DLL ~ZstdCompressor();
 
     PDAL_DLL void compress(const char *buf, size_t bufsize);