Chris@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@2: #define _USE_MATH_DEFINES
Chris@2: 
Chris@0: #include <iostream>
Chris@0: #include <cmath>
Chris@2: #include <cstdio>
Chris@0: 
Chris@2: #include "public.sdk/source/vst2.x/audioeffect.h"
Chris@2: 
Chris@2: #define snprintf _snprintf
Chris@2: #define alloca _alloca
Chris@0: 
Chris@8: #define FFTSIZE 1024
Chris@4: #define WINSIZE 1024
Chris@4: 
Chris@4: #include "median.h"
Chris@0: 
Chris@0: class Devuvuzelator : public AudioEffect
Chris@0: {
Chris@0:     enum {
Chris@6:         FundamentalParam = 0,
Chris@6:         BandwidthParam = 1,
Chris@6:         HarmonicsParam = 2,
Chris@6:         ReductionParam = 3,
Chris@6:         NumParams      = 4
Chris@0:     };
Chris@0: 
Chris@0: public:
Chris@0:     Devuvuzelator(audioMasterCallback cb);
Chris@0:     ~Devuvuzelator();
Chris@0: 
Chris@0:     virtual void getEffectName(char *n) {
Chris@0:         vst_strncpy(n, "Devuvuzelator", kVstMaxEffectNameLen);
Chris@0:     }
Chris@0:     virtual void getProductString(char *n) {
Chris@0:         vst_strncpy(n, "Devuvuzelator", kVstMaxProductStrLen);
Chris@0:     }
Chris@0:     virtual void getVendorString(char *n) {
Chris@0:         vst_strncpy(n, "Queen Mary, University of London", kVstMaxVendorStrLen);
Chris@0:     }
Chris@0: 
Chris@0:     virtual void setParameter(VstInt32 index, float value);
Chris@0:     virtual float getParameter(VstInt32 index);
Chris@0:     virtual void getParameterLabel(VstInt32 index, char* label);
Chris@0:     virtual void getParameterDisplay(VstInt32 index, char* text);
Chris@0:     virtual void getParameterName(VstInt32 index, char* text);
Chris@0:     
Chris@0:     virtual void setSampleRate (float sampleRate) {
Chris@0:         m_sampleRate = sampleRate;
Chris@0:         AudioEffect::setSampleRate(sampleRate);
Chris@0:     }
Chris@0: 
Chris@0:     virtual void processReplacing (float** inputs, float** outputs, VstInt32 sampleFrames) {
Chris@0:         m_input = inputs[0];
Chris@0:         m_output = outputs[0];
Chris@0:         runImpl(sampleFrames);
Chris@0:     }
Chris@0: 
Chris@0:     void reset();
Chris@0:     void window(float *);
Chris@0:     void runImpl(unsigned long);
Chris@0:     void processFrame();
Chris@0:     void processSpectralFrame();
Chris@0: 
Chris@0:     static void fft(unsigned int n, bool inverse,
Chris@0:                     double *ri, double *ii, double *ro, double *io);
Chris@0: 
Chris@2:     float m_sampleRate;
Chris@0:     float *m_input;
Chris@0:     float *m_output;
Chris@0: 
Chris@0:     float m_fundamental;
Chris@0:     float m_bandwidth;
Chris@2:     int m_harmonics;
Chris@0:     float m_reduction;
Chris@0: 
Chris@0:     const int m_fftsize;
Chris@4:     const int m_winsize;
Chris@0:     const int m_increment;
Chris@0:     int m_fill;
Chris@0:     int m_read;
Chris@0:     float *m_buffer;
Chris@0:     float *m_outacc;
Chris@8: 	double *m_frame;
Chris@8: 	double *m_spare;
Chris@0:     double *m_real;
Chris@0:     double *m_imag;
Chris@0:     double *m_window;
Chris@4:     MedianFilter<double> **m_medians;
Chris@0: };
Chris@0: 
Chris@1: // VST params 0->1
Chris@1: 
Chris@0: void
Chris@0: Devuvuzelator::setParameter(VstInt32 index, float value)
Chris@0: {
Chris@1:     switch (index) {
Chris@4:     case 0: m_fundamental = 50 + 720 * value; break;
Chris@4:     case 1: m_bandwidth = 20 + 80 * value; break;
Chris@4:     case 2: m_harmonics = int(value * 6 + 0.5); break;
Chris@4:     case 3: m_reduction = 100 * value; break;
Chris@1:     }
Chris@0: }
Chris@0: 
Chris@0: float
Chris@0: Devuvuzelator::getParameter(VstInt32 index)
Chris@0: {
Chris@1:     switch (index) {
Chris@4:     case 0: return (m_fundamental - 50) / 720;
Chris@4:     case 1: return (m_bandwidth - 20) / 80;
Chris@4:     case 2: return (m_harmonics / 6.f);
Chris@4:     case 3: return m_reduction / 100;
Chris@1:     }
Chris@4:     return 0;
Chris@0: }
Chris@0: 
Chris@0: // NB! The max name length for VST parameter names, labels
Chris@0: // (i.e. units) and display values (i.e. string renderings of current
Chris@0: // value) is a rather amazing 8 bytes
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::getParameterLabel(VstInt32 index, char *label)
Chris@0: {
Chris@0:     const char *units[NumParams] = {
Chris@0:         "Hz",
Chris@0:         "Hz",
Chris@0:         "",
Chris@4:         "%",
Chris@0:     };
Chris@0:     
Chris@0:     vst_strncpy(label, units[index], kVstMaxParamStrLen);
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::getParameterDisplay(VstInt32 index, char *label)
Chris@0: {
Chris@4:     switch (index) {
Chris@4:     case 0: snprintf(label, kVstMaxParamStrLen, "%f", m_fundamental); break;
Chris@4:     case 1: snprintf(label, kVstMaxParamStrLen, "%f", m_bandwidth); break;
Chris@4:     case 2: snprintf(label, kVstMaxParamStrLen, "%d", m_harmonics); break;
Chris@4:     case 3: snprintf(label, kVstMaxParamStrLen, "%f", m_reduction); break;
Chris@4:     }
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::getParameterName(VstInt32 index, char *label)
Chris@0: {
Chris@0:     const char *names[NumParams] = {
Chris@0:         "Pitch",
Chris@0:         "B/W",
Chris@0:         "Partials",
Chris@0:         "Reductn",
Chris@0:     };
Chris@0:     
Chris@0:     vst_strncpy(label, names[index], kVstMaxParamStrLen);
Chris@0: }
Chris@0: 
Chris@0: Devuvuzelator::Devuvuzelator(audioMasterCallback cb) :
Chris@0:     AudioEffect(cb, 0, NumParams),
Chris@0:     m_sampleRate(0),
Chris@0:     m_input(0),
Chris@0:     m_output(0),
Chris@0:     m_fftsize(FFTSIZE),
Chris@7:     m_winsize(WINSIZE),
Chris@5:     m_increment(m_winsize/2),
Chris@0:     m_fill(0),
Chris@0:     m_read(0)
Chris@0: {
Chris@4:     m_buffer = new float[m_winsize];
Chris@4:     m_outacc = new float[m_winsize * 2];
Chris@4:     m_window = new double[m_winsize];
Chris@8: 	m_frame = new double[m_fftsize];
Chris@8: 	m_spare = new double[m_fftsize];
Chris@0:     m_real = new double[m_fftsize];
Chris@0:     m_imag = new double[m_fftsize];
Chris@4:     m_medians = new MedianFilter<double> *[m_fftsize/2+1];
Chris@0: 
Chris@4:     for (int i = 0; i < m_winsize; ++i) {
Chris@4:         m_window[i] = 0.5 - 0.5 * cos(2 * M_PI * i / m_winsize);
Chris@4:     }
Chris@4:     for (int i = 0; i < m_fftsize/2+1; ++i) {
Chris@4:         m_medians[i] = 0;
Chris@0:     }
Chris@0: 
Chris@4:     m_fundamental = 230;
Chris@0:     m_bandwidth = 60;
Chris@0:     m_harmonics = 3;
Chris@4:     m_reduction = 30;
Chris@0:     
Chris@2:     setUniqueID('qmvz');
Chris@0:     setNumInputs(1);
Chris@0:     setNumOutputs(1);
Chris@0:     canProcessReplacing(true);
Chris@0:     canDoubleReplacing(false);
Chris@0: 
Chris@0:     reset();
Chris@0: }
Chris@0: 
Chris@0: Devuvuzelator::~Devuvuzelator()
Chris@0: {
Chris@0:     delete[] m_buffer;
Chris@0:     delete[] m_outacc;
Chris@8: 	delete[] m_frame;
Chris@8: 	delete[] m_spare;
Chris@0:     delete[] m_real;
Chris@0:     delete[] m_imag;
Chris@0:     delete[] m_window;
Chris@4:     for (int i = 0; i < m_fftsize/2+1; ++i) {
Chris@4:         delete m_medians[i];
Chris@4:     }
Chris@4:     delete[] m_medians;
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::reset()
Chris@0: {
Chris@4:     for (int i = 0; i < m_winsize; ++i) {
Chris@0:         m_buffer[i] = 0.f;
Chris@0:     }
Chris@4:     for (int i = 0; i < m_winsize*2; ++i) {
Chris@0:         m_outacc[i] = 0.f;
Chris@0:     }
Chris@0:     m_fill = 0;
Chris@0:     m_read = 0;
Chris@4:     for (int i = 0; i < m_fftsize/2+1; ++i) {
Chris@4:         if (m_medians[i]) m_medians[i]->reset();
Chris@4:     }
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::runImpl(unsigned long sampleCount)
Chris@0: {
Chris@0:     if (!m_input || !m_output) return;
Chris@0: 
Chris@4:     const int sc = sampleCount;
Chris@8: /*
Chris@8: 	static FILE *blah = 0;
Chris@8: 	if (!blah) {
Chris@8: 		blah = fopen("d:\\devuvu-counts.txt", "w");
Chris@8: 	}
Chris@0: 
Chris@8: 	if (m_input == m_output) fprintf(blah, "in-place\n");
Chris@8: */
Chris@8: 	float *output = m_output;
Chris@8: 	if (m_input == m_output) {
Chris@8: 		output = (float *)alloca(sampleCount * sizeof(float));
Chris@8: 	}
Chris@8: /*	
Chris@8: 	float inmean = 0;
Chris@8: 	for (int i = 0; i < sc; ++i) {
Chris@8: 		inmean += m_input[i] * m_input[i];
Chris@8: 		fprintf(blah, "i:%d:%f ", i, m_input[i]);
Chris@8: 	}
Chris@8: 	inmean/=sc;
Chris@8: 	inmean = sqrt(inmean);
Chris@0: 
Chris@8: 	fprintf(blah, "%d\n", (int)sampleCount);
Chris@8: 	fflush(blah);
Chris@8: */
Chris@8: 	int oi = 0;
Chris@8: 	for (int ii = 0; ii < sc; ++ii) {
Chris@0: 
Chris@8: 		output[oi++] = m_outacc[m_read++];
Chris@8: //	m_output[oi++] = inmean * float(ii%100)/100;
Chris@8: //	m_read++;
Chris@4:         if (m_fill == m_winsize) {
Chris@0: 
Chris@0:             processFrame();
Chris@8: /*
Chris@8: 	for (int i = 0; i < m_winsize; ++i) {
Chris@8: 		float v = m_buffer[i];
Chris@8: 		fprintf(blah, "%f ", v);
Chris@8: 		m_outacc[m_winsize + i] = m_buffer[i];//m_input[i];//m_buffer[i] ;//* m_window[i];
Chris@8: 	}
Chris@8: 	fprintf(blah, "\n");
Chris@8: 
Chris@8: //			for (int j = 0; j < m_winsize; ++j) {
Chris@8: //				m_outacc[m_winsize+j] = inmean * float(j%100)/100;
Chris@8: //			}
Chris@8: */
Chris@0: 
Chris@4:             for (int j = m_increment; j < m_winsize; ++j) {
Chris@0:                 m_buffer[j - m_increment] = m_buffer[j];
Chris@0:             }
Chris@0: 
Chris@4:             for (int j = m_increment; j < m_winsize*2; ++j) {
Chris@0:                 m_outacc[j - m_increment] = m_outacc[j];
Chris@0:             }
Chris@0: 
Chris@4:             for (int j = m_winsize*2 - m_increment; j < m_winsize*2; ++j) {
Chris@0:                 m_outacc[j] = 0.f;
Chris@0:             }
Chris@0: 
Chris@8:             m_fill = m_fill - m_increment;
Chris@8:             m_read = m_read - m_increment;
Chris@0:         }
Chris@8: /*
Chris@8: 		fprintf(blah, "%d:%f ", ii, m_input[ii]);
Chris@8: */
Chris@8: 		m_buffer[m_fill] = m_input[ii];
Chris@8: 		++m_fill;
Chris@8:     }
Chris@0: 
Chris@8: 	static int block = 0;
Chris@8: 	for (int i = 0; i < sc; ++i) {
Chris@8: //		m_output[i] = float(block % 100) / 100;
Chris@8: //		m_output[i] = inmean * float(i % 100) / 100;
Chris@8: 	}
Chris@8: 	++block;
Chris@8: 
Chris@8: 	if (m_input == m_output) {
Chris@8: 		for (int i = 0; i < sc; ++i) m_output[i] = output[i];
Chris@8: 	}
Chris@0: }
Chris@0: 
Chris@0: void
Chris@0: Devuvuzelator::processFrame()
Chris@0: {
Chris@8: /*	for (int i = 0; i < m_winsize; ++i) {
Chris@8: 		m_outacc[m_winsize + i] += m_buffer[i] ;//* m_window[i];
Chris@8: 	}
Chris@8: 	return;
Chris@8: */
Chris@8: 
Chris@0:     for (int i = 0; i < m_fftsize; ++i) {
Chris@8:         m_frame[i] = 0.0;
Chris@4:     }
Chris@4: 
Chris@4:     int ix = m_fftsize - m_winsize/2;
Chris@4:     while (ix < 0) ix += m_fftsize;
Chris@4:     for (int i = 0; i < m_winsize; ++i) {
Chris@8:         m_frame[ix++] += m_buffer[i] * m_window[i];
Chris@0:         if (ix == m_fftsize) ix = 0;
Chris@0:     }
Chris@0: 
Chris@8:     fft(m_fftsize, false, m_frame, 0, m_real, m_imag);
Chris@0: 
Chris@8: //    processSpectralFrame();
Chris@0: 
Chris@0:     for (int i = 0; i < m_fftsize/2-1; ++i) {
Chris@3:         m_real[m_fftsize-i-1] =  m_real[i+1];
Chris@3:         m_imag[m_fftsize-i-1] = -m_imag[i+1];
Chris@0:     }
Chris@0: 
Chris@8:     fft(m_fftsize, true, m_real, m_imag, m_frame, m_spare);
Chris@0: 
Chris@4:     ix = m_fftsize - m_winsize/2;
Chris@4:     while (ix < 0) ix += m_fftsize;
Chris@4:     for (int i = 0; i < m_winsize; ++i) {
Chris@8:         m_outacc[m_winsize + i] += m_frame[ix++];
Chris@0:         if (ix == m_fftsize) ix = 0;
Chris@0:     }
Chris@0: }
Chris@0: 
Chris@0: // FFT implementation by Don Cross, public domain.
Chris@0: // This version scales the forward transform.
Chris@0: 
Chris@0: void Devuvuzelator::fft(unsigned int n, bool inverse,
Chris@0:                         double *ri, double *ii, double *ro, double *io)
Chris@0: {
Chris@0:     if (!ri || !ro || !io) return;
Chris@0: 
Chris@0:     unsigned int bits;
Chris@0:     unsigned int i, j, k, m;
Chris@0:     unsigned int blockSize, blockEnd;
Chris@0: 
Chris@0:     double tr, ti;
Chris@0: 
Chris@0:     if (n < 2) return;
Chris@0:     if (n & (n-1)) return;
Chris@0: 
Chris@0:     double angle = 2.0 * M_PI;
Chris@0:     if (inverse) angle = -angle;
Chris@0: 
Chris@0:     for (i = 0; ; ++i) {
Chris@0: 	if (n & (1 << i)) {
Chris@0: 	    bits = i;
Chris@0: 	    break;
Chris@0: 	}
Chris@0:     }
Chris@0: 
Chris@0:     static unsigned int tableSize = 0;
Chris@0:     static int *table = 0;
Chris@0: 
Chris@0:     if (tableSize != n) {
Chris@0: 
Chris@0: 	delete[] table;
Chris@0: 
Chris@0: 	table = new int[n];
Chris@0: 
Chris@0: 	for (i = 0; i < n; ++i) {
Chris@0: 	
Chris@0: 	    m = i;
Chris@0: 
Chris@0: 	    for (j = k = 0; j < bits; ++j) {
Chris@0: 		k = (k << 1) | (m & 1);
Chris@0: 		m >>= 1;
Chris@0: 	    }
Chris@0: 
Chris@0: 	    table[i] = k;
Chris@0: 	}
Chris@0: 
Chris@0: 	tableSize = n;
Chris@0:     }
Chris@0: 
Chris@0:     if (ii) {
Chris@0: 	for (i = 0; i < n; ++i) {
Chris@0: 	    ro[table[i]] = ri[i];
Chris@0: 	    io[table[i]] = ii[i];
Chris@0: 	}
Chris@0:     } else {
Chris@0: 	for (i = 0; i < n; ++i) {
Chris@0: 	    ro[table[i]] = ri[i];
Chris@0: 	    io[table[i]] = 0.0;
Chris@0: 	}
Chris@0:     }
Chris@0: 
Chris@0:     blockEnd = 1;
Chris@0: 
Chris@0:     for (blockSize = 2; blockSize <= n; blockSize <<= 1) {
Chris@0: 
Chris@0: 	double delta = angle / (double)blockSize;
Chris@0: 	double sm2 = -sin(-2 * delta);
Chris@0: 	double sm1 = -sin(-delta);
Chris@0: 	double cm2 = cos(-2 * delta);
Chris@0: 	double cm1 = cos(-delta);
Chris@0: 	double w = 2 * cm1;
Chris@0: 	double ar[3], ai[3];
Chris@0: 
Chris@0: 	for (i = 0; i < n; i += blockSize) {
Chris@0: 
Chris@0: 	    ar[2] = cm2;
Chris@0: 	    ar[1] = cm1;
Chris@0: 
Chris@0: 	    ai[2] = sm2;
Chris@0: 	    ai[1] = sm1;
Chris@0: 
Chris@0: 	    for (j = i, m = 0; m < blockEnd; j++, m++) {
Chris@0: 
Chris@0: 		ar[0] = w * ar[1] - ar[2];
Chris@0: 		ar[2] = ar[1];
Chris@0: 		ar[1] = ar[0];
Chris@0: 
Chris@0: 		ai[0] = w * ai[1] - ai[2];
Chris@0: 		ai[2] = ai[1];
Chris@0: 		ai[1] = ai[0];
Chris@0: 
Chris@0: 		k = j + blockEnd;
Chris@0: 		tr = ar[0] * ro[k] - ai[0] * io[k];
Chris@0: 		ti = ar[0] * io[k] + ai[0] * ro[k];
Chris@0: 
Chris@0: 		ro[k] = ro[j] - tr;
Chris@0: 		io[k] = io[j] - ti;
Chris@0: 
Chris@0: 		ro[j] += tr;
Chris@0: 		io[j] += ti;
Chris@0: 	    }
Chris@0: 	}
Chris@0: 
Chris@0: 	blockEnd = blockSize;
Chris@0:     }
Chris@0: 
Chris@0:     if (!inverse) {
Chris@0: 
Chris@0: 	double denom = (double)n;
Chris@0: 
Chris@0: 	for (i = 0; i < n; i++) {
Chris@0: 	    ro[i] /= denom;
Chris@0: 	    io[i] /= denom;
Chris@0: 	}
Chris@0:     }
Chris@0: }
Chris@0: 
Chris@0: AudioEffect *createEffectInstance(audioMasterCallback audioMaster)
Chris@0: {
Chris@0:     return new Devuvuzelator(audioMaster);
Chris@0: }
Chris@0: 
Chris@1: #include "devuvuzelator.cpp"
Chris@1: