git.sesse.net Git - ffmpeg/blob - libavfilter/vf_fspp.h

   1 /*
   2  * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
   3  * Copyright (C) 2005 Nikolaj Poroshin <porosh3@psu.ru>
   4  * Copyright (c) 2014 Arwa Arif <arwaarif1994@gmail.com>
   5  *
   6  * This file is part of FFmpeg.
   7  *
   8  * FFmpeg is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * FFmpeg is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with FFmpeg; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22
  23 #ifndef AVFILTER_FSPP_H
  24 #define AVFILTER_FSPP_H
  25
  26 #include "avfilter.h"
  27
  28 #define BLOCKSZ 12
  29 #define MAX_LEVEL 5
  30
  31 #define DCTSIZE 8
  32 #define DCTSIZE_S "8"
  33
  34 #define FIX(x,s)  ((x) * (1 << s) + 0.5)
  35
  36 #define MULTIPLY16H(x,k)   (((x) * (k)) >> 16)
  37 #define THRESHOLD(r,x,t)                         \
  38     if(((unsigned)((x) + t)) > t * 2) r = (x);   \
  39     else r = 0;
  40 #define DESCALE(x,n)  (((x) + (1 << ((n) - 1))) >> n)
  41
  42 typedef int32_t int_simd16_t;
  43 static const int16_t FIX_0_382683433   = FIX(0.382683433, 14);
  44 static const int16_t FIX_0_541196100   = FIX(0.541196100, 14);
  45 static const int16_t FIX_0_707106781   = FIX(M_SQRT1_2  , 14);
  46 static const int16_t FIX_1_306562965   = FIX(1.306562965, 14);
  47 static const int16_t FIX_1_414213562_A = FIX(M_SQRT2    , 14);
  48 static const int16_t FIX_1_847759065   = FIX(1.847759065, 13);
  49 static const int16_t FIX_2_613125930   = FIX(-2.613125930, 13);
  50 static const int16_t FIX_1_414213562   = FIX(M_SQRT2    , 13);
  51 static const int16_t FIX_1_082392200   = FIX(1.082392200, 13);
  52
  53 typedef struct FSPPContext {
  54     AVClass *class;
  55     uint64_t threshold_mtx_noq[8 * 2];
  56     uint64_t threshold_mtx[8 * 2];        //used in both C & MMX (& later SSE2) versions
  57
  58     int log2_count;
  59     int strength;
  60     int hsub;
  61     int vsub;
  62     int temp_stride;
  63     int qp;
  64     int qscale_type;
  65     int prev_q;
  66     uint8_t *src;
  67     int16_t *temp;
  68     int8_t  *non_b_qp_table;
  69     int non_b_qp_stride;
  70     int use_bframe_qp;
  71
  72     void (*store_slice)(uint8_t *dst, int16_t *src,
  73                         ptrdiff_t dst_stride, ptrdiff_t src_stride,
  74                         ptrdiff_t width, ptrdiff_t height, ptrdiff_t log2_scale);
  75
  76     void (*store_slice2)(uint8_t *dst, int16_t *src,
  77                          ptrdiff_t dst_stride, ptrdiff_t src_stride,
  78                          ptrdiff_t width, ptrdiff_t height, ptrdiff_t log2_scale);
  79
  80     void (*mul_thrmat)(int16_t *thr_adr_noq, int16_t *thr_adr, int q);
  81
  82     void (*column_fidct)(int16_t *thr_adr, int16_t *data,
  83                          int16_t *output, int cnt);
  84
  85     void (*row_idct)(int16_t *workspace, int16_t *output_adr,
  86                      ptrdiff_t output_stride, int cnt);
  87
  88     void (*row_fdct)(int16_t *data, const uint8_t *pixels,
  89                      ptrdiff_t line_size, int cnt);
  90
  91 } FSPPContext;
  92
  93 void ff_fspp_init_x86(FSPPContext *fspp);
  94
  95 #endif /* AVFILTER_FSPP_H */