* now 0.6.0-cvs
[vlc.git] / modules / video_filter / transform.c
1 /*****************************************************************************
2  * transform.c : transform image plugin for vlc
3  *****************************************************************************
4  * Copyright (C) 2000, 2001, 2002, 2003 VideoLAN
5  * $Id: transform.c,v 1.12 2003/05/15 22:27:37 massiot Exp $
6  *
7  * Authors: Samuel Hocevar <sam@zoy.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <stdlib.h>                                      /* malloc(), free() */
28 #include <string.h>
29
30 #include <vlc/vlc.h>
31 #include <vlc/vout.h>
32
33 #include "filter_common.h"
34
35 #define TRANSFORM_MODE_HFLIP   1
36 #define TRANSFORM_MODE_VFLIP   2
37 #define TRANSFORM_MODE_90      3
38 #define TRANSFORM_MODE_180     4
39 #define TRANSFORM_MODE_270     5
40
41 /*****************************************************************************
42  * Local prototypes
43  *****************************************************************************/
44 static int  Create    ( vlc_object_t * );
45 static void Destroy   ( vlc_object_t * );
46
47 static int  Init      ( vout_thread_t * );
48 static void End       ( vout_thread_t * );
49 static void Render    ( vout_thread_t *, picture_t * );
50
51 static int  SendEvents( vlc_object_t *, char const *,
52                         vlc_value_t, vlc_value_t, void * );
53
54 /*****************************************************************************
55  * Module descriptor
56  *****************************************************************************/
57 #define TYPE_TEXT N_("Transform type")
58 #define TYPE_LONGTEXT N_("One of '90', '180', '270', 'hflip' and 'vflip'")
59
60 static char *type_list[] = { "90", "180", "270", "hflip", "vflip", NULL };
61
62 vlc_module_begin();
63     add_category_hint( N_("Miscellaneous"), NULL, VLC_FALSE );
64     add_string_from_list( "transform-type", "90", type_list, NULL,
65                           TYPE_TEXT, TYPE_LONGTEXT, VLC_FALSE);
66     set_description( _("video transformation filter") );
67     set_capability( "video filter", 0 );
68     add_shortcut( "transform" );
69     set_callbacks( Create, Destroy );
70 vlc_module_end();
71
72 /*****************************************************************************
73  * vout_sys_t: Transform video output method descriptor
74  *****************************************************************************
75  * This structure is part of the video output thread descriptor.
76  * It describes the Transform specific properties of an output thread.
77  *****************************************************************************/
78 struct vout_sys_t
79 {
80     int i_mode;
81     vlc_bool_t b_rotation;
82     vout_thread_t *p_vout;
83 };
84
85 /*****************************************************************************
86  * Create: allocates Transform video thread output method
87  *****************************************************************************
88  * This function allocates and initializes a Transform vout method.
89  *****************************************************************************/
90 static int Create( vlc_object_t *p_this )
91 {
92     vout_thread_t *p_vout = (vout_thread_t *)p_this;
93     char *psz_method;
94
95     /* Allocate structure */
96     p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
97     if( p_vout->p_sys == NULL )
98     {
99         msg_Err( p_vout, "out of memory" );
100         return VLC_ENOMEM;
101     }
102
103     p_vout->pf_init = Init;
104     p_vout->pf_end = End;
105     p_vout->pf_manage = NULL;
106     p_vout->pf_render = Render;
107     p_vout->pf_display = NULL;
108
109     /* Look what method was requested */
110     psz_method = config_GetPsz( p_vout, "transform-type" );
111
112     if( psz_method == NULL )
113     {
114         msg_Err( p_vout, "configuration variable %s empty", "transform-type" );
115         msg_Err( p_vout, "no valid transform mode provided, using '90'" );
116         p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
117         p_vout->p_sys->b_rotation = 1;
118     }
119     else
120     {
121         if( !strcmp( psz_method, "hflip" ) )
122         {
123             p_vout->p_sys->i_mode = TRANSFORM_MODE_HFLIP;
124             p_vout->p_sys->b_rotation = 0;
125         }
126         else if( !strcmp( psz_method, "vflip" ) )
127         {
128             p_vout->p_sys->i_mode = TRANSFORM_MODE_VFLIP;
129             p_vout->p_sys->b_rotation = 0;
130         }
131         else if( !strcmp( psz_method, "90" ) )
132         {
133             p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
134             p_vout->p_sys->b_rotation = 1;
135         }
136         else if( !strcmp( psz_method, "180" ) )
137         {
138             p_vout->p_sys->i_mode = TRANSFORM_MODE_180;
139             p_vout->p_sys->b_rotation = 0;
140         }
141         else if( !strcmp( psz_method, "270" ) )
142         {
143             p_vout->p_sys->i_mode = TRANSFORM_MODE_270;
144             p_vout->p_sys->b_rotation = 1;
145         }
146         else
147         {
148             msg_Err( p_vout, "no valid transform mode provided, using '90'" );
149             p_vout->p_sys->i_mode = TRANSFORM_MODE_90;
150             p_vout->p_sys->b_rotation = 1;
151         }
152
153         free( psz_method );
154     }
155
156     return VLC_SUCCESS;
157 }
158
159 /*****************************************************************************
160  * Init: initialize Transform video thread output method
161  *****************************************************************************/
162 static int Init( vout_thread_t *p_vout )
163 {
164     int i_index;
165     picture_t *p_pic;
166
167     I_OUTPUTPICTURES = 0;
168
169     /* Initialize the output structure */
170     p_vout->output.i_chroma = p_vout->render.i_chroma;
171     p_vout->output.i_width  = p_vout->render.i_width;
172     p_vout->output.i_height = p_vout->render.i_height;
173     p_vout->output.i_aspect = p_vout->render.i_aspect;
174
175     /* Try to open the real video output */
176     msg_Dbg( p_vout, "spawning the real video output" );
177
178     if( p_vout->p_sys->b_rotation )
179     {
180         p_vout->p_sys->p_vout = vout_Create( p_vout,
181                            p_vout->render.i_height, p_vout->render.i_width,
182                            p_vout->render.i_chroma,
183                            (uint64_t)VOUT_ASPECT_FACTOR
184                             * (uint64_t)VOUT_ASPECT_FACTOR
185                             / (uint64_t)p_vout->render.i_aspect );
186     }
187     else
188     {
189         p_vout->p_sys->p_vout = vout_Create( p_vout,
190                            p_vout->render.i_width, p_vout->render.i_height,
191                            p_vout->render.i_chroma, p_vout->render.i_aspect );
192     }
193
194     /* Everything failed */
195     if( p_vout->p_sys->p_vout == NULL )
196     {
197         msg_Err( p_vout, "cannot open vout, aborting" );
198         return VLC_EGENERIC;
199     }
200
201     ALLOCATE_DIRECTBUFFERS( VOUT_MAX_PICTURES );
202
203     ADD_CALLBACKS( p_vout->p_sys->p_vout, SendEvents );
204
205     return VLC_SUCCESS;
206 }
207
208 /*****************************************************************************
209  * End: terminate Transform video thread output method
210  *****************************************************************************/
211 static void End( vout_thread_t *p_vout )
212 {
213     int i_index;
214
215     /* Free the fake output buffers we allocated */
216     for( i_index = I_OUTPUTPICTURES ; i_index ; )
217     {
218         i_index--;
219         free( PP_OUTPUTPICTURE[ i_index ]->p_data_orig );
220     }
221 }
222
223 /*****************************************************************************
224  * Destroy: destroy Transform video thread output method
225  *****************************************************************************
226  * Terminate an output method created by TransformCreateOutputMethod
227  *****************************************************************************/
228 static void Destroy( vlc_object_t *p_this )
229 {
230     vout_thread_t *p_vout = (vout_thread_t *)p_this;
231
232     DEL_CALLBACKS( p_vout->p_sys->p_vout, SendEvents );
233     vlc_object_detach( p_vout->p_sys->p_vout );
234     vout_Destroy( p_vout->p_sys->p_vout );
235
236     free( p_vout->p_sys );
237 }
238
239 /*****************************************************************************
240  * Render: displays previously rendered output
241  *****************************************************************************
242  * This function send the currently rendered image to Transform image, waits
243  * until it is displayed and switch the two rendering buffers, preparing next
244  * frame.
245  *****************************************************************************/
246 static void Render( vout_thread_t *p_vout, picture_t *p_pic )
247 {
248     picture_t *p_outpic;
249     int i_index;
250
251     /* This is a new frame. Get a structure from the video_output. */
252     while( ( p_outpic = vout_CreatePicture( p_vout->p_sys->p_vout, 0, 0, 0 ) )
253               == NULL )
254     {
255         if( p_vout->b_die || p_vout->b_error )
256         {
257             return;
258         }
259         msleep( VOUT_OUTMEM_SLEEP );
260     }
261
262     vout_DatePicture( p_vout->p_sys->p_vout, p_outpic, p_pic->date );
263     vout_LinkPicture( p_vout->p_sys->p_vout, p_outpic );
264
265     switch( p_vout->p_sys->i_mode )
266     {
267         case TRANSFORM_MODE_90:
268             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
269             {
270                 int i_pitch = p_pic->p[i_index].i_pitch;
271
272                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
273
274                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
275                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
276                                               * p_outpic->p[i_index].i_pitch;
277
278                 for( ; p_out < p_out_end ; )
279                 {
280                     uint8_t *p_line_end;
281
282                     p_out_end -= p_outpic->p[i_index].i_pitch
283                                   - p_outpic->p[i_index].i_visible_pitch;
284                     p_line_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
285
286                     for( ; p_in < p_line_end ; )
287                     {
288                         p_line_end -= i_pitch;
289                         *(--p_out_end) = *p_line_end;
290                     }
291
292                     p_in++;
293                 }
294             }
295             break;
296
297         case TRANSFORM_MODE_180:
298             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
299             {
300                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
301                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
302                                             * p_pic->p[i_index].i_pitch;
303
304                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
305
306                 for( ; p_in < p_in_end ; )
307                 {
308                     uint8_t *p_line_start = p_in_end
309                                              - p_pic->p[i_index].i_pitch;
310                     p_in_end -= p_pic->p[i_index].i_pitch
311                                  - p_pic->p[i_index].i_visible_pitch;
312
313                     for( ; p_line_start < p_in_end ; )
314                     {
315                         *p_out++ = *(--p_in_end);
316                     }
317
318                     p_out += p_outpic->p[i_index].i_pitch
319                               - p_outpic->p[i_index].i_visible_pitch;
320                 }
321             }
322             break;
323
324         case TRANSFORM_MODE_270:
325             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
326             {
327                 int i_pitch = p_pic->p[i_index].i_pitch;
328
329                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
330
331                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
332                 uint8_t *p_out_end = p_out + p_outpic->p[i_index].i_lines
333                                               * p_outpic->p[i_index].i_pitch;
334
335                 for( ; p_out < p_out_end ; )
336                 {
337                     uint8_t *p_in_end;
338
339                     p_in_end = p_in + p_pic->p[i_index].i_lines * i_pitch;
340
341                     for( ; p_in < p_in_end ; )
342                     {
343                         p_in_end -= i_pitch;
344                         *p_out++ = *p_in_end;
345                     }
346
347                     p_out += p_outpic->p[i_index].i_pitch
348                               - p_outpic->p[i_index].i_visible_pitch;
349                     p_in++;
350                 }
351             }
352             break;
353
354         case TRANSFORM_MODE_VFLIP:
355             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
356             {
357                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
358                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
359                                             * p_pic->p[i_index].i_pitch;
360
361                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
362
363                 for( ; p_in < p_in_end ; )
364                 {
365                     p_in_end -= p_pic->p[i_index].i_pitch;
366                     p_vout->p_vlc->pf_memcpy( p_out, p_in_end,
367                                            p_pic->p[i_index].i_visible_pitch );
368                     p_out += p_pic->p[i_index].i_pitch;
369                 }
370             }
371             break;
372
373         case TRANSFORM_MODE_HFLIP:
374             for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
375             {
376                 uint8_t *p_in = p_pic->p[i_index].p_pixels;
377                 uint8_t *p_in_end = p_in + p_pic->p[i_index].i_lines
378                                             * p_pic->p[i_index].i_pitch;
379
380                 uint8_t *p_out = p_outpic->p[i_index].p_pixels;
381
382                 for( ; p_in < p_in_end ; )
383                 {
384                     uint8_t *p_line_end = p_in
385                                            + p_pic->p[i_index].i_visible_pitch;
386
387                     for( ; p_in < p_line_end ; )
388                     {
389                         *p_out++ = *(--p_line_end);
390                     }
391
392                     p_in += p_pic->p[i_index].i_pitch;
393                 }
394             }
395             break;
396
397         default:
398             break;
399     }
400
401     vout_UnlinkPicture( p_vout->p_sys->p_vout, p_outpic );
402
403     vout_DisplayPicture( p_vout->p_sys->p_vout, p_outpic );
404 }
405
406 /*****************************************************************************
407  * SendEvents: forward mouse and keyboard events to the parent p_vout
408  *****************************************************************************/
409 static int SendEvents( vlc_object_t *p_this, char const *psz_var,
410                        vlc_value_t oldval, vlc_value_t newval, void *_p_vout )
411 {
412     vout_thread_t *p_vout = (vout_thread_t *)_p_vout;
413     vlc_value_t sentval = newval;
414
415     /* Translate the mouse coordinates */
416     if( !strcmp( psz_var, "mouse-x" ) )
417     {
418         switch( p_vout->p_sys->i_mode )
419         {
420         case TRANSFORM_MODE_270:
421             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
422                              - sentval.i_int;
423         case TRANSFORM_MODE_90:
424             var_Set( p_vout, "mouse-y", sentval );
425             return VLC_SUCCESS;
426
427         case TRANSFORM_MODE_180:
428         case TRANSFORM_MODE_HFLIP:
429             sentval.i_int = p_vout->p_sys->p_vout->output.i_width
430                              - sentval.i_int;
431             break;
432
433         case TRANSFORM_MODE_VFLIP:
434         default:
435             break;
436         }
437     }
438     else if( !strcmp( psz_var, "mouse-y" ) )
439     {
440         switch( p_vout->p_sys->i_mode )
441         {
442         case TRANSFORM_MODE_90:
443             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
444                              - sentval.i_int;
445         case TRANSFORM_MODE_270:
446             var_Set( p_vout, "mouse-x", sentval );
447             return VLC_SUCCESS;
448
449         case TRANSFORM_MODE_180:
450         case TRANSFORM_MODE_VFLIP:
451             sentval.i_int = p_vout->p_sys->p_vout->output.i_height
452                              - sentval.i_int;
453             break;
454
455         case TRANSFORM_MODE_HFLIP:
456         default:
457             break;
458         }
459     }
460
461     var_Set( p_vout, psz_var, sentval );
462
463     return VLC_SUCCESS;
464 }
465