modules/video_filter/hqdn3d.h

   1 /*
   2  * Copyright (C) 2003 Daniel Moreno <comac@comac.darktech.org>
   3  *
   4  * This file is part of MPlayer.
   5  *
   6  * MPlayer is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * MPlayer is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <inttypes.h>
  25 #include <math.h>
  26
  27 #define PARAM1_DEFAULT 4.0
  28 #define PARAM2_DEFAULT 3.0
  29 #define PARAM3_DEFAULT 6.0
  30
  31 //===========================================================================//
  32
  33 struct vf_priv_s {
  34         int Coefs[4][512*16];
  35         unsigned int *Line;
  36         unsigned short *Frame[3];
  37 };
  38
  39
  40 /***************************************************************************/
  41
  42 static /*inline*/ unsigned int LowPassMul(unsigned int PrevMul, unsigned int CurrMul, int* Coef){
  43 //    int dMul= (PrevMul&0xFFFFFF)-(CurrMul&0xFFFFFF);
  44     int dMul= PrevMul-CurrMul;
  45     unsigned int d=((dMul+0x10007FF)>>12);
  46     return CurrMul + Coef[d];
  47 }
  48
  49 static void deNoiseTemporal(
  50                     unsigned char *Frame,        // mpi->planes[x]
  51                     unsigned char *FrameDest,    // dmpi->planes[x]
  52                     unsigned short *FrameAnt,
  53                     int W, int H, int sStride, int dStride,
  54                     int *Temporal)
  55 {
  56     unsigned int PixelDst;
  57
  58     for (long Y = 0; Y < H; Y++){
  59         for (long X = 0; X < W; X++){
  60             PixelDst = LowPassMul(FrameAnt[X]<<8, Frame[X]<<16, Temporal);
  61             FrameAnt[X] = ((PixelDst+0x1000007F)>>8);
  62             FrameDest[X]= ((PixelDst+0x10007FFF)>>16);
  63         }
  64         Frame += sStride;
  65         FrameDest += dStride;
  66         FrameAnt += W;
  67     }
  68 }
  69
  70 static void deNoiseSpacial(
  71                     unsigned char *Frame,        // mpi->planes[x]
  72                     unsigned char *FrameDest,    // dmpi->planes[x]
  73                     unsigned int *LineAnt,       // vf->priv->Line (width bytes)
  74                     int W, int H, int sStride, int dStride,
  75                     int *Horizontal, int *Vertical)
  76 {
  77     long sLineOffs = 0, dLineOffs = 0;
  78     unsigned int PixelAnt;
  79     unsigned int PixelDst;
  80
  81     /* First pixel has no left nor top neighbor. */
  82     PixelDst = LineAnt[0] = PixelAnt = Frame[0]<<16;
  83     FrameDest[0]= ((PixelDst+0x10007FFF)>>16);
  84
  85     /* First line has no top neighbor, only left. */
  86     for (long X = 1; X < W; X++){
  87         PixelDst = LineAnt[X] = LowPassMul(PixelAnt, Frame[X]<<16, Horizontal);
  88         FrameDest[X]= ((PixelDst+0x10007FFF)>>16);
  89     }
  90
  91     for (long Y = 1; Y < H; Y++){
  92         unsigned int PixelAnt;
  93         sLineOffs += sStride, dLineOffs += dStride;
  94         /* First pixel on each line doesn't have previous pixel */
  95         PixelAnt = Frame[sLineOffs]<<16;
  96         PixelDst = LineAnt[0] = LowPassMul(LineAnt[0], PixelAnt, Vertical);
  97         FrameDest[dLineOffs]= ((PixelDst+0x10007FFF)>>16);
  98
  99         for (long X = 1; X < W; X++){
 100             unsigned int PixelDst;
 101             /* The rest are normal */
 102             PixelAnt = LowPassMul(PixelAnt, Frame[sLineOffs+X]<<16, Horizontal);
 103             PixelDst = LineAnt[X] = LowPassMul(LineAnt[X], PixelAnt, Vertical);
 104             FrameDest[dLineOffs+X]= ((PixelDst+0x10007FFF)>>16);
 105         }
 106     }
 107 }
 108
 109 static void deNoise(unsigned char *Frame,        // mpi->planes[x]
 110                     unsigned char *FrameDest,    // dmpi->planes[x]
 111                     unsigned int *LineAnt,      // vf->priv->Line (width bytes)
 112                     unsigned short **FrameAntPtr,
 113                     int W, int H, int sStride, int dStride,
 114                     int *Horizontal, int *Vertical, int *Temporal)
 115 {
 116     long sLineOffs = 0, dLineOffs = 0;
 117     unsigned int PixelAnt;
 118     unsigned int PixelDst;
 119     unsigned short* FrameAnt=(*FrameAntPtr);
 120
 121     if(!FrameAnt){
 122         (*FrameAntPtr)=FrameAnt=malloc(W*H*sizeof(unsigned short));
 123         if(!FrameAnt)
 124             return;
 125         for (long Y = 0; Y < H; Y++){
 126             unsigned short* dst=&FrameAnt[Y*W];
 127             unsigned char* src=Frame+Y*sStride;
 128             for (long X = 0; X < W; X++) dst[X]=src[X]<<8;
 129         }
 130     }
 131
 132     if(!Horizontal[0] && !Vertical[0]){
 133         deNoiseTemporal(Frame, FrameDest, FrameAnt,
 134                         W, H, sStride, dStride, Temporal);
 135         return;
 136     }
 137     if(!Temporal[0]){
 138         deNoiseSpacial(Frame, FrameDest, LineAnt,
 139                        W, H, sStride, dStride, Horizontal, Vertical);
 140         return;
 141     }
 142
 143     /* First pixel has no left nor top neighbor. Only previous frame */
 144     LineAnt[0] = PixelAnt = Frame[0]<<16;
 145     PixelDst = LowPassMul(FrameAnt[0]<<8, PixelAnt, Temporal);
 146     FrameAnt[0] = ((PixelDst+0x1000007F)>>8);
 147     FrameDest[0]= ((PixelDst+0x10007FFF)>>16);
 148
 149     /* First line has no top neighbor. Only left one for each pixel and
 150      * last frame */
 151     for (long X = 1; X < W; X++){
 152         LineAnt[X] = PixelAnt = LowPassMul(PixelAnt, Frame[X]<<16, Horizontal);
 153         PixelDst = LowPassMul(FrameAnt[X]<<8, PixelAnt, Temporal);
 154         FrameAnt[X] = ((PixelDst+0x1000007F)>>8);
 155         FrameDest[X]= ((PixelDst+0x10007FFF)>>16);
 156     }
 157
 158     for (long Y = 1; Y < H; Y++){
 159         unsigned int PixelAnt;
 160         unsigned short* LinePrev=&FrameAnt[Y*W];
 161         sLineOffs += sStride, dLineOffs += dStride;
 162         /* First pixel on each line doesn't have previous pixel */
 163         PixelAnt = Frame[sLineOffs]<<16;
 164         LineAnt[0] = LowPassMul(LineAnt[0], PixelAnt, Vertical);
 165         PixelDst = LowPassMul(LinePrev[0]<<8, LineAnt[0], Temporal);
 166         LinePrev[0] = ((PixelDst+0x1000007F)>>8);
 167         FrameDest[dLineOffs]= ((PixelDst+0x10007FFF)>>16);
 168
 169         for (long X = 1; X < W; X++){
 170             unsigned int PixelDst;
 171             /* The rest are normal */
 172             PixelAnt = LowPassMul(PixelAnt, Frame[sLineOffs+X]<<16, Horizontal);
 173             LineAnt[X] = LowPassMul(LineAnt[X], PixelAnt, Vertical);
 174             PixelDst = LowPassMul(LinePrev[X]<<8, LineAnt[X], Temporal);
 175             LinePrev[X] = ((PixelDst+0x1000007F)>>8);
 176             FrameDest[dLineOffs+X]= ((PixelDst+0x10007FFF)>>16);
 177         }
 178     }
 179 }
 180
 181
 182 //===========================================================================//
 183
 184 static void PrecalcCoefs(int *Ct, double Dist25)
 185 {
 186     double Gamma, Simil, C;
 187
 188     Gamma = log(0.25) / log(1.0 - Dist25/255.0 - 0.00001);
 189
 190     for (int i = -255*16; i <= 255*16; i++)
 191     {
 192         Simil = 1.0 - abs(i) / (16*255.0);
 193         C = pow(Simil, Gamma) * 65536.0 * (double)i / 16.0;
 194         Ct[16*256+i] = (C<0) ? (C-0.5) : (C+0.5);
 195     }
 196
 197     Ct[0] = (Dist25 != 0);
 198 }
 199
 200
 201 //===========================================================================//