/*====================================================================================
    EVS Codec 3GPP TS26.442 Apr 03, 2018. Version 12.11.0 / 13.6.0 / 14.2.0
  ====================================================================================*/

#include "options.h"    /* Compilation switches                   */
#include "stl.h"        /* required for wmc_tool */
#include "prot_fx.h"    /* Function prototypes                    */
#include "cnst_fx.h"    /* Common constants                       */

/*--------------------------------------------------------------------------*
 * hq_bit_allocation_fx()
 *
 * Assign bits for HQ fine structure coding with PVQ
 *--------------------------------------------------------------------------*/

void hq_bit_allocation_fx(
    const Word32 core_brate,     /* i  : Core bit-rate                    Q0  */
    const Word16 length,         /* i  : Frame length                     Q0  */
    const Word16 hqswb_clas,     /* i  : HQ class                         Q0  */
    Word16 *num_bits,      /* i/o: Remaining bit budget             Q0  */
    const Word16 *normqlg2,      /* i  : Quantized norms                  Q0  */
    const Word16 nb_sfm,         /* i  : Number sub bands to be encoded   Q0  */
    const Word16 *sfmsize,       /* i  : Sub band bandwidths              Q0  */
    Word16 *noise_level,   /* o  : HVQ noise level                    */
    Word16 *R,             /* o  : Bit allocation per sub band      Q0  */
    Word16 *Rsubband,      /* o  : Fractional bit allocation        Q3  */
    Word16 *sum,           /* o  : Sum of allocated shape bits      Q0  */
    Word16 *core_sfm,      /* o  : Last coded band in core          Q0  */
    const Word16 num_env_bands   /* i  : Number sub bands to be encoded for HQ_GEN Q0  */
)
{
    Word16 i;
    Word16 idx[NB_SFM];
    Word16 wnorm[NB_SFM];
    Word16 avrg_wnorm;
    Word16 tmp, tmp2;
    Word16 E_low;
    Word16 E_hb_mean;
    Word16 E_max;
    Word16 i_max;
    /* Temp */

    Word16 sfm_limit = nb_sfm;
    move16();

    set16_fx( R, 0, NB_SFM);
    FOR( i = 0; i < nb_sfm; i++ )
    {
        idx[i] = i;
        move16();
    }
    test();
    test();
    test();
    if( sub(hqswb_clas, HQ_TRANSIENT) != 0 && sub(hqswb_clas, HQ_HVQ) != 0 && !(sub(length, L_FRAME16k) == 0 && L_sub(core_brate, HQ_32k) == 0))
    {
        /* 'nf_idx' 2-bits index written later */
        *num_bits = sub(*num_bits, 2);
    }

    test();
    IF ( sub(hqswb_clas, HQ_GEN_SWB) == 0 || sub(hqswb_clas, HQ_GEN_FB) == 0 )
    {
        IF ( L_sub(core_brate, HQ_32k) == 0 )
        {
            *num_bits = sub(*num_bits, HQ_GENERIC_SWB_NBITS2 );
        }
        ELSE
        {
            *num_bits = sub(*num_bits, HQ_GENERIC_SWB_NBITS );
        }

        if ( sub(hqswb_clas, HQ_GEN_FB) == 0 )
        {
            *num_bits = sub(*num_bits, HQ_GENERIC_FB_NBITS );
        }
    }

    IF( ( sub(length, L_FRAME48k) == 0 ) && (sub(hqswb_clas, HQ_HARMONIC) != 0) && (sub(hqswb_clas, HQ_HVQ) != 0))
    {
        tmp = 0;
        move16();
        if( sub(hqswb_clas,HQ_TRANSIENT) == 0 )
        {
            tmp = 1;
            move16();
        }
        map_quant_weight_fx( normqlg2, wnorm, tmp );
    }
    ELSE
    {
        Copy( normqlg2, wnorm, nb_sfm );
    }

    IF( sub(hqswb_clas, HQ_HARMONIC) == 0 )
    {
        /* classification and limit bandwidth for bit allocation */
        sfm_limit = sub(sfm_limit, 2);
        limit_band_noise_level_calc_fx( wnorm, &sfm_limit, core_brate, noise_level );

        /* Detect important band in high frequency region */
        E_low = sum16_fx(wnorm, SFM_G1);
        i_max = 0;
        move16();
        E_max = MIN16B;
        move16();
        E_hb_mean = 0;
        move16();
        FOR( i = SFM_G1; i < nb_sfm; i++)
        {
            E_hb_mean = add(E_hb_mean, wnorm[i]);
            IF( sub(wnorm[i], E_max) > 0)
            {
                E_max = wnorm[i];
                move16();
                i_max = i;
                move16();
            }
        }
        E_hb_mean = shr(E_hb_mean, 4); /* Truncated division by SFM_G1 */
        set16_fx( wnorm + sfm_limit, -20, sub(nb_sfm, sfm_limit) );
        IF (L_msu0(L_deposit_l(E_low), E_max, 15) <= 0)
        {
            IF (L_msu(L_deposit_h(E_hb_mean), E_max, 21955) <= 0)   /* 21955 = 0.67 (Q15) */
            {
                if (sub(i_max, sfm_limit) >= 0)
                {
                    wnorm[i_max] = E_max;
                    move16();
                }
            }
        }
    }
    test();
    test();
    test();
    test();
    IF( sub(hqswb_clas, HQ_HVQ) == 0 )
    {
        *sum = 0;
        move16();
    }
    ELSE IF ( sub(hqswb_clas, HQ_GEN_SWB) == 0 || (sub(hqswb_clas, HQ_TRANSIENT) == 0 && sub(length, L_FRAME32k) == 0 && L_sub(core_brate, HQ_32k) <= 0) )
    {
        *sum = BitAllocF_fx( wnorm, core_brate, *num_bits, nb_sfm, R, Rsubband, hqswb_clas, num_env_bands );
    }
    ELSE IF( sub(length, L_FRAME16k) == 0 && L_sub(core_brate, HQ_32k) == 0 )
    {
        IF( sub(hqswb_clas, HQ_TRANSIENT) != 0 )
        {
            avrg_wnorm = wnorm[10];
            move16();
            FOR( i=11; i<18; i++ )
            {
                avrg_wnorm = add(avrg_wnorm, wnorm[i]);
            }

            avrg_wnorm  = shr(avrg_wnorm, 3);
            FOR( i=0; i<4; i++ )
            {
                if( sub(wnorm[i], avrg_wnorm) < 0 )
                {
                    wnorm[i] = avrg_wnorm;
                    move16();
                }
            }

            /* Estimate number of bits per band */
            *sum = BitAllocWB_fx( wnorm, *num_bits, nb_sfm, R, Rsubband );
        }
        ELSE
        {
            reordvct_fx(wnorm, nb_sfm, idx);
            bitalloc_fx( wnorm, idx, *num_bits, nb_sfm, QBIT_MAX2, R, sfmsize, hqswb_clas );
            bitallocsum_fx( R, nb_sfm, sum, Rsubband, *num_bits, length, sfmsize );
        }
    }
    ELSE
    {
        reordvct_fx(wnorm, nb_sfm, idx);

        /* enlarge the wnorm value so that more bits can be allocated to (sfm_limit/2 ~ sfm_limit) range */
        IF( sub(hqswb_clas, HQ_HARMONIC) == 0 )
        {
            tmp = shr(sfm_limit,1);
            tmp2 = sub(tmp,1);
            FOR( i=tmp; i<sfm_limit; i++ )
            {
                wnorm[i] = wnorm[tmp2];
                move16();
            }
        }
        bitalloc_fx( wnorm, idx, *num_bits, nb_sfm, QBIT_MAX2, R, sfmsize, hqswb_clas );
        bitallocsum_fx( R, nb_sfm, sum, Rsubband, *num_bits, length, sfmsize );
    }

    /* Find last coded core band */
    *core_sfm = sub(nb_sfm, 1);
    test();
    test();
    IF( hqswb_clas == HQ_NORMAL || sub(hqswb_clas, HQ_HARMONIC) == 0 )
    {
        *core_sfm = find_last_band_fx(R, nb_sfm );
    }
    ELSE IF ( sub(hqswb_clas, HQ_GEN_SWB) == 0 || sub(hqswb_clas, HQ_GEN_FB) == 0 )
    {
        *core_sfm = find_last_band_fx( R, nb_sfm );
        IF ( sub(*core_sfm ,num_env_bands) <0 )
        {
            *core_sfm = sub(num_env_bands,1);
        }
    }

    *num_bits = sub(*num_bits, *sum);

    return;
}