2 Copyright (C) 2003 Daniel Moreno <comac@comac.darktech.org>
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program 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
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
32 #include "img_format.h"
36 #define PARAM1_DEFAULT 4.0
37 #define PARAM2_DEFAULT 3.0
38 #define PARAM3_DEFAULT 6.0
40 //===========================================================================//
45 unsigned short *Frame
[3];
49 /***************************************************************************/
51 static void uninit(struct vf_instance_s
* vf
){
52 if(vf
->priv
->Line
){free(vf
->priv
->Line
);vf
->priv
->Line
=NULL
;}
53 if(vf
->priv
->Frame
[0]){free(vf
->priv
->Frame
[0]);vf
->priv
->Frame
[0]=NULL
;}
54 if(vf
->priv
->Frame
[1]){free(vf
->priv
->Frame
[1]);vf
->priv
->Frame
[1]=NULL
;}
55 if(vf
->priv
->Frame
[2]){free(vf
->priv
->Frame
[2]);vf
->priv
->Frame
[2]=NULL
;}
58 static int config(struct vf_instance_s
* vf
,
59 int width
, int height
, int d_width
, int d_height
,
60 unsigned int flags
, unsigned int outfmt
){
63 vf
->priv
->Line
= malloc(width
*sizeof(int));
65 return vf_next_config(vf
,width
,height
,d_width
,d_height
,flags
,outfmt
);
68 static inline unsigned int LowPassMul(unsigned int PrevMul
, unsigned int CurrMul
, int* Coef
){
69 // int dMul= (PrevMul&0xFFFFFF)-(CurrMul&0xFFFFFF);
70 int dMul
= PrevMul
-CurrMul
;
71 int d
=((dMul
+0x10007FF)>>12);
72 return CurrMul
+ Coef
[d
];
75 static void deNoiseTemporal(
76 unsigned char *Frame
, // mpi->planes[x]
77 unsigned char *FrameDest
, // dmpi->planes[x]
78 unsigned short *FrameAnt
,
79 int W
, int H
, int sStride
, int dStride
,
83 unsigned int PixelDst
;
85 for (Y
= 0; Y
< H
; Y
++){
86 for (X
= 0; X
< W
; X
++){
87 PixelDst
= LowPassMul(FrameAnt
[X
]<<8, Frame
[X
]<<16, Temporal
);
88 FrameAnt
[X
] = ((PixelDst
+0x1000007F)>>8);
89 FrameDest
[X
]= ((PixelDst
+0x10007FFF)>>16);
97 static void deNoiseSpacial(
98 unsigned char *Frame
, // mpi->planes[x]
99 unsigned char *FrameDest
, // dmpi->planes[x]
100 unsigned int *LineAnt
, // vf->priv->Line (width bytes)
101 int W
, int H
, int sStride
, int dStride
,
102 int *Horizontal
, int *Vertical
)
105 int sLineOffs
= 0, dLineOffs
= 0;
106 unsigned int PixelAnt
;
107 unsigned int PixelDst
;
109 /* First pixel has no left nor top neighbor. */
110 PixelDst
= LineAnt
[0] = PixelAnt
= Frame
[0]<<16;
111 FrameDest
[0]= ((PixelDst
+0x10007FFF)>>16);
113 /* First line has no top neighbor, only left. */
114 for (X
= 1; X
< W
; X
++){
115 PixelDst
= LineAnt
[X
] = LowPassMul(PixelAnt
, Frame
[X
]<<16, Horizontal
);
116 FrameDest
[X
]= ((PixelDst
+0x10007FFF)>>16);
119 for (Y
= 1; Y
< H
; Y
++){
120 unsigned int PixelAnt
;
121 sLineOffs
+= sStride
, dLineOffs
+= dStride
;
122 /* First pixel on each line doesn't have previous pixel */
123 PixelAnt
= Frame
[sLineOffs
]<<16;
124 PixelDst
= LineAnt
[0] = LowPassMul(LineAnt
[0], PixelAnt
, Vertical
);
125 FrameDest
[dLineOffs
]= ((PixelDst
+0x10007FFF)>>16);
127 for (X
= 1; X
< W
; X
++){
128 unsigned int PixelDst
;
129 /* The rest are normal */
130 PixelAnt
= LowPassMul(PixelAnt
, Frame
[sLineOffs
+X
]<<16, Horizontal
);
131 PixelDst
= LineAnt
[X
] = LowPassMul(LineAnt
[X
], PixelAnt
, Vertical
);
132 FrameDest
[dLineOffs
+X
]= ((PixelDst
+0x10007FFF)>>16);
137 static void deNoise(unsigned char *Frame
, // mpi->planes[x]
138 unsigned char *FrameDest
, // dmpi->planes[x]
139 unsigned int *LineAnt
, // vf->priv->Line (width bytes)
140 unsigned short **FrameAntPtr
,
141 int W
, int H
, int sStride
, int dStride
,
142 int *Horizontal
, int *Vertical
, int *Temporal
)
145 int sLineOffs
= 0, dLineOffs
= 0;
146 unsigned int PixelAnt
;
147 unsigned int PixelDst
;
148 unsigned short* FrameAnt
=(*FrameAntPtr
);
151 (*FrameAntPtr
)=FrameAnt
=malloc(W
*H
*sizeof(unsigned short));
152 for (Y
= 0; Y
< H
; Y
++){
153 unsigned short* dst
=&FrameAnt
[Y
*W
];
154 unsigned char* src
=Frame
+Y
*sStride
;
155 for (X
= 0; X
< W
; X
++) dst
[X
]=src
[X
]<<8;
159 if(!Horizontal
[0] && !Vertical
[0]){
160 deNoiseTemporal(Frame
, FrameDest
, FrameAnt
,
161 W
, H
, sStride
, dStride
, Temporal
);
165 deNoiseSpacial(Frame
, FrameDest
, LineAnt
,
166 W
, H
, sStride
, dStride
, Horizontal
, Vertical
);
170 /* First pixel has no left nor top neighbor. Only previous frame */
171 LineAnt
[0] = PixelAnt
= Frame
[0]<<16;
172 PixelDst
= LowPassMul(FrameAnt
[0]<<8, PixelAnt
, Temporal
);
173 FrameAnt
[0] = ((PixelDst
+0x1000007F)>>8);
174 FrameDest
[0]= ((PixelDst
+0x10007FFF)>>16);
176 /* First line has no top neighbor. Only left one for each pixel and
178 for (X
= 1; X
< W
; X
++){
179 LineAnt
[X
] = PixelAnt
= LowPassMul(PixelAnt
, Frame
[X
]<<16, Horizontal
);
180 PixelDst
= LowPassMul(FrameAnt
[X
]<<8, PixelAnt
, Temporal
);
181 FrameAnt
[X
] = ((PixelDst
+0x1000007F)>>8);
182 FrameDest
[X
]= ((PixelDst
+0x10007FFF)>>16);
185 for (Y
= 1; Y
< H
; Y
++){
186 unsigned int PixelAnt
;
187 unsigned short* LinePrev
=&FrameAnt
[Y
*W
];
188 sLineOffs
+= sStride
, dLineOffs
+= dStride
;
189 /* First pixel on each line doesn't have previous pixel */
190 PixelAnt
= Frame
[sLineOffs
]<<16;
191 LineAnt
[0] = LowPassMul(LineAnt
[0], PixelAnt
, Vertical
);
192 PixelDst
= LowPassMul(LinePrev
[0]<<8, LineAnt
[0], Temporal
);
193 LinePrev
[0] = ((PixelDst
+0x1000007F)>>8);
194 FrameDest
[dLineOffs
]= ((PixelDst
+0x10007FFF)>>16);
196 for (X
= 1; X
< W
; X
++){
197 unsigned int PixelDst
;
198 /* The rest are normal */
199 PixelAnt
= LowPassMul(PixelAnt
, Frame
[sLineOffs
+X
]<<16, Horizontal
);
200 LineAnt
[X
] = LowPassMul(LineAnt
[X
], PixelAnt
, Vertical
);
201 PixelDst
= LowPassMul(LinePrev
[X
]<<8, LineAnt
[X
], Temporal
);
202 LinePrev
[X
] = ((PixelDst
+0x1000007F)>>8);
203 FrameDest
[dLineOffs
+X
]= ((PixelDst
+0x10007FFF)>>16);
209 static int put_image(struct vf_instance_s
* vf
, mp_image_t
*mpi
, double pts
){
210 int cw
= mpi
->w
>> mpi
->chroma_x_shift
;
211 int ch
= mpi
->h
>> mpi
->chroma_y_shift
;
212 int W
= mpi
->w
, H
= mpi
->h
;
214 mp_image_t
*dmpi
=vf_get_image(vf
->next
,mpi
->imgfmt
,
215 MP_IMGTYPE_TEMP
, MP_IMGFLAG_ACCEPT_STRIDE
,
220 deNoise(mpi
->planes
[0], dmpi
->planes
[0],
221 vf
->priv
->Line
, &vf
->priv
->Frame
[0], W
, H
,
222 mpi
->stride
[0], dmpi
->stride
[0],
226 deNoise(mpi
->planes
[1], dmpi
->planes
[1],
227 vf
->priv
->Line
, &vf
->priv
->Frame
[1], cw
, ch
,
228 mpi
->stride
[1], dmpi
->stride
[1],
232 deNoise(mpi
->planes
[2], dmpi
->planes
[2],
233 vf
->priv
->Line
, &vf
->priv
->Frame
[2], cw
, ch
,
234 mpi
->stride
[2], dmpi
->stride
[2],
239 return vf_next_put_image(vf
,dmpi
, pts
);
242 //===========================================================================//
244 static int query_format(struct vf_instance_s
* vf
, unsigned int fmt
){
254 return vf_next_query_format(vf
, fmt
);
260 #define ABS(A) ( (A) > 0 ? (A) : -(A) )
262 static void PrecalcCoefs(int *Ct
, double Dist25
)
265 double Gamma
, Simil
, C
;
267 Gamma
= log(0.25) / log(1.0 - Dist25
/255.0 - 0.00001);
269 for (i
= -255*16; i
<= 255*16; i
++)
271 Simil
= 1.0 - ABS(i
) / (16*255.0);
272 C
= pow(Simil
, Gamma
) * 65536.0 * (double)i
/ 16.0;
273 Ct
[16*256+i
] = (C
<0) ? (C
-0.5) : (C
+0.5);
276 Ct
[0] = (Dist25
!= 0);
280 static int open(vf_instance_t
*vf
, char* args
){
281 double LumSpac
, LumTmp
, ChromSpac
, ChromTmp
;
282 double Param1
, Param2
, Param3
, Param4
;
285 vf
->put_image
=put_image
;
286 vf
->query_format
=query_format
;
288 vf
->priv
=malloc(sizeof(struct vf_priv_s
));
289 memset(vf
->priv
, 0, sizeof(struct vf_priv_s
));
293 switch(sscanf(args
, "%lf:%lf:%lf:%lf",
294 &Param1
, &Param2
, &Param3
, &Param4
298 LumSpac
= PARAM1_DEFAULT
;
299 LumTmp
= PARAM3_DEFAULT
;
301 ChromSpac
= PARAM2_DEFAULT
;
302 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
307 LumTmp
= PARAM3_DEFAULT
* Param1
/ PARAM1_DEFAULT
;
309 ChromSpac
= PARAM2_DEFAULT
* Param1
/ PARAM1_DEFAULT
;
310 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
315 LumTmp
= PARAM3_DEFAULT
* Param1
/ PARAM1_DEFAULT
;
318 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
326 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
338 LumSpac
= PARAM1_DEFAULT
;
339 LumTmp
= PARAM3_DEFAULT
;
341 ChromSpac
= PARAM2_DEFAULT
;
342 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
347 LumSpac
= PARAM1_DEFAULT
;
348 LumTmp
= PARAM3_DEFAULT
;
350 ChromSpac
= PARAM2_DEFAULT
;
351 ChromTmp
= LumTmp
* ChromSpac
/ LumSpac
;
354 PrecalcCoefs(vf
->priv
->Coefs
[0], LumSpac
);
355 PrecalcCoefs(vf
->priv
->Coefs
[1], LumTmp
);
356 PrecalcCoefs(vf
->priv
->Coefs
[2], ChromSpac
);
357 PrecalcCoefs(vf
->priv
->Coefs
[3], ChromTmp
);
362 const vf_info_t vf_info_hqdn3d
= {
363 "High Quality 3D Denoiser",
365 "Daniel Moreno & A'rpi",
371 //===========================================================================//