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 and QMUL. 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_DATA_SERVER_H_ lbajardsilogic@0: #define _FFT_DATA_SERVER_H_ lbajardsilogic@0: lbajardsilogic@0: #include "base/Window.h" lbajardsilogic@0: #include "base/Thread.h" lbajardsilogic@0: lbajardsilogic@0: #include "FFTapi.h" lbajardsilogic@0: lbajardsilogic@0: #include lbajardsilogic@0: #include lbajardsilogic@0: #include lbajardsilogic@0: lbajardsilogic@0: #include lbajardsilogic@0: #include lbajardsilogic@0: lbajardsilogic@0: class DenseTimeValueModel; lbajardsilogic@0: class Model; lbajardsilogic@0: class FFTCache; lbajardsilogic@0: lbajardsilogic@0: class FFTDataServer lbajardsilogic@0: { lbajardsilogic@0: public: lbajardsilogic@0: static FFTDataServer *getInstance(const DenseTimeValueModel *model, lbajardsilogic@0: int channel, lbajardsilogic@0: WindowType windowType, lbajardsilogic@0: size_t windowSize, lbajardsilogic@0: size_t windowIncrement, lbajardsilogic@0: size_t fftSize, lbajardsilogic@0: bool polar, lbajardsilogic@0: size_t fillFromColumn = 0); lbajardsilogic@0: lbajardsilogic@0: static FFTDataServer *getFuzzyInstance(const DenseTimeValueModel *model, lbajardsilogic@0: int channel, lbajardsilogic@0: WindowType windowType, lbajardsilogic@0: size_t windowSize, lbajardsilogic@0: size_t windowIncrement, lbajardsilogic@0: size_t fftSize, lbajardsilogic@0: bool polar, lbajardsilogic@0: size_t fillFromColumn = 0); lbajardsilogic@0: lbajardsilogic@0: static void claimInstance(FFTDataServer *); lbajardsilogic@0: static void releaseInstance(FFTDataServer *); lbajardsilogic@0: lbajardsilogic@0: static void modelAboutToBeDeleted(Model *); lbajardsilogic@0: lbajardsilogic@0: const DenseTimeValueModel *getModel() const { return m_model; } lbajardsilogic@0: int getChannel() const { return m_channel; } lbajardsilogic@0: WindowType getWindowType() const { return m_windower.getType(); } lbajardsilogic@0: size_t getWindowSize() const { return m_windowSize; } lbajardsilogic@0: size_t getWindowIncrement() const { return m_windowIncrement; } lbajardsilogic@0: size_t getFFTSize() const { return m_fftSize; } lbajardsilogic@0: bool getPolar() const { return m_polar; } lbajardsilogic@0: lbajardsilogic@0: size_t getWidth() const { return m_width; } lbajardsilogic@0: size_t getHeight() const { return m_height; } lbajardsilogic@0: lbajardsilogic@0: float getMagnitudeAt(size_t x, size_t y); lbajardsilogic@0: float getNormalizedMagnitudeAt(size_t x, size_t y); lbajardsilogic@0: float getMaximumMagnitudeAt(size_t x); lbajardsilogic@0: float getPhaseAt(size_t x, size_t y); lbajardsilogic@0: void getValuesAt(size_t x, size_t y, float &real, float &imaginary); lbajardsilogic@0: bool isColumnReady(size_t x); lbajardsilogic@0: lbajardsilogic@0: void suspend(); lbajardsilogic@0: void suspendWrites(); lbajardsilogic@0: void resume(); // also happens automatically if new data needed lbajardsilogic@0: lbajardsilogic@0: // Convenience functions: lbajardsilogic@0: lbajardsilogic@0: bool isLocalPeak(size_t x, size_t y) { lbajardsilogic@0: float mag = getMagnitudeAt(x, y); lbajardsilogic@0: if (y > 0 && mag < getMagnitudeAt(x, y - 1)) return false; lbajardsilogic@0: if (y < getHeight()-1 && mag < getMagnitudeAt(x, y + 1)) return false; lbajardsilogic@0: return true; lbajardsilogic@0: } lbajardsilogic@0: bool isOverThreshold(size_t x, size_t y, float threshold) { lbajardsilogic@0: return getMagnitudeAt(x, y) > threshold; lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: size_t getFillCompletion() const; lbajardsilogic@0: size_t getFillExtent() const; lbajardsilogic@0: lbajardsilogic@0: private: lbajardsilogic@0: FFTDataServer(QString fileBaseName, lbajardsilogic@0: const DenseTimeValueModel *model, lbajardsilogic@0: int channel, lbajardsilogic@0: WindowType windowType, lbajardsilogic@0: size_t windowSize, lbajardsilogic@0: size_t windowIncrement, lbajardsilogic@0: size_t fftSize, lbajardsilogic@0: bool polar, lbajardsilogic@0: size_t fillFromColumn = 0); lbajardsilogic@0: lbajardsilogic@0: virtual ~FFTDataServer(); lbajardsilogic@0: lbajardsilogic@0: FFTDataServer(const FFTDataServer &); // not implemented lbajardsilogic@0: FFTDataServer &operator=(const FFTDataServer &); // not implemented lbajardsilogic@0: lbajardsilogic@0: typedef float fftsample; lbajardsilogic@0: lbajardsilogic@0: QString m_fileBaseName; lbajardsilogic@0: const DenseTimeValueModel *m_model; lbajardsilogic@0: int m_channel; lbajardsilogic@0: lbajardsilogic@0: Window m_windower; lbajardsilogic@0: lbajardsilogic@0: size_t m_windowSize; lbajardsilogic@0: size_t m_windowIncrement; lbajardsilogic@0: size_t m_fftSize; lbajardsilogic@0: bool m_polar; lbajardsilogic@0: lbajardsilogic@0: size_t m_width; lbajardsilogic@0: size_t m_height; lbajardsilogic@0: size_t m_cacheWidth; lbajardsilogic@0: size_t m_cacheWidthPower; lbajardsilogic@0: size_t m_cacheWidthMask; lbajardsilogic@0: bool m_memoryCache; lbajardsilogic@0: bool m_compactCache; lbajardsilogic@0: lbajardsilogic@0: typedef std::vector CacheVector; lbajardsilogic@0: CacheVector m_caches; lbajardsilogic@0: lbajardsilogic@0: typedef std::deque IntQueue; lbajardsilogic@0: IntQueue m_dormantCaches; lbajardsilogic@0: lbajardsilogic@0: int m_lastUsedCache; lbajardsilogic@0: FFTCache *getCache(size_t x, size_t &col) { lbajardsilogic@0: col = x % m_cacheWidth; lbajardsilogic@0: int c = x / m_cacheWidth; lbajardsilogic@0: // The only use of m_lastUsedCache without a lock is to lbajardsilogic@0: // establish whether a cache has been created at all (they're lbajardsilogic@0: // created on demand, but not destroyed until the server is). lbajardsilogic@0: if (c == m_lastUsedCache) return m_caches[c]; lbajardsilogic@0: else return getCacheAux(c); lbajardsilogic@0: } lbajardsilogic@0: bool haveCache(size_t x) { lbajardsilogic@0: int c = x / m_cacheWidth; lbajardsilogic@0: if (c == m_lastUsedCache) return true; lbajardsilogic@0: else return (m_caches[c] != 0); lbajardsilogic@0: } lbajardsilogic@0: lbajardsilogic@0: FFTCache *getCacheAux(size_t c); lbajardsilogic@0: QMutex m_writeMutex; lbajardsilogic@0: QWaitCondition m_condition; lbajardsilogic@0: lbajardsilogic@0: fftsample *m_fftInput; lbajardsilogic@0: fftf_complex *m_fftOutput; lbajardsilogic@0: float *m_workbuffer; lbajardsilogic@0: fftf_plan m_fftPlan; lbajardsilogic@0: lbajardsilogic@0: class FillThread : public Thread lbajardsilogic@0: { lbajardsilogic@0: public: lbajardsilogic@0: FillThread(FFTDataServer &server, size_t fillFromColumn) : lbajardsilogic@0: m_server(server), m_extent(0), m_completion(0), lbajardsilogic@0: m_fillFrom(fillFromColumn) { } lbajardsilogic@0: lbajardsilogic@0: size_t getExtent() const { return m_extent; } lbajardsilogic@0: size_t getCompletion() const { return m_completion ? m_completion : 1; } lbajardsilogic@0: virtual void run(); lbajardsilogic@0: lbajardsilogic@0: protected: lbajardsilogic@0: FFTDataServer &m_server; lbajardsilogic@0: size_t m_extent; lbajardsilogic@0: size_t m_completion; lbajardsilogic@0: size_t m_fillFrom; lbajardsilogic@0: }; lbajardsilogic@0: lbajardsilogic@0: bool m_exiting; lbajardsilogic@0: bool m_suspended; lbajardsilogic@0: FillThread *m_fillThread; lbajardsilogic@0: lbajardsilogic@0: void deleteProcessingData(); lbajardsilogic@0: void fillColumn(size_t x); lbajardsilogic@0: lbajardsilogic@0: QString generateFileBasename() const; lbajardsilogic@0: static QString generateFileBasename(const DenseTimeValueModel *model, lbajardsilogic@0: int channel, lbajardsilogic@0: WindowType windowType, lbajardsilogic@0: size_t windowSize, lbajardsilogic@0: size_t windowIncrement, lbajardsilogic@0: size_t fftSize, lbajardsilogic@0: bool polar); lbajardsilogic@0: lbajardsilogic@0: typedef std::pair ServerCountPair; lbajardsilogic@0: typedef std::map ServerMap; lbajardsilogic@0: typedef std::deque ServerQueue; lbajardsilogic@0: lbajardsilogic@0: static ServerMap m_servers; lbajardsilogic@0: static ServerQueue m_releasedServers; // these are still in m_servers as well, with zero refcount lbajardsilogic@0: static QMutex m_serverMapMutex; lbajardsilogic@0: static FFTDataServer *findServer(QString); // call with serverMapMutex held lbajardsilogic@0: static void purgeLimbo(int maxSize = 3); // call with serverMapMutex held lbajardsilogic@0: lbajardsilogic@0: static void claimInstance(FFTDataServer *, bool needLock); lbajardsilogic@0: static void releaseInstance(FFTDataServer *, bool needLock); lbajardsilogic@0: lbajardsilogic@0: }; lbajardsilogic@0: lbajardsilogic@0: #endif