annotate data/model/test/MockWaveModel.cpp @ 1086:9f4505ac9072

Tidy dense time-value model API a bit; add first simple unit test for FFT model
author Chris Cannam
date Wed, 10 Jun 2015 17:06:02 +0100
parents
children dcf54a6964d0
rev   line source
Chris@1086 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@1086 2
Chris@1086 3 /*
Chris@1086 4 Sonic Visualiser
Chris@1086 5 An audio file viewer and annotation editor.
Chris@1086 6 Centre for Digital Music, Queen Mary, University of London.
Chris@1086 7 This file copyright 2006 Chris Cannam.
Chris@1086 8
Chris@1086 9 This program is free software; you can redistribute it and/or
Chris@1086 10 modify it under the terms of the GNU General Public License as
Chris@1086 11 published by the Free Software Foundation; either version 2 of the
Chris@1086 12 License, or (at your option) any later version. See the file
Chris@1086 13 COPYING included with this distribution for more information.
Chris@1086 14 */
Chris@1086 15
Chris@1086 16 #include "MockWaveModel.h"
Chris@1086 17
Chris@1086 18 using namespace std;
Chris@1086 19
Chris@1086 20 MockWaveModel::MockWaveModel(vector<Sort> sorts, int length)
Chris@1086 21 {
Chris@1086 22 for (auto sort: sorts) {
Chris@1086 23 m_data.push_back(generate(sort, length));
Chris@1086 24 }
Chris@1086 25 }
Chris@1086 26
Chris@1086 27 sv_frame_t
Chris@1086 28 MockWaveModel::getData(int channel, sv_frame_t start, sv_frame_t count,
Chris@1086 29 float *buffer) const
Chris@1086 30 {
Chris@1086 31 sv_frame_t i = 0;
Chris@1086 32
Chris@1086 33 cerr << "MockWaveModel::getData(" << channel << "," << start << "," << count << "): ";
Chris@1086 34
Chris@1086 35 while (i < count) {
Chris@1086 36 sv_frame_t idx = start + i;
Chris@1086 37 if (!in_range_for(m_data[channel], idx)) break;
Chris@1086 38 buffer[i] = m_data[channel][idx];
Chris@1086 39 cerr << buffer[i] << " ";
Chris@1086 40 ++i;
Chris@1086 41 }
Chris@1086 42
Chris@1086 43 cerr << endl;
Chris@1086 44
Chris@1086 45 return i;
Chris@1086 46 }
Chris@1086 47
Chris@1086 48 sv_frame_t
Chris@1086 49 MockWaveModel::getMultiChannelData(int fromchannel, int tochannel,
Chris@1086 50 sv_frame_t start, sv_frame_t count,
Chris@1086 51 float **buffers) const
Chris@1086 52 {
Chris@1086 53 sv_frame_t min = count;
Chris@1086 54
Chris@1086 55 for (int c = fromchannel; c <= tochannel; ++c) {
Chris@1086 56 sv_frame_t n = getData(c, start, count, buffers[c]);
Chris@1086 57 if (n < min) min = n;
Chris@1086 58 }
Chris@1086 59
Chris@1086 60 return min;
Chris@1086 61 }
Chris@1086 62
Chris@1086 63 vector<float>
Chris@1086 64 MockWaveModel::generate(Sort sort, int length) const
Chris@1086 65 {
Chris@1086 66 vector<float> data;
Chris@1086 67
Chris@1086 68 for (int i = 0; i < length; ++i) {
Chris@1086 69
Chris@1086 70 float v = 0.f;
Chris@1086 71
Chris@1086 72 switch (sort) {
Chris@1086 73 case DC: v = 1.f; break;
Chris@1086 74 case Sine: v = (float)sin((2.0 * M_PI / 8.0) * i); break;
Chris@1086 75 case Cosine: v = (float)cos((2.0 * M_PI / 8.0) * i); break;
Chris@1086 76 case Nyquist: v = (i % 2) * 2 - 1; break;
Chris@1086 77 case Dirac: v = (i == 0) ? 1.f : 0.f; break;
Chris@1086 78 }
Chris@1086 79
Chris@1086 80 data.push_back(v);
Chris@1086 81 }
Chris@1086 82
Chris@1086 83 return data;
Chris@1086 84 }
Chris@1086 85