c@0: 
c@0: // This is a skeleton file for use in creating your own plugin
c@0: // libraries.  Replace MyPlugin and myPlugin throughout with the name
c@0: // of your first plugin class, and fill in the gaps as appropriate.
c@0: 
c@0: 
c@0: #include "Tempogram.h"
c@4: 
c@0: using Vamp::FFT;
c@7: using Vamp::RealTime;
c@0: using namespace std;
c@0: 
c@0: Tempogram::Tempogram(float inputSampleRate) :
c@0:     Plugin(inputSampleRate),
c@0:     m_blockSize(0),
c@1:     m_stepSize(0),
c@0:     compressionConstant(1000), //make param
c@3:     specMax(0),
c@3:     minDB(0),
c@7:     tN(128), //make param
c@7:     thopSize(64), //make param
c@0:     fftInput(NULL),
c@0:     fftOutputReal(NULL),
c@0:     fftOutputImag(NULL),
c@4:     numberOfBlocks(0)
c@0: 
c@0:     // Also be sure to set your plugin parameters (presumably stored
c@0:     // in member variables) to their default values here -- the host
c@0:     // will not do that for you
c@0: {
c@0: }
c@0: 
c@0: Tempogram::~Tempogram()
c@0: {
c@0:     //delete stuff
c@7:     cleanup();
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getIdentifier() const
c@0: {
c@0:     return "tempogram";
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getName() const
c@0: {
c@0:     return "Tempogram";
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getDescription() const
c@0: {
c@0:     // Return something helpful here!
c@0:     return "Cyclic Tempogram as described by Peter Grosche and Meinard Muller";
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getMaker() const
c@0: {
c@0:     //Your name here
c@0:     return "Carl Bussey";
c@0: }
c@0: 
c@0: int
c@0: Tempogram::getPluginVersion() const
c@0: {
c@0:     // Increment this each time you release a version that behaves
c@0:     // differently from the previous one
c@0:     return 1;
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getCopyright() const
c@0: {
c@0:     // This function is not ideally named.  It does not necessarily
c@0:     // need to say who made the plugin -- getMaker does that -- but it
c@0:     // should indicate the terms under which it is distributed.  For
c@0:     // example, "Copyright (year). All Rights Reserved", or "GPL"
c@0:     return "";
c@0: }
c@0: 
c@0: Tempogram::InputDomain
c@0: Tempogram::getInputDomain() const
c@0: {
c@0:     return FrequencyDomain;
c@0: }
c@0: 
c@0: size_t
c@0: Tempogram::getPreferredBlockSize() const
c@0: {
c@0:     return 0; // 0 means "I can handle any block size"
c@0: }
c@0: 
c@0: size_t 
c@0: Tempogram::getPreferredStepSize() const
c@0: {
c@0:     return 0; // 0 means "anything sensible"; in practice this
c@0:               // means the same as the block size for TimeDomain
c@0:               // plugins, or half of it for FrequencyDomain plugins
c@0: }
c@0: 
c@0: size_t
c@0: Tempogram::getMinChannelCount() const
c@0: {
c@0:     return 1;
c@0: }
c@0: 
c@0: size_t
c@0: Tempogram::getMaxChannelCount() const
c@0: {
c@0:     return 1;
c@0: }
c@0: 
c@0: Tempogram::ParameterList
c@0: Tempogram::getParameterDescriptors() const
c@0: {
c@0:     ParameterList list;
c@0: 
c@0:     // If the plugin has no adjustable parameters, return an empty
c@0:     // list here (and there's no need to provide implementations of
c@0:     // getParameter and setParameter in that case either).
c@0: 
c@0:     // Note that it is your responsibility to make sure the parameters
c@0:     // start off having their default values (e.g. in the constructor
c@0:     // above).  The host needs to know the default value so it can do
c@0:     // things like provide a "reset to default" function, but it will
c@0:     // not explicitly set your parameters to their defaults for you if
c@0:     // they have not changed in the mean time.
c@0: 
c@0:     ParameterDescriptor C;
c@0:     C.identifier = "C";
c@0:     C.name = "C";
c@0:     C.description = "Spectrogram compression constant, C";
c@0:     C.unit = "";
c@0:     C.minValue = 2;
c@0:     C.maxValue = 10000;
c@0:     C.defaultValue = 1000;
c@0:     C.isQuantized = false;
c@0:     list.push_back(C);
c@0:     
c@0:     ParameterDescriptor tN;
c@0:     tN.identifier = "tN";
c@0:     tN.name = "Tempogram FFT length";
c@0:     tN.description = "Tempogram FFT length.";
c@0:     tN.unit = "";
c@0:     tN.minValue = 128;
c@0:     tN.maxValue = 4096;
c@7:     tN.defaultValue = 128;
c@0:     tN.isQuantized = true;
c@0:     tN.quantizeStep = 128;
c@0:     list.push_back(tN);
c@0: 
c@0:     return list;
c@0: }
c@0: 
c@0: float
c@0: Tempogram::getParameter(string identifier) const
c@0: {
c@0:     if (identifier == "C") {
c@0:         return compressionConstant; // return the ACTUAL current value of your parameter here!
c@0:     }
c@0:     if (identifier == "tN"){
c@0:         return tN;
c@0:     }
c@0:     
c@0:     return 0;
c@0: }
c@0: 
c@0: void
c@0: Tempogram::setParameter(string identifier, float value) 
c@0: {
c@0:     if (identifier == "C") {
c@1:         compressionConstant = value; // set the actual value of your parameter
c@0:     }
c@0:     if (identifier == "tN") {
c@0:         tN = value;
c@0:     }
c@0: }
c@0: 
c@0: Tempogram::ProgramList
c@0: Tempogram::getPrograms() const
c@0: {
c@0:     ProgramList list;
c@0: 
c@0:     // If you have no programs, return an empty list (or simply don't
c@0:     // implement this function or getCurrentProgram/selectProgram)
c@0: 
c@0:     return list;
c@0: }
c@0: 
c@0: string
c@0: Tempogram::getCurrentProgram() const
c@0: {
c@0:     return ""; // no programs
c@0: }
c@0: 
c@0: void
c@0: Tempogram::selectProgram(string name)
c@0: {
c@0: }
c@0: 
c@0: Tempogram::OutputList
c@0: Tempogram::getOutputDescriptors() const
c@0: {
c@0:     OutputList list;
c@0: 
c@0:     // See OutputDescriptor documentation for the possibilities here.
c@0:     // Every plugin must have at least one output.
c@1:     
c@0:     OutputDescriptor d;
c@7:     float d_sampleRate;
c@7:     
c@1:     d.identifier = "tempogram";
c@7:     d.name = "Tempogram";
c@7:     d.description = "Tempogram";
c@0:     d.unit = "";
c@1:     d.hasFixedBinCount = true;
c@7:     d.binCount = tN/2 + 1;
c@0:     d.hasKnownExtents = false;
c@0:     d.isQuantized = false;
c@1:     d.sampleType = OutputDescriptor::FixedSampleRate;
c@7:     d_sampleRate = m_inputSampleRate/(m_stepSize * thopSize);
c@1:     d.sampleRate = d_sampleRate > 0.0 && !isnan(d_sampleRate) ? d_sampleRate : 0.0;
c@0:     d.hasDuration = false;
c@0:     list.push_back(d);
c@7:     
c@1:     d.identifier = "nc";
c@1:     d.name = "Novelty Curve";
c@1:     d.description = "Novelty Curve";
c@1:     d.unit = "";
c@1:     d.hasFixedBinCount = true;
c@1:     d.binCount = 1;
c@1:     d.hasKnownExtents = false;
c@1:     d.isQuantized = false;
c@1:     d.sampleType = OutputDescriptor::FixedSampleRate;
c@1:     d_sampleRate = m_inputSampleRate/m_stepSize;
c@1:     d.sampleRate = d_sampleRate > 0 && !isnan(d_sampleRate) ? d_sampleRate : 0.0;
c@1:     d.hasDuration = false;
c@1:     list.push_back(d);
c@1:     
c@0:     return list;
c@0: }
c@0: 
c@0: bool
c@0: Tempogram::initialise(size_t channels, size_t stepSize, size_t blockSize)
c@0: {
c@0:     if (channels < getMinChannelCount() ||
c@0: 	channels > getMaxChannelCount()) return false;
c@0: 
c@0:     // Real initialisation work goes here!
c@0:     m_blockSize = blockSize;
c@1:     m_stepSize = stepSize;
c@7:     minDB = pow(10,(float)-74/20);
c@0:     
c@4:     specData = vector< vector<float> >(m_blockSize/2 + 1);
c@4:     
c@0:     return true;
c@0: }
c@0: 
c@7: void Tempogram::cleanup(){
c@7: 
c@7: }
c@7: 
c@0: void
c@0: Tempogram::reset()
c@0: {
c@0:     // Clear buffers, reset stored values, etc
c@7:     cleanupForGRF();
c@7:     ncTimestamps.clear();
c@7:     specData.clear();
c@0: }
c@0: 
c@0: Tempogram::FeatureSet
c@0: Tempogram::process(const float *const *inputBuffers, Vamp::RealTime timestamp)
c@0: {
c@4:     
c@0:     size_t n = m_blockSize/2 + 1;
c@0:     
c@0:     FeatureSet featureSet;
c@0:     Feature feature;
c@0:     
c@0:     const float *in = inputBuffers[0];
c@3: 
c@0:     for (int i = 0; i < n; i++){
c@0:         float magnitude = sqrt(in[2*i] * in[2*i] + in[2*i + 1] * in[2*i + 1]);
c@3:         magnitude = magnitude > minDB ? magnitude : minDB;
c@4:         specData[i].push_back(magnitude);
c@0:     }
c@0:     
c@3:     numberOfBlocks++;
c@0:     ncTimestamps.push_back(timestamp);
c@7: 
c@2:     return featureSet;
c@0: }
c@0: 
c@0: void
c@0: Tempogram::initialiseForGRF(){
c@0:     hannWindowtN = new float[tN];
c@0:     fftInput = new double[tN];
c@0:     fftOutputReal = new double[tN];
c@0:     fftOutputImag = new double[tN];
c@0:     
c@7:     for (int i = 0; i < tN; i ++){
c@7:         hannWindowtN[i] = 0.0;
c@7:         fftInput[i] = 0.0;
c@7:         fftOutputReal[i] = 0.0;
c@7:         fftOutputImag[i] = 0.0;
c@4:     }
c@0: }
c@0: 
c@0: void
c@0: Tempogram::cleanupForGRF(){
c@0:     delete []hannWindowtN;
c@0:     hannWindowtN = NULL;
c@7:     fftInput = fftOutputReal = fftOutputImag = NULL;
c@0: }
c@0: 
c@0: Tempogram::FeatureSet
c@0: Tempogram::getRemainingFeatures()
c@0: {
c@0:     //Make sure this is called at the beginning of the function
c@0:     initialiseForGRF();
c@1:     FeatureSet featureSet;
c@0:     
c@4:     NoveltyCurve nc(m_inputSampleRate, m_blockSize, numberOfBlocks, compressionConstant);
c@7:     noveltyCurve = nc.spectrogramToNoveltyCurve(specData);
c@4:     
c@4:     for (int i = 0; i < numberOfBlocks; i++){
c@7:         Feature feature;
c@7:         feature.values.push_back(noveltyCurve[i]);
c@7:         feature.hasTimestamp = true;
c@7:         feature.timestamp = ncTimestamps[i];
c@7:         featureSet[1].push_back(feature);
c@4:     }
c@4:     
c@0:     WindowFunction::hanning(hannWindowtN, tN);
c@7:     Spectrogram * spectrogramProcessor = new Spectrogram(numberOfBlocks, tN, thopSize);
c@7:     vector< vector<float> > tempogram = spectrogramProcessor->audioToMagnitudeSpectrogram(&noveltyCurve[0], hannWindowtN);
c@0:     
c@7:     cout << "About to delete..." << endl;
c@7:     delete spectrogramProcessor;
c@7:     cout << "Deleted!" << endl;
c@7:     spectrogramProcessor = NULL;
c@0:     
c@7:     int timePointer = thopSize-tN/2;
c@7:     int tempogramLength = tempogram[0].size();
c@7:     
c@7:     for (int block = 0; block < tempogramLength; block++){
c@0:         Feature feature;
c@0:         
c@7:         int timeMS = floor(1000*(m_stepSize*timePointer)/m_inputSampleRate + 0.5);
c@7:         
c@7:         cout << timeMS << endl;
c@7:         
c@7:         for(int k = 0; k < tN/2 + 1; k++){
c@7:             feature.values.push_back(tempogram[k][block]);
c@0:         }
c@7:         feature.hasTimestamp = true;
c@7:         feature.timestamp = RealTime::fromMilliseconds(timeMS);
c@7:         featureSet[0].push_back(feature);
c@4:         
c@7:         timePointer += thopSize;
c@0:     }
c@0:     
c@0:     //Make sure this is called at the end of the function
c@0:     cleanupForGRF();
c@0:     
c@0:     return featureSet;
c@0: }