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