diff options
Diffstat (limited to 'src/common/data_compression.cpp')
| -rw-r--r-- | src/common/data_compression.cpp | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/src/common/data_compression.cpp b/src/common/data_compression.cpp index 624b8ac47..fdf4907d1 100644 --- a/src/common/data_compression.cpp +++ b/src/common/data_compression.cpp | |||
| @@ -4,14 +4,14 @@ | |||
| 4 | 4 | ||
| 5 | #pragma once | 5 | #pragma once |
| 6 | 6 | ||
| 7 | #include <lz4.h> | 7 | #include <lz4hc.h> |
| 8 | 8 | ||
| 9 | #include "data_compression.h" | 9 | #include "data_compression.h" |
| 10 | 10 | ||
| 11 | namespace Compression { | 11 | namespace Compression { |
| 12 | 12 | ||
| 13 | 13 | std::vector<u8> CompressDataLZ4(const u8* source, std::size_t source_size, | |
| 14 | std::vector<u8> CompressDataLZ4(const u8* source, std::size_t source_size) { | 14 | bool use_LZ4_high_compression) { |
| 15 | if (source_size > LZ4_MAX_INPUT_SIZE) { | 15 | if (source_size > LZ4_MAX_INPUT_SIZE) { |
| 16 | // Source size exceeds LZ4 maximum input size | 16 | // Source size exceeds LZ4 maximum input size |
| 17 | return {}; | 17 | return {}; |
| @@ -19,9 +19,18 @@ std::vector<u8> CompressDataLZ4(const u8* source, std::size_t source_size) { | |||
| 19 | const auto source_size_int = static_cast<int>(source_size); | 19 | const auto source_size_int = static_cast<int>(source_size); |
| 20 | const int max_compressed_size = LZ4_compressBound(source_size_int); | 20 | const int max_compressed_size = LZ4_compressBound(source_size_int); |
| 21 | std::vector<u8> compressed(max_compressed_size); | 21 | std::vector<u8> compressed(max_compressed_size); |
| 22 | const int compressed_size = LZ4_compress_default(reinterpret_cast<const char*>(source), | 22 | int compressed_size = 0; |
| 23 | reinterpret_cast<char*>(compressed.data()), | 23 | |
| 24 | source_size_int, max_compressed_size); | 24 | if (use_LZ4_high_compression) { |
| 25 | compressed_size = LZ4_compress_HC(reinterpret_cast<const char*>(source), | ||
| 26 | reinterpret_cast<char*>(compressed.data()), | ||
| 27 | source_size_int, max_compressed_size, LZ4HC_CLEVEL_MAX); | ||
| 28 | } else { | ||
| 29 | compressed_size = LZ4_compress_default(reinterpret_cast<const char*>(source), | ||
| 30 | reinterpret_cast<char*>(compressed.data()), | ||
| 31 | source_size_int, max_compressed_size); | ||
| 32 | } | ||
| 33 | |||
| 25 | if (compressed_size <= 0) { | 34 | if (compressed_size <= 0) { |
| 26 | // Compression failed | 35 | // Compression failed |
| 27 | return {}; | 36 | return {}; |
| @@ -30,7 +39,8 @@ std::vector<u8> CompressDataLZ4(const u8* source, std::size_t source_size) { | |||
| 30 | return compressed; | 39 | return compressed; |
| 31 | } | 40 | } |
| 32 | 41 | ||
| 33 | std::vector<u8> DecompressDataLZ4(const std::vector<u8>& compressed, std::size_t uncompressed_size) { | 42 | std::vector<u8> DecompressDataLZ4(const std::vector<u8>& compressed, |
| 43 | std::size_t uncompressed_size) { | ||
| 34 | std::vector<u8> uncompressed(uncompressed_size); | 44 | std::vector<u8> uncompressed(uncompressed_size); |
| 35 | const int size_check = LZ4_decompress_safe(reinterpret_cast<const char*>(compressed.data()), | 45 | const int size_check = LZ4_decompress_safe(reinterpret_cast<const char*>(compressed.data()), |
| 36 | reinterpret_cast<char*>(uncompressed.data()), | 46 | reinterpret_cast<char*>(uncompressed.data()), |