55 lines
1.4 KiB
C
55 lines
1.4 KiB
C
#include "ref.h"
|
|
|
|
void ref_rms_f32(float32_t *pSrc, uint32_t blockSize, float32_t *pResult)
|
|
{
|
|
uint32_t i;
|
|
float32_t sumsq = 0;
|
|
|
|
for (i = 0; i < blockSize; i++) {
|
|
sumsq += pSrc[i] * pSrc[i];
|
|
}
|
|
*pResult = sqrtf(sumsq / (float32_t)blockSize);
|
|
}
|
|
|
|
void ref_rms_q31(q31_t *pSrc, uint32_t blockSize, q31_t *pResult)
|
|
{
|
|
uint32_t i;
|
|
uint64_t sumsq = 0; /* accumulator (can get never negative. changed type
|
|
from q63 to uint64 */
|
|
q63_t tmp1;
|
|
q31_t tmp2;
|
|
|
|
float help_float;
|
|
|
|
for (i = 0; i < blockSize; i++) {
|
|
sumsq += (q63_t)pSrc[i] * pSrc[i];
|
|
}
|
|
tmp1 = (sumsq / (q63_t)blockSize) >> 31;
|
|
tmp2 = ref_sat_q31(tmp1);
|
|
|
|
/* GCC M0 problem: __aeabi_f2iz(QNAN) returns not 0 */
|
|
help_float = (sqrtf((float)tmp2 / 2147483648.0f) * 2147483648.0f);
|
|
/* Checking for a NAN value in help_float */
|
|
if (((*((int *)(&help_float))) & 0x7FC00000) == 0x7FC00000) {
|
|
help_float = 0;
|
|
}
|
|
*pResult = (q31_t)(help_float);
|
|
|
|
// *pResult = (q31_t)(sqrtf((float)tmp2 / 2147483648.0f) * 2147483648.0f);
|
|
}
|
|
|
|
void ref_rms_q15(q15_t *pSrc, uint32_t blockSize, q15_t *pResult)
|
|
{
|
|
uint32_t i;
|
|
q63_t sumsq = 0;
|
|
q31_t tmp1;
|
|
q15_t tmp2;
|
|
|
|
for (i = 0; i < blockSize; i++) {
|
|
sumsq += (q63_t)pSrc[i] * pSrc[i];
|
|
}
|
|
tmp1 = (sumsq / (q63_t)blockSize) >> 15;
|
|
tmp2 = ref_sat_q15(tmp1);
|
|
*pResult = (q15_t)(sqrtf((float)tmp2 / 32768.0f) * 32768.0f);
|
|
}
|