]> git.sesse.net Git - mlt/blob - src/modules/gtk2/filter_rescale.c
Refactor to use mlt_frame_set_image/_alpha.
[mlt] / src / modules / gtk2 / filter_rescale.c
1 /*
2  * filter_rescale.c -- scale the producer video frame size to match the consumer
3  * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
4  * Author: Dan Dennedy <dan@dennedy.org>
5  *
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.
10  *
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.
15  *
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
19  */
20
21 #include "pixops.h"
22
23 #include <framework/mlt_filter.h>
24 #include <framework/mlt_frame.h>
25 #include <framework/mlt_factory.h>
26
27 #include <stdio.h>
28 #include <string.h>
29 #include <stdlib.h>
30 #include <gdk-pixbuf/gdk-pixbuf.h>
31
32 static int filter_scale( mlt_frame this, uint8_t **image, mlt_image_format *format, int iwidth, int iheight, int owidth, int oheight )
33 {
34         // Get the properties
35         mlt_properties properties = MLT_FRAME_PROPERTIES( this );
36
37         // Get the requested interpolation method
38         char *interps = mlt_properties_get( properties, "rescale.interp" );
39
40         // Convert to the GTK flag
41         int interp = PIXOPS_INTERP_BILINEAR;
42
43         if ( strcmp( interps, "nearest" ) == 0 )
44                 interp = PIXOPS_INTERP_NEAREST;
45         else if ( strcmp( interps, "tiles" ) == 0 )
46                 interp = PIXOPS_INTERP_TILES;
47         else if ( strcmp( interps, "hyper" ) == 0 || strcmp( interps, "bicubic" ) == 0 )
48                 interp = PIXOPS_INTERP_HYPER;
49
50         // Carry out the rescaling
51         switch ( *format )
52         {
53         case mlt_image_yuv422:
54         {
55                 // Create the output image
56                 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * 2 );
57
58                 // Calculate strides
59                 int istride = iwidth * 2;
60                 int ostride = owidth * 2;
61
62                 yuv422_scale_simple( output, owidth, oheight, ostride, *image, iwidth, iheight, istride, interp );
63                 
64                 // Now update the frame
65                 mlt_frame_set_image( this, output, owidth * ( oheight + 1 ) * 2, mlt_pool_release );
66
67                 // Return the output
68                 *image = output;
69                 break;
70         }
71         case mlt_image_rgb24:
72         case mlt_image_rgb24a:
73         case mlt_image_opengl:
74         {
75                 int bpp = ( *format == mlt_image_rgb24 ? 3 : 4 );
76
77                 // Create the output image
78                 uint8_t *output = mlt_pool_alloc( owidth * ( oheight + 1 ) * bpp );
79
80                 if ( strcmp( interps, "none" ) && ( iwidth != owidth || iheight != oheight ) )
81                 {
82                         GdkPixbuf *pixbuf = gdk_pixbuf_new_from_data( *image, GDK_COLORSPACE_RGB,
83                                 ( *format == mlt_image_rgb24a || *format == mlt_image_opengl ), 8, iwidth, iheight,
84                                 iwidth * bpp, NULL, NULL );
85                         GdkPixbuf *scaled = gdk_pixbuf_scale_simple( pixbuf, owidth, oheight, interp );
86                         g_object_unref( pixbuf );
87
88                         int src_stride = gdk_pixbuf_get_rowstride( scaled );
89                         int dst_stride = owidth * bpp;
90                         if ( src_stride != dst_stride )
91                         {
92                                 int y = oheight;
93                                 uint8_t *src = gdk_pixbuf_get_pixels( scaled );
94                                 uint8_t *dst = output;
95                                 while ( y-- )
96                                 {
97                                         memcpy( dst, src, dst_stride );
98                                         dst += dst_stride;
99                                         src += src_stride;
100                                 }
101                         }
102                         else
103                         {
104                                 memcpy( output, gdk_pixbuf_get_pixels( scaled ), owidth * oheight * bpp );
105                         }
106
107                         g_object_unref( scaled );
108
109                         // Now update the frame
110                         mlt_frame_set_image( this, output, owidth * ( oheight + 1 ) * bpp, mlt_pool_release );
111         
112                         // Return the output
113                         *image = output;
114                 }
115                 break;
116         }
117         default:
118                 break;
119         }
120
121         return 0;
122 }
123
124 /** Constructor for the filter.
125 */
126
127 mlt_filter filter_rescale_init( mlt_profile profile, char *arg )
128 {
129         // Create a new scaler
130         mlt_filter this = mlt_factory_filter( profile, "rescale", arg );
131
132         // If successful, then initialise it
133         if ( this != NULL )
134         {
135                 // Get the properties
136                 mlt_properties properties = MLT_FILTER_PROPERTIES( this );
137
138                 // Set the inerpolation
139                 mlt_properties_set( properties, "interpolation", arg == NULL ? "bilinear" : arg );
140
141                 // Set the method
142                 mlt_properties_set_data( properties, "method", filter_scale, 0, NULL, NULL );
143         }
144
145         return this;
146 }
147