1cabdff1aSopenharmony_ci/* 2cabdff1aSopenharmony_ci * Copyright (c) 2021 Loongson Technology Corporation Limited 3cabdff1aSopenharmony_ci * Contributed by Hao Chen <chenhao@loongson.cn> 4cabdff1aSopenharmony_ci * 5cabdff1aSopenharmony_ci * This file is part of FFmpeg. 6cabdff1aSopenharmony_ci * 7cabdff1aSopenharmony_ci * FFmpeg is free software; you can redistribute it and/or 8cabdff1aSopenharmony_ci * modify it under the terms of the GNU Lesser General Public 9cabdff1aSopenharmony_ci * License as published by the Free Software Foundation; either 10cabdff1aSopenharmony_ci * version 2.1 of the License, or (at your option) any later version. 11cabdff1aSopenharmony_ci * 12cabdff1aSopenharmony_ci * FFmpeg is distributed in the hope that it will be useful, 13cabdff1aSopenharmony_ci * but WITHOUT ANY WARRANTY; without even the implied warranty of 14cabdff1aSopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15cabdff1aSopenharmony_ci * Lesser General Public License for more details. 16cabdff1aSopenharmony_ci * 17cabdff1aSopenharmony_ci * You should have received a copy of the GNU Lesser General Public 18cabdff1aSopenharmony_ci * License along with FFmpeg; if not, write to the Free Software 19cabdff1aSopenharmony_ci * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20cabdff1aSopenharmony_ci */ 21cabdff1aSopenharmony_ci 22cabdff1aSopenharmony_ci#include "libavutil/loongarch/cpu.h" 23cabdff1aSopenharmony_ci#include "libavcodec/h264pred.h" 24cabdff1aSopenharmony_ci#include "h264_intrapred_lasx.h" 25cabdff1aSopenharmony_ci 26cabdff1aSopenharmony_ciav_cold void ff_h264_pred_init_loongarch(H264PredContext *h, int codec_id, 27cabdff1aSopenharmony_ci const int bit_depth, 28cabdff1aSopenharmony_ci const int chroma_format_idc) 29cabdff1aSopenharmony_ci{ 30cabdff1aSopenharmony_ci int cpu_flags = av_get_cpu_flags(); 31cabdff1aSopenharmony_ci 32cabdff1aSopenharmony_ci if (bit_depth == 8) { 33cabdff1aSopenharmony_ci if (have_lasx(cpu_flags)) { 34cabdff1aSopenharmony_ci if (chroma_format_idc <= 1) { 35cabdff1aSopenharmony_ci } 36cabdff1aSopenharmony_ci if (codec_id == AV_CODEC_ID_VP7 || codec_id == AV_CODEC_ID_VP8) { 37cabdff1aSopenharmony_ci } else { 38cabdff1aSopenharmony_ci if (chroma_format_idc <= 1) { 39cabdff1aSopenharmony_ci } 40cabdff1aSopenharmony_ci if (codec_id == AV_CODEC_ID_SVQ3) { 41cabdff1aSopenharmony_ci h->pred16x16[PLANE_PRED8x8] = ff_h264_pred16x16_plane_svq3_8_lasx; 42cabdff1aSopenharmony_ci } else if (codec_id == AV_CODEC_ID_RV40) { 43cabdff1aSopenharmony_ci h->pred16x16[PLANE_PRED8x8] = ff_h264_pred16x16_plane_rv40_8_lasx; 44cabdff1aSopenharmony_ci } else { 45cabdff1aSopenharmony_ci h->pred16x16[PLANE_PRED8x8] = ff_h264_pred16x16_plane_h264_8_lasx; 46cabdff1aSopenharmony_ci } 47cabdff1aSopenharmony_ci } 48cabdff1aSopenharmony_ci } 49cabdff1aSopenharmony_ci } 50cabdff1aSopenharmony_ci} 51