Chris@58: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ Chris@0: Chris@0: /* Chris@59: Sonic Visualiser Chris@59: An audio file viewer and annotation editor. Chris@59: Centre for Digital Music, Queen Mary, University of London. Chris@484: This file copyright 2006-2009 Chris Cannam and QMUL. Chris@0: Chris@59: This program is free software; you can redistribute it and/or Chris@59: modify it under the terms of the GNU General Public License as Chris@59: published by the Free Software Foundation; either version 2 of the Chris@59: License, or (at your option) any later version. See the file Chris@59: COPYING included with this distribution for more information. Chris@0: */ Chris@0: Chris@0: #include "SpectrogramLayer.h" Chris@0: Chris@128: #include "view/View.h" Chris@0: #include "base/Profiler.h" Chris@0: #include "base/AudioLevel.h" Chris@0: #include "base/Window.h" Chris@24: #include "base/Pitch.h" Chris@118: #include "base/Preferences.h" Chris@167: #include "base/RangeMapper.h" Chris@253: #include "base/LogRange.h" Chris@376: #include "widgets/CommandHistory.h" Chris@376: #include "ColourMapper.h" Chris@283: #include "ImageRegionFinder.h" Chris@484: #include "data/model/Dense3DModelPeakCache.h" Chris@0: Chris@0: #include Chris@0: #include Chris@0: #include Chris@0: #include Chris@0: #include Chris@92: #include Chris@178: #include Chris@283: #include Chris@316: #include Chris@0: Chris@0: #include Chris@0: Chris@480: using std::cerr; Chris@480: using std::endl; Chris@0: #include Chris@0: #include Chris@0: Chris@545: #ifndef __GNUC__ Chris@545: #include Chris@545: #endif Chris@545: Chris@490: //#define DEBUG_SPECTROGRAM_REPAINT 1 Chris@0: Chris@44: SpectrogramLayer::SpectrogramLayer(Configuration config) : Chris@0: m_model(0), Chris@0: m_channel(0), Chris@0: m_windowSize(1024), Chris@0: m_windowType(HanningWindow), Chris@97: m_windowHopLevel(2), Chris@109: m_zeroPadLevel(0), Chris@107: m_fftSize(1024), Chris@0: m_gain(1.0), Chris@215: m_initialGain(1.0), Chris@37: m_threshold(0.0), Chris@215: m_initialThreshold(0.0), Chris@9: m_colourRotation(0), Chris@215: m_initialRotation(0), Chris@119: m_minFrequency(10), Chris@0: m_maxFrequency(8000), Chris@135: m_initialMaxFrequency(8000), Chris@0: m_colourScale(dBColourScale), Chris@197: m_colourMap(0), Chris@0: m_frequencyScale(LinearFrequencyScale), Chris@37: m_binDisplay(AllBins), Chris@36: m_normalizeColumns(false), Chris@120: m_normalizeVisibleArea(false), Chris@133: m_lastEmittedZoomStep(-1), Chris@390: m_synchronous(false), Chris@215: m_lastPaintBlockWidth(0), Chris@0: m_updateTimer(0), Chris@44: m_candidateFillStartFrame(0), Chris@193: m_exiting(false), Chris@193: m_sliceableModel(0) Chris@0: { Chris@215: if (config == FullRangeDb) { Chris@215: m_initialMaxFrequency = 0; Chris@215: setMaxFrequency(0); Chris@215: } else if (config == MelodicRange) { Chris@0: setWindowSize(8192); Chris@97: setWindowHopLevel(4); Chris@215: m_initialMaxFrequency = 1500; Chris@215: setMaxFrequency(1500); Chris@215: setMinFrequency(40); Chris@0: setColourScale(LinearColourScale); Chris@215: setColourMap(ColourMapper::Sunset); Chris@215: setFrequencyScale(LogFrequencyScale); Chris@224: // setGain(20); Chris@37: } else if (config == MelodicPeaks) { Chris@37: setWindowSize(4096); Chris@97: setWindowHopLevel(5); Chris@135: m_initialMaxFrequency = 2000; Chris@40: setMaxFrequency(2000); Chris@37: setMinFrequency(40); Chris@37: setFrequencyScale(LogFrequencyScale); Chris@215: setColourScale(LinearColourScale); Chris@37: setBinDisplay(PeakFrequencies); Chris@37: setNormalizeColumns(true); Chris@0: } Chris@110: Chris@122: Preferences *prefs = Preferences::getInstance(); Chris@122: connect(prefs, SIGNAL(propertyChanged(PropertyContainer::PropertyName)), Chris@122: this, SLOT(preferenceChanged(PropertyContainer::PropertyName))); Chris@122: setWindowType(prefs->getWindowType()); Chris@122: Chris@197: initialisePalette(); Chris@0: } Chris@0: Chris@0: SpectrogramLayer::~SpectrogramLayer() Chris@0: { Chris@0: delete m_updateTimer; Chris@0: m_updateTimer = 0; Chris@0: Chris@130: invalidateFFTModels(); Chris@0: } Chris@0: Chris@0: void Chris@0: SpectrogramLayer::setModel(const DenseTimeValueModel *model) Chris@0: { Chris@101: // std::cerr << "SpectrogramLayer(" << this << "): setModel(" << model << ")" << std::endl; Chris@34: Chris@110: if (model == m_model) return; Chris@110: Chris@0: m_model = model; Chris@130: invalidateFFTModels(); Chris@0: Chris@0: if (!m_model || !m_model->isOK()) return; Chris@0: Chris@320: connectSignals(m_model); Chris@0: Chris@0: connect(m_model, SIGNAL(modelChanged()), this, SLOT(cacheInvalid())); Chris@0: connect(m_model, SIGNAL(modelChanged(size_t, size_t)), Chris@0: this, SLOT(cacheInvalid(size_t, size_t))); Chris@0: Chris@0: emit modelReplaced(); Chris@110: } Chris@115: Chris@0: Layer::PropertyList Chris@0: SpectrogramLayer::getProperties() const Chris@0: { Chris@0: PropertyList list; Chris@87: list.push_back("Colour"); Chris@87: list.push_back("Colour Scale"); Chris@87: list.push_back("Window Size"); Chris@97: list.push_back("Window Increment"); Chris@87: list.push_back("Normalize Columns"); Chris@120: list.push_back("Normalize Visible Area"); Chris@87: list.push_back("Bin Display"); Chris@87: list.push_back("Threshold"); Chris@87: list.push_back("Gain"); Chris@87: list.push_back("Colour Rotation"); Chris@153: // list.push_back("Min Frequency"); Chris@153: // list.push_back("Max Frequency"); Chris@87: list.push_back("Frequency Scale"); Chris@153: //// list.push_back("Zero Padding"); Chris@0: return list; Chris@0: } Chris@0: Chris@87: QString Chris@87: SpectrogramLayer::getPropertyLabel(const PropertyName &name) const Chris@87: { Chris@87: if (name == "Colour") return tr("Colour"); Chris@87: if (name == "Colour Scale") return tr("Colour Scale"); Chris@87: if (name == "Window Size") return tr("Window Size"); Chris@112: if (name == "Window Increment") return tr("Window Overlap"); Chris@87: if (name == "Normalize Columns") return tr("Normalize Columns"); Chris@120: if (name == "Normalize Visible Area") return tr("Normalize Visible Area"); Chris@87: if (name == "Bin Display") return tr("Bin Display"); Chris@87: if (name == "Threshold") return tr("Threshold"); Chris@87: if (name == "Gain") return tr("Gain"); Chris@87: if (name == "Colour Rotation") return tr("Colour Rotation"); Chris@87: if (name == "Min Frequency") return tr("Min Frequency"); Chris@87: if (name == "Max Frequency") return tr("Max Frequency"); Chris@87: if (name == "Frequency Scale") return tr("Frequency Scale"); Chris@109: if (name == "Zero Padding") return tr("Smoothing"); Chris@87: return ""; Chris@87: } Chris@87: Chris@335: QString Chris@335: SpectrogramLayer::getPropertyIconName(const PropertyName &name) const Chris@335: { Chris@335: if (name == "Normalize Columns") return "normalise-columns"; Chris@335: if (name == "Normalize Visible Area") return "normalise"; Chris@335: return ""; Chris@335: } Chris@335: Chris@0: Layer::PropertyType Chris@0: SpectrogramLayer::getPropertyType(const PropertyName &name) const Chris@0: { Chris@87: if (name == "Gain") return RangeProperty; Chris@87: if (name == "Colour Rotation") return RangeProperty; Chris@87: if (name == "Normalize Columns") return ToggleProperty; Chris@120: if (name == "Normalize Visible Area") return ToggleProperty; Chris@87: if (name == "Threshold") return RangeProperty; Chris@109: if (name == "Zero Padding") return ToggleProperty; Chris@0: return ValueProperty; Chris@0: } Chris@0: Chris@0: QString Chris@0: SpectrogramLayer::getPropertyGroupName(const PropertyName &name) const Chris@0: { Chris@153: if (name == "Bin Display" || Chris@153: name == "Frequency Scale") return tr("Bins"); Chris@87: if (name == "Window Size" || Chris@109: name == "Window Increment" || Chris@109: name == "Zero Padding") return tr("Window"); Chris@87: if (name == "Colour" || Chris@87: name == "Threshold" || Chris@87: name == "Colour Rotation") return tr("Colour"); Chris@87: if (name == "Normalize Columns" || Chris@120: name == "Normalize Visible Area" || Chris@153: name == "Gain" || Chris@87: name == "Colour Scale") return tr("Scale"); Chris@0: return QString(); Chris@0: } Chris@0: Chris@0: int Chris@0: SpectrogramLayer::getPropertyRangeAndValue(const PropertyName &name, Chris@216: int *min, int *max, int *deflt) const Chris@0: { Chris@216: int val = 0; Chris@216: Chris@216: int garbage0, garbage1, garbage2; Chris@55: if (!min) min = &garbage0; Chris@55: if (!max) max = &garbage1; Chris@216: if (!deflt) deflt = &garbage2; Chris@10: Chris@87: if (name == "Gain") { Chris@0: Chris@0: *min = -50; Chris@0: *max = 50; Chris@0: Chris@216: *deflt = lrintf(log10(m_initialGain) * 20.0);; Chris@216: if (*deflt < *min) *deflt = *min; Chris@216: if (*deflt > *max) *deflt = *max; Chris@216: Chris@216: val = lrintf(log10(m_gain) * 20.0); Chris@216: if (val < *min) val = *min; Chris@216: if (val > *max) val = *max; Chris@0: Chris@87: } else if (name == "Threshold") { Chris@37: Chris@37: *min = -50; Chris@37: *max = 0; Chris@37: Chris@216: *deflt = lrintf(AudioLevel::multiplier_to_dB(m_initialThreshold)); Chris@216: if (*deflt < *min) *deflt = *min; Chris@216: if (*deflt > *max) *deflt = *max; Chris@216: Chris@216: val = lrintf(AudioLevel::multiplier_to_dB(m_threshold)); Chris@216: if (val < *min) val = *min; Chris@216: if (val > *max) val = *max; Chris@37: Chris@87: } else if (name == "Colour Rotation") { Chris@9: Chris@9: *min = 0; Chris@9: *max = 256; Chris@216: *deflt = m_initialRotation; Chris@216: Chris@216: val = m_colourRotation; Chris@9: Chris@87: } else if (name == "Colour Scale") { Chris@0: Chris@0: *min = 0; Chris@176: *max = 4; Chris@216: *deflt = int(dBColourScale); Chris@216: Chris@216: val = (int)m_colourScale; Chris@0: Chris@87: } else if (name == "Colour") { Chris@0: Chris@0: *min = 0; Chris@196: *max = ColourMapper::getColourMapCount() - 1; Chris@216: *deflt = 0; Chris@216: Chris@216: val = m_colourMap; Chris@0: Chris@87: } else if (name == "Window Size") { Chris@0: Chris@0: *min = 0; Chris@0: *max = 10; Chris@216: *deflt = 5; Chris@0: Chris@216: val = 0; Chris@0: int ws = m_windowSize; Chris@216: while (ws > 32) { ws >>= 1; val ++; } Chris@0: Chris@97: } else if (name == "Window Increment") { Chris@0: Chris@0: *min = 0; Chris@97: *max = 5; Chris@216: *deflt = 2; Chris@216: Chris@216: val = m_windowHopLevel; Chris@0: Chris@109: } else if (name == "Zero Padding") { Chris@109: Chris@109: *min = 0; Chris@109: *max = 1; Chris@216: *deflt = 0; Chris@109: Chris@216: val = m_zeroPadLevel > 0 ? 1 : 0; Chris@109: Chris@87: } else if (name == "Min Frequency") { Chris@37: Chris@37: *min = 0; Chris@37: *max = 9; Chris@216: *deflt = 1; Chris@37: Chris@37: switch (m_minFrequency) { Chris@216: case 0: default: val = 0; break; Chris@216: case 10: val = 1; break; Chris@216: case 20: val = 2; break; Chris@216: case 40: val = 3; break; Chris@216: case 100: val = 4; break; Chris@216: case 250: val = 5; break; Chris@216: case 500: val = 6; break; Chris@216: case 1000: val = 7; break; Chris@216: case 4000: val = 8; break; Chris@216: case 10000: val = 9; break; Chris@37: } Chris@37: Chris@87: } else if (name == "Max Frequency") { Chris@0: Chris@0: *min = 0; Chris@0: *max = 9; Chris@216: *deflt = 6; Chris@0: Chris@0: switch (m_maxFrequency) { Chris@216: case 500: val = 0; break; Chris@216: case 1000: val = 1; break; Chris@216: case 1500: val = 2; break; Chris@216: case 2000: val = 3; break; Chris@216: case 4000: val = 4; break; Chris@216: case 6000: val = 5; break; Chris@216: case 8000: val = 6; break; Chris@216: case 12000: val = 7; break; Chris@216: case 16000: val = 8; break; Chris@216: default: val = 9; break; Chris@0: } Chris@0: Chris@87: } else if (name == "Frequency Scale") { Chris@0: Chris@0: *min = 0; Chris@0: *max = 1; Chris@216: *deflt = int(LinearFrequencyScale); Chris@216: val = (int)m_frequencyScale; Chris@0: Chris@87: } else if (name == "Bin Display") { Chris@35: Chris@35: *min = 0; Chris@35: *max = 2; Chris@216: *deflt = int(AllBins); Chris@216: val = (int)m_binDisplay; Chris@35: Chris@87: } else if (name == "Normalize Columns") { Chris@36: Chris@216: *deflt = 0; Chris@216: val = (m_normalizeColumns ? 1 : 0); Chris@36: Chris@120: } else if (name == "Normalize Visible Area") { Chris@120: Chris@216: *deflt = 0; Chris@216: val = (m_normalizeVisibleArea ? 1 : 0); Chris@120: Chris@0: } else { Chris@216: val = Layer::getPropertyRangeAndValue(name, min, max, deflt); Chris@0: } Chris@0: Chris@216: return val; Chris@0: } Chris@0: Chris@0: QString Chris@0: SpectrogramLayer::getPropertyValueLabel(const PropertyName &name, Chris@9: int value) const Chris@0: { Chris@87: if (name == "Colour") { Chris@196: return ColourMapper::getColourMapName(value); Chris@0: } Chris@87: if (name == "Colour Scale") { Chris@0: switch (value) { Chris@0: default: Chris@37: case 0: return tr("Linear"); Chris@37: case 1: return tr("Meter"); Chris@215: case 2: return tr("dBV^2"); Chris@215: case 3: return tr("dBV"); Chris@119: case 4: return tr("Phase"); Chris@0: } Chris@0: } Chris@87: if (name == "Window Size") { Chris@0: return QString("%1").arg(32 << value); Chris@0: } Chris@97: if (name == "Window Increment") { Chris@0: switch (value) { Chris@0: default: Chris@112: case 0: return tr("None"); Chris@112: case 1: return tr("25 %"); Chris@112: case 2: return tr("50 %"); Chris@112: case 3: return tr("75 %"); Chris@112: case 4: return tr("87.5 %"); Chris@112: case 5: return tr("93.75 %"); Chris@0: } Chris@0: } Chris@109: if (name == "Zero Padding") { Chris@109: if (value == 0) return tr("None"); Chris@109: return QString("%1x").arg(value + 1); Chris@109: } Chris@87: if (name == "Min Frequency") { Chris@37: switch (value) { Chris@37: default: Chris@38: case 0: return tr("No min"); Chris@37: case 1: return tr("10 Hz"); Chris@37: case 2: return tr("20 Hz"); Chris@37: case 3: return tr("40 Hz"); Chris@37: case 4: return tr("100 Hz"); Chris@37: case 5: return tr("250 Hz"); Chris@37: case 6: return tr("500 Hz"); Chris@37: case 7: return tr("1 KHz"); Chris@37: case 8: return tr("4 KHz"); Chris@37: case 9: return tr("10 KHz"); Chris@37: } Chris@37: } Chris@87: if (name == "Max Frequency") { Chris@0: switch (value) { Chris@0: default: Chris@0: case 0: return tr("500 Hz"); Chris@0: case 1: return tr("1 KHz"); Chris@0: case 2: return tr("1.5 KHz"); Chris@0: case 3: return tr("2 KHz"); Chris@0: case 4: return tr("4 KHz"); Chris@0: case 5: return tr("6 KHz"); Chris@0: case 6: return tr("8 KHz"); Chris@0: case 7: return tr("12 KHz"); Chris@0: case 8: return tr("16 KHz"); Chris@38: case 9: return tr("No max"); Chris@0: } Chris@0: } Chris@87: if (name == "Frequency Scale") { Chris@0: switch (value) { Chris@0: default: Chris@0: case 0: return tr("Linear"); Chris@0: case 1: return tr("Log"); Chris@0: } Chris@0: } Chris@87: if (name == "Bin Display") { Chris@35: switch (value) { Chris@35: default: Chris@37: case 0: return tr("All Bins"); Chris@37: case 1: return tr("Peak Bins"); Chris@37: case 2: return tr("Frequencies"); Chris@35: } Chris@35: } Chris@0: return tr(""); Chris@0: } Chris@0: Chris@167: RangeMapper * Chris@167: SpectrogramLayer::getNewPropertyRangeMapper(const PropertyName &name) const Chris@167: { Chris@167: if (name == "Gain") { Chris@167: return new LinearRangeMapper(-50, 50, -25, 25, tr("dB")); Chris@167: } Chris@167: if (name == "Threshold") { Chris@167: return new LinearRangeMapper(-50, 0, -50, 0, tr("dB")); Chris@167: } Chris@167: return 0; Chris@167: } Chris@167: Chris@0: void Chris@0: SpectrogramLayer::setProperty(const PropertyName &name, int value) Chris@0: { Chris@87: if (name == "Gain") { Chris@0: setGain(pow(10, float(value)/20.0)); Chris@87: } else if (name == "Threshold") { Chris@37: if (value == -50) setThreshold(0.0); Chris@37: else setThreshold(AudioLevel::dB_to_multiplier(value)); Chris@87: } else if (name == "Colour Rotation") { Chris@9: setColourRotation(value); Chris@87: } else if (name == "Colour") { Chris@197: setColourMap(value); Chris@87: } else if (name == "Window Size") { Chris@0: setWindowSize(32 << value); Chris@97: } else if (name == "Window Increment") { Chris@97: setWindowHopLevel(value); Chris@109: } else if (name == "Zero Padding") { Chris@109: setZeroPadLevel(value > 0.1 ? 3 : 0); Chris@87: } else if (name == "Min Frequency") { Chris@37: switch (value) { Chris@37: default: Chris@37: case 0: setMinFrequency(0); break; Chris@37: case 1: setMinFrequency(10); break; Chris@37: case 2: setMinFrequency(20); break; Chris@37: case 3: setMinFrequency(40); break; Chris@37: case 4: setMinFrequency(100); break; Chris@37: case 5: setMinFrequency(250); break; Chris@37: case 6: setMinFrequency(500); break; Chris@37: case 7: setMinFrequency(1000); break; Chris@37: case 8: setMinFrequency(4000); break; Chris@37: case 9: setMinFrequency(10000); break; Chris@37: } Chris@133: int vs = getCurrentVerticalZoomStep(); Chris@133: if (vs != m_lastEmittedZoomStep) { Chris@133: emit verticalZoomChanged(); Chris@133: m_lastEmittedZoomStep = vs; Chris@133: } Chris@87: } else if (name == "Max Frequency") { Chris@0: switch (value) { Chris@0: case 0: setMaxFrequency(500); break; Chris@0: case 1: setMaxFrequency(1000); break; Chris@0: case 2: setMaxFrequency(1500); break; Chris@0: case 3: setMaxFrequency(2000); break; Chris@0: case 4: setMaxFrequency(4000); break; Chris@0: case 5: setMaxFrequency(6000); break; Chris@0: case 6: setMaxFrequency(8000); break; Chris@0: case 7: setMaxFrequency(12000); break; Chris@0: case 8: setMaxFrequency(16000); break; Chris@0: default: Chris@0: case 9: setMaxFrequency(0); break; Chris@0: } Chris@133: int vs = getCurrentVerticalZoomStep(); Chris@133: if (vs != m_lastEmittedZoomStep) { Chris@133: emit verticalZoomChanged(); Chris@133: m_lastEmittedZoomStep = vs; Chris@133: } Chris@87: } else if (name == "Colour Scale") { Chris@0: switch (value) { Chris@0: default: Chris@0: case 0: setColourScale(LinearColourScale); break; Chris@0: case 1: setColourScale(MeterColourScale); break; Chris@215: case 2: setColourScale(dBSquaredColourScale); break; Chris@215: case 3: setColourScale(dBColourScale); break; Chris@119: case 4: setColourScale(PhaseColourScale); break; Chris@0: } Chris@87: } else if (name == "Frequency Scale") { Chris@0: switch (value) { Chris@0: default: Chris@0: case 0: setFrequencyScale(LinearFrequencyScale); break; Chris@0: case 1: setFrequencyScale(LogFrequencyScale); break; Chris@0: } Chris@87: } else if (name == "Bin Display") { Chris@35: switch (value) { Chris@35: default: Chris@37: case 0: setBinDisplay(AllBins); break; Chris@37: case 1: setBinDisplay(PeakBins); break; Chris@37: case 2: setBinDisplay(PeakFrequencies); break; Chris@35: } Chris@82: } else if (name == "Normalize Columns") { Chris@36: setNormalizeColumns(value ? true : false); Chris@120: } else if (name == "Normalize Visible Area") { Chris@120: setNormalizeVisibleArea(value ? true : false); Chris@0: } Chris@0: } Chris@0: Chris@0: void Chris@478: SpectrogramLayer::invalidateImageCaches() Chris@95: { Chris@478: for (ViewImageCache::iterator i = m_imageCaches.begin(); Chris@478: i != m_imageCaches.end(); ++i) { Chris@95: i->second.validArea = QRect(); Chris@95: } Chris@95: } Chris@95: Chris@95: void Chris@478: SpectrogramLayer::invalidateImageCaches(size_t startFrame, size_t endFrame) Chris@95: { Chris@478: for (ViewImageCache::iterator i = m_imageCaches.begin(); Chris@478: i != m_imageCaches.end(); ++i) { Chris@131: Chris@95: //!!! when are views removed from the map? on setLayerDormant? Chris@95: const View *v = i->first; Chris@95: Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::invalidateImageCaches(" Chris@391: << startFrame << ", " << endFrame << "): view range is " Chris@391: << v->getStartFrame() << ", " << v->getEndFrame() Chris@585: << endl; Chris@391: Chris@391: std::cerr << "Valid area was: " << i->second.validArea.x() << ", " Chris@391: << i->second.validArea.y() << " " Chris@391: << i->second.validArea.width() << "x" Chris@391: << i->second.validArea.height() << std::endl; Chris@391: #endif Chris@391: Chris@391: if (long(startFrame) > v->getStartFrame()) { Chris@391: if (startFrame >= v->getEndFrame()) { Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@391: std::cerr << "Modified start frame is off right of view" << std::endl; Chris@391: #endif Chris@391: return; Chris@391: } Chris@391: int x = v->getXForFrame(startFrame); Chris@407: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "clipping from 0 to " << x-1 << endl; Chris@407: #endif Chris@391: if (x > 1) { Chris@391: i->second.validArea &= Chris@391: QRect(0, 0, x-1, v->height()); Chris@391: } else { Chris@391: i->second.validArea = QRect(); Chris@391: } Chris@391: } else { Chris@391: if (long(endFrame) < v->getStartFrame()) { Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@391: std::cerr << "Modified end frame is off left of view" << std::endl; Chris@391: #endif Chris@391: return; Chris@391: } Chris@391: int x = v->getXForFrame(endFrame); Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "clipping from " << x+1 << " to " << v->width() Chris@585: << endl; Chris@391: #endif Chris@391: if (x < v->width()) { Chris@391: i->second.validArea &= Chris@391: QRect(x+1, 0, v->width()-(x+1), v->height()); Chris@391: } else { Chris@391: i->second.validArea = QRect(); Chris@391: } Chris@95: } Chris@391: Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@391: std::cerr << "Valid area is now: " << i->second.validArea.x() << ", " Chris@391: << i->second.validArea.y() << " " Chris@391: << i->second.validArea.width() << "x" Chris@391: << i->second.validArea.height() << std::endl; Chris@391: #endif Chris@95: } Chris@95: } Chris@95: Chris@95: void Chris@122: SpectrogramLayer::preferenceChanged(PropertyContainer::PropertyName name) Chris@122: { Chris@587: SVDEBUG << "SpectrogramLayer::preferenceChanged(" << name << ")" << endl; Chris@122: Chris@122: if (name == "Window Type") { Chris@122: setWindowType(Preferences::getInstance()->getWindowType()); Chris@122: return; Chris@122: } Chris@490: if (name == "Spectrogram Y Smoothing") { Chris@490: invalidateImageCaches(); Chris@490: invalidateMagnitudes(); Chris@490: emit layerParametersChanged(); Chris@490: } Chris@490: if (name == "Spectrogram X Smoothing") { Chris@478: invalidateImageCaches(); Chris@122: invalidateMagnitudes(); Chris@122: emit layerParametersChanged(); Chris@122: } Chris@122: if (name == "Tuning Frequency") { Chris@122: emit layerParametersChanged(); Chris@122: } Chris@122: } Chris@122: Chris@122: void Chris@0: SpectrogramLayer::setChannel(int ch) Chris@0: { Chris@0: if (m_channel == ch) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: m_channel = ch; Chris@130: invalidateFFTModels(); Chris@9: Chris@0: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: int Chris@0: SpectrogramLayer::getChannel() const Chris@0: { Chris@0: return m_channel; Chris@0: } Chris@0: Chris@0: void Chris@0: SpectrogramLayer::setWindowSize(size_t ws) Chris@0: { Chris@0: if (m_windowSize == ws) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@0: m_windowSize = ws; Chris@109: m_fftSize = ws * (m_zeroPadLevel + 1); Chris@0: Chris@130: invalidateFFTModels(); Chris@9: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: size_t Chris@0: SpectrogramLayer::getWindowSize() const Chris@0: { Chris@0: return m_windowSize; Chris@0: } Chris@0: Chris@0: void Chris@97: SpectrogramLayer::setWindowHopLevel(size_t v) Chris@0: { Chris@97: if (m_windowHopLevel == v) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@97: m_windowHopLevel = v; Chris@0: Chris@130: invalidateFFTModels(); Chris@9: Chris@9: emit layerParametersChanged(); Chris@9: Chris@110: // fillCache(); Chris@0: } Chris@0: Chris@0: size_t Chris@97: SpectrogramLayer::getWindowHopLevel() const Chris@0: { Chris@97: return m_windowHopLevel; Chris@0: } Chris@0: Chris@0: void Chris@109: SpectrogramLayer::setZeroPadLevel(size_t v) Chris@109: { Chris@109: if (m_zeroPadLevel == v) return; Chris@109: Chris@478: invalidateImageCaches(); Chris@109: Chris@109: m_zeroPadLevel = v; Chris@109: m_fftSize = m_windowSize * (v + 1); Chris@110: Chris@130: invalidateFFTModels(); Chris@109: Chris@109: emit layerParametersChanged(); Chris@109: } Chris@109: Chris@109: size_t Chris@109: SpectrogramLayer::getZeroPadLevel() const Chris@109: { Chris@109: return m_zeroPadLevel; Chris@109: } Chris@109: Chris@109: void Chris@0: SpectrogramLayer::setWindowType(WindowType w) Chris@0: { Chris@0: if (m_windowType == w) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@0: m_windowType = w; Chris@110: Chris@130: invalidateFFTModels(); Chris@9: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: WindowType Chris@0: SpectrogramLayer::getWindowType() const Chris@0: { Chris@0: return m_windowType; Chris@0: } Chris@0: Chris@0: void Chris@0: SpectrogramLayer::setGain(float gain) Chris@0: { Chris@587: // SVDEBUG << "SpectrogramLayer::setGain(" << gain << ") (my gain is now " Chris@585: // << m_gain << ")" << endl; Chris@55: Chris@40: if (m_gain == gain) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@0: m_gain = gain; Chris@0: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: float Chris@0: SpectrogramLayer::getGain() const Chris@0: { Chris@0: return m_gain; Chris@0: } Chris@0: Chris@0: void Chris@37: SpectrogramLayer::setThreshold(float threshold) Chris@37: { Chris@40: if (m_threshold == threshold) return; Chris@37: Chris@478: invalidateImageCaches(); Chris@37: Chris@37: m_threshold = threshold; Chris@37: Chris@37: emit layerParametersChanged(); Chris@37: } Chris@37: Chris@37: float Chris@37: SpectrogramLayer::getThreshold() const Chris@37: { Chris@37: return m_threshold; Chris@37: } Chris@37: Chris@37: void Chris@37: SpectrogramLayer::setMinFrequency(size_t mf) Chris@37: { Chris@37: if (m_minFrequency == mf) return; Chris@37: Chris@587: // SVDEBUG << "SpectrogramLayer::setMinFrequency: " << mf << endl; Chris@187: Chris@478: invalidateImageCaches(); Chris@119: invalidateMagnitudes(); Chris@37: Chris@37: m_minFrequency = mf; Chris@37: Chris@37: emit layerParametersChanged(); Chris@37: } Chris@37: Chris@37: size_t Chris@37: SpectrogramLayer::getMinFrequency() const Chris@37: { Chris@37: return m_minFrequency; Chris@37: } Chris@37: Chris@37: void Chris@0: SpectrogramLayer::setMaxFrequency(size_t mf) Chris@0: { Chris@0: if (m_maxFrequency == mf) return; Chris@0: Chris@587: // SVDEBUG << "SpectrogramLayer::setMaxFrequency: " << mf << endl; Chris@187: Chris@478: invalidateImageCaches(); Chris@119: invalidateMagnitudes(); Chris@0: Chris@0: m_maxFrequency = mf; Chris@0: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: size_t Chris@0: SpectrogramLayer::getMaxFrequency() const Chris@0: { Chris@0: return m_maxFrequency; Chris@0: } Chris@0: Chris@0: void Chris@9: SpectrogramLayer::setColourRotation(int r) Chris@9: { Chris@478: invalidateImageCaches(); Chris@9: Chris@9: if (r < 0) r = 0; Chris@9: if (r > 256) r = 256; Chris@9: int distance = r - m_colourRotation; Chris@9: Chris@9: if (distance != 0) { Chris@197: rotatePalette(-distance); Chris@9: m_colourRotation = r; Chris@9: } Chris@9: Chris@9: emit layerParametersChanged(); Chris@9: } Chris@9: Chris@9: void Chris@0: SpectrogramLayer::setColourScale(ColourScale colourScale) Chris@0: { Chris@0: if (m_colourScale == colourScale) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@0: m_colourScale = colourScale; Chris@0: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: SpectrogramLayer::ColourScale Chris@0: SpectrogramLayer::getColourScale() const Chris@0: { Chris@0: return m_colourScale; Chris@0: } Chris@0: Chris@0: void Chris@197: SpectrogramLayer::setColourMap(int map) Chris@0: { Chris@197: if (m_colourMap == map) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: Chris@197: m_colourMap = map; Chris@197: initialisePalette(); Chris@9: Chris@0: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@196: int Chris@197: SpectrogramLayer::getColourMap() const Chris@0: { Chris@197: return m_colourMap; Chris@0: } Chris@0: Chris@0: void Chris@0: SpectrogramLayer::setFrequencyScale(FrequencyScale frequencyScale) Chris@0: { Chris@0: if (m_frequencyScale == frequencyScale) return; Chris@0: Chris@478: invalidateImageCaches(); Chris@0: m_frequencyScale = frequencyScale; Chris@9: Chris@9: emit layerParametersChanged(); Chris@0: } Chris@0: Chris@0: SpectrogramLayer::FrequencyScale Chris@0: SpectrogramLayer::getFrequencyScale() const Chris@0: { Chris@0: return m_frequencyScale; Chris@0: } Chris@0: Chris@0: void Chris@37: SpectrogramLayer::setBinDisplay(BinDisplay binDisplay) Chris@35: { Chris@37: if (m_binDisplay == binDisplay) return; Chris@35: Chris@478: invalidateImageCaches(); Chris@37: m_binDisplay = binDisplay; Chris@35: Chris@35: emit layerParametersChanged(); Chris@35: } Chris@35: Chris@37: SpectrogramLayer::BinDisplay Chris@37: SpectrogramLayer::getBinDisplay() const Chris@35: { Chris@37: return m_binDisplay; Chris@35: } Chris@35: Chris@35: void Chris@36: SpectrogramLayer::setNormalizeColumns(bool n) Chris@36: { Chris@36: if (m_normalizeColumns == n) return; Chris@36: Chris@478: invalidateImageCaches(); Chris@119: invalidateMagnitudes(); Chris@36: m_normalizeColumns = n; Chris@36: Chris@36: emit layerParametersChanged(); Chris@36: } Chris@36: Chris@36: bool Chris@36: SpectrogramLayer::getNormalizeColumns() const Chris@36: { Chris@36: return m_normalizeColumns; Chris@36: } Chris@36: Chris@36: void Chris@120: SpectrogramLayer::setNormalizeVisibleArea(bool n) Chris@120: { Chris@587: SVDEBUG << "SpectrogramLayer::setNormalizeVisibleArea(" << n Chris@585: << ") (from " << m_normalizeVisibleArea << ")" << endl; Chris@327: Chris@120: if (m_normalizeVisibleArea == n) return; Chris@120: Chris@478: invalidateImageCaches(); Chris@120: invalidateMagnitudes(); Chris@120: m_normalizeVisibleArea = n; Chris@120: Chris@120: emit layerParametersChanged(); Chris@120: } Chris@120: Chris@120: bool Chris@120: SpectrogramLayer::getNormalizeVisibleArea() const Chris@120: { Chris@120: return m_normalizeVisibleArea; Chris@120: } Chris@120: Chris@120: void Chris@47: SpectrogramLayer::setLayerDormant(const View *v, bool dormant) Chris@29: { Chris@33: if (dormant) { Chris@33: Chris@331: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::setLayerDormant(" << dormant << ")" Chris@585: << endl; Chris@331: #endif Chris@331: Chris@131: if (isLayerDormant(v)) { Chris@131: return; Chris@131: } Chris@131: Chris@131: Layer::setLayerDormant(v, true); Chris@33: Chris@478: invalidateImageCaches(); Chris@478: m_imageCaches.erase(v); Chris@114: Chris@130: if (m_fftModels.find(v) != m_fftModels.end()) { Chris@193: Chris@193: if (m_sliceableModel == m_fftModels[v].first) { Chris@193: bool replaced = false; Chris@193: for (ViewFFTMap::iterator i = m_fftModels.begin(); Chris@193: i != m_fftModels.end(); ++i) { Chris@193: if (i->second.first != m_sliceableModel) { Chris@193: emit sliceableModelReplaced(m_sliceableModel, i->second.first); Chris@193: replaced = true; Chris@193: break; Chris@193: } Chris@193: } Chris@193: if (!replaced) emit sliceableModelReplaced(m_sliceableModel, 0); Chris@193: } Chris@193: Chris@130: delete m_fftModels[v].first; Chris@130: m_fftModels.erase(v); Chris@484: Chris@484: delete m_peakCaches[v]; Chris@484: m_peakCaches.erase(v); Chris@114: } Chris@33: Chris@33: } else { Chris@33: Chris@131: Layer::setLayerDormant(v, false); Chris@33: } Chris@29: } Chris@29: Chris@29: void Chris@0: SpectrogramLayer::cacheInvalid() Chris@0: { Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::cacheInvalid()" << endl; Chris@391: #endif Chris@391: Chris@478: invalidateImageCaches(); Chris@119: invalidateMagnitudes(); Chris@0: } Chris@0: Chris@0: void Chris@391: SpectrogramLayer::cacheInvalid(size_t from, size_t to) Chris@0: { Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::cacheInvalid(" << from << ", " << to << ")" << endl; Chris@391: #endif Chris@391: Chris@478: invalidateImageCaches(from, to); Chris@391: invalidateMagnitudes(); Chris@0: } Chris@0: Chris@0: void Chris@0: SpectrogramLayer::fillTimerTimedOut() Chris@0: { Chris@115: if (!m_model) return; Chris@115: Chris@115: bool allDone = true; Chris@115: Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::fillTimerTimedOut: have " << m_fftModels.size() << " FFT models associated with views" << endl; Chris@184: #endif Chris@184: Chris@130: for (ViewFFTMap::iterator i = m_fftModels.begin(); Chris@130: i != m_fftModels.end(); ++i) { Chris@115: Chris@130: const FFTModel *model = i->second.first; Chris@115: size_t lastFill = i->second.second; Chris@115: Chris@130: if (model) { Chris@130: Chris@130: size_t fill = model->getFillExtent(); Chris@115: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::fillTimerTimedOut: extent for " << model << ": " << fill << ", last " << lastFill << ", total " << m_model->getEndFrame() << endl; Chris@0: #endif Chris@115: Chris@115: if (fill >= lastFill) { Chris@115: if (fill >= m_model->getEndFrame() && lastFill > 0) { Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@115: std::cerr << "complete!" << std::endl; Chris@0: #endif Chris@478: invalidateImageCaches(); Chris@184: i->second.second = -1; Chris@115: emit modelChanged(); Chris@115: Chris@115: } else if (fill > lastFill) { Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@115: std::cerr << "SpectrogramLayer: emitting modelChanged(" Chris@115: << lastFill << "," << fill << ")" << std::endl; Chris@0: #endif Chris@478: invalidateImageCaches(lastFill, fill); Chris@184: i->second.second = fill; Chris@115: emit modelChanged(lastFill, fill); Chris@115: } Chris@115: } else { Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@115: std::cerr << "SpectrogramLayer: going backwards, emitting modelChanged(" Chris@115: << m_model->getStartFrame() << "," << m_model->getEndFrame() << ")" << std::endl; Chris@0: #endif Chris@478: invalidateImageCaches(); Chris@184: i->second.second = fill; Chris@115: emit modelChanged(m_model->getStartFrame(), m_model->getEndFrame()); Chris@115: } Chris@115: Chris@115: if (i->second.second >= 0) { Chris@115: allDone = false; Chris@115: } Chris@115: } Chris@0: } Chris@115: Chris@115: if (allDone) { Chris@115: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@115: std::cerr << "SpectrogramLayer: all complete!" << std::endl; Chris@115: #endif Chris@115: delete m_updateTimer; Chris@115: m_updateTimer = 0; Chris@115: } Chris@0: } Chris@0: Chris@224: bool Chris@224: SpectrogramLayer::hasLightBackground() const Chris@224: { Chris@287: return ColourMapper(m_colourMap, 1.f, 255.f).hasLightBackground(); Chris@224: } Chris@224: Chris@0: void Chris@197: SpectrogramLayer::initialisePalette() Chris@0: { Chris@10: int formerRotation = m_colourRotation; Chris@10: Chris@197: if (m_colourMap == (int)ColourMapper::BlackOnWhite) { Chris@197: m_palette.setColour(NO_VALUE, Qt::white); Chris@38: } else { Chris@197: m_palette.setColour(NO_VALUE, Qt::black); Chris@38: } Chris@0: Chris@197: ColourMapper mapper(m_colourMap, 1.f, 255.f); Chris@196: Chris@0: for (int pixel = 1; pixel < 256; ++pixel) { Chris@197: m_palette.setColour(pixel, mapper.map(pixel)); Chris@0: } Chris@9: Chris@196: m_crosshairColour = mapper.getContrastingColour(); Chris@196: Chris@9: m_colourRotation = 0; Chris@197: rotatePalette(m_colourRotation - formerRotation); Chris@10: m_colourRotation = formerRotation; Chris@478: Chris@478: m_drawBuffer = QImage(); Chris@9: } Chris@9: Chris@9: void Chris@197: SpectrogramLayer::rotatePalette(int distance) Chris@9: { Chris@31: QColor newPixels[256]; Chris@9: Chris@197: newPixels[NO_VALUE] = m_palette.getColour(NO_VALUE); Chris@9: Chris@9: for (int pixel = 1; pixel < 256; ++pixel) { Chris@9: int target = pixel + distance; Chris@9: while (target < 1) target += 255; Chris@9: while (target > 255) target -= 255; Chris@197: newPixels[target] = m_palette.getColour(pixel); Chris@9: } Chris@9: Chris@9: for (int pixel = 0; pixel < 256; ++pixel) { Chris@197: m_palette.setColour(pixel, newPixels[pixel]); Chris@9: } Chris@478: Chris@478: m_drawBuffer = QImage(); Chris@0: } Chris@0: Chris@38: unsigned char Chris@119: SpectrogramLayer::getDisplayValue(View *v, float input) const Chris@38: { Chris@38: int value; Chris@37: Chris@120: float min = 0.f; Chris@120: float max = 1.f; Chris@120: Chris@120: if (m_normalizeVisibleArea) { Chris@120: min = m_viewMags[v].getMin(); Chris@120: max = m_viewMags[v].getMax(); Chris@120: } else if (!m_normalizeColumns) { Chris@224: if (m_colourScale == LinearColourScale //|| Chris@224: // m_colourScale == MeterColourScale) { Chris@224: ) { Chris@224: max = 0.1f; Chris@120: } Chris@120: } Chris@120: Chris@119: float thresh = -80.f; Chris@119: Chris@119: if (max == 0.f) max = 1.f; Chris@119: if (max == min) min = max - 0.0001f; Chris@119: Chris@40: switch (m_colourScale) { Chris@40: Chris@40: default: Chris@40: case LinearColourScale: Chris@119: value = int(((input - min) / (max - min)) * 255.f) + 1; Chris@40: break; Chris@40: Chris@40: case MeterColourScale: Chris@210: value = AudioLevel::multiplier_to_preview Chris@210: ((input - min) / (max - min), 254) + 1; Chris@40: break; Chris@119: Chris@210: case dBSquaredColourScale: Chris@215: input = ((input - min) * (input - min)) / ((max - min) * (max - min)); Chris@133: if (input > 0.f) { Chris@133: input = 10.f * log10f(input); Chris@133: } else { Chris@133: input = thresh; Chris@133: } Chris@119: if (min > 0.f) { Chris@119: thresh = 10.f * log10f(min * min); Chris@119: if (thresh < -80.f) thresh = -80.f; Chris@119: } Chris@119: input = (input - thresh) / (-thresh); Chris@119: if (input < 0.f) input = 0.f; Chris@119: if (input > 1.f) input = 1.f; Chris@119: value = int(input * 255.f) + 1; Chris@119: break; Chris@40: Chris@215: case dBColourScale: Chris@215: //!!! experiment with normalizing the visible area this way. Chris@215: //In any case, we need to have some indication of what the dB Chris@215: //scale is relative to. Chris@215: input = (input - min) / (max - min); Chris@215: if (input > 0.f) { Chris@215: input = 10.f * log10f(input); Chris@215: } else { Chris@215: input = thresh; Chris@215: } Chris@215: if (min > 0.f) { Chris@215: thresh = 10.f * log10f(min); Chris@215: if (thresh < -80.f) thresh = -80.f; Chris@215: } Chris@215: input = (input - thresh) / (-thresh); Chris@215: if (input < 0.f) input = 0.f; Chris@215: if (input > 1.f) input = 1.f; Chris@215: value = int(input * 255.f) + 1; Chris@215: break; Chris@215: Chris@40: case PhaseColourScale: Chris@40: value = int((input * 127.0 / M_PI) + 128); Chris@40: break; Chris@0: } Chris@210: Chris@38: if (value > UCHAR_MAX) value = UCHAR_MAX; Chris@38: if (value < 0) value = 0; Chris@38: return value; Chris@0: } Chris@0: Chris@40: float Chris@40: SpectrogramLayer::getInputForDisplayValue(unsigned char uc) const Chris@40: { Chris@153: //!!! unused Chris@153: Chris@40: int value = uc; Chris@40: float input; Chris@40: Chris@120: //!!! incorrect for normalizing visible area (and also out of date) Chris@120: Chris@40: switch (m_colourScale) { Chris@40: Chris@40: default: Chris@40: case LinearColourScale: Chris@40: input = float(value - 1) / 255.0 / (m_normalizeColumns ? 1 : 50); Chris@40: break; Chris@40: Chris@40: case MeterColourScale: Chris@40: input = AudioLevel::preview_to_multiplier(value - 1, 255) Chris@40: / (m_normalizeColumns ? 1.0 : 50.0); Chris@40: break; Chris@40: Chris@215: case dBSquaredColourScale: Chris@40: input = float(value - 1) / 255.0; Chris@40: input = (input * 80.0) - 80.0; Chris@40: input = powf(10.0, input) / 20.0; Chris@40: value = int(input); Chris@40: break; Chris@40: Chris@215: case dBColourScale: Chris@119: input = float(value - 1) / 255.0; Chris@119: input = (input * 80.0) - 80.0; Chris@119: input = powf(10.0, input) / 20.0; Chris@119: value = int(input); Chris@119: break; Chris@119: Chris@40: case PhaseColourScale: Chris@40: input = float(value - 128) * M_PI / 127.0; Chris@40: break; Chris@40: } Chris@40: Chris@40: return input; Chris@40: } Chris@40: Chris@40: float Chris@40: SpectrogramLayer::getEffectiveMinFrequency() const Chris@40: { Chris@40: int sr = m_model->getSampleRate(); Chris@107: float minf = float(sr) / m_fftSize; Chris@40: Chris@40: if (m_minFrequency > 0.0) { Chris@107: size_t minbin = size_t((double(m_minFrequency) * m_fftSize) / sr + 0.01); Chris@40: if (minbin < 1) minbin = 1; Chris@107: minf = minbin * sr / m_fftSize; Chris@40: } Chris@40: Chris@40: return minf; Chris@40: } Chris@40: Chris@40: float Chris@40: SpectrogramLayer::getEffectiveMaxFrequency() const Chris@40: { Chris@40: int sr = m_model->getSampleRate(); Chris@40: float maxf = float(sr) / 2; Chris@40: Chris@40: if (m_maxFrequency > 0.0) { Chris@107: size_t maxbin = size_t((double(m_maxFrequency) * m_fftSize) / sr + 0.1); Chris@107: if (maxbin > m_fftSize / 2) maxbin = m_fftSize / 2; Chris@107: maxf = maxbin * sr / m_fftSize; Chris@40: } Chris@40: Chris@40: return maxf; Chris@40: } Chris@40: Chris@0: bool Chris@44: SpectrogramLayer::getYBinRange(View *v, int y, float &q0, float &q1) const Chris@0: { Chris@382: Profiler profiler("SpectrogramLayer::getYBinRange"); Chris@382: Chris@44: int h = v->height(); Chris@0: if (y < 0 || y >= h) return false; Chris@0: Chris@38: int sr = m_model->getSampleRate(); Chris@40: float minf = getEffectiveMinFrequency(); Chris@40: float maxf = getEffectiveMaxFrequency(); Chris@0: Chris@38: bool logarithmic = (m_frequencyScale == LogFrequencyScale); Chris@38: Chris@44: q0 = v->getFrequencyForY(y, minf, maxf, logarithmic); Chris@44: q1 = v->getFrequencyForY(y - 1, minf, maxf, logarithmic); Chris@38: Chris@490: // Now map these on to ("proportions of") actual bins, using raw Chris@490: // FFT size (unsmoothed) Chris@490: Chris@490: q0 = (q0 * m_fftSize) / sr; Chris@490: q1 = (q1 * m_fftSize) / sr; Chris@0: Chris@0: return true; Chris@0: } Chris@486: Chris@486: bool Chris@486: SpectrogramLayer::getSmoothedYBinRange(View *v, int y, float &q0, float &q1) const Chris@486: { Chris@486: Profiler profiler("SpectrogramLayer::getSmoothedYBinRange"); Chris@486: Chris@486: int h = v->height(); Chris@486: if (y < 0 || y >= h) return false; Chris@486: Chris@486: int sr = m_model->getSampleRate(); Chris@486: float minf = getEffectiveMinFrequency(); Chris@486: float maxf = getEffectiveMaxFrequency(); Chris@486: Chris@486: bool logarithmic = (m_frequencyScale == LogFrequencyScale); Chris@486: Chris@486: q0 = v->getFrequencyForY(y, minf, maxf, logarithmic); Chris@486: q1 = v->getFrequencyForY(y - 1, minf, maxf, logarithmic); Chris@486: Chris@490: // Now map these on to ("proportions of") actual bins, using raw Chris@490: // FFT size (unsmoothed) Chris@490: Chris@490: q0 = (q0 * getFFTSize(v)) / sr; Chris@490: q1 = (q1 * getFFTSize(v)) / sr; Chris@486: Chris@486: return true; Chris@486: } Chris@38: Chris@0: bool Chris@44: SpectrogramLayer::getXBinRange(View *v, int x, float &s0, float &s1) const Chris@0: { Chris@21: size_t modelStart = m_model->getStartFrame(); Chris@21: size_t modelEnd = m_model->getEndFrame(); Chris@0: Chris@0: // Each pixel column covers an exact range of sample frames: Chris@44: int f0 = v->getFrameForX(x) - modelStart; Chris@44: int f1 = v->getFrameForX(x + 1) - modelStart - 1; Chris@20: Chris@41: if (f1 < int(modelStart) || f0 > int(modelEnd)) { Chris@41: return false; Chris@41: } Chris@20: Chris@0: // And that range may be drawn from a possibly non-integral Chris@0: // range of spectrogram windows: Chris@0: Chris@0: size_t windowIncrement = getWindowIncrement(); Chris@0: s0 = float(f0) / windowIncrement; Chris@0: s1 = float(f1) / windowIncrement; Chris@0: Chris@0: return true; Chris@0: } Chris@0: Chris@0: bool Chris@44: SpectrogramLayer::getXBinSourceRange(View *v, int x, RealTime &min, RealTime &max) const Chris@0: { Chris@0: float s0 = 0, s1 = 0; Chris@44: if (!getXBinRange(v, x, s0, s1)) return false; Chris@0: Chris@0: int s0i = int(s0 + 0.001); Chris@0: int s1i = int(s1); Chris@0: Chris@0: int windowIncrement = getWindowIncrement(); Chris@0: int w0 = s0i * windowIncrement - (m_windowSize - windowIncrement)/2; Chris@0: int w1 = s1i * windowIncrement + windowIncrement + Chris@0: (m_windowSize - windowIncrement)/2 - 1; Chris@0: Chris@0: min = RealTime::frame2RealTime(w0, m_model->getSampleRate()); Chris@0: max = RealTime::frame2RealTime(w1, m_model->getSampleRate()); Chris@0: return true; Chris@0: } Chris@0: Chris@0: bool Chris@44: SpectrogramLayer::getYBinSourceRange(View *v, int y, float &freqMin, float &freqMax) Chris@0: const Chris@0: { Chris@0: float q0 = 0, q1 = 0; Chris@44: if (!getYBinRange(v, y, q0, q1)) return false; Chris@0: Chris@0: int q0i = int(q0 + 0.001); Chris@0: int q1i = int(q1); Chris@0: Chris@0: int sr = m_model->getSampleRate(); Chris@0: Chris@0: for (int q = q0i; q <= q1i; ++q) { Chris@121: if (q == q0i) freqMin = (sr * q) / m_fftSize; Chris@121: if (q == q1i) freqMax = (sr * (q+1)) / m_fftSize; Chris@0: } Chris@0: return true; Chris@0: } Chris@35: Chris@35: bool Chris@44: SpectrogramLayer::getAdjustedYBinSourceRange(View *v, int x, int y, Chris@35: float &freqMin, float &freqMax, Chris@35: float &adjFreqMin, float &adjFreqMax) Chris@35: const Chris@35: { Chris@277: if (!m_model || !m_model->isOK() || !m_model->isReady()) { Chris@277: return false; Chris@277: } Chris@277: Chris@130: FFTModel *fft = getFFTModel(v); Chris@114: if (!fft) return false; Chris@110: Chris@35: float s0 = 0, s1 = 0; Chris@44: if (!getXBinRange(v, x, s0, s1)) return false; Chris@35: Chris@35: float q0 = 0, q1 = 0; Chris@44: if (!getYBinRange(v, y, q0, q1)) return false; Chris@35: Chris@35: int s0i = int(s0 + 0.001); Chris@35: int s1i = int(s1); Chris@35: Chris@35: int q0i = int(q0 + 0.001); Chris@35: int q1i = int(q1); Chris@35: Chris@35: int sr = m_model->getSampleRate(); Chris@35: Chris@38: size_t windowSize = m_windowSize; Chris@38: size_t windowIncrement = getWindowIncrement(); Chris@38: Chris@35: bool haveAdj = false; Chris@35: Chris@37: bool peaksOnly = (m_binDisplay == PeakBins || Chris@37: m_binDisplay == PeakFrequencies); Chris@37: Chris@35: for (int q = q0i; q <= q1i; ++q) { Chris@35: Chris@35: for (int s = s0i; s <= s1i; ++s) { Chris@35: Chris@160: if (!fft->isColumnAvailable(s)) continue; Chris@117: Chris@35: float binfreq = (sr * q) / m_windowSize; Chris@35: if (q == q0i) freqMin = binfreq; Chris@35: if (q == q1i) freqMax = binfreq; Chris@37: Chris@114: if (peaksOnly && !fft->isLocalPeak(s, q)) continue; Chris@38: Chris@253: if (!fft->isOverThreshold(s, q, m_threshold * (m_fftSize/2))) continue; Chris@38: Chris@38: float freq = binfreq; Chris@38: bool steady = false; Chris@40: Chris@114: if (s < int(fft->getWidth()) - 1) { Chris@38: Chris@277: fft->estimateStableFrequency(s, q, freq); Chris@35: Chris@38: if (!haveAdj || freq < adjFreqMin) adjFreqMin = freq; Chris@38: if (!haveAdj || freq > adjFreqMax) adjFreqMax = freq; Chris@35: Chris@35: haveAdj = true; Chris@35: } Chris@35: } Chris@35: } Chris@35: Chris@35: if (!haveAdj) { Chris@40: adjFreqMin = adjFreqMax = 0.0; Chris@35: } Chris@35: Chris@35: return haveAdj; Chris@35: } Chris@0: Chris@0: bool Chris@44: SpectrogramLayer::getXYBinSourceRange(View *v, int x, int y, Chris@38: float &min, float &max, Chris@38: float &phaseMin, float &phaseMax) const Chris@0: { Chris@277: if (!m_model || !m_model->isOK() || !m_model->isReady()) { Chris@277: return false; Chris@277: } Chris@277: Chris@0: float q0 = 0, q1 = 0; Chris@44: if (!getYBinRange(v, y, q0, q1)) return false; Chris@0: Chris@0: float s0 = 0, s1 = 0; Chris@44: if (!getXBinRange(v, x, s0, s1)) return false; Chris@0: Chris@0: int q0i = int(q0 + 0.001); Chris@0: int q1i = int(q1); Chris@0: Chris@0: int s0i = int(s0 + 0.001); Chris@0: int s1i = int(s1); Chris@0: Chris@37: bool rv = false; Chris@37: Chris@122: size_t zp = getZeroPadLevel(v); Chris@122: q0i *= zp + 1; Chris@122: q1i *= zp + 1; Chris@122: Chris@130: FFTModel *fft = getFFTModel(v); Chris@0: Chris@114: if (fft) { Chris@114: Chris@114: int cw = fft->getWidth(); Chris@114: int ch = fft->getHeight(); Chris@0: Chris@110: min = 0.0; Chris@110: max = 0.0; Chris@110: phaseMin = 0.0; Chris@110: phaseMax = 0.0; Chris@110: bool have = false; Chris@0: Chris@110: for (int q = q0i; q <= q1i; ++q) { Chris@110: for (int s = s0i; s <= s1i; ++s) { Chris@110: if (s >= 0 && q >= 0 && s < cw && q < ch) { Chris@117: Chris@160: if (!fft->isColumnAvailable(s)) continue; Chris@110: Chris@110: float value; Chris@38: Chris@114: value = fft->getPhaseAt(s, q); Chris@110: if (!have || value < phaseMin) { phaseMin = value; } Chris@110: if (!have || value > phaseMax) { phaseMax = value; } Chris@91: Chris@252: value = fft->getMagnitudeAt(s, q) / (m_fftSize/2); Chris@110: if (!have || value < min) { min = value; } Chris@110: if (!have || value > max) { max = value; } Chris@110: Chris@110: have = true; Chris@110: } Chris@110: } Chris@110: } Chris@110: Chris@110: if (have) { Chris@110: rv = true; Chris@110: } Chris@0: } Chris@0: Chris@37: return rv; Chris@0: } Chris@0: Chris@114: size_t Chris@114: SpectrogramLayer::getZeroPadLevel(const View *v) const Chris@114: { Chris@114: //!!! tidy all this stuff Chris@114: Chris@114: if (m_binDisplay != AllBins) return 0; Chris@221: Chris@221: Preferences::SpectrogramSmoothing smoothing = Chris@221: Preferences::getInstance()->getSpectrogramSmoothing(); Chris@221: Chris@221: if (smoothing == Preferences::NoSpectrogramSmoothing || Chris@221: smoothing == Preferences::SpectrogramInterpolated) return 0; Chris@221: Chris@114: if (m_frequencyScale == LogFrequencyScale) return 3; Chris@114: Chris@114: int sr = m_model->getSampleRate(); Chris@114: Chris@184: size_t maxbin = m_fftSize / 2; Chris@114: if (m_maxFrequency > 0) { Chris@184: maxbin = int((double(m_maxFrequency) * m_fftSize) / sr + 0.1); Chris@184: if (maxbin > m_fftSize / 2) maxbin = m_fftSize / 2; Chris@114: } Chris@114: Chris@114: size_t minbin = 1; Chris@114: if (m_minFrequency > 0) { Chris@114: minbin = int((double(m_minFrequency) * m_fftSize) / sr + 0.1); Chris@114: if (minbin < 1) minbin = 1; Chris@184: if (minbin >= maxbin) minbin = maxbin - 1; Chris@114: } Chris@114: Chris@118: float perPixel = Chris@118: float(v->height()) / Chris@184: float((maxbin - minbin) / (m_zeroPadLevel + 1)); Chris@118: Chris@118: if (perPixel > 2.8) { Chris@118: return 3; // 4x oversampling Chris@118: } else if (perPixel > 1.5) { Chris@118: return 1; // 2x Chris@114: } else { Chris@118: return 0; // 1x Chris@114: } Chris@114: } Chris@114: Chris@114: size_t Chris@114: SpectrogramLayer::getFFTSize(const View *v) const Chris@114: { Chris@114: return m_fftSize * (getZeroPadLevel(v) + 1); Chris@114: } Chris@114: Chris@130: FFTModel * Chris@130: SpectrogramLayer::getFFTModel(const View *v) const Chris@114: { Chris@114: if (!m_model) return 0; Chris@114: Chris@114: size_t fftSize = getFFTSize(v); Chris@114: Chris@130: if (m_fftModels.find(v) != m_fftModels.end()) { Chris@184: if (m_fftModels[v].first == 0) { Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::getFFTModel(" << v << "): Found null model" << endl; Chris@184: #endif Chris@184: return 0; Chris@184: } Chris@184: if (m_fftModels[v].first->getHeight() != fftSize / 2 + 1) { Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::getFFTModel(" << v << "): Found a model with the wrong height (" << m_fftModels[v].first->getHeight() << ", wanted " << (fftSize / 2 + 1) << ")" << endl; Chris@184: #endif Chris@130: delete m_fftModels[v].first; Chris@130: m_fftModels.erase(v); Chris@484: delete m_peakCaches[v]; Chris@484: m_peakCaches.erase(v); Chris@184: } else { Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::getFFTModel(" << v << "): Found a good model of height " << m_fftModels[v].first->getHeight() << endl; Chris@184: #endif Chris@184: return m_fftModels[v].first; Chris@114: } Chris@114: } Chris@114: Chris@130: if (m_fftModels.find(v) == m_fftModels.end()) { Chris@169: Chris@169: FFTModel *model = new FFTModel(m_model, Chris@169: m_channel, Chris@169: m_windowType, Chris@169: m_windowSize, Chris@169: getWindowIncrement(), Chris@169: fftSize, Chris@382: true, // polar Chris@327: StorageAdviser::SpeedCritical, Chris@169: m_candidateFillStartFrame); Chris@169: Chris@178: if (!model->isOK()) { Chris@178: QMessageBox::critical Chris@178: (0, tr("FFT cache failed"), Chris@178: tr("Failed to create the FFT model for this spectrogram.\n" Chris@178: "There may be insufficient memory or disc space to continue.")); Chris@178: delete model; Chris@178: m_fftModels[v] = FFTFillPair(0, 0); Chris@178: return 0; Chris@178: } Chris@178: Chris@193: if (!m_sliceableModel) { Chris@248: #ifdef DEBUG_SPECTROGRAM Chris@193: std::cerr << "SpectrogramLayer: emitting sliceableModelReplaced(0, " << model << ")" << std::endl; Chris@248: #endif Chris@193: ((SpectrogramLayer *)this)->sliceableModelReplaced(0, model); Chris@193: m_sliceableModel = model; Chris@193: } Chris@193: Chris@169: m_fftModels[v] = FFTFillPair(model, 0); Chris@169: Chris@169: model->resume(); Chris@114: Chris@114: delete m_updateTimer; Chris@114: m_updateTimer = new QTimer((SpectrogramLayer *)this); Chris@114: connect(m_updateTimer, SIGNAL(timeout()), Chris@114: this, SLOT(fillTimerTimedOut())); Chris@114: m_updateTimer->start(200); Chris@114: } Chris@114: Chris@130: return m_fftModels[v].first; Chris@114: } Chris@114: Chris@484: Dense3DModelPeakCache * Chris@484: SpectrogramLayer::getPeakCache(const View *v) const Chris@484: { Chris@484: if (!m_peakCaches[v]) { Chris@484: FFTModel *f = getFFTModel(v); Chris@484: if (!f) return 0; Chris@484: m_peakCaches[v] = new Dense3DModelPeakCache(f, 8); Chris@484: } Chris@484: return m_peakCaches[v]; Chris@484: } Chris@484: Chris@193: const Model * Chris@193: SpectrogramLayer::getSliceableModel() const Chris@193: { Chris@193: if (m_sliceableModel) return m_sliceableModel; Chris@193: if (m_fftModels.empty()) return 0; Chris@193: m_sliceableModel = m_fftModels.begin()->second.first; Chris@193: return m_sliceableModel; Chris@193: } Chris@193: Chris@114: void Chris@130: SpectrogramLayer::invalidateFFTModels() Chris@114: { Chris@130: for (ViewFFTMap::iterator i = m_fftModels.begin(); Chris@130: i != m_fftModels.end(); ++i) { Chris@115: delete i->second.first; Chris@114: } Chris@486: for (PeakCacheMap::iterator i = m_peakCaches.begin(); Chris@486: i != m_peakCaches.end(); ++i) { Chris@486: delete i->second; Chris@486: } Chris@114: Chris@130: m_fftModels.clear(); Chris@486: m_peakCaches.clear(); Chris@193: Chris@193: if (m_sliceableModel) { Chris@193: std::cerr << "SpectrogramLayer: emitting sliceableModelReplaced(" << m_sliceableModel << ", 0)" << std::endl; Chris@193: emit sliceableModelReplaced(m_sliceableModel, 0); Chris@193: m_sliceableModel = 0; Chris@193: } Chris@114: } Chris@114: Chris@0: void Chris@119: SpectrogramLayer::invalidateMagnitudes() Chris@119: { Chris@119: m_viewMags.clear(); Chris@119: for (std::vector::iterator i = m_columnMags.begin(); Chris@119: i != m_columnMags.end(); ++i) { Chris@119: *i = MagnitudeRange(); Chris@119: } Chris@119: } Chris@119: Chris@119: bool Chris@119: SpectrogramLayer::updateViewMagnitudes(View *v) const Chris@119: { Chris@119: MagnitudeRange mag; Chris@119: Chris@119: int x0 = 0, x1 = v->width(); Chris@119: float s00 = 0, s01 = 0, s10 = 0, s11 = 0; Chris@119: Chris@203: if (!getXBinRange(v, x0, s00, s01)) { Chris@203: s00 = s01 = m_model->getStartFrame() / getWindowIncrement(); Chris@203: } Chris@203: Chris@203: if (!getXBinRange(v, x1, s10, s11)) { Chris@203: s10 = s11 = m_model->getEndFrame() / getWindowIncrement(); Chris@203: } Chris@119: Chris@119: int s0 = int(std::min(s00, s10) + 0.0001); Chris@203: int s1 = int(std::max(s01, s11) + 0.0001); Chris@203: Chris@587: // SVDEBUG << "SpectrogramLayer::updateViewMagnitudes: x0 = " << x0 << ", x1 = " << x1 << ", s00 = " << s00 << ", s11 = " << s11 << " s0 = " << s0 << ", s1 = " << s1 << endl; Chris@119: Chris@248: if (int(m_columnMags.size()) <= s1) { Chris@119: m_columnMags.resize(s1 + 1); Chris@119: } Chris@119: Chris@119: for (int s = s0; s <= s1; ++s) { Chris@119: if (m_columnMags[s].isSet()) { Chris@119: mag.sample(m_columnMags[s]); Chris@119: } Chris@119: } Chris@119: Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::updateViewMagnitudes returning from cols " Chris@585: << s0 << " -> " << s1 << " inclusive" << endl; Chris@184: #endif Chris@119: Chris@119: if (!mag.isSet()) return false; Chris@119: if (mag == m_viewMags[v]) return false; Chris@119: m_viewMags[v] = mag; Chris@119: return true; Chris@119: } Chris@119: Chris@119: void Chris@389: SpectrogramLayer::setSynchronousPainting(bool synchronous) Chris@389: { Chris@389: m_synchronous = synchronous; Chris@389: } Chris@389: Chris@389: void Chris@44: SpectrogramLayer::paint(View *v, QPainter &paint, QRect rect) const Chris@0: { Chris@253: // What a lovely, old-fashioned function this is. Chris@253: // It's practically FORTRAN 77 in its clarity and linearity. Chris@253: Chris@334: Profiler profiler("SpectrogramLayer::paint", false); Chris@334: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::paint(): m_model is " << m_model << ", zoom level is " << v->getZoomLevel() << ", m_updateTimer " << m_updateTimer << endl; Chris@95: Chris@95: std::cerr << "rect is " << rect.x() << "," << rect.y() << " " << rect.width() << "x" << rect.height() << std::endl; Chris@0: #endif Chris@95: Chris@133: long startFrame = v->getStartFrame(); Chris@133: if (startFrame < 0) m_candidateFillStartFrame = 0; Chris@133: else m_candidateFillStartFrame = startFrame; Chris@44: Chris@0: if (!m_model || !m_model->isOK() || !m_model->isReady()) { Chris@0: return; Chris@0: } Chris@0: Chris@47: if (isLayerDormant(v)) { Chris@587: SVDEBUG << "SpectrogramLayer::paint(): Layer is dormant, making it undormant again" << endl; Chris@29: } Chris@29: Chris@48: // Need to do this even if !isLayerDormant, as that could mean v Chris@48: // is not in the dormancy map at all -- we need it to be present Chris@48: // and accountable for when determining whether we need the cache Chris@48: // in the cache-fill thread above. Chris@131: //!!! no longer use cache-fill thread Chris@131: const_cast(this)->Layer::setLayerDormant(v, false); Chris@48: Chris@114: size_t fftSize = getFFTSize(v); Chris@484: /* Chris@130: FFTModel *fft = getFFTModel(v); Chris@114: if (!fft) { Chris@130: std::cerr << "ERROR: SpectrogramLayer::paint(): No FFT model, returning" << std::endl; Chris@0: return; Chris@0: } Chris@484: */ Chris@478: ImageCache &cache = m_imageCaches[v]; Chris@95: Chris@95: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::paint(): image cache valid area " << cache. Chris@585: Chris@585: validArea.x() << ", " << cache.validArea.y() << ", " << cache.validArea.width() << "x" << cache.validArea.height() << endl; Chris@95: #endif Chris@95: Chris@248: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@0: bool stillCacheing = (m_updateTimer != 0); Chris@587: SVDEBUG << "SpectrogramLayer::paint(): Still cacheing = " << stillCacheing << endl; Chris@0: #endif Chris@0: Chris@44: int zoomLevel = v->getZoomLevel(); Chris@0: Chris@0: int x0 = 0; Chris@44: int x1 = v->width(); Chris@0: Chris@478: bool recreateWholeImageCache = true; Chris@0: Chris@95: x0 = rect.left(); Chris@95: x1 = rect.right() + 1; Chris@481: /* Chris@479: float xPixelRatio = float(fft->getResolution()) / float(zoomLevel); Chris@479: std::cerr << "xPixelRatio = " << xPixelRatio << std::endl; Chris@479: if (xPixelRatio < 1.f) xPixelRatio = 1.f; Chris@481: */ Chris@95: if (cache.validArea.width() > 0) { Chris@95: Chris@482: int cw = cache.image.width(); Chris@482: int ch = cache.image.height(); Chris@482: Chris@95: if (int(cache.zoomLevel) == zoomLevel && Chris@482: cw == v->width() && Chris@482: ch == v->height()) { Chris@95: Chris@95: if (v->getXForFrame(cache.startFrame) == Chris@95: v->getXForFrame(startFrame) && Chris@95: cache.validArea.x() <= x0 && Chris@95: cache.validArea.x() + cache.validArea.width() >= x1) { Chris@0: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@478: std::cerr << "SpectrogramLayer: image cache good" << std::endl; Chris@0: #endif Chris@0: Chris@478: paint.drawImage(rect, cache.image, rect); Chris@479: //!!! Chris@479: // paint.drawImage(v->rect(), cache.image, Chris@479: // QRect(QPoint(0, 0), cache.image.size())); Chris@479: Chris@121: illuminateLocalFeatures(v, paint); Chris@0: return; Chris@0: Chris@0: } else { Chris@0: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@478: std::cerr << "SpectrogramLayer: image cache partially OK" << std::endl; Chris@0: #endif Chris@0: Chris@478: recreateWholeImageCache = false; Chris@0: Chris@95: int dx = v->getXForFrame(cache.startFrame) - Chris@44: v->getXForFrame(startFrame); Chris@0: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@482: std::cerr << "SpectrogramLayer: dx = " << dx << " (image cache " << cw << "x" << ch << ")" << std::endl; Chris@0: #endif Chris@0: Chris@95: if (dx != 0 && Chris@482: dx > -cw && Chris@482: dx < cw) { Chris@482: Chris@482: int dxp = dx; Chris@482: if (dxp < 0) dxp = -dxp; Chris@482: int copy = (cw - dxp) * sizeof(QRgb); Chris@482: for (int y = 0; y < ch; ++y) { Chris@482: QRgb *line = (QRgb *)cache.image.scanLine(y); Chris@482: if (dx < 0) { Chris@482: memmove(line, line + dxp, copy); Chris@482: } else { Chris@482: memmove(line + dxp, line, copy); Chris@482: } Chris@331: } Chris@0: Chris@95: int px = cache.validArea.x(); Chris@95: int pw = cache.validArea.width(); Chris@0: Chris@0: if (dx < 0) { Chris@482: x0 = cw + dx; Chris@482: x1 = cw; Chris@95: px += dx; Chris@95: if (px < 0) { Chris@95: pw += px; Chris@95: px = 0; Chris@95: if (pw < 0) pw = 0; Chris@95: } Chris@0: } else { Chris@0: x0 = 0; Chris@0: x1 = dx; Chris@95: px += dx; Chris@482: if (px + pw > cw) { Chris@482: pw = int(cw) - px; Chris@95: if (pw < 0) pw = 0; Chris@95: } Chris@0: } Chris@95: Chris@95: cache.validArea = Chris@95: QRect(px, cache.validArea.y(), Chris@95: pw, cache.validArea.height()); Chris@95: Chris@331: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@331: std::cerr << "valid area now " Chris@331: << px << "," << cache.validArea.y() Chris@331: << " " << pw << "x" << cache.validArea.height() Chris@331: << std::endl; Chris@331: #endif Chris@479: /* Chris@478: paint.drawImage(rect & cache.validArea, Chris@478: cache.image, Chris@95: rect & cache.validArea); Chris@479: */ Chris@331: } else if (dx != 0) { Chris@331: Chris@331: // we scrolled too far to be of use Chris@331: Chris@391: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@391: std::cerr << "dx == " << dx << ": scrolled too far for cache to be useful" << std::endl; Chris@391: #endif Chris@391: Chris@331: cache.validArea = QRect(); Chris@478: recreateWholeImageCache = true; Chris@331: } Chris@0: } Chris@0: } else { Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@478: std::cerr << "SpectrogramLayer: image cache useless" << std::endl; Chris@224: if (int(cache.zoomLevel) != zoomLevel) { Chris@224: std::cerr << "(cache zoomLevel " << cache.zoomLevel Chris@224: << " != " << zoomLevel << ")" << std::endl; Chris@224: } Chris@482: if (cw != v->width()) { Chris@482: std::cerr << "(cache width " << cw Chris@224: << " != " << v->width(); Chris@224: } Chris@482: if (ch != v->height()) { Chris@482: std::cerr << "(cache height " << ch Chris@224: << " != " << v->height(); Chris@224: } Chris@0: #endif Chris@95: cache.validArea = QRect(); Chris@478: // recreateWholeImageCache = true; Chris@0: } Chris@0: } Chris@95: Chris@133: if (updateViewMagnitudes(v)) { Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@133: std::cerr << "SpectrogramLayer: magnitude range changed to [" << m_viewMags[v].getMin() << "->" << m_viewMags[v].getMax() << "]" << std::endl; Chris@184: #endif Chris@331: if (m_normalizeVisibleArea) { Chris@331: cache.validArea = QRect(); Chris@478: recreateWholeImageCache = true; Chris@331: } Chris@133: } else { Chris@184: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@133: std::cerr << "No change in magnitude range [" << m_viewMags[v].getMin() << "->" << m_viewMags[v].getMax() << "]" << std::endl; Chris@184: #endif Chris@133: } Chris@133: Chris@478: if (recreateWholeImageCache) { Chris@95: x0 = 0; Chris@95: x1 = v->width(); Chris@95: } Chris@95: Chris@215: struct timeval tv; Chris@215: (void)gettimeofday(&tv, 0); Chris@215: RealTime mainPaintStart = RealTime::fromTimeval(tv); Chris@215: Chris@215: int paintBlockWidth = m_lastPaintBlockWidth; Chris@215: Chris@389: if (m_synchronous) { Chris@389: if (paintBlockWidth < x1 - x0) { Chris@389: // always paint full width Chris@389: paintBlockWidth = x1 - x0; Chris@389: } Chris@215: } else { Chris@389: if (paintBlockWidth == 0) { Chris@389: paintBlockWidth = (300000 / zoomLevel); Chris@389: } else { Chris@389: RealTime lastTime = m_lastPaintTime; Chris@389: while (lastTime > RealTime::fromMilliseconds(200) && Chris@389: paintBlockWidth > 50) { Chris@389: paintBlockWidth /= 2; Chris@389: lastTime = lastTime / 2; Chris@389: } Chris@389: while (lastTime < RealTime::fromMilliseconds(90) && Chris@389: paintBlockWidth < 1500) { Chris@389: paintBlockWidth *= 2; Chris@389: lastTime = lastTime * 2; Chris@389: } Chris@215: } Chris@389: Chris@389: if (paintBlockWidth < 20) paintBlockWidth = 20; Chris@215: } Chris@215: Chris@525: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@215: std::cerr << "[" << this << "]: last paint width: " << m_lastPaintBlockWidth << ", last paint time: " << m_lastPaintTime << ", new paint width: " << paintBlockWidth << std::endl; Chris@525: #endif Chris@224: Chris@224: // We always paint the full height when refreshing the cache. Chris@224: // Smaller heights can be used when painting direct from cache Chris@224: // (further up in this function), but we want to ensure the cache Chris@224: // is coherent without having to worry about vertical matching of Chris@224: // required and valid areas as well as horizontal. Chris@224: Chris@224: int h = v->height(); Chris@215: Chris@96: if (cache.validArea.width() > 0) { Chris@96: Chris@331: // If part of the cache is known to be valid, select a strip Chris@331: // immediately to left or right of the valid part Chris@331: Chris@481: //!!! this really needs to be coordinated with the selection Chris@481: //!!! of m_drawBuffer boundaries in the bufferBinResolution Chris@481: //!!! case below Chris@481: Chris@96: int vx0 = 0, vx1 = 0; Chris@96: vx0 = cache.validArea.x(); Chris@96: vx1 = cache.validArea.x() + cache.validArea.width(); Chris@96: Chris@96: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@96: std::cerr << "x0 " << x0 << ", x1 " << x1 << ", vx0 " << vx0 << ", vx1 " << vx1 << ", paintBlockWidth " << paintBlockWidth << std::endl; Chris@331: #endif Chris@96: if (x0 < vx0) { Chris@96: if (x0 + paintBlockWidth < vx0) { Chris@96: x0 = vx0 - paintBlockWidth; Chris@331: } Chris@331: x1 = vx0; Chris@331: } else if (x0 >= vx1) { Chris@331: x0 = vx1; Chris@331: if (x1 > x0 + paintBlockWidth) { Chris@331: x1 = x0 + paintBlockWidth; Chris@331: } Chris@331: } else { Chris@331: // x0 is within the valid area Chris@331: if (x1 > vx1) { Chris@331: x0 = vx1; Chris@331: if (x0 + paintBlockWidth < x1) { Chris@331: x1 = x0 + paintBlockWidth; Chris@331: } Chris@96: } else { Chris@331: x1 = x0; // it's all valid, paint nothing Chris@95: } Chris@96: } Chris@331: Chris@96: cache.validArea = QRect Chris@96: (std::min(vx0, x0), cache.validArea.y(), Chris@96: std::max(vx1 - std::min(vx0, x0), Chris@337: x1 - std::min(vx0, x0)), Chris@96: cache.validArea.height()); Chris@337: Chris@337: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@337: std::cerr << "Valid area becomes " << cache.validArea.x() Chris@337: << ", " << cache.validArea.y() << ", " Chris@337: << cache.validArea.width() << "x" Chris@337: << cache.validArea.height() << std::endl; Chris@337: #endif Chris@95: Chris@96: } else { Chris@96: if (x1 > x0 + paintBlockWidth) { Chris@133: int sfx = x1; Chris@133: if (startFrame < 0) sfx = v->getXForFrame(0); Chris@133: if (sfx >= x0 && sfx + paintBlockWidth <= x1) { Chris@133: x0 = sfx; Chris@133: x1 = x0 + paintBlockWidth; Chris@133: } else { Chris@133: int mid = (x1 + x0) / 2; Chris@133: x0 = mid - paintBlockWidth/2; Chris@133: x1 = x0 + paintBlockWidth; Chris@133: } Chris@95: } Chris@337: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@337: std::cerr << "Valid area becomes " << x0 << ", 0, " << (x1-x0) Chris@337: << "x" << h << std::endl; Chris@337: #endif Chris@224: cache.validArea = QRect(x0, 0, x1 - x0, h); Chris@95: } Chris@95: Chris@481: /* Chris@480: if (xPixelRatio != 1.f) { Chris@480: x0 = int((int(x0 / xPixelRatio) - 4) * xPixelRatio + 0.0001); Chris@480: x1 = int((int(x1 / xPixelRatio) + 4) * xPixelRatio + 0.0001); Chris@480: } Chris@481: */ Chris@0: int w = x1 - x0; Chris@0: Chris@95: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@95: std::cerr << "x0 " << x0 << ", x1 " << x1 << ", w " << w << ", h " << h << std::endl; Chris@95: #endif Chris@95: Chris@37: int sr = m_model->getSampleRate(); Chris@122: Chris@122: // Set minFreq and maxFreq to the frequency extents of the possibly Chris@122: // zero-padded visible bin range, and displayMinFreq and displayMaxFreq Chris@122: // to the actual scale frequency extents (presumably not zero padded). Chris@253: Chris@253: // If we are zero padding, we want to use the zero-padded Chris@253: // equivalents of the bins that we would be using if not zero Chris@253: // padded, to avoid spaces at the top and bottom of the display. Chris@253: Chris@253: // Note fftSize is the actual zero-padded fft size, m_fftSize the Chris@253: // nominal fft size. Chris@35: Chris@253: size_t maxbin = m_fftSize / 2; Chris@35: if (m_maxFrequency > 0) { Chris@253: maxbin = int((double(m_maxFrequency) * m_fftSize) / sr + 0.001); Chris@253: if (maxbin > m_fftSize / 2) maxbin = m_fftSize / 2; Chris@35: } Chris@111: Chris@40: size_t minbin = 1; Chris@37: if (m_minFrequency > 0) { Chris@253: minbin = int((double(m_minFrequency) * m_fftSize) / sr + 0.001); Chris@253: // std::cerr << "m_minFrequency = " << m_minFrequency << " -> minbin = " << minbin << std::endl; Chris@40: if (minbin < 1) minbin = 1; Chris@184: if (minbin >= maxbin) minbin = maxbin - 1; Chris@37: } Chris@37: Chris@253: int zpl = getZeroPadLevel(v) + 1; Chris@253: minbin = minbin * zpl; Chris@253: maxbin = (maxbin + 1) * zpl - 1; Chris@253: Chris@114: float minFreq = (float(minbin) * sr) / fftSize; Chris@184: float maxFreq = (float(maxbin) * sr) / fftSize; Chris@0: Chris@122: float displayMinFreq = minFreq; Chris@122: float displayMaxFreq = maxFreq; Chris@122: Chris@122: if (fftSize != m_fftSize) { Chris@122: displayMinFreq = getEffectiveMinFrequency(); Chris@122: displayMaxFreq = getEffectiveMaxFrequency(); Chris@122: } Chris@122: Chris@253: // std::cerr << "(giving actual minFreq " << minFreq << " and display minFreq " << displayMinFreq << ")" << std::endl; Chris@253: Chris@518: int increment = getWindowIncrement(); Chris@40: Chris@40: bool logarithmic = (m_frequencyScale == LogFrequencyScale); Chris@488: /* Chris@481: float yforbin[maxbin - minbin + 1]; Chris@481: Chris@184: for (size_t q = minbin; q <= maxbin; ++q) { Chris@114: float f0 = (float(q) * sr) / fftSize; Chris@477: yforbin[q - minbin] = Chris@382: v->getYForFrequency(f0, displayMinFreq, displayMaxFreq, Chris@382: logarithmic); Chris@92: } Chris@488: */ Chris@119: MagnitudeRange overallMag = m_viewMags[v]; Chris@119: bool overallMagChanged = false; Chris@119: Chris@162: bool fftSuspended = false; Chris@131: Chris@137: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@224: std::cerr << ((float(v->getFrameForX(1) - v->getFrameForX(0))) / increment) << " bin(s) per pixel" << std::endl; Chris@137: #endif Chris@137: Chris@224: bool runOutOfData = false; Chris@224: Chris@331: if (w == 0) { Chris@587: SVDEBUG << "*** NOTE: w == 0" << endl; Chris@331: } Chris@331: Chris@331: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@331: size_t pixels = 0; Chris@331: #endif Chris@331: Chris@382: Profiler outerprof("SpectrogramLayer::paint: all cols"); Chris@382: Chris@481: // The draw buffer contains a fragment at either our pixel Chris@481: // resolution (if there is more than one time-bin per pixel) or Chris@481: // time-bin resolution (if a time-bin spans more than one pixel). Chris@481: // We need to ensure that it starts and ends at points where a Chris@481: // time-bin boundary occurs at an exact pixel boundary, and with a Chris@481: // certain amount of overlap across existing pixels so that we can Chris@481: // scale and draw from it without smoothing errors at the edges. Chris@481: Chris@481: // If (getFrameForX(x) / increment) * increment == Chris@481: // getFrameForX(x), then x is a time-bin boundary. We want two Chris@481: // such boundaries at either side of the draw buffer -- one which Chris@481: // we draw up to, and one which we subsequently crop at. Chris@481: Chris@481: bool bufferBinResolution = false; Chris@481: if (increment > zoomLevel) bufferBinResolution = true; Chris@481: Chris@481: long leftBoundaryFrame = -1, leftCropFrame = -1; Chris@481: long rightBoundaryFrame = -1, rightCropFrame = -1; Chris@481: Chris@481: int bufwid; Chris@481: Chris@481: if (bufferBinResolution) { Chris@481: Chris@482: for (int x = x0; ; --x) { Chris@481: long f = v->getFrameForX(x); Chris@481: if ((f / increment) * increment == f) { Chris@481: if (leftCropFrame == -1) leftCropFrame = f; Chris@482: else if (x < x0 - 2) { leftBoundaryFrame = f; break; } Chris@481: } Chris@481: } Chris@482: for (int x = x0 + w; ; ++x) { Chris@481: long f = v->getFrameForX(x); Chris@481: if ((f / increment) * increment == f) { Chris@481: if (rightCropFrame == -1) rightCropFrame = f; Chris@482: else if (x > x0 + w + 2) { rightBoundaryFrame = f; break; } Chris@481: } Chris@481: } Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@481: cerr << "Left: crop: " << leftCropFrame << " (bin " << leftCropFrame/increment << "); boundary: " << leftBoundaryFrame << " (bin " << leftBoundaryFrame/increment << ")" << endl; Chris@481: cerr << "Right: crop: " << rightCropFrame << " (bin " << rightCropFrame/increment << "); boundary: " << rightBoundaryFrame << " (bin " << rightBoundaryFrame/increment << ")" << endl; Chris@485: #endif Chris@481: Chris@481: bufwid = (rightBoundaryFrame - leftBoundaryFrame) / increment; Chris@481: Chris@481: } else { Chris@481: Chris@481: bufwid = w; Chris@481: } Chris@481: Chris@545: #ifdef __GNUC__ Chris@481: int binforx[bufwid]; Chris@490: float binfory[h]; Chris@545: #else Chris@545: int *binforx = (int *)alloca(bufwid * sizeof(int)); Chris@545: float *binfory = (float *)alloca(h * sizeof(float)); Chris@545: #endif Chris@481: Chris@484: bool usePeaksCache = false; Chris@484: Chris@481: if (bufferBinResolution) { Chris@481: for (int x = 0; x < bufwid; ++x) { Chris@481: binforx[x] = (leftBoundaryFrame / increment) + x; Chris@482: // cerr << "binforx[" << x << "] = " << binforx[x] << endl; Chris@481: } Chris@481: m_drawBuffer = QImage(bufwid, h, QImage::Format_Indexed8); Chris@481: } else { Chris@481: for (int x = 0; x < bufwid; ++x) { Chris@481: float s0 = 0, s1 = 0; Chris@481: if (getXBinRange(v, x + x0, s0, s1)) { Chris@481: binforx[x] = int(s0 + 0.0001); Chris@481: } else { Chris@487: binforx[x] = -1; //??? Chris@481: } Chris@481: } Chris@481: if (m_drawBuffer.width() < bufwid || m_drawBuffer.height() < h) { Chris@481: m_drawBuffer = QImage(bufwid, h, QImage::Format_Indexed8); Chris@480: } Chris@484: usePeaksCache = (increment * 8) < zoomLevel; Chris@487: if (m_colourScale == PhaseColourScale) usePeaksCache = false; Chris@480: } Chris@481: Chris@481: m_drawBuffer.setNumColors(256); Chris@481: for (int pixel = 0; pixel < 256; ++pixel) { Chris@481: m_drawBuffer.setColor(pixel, m_palette.getColour(pixel).rgb()); Chris@481: } Chris@481: Chris@481: m_drawBuffer.fill(0); Chris@480: Chris@488: if (m_binDisplay != PeakFrequencies) { Chris@488: Chris@488: for (int y = 0; y < h; ++y) { Chris@488: float q0 = 0, q1 = 0; Chris@488: if (!getSmoothedYBinRange(v, h-y-1, q0, q1)) { Chris@488: binfory[y] = -1; Chris@488: } else { Chris@490: binfory[y] = q0; Chris@488: // cerr << "binfory[" << y << "] = " << binfory[y] << endl; Chris@488: } Chris@480: } Chris@488: Chris@491: paintDrawBuffer(v, bufwid, h, binforx, binfory, usePeaksCache, Chris@491: overallMag, overallMagChanged); Chris@488: Chris@488: } else { Chris@488: Chris@488: paintDrawBufferPeakFrequencies(v, bufwid, h, binforx, Chris@488: minbin, maxbin, Chris@488: displayMinFreq, displayMaxFreq, Chris@491: logarithmic, Chris@491: overallMag, overallMagChanged); Chris@480: } Chris@481: Chris@480: /* Chris@479: for (int x = 0; x < w / xPixelRatio; ++x) { Chris@35: Chris@382: Profiler innerprof("SpectrogramLayer::paint: 1 pixel column"); Chris@382: Chris@478: runOutOfData = !paintColumnValues(v, fft, x0, x, Chris@478: minbin, maxbin, Chris@478: displayMinFreq, displayMaxFreq, Chris@479: xPixelRatio, Chris@478: h, yforbin); Chris@477: Chris@331: if (runOutOfData) { Chris@331: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@331: std::cerr << "Run out of data -- dropping out of loop" << std::endl; Chris@331: #endif Chris@331: break; Chris@331: } Chris@35: } Chris@480: */ Chris@331: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@480: // std::cerr << pixels << " pixels drawn" << std::endl; Chris@331: #endif Chris@331: Chris@119: if (overallMagChanged) { Chris@119: m_viewMags[v] = overallMag; Chris@209: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@119: std::cerr << "Overall mag is now [" << m_viewMags[v].getMin() << "->" << m_viewMags[v].getMax() << "] - will be updating" << std::endl; Chris@209: #endif Chris@119: } else { Chris@209: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@119: std::cerr << "Overall mag unchanged at [" << m_viewMags[v].getMin() << "->" << m_viewMags[v].getMax() << "]" << std::endl; Chris@209: #endif Chris@119: } Chris@119: Chris@382: outerprof.end(); Chris@382: Chris@382: Profiler profiler2("SpectrogramLayer::paint: draw image"); Chris@137: Chris@478: if (recreateWholeImageCache) { Chris@407: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Recreating image cache: width = " << v->width() Chris@585: << ", height = " << h << endl; Chris@407: #endif Chris@556: cache.image = QImage(v->width(), h, QImage::Format_ARGB32_Premultiplied); Chris@0: } Chris@0: Chris@331: if (w > 0) { Chris@224: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Painting " << w << "x" << h Chris@331: << " from draw buffer at " << 0 << "," << 0 Chris@480: << " to " << w << "x" << h << " on cache at " Chris@585: << x0 << "," << 0 << endl; Chris@224: #endif Chris@224: Chris@478: QPainter cachePainter(&cache.image); Chris@481: Chris@481: if (bufferBinResolution) { Chris@481: int scaledLeft = v->getXForFrame(leftBoundaryFrame); Chris@481: int scaledRight = v->getXForFrame(rightBoundaryFrame); Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Rescaling image from " << bufwid Chris@481: << "x" << h << " to " Chris@481: << scaledRight-scaledLeft << "x" << h << endl; Chris@485: #endif Chris@490: Preferences::SpectrogramXSmoothing xsmoothing = Chris@490: Preferences::getInstance()->getSpectrogramXSmoothing(); Chris@587: // SVDEBUG << "xsmoothing == " << xsmoothing << endl; Chris@481: QImage scaled = m_drawBuffer.scaled Chris@481: (scaledRight - scaledLeft, h, Chris@490: Qt::IgnoreAspectRatio, Chris@490: ((xsmoothing == Preferences::SpectrogramXInterpolated) ? Chris@490: Qt::SmoothTransformation : Qt::FastTransformation)); Chris@481: int scaledLeftCrop = v->getXForFrame(leftCropFrame); Chris@481: int scaledRightCrop = v->getXForFrame(rightCropFrame); Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Drawing image region of width " << scaledRightCrop - scaledLeftCrop << " to " Chris@481: << scaledLeftCrop << " from " << scaledLeftCrop - scaledLeft << endl; Chris@485: #endif Chris@481: cachePainter.drawImage Chris@481: (QRect(scaledLeftCrop, 0, Chris@481: scaledRightCrop - scaledLeftCrop, h), Chris@481: scaled, Chris@481: QRect(scaledLeftCrop - scaledLeft, 0, Chris@481: scaledRightCrop - scaledLeftCrop, h)); Chris@481: } else { Chris@481: cachePainter.drawImage(QRect(x0, 0, w, h), Chris@481: m_drawBuffer, Chris@481: QRect(0, 0, w, h)); Chris@481: } Chris@481: Chris@331: cachePainter.end(); Chris@331: } Chris@331: Chris@337: QRect pr = rect & cache.validArea; Chris@337: Chris@337: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Painting " << pr.width() << "x" << pr.height() Chris@337: << " from cache at " << pr.x() << "," << pr.y() Chris@585: << " to window" << endl; Chris@337: #endif Chris@337: Chris@478: paint.drawImage(pr.x(), pr.y(), cache.image, Chris@479: pr.x(), pr.y(), pr.width(), pr.height()); Chris@479: //!!! Chris@479: // paint.drawImage(v->rect(), cache.image, Chris@479: // QRect(QPoint(0, 0), cache.image.size())); Chris@337: Chris@331: cache.startFrame = startFrame; Chris@331: cache.zoomLevel = zoomLevel; Chris@119: Chris@389: if (!m_synchronous) { Chris@389: Chris@389: if (!m_normalizeVisibleArea || !overallMagChanged) { Chris@0: Chris@389: if (cache.validArea.x() > 0) { Chris@95: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::paint() updating left (0, " Chris@585: << cache.validArea.x() << ")" << endl; Chris@95: #endif Chris@389: v->update(0, 0, cache.validArea.x(), h); Chris@389: } Chris@389: Chris@389: if (cache.validArea.x() + cache.validArea.width() < Chris@478: cache.image.width()) { Chris@389: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::paint() updating right (" Chris@389: << cache.validArea.x() + cache.validArea.width() Chris@389: << ", " Chris@478: << cache.image.width() - (cache.validArea.x() + Chris@389: cache.validArea.width()) Chris@585: << ")" << endl; Chris@389: #endif Chris@389: v->update(cache.validArea.x() + cache.validArea.width(), Chris@389: 0, Chris@478: cache.image.width() - (cache.validArea.x() + Chris@389: cache.validArea.width()), Chris@389: h); Chris@389: } Chris@389: } else { Chris@389: // overallMagChanged Chris@389: std::cerr << "\noverallMagChanged - updating all\n" << std::endl; Chris@389: cache.validArea = QRect(); Chris@389: v->update(); Chris@119: } Chris@95: } Chris@0: Chris@121: illuminateLocalFeatures(v, paint); Chris@120: Chris@0: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::paint() returning" << endl; Chris@0: #endif Chris@131: Chris@389: if (!m_synchronous) { Chris@389: m_lastPaintBlockWidth = paintBlockWidth; Chris@389: (void)gettimeofday(&tv, 0); Chris@389: m_lastPaintTime = RealTime::fromTimeval(tv) - mainPaintStart; Chris@389: } Chris@215: Chris@473: //!!! if (fftSuspended) fft->resume(); Chris@0: } Chris@0: Chris@480: bool Chris@488: SpectrogramLayer::paintDrawBufferPeakFrequencies(View *v, Chris@488: int w, Chris@488: int h, Chris@488: int *binforx, Chris@488: int minbin, Chris@488: int maxbin, Chris@488: float displayMinFreq, Chris@488: float displayMaxFreq, Chris@491: bool logarithmic, Chris@491: MagnitudeRange &overallMag, Chris@491: bool &overallMagChanged) const Chris@488: { Chris@488: Profiler profiler("SpectrogramLayer::paintDrawBufferPeakFrequencies"); Chris@488: Chris@488: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@488: cerr << "minbin " << minbin << ", maxbin " << maxbin << "; w " << w << ", h " << h << endl; Chris@488: #endif Chris@488: if (minbin < 0) minbin = 0; Chris@488: if (maxbin < 0) maxbin = minbin+1; Chris@488: Chris@488: FFTModel *fft = getFFTModel(v); Chris@488: if (!fft) return false; Chris@488: Chris@488: FFTModel::PeakSet peakfreqs; Chris@488: Chris@488: int px = -1, psx = -1; Chris@545: Chris@545: #ifdef __GNUC__ Chris@488: float values[maxbin - minbin + 1]; Chris@545: #else Chris@545: float *values = (float *)alloca((maxbin - minbin + 1) * sizeof(float)); Chris@545: #endif Chris@488: Chris@488: for (int x = 0; x < w; ++x) { Chris@488: Chris@488: if (binforx[x] < 0) continue; Chris@488: Chris@488: float columnMax = 0.f; Chris@488: Chris@488: int sx0 = binforx[x]; Chris@488: int sx1 = sx0; Chris@488: if (x+1 < w) sx1 = binforx[x+1]; Chris@488: if (sx0 < 0) sx0 = sx1 - 1; Chris@488: if (sx0 < 0) continue; Chris@488: if (sx1 <= sx0) sx1 = sx0 + 1; Chris@488: Chris@488: for (int sx = sx0; sx < sx1; ++sx) { Chris@488: Chris@488: if (x == px && sx == psx) continue; Chris@488: if (sx < 0 || sx >= int(fft->getWidth())) continue; Chris@488: Chris@488: if (!m_synchronous) { Chris@488: if (!fft->isColumnAvailable(sx)) { Chris@488: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@488: std::cerr << "Met unavailable column at col " << sx << std::endl; Chris@488: #endif Chris@488: return false; Chris@488: } Chris@488: } Chris@488: Chris@488: MagnitudeRange mag; Chris@488: Chris@488: if (sx != psx) { Chris@488: peakfreqs = fft->getPeakFrequencies(FFTModel::AllPeaks, sx, Chris@488: minbin, maxbin - 1); Chris@488: if (m_colourScale == PhaseColourScale) { Chris@488: fft->getPhasesAt(sx, values, minbin, maxbin - minbin + 1); Chris@488: } else if (m_normalizeColumns) { Chris@488: fft->getNormalizedMagnitudesAt(sx, values, minbin, maxbin - minbin + 1); Chris@488: } else { Chris@488: fft->getMagnitudesAt(sx, values, minbin, maxbin - minbin + 1); Chris@488: } Chris@488: psx = sx; Chris@488: } Chris@488: Chris@488: for (FFTModel::PeakSet::const_iterator pi = peakfreqs.begin(); Chris@488: pi != peakfreqs.end(); ++pi) { Chris@488: Chris@488: int bin = pi->first; Chris@488: int freq = pi->second; Chris@488: Chris@488: if (bin < minbin) continue; Chris@488: if (bin > maxbin) break; Chris@488: Chris@488: float value = values[bin - minbin]; Chris@488: Chris@488: if (m_colourScale != PhaseColourScale) { Chris@488: if (!m_normalizeColumns) { Chris@488: value /= (m_fftSize/2.f); Chris@488: } Chris@488: mag.sample(value); Chris@488: value *= m_gain; Chris@488: } Chris@488: Chris@488: float y = v->getYForFrequency Chris@488: (freq, displayMinFreq, displayMaxFreq, logarithmic); Chris@488: Chris@558: int iy = int(y + 0.5); Chris@558: if (iy < 0 || iy >= h) continue; Chris@558: Chris@558: m_drawBuffer.setPixel(x, iy, getDisplayValue(v, value)); Chris@488: } Chris@488: Chris@488: if (mag.isSet()) { Chris@488: if (sx >= int(m_columnMags.size())) { Chris@540: #ifdef DEBUG_SPECTROGRAM Chris@488: std::cerr << "INTERNAL ERROR: " << sx << " >= " Chris@488: << m_columnMags.size() Chris@488: << " at SpectrogramLayer.cpp::paintDrawBuffer" Chris@488: << std::endl; Chris@540: #endif Chris@490: } else { Chris@490: m_columnMags[sx].sample(mag); Chris@491: if (overallMag.sample(mag)) overallMagChanged = true; Chris@488: } Chris@488: } Chris@488: } Chris@488: } Chris@488: Chris@488: return true; Chris@488: } Chris@488: Chris@488: bool Chris@481: SpectrogramLayer::paintDrawBuffer(View *v, Chris@481: int w, Chris@481: int h, Chris@481: int *binforx, Chris@490: float *binfory, Chris@491: bool usePeaksCache, Chris@491: MagnitudeRange &overallMag, Chris@491: bool &overallMagChanged) const Chris@480: { Chris@481: Profiler profiler("SpectrogramLayer::paintDrawBuffer"); Chris@480: Chris@490: int minbin = int(binfory[0] + 0.0001); Chris@480: int maxbin = binfory[h-1]; Chris@480: Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@481: cerr << "minbin " << minbin << ", maxbin " << maxbin << "; w " << w << ", h " << h << endl; Chris@485: #endif Chris@480: if (minbin < 0) minbin = 0; Chris@480: if (maxbin < 0) maxbin = minbin+1; Chris@480: Chris@484: DenseThreeDimensionalModel *sourceModel = 0; Chris@484: FFTModel *fft = 0; Chris@484: int divisor = 1; Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@484: cerr << "Note: bin display = " << m_binDisplay << ", w = " << w << ", binforx[" << w-1 << "] = " << binforx[w-1] << ", binforx[0] = " << binforx[0] << endl; Chris@485: #endif Chris@484: if (usePeaksCache) { //!!! Chris@484: sourceModel = getPeakCache(v); Chris@484: divisor = 8;//!!! Chris@484: minbin = 0; Chris@484: maxbin = sourceModel->getHeight(); Chris@484: } else { Chris@484: sourceModel = fft = getFFTModel(v); Chris@484: } Chris@484: Chris@484: if (!sourceModel) return false; Chris@484: Chris@490: bool interpolate = false; Chris@490: Preferences::SpectrogramSmoothing smoothing = Chris@490: Preferences::getInstance()->getSpectrogramSmoothing(); Chris@490: if (smoothing == Preferences::SpectrogramInterpolated || Chris@490: smoothing == Preferences::SpectrogramZeroPaddedAndInterpolated) { Chris@490: if (m_binDisplay != PeakBins && Chris@490: m_binDisplay != PeakFrequencies) { Chris@490: interpolate = true; Chris@490: } Chris@490: } Chris@490: Chris@480: int psx = -1; Chris@545: Chris@545: #ifdef __GNUC__ Chris@490: float autoarray[maxbin - minbin + 1]; Chris@545: float peaks[h]; Chris@545: #else Chris@545: float *autoarray = (float *)alloca((maxbin - minbin + 1) * sizeof(float)); Chris@545: float *peaks = (float *)alloca(h * sizeof(float)); Chris@545: #endif Chris@545: Chris@490: const float *values = autoarray; Chris@484: DenseThreeDimensionalModel::Column c; Chris@480: Chris@480: for (int x = 0; x < w; ++x) { Chris@480: Chris@482: if (binforx[x] < 0) continue; Chris@482: Chris@488: // float columnGain = m_gain; Chris@487: float columnMax = 0.f; Chris@487: Chris@484: int sx0 = binforx[x] / divisor; Chris@483: int sx1 = sx0; Chris@484: if (x+1 < w) sx1 = binforx[x+1] / divisor; Chris@483: if (sx0 < 0) sx0 = sx1 - 1; Chris@483: if (sx0 < 0) continue; Chris@483: if (sx1 <= sx0) sx1 = sx0 + 1; Chris@483: Chris@483: for (int y = 0; y < h; ++y) peaks[y] = 0.f; Chris@480: Chris@483: for (int sx = sx0; sx < sx1; ++sx) { Chris@483: Chris@518: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@518: // std::cerr << "sx = " << sx << std::endl; Chris@518: #endif Chris@518: Chris@484: if (sx < 0 || sx >= int(sourceModel->getWidth())) continue; Chris@483: Chris@483: if (!m_synchronous) { Chris@484: if (!sourceModel->isColumnAvailable(sx)) { Chris@480: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@483: std::cerr << "Met unavailable column at col " << sx << std::endl; Chris@480: #endif Chris@483: return false; Chris@480: } Chris@483: } Chris@483: Chris@488: MagnitudeRange mag; Chris@488: Chris@483: if (sx != psx) { Chris@484: if (fft) { Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Retrieving column " << sx << " from fft directly" << endl; Chris@485: #endif Chris@487: if (m_colourScale == PhaseColourScale) { Chris@490: fft->getPhasesAt(sx, autoarray, minbin, maxbin - minbin + 1); Chris@487: } else if (m_normalizeColumns) { Chris@490: fft->getNormalizedMagnitudesAt(sx, autoarray, minbin, maxbin - minbin + 1); Chris@487: } else { Chris@490: fft->getMagnitudesAt(sx, autoarray, minbin, maxbin - minbin + 1); Chris@487: } Chris@484: } else { Chris@485: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "Retrieving column " << sx << " from peaks cache" << endl; Chris@485: #endif Chris@484: c = sourceModel->getColumn(sx); Chris@487: if (m_normalizeColumns) { Chris@487: for (int y = 0; y < h; ++y) { Chris@487: if (c[y] > columnMax) columnMax = c[y]; Chris@487: } Chris@487: } Chris@490: values = c.constData() + minbin; Chris@484: } Chris@483: psx = sx; Chris@483: } Chris@483: Chris@483: for (int y = 0; y < h; ++y) { Chris@480: Chris@490: float sy0 = binfory[y]; Chris@490: float sy1 = sy0 + 1; Chris@481: if (y+1 < h) sy1 = binfory[y+1]; Chris@490: Chris@490: float value = 0.f; Chris@490: Chris@490: if (interpolate && fabsf(sy1 - sy0) < 1.f) { Chris@490: Chris@490: float centre = (sy0 + sy1) / 2; Chris@490: float dist = (centre - 0.5) - lrintf(centre - 0.5); Chris@490: int bin = int(centre); Chris@490: int other = (dist < 0 ? (bin-1) : (bin+1)); Chris@490: if (bin < minbin) bin = minbin; Chris@490: if (bin > maxbin) bin = maxbin; Chris@490: if (other < minbin || other > maxbin) other = bin; Chris@490: float prop = 1.f - fabsf(dist); Chris@490: Chris@490: float v0 = values[bin - minbin]; Chris@490: float v1 = values[other - minbin]; Chris@490: if (m_binDisplay == PeakBins) { Chris@490: if (bin == minbin || bin == maxbin || Chris@490: v0 < values[bin-minbin-1] || Chris@490: v0 < values[bin-minbin+1]) v0 = 0.f; Chris@490: if (other == minbin || other == maxbin || Chris@490: v1 < values[other-minbin-1] || Chris@490: v1 < values[other-minbin+1]) v1 = 0.f; Chris@489: } Chris@490: if (v0 == 0.f && v1 == 0.f) continue; Chris@490: value = prop * v0 + (1.f - prop) * v1; Chris@484: Chris@488: if (m_colourScale != PhaseColourScale) { Chris@488: if (!m_normalizeColumns) { Chris@488: value /= (m_fftSize/2.f); Chris@488: } Chris@488: mag.sample(value); Chris@488: value *= m_gain; Chris@488: } Chris@488: Chris@490: peaks[y] = value; Chris@490: Chris@490: } else { Chris@490: Chris@490: int by0 = int(sy0 + 0.0001); Chris@490: int by1 = int(sy1 + 0.0001); Chris@490: if (by1 < by0 + 1) by1 = by0 + 1; Chris@490: Chris@490: for (int bin = by0; bin < by1; ++bin) { Chris@490: Chris@490: value = values[bin - minbin]; Chris@490: if (m_binDisplay == PeakBins) { Chris@490: if (bin == minbin || bin == maxbin || Chris@490: value < values[bin-minbin-1] || Chris@490: value < values[bin-minbin+1]) continue; Chris@480: } Chris@490: Chris@490: if (m_colourScale != PhaseColourScale) { Chris@490: if (!m_normalizeColumns) { Chris@490: value /= (m_fftSize/2.f); Chris@490: } Chris@490: mag.sample(value); Chris@490: value *= m_gain; Chris@490: } Chris@490: Chris@490: if (value > peaks[y]) peaks[y] = value; //!!! not right for phase! Chris@480: } Chris@480: } Chris@483: } Chris@488: Chris@488: if (mag.isSet()) { Chris@488: if (sx >= int(m_columnMags.size())) { Chris@540: #ifdef DEBUG_SPECTROGRAM Chris@488: std::cerr << "INTERNAL ERROR: " << sx << " >= " Chris@488: << m_columnMags.size() Chris@488: << " at SpectrogramLayer.cpp::paintDrawBuffer" Chris@488: << std::endl; Chris@540: #endif Chris@490: } else { Chris@490: m_columnMags[sx].sample(mag); Chris@491: if (overallMag.sample(mag)) overallMagChanged = true; Chris@488: } Chris@488: } Chris@483: } Chris@483: Chris@483: for (int y = 0; y < h; ++y) { Chris@483: Chris@483: float peak = peaks[y]; Chris@483: Chris@488: if (m_colourScale != PhaseColourScale && Chris@488: m_normalizeColumns && Chris@488: columnMax > 0.f) { Chris@488: peak /= columnMax; Chris@480: } Chris@483: Chris@483: unsigned char peakpix = getDisplayValue(v, peak); Chris@480: Chris@480: m_drawBuffer.setPixel(x, h-y-1, peakpix); Chris@480: } Chris@480: } Chris@480: Chris@480: return true; Chris@480: } Chris@477: Chris@121: void Chris@121: SpectrogramLayer::illuminateLocalFeatures(View *v, QPainter &paint) const Chris@121: { Chris@382: Profiler profiler("SpectrogramLayer::illuminateLocalFeatures"); Chris@382: Chris@121: QPoint localPos; Chris@121: if (!v->shouldIlluminateLocalFeatures(this, localPos) || !m_model) { Chris@121: return; Chris@121: } Chris@121: Chris@180: // std::cerr << "SpectrogramLayer: illuminateLocalFeatures(" Chris@180: // << localPos.x() << "," << localPos.y() << ")" << std::endl; Chris@121: Chris@121: float s0, s1; Chris@121: float f0, f1; Chris@121: Chris@121: if (getXBinRange(v, localPos.x(), s0, s1) && Chris@121: getYBinSourceRange(v, localPos.y(), f0, f1)) { Chris@121: Chris@121: int s0i = int(s0 + 0.001); Chris@121: int s1i = int(s1); Chris@121: Chris@121: int x0 = v->getXForFrame(s0i * getWindowIncrement()); Chris@121: int x1 = v->getXForFrame((s1i + 1) * getWindowIncrement()); Chris@121: Chris@248: int y1 = int(getYForFrequency(v, f1)); Chris@248: int y0 = int(getYForFrequency(v, f0)); Chris@121: Chris@180: // std::cerr << "SpectrogramLayer: illuminate " Chris@180: // << x0 << "," << y1 << " -> " << x1 << "," << y0 << std::endl; Chris@121: Chris@287: paint.setPen(v->getForeground()); Chris@133: Chris@133: //!!! should we be using paintCrosshairs for this? Chris@133: Chris@121: paint.drawRect(x0, y1, x1 - x0 + 1, y0 - y1 + 1); Chris@121: } Chris@121: } Chris@121: Chris@42: float Chris@267: SpectrogramLayer::getYForFrequency(const View *v, float frequency) const Chris@42: { Chris@44: return v->getYForFrequency(frequency, Chris@44: getEffectiveMinFrequency(), Chris@44: getEffectiveMaxFrequency(), Chris@44: m_frequencyScale == LogFrequencyScale); Chris@42: } Chris@42: Chris@42: float Chris@267: SpectrogramLayer::getFrequencyForY(const View *v, int y) const Chris@42: { Chris@44: return v->getFrequencyForY(y, Chris@44: getEffectiveMinFrequency(), Chris@44: getEffectiveMaxFrequency(), Chris@44: m_frequencyScale == LogFrequencyScale); Chris@42: } Chris@42: Chris@0: int Chris@115: SpectrogramLayer::getCompletion(View *v) const Chris@0: { Chris@115: if (m_updateTimer == 0) return 100; Chris@130: if (m_fftModels.find(v) == m_fftModels.end()) return 100; Chris@130: Chris@130: size_t completion = m_fftModels[v].first->getCompletion(); Chris@224: #ifdef DEBUG_SPECTROGRAM_REPAINT Chris@587: SVDEBUG << "SpectrogramLayer::getCompletion: completion = " << completion << endl; Chris@224: #endif Chris@0: return completion; Chris@0: } Chris@0: Chris@583: QString Chris@583: SpectrogramLayer::getError(View *v) const Chris@583: { Chris@583: if (m_fftModels.find(v) == m_fftModels.end()) return ""; Chris@583: return m_fftModels[v].first->getError(); Chris@583: } Chris@583: Chris@28: bool Chris@101: SpectrogramLayer::getValueExtents(float &min, float &max, Chris@101: bool &logarithmic, QString &unit) const Chris@79: { Chris@133: if (!m_model) return false; Chris@133: Chris@133: int sr = m_model->getSampleRate(); Chris@133: min = float(sr) / m_fftSize; Chris@133: max = float(sr) / 2; Chris@133: Chris@101: logarithmic = (m_frequencyScale == LogFrequencyScale); Chris@79: unit = "Hz"; Chris@79: return true; Chris@79: } Chris@79: Chris@79: bool Chris@101: SpectrogramLayer::getDisplayExtents(float &min, float &max) const Chris@101: { Chris@101: min = getEffectiveMinFrequency(); Chris@101: max = getEffectiveMaxFrequency(); Chris@253: Chris@587: // SVDEBUG << "SpectrogramLayer::getDisplayExtents: " << min << "->" << max << endl; Chris@101: return true; Chris@101: } Chris@101: Chris@101: bool Chris@120: SpectrogramLayer::setDisplayExtents(float min, float max) Chris@120: { Chris@120: if (!m_model) return false; Chris@187: Chris@587: // SVDEBUG << "SpectrogramLayer::setDisplayExtents: " << min << "->" << max << endl; Chris@187: Chris@120: if (min < 0) min = 0; Chris@120: if (max > m_model->getSampleRate()/2) max = m_model->getSampleRate()/2; Chris@120: Chris@120: size_t minf = lrintf(min); Chris@120: size_t maxf = lrintf(max); Chris@120: Chris@120: if (m_minFrequency == minf && m_maxFrequency == maxf) return true; Chris@120: Chris@478: invalidateImageCaches(); Chris@120: invalidateMagnitudes(); Chris@120: Chris@120: m_minFrequency = minf; Chris@120: m_maxFrequency = maxf; Chris@120: Chris@120: emit layerParametersChanged(); Chris@120: Chris@133: int vs = getCurrentVerticalZoomStep(); Chris@133: if (vs != m_lastEmittedZoomStep) { Chris@133: emit verticalZoomChanged(); Chris@133: m_lastEmittedZoomStep = vs; Chris@133: } Chris@133: Chris@120: return true; Chris@120: } Chris@120: Chris@120: bool Chris@267: SpectrogramLayer::getYScaleValue(const View *v, int y, Chris@261: float &value, QString &unit) const Chris@261: { Chris@261: value = getFrequencyForY(v, y); Chris@261: unit = "Hz"; Chris@261: return true; Chris@261: } Chris@261: Chris@261: bool Chris@248: SpectrogramLayer::snapToFeatureFrame(View *, int &frame, Chris@28: size_t &resolution, Chris@28: SnapType snap) const Chris@13: { Chris@13: resolution = getWindowIncrement(); Chris@28: int left = (frame / resolution) * resolution; Chris@28: int right = left + resolution; Chris@28: Chris@28: switch (snap) { Chris@28: case SnapLeft: frame = left; break; Chris@28: case SnapRight: frame = right; break; Chris@28: case SnapNearest: Chris@28: case SnapNeighbouring: Chris@28: if (frame - left > right - frame) frame = right; Chris@28: else frame = left; Chris@28: break; Chris@28: } Chris@28: Chris@28: return true; Chris@28: } Chris@13: Chris@283: void Chris@283: SpectrogramLayer::measureDoubleClick(View *v, QMouseEvent *e) Chris@283: { Chris@478: ImageCache &cache = m_imageCaches[v]; Chris@478: Chris@478: std::cerr << "cache width: " << cache.image.width() << ", height: " Chris@478: << cache.image.height() << std::endl; Chris@478: Chris@478: QImage image = cache.image; Chris@283: Chris@283: ImageRegionFinder finder; Chris@283: QRect rect = finder.findRegionExtents(&image, e->pos()); Chris@283: if (rect.isValid()) { Chris@283: MeasureRect mr; Chris@283: setMeasureRectFromPixrect(v, mr, rect); Chris@283: CommandHistory::getInstance()->addCommand Chris@283: (new AddMeasurementRectCommand(this, mr)); Chris@283: } Chris@283: } Chris@283: Chris@77: bool Chris@264: SpectrogramLayer::getCrosshairExtents(View *v, QPainter &paint, Chris@77: QPoint cursorPos, Chris@77: std::vector &extents) const Chris@77: { Chris@77: QRect vertical(cursorPos.x() - 12, 0, 12, v->height()); Chris@77: extents.push_back(vertical); Chris@77: Chris@77: QRect horizontal(0, cursorPos.y(), cursorPos.x(), 1); Chris@77: extents.push_back(horizontal); Chris@77: Chris@264: int sw = getVerticalScaleWidth(v, paint); Chris@264: Chris@280: QRect freq(sw, cursorPos.y() - paint.fontMetrics().ascent() - 2, Chris@280: paint.fontMetrics().width("123456 Hz") + 2, Chris@280: paint.fontMetrics().height()); Chris@280: extents.push_back(freq); Chris@264: Chris@279: QRect pitch(sw, cursorPos.y() + 2, Chris@279: paint.fontMetrics().width("C#10+50c") + 2, Chris@279: paint.fontMetrics().height()); Chris@279: extents.push_back(pitch); Chris@279: Chris@280: QRect rt(cursorPos.x(), Chris@280: v->height() - paint.fontMetrics().height() - 2, Chris@280: paint.fontMetrics().width("1234.567 s"), Chris@280: paint.fontMetrics().height()); Chris@280: extents.push_back(rt); Chris@280: Chris@280: int w(paint.fontMetrics().width("1234567890") + 2); Chris@280: QRect frame(cursorPos.x() - w - 2, Chris@280: v->height() - paint.fontMetrics().height() - 2, Chris@280: w, Chris@280: paint.fontMetrics().height()); Chris@280: extents.push_back(frame); Chris@280: Chris@77: return true; Chris@77: } Chris@77: Chris@77: void Chris@77: SpectrogramLayer::paintCrosshairs(View *v, QPainter &paint, Chris@77: QPoint cursorPos) const Chris@77: { Chris@77: paint.save(); Chris@283: Chris@283: int sw = getVerticalScaleWidth(v, paint); Chris@283: Chris@282: QFont fn = paint.font(); Chris@282: if (fn.pointSize() > 8) { Chris@282: fn.setPointSize(fn.pointSize() - 1); Chris@282: paint.setFont(fn); Chris@282: } Chris@77: paint.setPen(m_crosshairColour); Chris@77: Chris@77: paint.drawLine(0, cursorPos.y(), cursorPos.x() - 1, cursorPos.y()); Chris@77: paint.drawLine(cursorPos.x(), 0, cursorPos.x(), v->height()); Chris@77: Chris@77: float fundamental = getFrequencyForY(v, cursorPos.y()); Chris@77: Chris@278: v->drawVisibleText(paint, Chris@278: sw + 2, Chris@278: cursorPos.y() - 2, Chris@278: QString("%1 Hz").arg(fundamental), Chris@278: View::OutlinedText); Chris@278: Chris@279: if (Pitch::isFrequencyInMidiRange(fundamental)) { Chris@279: QString pitchLabel = Pitch::getPitchLabelForFrequency(fundamental); Chris@279: v->drawVisibleText(paint, Chris@279: sw + 2, Chris@279: cursorPos.y() + paint.fontMetrics().ascent() + 2, Chris@279: pitchLabel, Chris@279: View::OutlinedText); Chris@279: } Chris@279: Chris@280: long frame = v->getFrameForX(cursorPos.x()); Chris@279: RealTime rt = RealTime::frame2RealTime(frame, m_model->getSampleRate()); Chris@280: QString rtLabel = QString("%1 s").arg(rt.toText(true).c_str()); Chris@280: QString frameLabel = QString("%1").arg(frame); Chris@280: v->drawVisibleText(paint, Chris@280: cursorPos.x() - paint.fontMetrics().width(frameLabel) - 2, Chris@280: v->height() - 2, Chris@280: frameLabel, Chris@280: View::OutlinedText); Chris@280: v->drawVisibleText(paint, Chris@280: cursorPos.x() + 2, Chris@280: v->height() - 2, Chris@280: rtLabel, Chris@280: View::OutlinedText); Chris@264: Chris@77: int harmonic = 2; Chris@77: Chris@77: while (harmonic < 100) { Chris@77: Chris@77: float hy = lrintf(getYForFrequency(v, fundamental * harmonic)); Chris@77: if (hy < 0 || hy > v->height()) break; Chris@77: Chris@77: int len = 7; Chris@77: Chris@77: if (harmonic % 2 == 0) { Chris@77: if (harmonic % 4 == 0) { Chris@77: len = 12; Chris@77: } else { Chris@77: len = 10; Chris@77: } Chris@77: } Chris@77: Chris@77: paint.drawLine(cursorPos.x() - len, Chris@248: int(hy), Chris@77: cursorPos.x(), Chris@248: int(hy)); Chris@77: Chris@77: ++harmonic; Chris@77: } Chris@77: Chris@77: paint.restore(); Chris@77: } Chris@77: Chris@25: QString Chris@44: SpectrogramLayer::getFeatureDescription(View *v, QPoint &pos) const Chris@25: { Chris@25: int x = pos.x(); Chris@25: int y = pos.y(); Chris@0: Chris@25: if (!m_model || !m_model->isOK()) return ""; Chris@0: Chris@38: float magMin = 0, magMax = 0; Chris@38: float phaseMin = 0, phaseMax = 0; Chris@0: float freqMin = 0, freqMax = 0; Chris@35: float adjFreqMin = 0, adjFreqMax = 0; Chris@25: QString pitchMin, pitchMax; Chris@0: RealTime rtMin, rtMax; Chris@0: Chris@38: bool haveValues = false; Chris@0: Chris@44: if (!getXBinSourceRange(v, x, rtMin, rtMax)) { Chris@38: return ""; Chris@38: } Chris@44: if (getXYBinSourceRange(v, x, y, magMin, magMax, phaseMin, phaseMax)) { Chris@38: haveValues = true; Chris@38: } Chris@0: Chris@35: QString adjFreqText = "", adjPitchText = ""; Chris@35: Chris@38: if (m_binDisplay == PeakFrequencies) { Chris@35: Chris@44: if (!getAdjustedYBinSourceRange(v, x, y, freqMin, freqMax, Chris@38: adjFreqMin, adjFreqMax)) { Chris@38: return ""; Chris@38: } Chris@35: Chris@35: if (adjFreqMin != adjFreqMax) { Chris@65: adjFreqText = tr("Peak Frequency:\t%1 - %2 Hz\n") Chris@35: .arg(adjFreqMin).arg(adjFreqMax); Chris@35: } else { Chris@65: adjFreqText = tr("Peak Frequency:\t%1 Hz\n") Chris@35: .arg(adjFreqMin); Chris@38: } Chris@38: Chris@38: QString pmin = Pitch::getPitchLabelForFrequency(adjFreqMin); Chris@38: QString pmax = Pitch::getPitchLabelForFrequency(adjFreqMax); Chris@38: Chris@38: if (pmin != pmax) { Chris@65: adjPitchText = tr("Peak Pitch:\t%3 - %4\n").arg(pmin).arg(pmax); Chris@38: } else { Chris@65: adjPitchText = tr("Peak Pitch:\t%2\n").arg(pmin); Chris@35: } Chris@35: Chris@35: } else { Chris@35: Chris@44: if (!getYBinSourceRange(v, y, freqMin, freqMax)) return ""; Chris@35: } Chris@35: Chris@25: QString text; Chris@25: Chris@25: if (rtMin != rtMax) { Chris@25: text += tr("Time:\t%1 - %2\n") Chris@25: .arg(rtMin.toText(true).c_str()) Chris@25: .arg(rtMax.toText(true).c_str()); Chris@25: } else { Chris@25: text += tr("Time:\t%1\n") Chris@25: .arg(rtMin.toText(true).c_str()); Chris@0: } Chris@0: Chris@25: if (freqMin != freqMax) { Chris@65: text += tr("%1Bin Frequency:\t%2 - %3 Hz\n%4Bin Pitch:\t%5 - %6\n") Chris@65: .arg(adjFreqText) Chris@25: .arg(freqMin) Chris@25: .arg(freqMax) Chris@65: .arg(adjPitchText) Chris@65: .arg(Pitch::getPitchLabelForFrequency(freqMin)) Chris@65: .arg(Pitch::getPitchLabelForFrequency(freqMax)); Chris@65: } else { Chris@65: text += tr("%1Bin Frequency:\t%2 Hz\n%3Bin Pitch:\t%4\n") Chris@35: .arg(adjFreqText) Chris@25: .arg(freqMin) Chris@65: .arg(adjPitchText) Chris@65: .arg(Pitch::getPitchLabelForFrequency(freqMin)); Chris@25: } Chris@25: Chris@38: if (haveValues) { Chris@38: float dbMin = AudioLevel::multiplier_to_dB(magMin); Chris@38: float dbMax = AudioLevel::multiplier_to_dB(magMax); Chris@43: QString dbMinString; Chris@43: QString dbMaxString; Chris@43: if (dbMin == AudioLevel::DB_FLOOR) { Chris@43: dbMinString = tr("-Inf"); Chris@43: } else { Chris@43: dbMinString = QString("%1").arg(lrintf(dbMin)); Chris@43: } Chris@43: if (dbMax == AudioLevel::DB_FLOOR) { Chris@43: dbMaxString = tr("-Inf"); Chris@43: } else { Chris@43: dbMaxString = QString("%1").arg(lrintf(dbMax)); Chris@43: } Chris@25: if (lrintf(dbMin) != lrintf(dbMax)) { Chris@199: text += tr("dB:\t%1 - %2").arg(dbMinString).arg(dbMaxString); Chris@25: } else { Chris@199: text += tr("dB:\t%1").arg(dbMinString); Chris@25: } Chris@38: if (phaseMin != phaseMax) { Chris@38: text += tr("\nPhase:\t%1 - %2").arg(phaseMin).arg(phaseMax); Chris@38: } else { Chris@38: text += tr("\nPhase:\t%1").arg(phaseMin); Chris@38: } Chris@25: } Chris@25: Chris@25: return text; Chris@0: } Chris@25: Chris@0: int Chris@40: SpectrogramLayer::getColourScaleWidth(QPainter &paint) const Chris@40: { Chris@40: int cw; Chris@40: Chris@119: cw = paint.fontMetrics().width("-80dB"); Chris@119: Chris@40: return cw; Chris@40: } Chris@40: Chris@40: int Chris@248: SpectrogramLayer::getVerticalScaleWidth(View *, QPainter &paint) const Chris@0: { Chris@0: if (!m_model || !m_model->isOK()) return 0; Chris@0: Chris@40: int cw = getColourScaleWidth(paint); Chris@40: Chris@0: int tw = paint.fontMetrics().width(QString("%1") Chris@0: .arg(m_maxFrequency > 0 ? Chris@0: m_maxFrequency - 1 : Chris@0: m_model->getSampleRate() / 2)); Chris@0: Chris@234: int fw = paint.fontMetrics().width(tr("43Hz")); Chris@0: if (tw < fw) tw = fw; Chris@40: Chris@40: int tickw = (m_frequencyScale == LogFrequencyScale ? 10 : 4); Chris@0: Chris@40: return cw + tickw + tw + 13; Chris@0: } Chris@0: Chris@0: void Chris@44: SpectrogramLayer::paintVerticalScale(View *v, QPainter &paint, QRect rect) const Chris@0: { Chris@0: if (!m_model || !m_model->isOK()) { Chris@0: return; Chris@0: } Chris@0: Chris@382: Profiler profiler("SpectrogramLayer::paintVerticalScale"); Chris@122: Chris@120: //!!! cache this? Chris@120: Chris@0: int h = rect.height(), w = rect.width(); Chris@0: Chris@40: int tickw = (m_frequencyScale == LogFrequencyScale ? 10 : 4); Chris@40: int pkw = (m_frequencyScale == LogFrequencyScale ? 10 : 0); Chris@40: Chris@107: size_t bins = m_fftSize / 2; Chris@0: int sr = m_model->getSampleRate(); Chris@0: Chris@0: if (m_maxFrequency > 0) { Chris@107: bins = int((double(m_maxFrequency) * m_fftSize) / sr + 0.1); Chris@107: if (bins > m_fftSize / 2) bins = m_fftSize / 2; Chris@0: } Chris@0: Chris@40: int cw = getColourScaleWidth(paint); Chris@119: int cbw = paint.fontMetrics().width("dB"); Chris@40: Chris@0: int py = -1; Chris@0: int textHeight = paint.fontMetrics().height(); Chris@0: int toff = -textHeight + paint.fontMetrics().ascent() + 2; Chris@0: Chris@119: if (h > textHeight * 3 + 10) { Chris@119: Chris@119: int topLines = 2; Chris@119: if (m_colourScale == PhaseColourScale) topLines = 1; Chris@119: Chris@119: int ch = h - textHeight * (topLines + 1) - 8; Chris@119: // paint.drawRect(4, textHeight + 4, cw - 1, ch + 1); Chris@119: paint.drawRect(4 + cw - cbw, textHeight * topLines + 4, cbw - 1, ch + 1); Chris@40: Chris@40: QString top, bottom; Chris@119: float min = m_viewMags[v].getMin(); Chris@119: float max = m_viewMags[v].getMax(); Chris@119: Chris@119: float dBmin = AudioLevel::multiplier_to_dB(min); Chris@119: float dBmax = AudioLevel::multiplier_to_dB(max); Chris@119: Chris@120: if (dBmax < -60.f) dBmax = -60.f; Chris@120: else top = QString("%1").arg(lrintf(dBmax)); Chris@120: Chris@120: if (dBmin < dBmax - 60.f) dBmin = dBmax - 60.f; Chris@119: bottom = QString("%1").arg(lrintf(dBmin)); Chris@119: Chris@119: //!!! & phase etc Chris@119: Chris@119: if (m_colourScale != PhaseColourScale) { Chris@119: paint.drawText((cw + 6 - paint.fontMetrics().width("dBFS")) / 2, Chris@119: 2 + textHeight + toff, "dBFS"); Chris@119: } Chris@119: Chris@119: // paint.drawText((cw + 6 - paint.fontMetrics().width(top)) / 2, Chris@119: paint.drawText(3 + cw - cbw - paint.fontMetrics().width(top), Chris@119: 2 + textHeight * topLines + toff + textHeight/2, top); Chris@119: Chris@119: paint.drawText(3 + cw - cbw - paint.fontMetrics().width(bottom), Chris@119: h + toff - 3 - textHeight/2, bottom); Chris@40: Chris@40: paint.save(); Chris@40: paint.setBrush(Qt::NoBrush); Chris@119: Chris@119: int lasty = 0; Chris@119: int lastdb = 0; Chris@119: Chris@40: for (int i = 0; i < ch; ++i) { Chris@119: Chris@119: float dBval = dBmin + (((dBmax - dBmin) * i) / (ch - 1)); Chris@119: int idb = int(dBval); Chris@119: Chris@119: float value = AudioLevel::dB_to_multiplier(dBval); Chris@119: int colour = getDisplayValue(v, value * m_gain); Chris@210: Chris@197: paint.setPen(m_palette.getColour(colour)); Chris@119: Chris@119: int y = textHeight * topLines + 4 + ch - i; Chris@119: Chris@119: paint.drawLine(5 + cw - cbw, y, cw + 2, y); Chris@119: Chris@119: if (i == 0) { Chris@119: lasty = y; Chris@119: lastdb = idb; Chris@119: } else if (i < ch - paint.fontMetrics().ascent() && Chris@120: idb != lastdb && Chris@119: ((abs(y - lasty) > textHeight && Chris@119: idb % 10 == 0) || Chris@119: (abs(y - lasty) > paint.fontMetrics().ascent() && Chris@119: idb % 5 == 0))) { Chris@287: paint.setPen(v->getBackground()); Chris@119: QString text = QString("%1").arg(idb); Chris@119: paint.drawText(3 + cw - cbw - paint.fontMetrics().width(text), Chris@119: y + toff + textHeight/2, text); Chris@287: paint.setPen(v->getForeground()); Chris@119: paint.drawLine(5 + cw - cbw, y, 8 + cw - cbw, y); Chris@119: lasty = y; Chris@119: lastdb = idb; Chris@119: } Chris@40: } Chris@40: paint.restore(); Chris@40: } Chris@40: Chris@40: paint.drawLine(cw + 7, 0, cw + 7, h); Chris@40: Chris@0: int bin = -1; Chris@0: Chris@44: for (int y = 0; y < v->height(); ++y) { Chris@0: Chris@0: float q0, q1; Chris@44: if (!getYBinRange(v, v->height() - y, q0, q1)) continue; Chris@0: Chris@0: int vy; Chris@0: Chris@0: if (int(q0) > bin) { Chris@0: vy = y; Chris@0: bin = int(q0); Chris@0: } else { Chris@0: continue; Chris@0: } Chris@0: Chris@107: int freq = (sr * bin) / m_fftSize; Chris@0: Chris@0: if (py >= 0 && (vy - py) < textHeight - 1) { Chris@40: if (m_frequencyScale == LinearFrequencyScale) { Chris@40: paint.drawLine(w - tickw, h - vy, w, h - vy); Chris@40: } Chris@0: continue; Chris@0: } Chris@0: Chris@0: QString text = QString("%1").arg(freq); Chris@234: if (bin == 1) text = tr("%1Hz").arg(freq); // bin 0 is DC Chris@40: paint.drawLine(cw + 7, h - vy, w - pkw - 1, h - vy); Chris@0: Chris@0: if (h - vy - textHeight >= -2) { Chris@40: int tx = w - 3 - paint.fontMetrics().width(text) - std::max(tickw, pkw); Chris@0: paint.drawText(tx, h - vy + toff, text); Chris@0: } Chris@0: Chris@0: py = vy; Chris@0: } Chris@40: Chris@40: if (m_frequencyScale == LogFrequencyScale) { Chris@40: Chris@277: // piano keyboard Chris@277: Chris@40: paint.drawLine(w - pkw - 1, 0, w - pkw - 1, h); Chris@40: Chris@40: float minf = getEffectiveMinFrequency(); Chris@40: float maxf = getEffectiveMaxFrequency(); Chris@40: Chris@122: int py = h, ppy = h; Chris@40: paint.setBrush(paint.pen().color()); Chris@40: Chris@40: for (int i = 0; i < 128; ++i) { Chris@40: Chris@40: float f = Pitch::getFrequencyForPitch(i); Chris@44: int y = lrintf(v->getYForFrequency(f, minf, maxf, true)); Chris@122: Chris@122: if (y < -2) break; Chris@122: if (y > h + 2) { Chris@122: continue; Chris@122: } Chris@122: Chris@40: int n = (i % 12); Chris@122: Chris@122: if (n == 1) { Chris@122: // C# -- fill the C from here Chris@284: QColor col = Qt::gray; Chris@284: if (i == 61) { // filling middle C Chris@284: col = Qt::blue; Chris@284: col = col.light(150); Chris@284: } Chris@122: if (ppy - y > 2) { Chris@122: paint.fillRect(w - pkw, Chris@122: y, Chris@122: pkw, Chris@122: (py + ppy) / 2 - y, Chris@284: col); Chris@122: } Chris@122: } Chris@122: Chris@40: if (n == 1 || n == 3 || n == 6 || n == 8 || n == 10) { Chris@40: // black notes Chris@40: paint.drawLine(w - pkw, y, w, y); Chris@41: int rh = ((py - y) / 4) * 2; Chris@41: if (rh < 2) rh = 2; Chris@41: paint.drawRect(w - pkw, y - (py-y)/4, pkw/2, rh); Chris@40: } else if (n == 0 || n == 5) { Chris@122: // C, F Chris@40: if (py < h) { Chris@40: paint.drawLine(w - pkw, (y + py) / 2, w, (y + py) / 2); Chris@40: } Chris@40: } Chris@40: Chris@122: ppy = py; Chris@40: py = y; Chris@40: } Chris@40: } Chris@0: } Chris@0: Chris@187: class SpectrogramRangeMapper : public RangeMapper Chris@187: { Chris@187: public: Chris@248: SpectrogramRangeMapper(int sr, int /* fftsize */) : Chris@187: m_dist(float(sr) / 2), Chris@187: m_s2(sqrtf(sqrtf(2))) { } Chris@187: ~SpectrogramRangeMapper() { } Chris@187: Chris@187: virtual int getPositionForValue(float value) const { Chris@187: Chris@187: float dist = m_dist; Chris@187: Chris@187: int n = 0; Chris@187: Chris@187: while (dist > (value + 0.00001) && dist > 0.1f) { Chris@187: dist /= m_s2; Chris@187: ++n; Chris@187: } Chris@187: Chris@187: return n; Chris@187: } Chris@187: Chris@187: virtual float getValueForPosition(int position) const { Chris@187: Chris@187: // Vertical zoom step 0 shows the entire range from DC -> Chris@187: // Nyquist frequency. Step 1 shows 2^(1/4) of the range of Chris@187: // step 0, and so on until the visible range is smaller than Chris@187: // the frequency step between bins at the current fft size. Chris@187: Chris@187: float dist = m_dist; Chris@187: Chris@187: int n = 0; Chris@187: while (n < position) { Chris@187: dist /= m_s2; Chris@187: ++n; Chris@187: } Chris@187: Chris@187: return dist; Chris@187: } Chris@187: Chris@187: virtual QString getUnit() const { return "Hz"; } Chris@187: Chris@187: protected: Chris@187: float m_dist; Chris@187: float m_s2; Chris@187: }; Chris@187: Chris@133: int Chris@133: SpectrogramLayer::getVerticalZoomSteps(int &defaultStep) const Chris@133: { Chris@135: if (!m_model) return 0; Chris@187: Chris@187: int sr = m_model->getSampleRate(); Chris@187: Chris@187: SpectrogramRangeMapper mapper(sr, m_fftSize); Chris@187: Chris@187: // int maxStep = mapper.getPositionForValue((float(sr) / m_fftSize) + 0.001); Chris@187: int maxStep = mapper.getPositionForValue(0); Chris@187: int minStep = mapper.getPositionForValue(float(sr) / 2); Chris@250: Chris@250: size_t initialMax = m_initialMaxFrequency; Chris@250: if (initialMax == 0) initialMax = sr / 2; Chris@250: Chris@250: defaultStep = mapper.getPositionForValue(initialMax) - minStep; Chris@250: Chris@587: // SVDEBUG << "SpectrogramLayer::getVerticalZoomSteps: " << maxStep - minStep << " (" << maxStep <<"-" << minStep << "), default is " << defaultStep << " (from initial max freq " << initialMax << ")" << endl; Chris@187: Chris@187: return maxStep - minStep; Chris@133: } Chris@133: Chris@133: int Chris@133: SpectrogramLayer::getCurrentVerticalZoomStep() const Chris@133: { Chris@133: if (!m_model) return 0; Chris@133: Chris@133: float dmin, dmax; Chris@133: getDisplayExtents(dmin, dmax); Chris@133: Chris@187: SpectrogramRangeMapper mapper(m_model->getSampleRate(), m_fftSize); Chris@187: int n = mapper.getPositionForValue(dmax - dmin); Chris@587: // SVDEBUG << "SpectrogramLayer::getCurrentVerticalZoomStep: " << n << endl; Chris@133: return n; Chris@133: } Chris@133: Chris@133: void Chris@133: SpectrogramLayer::setVerticalZoomStep(int step) Chris@133: { Chris@187: if (!m_model) return; Chris@187: Chris@253: float dmin = m_minFrequency, dmax = m_maxFrequency; Chris@253: // getDisplayExtents(dmin, dmax); Chris@253: Chris@253: // std::cerr << "current range " << dmin << " -> " << dmax << ", range " << dmax-dmin << ", mid " << (dmax + dmin)/2 << std::endl; Chris@133: Chris@133: int sr = m_model->getSampleRate(); Chris@187: SpectrogramRangeMapper mapper(sr, m_fftSize); Chris@253: float newdist = mapper.getValueForPosition(step); Chris@253: Chris@253: float newmin, newmax; Chris@253: Chris@253: if (m_frequencyScale == LogFrequencyScale) { Chris@253: Chris@253: // need to pick newmin and newmax such that Chris@253: // Chris@253: // (log(newmin) + log(newmax)) / 2 == logmid Chris@253: // and Chris@253: // newmax - newmin = newdist Chris@253: // Chris@253: // so log(newmax - newdist) + log(newmax) == 2logmid Chris@253: // log(newmax(newmax - newdist)) == 2logmid Chris@253: // newmax.newmax - newmax.newdist == exp(2logmid) Chris@253: // newmax^2 + (-newdist)newmax + -exp(2logmid) == 0 Chris@253: // quadratic with a = 1, b = -newdist, c = -exp(2logmid), all known Chris@253: // Chris@253: // positive root Chris@253: // newmax = (newdist + sqrt(newdist^2 + 4exp(2logmid))) / 2 Chris@253: // Chris@253: // but logmid = (log(dmin) + log(dmax)) / 2 Chris@253: // so exp(2logmid) = exp(log(dmin) + log(dmax)) Chris@253: // = exp(log(dmin.dmax)) Chris@253: // = dmin.dmax Chris@253: // so newmax = (newdist + sqrtf(newdist^2 + 4dmin.dmax)) / 2 Chris@253: Chris@253: newmax = (newdist + sqrtf(newdist*newdist + 4*dmin*dmax)) / 2; Chris@253: newmin = newmax - newdist; Chris@253: Chris@253: // std::cerr << "newmin = " << newmin << ", newmax = " << newmax << std::endl; Chris@253: Chris@253: } else { Chris@253: float dmid = (dmax + dmin) / 2; Chris@253: newmin = dmid - newdist / 2; Chris@253: newmax = dmid + newdist / 2; Chris@253: } Chris@187: Chris@187: float mmin, mmax; Chris@187: mmin = 0; Chris@187: mmax = float(sr) / 2; Chris@133: Chris@187: if (newmin < mmin) { Chris@187: newmax += (mmin - newmin); Chris@187: newmin = mmin; Chris@187: } Chris@187: if (newmax > mmax) { Chris@187: newmax = mmax; Chris@187: } Chris@133: Chris@587: // SVDEBUG << "SpectrogramLayer::setVerticalZoomStep: " << step << ": " << newmin << " -> " << newmax << " (range " << newdist << ")" << endl; Chris@253: Chris@253: setMinFrequency(lrintf(newmin)); Chris@253: setMaxFrequency(lrintf(newmax)); Chris@187: } Chris@187: Chris@187: RangeMapper * Chris@187: SpectrogramLayer::getNewVerticalZoomRangeMapper() const Chris@187: { Chris@187: if (!m_model) return 0; Chris@187: return new SpectrogramRangeMapper(m_model->getSampleRate(), m_fftSize); Chris@133: } Chris@133: Chris@273: void Chris@273: SpectrogramLayer::updateMeasureRectYCoords(View *v, const MeasureRect &r) const Chris@273: { Chris@273: int y0 = 0; Chris@273: if (r.startY > 0.0) y0 = getYForFrequency(v, r.startY); Chris@273: Chris@273: int y1 = y0; Chris@273: if (r.endY > 0.0) y1 = getYForFrequency(v, r.endY); Chris@273: Chris@587: // SVDEBUG << "SpectrogramLayer::updateMeasureRectYCoords: start " << r.startY << " -> " << y0 << ", end " << r.endY << " -> " << y1 << endl; Chris@273: Chris@273: r.pixrect = QRect(r.pixrect.x(), y0, r.pixrect.width(), y1 - y0); Chris@273: } Chris@273: Chris@273: void Chris@273: SpectrogramLayer::setMeasureRectYCoord(View *v, MeasureRect &r, bool start, int y) const Chris@273: { Chris@273: if (start) { Chris@273: r.startY = getFrequencyForY(v, y); Chris@273: r.endY = r.startY; Chris@273: } else { Chris@273: r.endY = getFrequencyForY(v, y); Chris@273: } Chris@587: // SVDEBUG << "SpectrogramLayer::setMeasureRectYCoord: start " << r.startY << " <- " << y << ", end " << r.endY << " <- " << y << endl; Chris@273: Chris@273: } Chris@273: Chris@316: void Chris@316: SpectrogramLayer::toXml(QTextStream &stream, Chris@316: QString indent, QString extraAttributes) const Chris@6: { Chris@6: QString s; Chris@6: Chris@6: s += QString("channel=\"%1\" " Chris@6: "windowSize=\"%2\" " Chris@153: "windowHopLevel=\"%3\" " Chris@153: "gain=\"%4\" " Chris@153: "threshold=\"%5\" ") Chris@6: .arg(m_channel) Chris@6: .arg(m_windowSize) Chris@97: .arg(m_windowHopLevel) Chris@37: .arg(m_gain) Chris@37: .arg(m_threshold); Chris@37: Chris@37: s += QString("minFrequency=\"%1\" " Chris@37: "maxFrequency=\"%2\" " Chris@37: "colourScale=\"%3\" " Chris@37: "colourScheme=\"%4\" " Chris@37: "colourRotation=\"%5\" " Chris@37: "frequencyScale=\"%6\" " Chris@37: "binDisplay=\"%7\" " Chris@153: "normalizeColumns=\"%8\" " Chris@153: "normalizeVisibleArea=\"%9\"") Chris@37: .arg(m_minFrequency) Chris@6: .arg(m_maxFrequency) Chris@6: .arg(m_colourScale) Chris@197: .arg(m_colourMap) Chris@37: .arg(m_colourRotation) Chris@35: .arg(m_frequencyScale) Chris@37: .arg(m_binDisplay) Chris@153: .arg(m_normalizeColumns ? "true" : "false") Chris@153: .arg(m_normalizeVisibleArea ? "true" : "false"); Chris@6: Chris@316: Layer::toXml(stream, indent, extraAttributes + " " + s); Chris@6: } Chris@6: Chris@11: void Chris@11: SpectrogramLayer::setProperties(const QXmlAttributes &attributes) Chris@11: { Chris@11: bool ok = false; Chris@11: Chris@11: int channel = attributes.value("channel").toInt(&ok); Chris@11: if (ok) setChannel(channel); Chris@11: Chris@11: size_t windowSize = attributes.value("windowSize").toUInt(&ok); Chris@11: if (ok) setWindowSize(windowSize); Chris@11: Chris@97: size_t windowHopLevel = attributes.value("windowHopLevel").toUInt(&ok); Chris@97: if (ok) setWindowHopLevel(windowHopLevel); Chris@97: else { Chris@97: size_t windowOverlap = attributes.value("windowOverlap").toUInt(&ok); Chris@97: // a percentage value Chris@97: if (ok) { Chris@97: if (windowOverlap == 0) setWindowHopLevel(0); Chris@97: else if (windowOverlap == 25) setWindowHopLevel(1); Chris@97: else if (windowOverlap == 50) setWindowHopLevel(2); Chris@97: else if (windowOverlap == 75) setWindowHopLevel(3); Chris@97: else if (windowOverlap == 90) setWindowHopLevel(4); Chris@97: } Chris@97: } Chris@11: Chris@11: float gain = attributes.value("gain").toFloat(&ok); Chris@11: if (ok) setGain(gain); Chris@11: Chris@37: float threshold = attributes.value("threshold").toFloat(&ok); Chris@37: if (ok) setThreshold(threshold); Chris@37: Chris@37: size_t minFrequency = attributes.value("minFrequency").toUInt(&ok); Chris@187: if (ok) { Chris@587: SVDEBUG << "SpectrogramLayer::setProperties: setting min freq to " << minFrequency << endl; Chris@187: setMinFrequency(minFrequency); Chris@187: } Chris@37: Chris@11: size_t maxFrequency = attributes.value("maxFrequency").toUInt(&ok); Chris@187: if (ok) { Chris@587: SVDEBUG << "SpectrogramLayer::setProperties: setting max freq to " << maxFrequency << endl; Chris@187: setMaxFrequency(maxFrequency); Chris@187: } Chris@11: Chris@11: ColourScale colourScale = (ColourScale) Chris@11: attributes.value("colourScale").toInt(&ok); Chris@11: if (ok) setColourScale(colourScale); Chris@11: Chris@197: int colourMap = attributes.value("colourScheme").toInt(&ok); Chris@197: if (ok) setColourMap(colourMap); Chris@11: Chris@37: int colourRotation = attributes.value("colourRotation").toInt(&ok); Chris@37: if (ok) setColourRotation(colourRotation); Chris@37: Chris@11: FrequencyScale frequencyScale = (FrequencyScale) Chris@11: attributes.value("frequencyScale").toInt(&ok); Chris@11: if (ok) setFrequencyScale(frequencyScale); Chris@35: Chris@37: BinDisplay binDisplay = (BinDisplay) Chris@37: attributes.value("binDisplay").toInt(&ok); Chris@37: if (ok) setBinDisplay(binDisplay); Chris@36: Chris@36: bool normalizeColumns = Chris@36: (attributes.value("normalizeColumns").trimmed() == "true"); Chris@36: setNormalizeColumns(normalizeColumns); Chris@153: Chris@153: bool normalizeVisibleArea = Chris@153: (attributes.value("normalizeVisibleArea").trimmed() == "true"); Chris@153: setNormalizeVisibleArea(normalizeVisibleArea); Chris@11: } Chris@11: