test_pcm_util.hxx 2.42 KB
Newer Older
1
/*
Max Kellermann's avatar
Max Kellermann committed
2
 * Copyright 2003-2019 The Music Player Daemon Project
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
 * 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.
 */

20 21
#include "util/ConstBuffer.hxx"

22 23
#include <gtest/gtest.h>

24
#include <array>
25
#include <random>
26 27 28 29 30

#include <stddef.h>
#include <stdint.h>

template<typename T>
31 32 33 34 35 36 37 38
struct RandomInt {
	typedef std::minstd_rand Engine;
	Engine engine;

	static_assert(sizeof(T) <= sizeof(Engine::result_type),
		      "RNG result type too small");

	T operator()() {
39
		return T(engine());
40 41 42
	}
};

43 44 45
struct RandomInt24 : RandomInt<int32_t> {
	int32_t operator()() {
		auto t = RandomInt::operator()();
46 47 48 49 50 51 52
		t &= 0xffffff;
		if (t & 0x800000)
			t |= 0xff000000;
		return t;
	}
};

53 54 55 56 57 58 59 60
struct RandomFloat {
	std::mt19937 gen;
	std::uniform_real_distribution<float> dis;

	RandomFloat():gen(std::random_device()()), dis(-1.0, 1.0) {}

	float operator()() {
		return dis(gen);
61 62 63 64 65 66 67
	}
};

template<typename T, size_t N>
class TestDataBuffer : std::array<T, N> {
public:
	using typename std::array<T, N>::const_pointer;
68
	using std::array<T, N>::size;
69 70 71 72
	using std::array<T, N>::begin;
	using std::array<T, N>::end;
	using std::array<T, N>::operator[];

73
	template<typename G=RandomInt<T>>
74 75 76 77 78 79 80 81 82
	TestDataBuffer(G g=G()):std::array<T, N>() {
		for (auto &i : *this)
			i = g();

	}

	operator typename std::array<T, N>::const_pointer() const {
		return begin();
	}
83 84 85 86 87 88 89 90

	operator ConstBuffer<T>() const {
		return { begin(), size() };
	}

	operator ConstBuffer<void>() const {
		return { begin(), size() * sizeof(T) };
	}
91
};
92 93 94 95 96

template<typename T>
bool
AssertEqualWithTolerance(const T &a, const T &b, unsigned tolerance)
{
97
	EXPECT_EQ(a.size(), b.size());
98 99 100 101

	for (unsigned i = 0; i < a.size(); ++i) {
		int64_t x = a[i], y = b[i];

102 103
		EXPECT_GE(x, y - int64_t(tolerance));
		EXPECT_LE(x, y + int64_t(tolerance));
104 105 106 107
	}

	return true;
}