Chris@137: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@137: 
Chris@137: #include "Resampler.h"
Chris@137: 
Chris@137: #include <iostream>
Chris@137: 
Chris@137: #include <cmath>
Chris@137: 
Chris@137: #define BOOST_TEST_DYN_LINK
Chris@137: #define BOOST_TEST_MAIN
Chris@137: 
Chris@137: #include <boost/test/unit_test.hpp>
Chris@137: 
Chris@137: BOOST_AUTO_TEST_SUITE(TestResampler)
Chris@137: 
Chris@137: using std::cout;
Chris@137: using std::endl;
Chris@138: using std::vector;
Chris@138: 
Chris@138: void
Chris@138: testResamplerOneShot(int sourceRate,
Chris@138: 		     int targetRate,
Chris@138: 		     int n,
Chris@138: 		     double *in,
Chris@138: 		     int m,
Chris@138: 		     double *expected)
Chris@138: {
Chris@138:     vector<double> resampled = Resampler::resample(sourceRate, targetRate,
Chris@138: 						   in, n);
Chris@138:     BOOST_CHECK_EQUAL(resampled.size(), m);
Chris@138:     for (int i = 0; i < m; ++i) {
Chris@138: 	BOOST_CHECK_SMALL(resampled[i] - expected[i], 1e-8);
Chris@138:     }
Chris@138: }
Chris@137: 
Chris@137: void
Chris@137: testResampler(int sourceRate,
Chris@137: 	      int targetRate,
Chris@137: 	      int n,
Chris@137: 	      double *in,
Chris@137: 	      int m,
Chris@137: 	      double *expected)
Chris@137: {
Chris@139:     // Here we provide the input in chunks (of varying size)
Chris@138: 
Chris@137:     Resampler r(sourceRate, targetRate);
Chris@137:     int latency = r.getLatency();
Chris@137:     std::cerr << "latency = " << latency << std::endl;
Chris@137: 
Chris@137:     int m1 = m + latency;
Chris@137:     int n1 = int((m1 * sourceRate) / targetRate);
Chris@137: 
Chris@137:     double *inPadded = new double[n1];
Chris@137:     double *outPadded = new double[m1];
Chris@137: 
Chris@137:     for (int i = 0; i < n1; ++i) {
Chris@137: 	if (i < n) inPadded[i] = in[i];
Chris@137: 	else inPadded[i] = 0.0;
Chris@137:     }
Chris@137:     
Chris@137:     for (int i = 0; i < m1; ++i) {
Chris@137: 	outPadded[i] = -999.0;
Chris@137:     }
Chris@137: 
Chris@139:     int chunkSize = 1;
Chris@139:     int got = 0;
Chris@139:     int i = 0;
Chris@137: 
Chris@139:     while (true) {
Chris@139: 	std::cerr << "i = " << i << ", n1 = " << n1 << ", chunkSize = " << chunkSize << std::endl;
Chris@139: 	got += r.process(inPadded + i, outPadded + got, chunkSize);
Chris@139: 	i = i + chunkSize;
Chris@139: 	chunkSize = chunkSize + 1;
Chris@139: 	if (i + 1 >= n1) {
Chris@139: 	    break;
Chris@139: 	} else if (i + chunkSize >= n1) {
Chris@139: 	    chunkSize = n1 - i;
Chris@139: 	}
Chris@139:     }
Chris@139: 
Chris@139: //    int got = r.process(inPadded, outPadded, n1);
Chris@139:     std::cerr << i << " in, " << got << " out" << std::endl;
Chris@137: 
Chris@137:     BOOST_CHECK_EQUAL(got, m1);
Chris@139: /*
Chris@137:     std::cerr << "results including latency padding:" << std::endl;
Chris@137:     for (int i = 0; i < m1; ++i) {
Chris@137: 	std::cerr << outPadded[i] << " ";
Chris@137: 	if (i % 6 == 5) std::cerr << "\n";
Chris@137:     }
Chris@137:     std::cerr << "\n";
Chris@139: */
Chris@137:     for (int i = latency; i < m1; ++i) {
Chris@138: 	BOOST_CHECK_SMALL(outPadded[i] - expected[i-latency], 1e-8);
Chris@137:     }
Chris@137:     delete[] outPadded;
Chris@137:     delete[] inPadded;
Chris@137: }
Chris@137: 
Chris@140: BOOST_AUTO_TEST_CASE(sameRateOneShot)
Chris@140: {
Chris@140:     double d[] = { 0, 0.1, -0.3, -0.4, -0.3, 0, 0.5, 0.2, 0.8, -0.1 };
Chris@140:     testResamplerOneShot(4, 4, 10, d, 10, d);
Chris@140: }
Chris@140: 
Chris@137: BOOST_AUTO_TEST_CASE(sameRate)
Chris@137: {
Chris@137:     double d[] = { 0, 0.1, -0.3, -0.4, -0.3, 0, 0.5, 0.2, 0.8, -0.1 };
Chris@137:     testResampler(4, 4, 10, d, 10, d);
Chris@137: }
Chris@137: 
Chris@137: BOOST_AUTO_TEST_SUITE_END()
Chris@137: