Chris@112: 
Chris@112: #include "dsp/phasevocoder/PhaseVocoder.h"
Chris@112: 
Chris@112: #include "base/Window.h"
Chris@112: 
Chris@112: #define BOOST_TEST_DYN_LINK
Chris@112: #define BOOST_TEST_MAIN
Chris@112: 
Chris@112: #include <boost/test/unit_test.hpp>
Chris@112: 
Chris@112: BOOST_AUTO_TEST_SUITE(TestFFT)
Chris@112: 
Chris@112: #define COMPARE_CONST(a, n) \
Chris@112:     for (int cmp_i = 0; cmp_i < (int)(sizeof(a)/sizeof(a[0])); ++cmp_i) { \
Chris@112:         BOOST_CHECK_SMALL(a[cmp_i] - n, 1e-14);				\
Chris@112:     }
Chris@112: 
Chris@112: #define COMPARE_ARRAY(a, b)						\
Chris@112:     for (int cmp_i = 0; cmp_i < (int)(sizeof(a)/sizeof(a[0])); ++cmp_i) { \
Chris@112:         BOOST_CHECK_SMALL(a[cmp_i] - b[cmp_i], 1e-14);			\
Chris@112:     }
Chris@112: 
Chris@112: #define COMPARE_ARRAY_EXACT(a, b)						\
Chris@112:     for (int cmp_i = 0; cmp_i < (int)(sizeof(a)/sizeof(a[0])); ++cmp_i) { \
Chris@112:         BOOST_CHECK_EQUAL(a[cmp_i], b[cmp_i]);			\
Chris@112:     }
Chris@112: 
Chris@112: BOOST_AUTO_TEST_CASE(fullcycle)
Chris@112: {
Chris@112:     // Cosine with one cycle exactly equal to pvoc hopsize. We aren't
Chris@112:     // windowing the input frame because (for once) it actually *is*
Chris@112:     // just a short part of a continuous infinite sinusoid.
Chris@112: 
Chris@112:     double frame[] = { 1, 0, -1, 0, 1, 0, -1, 0 };
Chris@112: 
Chris@112:     PhaseVocoder pvoc(8);
Chris@112: 
Chris@112:     // Make these arrays one element too long at each end, so as to
Chris@112:     // test for overruns. For frame size 8, we expect 8/2+1 = 5
Chris@112:     // mag/phase pairs.
Chris@112:     double mag[]   = { 999, 999, 999, 999, 999, 999, 999 };
Chris@112:     double phase[] = { 999, 999, 999, 999, 999, 999, 999 };
Chris@112: 
Chris@112:     pvoc.process(frame, mag + 1, phase + 1);
Chris@112: 
Chris@112:     double magExpected0[] = { 999, 0, 0, 4, 0, 0, 999 };
Chris@112:     COMPARE_ARRAY_EXACT(mag, magExpected0);
Chris@112: 
Chris@112:     double phaseExpected0[] = { 999, 0, 0, 0, 0, 0, 999 };
Chris@112:     COMPARE_ARRAY_EXACT(phase, phaseExpected0);
Chris@112: 
Chris@112:     pvoc.process(frame, mag + 1, phase + 1);
Chris@112: 
Chris@112:     double magExpected1[] = { 999, 0, 0, 4, 0, 0, 999 };
Chris@112:     COMPARE_ARRAY_EXACT(mag, magExpected1);
Chris@112: 
Chris@112:     double phaseExpected1[] = { 999, 0, 0, 2 * M_PI, 0, 0, 999 };
Chris@112:     COMPARE_ARRAY(phase, phaseExpected1);
Chris@112: }
Chris@112: 
Chris@112: BOOST_AUTO_TEST_SUITE_END()
Chris@112: