xref: /aosp_15_r20/external/libavc/encoder/ime_distortion_metrics.h (revision 495ae853bb871d1e5a258cb02c2cc13cde8ddb9a)
1 /******************************************************************************
2  *
3  * Copyright (C) 2015 The Android Open Source Project
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *****************************************************************************
18  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19 */
20 
21 /**
22 ******************************************************************************
23 * @file
24 *  ime_distortion_metrics.h
25 *
26 * @brief
27 *  This file contains declarations of routines that compute distortion
28 *  between two macro/sub blocks of identical dimensions
29 *
30 * @author
31 *  Ittiam
32 *
33 * @remarks
34 *  None
35 *
36 *******************************************************************************
37 */
38 
39 #ifndef _IME_DISTORTION_METRICS_H_
40 #define _IME_DISTORTION_METRICS_H_
41 
42 
43 /*****************************************************************************/
44 /* Type definitions for function prototypes                                  */
45 /*****************************************************************************/
46 
47 typedef void ime_compute_sad_ft(UWORD8 *pu1_src,
48                                 UWORD8 *pu1_est,
49                                 WORD32 src_strd,
50                                 WORD32 est_strd,
51                                 WORD32 i4_max_sad,
52                                 WORD32 *pi4_mb_distortion);
53 
54 typedef void ime_compute_sad4_diamond(UWORD8 *pu1_ref,
55                                       UWORD8 *pu1_src,
56                                       WORD32 ref_strd,
57                                       WORD32 src_strd,
58                                       WORD32 *pi4_sad);
59 
60 typedef void ime_compute_sad3_diamond(UWORD8 *pu1_ref1,
61                                       UWORD8 *pu1_ref2,
62                                       UWORD8 *pu1_ref3,
63                                       UWORD8 *pu1_src,
64                                       WORD32 ref_strd,
65                                       WORD32 src_strd,
66                                       WORD32 *pi4_sad);
67 
68 typedef void ime_compute_sad2_diamond(UWORD8 *pu1_ref1,
69                                       UWORD8 *pu1_ref2,
70                                       UWORD8 *pu1_src,
71                                       WORD32 ref_strd,
72                                       WORD32 src_strd,
73                                       WORD32 *pi4_sad);
74 
75 typedef void ime_sub_pel_compute_sad_16x16_ft(UWORD8 *pu1_src,
76                                               UWORD8 *pu1_ref_half_x,
77                                               UWORD8 *pu1_ref_half_y,
78                                               UWORD8 *pu1_ref_half_xy,
79                                               WORD32 src_strd,
80                                               WORD32 ref_strd,
81                                               WORD32 *pi4_sad);
82 
83 typedef void ime_compute_sad_stat(UWORD8 *pu1_src,
84                                   UWORD8 *pu1_est,
85                                   WORD32 src_strd,
86                                   WORD32 est_strd,
87                                   UWORD16 *pu2_thrsh,
88                                   WORD32 *pi4_mb_distortion,
89                                   UWORD32 *pu4_is_zero);
90 
91 typedef void ime_compute_satqd_16x16_lumainter_ft(UWORD8 *pu1_src,
92                                                   UWORD8 *pu1_est,
93                                                   WORD32 src_strd,
94                                                   WORD32 est_strd,
95                                                   UWORD16 *pu2_thrsh,
96                                                   WORD32 *pi4_mb_distortion,
97                                                   UWORD32 *pu4_is_zero);
98 
99 typedef void ime_compute_satqd_8x16_chroma_ft(UWORD8 *pu1_src,
100                                               UWORD8 *pu1_est,
101                                               WORD32 src_strd,
102                                               WORD32 est_strd,
103                                               WORD32 i4_max_sad,
104                                               UWORD16 *thrsh);
105 
106 typedef void ime_compute_satqd_16x16_lumaintra_ft(UWORD8 *pu1_src,
107                                                   UWORD8 *pu1_est,
108                                                   WORD32 src_strd,
109                                                   WORD32 est_strd,
110                                                   WORD32 i4_max_sad,
111                                                   UWORD16 *thrsh,
112                                                   WORD32 *pi4_mb_distortion,
113                                                   UWORD8 *sig_nz_sad);
114 
115 /*****************************************************************************/
116 /* Function Declarations                                                     */
117 /*****************************************************************************/
118 
119 /* C declarations */
120 ime_compute_sad_ft ime_compute_sad_16x16;
121 ime_compute_sad_ft ime_compute_sad_16x16_fast;
122 ime_compute_sad_ft ime_compute_sad_16x8;
123 ime_compute_sad_ft ime_compute_sad_16x16_ea8;
124 ime_compute_sad_ft ime_compute_sad_8x8;
125 ime_compute_sad_ft ime_compute_sad_4x4;
126 ime_compute_sad4_diamond ime_calculate_sad4_prog;
127 ime_compute_sad3_diamond ime_calculate_sad3_prog;
128 ime_compute_sad2_diamond ime_calculate_sad2_prog;
129 ime_sub_pel_compute_sad_16x16_ft ime_sub_pel_compute_sad_16x16;
130 ime_compute_sad_stat ime_compute_16x16_sad_stat;
131 ime_compute_satqd_16x16_lumainter_ft ime_compute_satqd_16x16_lumainter;
132 ime_compute_satqd_8x16_chroma_ft ime_compute_satqd_8x16_chroma;
133 ime_compute_satqd_16x16_lumaintra_ft ime_compute_satqd_16x16_lumaintra;
134 
135 /* SSE4.2 Declarations */
136 ime_compute_sad_ft ime_compute_sad_16x16_sse42;
137 ime_compute_sad_ft ime_compute_sad_16x16_fast_sse42;
138 ime_compute_sad_ft ime_compute_sad_16x8_sse42;
139 ime_compute_sad_ft ime_compute_sad_16x16_ea8_sse42;
140 ime_sub_pel_compute_sad_16x16_ft ime_sub_pel_compute_sad_16x16_sse42;
141 ime_compute_sad4_diamond ime_calculate_sad4_prog_sse42;
142 ime_compute_satqd_16x16_lumainter_ft ime_compute_satqd_16x16_lumainter_sse42;
143 
144 /* A9 Declarations */
145 ime_compute_sad_ft ime_compute_sad_16x16_a9q;
146 ime_compute_sad_ft ime_compute_sad_16x16_fast_a9q;
147 ime_compute_sad_ft ime_compute_sad_16x8_a9q;
148 ime_compute_sad_ft ime_compute_sad_16x16_ea8_a9q;
149 ime_compute_sad4_diamond ime_calculate_sad4_prog_a9q;
150 ime_compute_sad3_diamond ime_calculate_sad3_prog_a9q;
151 ime_compute_sad2_diamond ime_calculate_sad2_prog_a9q;
152 ime_sub_pel_compute_sad_16x16_ft ime_sub_pel_compute_sad_16x16_a9q;
153 ime_compute_sad_stat ime_compute_16x16_sad_stat_a9;
154 ime_compute_satqd_16x16_lumainter_ft ime_compute_satqd_16x16_lumainter_a9q;
155 
156 /* AV8 declarations */
157 ime_compute_sad_ft ime_compute_sad_16x16_av8;
158 ime_compute_sad_ft ime_compute_sad_16x16_fast_av8;
159 ime_compute_sad_ft ime_compute_sad_16x8_av8;
160 ime_compute_sad_ft ime_compute_sad_16x16_ea8_av8;
161 ime_compute_sad4_diamond ime_calculate_sad4_prog_av8;
162 ime_compute_sad3_diamond ime_calculate_sad3_prog_av8;
163 ime_compute_sad2_diamond ime_calculate_sad2_prog_av8;
164 ime_sub_pel_compute_sad_16x16_ft ime_sub_pel_compute_sad_16x16_av8;
165 ime_compute_sad_stat ime_compute_16x16_sad_stat_av8;
166 ime_compute_satqd_16x16_lumainter_ft ime_compute_satqd_16x16_lumainter_av8;
167 
168 #endif /* _IME_DISTORTION_METRICS_H_ */
169 
170 
171