git.sesse.net Git - ffmpeg/blob - 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 }