From 2e511246fa99ad6a9f9cf1bbb8cf35a51382cd19 Mon Sep 17 00:00:00 2001 From: Lioncash Date: Sat, 25 Jul 2020 03:17:01 -0400 Subject: [PATCH] lz4_compression: Make use of std::span in interfaces Allows compressing the data and size parameters into one. --- src/common/lz4_compression.cpp | 21 ++++++++++----------- src/common/lz4_compression.h | 10 ++++------ 2 files changed, 14 insertions(+), 17 deletions(-) diff --git a/src/common/lz4_compression.cpp b/src/common/lz4_compression.cpp index ade6759bb..8e2e4094b 100644 --- a/src/common/lz4_compression.cpp +++ b/src/common/lz4_compression.cpp @@ -10,14 +10,14 @@ namespace Common::Compression { -std::vector CompressDataLZ4(const u8* source, std::size_t source_size) { - ASSERT_MSG(source_size <= LZ4_MAX_INPUT_SIZE, "Source size exceeds LZ4 maximum input size"); +std::vector CompressDataLZ4(std::span source) { + ASSERT_MSG(source.size() <= LZ4_MAX_INPUT_SIZE, "Source size exceeds LZ4 maximum input size"); - const auto source_size_int = static_cast(source_size); + const auto source_size_int = static_cast(source.size()); const int max_compressed_size = LZ4_compressBound(source_size_int); std::vector compressed(max_compressed_size); - const int compressed_size = LZ4_compress_default(reinterpret_cast(source), + const int compressed_size = LZ4_compress_default(reinterpret_cast(source.data()), reinterpret_cast(compressed.data()), source_size_int, max_compressed_size); @@ -31,18 +31,17 @@ std::vector CompressDataLZ4(const u8* source, std::size_t source_size) { return compressed; } -std::vector CompressDataLZ4HC(const u8* source, std::size_t source_size, - s32 compression_level) { - ASSERT_MSG(source_size <= LZ4_MAX_INPUT_SIZE, "Source size exceeds LZ4 maximum input size"); +std::vector CompressDataLZ4HC(std::span source, s32 compression_level) { + ASSERT_MSG(source.size() <= LZ4_MAX_INPUT_SIZE, "Source size exceeds LZ4 maximum input size"); compression_level = std::clamp(compression_level, LZ4HC_CLEVEL_MIN, LZ4HC_CLEVEL_MAX); - const auto source_size_int = static_cast(source_size); + const auto source_size_int = static_cast(source.size()); const int max_compressed_size = LZ4_compressBound(source_size_int); std::vector compressed(max_compressed_size); const int compressed_size = LZ4_compress_HC( - reinterpret_cast(source), reinterpret_cast(compressed.data()), + reinterpret_cast(source.data()), reinterpret_cast(compressed.data()), source_size_int, max_compressed_size, compression_level); if (compressed_size <= 0) { @@ -55,8 +54,8 @@ std::vector CompressDataLZ4HC(const u8* source, std::size_t source_size, return compressed; } -std::vector CompressDataLZ4HCMax(const u8* source, std::size_t source_size) { - return CompressDataLZ4HC(source, source_size, LZ4HC_CLEVEL_MAX); +std::vector CompressDataLZ4HCMax(std::span source) { + return CompressDataLZ4HC(source, LZ4HC_CLEVEL_MAX); } std::vector DecompressDataLZ4(const std::vector& compressed, diff --git a/src/common/lz4_compression.h b/src/common/lz4_compression.h index 4c16f6e03..173f9b9ad 100644 --- a/src/common/lz4_compression.h +++ b/src/common/lz4_compression.h @@ -4,6 +4,7 @@ #pragma once +#include #include #include "common/common_types.h" @@ -14,11 +15,10 @@ namespace Common::Compression { * Compresses a source memory region with LZ4 and returns the compressed data in a vector. * * @param source the uncompressed source memory region. - * @param source_size the size in bytes of the uncompressed source memory region. * * @return the compressed data. */ -std::vector CompressDataLZ4(const u8* source, std::size_t source_size); +std::vector CompressDataLZ4(std::span source); /** * Utilizes the LZ4 subalgorithm LZ4HC with the specified compression level. Higher compression @@ -27,22 +27,20 @@ std::vector CompressDataLZ4(const u8* source, std::size_t source_size); * also be decompressed with the default LZ4 decompression. * * @param source the uncompressed source memory region. - * @param source_size the size in bytes of the uncompressed source memory region. * @param compression_level the used compression level. Should be between 3 and 12. * * @return the compressed data. */ -std::vector CompressDataLZ4HC(const u8* source, std::size_t source_size, s32 compression_level); +std::vector CompressDataLZ4HC(std::span source, s32 compression_level); /** * Utilizes the LZ4 subalgorithm LZ4HC with the highest possible compression level. * * @param source the uncompressed source memory region. - * @param source_size the size in bytes of the uncompressed source memory region. * * @return the compressed data. */ -std::vector CompressDataLZ4HCMax(const u8* source, std::size_t source_size); +std::vector CompressDataLZ4HCMax(std::span source); /** * Decompresses a source memory region with LZ4 and returns the uncompressed data in a vector.