xref: /aosp_15_r20/external/libhevc/encoder/sqrt_interp.c (revision c83a76b084498d55f252f48b2e3786804cdf24b7)
1*c83a76b0SSuyog Pawar /******************************************************************************
2*c83a76b0SSuyog Pawar  *
3*c83a76b0SSuyog Pawar  * Copyright (C) 2018 The Android Open Source Project
4*c83a76b0SSuyog Pawar  *
5*c83a76b0SSuyog Pawar  * Licensed under the Apache License, Version 2.0 (the "License");
6*c83a76b0SSuyog Pawar  * you may not use this file except in compliance with the License.
7*c83a76b0SSuyog Pawar  * You may obtain a copy of the License at:
8*c83a76b0SSuyog Pawar  *
9*c83a76b0SSuyog Pawar  * http://www.apache.org/licenses/LICENSE-2.0
10*c83a76b0SSuyog Pawar  *
11*c83a76b0SSuyog Pawar  * Unless required by applicable law or agreed to in writing, software
12*c83a76b0SSuyog Pawar  * distributed under the License is distributed on an "AS IS" BASIS,
13*c83a76b0SSuyog Pawar  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*c83a76b0SSuyog Pawar  * See the License for the specific language governing permissions and
15*c83a76b0SSuyog Pawar  * limitations under the License.
16*c83a76b0SSuyog Pawar  *
17*c83a76b0SSuyog Pawar  *****************************************************************************
18*c83a76b0SSuyog Pawar  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19*c83a76b0SSuyog Pawar */
20*c83a76b0SSuyog Pawar /*!
21*c83a76b0SSuyog Pawar ******************************************************************************
22*c83a76b0SSuyog Pawar * \file squrt_interp.c
23*c83a76b0SSuyog Pawar *
24*c83a76b0SSuyog Pawar * \brief
25*c83a76b0SSuyog Pawar *    This file contain square root interpolate function
26*c83a76b0SSuyog Pawar *
27*c83a76b0SSuyog Pawar * \date
28*c83a76b0SSuyog Pawar *
29*c83a76b0SSuyog Pawar * \author
30*c83a76b0SSuyog Pawar *    ittiam
31*c83a76b0SSuyog Pawar *
32*c83a76b0SSuyog Pawar ******************************************************************************
33*c83a76b0SSuyog Pawar */
34*c83a76b0SSuyog Pawar /*****************************************************************************/
35*c83a76b0SSuyog Pawar /* File Includes                                                             */
36*c83a76b0SSuyog Pawar /*****************************************************************************/
37*c83a76b0SSuyog Pawar /* System include files */
38*c83a76b0SSuyog Pawar #include <stdio.h>
39*c83a76b0SSuyog Pawar #include <stdlib.h>
40*c83a76b0SSuyog Pawar 
41*c83a76b0SSuyog Pawar /* User include files */
42*c83a76b0SSuyog Pawar #include "ia_type_def.h"
43*c83a76b0SSuyog Pawar #include "defs.h"
44*c83a76b0SSuyog Pawar /* #include "constants.h" */
45*c83a76b0SSuyog Pawar #include "ia_basic_ops32.h"
46*c83a76b0SSuyog Pawar /* #include "ia_basic_ops16.h" */
47*c83a76b0SSuyog Pawar #include "ia_basic_ops40.h"
48*c83a76b0SSuyog Pawar #include "sqrt_interp.h"
49*c83a76b0SSuyog Pawar /* #include "ia_enhaacplus_enc_basicops.h" */
50*c83a76b0SSuyog Pawar #include "common_rom.h"
51*c83a76b0SSuyog Pawar 
sqrtFix_interpolate(WORD32 num,WORD * q,const WORD32 * sqrt_tab)52*c83a76b0SSuyog Pawar WORD32 sqrtFix_interpolate(WORD32 num, WORD *q, const WORD32 *sqrt_tab)
53*c83a76b0SSuyog Pawar {
54*c83a76b0SSuyog Pawar     WORD32 index, answer, temp, delta, step_size;
55*c83a76b0SSuyog Pawar     WORD q_temp = *q;
56*c83a76b0SSuyog Pawar     WORD k;
57*c83a76b0SSuyog Pawar 
58*c83a76b0SSuyog Pawar     if(num == 0)
59*c83a76b0SSuyog Pawar         return 0;
60*c83a76b0SSuyog Pawar 
61*c83a76b0SSuyog Pawar     k = norm32(num);
62*c83a76b0SSuyog Pawar     temp = shr32(shl32(num, k), 21);
63*c83a76b0SSuyog Pawar     q_temp += k;
64*c83a76b0SSuyog Pawar     index = temp & 0x1FF; /* Leave leading 1 */
65*c83a76b0SSuyog Pawar     {
66*c83a76b0SSuyog Pawar         delta = shl32((shl32(num, k) - shl32(temp, 21)), 10);
67*c83a76b0SSuyog Pawar         step_size = sub32(sqrt_tab[index + 1], sqrt_tab[index]);
68*c83a76b0SSuyog Pawar         step_size = mult32_shl_sat(step_size, delta); /* Q:Q_SQRT_TAB + 21 + 10 - 31 */
69*c83a76b0SSuyog Pawar         answer = add32(sqrt_tab[index], step_size);
70*c83a76b0SSuyog Pawar     }
71*c83a76b0SSuyog Pawar     if(q_temp & 1)
72*c83a76b0SSuyog Pawar     {
73*c83a76b0SSuyog Pawar         q_temp -= 1;
74*c83a76b0SSuyog Pawar         answer = mult32_shl_sat(answer, INV_SQRT_2_Q31);
75*c83a76b0SSuyog Pawar     }
76*c83a76b0SSuyog Pawar 
77*c83a76b0SSuyog Pawar     q_temp = q_temp >> 1;
78*c83a76b0SSuyog Pawar     q_temp += (Q_SQRT_TAB);
79*c83a76b0SSuyog Pawar     *q = q_temp;
80*c83a76b0SSuyog Pawar 
81*c83a76b0SSuyog Pawar     answer >>= 1;
82*c83a76b0SSuyog Pawar     *q -= 1;
83*c83a76b0SSuyog Pawar 
84*c83a76b0SSuyog Pawar     return answer;
85*c83a76b0SSuyog Pawar }
86*c83a76b0SSuyog Pawar 
sqrtFix(WORD32 num,WORD * q,const WORD32 * sqrt_tab)87*c83a76b0SSuyog Pawar WORD32 sqrtFix(WORD32 num, WORD *q, const WORD32 *sqrt_tab)
88*c83a76b0SSuyog Pawar {
89*c83a76b0SSuyog Pawar     WORD32 index, answer, temp;
90*c83a76b0SSuyog Pawar     WORD k;
91*c83a76b0SSuyog Pawar     WORD q_temp = *q;
92*c83a76b0SSuyog Pawar 
93*c83a76b0SSuyog Pawar     if(num == 0)
94*c83a76b0SSuyog Pawar         return 0;
95*c83a76b0SSuyog Pawar 
96*c83a76b0SSuyog Pawar     k = norm32(num);
97*c83a76b0SSuyog Pawar     temp = shr32(shl32(num, k), 21);
98*c83a76b0SSuyog Pawar     q_temp += k;
99*c83a76b0SSuyog Pawar     index = temp & 0x1FF; /* Leave leading 1 */
100*c83a76b0SSuyog Pawar     answer = sqrt_tab[index];
101*c83a76b0SSuyog Pawar     if(q_temp & 1)
102*c83a76b0SSuyog Pawar     {
103*c83a76b0SSuyog Pawar         q_temp -= 1;
104*c83a76b0SSuyog Pawar         answer = mult32x16in32_shl(answer, INV_SQRT_2_Q15);
105*c83a76b0SSuyog Pawar     }
106*c83a76b0SSuyog Pawar     q_temp = q_temp >> 1;
107*c83a76b0SSuyog Pawar     q_temp += Q_SQRT_TAB;
108*c83a76b0SSuyog Pawar     *q = q_temp;
109*c83a76b0SSuyog Pawar     return answer;
110*c83a76b0SSuyog Pawar }
111*c83a76b0SSuyog Pawar 
112*c83a76b0SSuyog Pawar #ifdef ITT_C6678
113*c83a76b0SSuyog Pawar #pragma CODE_SECTION(sqrtFix_interpolate, "itt_varq_l1pram");
114*c83a76b0SSuyog Pawar #pragma CODE_SECTION(sqrtFix, "itt_varq_l1pram");
115*c83a76b0SSuyog Pawar #endif
116