13
0
livetrax/libs/qm-dsp/dsp/onsets/DetectionFunction.h

97 lines
2.9 KiB
C
Raw Normal View History

/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
/*
QM DSP Library
Centre for Digital Music, Queen Mary, University of London.
This file 2005-2006 Christian Landone.
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of the
License, or (at your option) any later version. See the file
COPYING included with this distribution for more information.
*/
#ifndef DETECTIONFUNCTION_H
#define DETECTIONFUNCTION_H
#include "maths/MathUtilities.h"
#include "maths/MathAliases.h"
#include "dsp/phasevocoder/PhaseVocoder.h"
#include "base/Window.h"
#define DF_HFC (1)
#define DF_SPECDIFF (2)
#define DF_PHASEDEV (3)
#define DF_COMPLEXSD (4)
#define DF_BROADBAND (5)
struct DFConfig{
unsigned int stepSize; // DF step in samples
2016-10-05 18:16:44 -04:00
unsigned int frameLength; // DF analysis window - usually 2*step. Must be even!
int DFType; // type of detection function ( see defines )
double dbRise; // only used for broadband df (and required for it)
bool adaptiveWhitening; // perform adaptive whitening
double whiteningRelaxCoeff; // if < 0, a sensible default will be used
double whiteningFloor; // if < 0, a sensible default will be used
};
2016-10-05 18:16:44 -04:00
class DetectionFunction
{
public:
double* getSpectrumMagnitude();
DetectionFunction( DFConfig Config );
virtual ~DetectionFunction();
2016-10-05 18:16:44 -04:00
/**
* Process a single time-domain frame of audio, provided as
* frameLength samples.
*/
double processTimeDomain(const double* samples);
/**
* Process a single frequency-domain frame, provided as
* frameLength/2+1 real and imaginary component values.
*/
double processFrequencyDomain(const double* reals, const double* imags);
private:
void whiten();
double runDF();
double HFC( unsigned int length, double* src);
double specDiff( unsigned int length, double* src);
double phaseDev(unsigned int length, double *srcPhase);
double complexSD(unsigned int length, double *srcMagnitude, double *srcPhase);
double broadband(unsigned int length, double *srcMagnitude);
2016-10-05 18:16:44 -04:00
private:
void initialise( DFConfig Config );
void deInitialise();
int m_DFType;
unsigned int m_dataLength;
unsigned int m_halfLength;
unsigned int m_stepSize;
double m_dbRise;
bool m_whiten;
double m_whitenRelaxCoeff;
double m_whitenFloor;
double* m_magHistory;
double* m_phaseHistory;
double* m_phaseHistoryOld;
double* m_magPeaks;
2016-10-05 18:16:44 -04:00
double* m_windowed; // Array for windowed analysis frame
double* m_magnitude; // Magnitude of analysis frame ( frequency domain )
double* m_thetaAngle;// Phase of analysis frame ( frequency domain )
2016-10-05 18:16:44 -04:00
double* m_unwrapped; // Unwrapped phase of analysis frame
Window<double> *m_window;
PhaseVocoder* m_phaseVoc; // Phase Vocoder
};
2016-10-05 18:16:44 -04:00
#endif