lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ lbajardsilogic@0: lbajardsilogic@0: /* lbajardsilogic@0: Sonic Visualiser lbajardsilogic@0: An audio file viewer and annotation editor. lbajardsilogic@0: Centre for Digital Music, Queen Mary, University of London. lbajardsilogic@0: This file copyright 2006 Chris Cannam. lbajardsilogic@0: lbajardsilogic@0: This program is free software; you can redistribute it and/or lbajardsilogic@0: modify it under the terms of the GNU General Public License as lbajardsilogic@0: published by the Free Software Foundation; either version 2 of the lbajardsilogic@0: License, or (at your option) any later version. See the file lbajardsilogic@0: COPYING included with this distribution for more information. lbajardsilogic@0: */ lbajardsilogic@0: lbajardsilogic@0: #ifndef _FFT_MEMORY_CACHE_H_ lbajardsilogic@0: #define _FFT_MEMORY_CACHE_H_ lbajardsilogic@0: lbajardsilogic@0: #include "FFTCache.h" lbajardsilogic@0: lbajardsilogic@0: #include "base/ResizeableBitset.h" lbajardsilogic@0: lbajardsilogic@0: /** lbajardsilogic@0: * In-memory FFT cache. For this we want to cache magnitude with lbajardsilogic@0: * enough resolution to have gain applied afterwards and determine lbajardsilogic@0: * whether something is a peak or not, and also cache phase rather lbajardsilogic@0: * than only phase-adjusted frequency so that we don't have to lbajardsilogic@0: * recalculate if switching between phase and magnitude displays. At lbajardsilogic@0: * the same time, we don't want to take up too much memory. It's not lbajardsilogic@0: * expected to be accurate enough to be used as input for DSP or lbajardsilogic@0: * resynthesis code. lbajardsilogic@0: * lbajardsilogic@0: * This implies probably 16 bits for a normalized magnitude and at lbajardsilogic@0: * most 16 bits for phase. lbajardsilogic@0: * lbajardsilogic@0: * Each column's magnitudes are expected to be stored normalized lbajardsilogic@0: * to [0,1] with respect to the column, so the normalization lbajardsilogic@0: * factor should be calculated before all values in a column, and lbajardsilogic@0: * set appropriately. lbajardsilogic@0: */ lbajardsilogic@0: lbajardsilogic@0: class FFTMemoryCache : public FFTCache lbajardsilogic@0: { lbajardsilogic@0: public: lbajardsilogic@0: FFTMemoryCache(); // of size zero, call resize() before using lbajardsilogic@0: virtual ~FFTMemoryCache(); lbajardsilogic@0: lbajardsilogic@0: virtual size_t getWidth() const { return m_width; } lbajardsilogic@0: virtual size_t getHeight() const { return m_height; } lbajardsilogic@0: lbajardsilogic@0: virtual void resize(size_t width, size_t height); lbajardsilogic@0: virtual void reset(); // zero-fill or 1-fill as appropriate without changing size lbajardsilogic@0: lbajardsilogic@0: virtual float getMagnitudeAt(size_t x, size_t y) const { lbajardsilogic@0: return getNormalizedMagnitudeAt(x, y) * m_factor[x]; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual float getNormalizedMagnitudeAt(size_t x, size_t y) const { lbajardsilogic@0: return float(m_magnitude[x][y]) / 65535.0; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual float getMaximumMagnitudeAt(size_t x) const { lbajardsilogic@0: return m_factor[x]; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual float getPhaseAt(size_t x, size_t y) const { lbajardsilogic@0: int16_t i = (int16_t)m_phase[x][y]; lbajardsilogic@0: return (float(i) / 32767.0) * M_PI; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void getValuesAt(size_t x, size_t y, float &real, float &imag) const { lbajardsilogic@0: float mag = getMagnitudeAt(x, y); lbajardsilogic@0: float phase = getPhaseAt(x, y); lbajardsilogic@0: real = mag * cosf(phase); lbajardsilogic@0: imag = mag * sinf(phase); lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setNormalizationFactor(size_t x, float factor) { lbajardsilogic@0: if (x < m_width) m_factor[x] = factor; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setMagnitudeAt(size_t x, size_t y, float mag) { lbajardsilogic@0: // norm factor must already be set lbajardsilogic@0: setNormalizedMagnitudeAt(x, y, mag / m_factor[x]); lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setNormalizedMagnitudeAt(size_t x, size_t y, float norm) { lbajardsilogic@0: if (x < m_width && y < m_height) { lbajardsilogic@0: m_magnitude[x][y] = uint16_t(norm * 65535.0); lbajardsilogic@0: } lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setPhaseAt(size_t x, size_t y, float phase) { lbajardsilogic@0: // phase in range -pi -> pi lbajardsilogic@0: if (x < m_width && y < m_height) { lbajardsilogic@0: m_phase[x][y] = uint16_t(int16_t((phase * 32767) / M_PI)); lbajardsilogic@0: } lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual bool haveSetColumnAt(size_t x) const { lbajardsilogic@0: return m_colset.get(x); lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setColumnAt(size_t x, float *mags, float *phases, float factor) { lbajardsilogic@0: setNormalizationFactor(x, factor); lbajardsilogic@0: for (size_t y = 0; y < m_height; ++y) { lbajardsilogic@0: setMagnitudeAt(x, y, mags[y]); lbajardsilogic@0: setPhaseAt(x, y, phases[y]); lbajardsilogic@0: } lbajardsilogic@0: m_colset.set(x); lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: virtual void setColumnAt(size_t x, float *reals, float *imags); lbajardsilogic@0: lbajardsilogic@0: static size_t getCacheSize(size_t width, size_t height); lbajardsilogic@0: lbajardsilogic@0: private: lbajardsilogic@0: size_t m_width; lbajardsilogic@0: size_t m_height; lbajardsilogic@0: uint16_t **m_magnitude; lbajardsilogic@0: uint16_t **m_phase; lbajardsilogic@0: float *m_factor; lbajardsilogic@0: ResizeableBitset m_colset; lbajardsilogic@0: lbajardsilogic@0: void resize(uint16_t **&, size_t, size_t); lbajardsilogic@0: }; lbajardsilogic@0: lbajardsilogic@0: lbajardsilogic@0: #endif lbajardsilogic@0: