PeerStreamer

/*

 * Copyright (C) 2001-2002 Michael Niedermayer (michaelni@gmx.at)

 *

 * This file is part of FFmpeg.

 *

 * FFmpeg is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * FFmpeg is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with FFmpeg; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

/**

 * @file postprocess_internal.h

 * internal api header.

 */

#ifndef FFMPEG_POSTPROCESS_INTERNAL_H

#define FFMPEG_POSTPROCESS_INTERNAL_H

#include "avutil.h"

#include "postprocess.h"

#define V_DEBLOCK       0x01

#define H_DEBLOCK       0x02

#define DERING          0x04

#define LEVEL_FIX       0x08 ///< Brightness & Contrast

#define LUM_V_DEBLOCK   V_DEBLOCK               //   1

#define LUM_H_DEBLOCK   H_DEBLOCK               //   2

#define CHROM_V_DEBLOCK (V_DEBLOCK<<4)          //  16

#define CHROM_H_DEBLOCK (H_DEBLOCK<<4)          //  32

#define LUM_DERING      DERING                  //   4

#define CHROM_DERING    (DERING<<4)             //  64

#define LUM_LEVEL_FIX   LEVEL_FIX               //   8

#define CHROM_LEVEL_FIX (LEVEL_FIX<<4)          // 128 (not implemented yet)

// Experimental vertical filters

#define V_X1_FILTER     0x0200                  // 512

#define V_A_DEBLOCK     0x0400

// Experimental horizontal filters

#define H_X1_FILTER     0x2000                  // 8192

#define H_A_DEBLOCK     0x4000

/// select between full y range (255-0) or standart one (234-16)

#define FULL_Y_RANGE    0x8000                  // 32768

//Deinterlacing Filters

#define        LINEAR_IPOL_DEINT_FILTER         0x10000 // 65536

#define        LINEAR_BLEND_DEINT_FILTER        0x20000 // 131072

#define        CUBIC_BLEND_DEINT_FILTER         0x8000  // (not implemented yet)

#define        CUBIC_IPOL_DEINT_FILTER          0x40000 // 262144

#define        MEDIAN_DEINT_FILTER              0x80000 // 524288

#define        FFMPEG_DEINT_FILTER              0x400000

#define        LOWPASS5_DEINT_FILTER            0x800000

#define TEMP_NOISE_FILTER               0x100000

#define FORCE_QUANT                     0x200000

#if ( defined(__PIC__) || defined(__pic__) ) && ! defined(PIC)

#    define PIC

#endif

//use if you want a faster postprocessing code

//cannot differentiate between chroma & luma filters (both on or both off)

//obviously the -pp option on the command line has no effect except turning the here selected

//filters on

//#define COMPILE_TIME_MODE 0x77

#if 1

static inline int CLIP(int a){

        if(a&256) return ((a)>>31)^(-1);

        else      return a;

}

//#define CLIP(a) (((a)&256) ? ((a)>>31)^(-1) : (a))

#elif 0

#define CLIP(a) clip_tab[a]

#else

#define CLIP(a) (a)

#endif

/**

 * Postprocessng filter.

 */

struct PPFilter{

        const char *shortName;

        const char *longName;

        int chromDefault;       ///< is chrominance filtering on by default if this filter is manually activated

        int minLumQuality;      ///< minimum quality to turn luminance filtering on

        int minChromQuality;    ///< minimum quality to turn chrominance filtering on

        int mask;               ///< Bitmask to turn this filter on

};

/**

 * Postprocessng mode.

 */

typedef struct PPMode{

        int lumMode;                    ///< acivates filters for luminance

        int chromMode;                  ///< acivates filters for chrominance

        int error;                      ///< non zero on error

        int minAllowedY;                ///< for brigtness correction

        int maxAllowedY;                ///< for brihtness correction

        float maxClippedThreshold;      ///< amount of "black" u r willing to loose to get a brightness corrected picture

        int maxTmpNoise[3];             ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)

        int baseDcDiff;

        int flatnessThreshold;

        int forcedQuant;                ///< quantizer if FORCE_QUANT is used

} PPMode;

/**

 * postprocess context.

 */

typedef struct PPContext{

        /**

         * info on struct for av_log

         */

        AVClass *av_class;

        uint8_t *tempBlocks; ///<used for the horizontal code

        /**

         * luma histogram.

         * we need 64bit here otherwise we'll going to have a problem

         * after watching a black picture for 5 hours

         */

        uint64_t *yHistogram;

        DECLARE_ALIGNED(8, uint64_t, packedYOffset);

        DECLARE_ALIGNED(8, uint64_t, packedYScale);

        /** Temporal noise reducing buffers */

        uint8_t *tempBlured[3];

        int32_t *tempBluredPast[3];

        /** Temporary buffers for handling the last row(s) */

        uint8_t *tempDst;

        uint8_t *tempSrc;

        uint8_t *deintTemp;

        DECLARE_ALIGNED(8, uint64_t, pQPb);

        DECLARE_ALIGNED(8, uint64_t, pQPb2);

        DECLARE_ALIGNED(8, uint64_t, mmxDcOffset[64]);

        DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold[64]);

        QP_STORE_T *stdQPTable;       ///< used to fix MPEG2 style qscale

        QP_STORE_T *nonBQPTable;

        QP_STORE_T *forcedQPTable;

        int QP;

        int nonBQP;

        int frameNum;

        int cpuCaps;

        int qpStride; ///<size of qp buffers (needed to realloc them if needed)

        int stride;   ///<size of some buffers (needed to realloc them if needed)

        int hChromaSubSample;

        int vChromaSubSample;

        PPMode ppMode;

} PPContext;

static inline void linecpy(void *dest, void *src, int lines, int stride)

{

        if (stride > 0) {

                memcpy(dest, src, lines*stride);

        } else {

                memcpy(dest+(lines-1)*stride, src+(lines-1)*stride, -lines*stride);

        }

}

#endif /* FFMPEG_POSTPROCESS_INTERNAL_H */