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: #include "FFTModel.h"
lbajardsilogic@0: #include "DenseTimeValueModel.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include "base/Profiler.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <cassert>
lbajardsilogic@0: 
lbajardsilogic@0: FFTModel::FFTModel(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) :
lbajardsilogic@0:     //!!! ZoomConstraint!
lbajardsilogic@0:     m_server(0),
lbajardsilogic@0:     m_xshift(0),
lbajardsilogic@0:     m_yshift(0)
lbajardsilogic@0: {
lbajardsilogic@0:     m_server = FFTDataServer::getFuzzyInstance(model,
lbajardsilogic@0:                                                channel,
lbajardsilogic@0:                                                windowType,
lbajardsilogic@0:                                                windowSize,
lbajardsilogic@0:                                                windowIncrement,
lbajardsilogic@0:                                                fftSize,
lbajardsilogic@0:                                                polar,
lbajardsilogic@0:                                                fillFromColumn);
lbajardsilogic@0: 
lbajardsilogic@0:     if (!m_server) return; // caller should check isOK()
lbajardsilogic@0: 
lbajardsilogic@0:     size_t xratio = windowIncrement / m_server->getWindowIncrement();
lbajardsilogic@0:     size_t yratio = m_server->getFFTSize() / fftSize;
lbajardsilogic@0: 
lbajardsilogic@0:     while (xratio > 1) {
lbajardsilogic@0:         if (xratio & 0x1) {
lbajardsilogic@0:             std::cerr << "ERROR: FFTModel: Window increment ratio "
lbajardsilogic@0:                       << windowIncrement << " / "
lbajardsilogic@0:                       << m_server->getWindowIncrement()
lbajardsilogic@0:                       << " must be a power of two" << std::endl;
lbajardsilogic@0:             assert(!(xratio & 0x1));
lbajardsilogic@0:         }
lbajardsilogic@0:         ++m_xshift;
lbajardsilogic@0:         xratio >>= 1;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     while (yratio > 1) {
lbajardsilogic@0:         if (yratio & 0x1) {
lbajardsilogic@0:             std::cerr << "ERROR: FFTModel: FFT size ratio "
lbajardsilogic@0:                       << m_server->getFFTSize() << " / " << fftSize
lbajardsilogic@0:                       << " must be a power of two" << std::endl;
lbajardsilogic@0:             assert(!(yratio & 0x1));
lbajardsilogic@0:         }
lbajardsilogic@0:         ++m_yshift;
lbajardsilogic@0:         yratio >>= 1;
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: FFTModel::~FFTModel()
lbajardsilogic@0: {
lbajardsilogic@0:     if (m_server) FFTDataServer::releaseInstance(m_server);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: size_t
lbajardsilogic@0: FFTModel::getSampleRate() const
lbajardsilogic@0: {
lbajardsilogic@0:     return isOK() ? m_server->getModel()->getSampleRate() : 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: FFTModel::getColumn(size_t x, Column &result) const
lbajardsilogic@0: {
lbajardsilogic@0:     Profiler profiler("FFTModel::getColumn", false);
lbajardsilogic@0: 
lbajardsilogic@0:     result.clear();
lbajardsilogic@0:     size_t height(getHeight());
lbajardsilogic@0:     for (size_t y = 0; y < height; ++y) {
lbajardsilogic@0:         result.push_back(const_cast<FFTModel *>(this)->getMagnitudeAt(x, y));
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: QString
lbajardsilogic@0: FFTModel::getBinName(size_t n) const
lbajardsilogic@0: {
lbajardsilogic@0:     size_t sr = getSampleRate();
lbajardsilogic@0:     if (!sr) return "";
lbajardsilogic@0:     QString name = tr("%1 Hz").arg((n * sr) / ((getHeight()-1) * 2));
lbajardsilogic@0:     return name;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: Model *
lbajardsilogic@0: FFTModel::clone() const
lbajardsilogic@0: {
lbajardsilogic@0:     return new FFTModel(*this);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: FFTModel::FFTModel(const FFTModel &model) :
lbajardsilogic@0:     DenseThreeDimensionalModel(),
lbajardsilogic@0:     m_server(model.m_server),
lbajardsilogic@0:     m_xshift(model.m_xshift),
lbajardsilogic@0:     m_yshift(model.m_yshift)
lbajardsilogic@0: {
lbajardsilogic@0:     FFTDataServer::claimInstance(m_server);
lbajardsilogic@0: }
lbajardsilogic@0: