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

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg 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  * FFmpeg 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 FFmpeg; 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 const uint32_t ff_square_tab[512];
  27
  28
  29 /* minimum alignment rules ;)
  30  * If you notice errors in the align stuff, need more alignment for some ASM code
  31  * for some CPU or need to use a function with less aligned data then send a mail
  32  * to the ffmpeg-devel mailing list, ...
  33  *
  34  * !warning These alignments might not match reality, (missing attribute((align))
  35  * stuff somewhere possible).
  36  * I (Michael) did not check them, these are just the alignments which I think
  37  * could be reached easily ...
  38  *
  39  * !future video codecs might need functions with less strict alignment
  40  */
  41
  42 struct MpegEncContext;
  43 /* Motion estimation:
  44  * h is limited to { width / 2, width, 2 * width },
  45  * but never larger than 16 and never smaller than 2.
  46  * Although currently h < 4 is not used as functions with
  47  * width < 8 are neither used nor implemented. */
  48 typedef int (*me_cmp_func)(struct MpegEncContext *c,
  49                            uint8_t *blk1 /* align width (8 or 16) */,
  50                            uint8_t *blk2 /* align 1 */, ptrdiff_t stride,
  51                            int h);
  52
  53 typedef struct MECmpContext {
  54     int (*sum_abs_dctelem)(int16_t *block /* align 16 */);
  55
  56     me_cmp_func sad[6]; /* identical to pix_absAxA except additional void * */
  57     me_cmp_func sse[6];
  58     me_cmp_func hadamard8_diff[6];
  59     me_cmp_func dct_sad[6];
  60     me_cmp_func quant_psnr[6];
  61     me_cmp_func bit[6];
  62     me_cmp_func rd[6];
  63     me_cmp_func vsad[6];
  64     me_cmp_func vsse[6];
  65     me_cmp_func nsse[6];
  66     me_cmp_func w53[6];
  67     me_cmp_func w97[6];
  68     me_cmp_func dct_max[6];
  69     me_cmp_func dct264_sad[6];
  70
  71     me_cmp_func me_pre_cmp[6];
  72     me_cmp_func me_cmp[6];
  73     me_cmp_func me_sub_cmp[6];
  74     me_cmp_func mb_cmp[6];
  75     me_cmp_func ildct_cmp[6]; // only width 16 used
  76     me_cmp_func frame_skip_cmp[6]; // only width 8 used
  77
  78     me_cmp_func pix_abs[2][4];
  79     me_cmp_func median_sad[6];
  80 } MECmpContext;
  81
  82 void ff_me_cmp_init(MECmpContext *c, AVCodecContext *avctx);
  83 void ff_me_cmp_init_alpha(MECmpContext *c, AVCodecContext *avctx);
  84 void ff_me_cmp_init_arm(MECmpContext *c, AVCodecContext *avctx);
  85 void ff_me_cmp_init_ppc(MECmpContext *c, AVCodecContext *avctx);
  86 void ff_me_cmp_init_x86(MECmpContext *c, AVCodecContext *avctx);
  87 void ff_me_cmp_init_mips(MECmpContext *c, AVCodecContext *avctx);
  88
  89 void ff_set_cmp(MECmpContext *c, me_cmp_func *cmp, int type);
  90
  91 void ff_dsputil_init_dwt(MECmpContext *c);
  92
  93 #endif /* AVCODEC_ME_CMP_H */