Mercurial > hg > beaglert
diff projects/filter_FIR/render.cpp @ 2:021ac8a1a4f9
_new FIR filter example
author | Victor Zappi <victor.zappi@qmul.ac.uk> |
---|---|
date | Thu, 06 Nov 2014 15:59:16 +0000 |
parents | |
children | 09f03ac40fcc |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/projects/filter_FIR/render.cpp Thu Nov 06 15:59:16 2014 +0000 @@ -0,0 +1,174 @@ +/* + * render.cpp + * + * Created on: Oct 24, 2014 + * Author: Andrew McPherson and Victor Zappi + */ + + +#include "../../include/render.h" +#include "../../include/RTAudio.h" // to schedule lower prio parallel process +#include <rtdk.h> +#include <cmath> +#include <stdio.h> +#include <NE10.h> // neon library +#include "SampleData.h" +#include "FIRfilter.h" + +SampleData gSampleData; // User defined structure to get complex data from main +int gReadPtr; // Position of last read sample from file +int gNumChannels; + + +// filter vars +ne10_fir_instance_f32_t gFIRfilter; +ne10_float32_t *gFIRfilterIn; +ne10_float32_t *gFIRfilterOut; +ne10_uint32_t blockSize; +ne10_float32_t *gFIRfilterState; + +void initialise_filter(); + + +// Task for handling the update of the frequencies using the matrix +AuxiliaryTask gTriggerSamplesTask; + +bool initialise_trigger(); +void trigger_samples(); + +extern int gPeriodSize; // Period size in sensor frames + + +// initialise_render() is called once before the audio rendering starts. +// Use it to perform any initialisation and allocation which is dependent +// on the period size or sample rate. +// +// userData holds an opaque pointer to a data structure that was passed +// in from the call to initAudio(). +// +// Return true on success; returning false halts the program. + +bool initialise_render(int numChannels, int numMatrixFramesPerPeriod, + int numAudioFramesPerPeriod, float matrixSampleRate, + float audioSampleRate, void *userData) +{ + + // Retrieve a parameter passed in from the initAudio() call + gSampleData = *(SampleData *)userData; + + gReadPtr = -1; + gNumChannels = numChannels; + + initialise_filter(); + + // Initialise auxiliary tasks + if(!initialise_trigger()) + return false; + + return true; +} + +// render() is called regularly at the highest priority by the audio engine. +// Input and output are given from the audio hardware and the other +// ADCs and DACs (if available). If only audio is available, numMatrixFrames +// will be 0. + +void render(int numMatrixFrames, int numAudioFrames, float *audioIn, float *audioOut, + uint16_t *matrixIn, uint16_t *matrixOut) +{ + for(int n = 0; n < numAudioFrames; n++) { + float in = 0; + + // If triggered... + if(gReadPtr != -1) + in += gSampleData.samples[gReadPtr++]; // ...read each sample... + + if(gReadPtr >= gSampleData.sampleLen) + gReadPtr = -1; + + gFIRfilterIn[n] = in; + } + + ne10_fir_float_neon(&gFIRfilter, gFIRfilterIn, gFIRfilterOut, blockSize); + + for(int n = 0; n < numAudioFrames; n++) { + for(int channel = 0; channel < gNumChannels; channel++) + audioOut[n * gNumChannels + channel] = gFIRfilterOut[n]; // ...and put it in both left and right channel + } + + + // Request that the lower-priority task run at next opportunity + scheduleAuxiliaryTask(gTriggerSamplesTask); +} + +// Initialise NE10 data structures to define FIR filter + +void initialise_filter() +{ + blockSize = 2*gPeriodSize; + gFIRfilterState = (ne10_float32_t *) NE10_MALLOC ((FILTER_TAP_NUM+blockSize-1) * sizeof (ne10_float32_t)); + gFIRfilterIn = (ne10_float32_t *) NE10_MALLOC (blockSize * sizeof (ne10_float32_t)); + gFIRfilterOut = (ne10_float32_t *) NE10_MALLOC (blockSize * sizeof (ne10_float32_t)); + ne10_fir_init_float(&gFIRfilter, FILTER_TAP_NUM, filterTaps, gFIRfilterState, blockSize); +} + + +// Initialise the auxiliary task +// and print info + +bool initialise_trigger() +{ + if((gTriggerSamplesTask = createAuxiliaryTaskLoop(&trigger_samples, 50, "beaglert-trigger-samples")) == 0) + return false; + + rt_printf("Press 'a' to trigger sample, 's' to stop\n"); + rt_printf("Press 'q' to quit\n"); + + return true; +} + +// This is a lower-priority call to periodically read keyboard input +// and trigger samples. By placing it at a lower priority, +// it has minimal effect on the audio performance but it will take longer to +// complete if the system is under heavy audio load. + +void trigger_samples() +{ + // This is not a real-time task! + // Cos getchar is a system call, not handled by Xenomai. + // This task will be automatically down graded. + + char keyStroke = '.'; + + keyStroke = getchar(); + while(getchar()!='\n'); // to read the first stroke + + switch (keyStroke) + { + case 'a': + gReadPtr = 0; + break; + case 's': + gReadPtr = -1; + break; + case 'q': + gShouldStop = true; + break; + default: + break; + } +} + + + +// cleanup_render() is called once at the end, after the audio has stopped. +// Release any resources that were allocated in initialise_render(). + +void cleanup_render() +{ + delete[] gSampleData.samples; + + NE10_FREE(gFIRfilterState); + NE10_FREE(gFIRfilterIn); + NE10_FREE(gFIRfilterOut); +}