38 lines
1017 B
C
38 lines
1017 B
C
#include "ref.h"
|
|
|
|
void ref_cmplx_mag_f32(float32_t *pSrc, float32_t *pDst, uint32_t numSamples)
|
|
{
|
|
uint32_t i;
|
|
|
|
for (i = 0; i < numSamples * 2; i += 2) {
|
|
*pDst++ = sqrtf(pSrc[i] * pSrc[i] + pSrc[i + 1] * pSrc[i + 1]);
|
|
}
|
|
}
|
|
|
|
void ref_cmplx_mag_q31(q31_t *pSrc, q31_t *pDst, uint32_t numSamples)
|
|
{
|
|
uint32_t i;
|
|
q31_t acc0, acc1, out;
|
|
|
|
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);
|
|
out = acc0 + acc1;
|
|
*pDst++ = (q31_t)(sqrtf((float)out / 2147483648.0f) * 2147483648.0f);
|
|
}
|
|
}
|
|
|
|
void ref_cmplx_mag_q15(q15_t *pSrc, q15_t *pDst, uint32_t numSamples)
|
|
{
|
|
uint32_t i;
|
|
q31_t acc0, acc1;
|
|
q15_t out;
|
|
|
|
for (i = 0; i < numSamples * 2; i += 2) {
|
|
acc0 = pSrc[i] * pSrc[i];
|
|
acc1 = pSrc[i + 1] * pSrc[i + 1];
|
|
out = (q15_t)(((q63_t)acc0 + acc1) >> 17);
|
|
*pDst++ = (q15_t)(sqrtf((float)out / 32768.0f) * 32768.0f);
|
|
}
|
|
}
|