2 * filter_swscale.c -- image scaling filter
3 * Copyright (C) 2008-2009 Ushodaya Enterprises Limited
4 * Author: Dan Dennedy <dan@dennedy.org>
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library 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 GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #include <framework/mlt_filter.h>
22 #include <framework/mlt_frame.h>
23 #include <framework/mlt_factory.h>
24 #include <framework/mlt_factory.h>
27 // ffmpeg Header files
36 #if LIBAVUTIL_VERSION_INT < (50<<16)
37 #define PIX_FMT_RGB32 PIX_FMT_RGBA32
38 #define PIX_FMT_YUYV422 PIX_FMT_YUV422
41 static inline int is_big_endian( )
43 union { int i; char c[ 4 ]; } big_endian_test;
44 big_endian_test.i = 1;
46 return big_endian_test.c[ 0 ] != 1;
49 static inline int convert_mlt_to_av_cs( mlt_image_format format )
56 value = PIX_FMT_RGB24;
58 case mlt_image_rgb24a:
59 value = PIX_FMT_RGB32;
61 case mlt_image_yuv422:
62 value = PIX_FMT_YUYV422;
64 case mlt_image_yuv420p:
65 value = PIX_FMT_YUV420P;
67 case mlt_image_opengl:
69 fprintf( stderr, "Invalid format...\n" );
76 static int filter_scale( mlt_frame this, uint8_t **image, mlt_image_format iformat, mlt_image_format oformat, int iwidth, int iheight, int owidth, int oheight )
79 mlt_properties properties = MLT_FRAME_PROPERTIES( this );
81 // Get the requested interpolation method
82 char *interps = mlt_properties_get( properties, "rescale.interp" );
84 // Convert to the SwScale flag
85 int interp = SWS_BILINEAR;
86 if ( strcmp( interps, "nearest" ) == 0 || strcmp( interps, "neighbor" ) == 0 )
88 else if ( strcmp( interps, "tiles" ) == 0 || strcmp( interps, "fast_bilinear" ) == 0 )
89 interp = SWS_FAST_BILINEAR;
90 else if ( strcmp( interps, "bilinear" ) == 0 )
91 interp = SWS_BILINEAR;
92 else if ( strcmp( interps, "bicubic" ) == 0 )
94 else if ( strcmp( interps, "bicublin" ) == 0 )
95 interp = SWS_BICUBLIN;
96 else if ( strcmp( interps, "gauss" ) == 0 )
98 else if ( strcmp( interps, "sinc" ) == 0 )
100 else if ( strcmp( interps, "hyper" ) == 0 || strcmp( interps, "lanczos" ) == 0 )
101 interp = SWS_LANCZOS;
102 else if ( strcmp( interps, "spline" ) == 0 )
107 uint8_t *outbuf = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
109 // Convert the pixel formats
110 iformat = convert_mlt_to_av_cs( iformat );
111 oformat = convert_mlt_to_av_cs( oformat );
113 avpicture_fill( &input, *image, iformat, iwidth, iheight );
114 avpicture_fill( &output, outbuf, oformat, owidth, oheight );
116 // Extract the alpha channel
117 if ( iformat == PIX_FMT_RGB32 && oformat == PIX_FMT_YUYV422 )
119 // Allocate the alpha mask
120 uint8_t *alpha = mlt_pool_alloc( iwidth * ( iheight + 1 ) );
123 // Convert the image and extract alpha
124 mlt_convert_rgb24a_to_yuv422( *image, iwidth, iheight, iwidth * 4, outbuf, alpha );
125 mlt_properties_set_data( properties, "alpha", alpha, iwidth * ( iheight + 1 ), ( mlt_destructor )mlt_pool_release, NULL );
126 iformat = PIX_FMT_YUYV422;
127 avpicture_fill( &input, outbuf, iformat, iwidth, iheight );
128 avpicture_fill( &output, *image, oformat, owidth, oheight );
132 // Create the context and output image
133 struct SwsContext *context = sws_getContext( iwidth, iheight, iformat, owidth, oheight, oformat, interp, NULL, NULL, NULL);
136 // Perform the scaling
137 sws_scale( context, input.data, input.linesize, 0, iheight, output.data, output.linesize);
138 sws_freeContext( context );
140 // Now update the frame
141 mlt_properties_set_data( properties, "image", output.data[0], owidth * ( oheight + 1 ) * 2, ( mlt_destructor )mlt_pool_release, NULL );
142 mlt_properties_set_int( properties, "width", owidth );
143 mlt_properties_set_int( properties, "height", oheight );
146 *image = output.data[0];
148 // Scale the alpha channel only if exists and not correct size
150 mlt_properties_get_data( properties, "alpha", &alpha_size );
151 if ( alpha_size > 0 && alpha_size != ( owidth * oheight ) )
153 // Create the context and output image
154 uint8_t *alpha = mlt_frame_get_alpha_mask( this );
157 iformat = oformat = PIX_FMT_GRAY8;
158 struct SwsContext *context = sws_getContext( iwidth, iheight, iformat, owidth, oheight, oformat, interp, NULL, NULL, NULL);
159 avpicture_fill( &input, alpha, iformat, iwidth, iheight );
160 outbuf = mlt_pool_alloc( owidth * oheight );
161 avpicture_fill( &output, outbuf, oformat, owidth, oheight );
163 // Perform the scaling
164 sws_scale( context, input.data, input.linesize, 0, iheight, output.data, output.linesize);
165 sws_freeContext( context );
167 // Set it back on the frame
168 mlt_properties_set_data( MLT_FRAME_PROPERTIES( this ), "alpha", output.data[0], owidth * oheight, mlt_pool_release, NULL );
175 /** Constructor for the filter.
178 mlt_filter filter_swscale_init( mlt_profile profile, void *arg )
180 // Create a new scaler
181 mlt_filter this = mlt_factory_filter( profile, "rescale", arg );
183 // If successful, then initialise it
186 // Get the properties
187 mlt_properties properties = MLT_FILTER_PROPERTIES( this );
189 // Set the inerpolation
190 mlt_properties_set( properties, "interpolation", arg == NULL ? "bilinear" : arg );
193 mlt_properties_set_data( properties, "method", filter_scale, 0, NULL, NULL );