annotate layer/SpectrogramLayer.cpp @ 111:ab55c427e068

* Split large FFT caches up into several files. Ongoing.
author Chris Cannam
date Wed, 28 Jun 2006 13:41:12 +0000
parents f262aa8973e3
children 43b1ca9647e1
rev   line source
Chris@58 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@0 2
Chris@0 3 /*
Chris@59 4 Sonic Visualiser
Chris@59 5 An audio file viewer and annotation editor.
Chris@59 6 Centre for Digital Music, Queen Mary, University of London.
Chris@59 7 This file copyright 2006 Chris Cannam.
Chris@0 8
Chris@59 9 This program is free software; you can redistribute it and/or
Chris@59 10 modify it under the terms of the GNU General Public License as
Chris@59 11 published by the Free Software Foundation; either version 2 of the
Chris@59 12 License, or (at your option) any later version. See the file
Chris@59 13 COPYING included with this distribution for more information.
Chris@0 14 */
Chris@0 15
Chris@0 16 #include "SpectrogramLayer.h"
Chris@0 17
Chris@0 18 #include "base/View.h"
Chris@0 19 #include "base/Profiler.h"
Chris@0 20 #include "base/AudioLevel.h"
Chris@0 21 #include "base/Window.h"
Chris@24 22 #include "base/Pitch.h"
Chris@110 23 #include "fileio/FFTDataServer.h"
Chris@0 24
Chris@0 25 #include <QPainter>
Chris@0 26 #include <QImage>
Chris@0 27 #include <QPixmap>
Chris@0 28 #include <QRect>
Chris@0 29 #include <QTimer>
Chris@92 30 #include <QApplication>
Chris@0 31
Chris@0 32 #include <iostream>
Chris@0 33
Chris@0 34 #include <cassert>
Chris@0 35 #include <cmath>
Chris@0 36
Chris@101 37 //#define DEBUG_SPECTROGRAM_REPAINT 1
Chris@0 38
Chris@44 39 SpectrogramLayer::SpectrogramLayer(Configuration config) :
Chris@44 40 Layer(),
Chris@0 41 m_model(0),
Chris@0 42 m_channel(0),
Chris@0 43 m_windowSize(1024),
Chris@0 44 m_windowType(HanningWindow),
Chris@97 45 m_windowHopLevel(2),
Chris@109 46 m_zeroPadLevel(0),
Chris@107 47 m_fftSize(1024),
Chris@0 48 m_gain(1.0),
Chris@37 49 m_threshold(0.0),
Chris@9 50 m_colourRotation(0),
Chris@37 51 m_minFrequency(0),
Chris@0 52 m_maxFrequency(8000),
Chris@0 53 m_colourScale(dBColourScale),
Chris@0 54 m_colourScheme(DefaultColours),
Chris@0 55 m_frequencyScale(LinearFrequencyScale),
Chris@37 56 m_binDisplay(AllBins),
Chris@36 57 m_normalizeColumns(false),
Chris@110 58 m_fftServer(0),
Chris@110 59 // m_fftServerChanged(true),
Chris@0 60 m_updateTimer(0),
Chris@44 61 m_candidateFillStartFrame(0),
Chris@0 62 m_lastFillExtent(0),
Chris@0 63 m_exiting(false)
Chris@0 64 {
Chris@0 65 if (config == MelodicRange) {
Chris@0 66 setWindowSize(8192);
Chris@97 67 setWindowHopLevel(4);
Chris@109 68 // setWindowType(ParzenWindow);
Chris@0 69 setMaxFrequency(1000);
Chris@0 70 setColourScale(LinearColourScale);
Chris@37 71 } else if (config == MelodicPeaks) {
Chris@37 72 setWindowSize(4096);
Chris@97 73 setWindowHopLevel(5);
Chris@109 74 // setWindowType(BlackmanWindow);
Chris@40 75 setMaxFrequency(2000);
Chris@37 76 setMinFrequency(40);
Chris@37 77 setFrequencyScale(LogFrequencyScale);
Chris@41 78 setColourScale(MeterColourScale);
Chris@37 79 setBinDisplay(PeakFrequencies);
Chris@37 80 setNormalizeColumns(true);
Chris@0 81 }
Chris@110 82
Chris@110 83 setColourmap();
Chris@0 84 }
Chris@0 85
Chris@0 86 SpectrogramLayer::~SpectrogramLayer()
Chris@0 87 {
Chris@0 88 delete m_updateTimer;
Chris@0 89 m_updateTimer = 0;
Chris@0 90
Chris@110 91 if (m_fftServer) {
Chris@110 92 FFTDataServer::releaseInstance(m_fftServer);
Chris@110 93 m_fftServer = 0;
Chris@110 94 }
Chris@0 95 }
Chris@0 96
Chris@0 97 void
Chris@0 98 SpectrogramLayer::setModel(const DenseTimeValueModel *model)
Chris@0 99 {
Chris@101 100 // std::cerr << "SpectrogramLayer(" << this << "): setModel(" << model << ")" << std::endl;
Chris@34 101
Chris@110 102 if (model == m_model) return;
Chris@110 103
Chris@0 104 m_model = model;
Chris@110 105 getFFTServer();
Chris@0 106
Chris@0 107 if (!m_model || !m_model->isOK()) return;
Chris@0 108
Chris@0 109 connect(m_model, SIGNAL(modelChanged()), this, SIGNAL(modelChanged()));
Chris@0 110 connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
Chris@0 111 this, SIGNAL(modelChanged(size_t, size_t)));
Chris@0 112
Chris@0 113 connect(m_model, SIGNAL(completionChanged()),
Chris@0 114 this, SIGNAL(modelCompletionChanged()));
Chris@0 115
Chris@0 116 connect(m_model, SIGNAL(modelChanged()), this, SLOT(cacheInvalid()));
Chris@0 117 connect(m_model, SIGNAL(modelChanged(size_t, size_t)),
Chris@0 118 this, SLOT(cacheInvalid(size_t, size_t)));
Chris@0 119
Chris@0 120 emit modelReplaced();
Chris@110 121 }
Chris@110 122
Chris@110 123 void
Chris@110 124 SpectrogramLayer::getFFTServer()
Chris@110 125 {
Chris@110 126 //!!! really want to check that params differ from previous ones
Chris@110 127
Chris@110 128 if (m_fftServer) {
Chris@110 129 FFTDataServer::releaseInstance(m_fftServer);
Chris@110 130 m_fftServer = 0;
Chris@110 131 }
Chris@110 132
Chris@110 133 if (m_model) {
Chris@110 134 m_fftServer = FFTDataServer::getInstance(m_model,
Chris@110 135 m_channel,
Chris@110 136 m_windowType,
Chris@110 137 m_windowSize,
Chris@110 138 getWindowIncrement(),
Chris@110 139 m_fftSize,
Chris@110 140 true,
Chris@110 141 m_candidateFillStartFrame);
Chris@110 142
Chris@110 143 m_lastFillExtent = 0;
Chris@110 144
Chris@110 145 delete m_updateTimer;
Chris@110 146 m_updateTimer = new QTimer(this);
Chris@110 147 connect(m_updateTimer, SIGNAL(timeout()), this, SLOT(fillTimerTimedOut()));
Chris@110 148 m_updateTimer->start(200);
Chris@110 149 }
Chris@0 150 }
Chris@0 151
Chris@0 152 Layer::PropertyList
Chris@0 153 SpectrogramLayer::getProperties() const
Chris@0 154 {
Chris@0 155 PropertyList list;
Chris@87 156 list.push_back("Colour");
Chris@87 157 list.push_back("Colour Scale");
Chris@109 158 // list.push_back("Window Type");
Chris@87 159 list.push_back("Window Size");
Chris@97 160 list.push_back("Window Increment");
Chris@87 161 list.push_back("Normalize Columns");
Chris@87 162 list.push_back("Bin Display");
Chris@87 163 list.push_back("Threshold");
Chris@87 164 list.push_back("Gain");
Chris@87 165 list.push_back("Colour Rotation");
Chris@87 166 list.push_back("Min Frequency");
Chris@87 167 list.push_back("Max Frequency");
Chris@87 168 list.push_back("Frequency Scale");
Chris@111 169 // list.push_back("Zero Padding");
Chris@0 170 return list;
Chris@0 171 }
Chris@0 172
Chris@87 173 QString
Chris@87 174 SpectrogramLayer::getPropertyLabel(const PropertyName &name) const
Chris@87 175 {
Chris@87 176 if (name == "Colour") return tr("Colour");
Chris@87 177 if (name == "Colour Scale") return tr("Colour Scale");
Chris@87 178 if (name == "Window Type") return tr("Window Type");
Chris@87 179 if (name == "Window Size") return tr("Window Size");
Chris@97 180 if (name == "Window Increment") return tr("Window Increment");
Chris@87 181 if (name == "Normalize Columns") return tr("Normalize Columns");
Chris@87 182 if (name == "Bin Display") return tr("Bin Display");
Chris@87 183 if (name == "Threshold") return tr("Threshold");
Chris@87 184 if (name == "Gain") return tr("Gain");
Chris@87 185 if (name == "Colour Rotation") return tr("Colour Rotation");
Chris@87 186 if (name == "Min Frequency") return tr("Min Frequency");
Chris@87 187 if (name == "Max Frequency") return tr("Max Frequency");
Chris@87 188 if (name == "Frequency Scale") return tr("Frequency Scale");
Chris@109 189 if (name == "Zero Padding") return tr("Smoothing");
Chris@87 190 return "";
Chris@87 191 }
Chris@87 192
Chris@0 193 Layer::PropertyType
Chris@0 194 SpectrogramLayer::getPropertyType(const PropertyName &name) const
Chris@0 195 {
Chris@87 196 if (name == "Gain") return RangeProperty;
Chris@87 197 if (name == "Colour Rotation") return RangeProperty;
Chris@87 198 if (name == "Normalize Columns") return ToggleProperty;
Chris@87 199 if (name == "Threshold") return RangeProperty;
Chris@109 200 if (name == "Zero Padding") return ToggleProperty;
Chris@0 201 return ValueProperty;
Chris@0 202 }
Chris@0 203
Chris@0 204 QString
Chris@0 205 SpectrogramLayer::getPropertyGroupName(const PropertyName &name) const
Chris@0 206 {
Chris@87 207 if (name == "Window Size" ||
Chris@87 208 name == "Window Type" ||
Chris@109 209 name == "Window Increment" ||
Chris@109 210 name == "Zero Padding") return tr("Window");
Chris@87 211 if (name == "Colour" ||
Chris@87 212 name == "Gain" ||
Chris@87 213 name == "Threshold" ||
Chris@87 214 name == "Colour Rotation") return tr("Colour");
Chris@87 215 if (name == "Normalize Columns" ||
Chris@87 216 name == "Bin Display" ||
Chris@87 217 name == "Colour Scale") return tr("Scale");
Chris@87 218 if (name == "Max Frequency" ||
Chris@87 219 name == "Min Frequency" ||
Chris@87 220 name == "Frequency Scale" ||
Chris@87 221 name == "Frequency Adjustment") return tr("Range");
Chris@0 222 return QString();
Chris@0 223 }
Chris@0 224
Chris@0 225 int
Chris@0 226 SpectrogramLayer::getPropertyRangeAndValue(const PropertyName &name,
Chris@55 227 int *min, int *max) const
Chris@0 228 {
Chris@0 229 int deft = 0;
Chris@0 230
Chris@55 231 int garbage0, garbage1;
Chris@55 232 if (!min) min = &garbage0;
Chris@55 233 if (!max) max = &garbage1;
Chris@10 234
Chris@87 235 if (name == "Gain") {
Chris@0 236
Chris@0 237 *min = -50;
Chris@0 238 *max = 50;
Chris@0 239
Chris@0 240 deft = lrint(log10(m_gain) * 20.0);
Chris@0 241 if (deft < *min) deft = *min;
Chris@0 242 if (deft > *max) deft = *max;
Chris@0 243
Chris@87 244 } else if (name == "Threshold") {
Chris@37 245
Chris@37 246 *min = -50;
Chris@37 247 *max = 0;
Chris@37 248
Chris@37 249 deft = lrintf(AudioLevel::multiplier_to_dB(m_threshold));
Chris@37 250 if (deft < *min) deft = *min;
Chris@37 251 if (deft > *max) deft = *max;
Chris@37 252
Chris@87 253 } else if (name == "Colour Rotation") {
Chris@9 254
Chris@9 255 *min = 0;
Chris@9 256 *max = 256;
Chris@9 257
Chris@9 258 deft = m_colourRotation;
Chris@9 259
Chris@87 260 } else if (name == "Colour Scale") {
Chris@0 261
Chris@0 262 *min = 0;
Chris@0 263 *max = 3;
Chris@0 264
Chris@0 265 deft = (int)m_colourScale;
Chris@0 266
Chris@87 267 } else if (name == "Colour") {
Chris@0 268
Chris@0 269 *min = 0;
Chris@71 270 *max = 6;
Chris@0 271
Chris@0 272 deft = (int)m_colourScheme;
Chris@0 273
Chris@87 274 } else if (name == "Window Type") {
Chris@0 275
Chris@0 276 *min = 0;
Chris@0 277 *max = 6;
Chris@0 278
Chris@0 279 deft = (int)m_windowType;
Chris@0 280
Chris@87 281 } else if (name == "Window Size") {
Chris@0 282
Chris@0 283 *min = 0;
Chris@0 284 *max = 10;
Chris@0 285
Chris@0 286 deft = 0;
Chris@0 287 int ws = m_windowSize;
Chris@0 288 while (ws > 32) { ws >>= 1; deft ++; }
Chris@0 289
Chris@97 290 } else if (name == "Window Increment") {
Chris@0 291
Chris@0 292 *min = 0;
Chris@97 293 *max = 5;
Chris@0 294
Chris@97 295 deft = m_windowHopLevel;
Chris@0 296
Chris@109 297 } else if (name == "Zero Padding") {
Chris@109 298
Chris@109 299 *min = 0;
Chris@109 300 *max = 1;
Chris@109 301
Chris@109 302 deft = m_zeroPadLevel > 0 ? 1 : 0;
Chris@109 303
Chris@87 304 } else if (name == "Min Frequency") {
Chris@37 305
Chris@37 306 *min = 0;
Chris@37 307 *max = 9;
Chris@37 308
Chris@37 309 switch (m_minFrequency) {
Chris@37 310 case 0: default: deft = 0; break;
Chris@37 311 case 10: deft = 1; break;
Chris@37 312 case 20: deft = 2; break;
Chris@37 313 case 40: deft = 3; break;
Chris@37 314 case 100: deft = 4; break;
Chris@37 315 case 250: deft = 5; break;
Chris@37 316 case 500: deft = 6; break;
Chris@37 317 case 1000: deft = 7; break;
Chris@37 318 case 4000: deft = 8; break;
Chris@37 319 case 10000: deft = 9; break;
Chris@37 320 }
Chris@37 321
Chris@87 322 } else if (name == "Max Frequency") {
Chris@0 323
Chris@0 324 *min = 0;
Chris@0 325 *max = 9;
Chris@0 326
Chris@0 327 switch (m_maxFrequency) {
Chris@0 328 case 500: deft = 0; break;
Chris@0 329 case 1000: deft = 1; break;
Chris@0 330 case 1500: deft = 2; break;
Chris@0 331 case 2000: deft = 3; break;
Chris@0 332 case 4000: deft = 4; break;
Chris@0 333 case 6000: deft = 5; break;
Chris@0 334 case 8000: deft = 6; break;
Chris@0 335 case 12000: deft = 7; break;
Chris@0 336 case 16000: deft = 8; break;
Chris@0 337 default: deft = 9; break;
Chris@0 338 }
Chris@0 339
Chris@87 340 } else if (name == "Frequency Scale") {
Chris@0 341
Chris@0 342 *min = 0;
Chris@0 343 *max = 1;
Chris@0 344 deft = (int)m_frequencyScale;
Chris@0 345
Chris@87 346 } else if (name == "Bin Display") {
Chris@35 347
Chris@35 348 *min = 0;
Chris@35 349 *max = 2;
Chris@37 350 deft = (int)m_binDisplay;
Chris@35 351
Chris@87 352 } else if (name == "Normalize Columns") {
Chris@36 353
Chris@36 354 deft = (m_normalizeColumns ? 1 : 0);
Chris@36 355
Chris@0 356 } else {
Chris@0 357 deft = Layer::getPropertyRangeAndValue(name, min, max);
Chris@0 358 }
Chris@0 359
Chris@0 360 return deft;
Chris@0 361 }
Chris@0 362
Chris@0 363 QString
Chris@0 364 SpectrogramLayer::getPropertyValueLabel(const PropertyName &name,
Chris@9 365 int value) const
Chris@0 366 {
Chris@87 367 if (name == "Colour") {
Chris@0 368 switch (value) {
Chris@0 369 default:
Chris@0 370 case 0: return tr("Default");
Chris@0 371 case 1: return tr("White on Black");
Chris@0 372 case 2: return tr("Black on White");
Chris@0 373 case 3: return tr("Red on Blue");
Chris@0 374 case 4: return tr("Yellow on Black");
Chris@71 375 case 5: return tr("Blue on Black");
Chris@71 376 case 6: return tr("Fruit Salad");
Chris@0 377 }
Chris@0 378 }
Chris@87 379 if (name == "Colour Scale") {
Chris@0 380 switch (value) {
Chris@0 381 default:
Chris@37 382 case 0: return tr("Linear");
Chris@37 383 case 1: return tr("Meter");
Chris@37 384 case 2: return tr("dB");
Chris@0 385 case 3: return tr("Phase");
Chris@0 386 }
Chris@0 387 }
Chris@87 388 if (name == "Window Type") {
Chris@0 389 switch ((WindowType)value) {
Chris@0 390 default:
Chris@35 391 case RectangularWindow: return tr("Rectangle");
Chris@0 392 case BartlettWindow: return tr("Bartlett");
Chris@0 393 case HammingWindow: return tr("Hamming");
Chris@0 394 case HanningWindow: return tr("Hanning");
Chris@0 395 case BlackmanWindow: return tr("Blackman");
Chris@0 396 case GaussianWindow: return tr("Gaussian");
Chris@0 397 case ParzenWindow: return tr("Parzen");
Chris@0 398 }
Chris@0 399 }
Chris@87 400 if (name == "Window Size") {
Chris@0 401 return QString("%1").arg(32 << value);
Chris@0 402 }
Chris@97 403 if (name == "Window Increment") {
Chris@0 404 switch (value) {
Chris@0 405 default:
Chris@97 406 case 0: return tr("1/1");
Chris@97 407 case 1: return tr("3/4");
Chris@97 408 case 2: return tr("1/2");
Chris@97 409 case 3: return tr("1/4");
Chris@97 410 case 4: return tr("1/8");
Chris@97 411 case 5: return tr("1/16");
Chris@0 412 }
Chris@0 413 }
Chris@109 414 if (name == "Zero Padding") {
Chris@109 415 if (value == 0) return tr("None");
Chris@109 416 return QString("%1x").arg(value + 1);
Chris@109 417 }
Chris@87 418 if (name == "Min Frequency") {
Chris@37 419 switch (value) {
Chris@37 420 default:
Chris@38 421 case 0: return tr("No min");
Chris@37 422 case 1: return tr("10 Hz");
Chris@37 423 case 2: return tr("20 Hz");
Chris@37 424 case 3: return tr("40 Hz");
Chris@37 425 case 4: return tr("100 Hz");
Chris@37 426 case 5: return tr("250 Hz");
Chris@37 427 case 6: return tr("500 Hz");
Chris@37 428 case 7: return tr("1 KHz");
Chris@37 429 case 8: return tr("4 KHz");
Chris@37 430 case 9: return tr("10 KHz");
Chris@37 431 }
Chris@37 432 }
Chris@87 433 if (name == "Max Frequency") {
Chris@0 434 switch (value) {
Chris@0 435 default:
Chris@0 436 case 0: return tr("500 Hz");
Chris@0 437 case 1: return tr("1 KHz");
Chris@0 438 case 2: return tr("1.5 KHz");
Chris@0 439 case 3: return tr("2 KHz");
Chris@0 440 case 4: return tr("4 KHz");
Chris@0 441 case 5: return tr("6 KHz");
Chris@0 442 case 6: return tr("8 KHz");
Chris@0 443 case 7: return tr("12 KHz");
Chris@0 444 case 8: return tr("16 KHz");
Chris@38 445 case 9: return tr("No max");
Chris@0 446 }
Chris@0 447 }
Chris@87 448 if (name == "Frequency Scale") {
Chris@0 449 switch (value) {
Chris@0 450 default:
Chris@0 451 case 0: return tr("Linear");
Chris@0 452 case 1: return tr("Log");
Chris@0 453 }
Chris@0 454 }
Chris@87 455 if (name == "Bin Display") {
Chris@35 456 switch (value) {
Chris@35 457 default:
Chris@37 458 case 0: return tr("All Bins");
Chris@37 459 case 1: return tr("Peak Bins");
Chris@37 460 case 2: return tr("Frequencies");
Chris@35 461 }
Chris@35 462 }
Chris@0 463 return tr("<unknown>");
Chris@0 464 }
Chris@0 465
Chris@0 466 void
Chris@0 467 SpectrogramLayer::setProperty(const PropertyName &name, int value)
Chris@0 468 {
Chris@87 469 if (name == "Gain") {
Chris@0 470 setGain(pow(10, float(value)/20.0));
Chris@87 471 } else if (name == "Threshold") {
Chris@37 472 if (value == -50) setThreshold(0.0);
Chris@37 473 else setThreshold(AudioLevel::dB_to_multiplier(value));
Chris@87 474 } else if (name == "Colour Rotation") {
Chris@9 475 setColourRotation(value);
Chris@87 476 } else if (name == "Colour") {
Chris@0 477 switch (value) {
Chris@0 478 default:
Chris@0 479 case 0: setColourScheme(DefaultColours); break;
Chris@0 480 case 1: setColourScheme(WhiteOnBlack); break;
Chris@0 481 case 2: setColourScheme(BlackOnWhite); break;
Chris@0 482 case 3: setColourScheme(RedOnBlue); break;
Chris@0 483 case 4: setColourScheme(YellowOnBlack); break;
Chris@71 484 case 5: setColourScheme(BlueOnBlack); break;
Chris@71 485 case 6: setColourScheme(Rainbow); break;
Chris@0 486 }
Chris@87 487 } else if (name == "Window Type") {
Chris@0 488 setWindowType(WindowType(value));
Chris@87 489 } else if (name == "Window Size") {
Chris@0 490 setWindowSize(32 << value);
Chris@97 491 } else if (name == "Window Increment") {
Chris@97 492 setWindowHopLevel(value);
Chris@109 493 } else if (name == "Zero Padding") {
Chris@109 494 setZeroPadLevel(value > 0.1 ? 3 : 0);
Chris@87 495 } else if (name == "Min Frequency") {
Chris@37 496 switch (value) {
Chris@37 497 default:
Chris@37 498 case 0: setMinFrequency(0); break;
Chris@37 499 case 1: setMinFrequency(10); break;
Chris@37 500 case 2: setMinFrequency(20); break;
Chris@37 501 case 3: setMinFrequency(40); break;
Chris@37 502 case 4: setMinFrequency(100); break;
Chris@37 503 case 5: setMinFrequency(250); break;
Chris@37 504 case 6: setMinFrequency(500); break;
Chris@37 505 case 7: setMinFrequency(1000); break;
Chris@37 506 case 8: setMinFrequency(4000); break;
Chris@37 507 case 9: setMinFrequency(10000); break;
Chris@37 508 }
Chris@87 509 } else if (name == "Max Frequency") {
Chris@0 510 switch (value) {
Chris@0 511 case 0: setMaxFrequency(500); break;
Chris@0 512 case 1: setMaxFrequency(1000); break;
Chris@0 513 case 2: setMaxFrequency(1500); break;
Chris@0 514 case 3: setMaxFrequency(2000); break;
Chris@0 515 case 4: setMaxFrequency(4000); break;
Chris@0 516 case 5: setMaxFrequency(6000); break;
Chris@0 517 case 6: setMaxFrequency(8000); break;
Chris@0 518 case 7: setMaxFrequency(12000); break;
Chris@0 519 case 8: setMaxFrequency(16000); break;
Chris@0 520 default:
Chris@0 521 case 9: setMaxFrequency(0); break;
Chris@0 522 }
Chris@87 523 } else if (name == "Colour Scale") {
Chris@0 524 switch (value) {
Chris@0 525 default:
Chris@0 526 case 0: setColourScale(LinearColourScale); break;
Chris@0 527 case 1: setColourScale(MeterColourScale); break;
Chris@0 528 case 2: setColourScale(dBColourScale); break;
Chris@0 529 case 3: setColourScale(PhaseColourScale); break;
Chris@0 530 }
Chris@87 531 } else if (name == "Frequency Scale") {
Chris@0 532 switch (value) {
Chris@0 533 default:
Chris@0 534 case 0: setFrequencyScale(LinearFrequencyScale); break;
Chris@0 535 case 1: setFrequencyScale(LogFrequencyScale); break;
Chris@0 536 }
Chris@87 537 } else if (name == "Bin Display") {
Chris@35 538 switch (value) {
Chris@35 539 default:
Chris@37 540 case 0: setBinDisplay(AllBins); break;
Chris@37 541 case 1: setBinDisplay(PeakBins); break;
Chris@37 542 case 2: setBinDisplay(PeakFrequencies); break;
Chris@35 543 }
Chris@82 544 } else if (name == "Normalize Columns") {
Chris@36 545 setNormalizeColumns(value ? true : false);
Chris@0 546 }
Chris@0 547 }
Chris@0 548
Chris@0 549 void
Chris@95 550 SpectrogramLayer::invalidatePixmapCaches()
Chris@95 551 {
Chris@95 552 for (ViewPixmapCache::iterator i = m_pixmapCaches.begin();
Chris@95 553 i != m_pixmapCaches.end(); ++i) {
Chris@95 554 i->second.validArea = QRect();
Chris@95 555 }
Chris@95 556 }
Chris@95 557
Chris@95 558 void
Chris@95 559 SpectrogramLayer::invalidatePixmapCaches(size_t startFrame, size_t endFrame)
Chris@95 560 {
Chris@95 561 for (ViewPixmapCache::iterator i = m_pixmapCaches.begin();
Chris@95 562 i != m_pixmapCaches.end(); ++i) {
Chris@95 563 //!!! when are views removed from the map? on setLayerDormant?
Chris@95 564 const View *v = i->first;
Chris@95 565
Chris@95 566 if (startFrame < v->getEndFrame() && endFrame >= v->getStartFrame()) {
Chris@95 567 i->second.validArea = QRect();
Chris@95 568 }
Chris@95 569 }
Chris@95 570 }
Chris@95 571
Chris@95 572 void
Chris@0 573 SpectrogramLayer::setChannel(int ch)
Chris@0 574 {
Chris@0 575 if (m_channel == ch) return;
Chris@0 576
Chris@95 577 invalidatePixmapCaches();
Chris@0 578 m_channel = ch;
Chris@110 579 getFFTServer();
Chris@9 580
Chris@0 581 emit layerParametersChanged();
Chris@0 582 }
Chris@0 583
Chris@0 584 int
Chris@0 585 SpectrogramLayer::getChannel() const
Chris@0 586 {
Chris@0 587 return m_channel;
Chris@0 588 }
Chris@0 589
Chris@0 590 void
Chris@0 591 SpectrogramLayer::setWindowSize(size_t ws)
Chris@0 592 {
Chris@0 593 if (m_windowSize == ws) return;
Chris@0 594
Chris@95 595 invalidatePixmapCaches();
Chris@0 596
Chris@0 597 m_windowSize = ws;
Chris@109 598 m_fftSize = ws * (m_zeroPadLevel + 1);
Chris@0 599
Chris@110 600 getFFTServer();
Chris@9 601
Chris@9 602 emit layerParametersChanged();
Chris@0 603 }
Chris@0 604
Chris@0 605 size_t
Chris@0 606 SpectrogramLayer::getWindowSize() const
Chris@0 607 {
Chris@0 608 return m_windowSize;
Chris@0 609 }
Chris@0 610
Chris@0 611 void
Chris@97 612 SpectrogramLayer::setWindowHopLevel(size_t v)
Chris@0 613 {
Chris@97 614 if (m_windowHopLevel == v) return;
Chris@0 615
Chris@95 616 invalidatePixmapCaches();
Chris@0 617
Chris@97 618 m_windowHopLevel = v;
Chris@0 619
Chris@110 620 getFFTServer();
Chris@9 621
Chris@9 622 emit layerParametersChanged();
Chris@9 623
Chris@110 624 // fillCache();
Chris@0 625 }
Chris@0 626
Chris@0 627 size_t
Chris@97 628 SpectrogramLayer::getWindowHopLevel() const
Chris@0 629 {
Chris@97 630 return m_windowHopLevel;
Chris@0 631 }
Chris@0 632
Chris@0 633 void
Chris@109 634 SpectrogramLayer::setZeroPadLevel(size_t v)
Chris@109 635 {
Chris@109 636 if (m_zeroPadLevel == v) return;
Chris@109 637
Chris@109 638 invalidatePixmapCaches();
Chris@109 639
Chris@109 640 m_zeroPadLevel = v;
Chris@109 641 m_fftSize = m_windowSize * (v + 1);
Chris@110 642
Chris@110 643 getFFTServer();
Chris@109 644
Chris@109 645 emit layerParametersChanged();
Chris@109 646 }
Chris@109 647
Chris@109 648 size_t
Chris@109 649 SpectrogramLayer::getZeroPadLevel() const
Chris@109 650 {
Chris@109 651 return m_zeroPadLevel;
Chris@109 652 }
Chris@109 653
Chris@109 654 void
Chris@0 655 SpectrogramLayer::setWindowType(WindowType w)
Chris@0 656 {
Chris@0 657 if (m_windowType == w) return;
Chris@0 658
Chris@95 659 invalidatePixmapCaches();
Chris@0 660
Chris@0 661 m_windowType = w;
Chris@110 662
Chris@110 663 getFFTServer();
Chris@9 664
Chris@9 665 emit layerParametersChanged();
Chris@0 666 }
Chris@0 667
Chris@0 668 WindowType
Chris@0 669 SpectrogramLayer::getWindowType() const
Chris@0 670 {
Chris@0 671 return m_windowType;
Chris@0 672 }
Chris@0 673
Chris@0 674 void
Chris@0 675 SpectrogramLayer::setGain(float gain)
Chris@0 676 {
Chris@101 677 // std::cerr << "SpectrogramLayer::setGain(" << gain << ") (my gain is now "
Chris@101 678 // << m_gain << ")" << std::endl;
Chris@55 679
Chris@40 680 if (m_gain == gain) return;
Chris@0 681
Chris@95 682 invalidatePixmapCaches();
Chris@0 683
Chris@0 684 m_gain = gain;
Chris@0 685
Chris@9 686 emit layerParametersChanged();
Chris@0 687 }
Chris@0 688
Chris@0 689 float
Chris@0 690 SpectrogramLayer::getGain() const
Chris@0 691 {
Chris@0 692 return m_gain;
Chris@0 693 }
Chris@0 694
Chris@0 695 void
Chris@37 696 SpectrogramLayer::setThreshold(float threshold)
Chris@37 697 {
Chris@40 698 if (m_threshold == threshold) return;
Chris@37 699
Chris@95 700 invalidatePixmapCaches();
Chris@37 701
Chris@37 702 m_threshold = threshold;
Chris@37 703
Chris@37 704 emit layerParametersChanged();
Chris@37 705 }
Chris@37 706
Chris@37 707 float
Chris@37 708 SpectrogramLayer::getThreshold() const
Chris@37 709 {
Chris@37 710 return m_threshold;
Chris@37 711 }
Chris@37 712
Chris@37 713 void
Chris@37 714 SpectrogramLayer::setMinFrequency(size_t mf)
Chris@37 715 {
Chris@37 716 if (m_minFrequency == mf) return;
Chris@37 717
Chris@95 718 invalidatePixmapCaches();
Chris@37 719
Chris@37 720 m_minFrequency = mf;
Chris@37 721
Chris@37 722 emit layerParametersChanged();
Chris@37 723 }
Chris@37 724
Chris@37 725 size_t
Chris@37 726 SpectrogramLayer::getMinFrequency() const
Chris@37 727 {
Chris@37 728 return m_minFrequency;
Chris@37 729 }
Chris@37 730
Chris@37 731 void
Chris@0 732 SpectrogramLayer::setMaxFrequency(size_t mf)
Chris@0 733 {
Chris@0 734 if (m_maxFrequency == mf) return;
Chris@0 735
Chris@95 736 invalidatePixmapCaches();
Chris@0 737
Chris@0 738 m_maxFrequency = mf;
Chris@0 739
Chris@9 740 emit layerParametersChanged();
Chris@0 741 }
Chris@0 742
Chris@0 743 size_t
Chris@0 744 SpectrogramLayer::getMaxFrequency() const
Chris@0 745 {
Chris@0 746 return m_maxFrequency;
Chris@0 747 }
Chris@0 748
Chris@0 749 void
Chris@9 750 SpectrogramLayer::setColourRotation(int r)
Chris@9 751 {
Chris@95 752 invalidatePixmapCaches();
Chris@9 753
Chris@9 754 if (r < 0) r = 0;
Chris@9 755 if (r > 256) r = 256;
Chris@9 756 int distance = r - m_colourRotation;
Chris@9 757
Chris@9 758 if (distance != 0) {
Chris@90 759 rotateColourmap(-distance);
Chris@9 760 m_colourRotation = r;
Chris@9 761 }
Chris@9 762
Chris@9 763 emit layerParametersChanged();
Chris@9 764 }
Chris@9 765
Chris@9 766 void
Chris@0 767 SpectrogramLayer::setColourScale(ColourScale colourScale)
Chris@0 768 {
Chris@0 769 if (m_colourScale == colourScale) return;
Chris@0 770
Chris@95 771 invalidatePixmapCaches();
Chris@0 772
Chris@0 773 m_colourScale = colourScale;
Chris@0 774
Chris@9 775 emit layerParametersChanged();
Chris@0 776 }
Chris@0 777
Chris@0 778 SpectrogramLayer::ColourScale
Chris@0 779 SpectrogramLayer::getColourScale() const
Chris@0 780 {
Chris@0 781 return m_colourScale;
Chris@0 782 }
Chris@0 783
Chris@0 784 void
Chris@0 785 SpectrogramLayer::setColourScheme(ColourScheme scheme)
Chris@0 786 {
Chris@0 787 if (m_colourScheme == scheme) return;
Chris@0 788
Chris@95 789 invalidatePixmapCaches();
Chris@0 790
Chris@0 791 m_colourScheme = scheme;
Chris@90 792 setColourmap();
Chris@9 793
Chris@0 794 emit layerParametersChanged();
Chris@0 795 }
Chris@0 796
Chris@0 797 SpectrogramLayer::ColourScheme
Chris@0 798 SpectrogramLayer::getColourScheme() const
Chris@0 799 {
Chris@0 800 return m_colourScheme;
Chris@0 801 }
Chris@0 802
Chris@0 803 void
Chris@0 804 SpectrogramLayer::setFrequencyScale(FrequencyScale frequencyScale)
Chris@0 805 {
Chris@0 806 if (m_frequencyScale == frequencyScale) return;
Chris@0 807
Chris@95 808 invalidatePixmapCaches();
Chris@0 809 m_frequencyScale = frequencyScale;
Chris@9 810
Chris@9 811 emit layerParametersChanged();
Chris@0 812 }
Chris@0 813
Chris@0 814 SpectrogramLayer::FrequencyScale
Chris@0 815 SpectrogramLayer::getFrequencyScale() const
Chris@0 816 {
Chris@0 817 return m_frequencyScale;
Chris@0 818 }
Chris@0 819
Chris@0 820 void
Chris@37 821 SpectrogramLayer::setBinDisplay(BinDisplay binDisplay)
Chris@35 822 {
Chris@37 823 if (m_binDisplay == binDisplay) return;
Chris@35 824
Chris@95 825 invalidatePixmapCaches();
Chris@37 826 m_binDisplay = binDisplay;
Chris@35 827
Chris@35 828 emit layerParametersChanged();
Chris@35 829 }
Chris@35 830
Chris@37 831 SpectrogramLayer::BinDisplay
Chris@37 832 SpectrogramLayer::getBinDisplay() const
Chris@35 833 {
Chris@37 834 return m_binDisplay;
Chris@35 835 }
Chris@35 836
Chris@35 837 void
Chris@36 838 SpectrogramLayer::setNormalizeColumns(bool n)
Chris@36 839 {
Chris@36 840 if (m_normalizeColumns == n) return;
Chris@36 841
Chris@95 842 invalidatePixmapCaches();
Chris@36 843 m_normalizeColumns = n;
Chris@36 844
Chris@36 845 emit layerParametersChanged();
Chris@36 846 }
Chris@36 847
Chris@36 848 bool
Chris@36 849 SpectrogramLayer::getNormalizeColumns() const
Chris@36 850 {
Chris@36 851 return m_normalizeColumns;
Chris@36 852 }
Chris@36 853
Chris@36 854 void
Chris@47 855 SpectrogramLayer::setLayerDormant(const View *v, bool dormant)
Chris@29 856 {
Chris@47 857 if (dormant == m_dormancy[v]) return;
Chris@33 858
Chris@33 859 if (dormant) {
Chris@33 860
Chris@47 861 m_dormancy[v] = true;
Chris@33 862
Chris@95 863 invalidatePixmapCaches();
Chris@95 864 m_pixmapCaches.erase(v);
Chris@33 865
Chris@33 866 } else {
Chris@33 867
Chris@47 868 m_dormancy[v] = false;
Chris@33 869 }
Chris@29 870 }
Chris@29 871
Chris@29 872 void
Chris@0 873 SpectrogramLayer::cacheInvalid()
Chris@0 874 {
Chris@95 875 invalidatePixmapCaches();
Chris@0 876 }
Chris@0 877
Chris@0 878 void
Chris@0 879 SpectrogramLayer::cacheInvalid(size_t, size_t)
Chris@0 880 {
Chris@0 881 // for now (or forever?)
Chris@0 882 cacheInvalid();
Chris@0 883 }
Chris@0 884
Chris@0 885 void
Chris@0 886 SpectrogramLayer::fillTimerTimedOut()
Chris@0 887 {
Chris@110 888 if (m_fftServer && m_model) {
Chris@110 889
Chris@110 890 size_t fillExtent = m_fftServer->getFillExtent();
Chris@0 891 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 892 std::cerr << "SpectrogramLayer::fillTimerTimedOut: extent " << fillExtent << ", last " << m_lastFillExtent << ", total " << m_model->getEndFrame() << std::endl;
Chris@0 893 #endif
Chris@0 894 if (fillExtent >= m_lastFillExtent) {
Chris@0 895 if (fillExtent >= m_model->getEndFrame() && m_lastFillExtent > 0) {
Chris@0 896 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 897 std::cerr << "complete!" << std::endl;
Chris@0 898 #endif
Chris@95 899 invalidatePixmapCaches();
Chris@0 900 emit modelChanged();
Chris@0 901 delete m_updateTimer;
Chris@0 902 m_updateTimer = 0;
Chris@0 903 m_lastFillExtent = 0;
Chris@0 904 } else if (fillExtent > m_lastFillExtent) {
Chris@0 905 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 906 std::cerr << "SpectrogramLayer: emitting modelChanged("
Chris@0 907 << m_lastFillExtent << "," << fillExtent << ")" << std::endl;
Chris@0 908 #endif
Chris@95 909 invalidatePixmapCaches(m_lastFillExtent, fillExtent);
Chris@0 910 emit modelChanged(m_lastFillExtent, fillExtent);
Chris@0 911 m_lastFillExtent = fillExtent;
Chris@0 912 }
Chris@0 913 } else {
Chris@0 914 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@110 915 std::cerr << "SpectrogramLayer: going backwards, emitting modelChanged("
Chris@110 916 << m_model->getStartFrame() << "," << m_model->getEndFrame() << ")" << std::endl;
Chris@0 917 #endif
Chris@110 918 invalidatePixmapCaches();
Chris@110 919 emit modelChanged(m_model->getStartFrame(), m_model->getEndFrame());
Chris@0 920 m_lastFillExtent = fillExtent;
Chris@0 921 }
Chris@0 922 }
Chris@0 923 }
Chris@0 924
Chris@0 925 void
Chris@90 926 SpectrogramLayer::setColourmap()
Chris@0 927 {
Chris@10 928 int formerRotation = m_colourRotation;
Chris@10 929
Chris@38 930 if (m_colourScheme == BlackOnWhite) {
Chris@86 931 m_colourMap.setColour(NO_VALUE, Qt::white);
Chris@38 932 } else {
Chris@86 933 m_colourMap.setColour(NO_VALUE, Qt::black);
Chris@38 934 }
Chris@0 935
Chris@0 936 for (int pixel = 1; pixel < 256; ++pixel) {
Chris@0 937
Chris@0 938 QColor colour;
Chris@0 939 int hue, px;
Chris@0 940
Chris@0 941 switch (m_colourScheme) {
Chris@0 942
Chris@0 943 default:
Chris@0 944 case DefaultColours:
Chris@0 945 hue = 256 - pixel;
Chris@0 946 colour = QColor::fromHsv(hue, pixel/2 + 128, pixel);
Chris@77 947 m_crosshairColour = QColor(255, 150, 50);
Chris@77 948 // m_crosshairColour = QColor::fromHsv(240, 160, 255);
Chris@0 949 break;
Chris@0 950
Chris@0 951 case WhiteOnBlack:
Chris@0 952 colour = QColor(pixel, pixel, pixel);
Chris@77 953 m_crosshairColour = Qt::red;
Chris@0 954 break;
Chris@0 955
Chris@0 956 case BlackOnWhite:
Chris@0 957 colour = QColor(256-pixel, 256-pixel, 256-pixel);
Chris@77 958 m_crosshairColour = Qt::darkGreen;
Chris@0 959 break;
Chris@0 960
Chris@0 961 case RedOnBlue:
Chris@0 962 colour = QColor(pixel > 128 ? (pixel - 128) * 2 : 0, 0,
Chris@0 963 pixel < 128 ? pixel : (256 - pixel));
Chris@77 964 m_crosshairColour = Qt::green;
Chris@0 965 break;
Chris@0 966
Chris@0 967 case YellowOnBlack:
Chris@0 968 px = 256 - pixel;
Chris@0 969 colour = QColor(px < 64 ? 255 - px/2 :
Chris@0 970 px < 128 ? 224 - (px - 64) :
Chris@0 971 px < 192 ? 160 - (px - 128) * 3 / 2 :
Chris@0 972 256 - px,
Chris@0 973 pixel,
Chris@0 974 pixel / 4);
Chris@77 975 m_crosshairColour = QColor::fromHsv(240, 255, 255);
Chris@0 976 break;
Chris@0 977
Chris@71 978 case BlueOnBlack:
Chris@71 979 colour = QColor::fromHsv
Chris@71 980 (240, pixel > 226 ? 256 - (pixel - 226) * 8 : 255,
Chris@71 981 (pixel * pixel) / 255);
Chris@77 982 m_crosshairColour = Qt::red;
Chris@71 983 break;
Chris@71 984
Chris@40 985 case Rainbow:
Chris@40 986 hue = 250 - pixel;
Chris@40 987 if (hue < 0) hue += 256;
Chris@40 988 colour = QColor::fromHsv(pixel, 255, 255);
Chris@77 989 m_crosshairColour = Qt::white;
Chris@0 990 break;
Chris@0 991 }
Chris@0 992
Chris@86 993 m_colourMap.setColour(pixel, colour);
Chris@0 994 }
Chris@9 995
Chris@9 996 m_colourRotation = 0;
Chris@90 997 rotateColourmap(m_colourRotation - formerRotation);
Chris@10 998 m_colourRotation = formerRotation;
Chris@9 999 }
Chris@9 1000
Chris@9 1001 void
Chris@90 1002 SpectrogramLayer::rotateColourmap(int distance)
Chris@9 1003 {
Chris@110 1004 if (!m_fftServer) return;
Chris@10 1005
Chris@31 1006 QColor newPixels[256];
Chris@9 1007
Chris@86 1008 newPixels[NO_VALUE] = m_colourMap.getColour(NO_VALUE);
Chris@9 1009
Chris@9 1010 for (int pixel = 1; pixel < 256; ++pixel) {
Chris@9 1011 int target = pixel + distance;
Chris@9 1012 while (target < 1) target += 255;
Chris@9 1013 while (target > 255) target -= 255;
Chris@86 1014 newPixels[target] = m_colourMap.getColour(pixel);
Chris@9 1015 }
Chris@9 1016
Chris@9 1017 for (int pixel = 0; pixel < 256; ++pixel) {
Chris@86 1018 m_colourMap.setColour(pixel, newPixels[pixel]);
Chris@9 1019 }
Chris@0 1020 }
Chris@0 1021
Chris@38 1022 float
Chris@38 1023 SpectrogramLayer::calculateFrequency(size_t bin,
Chris@38 1024 size_t windowSize,
Chris@38 1025 size_t windowIncrement,
Chris@38 1026 size_t sampleRate,
Chris@38 1027 float oldPhase,
Chris@38 1028 float newPhase,
Chris@38 1029 bool &steadyState)
Chris@38 1030 {
Chris@38 1031 // At frequency f, phase shift of 2pi (one cycle) happens in 1/f sec.
Chris@38 1032 // At hopsize h and sample rate sr, one hop happens in h/sr sec.
Chris@38 1033 // At window size w, for bin b, f is b*sr/w.
Chris@38 1034 // thus 2pi phase shift happens in w/(b*sr) sec.
Chris@38 1035 // We need to know what phase shift we expect from h/sr sec.
Chris@38 1036 // -> 2pi * ((h/sr) / (w/(b*sr)))
Chris@38 1037 // = 2pi * ((h * b * sr) / (w * sr))
Chris@38 1038 // = 2pi * (h * b) / w.
Chris@38 1039
Chris@38 1040 float frequency = (float(bin) * sampleRate) / windowSize;
Chris@38 1041
Chris@38 1042 float expectedPhase =
Chris@38 1043 oldPhase + (2.0 * M_PI * bin * windowIncrement) / windowSize;
Chris@38 1044
Chris@104 1045 float phaseError = princargf(newPhase - expectedPhase);
Chris@38 1046
Chris@38 1047 if (fabs(phaseError) < (1.1 * (windowIncrement * M_PI) / windowSize)) {
Chris@38 1048
Chris@38 1049 // The new frequency estimate based on the phase error
Chris@38 1050 // resulting from assuming the "native" frequency of this bin
Chris@38 1051
Chris@38 1052 float newFrequency =
Chris@38 1053 (sampleRate * (expectedPhase + phaseError - oldPhase)) /
Chris@38 1054 (2 * M_PI * windowIncrement);
Chris@38 1055
Chris@38 1056 steadyState = true;
Chris@38 1057 return newFrequency;
Chris@38 1058 }
Chris@38 1059
Chris@38 1060 steadyState = false;
Chris@38 1061 return frequency;
Chris@38 1062 }
Chris@38 1063
Chris@38 1064 unsigned char
Chris@38 1065 SpectrogramLayer::getDisplayValue(float input) const
Chris@38 1066 {
Chris@38 1067 int value;
Chris@37 1068
Chris@40 1069 switch (m_colourScale) {
Chris@40 1070
Chris@40 1071 default:
Chris@40 1072 case LinearColourScale:
Chris@40 1073 value = int
Chris@40 1074 (input * (m_normalizeColumns ? 1.0 : 50.0) * 255.0) + 1;
Chris@40 1075 break;
Chris@40 1076
Chris@40 1077 case MeterColourScale:
Chris@40 1078 value = AudioLevel::multiplier_to_preview
Chris@40 1079 (input * (m_normalizeColumns ? 1.0 : 50.0), 255) + 1;
Chris@40 1080 break;
Chris@40 1081
Chris@40 1082 case dBColourScale:
Chris@40 1083 input = 20.0 * log10(input);
Chris@40 1084 input = (input + 80.0) / 80.0;
Chris@40 1085 if (input < 0.0) input = 0.0;
Chris@40 1086 if (input > 1.0) input = 1.0;
Chris@40 1087 value = int(input * 255.0) + 1;
Chris@40 1088 break;
Chris@40 1089
Chris@40 1090 case PhaseColourScale:
Chris@40 1091 value = int((input * 127.0 / M_PI) + 128);
Chris@40 1092 break;
Chris@0 1093 }
Chris@38 1094
Chris@38 1095 if (value > UCHAR_MAX) value = UCHAR_MAX;
Chris@38 1096 if (value < 0) value = 0;
Chris@38 1097 return value;
Chris@0 1098 }
Chris@0 1099
Chris@40 1100 float
Chris@40 1101 SpectrogramLayer::getInputForDisplayValue(unsigned char uc) const
Chris@40 1102 {
Chris@40 1103 int value = uc;
Chris@40 1104 float input;
Chris@40 1105
Chris@40 1106 switch (m_colourScale) {
Chris@40 1107
Chris@40 1108 default:
Chris@40 1109 case LinearColourScale:
Chris@40 1110 input = float(value - 1) / 255.0 / (m_normalizeColumns ? 1 : 50);
Chris@40 1111 break;
Chris@40 1112
Chris@40 1113 case MeterColourScale:
Chris@40 1114 input = AudioLevel::preview_to_multiplier(value - 1, 255)
Chris@40 1115 / (m_normalizeColumns ? 1.0 : 50.0);
Chris@40 1116 break;
Chris@40 1117
Chris@40 1118 case dBColourScale:
Chris@40 1119 input = float(value - 1) / 255.0;
Chris@40 1120 input = (input * 80.0) - 80.0;
Chris@40 1121 input = powf(10.0, input) / 20.0;
Chris@40 1122 value = int(input);
Chris@40 1123 break;
Chris@40 1124
Chris@40 1125 case PhaseColourScale:
Chris@40 1126 input = float(value - 128) * M_PI / 127.0;
Chris@40 1127 break;
Chris@40 1128 }
Chris@40 1129
Chris@40 1130 return input;
Chris@40 1131 }
Chris@40 1132
Chris@40 1133 float
Chris@40 1134 SpectrogramLayer::getEffectiveMinFrequency() const
Chris@40 1135 {
Chris@40 1136 int sr = m_model->getSampleRate();
Chris@107 1137 float minf = float(sr) / m_fftSize;
Chris@40 1138
Chris@40 1139 if (m_minFrequency > 0.0) {
Chris@107 1140 size_t minbin = size_t((double(m_minFrequency) * m_fftSize) / sr + 0.01);
Chris@40 1141 if (minbin < 1) minbin = 1;
Chris@107 1142 minf = minbin * sr / m_fftSize;
Chris@40 1143 }
Chris@40 1144
Chris@40 1145 return minf;
Chris@40 1146 }
Chris@40 1147
Chris@40 1148 float
Chris@40 1149 SpectrogramLayer::getEffectiveMaxFrequency() const
Chris@40 1150 {
Chris@40 1151 int sr = m_model->getSampleRate();
Chris@40 1152 float maxf = float(sr) / 2;
Chris@40 1153
Chris@40 1154 if (m_maxFrequency > 0.0) {
Chris@107 1155 size_t maxbin = size_t((double(m_maxFrequency) * m_fftSize) / sr + 0.1);
Chris@107 1156 if (maxbin > m_fftSize / 2) maxbin = m_fftSize / 2;
Chris@107 1157 maxf = maxbin * sr / m_fftSize;
Chris@40 1158 }
Chris@40 1159
Chris@40 1160 return maxf;
Chris@40 1161 }
Chris@40 1162
Chris@0 1163 bool
Chris@44 1164 SpectrogramLayer::getYBinRange(View *v, int y, float &q0, float &q1) const
Chris@0 1165 {
Chris@44 1166 int h = v->height();
Chris@0 1167 if (y < 0 || y >= h) return false;
Chris@0 1168
Chris@38 1169 int sr = m_model->getSampleRate();
Chris@40 1170 float minf = getEffectiveMinFrequency();
Chris@40 1171 float maxf = getEffectiveMaxFrequency();
Chris@0 1172
Chris@38 1173 bool logarithmic = (m_frequencyScale == LogFrequencyScale);
Chris@38 1174
Chris@44 1175 q0 = v->getFrequencyForY(y, minf, maxf, logarithmic);
Chris@44 1176 q1 = v->getFrequencyForY(y - 1, minf, maxf, logarithmic);
Chris@38 1177
Chris@38 1178 // Now map these on to actual bins
Chris@38 1179
Chris@107 1180 int b0 = int((q0 * m_fftSize) / sr);
Chris@107 1181 int b1 = int((q1 * m_fftSize) / sr);
Chris@0 1182
Chris@40 1183 //!!! this is supposed to return fractions-of-bins, as it were, hence the floats
Chris@38 1184 q0 = b0;
Chris@38 1185 q1 = b1;
Chris@38 1186
Chris@107 1187 // q0 = (b0 * sr) / m_fftSize;
Chris@107 1188 // q1 = (b1 * sr) / m_fftSize;
Chris@0 1189
Chris@0 1190 return true;
Chris@0 1191 }
Chris@38 1192
Chris@0 1193 bool
Chris@44 1194 SpectrogramLayer::getXBinRange(View *v, int x, float &s0, float &s1) const
Chris@0 1195 {
Chris@21 1196 size_t modelStart = m_model->getStartFrame();
Chris@21 1197 size_t modelEnd = m_model->getEndFrame();
Chris@0 1198
Chris@0 1199 // Each pixel column covers an exact range of sample frames:
Chris@44 1200 int f0 = v->getFrameForX(x) - modelStart;
Chris@44 1201 int f1 = v->getFrameForX(x + 1) - modelStart - 1;
Chris@20 1202
Chris@41 1203 if (f1 < int(modelStart) || f0 > int(modelEnd)) {
Chris@41 1204 return false;
Chris@41 1205 }
Chris@20 1206
Chris@0 1207 // And that range may be drawn from a possibly non-integral
Chris@0 1208 // range of spectrogram windows:
Chris@0 1209
Chris@0 1210 size_t windowIncrement = getWindowIncrement();
Chris@0 1211 s0 = float(f0) / windowIncrement;
Chris@0 1212 s1 = float(f1) / windowIncrement;
Chris@0 1213
Chris@0 1214 return true;
Chris@0 1215 }
Chris@0 1216
Chris@0 1217 bool
Chris@44 1218 SpectrogramLayer::getXBinSourceRange(View *v, int x, RealTime &min, RealTime &max) const
Chris@0 1219 {
Chris@0 1220 float s0 = 0, s1 = 0;
Chris@44 1221 if (!getXBinRange(v, x, s0, s1)) return false;
Chris@0 1222
Chris@0 1223 int s0i = int(s0 + 0.001);
Chris@0 1224 int s1i = int(s1);
Chris@0 1225
Chris@0 1226 int windowIncrement = getWindowIncrement();
Chris@0 1227 int w0 = s0i * windowIncrement - (m_windowSize - windowIncrement)/2;
Chris@0 1228 int w1 = s1i * windowIncrement + windowIncrement +
Chris@0 1229 (m_windowSize - windowIncrement)/2 - 1;
Chris@0 1230
Chris@0 1231 min = RealTime::frame2RealTime(w0, m_model->getSampleRate());
Chris@0 1232 max = RealTime::frame2RealTime(w1, m_model->getSampleRate());
Chris@0 1233 return true;
Chris@0 1234 }
Chris@0 1235
Chris@0 1236 bool
Chris@44 1237 SpectrogramLayer::getYBinSourceRange(View *v, int y, float &freqMin, float &freqMax)
Chris@0 1238 const
Chris@0 1239 {
Chris@0 1240 float q0 = 0, q1 = 0;
Chris@44 1241 if (!getYBinRange(v, y, q0, q1)) return false;
Chris@0 1242
Chris@0 1243 int q0i = int(q0 + 0.001);
Chris@0 1244 int q1i = int(q1);
Chris@0 1245
Chris@0 1246 int sr = m_model->getSampleRate();
Chris@0 1247
Chris@0 1248 for (int q = q0i; q <= q1i; ++q) {
Chris@107 1249 int binfreq = (sr * q) / m_fftSize;
Chris@0 1250 if (q == q0i) freqMin = binfreq;
Chris@0 1251 if (q == q1i) freqMax = binfreq;
Chris@0 1252 }
Chris@0 1253 return true;
Chris@0 1254 }
Chris@35 1255
Chris@35 1256 bool
Chris@44 1257 SpectrogramLayer::getAdjustedYBinSourceRange(View *v, int x, int y,
Chris@35 1258 float &freqMin, float &freqMax,
Chris@35 1259 float &adjFreqMin, float &adjFreqMax)
Chris@35 1260 const
Chris@35 1261 {
Chris@110 1262 if (!m_fftServer) return false;
Chris@110 1263
Chris@35 1264 float s0 = 0, s1 = 0;
Chris@44 1265 if (!getXBinRange(v, x, s0, s1)) return false;
Chris@35 1266
Chris@35 1267 float q0 = 0, q1 = 0;
Chris@44 1268 if (!getYBinRange(v, y, q0, q1)) return false;
Chris@35 1269
Chris@35 1270 int s0i = int(s0 + 0.001);
Chris@35 1271 int s1i = int(s1);
Chris@35 1272
Chris@35 1273 int q0i = int(q0 + 0.001);
Chris@35 1274 int q1i = int(q1);
Chris@35 1275
Chris@35 1276 int sr = m_model->getSampleRate();
Chris@35 1277
Chris@38 1278 size_t windowSize = m_windowSize;
Chris@38 1279 size_t windowIncrement = getWindowIncrement();
Chris@38 1280
Chris@35 1281 bool haveAdj = false;
Chris@35 1282
Chris@37 1283 bool peaksOnly = (m_binDisplay == PeakBins ||
Chris@37 1284 m_binDisplay == PeakFrequencies);
Chris@37 1285
Chris@35 1286 for (int q = q0i; q <= q1i; ++q) {
Chris@35 1287
Chris@35 1288 for (int s = s0i; s <= s1i; ++s) {
Chris@35 1289
Chris@35 1290 float binfreq = (sr * q) / m_windowSize;
Chris@35 1291 if (q == q0i) freqMin = binfreq;
Chris@35 1292 if (q == q1i) freqMax = binfreq;
Chris@37 1293
Chris@110 1294 if (peaksOnly && !m_fftServer->isLocalPeak(s, q)) continue;
Chris@38 1295
Chris@110 1296 if (!m_fftServer->isOverThreshold(s, q, m_threshold)) continue;
Chris@38 1297
Chris@38 1298 float freq = binfreq;
Chris@38 1299 bool steady = false;
Chris@40 1300
Chris@110 1301 if (s < int(m_fftServer->getWidth()) - 1) {
Chris@38 1302
Chris@38 1303 freq = calculateFrequency(q,
Chris@38 1304 windowSize,
Chris@38 1305 windowIncrement,
Chris@38 1306 sr,
Chris@110 1307 m_fftServer->getPhaseAt(s, q),
Chris@110 1308 m_fftServer->getPhaseAt(s+1, q),
Chris@38 1309 steady);
Chris@35 1310
Chris@38 1311 if (!haveAdj || freq < adjFreqMin) adjFreqMin = freq;
Chris@38 1312 if (!haveAdj || freq > adjFreqMax) adjFreqMax = freq;
Chris@35 1313
Chris@35 1314 haveAdj = true;
Chris@35 1315 }
Chris@35 1316 }
Chris@35 1317 }
Chris@35 1318
Chris@35 1319 if (!haveAdj) {
Chris@40 1320 adjFreqMin = adjFreqMax = 0.0;
Chris@35 1321 }
Chris@35 1322
Chris@35 1323 return haveAdj;
Chris@35 1324 }
Chris@0 1325
Chris@0 1326 bool
Chris@44 1327 SpectrogramLayer::getXYBinSourceRange(View *v, int x, int y,
Chris@38 1328 float &min, float &max,
Chris@38 1329 float &phaseMin, float &phaseMax) const
Chris@0 1330 {
Chris@0 1331 float q0 = 0, q1 = 0;
Chris@44 1332 if (!getYBinRange(v, y, q0, q1)) return false;
Chris@0 1333
Chris@0 1334 float s0 = 0, s1 = 0;
Chris@44 1335 if (!getXBinRange(v, x, s0, s1)) return false;
Chris@0 1336
Chris@0 1337 int q0i = int(q0 + 0.001);
Chris@0 1338 int q1i = int(q1);
Chris@0 1339
Chris@0 1340 int s0i = int(s0 + 0.001);
Chris@0 1341 int s1i = int(s1);
Chris@0 1342
Chris@37 1343 bool rv = false;
Chris@37 1344
Chris@110 1345 if (m_fftServer) {
Chris@0 1346
Chris@110 1347 int cw = m_fftServer->getWidth();
Chris@110 1348 int ch = m_fftServer->getHeight();
Chris@0 1349
Chris@110 1350 min = 0.0;
Chris@110 1351 max = 0.0;
Chris@110 1352 phaseMin = 0.0;
Chris@110 1353 phaseMax = 0.0;
Chris@110 1354 bool have = false;
Chris@0 1355
Chris@110 1356 for (int q = q0i; q <= q1i; ++q) {
Chris@110 1357 for (int s = s0i; s <= s1i; ++s) {
Chris@110 1358 if (s >= 0 && q >= 0 && s < cw && q < ch) {
Chris@110 1359
Chris@110 1360 float value;
Chris@38 1361
Chris@110 1362 value = m_fftServer->getPhaseAt(s, q);
Chris@110 1363 if (!have || value < phaseMin) { phaseMin = value; }
Chris@110 1364 if (!have || value > phaseMax) { phaseMax = value; }
Chris@91 1365
Chris@110 1366 value = m_fftServer->getMagnitudeAt(s, q);
Chris@110 1367 if (!have || value < min) { min = value; }
Chris@110 1368 if (!have || value > max) { max = value; }
Chris@110 1369
Chris@110 1370 have = true;
Chris@110 1371 }
Chris@110 1372 }
Chris@110 1373 }
Chris@110 1374
Chris@110 1375 if (have) {
Chris@110 1376 rv = true;
Chris@110 1377 }
Chris@0 1378 }
Chris@0 1379
Chris@37 1380 return rv;
Chris@0 1381 }
Chris@0 1382
Chris@0 1383 void
Chris@44 1384 SpectrogramLayer::paint(View *v, QPainter &paint, QRect rect) const
Chris@0 1385 {
Chris@55 1386 if (m_colourScheme == BlackOnWhite) {
Chris@55 1387 v->setLightBackground(true);
Chris@55 1388 } else {
Chris@55 1389 v->setLightBackground(false);
Chris@55 1390 }
Chris@55 1391
Chris@0 1392 // Profiler profiler("SpectrogramLayer::paint", true);
Chris@0 1393 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1394 std::cerr << "SpectrogramLayer::paint(): m_model is " << m_model << ", zoom level is " << v->getZoomLevel() << ", m_updateTimer " << m_updateTimer << std::endl;
Chris@95 1395
Chris@95 1396 std::cerr << "rect is " << rect.x() << "," << rect.y() << " " << rect.width() << "x" << rect.height() << std::endl;
Chris@0 1397 #endif
Chris@95 1398
Chris@45 1399 long sf = v->getStartFrame();
Chris@45 1400 if (sf < 0) m_candidateFillStartFrame = 0;
Chris@45 1401 else m_candidateFillStartFrame = sf;
Chris@44 1402
Chris@0 1403 if (!m_model || !m_model->isOK() || !m_model->isReady()) {
Chris@0 1404 return;
Chris@0 1405 }
Chris@0 1406
Chris@47 1407 if (isLayerDormant(v)) {
Chris@48 1408 std::cerr << "SpectrogramLayer::paint(): Layer is dormant, making it undormant again" << std::endl;
Chris@29 1409 }
Chris@29 1410
Chris@48 1411 // Need to do this even if !isLayerDormant, as that could mean v
Chris@48 1412 // is not in the dormancy map at all -- we need it to be present
Chris@48 1413 // and accountable for when determining whether we need the cache
Chris@48 1414 // in the cache-fill thread above.
Chris@48 1415 m_dormancy[v] = false;
Chris@48 1416
Chris@110 1417 if (!m_fftServer) { // lock the mutex before checking this
Chris@0 1418 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@110 1419 std::cerr << "SpectrogramLayer::paint(): No FFT server, returning" << std::endl;
Chris@0 1420 #endif
Chris@0 1421 return;
Chris@0 1422 }
Chris@0 1423
Chris@95 1424 PixmapCache &cache = m_pixmapCaches[v];
Chris@95 1425
Chris@95 1426 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1427 std::cerr << "SpectrogramLayer::paint(): pixmap cache valid area " << cache.validArea.x() << ", " << cache.validArea.y() << ", " << cache.validArea.width() << "x" << cache.validArea.height() << std::endl;
Chris@95 1428 #endif
Chris@95 1429
Chris@0 1430 bool stillCacheing = (m_updateTimer != 0);
Chris@0 1431
Chris@0 1432 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 1433 std::cerr << "SpectrogramLayer::paint(): Still cacheing = " << stillCacheing << std::endl;
Chris@0 1434 #endif
Chris@0 1435
Chris@44 1436 long startFrame = v->getStartFrame();
Chris@44 1437 int zoomLevel = v->getZoomLevel();
Chris@0 1438
Chris@0 1439 int x0 = 0;
Chris@44 1440 int x1 = v->width();
Chris@0 1441 int y0 = 0;
Chris@44 1442 int y1 = v->height();
Chris@0 1443
Chris@0 1444 bool recreateWholePixmapCache = true;
Chris@0 1445
Chris@95 1446 x0 = rect.left();
Chris@95 1447 x1 = rect.right() + 1;
Chris@95 1448 y0 = rect.top();
Chris@95 1449 y1 = rect.bottom() + 1;
Chris@95 1450
Chris@95 1451 if (cache.validArea.width() > 0) {
Chris@95 1452
Chris@95 1453 if (int(cache.zoomLevel) == zoomLevel &&
Chris@95 1454 cache.pixmap.width() == v->width() &&
Chris@95 1455 cache.pixmap.height() == v->height()) {
Chris@95 1456
Chris@95 1457 if (v->getXForFrame(cache.startFrame) ==
Chris@95 1458 v->getXForFrame(startFrame) &&
Chris@95 1459 cache.validArea.x() <= x0 &&
Chris@95 1460 cache.validArea.x() + cache.validArea.width() >= x1) {
Chris@0 1461
Chris@0 1462 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 1463 std::cerr << "SpectrogramLayer: pixmap cache good" << std::endl;
Chris@0 1464 #endif
Chris@0 1465
Chris@95 1466 paint.drawPixmap(rect, cache.pixmap, rect);
Chris@0 1467 return;
Chris@0 1468
Chris@0 1469 } else {
Chris@0 1470
Chris@0 1471 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 1472 std::cerr << "SpectrogramLayer: pixmap cache partially OK" << std::endl;
Chris@0 1473 #endif
Chris@0 1474
Chris@0 1475 recreateWholePixmapCache = false;
Chris@0 1476
Chris@95 1477 int dx = v->getXForFrame(cache.startFrame) -
Chris@44 1478 v->getXForFrame(startFrame);
Chris@0 1479
Chris@0 1480 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1481 std::cerr << "SpectrogramLayer: dx = " << dx << " (pixmap cache " << cache.pixmap.width() << "x" << cache.pixmap.height() << ")" << std::endl;
Chris@0 1482 #endif
Chris@0 1483
Chris@95 1484 if (dx != 0 &&
Chris@95 1485 dx > -cache.pixmap.width() &&
Chris@95 1486 dx < cache.pixmap.width()) {
Chris@0 1487
Chris@0 1488 #if defined(Q_WS_WIN32) || defined(Q_WS_MAC)
Chris@0 1489 // Copying a pixmap to itself doesn't work
Chris@0 1490 // properly on Windows or Mac (it only works when
Chris@0 1491 // moving in one direction).
Chris@0 1492
Chris@0 1493 //!!! Need a utility function for this
Chris@0 1494
Chris@0 1495 static QPixmap *tmpPixmap = 0;
Chris@0 1496 if (!tmpPixmap ||
Chris@95 1497 tmpPixmap->width() != cache.pixmap.width() ||
Chris@95 1498 tmpPixmap->height() != cache.pixmap.height()) {
Chris@0 1499 delete tmpPixmap;
Chris@95 1500 tmpPixmap = new QPixmap(cache.pixmap.width(),
Chris@95 1501 cache.pixmap.height());
Chris@0 1502 }
Chris@0 1503 QPainter cachePainter;
Chris@0 1504 cachePainter.begin(tmpPixmap);
Chris@95 1505 cachePainter.drawPixmap(0, 0, cache.pixmap);
Chris@0 1506 cachePainter.end();
Chris@95 1507 cachePainter.begin(&cache.pixmap);
Chris@0 1508 cachePainter.drawPixmap(dx, 0, *tmpPixmap);
Chris@0 1509 cachePainter.end();
Chris@0 1510 #else
Chris@95 1511 QPainter cachePainter(&cache.pixmap);
Chris@95 1512 cachePainter.drawPixmap(dx, 0, cache.pixmap);
Chris@0 1513 cachePainter.end();
Chris@0 1514 #endif
Chris@0 1515
Chris@95 1516 int px = cache.validArea.x();
Chris@95 1517 int pw = cache.validArea.width();
Chris@0 1518
Chris@0 1519 if (dx < 0) {
Chris@95 1520 x0 = cache.pixmap.width() + dx;
Chris@95 1521 x1 = cache.pixmap.width();
Chris@95 1522 px += dx;
Chris@95 1523 if (px < 0) {
Chris@95 1524 pw += px;
Chris@95 1525 px = 0;
Chris@95 1526 if (pw < 0) pw = 0;
Chris@95 1527 }
Chris@0 1528 } else {
Chris@0 1529 x0 = 0;
Chris@0 1530 x1 = dx;
Chris@95 1531 px += dx;
Chris@95 1532 if (px + pw > cache.pixmap.width()) {
Chris@95 1533 pw = int(cache.pixmap.width()) - px;
Chris@95 1534 if (pw < 0) pw = 0;
Chris@95 1535 }
Chris@0 1536 }
Chris@95 1537
Chris@95 1538 cache.validArea =
Chris@95 1539 QRect(px, cache.validArea.y(),
Chris@95 1540 pw, cache.validArea.height());
Chris@95 1541
Chris@95 1542 paint.drawPixmap(rect & cache.validArea,
Chris@95 1543 cache.pixmap,
Chris@95 1544 rect & cache.validArea);
Chris@0 1545 }
Chris@0 1546 }
Chris@0 1547 } else {
Chris@0 1548 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 1549 std::cerr << "SpectrogramLayer: pixmap cache useless" << std::endl;
Chris@0 1550 #endif
Chris@95 1551 cache.validArea = QRect();
Chris@0 1552 }
Chris@0 1553 }
Chris@95 1554
Chris@92 1555 /*
Chris@0 1556 if (stillCacheing) {
Chris@0 1557 x0 = rect.left();
Chris@0 1558 x1 = rect.right() + 1;
Chris@0 1559 y0 = rect.top();
Chris@0 1560 y1 = rect.bottom() + 1;
Chris@0 1561 }
Chris@92 1562 */
Chris@95 1563
Chris@95 1564 if (recreateWholePixmapCache) {
Chris@95 1565 x0 = 0;
Chris@95 1566 x1 = v->width();
Chris@95 1567 }
Chris@95 1568
Chris@96 1569 int paintBlockWidth = (300000 / zoomLevel);
Chris@96 1570 if (paintBlockWidth < 20) paintBlockWidth = 20;
Chris@96 1571
Chris@96 1572 if (cache.validArea.width() > 0) {
Chris@96 1573
Chris@96 1574 int vx0 = 0, vx1 = 0;
Chris@96 1575 vx0 = cache.validArea.x();
Chris@96 1576 vx1 = cache.validArea.x() + cache.validArea.width();
Chris@96 1577
Chris@96 1578 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@96 1579 std::cerr << "x0 " << x0 << ", x1 " << x1 << ", vx0 " << vx0 << ", vx1 " << vx1 << ", paintBlockWidth " << paintBlockWidth << std::endl;
Chris@96 1580 #endif
Chris@96 1581 if (x0 < vx0) {
Chris@96 1582 if (x0 + paintBlockWidth < vx0) {
Chris@96 1583 x0 = vx0 - paintBlockWidth;
Chris@96 1584 } else {
Chris@96 1585 x0 = 0;
Chris@96 1586 }
Chris@96 1587 } else if (x0 > vx1) {
Chris@96 1588 x0 = vx1;
Chris@96 1589 }
Chris@95 1590
Chris@96 1591 if (x1 < vx0) {
Chris@96 1592 x1 = vx0;
Chris@96 1593 } else if (x1 > vx1) {
Chris@96 1594 if (vx1 + paintBlockWidth < x1) {
Chris@96 1595 x1 = vx1 + paintBlockWidth;
Chris@96 1596 } else {
Chris@96 1597 x1 = v->width();
Chris@95 1598 }
Chris@96 1599 }
Chris@95 1600
Chris@96 1601 cache.validArea = QRect
Chris@96 1602 (std::min(vx0, x0), cache.validArea.y(),
Chris@96 1603 std::max(vx1 - std::min(vx0, x0),
Chris@96 1604 x1 - std::min(vx0, x0)),
Chris@96 1605 cache.validArea.height());
Chris@95 1606
Chris@96 1607 } else {
Chris@96 1608 if (x1 > x0 + paintBlockWidth) {
Chris@96 1609 x1 = x0 + paintBlockWidth;
Chris@95 1610 }
Chris@96 1611 cache.validArea = QRect(x0, 0, x1 - x0, v->height());
Chris@95 1612 }
Chris@95 1613
Chris@0 1614 int w = x1 - x0;
Chris@0 1615 int h = y1 - y0;
Chris@0 1616
Chris@95 1617 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1618 std::cerr << "x0 " << x0 << ", x1 " << x1 << ", w " << w << ", h " << h << std::endl;
Chris@95 1619 #endif
Chris@95 1620
Chris@95 1621 if (m_drawBuffer.width() < w || m_drawBuffer.height() < h) {
Chris@95 1622 m_drawBuffer = QImage(w, h, QImage::Format_RGB32);
Chris@95 1623 }
Chris@95 1624
Chris@97 1625 m_drawBuffer.fill(m_colourMap.getColour(0).rgb());
Chris@35 1626
Chris@37 1627 int sr = m_model->getSampleRate();
Chris@35 1628
Chris@107 1629 size_t bins = m_fftSize / 2;
Chris@35 1630 if (m_maxFrequency > 0) {
Chris@107 1631 bins = int((double(m_maxFrequency) * m_fftSize) / sr + 0.1);
Chris@107 1632 if (bins > m_fftSize / 2) bins = m_fftSize / 2;
Chris@35 1633 }
Chris@111 1634
Chris@40 1635 size_t minbin = 1;
Chris@37 1636 if (m_minFrequency > 0) {
Chris@107 1637 minbin = int((double(m_minFrequency) * m_fftSize) / sr + 0.1);
Chris@40 1638 if (minbin < 1) minbin = 1;
Chris@37 1639 if (minbin >= bins) minbin = bins - 1;
Chris@37 1640 }
Chris@37 1641
Chris@111 1642 //!!! quite wrong and won't work for layers that may be in more than one view
Chris@111 1643 if (v->height() > (bins - minbin) / (m_zeroPadLevel + 1)) {
Chris@111 1644 if (m_zeroPadLevel != 3) {
Chris@111 1645 ((SpectrogramLayer *)this)->setZeroPadLevel(3);
Chris@111 1646 return;
Chris@111 1647 }
Chris@111 1648 } else {
Chris@111 1649 if (m_zeroPadLevel != 0) {
Chris@111 1650 ((SpectrogramLayer *)this)->setZeroPadLevel(0);
Chris@111 1651 return;
Chris@111 1652 }
Chris@111 1653 }
Chris@111 1654
Chris@107 1655 float minFreq = (float(minbin) * sr) / m_fftSize;
Chris@107 1656 float maxFreq = (float(bins) * sr) / m_fftSize;
Chris@0 1657
Chris@92 1658 float ymag[h];
Chris@92 1659 float ydiv[h];
Chris@92 1660 float yval[bins + 1];
Chris@92 1661
Chris@38 1662 size_t increment = getWindowIncrement();
Chris@40 1663
Chris@40 1664 bool logarithmic = (m_frequencyScale == LogFrequencyScale);
Chris@38 1665
Chris@92 1666 for (size_t q = minbin; q <= bins; ++q) {
Chris@107 1667 float f0 = (float(q) * sr) / m_fftSize;
Chris@92 1668 yval[q] = v->getYForFrequency(f0, minFreq, maxFreq, logarithmic);
Chris@92 1669 }
Chris@92 1670
Chris@35 1671 for (int x = 0; x < w; ++x) {
Chris@35 1672
Chris@35 1673 for (int y = 0; y < h; ++y) {
Chris@40 1674 ymag[y] = 0.0;
Chris@40 1675 ydiv[y] = 0.0;
Chris@35 1676 }
Chris@35 1677
Chris@35 1678 float s0 = 0, s1 = 0;
Chris@35 1679
Chris@44 1680 if (!getXBinRange(v, x0 + x, s0, s1)) {
Chris@95 1681 assert(x <= m_drawBuffer.width());
Chris@35 1682 continue;
Chris@35 1683 }
Chris@35 1684
Chris@35 1685 int s0i = int(s0 + 0.001);
Chris@35 1686 int s1i = int(s1);
Chris@35 1687
Chris@110 1688 if (s1i >= m_fftServer->getWidth()) {
Chris@110 1689 if (s0i >= m_fftServer->getWidth()) {
Chris@45 1690 continue;
Chris@45 1691 } else {
Chris@45 1692 s1i = s0i;
Chris@45 1693 }
Chris@45 1694 }
Chris@92 1695
Chris@92 1696 for (int s = s0i; s <= s1i; ++s) {
Chris@92 1697
Chris@110 1698 // if (!m_fftServer->haveSetColumnAt(s)) continue;
Chris@92 1699
Chris@92 1700 for (size_t q = minbin; q < bins; ++q) {
Chris@92 1701
Chris@92 1702 float y0 = yval[q + 1];
Chris@92 1703 float y1 = yval[q];
Chris@92 1704
Chris@40 1705 if (m_binDisplay == PeakBins ||
Chris@40 1706 m_binDisplay == PeakFrequencies) {
Chris@110 1707 if (!m_fftServer->isLocalPeak(s, q)) continue;
Chris@40 1708 }
Chris@40 1709
Chris@110 1710 if (!m_fftServer->isOverThreshold(s, q, m_threshold)) continue;
Chris@40 1711
Chris@35 1712 float sprop = 1.0;
Chris@35 1713 if (s == s0i) sprop *= (s + 1) - s0;
Chris@35 1714 if (s == s1i) sprop *= s1 - s;
Chris@35 1715
Chris@38 1716 if (m_binDisplay == PeakFrequencies &&
Chris@110 1717 s < int(m_fftServer->getWidth()) - 1) {
Chris@35 1718
Chris@38 1719 bool steady = false;
Chris@92 1720 float f = calculateFrequency(q,
Chris@38 1721 m_windowSize,
Chris@38 1722 increment,
Chris@38 1723 sr,
Chris@110 1724 m_fftServer->getPhaseAt(s, q),
Chris@110 1725 m_fftServer->getPhaseAt(s+1, q),
Chris@38 1726 steady);
Chris@40 1727
Chris@44 1728 y0 = y1 = v->getYForFrequency
Chris@92 1729 (f, minFreq, maxFreq, logarithmic);
Chris@35 1730 }
Chris@38 1731
Chris@35 1732 int y0i = int(y0 + 0.001);
Chris@35 1733 int y1i = int(y1);
Chris@35 1734
Chris@92 1735 float value;
Chris@92 1736
Chris@92 1737 if (m_colourScale == PhaseColourScale) {
Chris@110 1738 value = m_fftServer->getPhaseAt(s, q);
Chris@92 1739 } else if (m_normalizeColumns) {
Chris@110 1740 value = m_fftServer->getNormalizedMagnitudeAt(s, q) * m_gain;
Chris@92 1741 } else {
Chris@110 1742 value = m_fftServer->getMagnitudeAt(s, q) * m_gain;
Chris@92 1743 }
Chris@92 1744
Chris@35 1745 for (int y = y0i; y <= y1i; ++y) {
Chris@35 1746
Chris@35 1747 if (y < 0 || y >= h) continue;
Chris@35 1748
Chris@35 1749 float yprop = sprop;
Chris@35 1750 if (y == y0i) yprop *= (y + 1) - y0;
Chris@35 1751 if (y == y1i) yprop *= y1 - y;
Chris@37 1752 ymag[y] += yprop * value;
Chris@35 1753 ydiv[y] += yprop;
Chris@35 1754 }
Chris@35 1755 }
Chris@35 1756 }
Chris@35 1757
Chris@35 1758 for (int y = 0; y < h; ++y) {
Chris@35 1759
Chris@35 1760 if (ydiv[y] > 0.0) {
Chris@40 1761
Chris@40 1762 unsigned char pixel = 0;
Chris@40 1763
Chris@38 1764 float avg = ymag[y] / ydiv[y];
Chris@38 1765 pixel = getDisplayValue(avg);
Chris@40 1766
Chris@95 1767 assert(x <= m_drawBuffer.width());
Chris@86 1768 QColor c = m_colourMap.getColour(pixel);
Chris@95 1769 m_drawBuffer.setPixel(x, y,
Chris@95 1770 qRgb(c.red(), c.green(), c.blue()));
Chris@35 1771 }
Chris@35 1772 }
Chris@35 1773 }
Chris@35 1774
Chris@95 1775 paint.drawImage(x0, y0, m_drawBuffer, 0, 0, w, h);
Chris@0 1776
Chris@0 1777 if (recreateWholePixmapCache) {
Chris@95 1778 cache.pixmap = QPixmap(v->width(), v->height());
Chris@0 1779 }
Chris@0 1780
Chris@95 1781 QPainter cachePainter(&cache.pixmap);
Chris@95 1782 cachePainter.drawImage(x0, y0, m_drawBuffer, 0, 0, w, h);
Chris@0 1783 cachePainter.end();
Chris@0 1784
Chris@95 1785 cache.startFrame = startFrame;
Chris@95 1786 cache.zoomLevel = zoomLevel;
Chris@95 1787
Chris@95 1788 if (cache.validArea.x() > 0) {
Chris@95 1789 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1790 std::cerr << "SpectrogramLayer::paint() updating left" << std::endl;
Chris@95 1791 #endif
Chris@95 1792 v->update(0, 0, cache.validArea.x(), v->height());
Chris@95 1793 }
Chris@95 1794
Chris@95 1795 if (cache.validArea.x() + cache.validArea.width() <
Chris@95 1796 cache.pixmap.width()) {
Chris@95 1797 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@95 1798 std::cerr << "SpectrogramLayer::paint() updating right ("
Chris@95 1799 << cache.validArea.x() + cache.validArea.width()
Chris@95 1800 << ", "
Chris@95 1801 << cache.pixmap.width() - (cache.validArea.x() +
Chris@95 1802 cache.validArea.width())
Chris@95 1803 << ")" << std::endl;
Chris@95 1804 #endif
Chris@95 1805 v->update(cache.validArea.x() + cache.validArea.width(),
Chris@95 1806 0,
Chris@95 1807 cache.pixmap.width() - (cache.validArea.x() +
Chris@95 1808 cache.validArea.width()),
Chris@95 1809 v->height());
Chris@95 1810 }
Chris@0 1811
Chris@0 1812 #ifdef DEBUG_SPECTROGRAM_REPAINT
Chris@0 1813 std::cerr << "SpectrogramLayer::paint() returning" << std::endl;
Chris@0 1814 #endif
Chris@0 1815 }
Chris@0 1816
Chris@42 1817 float
Chris@44 1818 SpectrogramLayer::getYForFrequency(View *v, float frequency) const
Chris@42 1819 {
Chris@44 1820 return v->getYForFrequency(frequency,
Chris@44 1821 getEffectiveMinFrequency(),
Chris@44 1822 getEffectiveMaxFrequency(),
Chris@44 1823 m_frequencyScale == LogFrequencyScale);
Chris@42 1824 }
Chris@42 1825
Chris@42 1826 float
Chris@44 1827 SpectrogramLayer::getFrequencyForY(View *v, int y) const
Chris@42 1828 {
Chris@44 1829 return v->getFrequencyForY(y,
Chris@44 1830 getEffectiveMinFrequency(),
Chris@44 1831 getEffectiveMaxFrequency(),
Chris@44 1832 m_frequencyScale == LogFrequencyScale);
Chris@42 1833 }
Chris@42 1834
Chris@0 1835 int
Chris@0 1836 SpectrogramLayer::getCompletion() const
Chris@0 1837 {
Chris@110 1838 if (m_updateTimer == 0 || !m_fftServer) return 100;
Chris@110 1839 size_t completion = m_fftServer->getFillCompletion();
Chris@0 1840 // std::cerr << "SpectrogramLayer::getCompletion: completion = " << completion << std::endl;
Chris@0 1841 return completion;
Chris@0 1842 }
Chris@0 1843
Chris@28 1844 bool
Chris@101 1845 SpectrogramLayer::getValueExtents(float &min, float &max,
Chris@101 1846 bool &logarithmic, QString &unit) const
Chris@79 1847 {
Chris@79 1848 min = getEffectiveMinFrequency();
Chris@79 1849 max = getEffectiveMaxFrequency();
Chris@101 1850 logarithmic = (m_frequencyScale == LogFrequencyScale);
Chris@79 1851 unit = "Hz";
Chris@79 1852 return true;
Chris@79 1853 }
Chris@79 1854
Chris@79 1855 bool
Chris@101 1856 SpectrogramLayer::getDisplayExtents(float &min, float &max) const
Chris@101 1857 {
Chris@101 1858 min = getEffectiveMinFrequency();
Chris@101 1859 max = getEffectiveMaxFrequency();
Chris@101 1860 return true;
Chris@101 1861 }
Chris@101 1862
Chris@101 1863 bool
Chris@44 1864 SpectrogramLayer::snapToFeatureFrame(View *v, int &frame,
Chris@28 1865 size_t &resolution,
Chris@28 1866 SnapType snap) const
Chris@13 1867 {
Chris@13 1868 resolution = getWindowIncrement();
Chris@28 1869 int left = (frame / resolution) * resolution;
Chris@28 1870 int right = left + resolution;
Chris@28 1871
Chris@28 1872 switch (snap) {
Chris@28 1873 case SnapLeft: frame = left; break;
Chris@28 1874 case SnapRight: frame = right; break;
Chris@28 1875 case SnapNearest:
Chris@28 1876 case SnapNeighbouring:
Chris@28 1877 if (frame - left > right - frame) frame = right;
Chris@28 1878 else frame = left;
Chris@28 1879 break;
Chris@28 1880 }
Chris@28 1881
Chris@28 1882 return true;
Chris@28 1883 }
Chris@13 1884
Chris@77 1885 bool
Chris@77 1886 SpectrogramLayer::getCrosshairExtents(View *v, QPainter &paint,
Chris@77 1887 QPoint cursorPos,
Chris@77 1888 std::vector<QRect> &extents) const
Chris@77 1889 {
Chris@77 1890 QRect vertical(cursorPos.x() - 12, 0, 12, v->height());
Chris@77 1891 extents.push_back(vertical);
Chris@77 1892
Chris@77 1893 QRect horizontal(0, cursorPos.y(), cursorPos.x(), 1);
Chris@77 1894 extents.push_back(horizontal);
Chris@77 1895
Chris@77 1896 return true;
Chris@77 1897 }
Chris@77 1898
Chris@77 1899 void
Chris@77 1900 SpectrogramLayer::paintCrosshairs(View *v, QPainter &paint,
Chris@77 1901 QPoint cursorPos) const
Chris@77 1902 {
Chris@77 1903 paint.save();
Chris@77 1904 paint.setPen(m_crosshairColour);
Chris@77 1905
Chris@77 1906 paint.drawLine(0, cursorPos.y(), cursorPos.x() - 1, cursorPos.y());
Chris@77 1907 paint.drawLine(cursorPos.x(), 0, cursorPos.x(), v->height());
Chris@77 1908
Chris@77 1909 float fundamental = getFrequencyForY(v, cursorPos.y());
Chris@77 1910
Chris@77 1911 int harmonic = 2;
Chris@77 1912
Chris@77 1913 while (harmonic < 100) {
Chris@77 1914
Chris@77 1915 float hy = lrintf(getYForFrequency(v, fundamental * harmonic));
Chris@77 1916 if (hy < 0 || hy > v->height()) break;
Chris@77 1917
Chris@77 1918 int len = 7;
Chris@77 1919
Chris@77 1920 if (harmonic % 2 == 0) {
Chris@77 1921 if (harmonic % 4 == 0) {
Chris@77 1922 len = 12;
Chris@77 1923 } else {
Chris@77 1924 len = 10;
Chris@77 1925 }
Chris@77 1926 }
Chris@77 1927
Chris@77 1928 paint.drawLine(cursorPos.x() - len,
Chris@77 1929 hy,
Chris@77 1930 cursorPos.x(),
Chris@77 1931 hy);
Chris@77 1932
Chris@77 1933 ++harmonic;
Chris@77 1934 }
Chris@77 1935
Chris@77 1936 paint.restore();
Chris@77 1937 }
Chris@77 1938
Chris@25 1939 QString
Chris@44 1940 SpectrogramLayer::getFeatureDescription(View *v, QPoint &pos) const
Chris@25 1941 {
Chris@25 1942 int x = pos.x();
Chris@25 1943 int y = pos.y();
Chris@0 1944
Chris@25 1945 if (!m_model || !m_model->isOK()) return "";
Chris@0 1946
Chris@38 1947 float magMin = 0, magMax = 0;
Chris@38 1948 float phaseMin = 0, phaseMax = 0;
Chris@0 1949 float freqMin = 0, freqMax = 0;
Chris@35 1950 float adjFreqMin = 0, adjFreqMax = 0;
Chris@25 1951 QString pitchMin, pitchMax;
Chris@0 1952 RealTime rtMin, rtMax;
Chris@0 1953
Chris@38 1954 bool haveValues = false;
Chris@0 1955
Chris@44 1956 if (!getXBinSourceRange(v, x, rtMin, rtMax)) {
Chris@38 1957 return "";
Chris@38 1958 }
Chris@44 1959 if (getXYBinSourceRange(v, x, y, magMin, magMax, phaseMin, phaseMax)) {
Chris@38 1960 haveValues = true;
Chris@38 1961 }
Chris@0 1962
Chris@35 1963 QString adjFreqText = "", adjPitchText = "";
Chris@35 1964
Chris@38 1965 if (m_binDisplay == PeakFrequencies) {
Chris@35 1966
Chris@44 1967 if (!getAdjustedYBinSourceRange(v, x, y, freqMin, freqMax,
Chris@38 1968 adjFreqMin, adjFreqMax)) {
Chris@38 1969 return "";
Chris@38 1970 }
Chris@35 1971
Chris@35 1972 if (adjFreqMin != adjFreqMax) {
Chris@65 1973 adjFreqText = tr("Peak Frequency:\t%1 - %2 Hz\n")
Chris@35 1974 .arg(adjFreqMin).arg(adjFreqMax);
Chris@35 1975 } else {
Chris@65 1976 adjFreqText = tr("Peak Frequency:\t%1 Hz\n")
Chris@35 1977 .arg(adjFreqMin);
Chris@38 1978 }
Chris@38 1979
Chris@38 1980 QString pmin = Pitch::getPitchLabelForFrequency(adjFreqMin);
Chris@38 1981 QString pmax = Pitch::getPitchLabelForFrequency(adjFreqMax);
Chris@38 1982
Chris@38 1983 if (pmin != pmax) {
Chris@65 1984 adjPitchText = tr("Peak Pitch:\t%3 - %4\n").arg(pmin).arg(pmax);
Chris@38 1985 } else {
Chris@65 1986 adjPitchText = tr("Peak Pitch:\t%2\n").arg(pmin);
Chris@35 1987 }
Chris@35 1988
Chris@35 1989 } else {
Chris@35 1990
Chris@44 1991 if (!getYBinSourceRange(v, y, freqMin, freqMax)) return "";
Chris@35 1992 }
Chris@35 1993
Chris@25 1994 QString text;
Chris@25 1995
Chris@25 1996 if (rtMin != rtMax) {
Chris@25 1997 text += tr("Time:\t%1 - %2\n")
Chris@25 1998 .arg(rtMin.toText(true).c_str())
Chris@25 1999 .arg(rtMax.toText(true).c_str());
Chris@25 2000 } else {
Chris@25 2001 text += tr("Time:\t%1\n")
Chris@25 2002 .arg(rtMin.toText(true).c_str());
Chris@0 2003 }
Chris@0 2004
Chris@25 2005 if (freqMin != freqMax) {
Chris@65 2006 text += tr("%1Bin Frequency:\t%2 - %3 Hz\n%4Bin Pitch:\t%5 - %6\n")
Chris@65 2007 .arg(adjFreqText)
Chris@25 2008 .arg(freqMin)
Chris@25 2009 .arg(freqMax)
Chris@65 2010 .arg(adjPitchText)
Chris@65 2011 .arg(Pitch::getPitchLabelForFrequency(freqMin))
Chris@65 2012 .arg(Pitch::getPitchLabelForFrequency(freqMax));
Chris@65 2013 } else {
Chris@65 2014 text += tr("%1Bin Frequency:\t%2 Hz\n%3Bin Pitch:\t%4\n")
Chris@35 2015 .arg(adjFreqText)
Chris@25 2016 .arg(freqMin)
Chris@65 2017 .arg(adjPitchText)
Chris@65 2018 .arg(Pitch::getPitchLabelForFrequency(freqMin));
Chris@25 2019 }
Chris@25 2020
Chris@38 2021 if (haveValues) {
Chris@38 2022 float dbMin = AudioLevel::multiplier_to_dB(magMin);
Chris@38 2023 float dbMax = AudioLevel::multiplier_to_dB(magMax);
Chris@43 2024 QString dbMinString;
Chris@43 2025 QString dbMaxString;
Chris@43 2026 if (dbMin == AudioLevel::DB_FLOOR) {
Chris@43 2027 dbMinString = tr("-Inf");
Chris@43 2028 } else {
Chris@43 2029 dbMinString = QString("%1").arg(lrintf(dbMin));
Chris@43 2030 }
Chris@43 2031 if (dbMax == AudioLevel::DB_FLOOR) {
Chris@43 2032 dbMaxString = tr("-Inf");
Chris@43 2033 } else {
Chris@43 2034 dbMaxString = QString("%1").arg(lrintf(dbMax));
Chris@43 2035 }
Chris@25 2036 if (lrintf(dbMin) != lrintf(dbMax)) {
Chris@25 2037 text += tr("dB:\t%1 - %2").arg(lrintf(dbMin)).arg(lrintf(dbMax));
Chris@25 2038 } else {
Chris@25 2039 text += tr("dB:\t%1").arg(lrintf(dbMin));
Chris@25 2040 }
Chris@38 2041 if (phaseMin != phaseMax) {
Chris@38 2042 text += tr("\nPhase:\t%1 - %2").arg(phaseMin).arg(phaseMax);
Chris@38 2043 } else {
Chris@38 2044 text += tr("\nPhase:\t%1").arg(phaseMin);
Chris@38 2045 }
Chris@25 2046 }
Chris@25 2047
Chris@25 2048 return text;
Chris@0 2049 }
Chris@25 2050
Chris@0 2051 int
Chris@40 2052 SpectrogramLayer::getColourScaleWidth(QPainter &paint) const
Chris@40 2053 {
Chris@40 2054 int cw;
Chris@40 2055
Chris@40 2056 switch (m_colourScale) {
Chris@40 2057 default:
Chris@40 2058 case LinearColourScale:
Chris@40 2059 cw = paint.fontMetrics().width(QString("0.00"));
Chris@40 2060 break;
Chris@40 2061
Chris@40 2062 case MeterColourScale:
Chris@40 2063 case dBColourScale:
Chris@40 2064 cw = std::max(paint.fontMetrics().width(tr("-Inf")),
Chris@40 2065 paint.fontMetrics().width(tr("-90")));
Chris@40 2066 break;
Chris@40 2067
Chris@40 2068 case PhaseColourScale:
Chris@40 2069 cw = paint.fontMetrics().width(QString("-") + QChar(0x3c0));
Chris@40 2070 break;
Chris@40 2071 }
Chris@40 2072
Chris@40 2073 return cw;
Chris@40 2074 }
Chris@40 2075
Chris@40 2076 int
Chris@44 2077 SpectrogramLayer::getVerticalScaleWidth(View *v, QPainter &paint) const
Chris@0 2078 {
Chris@0 2079 if (!m_model || !m_model->isOK()) return 0;
Chris@0 2080
Chris@40 2081 int cw = getColourScaleWidth(paint);
Chris@40 2082
Chris@0 2083 int tw = paint.fontMetrics().width(QString("%1")
Chris@0 2084 .arg(m_maxFrequency > 0 ?
Chris@0 2085 m_maxFrequency - 1 :
Chris@0 2086 m_model->getSampleRate() / 2));
Chris@0 2087
Chris@0 2088 int fw = paint.fontMetrics().width(QString("43Hz"));
Chris@0 2089 if (tw < fw) tw = fw;
Chris@40 2090
Chris@40 2091 int tickw = (m_frequencyScale == LogFrequencyScale ? 10 : 4);
Chris@0 2092
Chris@40 2093 return cw + tickw + tw + 13;
Chris@0 2094 }
Chris@0 2095
Chris@0 2096 void
Chris@44 2097 SpectrogramLayer::paintVerticalScale(View *v, QPainter &paint, QRect rect) const
Chris@0 2098 {
Chris@0 2099 if (!m_model || !m_model->isOK()) {
Chris@0 2100 return;
Chris@0 2101 }
Chris@0 2102
Chris@0 2103 int h = rect.height(), w = rect.width();
Chris@0 2104
Chris@40 2105 int tickw = (m_frequencyScale == LogFrequencyScale ? 10 : 4);
Chris@40 2106 int pkw = (m_frequencyScale == LogFrequencyScale ? 10 : 0);
Chris@40 2107
Chris@107 2108 size_t bins = m_fftSize / 2;
Chris@0 2109 int sr = m_model->getSampleRate();
Chris@0 2110
Chris@0 2111 if (m_maxFrequency > 0) {
Chris@107 2112 bins = int((double(m_maxFrequency) * m_fftSize) / sr + 0.1);
Chris@107 2113 if (bins > m_fftSize / 2) bins = m_fftSize / 2;
Chris@0 2114 }
Chris@0 2115
Chris@40 2116 int cw = getColourScaleWidth(paint);
Chris@40 2117
Chris@0 2118 int py = -1;
Chris@0 2119 int textHeight = paint.fontMetrics().height();
Chris@0 2120 int toff = -textHeight + paint.fontMetrics().ascent() + 2;
Chris@0 2121
Chris@110 2122 if (m_fftServer && h > textHeight * 2 + 10) {
Chris@40 2123
Chris@40 2124 int ch = h - textHeight * 2 - 8;
Chris@40 2125 paint.drawRect(4, textHeight + 4, cw - 1, ch + 1);
Chris@40 2126
Chris@40 2127 QString top, bottom;
Chris@40 2128
Chris@40 2129 switch (m_colourScale) {
Chris@40 2130 default:
Chris@40 2131 case LinearColourScale:
Chris@40 2132 top = (m_normalizeColumns ? "1.0" : "0.02");
Chris@40 2133 bottom = (m_normalizeColumns ? "0.0" : "0.00");
Chris@40 2134 break;
Chris@40 2135
Chris@40 2136 case MeterColourScale:
Chris@40 2137 top = (m_normalizeColumns ? QString("0") :
Chris@40 2138 QString("%1").arg(int(AudioLevel::multiplier_to_dB(0.02))));
Chris@40 2139 bottom = QString("%1").
Chris@40 2140 arg(int(AudioLevel::multiplier_to_dB
Chris@40 2141 (AudioLevel::preview_to_multiplier(0, 255))));
Chris@40 2142 break;
Chris@40 2143
Chris@40 2144 case dBColourScale:
Chris@40 2145 top = "0";
Chris@40 2146 bottom = "-80";
Chris@40 2147 break;
Chris@40 2148
Chris@40 2149 case PhaseColourScale:
Chris@40 2150 top = QChar(0x3c0);
Chris@40 2151 bottom = "-" + top;
Chris@40 2152 break;
Chris@40 2153 }
Chris@40 2154
Chris@40 2155 paint.drawText((cw + 6 - paint.fontMetrics().width(top)) / 2,
Chris@40 2156 2 + textHeight + toff, top);
Chris@40 2157
Chris@40 2158 paint.drawText((cw + 6 - paint.fontMetrics().width(bottom)) / 2,
Chris@40 2159 h + toff - 3, bottom);
Chris@40 2160
Chris@40 2161 paint.save();
Chris@40 2162 paint.setBrush(Qt::NoBrush);
Chris@40 2163 for (int i = 0; i < ch; ++i) {
Chris@40 2164 int v = (i * 255) / ch + 1;
Chris@86 2165 paint.setPen(m_colourMap.getColour(v));
Chris@40 2166 paint.drawLine(5, 4 + textHeight + ch - i,
Chris@40 2167 cw + 2, 4 + textHeight + ch - i);
Chris@40 2168 }
Chris@40 2169 paint.restore();
Chris@40 2170 }
Chris@40 2171
Chris@40 2172 paint.drawLine(cw + 7, 0, cw + 7, h);
Chris@40 2173
Chris@0 2174 int bin = -1;
Chris@0 2175
Chris@44 2176 for (int y = 0; y < v->height(); ++y) {
Chris@0 2177
Chris@0 2178 float q0, q1;
Chris@44 2179 if (!getYBinRange(v, v->height() - y, q0, q1)) continue;
Chris@0 2180
Chris@0 2181 int vy;
Chris@0 2182
Chris@0 2183 if (int(q0) > bin) {
Chris@0 2184 vy = y;
Chris@0 2185 bin = int(q0);
Chris@0 2186 } else {
Chris@0 2187 continue;
Chris@0 2188 }
Chris@0 2189
Chris@107 2190 int freq = (sr * bin) / m_fftSize;
Chris@0 2191
Chris@0 2192 if (py >= 0 && (vy - py) < textHeight - 1) {
Chris@40 2193 if (m_frequencyScale == LinearFrequencyScale) {
Chris@40 2194 paint.drawLine(w - tickw, h - vy, w, h - vy);
Chris@40 2195 }
Chris@0 2196 continue;
Chris@0 2197 }
Chris@0 2198
Chris@0 2199 QString text = QString("%1").arg(freq);
Chris@40 2200 if (bin == 1) text = QString("%1Hz").arg(freq); // bin 0 is DC
Chris@40 2201 paint.drawLine(cw + 7, h - vy, w - pkw - 1, h - vy);
Chris@0 2202
Chris@0 2203 if (h - vy - textHeight >= -2) {
Chris@40 2204 int tx = w - 3 - paint.fontMetrics().width(text) - std::max(tickw, pkw);
Chris@0 2205 paint.drawText(tx, h - vy + toff, text);
Chris@0 2206 }
Chris@0 2207
Chris@0 2208 py = vy;
Chris@0 2209 }
Chris@40 2210
Chris@40 2211 if (m_frequencyScale == LogFrequencyScale) {
Chris@40 2212
Chris@40 2213 paint.drawLine(w - pkw - 1, 0, w - pkw - 1, h);
Chris@40 2214
Chris@110 2215 int sr = m_model->getSampleRate();
Chris@40 2216 float minf = getEffectiveMinFrequency();
Chris@40 2217 float maxf = getEffectiveMaxFrequency();
Chris@40 2218
Chris@40 2219 int py = h;
Chris@40 2220 paint.setBrush(paint.pen().color());
Chris@40 2221
Chris@40 2222 for (int i = 0; i < 128; ++i) {
Chris@40 2223
Chris@40 2224 float f = Pitch::getFrequencyForPitch(i);
Chris@44 2225 int y = lrintf(v->getYForFrequency(f, minf, maxf, true));
Chris@40 2226 int n = (i % 12);
Chris@40 2227 if (n == 1 || n == 3 || n == 6 || n == 8 || n == 10) {
Chris@40 2228 // black notes
Chris@40 2229 paint.drawLine(w - pkw, y, w, y);
Chris@41 2230 int rh = ((py - y) / 4) * 2;
Chris@41 2231 if (rh < 2) rh = 2;
Chris@41 2232 paint.drawRect(w - pkw, y - (py-y)/4, pkw/2, rh);
Chris@40 2233 } else if (n == 0 || n == 5) {
Chris@40 2234 // C, A
Chris@40 2235 if (py < h) {
Chris@40 2236 paint.drawLine(w - pkw, (y + py) / 2, w, (y + py) / 2);
Chris@40 2237 }
Chris@40 2238 }
Chris@40 2239
Chris@40 2240 py = y;
Chris@40 2241 }
Chris@40 2242 }
Chris@0 2243 }
Chris@0 2244
Chris@6 2245 QString
Chris@6 2246 SpectrogramLayer::toXmlString(QString indent, QString extraAttributes) const
Chris@6 2247 {
Chris@6 2248 QString s;
Chris@6 2249
Chris@6 2250 s += QString("channel=\"%1\" "
Chris@6 2251 "windowSize=\"%2\" "
Chris@6 2252 "windowType=\"%3\" "
Chris@97 2253 "windowHopLevel=\"%4\" "
Chris@37 2254 "gain=\"%5\" "
Chris@37 2255 "threshold=\"%6\" ")
Chris@6 2256 .arg(m_channel)
Chris@6 2257 .arg(m_windowSize)
Chris@6 2258 .arg(m_windowType)
Chris@97 2259 .arg(m_windowHopLevel)
Chris@37 2260 .arg(m_gain)
Chris@37 2261 .arg(m_threshold);
Chris@37 2262
Chris@37 2263 s += QString("minFrequency=\"%1\" "
Chris@37 2264 "maxFrequency=\"%2\" "
Chris@37 2265 "colourScale=\"%3\" "
Chris@37 2266 "colourScheme=\"%4\" "
Chris@37 2267 "colourRotation=\"%5\" "
Chris@37 2268 "frequencyScale=\"%6\" "
Chris@37 2269 "binDisplay=\"%7\" "
Chris@37 2270 "normalizeColumns=\"%8\"")
Chris@37 2271 .arg(m_minFrequency)
Chris@6 2272 .arg(m_maxFrequency)
Chris@6 2273 .arg(m_colourScale)
Chris@6 2274 .arg(m_colourScheme)
Chris@37 2275 .arg(m_colourRotation)
Chris@35 2276 .arg(m_frequencyScale)
Chris@37 2277 .arg(m_binDisplay)
Chris@36 2278 .arg(m_normalizeColumns ? "true" : "false");
Chris@6 2279
Chris@6 2280 return Layer::toXmlString(indent, extraAttributes + " " + s);
Chris@6 2281 }
Chris@6 2282
Chris@11 2283 void
Chris@11 2284 SpectrogramLayer::setProperties(const QXmlAttributes &attributes)
Chris@11 2285 {
Chris@11 2286 bool ok = false;
Chris@11 2287
Chris@11 2288 int channel = attributes.value("channel").toInt(&ok);
Chris@11 2289 if (ok) setChannel(channel);
Chris@11 2290
Chris@11 2291 size_t windowSize = attributes.value("windowSize").toUInt(&ok);
Chris@11 2292 if (ok) setWindowSize(windowSize);
Chris@11 2293
Chris@11 2294 WindowType windowType = (WindowType)
Chris@11 2295 attributes.value("windowType").toInt(&ok);
Chris@11 2296 if (ok) setWindowType(windowType);
Chris@11 2297
Chris@97 2298 size_t windowHopLevel = attributes.value("windowHopLevel").toUInt(&ok);
Chris@97 2299 if (ok) setWindowHopLevel(windowHopLevel);
Chris@97 2300 else {
Chris@97 2301 size_t windowOverlap = attributes.value("windowOverlap").toUInt(&ok);
Chris@97 2302 // a percentage value
Chris@97 2303 if (ok) {
Chris@97 2304 if (windowOverlap == 0) setWindowHopLevel(0);
Chris@97 2305 else if (windowOverlap == 25) setWindowHopLevel(1);
Chris@97 2306 else if (windowOverlap == 50) setWindowHopLevel(2);
Chris@97 2307 else if (windowOverlap == 75) setWindowHopLevel(3);
Chris@97 2308 else if (windowOverlap == 90) setWindowHopLevel(4);
Chris@97 2309 }
Chris@97 2310 }
Chris@11 2311
Chris@11 2312 float gain = attributes.value("gain").toFloat(&ok);
Chris@11 2313 if (ok) setGain(gain);
Chris@11 2314
Chris@37 2315 float threshold = attributes.value("threshold").toFloat(&ok);
Chris@37 2316 if (ok) setThreshold(threshold);
Chris@37 2317
Chris@37 2318 size_t minFrequency = attributes.value("minFrequency").toUInt(&ok);
Chris@37 2319 if (ok) setMinFrequency(minFrequency);
Chris@37 2320
Chris@11 2321 size_t maxFrequency = attributes.value("maxFrequency").toUInt(&ok);
Chris@11 2322 if (ok) setMaxFrequency(maxFrequency);
Chris@11 2323
Chris@11 2324 ColourScale colourScale = (ColourScale)
Chris@11 2325 attributes.value("colourScale").toInt(&ok);
Chris@11 2326 if (ok) setColourScale(colourScale);
Chris@11 2327
Chris@11 2328 ColourScheme colourScheme = (ColourScheme)
Chris@11 2329 attributes.value("colourScheme").toInt(&ok);
Chris@11 2330 if (ok) setColourScheme(colourScheme);
Chris@11 2331
Chris@37 2332 int colourRotation = attributes.value("colourRotation").toInt(&ok);
Chris@37 2333 if (ok) setColourRotation(colourRotation);
Chris@37 2334
Chris@11 2335 FrequencyScale frequencyScale = (FrequencyScale)
Chris@11 2336 attributes.value("frequencyScale").toInt(&ok);
Chris@11 2337 if (ok) setFrequencyScale(frequencyScale);
Chris@35 2338
Chris@37 2339 BinDisplay binDisplay = (BinDisplay)
Chris@37 2340 attributes.value("binDisplay").toInt(&ok);
Chris@37 2341 if (ok) setBinDisplay(binDisplay);
Chris@36 2342
Chris@36 2343 bool normalizeColumns =
Chris@36 2344 (attributes.value("normalizeColumns").trimmed() == "true");
Chris@36 2345 setNormalizeColumns(normalizeColumns);
Chris@11 2346 }
Chris@11 2347
Chris@11 2348
Chris@0 2349 #ifdef INCLUDE_MOCFILES
Chris@0 2350 #include "SpectrogramLayer.moc.cpp"
Chris@0 2351 #endif
Chris@0 2352