git.sesse.net Git - ffmpeg/blob - libavcodec/me_cmp.h

   1 /*
   2  * This file is part of Libav.
   3  *
   4  * Libav is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * Libav is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with Libav; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 #ifndef AVCODEC_ME_CMP_H
  20 #define AVCODEC_ME_CMP_H
  21
  22 #include <stdint.h>
  23
  24 #include "avcodec.h"
  25
  26 extern uint32_t ff_square_tab[512];
  27
  28 struct MpegEncContext;
  29 /* Motion estimation:
  30  * h is limited to { width / 2, width, 2 * width },
  31  * but never larger than 16 and never smaller than 2.
  32  * Although currently h < 4 is not used as functions with
  33  * width < 8 are neither used nor implemented. */
  34 typedef int (*me_cmp_func)(struct MpegEncContext *c,
  35                            uint8_t *blk1 /* align width (8 or 16) */,
  36                            uint8_t *blk2 /* align 1 */, ptrdiff_t stride,
  37                            int h);
  38
  39 typedef struct MECmpContext {
  40     int (*sum_abs_dctelem)(int16_t *block /* align 16 */);
  41
  42     me_cmp_func sad[6]; /* identical to pix_absAxA except additional void * */
  43     me_cmp_func sse[6];
  44     me_cmp_func hadamard8_diff[6];
  45     me_cmp_func dct_sad[6];
  46     me_cmp_func quant_psnr[6];
  47     me_cmp_func bit[6];
  48     me_cmp_func rd[6];
  49     me_cmp_func vsad[6];
  50     me_cmp_func vsse[6];
  51     me_cmp_func nsse[6];
  52     me_cmp_func dct_max[6];
  53     me_cmp_func dct264_sad[6];
  54
  55     me_cmp_func me_pre_cmp[6];
  56     me_cmp_func me_cmp[6];
  57     me_cmp_func me_sub_cmp[6];
  58     me_cmp_func mb_cmp[6];
  59     me_cmp_func ildct_cmp[6]; // only width 16 used
  60     me_cmp_func frame_skip_cmp[6]; // only width 8 used
  61
  62     me_cmp_func pix_abs[2][4];
  63 } MECmpContext;
  64
  65 void ff_me_cmp_init_static(void);
  66
  67 void ff_me_cmp_init(MECmpContext *c, AVCodecContext *avctx);
  68 void ff_me_cmp_init_arm(MECmpContext *c, AVCodecContext *avctx);
  69 void ff_me_cmp_init_ppc(MECmpContext *c, AVCodecContext *avctx);
  70 void ff_me_cmp_init_x86(MECmpContext *c, AVCodecContext *avctx);
  71
  72 void ff_set_cmp(MECmpContext *c, me_cmp_func *cmp, int type);
  73
  74 #endif /* AVCODEC_ME_CMP_H */