annotate layer/SpectrumLayer.cpp @ 1155:d1bab6a99100 project-file-rework

Cut down vastly on the number of config.pri files and places where their contents has to be effectively duplicated without them
author Chris Cannam
date Mon, 24 Oct 2016 17:53:33 +0100
parents 1badacff7ab2
children ff97318e993c
rev   line source
Chris@133 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@133 2
Chris@133 3 /*
Chris@133 4 Sonic Visualiser
Chris@133 5 An audio file viewer and annotation editor.
Chris@133 6 Centre for Digital Music, Queen Mary, University of London.
Chris@195 7 This file copyright 2006-2007 QMUL.
Chris@133 8
Chris@133 9 This program is free software; you can redistribute it and/or
Chris@133 10 modify it under the terms of the GNU General Public License as
Chris@133 11 published by the Free Software Foundation; either version 2 of the
Chris@133 12 License, or (at your option) any later version. See the file
Chris@133 13 COPYING included with this distribution for more information.
Chris@133 14 */
Chris@133 15
Chris@133 16 #include "SpectrumLayer.h"
Chris@133 17
Chris@133 18 #include "data/model/FFTModel.h"
Chris@133 19 #include "view/View.h"
Chris@153 20 #include "base/AudioLevel.h"
Chris@153 21 #include "base/Preferences.h"
Chris@167 22 #include "base/RangeMapper.h"
Chris@277 23 #include "base/Pitch.h"
Chris@1147 24 #include "base/Strings.h"
Chris@1078 25
Chris@376 26 #include "ColourMapper.h"
Chris@1078 27 #include "PaintAssistant.h"
Chris@254 28
Chris@254 29 #include <QPainter>
Chris@316 30 #include <QTextStream>
Chris@316 31
Chris@133 32
Chris@133 33 SpectrumLayer::SpectrumLayer() :
Chris@193 34 m_originModel(0),
Chris@153 35 m_channel(-1),
Chris@153 36 m_channelSet(false),
Chris@290 37 m_windowSize(4096),
Chris@153 38 m_windowType(HanningWindow),
Chris@290 39 m_windowHopLevel(3),
Chris@284 40 m_showPeaks(false),
Chris@275 41 m_newFFTNeeded(true)
Chris@133 42 {
Chris@153 43 Preferences *prefs = Preferences::getInstance();
Chris@153 44 connect(prefs, SIGNAL(propertyChanged(PropertyContainer::PropertyName)),
Chris@153 45 this, SLOT(preferenceChanged(PropertyContainer::PropertyName)));
Chris@153 46 setWindowType(prefs->getWindowType());
Chris@195 47
Chris@195 48 setBinScale(LogBins);
Chris@133 49 }
Chris@133 50
Chris@133 51 SpectrumLayer::~SpectrumLayer()
Chris@133 52 {
Chris@349 53 Model *m = const_cast<Model *>
Chris@349 54 (static_cast<const Model *>(m_sliceableModel));
Chris@458 55 if (m) m->aboutToDelete();
Chris@349 56 m_sliceableModel = 0;
Chris@349 57 delete m;
Chris@133 58 }
Chris@133 59
Chris@133 60 void
Chris@133 61 SpectrumLayer::setModel(DenseTimeValueModel *model)
Chris@133 62 {
Chris@587 63 SVDEBUG << "SpectrumLayer::setModel(" << model << ") from " << m_originModel << endl;
Chris@345 64
Chris@193 65 if (m_originModel == model) return;
Chris@349 66
Chris@193 67 m_originModel = model;
Chris@277 68
Chris@350 69 if (m_sliceableModel) {
Chris@350 70 Model *m = const_cast<Model *>
Chris@350 71 (static_cast<const Model *>(m_sliceableModel));
Chris@350 72 m->aboutToDelete();
Chris@350 73 setSliceableModel(0);
Chris@350 74 delete m;
Chris@350 75 }
Chris@350 76
Chris@349 77 m_newFFTNeeded = true;
Chris@349 78
Chris@349 79 emit layerParametersChanged();
Chris@349 80 }
Chris@349 81
Chris@349 82 void
Chris@349 83 SpectrumLayer::setChannel(int channel)
Chris@349 84 {
Chris@587 85 SVDEBUG << "SpectrumLayer::setChannel(" << channel << ") from " << m_channel << endl;
Chris@349 86
Chris@349 87 m_channelSet = true;
Chris@349 88
Chris@349 89 if (m_channel == channel) return;
Chris@349 90
Chris@349 91 m_channel = channel;
Chris@349 92
Chris@349 93 m_newFFTNeeded = true;
Chris@349 94
Chris@349 95 emit layerParametersChanged();
Chris@153 96 }
Chris@153 97
Chris@153 98 void
Chris@193 99 SpectrumLayer::setupFFT()
Chris@153 100 {
Chris@349 101 if (m_sliceableModel) {
Chris@349 102 Model *m = const_cast<Model *>
Chris@349 103 (static_cast<const Model *>(m_sliceableModel));
Chris@349 104 m->aboutToDelete();
Chris@193 105 setSliceableModel(0);
Chris@349 106 delete m;
Chris@349 107 }
Chris@349 108
Chris@349 109 if (!m_originModel) {
Chris@349 110 return;
Chris@153 111 }
Chris@153 112
Chris@193 113 FFTModel *newFFT = new FFTModel(m_originModel,
Chris@193 114 m_channel,
Chris@193 115 m_windowType,
Chris@193 116 m_windowSize,
Chris@193 117 getWindowIncrement(),
Chris@975 118 m_windowSize);
Chris@153 119
Chris@193 120 setSliceableModel(newFFT);
Chris@193 121
Chris@254 122 m_biasCurve.clear();
Chris@805 123 for (int i = 0; i < m_windowSize; ++i) {
Chris@254 124 m_biasCurve.push_back(1.f / (float(m_windowSize)/2.f));
Chris@254 125 }
Chris@254 126
Chris@349 127 m_newFFTNeeded = false;
Chris@133 128 }
Chris@133 129
Chris@153 130 Layer::PropertyList
Chris@153 131 SpectrumLayer::getProperties() const
Chris@153 132 {
Chris@193 133 PropertyList list = SliceLayer::getProperties();
Chris@153 134 list.push_back("Window Size");
Chris@153 135 list.push_back("Window Increment");
Chris@284 136 list.push_back("Show Peak Frequencies");
Chris@153 137 return list;
Chris@153 138 }
Chris@153 139
Chris@153 140 QString
Chris@153 141 SpectrumLayer::getPropertyLabel(const PropertyName &name) const
Chris@153 142 {
Chris@153 143 if (name == "Window Size") return tr("Window Size");
Chris@153 144 if (name == "Window Increment") return tr("Window Overlap");
Chris@284 145 if (name == "Show Peak Frequencies") return tr("Show Peak Frequencies");
Chris@193 146 return SliceLayer::getPropertyLabel(name);
Chris@153 147 }
Chris@153 148
Chris@335 149 QString
Chris@335 150 SpectrumLayer::getPropertyIconName(const PropertyName &name) const
Chris@335 151 {
Chris@335 152 if (name == "Show Peak Frequencies") return "show-peaks";
Chris@335 153 return SliceLayer::getPropertyIconName(name);
Chris@335 154 }
Chris@335 155
Chris@153 156 Layer::PropertyType
Chris@153 157 SpectrumLayer::getPropertyType(const PropertyName &name) const
Chris@153 158 {
Chris@193 159 if (name == "Window Size") return ValueProperty;
Chris@193 160 if (name == "Window Increment") return ValueProperty;
Chris@284 161 if (name == "Show Peak Frequencies") return ToggleProperty;
Chris@193 162 return SliceLayer::getPropertyType(name);
Chris@153 163 }
Chris@153 164
Chris@153 165 QString
Chris@153 166 SpectrumLayer::getPropertyGroupName(const PropertyName &name) const
Chris@153 167 {
Chris@153 168 if (name == "Window Size" ||
Chris@153 169 name == "Window Increment") return tr("Window");
Chris@557 170 if (name == "Show Peak Frequencies") return tr("Bins");
Chris@193 171 return SliceLayer::getPropertyGroupName(name);
Chris@153 172 }
Chris@153 173
Chris@153 174 int
Chris@153 175 SpectrumLayer::getPropertyRangeAndValue(const PropertyName &name,
Chris@216 176 int *min, int *max, int *deflt) const
Chris@153 177 {
Chris@216 178 int val = 0;
Chris@153 179
Chris@216 180 int garbage0, garbage1, garbage2;
Chris@153 181 if (!min) min = &garbage0;
Chris@153 182 if (!max) max = &garbage1;
Chris@216 183 if (!deflt) deflt = &garbage2;
Chris@153 184
Chris@193 185 if (name == "Window Size") {
Chris@153 186
Chris@153 187 *min = 0;
Chris@254 188 *max = 15;
Chris@216 189 *deflt = 5;
Chris@153 190
Chris@216 191 val = 0;
Chris@153 192 int ws = m_windowSize;
Chris@216 193 while (ws > 32) { ws >>= 1; val ++; }
Chris@153 194
Chris@153 195 } else if (name == "Window Increment") {
Chris@153 196
Chris@153 197 *min = 0;
Chris@153 198 *max = 5;
Chris@216 199 *deflt = 2;
Chris@153 200
Chris@216 201 val = m_windowHopLevel;
Chris@153 202
Chris@284 203 } else if (name == "Show Peak Frequencies") {
Chris@284 204
Chris@284 205 return m_showPeaks ? 1 : 0;
Chris@284 206
Chris@153 207 } else {
Chris@193 208
Chris@216 209 val = SliceLayer::getPropertyRangeAndValue(name, min, max, deflt);
Chris@153 210 }
Chris@153 211
Chris@216 212 return val;
Chris@153 213 }
Chris@153 214
Chris@153 215 QString
Chris@153 216 SpectrumLayer::getPropertyValueLabel(const PropertyName &name,
Chris@153 217 int value) const
Chris@153 218 {
Chris@153 219 if (name == "Window Size") {
Chris@153 220 return QString("%1").arg(32 << value);
Chris@153 221 }
Chris@153 222 if (name == "Window Increment") {
Chris@153 223 switch (value) {
Chris@153 224 default:
Chris@153 225 case 0: return tr("None");
Chris@153 226 case 1: return tr("25 %");
Chris@153 227 case 2: return tr("50 %");
Chris@153 228 case 3: return tr("75 %");
Chris@153 229 case 4: return tr("87.5 %");
Chris@153 230 case 5: return tr("93.75 %");
Chris@153 231 }
Chris@153 232 }
Chris@193 233 return SliceLayer::getPropertyValueLabel(name, value);
Chris@153 234 }
Chris@153 235
Chris@167 236 RangeMapper *
Chris@167 237 SpectrumLayer::getNewPropertyRangeMapper(const PropertyName &name) const
Chris@167 238 {
Chris@193 239 return SliceLayer::getNewPropertyRangeMapper(name);
Chris@167 240 }
Chris@167 241
Chris@133 242 void
Chris@153 243 SpectrumLayer::setProperty(const PropertyName &name, int value)
Chris@133 244 {
Chris@193 245 if (name == "Window Size") {
Chris@153 246 setWindowSize(32 << value);
Chris@153 247 } else if (name == "Window Increment") {
Chris@153 248 setWindowHopLevel(value);
Chris@284 249 } else if (name == "Show Peak Frequencies") {
Chris@284 250 setShowPeaks(value ? true : false);
Chris@193 251 } else {
Chris@193 252 SliceLayer::setProperty(name, value);
Chris@153 253 }
Chris@153 254 }
Chris@153 255
Chris@153 256 void
Chris@805 257 SpectrumLayer::setWindowSize(int ws)
Chris@153 258 {
Chris@153 259 if (m_windowSize == ws) return;
Chris@153 260 m_windowSize = ws;
Chris@275 261 m_newFFTNeeded = true;
Chris@153 262 emit layerParametersChanged();
Chris@153 263 }
Chris@153 264
Chris@153 265 void
Chris@805 266 SpectrumLayer::setWindowHopLevel(int v)
Chris@153 267 {
Chris@153 268 if (m_windowHopLevel == v) return;
Chris@153 269 m_windowHopLevel = v;
Chris@275 270 m_newFFTNeeded = true;
Chris@153 271 emit layerParametersChanged();
Chris@153 272 }
Chris@153 273
Chris@153 274 void
Chris@153 275 SpectrumLayer::setWindowType(WindowType w)
Chris@153 276 {
Chris@153 277 if (m_windowType == w) return;
Chris@153 278 m_windowType = w;
Chris@275 279 m_newFFTNeeded = true;
Chris@153 280 emit layerParametersChanged();
Chris@153 281 }
Chris@153 282
Chris@153 283 void
Chris@284 284 SpectrumLayer::setShowPeaks(bool show)
Chris@284 285 {
Chris@284 286 if (m_showPeaks == show) return;
Chris@284 287 m_showPeaks = show;
Chris@284 288 emit layerParametersChanged();
Chris@284 289 }
Chris@284 290
Chris@284 291 void
Chris@153 292 SpectrumLayer::preferenceChanged(PropertyContainer::PropertyName name)
Chris@153 293 {
Chris@153 294 if (name == "Window Type") {
Chris@153 295 setWindowType(Preferences::getInstance()->getWindowType());
Chris@153 296 return;
Chris@153 297 }
Chris@153 298 }
Chris@153 299
Chris@133 300 bool
Chris@908 301 SpectrumLayer::getValueExtents(double &, double &, bool &, QString &) const
Chris@133 302 {
Chris@133 303 return false;
Chris@133 304 }
Chris@133 305
Chris@908 306 double
Chris@908 307 SpectrumLayer::getXForBin(int bin, int totalBins, double w) const
Chris@265 308 {
Chris@265 309 if (!m_sliceableModel) return SliceLayer::getXForBin(bin, totalBins, w);
Chris@265 310
Chris@908 311 sv_samplerate_t sampleRate = m_sliceableModel->getSampleRate();
Chris@908 312 double binfreq = (sampleRate * bin) / (totalBins * 2);
Chris@265 313
Chris@265 314 return getXForFrequency(binfreq, w);
Chris@265 315 }
Chris@265 316
Chris@265 317 int
Chris@908 318 SpectrumLayer::getBinForX(double x, int totalBins, double w) const
Chris@265 319 {
Chris@265 320 if (!m_sliceableModel) return SliceLayer::getBinForX(x, totalBins, w);
Chris@265 321
Chris@908 322 sv_samplerate_t sampleRate = m_sliceableModel->getSampleRate();
Chris@908 323 double binfreq = getFrequencyForX(x, w);
Chris@265 324
Chris@265 325 return int((binfreq * totalBins * 2) / sampleRate);
Chris@265 326 }
Chris@265 327
Chris@908 328 double
Chris@908 329 SpectrumLayer::getFrequencyForX(double x, double w) const
Chris@254 330 {
Chris@908 331 double freq = 0;
Chris@280 332 if (!m_sliceableModel) return 0;
Chris@254 333
Chris@908 334 sv_samplerate_t sampleRate = m_sliceableModel->getSampleRate();
Chris@254 335
Chris@908 336 double maxfreq = double(sampleRate) / 2;
Chris@254 337
Chris@254 338 switch (m_binScale) {
Chris@254 339
Chris@254 340 case LinearBins:
Chris@254 341 freq = ((x * maxfreq) / w);
Chris@254 342 break;
Chris@254 343
Chris@254 344 case LogBins:
Chris@908 345 freq = pow(10.0, (x * log10(maxfreq)) / w);
Chris@254 346 break;
Chris@254 347
Chris@254 348 case InvertedLogBins:
Chris@908 349 freq = maxfreq - pow(10.0, ((w - x) * log10(maxfreq)) / w);
Chris@254 350 break;
Chris@254 351 }
Chris@254 352
Chris@254 353 return freq;
Chris@254 354 }
Chris@254 355
Chris@908 356 double
Chris@908 357 SpectrumLayer::getXForFrequency(double freq, double w) const
Chris@254 358 {
Chris@908 359 double x = 0;
Chris@280 360 if (!m_sliceableModel) return x;
Chris@254 361
Chris@908 362 sv_samplerate_t sampleRate = m_sliceableModel->getSampleRate();
Chris@254 363
Chris@908 364 double maxfreq = double(sampleRate) / 2;
Chris@254 365
Chris@254 366 switch (m_binScale) {
Chris@254 367
Chris@254 368 case LinearBins:
Chris@254 369 x = (freq * w) / maxfreq;
Chris@254 370 break;
Chris@254 371
Chris@254 372 case LogBins:
Chris@908 373 x = (log10(freq) * w) / log10(maxfreq);
Chris@254 374 break;
Chris@254 375
Chris@254 376 case InvertedLogBins:
Chris@274 377 if (maxfreq == freq) x = w;
Chris@908 378 else x = w - (log10(maxfreq - freq) * w) / log10(maxfreq);
Chris@254 379 break;
Chris@254 380 }
Chris@254 381
Chris@254 382 return x;
Chris@254 383 }
Chris@254 384
Chris@260 385 bool
Chris@918 386 SpectrumLayer::getXScaleValue(const LayerGeometryProvider *v, int x,
Chris@908 387 double &value, QString &unit) const
Chris@260 388 {
Chris@267 389 if (m_xorigins.find(v) == m_xorigins.end()) return false;
Chris@267 390 int xorigin = m_xorigins.find(v)->second;
Chris@918 391 value = getFrequencyForX(x - xorigin, v->getPaintWidth() - xorigin - 1);
Chris@260 392 unit = "Hz";
Chris@260 393 return true;
Chris@260 394 }
Chris@260 395
Chris@264 396 bool
Chris@918 397 SpectrumLayer::getYScaleValue(const LayerGeometryProvider *v, int y,
Chris@908 398 double &value, QString &unit) const
Chris@274 399 {
Chris@274 400 value = getValueForY(y, v);
Chris@274 401
Chris@274 402 if (m_energyScale == dBScale || m_energyScale == MeterScale) {
Chris@274 403
Chris@908 404 if (value > 0.0) {
Chris@908 405 value = 10.0 * log10(value);
Chris@284 406 if (value < m_threshold) value = m_threshold;
Chris@284 407 } else value = m_threshold;
Chris@274 408
Chris@274 409 unit = "dBV";
Chris@274 410
Chris@274 411 } else {
Chris@274 412 unit = "V";
Chris@274 413 }
Chris@274 414
Chris@274 415 return true;
Chris@274 416 }
Chris@274 417
Chris@274 418 bool
Chris@918 419 SpectrumLayer::getYScaleDifference(const LayerGeometryProvider *v, int y0, int y1,
Chris@908 420 double &diff, QString &unit) const
Chris@274 421 {
Chris@274 422 bool rv = SliceLayer::getYScaleDifference(v, y0, y1, diff, unit);
Chris@274 423 if (rv && (unit == "dBV")) unit = "dB";
Chris@274 424 return rv;
Chris@274 425 }
Chris@274 426
Chris@274 427
Chris@274 428 bool
Chris@918 429 SpectrumLayer::getCrosshairExtents(LayerGeometryProvider *v, QPainter &paint,
Chris@264 430 QPoint cursorPos,
Chris@264 431 std::vector<QRect> &extents) const
Chris@264 432 {
Chris@918 433 QRect vertical(cursorPos.x(), cursorPos.y(), 1, v->getPaintHeight() - cursorPos.y());
Chris@264 434 extents.push_back(vertical);
Chris@264 435
Chris@918 436 QRect horizontal(0, cursorPos.y(), v->getPaintWidth(), 12);
Chris@264 437 extents.push_back(horizontal);
Chris@264 438
Chris@280 439 int hoffset = 2;
Chris@280 440 if (m_binScale == LogBins) hoffset = 13;
Chris@278 441
Chris@607 442 int sw = getVerticalScaleWidth(v, false, paint);
Chris@280 443
Chris@280 444 QRect value(sw, cursorPos.y() - paint.fontMetrics().ascent() - 2,
Chris@280 445 paint.fontMetrics().width("0.0000001 V") + 2,
Chris@264 446 paint.fontMetrics().height());
Chris@280 447 extents.push_back(value);
Chris@280 448
Chris@280 449 QRect log(sw, cursorPos.y() + 2,
Chris@280 450 paint.fontMetrics().width("-80.000 dBV") + 2,
Chris@280 451 paint.fontMetrics().height());
Chris@280 452 extents.push_back(log);
Chris@280 453
Chris@280 454 QRect freq(cursorPos.x(),
Chris@918 455 v->getPaintHeight() - paint.fontMetrics().height() - hoffset,
Chris@280 456 paint.fontMetrics().width("123456 Hz") + 2,
Chris@280 457 paint.fontMetrics().height());
Chris@280 458 extents.push_back(freq);
Chris@264 459
Chris@278 460 int w(paint.fontMetrics().width("C#10+50c") + 2);
Chris@278 461 QRect pitch(cursorPos.x() - w,
Chris@918 462 v->getPaintHeight() - paint.fontMetrics().height() - hoffset,
Chris@278 463 w,
Chris@278 464 paint.fontMetrics().height());
Chris@278 465 extents.push_back(pitch);
Chris@278 466
Chris@264 467 return true;
Chris@264 468 }
Chris@264 469
Chris@254 470 void
Chris@918 471 SpectrumLayer::paintCrosshairs(LayerGeometryProvider *v, QPainter &paint,
Chris@254 472 QPoint cursorPos) const
Chris@254 473 {
Chris@280 474 if (!m_sliceableModel) return;
Chris@280 475
Chris@254 476 paint.save();
Chris@282 477 QFont fn = paint.font();
Chris@282 478 if (fn.pointSize() > 8) {
Chris@282 479 fn.setPointSize(fn.pointSize() - 1);
Chris@282 480 paint.setFont(fn);
Chris@282 481 }
Chris@254 482
Chris@254 483 ColourMapper mapper(m_colourMap, 0, 1);
Chris@254 484 paint.setPen(mapper.getContrastingColour());
Chris@254 485
Chris@254 486 int xorigin = m_xorigins[v];
Chris@918 487 int w = v->getPaintWidth() - xorigin - 1;
Chris@254 488
Chris@918 489 paint.drawLine(xorigin, cursorPos.y(), v->getPaintWidth(), cursorPos.y());
Chris@918 490 paint.drawLine(cursorPos.x(), cursorPos.y(), cursorPos.x(), v->getPaintHeight());
Chris@254 491
Chris@908 492 double fundamental = getFrequencyForX(cursorPos.x() - xorigin, w);
Chris@254 493
Chris@280 494 int hoffset = 2;
Chris@280 495 if (m_binScale == LogBins) hoffset = 13;
Chris@278 496
Chris@1078 497 PaintAssistant::drawVisibleText(v, paint,
Chris@278 498 cursorPos.x() + 2,
Chris@918 499 v->getPaintHeight() - 2 - hoffset,
Chris@278 500 QString("%1 Hz").arg(fundamental),
Chris@1078 501 PaintAssistant::OutlinedText);
Chris@278 502
Chris@278 503 if (Pitch::isFrequencyInMidiRange(fundamental)) {
Chris@278 504 QString pitchLabel = Pitch::getPitchLabelForFrequency(fundamental);
Chris@1078 505 PaintAssistant::drawVisibleText(v, paint,
Chris@278 506 cursorPos.x() - paint.fontMetrics().width(pitchLabel) - 2,
Chris@918 507 v->getPaintHeight() - 2 - hoffset,
Chris@278 508 pitchLabel,
Chris@1078 509 PaintAssistant::OutlinedText);
Chris@278 510 }
Chris@264 511
Chris@908 512 double value = getValueForY(cursorPos.y(), v);
Chris@908 513 double thresh = m_threshold;
Chris@908 514 double db = thresh;
Chris@908 515 if (value > 0.0) db = 10.0 * log10(value);
Chris@280 516 if (db < thresh) db = thresh;
Chris@280 517
Chris@1078 518 PaintAssistant::drawVisibleText(v, paint,
Chris@280 519 xorigin + 2,
Chris@280 520 cursorPos.y() - 2,
Chris@280 521 QString("%1 V").arg(value),
Chris@1078 522 PaintAssistant::OutlinedText);
Chris@280 523
Chris@1078 524 PaintAssistant::drawVisibleText(v, paint,
Chris@280 525 xorigin + 2,
Chris@280 526 cursorPos.y() + 2 + paint.fontMetrics().ascent(),
Chris@280 527 QString("%1 dBV").arg(db),
Chris@1078 528 PaintAssistant::OutlinedText);
Chris@280 529
Chris@254 530 int harmonic = 2;
Chris@254 531
Chris@254 532 while (harmonic < 100) {
Chris@254 533
Chris@908 534 int hx = int(lrint(getXForFrequency(fundamental * harmonic, w)));
Chris@254 535 hx += xorigin;
Chris@254 536
Chris@918 537 if (hx < xorigin || hx > v->getPaintWidth()) break;
Chris@254 538
Chris@254 539 int len = 7;
Chris@254 540
Chris@254 541 if (harmonic % 2 == 0) {
Chris@254 542 if (harmonic % 4 == 0) {
Chris@254 543 len = 12;
Chris@254 544 } else {
Chris@254 545 len = 10;
Chris@254 546 }
Chris@254 547 }
Chris@254 548
Chris@908 549 paint.drawLine(hx,
Chris@254 550 cursorPos.y(),
Chris@908 551 hx,
Chris@254 552 cursorPos.y() + len);
Chris@254 553
Chris@254 554 ++harmonic;
Chris@254 555 }
Chris@254 556
Chris@254 557 paint.restore();
Chris@254 558 }
Chris@254 559
Chris@199 560 QString
Chris@918 561 SpectrumLayer::getFeatureDescription(LayerGeometryProvider *v, QPoint &p) const
Chris@199 562 {
Chris@199 563 if (!m_sliceableModel) return "";
Chris@199 564
Chris@199 565 int minbin = 0, maxbin = 0, range = 0;
Chris@805 566 QString genericDesc = SliceLayer::getFeatureDescriptionAux
Chris@199 567 (v, p, false, minbin, maxbin, range);
Chris@199 568
Chris@199 569 if (genericDesc == "") return "";
Chris@199 570
Chris@908 571 double minvalue = 0.f;
Chris@248 572 if (minbin < int(m_values.size())) minvalue = m_values[minbin];
Chris@199 573
Chris@908 574 double maxvalue = minvalue;
Chris@248 575 if (maxbin < int(m_values.size())) maxvalue = m_values[maxbin];
Chris@199 576
Chris@199 577 if (minvalue > maxvalue) std::swap(minvalue, maxvalue);
Chris@199 578
Chris@199 579 QString binstr;
Chris@199 580 QString hzstr;
Chris@908 581 int minfreq = int(lrint((minbin * m_sliceableModel->getSampleRate()) /
Chris@908 582 m_windowSize));
Chris@908 583 int maxfreq = int(lrint((std::max(maxbin, minbin+1)
Chris@908 584 * m_sliceableModel->getSampleRate()) /
Chris@908 585 m_windowSize));
Chris@199 586
Chris@199 587 if (maxbin != minbin) {
Chris@199 588 binstr = tr("%1 - %2").arg(minbin+1).arg(maxbin+1);
Chris@199 589 } else {
Chris@199 590 binstr = QString("%1").arg(minbin+1);
Chris@199 591 }
Chris@199 592 if (minfreq != maxfreq) {
Chris@199 593 hzstr = tr("%1 - %2 Hz").arg(minfreq).arg(maxfreq);
Chris@199 594 } else {
Chris@199 595 hzstr = tr("%1 Hz").arg(minfreq);
Chris@199 596 }
Chris@199 597
Chris@199 598 QString valuestr;
Chris@199 599 if (maxvalue != minvalue) {
Chris@199 600 valuestr = tr("%1 - %2").arg(minvalue).arg(maxvalue);
Chris@199 601 } else {
Chris@199 602 valuestr = QString("%1").arg(minvalue);
Chris@199 603 }
Chris@199 604
Chris@199 605 QString dbstr;
Chris@908 606 double mindb = AudioLevel::multiplier_to_dB(minvalue);
Chris@908 607 double maxdb = AudioLevel::multiplier_to_dB(maxvalue);
Chris@199 608 QString mindbstr;
Chris@199 609 QString maxdbstr;
Chris@199 610 if (mindb == AudioLevel::DB_FLOOR) {
Chris@1147 611 mindbstr = Strings::minus_infinity;
Chris@199 612 } else {
Chris@908 613 mindbstr = QString("%1").arg(lrint(mindb));
Chris@199 614 }
Chris@199 615 if (maxdb == AudioLevel::DB_FLOOR) {
Chris@1147 616 maxdbstr = Strings::minus_infinity;
Chris@199 617 } else {
Chris@908 618 maxdbstr = QString("%1").arg(lrint(maxdb));
Chris@199 619 }
Chris@908 620 if (lrint(mindb) != lrint(maxdb)) {
Chris@199 621 dbstr = tr("%1 - %2").arg(mindbstr).arg(maxdbstr);
Chris@199 622 } else {
Chris@199 623 dbstr = tr("%1").arg(mindbstr);
Chris@199 624 }
Chris@199 625
Chris@199 626 QString description;
Chris@199 627
Chris@248 628 if (range > int(m_sliceableModel->getResolution())) {
Chris@199 629 description = tr("%1\nBin:\t%2 (%3)\n%4 value:\t%5\ndB:\t%6")
Chris@199 630 .arg(genericDesc)
Chris@199 631 .arg(binstr)
Chris@199 632 .arg(hzstr)
Chris@199 633 .arg(m_samplingMode == NearestSample ? tr("First") :
Chris@199 634 m_samplingMode == SampleMean ? tr("Mean") : tr("Peak"))
Chris@199 635 .arg(valuestr)
Chris@199 636 .arg(dbstr);
Chris@199 637 } else {
Chris@199 638 description = tr("%1\nBin:\t%2 (%3)\nValue:\t%4\ndB:\t%5")
Chris@199 639 .arg(genericDesc)
Chris@199 640 .arg(binstr)
Chris@199 641 .arg(hzstr)
Chris@199 642 .arg(valuestr)
Chris@199 643 .arg(dbstr);
Chris@199 644 }
Chris@199 645
Chris@199 646 return description;
Chris@199 647 }
Chris@199 648
Chris@254 649 void
Chris@916 650 SpectrumLayer::paint(LayerGeometryProvider *v, QPainter &paint, QRect rect) const
Chris@275 651 {
Chris@275 652 if (!m_originModel || !m_originModel->isOK() ||
Chris@349 653 !m_originModel->isReady()) {
Chris@587 654 SVDEBUG << "SpectrumLayer::paint: no origin model, or origin model not OK or not ready" << endl;
Chris@349 655 return;
Chris@349 656 }
Chris@275 657
Chris@275 658 if (m_newFFTNeeded) {
Chris@587 659 SVDEBUG << "SpectrumLayer::paint: new FFT needed, calling setupFFT" << endl;
Chris@275 660 const_cast<SpectrumLayer *>(this)->setupFFT(); //ugh
Chris@275 661 }
Chris@277 662
Chris@277 663 FFTModel *fft = dynamic_cast<FFTModel *>
Chris@277 664 (const_cast<DenseThreeDimensionalModel *>(m_sliceableModel));
Chris@277 665
Chris@908 666 double thresh = (pow(10, -6) / m_gain) * (m_windowSize / 2.0); // -60dB adj
Chris@277 667
Chris@607 668 int xorigin = getVerticalScaleWidth(v, false, paint) + 1;
Chris@918 669 int w = v->getPaintWidth() - xorigin - 1;
Chris@277 670
Chris@278 671 int pkh = 0;
Chris@287 672 //!!! if (m_binScale == LogBins) {
Chris@287 673 pkh = 10;
Chris@287 674 //!!! }
Chris@278 675
Chris@345 676 paint.save();
Chris@345 677
Chris@284 678 if (fft && m_showPeaks) {
Chris@277 679
Chris@277 680 // draw peak lines
Chris@277 681
Chris@587 682 // SVDEBUG << "Showing peaks..." << endl;
Chris@456 683
Chris@908 684 int col = int(v->getCentreFrame() / fft->getResolution());
Chris@277 685
Chris@277 686 paint.save();
Chris@277 687 paint.setRenderHint(QPainter::Antialiasing, false);
Chris@277 688 paint.setPen(QColor(160, 160, 160)); //!!!
Chris@277 689
Chris@290 690 int peakminbin = 0;
Chris@290 691 int peakmaxbin = fft->getHeight() - 1;
Chris@908 692 double peakmaxfreq = Pitch::getFrequencyForPitch(128);
Chris@908 693 peakmaxbin = int(((peakmaxfreq * fft->getHeight() * 2) / fft->getSampleRate()));
Chris@290 694
Chris@280 695 FFTModel::PeakSet peaks = fft->getPeakFrequencies
Chris@290 696 (FFTModel::MajorPitchAdaptivePeaks, col, peakminbin, peakmaxbin);
Chris@280 697
Chris@280 698 ColourMapper mapper(ColourMapper::BlackOnWhite, 0, 1);
Chris@277 699
Chris@277 700 BiasCurve curve;
Chris@277 701 getBiasCurve(curve);
Chris@908 702 int cs = int(curve.size());
Chris@280 703
Chris@908 704 std::vector<double> values;
Chris@277 705
Chris@805 706 for (int bin = 0; bin < fft->getHeight(); ++bin) {
Chris@908 707 double value = m_sliceableModel->getValueAt(col, bin);
Chris@280 708 if (bin < cs) value *= curve[bin];
Chris@280 709 values.push_back(value);
Chris@280 710 }
Chris@280 711
Chris@280 712 for (FFTModel::PeakSet::iterator i = peaks.begin();
Chris@280 713 i != peaks.end(); ++i) {
Chris@280 714
Chris@805 715 int bin = i->first;
Chris@277 716
Chris@682 717 // cerr << "bin = " << bin << ", thresh = " << thresh << ", value = " << fft->getMagnitudeAt(col, bin) << endl;
Chris@280 718
Chris@908 719 if (!fft->isOverThreshold(col, bin, float(thresh))) continue;
Chris@277 720
Chris@908 721 double freq = i->second;
Chris@280 722
Chris@908 723 int x = int(lrint(getXForFrequency(freq, w)));
Chris@277 724
Chris@908 725 double norm = 0.f;
Chris@805 726 (void)getYForValue(values[bin], v, norm); // don't need return value, need norm
Chris@277 727
Chris@277 728 paint.setPen(mapper.map(norm));
Chris@918 729 paint.drawLine(xorigin + x, 0, xorigin + x, v->getPaintHeight() - pkh - 1);
Chris@277 730 }
Chris@277 731
Chris@277 732 paint.restore();
Chris@277 733 }
Chris@275 734
Chris@275 735 SliceLayer::paint(v, paint, rect);
Chris@277 736
Chris@278 737 //!!! All of this stuff relating to depicting frequencies
Chris@278 738 //(keyboard, crosshairs etc) should be applicable to any slice
Chris@278 739 //layer whose model has a vertical scale unit of Hz. However, the
Chris@278 740 //dense 3d model at the moment doesn't record its vertical scale
Chris@278 741 //unit -- we need to fix that and hoist this code as appropriate.
Chris@278 742 //Same really goes for any code in SpectrogramLayer that could be
Chris@278 743 //relevant to Colour3DPlotLayer with unit Hz, but that's a bigger
Chris@278 744 //proposition.
Chris@278 745
Chris@287 746 // if (m_binScale == LogBins) {
Chris@277 747
Chris@287 748 // int pkh = 10;
Chris@918 749 int h = v->getPaintHeight();
Chris@277 750
Chris@277 751 // piano keyboard
Chris@277 752 //!!! should be in a new paintHorizontalScale()?
Chris@278 753 // nice to have a piano keyboard class, of course
Chris@277 754
Chris@278 755 paint.drawLine(xorigin, h - pkh - 1, w + xorigin, h - pkh - 1);
Chris@277 756
Chris@277 757 int px = xorigin, ppx = xorigin;
Chris@278 758 paint.setBrush(paint.pen().color());
Chris@277 759
Chris@277 760 for (int i = 0; i < 128; ++i) {
Chris@277 761
Chris@908 762 double f = Pitch::getFrequencyForPitch(i);
Chris@908 763 int x = int(lrint(getXForFrequency(f, w)));
Chris@278 764
Chris@278 765 x += xorigin;
Chris@277 766
Chris@278 767 if (i == 0) {
Chris@278 768 px = ppx = x;
Chris@278 769 }
Chris@278 770 if (i == 1) {
Chris@278 771 ppx = px - (x - px);
Chris@278 772 }
Chris@278 773
Chris@278 774 if (x < xorigin) {
Chris@278 775 ppx = px;
Chris@278 776 px = x;
Chris@277 777 continue;
Chris@277 778 }
Chris@278 779
Chris@278 780 if (x > w) {
Chris@278 781 break;
Chris@278 782 }
Chris@277 783
Chris@277 784 int n = (i % 12);
Chris@277 785
Chris@277 786 if (n == 1) {
Chris@277 787 // C# -- fill the C from here
Chris@284 788 QColor col = Qt::gray;
Chris@284 789 if (i == 61) { // filling middle C
Chris@284 790 col = Qt::blue;
Chris@284 791 col = col.light(150);
Chris@284 792 }
Chris@277 793 if (x - ppx > 2) {
Chris@278 794 paint.fillRect((px + ppx) / 2 + 1,
Chris@277 795 h - pkh,
Chris@278 796 x - (px + ppx) / 2 - 1,
Chris@277 797 pkh,
Chris@284 798 col);
Chris@277 799 }
Chris@277 800 }
Chris@277 801
Chris@277 802 if (n == 1 || n == 3 || n == 6 || n == 8 || n == 10) {
Chris@277 803 // black notes
Chris@277 804 paint.drawLine(x, h - pkh, x, h);
Chris@908 805 int rw = int(lrint(double(x - px) / 4) * 2);
Chris@277 806 if (rw < 2) rw = 2;
Chris@278 807 paint.drawRect(x - rw/2, h - pkh, rw, pkh/2);
Chris@277 808 } else if (n == 0 || n == 5) {
Chris@277 809 // C, F
Chris@277 810 if (px < w) {
Chris@277 811 paint.drawLine((x + px) / 2, h - pkh, (x + px) / 2, h);
Chris@277 812 }
Chris@277 813 }
Chris@277 814
Chris@277 815 ppx = px;
Chris@277 816 px = x;
Chris@277 817 }
Chris@287 818 // }
Chris@345 819
Chris@345 820 paint.restore();
Chris@275 821 }
Chris@275 822
Chris@275 823 void
Chris@254 824 SpectrumLayer::getBiasCurve(BiasCurve &curve) const
Chris@254 825 {
Chris@254 826 curve = m_biasCurve;
Chris@254 827 }
Chris@199 828
Chris@316 829 void
Chris@316 830 SpectrumLayer::toXml(QTextStream &stream,
Chris@316 831 QString indent, QString extraAttributes) const
Chris@220 832 {
Chris@316 833 QString s = QString("windowSize=\"%1\" "
Chris@456 834 "windowHopLevel=\"%2\" "
Chris@456 835 "showPeaks=\"%3\" ")
Chris@220 836 .arg(m_windowSize)
Chris@456 837 .arg(m_windowHopLevel)
Chris@456 838 .arg(m_showPeaks ? "true" : "false");
Chris@220 839
Chris@316 840 SliceLayer::toXml(stream, indent, extraAttributes + " " + s);
Chris@220 841 }
Chris@220 842
Chris@220 843 void
Chris@220 844 SpectrumLayer::setProperties(const QXmlAttributes &attributes)
Chris@220 845 {
Chris@220 846 SliceLayer::setProperties(attributes);
Chris@220 847
Chris@220 848 bool ok = false;
Chris@220 849
Chris@805 850 int windowSize = attributes.value("windowSize").toUInt(&ok);
Chris@220 851 if (ok) setWindowSize(windowSize);
Chris@220 852
Chris@805 853 int windowHopLevel = attributes.value("windowHopLevel").toUInt(&ok);
Chris@220 854 if (ok) setWindowHopLevel(windowHopLevel);
Chris@456 855
Chris@456 856 bool showPeaks = (attributes.value("showPeaks").trimmed() == "true");
Chris@456 857 setShowPeaks(showPeaks);
Chris@220 858 }
Chris@220 859
Chris@220 860