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/attributes.h"
  23 #include "libavutil/mips/cpu.h"
  24 #include "config.h"
  25 #include "h264dsp_mips.h"
  26 #include "h264pred_mips.h"
  27
  28 av_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 }