annotate data/model/FFTModel.h @ 1128:50210da3997c

Simple MIDI writer test
author Chris Cannam
date Tue, 01 Sep 2015 15:51:07 +0100
parents 9f4505ac9072
children 420fc961c0c4
rev   line source
Chris@152 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@152 2
Chris@152 3 /*
Chris@152 4 Sonic Visualiser
Chris@152 5 An audio file viewer and annotation editor.
Chris@152 6 Centre for Digital Music, Queen Mary, University of London.
Chris@152 7 This file copyright 2006 Chris Cannam.
Chris@152 8
Chris@152 9 This program is free software; you can redistribute it and/or
Chris@152 10 modify it under the terms of the GNU General Public License as
Chris@152 11 published by the Free Software Foundation; either version 2 of the
Chris@152 12 License, or (at your option) any later version. See the file
Chris@152 13 COPYING included with this distribution for more information.
Chris@152 14 */
Chris@152 15
Chris@1086 16 #ifndef FFT_MODEL_H
Chris@1086 17 #define FFT_MODEL_H
Chris@152 18
Chris@152 19 #include "data/fft/FFTDataServer.h"
Chris@152 20 #include "DenseThreeDimensionalModel.h"
Chris@152 21
Chris@275 22 #include <set>
Chris@275 23 #include <map>
Chris@275 24
Chris@254 25 /**
Chris@254 26 * An implementation of DenseThreeDimensionalModel that makes FFT data
Chris@387 27 * derived from a DenseTimeValueModel available as a generic data
Chris@387 28 * grid. The FFT data is acquired using FFTDataServer. Note that any
Chris@387 29 * of the accessor functions may throw AllocationFailed if a cache
Chris@387 30 * resize fails.
Chris@254 31 */
Chris@254 32
Chris@152 33 class FFTModel : public DenseThreeDimensionalModel
Chris@152 34 {
Chris@247 35 Q_OBJECT
Chris@247 36
Chris@152 37 public:
Chris@254 38 /**
Chris@254 39 * Construct an FFT model derived from the given
Chris@254 40 * DenseTimeValueModel, with the given window parameters and FFT
Chris@254 41 * size (which may exceed the window size, for zero-padded FFTs).
Chris@254 42 *
Chris@254 43 * If the model has multiple channels use only the given channel,
Chris@254 44 * unless the channel is -1 in which case merge all available
Chris@254 45 * channels.
Chris@254 46 *
Chris@254 47 * If polar is true, the data will normally be retrieved from the
Chris@254 48 * FFT model in magnitude/phase form; otherwise it will normally
Chris@254 49 * be retrieved in "cartesian" real/imaginary form. The results
Chris@254 50 * should be the same either way, but a "polar" model addressed in
Chris@254 51 * "cartesian" form or vice versa may suffer a performance
Chris@254 52 * penalty.
Chris@254 53 *
Chris@254 54 * The fillFromColumn argument gives a hint that the FFT data
Chris@254 55 * server should aim to start calculating FFT data at that column
Chris@254 56 * number if possible, as that is likely to be requested first.
Chris@254 57 */
Chris@152 58 FFTModel(const DenseTimeValueModel *model,
Chris@152 59 int channel,
Chris@152 60 WindowType windowType,
Chris@929 61 int windowSize,
Chris@929 62 int windowIncrement,
Chris@929 63 int fftSize,
Chris@152 64 bool polar,
Chris@334 65 StorageAdviser::Criteria criteria = StorageAdviser::NoCriteria,
Chris@1045 66 sv_frame_t fillFromFrame = 0);
Chris@152 67 ~FFTModel();
Chris@152 68
Chris@929 69 inline float getMagnitudeAt(int x, int y) {
Chris@152 70 return m_server->getMagnitudeAt(x << m_xshift, y << m_yshift);
Chris@152 71 }
Chris@929 72 inline float getNormalizedMagnitudeAt(int x, int y) {
Chris@152 73 return m_server->getNormalizedMagnitudeAt(x << m_xshift, y << m_yshift);
Chris@152 74 }
Chris@929 75 inline float getMaximumMagnitudeAt(int x) {
Chris@152 76 return m_server->getMaximumMagnitudeAt(x << m_xshift);
Chris@152 77 }
Chris@929 78 inline float getPhaseAt(int x, int y) {
Chris@152 79 return m_server->getPhaseAt(x << m_xshift, y << m_yshift);
Chris@152 80 }
Chris@929 81 inline void getValuesAt(int x, int y, float &real, float &imaginary) {
Chris@152 82 m_server->getValuesAt(x << m_xshift, y << m_yshift, real, imaginary);
Chris@152 83 }
Chris@929 84 inline bool isColumnAvailable(int x) const {
Chris@152 85 return m_server->isColumnReady(x << m_xshift);
Chris@152 86 }
Chris@152 87
Chris@929 88 inline bool getMagnitudesAt(int x, float *values, int minbin = 0, int count = 0) {
Chris@408 89 return m_server->getMagnitudesAt(x << m_xshift, values, minbin << m_yshift, count, getYRatio());
Chris@408 90 }
Chris@929 91 inline bool getNormalizedMagnitudesAt(int x, float *values, int minbin = 0, int count = 0) {
Chris@408 92 return m_server->getNormalizedMagnitudesAt(x << m_xshift, values, minbin << m_yshift, count, getYRatio());
Chris@408 93 }
Chris@929 94 inline bool getPhasesAt(int x, float *values, int minbin = 0, int count = 0) {
Chris@408 95 return m_server->getPhasesAt(x << m_xshift, values, minbin << m_yshift, count, getYRatio());
Chris@408 96 }
Chris@929 97 inline bool getValuesAt(int x, float *reals, float *imaginaries, int minbin = 0, int count = 0) {
Chris@556 98 return m_server->getValuesAt(x << m_xshift, reals, imaginaries, minbin << m_yshift, count, getYRatio());
Chris@556 99 }
Chris@408 100
Chris@1038 101 inline sv_frame_t getFillExtent() const { return m_server->getFillExtent(); }
Chris@152 102
Chris@152 103 // DenseThreeDimensionalModel and Model methods:
Chris@152 104 //
Chris@929 105 inline virtual int getWidth() const {
Chris@182 106 return m_server->getWidth() >> m_xshift;
Chris@182 107 }
Chris@929 108 inline virtual int getHeight() const {
Chris@212 109 // If there is no y-shift, the server's height (based on its
Chris@212 110 // fftsize/2 + 1) is correct. If there is a shift, then the
Chris@212 111 // server is using a larger fft size than we want, so we shift
Chris@212 112 // it right as many times as necessary, but then we need to
Chris@212 113 // re-add the "+1" part (because ((fftsize*2)/2 + 1) / 2 !=
Chris@212 114 // fftsize/2 + 1).
Chris@212 115 return (m_server->getHeight() >> m_yshift) + (m_yshift > 0 ? 1 : 0);
Chris@182 116 }
Chris@929 117 virtual float getValueAt(int x, int y) const {
Chris@182 118 return const_cast<FFTModel *>(this)->getMagnitudeAt(x, y);
Chris@182 119 }
Chris@152 120 virtual bool isOK() const {
Chris@1077 121 // Return true if the model was constructed successfully (not
Chris@1077 122 // necessarily whether an error has occurred since
Chris@1077 123 // construction, use getError for that)
Chris@152 124 return m_server && m_server->getModel();
Chris@152 125 }
Chris@1038 126 virtual sv_frame_t getStartFrame() const {
Chris@152 127 return 0;
Chris@152 128 }
Chris@1038 129 virtual sv_frame_t getEndFrame() const {
Chris@1038 130 return sv_frame_t(getWidth()) * getResolution() + getResolution();
Chris@152 131 }
Chris@1040 132 virtual sv_samplerate_t getSampleRate() const;
Chris@929 133 virtual int getResolution() const {
Chris@152 134 return m_server->getWindowIncrement() << m_xshift;
Chris@152 135 }
Chris@929 136 virtual int getYBinCount() const {
Chris@152 137 return getHeight();
Chris@152 138 }
Chris@152 139 virtual float getMinimumLevel() const {
Chris@152 140 return 0.f; // Can't provide
Chris@152 141 }
Chris@152 142 virtual float getMaximumLevel() const {
Chris@152 143 return 1.f; // Can't provide
Chris@152 144 }
Chris@929 145 virtual Column getColumn(int x) const;
Chris@929 146 virtual QString getBinName(int n) const;
Chris@152 147
Chris@478 148 virtual bool shouldUseLogValueScale() const {
Chris@478 149 return true; // Although obviously it's up to the user...
Chris@478 150 }
Chris@478 151
Chris@275 152 /**
Chris@275 153 * Calculate an estimated frequency for a stable signal in this
Chris@275 154 * bin, using phase unwrapping. This will be completely wrong if
Chris@275 155 * the signal is not stable here.
Chris@275 156 */
Chris@1045 157 virtual bool estimateStableFrequency(int x, int y, double &frequency);
Chris@275 158
Chris@275 159 enum PeakPickType
Chris@275 160 {
Chris@275 161 AllPeaks, /// Any bin exceeding its immediate neighbours
Chris@275 162 MajorPeaks, /// Peaks picked using sliding median window
Chris@275 163 MajorPitchAdaptivePeaks /// Bigger window for higher frequencies
Chris@275 164 };
Chris@275 165
Chris@929 166 typedef std::set<int> PeakLocationSet; // bin
Chris@1045 167 typedef std::map<int, double> PeakSet; // bin -> freq
Chris@275 168
Chris@275 169 /**
Chris@275 170 * Return locations of peak bins in the range [ymin,ymax]. If
Chris@275 171 * ymax is zero, getHeight()-1 will be used.
Chris@275 172 */
Chris@929 173 virtual PeakLocationSet getPeaks(PeakPickType type, int x,
Chris@929 174 int ymin = 0, int ymax = 0);
Chris@275 175
Chris@275 176 /**
Chris@275 177 * Return locations and estimated stable frequencies of peak bins.
Chris@275 178 */
Chris@929 179 virtual PeakSet getPeakFrequencies(PeakPickType type, int x,
Chris@929 180 int ymin = 0, int ymax = 0);
Chris@273 181
Chris@152 182 virtual int getCompletion() const { return m_server->getFillCompletion(); }
Chris@678 183 virtual QString getError() const { return m_server->getError(); }
Chris@152 184
Chris@154 185 virtual void suspend() { m_server->suspend(); }
Chris@155 186 virtual void suspendWrites() { m_server->suspendWrites(); }
Chris@154 187 virtual void resume() { m_server->resume(); }
Chris@154 188
Chris@345 189 QString getTypeName() const { return tr("FFT"); }
Chris@345 190
Chris@360 191 public slots:
Chris@360 192 void sourceModelAboutToBeDeleted();
Chris@360 193
Chris@152 194 private:
Chris@297 195 FFTModel(const FFTModel &); // not implemented
Chris@152 196 FFTModel &operator=(const FFTModel &); // not implemented
Chris@152 197
Chris@152 198 FFTDataServer *m_server;
Chris@152 199 int m_xshift;
Chris@152 200 int m_yshift;
Chris@275 201
Chris@297 202 FFTDataServer *getServer(const DenseTimeValueModel *,
Chris@929 203 int, WindowType, int, int, int,
Chris@1045 204 bool, StorageAdviser::Criteria, sv_frame_t);
Chris@297 205
Chris@1040 206 int getPeakPickWindowSize(PeakPickType type, sv_samplerate_t sampleRate,
Chris@1040 207 int bin, float &percentile) const;
Chris@408 208
Chris@929 209 int getYRatio() {
Chris@929 210 int ys = m_yshift;
Chris@929 211 int r = 1;
Chris@408 212 while (ys) { --ys; r <<= 1; }
Chris@408 213 return r;
Chris@408 214 }
Chris@152 215 };
Chris@152 216
Chris@152 217 #endif