// Copyright 2020-2026 XMOS LIMITED.
// This Software is subject to the terms of the XMOS Public Licence: Version 1.

#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>

#include "xmath/xmath.h"

#include "../../tst_common.h"

#include "unity_fixture.h"


TEST_GROUP_RUNNER(bfp_mean) {
  RUN_TEST_CASE(bfp_mean, bfp_s16_mean);
  RUN_TEST_CASE(bfp_mean, bfp_s32_mean);
}

TEST_GROUP(bfp_mean);
TEST_SETUP(bfp_mean) { fflush(stdout); }
TEST_TEAR_DOWN(bfp_mean) {}

#if SMOKE_TEST
#  define REPS       (100)
#  define MAX_LEN    (128)
#else
#  define REPS       (1000)
#  define MAX_LEN    (512)
#endif


TEST(bfp_mean, bfp_s16_mean)
{


    unsigned seed = SEED_FROM_FUNC_NAME();

    int16_t dataB[MAX_LEN];
    bfp_s16_t B;

    B.data = dataB;

    for(int r = 0; r < REPS; r++){
        setExtraInfo_RS(r, seed);

        B.length = pseudo_rand_uint(&seed, 1, MAX_LEN+1);
        B.exp = pseudo_rand_int(&seed, -100, 100);
        B.hr = pseudo_rand_uint(&seed, 0, 15);


        double sum = 0;

        for(unsigned int i = 0; i < B.length; i++){
            B.data[i] = (pseudo_rand_int16(&seed) >> B.hr);

            //This is a simple way of bounding the error due to the new rounding mode in VX4B
            #if defined(__VX4B__)
                B.data[i] = B.data[i]&~1;
            #endif

            sum += B.data[i];
        }

        bfp_s16_headroom(&B);

        float result = bfp_s16_mean(&B);

        int16_t Q;
        exponent_t R;
        f32_unpack_s16(&Q, &R, result);

        double meanf = ldexp(sum, B.exp) / B.length;
        
        double diff = meanf - result;

        TEST_ASSERT( fabs(diff) <= ldexp(1, R) );
    }
}

TEST(bfp_mean, bfp_s32_mean)
{


    unsigned seed = SEED_FROM_FUNC_NAME();

    int32_t dataB[MAX_LEN];
    bfp_s32_t B;

    B.data = dataB;

    for(int r = 0; r < REPS; r++){
        setExtraInfo_RS(r, seed);

        B.length = pseudo_rand_uint(&seed, 1, MAX_LEN+1);
        B.exp = pseudo_rand_int(&seed, -100, 100);
        B.hr = pseudo_rand_uint(&seed, 0, 31);

        double sum = 0;

        for(unsigned int i = 0; i < B.length; i++){
            B.data[i] = pseudo_rand_int32(&seed) >> B.hr;

            sum += B.data[i];
        }

        bfp_s32_headroom(&B);

        float_s32_t result = bfp_s32_mean(&B);
        double meanf = ldexp(sum, B.exp) / B.length;
        double diff = meanf - ldexp(result.mant, result.exp);

        TEST_ASSERT( fabs(diff) <= ldexp(1, result.exp) );
    }
}
