]> git.sesse.net Git - vlc/blob - src/misc/image.c
Fix memleak.
[vlc] / src / misc / image.c
1 /*****************************************************************************
2  * image.c : wrapper for image reading/writing facilities
3  *****************************************************************************
4  * Copyright (C) 2004-2007 the VideoLAN team
5  * $Id$
6  *
7  * Author: Gildas Bazin <gbazin@videolan.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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /**
25  * \file
26  * This file contains the functions to handle the image_handler_t type
27  */
28
29 /*****************************************************************************
30  * Preamble
31  *****************************************************************************/
32 #include <ctype.h>
33 #include <errno.h>
34 #include <vlc/vlc.h>
35 #include <vlc_codec.h>
36 #include <vlc_filter.h>
37 #include <vlc_es.h>
38 #include <vlc_image.h>
39 #include <vlc_stream.h>
40 #include <vlc_charset.h>
41
42 static picture_t *ImageRead( image_handler_t *, block_t *,
43                              video_format_t *, video_format_t * );
44 static picture_t *ImageReadUrl( image_handler_t *, const char *,
45                                 video_format_t *, video_format_t * );
46 static block_t *ImageWrite( image_handler_t *, picture_t *,
47                             video_format_t *, video_format_t * );
48 static int ImageWriteUrl( image_handler_t *, picture_t *,
49                           video_format_t *, video_format_t *, const char * );
50
51 static picture_t *ImageConvert( image_handler_t *, picture_t *,
52                                 video_format_t *, video_format_t * );
53 static picture_t *ImageFilter( image_handler_t *, picture_t *,
54                                video_format_t *, const char *psz_module );
55
56 static decoder_t *CreateDecoder( vlc_object_t *, video_format_t * );
57 static void DeleteDecoder( decoder_t * );
58 static encoder_t *CreateEncoder( vlc_object_t *, video_format_t *,
59                                  video_format_t * );
60 static void DeleteEncoder( encoder_t * );
61 static filter_t *CreateFilter( vlc_object_t *, es_format_t *,
62                                video_format_t *, const char * );
63 static void DeleteFilter( filter_t * );
64
65 static vlc_fourcc_t Ext2Fourcc( const char * );
66 /*static const char *Fourcc2Ext( vlc_fourcc_t );*/
67
68 /**
69  * Create an image_handler_t instance
70  *
71  */
72 image_handler_t *__image_HandlerCreate( vlc_object_t *p_this )
73 {
74     image_handler_t *p_image = malloc( sizeof(image_handler_t) );
75
76     memset( p_image, 0, sizeof(image_handler_t) );
77     p_image->p_parent = p_this;
78
79     p_image->pf_read = ImageRead;
80     p_image->pf_read_url = ImageReadUrl;
81     p_image->pf_write = ImageWrite;
82     p_image->pf_write_url = ImageWriteUrl;
83     p_image->pf_convert = ImageConvert;
84     p_image->pf_filter = ImageFilter;
85
86     return p_image;
87 }
88
89 /**
90  * Delete the image_handler_t instance
91  *
92  */
93 void image_HandlerDelete( image_handler_t *p_image )
94 {
95     if( !p_image ) return;
96
97     if( p_image->p_dec ) DeleteDecoder( p_image->p_dec );
98     if( p_image->p_enc ) DeleteEncoder( p_image->p_enc );
99     if( p_image->p_filter ) DeleteFilter( p_image->p_filter );
100
101     free( p_image );
102     p_image = NULL;
103 }
104
105 /**
106  * Read an image
107  *
108  */
109
110 static picture_t *ImageRead( image_handler_t *p_image, block_t *p_block,
111                              video_format_t *p_fmt_in,
112                              video_format_t *p_fmt_out )
113 {
114     picture_t *p_pic = NULL, *p_tmp;
115
116     /* Check if we can reuse the current decoder */
117     if( p_image->p_dec &&
118         p_image->p_dec->fmt_in.i_codec != p_fmt_in->i_chroma )
119     {
120         DeleteDecoder( p_image->p_dec );
121         p_image->p_dec = 0;
122     }
123
124     /* Start a decoder */
125     if( !p_image->p_dec )
126     {
127         p_image->p_dec = CreateDecoder( p_image->p_parent, p_fmt_in );
128         if( !p_image->p_dec ) return NULL;
129     }
130
131     p_block->i_pts = p_block->i_dts = mdate();
132     while( (p_tmp = p_image->p_dec->pf_decode_video( p_image->p_dec, &p_block ))
133              != NULL )
134     {
135         if ( p_pic != NULL )
136             p_pic->pf_release( p_pic );
137         p_pic = p_tmp;
138     }
139
140     if( p_pic == NULL )
141     {
142         msg_Warn( p_image->p_parent, "no image decoded" );
143         return 0;
144     }
145
146     if( !p_fmt_out->i_chroma )
147         p_fmt_out->i_chroma = p_image->p_dec->fmt_out.video.i_chroma;
148     if( !p_fmt_out->i_width && p_fmt_out->i_height )
149         p_fmt_out->i_width = p_fmt_out->i_height
150                               * p_image->p_dec->fmt_out.video.i_aspect
151                               / VOUT_ASPECT_FACTOR;
152     if( !p_fmt_out->i_height && p_fmt_out->i_width )
153         p_fmt_out->i_height = p_fmt_out->i_width * VOUT_ASPECT_FACTOR
154                                / p_image->p_dec->fmt_out.video.i_aspect;
155     if( !p_fmt_out->i_width )
156         p_fmt_out->i_width = p_image->p_dec->fmt_out.video.i_width;
157     if( !p_fmt_out->i_height )
158         p_fmt_out->i_height = p_image->p_dec->fmt_out.video.i_height;
159
160     /* Check if we need chroma conversion or resizing */
161     if( p_image->p_dec->fmt_out.video.i_chroma != p_fmt_out->i_chroma ||
162         p_image->p_dec->fmt_out.video.i_width != p_fmt_out->i_width ||
163         p_image->p_dec->fmt_out.video.i_height != p_fmt_out->i_height )
164     {
165         if( p_image->p_filter )
166         if( p_image->p_filter->fmt_in.video.i_chroma !=
167             p_image->p_dec->fmt_out.video.i_chroma ||
168             p_image->p_filter->fmt_out.video.i_chroma != p_fmt_out->i_chroma )
169         {
170             /* We need to restart a new filter */
171             DeleteFilter( p_image->p_filter );
172             p_image->p_filter = 0;
173         }
174
175         /* Start a filter */
176         if( !p_image->p_filter )
177         {
178             p_image->p_filter =
179                 CreateFilter( p_image->p_parent, &p_image->p_dec->fmt_out,
180                               p_fmt_out, NULL );
181
182             if( !p_image->p_filter )
183             {
184                 p_pic->pf_release( p_pic );
185                 return NULL;
186             }
187         }
188         else
189         {
190             /* Filters should handle on-the-fly size changes */
191             p_image->p_filter->fmt_in = p_image->p_dec->fmt_out;
192             p_image->p_filter->fmt_out = p_image->p_dec->fmt_out;
193             p_image->p_filter->fmt_out.i_codec = p_fmt_out->i_chroma;
194             p_image->p_filter->fmt_out.video = *p_fmt_out;
195         }
196
197         p_pic = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
198         *p_fmt_out = p_image->p_filter->fmt_out.video;
199     }
200     else *p_fmt_out = p_image->p_dec->fmt_out.video;
201
202     return p_pic;
203 }
204
205 static picture_t *ImageReadUrl( image_handler_t *p_image, const char *psz_url,
206                                 video_format_t *p_fmt_in,
207                                 video_format_t *p_fmt_out )
208 {
209     block_t *p_block;
210     picture_t *p_pic;
211     stream_t *p_stream = NULL;
212     int i_size;
213
214     p_stream = stream_UrlNew( p_image->p_parent, psz_url );
215
216     if( !p_stream )
217     {
218         msg_Dbg( p_image->p_parent, "could not open %s for reading",
219                  psz_url );
220         return NULL;
221     }
222
223     i_size = stream_Size( p_stream );
224
225     p_block = block_New( p_image->p_parent, i_size );
226
227     stream_Read( p_stream, p_block->p_buffer, i_size );
228     stream_Delete( p_stream );
229
230     if( !p_fmt_in->i_chroma )
231     {
232         /* Try to guess format from file name */
233         p_fmt_in->i_chroma = Ext2Fourcc( psz_url );
234     }
235
236     p_pic = ImageRead( p_image, p_block, p_fmt_in, p_fmt_out );
237
238     return p_pic;
239 }
240
241 /**
242  * Write an image
243  *
244  */
245
246 static void PicRelease( picture_t *p_pic ) { (void)p_pic; }
247
248 static block_t *ImageWrite( image_handler_t *p_image, picture_t *p_pic,
249                             video_format_t *p_fmt_in,
250                             video_format_t *p_fmt_out )
251 {
252     block_t *p_block;
253     void (*pf_release)( picture_t * );
254
255     /* Check if we can reuse the current encoder */
256     if( p_image->p_enc &&
257         ( p_image->p_enc->fmt_out.i_codec != p_fmt_out->i_chroma ||
258           p_image->p_enc->fmt_out.video.i_width != p_fmt_out->i_width ||
259           p_image->p_enc->fmt_out.video.i_height != p_fmt_out->i_height ) )
260     {
261         DeleteEncoder( p_image->p_enc );
262         p_image->p_enc = 0;
263     }
264
265     /* Start an encoder */
266     if( !p_image->p_enc )
267     {
268         p_image->p_enc = CreateEncoder( p_image->p_parent,
269                                         p_fmt_in, p_fmt_out );
270         if( !p_image->p_enc ) return NULL;
271     }
272
273     /* Check if we need chroma conversion or resizing */
274     if( p_image->p_enc->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
275         p_image->p_enc->fmt_in.video.i_width != p_fmt_in->i_width ||
276         p_image->p_enc->fmt_in.video.i_height != p_fmt_in->i_height )
277     {
278         picture_t *p_tmp_pic;
279
280         if( p_image->p_filter )
281         if( p_image->p_filter->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
282             p_image->p_filter->fmt_out.video.i_chroma !=
283             p_image->p_enc->fmt_in.video.i_chroma )
284         {
285             /* We need to restart a new filter */
286             DeleteFilter( p_image->p_filter );
287             p_image->p_filter = 0;
288         }
289
290         /* Start a filter */
291         if( !p_image->p_filter )
292         {
293             es_format_t fmt_in;
294             es_format_Init( &fmt_in, VIDEO_ES, p_fmt_in->i_chroma );
295             fmt_in.video = *p_fmt_in;
296
297             p_image->p_filter =
298                 CreateFilter( p_image->p_parent, &fmt_in,
299                               &p_image->p_enc->fmt_in.video, NULL );
300
301             if( !p_image->p_filter )
302             {
303                 return NULL;
304             }
305         }
306         else
307         {
308             /* Filters should handle on-the-fly size changes */
309             p_image->p_filter->fmt_in.i_codec = p_fmt_in->i_chroma;
310             p_image->p_filter->fmt_out.video = *p_fmt_in;
311             p_image->p_filter->fmt_out.i_codec =p_image->p_enc->fmt_in.i_codec;
312             p_image->p_filter->fmt_out.video = p_image->p_enc->fmt_in.video;
313         }
314
315         pf_release = p_pic->pf_release;
316         p_pic->pf_release = PicRelease; /* Small hack */
317         p_tmp_pic =
318             p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
319         p_pic->pf_release = pf_release;
320
321         p_block = p_image->p_enc->pf_encode_video( p_image->p_enc, p_tmp_pic );
322
323         p_image->p_filter->pf_vout_buffer_del( p_image->p_filter, p_tmp_pic );
324     }
325     else
326     {
327         p_block = p_image->p_enc->pf_encode_video( p_image->p_enc, p_pic );
328     }
329
330     if( !p_block )
331     {
332         msg_Dbg( p_image->p_parent, "no image encoded" );
333         return 0;
334     }
335
336     return p_block;
337 }
338
339 static int ImageWriteUrl( image_handler_t *p_image, picture_t *p_pic,
340                           video_format_t *p_fmt_in, video_format_t *p_fmt_out,
341                           const char *psz_url )
342 {
343     block_t *p_block;
344     FILE *file;
345
346     if( !p_fmt_out->i_chroma )
347     {
348         /* Try to guess format from file name */
349         p_fmt_out->i_chroma = Ext2Fourcc( psz_url );
350     }
351
352     file = utf8_fopen( psz_url, "wb" );
353     if( !file )
354     {
355         msg_Err( p_image->p_parent, "%s: %s", psz_url, strerror( errno ) );
356         return VLC_EGENERIC;
357     }
358
359     p_block = ImageWrite( p_image, p_pic, p_fmt_in, p_fmt_out );
360
361     int err = 0;
362     if( p_block )
363     {
364         if( fwrite( p_block->p_buffer, p_block->i_buffer, 1, file ) != 1 )
365             err = errno;
366         block_Release( p_block );
367     }
368
369     if( fclose( file ) && !err )
370         err = errno;
371
372     if( err )
373        msg_Err( p_image->p_parent, "%s: %s", psz_url, strerror( err ) );
374
375     return err ? VLC_EGENERIC : VLC_SUCCESS;
376 }
377
378 /**
379  * Convert an image to a different format
380  *
381  */
382
383 static picture_t *ImageConvert( image_handler_t *p_image, picture_t *p_pic,
384                                 video_format_t *p_fmt_in,
385                                 video_format_t *p_fmt_out )
386 {
387     void (*pf_release)( picture_t * );
388     picture_t *p_pif;
389
390     if( !p_fmt_out->i_width && !p_fmt_out->i_height &&
391         p_fmt_out->i_sar_num && p_fmt_out->i_sar_den &&
392         p_fmt_out->i_sar_num * p_fmt_in->i_sar_den !=
393         p_fmt_out->i_sar_den * p_fmt_in->i_sar_num )
394     {
395         p_fmt_out->i_width =
396             p_fmt_in->i_sar_num * (int64_t)p_fmt_out->i_sar_den *
397             p_fmt_in->i_width / p_fmt_in->i_sar_den / p_fmt_out->i_sar_num;
398         p_fmt_out->i_visible_width =
399             p_fmt_in->i_sar_num * (int64_t)p_fmt_out->i_sar_den *
400             p_fmt_in->i_visible_width / p_fmt_in->i_sar_den /
401             p_fmt_out->i_sar_num;
402     }
403
404     if( !p_fmt_out->i_chroma ) p_fmt_out->i_chroma = p_fmt_in->i_chroma;
405     if( !p_fmt_out->i_width )
406         p_fmt_out->i_width = p_fmt_out->i_visible_width = p_fmt_in->i_width;
407     if( !p_fmt_out->i_height )
408         p_fmt_out->i_height = p_fmt_out->i_visible_height = p_fmt_in->i_height;
409     if( !p_fmt_out->i_sar_num ) p_fmt_out->i_sar_num = p_fmt_in->i_sar_num;
410     if( !p_fmt_out->i_sar_den ) p_fmt_out->i_sar_den = p_fmt_in->i_sar_den;
411     if( !p_fmt_out->i_aspect ) p_fmt_out->i_aspect = p_fmt_in->i_aspect;
412
413     if( p_image->p_filter )
414     if( p_image->p_filter->fmt_in.video.i_chroma != p_fmt_in->i_chroma ||
415         p_image->p_filter->fmt_out.video.i_chroma != p_fmt_out->i_chroma )
416     {
417         /* We need to restart a new filter */
418         DeleteFilter( p_image->p_filter );
419         p_image->p_filter = NULL;
420     }
421
422     /* Start a filter */
423     if( !p_image->p_filter )
424     {
425         es_format_t fmt_in;
426         es_format_Init( &fmt_in, VIDEO_ES, p_fmt_in->i_chroma );
427         fmt_in.video = *p_fmt_in;
428
429         p_image->p_filter =
430             CreateFilter( p_image->p_parent, &fmt_in, p_fmt_out, NULL );
431
432         if( !p_image->p_filter )
433         {
434             return NULL;
435         }
436     }
437     else
438     {
439         /* Filters should handle on-the-fly size changes */
440         p_image->p_filter->fmt_in.video = *p_fmt_in;
441         p_image->p_filter->fmt_out.video = *p_fmt_out;
442     }
443
444     pf_release = p_pic->pf_release;
445     p_pic->pf_release = PicRelease; /* Small hack */
446     p_pif = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
447     p_pic->pf_release = pf_release;
448
449     if( p_fmt_in->i_chroma == p_fmt_out->i_chroma &&
450         p_fmt_in->i_width == p_fmt_out->i_width &&
451         p_fmt_in->i_height == p_fmt_out->i_height )
452     {
453         /* Duplicate image */
454         p_pif->pf_release( p_pif ); /* XXX: Better fix must be possible */
455         p_pif = p_image->p_filter->pf_vout_buffer_new( p_image->p_filter );
456         if( p_pif ) vout_CopyPicture( p_image->p_parent, p_pif, p_pic );
457     }
458
459     return p_pif;
460 }
461
462 /**
463  * Filter an image with a psz_module filter
464  *
465  */
466
467 static picture_t *ImageFilter( image_handler_t *p_image, picture_t *p_pic,
468                                video_format_t *p_fmt, const char *psz_module )
469 {
470     void (*pf_release)( picture_t * );
471     picture_t *p_pif;
472
473     /* Start a filter */
474     if( !p_image->p_filter )
475     {
476         es_format_t fmt;
477         es_format_Init( &fmt, VIDEO_ES, p_fmt->i_chroma );
478         fmt.video = *p_fmt;
479
480         p_image->p_filter =
481             CreateFilter( p_image->p_parent, &fmt, &fmt.video, psz_module );
482
483         if( !p_image->p_filter )
484         {
485             return NULL;
486         }
487     }
488     else
489     {
490         /* Filters should handle on-the-fly size changes */
491         p_image->p_filter->fmt_in.video = *p_fmt;
492         p_image->p_filter->fmt_out.video = *p_fmt;
493     }
494
495     pf_release = p_pic->pf_release;
496     p_pic->pf_release = PicRelease; /* Small hack */
497     p_pif = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
498     p_pic->pf_release = pf_release;
499
500     return p_pif;
501 }
502
503 /**
504  * Misc functions
505  *
506  */
507 static struct
508 {
509     vlc_fourcc_t i_codec;
510     const char *psz_ext;
511
512 } ext_table[] =
513 {
514     { VLC_FOURCC('j','p','e','g'), "jpeg" },
515     { VLC_FOURCC('j','p','e','g'), "jpg"  },
516     { VLC_FOURCC('l','j','p','g'), "ljpg" },
517     { VLC_FOURCC('p','n','g',' '), "png" },
518     { VLC_FOURCC('p','g','m',' '), "pgm" },
519     { VLC_FOURCC('p','g','m','y'), "pgmyuv" },
520     { VLC_FOURCC('p','b','m',' '), "pbm" },
521     { VLC_FOURCC('p','a','m',' '), "pam" },
522     { VLC_FOURCC('t','g','a',' '), "tga" },
523     { VLC_FOURCC('b','m','p',' '), "bmp" },
524     { VLC_FOURCC('p','n','m',' '), "pnm" },
525     { VLC_FOURCC('x','p','m',' '), "xpm" },
526     { VLC_FOURCC('x','c','f',' '), "xcf" },
527     { VLC_FOURCC('p','c','x',' '), "pcx" },
528     { VLC_FOURCC('g','i','f',' '), "gif" },
529     { VLC_FOURCC('t','i','f','f'), "tif" },
530     { VLC_FOURCC('t','i','f','f'), "tiff" },
531     { VLC_FOURCC('l','b','m',' '), "lbm" },
532     { 0, NULL }
533 };
534
535 static vlc_fourcc_t Ext2Fourcc( const char *psz_name )
536 {
537     int i;
538
539     psz_name = strrchr( psz_name, '.' );
540     if( !psz_name ) return 0;
541     psz_name++;
542
543     for( i = 0; ext_table[i].i_codec; i++ )
544     {
545         int j;
546         for( j = 0; toupper(ext_table[i].psz_ext[j]) == toupper(psz_name[j]);
547              j++ )
548         {
549             if( !ext_table[i].psz_ext[j] && !psz_name[j] )
550                 return ext_table[i].i_codec;
551         }
552     }
553
554     return 0;
555 }
556
557 /*
558 static const char *Fourcc2Ext( vlc_fourcc_t i_codec )
559 {
560     int i;
561
562     for( i = 0; ext_table[i].i_codec != 0; i++ )
563     {
564         if( ext_table[i].i_codec == i_codec ) return ext_table[i].psz_ext;
565     }
566
567     return NULL;
568 }
569 */
570
571 static void video_release_buffer( picture_t *p_pic )
572 {
573     if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig );
574     if( p_pic && p_pic->p_sys ) free( p_pic->p_sys );
575     if( p_pic ) free( p_pic );
576 }
577
578 static picture_t *video_new_buffer( decoder_t *p_dec )
579 {
580     picture_t *p_pic = malloc( sizeof(picture_t) );
581
582     p_dec->fmt_out.video.i_chroma = p_dec->fmt_out.i_codec;
583     vout_AllocatePicture( VLC_OBJECT(p_dec), p_pic,
584                           p_dec->fmt_out.video.i_chroma,
585                           p_dec->fmt_out.video.i_width,
586                           p_dec->fmt_out.video.i_height,
587                           p_dec->fmt_out.video.i_aspect );
588
589     if( !p_pic->i_planes )
590     {
591         free( p_pic );
592         return 0;
593     }
594
595     p_pic->pf_release = video_release_buffer;
596     p_pic->i_status = RESERVED_PICTURE;
597     p_pic->p_sys = NULL;
598
599     return p_pic;
600 }
601
602 static void video_del_buffer( decoder_t *p_dec, picture_t *p_pic )
603 {
604     (void)p_dec;
605     if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig );
606     if( p_pic && p_pic->p_sys ) free( p_pic->p_sys );
607     if( p_pic ) free( p_pic );
608 }
609
610 static void video_link_picture( decoder_t *p_dec, picture_t *p_pic )
611 {
612     (void)p_dec; (void)p_pic;
613 }
614
615 static void video_unlink_picture( decoder_t *p_dec, picture_t *p_pic )
616 {
617     (void)p_dec; (void)p_pic;
618 }
619
620 static decoder_t *CreateDecoder( vlc_object_t *p_this, video_format_t *fmt )
621 {
622     decoder_t *p_dec;
623
624     p_dec = vlc_object_create( p_this, VLC_OBJECT_DECODER );
625     if( p_dec == NULL )
626     {
627         msg_Err( p_this, "out of memory" );
628         return NULL;
629     }
630
631     p_dec->p_module = NULL;
632     es_format_Init( &p_dec->fmt_in, VIDEO_ES, fmt->i_chroma );
633     es_format_Init( &p_dec->fmt_out, VIDEO_ES, 0 );
634     p_dec->fmt_in.video = *fmt;
635     p_dec->b_pace_control = VLC_TRUE;
636
637     p_dec->pf_vout_buffer_new = video_new_buffer;
638     p_dec->pf_vout_buffer_del = video_del_buffer;
639     p_dec->pf_picture_link    = video_link_picture;
640     p_dec->pf_picture_unlink  = video_unlink_picture;
641
642     vlc_object_attach( p_dec, p_this );
643
644     /* Find a suitable decoder module */
645     p_dec->p_module = module_Need( p_dec, "decoder", "$codec", 0 );
646     if( !p_dec->p_module )
647     {
648         msg_Err( p_dec, "no suitable decoder module for fourcc `%4.4s'.\n"
649                  "VLC probably does not support this image format.",
650                  (char*)&p_dec->fmt_in.i_codec );
651
652         DeleteDecoder( p_dec );
653         return NULL;
654     }
655
656     return p_dec;
657 }
658
659 static void DeleteDecoder( decoder_t * p_dec )
660 {
661     vlc_object_detach( p_dec );
662
663     if( p_dec->p_module ) module_Unneed( p_dec, p_dec->p_module );
664
665     es_format_Clean( &p_dec->fmt_in );
666     es_format_Clean( &p_dec->fmt_out );
667
668     vlc_object_destroy( p_dec );
669     p_dec = NULL;
670 }
671
672 static encoder_t *CreateEncoder( vlc_object_t *p_this, video_format_t *fmt_in,
673                                  video_format_t *fmt_out )
674 {
675     encoder_t *p_enc;
676
677     p_enc = vlc_object_create( p_this, VLC_OBJECT_ENCODER );
678     if( p_enc == NULL )
679     {
680         msg_Err( p_this, "out of memory" );
681         return NULL;
682     }
683
684     p_enc->p_module = NULL;
685     es_format_Init( &p_enc->fmt_in, VIDEO_ES, fmt_in->i_chroma );
686     p_enc->fmt_in.video = *fmt_in;
687     if( fmt_out->i_width > 0 && fmt_out->i_height > 0 )
688     {
689         p_enc->fmt_in.video.i_width = fmt_out->i_width;
690         p_enc->fmt_in.video.i_height = fmt_out->i_height;
691
692         if( fmt_out->i_visible_width > 0 &&
693             fmt_out->i_visible_height > 0 )
694         {
695             p_enc->fmt_in.video.i_visible_width = fmt_out->i_visible_width;
696             p_enc->fmt_in.video.i_visible_height = fmt_out->i_visible_height;
697         }
698         else
699         {
700             p_enc->fmt_in.video.i_visible_width = fmt_out->i_width;
701             p_enc->fmt_in.video.i_visible_height = fmt_out->i_height;
702         }
703     }
704     else if( fmt_out->i_sar_num && fmt_out->i_sar_den &&
705              fmt_out->i_sar_num * fmt_in->i_sar_den !=
706              fmt_out->i_sar_den * fmt_in->i_sar_num )
707     {
708         p_enc->fmt_in.video.i_width =
709             fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den * fmt_in->i_width /
710             fmt_in->i_sar_den / fmt_out->i_sar_num;
711         p_enc->fmt_in.video.i_visible_width =
712             fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den *
713             fmt_in->i_visible_width / fmt_in->i_sar_den / fmt_out->i_sar_num;
714     }
715
716     p_enc->fmt_in.video.i_frame_rate = 25;
717     p_enc->fmt_in.video.i_frame_rate_base = 1;
718
719     es_format_Init( &p_enc->fmt_out, VIDEO_ES, fmt_out->i_chroma );
720     p_enc->fmt_out.video = *fmt_out;
721     p_enc->fmt_out.video.i_width = p_enc->fmt_in.video.i_width;
722     p_enc->fmt_out.video.i_height = p_enc->fmt_in.video.i_height;
723
724     vlc_object_attach( p_enc, p_this );
725
726     /* Find a suitable decoder module */
727     p_enc->p_module = module_Need( p_enc, "encoder", 0, 0 );
728     if( !p_enc->p_module )
729     {
730         msg_Err( p_enc, "no suitable encoder module for fourcc `%4.4s'.\n"
731                  "VLC probably does not support this image format.",
732                  (char*)&p_enc->fmt_out.i_codec );
733
734         DeleteEncoder( p_enc );
735         return NULL;
736     }
737     p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec;
738
739     return p_enc;
740 }
741
742 static void DeleteEncoder( encoder_t * p_enc )
743 {
744     vlc_object_detach( p_enc );
745
746     if( p_enc->p_module ) module_Unneed( p_enc, p_enc->p_module );
747
748     es_format_Clean( &p_enc->fmt_in );
749     es_format_Clean( &p_enc->fmt_out );
750
751     vlc_object_destroy( p_enc );
752     p_enc = NULL;
753 }
754
755 static filter_t *CreateFilter( vlc_object_t *p_this, es_format_t *p_fmt_in,
756                                video_format_t *p_fmt_out,
757                                const char *psz_module )
758 {
759     filter_t *p_filter;
760
761     p_filter = vlc_object_create( p_this, VLC_OBJECT_FILTER );
762     vlc_object_attach( p_filter, p_this );
763
764     p_filter->pf_vout_buffer_new =
765         (picture_t *(*)(filter_t *))video_new_buffer;
766     p_filter->pf_vout_buffer_del =
767         (void (*)(filter_t *, picture_t *))video_del_buffer;
768
769     p_filter->fmt_in = *p_fmt_in;
770     p_filter->fmt_out = *p_fmt_in;
771     p_filter->fmt_out.i_codec = p_fmt_out->i_chroma;
772     p_filter->fmt_out.video = *p_fmt_out;
773     p_filter->p_module = module_Need( p_filter, "video filter2",
774                                       psz_module, 0 );
775
776     if( !p_filter->p_module )
777     {
778         msg_Dbg( p_filter, "no video filter found" );
779         DeleteFilter( p_filter );
780         return NULL;
781     }
782
783     return p_filter;
784 }
785
786 static void DeleteFilter( filter_t * p_filter )
787 {
788     vlc_object_detach( p_filter );
789
790     if( p_filter->p_module ) module_Unneed( p_filter, p_filter->p_module );
791
792     es_format_Clean( &p_filter->fmt_in );
793     es_format_Clean( &p_filter->fmt_out );
794
795     vlc_object_destroy( p_filter );
796     p_filter = NULL;
797 }