1 /*****************************************************************************
2 * postproc.c: video postprocessing using libpostproc
3 *****************************************************************************
4 * Copyright (C) 1999-2008 the VideoLAN team
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Gildas Bazin <gbazin@netcourrier.com>
9 * Antoine Cellerier <dionoea at videolan dot org>
11 * This program is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation; either version 2 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
21 * You should have received a copy of the GNU General Public License
22 * along with this program; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24 *****************************************************************************/
30 #include <vlc_common.h>
31 #include <vlc_plugin.h>
34 #include "filter_picture.h"
36 #ifdef HAVE_POSTPROC_POSTPROCESS_H
37 # include <postproc/postprocess.h>
39 # include <libpostproc/postprocess.h>
42 #ifndef PP_CPU_CAPS_ALTIVEC
43 # define PP_CPU_CAPS_ALTIVEC 0
46 /*****************************************************************************
48 *****************************************************************************/
49 static int OpenPostproc( vlc_object_t * );
50 static void ClosePostproc( vlc_object_t * );
52 static picture_t *PostprocPict( filter_t *, picture_t * );
54 static int PPQCallback( vlc_object_t *, char const *,
55 vlc_value_t, vlc_value_t, void * );
56 static int PPNameCallback( vlc_object_t *, char const *,
57 vlc_value_t, vlc_value_t, void * );
59 #define Q_TEXT N_("Post processing quality")
60 #define Q_LONGTEXT N_( \
61 "Quality of post processing. Valid range is 0 to 6\n" \
62 "Higher levels require considerable more CPU power, but produce " \
63 "better looking pictures." )
65 #define NAME_TEXT N_("FFmpeg post processing filter chains")
66 /* defined by libpostproc */
67 #define NAME_LONGTEXT pp_help
69 #define FILTER_PREFIX "postproc-"
71 /*****************************************************************************
73 *****************************************************************************/
75 set_description( N_("Video post processing filter") );
76 add_shortcut( "postprocess" ); /* name is "postproc" */
78 set_category( CAT_VIDEO );
79 set_subcategory( SUBCAT_VIDEO_VFILTER );
81 set_capability( "video filter2", 0 );
83 set_callbacks( OpenPostproc, ClosePostproc );
85 add_integer_with_range( FILTER_PREFIX "q", PP_QUALITY_MAX, 0,
86 PP_QUALITY_MAX, NULL, Q_TEXT, Q_LONGTEXT, false );
87 add_deprecated_alias( "ffmpeg-pp-q" );
88 add_string( FILTER_PREFIX "name", "default", NULL, NAME_TEXT,
89 NAME_LONGTEXT, true );
90 add_deprecated_alias( "ffmpeg-pp-name" );
93 static const char *const ppsz_filter_options[] = {
97 /*****************************************************************************
98 * filter_sys_t : libpostproc video postprocessing descriptor
99 *****************************************************************************/
102 pp_context_t *pp_context; /* Never changes after init */
103 pp_mode_t *pp_mode; /* Set to NULL if post processing is disabled */
105 vlc_mutex_t lock; /* Lock when using or changing pp_mode */
109 /*****************************************************************************
110 * OpenPostproc: probe and open the postproc
111 *****************************************************************************/
112 static int OpenPostproc( vlc_object_t *p_this )
114 filter_t *p_filter = (filter_t *)p_this;
116 vlc_value_t val, val_orig, text;
117 unsigned i_cpu = vlc_CPU();
120 if( p_filter->fmt_in.video.i_chroma != p_filter->fmt_out.video.i_chroma ||
121 p_filter->fmt_in.video.i_height != p_filter->fmt_out.video.i_height ||
122 p_filter->fmt_in.video.i_width != p_filter->fmt_out.video.i_width )
124 msg_Err( p_filter, "Filter input and output formats must be identical" );
128 /* Set CPU capabilities */
129 if( i_cpu & CPU_CAPABILITY_MMX )
130 i_flags |= PP_CPU_CAPS_MMX;
131 if( i_cpu & CPU_CAPABILITY_MMXEXT )
132 i_flags |= PP_CPU_CAPS_MMX2;
133 if( i_cpu & CPU_CAPABILITY_3DNOW )
134 i_flags |= PP_CPU_CAPS_3DNOW;
135 if( i_cpu & CPU_CAPABILITY_ALTIVEC )
136 i_flags |= PP_CPU_CAPS_ALTIVEC;
138 switch( p_filter->fmt_in.video.i_chroma )
140 case VLC_FOURCC('I','4','4','4'):
141 case VLC_FOURCC('J','4','4','4'):
142 i_flags |= PP_FORMAT_444;
144 case VLC_FOURCC('I','4','2','2'):
145 case VLC_FOURCC('J','4','2','2'):
146 i_flags |= PP_FORMAT_422;
148 case VLC_FOURCC('I','4','1','1'):
149 i_flags |= PP_FORMAT_411;
151 case VLC_FOURCC('I','4','2','0'):
152 case VLC_FOURCC('I','Y','U','V'):
153 case VLC_FOURCC('J','4','2','0'):
154 case VLC_FOURCC('Y','V','1','2'):
155 /* case VLC_FOURCC('Y','U','V','A'): FIXME Should work but alpha plane needs to be copied manually and I'm kind of feeling too lazy to write the code to do that ATM (i_pitch vs i_visible_pitch...). */
156 i_flags |= PP_FORMAT_420;
159 msg_Err( p_filter, "Unsupported input chroma (%4s)",
160 (char*)&p_filter->fmt_in.video.i_chroma );
164 p_sys = malloc( sizeof( filter_sys_t ) );
165 if( !p_sys ) return VLC_ENOMEM;
166 p_filter->p_sys = p_sys;
168 p_sys->pp_context = pp_get_context( p_filter->fmt_in.video.i_width,
169 p_filter->fmt_in.video.i_height,
171 if( !p_sys->pp_context )
173 msg_Err( p_filter, "Error while creating post processing context." );
178 config_ChainParse( p_filter, FILTER_PREFIX, ppsz_filter_options,
181 var_Create( p_filter, FILTER_PREFIX "q",
182 VLC_VAR_INTEGER | VLC_VAR_HASCHOICE | VLC_VAR_DOINHERIT |
184 var_Change( p_filter, FILTER_PREFIX "q", VLC_VAR_SETISCOMMAND, NULL, NULL ); /* For some obscure reason the VLC_VAR_ISCOMMAND isn't taken into account in during var_Create */
185 var_AddCallback( p_filter, FILTER_PREFIX "q", PPQCallback, NULL );
186 text.psz_string = _("Post processing");
187 var_Change( p_filter, FILTER_PREFIX "q", VLC_VAR_SETTEXT, &text, NULL );
189 var_Get( p_filter, FILTER_PREFIX "q", &val_orig );
190 var_Change( p_filter, FILTER_PREFIX "q", VLC_VAR_DELCHOICE, &val_orig, NULL );
192 val.psz_string = var_CreateGetNonEmptyStringCommand(
193 p_filter, FILTER_PREFIX "name" );
194 var_AddCallback( p_filter, FILTER_PREFIX "name", PPNameCallback, NULL );
197 p_sys->pp_mode = pp_get_mode_by_name_and_quality( val.psz_string?:
201 if( !p_sys->pp_mode )
203 msg_Err( p_filter, "Error while creating post processing mode." );
204 free( val.psz_string );
205 var_Destroy( p_filter, FILTER_PREFIX "q" );
206 pp_free_context( p_sys->pp_context );
213 p_sys->pp_mode = NULL;
215 free( val.psz_string );
217 for( val.i_int = 0; val.i_int <= PP_QUALITY_MAX; val.i_int++ )
222 text.psz_string = _("Disable");
225 text.psz_string = _("Lowest");
228 text.psz_string = _("Highest");
231 text.psz_string = NULL;
234 var_Change( p_filter, FILTER_PREFIX "q", VLC_VAR_ADDCHOICE,
235 &val, text.psz_string?&text:NULL );
238 vlc_mutex_init( &p_sys->lock );
240 p_filter->pf_video_filter = PostprocPict;
245 /*****************************************************************************
247 *****************************************************************************/
248 static void ClosePostproc( vlc_object_t *p_this )
250 filter_t *p_filter = (filter_t *)p_this;
251 filter_sys_t *p_sys = p_filter->p_sys;
252 vlc_mutex_destroy( &p_sys->lock );
253 pp_free_context( p_sys->pp_context );
254 if( p_sys->pp_mode ) pp_free_mode( p_sys->pp_mode );
258 /*****************************************************************************
260 *****************************************************************************/
261 static picture_t *PostprocPict( filter_t *p_filter, picture_t *p_pic )
263 filter_sys_t *p_sys = p_filter->p_sys;
265 const uint8_t *src[3];
268 int i_src_stride[3], i_dst_stride[3];
270 vlc_mutex_lock( &p_sys->lock ); /* Lock to prevent issues if pp_mode is changed */
271 if( !p_sys->pp_mode )
273 vlc_mutex_unlock( &p_sys->lock );
277 picture_t *p_outpic = filter_NewPicture( p_filter );
280 picture_Release( p_pic );
281 vlc_mutex_unlock( &p_sys->lock );
285 for( i_plane = 0; i_plane < p_pic->i_planes; i_plane++ )
287 src[i_plane] = p_pic->p[i_plane].p_pixels;
288 dst[i_plane] = p_outpic->p[i_plane].p_pixels;
290 /* I'm not sure what happens if i_pitch != i_visible_pitch ...
291 * at least it shouldn't crash. */
292 i_src_stride[i_plane] = p_pic->p[i_plane].i_pitch;
293 i_dst_stride[i_plane] = p_outpic->p[i_plane].i_pitch;
296 pp_postprocess( src, i_src_stride, dst, i_dst_stride,
297 p_filter->fmt_in.video.i_width,
298 p_filter->fmt_in.video.i_height,
299 NULL /* FIXME ? works by selecting a default table. But maybe setting our own might help improve post processing quality ... */,
301 p_sys->pp_mode, p_sys->pp_context,
302 PP_PICT_TYPE_QP2 /* FIXME ? This should be set only for mpeg2 type codecs if I understand correctly. */ );
303 vlc_mutex_unlock( &p_sys->lock );
305 return CopyInfoAndRelease( p_outpic, p_pic );
308 /*****************************************************************************
310 *****************************************************************************/
311 static void PPChangeMode( filter_t *p_filter, const char *psz_name,
314 filter_sys_t *p_sys = p_filter->p_sys;
315 vlc_mutex_lock( &p_sys->lock );
318 pp_mode_t *pp_mode = pp_get_mode_by_name_and_quality( psz_name?:
323 pp_free_mode( p_sys->pp_mode );
324 p_sys->pp_mode = pp_mode;
327 msg_Warn( p_filter, "Error while changing post processing mode. "
328 "Keeping previous mode." );
332 pp_free_mode( p_sys->pp_mode );
333 p_sys->pp_mode = NULL;
335 vlc_mutex_unlock( &p_sys->lock );
338 static int PPQCallback( vlc_object_t *p_this, const char *psz_var,
339 vlc_value_t oldval, vlc_value_t newval, void *p_data )
341 VLC_UNUSED(psz_var); VLC_UNUSED(oldval); VLC_UNUSED(p_data);
342 filter_t *p_filter = (filter_t *)p_this;
344 char *psz_name = var_GetNonEmptyString( p_filter, FILTER_PREFIX "name" );
345 PPChangeMode( p_filter, psz_name, newval.i_int );
350 static int PPNameCallback( vlc_object_t *p_this, const char *psz_var,
351 vlc_value_t oldval, vlc_value_t newval, void *p_data )
353 VLC_UNUSED(psz_var); VLC_UNUSED(oldval); VLC_UNUSED(p_data);
354 filter_t *p_filter = (filter_t *)p_this;
356 int i_quality = var_GetInteger( p_filter, FILTER_PREFIX "q" );
357 PPChangeMode( p_filter, *newval.psz_string?newval.psz_string:NULL, i_quality );