annotate data/model/AggregateWaveModel.cpp @ 1575:054bbf17ac17 spectrogramparam

Slightly simplify median-window size calculation - but this is not the best way to do this for spectrum views, need to reconsider
author Chris Cannam
date Mon, 12 Nov 2018 14:48:15 +0000
parents 7b4d56b01440
children f97d64b8674f 901f37d32060
rev   line source
Chris@297 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@297 2
Chris@297 3 /*
Chris@297 4 Sonic Visualiser
Chris@297 5 An audio file viewer and annotation editor.
Chris@297 6 Centre for Digital Music, Queen Mary, University of London.
Chris@297 7 This file copyright 2007 QMUL.
Chris@297 8
Chris@297 9 This program is free software; you can redistribute it and/or
Chris@297 10 modify it under the terms of the GNU General Public License as
Chris@297 11 published by the Free Software Foundation; either version 2 of the
Chris@297 12 License, or (at your option) any later version. See the file
Chris@297 13 COPYING included with this distribution for more information.
Chris@297 14 */
Chris@297 15
Chris@297 16 #include "AggregateWaveModel.h"
Chris@297 17
Chris@297 18 #include <iostream>
Chris@297 19
Chris@314 20 #include <QTextStream>
Chris@314 21
Chris@1096 22 using namespace std;
Chris@1096 23
Chris@1556 24 //#define DEBUG_AGGREGATE_WAVE_FILE_MODEL 1
Chris@1556 25
Chris@297 26 PowerOfSqrtTwoZoomConstraint
Chris@297 27 AggregateWaveModel::m_zoomConstraint;
Chris@297 28
Chris@297 29 AggregateWaveModel::AggregateWaveModel(ChannelSpecList channelSpecs) :
Chris@1389 30 m_components(channelSpecs),
Chris@1389 31 m_invalidated(false)
Chris@297 32 {
Chris@297 33 for (ChannelSpecList::const_iterator i = channelSpecs.begin();
Chris@297 34 i != channelSpecs.end(); ++i) {
Chris@1389 35
Chris@1389 36 connect(i->model, SIGNAL(aboutToBeDeleted()),
Chris@1389 37 this, SLOT(componentModelAboutToBeDeleted()));
Chris@1389 38
Chris@297 39 if (i->model->getSampleRate() !=
Chris@297 40 channelSpecs.begin()->model->getSampleRate()) {
Chris@690 41 SVDEBUG << "AggregateWaveModel::AggregateWaveModel: WARNING: Component models do not all have the same sample rate" << endl;
Chris@297 42 break;
Chris@297 43 }
Chris@297 44 }
Chris@297 45 }
Chris@297 46
Chris@297 47 AggregateWaveModel::~AggregateWaveModel()
Chris@297 48 {
Chris@297 49 }
Chris@297 50
Chris@1389 51 void
Chris@1389 52 AggregateWaveModel::componentModelAboutToBeDeleted()
Chris@1389 53 {
Chris@1389 54 SVDEBUG << "AggregateWaveModel::componentModelAboutToBeDeleted: invalidating"
Chris@1389 55 << endl;
Chris@1389 56 m_components.clear();
Chris@1389 57 m_invalidated = true;
Chris@1390 58 emit modelInvalidated();
Chris@1389 59 }
Chris@1389 60
Chris@297 61 bool
Chris@297 62 AggregateWaveModel::isOK() const
Chris@297 63 {
Chris@1389 64 if (m_invalidated) {
Chris@1389 65 return false;
Chris@1389 66 }
Chris@297 67 for (ChannelSpecList::const_iterator i = m_components.begin();
Chris@297 68 i != m_components.end(); ++i) {
Chris@1389 69 if (!i->model->isOK()) {
Chris@1389 70 return false;
Chris@1389 71 }
Chris@297 72 }
Chris@297 73 return true;
Chris@297 74 }
Chris@297 75
Chris@297 76 bool
Chris@297 77 AggregateWaveModel::isReady(int *completion) const
Chris@297 78 {
Chris@297 79 if (completion) *completion = 100;
Chris@1389 80
Chris@297 81 bool ready = true;
Chris@297 82 for (ChannelSpecList::const_iterator i = m_components.begin();
Chris@297 83 i != m_components.end(); ++i) {
Chris@297 84 int completionHere = 100;
Chris@1556 85 if (!i->model->isReady(&completionHere)) {
Chris@1556 86 ready = false;
Chris@1556 87 }
Chris@297 88 if (completion && completionHere < *completion) {
Chris@297 89 *completion = completionHere;
Chris@297 90 }
Chris@297 91 }
Chris@1556 92
Chris@1556 93 #ifdef DEBUG_AGGREGATE_WAVE_FILE_MODEL
Chris@1556 94 SVDEBUG << "AggregateWaveModel(" << objectName()
Chris@1556 95 << ")::isReady: returning " << ready << endl;
Chris@1556 96 #endif
Chris@1556 97
Chris@297 98 return ready;
Chris@297 99 }
Chris@297 100
Chris@1038 101 sv_frame_t
Chris@297 102 AggregateWaveModel::getFrameCount() const
Chris@297 103 {
Chris@1038 104 sv_frame_t count = 0;
Chris@297 105 for (ChannelSpecList::const_iterator i = m_components.begin();
Chris@297 106 i != m_components.end(); ++i) {
Chris@1389 107 sv_frame_t thisCount =
Chris@1389 108 i->model->getEndFrame() - i->model->getStartFrame();
Chris@297 109 if (thisCount > count) count = thisCount;
Chris@297 110 }
Chris@297 111 return count;
Chris@297 112 }
Chris@297 113
Chris@929 114 int
Chris@297 115 AggregateWaveModel::getChannelCount() const
Chris@297 116 {
Chris@1038 117 return int(m_components.size());
Chris@297 118 }
Chris@297 119
Chris@1040 120 sv_samplerate_t
Chris@297 121 AggregateWaveModel::getSampleRate() const
Chris@297 122 {
Chris@297 123 if (m_components.empty()) return 0;
Chris@297 124 return m_components.begin()->model->getSampleRate();
Chris@297 125 }
Chris@297 126
Chris@1326 127 floatvec_t
Chris@1096 128 AggregateWaveModel::getData(int channel, sv_frame_t start, sv_frame_t count) const
Chris@297 129 {
Chris@297 130 int ch0 = channel, ch1 = channel;
Chris@297 131 if (channel == -1) {
Chris@297 132 ch0 = 0;
Chris@297 133 ch1 = getChannelCount()-1;
Chris@297 134 }
Chris@297 135
Chris@1326 136 floatvec_t result(count, 0.f);
Chris@1038 137 sv_frame_t longest = 0;
Chris@1008 138
Chris@297 139 for (int c = ch0; c <= ch1; ++c) {
Chris@1096 140
Chris@1096 141 auto here = m_components[c].model->getData(m_components[c].channel,
Chris@1096 142 start, count);
Chris@1100 143 if (sv_frame_t(here.size()) > longest) {
Chris@1100 144 longest = sv_frame_t(here.size());
Chris@1008 145 }
Chris@1096 146 for (sv_frame_t i = 0; in_range_for(here, i); ++i) {
Chris@1096 147 result[i] += here[i];
Chris@297 148 }
Chris@297 149 }
Chris@297 150
Chris@1096 151 result.resize(longest);
Chris@1096 152 return result;
Chris@297 153 }
Chris@363 154
Chris@1326 155 vector<floatvec_t>
Chris@1086 156 AggregateWaveModel::getMultiChannelData(int fromchannel, int tochannel,
Chris@1096 157 sv_frame_t start, sv_frame_t count) const
Chris@363 158 {
Chris@1038 159 sv_frame_t min = count;
Chris@363 160
Chris@1326 161 vector<floatvec_t> result;
Chris@1096 162
Chris@929 163 for (int c = fromchannel; c <= tochannel; ++c) {
Chris@1096 164 auto here = getData(c, start, count);
Chris@1100 165 if (sv_frame_t(here.size()) < min) {
Chris@1100 166 min = sv_frame_t(here.size());
Chris@1100 167 }
Chris@1096 168 result.push_back(here);
Chris@1096 169 }
Chris@1096 170
Chris@1096 171 if (min < count) {
Chris@1096 172 for (auto &v : result) v.resize(min);
Chris@363 173 }
Chris@363 174
Chris@1096 175 return result;
Chris@363 176 }
Chris@377 177
Chris@929 178 int
Chris@929 179 AggregateWaveModel::getSummaryBlockSize(int desired) const
Chris@377 180 {
Chris@377 181 //!!! complete
Chris@377 182 return desired;
Chris@377 183 }
Chris@297 184
Chris@297 185 void
Chris@1038 186 AggregateWaveModel::getSummaries(int, sv_frame_t, sv_frame_t,
Chris@929 187 RangeBlock &, int &) const
Chris@297 188 {
Chris@297 189 //!!! complete
Chris@297 190 }
Chris@297 191
Chris@297 192 AggregateWaveModel::Range
Chris@1038 193 AggregateWaveModel::getSummary(int, sv_frame_t, sv_frame_t) const
Chris@297 194 {
Chris@297 195 //!!! complete
Chris@297 196 return Range();
Chris@297 197 }
Chris@297 198
Chris@929 199 int
Chris@297 200 AggregateWaveModel::getComponentCount() const
Chris@297 201 {
Chris@1038 202 return int(m_components.size());
Chris@297 203 }
Chris@297 204
Chris@297 205 AggregateWaveModel::ModelChannelSpec
Chris@929 206 AggregateWaveModel::getComponent(int c) const
Chris@297 207 {
Chris@297 208 return m_components[c];
Chris@297 209 }
Chris@297 210
Chris@297 211 void
Chris@297 212 AggregateWaveModel::componentModelChanged()
Chris@297 213 {
Chris@297 214 emit modelChanged();
Chris@297 215 }
Chris@297 216
Chris@297 217 void
Chris@1038 218 AggregateWaveModel::componentModelChangedWithin(sv_frame_t start, sv_frame_t end)
Chris@297 219 {
Chris@947 220 emit modelChangedWithin(start, end);
Chris@297 221 }
Chris@297 222
Chris@297 223 void
Chris@297 224 AggregateWaveModel::componentModelCompletionChanged()
Chris@297 225 {
Chris@297 226 emit completionChanged();
Chris@297 227 }
Chris@297 228
Chris@297 229 void
Chris@1391 230 AggregateWaveModel::toXml(QTextStream &out,
Chris@1391 231 QString indent,
Chris@1391 232 QString extraAttributes) const
Chris@297 233 {
Chris@1391 234 QStringList componentStrings;
Chris@1391 235 for (const auto &c: m_components) {
Chris@1391 236 componentStrings.push_back(QString("%1").arg(getObjectExportId(c.model)));
Chris@1391 237 }
Chris@1391 238 Model::toXml(out, indent,
Chris@1391 239 QString("type=\"aggregatewave\" components=\"%1\" %2")
Chris@1391 240 .arg(componentStrings.join(","))
Chris@1391 241 .arg(extraAttributes));
Chris@297 242 }
Chris@297 243