cabac.h
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
3  *
4  * This file is part of Libav.
5  *
6  * Libav is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * Libav is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with Libav; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 #ifndef AVCODEC_X86_CABAC_H
22 #define AVCODEC_X86_CABAC_H
23 
24 #include "libavcodec/cabac.h"
25 #include "libavutil/attributes.h"
26 #include "libavutil/x86/asm.h"
27 #include "libavutil/internal.h"
28 #include "config.h"
29 
30 #if HAVE_INLINE_ASM
31 
32 #ifdef BROKEN_RELOCATIONS
33 #define TABLES_ARG , "r"(tables)
34 
35 #if HAVE_FAST_CMOV
36 #define BRANCHLESS_GET_CABAC_UPDATE(ret, retq, low, range, tmp) \
37  "cmp "low" , "tmp" \n\t"\
38  "cmova %%ecx , "range" \n\t"\
39  "sbb %%rcx , %%rcx \n\t"\
40  "and %%ecx , "tmp" \n\t"\
41  "xor %%rcx , "retq" \n\t"\
42  "sub "tmp" , "low" \n\t"
43 #else /* HAVE_FAST_CMOV */
44 #define BRANCHLESS_GET_CABAC_UPDATE(ret, retq, low, range, tmp) \
45 /* P4 Prescott has crappy cmov,sbb,64bit shift so avoid them */ \
46  "sub "low" , "tmp" \n\t"\
47  "sar $31 , "tmp" \n\t"\
48  "sub %%ecx , "range" \n\t"\
49  "and "tmp" , "range" \n\t"\
50  "add %%ecx , "range" \n\t"\
51  "shl $17 , %%ecx \n\t"\
52  "and "tmp" , %%ecx \n\t"\
53  "sub %%ecx , "low" \n\t"\
54  "xor "tmp" , "ret" \n\t"\
55  "movslq "ret" , "retq" \n\t"
56 #endif /* HAVE_FAST_CMOV */
57 
58 #define BRANCHLESS_GET_CABAC(ret, retq, statep, low, lowword, range, rangeq, tmp, tmpbyte, byte, end, norm_off, lps_off, mlps_off, tables) \
59  "movzbl "statep" , "ret" \n\t"\
60  "mov "range" , "tmp" \n\t"\
61  "and $0xC0 , "range" \n\t"\
62  "lea ("ret", "range", 2), %%ecx \n\t"\
63  "movzbl "lps_off"("tables", %%rcx), "range" \n\t"\
64  "sub "range" , "tmp" \n\t"\
65  "mov "tmp" , %%ecx \n\t"\
66  "shl $17 , "tmp" \n\t"\
67  BRANCHLESS_GET_CABAC_UPDATE(ret, retq, low, range, tmp) \
68  "movzbl "norm_off"("tables", "rangeq"), %%ecx \n\t"\
69  "shl %%cl , "range" \n\t"\
70  "movzbl "mlps_off"+128("tables", "retq"), "tmp" \n\t"\
71  "shl %%cl , "low" \n\t"\
72  "mov "tmpbyte" , "statep" \n\t"\
73  "test "lowword" , "lowword" \n\t"\
74  "jnz 2f \n\t"\
75  "mov "byte" , %%"REG_c" \n\t"\
76  "cmp "end" , %%"REG_c" \n\t"\
77  "jge 1f \n\t"\
78  "add"OPSIZE" $2 , "byte" \n\t"\
79  "1: \n\t"\
80  "movzwl (%%"REG_c") , "tmp" \n\t"\
81  "lea -1("low") , %%ecx \n\t"\
82  "xor "low" , %%ecx \n\t"\
83  "shr $15 , %%ecx \n\t"\
84  "bswap "tmp" \n\t"\
85  "shr $15 , "tmp" \n\t"\
86  "movzbl "norm_off"("tables", %%rcx), %%ecx \n\t"\
87  "sub $0xFFFF , "tmp" \n\t"\
88  "neg %%ecx \n\t"\
89  "add $7 , %%ecx \n\t"\
90  "shl %%cl , "tmp" \n\t"\
91  "add "tmp" , "low" \n\t"\
92  "2: \n\t"
93 
94 #else /* BROKEN_RELOCATIONS */
95 #define TABLES_ARG
96 
97 #if HAVE_FAST_CMOV
98 #define BRANCHLESS_GET_CABAC_UPDATE(ret, low, range, tmp)\
99  "mov "tmp" , %%ecx \n\t"\
100  "shl $17 , "tmp" \n\t"\
101  "cmp "low" , "tmp" \n\t"\
102  "cmova %%ecx , "range" \n\t"\
103  "sbb %%ecx , %%ecx \n\t"\
104  "and %%ecx , "tmp" \n\t"\
105  "xor %%ecx , "ret" \n\t"\
106  "sub "tmp" , "low" \n\t"
107 #else /* HAVE_FAST_CMOV */
108 #define BRANCHLESS_GET_CABAC_UPDATE(ret, low, range, tmp)\
109  "mov "tmp" , %%ecx \n\t"\
110  "shl $17 , "tmp" \n\t"\
111  "sub "low" , "tmp" \n\t"\
112  "sar $31 , "tmp" \n\t" /*lps_mask*/\
113  "sub %%ecx , "range" \n\t" /*RangeLPS - range*/\
114  "and "tmp" , "range" \n\t" /*(RangeLPS - range)&lps_mask*/\
115  "add %%ecx , "range" \n\t" /*new range*/\
116  "shl $17 , %%ecx \n\t"\
117  "and "tmp" , %%ecx \n\t"\
118  "sub %%ecx , "low" \n\t"\
119  "xor "tmp" , "ret" \n\t"
120 #endif /* HAVE_FAST_CMOV */
121 
122 #define BRANCHLESS_GET_CABAC(ret, retq, statep, low, lowword, range, rangeq, tmp, tmpbyte, byte, end, norm_off, lps_off, mlps_off, tables) \
123  "movzbl "statep" , "ret" \n\t"\
124  "mov "range" , "tmp" \n\t"\
125  "and $0xC0 , "range" \n\t"\
126  "movzbl "MANGLE(ff_h264_cabac_tables)"+"lps_off"("ret", "range", 2), "range" \n\t"\
127  "sub "range" , "tmp" \n\t"\
128  BRANCHLESS_GET_CABAC_UPDATE(ret, low, range, tmp) \
129  "movzbl "MANGLE(ff_h264_cabac_tables)"+"norm_off"("range"), %%ecx \n\t"\
130  "shl %%cl , "range" \n\t"\
131  "movzbl "MANGLE(ff_h264_cabac_tables)"+"mlps_off"+128("ret"), "tmp" \n\t"\
132  "shl %%cl , "low" \n\t"\
133  "mov "tmpbyte" , "statep" \n\t"\
134  "test "lowword" , "lowword" \n\t"\
135  " jnz 2f \n\t"\
136  "mov "byte" , %%"REG_c" \n\t"\
137  "cmp "end" , %%"REG_c" \n\t"\
138  "jge 1f \n\t"\
139  "add"OPSIZE" $2 , "byte" \n\t"\
140  "1: \n\t"\
141  "movzwl (%%"REG_c") , "tmp" \n\t"\
142  "lea -1("low") , %%ecx \n\t"\
143  "xor "low" , %%ecx \n\t"\
144  "shr $15 , %%ecx \n\t"\
145  "bswap "tmp" \n\t"\
146  "shr $15 , "tmp" \n\t"\
147  "movzbl "MANGLE(ff_h264_cabac_tables)"+"norm_off"(%%ecx), %%ecx \n\t"\
148  "sub $0xFFFF , "tmp" \n\t"\
149  "neg %%ecx \n\t"\
150  "add $7 , %%ecx \n\t"\
151  "shl %%cl , "tmp" \n\t"\
152  "add "tmp" , "low" \n\t"\
153  "2: \n\t"
154 
155 #endif /* BROKEN_RELOCATIONS */
156 
157 
158 #if HAVE_7REGS
159 #define get_cabac_inline get_cabac_inline_x86
160 static av_always_inline int get_cabac_inline_x86(CABACContext *c,
161  uint8_t *const state)
162 {
163  int bit, tmp;
164 #ifdef BROKEN_RELOCATIONS
165  void *tables;
166 
167  __asm__ volatile(
168  "lea "MANGLE(ff_h264_cabac_tables)", %0 \n\t"
169  : "=&r"(tables)
170  );
171 #endif
172 
173  __asm__ volatile(
174  BRANCHLESS_GET_CABAC("%0", "%q0", "(%4)", "%1", "%w1",
175  "%2", "%q2", "%3", "%b3",
176  "%c6(%5)", "%c7(%5)",
180  "%8")
181  : "=&r"(bit), "+&r"(c->low), "+&r"(c->range), "=&q"(tmp)
182  : "r"(state), "r"(c),
183  "i"(offsetof(CABACContext, bytestream)),
184  "i"(offsetof(CABACContext, bytestream_end))
185  TABLES_ARG
186  : "%"REG_c, "memory"
187  );
188  return bit & 1;
189 }
190 #endif /* HAVE_7REGS */
191 
192 #define get_cabac_bypass_sign get_cabac_bypass_sign_x86
193 static av_always_inline int get_cabac_bypass_sign_x86(CABACContext *c, int val)
194 {
195  x86_reg tmp;
196  __asm__ volatile(
197  "movl %c6(%2), %k1 \n\t"
198  "movl %c3(%2), %%eax \n\t"
199  "shl $17, %k1 \n\t"
200  "add %%eax, %%eax \n\t"
201  "sub %k1, %%eax \n\t"
202  "cltd \n\t"
203  "and %%edx, %k1 \n\t"
204  "add %k1, %%eax \n\t"
205  "xor %%edx, %%ecx \n\t"
206  "sub %%edx, %%ecx \n\t"
207  "test %%ax, %%ax \n\t"
208  "jnz 1f \n\t"
209  "mov %c4(%2), %1 \n\t"
210  "subl $0xFFFF, %%eax \n\t"
211  "movzwl (%1), %%edx \n\t"
212  "bswap %%edx \n\t"
213  "shrl $15, %%edx \n\t"
214  "addl %%edx, %%eax \n\t"
215  "cmp %c5(%2), %1 \n\t"
216  "jge 1f \n\t"
217  "add"OPSIZE" $2, %c4(%2) \n\t"
218  "1: \n\t"
219  "movl %%eax, %c3(%2) \n\t"
220 
221  : "+c"(val), "=&r"(tmp)
222  : "r"(c),
223  "i"(offsetof(CABACContext, low)),
224  "i"(offsetof(CABACContext, bytestream)),
225  "i"(offsetof(CABACContext, bytestream_end)),
226  "i"(offsetof(CABACContext, range))
227  : "%eax", "%edx", "memory"
228  );
229  return val;
230 }
231 
232 #endif /* HAVE_INLINE_ASM */
233 #endif /* AVCODEC_X86_CABAC_H */
#define MANGLE(a)
Definition: asm.h:108
uint8_t ff_h264_cabac_tables[512+4 *2 *64+4 *64+63]
Definition: cabac.c:34
Macro definitions for various function/variable attributes.
uint8_t
static const uint8_t *const tables[]
common internal API header
#define H264_MLPS_STATE_OFFSET
Definition: cabac.h:36
static uint32_t state
Definition: trasher.c:27
#define H264_NORM_SHIFT_OFFSET
Definition: cabac.h:34
int x86_reg
Definition: asm.h:68
Context Adaptive Binary Arithmetic Coder.
#define H264_LPS_RANGE_OFFSET
Definition: cabac.h:35