161 lines
4.0 KiB
C++
161 lines
4.0 KiB
C++
/*
|
|
* Copyright 2003-2022 The Music Player Daemon Project
|
|
* http://www.musicpd.org
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#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();
|
|
}
|