135 lines
3.9 KiB
C
135 lines
3.9 KiB
C
/*********************************************************************
|
|
Blosc - Blocked Shuffling and Compression Library
|
|
|
|
Roundtrip compression/decompression tests.
|
|
|
|
Creation date: 2010-06-07
|
|
Author: Francesc Alted <francesc@blosc.org>
|
|
|
|
See LICENSES/BLOSC.txt for details about copyright and rights to use.
|
|
**********************************************************************/
|
|
|
|
#include "test_common.h"
|
|
#include "../blosc/shuffle.h"
|
|
#include "../blosc/shuffle-generic.h"
|
|
|
|
|
|
/** Perform a compress + decompress round trip. */
|
|
static int test_compress_roundtrip(size_t type_size, size_t num_elements,
|
|
size_t buffer_alignment, int compression_level, int do_shuffle)
|
|
{
|
|
size_t buffer_size = type_size * num_elements;
|
|
int exit_code;
|
|
|
|
/* Allocate memory for the test. */
|
|
void* original = blosc_test_malloc(buffer_alignment, buffer_size);
|
|
void* intermediate = blosc_test_malloc(buffer_alignment, buffer_size + BLOSC_MAX_OVERHEAD);
|
|
void* result = blosc_test_malloc(buffer_alignment, buffer_size);
|
|
|
|
/* Fill the input data buffer with random values. */
|
|
blosc_test_fill_random(original, buffer_size);
|
|
|
|
/* Compress the input data and store it in an intermediate buffer.
|
|
Decompress the data from the intermediate buffer into a result buffer. */
|
|
blosc_compress(compression_level, do_shuffle, type_size, buffer_size,
|
|
original, intermediate, buffer_size + BLOSC_MAX_OVERHEAD);
|
|
blosc_decompress(intermediate, result, buffer_size);
|
|
|
|
/* The round-tripped data matches the original data when the
|
|
result of memcmp is 0. */
|
|
exit_code = memcmp(original, result, buffer_size) ?
|
|
EXIT_FAILURE : EXIT_SUCCESS;
|
|
|
|
/* Free allocated memory. */
|
|
blosc_test_free(original);
|
|
blosc_test_free(intermediate);
|
|
blosc_test_free(result);
|
|
|
|
return exit_code;
|
|
}
|
|
|
|
/** Required number of arguments to this test, including the executable name. */
|
|
#define TEST_ARG_COUNT 7
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int shuffle_enabled;
|
|
uint32_t blosc_thread_count;
|
|
uint32_t type_size;
|
|
uint32_t num_elements;
|
|
uint32_t buffer_align_size;
|
|
uint32_t compression_level;
|
|
int result;
|
|
|
|
/* argv[1]: sizeof(element type)
|
|
argv[2]: number of elements
|
|
argv[3]: buffer alignment
|
|
argv[4]: compression level
|
|
argv[5]: shuffle enabled
|
|
argv[6]: thread count
|
|
*/
|
|
|
|
/* Verify the correct number of command-line args have been specified. */
|
|
if (TEST_ARG_COUNT != argc)
|
|
{
|
|
blosc_test_print_bad_argcount_msg(TEST_ARG_COUNT, argc);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
/* Parse arguments */
|
|
if (!blosc_test_parse_uint32_t(argv[1], &type_size) || (type_size < 1))
|
|
{
|
|
blosc_test_print_bad_arg_msg(1);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!blosc_test_parse_uint32_t(argv[2], &num_elements) || (num_elements < 0))
|
|
{
|
|
blosc_test_print_bad_arg_msg(2);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!blosc_test_parse_uint32_t(argv[3], &buffer_align_size)
|
|
|| (buffer_align_size & (buffer_align_size - 1))
|
|
|| (buffer_align_size < sizeof(void*)))
|
|
{
|
|
blosc_test_print_bad_arg_msg(3);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!blosc_test_parse_uint32_t(argv[4], &compression_level) || (compression_level > 9))
|
|
{
|
|
blosc_test_print_bad_arg_msg(4);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
{
|
|
uint32_t shuffle_enabled_raw;
|
|
if (!blosc_test_parse_uint32_t(argv[5], &shuffle_enabled_raw) || (shuffle_enabled_raw > 1))
|
|
{
|
|
blosc_test_print_bad_arg_msg(5);
|
|
return EXIT_FAILURE;
|
|
}
|
|
shuffle_enabled = shuffle_enabled_raw == 0 ? 0 : 1;
|
|
}
|
|
|
|
if (!blosc_test_parse_uint32_t(argv[6], &blosc_thread_count) || (blosc_thread_count < 1))
|
|
{
|
|
blosc_test_print_bad_arg_msg(6);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
/* Initialize blosc before running tests. */
|
|
blosc_init();
|
|
blosc_set_nthreads(blosc_thread_count);
|
|
|
|
/* Run the test. */
|
|
result = test_compress_roundtrip(type_size, num_elements, buffer_align_size,
|
|
compression_level, shuffle_enabled);
|
|
|
|
/* Cleanup blosc resources. */
|
|
blosc_destroy();
|
|
|
|
return result;
|
|
}
|