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/attributes.h"
23#include "libavutil/mips/cpu.h"
24#include "config.h"
25#include "h264dsp_mips.h"
26#include "h264pred_mips.h"
27
28av_cold void ff_h264_pred_init_mips(H264PredContext *h, int codec_id,
29                                    int bit_depth,
30                                    const int chroma_format_idc)
31{
32    int cpu_flags = av_get_cpu_flags();
33
34    if (have_mmi(cpu_flags)) {
35        if (bit_depth == 8) {
36            if (chroma_format_idc == 1) {
37                h->pred8x8  [VERT_PRED8x8       ] = ff_pred8x8_vertical_8_mmi;
38                h->pred8x8  [HOR_PRED8x8        ] = ff_pred8x8_horizontal_8_mmi;
39            } else {
40                h->pred8x8  [VERT_PRED8x8       ] = ff_pred8x16_vertical_8_mmi;
41                h->pred8x8  [HOR_PRED8x8        ] = ff_pred8x16_horizontal_8_mmi;
42            }
43
44            h->pred16x16[DC_PRED8x8             ] = ff_pred16x16_dc_8_mmi;
45            h->pred16x16[VERT_PRED8x8           ] = ff_pred16x16_vertical_8_mmi;
46            h->pred16x16[HOR_PRED8x8            ] = ff_pred16x16_horizontal_8_mmi;
47            h->pred8x8l [TOP_DC_PRED            ] = ff_pred8x8l_top_dc_8_mmi;
48            h->pred8x8l [DC_PRED                ] = ff_pred8x8l_dc_8_mmi;
49
50    #if ARCH_MIPS64
51            switch (codec_id) {
52            case AV_CODEC_ID_SVQ3:
53                h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_svq3_8_mmi;
54                break;
55            case AV_CODEC_ID_RV40:
56                h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_rv40_8_mmi;
57                break;
58            case AV_CODEC_ID_VP7:
59            case AV_CODEC_ID_VP8:
60                break;
61            default:
62                h->pred16x16[PLANE_PRED8x8      ] = ff_pred16x16_plane_h264_8_mmi;
63                break;
64            }
65    #endif
66
67            if (codec_id == AV_CODEC_ID_SVQ3 || codec_id == AV_CODEC_ID_H264) {
68                if (chroma_format_idc == 1) {
69                    h->pred8x8[TOP_DC_PRED8x8   ] = ff_pred8x8_top_dc_8_mmi;
70                    h->pred8x8[DC_PRED8x8       ] = ff_pred8x8_dc_8_mmi;
71                }
72            }
73        }
74    }
75
76    if (have_msa(cpu_flags)) {
77        if (8 == bit_depth) {
78            if (chroma_format_idc == 1) {
79                h->pred8x8[VERT_PRED8x8] = ff_h264_intra_pred_vert_8x8_msa;
80                h->pred8x8[HOR_PRED8x8] = ff_h264_intra_pred_horiz_8x8_msa;
81            }
82
83            if (codec_id != AV_CODEC_ID_VP7 && codec_id != AV_CODEC_ID_VP8) {
84                if (chroma_format_idc == 1) {
85                    h->pred8x8[PLANE_PRED8x8] = ff_h264_intra_predict_plane_8x8_msa;
86                }
87            }
88            if (codec_id != AV_CODEC_ID_RV40 && codec_id != AV_CODEC_ID_VP7
89                && codec_id != AV_CODEC_ID_VP8) {
90                if (chroma_format_idc == 1) {
91                    h->pred8x8[DC_PRED8x8] = ff_h264_intra_predict_dc_4blk_8x8_msa;
92                    h->pred8x8[LEFT_DC_PRED8x8] =
93                        ff_h264_intra_predict_hor_dc_8x8_msa;
94                    h->pred8x8[TOP_DC_PRED8x8] =
95                        ff_h264_intra_predict_vert_dc_8x8_msa;
96                    h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8] =
97                        ff_h264_intra_predict_mad_cow_dc_l0t_8x8_msa;
98                    h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8] =
99                        ff_h264_intra_predict_mad_cow_dc_0lt_8x8_msa;
100                    h->pred8x8[ALZHEIMER_DC_L00_PRED8x8] =
101                        ff_h264_intra_predict_mad_cow_dc_l00_8x8_msa;
102                    h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8] =
103                        ff_h264_intra_predict_mad_cow_dc_0l0_8x8_msa;
104                }
105            } else {
106                if (codec_id == AV_CODEC_ID_VP7 || codec_id == AV_CODEC_ID_VP8) {
107                    h->pred8x8[7] = ff_vp8_pred8x8_127_dc_8_msa;
108                    h->pred8x8[8] = ff_vp8_pred8x8_129_dc_8_msa;
109                }
110            }
111
112            if (chroma_format_idc == 1) {
113                h->pred8x8[DC_128_PRED8x8] = ff_h264_intra_pred_dc_128_8x8_msa;
114            }
115
116            h->pred16x16[DC_PRED8x8] = ff_h264_intra_pred_dc_16x16_msa;
117            h->pred16x16[VERT_PRED8x8] = ff_h264_intra_pred_vert_16x16_msa;
118            h->pred16x16[HOR_PRED8x8] = ff_h264_intra_pred_horiz_16x16_msa;
119
120            switch (codec_id) {
121            case AV_CODEC_ID_SVQ3:
122            case AV_CODEC_ID_RV40:
123                break;
124            case AV_CODEC_ID_VP7:
125            case AV_CODEC_ID_VP8:
126                h->pred16x16[7] = ff_vp8_pred16x16_127_dc_8_msa;
127                h->pred16x16[8] = ff_vp8_pred16x16_129_dc_8_msa;
128                break;
129            default:
130                h->pred16x16[PLANE_PRED8x8] =
131                    ff_h264_intra_predict_plane_16x16_msa;
132                break;
133            }
134
135            h->pred16x16[LEFT_DC_PRED8x8] = ff_h264_intra_pred_dc_left_16x16_msa;
136            h->pred16x16[TOP_DC_PRED8x8] = ff_h264_intra_pred_dc_top_16x16_msa;
137            h->pred16x16[DC_128_PRED8x8] = ff_h264_intra_pred_dc_128_16x16_msa;
138        }
139    }
140}
141