Line data Source code
1 : /******************************************************************************************************
2 :
3 : (C) 2022-2025 IVAS codec Public Collaboration with portions copyright Dolby International AB, Ericsson AB,
4 : Fraunhofer-Gesellschaft zur Foerderung der angewandten Forschung e.V., Huawei Technologies Co. LTD.,
5 : Koninklijke Philips N.V., Nippon Telegraph and Telephone Corporation, Nokia Technologies Oy, Orange,
6 : Panasonic Holdings Corporation, Qualcomm Technologies, Inc., VoiceAge Corporation, and other
7 : contributors to this repository. All Rights Reserved.
8 :
9 : This software is protected by copyright law and by international treaties.
10 : The IVAS codec Public Collaboration consisting of Dolby International AB, Ericsson AB,
11 : Fraunhofer-Gesellschaft zur Foerderung der angewandten Forschung e.V., Huawei Technologies Co. LTD.,
12 : Koninklijke Philips N.V., Nippon Telegraph and Telephone Corporation, Nokia Technologies Oy, Orange,
13 : Panasonic Holdings Corporation, Qualcomm Technologies, Inc., VoiceAge Corporation, and other
14 : contributors to this repository retain full ownership rights in their respective contributions in
15 : the software. This notice grants no license of any kind, including but not limited to patent
16 : license, nor is any license granted by implication, estoppel or otherwise.
17 :
18 : Contributors are required to enter into the IVAS codec Public Collaboration agreement before making
19 : contributions.
20 :
21 : This software is provided "AS IS", without any express or implied warranties. The software is in the
22 : development stage. It is intended exclusively for experts who have experience with such software and
23 : solely for the purpose of inspection. All implied warranties of non-infringement, merchantability
24 : and fitness for a particular purpose are hereby disclaimed and excluded.
25 :
26 : Any dispute, controversy or claim arising under or in relation to providing this software shall be
27 : submitted to and settled by the final, binding jurisdiction of the courts of Munich, Germany in
28 : accordance with the laws of the Federal Republic of Germany excluding its conflict of law rules and
29 : the United Nations Convention on Contracts on the International Sales of Goods.
30 :
31 : *******************************************************************************************************/
32 :
33 : #include <assert.h>
34 : #include <stdint.h>
35 : #include "options.h"
36 : #include <math.h>
37 : #include "ivas_cnst.h"
38 : #include "ivas_rom_com.h"
39 : #include "ivas_prot.h"
40 : #include "prot.h"
41 : #include "cnst.h"
42 : #include "wmc_auto.h"
43 :
44 :
45 : /*-----------------------------------------------------------------------*
46 : * Local function prototypes
47 : *-----------------------------------------------------------------------*/
48 :
49 : static uint16_t deindex_sph_idx_general( const int16_t idx_sph, const int16_t no_bits, float *theta_dec, float *phi_dec, uint16_t *p_id_phi, const MC_LS_SETUP mc_format );
50 :
51 :
52 : /*-------------------------------------------------------------------------
53 : * ivas_get_hodirac_flag()
54 : *
55 : * Return flag for HO-DirAC method at high bitrates
56 : *------------------------------------------------------------------------*/
57 :
58 : /*! r: HO-DirAC flag */
59 36654167 : int16_t ivas_get_hodirac_flag(
60 : const int32_t ivas_total_brate, /* i : IVAS total bitrate */
61 : const int16_t sba_order /* i : Ambisonic (SBA) order */
62 : )
63 : {
64 36654167 : if ( sba_order > 1 && ivas_total_brate > IVAS_256k )
65 : {
66 3649872 : return 1;
67 : }
68 : else
69 : {
70 33004295 : return 0;
71 : }
72 : }
73 :
74 :
75 : /*-------------------------------------------------------------------------
76 : * ivas_dirac_sba_config()
77 : *
78 : * DirAC Configuration function; used also in MASA decoder
79 : *------------------------------------------------------------------------*/
80 :
81 368728 : ivas_error ivas_dirac_config(
82 : void *st_ivas, /* i/o: IVAS encoder/decoder state structure */
83 : const int16_t enc_dec /* i : encoder or decoder flag */
84 : )
85 : {
86 : IVAS_FORMAT ivas_format;
87 : int16_t sba_order;
88 : int16_t *element_mode;
89 : int32_t ivas_total_brate;
90 : DIRAC_CONFIG_DATA_HANDLE hConfig;
91 : IVAS_QMETADATA_HANDLE hQMetaData;
92 : int32_t Fs;
93 : int16_t *band_grouping;
94 : ivas_error error;
95 : int16_t spar_dirac_split_band;
96 : IVAS_FB_MIXER_HANDLE hFbMdft;
97 : int16_t *dirac_to_spar_md_bands;
98 :
99 368728 : error = IVAS_ERR_OK;
100 :
101 368728 : if ( enc_dec == ENC )
102 : {
103 21895 : ivas_format = ( (Encoder_Struct *) st_ivas )->hEncoderConfig->ivas_format;
104 21895 : element_mode = &( (Encoder_Struct *) st_ivas )->hEncoderConfig->element_mode_init;
105 21895 : sba_order = ( (Encoder_Struct *) st_ivas )->sba_analysis_order;
106 21895 : ivas_total_brate = ( (Encoder_Struct *) st_ivas )->hEncoderConfig->ivas_total_brate;
107 21895 : Fs = ( (Encoder_Struct *) st_ivas )->hEncoderConfig->input_Fs;
108 21895 : band_grouping = ( (Encoder_Struct *) st_ivas )->hDirAC->band_grouping;
109 21895 : hConfig = ( (Encoder_Struct *) st_ivas )->hDirAC->hConfig;
110 21895 : hQMetaData = ( (Encoder_Struct *) st_ivas )->hQMetaData;
111 21895 : if ( ( (Encoder_Struct *) st_ivas )->hSpar != NULL )
112 : {
113 21895 : hFbMdft = ( (Encoder_Struct *) st_ivas )->hSpar->hFbMixer;
114 21895 : dirac_to_spar_md_bands = ( (Encoder_Struct *) st_ivas )->hSpar->dirac_to_spar_md_bands;
115 : }
116 : else
117 : {
118 0 : hFbMdft = NULL;
119 0 : dirac_to_spar_md_bands = NULL;
120 : }
121 : }
122 : else
123 : {
124 346833 : ivas_format = ( (Decoder_Struct *) st_ivas )->ivas_format;
125 346833 : element_mode = &( (Decoder_Struct *) st_ivas )->element_mode_init;
126 346833 : sba_order = ( (Decoder_Struct *) st_ivas )->sba_analysis_order;
127 346833 : ivas_total_brate = ( (Decoder_Struct *) st_ivas )->hDecoderConfig->ivas_total_brate;
128 346833 : Fs = ( (Decoder_Struct *) st_ivas )->hDecoderConfig->output_Fs;
129 346833 : band_grouping = ( (Decoder_Struct *) st_ivas )->hDirAC->band_grouping;
130 346833 : hConfig = ( (Decoder_Struct *) st_ivas )->hDirAC->hConfig;
131 346833 : hQMetaData = ( (Decoder_Struct *) st_ivas )->hQMetaData;
132 346833 : if ( ( (Decoder_Struct *) st_ivas )->hSpar != NULL )
133 : {
134 85275 : hFbMdft = ( (Decoder_Struct *) st_ivas )->hSpar->hFbMixer;
135 85275 : dirac_to_spar_md_bands = ( (Decoder_Struct *) st_ivas )->hSpar->dirac_to_spar_md_bands;
136 : }
137 : else
138 : {
139 261558 : hFbMdft = NULL;
140 261558 : dirac_to_spar_md_bands = NULL;
141 : }
142 346833 : ( (Decoder_Struct *) st_ivas )->hDirAC->hFbMdft = hFbMdft;
143 : }
144 :
145 368728 : if ( ivas_format == SBA_FORMAT || ivas_format == SBA_ISM_FORMAT )
146 : {
147 107170 : hConfig->nbands = IVAS_MAX_NUM_BANDS;
148 :
149 107170 : spar_dirac_split_band = min( IVAS_MAX_NUM_BANDS, SPAR_DIRAC_SPLIT_START_BAND );
150 :
151 107170 : if ( ivas_get_hodirac_flag( ivas_total_brate, sba_order ) )
152 : {
153 12717 : spar_dirac_split_band = 0;
154 : }
155 : }
156 : else
157 : {
158 261558 : hConfig->nbands = 5;
159 261558 : spar_dirac_split_band = 0;
160 : }
161 368728 : hConfig->enc_param_start_band = 0;
162 368728 : hConfig->dec_param_estim = FALSE;
163 368728 : if ( ivas_format == SBA_FORMAT || ivas_format == SBA_ISM_FORMAT ) /* skip for MASA decoder */
164 : {
165 107170 : if ( ( error = ivas_dirac_sba_config( hQMetaData, element_mode, ivas_total_brate, sba_order, hConfig->nbands - spar_dirac_split_band, ivas_format ) ) != IVAS_ERR_OK )
166 : {
167 0 : return error;
168 : }
169 :
170 107170 : if ( hQMetaData != NULL )
171 : {
172 107170 : if ( enc_dec == ENC )
173 : {
174 21895 : hConfig->nbands = hQMetaData->q_direction[0].cfg.nbands;
175 : }
176 107170 : hConfig->enc_param_start_band = hQMetaData->q_direction[0].cfg.start_band + spar_dirac_split_band;
177 : }
178 :
179 107170 : hConfig->dec_param_estim = TRUE;
180 107170 : if ( hConfig->dec_param_estim == TRUE )
181 : {
182 107170 : hConfig->enc_param_start_band = spar_dirac_split_band;
183 : }
184 :
185 107170 : if ( ivas_get_hodirac_flag( ivas_total_brate, sba_order ) )
186 : {
187 12717 : hConfig->dec_param_estim = FALSE;
188 12717 : hConfig->enc_param_start_band = 0;
189 :
190 12717 : set_c( (int8_t *) hQMetaData->twoDirBands, (int8_t) 1, hQMetaData->q_direction[0].cfg.nbands );
191 12717 : hQMetaData->numTwoDirBands = (uint8_t) hQMetaData->q_direction[0].cfg.nbands;
192 : }
193 : }
194 :
195 368728 : if ( ivas_format == SBA_FORMAT || ivas_format == SBA_ISM_FORMAT )
196 : {
197 107170 : ivas_dirac_config_bands( band_grouping, IVAS_MAX_NUM_BANDS, (int16_t) ( Fs * INV_CLDFB_BANDWIDTH + 0.5f ), dirac_to_spar_md_bands, hQMetaData->useLowerBandRes, hConfig->enc_param_start_band, hFbMdft, 1 );
198 : }
199 : else
200 : {
201 261558 : ivas_dirac_config_bands( band_grouping, hConfig->nbands, (int16_t) ( Fs * INV_CLDFB_BANDWIDTH + 0.5f ), NULL, 0, 0, hFbMdft, 1 );
202 : }
203 :
204 368728 : return error;
205 : }
206 :
207 :
208 : /*-------------------------------------------------------------------------
209 : * ivas_dirac_config_bands()
210 : *
211 : * DirAC Configuration freq. band function; used also in MASA decoder
212 : *------------------------------------------------------------------------*/
213 :
214 5428339 : void ivas_dirac_config_bands(
215 : int16_t *band_grouping, /* o : band grouping */
216 : const int16_t nbands, /* i : number of bands */
217 : const int16_t max_band, /* i : maximal band index +1 */
218 : int16_t *dirac_to_spar_md_bands,
219 : const int8_t useLowerBandRes,
220 : const int16_t enc_param_start_band,
221 : IVAS_FB_MIXER_HANDLE hFbMdft,
222 : const int8_t BandGroupLowRes )
223 : {
224 : int16_t i;
225 : {
226 5428339 : if ( nbands == 5 )
227 : {
228 276926 : mvs2s( DirAC_band_grouping_5, band_grouping, 5 + 1 );
229 : }
230 5151413 : else if ( nbands == 6 )
231 : {
232 0 : mvs2s( DirAC_band_grouping_6, band_grouping, 6 + 1 );
233 : }
234 5151413 : else if ( nbands == 12 )
235 : {
236 : int16_t band;
237 66968369 : for ( band = 0; band < DIRAC_MAX_NBANDS; band++ )
238 : {
239 61816956 : dirac_to_spar_md_bands[band] = band;
240 : }
241 5151413 : if ( hFbMdft != NULL )
242 : {
243 : int16_t sb, idx1, idx2, b;
244 :
245 5128166 : idx1 = -1;
246 5128166 : sb = 0;
247 242865366 : for ( b = 0; b < max_band; b++ )
248 : {
249 237737200 : idx2 = hFbMdft->pFb->fb_bin_to_band.p_cldfb_map_to_spar_band[b];
250 237737200 : if ( idx2 > idx1 )
251 : {
252 58040354 : band_grouping[sb++] = b;
253 58040354 : idx1 = idx2;
254 : }
255 : }
256 5128166 : band_grouping[sb] = max_band;
257 :
258 : /* set the remaining bands to max_band to avoid problems for the DirAC parameter estimation with bw < FB */
259 13753970 : for ( b = sb; b <= nbands; b++ )
260 : {
261 8625804 : band_grouping[b] = max_band;
262 : }
263 : }
264 : else
265 : {
266 23247 : mvs2s( DirAC_band_grouping_12, band_grouping, 12 + 1 );
267 : }
268 :
269 5151413 : if ( useLowerBandRes )
270 : {
271 569158 : int16_t step = DIRAC_LOW_BANDRES_STEP;
272 : int16_t reduced_band;
273 :
274 569158 : if ( BandGroupLowRes )
275 : {
276 74937 : for ( band = enc_param_start_band + 2, reduced_band = enc_param_start_band + 1; band <= DIRAC_MAX_NBANDS; band += step, reduced_band++ )
277 : {
278 49958 : band_grouping[reduced_band] = band_grouping[band];
279 : }
280 74937 : for ( ; reduced_band <= DIRAC_MAX_NBANDS; reduced_band++ )
281 : {
282 49958 : band_grouping[reduced_band] = max_band;
283 : }
284 : }
285 :
286 1707474 : for ( band = enc_param_start_band + ( DIRAC_MAX_NBANDS - enc_param_start_band ) / 2 - 1, reduced_band = DIRAC_MAX_NBANDS - 1; band >= enc_param_start_band; band--, reduced_band -= step )
287 : {
288 1138316 : dirac_to_spar_md_bands[reduced_band] = dirac_to_spar_md_bands[band];
289 1138316 : dirac_to_spar_md_bands[reduced_band - 1] = dirac_to_spar_md_bands[band];
290 : }
291 : }
292 : else
293 : {
294 : /* always code the last two fb bands together */
295 4582255 : band_grouping[DIRAC_MAX_NBANDS - 1] = max_band;
296 4582255 : dirac_to_spar_md_bands[DIRAC_MAX_NBANDS - 1] = DIRAC_MAX_NBANDS - 2;
297 : }
298 : }
299 : else
300 : {
301 0 : assert( 0 && "nbands must be 5 or 6!" );
302 : }
303 : }
304 :
305 : /* Limit the band range to band max */
306 74058264 : for ( i = 0; i < nbands + 1; i++ )
307 : {
308 68629925 : if ( band_grouping[i] > max_band )
309 : {
310 82702 : band_grouping[i] = max_band;
311 : }
312 : }
313 :
314 5428339 : return;
315 : }
316 :
317 : /*-------------------------------------------------------------------*
318 : * ivas_get_dirac_sba_max_md_bits()
319 : *
320 : * Return maximum SBA DirAC metadata bit-budget and nominal bit-budget
321 : *-------------------------------------------------------------------*/
322 :
323 215692 : void ivas_get_dirac_sba_max_md_bits(
324 : const int32_t sba_total_brate,
325 : int16_t *bits_frame_nominal,
326 : int16_t *metadata_max_bits,
327 : int16_t *qmetadata_max_bit_req,
328 : const int16_t nbands,
329 : IVAS_FORMAT ivas_format )
330 : {
331 215692 : if ( sba_total_brate <= IVAS_13k2 )
332 : {
333 21153 : *bits_frame_nominal = ACELP_9k60 / FRAMES_PER_SEC;
334 21153 : *metadata_max_bits = 70;
335 : }
336 194539 : else if ( sba_total_brate <= IVAS_16k4 )
337 : {
338 20208 : *bits_frame_nominal = ACELP_13k20 / FRAMES_PER_SEC;
339 20208 : *metadata_max_bits = 80;
340 : }
341 174331 : else if ( sba_total_brate <= IVAS_24k4 )
342 : {
343 14412 : *bits_frame_nominal = ACELP_16k40 / FRAMES_PER_SEC;
344 14412 : *metadata_max_bits = 103;
345 : /* OSBA needs an additional 5-bits safety margin to avoid acelp crashes */
346 14412 : if ( ivas_format == SBA_ISM_FORMAT )
347 : {
348 7172 : ( *metadata_max_bits ) -= 7;
349 : }
350 : }
351 159919 : else if ( sba_total_brate <= IVAS_32k )
352 : {
353 15189 : *bits_frame_nominal = ACELP_32k / FRAMES_PER_SEC;
354 15189 : *metadata_max_bits = 214;
355 : }
356 144730 : else if ( sba_total_brate <= IVAS_48k )
357 : {
358 14013 : *bits_frame_nominal = IVAS_48k / FRAMES_PER_SEC;
359 14013 : *metadata_max_bits = 240;
360 : }
361 130717 : else if ( sba_total_brate <= IVAS_64k )
362 : {
363 14500 : *bits_frame_nominal = IVAS_64k / FRAMES_PER_SEC;
364 14500 : *metadata_max_bits = 200;
365 : }
366 116217 : else if ( sba_total_brate <= IVAS_80k )
367 : {
368 15100 : *bits_frame_nominal = IVAS_80k / FRAMES_PER_SEC;
369 15100 : *metadata_max_bits = 200;
370 : }
371 101117 : else if ( sba_total_brate <= IVAS_96k )
372 : {
373 14964 : *bits_frame_nominal = IVAS_96k / FRAMES_PER_SEC;
374 14964 : *metadata_max_bits = 200;
375 : }
376 86153 : else if ( sba_total_brate <= IVAS_128k )
377 : {
378 15048 : *bits_frame_nominal = IVAS_128k / FRAMES_PER_SEC;
379 15048 : *metadata_max_bits = 250;
380 : }
381 : else
382 : {
383 71105 : *bits_frame_nominal = (int16_t) ( sba_total_brate / FRAMES_PER_SEC );
384 71105 : *metadata_max_bits = MAX16B; /* no limit */
385 : }
386 215692 : *metadata_max_bits = (int16_t) min( (float) MAX16B, ceilf( (float) *metadata_max_bits * nbands / 5 ) );
387 215692 : *qmetadata_max_bit_req = QMETADATA_MAXBIT_REQ_SBA >> 1;
388 :
389 215692 : return;
390 : }
391 :
392 :
393 : /*-------------------------------------------------------------------------
394 : * ivas_dirac_sba_config()
395 : *
396 : * DirAC Configuration function for SBA
397 : *------------------------------------------------------------------------*/
398 :
399 215692 : ivas_error ivas_dirac_sba_config(
400 : IVAS_QMETADATA_HANDLE hQMetaData, /* i/o: q_metadata handle */
401 : int16_t *element_mode, /* i/o: element mode of the core coder */
402 : int32_t sba_total_brate, /* i : SBA total bitrate */
403 : const int16_t sba_order, /* i : Ambisonic (SBA) order */
404 : const int16_t nbands, /* i : number of frequency bands */
405 : IVAS_FORMAT ivas_format )
406 : {
407 : int16_t nbands_coded;
408 : int16_t hodirac_flag;
409 : ivas_error error;
410 :
411 215692 : error = IVAS_ERR_OK;
412 215692 : hQMetaData->is_masa_ivas_format = 0;
413 215692 : hodirac_flag = ivas_get_hodirac_flag( sba_total_brate, sba_order );
414 :
415 : /* map the bitrate for SID frame */
416 215692 : if ( sba_total_brate == IVAS_SID_5k2 )
417 : {
418 0 : if ( *element_mode == IVAS_SCE )
419 : {
420 0 : sba_total_brate = ACELP_24k40;
421 : }
422 : else
423 : {
424 0 : sba_total_brate = ACELP_48k;
425 : }
426 : }
427 :
428 215692 : ivas_set_qmetadata_maxbit_req( hQMetaData, SBA_FORMAT );
429 :
430 215692 : if ( sba_total_brate <= IVAS_16k4 )
431 : {
432 41361 : hQMetaData->useLowerRes = 1;
433 : }
434 : else
435 : {
436 174331 : hQMetaData->useLowerRes = 0;
437 : }
438 :
439 215692 : nbands_coded = nbands;
440 215692 : if ( sba_total_brate <= (int32_t) ( (float) IVAS_192k / (float) SPAR_DIRAC_SPLIT_START_BAND ) )
441 : {
442 41361 : hQMetaData->useLowerBandRes = 1;
443 41361 : nbands_coded = nbands / 2 + nbands % 2;
444 : }
445 : else
446 : {
447 174331 : hQMetaData->useLowerBandRes = 0;
448 174331 : nbands_coded = nbands - 1; /* always combine the last two bands */
449 : }
450 :
451 : {
452 215692 : int16_t no_dirs = 1;
453 215692 : if ( hodirac_flag )
454 : {
455 25449 : no_dirs = 2;
456 : }
457 :
458 215692 : if ( ( error = ivas_qmetadata_allocate_memory( hQMetaData, nbands_coded, no_dirs, 0 ) ) != IVAS_ERR_OK )
459 : {
460 0 : return error;
461 : }
462 : }
463 :
464 215692 : ivas_get_dirac_sba_max_md_bits( sba_total_brate, &hQMetaData->bits_frame_nominal, &hQMetaData->metadata_max_bits, &hQMetaData->qmetadata_max_bit_req, hQMetaData->q_direction[0].cfg.nbands, ivas_format );
465 :
466 215692 : return error;
467 : }
468 :
469 :
470 : /*-------------------------------------------------------------------------
471 : * computeDirectionVectors()
472 : *
473 : *
474 : *------------------------------------------------------------------------*/
475 :
476 18743788 : void computeDirectionVectors(
477 : float *intensity_real_x,
478 : float *intensity_real_y,
479 : float *intensity_real_z,
480 : const int16_t enc_param_start_band,
481 : const int16_t num_frequency_bands,
482 : float *direction_vector_x,
483 : float *direction_vector_y,
484 : float *direction_vector_z )
485 : {
486 : int16_t i;
487 : float intensityNorm;
488 :
489 230928160 : for ( i = enc_param_start_band; i < enc_param_start_band + num_frequency_bands; ++i )
490 : {
491 212184372 : intensityNorm = *( intensity_real_x ) * *( intensity_real_x ) +
492 212184372 : *( intensity_real_y ) * *( intensity_real_y ) +
493 212184372 : *( intensity_real_z ) * *( intensity_real_z );
494 :
495 212184372 : if ( intensityNorm <= EPSILON )
496 : {
497 29522793 : intensityNorm = 1.0f;
498 29522793 : *( direction_vector_x++ ) = 1.0f;
499 29522793 : *( direction_vector_y++ ) = 0.0f;
500 29522793 : *( direction_vector_z++ ) = 0.0f;
501 29522793 : intensity_real_x++;
502 29522793 : intensity_real_y++;
503 29522793 : intensity_real_z++;
504 : }
505 : else
506 : {
507 182661579 : intensityNorm = sqrtf( 1.f / intensityNorm );
508 182661579 : *( direction_vector_x++ ) = *( intensity_real_x++ ) * intensityNorm;
509 182661579 : *( direction_vector_y++ ) = *( intensity_real_y++ ) * intensityNorm;
510 182661579 : *( direction_vector_z++ ) = *( intensity_real_z++ ) * intensityNorm;
511 : }
512 : }
513 :
514 18743788 : return;
515 : }
516 :
517 :
518 : /*-------------------------------------------------------------------------
519 : * computeDiffuseness()
520 : *
521 : *
522 : *------------------------------------------------------------------------*/
523 :
524 50334300 : void computeDiffuseness(
525 : float *buffer_intensity[DIRAC_NUM_DIMS][DIRAC_NO_COL_AVG_DIFF],
526 : const float *buffer_energy,
527 : const int16_t num_freq_bands,
528 : float *diffuseness )
529 : {
530 : float intensity_slow[DIRAC_NUM_DIMS * CLDFB_NO_CHANNELS_MAX];
531 : float intensity_slow_abs[CLDFB_NO_CHANNELS_MAX];
532 : float energy_slow[CLDFB_NO_CHANNELS_MAX];
533 : int16_t i, j, k;
534 50334300 : float tmp = 0;
535 : float *p_tmp;
536 : const float *p_tmp_c;
537 :
538 : /* Compute Intensity slow and energy slow */
539 :
540 50334300 : set_f( intensity_slow, 0.0f, DIRAC_NUM_DIMS * CLDFB_NO_CHANNELS_MAX );
541 50334300 : set_f( intensity_slow_abs, 0.0f, CLDFB_NO_CHANNELS_MAX );
542 50334300 : set_f( energy_slow, 0.0f, CLDFB_NO_CHANNELS_MAX );
543 :
544 1661031900 : for ( i = 0; i < DIRAC_NO_COL_AVG_DIFF; ++i )
545 : {
546 : /* Energy slow */
547 1610697600 : p_tmp_c = buffer_energy + i * num_freq_bands;
548 22293287680 : for ( k = 0; k < num_freq_bands; k++ )
549 : {
550 20682590080 : energy_slow[k] += *( p_tmp_c++ );
551 : }
552 :
553 : /* Intensity slow */
554 6442790400 : for ( j = 0; j < DIRAC_NUM_DIMS; ++j )
555 : {
556 4832092800 : p_tmp = buffer_intensity[j][i];
557 :
558 66879863040 : for ( k = 0; k < num_freq_bands; k++ )
559 : {
560 62047770240 : intensity_slow[j * num_freq_bands + k] += *( p_tmp++ );
561 : }
562 : }
563 : }
564 :
565 : /* intensity_slow.^2 + intensity_slow_abs*/
566 201337200 : for ( j = 0; j < DIRAC_NUM_DIMS; ++j )
567 : {
568 151002900 : p_tmp = intensity_slow + j * num_freq_bands;
569 :
570 2089995720 : for ( k = 0; k < num_freq_bands; k++ )
571 : {
572 1938992820 : *( p_tmp ) *= ( *p_tmp );
573 1938992820 : intensity_slow_abs[k] += *( p_tmp++ );
574 : }
575 : }
576 :
577 : /* Compute Diffuseness */
578 50334300 : p_tmp = intensity_slow_abs;
579 696665240 : for ( i = 0; i < num_freq_bands; ++i )
580 : {
581 646330940 : tmp = sqrtf( *( p_tmp++ ) ) / ( energy_slow[i] + EPSILON );
582 646330940 : tmp = 1.0f - tmp;
583 646330940 : diffuseness[i] = ( ( tmp < 1.0f ) ? ( ( tmp < 0.0f ) ? 0.f : tmp ) : 1.0f );
584 : }
585 :
586 50334300 : return;
587 : }
588 :
589 :
590 : /*-------------------------------------------------------------------*
591 : * deindex_azimuth()
592 : *
593 : * Deindex one azimuth index
594 : *-------------------------------------------------------------------*/
595 :
596 : /*! r: azimuth value */
597 236337616 : float deindex_azimuth(
598 : int16_t id_phi, /* i : index */
599 : const int16_t no_bits, /* i : number of bits for the spherical grid */
600 : const int16_t id_th, /* i : elevation index */
601 : const int16_t remap, /* i : remapping flag */
602 : const MC_LS_SETUP mc_format /* i : channel format if in MC-mode */
603 : )
604 : {
605 : int16_t flag_delta;
606 : float dd, delta_phi;
607 : float phi_hat;
608 :
609 236337616 : if ( ( mc_format != MC_LS_SETUP_INVALID ) && ( no_bits == 2 ) )
610 : {
611 7148 : if ( id_phi % 2 == 0 )
612 : {
613 5251 : phi_hat = cb_azi_chan[id_phi / 2];
614 : }
615 : else
616 : {
617 1897 : phi_hat = -cb_azi_chan[( id_phi + 1 ) / 2];
618 : }
619 7148 : return phi_hat;
620 : }
621 236330468 : flag_delta = ( id_th % 2 == 1 );
622 :
623 236330468 : if ( remap )
624 : {
625 14871492 : id_phi = ivas_qmetadata_dereorder_generic( id_phi ) + ( no_phi_masa[no_bits - 1][id_th] >> 1 );
626 : }
627 :
628 236330468 : delta_phi = 360 / (float) ( no_phi_masa[no_bits - 1][id_th] );
629 236330468 : if ( ( flag_delta == 1 ) && ( no_phi_masa[no_bits - 1][id_th] > 2 ) && mc_format == MC_LS_SETUP_INVALID )
630 : {
631 53054232 : dd = delta_phi / 2.0f;
632 : }
633 : else
634 : {
635 183276236 : dd = 0;
636 : }
637 :
638 236330468 : id_phi -= ( ( no_phi_masa[no_bits - 1][id_th] ) >> 1 );
639 236330468 : phi_hat = id_phi * delta_phi + dd + 180 - 180;
640 236330468 : if ( mc_format != MC_LS_SETUP_INVALID )
641 : {
642 7649085 : phi_hat = companding_azimuth( phi_hat, mc_format, ( id_th * delta_theta_masa[no_bits - 3] > MC_MASA_THR_ELEVATION ), -1 );
643 : }
644 :
645 236330468 : return phi_hat;
646 : }
647 :
648 :
649 : /*----------------------------------------------------------------
650 : * deindex_spherical_component()
651 : *
652 : * decoding the spherical index for one tile
653 : *-----------------------------------------------------------------*/
654 :
655 101007776 : void deindex_spherical_component(
656 : const uint16_t sph_idx, /* i : spherical index */
657 : float *az, /* o : decoded azimuth value */
658 : float *el, /* o : decoded elevation value */
659 : uint16_t *az_idx, /* o : azimuth index */
660 : uint16_t *el_idx, /* o : elevation index */
661 : const uint16_t no_bits, /* i : number of bits for the spherical grid */
662 : const MC_LS_SETUP mc_format /* i : channel format if in MC-mode */
663 : )
664 : {
665 101007776 : assert( sph_idx < ( 1 << no_bits ) );
666 101007776 : switch ( no_bits )
667 : {
668 0 : case 0:
669 0 : *az = 0;
670 0 : *el = 0;
671 0 : *az_idx = 0;
672 0 : *el_idx = 0;
673 0 : break;
674 9 : case 1:
675 9 : *az_idx = sph_idx;
676 9 : *az = ( *az_idx ) * ( -180.f );
677 9 : *el = 0;
678 9 : *el_idx = 0;
679 9 : break;
680 52519 : case 2:
681 52519 : *el = 0;
682 52519 : *el_idx = 0;
683 52519 : *az_idx = sph_idx;
684 52519 : *az = deindex_azimuth( *az_idx, no_bits, 0, 0, mc_format );
685 52519 : break;
686 100955248 : default:
687 100955248 : *el_idx = deindex_sph_idx_general( sph_idx, no_bits, el, az, az_idx, mc_format );
688 100955248 : break;
689 : }
690 :
691 101007776 : return;
692 : }
693 :
694 :
695 : /*----------------------------------------------------------------
696 : * calculate_hodirac_sector_parameters()
697 : *
698 : *
699 : *-----------------------------------------------------------------*/
700 :
701 754692 : void calculate_hodirac_sector_parameters(
702 : DIRAC_ENC_HANDLE hDirAC, /* i : DirAC handle */
703 : float RealBuffer[DIRAC_MAX_ANA_CHANS][DIRAC_NO_FB_BANDS_MAX], /* i : signal vector (L+1)^2 x N_bins, real part */
704 : float ImagBuffer[DIRAC_MAX_ANA_CHANS][DIRAC_NO_FB_BANDS_MAX], /* i : signal vector, imaginary part */
705 : const float beta, /* i : forgetting factor for average filtering */
706 : const int16_t *band_grouping, /* i : indices of band groups */
707 : const int16_t N_bands, /* i : number of bands (groups) */
708 : const int16_t enc_param_start_band, /* i : first band to process */
709 : float *azi, /* o : array of sector azimuth angles, flat */
710 : float *ele, /* o : array of sector elevation angles, flat */
711 : float *diff, /* o : array of sector diffuseness values, flat */
712 : float *ene /* o : array of sector energy values, flat */
713 : )
714 : {
715 : int16_t i_sec, i_bin, i_band;
716 : float p_real, p_imag, normI, energy, tmp_diff;
717 : float sec_I_vec_x[NUM_ANA_SECTORS];
718 : float sec_I_vec_y[NUM_ANA_SECTORS];
719 : float sec_I_vec_z[NUM_ANA_SECTORS];
720 :
721 :
722 2264076 : for ( i_sec = 0; i_sec < NUM_ANA_SECTORS; i_sec++ )
723 : {
724 :
725 1509384 : float *p_sec_I_vec_x = &sec_I_vec_x[i_sec];
726 1509384 : float *p_sec_I_vec_y = &sec_I_vec_y[i_sec];
727 1509384 : float *p_sec_I_vec_z = &sec_I_vec_z[i_sec];
728 :
729 1509384 : const float *p_c_weights = c_weights;
730 :
731 1509384 : float *p_ImagBuffer_0 = ImagBuffer[0];
732 1509384 : float *p_ImagBuffer_1 = ImagBuffer[1];
733 1509384 : float *p_ImagBuffer_2 = ImagBuffer[2];
734 1509384 : float *p_ImagBuffer_3 = ImagBuffer[3];
735 1509384 : float *p_ImagBuffer_4 = ImagBuffer[4];
736 1509384 : float *p_ImagBuffer_5 = ImagBuffer[5];
737 1509384 : float *p_ImagBuffer_6 = ImagBuffer[6];
738 1509384 : float *p_ImagBuffer_8 = ImagBuffer[8];
739 :
740 1509384 : float *p_RealBuffer_0 = RealBuffer[0];
741 1509384 : float *p_RealBuffer_1 = RealBuffer[1];
742 1509384 : float *p_RealBuffer_2 = RealBuffer[2];
743 1509384 : float *p_RealBuffer_3 = RealBuffer[3];
744 1509384 : float *p_RealBuffer_4 = RealBuffer[4];
745 1509384 : float *p_RealBuffer_5 = RealBuffer[5];
746 1509384 : float *p_RealBuffer_6 = RealBuffer[6];
747 1509384 : float *p_RealBuffer_8 = RealBuffer[8];
748 :
749 18112608 : for ( i_band = enc_param_start_band; i_band < N_bands; i_band++ )
750 : {
751 16603224 : float *p_azi = &azi[i_sec * N_bands + i_band];
752 16603224 : float *p_ele = &ele[i_sec * N_bands + i_band];
753 16603224 : float *p_ene = &ene[i_sec * N_bands + i_band];
754 :
755 16603224 : float *p_diff = &diff[i_sec * N_bands + i_band];
756 16603224 : float *p_azi_prev = &hDirAC->azi_prev[i_sec * N_bands + i_band];
757 16603224 : float *p_ele_prev = &hDirAC->ele_prev[i_sec * N_bands + i_band];
758 :
759 16603224 : float *p_energy_smth = &hDirAC->energy_smth[i_sec][i_band];
760 16603224 : float *p_sec_I_vec_smth_x = &hDirAC->sec_I_vec_smth_x[i_sec][i_band];
761 16603224 : float *p_sec_I_vec_smth_y = &hDirAC->sec_I_vec_smth_y[i_sec][i_band];
762 16603224 : float *p_sec_I_vec_smth_z = &hDirAC->sec_I_vec_smth_z[i_sec][i_band];
763 16603224 : *p_sec_I_vec_x = 0.f;
764 16603224 : *p_sec_I_vec_y = 0.f;
765 16603224 : *p_sec_I_vec_z = 0.f;
766 16603224 : energy = 0.f;
767 :
768 16603224 : if ( i_sec == 0 )
769 : {
770 161724332 : for ( i_bin = band_grouping[i_band]; i_bin < band_grouping[i_band + 1]; i_bin++ )
771 : {
772 153422720 : float w = *( p_c_weights++ );
773 : float sec_w_imag, sec_x_imag, sec_y_imag, sec_z_imag;
774 : float sec_w_real, sec_x_real, sec_y_real, sec_z_real;
775 :
776 153422720 : sec_w_imag = 1.772454f * *( p_ImagBuffer_0 ) + 1.772454f * *( p_ImagBuffer_1 );
777 153422720 : sec_x_imag = 1.772454f * *( p_ImagBuffer_3++ ) + 1.023326f * *( p_ImagBuffer_4++ );
778 153422720 : sec_y_imag = 0.590818f * *( p_ImagBuffer_0++ ) + 1.772454f * *( p_ImagBuffer_1++ ) - 0.590817f * *( p_ImagBuffer_6++ ) - 1.023326f * *( p_ImagBuffer_8++ );
779 153422720 : sec_z_imag = 1.772454f * *( p_ImagBuffer_2++ ) + 1.023326f * *( p_ImagBuffer_5++ );
780 :
781 153422720 : sec_w_real = 1.772454f * *( p_RealBuffer_0 ) + 1.772454f * *( p_RealBuffer_1 );
782 153422720 : sec_x_real = 1.772454f * *( p_RealBuffer_3++ ) + 1.023326f * *( p_RealBuffer_4++ );
783 153422720 : sec_y_real = 0.590818f * *( p_RealBuffer_0++ ) + 1.772454f * *( p_RealBuffer_1++ ) - 0.590817f * *( p_RealBuffer_6++ ) - 1.023326f * *( p_RealBuffer_8++ );
784 153422720 : sec_z_real = 1.772454f * *( p_RealBuffer_2++ ) + 1.023326f * *( p_RealBuffer_5++ );
785 :
786 153422720 : p_real = sec_w_real * w;
787 153422720 : p_imag = sec_w_imag * w;
788 :
789 153422720 : *p_sec_I_vec_x += p_real * sec_x_real + p_imag * sec_x_imag;
790 153422720 : *p_sec_I_vec_y += p_real * sec_y_real + p_imag * sec_y_imag;
791 153422720 : *p_sec_I_vec_z += p_real * sec_z_real + p_imag * sec_z_imag;
792 :
793 153422720 : energy += 0.5f * ( p_real * p_real + p_imag * p_imag + sec_x_real * sec_x_real + sec_x_imag * sec_x_imag +
794 153422720 : sec_y_real * sec_y_real + sec_y_imag * sec_y_imag +
795 153422720 : sec_z_real * sec_z_real + sec_z_imag * sec_z_imag );
796 : }
797 : }
798 : else
799 : {
800 161724332 : for ( i_bin = band_grouping[i_band]; i_bin < band_grouping[i_band + 1]; i_bin++ )
801 : {
802 153422720 : float w = *( p_c_weights++ );
803 : float sec_w_imag, sec_x_imag, sec_y_imag, sec_z_imag;
804 : float sec_w_real, sec_x_real, sec_y_real, sec_z_real;
805 :
806 153422720 : sec_w_imag = 1.772454f * *(p_ImagBuffer_0) -1.772454f * *( p_ImagBuffer_1 );
807 153422720 : sec_x_imag = 1.772454f * *( p_ImagBuffer_3++ ) - 1.023326f * *( p_ImagBuffer_4++ );
808 153422720 : sec_y_imag = -0.590818f * *( p_ImagBuffer_0++ ) + 1.772454f * *( p_ImagBuffer_1++ ) + 0.590817f * *( p_ImagBuffer_6++ ) + 1.023326f * *( p_ImagBuffer_8++ );
809 153422720 : sec_z_imag = 1.772454f * *( p_ImagBuffer_2++ ) - 1.023326f * *( p_ImagBuffer_5++ );
810 :
811 153422720 : sec_w_real = 1.772454f * *(p_RealBuffer_0) -1.772454f * *( p_RealBuffer_1 );
812 153422720 : sec_x_real = 1.772454f * *( p_RealBuffer_3++ ) - 1.023326f * *( p_RealBuffer_4++ );
813 153422720 : sec_y_real = -0.590818f * *( p_RealBuffer_0++ ) + 1.772454f * *( p_RealBuffer_1++ ) + 0.590817f * *( p_RealBuffer_6++ ) + 1.023326f * *( p_RealBuffer_8++ );
814 153422720 : sec_z_real = 1.772454f * *( p_RealBuffer_2++ ) - 1.023326f * *( p_RealBuffer_5++ );
815 :
816 153422720 : p_real = sec_w_real * w;
817 153422720 : p_imag = sec_w_imag * w;
818 :
819 153422720 : *p_sec_I_vec_x += p_real * sec_x_real + p_imag * sec_x_imag;
820 153422720 : *p_sec_I_vec_y += p_real * sec_y_real + p_imag * sec_y_imag;
821 153422720 : *p_sec_I_vec_z += p_real * sec_z_real + p_imag * sec_z_imag;
822 :
823 153422720 : energy += 0.5f * ( p_real * p_real + p_imag * p_imag + sec_x_real * sec_x_real + sec_x_imag * sec_x_imag +
824 153422720 : sec_y_real * sec_y_real + sec_y_imag * sec_y_imag +
825 153422720 : sec_z_real * sec_z_real + sec_z_imag * sec_z_imag );
826 : }
827 : }
828 16603224 : if ( hDirAC->firstrun_sector_params )
829 : {
830 8800 : *p_sec_I_vec_smth_x = *p_sec_I_vec_x;
831 8800 : *p_sec_I_vec_smth_y = *p_sec_I_vec_y;
832 8800 : *p_sec_I_vec_smth_z = *p_sec_I_vec_z;
833 8800 : *p_energy_smth = energy;
834 : }
835 : else
836 : {
837 16594424 : float w = ( 1.0f - beta );
838 16594424 : *p_sec_I_vec_smth_x = w * *p_sec_I_vec_x + beta * *p_sec_I_vec_smth_x;
839 16594424 : *p_sec_I_vec_smth_y = w * *p_sec_I_vec_y + beta * *p_sec_I_vec_smth_y;
840 16594424 : *p_sec_I_vec_smth_z = w * *p_sec_I_vec_z + beta * *p_sec_I_vec_smth_z;
841 16594424 : *p_energy_smth = w * energy + beta * *p_energy_smth;
842 : }
843 :
844 16603224 : if ( energy < EPSILON )
845 : {
846 126944 : *p_azi = 0.f;
847 126944 : *p_ele = 0.f;
848 126944 : *p_ene = 0.f;
849 126944 : *p_diff = 1.f;
850 : }
851 : else
852 : {
853 16476280 : normI = sqrtf( *p_sec_I_vec_smth_x * *p_sec_I_vec_smth_x +
854 16476280 : *p_sec_I_vec_smth_y * *p_sec_I_vec_smth_y +
855 16476280 : *p_sec_I_vec_smth_z * *p_sec_I_vec_smth_z );
856 16476280 : *p_azi = atan2f( *p_sec_I_vec_smth_y, *p_sec_I_vec_smth_x ) * _180_OVER_PI;
857 16476280 : *p_ele = asinf( *p_sec_I_vec_smth_z / ( normI + EPSILON ) ) * _180_OVER_PI;
858 16476280 : *p_ene = *p_energy_smth;
859 16476280 : *p_diff = 1.f - normI / ( *p_energy_smth + EPSILON );
860 : }
861 :
862 16603224 : tmp_diff = *p_diff;
863 :
864 16603224 : if ( tmp_diff < 0.0f )
865 : {
866 0 : *p_diff = 0.f;
867 : }
868 16603224 : if ( tmp_diff > 0.5f )
869 : {
870 5511146 : if ( hDirAC->firstrun_sector_params )
871 : {
872 2130 : *p_azi = 0.f;
873 2130 : *p_ele = 0.f;
874 : }
875 : else
876 : {
877 5509016 : *p_azi = 2.f * ( 1.f - tmp_diff ) * *p_azi + ( 2.f * tmp_diff - 1.f ) * *p_azi_prev;
878 5509016 : *p_ele = 2.f * ( 1.f - tmp_diff ) * *p_ele + ( 2.f * tmp_diff - 1.f ) * *p_ele_prev;
879 : }
880 : }
881 : else
882 : {
883 11092078 : *p_azi_prev = *p_azi;
884 11092078 : *p_ele_prev = *p_ele;
885 : }
886 : }
887 : }
888 :
889 754692 : hDirAC->firstrun_sector_params = 0;
890 :
891 754692 : return;
892 : }
893 :
894 :
895 : /*-----------------------------------------------------------------------*
896 : * Local functions
897 : *-----------------------------------------------------------------------*/
898 :
899 :
900 : /*-------------------------------------------------------------------*
901 : * deindex_sph_idx_general()
902 : *
903 : * deindex the spherical index for more than 2 bits for the spherical grid
904 : *----------------------------------------------------------------------*/
905 :
906 : /*! r: decoded elevation index */
907 100955248 : static uint16_t deindex_sph_idx_general(
908 : const int16_t idx_sph, /* i : spherical index */
909 : const int16_t no_bits, /* i : number of bits in the spherical grid*/
910 : float *theta_dec, /* o : decoded elevation value */
911 : float *phi_dec, /* o : decoded azimuth value */
912 : uint16_t *p_id_phi, /* o : decoded azimuth index */
913 : const MC_LS_SETUP mc_format /* i : channel format if in MC-mode */
914 : )
915 : {
916 : int16_t i;
917 : uint16_t id_th;
918 : int16_t sign_theta;
919 : int16_t id_phi;
920 : int16_t cum_n[250];
921 : int16_t no_th;
922 : const int16_t *n;
923 :
924 100955248 : id_th = 0;
925 100955248 : id_phi = 0;
926 100955248 : sign_theta = 1;
927 :
928 100955248 : no_th = no_theta_masa[no_bits - 3];
929 100955248 : n = no_phi_masa[no_bits - 1];
930 100955248 : if ( mc_format != MC_LS_SETUP_INVALID )
931 : {
932 : /* indexing */
933 :
934 2531610 : cum_n[0] = n[0];
935 2531610 : if ( idx_sph >= cum_n[0] )
936 : {
937 2967780 : for ( i = 1; i < no_th; i++ )
938 : {
939 2967780 : cum_n[i] = cum_n[i - 1] + n[i];
940 2967780 : if ( idx_sph < cum_n[i] )
941 : {
942 1977239 : id_th = i;
943 1977239 : id_phi = idx_sph - cum_n[i - 1];
944 1977239 : break;
945 : }
946 : }
947 : }
948 : }
949 : else
950 : {
951 : /* Starting from Equator, alternating positive and negative */
952 98423638 : cum_n[0] = n[0];
953 98423638 : if ( idx_sph >= cum_n[0] )
954 : {
955 83038747 : for ( i = 1; i < no_th; i++ )
956 : {
957 83038747 : cum_n[2 * i - 1] = cum_n[2 * i - 2] + n[i];
958 :
959 83038747 : if ( idx_sph < cum_n[2 * i - 1] )
960 : {
961 25550881 : id_th = i;
962 25550881 : sign_theta = 1;
963 25550881 : id_phi = idx_sph - cum_n[2 * i - 2];
964 25550881 : break;
965 : }
966 :
967 57487866 : cum_n[2 * i] = cum_n[2 * i - 1] + n[i];
968 :
969 57487866 : if ( idx_sph < cum_n[2 * i] )
970 : {
971 20469533 : id_th = i;
972 20469533 : sign_theta = -1;
973 20469533 : id_phi = idx_sph - cum_n[2 * i - 1];
974 20469533 : break;
975 : }
976 :
977 37018333 : if ( i == ( no_th - 1 ) )
978 : {
979 0 : id_th = i;
980 0 : sign_theta = -1;
981 0 : id_phi = 0; /* idx_sph - cum_n[2*i-1]; */
982 0 : break;
983 : }
984 : }
985 : }
986 : }
987 :
988 100955248 : if ( id_th == 0 )
989 : {
990 52957595 : id_phi = idx_sph;
991 : }
992 : else
993 : {
994 47997653 : if ( ( id_th == no_th - 1 ) && ( no_bits > 4 ) )
995 : {
996 246591 : id_phi = 0;
997 : }
998 : }
999 :
1000 100955248 : *theta_dec = id_th * delta_theta_masa[no_bits - 3];
1001 :
1002 100955248 : if ( *theta_dec >= 90 )
1003 : {
1004 243639 : *theta_dec = 90.f * sign_theta;
1005 243639 : *phi_dec = 0;
1006 243639 : *p_id_phi = 0;
1007 : }
1008 : else
1009 : {
1010 100711609 : *theta_dec *= sign_theta;
1011 :
1012 100711609 : *phi_dec = deindex_azimuth( id_phi, no_bits, id_th, 0, mc_format );
1013 100711609 : *p_id_phi = id_phi;
1014 : }
1015 :
1016 100955248 : return id_th;
1017 : }
|
