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