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@152
|
16 #ifndef _FFT_MODEL_H_
|
Chris@152
|
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@152
|
22 class FFTModel : public DenseThreeDimensionalModel
|
Chris@152
|
23 {
|
Chris@247
|
24 Q_OBJECT
|
Chris@247
|
25
|
Chris@152
|
26 public:
|
Chris@152
|
27 FFTModel(const DenseTimeValueModel *model,
|
Chris@152
|
28 int channel,
|
Chris@152
|
29 WindowType windowType,
|
Chris@152
|
30 size_t windowSize,
|
Chris@152
|
31 size_t windowIncrement,
|
Chris@152
|
32 size_t fftSize,
|
Chris@152
|
33 bool polar,
|
Chris@152
|
34 size_t fillFromColumn = 0);
|
Chris@152
|
35 ~FFTModel();
|
Chris@152
|
36
|
Chris@152
|
37 float getMagnitudeAt(size_t x, size_t y) {
|
Chris@152
|
38 return m_server->getMagnitudeAt(x << m_xshift, y << m_yshift);
|
Chris@152
|
39 }
|
Chris@152
|
40 float getNormalizedMagnitudeAt(size_t x, size_t y) {
|
Chris@152
|
41 return m_server->getNormalizedMagnitudeAt(x << m_xshift, y << m_yshift);
|
Chris@152
|
42 }
|
Chris@152
|
43 float getMaximumMagnitudeAt(size_t x) {
|
Chris@152
|
44 return m_server->getMaximumMagnitudeAt(x << m_xshift);
|
Chris@152
|
45 }
|
Chris@152
|
46 float getPhaseAt(size_t x, size_t y) {
|
Chris@152
|
47 return m_server->getPhaseAt(x << m_xshift, y << m_yshift);
|
Chris@152
|
48 }
|
Chris@152
|
49 void getValuesAt(size_t x, size_t y, float &real, float &imaginary) {
|
Chris@152
|
50 m_server->getValuesAt(x << m_xshift, y << m_yshift, real, imaginary);
|
Chris@152
|
51 }
|
Chris@182
|
52 bool isColumnAvailable(size_t x) const {
|
Chris@152
|
53 return m_server->isColumnReady(x << m_xshift);
|
Chris@152
|
54 }
|
Chris@152
|
55
|
Chris@152
|
56 size_t getFillExtent() const { return m_server->getFillExtent(); }
|
Chris@152
|
57
|
Chris@152
|
58 // DenseThreeDimensionalModel and Model methods:
|
Chris@152
|
59 //
|
Chris@182
|
60 virtual size_t getWidth() const {
|
Chris@182
|
61 return m_server->getWidth() >> m_xshift;
|
Chris@182
|
62 }
|
Chris@182
|
63 virtual size_t getHeight() const {
|
Chris@212
|
64 // If there is no y-shift, the server's height (based on its
|
Chris@212
|
65 // fftsize/2 + 1) is correct. If there is a shift, then the
|
Chris@212
|
66 // server is using a larger fft size than we want, so we shift
|
Chris@212
|
67 // it right as many times as necessary, but then we need to
|
Chris@212
|
68 // re-add the "+1" part (because ((fftsize*2)/2 + 1) / 2 !=
|
Chris@212
|
69 // fftsize/2 + 1).
|
Chris@212
|
70 return (m_server->getHeight() >> m_yshift) + (m_yshift > 0 ? 1 : 0);
|
Chris@182
|
71 }
|
Chris@182
|
72 virtual float getValueAt(size_t x, size_t y) const {
|
Chris@182
|
73 return const_cast<FFTModel *>(this)->getMagnitudeAt(x, y);
|
Chris@182
|
74 }
|
Chris@152
|
75 virtual bool isOK() const {
|
Chris@152
|
76 return m_server && m_server->getModel();
|
Chris@152
|
77 }
|
Chris@152
|
78 virtual size_t getStartFrame() const {
|
Chris@152
|
79 return 0;
|
Chris@152
|
80 }
|
Chris@152
|
81 virtual size_t getEndFrame() const {
|
Chris@152
|
82 return getWidth() * getResolution() + getResolution();
|
Chris@152
|
83 }
|
Chris@152
|
84 virtual size_t getSampleRate() const;
|
Chris@152
|
85 virtual size_t getResolution() const {
|
Chris@152
|
86 return m_server->getWindowIncrement() << m_xshift;
|
Chris@152
|
87 }
|
Chris@152
|
88 virtual size_t getYBinCount() const {
|
Chris@152
|
89 return getHeight();
|
Chris@152
|
90 }
|
Chris@152
|
91 virtual float getMinimumLevel() const {
|
Chris@152
|
92 return 0.f; // Can't provide
|
Chris@152
|
93 }
|
Chris@152
|
94 virtual float getMaximumLevel() const {
|
Chris@152
|
95 return 1.f; // Can't provide
|
Chris@152
|
96 }
|
Chris@182
|
97 virtual void getColumn(size_t x, Column &result) const;
|
Chris@152
|
98 virtual QString getBinName(size_t n) const;
|
Chris@152
|
99
|
Chris@152
|
100 virtual int getCompletion() const { return m_server->getFillCompletion(); }
|
Chris@152
|
101
|
Chris@152
|
102 virtual Model *clone() const;
|
Chris@152
|
103
|
Chris@154
|
104 virtual void suspend() { m_server->suspend(); }
|
Chris@155
|
105 virtual void suspendWrites() { m_server->suspendWrites(); }
|
Chris@154
|
106 virtual void resume() { m_server->resume(); }
|
Chris@154
|
107
|
Chris@152
|
108 private:
|
Chris@152
|
109 FFTModel(const FFTModel &);
|
Chris@152
|
110 FFTModel &operator=(const FFTModel &); // not implemented
|
Chris@152
|
111
|
Chris@152
|
112 FFTDataServer *m_server;
|
Chris@152
|
113 int m_xshift;
|
Chris@152
|
114 int m_yshift;
|
Chris@152
|
115 };
|
Chris@152
|
116
|
Chris@152
|
117 #endif
|