Chris@148: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@148:
Chris@148: /*
Chris@148: Sonic Visualiser
Chris@148: An audio file viewer and annotation editor.
Chris@148: Centre for Digital Music, Queen Mary, University of London.
Chris@202: This file copyright 2006 Chris Cannam and QMUL.
Chris@148:
Chris@148: This program is free software; you can redistribute it and/or
Chris@148: modify it under the terms of the GNU General Public License as
Chris@148: published by the Free Software Foundation; either version 2 of the
Chris@148: License, or (at your option) any later version. See the file
Chris@148: COPYING included with this distribution for more information.
Chris@148: */
Chris@148:
Chris@148: #include "FFTDataServer.h"
Chris@148:
Chris@537: #include "FFTFileCacheReader.h"
Chris@537: #include "FFTFileCacheWriter.h"
Chris@159: #include "FFTMemoryCache.h"
Chris@148:
Chris@148: #include "model/DenseTimeValueModel.h"
Chris@148:
Chris@150: #include "system/System.h"
Chris@148:
Chris@168: #include "base/StorageAdviser.h"
Chris@200: #include "base/Exceptions.h"
Chris@183: #include "base/Profiler.h"
Chris@244: #include "base/Thread.h" // for debug mutex locker
Chris@168:
Chris@537: #include <QWriteLocker>
Chris@537:
Chris@1038: #include <stdexcept>
Chris@1038:
Chris@571: //#define DEBUG_FFT_SERVER 1
Chris@536: //#define DEBUG_FFT_SERVER_FILL 1
Chris@148:
Chris@148: #ifdef DEBUG_FFT_SERVER_FILL
Chris@153: #ifndef DEBUG_FFT_SERVER
Chris@153: #define DEBUG_FFT_SERVER 1
Chris@153: #endif
Chris@148: #endif
Chris@148:
Chris@244:
Chris@148: FFTDataServer::ServerMap FFTDataServer::m_servers;
Chris@215: FFTDataServer::ServerQueue FFTDataServer::m_releasedServers;
Chris@148: QMutex FFTDataServer::m_serverMapMutex;
Chris@148:
Chris@148: FFTDataServer *
Chris@148: FFTDataServer::getInstance(const DenseTimeValueModel *model,
Chris@148: int channel,
Chris@148: WindowType windowType,
Chris@929: int windowSize,
Chris@929: int windowIncrement,
Chris@929: int fftSize,
Chris@148: bool polar,
Chris@334: StorageAdviser::Criteria criteria,
Chris@1045: sv_frame_t fillFromFrame)
Chris@148: {
Chris@148: QString n = generateFileBasename(model,
Chris@148: channel,
Chris@148: windowType,
Chris@148: windowSize,
Chris@148: windowIncrement,
Chris@148: fftSize,
Chris@148: polar);
Chris@148:
Chris@148: FFTDataServer *server = 0;
Chris@148:
Chris@408: MutexLocker locker(&m_serverMapMutex, "FFTDataServer::getInstance::m_serverMapMutex");
Chris@148:
Chris@148: if ((server = findServer(n))) {
Chris@148: return server;
Chris@148: }
Chris@148:
Chris@148: QString npn = generateFileBasename(model,
Chris@148: channel,
Chris@148: windowType,
Chris@148: windowSize,
Chris@148: windowIncrement,
Chris@148: fftSize,
Chris@148: !polar);
Chris@148:
Chris@148: if ((server = findServer(npn))) {
Chris@148: return server;
Chris@148: }
Chris@148:
Chris@200: try {
Chris@200: server = new FFTDataServer(n,
Chris@200: model,
Chris@200: channel,
Chris@200: windowType,
Chris@200: windowSize,
Chris@200: windowIncrement,
Chris@200: fftSize,
Chris@200: polar,
Chris@334: criteria,
Chris@1045: fillFromFrame);
Chris@200: } catch (InsufficientDiscSpace) {
Chris@200: delete server;
Chris@200: server = 0;
Chris@200: }
Chris@148:
Chris@200: if (server) {
Chris@200: m_servers[n] = ServerCountPair(server, 1);
Chris@200: }
Chris@200:
Chris@200: return server;
Chris@148: }
Chris@148:
Chris@148: FFTDataServer *
Chris@148: FFTDataServer::getFuzzyInstance(const DenseTimeValueModel *model,
Chris@148: int channel,
Chris@148: WindowType windowType,
Chris@929: int windowSize,
Chris@929: int windowIncrement,
Chris@929: int fftSize,
Chris@148: bool polar,
Chris@334: StorageAdviser::Criteria criteria,
Chris@1045: sv_frame_t fillFromFrame)
Chris@148: {
Chris@148: // Fuzzy matching:
Chris@148: //
Chris@148: // -- if we're asked for polar and have non-polar, use it (and
Chris@148: // vice versa). This one is vital, and we do it for non-fuzzy as
Chris@148: // well (above).
Chris@148: //
Chris@148: // -- if we're asked for an instance with a given fft size and we
Chris@148: // have one already with a multiple of that fft size but the same
Chris@148: // window size and type (and model), we can draw the results from
Chris@148: // it (e.g. the 1st, 2nd, 3rd etc bins of a 512-sample FFT are the
Chris@148: // same as the the 1st, 5th, 9th etc of a 2048-sample FFT of the
Chris@148: // same window plus zero padding).
Chris@148: //
Chris@148: // -- if we're asked for an instance with a given window type and
Chris@148: // size and fft size and we have one already the same but with a
Chris@148: // smaller increment, we can draw the results from it (provided
Chris@148: // our increment is a multiple of its)
Chris@148: //
Chris@152: // The FFTModel knows how to interpret these things. In
Chris@148: // both cases we require that the larger one is a power-of-two
Chris@148: // multiple of the smaller (e.g. even though in principle you can
Chris@148: // draw the results at increment 256 from those at increment 768
Chris@152: // or 1536, the model doesn't support this).
Chris@148:
Chris@148: {
Chris@408: MutexLocker locker(&m_serverMapMutex, "FFTDataServer::getFuzzyInstance::m_serverMapMutex");
Chris@148:
Chris@148: ServerMap::iterator best = m_servers.end();
Chris@148: int bestdist = -1;
Chris@148:
Chris@148: for (ServerMap::iterator i = m_servers.begin(); i != m_servers.end(); ++i) {
Chris@148:
Chris@148: FFTDataServer *server = i->second.first;
Chris@148:
Chris@148: if (server->getModel() == model &&
Chris@148: (server->getChannel() == channel || model->getChannelCount() == 1) &&
Chris@148: server->getWindowType() == windowType &&
Chris@148: server->getWindowSize() == windowSize &&
Chris@148: server->getWindowIncrement() <= windowIncrement &&
Chris@148: server->getFFTSize() >= fftSize) {
Chris@148:
Chris@148: if ((windowIncrement % server->getWindowIncrement()) != 0) continue;
Chris@148: int ratio = windowIncrement / server->getWindowIncrement();
Chris@148: bool poweroftwo = true;
Chris@148: while (ratio > 1) {
Chris@148: if (ratio & 0x1) {
Chris@148: poweroftwo = false;
Chris@148: break;
Chris@148: }
Chris@148: ratio >>= 1;
Chris@148: }
Chris@148: if (!poweroftwo) continue;
Chris@148:
Chris@148: if ((server->getFFTSize() % fftSize) != 0) continue;
Chris@148: ratio = server->getFFTSize() / fftSize;
Chris@148: while (ratio > 1) {
Chris@148: if (ratio & 0x1) {
Chris@148: poweroftwo = false;
Chris@148: break;
Chris@148: }
Chris@148: ratio >>= 1;
Chris@148: }
Chris@148: if (!poweroftwo) continue;
Chris@148:
Chris@148: int distance = 0;
Chris@148:
Chris@148: if (server->getPolar() != polar) distance += 1;
Chris@148:
Chris@148: distance += ((windowIncrement / server->getWindowIncrement()) - 1) * 15;
Chris@148: distance += ((server->getFFTSize() / fftSize) - 1) * 10;
Chris@148:
Chris@148: if (server->getFillCompletion() < 50) distance += 100;
Chris@148:
Chris@148: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::getFuzzyInstance: Distance for server " << server << " is " << distance << ", best is " << bestdist << std::endl;
Chris@148: #endif
Chris@148:
Chris@148: if (bestdist == -1 || distance < bestdist) {
Chris@148: bestdist = distance;
Chris@148: best = i;
Chris@148: }
Chris@148: }
Chris@148: }
Chris@148:
Chris@148: if (bestdist >= 0) {
Chris@216: FFTDataServer *server = best->second.first;
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::getFuzzyInstance: We like server " << server << " (with distance " << bestdist << ")" << std::endl;
Chris@216: #endif
Chris@216: claimInstance(server, false);
Chris@216: return server;
Chris@148: }
Chris@148: }
Chris@148:
Chris@148: // Nothing found, make a new one
Chris@148:
Chris@148: return getInstance(model,
Chris@148: channel,
Chris@148: windowType,
Chris@148: windowSize,
Chris@148: windowIncrement,
Chris@148: fftSize,
Chris@148: polar,
Chris@334: criteria,
Chris@1045: fillFromFrame);
Chris@148: }
Chris@148:
Chris@148: FFTDataServer *
Chris@148: FFTDataServer::findServer(QString n)
Chris@148: {
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::findServer(\"" << n << "\")" << std::endl;
Chris@216: #endif
Chris@216:
Chris@148: if (m_servers.find(n) != m_servers.end()) {
Chris@216:
Chris@216: FFTDataServer *server = m_servers[n].first;
Chris@216:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::findServer(\"" << n << "\"): found " << server << std::endl;
Chris@216: #endif
Chris@216:
Chris@216: claimInstance(server, false);
Chris@216:
Chris@216: return server;
Chris@148: }
Chris@148:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::findServer(\"" << n << "\"): not found" << std::endl;
Chris@216: #endif
Chris@216:
Chris@148: return 0;
Chris@148: }
Chris@148:
Chris@148: void
Chris@152: FFTDataServer::claimInstance(FFTDataServer *server)
Chris@152: {
Chris@216: claimInstance(server, true);
Chris@216: }
Chris@216:
Chris@216: void
Chris@216: FFTDataServer::claimInstance(FFTDataServer *server, bool needLock)
Chris@216: {
Chris@244: MutexLocker locker(needLock ? &m_serverMapMutex : 0,
Chris@408: "FFTDataServer::claimInstance::m_serverMapMutex");
Chris@216:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::claimInstance(" << server << ")" << std::endl;
Chris@216: #endif
Chris@152:
Chris@152: for (ServerMap::iterator i = m_servers.begin(); i != m_servers.end(); ++i) {
Chris@152: if (i->second.first == server) {
Chris@215:
Chris@215: for (ServerQueue::iterator j = m_releasedServers.begin();
Chris@215: j != m_releasedServers.end(); ++j) {
Chris@216:
Chris@215: if (*j == server) {
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::claimInstance: found in released server list, removing from it" << std::endl;
Chris@216: #endif
Chris@215: m_releasedServers.erase(j);
Chris@215: break;
Chris@215: }
Chris@215: }
Chris@215:
Chris@152: ++i->second.second;
Chris@216:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::claimInstance: new refcount is " << i->second.second << std::endl;
Chris@216: #endif
Chris@216:
Chris@152: return;
Chris@152: }
Chris@152: }
Chris@152:
Chris@843: cerr << "ERROR: FFTDataServer::claimInstance: instance "
Chris@843: << server << " unknown!" << endl;
Chris@152: }
Chris@152:
Chris@152: void
Chris@148: FFTDataServer::releaseInstance(FFTDataServer *server)
Chris@148: {
Chris@216: releaseInstance(server, true);
Chris@216: }
Chris@216:
Chris@216: void
Chris@216: FFTDataServer::releaseInstance(FFTDataServer *server, bool needLock)
Chris@216: {
Chris@244: MutexLocker locker(needLock ? &m_serverMapMutex : 0,
Chris@408: "FFTDataServer::releaseInstance::m_serverMapMutex");
Chris@216:
Chris@148: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::releaseInstance(" << server << ")" << std::endl;
Chris@148: #endif
Chris@148:
Chris@148: // -- if ref count > 0, decrement and return
Chris@148: // -- if the instance hasn't been used at all, delete it immediately
Chris@148: // -- if fewer than N instances (N = e.g. 3) remain with zero refcounts,
Chris@148: // leave them hanging around
Chris@148: // -- if N instances with zero refcounts remain, delete the one that
Chris@148: // was last released first
Chris@148: // -- if we run out of disk space when allocating an instance, go back
Chris@148: // and delete the spare N instances before trying again
Chris@148: // -- have an additional method to indicate that a model has been
Chris@148: // destroyed, so that we can delete all of its fft server instances
Chris@148:
Chris@148: for (ServerMap::iterator i = m_servers.begin(); i != m_servers.end(); ++i) {
Chris@148: if (i->second.first == server) {
Chris@148: if (i->second.second == 0) {
Chris@843: cerr << "ERROR: FFTDataServer::releaseInstance("
Chris@843: << server << "): instance not allocated" << endl;
Chris@148: } else if (--i->second.second == 0) {
Chris@537: /*!!!
Chris@148: if (server->m_lastUsedCache == -1) { // never used
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::releaseInstance: instance "
Chris@216: << server << " has never been used, erasing"
chris@828: << std::endl;
Chris@216: #endif
Chris@148: delete server;
Chris@148: m_servers.erase(i);
Chris@148: } else {
Chris@537: */
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::releaseInstance: instance "
Chris@216: << server << " no longer in use, marking for possible collection"
chris@828: << std::endl;
Chris@216: #endif
Chris@216: bool found = false;
Chris@216: for (ServerQueue::iterator j = m_releasedServers.begin();
Chris@216: j != m_releasedServers.end(); ++j) {
Chris@216: if (*j == server) {
Chris@843: cerr << "ERROR: FFTDataServer::releaseInstance("
Chris@216: << server << "): server is already in "
Chris@843: << "released servers list" << endl;
Chris@216: found = true;
Chris@216: }
Chris@216: }
Chris@216: if (!found) m_releasedServers.push_back(server);
Chris@148: server->suspend();
Chris@148: purgeLimbo();
Chris@537: //!!! }
Chris@216: } else {
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::releaseInstance: instance "
Chris@216: << server << " now has refcount " << i->second.second
chris@828: << std::endl;
Chris@216: #endif
Chris@148: }
Chris@148: return;
Chris@148: }
Chris@148: }
Chris@148:
Chris@843: cerr << "ERROR: FFTDataServer::releaseInstance(" << server << "): "
Chris@843: << "instance not found" << endl;
Chris@148: }
Chris@148:
Chris@148: void
Chris@148: FFTDataServer::purgeLimbo(int maxSize)
Chris@148: {
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::purgeLimbo(" << maxSize << "): "
chris@828: << m_releasedServers.size() << " candidates" << std::endl;
Chris@216: #endif
Chris@216:
Chris@259: while (int(m_releasedServers.size()) > maxSize) {
Chris@148:
Chris@215: FFTDataServer *server = *m_releasedServers.begin();
Chris@148:
Chris@215: bool found = false;
Chris@215:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::purgeLimbo: considering candidate "
chris@828: << server << std::endl;
Chris@216: #endif
Chris@216:
Chris@215: for (ServerMap::iterator i = m_servers.begin(); i != m_servers.end(); ++i) {
Chris@215:
Chris@215: if (i->second.first == server) {
Chris@215: found = true;
Chris@215: if (i->second.second > 0) {
Chris@843: cerr << "ERROR: FFTDataServer::purgeLimbo: Server "
Chris@215: << server << " is in released queue, but still has non-zero refcount "
Chris@843: << i->second.second << endl;
Chris@215: // ... so don't delete it
Chris@215: break;
Chris@215: }
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::purgeLimbo: looks OK, erasing it"
chris@828: << std::endl;
Chris@216: #endif
Chris@216:
Chris@148: m_servers.erase(i);
Chris@215: delete server;
Chris@215: break;
Chris@148: }
Chris@148: }
Chris@215:
Chris@215: if (!found) {
Chris@843: cerr << "ERROR: FFTDataServer::purgeLimbo: Server "
Chris@215: << server << " is in released queue, but not in server map!"
Chris@843: << endl;
Chris@215: delete server;
Chris@215: }
Chris@215:
Chris@215: m_releasedServers.pop_front();
Chris@215: }
Chris@216:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::purgeLimbo(" << maxSize << "): "
chris@828: << m_releasedServers.size() << " remain" << std::endl;
Chris@216: #endif
Chris@216:
Chris@215: }
Chris@215:
Chris@215: void
Chris@215: FFTDataServer::modelAboutToBeDeleted(Model *model)
Chris@215: {
Chris@244: MutexLocker locker(&m_serverMapMutex,
Chris@408: "FFTDataServer::modelAboutToBeDeleted::m_serverMapMutex");
Chris@215:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::modelAboutToBeDeleted(" << model << ")"
chris@828: << std::endl;
Chris@216: #endif
Chris@216:
Chris@215: for (ServerMap::iterator i = m_servers.begin(); i != m_servers.end(); ++i) {
Chris@215:
Chris@215: FFTDataServer *server = i->second.first;
Chris@215:
Chris@215: if (server->getModel() == model) {
Chris@216:
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::modelAboutToBeDeleted: server is "
chris@828: << server << std::endl;
Chris@216: #endif
Chris@216:
Chris@215: if (i->second.second > 0) {
Chris@843: cerr << "WARNING: FFTDataServer::modelAboutToBeDeleted: Model " << model << " (\"" << model->objectName() << "\") is about to be deleted, but is still being referred to by FFT server " << server << " with non-zero refcount " << i->second.second << endl;
Chris@497: server->suspendWrites();
Chris@362: return;
Chris@215: }
Chris@215: for (ServerQueue::iterator j = m_releasedServers.begin();
Chris@215: j != m_releasedServers.end(); ++j) {
Chris@215: if (*j == server) {
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::modelAboutToBeDeleted: erasing from released servers" << std::endl;
Chris@216: #endif
Chris@215: m_releasedServers.erase(j);
Chris@215: break;
Chris@215: }
Chris@215: }
Chris@216: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::modelAboutToBeDeleted: erasing server" << std::endl;
Chris@216: #endif
Chris@215: m_servers.erase(i);
Chris@215: delete server;
Chris@215: return;
Chris@215: }
Chris@148: }
Chris@148: }
Chris@148:
Chris@148: FFTDataServer::FFTDataServer(QString fileBaseName,
Chris@148: const DenseTimeValueModel *model,
Chris@148: int channel,
Chris@148: WindowType windowType,
Chris@929: int windowSize,
Chris@929: int windowIncrement,
Chris@929: int fftSize,
Chris@148: bool polar,
Chris@334: StorageAdviser::Criteria criteria,
Chris@1045: sv_frame_t fillFromFrame) :
Chris@148: m_fileBaseName(fileBaseName),
Chris@148: m_model(model),
Chris@148: m_channel(channel),
Chris@148: m_windower(windowType, windowSize),
Chris@148: m_windowSize(windowSize),
Chris@148: m_windowIncrement(windowIncrement),
Chris@148: m_fftSize(fftSize),
Chris@148: m_polar(polar),
Chris@183: m_width(0),
Chris@183: m_height(0),
Chris@183: m_cacheWidth(0),
Chris@359: m_cacheWidthPower(0),
Chris@359: m_cacheWidthMask(0),
Chris@359: m_criteria(criteria),
Chris@148: m_fftInput(0),
Chris@148: m_exiting(false),
Chris@153: m_suspended(true), //!!! or false?
Chris@148: m_fillThread(0)
Chris@148: {
Chris@193: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "])::FFTDataServer" << endl;
Chris@193: #endif
Chris@193:
Chris@272: //!!! end is not correct until model finished reading -- what to do???
Chris@272:
Chris@1038: sv_frame_t start = m_model->getStartFrame();
Chris@1038: sv_frame_t end = m_model->getEndFrame();
Chris@148:
Chris@1038: m_width = int((end - start) / m_windowIncrement) + 1;
Chris@203: m_height = m_fftSize / 2 + 1; // DC == 0, Nyquist == fftsize/2
Chris@148:
Chris@216: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << "): dimensions are "
Chris@843: << m_width << "x" << m_height << endl;
Chris@216: #endif
Chris@216:
Chris@929: int maxCacheSize = 20 * 1024 * 1024;
Chris@1038: int columnSize = int(m_height * sizeof(fftsample) * 2 + sizeof(fftsample));
Chris@1084: if (m_width < ((maxCacheSize * 2) / columnSize)) m_cacheWidth = m_width;
Chris@148: else m_cacheWidth = maxCacheSize / columnSize;
Chris@148:
Chris@359: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << "): cache width nominal "
Chris@359: << m_cacheWidth << ", actual ";
Chris@359: #endif
Chris@359:
Chris@148: int bits = 0;
Chris@359: while (m_cacheWidth > 1) { m_cacheWidth >>= 1; ++bits; }
Chris@359: m_cacheWidthPower = bits + 1;
Chris@148: m_cacheWidth = 2;
Chris@148: while (bits) { m_cacheWidth <<= 1; --bits; }
Chris@359: m_cacheWidthMask = m_cacheWidth - 1;
Chris@172:
Chris@359: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << m_cacheWidth << " (power " << m_cacheWidthPower << ", mask "
Chris@843: << m_cacheWidthMask << ")" << endl;
Chris@359: #endif
Chris@359:
Chris@359: if (m_criteria == StorageAdviser::NoCriteria) {
Chris@172:
Chris@334: // assume "spectrogram" criteria for polar ffts, and "feature
Chris@334: // extraction" criteria for rectangular ones.
Chris@334:
Chris@334: if (m_polar) {
Chris@359: m_criteria = StorageAdviser::Criteria
Chris@334: (StorageAdviser::SpeedCritical |
Chris@334: StorageAdviser::LongRetentionLikely);
Chris@334: } else {
Chris@359: m_criteria = StorageAdviser::Criteria
Chris@334: (StorageAdviser::PrecisionCritical);
Chris@334: }
Chris@172: }
Chris@172:
Chris@929: for (int i = 0; i <= m_width / m_cacheWidth; ++i) {
Chris@148: m_caches.push_back(0);
Chris@148: }
Chris@148:
Chris@148: m_fftInput = (fftsample *)
Chris@226: fftf_malloc(fftSize * sizeof(fftsample));
Chris@148:
Chris@226: m_fftOutput = (fftf_complex *)
Chris@226: fftf_malloc((fftSize/2 + 1) * sizeof(fftf_complex));
Chris@148:
Chris@148: m_workbuffer = (float *)
Chris@226: fftf_malloc((fftSize+2) * sizeof(float));
Chris@148:
Chris@226: m_fftPlan = fftf_plan_dft_r2c_1d(m_fftSize,
Chris@334: m_fftInput,
Chris@334: m_fftOutput,
Chris@334: FFTW_MEASURE);
Chris@148:
Chris@148: if (!m_fftPlan) {
Chris@843: cerr << "ERROR: fftf_plan_dft_r2c_1d(" << m_windowSize << ") failed!" << endl;
Chris@148: throw(0);
Chris@148: }
Chris@148:
Chris@1045: m_fillThread = new FillThread(*this, fillFromFrame);
Chris@148: }
Chris@148:
Chris@148: FFTDataServer::~FFTDataServer()
Chris@148: {
Chris@148: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "])::~FFTDataServer()" << endl;
Chris@148: #endif
Chris@148:
Chris@155: m_suspended = false;
Chris@148: m_exiting = true;
Chris@148: m_condition.wakeAll();
Chris@148: if (m_fillThread) {
Chris@148: m_fillThread->wait();
Chris@148: delete m_fillThread;
Chris@148: }
Chris@148:
Chris@549: // MutexLocker locker(&m_writeMutex,
Chris@549: // "FFTDataServer::~FFTDataServer::m_writeMutex");
Chris@549:
Chris@549: QMutexLocker mlocker(&m_fftBuffersLock);
Chris@549: QWriteLocker wlocker(&m_cacheVectorLock);
Chris@148:
Chris@148: for (CacheVector::iterator i = m_caches.begin(); i != m_caches.end(); ++i) {
Chris@205: if (*i) {
Chris@205: delete *i;
Chris@205: }
Chris@148: }
Chris@148:
Chris@148: deleteProcessingData();
Chris@148: }
Chris@148:
Chris@148: void
Chris@148: FFTDataServer::deleteProcessingData()
Chris@148: {
Chris@193: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): deleteProcessingData" << endl;
Chris@193: #endif
Chris@148: if (m_fftInput) {
Chris@226: fftf_destroy_plan(m_fftPlan);
Chris@226: fftf_free(m_fftInput);
Chris@226: fftf_free(m_fftOutput);
Chris@226: fftf_free(m_workbuffer);
Chris@148: }
Chris@148: m_fftInput = 0;
Chris@148: }
Chris@148:
Chris@148: void
Chris@148: FFTDataServer::suspend()
Chris@148: {
Chris@148: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): suspend" << endl;
Chris@148: #endif
Chris@183: Profiler profiler("FFTDataServer::suspend", false);
Chris@183:
Chris@549: QMutexLocker locker(&m_fftBuffersLock);
Chris@148: m_suspended = true;
Chris@148: }
Chris@148:
Chris@148: void
Chris@155: FFTDataServer::suspendWrites()
Chris@155: {
Chris@155: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): suspendWrites" << endl;
Chris@155: #endif
Chris@183: Profiler profiler("FFTDataServer::suspendWrites", false);
Chris@183:
Chris@155: m_suspended = true;
Chris@155: }
Chris@155:
Chris@155: void
Chris@148: FFTDataServer::resume()
Chris@148: {
Chris@154: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): resume" << endl;
Chris@154: #endif
Chris@183: Profiler profiler("FFTDataServer::resume", false);
Chris@183:
Chris@148: m_suspended = false;
Chris@157: if (m_fillThread) {
Chris@157: if (m_fillThread->isFinished()) {
Chris@157: delete m_fillThread;
Chris@157: m_fillThread = 0;
Chris@157: deleteProcessingData();
Chris@411: } else if (!m_fillThread->isRunning()) {
Chris@411: m_fillThread->start();
Chris@157: } else {
Chris@157: m_condition.wakeAll();
Chris@157: }
Chris@157: }
Chris@148: }
Chris@148:
Chris@359: void
Chris@929: FFTDataServer::getStorageAdvice(int w, int h,
Chris@359: bool &memoryCache, bool &compactCache)
Chris@359: {
Chris@1038: if (w < 0 || h < 0) throw std::domain_error("width & height must be non-negative");
Chris@1038: size_t cells = size_t(w) * h;
Chris@1038: size_t minimumSize = (cells / 1024) * sizeof(uint16_t); // kb
Chris@1038: size_t maximumSize = (cells / 1024) * sizeof(float); // kb
Chris@359:
Chris@359: // We don't have a compact rectangular representation, and compact
Chris@359: // of course is never precision-critical
Chris@359:
Chris@359: bool canCompact = true;
Chris@359: if ((m_criteria & StorageAdviser::PrecisionCritical) || !m_polar) {
Chris@359: canCompact = false;
Chris@359: minimumSize = maximumSize; // don't use compact
Chris@359: }
Chris@359:
Chris@359: StorageAdviser::Recommendation recommendation;
Chris@359:
Chris@359: try {
Chris@359:
Chris@359: recommendation =
Chris@359: StorageAdviser::recommend(m_criteria, minimumSize, maximumSize);
Chris@359:
Chris@359: } catch (InsufficientDiscSpace s) {
Chris@359:
Chris@359: // Delete any unused servers we may have been leaving around
Chris@359: // in case we wanted them again
Chris@359:
Chris@359: purgeLimbo(0);
Chris@359:
Chris@359: // This time we don't catch InsufficientDiscSpace -- we
Chris@359: // haven't allocated anything yet and can safely let the
Chris@359: // exception out to indicate to the caller that we can't
Chris@359: // handle it.
Chris@359:
Chris@359: recommendation =
Chris@359: StorageAdviser::recommend(m_criteria, minimumSize, maximumSize);
Chris@359: }
Chris@359:
Chris@843: // cerr << "Recommendation was: " << recommendation << endl;
Chris@359:
Chris@359: memoryCache = false;
Chris@359:
Chris@359: if ((recommendation & StorageAdviser::UseMemory) ||
Chris@359: (recommendation & StorageAdviser::PreferMemory)) {
Chris@538: memoryCache = true;
Chris@359: }
Chris@359:
Chris@359: compactCache = canCompact &&
Chris@359: (recommendation & StorageAdviser::ConserveSpace);
Chris@359:
Chris@374: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer: memory cache = " << memoryCache << ", compact cache = " << compactCache << endl;
Chris@359:
Chris@843: cerr << "Width " << w << " of " << m_width << ", height " << h << ", size " << w * h << endl;
Chris@359: #endif
Chris@359: }
Chris@359:
Chris@537: bool
Chris@537: FFTDataServer::makeCache(int c)
Chris@148: {
Chris@548: // Creating the cache could take a significant amount of time. We
Chris@548: // don't want to block readers on m_cacheVectorLock while this is
Chris@548: // happening, but we do want to block any further calls to
Chris@548: // makeCache. So we use this lock solely to serialise this
Chris@548: // particular function -- it isn't used anywhere else.
Chris@183:
Chris@548: QMutexLocker locker(&m_cacheCreationMutex);
Chris@548:
Chris@548: m_cacheVectorLock.lockForRead();
Chris@537: if (m_caches[c]) {
Chris@537: // someone else must have created the cache between our
Chris@548: // testing for it and taking the mutex
Chris@548: m_cacheVectorLock.unlock();
Chris@537: return true;
Chris@148: }
Chris@548: m_cacheVectorLock.unlock();
Chris@548:
Chris@548: // Now m_cacheCreationMutex is held, but m_cacheVectorLock is not
Chris@548: // -- readers can proceed, but callers to this function will block
Chris@148:
Chris@537: CacheBlock *cb = new CacheBlock;
Chris@148:
Chris@148: QString name = QString("%1-%2").arg(m_fileBaseName).arg(c);
Chris@148:
Chris@929: int width = m_cacheWidth;
Chris@213: if (c * m_cacheWidth + width > m_width) {
Chris@213: width = m_width - c * m_cacheWidth;
Chris@213: }
Chris@213:
Chris@359: bool memoryCache = false;
Chris@359: bool compactCache = false;
Chris@359:
Chris@359: getStorageAdvice(width, m_height, memoryCache, compactCache);
Chris@359:
Chris@537: bool success = false;
Chris@264:
Chris@537: if (memoryCache) {
Chris@172:
Chris@537: try {
Chris@172:
Chris@537: cb->memoryCache = new FFTMemoryCache
Chris@537: (compactCache ? FFTCache::Compact :
Chris@537: m_polar ? FFTCache::Polar :
Chris@537: FFTCache::Rectangular,
Chris@537: width, m_height);
Chris@172:
Chris@537: success = true;
Chris@200:
Chris@537: } catch (std::bad_alloc) {
Chris@200:
Chris@537: delete cb->memoryCache;
Chris@537: cb->memoryCache = 0;
Chris@213:
Chris@843: cerr << "WARNING: Memory allocation failed when creating"
Chris@537: << " FFT memory cache no. " << c << " of " << width
Chris@213: << "x" << m_height << " (of total width " << m_width
Chris@843: << "): falling back to disc cache" << endl;
Chris@213:
Chris@537: memoryCache = false;
Chris@213: }
Chris@172: }
Chris@148:
Chris@537: if (!memoryCache) {
Chris@537:
Chris@537: try {
Chris@537:
Chris@537: cb->fileCacheWriter = new FFTFileCacheWriter
Chris@537: (name,
Chris@537: compactCache ? FFTCache::Compact :
Chris@537: m_polar ? FFTCache::Polar :
Chris@537: FFTCache::Rectangular,
Chris@537: width, m_height);
Chris@537:
Chris@537: success = true;
Chris@537:
Chris@678: } catch (std::exception &e) {
Chris@537:
Chris@537: delete cb->fileCacheWriter;
Chris@537: cb->fileCacheWriter = 0;
Chris@537:
Chris@843: cerr << "ERROR: Failed to construct disc cache for FFT data: "
Chris@843: << e.what() << endl;
Chris@678:
Chris@678: throw;
Chris@537: }
Chris@537: }
Chris@537:
Chris@548: m_cacheVectorLock.lockForWrite();
Chris@548:
Chris@537: m_caches[c] = cb;
Chris@537:
Chris@548: m_cacheVectorLock.unlock();
Chris@548:
Chris@537: return success;
Chris@148: }
Chris@537:
Chris@537: bool
Chris@537: FFTDataServer::makeCacheReader(int c)
Chris@537: {
Chris@537: // preconditions: m_caches[c] exists and contains a file writer;
Chris@537: // m_cacheVectorLock is not locked by this thread
Chris@537: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::makeCacheReader(" << c << ")" << std::endl;
Chris@537: #endif
Chris@148:
Chris@537: QThread *me = QThread::currentThread();
Chris@537: QWriteLocker locker(&m_cacheVectorLock);
Chris@537: CacheBlock *cb(m_caches.at(c));
Chris@537: if (!cb || !cb->fileCacheWriter) return false;
Chris@537:
Chris@537: try {
Chris@537:
Chris@537: cb->fileCacheReader[me] = new FFTFileCacheReader(cb->fileCacheWriter);
Chris@537:
Chris@678: } catch (std::exception &e) {
Chris@537:
Chris@537: delete cb->fileCacheReader[me];
Chris@537: cb->fileCacheReader.erase(me);
Chris@537:
Chris@843: cerr << "ERROR: Failed to construct disc cache reader for FFT data: "
Chris@843: << e.what() << endl;
Chris@537: return false;
Chris@537: }
Chris@537:
Chris@537: // erase a reader that looks like it may no longer going to be
Chris@537: // used by this thread for a while (leaving alone the current
Chris@537: // and previous cache readers)
Chris@537: int deleteCandidate = c - 2;
Chris@537: if (deleteCandidate < 0) deleteCandidate = c + 2;
Chris@897: if (deleteCandidate >= (int)m_caches.size()) {
Chris@537: return true;
Chris@537: }
Chris@537:
Chris@537: cb = m_caches.at(deleteCandidate);
Chris@537: if (cb && cb->fileCacheReader.find(me) != cb->fileCacheReader.end()) {
Chris@537: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::makeCacheReader: Deleting probably unpopular reader " << deleteCandidate << " for this thread (as I create reader " << c << ")" << std::endl;
Chris@537: #endif
Chris@537: delete cb->fileCacheReader[me];
Chris@537: cb->fileCacheReader.erase(me);
Chris@537: }
Chris@537:
Chris@537: return true;
Chris@537: }
Chris@537:
Chris@148: float
Chris@929: FFTDataServer::getMagnitudeAt(int x, int y)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::getMagnitudeAt", false);
Chris@183:
Chris@217: if (x >= m_width || y >= m_height) return 0;
Chris@217:
Chris@678: float val = 0;
Chris@148:
Chris@678: try {
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return 0;
Chris@678:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getMagnitudeAt: filling");
Chris@280: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::getMagnitudeAt: calling fillColumn("
chris@828: << x << ")" << std::endl;
Chris@280: #endif
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@678: val = cache->getMagnitudeAt(col, y);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@148: }
Chris@678:
Chris@678: return val;
Chris@148: }
Chris@148:
Chris@408: bool
Chris@929: FFTDataServer::getMagnitudesAt(int x, float *values, int minbin, int count, int step)
Chris@408: {
Chris@408: Profiler profiler("FFTDataServer::getMagnitudesAt", false);
Chris@408:
Chris@408: if (x >= m_width) return false;
Chris@408:
Chris@408: if (minbin >= m_height) minbin = m_height - 1;
Chris@408: if (count == 0) count = (m_height - minbin) / step;
Chris@408: else if (minbin + count * step > m_height) {
Chris@408: count = (m_height - minbin) / step;
Chris@408: }
Chris@408:
Chris@678: try {
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return false;
Chris@408:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getMagnitudesAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@678: cache->getMagnitudesAt(col, values, minbin, count, step);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@678: return false;
Chris@408: }
Chris@408:
Chris@408: return true;
Chris@408: }
Chris@408:
Chris@148: float
Chris@929: FFTDataServer::getNormalizedMagnitudeAt(int x, int y)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::getNormalizedMagnitudeAt", false);
Chris@183:
Chris@217: if (x >= m_width || y >= m_height) return 0;
Chris@217:
Chris@678: float val = 0;
Chris@148:
Chris@678: try {
Chris@678:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return 0;
Chris@678:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getNormalizedMagnitudeAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678: val = cache->getNormalizedMagnitudeAt(col, y);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@148: }
Chris@678:
Chris@678: return val;
Chris@148: }
Chris@148:
Chris@408: bool
Chris@929: FFTDataServer::getNormalizedMagnitudesAt(int x, float *values, int minbin, int count, int step)
Chris@408: {
Chris@408: Profiler profiler("FFTDataServer::getNormalizedMagnitudesAt", false);
Chris@408:
Chris@408: if (x >= m_width) return false;
Chris@408:
Chris@408: if (minbin >= m_height) minbin = m_height - 1;
Chris@408: if (count == 0) count = (m_height - minbin) / step;
Chris@408: else if (minbin + count * step > m_height) {
Chris@408: count = (m_height - minbin) / step;
Chris@408: }
Chris@408:
Chris@678: try {
Chris@408:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return false;
Chris@408:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getNormalizedMagnitudesAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@929: for (int i = 0; i < count; ++i) {
Chris@678: values[i] = cache->getNormalizedMagnitudeAt(col, i * step + minbin);
Chris@678: }
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@678: return false;
Chris@408: }
Chris@408:
Chris@408: return true;
Chris@408: }
Chris@408:
Chris@148: float
Chris@929: FFTDataServer::getMaximumMagnitudeAt(int x)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::getMaximumMagnitudeAt", false);
Chris@183:
Chris@217: if (x >= m_width) return 0;
Chris@217:
Chris@678: float val = 0;
Chris@148:
Chris@678: try {
Chris@678:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return 0;
Chris@678:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getMaximumMagnitudeAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678: val = cache->getMaximumMagnitudeAt(col);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@148: }
Chris@678:
Chris@678: return val;
Chris@148: }
Chris@148:
Chris@148: float
Chris@929: FFTDataServer::getPhaseAt(int x, int y)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::getPhaseAt", false);
Chris@183:
Chris@217: if (x >= m_width || y >= m_height) return 0;
Chris@217:
Chris@678: float val = 0;
Chris@148:
Chris@678: try {
Chris@678:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return 0;
Chris@678:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getPhaseAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678: val = cache->getPhaseAt(col, y);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@148: }
Chris@678:
Chris@678: return val;
Chris@148: }
Chris@148:
Chris@408: bool
Chris@929: FFTDataServer::getPhasesAt(int x, float *values, int minbin, int count, int step)
Chris@408: {
Chris@408: Profiler profiler("FFTDataServer::getPhasesAt", false);
Chris@408:
Chris@408: if (x >= m_width) return false;
Chris@408:
Chris@408: if (minbin >= m_height) minbin = m_height - 1;
Chris@408: if (count == 0) count = (m_height - minbin) / step;
Chris@408: else if (minbin + count * step > m_height) {
Chris@408: count = (m_height - minbin) / step;
Chris@408: }
Chris@408:
Chris@678: try {
Chris@408:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return false;
Chris@408:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getPhasesAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@929: for (int i = 0; i < count; ++i) {
Chris@678: values[i] = cache->getPhaseAt(col, i * step + minbin);
Chris@678: }
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@678: return false;
Chris@408: }
Chris@408:
Chris@408: return true;
Chris@408: }
Chris@408:
Chris@148: void
Chris@929: FFTDataServer::getValuesAt(int x, int y, float &real, float &imaginary)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::getValuesAt", false);
Chris@183:
Chris@216: if (x >= m_width || y >= m_height) {
Chris@216: real = 0;
Chris@216: imaginary = 0;
Chris@216: return;
Chris@216: }
Chris@216:
Chris@678: try {
Chris@216:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678:
Chris@678: if (!cache) {
Chris@678: real = 0;
Chris@678: imaginary = 0;
Chris@678: return;
Chris@678: }
Chris@678:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") {
Chris@1081: real = 0;
Chris@1081: imaginary = 0;
Chris@1081: return;
Chris@1081: }
Chris@678: Profiler profiler("FFTDataServer::getValuesAt: filling");
Chris@678: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::getValuesAt(" << x << ", " << y << "): filling" << std::endl;
Chris@678: #endif
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@678: cache->getValuesAt(col, y, real, imaginary);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@216: }
Chris@148: }
Chris@148:
Chris@148: bool
Chris@929: FFTDataServer::getValuesAt(int x, float *reals, float *imaginaries, int minbin, int count, int step)
Chris@556: {
Chris@556: Profiler profiler("FFTDataServer::getValuesAt", false);
Chris@556:
Chris@556: if (x >= m_width) return false;
Chris@556:
Chris@556: if (minbin >= m_height) minbin = m_height - 1;
Chris@556: if (count == 0) count = (m_height - minbin) / step;
Chris@556: else if (minbin + count * step > m_height) {
Chris@556: count = (m_height - minbin) / step;
Chris@556: }
Chris@556:
Chris@678: try {
Chris@556:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return false;
Chris@556:
Chris@678: if (!cache->haveSetColumnAt(col)) {
Chris@1081: if (getError() != "") return false;
Chris@678: Profiler profiler("FFTDataServer::getValuesAt: filling");
Chris@678: fillColumn(x);
Chris@678: }
Chris@678:
Chris@929: for (int i = 0; i < count; ++i) {
Chris@678: cache->getValuesAt(col, i * step + minbin, reals[i], imaginaries[i]);
Chris@678: }
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@678: return false;
Chris@556: }
Chris@556:
Chris@556: return true;
Chris@556: }
Chris@556:
Chris@556: bool
Chris@929: FFTDataServer::isColumnReady(int x)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::isColumnReady", false);
Chris@183:
Chris@217: if (x >= m_width) return true;
Chris@217:
Chris@148: if (!haveCache(x)) {
Chris@537: /*!!!
Chris@148: if (m_lastUsedCache == -1) {
Chris@183: if (m_suspended) {
chris@828: std::cerr << "FFTDataServer::isColumnReady(" << x << "): no cache, calling resume" << std::endl;
Chris@183: resume();
Chris@183: }
Chris@148: m_fillThread->start();
Chris@148: }
Chris@537: */
Chris@148: return false;
Chris@148: }
Chris@148:
Chris@678: try {
Chris@148:
Chris@929: int col;
Chris@678: FFTCacheReader *cache = getCacheReader(x, col);
Chris@678: if (!cache) return true;
Chris@678:
Chris@678: return cache->haveSetColumnAt(col);
Chris@678:
Chris@678: } catch (std::exception &e) {
Chris@678: m_error = e.what();
Chris@678: return false;
Chris@678: }
Chris@148: }
Chris@148:
Chris@148: void
Chris@929: FFTDataServer::fillColumn(int x)
Chris@148: {
Chris@183: Profiler profiler("FFTDataServer::fillColumn", false);
Chris@183:
Chris@272: if (!m_model->isReady()) {
Chris@843: cerr << "WARNING: FFTDataServer::fillColumn("
Chris@843: << x << "): model not yet ready" << endl;
Chris@272: return;
Chris@272: }
Chris@549: /*
Chris@217: if (!m_fftInput) {
Chris@843: cerr << "WARNING: FFTDataServer::fillColumn(" << x << "): "
Chris@217: << "input has already been completed and discarded?"
Chris@843: << endl;
Chris@217: return;
Chris@217: }
Chris@549: */
Chris@217: if (x >= m_width) {
Chris@843: cerr << "WARNING: FFTDataServer::fillColumn(" << x << "): "
Chris@217: << "x > width (" << x << " > " << m_width << ")"
Chris@843: << endl;
Chris@217: return;
Chris@217: }
Chris@217:
Chris@929: int col;
Chris@148: #ifdef DEBUG_FFT_SERVER_FILL
Chris@843: cout << "FFTDataServer::fillColumn(" << x << ")" << endl;
Chris@148: #endif
Chris@537: FFTCacheWriter *cache = getCacheWriter(x, col);
Chris@200: if (!cache) return;
Chris@148:
Chris@408: int winsize = m_windowSize;
Chris@408: int fftsize = m_fftSize;
Chris@408: int hs = fftsize/2;
Chris@408:
Chris@1038: sv_frame_t pfx = 0;
Chris@408: int off = (fftsize - winsize) / 2;
Chris@148:
Chris@1038: sv_frame_t startFrame = m_windowIncrement * sv_frame_t(x);
Chris@1038: sv_frame_t endFrame = startFrame + m_windowSize;
Chris@148:
Chris@1088: // FFT windows are centred at the respective audio sample frame,
Chris@1088: // so the first one is centred at 0
Chris@1088: startFrame -= winsize / 2;
Chris@1088: endFrame -= winsize / 2;
Chris@148:
Chris@549: #ifdef DEBUG_FFT_SERVER_FILL
chris@828: std::cerr << "FFTDataServer::fillColumn: requesting frames "
Chris@549: << startFrame + pfx << " -> " << endFrame << " ( = "
Chris@549: << endFrame - (startFrame + pfx) << ") at index "
Chris@549: << off + pfx << " in buffer of size " << m_fftSize
Chris@549: << " with window size " << m_windowSize
chris@828: << " from channel " << m_channel << std::endl;
Chris@549: #endif
Chris@549:
Chris@549: QMutexLocker locker(&m_fftBuffersLock);
Chris@549:
Chris@550: // We may have been called from a function that wanted to obtain a
Chris@550: // column using an FFTCacheReader. Before calling us, it checked
Chris@550: // whether the column was available already, and the reader
Chris@550: // reported that it wasn't. Now we test again, with the mutex
Chris@550: // held, to avoid a race condition in case another thread has
Chris@550: // called fillColumn at the same time.
Chris@550: if (cache->haveSetColumnAt(x & m_cacheWidthMask)) {
Chris@550: return;
Chris@549: }
Chris@549:
Chris@549: if (!m_fftInput) {
Chris@843: cerr << "WARNING: FFTDataServer::fillColumn(" << x << "): "
Chris@549: << "input has already been completed and discarded?"
Chris@843: << endl;
Chris@549: return;
Chris@549: }
Chris@549:
Chris@408: for (int i = 0; i < off; ++i) {
Chris@408: m_fftInput[i] = 0.0;
Chris@408: }
Chris@148:
Chris@408: for (int i = 0; i < off; ++i) {
Chris@408: m_fftInput[fftsize - i - 1] = 0.0;
Chris@148: }
Chris@148:
Chris@148: if (startFrame < 0) {
Chris@408: pfx = -startFrame;
Chris@408: for (int i = 0; i < pfx; ++i) {
Chris@148: m_fftInput[off + i] = 0.0;
Chris@148: }
Chris@148: }
Chris@148:
Chris@1038: sv_frame_t count = 0;
Chris@300: if (endFrame > startFrame + pfx) count = endFrame - (startFrame + pfx);
Chris@300:
Chris@1038: sv_frame_t got = m_model->getData(m_channel, startFrame + pfx,
Chris@1038: count, m_fftInput + off + pfx);
Chris@148:
Chris@408: while (got + pfx < winsize) {
Chris@148: m_fftInput[off + got + pfx] = 0.0;
Chris@148: ++got;
Chris@148: }
Chris@148:
Chris@148: if (m_channel == -1) {
Chris@148: int channels = m_model->getChannelCount();
Chris@148: if (channels > 1) {
Chris@408: for (int i = 0; i < winsize; ++i) {
Chris@1038: m_fftInput[off + i] /= float(channels);
Chris@148: }
Chris@148: }
Chris@148: }
Chris@148:
Chris@148: m_windower.cut(m_fftInput + off);
Chris@148:
Chris@408: for (int i = 0; i < hs; ++i) {
Chris@148: fftsample temp = m_fftInput[i];
Chris@408: m_fftInput[i] = m_fftInput[i + hs];
Chris@408: m_fftInput[i + hs] = temp;
Chris@148: }
Chris@148:
Chris@226: fftf_execute(m_fftPlan);
Chris@148:
Chris@408: float factor = 0.f;
Chris@148:
Chris@408: if (cache->getStorageType() == FFTCache::Compact ||
Chris@408: cache->getStorageType() == FFTCache::Polar) {
Chris@408:
Chris@408: for (int i = 0; i <= hs; ++i) {
Chris@408: fftsample real = m_fftOutput[i][0];
Chris@408: fftsample imag = m_fftOutput[i][1];
Chris@408: float mag = sqrtf(real * real + imag * imag);
Chris@408: m_workbuffer[i] = mag;
Chris@408: m_workbuffer[i + hs + 1] = atan2f(imag, real);
Chris@408: if (mag > factor) factor = mag;
Chris@408: }
Chris@408:
Chris@408: } else {
Chris@408:
Chris@408: for (int i = 0; i <= hs; ++i) {
Chris@408: m_workbuffer[i] = m_fftOutput[i][0];
Chris@408: m_workbuffer[i + hs + 1] = m_fftOutput[i][1];
Chris@408: }
Chris@148: }
Chris@148:
Chris@408: Profiler subprof("FFTDataServer::fillColumn: set to cache");
Chris@408:
Chris@549: if (cache->getStorageType() == FFTCache::Compact ||
Chris@549: cache->getStorageType() == FFTCache::Polar) {
Chris@549:
Chris@549: cache->setColumnAt(col,
Chris@549: m_workbuffer,
Chris@549: m_workbuffer + hs + 1,
Chris@549: factor);
Chris@408:
Chris@549: } else {
Chris@408:
Chris@549: cache->setColumnAt(col,
Chris@549: m_workbuffer,
Chris@549: m_workbuffer + hs + 1);
Chris@408: }
Chris@154:
Chris@183: if (m_suspended) {
chris@828: // std::cerr << "FFTDataServer::fillColumn(" << x << "): calling resume" << std::endl;
Chris@183: // resume();
Chris@183: }
Chris@148: }
Chris@148:
Chris@537: void
Chris@537: FFTDataServer::fillComplete()
Chris@537: {
Chris@537: for (int i = 0; i < int(m_caches.size()); ++i) {
Chris@678: if (!m_caches[i]) continue;
Chris@537: if (m_caches[i]->memoryCache) {
Chris@537: m_caches[i]->memoryCache->allColumnsWritten();
Chris@537: }
Chris@537: if (m_caches[i]->fileCacheWriter) {
Chris@537: m_caches[i]->fileCacheWriter->allColumnsWritten();
Chris@537: }
Chris@537: }
Chris@537: }
Chris@537:
Chris@678: QString
Chris@678: FFTDataServer::getError() const
Chris@678: {
Chris@1081: QString err;
Chris@1081: if (m_error != "") {
Chris@1081: err = m_error;
Chris@1084: // cerr << "FFTDataServer::getError: err (server " << this << ") = " << err << endl;
Chris@1081: } else {
Chris@1081: MutexLocker locker(&m_fftBuffersLock, "FFTDataServer::getError");
Chris@1081: if (m_fillThread) {
Chris@1081: err = m_fillThread->getError();
Chris@1084: // cerr << "FFTDataServer::getError: err (server " << this << ", from thread " << m_fillThread
Chris@1084: // << ") = " << err << endl;
Chris@1081: }
Chris@1081: }
Chris@1081: return err;
Chris@678: }
Chris@678:
Chris@929: int
Chris@148: FFTDataServer::getFillCompletion() const
Chris@148: {
Chris@148: if (m_fillThread) return m_fillThread->getCompletion();
Chris@148: else return 100;
Chris@148: }
Chris@148:
Chris@1038: sv_frame_t
Chris@148: FFTDataServer::getFillExtent() const
Chris@148: {
Chris@148: if (m_fillThread) return m_fillThread->getExtent();
Chris@148: else return m_model->getEndFrame();
Chris@148: }
Chris@148:
Chris@148: QString
Chris@148: FFTDataServer::generateFileBasename() const
Chris@148: {
Chris@148: return generateFileBasename(m_model, m_channel, m_windower.getType(),
Chris@148: m_windowSize, m_windowIncrement, m_fftSize,
Chris@148: m_polar);
Chris@148: }
Chris@148:
Chris@148: QString
Chris@148: FFTDataServer::generateFileBasename(const DenseTimeValueModel *model,
Chris@148: int channel,
Chris@148: WindowType windowType,
Chris@929: int windowSize,
Chris@929: int windowIncrement,
Chris@929: int fftSize,
Chris@148: bool polar)
Chris@148: {
Chris@986: return QString("%1-%2-%3-%4-%5-%6%7")
Chris@986: .arg(XmlExportable::getObjectExportId(model))
Chris@986: .arg(channel + 1)
Chris@986: .arg((int)windowType)
Chris@986: .arg(windowSize)
Chris@986: .arg(windowIncrement)
Chris@986: .arg(fftSize)
Chris@986: .arg(polar ? "-p" : "-r");
Chris@148: }
Chris@148:
Chris@148: void
Chris@148: FFTDataServer::FillThread::run()
Chris@148: {
Chris@411: #ifdef DEBUG_FFT_SERVER_FILL
chris@828: std::cerr << "FFTDataServer::FillThread::run()" << std::endl;
Chris@411: #endif
Chris@411:
Chris@148: m_extent = 0;
Chris@148: m_completion = 0;
Chris@148:
Chris@272: while (!m_server.m_model->isReady() && !m_server.m_exiting) {
Chris@411: #ifdef DEBUG_FFT_SERVER_FILL
chris@828: std::cerr << "FFTDataServer::FillThread::run(): waiting for model " << m_server.m_model << " to be ready" << std::endl;
Chris@411: #endif
Chris@272: sleep(1);
Chris@272: }
Chris@272: if (m_server.m_exiting) return;
Chris@272:
Chris@1038: sv_frame_t start = m_server.m_model->getStartFrame();
Chris@1038: sv_frame_t end = m_server.m_model->getEndFrame();
Chris@1038: sv_frame_t remainingEnd = end;
Chris@148:
Chris@148: int counter = 0;
Chris@246: int updateAt = 1;
Chris@1038: int maxUpdateAt = int(end / m_server.m_windowIncrement) / 20;
Chris@246: if (maxUpdateAt < 100) maxUpdateAt = 100;
Chris@148:
Chris@148: if (m_fillFrom > start) {
Chris@148:
Chris@1038: for (sv_frame_t f = m_fillFrom; f < end; f += m_server.m_windowIncrement) {
Chris@148:
Chris@678: try {
Chris@678: m_server.fillColumn(int((f - start) / m_server.m_windowIncrement));
Chris@678: } catch (std::exception &e) {
Chris@1081: MutexLocker locker(&m_server.m_fftBuffersLock,
Chris@1081: "FFTDataServer::run::m_fftBuffersLock [err]");
Chris@1081: m_threadError = e.what();
Chris@1081: std::cerr << "FFTDataServer::FillThread::run: exception: " << m_threadError << " (thread = " << this << " from server " << &m_server << ")" << std::endl;
Chris@678: m_server.fillComplete();
Chris@678: m_completion = 100;
Chris@678: m_extent = end;
Chris@678: return;
Chris@678: }
Chris@148:
Chris@148: if (m_server.m_exiting) return;
Chris@148:
Chris@148: while (m_server.m_suspended) {
Chris@148: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): suspended, waiting..." << endl;
Chris@148: #endif
Chris@549: MutexLocker locker(&m_server.m_fftBuffersLock,
Chris@549: "FFTDataServer::run::m_fftBuffersLock [1]");
Chris@549: if (m_server.m_suspended && !m_server.m_exiting) {
Chris@549: m_server.m_condition.wait(&m_server.m_fftBuffersLock, 10000);
Chris@244: }
Chris@159: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): waited" << endl;
Chris@159: #endif
Chris@148: if (m_server.m_exiting) return;
Chris@148: }
Chris@148:
Chris@148: if (++counter == updateAt) {
Chris@148: m_extent = f;
Chris@929: m_completion = int(100 * fabsf(float(f - m_fillFrom) /
Chris@148: float(end - start)));
Chris@148: counter = 0;
Chris@246: if (updateAt < maxUpdateAt) {
Chris@246: updateAt *= 2;
Chris@246: if (updateAt > maxUpdateAt) updateAt = maxUpdateAt;
Chris@246: }
Chris@148: }
Chris@148: }
Chris@148:
Chris@148: remainingEnd = m_fillFrom;
Chris@148: if (remainingEnd > start) --remainingEnd;
Chris@148: else remainingEnd = start;
Chris@148: }
Chris@148:
Chris@929: int baseCompletion = m_completion;
Chris@148:
Chris@1038: for (sv_frame_t f = start; f < remainingEnd; f += m_server.m_windowIncrement) {
Chris@148:
Chris@678: try {
Chris@678: m_server.fillColumn(int((f - start) / m_server.m_windowIncrement));
Chris@678: } catch (std::exception &e) {
Chris@1081: MutexLocker locker(&m_server.m_fftBuffersLock,
Chris@1081: "FFTDataServer::run::m_fftBuffersLock [err]");
Chris@1081: m_threadError = e.what();
Chris@1081: std::cerr << "FFTDataServer::FillThread::run: exception: " << m_threadError << " (thread = " << this << " from server " << &m_server << ")" << std::endl;
Chris@678: m_server.fillComplete();
Chris@678: m_completion = 100;
Chris@678: m_extent = end;
Chris@678: return;
Chris@678: }
Chris@148:
Chris@148: if (m_server.m_exiting) return;
Chris@148:
Chris@148: while (m_server.m_suspended) {
Chris@148: #ifdef DEBUG_FFT_SERVER
Chris@843: cerr << "FFTDataServer(" << this << " [" << (void *)QThread::currentThreadId() << "]): suspended, waiting..." << endl;
Chris@148: #endif
Chris@244: {
Chris@549: MutexLocker locker(&m_server.m_fftBuffersLock,
Chris@549: "FFTDataServer::run::m_fftBuffersLock [2]");
Chris@549: if (m_server.m_suspended && !m_server.m_exiting) {
Chris@549: m_server.m_condition.wait(&m_server.m_fftBuffersLock, 10000);
Chris@549: }
Chris@244: }
Chris@148: if (m_server.m_exiting) return;
Chris@148: }
Chris@148:
Chris@148: if (++counter == updateAt) {
Chris@148: m_extent = f;
Chris@148: m_completion = baseCompletion +
Chris@929: int(100 * fabsf(float(f - start) /
Chris@148: float(end - start)));
Chris@148: counter = 0;
Chris@246: if (updateAt < maxUpdateAt) {
Chris@246: updateAt *= 2;
Chris@246: if (updateAt > maxUpdateAt) updateAt = maxUpdateAt;
Chris@246: }
Chris@148: }
Chris@148: }
Chris@148:
Chris@537: m_server.fillComplete();
Chris@148: m_completion = 100;
Chris@148: m_extent = end;
Chris@537:
Chris@537: #ifdef DEBUG_FFT_SERVER
chris@828: std::cerr << "FFTDataServer::FillThread::run exiting" << std::endl;
Chris@537: #endif
Chris@148: }
Chris@148: