36 lines
907 B
C
36 lines
907 B
C
|
#include "ref.h"
|
||
|
|
|
||
|
|
void ref_cmplx_mag_squared_f32(float32_t *pSrc, float32_t *pDst,
|
||
|
|
uint32_t numSamples)
|
||
|
|
{
|
||
|
|
uint32_t i;
|
||
|
|
|
||
|
|
for (i = 0; i < numSamples * 2; i += 2) {
|
||
|
|
*pDst++ = pSrc[i] * pSrc[i] + pSrc[i + 1] * pSrc[i + 1];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ref_cmplx_mag_squared_q31(q31_t *pSrc, q31_t *pDst, uint32_t numSamples)
|
||
|
|
{
|
||
|
|
uint32_t i;
|
||
|
|
q31_t acc0, acc1;
|
||
|
|
|
||
|
|
for (i = 0; i < numSamples * 2; i += 2) {
|
||
|
|
acc0 = (q31_t)(((q63_t)pSrc[i] * pSrc[i]) >> 33);
|
||
|
|
acc1 = (q31_t)(((q63_t)pSrc[i + 1] * pSrc[i + 1]) >> 33);
|
||
|
|
*pDst++ = acc0 + acc1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ref_cmplx_mag_squared_q15(q15_t *pSrc, q15_t *pDst, uint32_t numSamples)
|
||
|
|
{
|
||
|
|
uint32_t i;
|
||
|
|
q31_t acc0, acc1;
|
||
|
|
|
||
|
|
for (i = 0; i < numSamples * 2; i += 2) {
|
||
|
|
acc0 = pSrc[i] * pSrc[i];
|
||
|
|
acc1 = pSrc[i + 1] * pSrc[i + 1];
|
||
|
|
*pDst++ = (q15_t)(((q63_t)acc0 + acc1) >> 17);
|
||
|
|
}
|
||
|
|
}
|