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_FILE_CACHE_H_
lbajardsilogic@0: #define _FFT_FILE_CACHE_H_
lbajardsilogic@0: 
lbajardsilogic@0: #include "FFTCache.h"
lbajardsilogic@0: #include "fileio/MatrixFile.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <QMutex>
lbajardsilogic@0: 
lbajardsilogic@0: class FFTFileCache : public FFTCache
lbajardsilogic@0: {
lbajardsilogic@0: public:
lbajardsilogic@0:     enum StorageType {
lbajardsilogic@0:         Compact, // 16 bits normalized polar
lbajardsilogic@0:         Rectangular, // floating point real+imag
lbajardsilogic@0:         Polar, // floating point mag+phase
lbajardsilogic@0:     };
lbajardsilogic@0: 
lbajardsilogic@0:     FFTFileCache(QString fileBase, MatrixFile::Mode mode,
lbajardsilogic@0:                  StorageType storageType);
lbajardsilogic@0:     virtual ~FFTFileCache();
lbajardsilogic@0: 
lbajardsilogic@0:     MatrixFile::Mode getMode() const { return m_mfc->getMode(); }
lbajardsilogic@0: 
lbajardsilogic@0:     virtual size_t getWidth() const;
lbajardsilogic@0:     virtual size_t getHeight() const;
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:     virtual float getNormalizedMagnitudeAt(size_t x, size_t y) const;
lbajardsilogic@0:     virtual float getMaximumMagnitudeAt(size_t x) const;
lbajardsilogic@0:     virtual float getPhaseAt(size_t x, size_t y) const;
lbajardsilogic@0: 
lbajardsilogic@0:     virtual void getValuesAt(size_t x, size_t y, float &real, float &imag) const;
lbajardsilogic@0: 
lbajardsilogic@0:     virtual bool haveSetColumnAt(size_t x) const;
lbajardsilogic@0: 
lbajardsilogic@0:     virtual void setColumnAt(size_t x, float *mags, float *phases, float factor);
lbajardsilogic@0:     virtual void setColumnAt(size_t x, float *reals, float *imags);
lbajardsilogic@0: 
lbajardsilogic@0:     virtual void suspend() { m_mfc->suspend(); }
lbajardsilogic@0: 
lbajardsilogic@0:     static size_t getCacheSize(size_t width, size_t height, StorageType type);
lbajardsilogic@0: 
lbajardsilogic@0: protected:
lbajardsilogic@0:     char *m_writebuf;
lbajardsilogic@0:     mutable char *m_readbuf;
lbajardsilogic@0:     mutable size_t m_readbufCol;
lbajardsilogic@0:     mutable size_t m_readbufWidth;
lbajardsilogic@0: 
lbajardsilogic@0:     float getFromReadBufStandard(size_t x, size_t y) const {
lbajardsilogic@0:         if (m_readbuf &&
lbajardsilogic@0:             (m_readbufCol == x || (m_readbufWidth > 1 && m_readbufCol+1 == x))) {
lbajardsilogic@0:             return ((float *)m_readbuf)[(x - m_readbufCol) * m_mfc->getHeight() + y];
lbajardsilogic@0:         } else {
lbajardsilogic@0:             populateReadBuf(x);
lbajardsilogic@0:             return getFromReadBufStandard(x, y);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     float getFromReadBufCompactUnsigned(size_t x, size_t y) const {
lbajardsilogic@0:         if (m_readbuf &&
lbajardsilogic@0:             (m_readbufCol == x || (m_readbufWidth > 1 && m_readbufCol+1 == x))) {
lbajardsilogic@0:             return ((uint16_t *)m_readbuf)[(x - m_readbufCol) * m_mfc->getHeight() + y];
lbajardsilogic@0:         } else {
lbajardsilogic@0:             populateReadBuf(x);
lbajardsilogic@0:             return getFromReadBufCompactUnsigned(x, y);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     float getFromReadBufCompactSigned(size_t x, size_t y) const {
lbajardsilogic@0:         if (m_readbuf &&
lbajardsilogic@0:             (m_readbufCol == x || (m_readbufWidth > 1 && m_readbufCol+1 == x))) {
lbajardsilogic@0:             return ((int16_t *)m_readbuf)[(x - m_readbufCol) * m_mfc->getHeight() + y];
lbajardsilogic@0:         } else {
lbajardsilogic@0:             populateReadBuf(x);
lbajardsilogic@0:             return getFromReadBufCompactSigned(x, y);
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     void populateReadBuf(size_t x) const;
lbajardsilogic@0: 
lbajardsilogic@0:     float getNormalizationFactor(size_t col) const {
lbajardsilogic@0:         if (m_storageType != Compact) {
lbajardsilogic@0:             return getFromReadBufStandard(col, m_mfc->getHeight() - 1);
lbajardsilogic@0:         } else {
lbajardsilogic@0:             float factor;
lbajardsilogic@0:             factor = getFromReadBufCompactUnsigned(col, m_mfc->getHeight() - 1);
lbajardsilogic@0:             return factor / 65535.0;
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     MatrixFile *m_mfc;
lbajardsilogic@0:     QMutex m_writeMutex;
lbajardsilogic@0:     StorageType m_storageType;
lbajardsilogic@0: };
lbajardsilogic@0: 
lbajardsilogic@0: #endif