annotate layer/SpectrogramLayer.cpp @ 86:93a7efc75fb7

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