annotate dsp/signalconditioning/DFProcess.cpp @ 47:38bf09927942

* Build fixes for gcc 4.3.2 * _Maybe_, but probably not, fix crash in tempo tracker... let's see how we get on
author cannam
date Mon, 10 Nov 2008 14:01:55 +0000
parents 2e3f5d2d62c1
children d72fcd34d9a7
rev   line source
cannam@0 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
cannam@0 2
cannam@0 3 /*
cannam@0 4 QM DSP Library
cannam@0 5
cannam@0 6 Centre for Digital Music, Queen Mary, University of London.
cannam@0 7 This file copyright 2005-2006 Christian Landone.
cannam@0 8 All rights reserved.
cannam@0 9 */
cannam@0 10
cannam@0 11 #include "DFProcess.h"
cannam@16 12 #include "maths/MathUtilities.h"
cannam@0 13
cannam@47 14 #include <cstring>
cannam@47 15
cannam@0 16 //////////////////////////////////////////////////////////////////////
cannam@0 17 // Construction/Destruction
cannam@0 18 //////////////////////////////////////////////////////////////////////
cannam@0 19
cannam@0 20 DFProcess::DFProcess( DFProcConfig Config )
cannam@0 21 {
cannam@0 22 filtSrc = NULL;
cannam@0 23 filtDst = NULL;
cannam@0 24 m_filtScratchIn = NULL;
cannam@0 25 m_filtScratchOut = NULL;
cannam@0 26
cannam@0 27 m_FFOrd = 0;
cannam@0 28
cannam@0 29 initialise( Config );
cannam@0 30 }
cannam@0 31
cannam@0 32 DFProcess::~DFProcess()
cannam@0 33 {
cannam@0 34 deInitialise();
cannam@0 35 }
cannam@0 36
cannam@0 37 void DFProcess::initialise( DFProcConfig Config )
cannam@0 38 {
cannam@0 39 m_length = Config.length;
cannam@0 40 m_winPre = Config.winPre;
cannam@0 41 m_winPost = Config.winPost;
cannam@0 42 m_alphaNormParam = Config.AlphaNormParam;
cannam@0 43
cannam@0 44 m_isMedianPositive = Config.isMedianPositive;
cannam@0 45
cannam@0 46 filtSrc = new double[ m_length ];
cannam@0 47 filtDst = new double[ m_length ];
cannam@0 48
cannam@0 49
cannam@0 50 //Low Pass Smoothing Filter Config
cannam@0 51 m_FilterConfigParams.ord = Config.LPOrd;
cannam@0 52 m_FilterConfigParams.ACoeffs = Config.LPACoeffs;
cannam@0 53 m_FilterConfigParams.BCoeffs = Config.LPBCoeffs;
cannam@0 54
cannam@0 55 m_FiltFilt = new FiltFilt( m_FilterConfigParams );
cannam@0 56 }
cannam@0 57
cannam@0 58 void DFProcess::deInitialise()
cannam@0 59 {
cannam@0 60 delete [] filtSrc;
cannam@0 61
cannam@0 62 delete [] filtDst;
cannam@0 63
cannam@0 64 delete [] m_filtScratchIn;
cannam@0 65
cannam@0 66 delete [] m_filtScratchOut;
cannam@0 67
cannam@0 68 delete m_FiltFilt;
cannam@0 69 }
cannam@0 70
cannam@0 71 void DFProcess::process(double *src, double* dst)
cannam@0 72 {
cannam@0 73 removeDCNormalize( src, filtSrc );
cannam@0 74
cannam@0 75 m_FiltFilt->process( filtSrc, filtDst, m_length );
cannam@0 76
cannam@0 77 medianFilter( filtDst, dst );
cannam@0 78 }
cannam@0 79
cannam@0 80
cannam@0 81 void DFProcess::medianFilter(double *src, double *dst)
cannam@0 82 {
cannam@0 83 unsigned int i,k,j,l;
cannam@0 84 unsigned int index = 0;
cannam@0 85
cannam@0 86 double val = 0;
cannam@0 87
cannam@0 88 double* y = new double[ m_winPost + m_winPre + 1];
cannam@0 89 memset( y, 0, sizeof( double ) * ( m_winPost + m_winPre + 1) );
cannam@0 90
cannam@0 91 double* scratch = new double[ m_length ];
cannam@0 92
cannam@0 93 for( i = 0; i < m_winPre; i++)
cannam@0 94 {
cannam@0 95 k = i + m_winPost + 1;
cannam@0 96
cannam@0 97 for( j = 0; j < k; j++)
cannam@0 98 {
cannam@0 99 y[ j ] = src[ j ];
cannam@0 100 }
cannam@0 101 scratch[ index ] = MathUtilities::median( y, k );
cannam@0 102 index++;
cannam@0 103 }
cannam@0 104
cannam@0 105 for( i = 0; i < ( m_length - ( m_winPost + m_winPre ) ); i ++)
cannam@0 106 {
cannam@0 107
cannam@0 108 l = 0;
cannam@0 109 for( j = i; j < ( i + m_winPost + m_winPre + 1); j++)
cannam@0 110 {
cannam@0 111 y[ l ] = src[ j ];
cannam@0 112 l++;
cannam@0 113 }
cannam@0 114
cannam@0 115 scratch[ index++ ] = MathUtilities::median( y, (m_winPost + m_winPre + 1 ));
cannam@0 116 }
cannam@0 117
cannam@0 118 for( i = std::max( m_length - m_winPost, (unsigned)1); i < m_length; i++)
cannam@0 119 {
cannam@0 120 k = std::max( i - m_winPre, (unsigned)1);
cannam@0 121
cannam@0 122 l = 0;
cannam@0 123 for( j = k; j < m_length; j++)
cannam@0 124 {
cannam@0 125 y[ l ] = src[ j ];
cannam@0 126
cannam@0 127 l++;
cannam@0 128 }
cannam@0 129
cannam@0 130 scratch[ index++ ] = MathUtilities::median( y, l);
cannam@0 131 }
cannam@0 132
cannam@0 133
cannam@0 134 for( i = 0; i < m_length; i++ )
cannam@0 135 {
cannam@0 136 val = src[ i ] - scratch[ i ];// - 0.033;
cannam@0 137
cannam@0 138 if( m_isMedianPositive )
cannam@0 139 {
cannam@0 140 if( val > 0 )
cannam@0 141 {
cannam@0 142 dst[ i ] = val;
cannam@0 143 }
cannam@0 144 else
cannam@0 145 {
cannam@0 146 dst[ i ] = 0;
cannam@0 147 }
cannam@0 148 }
cannam@0 149 else
cannam@0 150 {
cannam@0 151 dst[ i ] = val;
cannam@0 152 }
cannam@0 153 }
cannam@0 154
cannam@0 155 delete [] y;
cannam@0 156 delete [] scratch;
cannam@0 157 }
cannam@0 158
cannam@0 159
cannam@0 160 void DFProcess::removeDCNormalize( double *src, double*dst )
cannam@0 161 {
cannam@0 162 double DFmax = 0;
cannam@0 163 double DFMin = 0;
cannam@0 164 double DFAlphaNorm = 0;
cannam@0 165
cannam@0 166 MathUtilities::getFrameMinMax( src, m_length, &DFMin, &DFmax );
cannam@0 167
cannam@0 168 MathUtilities::getAlphaNorm( src, m_length, m_alphaNormParam, &DFAlphaNorm );
cannam@0 169
cannam@0 170 for( unsigned int i = 0; i< m_length; i++)
cannam@0 171 {
cannam@0 172 dst[ i ] = ( src[ i ] - DFMin ) / DFAlphaNorm;
cannam@0 173 }
cannam@0 174 }