decoder/ixheaacd_mps_temp_process.c - platform/external/libxaac - Git at Google

 /******************************************************************************
  *                                                                            *
  * Copyright (C) 2018 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at:
  *
  * http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  *
  *****************************************************************************
  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
 */
 #include <ixheaacd_type_def.h>
 #include "ixheaacd_bitbuffer.h"
 #include "ixheaacd_config.h"
 #include "ixheaacd_mps_polyphase.h"
 #include "ixheaacd_mps_dec.h"
 #include "ixheaacd_mps_interface.h"

 #include "ixheaacd_mps_process.h"

 #include <math.h>
 #include <float.h>
 #include <memory.h>

 #include <assert.h>

 #include "ixheaacd_common_rom.h"
 #include "ixheaacd_defines.h"

 #include "ixheaacd_pns.h"

 #include <ixheaacd_aac_rom.h>
 #include "ixheaacd_pulsedata.h"

 #include "ixheaacd_sbrdecsettings.h"
 #include "ixheaacd_sbr_scale.h"
 #include "ixheaacd_lpp_tran.h"
 #include "ixheaacd_env_extr_part.h"
 #include <ixheaacd_sbr_rom.h>
 #include "ixheaacd_hybrid.h"
 #include "ixheaacd_ps_dec.h"
 #include "ixheaacd_env_extr.h"

 #include "ixheaacd_qmf_dec.h"

 #include "ixheaacd_env_calc.h"
 #include "ixheaacd_sbr_const.h"
 #include "ixheaacd_pvc_dec.h"
 #include "ixheaacd_sbr_dec.h"

 #define HP_SIZE (9)

 #define STP_LPF_COEFF1 (0.950f)
 #define STP_LPF_COEFF2 (0.450f)
 #define STP_UPDATE_ENERGY_RATE (32)
 #define STP_SCALE_LIMIT (2.82f)
 #define STP_DAMP (0.1f)

 #define max(a, b) ((a > b) ? (a) : (b))
 #define min(a, b) ((a < b) ? (a) : (b))

 static FLOAT32 ixheaacd_bp[BP_SIZE] = {
     0.0000f, 0.0005f, 0.0092f, 0.0587f, 0.2580f, 0.7392f, 0.9791f,
     0.9993f, 1.0000f, 1.0000f, 1.0000f, 1.0000f, 0.9999f, 0.9984f,
     0.9908f, 0.9639f, 0.8952f, 0.7711f, 0.6127f, 0.4609f, 0.3391f,
     0.2493f, 0.1848f, 0.1387f, 0.1053f};

 static FLOAT32 ixheaacd_gf[BP_SIZE] = {
     0.f,         0.f,         0.f,         0.f,         0.f,
     0.f,         1e-008f,     8.1e-007f,   3.61e-006f,  8.41e-006f,
     1.6e-005f,   2.704e-005f, 3.969e-005f, 5.625e-005f, 7.396e-005f,
     9.801e-005f, 0.00012321f, 0.00015625f, 0.00019881f, 0.00024964f,
     0.00032041f, 0.00041209f, 0.00053824f, 0.00070756f, 0.00094249f};

 static VOID ixheaacd_mps_temp_process_scale_calc(ia_mps_dec_state_struct* self,
                                                  WORD32 ts, FLOAT32* scale) {
   FLOAT32 dir_energy;
   FLOAT32 diff_energy[2];
   FLOAT32 temp;

   WORD32 ch, n;
   WORD32 left_ch = 0, right_ch = 1;

   if (self->subband_var.init_flag == 0) {
     for (ch = 0; ch < 2; ch++) {
       self->subband_var.tp_scale_last[ch] = 1.0f;
       self->subband_var.nrg_diff_prev[ch] = 32768 * 32768;
     }

     self->subband_var.nrg_dir_prev = 32768 * 32768;
     self->subband_var.init_flag = 1;
   }

   if (self->subband_var.update_old_ener == STP_UPDATE_ENERGY_RATE) {
     self->subband_var.update_old_ener = 1;
     self->subband_var.nrg_dir_prev = self->subband_var.nrg_dir;
     for (ch = 0; ch < self->out_ch_count; ch++)
       self->subband_var.nrg_diff_prev[ch] = self->subband_var.nrg_diff[ch];
   } else
     self->subband_var.update_old_ener++;

   dir_energy = 0;

   for (n = 6; n < BP_SIZE; n++) {
     FLOAT32 dir_left_re = self->hyb_dir_out[left_ch][ts][n + 7].re;
     FLOAT32 dir_right_re = self->hyb_dir_out[right_ch][ts][n + 7].re;
     FLOAT32 dir_left_im = self->hyb_dir_out[left_ch][ts][n + 7].im;
     FLOAT32 dir_right_im = self->hyb_dir_out[right_ch][ts][n + 7].im;

     temp = ((dir_left_re + dir_right_re) * (dir_left_re + dir_right_re)) +
            ((dir_left_im + dir_right_im) * (dir_left_im + dir_right_im));
     dir_energy += temp * ixheaacd_bp[n] * ixheaacd_bp[n] * ixheaacd_gf[n] *
                   ixheaacd_gf[n];
   }

   self->subband_var.nrg_dir =
       (FLOAT32)(STP_LPF_COEFF1 * self->subband_var.nrg_dir +
                 (1.0 - STP_LPF_COEFF1) * dir_energy);

   dir_energy /= (self->subband_var.nrg_dir_prev + ABS_THR);

   for (ch = 0; ch < self->out_ch_count; ch++) {
     diff_energy[ch] = 0;
     for (n = 6; n < BP_SIZE; n++) {
       FLOAT32 diff_re = self->hyb_diff_out[ch][ts][n + 7].re;
       FLOAT32 diff_im = self->hyb_diff_out[ch][ts][n + 7].im;

       temp = (diff_re * diff_re) + (diff_im * diff_im);
       diff_energy[ch] += temp * ixheaacd_bp[n] * ixheaacd_bp[n] *
                          ixheaacd_gf[n] * ixheaacd_gf[n];
     }

     self->subband_var.nrg_diff[ch] =
         (FLOAT32)(STP_LPF_COEFF1 * self->subband_var.nrg_diff[ch] +
                   (1.0 - STP_LPF_COEFF1) * diff_energy[ch]);
     diff_energy[ch] /= (self->subband_var.nrg_diff_prev[ch] + ABS_THR);
   }

   scale[left_ch] = (FLOAT32)sqrt((dir_energy) / (diff_energy[left_ch] + 1e-9));
   scale[right_ch] =
       (FLOAT32)sqrt((dir_energy) / (diff_energy[right_ch] + 1e-9));

   for (ch = 0; ch < self->out_ch_count; ch++) {
     scale[ch] = STP_DAMP + (1 - STP_DAMP) * scale[ch];
   }

   for (ch = 0; ch < self->out_ch_count; ch++) {
     scale[ch] =
         min(max(scale[ch], (FLOAT32)(1.0 / STP_SCALE_LIMIT)), STP_SCALE_LIMIT);
   }

   for (ch = 0; ch < self->out_ch_count; ch++) {
     scale[ch] =
         (FLOAT32)(STP_LPF_COEFF2 * scale[ch] +
                   (1.0 - STP_LPF_COEFF2) * self->subband_var.tp_scale_last[ch]);
     self->subband_var.tp_scale_last[ch] = scale[ch];
   }
 }

 static VOID ixheaacd_mps_subbandtp(ia_mps_dec_state_struct* self, WORD32 ts) {
   FLOAT32 scale[2];
   WORD32 ch, n;
   WORD32 no_scaling;
   FLOAT32 temp;
   const WORD32 ixheaacd_hybrid_to_qmf_map[] = {0, 0, 0, 0, 0, 0, 1, 1, 2, 2};

   ixheaacd_mps_temp_process_scale_calc(self, ts, scale);

   for (ch = 0; ch < self->out_ch_count; ch++) {
     no_scaling = 1;

     if ((self->config->bs_temp_shape_config == 1) ||
         (self->config->bs_temp_shape_config == 2))
       no_scaling = !self->temp_shape_enable_ch_stp[ch];

     if (no_scaling == 1) {
       for (n = 0; n < self->hyb_band_count; n++) {
         self->hyb_dir_out[ch][ts][n].re += self->hyb_diff_out[ch][ts][n].re;
         self->hyb_dir_out[ch][ts][n].im += self->hyb_diff_out[ch][ts][n].im;
       }
     } else {
       for (n = 0; n < 10; n++) {
         temp =
             (FLOAT32)(scale[ch] * ixheaacd_bp[ixheaacd_hybrid_to_qmf_map[n]]);
         self->hyb_dir_out[ch][ts][n].re +=
             (self->hyb_diff_out[ch][ts][n].re * temp);
         self->hyb_dir_out[ch][ts][n].im +=
             (self->hyb_diff_out[ch][ts][n].im * temp);
       }
       for (; n < HP_SIZE - 3 + 10; n++) {
         temp = (FLOAT32)(scale[ch] * ixheaacd_bp[n + 3 - 10]);
         self->hyb_dir_out[ch][ts][n].re +=
             (self->hyb_diff_out[ch][ts][n].re * temp);
         self->hyb_dir_out[ch][ts][n].im +=
             (self->hyb_diff_out[ch][ts][n].im * temp);
       }
       for (; n < self->hyb_band_count; n++) {
         temp = (FLOAT32)(scale[ch]);
         self->hyb_dir_out[ch][ts][n].re +=
             (self->hyb_diff_out[ch][ts][n].re * temp);
         self->hyb_dir_out[ch][ts][n].im +=
             (self->hyb_diff_out[ch][ts][n].im * temp);
       }
     }
   }
 }

 WORD32 ixheaacd_mps_temp_process(ia_mps_dec_state_struct* self) {
   WORD32 ch, ts, hyb;
   WORD32 err = 0;

   for (ch = 0; ch < self->out_ch_count; ch++) {
     for (ts = 0; ts < self->time_slots; ts++) {
       for (hyb = 0; hyb < HYBRID_BAND_BORDER; hyb++) {
         self->hyb_dir_out[ch][ts][hyb].re += self->hyb_diff_out[ch][ts][hyb].re;
         self->hyb_dir_out[ch][ts][hyb].im += self->hyb_diff_out[ch][ts][hyb].im;
         self->hyb_diff_out[ch][ts][hyb].re = 0;
         self->hyb_diff_out[ch][ts][hyb].im = 0;
       }
     }
   }

   for (ts = 0; ts < self->time_slots; ts++) ixheaacd_mps_subbandtp(self, ts);

   ixheaacd_mps_qmf_hyb_synthesis(self);

   for (ch = 0; ch < self->out_ch_count; ch++) {
     err = ixheaacd_sbr_dec_from_mps(&self->qmf_out_dir[ch][0][0].re,
                                     self->p_sbr_dec[ch], self->p_sbr_frame[ch],
                                     self->p_sbr_header[ch]);
     if (err) return err;
   }

   ixheaacd_mps_synt_calc(self);
   return err;
 }
	/******************************************************************************
	* *
	* Copyright (C) 2018 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at:
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*
	*****************************************************************************
	* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
	*/
	#include <ixheaacd_type_def.h>
	#include "ixheaacd_bitbuffer.h"
	#include "ixheaacd_config.h"
	#include "ixheaacd_mps_polyphase.h"
	#include "ixheaacd_mps_dec.h"
	#include "ixheaacd_mps_interface.h"

	#include "ixheaacd_mps_process.h"

	#include <math.h>
	#include <float.h>
	#include <memory.h>

	#include <assert.h>

	#include "ixheaacd_common_rom.h"
	#include "ixheaacd_defines.h"

	#include "ixheaacd_pns.h"

	#include <ixheaacd_aac_rom.h>
	#include "ixheaacd_pulsedata.h"

	#include "ixheaacd_sbrdecsettings.h"
	#include "ixheaacd_sbr_scale.h"
	#include "ixheaacd_lpp_tran.h"
	#include "ixheaacd_env_extr_part.h"
	#include <ixheaacd_sbr_rom.h>
	#include "ixheaacd_hybrid.h"
	#include "ixheaacd_ps_dec.h"
	#include "ixheaacd_env_extr.h"

	#include "ixheaacd_qmf_dec.h"

	#include "ixheaacd_env_calc.h"
	#include "ixheaacd_sbr_const.h"
	#include "ixheaacd_pvc_dec.h"
	#include "ixheaacd_sbr_dec.h"

	#define HP_SIZE (9)

	#define STP_LPF_COEFF1 (0.950f)
	#define STP_LPF_COEFF2 (0.450f)
	#define STP_UPDATE_ENERGY_RATE (32)
	#define STP_SCALE_LIMIT (2.82f)
	#define STP_DAMP (0.1f)

	#define max(a, b) ((a > b) ? (a) : (b))
	#define min(a, b) ((a < b) ? (a) : (b))

	static FLOAT32 ixheaacd_bp[BP_SIZE] = {
	0.0000f, 0.0005f, 0.0092f, 0.0587f, 0.2580f, 0.7392f, 0.9791f,
	0.9993f, 1.0000f, 1.0000f, 1.0000f, 1.0000f, 0.9999f, 0.9984f,
	0.9908f, 0.9639f, 0.8952f, 0.7711f, 0.6127f, 0.4609f, 0.3391f,
	0.2493f, 0.1848f, 0.1387f, 0.1053f};

	static FLOAT32 ixheaacd_gf[BP_SIZE] = {
	0.f, 0.f, 0.f, 0.f, 0.f,
	0.f, 1e-008f, 8.1e-007f, 3.61e-006f, 8.41e-006f,
	1.6e-005f, 2.704e-005f, 3.969e-005f, 5.625e-005f, 7.396e-005f,
	9.801e-005f, 0.00012321f, 0.00015625f, 0.00019881f, 0.00024964f,
	0.00032041f, 0.00041209f, 0.00053824f, 0.00070756f, 0.00094249f};

	static VOID ixheaacd_mps_temp_process_scale_calc(ia_mps_dec_state_struct* self,
	WORD32 ts, FLOAT32* scale) {
	FLOAT32 dir_energy;
	FLOAT32 diff_energy[2];
	FLOAT32 temp;

	WORD32 ch, n;
	WORD32 left_ch = 0, right_ch = 1;

	if (self->subband_var.init_flag == 0) {
	for (ch = 0; ch < 2; ch++) {
	self->subband_var.tp_scale_last[ch] = 1.0f;
	self->subband_var.nrg_diff_prev[ch] = 32768 * 32768;
	}

	self->subband_var.nrg_dir_prev = 32768 * 32768;
	self->subband_var.init_flag = 1;
	}

	if (self->subband_var.update_old_ener == STP_UPDATE_ENERGY_RATE) {
	self->subband_var.update_old_ener = 1;
	self->subband_var.nrg_dir_prev = self->subband_var.nrg_dir;
	for (ch = 0; ch < self->out_ch_count; ch++)
	self->subband_var.nrg_diff_prev[ch] = self->subband_var.nrg_diff[ch];
	} else
	self->subband_var.update_old_ener++;

	dir_energy = 0;

	for (n = 6; n < BP_SIZE; n++) {
	FLOAT32 dir_left_re = self->hyb_dir_out[left_ch][ts][n + 7].re;
	FLOAT32 dir_right_re = self->hyb_dir_out[right_ch][ts][n + 7].re;
	FLOAT32 dir_left_im = self->hyb_dir_out[left_ch][ts][n + 7].im;
	FLOAT32 dir_right_im = self->hyb_dir_out[right_ch][ts][n + 7].im;

	temp = ((dir_left_re + dir_right_re) * (dir_left_re + dir_right_re)) +
	((dir_left_im + dir_right_im) * (dir_left_im + dir_right_im));
	dir_energy += temp * ixheaacd_bp[n] * ixheaacd_bp[n] * ixheaacd_gf[n] *
	ixheaacd_gf[n];
	}

	self->subband_var.nrg_dir =
	(FLOAT32)(STP_LPF_COEFF1 * self->subband_var.nrg_dir +
	(1.0 - STP_LPF_COEFF1) * dir_energy);

	dir_energy /= (self->subband_var.nrg_dir_prev + ABS_THR);

	for (ch = 0; ch < self->out_ch_count; ch++) {
	diff_energy[ch] = 0;
	for (n = 6; n < BP_SIZE; n++) {
	FLOAT32 diff_re = self->hyb_diff_out[ch][ts][n + 7].re;
	FLOAT32 diff_im = self->hyb_diff_out[ch][ts][n + 7].im;

	temp = (diff_re * diff_re) + (diff_im * diff_im);
	diff_energy[ch] += temp * ixheaacd_bp[n] * ixheaacd_bp[n] *
	ixheaacd_gf[n] * ixheaacd_gf[n];
	}

	self->subband_var.nrg_diff[ch] =
	(FLOAT32)(STP_LPF_COEFF1 * self->subband_var.nrg_diff[ch] +
	(1.0 - STP_LPF_COEFF1) * diff_energy[ch]);
	diff_energy[ch] /= (self->subband_var.nrg_diff_prev[ch] + ABS_THR);
	}

	scale[left_ch] = (FLOAT32)sqrt((dir_energy) / (diff_energy[left_ch] + 1e-9));
	scale[right_ch] =
	(FLOAT32)sqrt((dir_energy) / (diff_energy[right_ch] + 1e-9));

	for (ch = 0; ch < self->out_ch_count; ch++) {
	scale[ch] = STP_DAMP + (1 - STP_DAMP) * scale[ch];
	}

	for (ch = 0; ch < self->out_ch_count; ch++) {
	scale[ch] =
	min(max(scale[ch], (FLOAT32)(1.0 / STP_SCALE_LIMIT)), STP_SCALE_LIMIT);
	}

	for (ch = 0; ch < self->out_ch_count; ch++) {
	scale[ch] =
	(FLOAT32)(STP_LPF_COEFF2 * scale[ch] +
	(1.0 - STP_LPF_COEFF2) * self->subband_var.tp_scale_last[ch]);
	self->subband_var.tp_scale_last[ch] = scale[ch];
	}
	}

	static VOID ixheaacd_mps_subbandtp(ia_mps_dec_state_struct* self, WORD32 ts) {
	FLOAT32 scale[2];
	WORD32 ch, n;
	WORD32 no_scaling;
	FLOAT32 temp;
	const WORD32 ixheaacd_hybrid_to_qmf_map[] = {0, 0, 0, 0, 0, 0, 1, 1, 2, 2};

	ixheaacd_mps_temp_process_scale_calc(self, ts, scale);

	for (ch = 0; ch < self->out_ch_count; ch++) {
	no_scaling = 1;

	if ((self->config->bs_temp_shape_config == 1) \|\|
	(self->config->bs_temp_shape_config == 2))
	no_scaling = !self->temp_shape_enable_ch_stp[ch];

	if (no_scaling == 1) {
	for (n = 0; n < self->hyb_band_count; n++) {
	self->hyb_dir_out[ch][ts][n].re += self->hyb_diff_out[ch][ts][n].re;
	self->hyb_dir_out[ch][ts][n].im += self->hyb_diff_out[ch][ts][n].im;
	}
	} else {
	for (n = 0; n < 10; n++) {
	temp =
	(FLOAT32)(scale[ch] * ixheaacd_bp[ixheaacd_hybrid_to_qmf_map[n]]);
	self->hyb_dir_out[ch][ts][n].re +=
	(self->hyb_diff_out[ch][ts][n].re * temp);
	self->hyb_dir_out[ch][ts][n].im +=
	(self->hyb_diff_out[ch][ts][n].im * temp);
	}
	for (; n < HP_SIZE - 3 + 10; n++) {
	temp = (FLOAT32)(scale[ch] * ixheaacd_bp[n + 3 - 10]);
	self->hyb_dir_out[ch][ts][n].re +=
	(self->hyb_diff_out[ch][ts][n].re * temp);
	self->hyb_dir_out[ch][ts][n].im +=
	(self->hyb_diff_out[ch][ts][n].im * temp);
	}
	for (; n < self->hyb_band_count; n++) {
	temp = (FLOAT32)(scale[ch]);
	self->hyb_dir_out[ch][ts][n].re +=
	(self->hyb_diff_out[ch][ts][n].re * temp);
	self->hyb_dir_out[ch][ts][n].im +=
	(self->hyb_diff_out[ch][ts][n].im * temp);
	}
	}
	}
	}

	WORD32 ixheaacd_mps_temp_process(ia_mps_dec_state_struct* self) {
	WORD32 ch, ts, hyb;
	WORD32 err = 0;

	for (ch = 0; ch < self->out_ch_count; ch++) {
	for (ts = 0; ts < self->time_slots; ts++) {
	for (hyb = 0; hyb < HYBRID_BAND_BORDER; hyb++) {
	self->hyb_dir_out[ch][ts][hyb].re += self->hyb_diff_out[ch][ts][hyb].re;
	self->hyb_dir_out[ch][ts][hyb].im += self->hyb_diff_out[ch][ts][hyb].im;
	self->hyb_diff_out[ch][ts][hyb].re = 0;
	self->hyb_diff_out[ch][ts][hyb].im = 0;
	}
	}
	}

	for (ts = 0; ts < self->time_slots; ts++) ixheaacd_mps_subbandtp(self, ts);

	ixheaacd_mps_qmf_hyb_synthesis(self);

	for (ch = 0; ch < self->out_ch_count; ch++) {
	err = ixheaacd_sbr_dec_from_mps(&self->qmf_out_dir[ch][0][0].re,
	self->p_sbr_dec[ch], self->p_sbr_frame[ch],
	self->p_sbr_header[ch]);
	if (err) return err;
	}

	ixheaacd_mps_synt_calc(self);
	return err;
	}