13
0
livetrax/libs/audiographer/tests/utils.h

138 lines
3.0 KiB
C
Raw Normal View History

#ifndef AUDIOGRAPHER_TESTS_UTILS_H
#define AUDIOGRAPHER_TESTS_UTILS_H
// Includes we want almost always
#include <memory>
#include <cppunit/extensions/HelperMacros.h>
// includes used in this file
#include "audiographer/sink.h"
#include "audiographer/exception.h"
#include <vector>
#include <list>
#include <cstring>
#include <cstdlib>
#include <ctime>
using AudioGrapher::samplecnt_t;
struct TestUtils
{
template<typename T>
static bool array_equals (T const * a, T const * b, samplecnt_t samples)
{
for (samplecnt_t i = 0; i < samples; ++i) {
if (a[i] != b[i]) {
return false;
}
}
return true;
}
template<typename T>
static bool array_filled (T const * array, samplecnt_t samples)
{
for (samplecnt_t i = 0; i < samples; ++i) {
if (array[i] == static_cast<T> (0.0)) {
return false;
}
}
return true;
}
/// Generate random data, all samples guaranteed not to be 0.0, 1.0 or -1.0
static float * init_random_data (samplecnt_t samples, float range = 1.0)
{
unsigned int const granularity = 4096;
float * data = new float[samples];
srand (std::time (NULL));
2015-10-05 10:17:49 -04:00
for (samplecnt_t i = 0; i < samples; ++i) {
do {
int biased_int = (rand() % granularity) - (granularity / 2);
data[i] = (range * biased_int) / granularity;
} while (data[i] == 0.0 || data[i] == 1.0 || data[i] == -1.0);
}
return data;
}
};
template<typename T>
class VectorSink : public AudioGrapher::Sink<T>
{
public:
virtual void process (AudioGrapher::ProcessContext<T> const & c)
{
data.resize (c.samples());
memcpy (&data[0], c.data(), c.samples() * sizeof(T));
}
void process (AudioGrapher::ProcessContext<T> & c) { AudioGrapher::Sink<T>::process (c); }
using AudioGrapher::Sink<T>::process;
std::vector<T> const & get_data() const { return data; }
T const * get_array() const { return &data[0]; }
void reset() { data.clear(); }
protected:
std::vector<T> data;
};
template<typename T>
class AppendingVectorSink : public VectorSink<T>
{
public:
AppendingVectorSink() { reset (); }
void process (AudioGrapher::ProcessContext<T> const & c)
{
std::vector<T> & data (VectorSink<T>::data);
data.resize (total_samples + c.samples());
memcpy (&data[total_samples], c.data(), c.samples() * sizeof(T));
total_samples += c.samples();
}
using AudioGrapher::Sink<T>::process;
void reset ()
{
total_samples = 0;
VectorSink<T>::reset();
}
private:
samplecnt_t total_samples;
};
template<typename T>
class ThrowingSink : public AudioGrapher::Sink<T>
{
public:
void process (AudioGrapher::ProcessContext<T> const &)
{
throw AudioGrapher::Exception(*this, "ThrowingSink threw!");
}
using AudioGrapher::Sink<T>::process;
};
template<typename T>
class ProcessContextGrabber : public AudioGrapher::Sink<T>
{
public:
void process (AudioGrapher::ProcessContext<T> const & c)
{
contexts.push_back (c);
}
using AudioGrapher::Sink<T>::process;
2015-10-05 10:17:49 -04:00
typedef std::list<AudioGrapher::ProcessContext<T> > ContextList;
ContextList contexts;
2015-10-05 10:17:49 -04:00
};
#endif // AUDIOGRAPHER_TESTS_UTILS_H