xref: /aosp_15_r20/external/libopus/silk/shell_coder.c (revision a58d3d2adb790c104798cd88c8a3aff4fa8b82cc)
1*a58d3d2aSXin Li /***********************************************************************
2*a58d3d2aSXin Li Copyright (c) 2006-2011, Skype Limited. All rights reserved.
3*a58d3d2aSXin Li Redistribution and use in source and binary forms, with or without
4*a58d3d2aSXin Li modification, are permitted provided that the following conditions
5*a58d3d2aSXin Li are met:
6*a58d3d2aSXin Li - Redistributions of source code must retain the above copyright notice,
7*a58d3d2aSXin Li this list of conditions and the following disclaimer.
8*a58d3d2aSXin Li - Redistributions in binary form must reproduce the above copyright
9*a58d3d2aSXin Li notice, this list of conditions and the following disclaimer in the
10*a58d3d2aSXin Li documentation and/or other materials provided with the distribution.
11*a58d3d2aSXin Li - Neither the name of Internet Society, IETF or IETF Trust, nor the
12*a58d3d2aSXin Li names of specific contributors, may be used to endorse or promote
13*a58d3d2aSXin Li products derived from this software without specific prior written
14*a58d3d2aSXin Li permission.
15*a58d3d2aSXin Li THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16*a58d3d2aSXin Li AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17*a58d3d2aSXin Li IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18*a58d3d2aSXin Li ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
19*a58d3d2aSXin Li LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
20*a58d3d2aSXin Li CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
21*a58d3d2aSXin Li SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
22*a58d3d2aSXin Li INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23*a58d3d2aSXin Li CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
24*a58d3d2aSXin Li ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
25*a58d3d2aSXin Li POSSIBILITY OF SUCH DAMAGE.
26*a58d3d2aSXin Li ***********************************************************************/
27*a58d3d2aSXin Li 
28*a58d3d2aSXin Li #ifdef HAVE_CONFIG_H
29*a58d3d2aSXin Li #include "config.h"
30*a58d3d2aSXin Li #endif
31*a58d3d2aSXin Li 
32*a58d3d2aSXin Li #include "main.h"
33*a58d3d2aSXin Li 
34*a58d3d2aSXin Li /* shell coder; pulse-subframe length is hardcoded */
35*a58d3d2aSXin Li 
combine_pulses(opus_int * out,const opus_int * in,const opus_int len)36*a58d3d2aSXin Li static OPUS_INLINE void combine_pulses(
37*a58d3d2aSXin Li     opus_int         *out,   /* O    combined pulses vector [len] */
38*a58d3d2aSXin Li     const opus_int   *in,    /* I    input vector       [2 * len] */
39*a58d3d2aSXin Li     const opus_int   len     /* I    number of OUTPUT samples     */
40*a58d3d2aSXin Li )
41*a58d3d2aSXin Li {
42*a58d3d2aSXin Li     opus_int k;
43*a58d3d2aSXin Li     for( k = 0; k < len; k++ ) {
44*a58d3d2aSXin Li         out[ k ] = in[ 2 * k ] + in[ 2 * k + 1 ];
45*a58d3d2aSXin Li     }
46*a58d3d2aSXin Li }
47*a58d3d2aSXin Li 
encode_split(ec_enc * psRangeEnc,const opus_int p_child1,const opus_int p,const opus_uint8 * shell_table)48*a58d3d2aSXin Li static OPUS_INLINE void encode_split(
49*a58d3d2aSXin Li     ec_enc                      *psRangeEnc,    /* I/O  compressor data structure                   */
50*a58d3d2aSXin Li     const opus_int              p_child1,       /* I    pulse amplitude of first child subframe     */
51*a58d3d2aSXin Li     const opus_int              p,              /* I    pulse amplitude of current subframe         */
52*a58d3d2aSXin Li     const opus_uint8            *shell_table    /* I    table of shell cdfs                         */
53*a58d3d2aSXin Li )
54*a58d3d2aSXin Li {
55*a58d3d2aSXin Li     if( p > 0 ) {
56*a58d3d2aSXin Li         ec_enc_icdf( psRangeEnc, p_child1, &shell_table[ silk_shell_code_table_offsets[ p ] ], 8 );
57*a58d3d2aSXin Li     }
58*a58d3d2aSXin Li }
59*a58d3d2aSXin Li 
decode_split(opus_int16 * p_child1,opus_int16 * p_child2,ec_dec * psRangeDec,const opus_int p,const opus_uint8 * shell_table)60*a58d3d2aSXin Li static OPUS_INLINE void decode_split(
61*a58d3d2aSXin Li     opus_int16                  *p_child1,      /* O    pulse amplitude of first child subframe     */
62*a58d3d2aSXin Li     opus_int16                  *p_child2,      /* O    pulse amplitude of second child subframe    */
63*a58d3d2aSXin Li     ec_dec                      *psRangeDec,    /* I/O  Compressor data structure                   */
64*a58d3d2aSXin Li     const opus_int              p,              /* I    pulse amplitude of current subframe         */
65*a58d3d2aSXin Li     const opus_uint8            *shell_table    /* I    table of shell cdfs                         */
66*a58d3d2aSXin Li )
67*a58d3d2aSXin Li {
68*a58d3d2aSXin Li     if( p > 0 ) {
69*a58d3d2aSXin Li         p_child1[ 0 ] = ec_dec_icdf( psRangeDec, &shell_table[ silk_shell_code_table_offsets[ p ] ], 8 );
70*a58d3d2aSXin Li         p_child2[ 0 ] = p - p_child1[ 0 ];
71*a58d3d2aSXin Li     } else {
72*a58d3d2aSXin Li         p_child1[ 0 ] = 0;
73*a58d3d2aSXin Li         p_child2[ 0 ] = 0;
74*a58d3d2aSXin Li     }
75*a58d3d2aSXin Li }
76*a58d3d2aSXin Li 
77*a58d3d2aSXin Li /* Shell encoder, operates on one shell code frame of 16 pulses */
silk_shell_encoder(ec_enc * psRangeEnc,const opus_int * pulses0)78*a58d3d2aSXin Li void silk_shell_encoder(
79*a58d3d2aSXin Li     ec_enc                      *psRangeEnc,                    /* I/O  compressor data structure                   */
80*a58d3d2aSXin Li     const opus_int              *pulses0                        /* I    data: nonnegative pulse amplitudes          */
81*a58d3d2aSXin Li )
82*a58d3d2aSXin Li {
83*a58d3d2aSXin Li     opus_int pulses1[ 8 ], pulses2[ 4 ], pulses3[ 2 ], pulses4[ 1 ];
84*a58d3d2aSXin Li 
85*a58d3d2aSXin Li     /* this function operates on one shell code frame of 16 pulses */
86*a58d3d2aSXin Li     silk_assert( SHELL_CODEC_FRAME_LENGTH == 16 );
87*a58d3d2aSXin Li 
88*a58d3d2aSXin Li     /* tree representation per pulse-subframe */
89*a58d3d2aSXin Li     combine_pulses( pulses1, pulses0, 8 );
90*a58d3d2aSXin Li     combine_pulses( pulses2, pulses1, 4 );
91*a58d3d2aSXin Li     combine_pulses( pulses3, pulses2, 2 );
92*a58d3d2aSXin Li     combine_pulses( pulses4, pulses3, 1 );
93*a58d3d2aSXin Li 
94*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses3[  0 ], pulses4[ 0 ], silk_shell_code_table3 );
95*a58d3d2aSXin Li 
96*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses2[  0 ], pulses3[ 0 ], silk_shell_code_table2 );
97*a58d3d2aSXin Li 
98*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses1[  0 ], pulses2[ 0 ], silk_shell_code_table1 );
99*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[  0 ], pulses1[ 0 ], silk_shell_code_table0 );
100*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[  2 ], pulses1[ 1 ], silk_shell_code_table0 );
101*a58d3d2aSXin Li 
102*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses1[  2 ], pulses2[ 1 ], silk_shell_code_table1 );
103*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[  4 ], pulses1[ 2 ], silk_shell_code_table0 );
104*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[  6 ], pulses1[ 3 ], silk_shell_code_table0 );
105*a58d3d2aSXin Li 
106*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses2[  2 ], pulses3[ 1 ], silk_shell_code_table2 );
107*a58d3d2aSXin Li 
108*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses1[  4 ], pulses2[ 2 ], silk_shell_code_table1 );
109*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[  8 ], pulses1[ 4 ], silk_shell_code_table0 );
110*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[ 10 ], pulses1[ 5 ], silk_shell_code_table0 );
111*a58d3d2aSXin Li 
112*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses1[  6 ], pulses2[ 3 ], silk_shell_code_table1 );
113*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[ 12 ], pulses1[ 6 ], silk_shell_code_table0 );
114*a58d3d2aSXin Li     encode_split( psRangeEnc, pulses0[ 14 ], pulses1[ 7 ], silk_shell_code_table0 );
115*a58d3d2aSXin Li }
116*a58d3d2aSXin Li 
117*a58d3d2aSXin Li 
118*a58d3d2aSXin Li /* Shell decoder, operates on one shell code frame of 16 pulses */
silk_shell_decoder(opus_int16 * pulses0,ec_dec * psRangeDec,const opus_int pulses4)119*a58d3d2aSXin Li void silk_shell_decoder(
120*a58d3d2aSXin Li     opus_int16                  *pulses0,                       /* O    data: nonnegative pulse amplitudes          */
121*a58d3d2aSXin Li     ec_dec                      *psRangeDec,                    /* I/O  Compressor data structure                   */
122*a58d3d2aSXin Li     const opus_int              pulses4                         /* I    number of pulses per pulse-subframe         */
123*a58d3d2aSXin Li )
124*a58d3d2aSXin Li {
125*a58d3d2aSXin Li     opus_int16 pulses3[ 2 ], pulses2[ 4 ], pulses1[ 8 ];
126*a58d3d2aSXin Li 
127*a58d3d2aSXin Li     /* this function operates on one shell code frame of 16 pulses */
128*a58d3d2aSXin Li     silk_assert( SHELL_CODEC_FRAME_LENGTH == 16 );
129*a58d3d2aSXin Li 
130*a58d3d2aSXin Li     decode_split( &pulses3[  0 ], &pulses3[  1 ], psRangeDec, pulses4,      silk_shell_code_table3 );
131*a58d3d2aSXin Li 
132*a58d3d2aSXin Li     decode_split( &pulses2[  0 ], &pulses2[  1 ], psRangeDec, pulses3[ 0 ], silk_shell_code_table2 );
133*a58d3d2aSXin Li 
134*a58d3d2aSXin Li     decode_split( &pulses1[  0 ], &pulses1[  1 ], psRangeDec, pulses2[ 0 ], silk_shell_code_table1 );
135*a58d3d2aSXin Li     decode_split( &pulses0[  0 ], &pulses0[  1 ], psRangeDec, pulses1[ 0 ], silk_shell_code_table0 );
136*a58d3d2aSXin Li     decode_split( &pulses0[  2 ], &pulses0[  3 ], psRangeDec, pulses1[ 1 ], silk_shell_code_table0 );
137*a58d3d2aSXin Li 
138*a58d3d2aSXin Li     decode_split( &pulses1[  2 ], &pulses1[  3 ], psRangeDec, pulses2[ 1 ], silk_shell_code_table1 );
139*a58d3d2aSXin Li     decode_split( &pulses0[  4 ], &pulses0[  5 ], psRangeDec, pulses1[ 2 ], silk_shell_code_table0 );
140*a58d3d2aSXin Li     decode_split( &pulses0[  6 ], &pulses0[  7 ], psRangeDec, pulses1[ 3 ], silk_shell_code_table0 );
141*a58d3d2aSXin Li 
142*a58d3d2aSXin Li     decode_split( &pulses2[  2 ], &pulses2[  3 ], psRangeDec, pulses3[ 1 ], silk_shell_code_table2 );
143*a58d3d2aSXin Li 
144*a58d3d2aSXin Li     decode_split( &pulses1[  4 ], &pulses1[  5 ], psRangeDec, pulses2[ 2 ], silk_shell_code_table1 );
145*a58d3d2aSXin Li     decode_split( &pulses0[  8 ], &pulses0[  9 ], psRangeDec, pulses1[ 4 ], silk_shell_code_table0 );
146*a58d3d2aSXin Li     decode_split( &pulses0[ 10 ], &pulses0[ 11 ], psRangeDec, pulses1[ 5 ], silk_shell_code_table0 );
147*a58d3d2aSXin Li 
148*a58d3d2aSXin Li     decode_split( &pulses1[  6 ], &pulses1[  7 ], psRangeDec, pulses2[ 3 ], silk_shell_code_table1 );
149*a58d3d2aSXin Li     decode_split( &pulses0[ 12 ], &pulses0[ 13 ], psRangeDec, pulses1[ 6 ], silk_shell_code_table0 );
150*a58d3d2aSXin Li     decode_split( &pulses0[ 14 ], &pulses0[ 15 ], psRangeDec, pulses1[ 7 ], silk_shell_code_table0 );
151*a58d3d2aSXin Li }
152