Chris@366: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@366: /*
Chris@366:     Constant-Q library
Chris@366:     Copyright (c) 2013-2015 Queen Mary, University of London
Chris@366: 
Chris@366:     Permission is hereby granted, free of charge, to any person
Chris@366:     obtaining a copy of this software and associated documentation
Chris@366:     files (the "Software"), to deal in the Software without
Chris@366:     restriction, including without limitation the rights to use, copy,
Chris@366:     modify, merge, publish, distribute, sublicense, and/or sell copies
Chris@366:     of the Software, and to permit persons to whom the Software is
Chris@366:     furnished to do so, subject to the following conditions:
Chris@366: 
Chris@366:     The above copyright notice and this permission notice shall be
Chris@366:     included in all copies or substantial portions of the Software.
Chris@366: 
Chris@366:     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
Chris@366:     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
Chris@366:     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
Chris@366:     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
Chris@366:     CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
Chris@366:     CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
Chris@366:     WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Chris@366: 
Chris@366:     Except as contained in this notice, the names of the Centre for
Chris@366:     Digital Music; Queen Mary, University of London; and Chris Cannam
Chris@366:     shall not be used in advertising or otherwise to promote the sale,
Chris@366:     use or other dealings in this Software without prior written
Chris@366:     authorization.
Chris@366: */
Chris@366: 
Chris@366: #include "Chromagram.h"
Chris@366: #include "CQSpectrogram.h"
Chris@366: #include "Pitch.h"
Chris@366: 
Chris@366: #include <cstdio>
Chris@366: 
Chris@366: using namespace std;
Chris@366: 
Chris@366: Chromagram::Chromagram(Parameters params) :
Chris@366:     m_params(params),
Chris@366:     m_cq(0)
Chris@366: {
Chris@366:     int highestOctave = m_params.lowestOctave + m_params.octaveCount - 1;
Chris@366: 
Chris@366:     int midiPitchLimit = (1 + highestOctave) * 12 + 12; // C just beyond top
Chris@366:     double midiPitchLimitFreq = Pitch::getFrequencyForPitch
Chris@366:         (midiPitchLimit, 0, m_params.tuningFrequency);
Chris@366: 
Chris@366:     // Max frequency is frequency of the MIDI pitch just beyond the
Chris@366:     // top octave range (midiPitchLimit) minus one bin, then minus
Chris@366:     // floor(bins per semitone / 2)
Chris@366:     int bps = m_params.binsPerOctave / 12;
Chris@366:     m_maxFrequency = midiPitchLimitFreq /
Chris@366:         pow(2.0, (1.0 + floor(bps/2)) / m_params.binsPerOctave);
Chris@366: 
Chris@366:     // Min frequency is frequency of midiPitchLimit lowered by the
Chris@366:     // appropriate number of octaveCount.
Chris@366:     m_minFrequency = midiPitchLimitFreq /
Chris@366:         pow(2.0, m_params.octaveCount + 1);
Chris@366: 
Chris@366:     CQParameters p
Chris@366:         (params.sampleRate, m_minFrequency, m_maxFrequency, params.binsPerOctave);
Chris@366: 
Chris@366:     p.q = params.q;
Chris@366:     p.atomHopFactor = params.atomHopFactor;
Chris@366:     p.threshold = params.threshold;
Chris@366:     p.window = params.window;
Chris@366:     
Chris@366:     m_cq = new CQSpectrogram(p, CQSpectrogram::InterpolateLinear);
Chris@366: }
Chris@366: 
Chris@366: Chromagram::~Chromagram()
Chris@366: {
Chris@366:     delete m_cq;
Chris@366: }
Chris@366: 
Chris@366: bool
Chris@366: Chromagram::isValid() const
Chris@366: {
Chris@366:     return m_cq->isValid();
Chris@366: }
Chris@366: 
Chris@366: int
Chris@366: Chromagram::getColumnHop() const
Chris@366: {
Chris@366:     return m_cq->getColumnHop();
Chris@366: }
Chris@366: 
Chris@366: int
Chris@366: Chromagram::getLatency() const
Chris@366: {
Chris@366:     return m_cq->getLatency();
Chris@366: }
Chris@366: 
Chris@366: string
Chris@366: Chromagram::getBinName(int bin) const
Chris@366: {
Chris@366:     static const char *names[] = {
Chris@366:         "C", "C#", "D", "D#", "E", "F", "F#", "G", "G#", "A", "A#", "B"
Chris@366:     };
Chris@366: 
Chris@366:     float freq = m_cq->getBinFrequency(m_params.binsPerOctave - bin - 1);
Chris@366:     int note = Pitch::getPitchForFrequency(freq, 0, m_params.tuningFrequency);
Chris@366:     float nearestFreq =
Chris@366:         Pitch::getFrequencyForPitch(note, 0, m_params.tuningFrequency);
Chris@366:     
Chris@366:     char name[40];
Chris@366:     sprintf(name, "%d", bin);
Chris@366:     if (fabs(freq - nearestFreq) < 0.01) {
Chris@366:         return (name + std::string(" ") + names[note % 12]);
Chris@366:     } else {
Chris@366:         return (name);
Chris@366:     }
Chris@366: }
Chris@366: 
Chris@366: CQBase::RealBlock
Chris@366: Chromagram::process(const CQBase::RealSequence &data)
Chris@366: {
Chris@366:     return convert(m_cq->process(data));
Chris@366: }
Chris@366: 
Chris@366: CQBase::RealBlock
Chris@366: Chromagram::getRemainingOutput()
Chris@366: {
Chris@366:     return convert(m_cq->getRemainingOutput());
Chris@366: }
Chris@366: 
Chris@366: CQBase::RealBlock
Chris@366: Chromagram::convert(const CQBase::RealBlock &cqout)
Chris@366: {    
Chris@366:     CQBase::RealBlock chroma;
Chris@366: 
Chris@366:     int width = cqout.size();
Chris@366: 
Chris@366:     for (int i = 0; i < width; ++i) {
Chris@366: 
Chris@366:         CQBase::RealSequence column(m_params.binsPerOctave, 0.);
Chris@366: 
Chris@366:         // fold and invert to put low frequencies at the start
Chris@366: 
Chris@366:         int thisHeight = cqout[i].size();
Chris@366: 
Chris@366: 	for (int j = 0; j < thisHeight; ++j) {
Chris@366: 	    column[m_params.binsPerOctave - (j % m_params.binsPerOctave) - 1]
Chris@366:                 += cqout[i][j];
Chris@366: 	}
Chris@366: 
Chris@366:         chroma.push_back(column);
Chris@366:     }
Chris@366: 
Chris@366:     return chroma;
Chris@366: }
Chris@366: