Support for 9 and 10-bit encoding
[x262.git] / common / arm / predict-c.c
1 /*****************************************************************************
2  * predict.c: h264 encoder
3  *****************************************************************************
4  * Copyright (C) 2009 x264 project
5  *
6  * Authors: David Conrad <lessen42@gmail.com>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
21  *****************************************************************************/
22
23 #include "common/common.h"
24 #include "predict.h"
25 #include "pixel.h"
26
27 void x264_predict_4x4_dc_armv6( uint8_t *src );
28 void x264_predict_4x4_h_armv6( uint8_t *src );
29 void x264_predict_4x4_ddr_armv6( uint8_t *src );
30 void x264_predict_4x4_ddl_neon( uint8_t *src );
31
32 void x264_predict_8x8c_dc_neon( uint8_t *src );
33 void x264_predict_8x8c_dc_top_neon( uint8_t *src );
34 void x264_predict_8x8c_dc_left_neon( uint8_t *src );
35 void x264_predict_8x8c_h_neon( uint8_t *src );
36 void x264_predict_8x8c_v_neon( uint8_t *src );
37 void x264_predict_8x8c_p_neon( uint8_t *src );
38
39 void x264_predict_8x8_dc_neon( uint8_t *src, uint8_t edge[33] );
40 void x264_predict_8x8_h_neon( uint8_t *src, uint8_t edge[33] );
41
42 void x264_predict_16x16_dc_neon( uint8_t *src );
43 void x264_predict_16x16_dc_top_neon( uint8_t *src );
44 void x264_predict_16x16_dc_left_neon( uint8_t *src );
45 void x264_predict_16x16_h_neon( uint8_t *src );
46 void x264_predict_16x16_v_neon( uint8_t *src );
47 void x264_predict_16x16_p_neon( uint8_t *src );
48
49 void x264_predict_4x4_init_arm( int cpu, x264_predict_t pf[12] )
50 {
51     if (!(cpu&X264_CPU_ARMV6))
52         return;
53
54 #if !X264_HIGH_BIT_DEPTH
55     pf[I_PRED_4x4_H]   = x264_predict_4x4_h_armv6;
56     pf[I_PRED_4x4_DC]  = x264_predict_4x4_dc_armv6;
57     pf[I_PRED_4x4_DDR] = x264_predict_4x4_ddr_armv6;
58
59     if (!(cpu&X264_CPU_NEON))
60         return;
61
62     pf[I_PRED_4x4_DDL] = x264_predict_4x4_ddl_neon;
63 #endif // !X264_HIGH_BIT_DEPTH
64 }
65
66 void x264_predict_8x8c_init_arm( int cpu, x264_predict_t pf[7] )
67 {
68     if (!(cpu&X264_CPU_NEON))
69         return;
70
71 #if !X264_HIGH_BIT_DEPTH
72     pf[I_PRED_CHROMA_DC]      = x264_predict_8x8c_dc_neon;
73     pf[I_PRED_CHROMA_DC_TOP]  = x264_predict_8x8c_dc_top_neon;
74     pf[I_PRED_CHROMA_DC_LEFT] = x264_predict_8x8c_dc_left_neon;
75     pf[I_PRED_CHROMA_H] = x264_predict_8x8c_h_neon;
76     pf[I_PRED_CHROMA_V] = x264_predict_8x8c_v_neon;
77     pf[I_PRED_CHROMA_P] = x264_predict_8x8c_p_neon;
78 #endif // !X264_HIGH_BIT_DEPTH
79 }
80
81 void x264_predict_8x8_init_arm( int cpu, x264_predict8x8_t pf[12], x264_predict_8x8_filter_t *predict_filter )
82 {
83     if (!(cpu&X264_CPU_NEON))
84         return;
85
86 #if !X264_HIGH_BIT_DEPTH
87     pf[I_PRED_8x8_DC]  = x264_predict_8x8_dc_neon;
88     pf[I_PRED_8x8_H]   = x264_predict_8x8_h_neon;
89 #endif // !X264_HIGH_BIT_DEPTH
90 }
91
92 void x264_predict_16x16_init_arm( int cpu, x264_predict_t pf[7] )
93 {
94     if (!(cpu&X264_CPU_NEON))
95         return;
96
97 #if !X264_HIGH_BIT_DEPTH
98     pf[I_PRED_16x16_DC ]    = x264_predict_16x16_dc_neon;
99     pf[I_PRED_16x16_DC_TOP] = x264_predict_16x16_dc_top_neon;
100     pf[I_PRED_16x16_DC_LEFT]= x264_predict_16x16_dc_left_neon;
101     pf[I_PRED_16x16_H ]     = x264_predict_16x16_h_neon;
102     pf[I_PRED_16x16_V ]     = x264_predict_16x16_v_neon;
103     pf[I_PRED_16x16_P ]     = x264_predict_16x16_p_neon;
104 #endif // !X264_HIGH_BIT_DEPTH
105 }