Combined Workspace for Encoder and Decoder (#33)

* Combined Workspace for Encoder and Decoder

* Addressed review comments and some minor edits

---------

Co-authored-by: Divya B M <100655@ittiam.com>

1 files changed, 212 insertions, 0 deletions

diff --git a/encoder/ixheaace_mps_dct.c b/encoder/ixheaace_mps_dct.c
new file mode 100644
index 0000000..064616c
--- /dev/null
+++ b/encoder/ixheaace_mps_dct.c
@@ -0,0 +1,212 @@
+/******************************************************************************
+ *                                                                            *
+ * Copyright (C) 2023 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 "ixheaac_type_def.h"
+#include "ixheaace_error_codes.h"
+#include "ixheaac_error_standards.h"
+#include "ixheaace_mps_common_fix.h"
+#include "ixheaace_mps_defines.h"
+#include "ixheaace_mps_common_define.h"
+#include "ixheaace_bitbuffer.h"
+
+#include "ixheaace_psy_const.h"
+#include "ixheaace_tns.h"
+#include "ixheaace_tns_params.h"
+#include "ixheaace_rom.h"
+#include "ixheaace_common_rom.h"
+#include "ixheaace_enc_main.h"
+#include "ixheaace_fft.h"
+
+#include "ixheaace_mps_dct.h"
+#include "ixheaace_mps_buf.h"
+#include "ixheaace_mps_lib.h"
+#include "ixheaace_mps_main_structure.h"
+
+#include "ixheaace_mps_bitstream.h"
+#include "ixheaace_mps_frame_windowing.h"
+#include "ixheaace_mps_param_extract.h"
+#include "ixheaace_mps_static_gain.h"
+#include "ixheaace_mps_filter.h"
+#include "ixheaace_mps_delay.h"
+#include "ixheaace_mps_dmx_tdom_enh.h"
+#include "ixheaace_mps_tools_rom.h"
+
+IA_ERRORCODE ixheaace_mps_212_dct_iv(FLOAT32 *ptr_data, WORD32 length, WORD8 *ptr_scratch) {
+  WORD32 sin_step = 0;
+  WORD32 length_by_2 = length >> 1;
+  WORD32 step, idx;
+  FLOAT32 accu_1, accu_2, accu_3, accu_4;
+  FLOAT32 *ptr_data_0;
+  FLOAT32 *ptr_data_1;
+  const ixheaace_cmplx_str *ptr_cmplx_twiddle;
+  const ixheaace_cmplx_str *ptr_cmplx_sin_twiddle;
+  ixheaace_scratch_mem *pstr_scratch = (ixheaace_scratch_mem *)ptr_scratch;
+
+  if (length == 64) {
+    ptr_cmplx_twiddle = sine_window_64;
+    ptr_cmplx_sin_twiddle = sine_table_1024;
+    sin_step = 32;
+  } else {
+    ptr_cmplx_twiddle = sine_window_32;
+    ptr_cmplx_sin_twiddle = sine_table_1024;
+    sin_step = 64;
+  }
+
+  ptr_data_0 = &ptr_data[0];
+  ptr_data_1 = &ptr_data[length - 2];
+  for (idx = 0; idx < length_by_2 - 1; idx += 2, ptr_data_0 += 2, ptr_data_1 -= 2) {
+    accu_1 =
+        (ptr_data_1[1] * ptr_cmplx_twiddle[idx].re) - (ptr_data_0[0] * ptr_cmplx_twiddle[idx].im);
+    accu_2 =
+        (ptr_data_1[1] * ptr_cmplx_twiddle[idx].im) + (ptr_data_0[0] * ptr_cmplx_twiddle[idx].re);
+    accu_3 = (ptr_data_1[0] * ptr_cmplx_twiddle[idx + 1].re) -
+             (ptr_data_0[1] * ptr_cmplx_twiddle[idx + 1].im);
+    accu_4 = (ptr_data_1[0] * ptr_cmplx_twiddle[idx + 1].im) +
+             (ptr_data_0[1] * ptr_cmplx_twiddle[idx + 1].re);
+
+    ptr_data_0[0] = accu_2 / 4;
+    ptr_data_0[1] = accu_1 / 4;
+    ptr_data_1[0] = accu_4 / 4;
+    ptr_data_1[1] = -(accu_3 / 4);
+  }
+  ia_enhaacplus_enc_complex_fft(ptr_data, length_by_2, pstr_scratch);
+  if (length_by_2 == 32) {
+    for (idx = 0; idx < length; idx++) {
+      ptr_data[idx] = (ptr_data[idx] / (1 << (4)));
+    }
+  } else {
+    for (idx = 0; idx < length; idx++) {
+      ptr_data[idx] = (ptr_data[idx] / (1 << (3)));
+    }
+  }
+
+  ptr_data_0 = &ptr_data[0];
+  ptr_data_1 = &ptr_data[length - 2];
+  accu_1 = ptr_data_1[0];
+  accu_2 = ptr_data_1[1];
+  ptr_data_1[1] = -ptr_data_0[1];
+
+  for (step = sin_step, idx = 1; idx < (length_by_2 + 1) >> 1; idx++, step += sin_step) {
+    ixheaace_cmplx_str twd = ptr_cmplx_sin_twiddle[step];
+    accu_3 = (accu_1 * twd.re) - (accu_2 * twd.im);
+    accu_4 = (accu_1 * twd.im) + (accu_2 * twd.re);
+    ptr_data_0[1] = accu_3;
+    ptr_data_1[0] = accu_4;
+
+    ptr_data_0 += 2;
+    ptr_data_1 -= 2;
+
+    accu_3 = (ptr_data_0[1] * twd.re) - (ptr_data_0[0] * twd.im);
+    accu_4 = (ptr_data_0[1] * twd.im) + (ptr_data_0[0] * twd.re);
+    accu_1 = ptr_data_1[0];
+    accu_2 = ptr_data_1[1];
+
+    ptr_data_1[1] = -accu_3;
+    ptr_data_0[0] = accu_4;
+  }
+  accu_1 = (accu_1 * SQUARE_ROOT_TWO);
+  accu_2 = (accu_2 * SQUARE_ROOT_TWO);
+
+  ptr_data_1[0] = (accu_1 + accu_2) / 2;
+  ptr_data_0[1] = (accu_1 - accu_2) / 2;
+
+  return IA_NO_ERROR;
+}
+
+IA_ERRORCODE ixheaace_mps_212_dst_iv(FLOAT32 *ptr_data, WORD32 length, WORD8 *ptr_scratch) {
+  WORD32 sin_step = 0;
+  WORD32 step, idx;
+  WORD32 length_by_2 = length >> 1;
+  FLOAT32 accu_1, accu_2, accu_3, accu_4;
+  FLOAT32 *ptr_data_0;
+  FLOAT32 *ptr_data_1;
+  const ixheaace_cmplx_str *ptr_cmplx_twiddle;
+  const ixheaace_cmplx_str *ptr_cmplx_sin_twiddle;
+  ixheaace_scratch_mem *pstr_scratch = (ixheaace_scratch_mem *)ptr_scratch;
+  if (length == 64) {
+    ptr_cmplx_twiddle = sine_window_64;
+    ptr_cmplx_sin_twiddle = sine_table_1024;
+    sin_step = 32;
+  } else {
+    ptr_cmplx_twiddle = sine_window_32;
+    ptr_cmplx_sin_twiddle = sine_table_1024;
+    sin_step = 64;
+  }
+
+  ptr_data_0 = &ptr_data[0];
+  ptr_data_1 = &ptr_data[length - 2];
+  for (idx = 0; idx < length_by_2 - 1; idx += 2, ptr_data_0 += 2, ptr_data_1 -= 2) {
+    accu_1 = (ptr_data_1[1] * ptr_cmplx_twiddle[idx].re) -
+             ((-ptr_data_0[0]) * ptr_cmplx_twiddle[idx].im);
+    accu_2 = (ptr_data_1[1] * ptr_cmplx_twiddle[idx].im) +
+             ((-ptr_data_0[0]) * ptr_cmplx_twiddle[idx].re);
+    accu_3 = ((-ptr_data_1[0]) * ptr_cmplx_twiddle[idx + 1].re) -
+             (ptr_data_0[1] * ptr_cmplx_twiddle[idx + 1].im);
+    accu_4 = ((-ptr_data_1[0]) * ptr_cmplx_twiddle[idx + 1].im) +
+             (ptr_data_0[1] * ptr_cmplx_twiddle[idx + 1].re);
+    ptr_data_0[0] = accu_2 / 4;
+    ptr_data_0[1] = accu_1 / 4;
+    ptr_data_1[0] = accu_4 / 4;
+    ptr_data_1[1] = -(accu_3) / 4;
+  }
+  ia_enhaacplus_enc_complex_fft(ptr_data, length_by_2, pstr_scratch);
+  if (length_by_2 == 32) {
+    for (idx = 0; idx < length; idx++) {
+      ptr_data[idx] = (ptr_data[idx] / (1 << (4)));
+    }
+  } else {
+    for (idx = 0; idx < length; idx++) {
+      ptr_data[idx] = (ptr_data[idx] / (1 << (3)));
+    }
+  }
+  ptr_data_0 = &ptr_data[0];
+  ptr_data_1 = &ptr_data[length - 2];
+  accu_1 = ptr_data_1[0];
+  accu_2 = ptr_data_1[1];
+  ptr_data_1[1] = -ptr_data_0[0];
+  ptr_data_0[0] = ptr_data_0[1];
+
+  for (step = sin_step, idx = 1; idx < (length_by_2 + 1) >> 1; idx++, step += sin_step) {
+    ixheaace_cmplx_str twd = ptr_cmplx_sin_twiddle[step];
+
+    accu_3 = (accu_1 * twd.re) - (accu_2 * twd.im);
+    accu_4 = (accu_1 * twd.im) + (accu_2 * twd.re);
+    ptr_data_1[0] = -accu_3;
+    ptr_data_0[1] = -accu_4;
+
+    ptr_data_0 += 2;
+    ptr_data_1 -= 2;
+
+    accu_3 = (ptr_data_0[1] * twd.re) - (ptr_data_0[0] * twd.im);
+    accu_4 = (ptr_data_0[1] * twd.im) + (ptr_data_0[0] * twd.re);
+    accu_1 = ptr_data_1[0];
+    accu_2 = ptr_data_1[1];
+
+    ptr_data_0[0] = accu_3;
+    ptr_data_1[1] = -accu_4;
+  }
+
+  accu_1 = (accu_1 * SQUARE_ROOT_TWO);
+  accu_2 = (accu_2 * SQUARE_ROOT_TWO);
+  ptr_data_1[0] = (accu_1 + accu_2) / 2;
+  ptr_data_0[1] = (accu_1 - accu_2) / 2;
+
+  return IA_NO_ERROR;
+}