mirror of
https://github.com/aaru-dps/libaaruformat.git
synced 2025-12-16 19:24:40 +00:00
120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
/*
|
|
* This file is part of the Aaru Data Preservation Suite.
|
|
* Copyright (c) 2019-2025 Natalia Portillo.
|
|
*
|
|
* This library is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as
|
|
* published by the Free Software Foundation; either version 2.1 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <climits>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
#include <aaruformat.h>
|
|
|
|
#include "../include/aaruformat/flac.h"
|
|
#include "crc32.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#define EXPECTED_CRC32 0xdfbc99bb
|
|
|
|
static uint8_t *buffer;
|
|
|
|
class flacFixture : public ::testing::Test
|
|
{
|
|
public:
|
|
flacFixture() = default;
|
|
|
|
protected:
|
|
void SetUp() override
|
|
{
|
|
char path[PATH_MAX];
|
|
char filename[PATH_MAX];
|
|
|
|
getcwd(path, PATH_MAX);
|
|
snprintf(filename, PATH_MAX, "%s/data/flac.flac", path);
|
|
|
|
FILE *file = fopen(filename, "rb");
|
|
buffer = static_cast<uint8_t *>(malloc(6534197));
|
|
fread(buffer, 1, 6534197, file);
|
|
fclose(file);
|
|
}
|
|
|
|
void TearDown() override { free(buffer); }
|
|
|
|
~flacFixture() override = default;
|
|
|
|
// shared user data
|
|
};
|
|
|
|
TEST_F(flacFixture, flac)
|
|
{
|
|
auto *outBuf = static_cast<uint8_t *>(malloc(9633792));
|
|
|
|
auto decoded = aaruf_flac_decode_redbook_buffer(outBuf, 9633792, buffer, 6534197);
|
|
|
|
EXPECT_EQ(decoded, 9633792);
|
|
|
|
auto crc = crc32_data(outBuf, 9633792);
|
|
|
|
free(outBuf);
|
|
|
|
EXPECT_EQ(crc, EXPECTED_CRC32);
|
|
}
|
|
|
|
TEST_F(flacFixture, flacCompress)
|
|
{
|
|
size_t original_len = 9633792;
|
|
uint cmp_len = original_len;
|
|
uint decmp_len = original_len;
|
|
char path[PATH_MAX];
|
|
char filename[PATH_MAX * 2];
|
|
|
|
// Allocate buffers
|
|
uint8_t *original = static_cast<uint8_t *>(malloc(original_len));
|
|
uint8_t *cmp_buffer = static_cast<uint8_t *>(malloc(cmp_len));
|
|
uint8_t *decmp_buffer = static_cast<uint8_t *>(malloc(decmp_len));
|
|
|
|
// Read the file
|
|
getcwd(path, PATH_MAX);
|
|
snprintf(filename, PATH_MAX, "%s/data/audio.bin", path);
|
|
|
|
FILE *file = fopen(filename, "rb");
|
|
fread(original, 1, original_len, file);
|
|
fclose(file);
|
|
|
|
// Calculate the CRC
|
|
uint32_t original_crc = crc32_data(original, original_len);
|
|
|
|
// Compress
|
|
size_t newSize = aaruf_flac_encode_redbook_buffer(
|
|
cmp_buffer, cmp_len, original, original_len, 4608, 1, 0, "partial_tukey(0/1.0/1.0)", 12, 0, 1, false, 0, 8,
|
|
"Aaru.Compression.Native.Tests", strlen("Aaru.Compression.Native.Tests"));
|
|
cmp_len = newSize;
|
|
|
|
// Decompress
|
|
newSize = aaruf_flac_decode_redbook_buffer(decmp_buffer, decmp_len, cmp_buffer, cmp_len);
|
|
decmp_len = newSize;
|
|
|
|
EXPECT_EQ(decmp_len, original_len);
|
|
|
|
uint32_t decmp_crc = crc32_data(decmp_buffer, decmp_len);
|
|
|
|
// Free buffers
|
|
free(original);
|
|
free(cmp_buffer);
|
|
free(decmp_buffer);
|
|
|
|
EXPECT_EQ(decmp_crc, original_crc);
|
|
}
|