Mercurial > hg > svcore
view data/fft/FFTFileCache.cpp @ 370:e73a1a1e8f24
* more reference frame records
author | Chris Cannam |
---|---|
date | Mon, 04 Feb 2008 16:48:52 +0000 |
parents | 92e8dbde73cd |
children | 7cc6b7b0d819 |
line wrap: on
line source
/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ /* Sonic Visualiser An audio file viewer and annotation editor. Centre for Digital Music, Queen Mary, University of London. This file copyright 2006 Chris Cannam and QMUL. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. See the file COPYING included with this distribution for more information. */ #include "FFTFileCache.h" #include "fileio/MatrixFile.h" #include "base/Profiler.h" #include <iostream> #include <QMutexLocker> // The underlying matrix has height (m_height * 2 + 1). In each // column we store magnitude at [0], [2] etc and phase at [1], [3] // etc, and then store the normalization factor (maximum magnitude) at // [m_height * 2]. In compact mode, the factor takes two cells. FFTFileCache::FFTFileCache(QString fileBase, MatrixFile::Mode mode, StorageType storageType) : m_writebuf(0), m_readbuf(0), m_readbufCol(0), m_readbufWidth(0), m_mfc(new MatrixFile (fileBase, mode, storageType == Compact ? sizeof(uint16_t) : sizeof(float), mode == MatrixFile::ReadOnly)), m_storageType(storageType), m_factorSize(storageType == Compact ? 2 : 1) { // std::cerr << "FFTFileCache: storage type is " << (storageType == Compact ? "Compact" : storageType == Polar ? "Polar" : "Rectangular") << std::endl; } FFTFileCache::~FFTFileCache() { if (m_readbuf) delete[] m_readbuf; if (m_writebuf) delete[] m_writebuf; delete m_mfc; } size_t FFTFileCache::getWidth() const { return m_mfc->getWidth(); } size_t FFTFileCache::getHeight() const { size_t mh = m_mfc->getHeight(); if (mh > m_factorSize) return (mh - m_factorSize) / 2; else return 0; } void FFTFileCache::resize(size_t width, size_t height) { QMutexLocker locker(&m_writeMutex); m_mfc->resize(width, height * 2 + m_factorSize); if (m_readbuf) { delete[] m_readbuf; m_readbuf = 0; } if (m_writebuf) { delete[] m_writebuf; } m_writebuf = new char[(height * 2 + m_factorSize) * m_mfc->getCellSize()]; } void FFTFileCache::reset() { m_mfc->reset(); } float FFTFileCache::getMagnitudeAt(size_t x, size_t y) const { Profiler profiler("FFTFileCache::getMagnitudeAt", false); float value = 0.f; switch (m_storageType) { case Compact: value = (getFromReadBufCompactUnsigned(x, y * 2) / 65535.0) * getNormalizationFactor(x); break; case Rectangular: { float real, imag; getValuesAt(x, y, real, imag); value = sqrtf(real * real + imag * imag); break; } case Polar: value = getFromReadBufStandard(x, y * 2); break; } return value; } float FFTFileCache::getNormalizedMagnitudeAt(size_t x, size_t y) const { float value = 0.f; switch (m_storageType) { case Compact: value = getFromReadBufCompactUnsigned(x, y * 2) / 65535.0; break; default: { float mag = getMagnitudeAt(x, y); float factor = getNormalizationFactor(x); if (factor != 0) value = mag / factor; else value = 0.f; break; } } return value; } float FFTFileCache::getMaximumMagnitudeAt(size_t x) const { return getNormalizationFactor(x); } float FFTFileCache::getPhaseAt(size_t x, size_t y) const { float value = 0.f; switch (m_storageType) { case Compact: value = (getFromReadBufCompactSigned(x, y * 2 + 1) / 32767.0) * M_PI; break; case Rectangular: { float real, imag; getValuesAt(x, y, real, imag); value = princargf(atan2f(imag, real)); break; } case Polar: value = getFromReadBufStandard(x, y * 2 + 1); break; } return value; } void FFTFileCache::getValuesAt(size_t x, size_t y, float &real, float &imag) const { switch (m_storageType) { case Rectangular: real = getFromReadBufStandard(x, y * 2); imag = getFromReadBufStandard(x, y * 2 + 1); return; default: float mag = getMagnitudeAt(x, y); float phase = getPhaseAt(x, y); real = mag * cosf(phase); imag = mag * sinf(phase); return; } } bool FFTFileCache::haveSetColumnAt(size_t x) const { return m_mfc->haveSetColumnAt(x); } void FFTFileCache::setColumnAt(size_t x, float *mags, float *phases, float factor) { QMutexLocker locker(&m_writeMutex); size_t h = getHeight(); switch (m_storageType) { case Compact: for (size_t y = 0; y < h; ++y) { ((uint16_t *)m_writebuf)[y * 2] = uint16_t((mags[y] / factor) * 65535.0); ((uint16_t *)m_writebuf)[y * 2 + 1] = uint16_t(int16_t((phases[y] * 32767) / M_PI)); } break; case Rectangular: for (size_t y = 0; y < h; ++y) { ((float *)m_writebuf)[y * 2] = mags[y] * cosf(phases[y]); ((float *)m_writebuf)[y * 2 + 1] = mags[y] * sinf(phases[y]); } break; case Polar: for (size_t y = 0; y < h; ++y) { ((float *)m_writebuf)[y * 2] = mags[y]; ((float *)m_writebuf)[y * 2 + 1] = phases[y]; } break; } // static float maxFactor = 0; // if (factor > maxFactor) maxFactor = factor; // std::cerr << "Column " << x << ": normalization factor: " << factor << ", max " << maxFactor << " (height " << getHeight() << ")" << std::endl; setNormalizationFactorToWritebuf(factor); m_mfc->setColumnAt(x, m_writebuf); } void FFTFileCache::setColumnAt(size_t x, float *real, float *imag) { QMutexLocker locker(&m_writeMutex); size_t h = getHeight(); float factor = 0.0f; switch (m_storageType) { case Compact: for (size_t y = 0; y < h; ++y) { float mag = sqrtf(real[y] * real[y] + imag[y] * imag[y]); if (mag > factor) factor = mag; } for (size_t y = 0; y < h; ++y) { float mag = sqrtf(real[y] * real[y] + imag[y] * imag[y]); float phase = princargf(atan2f(imag[y], real[y])); ((uint16_t *)m_writebuf)[y * 2] = uint16_t((mag / factor) * 65535.0); ((uint16_t *)m_writebuf)[y * 2 + 1] = uint16_t(int16_t((phase * 32767) / M_PI)); } break; case Rectangular: for (size_t y = 0; y < h; ++y) { ((float *)m_writebuf)[y * 2] = real[y]; ((float *)m_writebuf)[y * 2 + 1] = imag[y]; float mag = sqrtf(real[y] * real[y] + imag[y] * imag[y]); if (mag > factor) factor = mag; } break; case Polar: for (size_t y = 0; y < h; ++y) { float mag = sqrtf(real[y] * real[y] + imag[y] * imag[y]); if (mag > factor) factor = mag; ((float *)m_writebuf)[y * 2] = mag; float phase = princargf(atan2f(imag[y], real[y])); ((float *)m_writebuf)[y * 2 + 1] = phase; } break; } // static float maxFactor = 0; // if (factor > maxFactor) maxFactor = factor; // std::cerr << "[RI] Column " << x << ": normalization factor: " << factor << ", max " << maxFactor << " (height " << getHeight() << ")" << std::endl; setNormalizationFactorToWritebuf(factor); m_mfc->setColumnAt(x, m_writebuf); } size_t FFTFileCache::getCacheSize(size_t width, size_t height, StorageType type) { return (height * 2 + (type == Compact ? 2 : 1)) * width * (type == Compact ? sizeof(uint16_t) : sizeof(float)) + 2 * sizeof(size_t); // matrix file header size } void FFTFileCache::populateReadBuf(size_t x) const { Profiler profiler("FFTFileCache::populateReadBuf", false); if (!m_readbuf) { m_readbuf = new char[m_mfc->getHeight() * 2 * m_mfc->getCellSize()]; } m_mfc->getColumnAt(x, m_readbuf); if (m_mfc->haveSetColumnAt(x + 1)) { m_mfc->getColumnAt (x + 1, m_readbuf + m_mfc->getCellSize() * m_mfc->getHeight()); m_readbufWidth = 2; } else { m_readbufWidth = 1; } m_readbufCol = x; }