robert@285
|
1 /*
|
robert@285
|
2 ____ _____ _ _
|
robert@285
|
3 | __ )| ____| | / \
|
robert@285
|
4 | _ \| _| | | / _ \
|
robert@285
|
5 | |_) | |___| |___ / ___ \
|
robert@285
|
6 |____/|_____|_____/_/ \_\.io
|
robert@285
|
7
|
robert@285
|
8 */
|
robert@285
|
9
|
victor@2
|
10 /*
|
victor@2
|
11 * render.cpp
|
victor@2
|
12 *
|
victor@2
|
13 * Created on: Oct 24, 2014
|
victor@2
|
14 * Author: Andrew McPherson and Victor Zappi
|
victor@2
|
15 */
|
victor@2
|
16
|
robert@285
|
17 /**
|
robert@285
|
18 \example 4_filter_FIR
|
robert@285
|
19
|
robert@285
|
20 Finite Impulse Response Filter
|
robert@285
|
21 ------------------------------
|
robert@285
|
22
|
robert@285
|
23 This is an example of a finite impulse response filter implementation.
|
robert@285
|
24
|
robert@285
|
25 */
|
robert@285
|
26
|
victor@2
|
27
|
andrewm@56
|
28 #include <BeagleRT.h>
|
victor@2
|
29 #include <cmath>
|
victor@2
|
30 #include <NE10.h> // neon library
|
victor@2
|
31 #include "SampleData.h"
|
victor@2
|
32 #include "FIRfilter.h"
|
victor@2
|
33
|
victor@2
|
34 SampleData gSampleData; // User defined structure to get complex data from main
|
victor@2
|
35 int gReadPtr; // Position of last read sample from file
|
victor@2
|
36
|
victor@2
|
37 // filter vars
|
victor@2
|
38 ne10_fir_instance_f32_t gFIRfilter;
|
victor@2
|
39 ne10_float32_t *gFIRfilterIn;
|
victor@2
|
40 ne10_float32_t *gFIRfilterOut;
|
victor@2
|
41 ne10_uint32_t blockSize;
|
victor@2
|
42 ne10_float32_t *gFIRfilterState;
|
victor@2
|
43
|
andrewm@52
|
44 void initialise_filter(BeagleRTContext *context);
|
victor@2
|
45
|
victor@2
|
46 // Task for handling the update of the frequencies using the matrix
|
victor@2
|
47 AuxiliaryTask gTriggerSamplesTask;
|
victor@2
|
48
|
victor@2
|
49 bool initialise_trigger();
|
victor@2
|
50 void trigger_samples();
|
victor@2
|
51
|
andrewm@56
|
52 // setup() is called once before the audio rendering starts.
|
victor@2
|
53 // Use it to perform any initialisation and allocation which is dependent
|
victor@2
|
54 // on the period size or sample rate.
|
victor@2
|
55 //
|
victor@2
|
56 // userData holds an opaque pointer to a data structure that was passed
|
victor@2
|
57 // in from the call to initAudio().
|
victor@2
|
58 //
|
victor@2
|
59 // Return true on success; returning false halts the program.
|
victor@2
|
60
|
andrewm@56
|
61 bool setup(BeagleRTContext *context, void *userData)
|
victor@2
|
62 {
|
victor@2
|
63
|
victor@2
|
64 // Retrieve a parameter passed in from the initAudio() call
|
victor@2
|
65 gSampleData = *(SampleData *)userData;
|
victor@2
|
66
|
victor@2
|
67 gReadPtr = -1;
|
victor@2
|
68
|
andrewm@52
|
69 initialise_filter(context);
|
victor@2
|
70
|
victor@2
|
71 // Initialise auxiliary tasks
|
victor@2
|
72 if(!initialise_trigger())
|
victor@2
|
73 return false;
|
victor@2
|
74
|
victor@2
|
75 return true;
|
victor@2
|
76 }
|
victor@2
|
77
|
victor@2
|
78 // render() is called regularly at the highest priority by the audio engine.
|
victor@2
|
79 // Input and output are given from the audio hardware and the other
|
victor@2
|
80 // ADCs and DACs (if available). If only audio is available, numMatrixFrames
|
victor@2
|
81 // will be 0.
|
victor@2
|
82
|
andrewm@52
|
83 void render(BeagleRTContext *context, void *userData)
|
victor@2
|
84 {
|
andrewm@52
|
85 for(unsigned int n = 0; n < context->audioFrames; n++) {
|
victor@2
|
86 float in = 0;
|
victor@2
|
87
|
victor@2
|
88 // If triggered...
|
victor@2
|
89 if(gReadPtr != -1)
|
victor@2
|
90 in += gSampleData.samples[gReadPtr++]; // ...read each sample...
|
victor@2
|
91
|
victor@2
|
92 if(gReadPtr >= gSampleData.sampleLen)
|
victor@2
|
93 gReadPtr = -1;
|
victor@2
|
94
|
victor@2
|
95 gFIRfilterIn[n] = in;
|
victor@2
|
96 }
|
victor@2
|
97
|
victor@2
|
98 ne10_fir_float_neon(&gFIRfilter, gFIRfilterIn, gFIRfilterOut, blockSize);
|
victor@2
|
99
|
andrewm@52
|
100 for(unsigned int n = 0; n < context->audioFrames; n++) {
|
andrewm@52
|
101 for(unsigned int channel = 0; channel < context->audioChannels; channel++)
|
andrewm@52
|
102 context->audioOut[n * context->audioChannels + channel] = gFIRfilterOut[n]; // ...and put it in both left and right channel
|
victor@2
|
103 }
|
victor@2
|
104
|
victor@2
|
105
|
victor@2
|
106 // Request that the lower-priority task run at next opportunity
|
andrewm@52
|
107 BeagleRT_scheduleAuxiliaryTask(gTriggerSamplesTask);
|
victor@2
|
108 }
|
victor@2
|
109
|
victor@2
|
110 // Initialise NE10 data structures to define FIR filter
|
victor@2
|
111
|
andrewm@52
|
112 void initialise_filter(BeagleRTContext *context)
|
victor@2
|
113 {
|
andrewm@52
|
114 blockSize = context->audioFrames;
|
victor@2
|
115 gFIRfilterState = (ne10_float32_t *) NE10_MALLOC ((FILTER_TAP_NUM+blockSize-1) * sizeof (ne10_float32_t));
|
victor@2
|
116 gFIRfilterIn = (ne10_float32_t *) NE10_MALLOC (blockSize * sizeof (ne10_float32_t));
|
victor@2
|
117 gFIRfilterOut = (ne10_float32_t *) NE10_MALLOC (blockSize * sizeof (ne10_float32_t));
|
victor@2
|
118 ne10_fir_init_float(&gFIRfilter, FILTER_TAP_NUM, filterTaps, gFIRfilterState, blockSize);
|
victor@2
|
119 }
|
victor@2
|
120
|
victor@2
|
121
|
victor@2
|
122 // Initialise the auxiliary task
|
victor@2
|
123 // and print info
|
victor@2
|
124
|
victor@2
|
125 bool initialise_trigger()
|
victor@2
|
126 {
|
andrewm@52
|
127 if((gTriggerSamplesTask = BeagleRT_createAuxiliaryTask(&trigger_samples, 50, "beaglert-trigger-samples")) == 0)
|
victor@2
|
128 return false;
|
victor@2
|
129
|
victor@2
|
130 rt_printf("Press 'a' to trigger sample, 's' to stop\n");
|
victor@2
|
131 rt_printf("Press 'q' to quit\n");
|
victor@2
|
132
|
victor@2
|
133 return true;
|
victor@2
|
134 }
|
victor@2
|
135
|
victor@2
|
136 // This is a lower-priority call to periodically read keyboard input
|
victor@2
|
137 // and trigger samples. By placing it at a lower priority,
|
victor@2
|
138 // it has minimal effect on the audio performance but it will take longer to
|
victor@2
|
139 // complete if the system is under heavy audio load.
|
victor@2
|
140
|
victor@2
|
141 void trigger_samples()
|
victor@2
|
142 {
|
victor@2
|
143 // This is not a real-time task!
|
victor@2
|
144 // Cos getchar is a system call, not handled by Xenomai.
|
victor@2
|
145 // This task will be automatically down graded.
|
victor@2
|
146
|
victor@2
|
147 char keyStroke = '.';
|
victor@2
|
148
|
victor@2
|
149 keyStroke = getchar();
|
victor@2
|
150 while(getchar()!='\n'); // to read the first stroke
|
victor@2
|
151
|
victor@2
|
152 switch (keyStroke)
|
victor@2
|
153 {
|
victor@2
|
154 case 'a':
|
victor@2
|
155 gReadPtr = 0;
|
victor@2
|
156 break;
|
victor@2
|
157 case 's':
|
victor@2
|
158 gReadPtr = -1;
|
victor@2
|
159 break;
|
victor@2
|
160 case 'q':
|
victor@2
|
161 gShouldStop = true;
|
victor@2
|
162 break;
|
victor@2
|
163 default:
|
victor@2
|
164 break;
|
victor@2
|
165 }
|
victor@2
|
166 }
|
victor@2
|
167
|
victor@2
|
168
|
victor@2
|
169
|
andrewm@56
|
170 // cleanup() is called once at the end, after the audio has stopped.
|
andrewm@56
|
171 // Release any resources that were allocated in setup().
|
victor@2
|
172
|
andrewm@56
|
173 void cleanup(BeagleRTContext *context, void *userData)
|
victor@2
|
174 {
|
victor@2
|
175 delete[] gSampleData.samples;
|
victor@2
|
176
|
victor@2
|
177 NE10_FREE(gFIRfilterState);
|
victor@2
|
178 NE10_FREE(gFIRfilterIn);
|
victor@2
|
179 NE10_FREE(gFIRfilterOut);
|
victor@2
|
180 }
|