ffmpeg/libavcodec/rv34dsp.h

   1 /*
   2  * RV30/40 decoder motion compensation functions
   3  * Copyright (c) 2008 Konstantin Shishkov
   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 /**
  23  * @file
  24  * RV30/40 decoder motion compensation functions
  25  */
  26
  27 #ifndef AVCODEC_RV34DSP_H
  28 #define AVCODEC_RV34DSP_H
  29
  30 #include "h264chroma.h"
  31 #include "qpeldsp.h"
  32
  33 typedef void (*rv40_weight_func)(uint8_t *dst/*align width (8 or 16)*/,
  34                                  uint8_t *src1/*align width (8 or 16)*/,
  35                                  uint8_t *src2/*align width (8 or 16)*/,
  36                                  int w1, int w2, ptrdiff_t stride);
  37
  38 typedef void (*rv34_inv_transform_func)(int16_t *block);
  39
  40 typedef void (*rv34_idct_add_func)(uint8_t *dst, ptrdiff_t stride, int16_t *block);
  41 typedef void (*rv34_idct_dc_add_func)(uint8_t *dst, ptrdiff_t stride,
  42                                       int   dc);
  43
  44 typedef void (*rv40_weak_loop_filter_func)(uint8_t *src, ptrdiff_t stride,
  45                                            int filter_p1, int filter_q1,
  46                                            int alpha, int beta,
  47                                            int lims, int lim_q1, int lim_p1);
  48
  49 typedef void (*rv40_strong_loop_filter_func)(uint8_t *src, ptrdiff_t stride,
  50                                              int alpha, int lims,
  51                                              int dmode, int chroma);
  52
  53 typedef int (*rv40_loop_filter_strength_func)(uint8_t *src, ptrdiff_t stride,
  54                                               int beta, int beta2, int edge,
  55                                               int *p1, int *q1);
  56
  57 typedef struct RV34DSPContext {
  58     qpel_mc_func put_pixels_tab[4][16];
  59     qpel_mc_func avg_pixels_tab[4][16];
  60     h264_chroma_mc_func put_chroma_pixels_tab[3];
  61     h264_chroma_mc_func avg_chroma_pixels_tab[3];
  62     /**
  63      * Biweight functions, first dimension is transform size (16/8),
  64      * second is whether the weight is prescaled by 1/512 to skip
  65      * the intermediate shifting.
  66      */
  67     rv40_weight_func rv40_weight_pixels_tab[2][2];
  68     rv34_inv_transform_func rv34_inv_transform;
  69     rv34_inv_transform_func rv34_inv_transform_dc;
  70     rv34_idct_add_func rv34_idct_add;
  71     rv34_idct_dc_add_func rv34_idct_dc_add;
  72     rv40_weak_loop_filter_func rv40_weak_loop_filter[2];
  73     rv40_strong_loop_filter_func rv40_strong_loop_filter[2];
  74     rv40_loop_filter_strength_func rv40_loop_filter_strength[2];
  75 } RV34DSPContext;
  76
  77 void ff_rv30dsp_init(RV34DSPContext *c);
  78 void ff_rv34dsp_init(RV34DSPContext *c);
  79 void ff_rv40dsp_init(RV34DSPContext *c);
  80
  81 void ff_rv34dsp_init_arm(RV34DSPContext *c);
  82 void ff_rv34dsp_init_x86(RV34DSPContext *c);
  83
  84 void ff_rv40dsp_init_aarch64(RV34DSPContext *c);
  85 void ff_rv40dsp_init_x86(RV34DSPContext *c);
  86 void ff_rv40dsp_init_arm(RV34DSPContext *c);
  87
  88 #endif /* AVCODEC_RV34DSP_H */