]> git.sesse.net Git - ffmpeg/blob - libavcodec/mips/h264pred_init_mips.c
avcodec/avcodec: Move decoder channel count check to ff_decode_preinit
[ffmpeg] / libavcodec / mips / h264pred_init_mips.c
1 /*
2  * Copyright (c) 2015 Shivraj Patil (Shivraj.Patil@imgtec.com)
3  *                    Zhou Xiaoyong <zhouxiaoyong@loongson.cn>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "libavutil/mips/cpu.h"
23 #include "config.h"
24 #include "h264dsp_mips.h"
25 #include "h264pred_mips.h"
26
27 av_cold void ff_h264_pred_init_mips(H264PredContext *h, int codec_id,
28                                     int bit_depth,
29                                     const int chroma_format_idc)
30 {
31     int cpu_flags = av_get_cpu_flags();
32
33     if (have_mmi(cpu_flags)) {
34         if (bit_depth == 8) {
35             if (chroma_format_idc == 1) {
36                 h->pred8x8  [VERT_PRED8x8       ] = ff_pred8x8_vertical_8_mmi;
37                 h->pred8x8  [HOR_PRED8x8        ] = ff_pred8x8_horizontal_8_mmi;
38             } else {
39                 h->pred8x8  [VERT_PRED8x8       ] = ff_pred8x16_vertical_8_mmi;
40                 h->pred8x8  [HOR_PRED8x8        ] = ff_pred8x16_horizontal_8_mmi;
41             }
42
43             h->pred16x16[DC_PRED8x8             ] = ff_pred16x16_dc_8_mmi;
44             h->pred16x16[VERT_PRED8x8           ] = ff_pred16x16_vertical_8_mmi;
45             h->pred16x16[HOR_PRED8x8            ] = ff_pred16x16_horizontal_8_mmi;
46             h->pred8x8l [TOP_DC_PRED            ] = ff_pred8x8l_top_dc_8_mmi;
47             h->pred8x8l [DC_PRED                ] = ff_pred8x8l_dc_8_mmi;
48
49     #if ARCH_MIPS64
50             switch (codec_id) {
51             case AV_CODEC_ID_SVQ3:
52                 h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_svq3_8_mmi;
53                 break;
54             case AV_CODEC_ID_RV40:
55                 h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_rv40_8_mmi;
56                 break;
57             case AV_CODEC_ID_VP7:
58             case AV_CODEC_ID_VP8:
59                 break;
60             default:
61                 h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_h264_8_mmi;
62                 break;
63             }
64     #endif
65
66             if (codec_id == AV_CODEC_ID_SVQ3 || codec_id == AV_CODEC_ID_H264) {
67                 if (chroma_format_idc == 1) {
68                     h->pred8x8[TOP_DC_PRED8x8   ] = ff_pred8x8_top_dc_8_mmi;
69                     h->pred8x8[DC_PRED8x8       ] = ff_pred8x8_dc_8_mmi;
70                 }
71             }
72         }
73     }
74
75     if (have_msa(cpu_flags)) {
76         if (8 == bit_depth) {
77             if (chroma_format_idc == 1) {
78                 h->pred8x8[VERT_PRED8x8] = ff_h264_intra_pred_vert_8x8_msa;
79                 h->pred8x8[HOR_PRED8x8] = ff_h264_intra_pred_horiz_8x8_msa;
80             }
81
82             if (codec_id != AV_CODEC_ID_VP7 && codec_id != AV_CODEC_ID_VP8) {
83                 if (chroma_format_idc == 1) {
84                     h->pred8x8[PLANE_PRED8x8] = ff_h264_intra_predict_plane_8x8_msa;
85                 }
86             }
87             if (codec_id != AV_CODEC_ID_RV40 && codec_id != AV_CODEC_ID_VP7
88                 && codec_id != AV_CODEC_ID_VP8) {
89                 if (chroma_format_idc == 1) {
90                     h->pred8x8[DC_PRED8x8] = ff_h264_intra_predict_dc_4blk_8x8_msa;
91                     h->pred8x8[LEFT_DC_PRED8x8] =
92                         ff_h264_intra_predict_hor_dc_8x8_msa;
93                     h->pred8x8[TOP_DC_PRED8x8] =
94                         ff_h264_intra_predict_vert_dc_8x8_msa;
95                     h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8] =
96                         ff_h264_intra_predict_mad_cow_dc_l0t_8x8_msa;
97                     h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8] =
98                         ff_h264_intra_predict_mad_cow_dc_0lt_8x8_msa;
99                     h->pred8x8[ALZHEIMER_DC_L00_PRED8x8] =
100                         ff_h264_intra_predict_mad_cow_dc_l00_8x8_msa;
101                     h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8] =
102                         ff_h264_intra_predict_mad_cow_dc_0l0_8x8_msa;
103                 }
104             } else {
105                 if (codec_id == AV_CODEC_ID_VP7 || codec_id == AV_CODEC_ID_VP8) {
106                     h->pred8x8[7] = ff_vp8_pred8x8_127_dc_8_msa;
107                     h->pred8x8[8] = ff_vp8_pred8x8_129_dc_8_msa;
108                 }
109             }
110
111             if (chroma_format_idc == 1) {
112                 h->pred8x8[DC_128_PRED8x8] = ff_h264_intra_pred_dc_128_8x8_msa;
113             }
114
115             h->pred16x16[DC_PRED8x8] = ff_h264_intra_pred_dc_16x16_msa;
116             h->pred16x16[VERT_PRED8x8] = ff_h264_intra_pred_vert_16x16_msa;
117             h->pred16x16[HOR_PRED8x8] = ff_h264_intra_pred_horiz_16x16_msa;
118
119             switch (codec_id) {
120             case AV_CODEC_ID_SVQ3:
121             case AV_CODEC_ID_RV40:
122                 break;
123             case AV_CODEC_ID_VP7:
124             case AV_CODEC_ID_VP8:
125                 h->pred16x16[7] = ff_vp8_pred16x16_127_dc_8_msa;
126                 h->pred16x16[8] = ff_vp8_pred16x16_129_dc_8_msa;
127                 break;
128             default:
129                 h->pred16x16[PLANE_PRED8x8] =
130                     ff_h264_intra_predict_plane_16x16_msa;
131                 break;
132             }
133
134             h->pred16x16[LEFT_DC_PRED8x8] = ff_h264_intra_pred_dc_left_16x16_msa;
135             h->pred16x16[TOP_DC_PRED8x8] = ff_h264_intra_pred_dc_top_16x16_msa;
136             h->pred16x16[DC_128_PRED8x8] = ff_h264_intra_pred_dc_128_16x16_msa;
137         }
138     }
139 }