Mercurial > hg > silvet
diff constant-q-cpp/cq/CQSpectrogram.h @ 366:5d0a2ebb4d17
Bring dependent libraries in to repo
author | Chris Cannam |
---|---|
date | Fri, 24 Jun 2016 14:47:45 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/constant-q-cpp/cq/CQSpectrogram.h Fri Jun 24 14:47:45 2016 +0100 @@ -0,0 +1,122 @@ +/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ +/* + Constant-Q library + Copyright (c) 2013-2014 Queen Mary, University of London + + Permission is hereby granted, free of charge, to any person + obtaining a copy of this software and associated documentation + files (the "Software"), to deal in the Software without + restriction, including without limitation the rights to use, copy, + modify, merge, publish, distribute, sublicense, and/or sell copies + of the Software, and to permit persons to whom the Software is + furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be + included in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY + CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF + CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION + WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + + Except as contained in this notice, the names of the Centre for + Digital Music; Queen Mary, University of London; and Chris Cannam + shall not be used in advertising or otherwise to promote the sale, + use or other dealings in this Software without prior written + authorization. +*/ + +#ifndef CQSPECTROGRAM_H +#define CQSPECTROGRAM_H + +#include "ConstantQ.h" + +/** + * Calculate a dense constant-Q magnitude spectrogram from time-domain + * input. The input of each \ref process call is a single frame of + * time-domain samples; the output is a series of fixed-height + * columns. See \ref process for details. + * + * If you need the full complex-valued constant-Q output, you must use + * the \ref ConstantQ class instead. + */ +class CQSpectrogram : public CQBase +{ +public: + enum Interpolation { + /// leave empty cells as zero + InterpolateZeros, + /// replace empty cells with a repeat of the previous column + InterpolateHold, + /// perform linear interpolation between consecutive time cells + InterpolateLinear, + }; + + /** + * Construct a Constant-Q magnitude spectrogram object using the + * given transform parameters. + */ + CQSpectrogram(CQParameters params, Interpolation interpolation); + virtual ~CQSpectrogram(); + + // CQBase methods, see CQBase.h for documentation + virtual bool isValid() const { return m_cq.isValid(); } + virtual double getSampleRate() const { return m_cq.getSampleRate(); } + virtual int getBinsPerOctave() const { return m_cq.getBinsPerOctave(); } + virtual int getOctaves() const { return m_cq.getOctaves(); } + virtual int getTotalBins() const { return m_cq.getTotalBins(); } + virtual int getColumnHop() const { return m_cq.getColumnHop(); } + virtual int getLatency() const { return m_cq.getLatency(); } + virtual double getMaxFrequency() const { return m_cq.getMaxFrequency(); } + virtual double getMinFrequency() const { return m_cq.getMinFrequency(); } + virtual double getBinFrequency(double bin) const { return m_cq.getBinFrequency(bin); } + + /** + * Given a series of time-domain samples, return a series of + * constant-Q magnitude columns. Any samples left over (that did + * not fit into a constant-Q processing block) are saved for the + * next call to process or getRemainingBlocks. + * + * The input is assumed to be a single frame of time-domain sample + * values, such that consecutive calls to \ref process receive + * contiguous frames from the source signal. Each frame may be of + * any length in samples. + * + * Each output column contains a series of constant-Q bin value + * magnitudes, ordered from highest to lowest frequency. + * + * The columns are all of the same height, but they might not all + * be populated, depending on the interpolation mode: in + * InterpolateZeros mode, the lower octaves (which are spaced more + * widely in the raw constant-Q than the highest octave) will + * contain zeros for the undefined values, but in the other + * interpolation modes every cell will be filled. + * + * To obtain raw, complex constant-Q bin values, use the ConstantQ + * class. + */ + RealBlock process(const RealSequence &); + + /** + * Return the remaining constant-Q magnitude columns following the + * end of processing. Any buffered input is padded so as to ensure + * that all input provided to process() will have been returned. + */ + RealBlock getRemainingOutput(); + +private: + ConstantQ m_cq; + Interpolation m_interpolation; + + RealBlock m_buffer; + RealBlock postProcess(const ComplexBlock &, bool insist); + RealBlock fetchHold(bool insist); + RealBlock fetchLinear(bool insist); + RealBlock linearInterpolated(const RealBlock &, int, int); + RealColumn m_prevColumn; +}; + +#endif