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);
+}