diff data/model/test/MockWaveModel.cpp @ 1126:39019ce29178 tony-2.0-integration

Merge through to branch for Tony 2.0
author Chris Cannam
date Thu, 20 Aug 2015 14:54:21 +0100
parents bead35921256
children 1309b66eff53
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/data/model/test/MockWaveModel.cpp	Thu Aug 20 14:54:21 2015 +0100
@@ -0,0 +1,94 @@
+/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
+
+/*
+    Sonic Visualiser
+    An audio file viewer and annotation editor.
+    Centre for Digital Music, Queen Mary, University of London.
+    This file copyright 2006 Chris Cannam.
+    
+    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.  See the file
+    COPYING included with this distribution for more information.
+*/
+
+#include "MockWaveModel.h"
+
+#include <cmath>
+
+using namespace std;
+
+MockWaveModel::MockWaveModel(vector<Sort> sorts, int length, int pad)
+{
+    for (auto sort: sorts) {
+	m_data.push_back(generate(sort, length, pad));
+    }
+}
+
+vector<float>
+MockWaveModel::getData(int channel, sv_frame_t start, sv_frame_t count) const
+{
+    sv_frame_t i = 0;
+
+//    cerr << "MockWaveModel::getData(" << channel << "," << start << "," << count << "): ";
+
+    vector<float> data;
+    
+    while (i < count) {
+	sv_frame_t idx = start + i;
+	if (!in_range_for(m_data[channel], idx)) break;
+	data.push_back(m_data[channel][idx]);
+//	cerr << data[i] << " ";
+	++i;
+    }
+
+//    cerr << endl;
+    
+    return data;
+}
+
+vector<vector<float>>
+MockWaveModel::getMultiChannelData(int fromchannel, int tochannel,
+				   sv_frame_t start, sv_frame_t count) const
+{
+    vector<vector<float>> data(tochannel - fromchannel + 1);
+    
+    for (int c = fromchannel; c <= tochannel; ++c) {
+        data.push_back(getData(c, start, count));
+    }
+
+    return data;
+}
+
+vector<float>
+MockWaveModel::generate(Sort sort, int length, int pad) const
+{
+    vector<float> data;
+
+    for (int i = 0; i < pad; ++i) {
+        data.push_back(0.f);
+    }
+    
+    for (int i = 0; i < length; ++i) {
+
+	double v = 0.0;
+	
+	switch (sort) {
+	case DC: v = 1.0; break;
+	case Sine: v = sin((2.0 * M_PI / 8.0) * i); break;
+	case Cosine: v = cos((2.0 * M_PI / 8.0) * i); break;
+	case Nyquist: v = (i % 2) * 2 - 1; break;
+	case Dirac: v = (i == 0) ? 1.0 : 0.0; break;
+	}
+
+	data.push_back(float(v));
+    }
+
+    for (int i = 0; i < pad; ++i) {
+        data.push_back(0.f);
+    }
+
+    return data;
+}
+