145 lines
3.3 KiB
C++
145 lines
3.3 KiB
C++
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
// Copyright The Music Player Daemon Project
|
|
|
|
#include "pcm/Volume.hxx"
|
|
#include "pcm/Traits.hxx"
|
|
#include "util/SpanCast.hxx"
|
|
#include "test_pcm_util.hxx"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <algorithm>
|
|
|
|
#include <string.h>
|
|
|
|
template<SampleFormat F, class Traits=SampleTraits<F>,
|
|
typename G=RandomInt<typename Traits::value_type>>
|
|
static void
|
|
TestVolume(G g=G())
|
|
{
|
|
using value_type = typename Traits::value_type;
|
|
|
|
PcmVolume pv;
|
|
EXPECT_EQ(pv.Open(F, false), F);
|
|
|
|
constexpr size_t N = 509;
|
|
static value_type zero[N];
|
|
const auto _src = TestDataBuffer<value_type, N>(g);
|
|
const std::span<const std::byte> src = _src;
|
|
|
|
pv.SetVolume(0);
|
|
auto dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
EXPECT_EQ(0, memcmp(dest.data(), zero, sizeof(zero)));
|
|
|
|
pv.SetVolume(PCM_VOLUME_1);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
EXPECT_EQ(0, memcmp(dest.data(), src.data(), src.size()));
|
|
|
|
pv.SetVolume(PCM_VOLUME_1 / 2);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
|
|
const auto _dest = FromBytesStrict<const value_type>(dest);
|
|
for (unsigned i = 0; i < N; ++i) {
|
|
const auto expected = (_src[i] + 1) / 2;
|
|
EXPECT_GE(_dest[i], expected - 4);
|
|
EXPECT_LE(_dest[i], expected + 4);
|
|
}
|
|
|
|
pv.Close();
|
|
}
|
|
|
|
TEST(PcmTest, Volume8)
|
|
{
|
|
TestVolume<SampleFormat::S8>();
|
|
}
|
|
|
|
TEST(PcmTest, Volume16)
|
|
{
|
|
TestVolume<SampleFormat::S16>();
|
|
}
|
|
|
|
TEST(PcmTest, Volume16to32)
|
|
{
|
|
constexpr SampleFormat F = SampleFormat::S16;
|
|
using value_type = int16_t;
|
|
RandomInt<value_type> g;
|
|
|
|
PcmVolume pv;
|
|
EXPECT_EQ(pv.Open(F, true), SampleFormat::S24_P32);
|
|
|
|
constexpr size_t N = 509;
|
|
static value_type zero[N];
|
|
const auto _src = TestDataBuffer<value_type, N>(g);
|
|
const std::span<const std::byte> src = _src;
|
|
|
|
pv.SetVolume(0);
|
|
auto dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size() * 2, dest.size());
|
|
EXPECT_EQ(0, memcmp(dest.data(), zero, sizeof(zero)));
|
|
|
|
pv.SetVolume(PCM_VOLUME_1);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size() * 2, dest.size());
|
|
auto s = FromBytesStrict<const int16_t>(src);
|
|
auto d = FromBytesStrict<const int32_t>(dest);
|
|
for (size_t i = 0; i < N; ++i)
|
|
EXPECT_EQ(d[i], s[i] << 8);
|
|
|
|
pv.SetVolume(PCM_VOLUME_1 / 2);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size() * 2, dest.size());
|
|
|
|
s = FromBytesStrict<const int16_t>(src);
|
|
d = FromBytesStrict<const int32_t>(dest);
|
|
for (unsigned i = 0; i < N; ++i) {
|
|
const int32_t expected = (s[i] << 8) / 2;
|
|
EXPECT_EQ(d[i], expected);
|
|
}
|
|
|
|
pv.Close();
|
|
}
|
|
|
|
TEST(PcmTest, Volume24)
|
|
{
|
|
TestVolume<SampleFormat::S24_P32>(RandomInt24());
|
|
}
|
|
|
|
TEST(PcmTest, Volume32)
|
|
{
|
|
TestVolume<SampleFormat::S32>();
|
|
}
|
|
|
|
TEST(PcmTest, VolumeFloat)
|
|
{
|
|
PcmVolume pv;
|
|
pv.Open(SampleFormat::FLOAT, false);
|
|
|
|
constexpr size_t N = 509;
|
|
static float zero[N];
|
|
const auto _src = TestDataBuffer<float, N>(RandomFloat());
|
|
const std::span<const std::byte> src = _src;
|
|
|
|
pv.SetVolume(0);
|
|
auto dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
EXPECT_EQ(0, memcmp(dest.data(), zero, sizeof(zero)));
|
|
|
|
pv.SetVolume(PCM_VOLUME_1);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
EXPECT_EQ(0, memcmp(dest.data(), src.data(), src.size()));
|
|
|
|
pv.SetVolume(PCM_VOLUME_1 / 2);
|
|
dest = pv.Apply(src);
|
|
EXPECT_EQ(src.size(), dest.size());
|
|
|
|
const auto _dest = FromBytesStrict<const float>(dest);
|
|
for (unsigned i = 0; i < N; ++i)
|
|
EXPECT_NEAR((double)_src[i] / 2., (double)_dest[i], 1.);
|
|
|
|
pv.Close();
|
|
}
|