95 lines
3.4 KiB
C++
95 lines
3.4 KiB
C++
#include <gtest/gtest.h>
|
|
#include <stdlib.h>
|
|
#include <tcompression.h>
|
|
#include <random>
|
|
|
|
namespace {} // namespace
|
|
|
|
TEST(utilTest, decompress_test) {
|
|
int64_t tsList[10] = {1700000000, 1700000100, 1700000200, 1700000300, 1700000400,
|
|
1700000500, 1700000600, 1700000700, 1700000800, 1700000900};
|
|
|
|
char* pOutput[10 * sizeof(int64_t)] = {0};
|
|
int32_t len = tsCompressTimestamp(tsList, sizeof(tsList), sizeof(tsList) / sizeof(tsList[0]), pOutput, 10, ONE_STAGE_COMP, NULL, 0);
|
|
|
|
char* decompOutput[10 * 8] = {0};
|
|
tsDecompressTimestamp(pOutput, len, 10, decompOutput, sizeof(int64_t)*10, ONE_STAGE_COMP, NULL, 0);
|
|
|
|
for(int32_t i = 0; i < 10; ++i) {
|
|
std::cout<< ((int64_t*)decompOutput)[i] << std::endl;
|
|
}
|
|
|
|
memset(decompOutput, 0, 10*8);
|
|
tsDecompressTimestampAvx512(reinterpret_cast<const char* const>(pOutput), 10,
|
|
reinterpret_cast<char* const>(decompOutput), false);
|
|
|
|
for(int32_t i = 0; i < 10; ++i) {
|
|
std::cout<<((int64_t*)decompOutput)[i] << std::endl;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
int64_t tsList1[7] = {1700000000, 1700000000, 1700000000, 1700000000, 1700000000, 1700000000, 1700000900};
|
|
int32_t len1 = tsCompressTimestamp(tsList1, sizeof(tsList1), sizeof(tsList1) / sizeof(tsList1[0]), pOutput, 7, ONE_STAGE_COMP, NULL, 0);
|
|
|
|
memset(decompOutput, 0, 10*8);
|
|
tsDecompressTimestampAvx512(reinterpret_cast<const char* const>(pOutput), 7,
|
|
reinterpret_cast<char* const>(decompOutput), false);
|
|
|
|
for(int32_t i = 0; i < 7; ++i) {
|
|
std::cout<<((int64_t*)decompOutput)[i] << std::endl;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
int64_t tsList2[1] = {1700000000};
|
|
int32_t len2 = tsCompressTimestamp(tsList2, sizeof(tsList2), sizeof(tsList2) / sizeof(tsList2[0]), pOutput, 1, ONE_STAGE_COMP, NULL, 0);
|
|
|
|
memset(decompOutput, 0, 10*8);
|
|
tsDecompressTimestampAvx512(reinterpret_cast<const char* const>(pOutput), 1,
|
|
reinterpret_cast<char* const>(decompOutput), false);
|
|
|
|
for(int32_t i = 0; i < 1; ++i) {
|
|
std::cout<<((int64_t*)decompOutput)[i] << std::endl;
|
|
}
|
|
}
|
|
|
|
TEST(utilTest, decompress_perf_test) {
|
|
int32_t num = 10000;
|
|
|
|
int64_t* pList = static_cast<int64_t*>(taosMemoryCalloc(num, sizeof(int64_t)));
|
|
int64_t iniVal = 1700000000;
|
|
|
|
uint32_t v = 100;
|
|
|
|
for(int32_t i = 0; i < num; ++i) {
|
|
iniVal += taosRandR(&v)%10;
|
|
pList[i] = iniVal;
|
|
}
|
|
|
|
char* px = static_cast<char*>(taosMemoryMalloc(num * sizeof(int64_t)));
|
|
int32_t len = tsCompressTimestamp(pList, num * sizeof(int64_t), num, px, num, ONE_STAGE_COMP, NULL, 0);
|
|
|
|
char* pOutput = static_cast<char*>(taosMemoryMalloc(num * sizeof(int64_t)));
|
|
|
|
int64_t st = taosGetTimestampUs();
|
|
for(int32_t k = 0; k < 10000; ++k) {
|
|
tsDecompressTimestamp(px, len, num, pOutput, sizeof(int64_t) * num, ONE_STAGE_COMP, NULL, 0);
|
|
}
|
|
|
|
int64_t el1 = taosGetTimestampUs() - st;
|
|
std::cout << "soft decompress elapsed time:" << el1 << " us" << std::endl;
|
|
|
|
memset(pOutput, 0, num * sizeof(int64_t));
|
|
st = taosGetTimestampUs();
|
|
for(int32_t k = 0; k < 10000; ++k) {
|
|
tsDecompressTimestampAvx512(px, num, pOutput, false);
|
|
}
|
|
|
|
int64_t el2 = taosGetTimestampUs() - st;
|
|
std::cout << "SIMD decompress elapsed time:" << el2 << " us" << std::endl;
|
|
|
|
taosMemoryFree(pList);
|
|
taosMemoryFree(pOutput);
|
|
taosMemoryFree(px);
|
|
}
|
|
|