git.sesse.net Git - vlc/blob - 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 = p_image->p_filter->pf_vout_buffer_new( p_image->p_filter );
 455         if( p_pif ) vout_CopyPicture( p_image->p_parent, p_pif, p_pic );
 456     }
 457
 458     return p_pif;
 459 }
 460
 461 /**
 462  * Filter an image with a psz_module filter
 463  *
 464  */
 465
 466 static picture_t *ImageFilter( image_handler_t *p_image, picture_t *p_pic,
 467                                video_format_t *p_fmt, const char *psz_module )
 468 {
 469     void (*pf_release)( picture_t * );
 470     picture_t *p_pif;
 471
 472     /* Start a filter */
 473     if( !p_image->p_filter )
 474     {
 475         es_format_t fmt;
 476         es_format_Init( &fmt, VIDEO_ES, p_fmt->i_chroma );
 477         fmt.video = *p_fmt;
 478
 479         p_image->p_filter =
 480             CreateFilter( p_image->p_parent, &fmt, &fmt.video, psz_module );
 481
 482         if( !p_image->p_filter )
 483         {
 484             return NULL;
 485         }
 486     }
 487     else
 488     {
 489         /* Filters should handle on-the-fly size changes */
 490         p_image->p_filter->fmt_in.video = *p_fmt;
 491         p_image->p_filter->fmt_out.video = *p_fmt;
 492     }
 493
 494     pf_release = p_pic->pf_release;
 495     p_pic->pf_release = PicRelease; /* Small hack */
 496     p_pif = p_image->p_filter->pf_video_filter( p_image->p_filter, p_pic );
 497     p_pic->pf_release = pf_release;
 498
 499     return p_pif;
 500 }
 501
 502 /**
 503  * Misc functions
 504  *
 505  */
 506 static struct
 507 {
 508     vlc_fourcc_t i_codec;
 509     const char *psz_ext;
 510
 511 } ext_table[] =
 512 {
 513     { VLC_FOURCC('j','p','e','g'), "jpeg" },
 514     { VLC_FOURCC('j','p','e','g'), "jpg"  },
 515     { VLC_FOURCC('l','j','p','g'), "ljpg" },
 516     { VLC_FOURCC('p','n','g',' '), "png" },
 517     { VLC_FOURCC('p','g','m',' '), "pgm" },
 518     { VLC_FOURCC('p','g','m','y'), "pgmyuv" },
 519     { VLC_FOURCC('p','b','m',' '), "pbm" },
 520     { VLC_FOURCC('p','a','m',' '), "pam" },
 521     { VLC_FOURCC('t','g','a',' '), "tga" },
 522     { VLC_FOURCC('b','m','p',' '), "bmp" },
 523     { VLC_FOURCC('p','n','m',' '), "pnm" },
 524     { VLC_FOURCC('x','p','m',' '), "xpm" },
 525     { VLC_FOURCC('x','c','f',' '), "xcf" },
 526     { VLC_FOURCC('p','c','x',' '), "pcx" },
 527     { VLC_FOURCC('g','i','f',' '), "gif" },
 528     { VLC_FOURCC('t','i','f','f'), "tif" },
 529     { VLC_FOURCC('t','i','f','f'), "tiff" },
 530     { VLC_FOURCC('l','b','m',' '), "lbm" },
 531     { 0, NULL }
 532 };
 533
 534 static vlc_fourcc_t Ext2Fourcc( const char *psz_name )
 535 {
 536     int i;
 537
 538     psz_name = strrchr( psz_name, '.' );
 539     if( !psz_name ) return 0;
 540     psz_name++;
 541
 542     for( i = 0; ext_table[i].i_codec; i++ )
 543     {
 544         int j;
 545         for( j = 0; toupper(ext_table[i].psz_ext[j]) == toupper(psz_name[j]);
 546              j++ )
 547         {
 548             if( !ext_table[i].psz_ext[j] && !psz_name[j] )
 549                 return ext_table[i].i_codec;
 550         }
 551     }
 552
 553     return 0;
 554 }
 555
 556 /*
 557 static const char *Fourcc2Ext( vlc_fourcc_t i_codec )
 558 {
 559     int i;
 560
 561     for( i = 0; ext_table[i].i_codec != 0; i++ )
 562     {
 563         if( ext_table[i].i_codec == i_codec ) return ext_table[i].psz_ext;
 564     }
 565
 566     return NULL;
 567 }
 568 */
 569
 570 static void video_release_buffer( picture_t *p_pic )
 571 {
 572     if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig );
 573     if( p_pic && p_pic->p_sys ) free( p_pic->p_sys );
 574     if( p_pic ) free( p_pic );
 575 }
 576
 577 static picture_t *video_new_buffer( decoder_t *p_dec )
 578 {
 579     picture_t *p_pic = malloc( sizeof(picture_t) );
 580
 581     p_dec->fmt_out.video.i_chroma = p_dec->fmt_out.i_codec;
 582     vout_AllocatePicture( VLC_OBJECT(p_dec), p_pic,
 583                           p_dec->fmt_out.video.i_chroma,
 584                           p_dec->fmt_out.video.i_width,
 585                           p_dec->fmt_out.video.i_height,
 586                           p_dec->fmt_out.video.i_aspect );
 587
 588     if( !p_pic->i_planes )
 589     {
 590         free( p_pic );
 591         return 0;
 592     }
 593
 594     p_pic->pf_release = video_release_buffer;
 595     p_pic->i_status = RESERVED_PICTURE;
 596     p_pic->p_sys = NULL;
 597
 598     return p_pic;
 599 }
 600
 601 static void video_del_buffer( decoder_t *p_dec, picture_t *p_pic )
 602 {
 603     (void)p_dec;
 604     if( p_pic && p_pic->p_data_orig ) free( p_pic->p_data_orig );
 605     if( p_pic && p_pic->p_sys ) free( p_pic->p_sys );
 606     if( p_pic ) free( p_pic );
 607 }
 608
 609 static void video_link_picture( decoder_t *p_dec, picture_t *p_pic )
 610 {
 611     (void)p_dec; (void)p_pic;
 612 }
 613
 614 static void video_unlink_picture( decoder_t *p_dec, picture_t *p_pic )
 615 {
 616     (void)p_dec; (void)p_pic;
 617 }
 618
 619 static decoder_t *CreateDecoder( vlc_object_t *p_this, video_format_t *fmt )
 620 {
 621     decoder_t *p_dec;
 622
 623     p_dec = vlc_object_create( p_this, VLC_OBJECT_DECODER );
 624     if( p_dec == NULL )
 625     {
 626         msg_Err( p_this, "out of memory" );
 627         return NULL;
 628     }
 629
 630     p_dec->p_module = NULL;
 631     es_format_Init( &p_dec->fmt_in, VIDEO_ES, fmt->i_chroma );
 632     es_format_Init( &p_dec->fmt_out, VIDEO_ES, 0 );
 633     p_dec->fmt_in.video = *fmt;
 634     p_dec->b_pace_control = VLC_TRUE;
 635
 636     p_dec->pf_vout_buffer_new = video_new_buffer;
 637     p_dec->pf_vout_buffer_del = video_del_buffer;
 638     p_dec->pf_picture_link    = video_link_picture;
 639     p_dec->pf_picture_unlink  = video_unlink_picture;
 640
 641     vlc_object_attach( p_dec, p_this );
 642
 643     /* Find a suitable decoder module */
 644     p_dec->p_module = module_Need( p_dec, "decoder", "$codec", 0 );
 645     if( !p_dec->p_module )
 646     {
 647         msg_Err( p_dec, "no suitable decoder module for fourcc `%4.4s'.\n"
 648                  "VLC probably does not support this image format.",
 649                  (char*)&p_dec->fmt_in.i_codec );
 650
 651         DeleteDecoder( p_dec );
 652         return NULL;
 653     }
 654
 655     return p_dec;
 656 }
 657
 658 static void DeleteDecoder( decoder_t * p_dec )
 659 {
 660     vlc_object_detach( p_dec );
 661
 662     if( p_dec->p_module ) module_Unneed( p_dec, p_dec->p_module );
 663
 664     es_format_Clean( &p_dec->fmt_in );
 665     es_format_Clean( &p_dec->fmt_out );
 666
 667     vlc_object_destroy( p_dec );
 668     p_dec = NULL;
 669 }
 670
 671 static encoder_t *CreateEncoder( vlc_object_t *p_this, video_format_t *fmt_in,
 672                                  video_format_t *fmt_out )
 673 {
 674     encoder_t *p_enc;
 675
 676     p_enc = vlc_object_create( p_this, VLC_OBJECT_ENCODER );
 677     if( p_enc == NULL )
 678     {
 679         msg_Err( p_this, "out of memory" );
 680         return NULL;
 681     }
 682
 683     p_enc->p_module = NULL;
 684     es_format_Init( &p_enc->fmt_in, VIDEO_ES, fmt_in->i_chroma );
 685     p_enc->fmt_in.video = *fmt_in;
 686     if( fmt_out->i_width > 0 && fmt_out->i_height > 0 )
 687     {
 688         p_enc->fmt_in.video.i_width = fmt_out->i_width;
 689         p_enc->fmt_in.video.i_height = fmt_out->i_height;
 690
 691         if( fmt_out->i_visible_width > 0 &&
 692             fmt_out->i_visible_height > 0 )
 693         {
 694             p_enc->fmt_in.video.i_visible_width = fmt_out->i_visible_width;
 695             p_enc->fmt_in.video.i_visible_height = fmt_out->i_visible_height;
 696         }
 697         else
 698         {
 699             p_enc->fmt_in.video.i_visible_width = fmt_out->i_width;
 700             p_enc->fmt_in.video.i_visible_height = fmt_out->i_height;
 701         }
 702     }
 703     else if( fmt_out->i_sar_num && fmt_out->i_sar_den &&
 704              fmt_out->i_sar_num * fmt_in->i_sar_den !=
 705              fmt_out->i_sar_den * fmt_in->i_sar_num )
 706     {
 707         p_enc->fmt_in.video.i_width =
 708             fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den * fmt_in->i_width /
 709             fmt_in->i_sar_den / fmt_out->i_sar_num;
 710         p_enc->fmt_in.video.i_visible_width =
 711             fmt_in->i_sar_num * (int64_t)fmt_out->i_sar_den *
 712             fmt_in->i_visible_width / fmt_in->i_sar_den / fmt_out->i_sar_num;
 713     }
 714
 715     p_enc->fmt_in.video.i_frame_rate = 25;
 716     p_enc->fmt_in.video.i_frame_rate_base = 1;
 717
 718     es_format_Init( &p_enc->fmt_out, VIDEO_ES, fmt_out->i_chroma );
 719     p_enc->fmt_out.video = *fmt_out;
 720     p_enc->fmt_out.video.i_width = p_enc->fmt_in.video.i_width;
 721     p_enc->fmt_out.video.i_height = p_enc->fmt_in.video.i_height;
 722
 723     vlc_object_attach( p_enc, p_this );
 724
 725     /* Find a suitable decoder module */
 726     p_enc->p_module = module_Need( p_enc, "encoder", 0, 0 );
 727     if( !p_enc->p_module )
 728     {
 729         msg_Err( p_enc, "no suitable encoder module for fourcc `%4.4s'.\n"
 730                  "VLC probably does not support this image format.",
 731                  (char*)&p_enc->fmt_out.i_codec );
 732
 733         DeleteEncoder( p_enc );
 734         return NULL;
 735     }
 736     p_enc->fmt_in.video.i_chroma = p_enc->fmt_in.i_codec;
 737
 738     return p_enc;
 739 }
 740
 741 static void DeleteEncoder( encoder_t * p_enc )
 742 {
 743     vlc_object_detach( p_enc );
 744
 745     if( p_enc->p_module ) module_Unneed( p_enc, p_enc->p_module );
 746
 747     es_format_Clean( &p_enc->fmt_in );
 748     es_format_Clean( &p_enc->fmt_out );
 749
 750     vlc_object_destroy( p_enc );
 751     p_enc = NULL;
 752 }
 753
 754 static filter_t *CreateFilter( vlc_object_t *p_this, es_format_t *p_fmt_in,
 755                                video_format_t *p_fmt_out,
 756                                const char *psz_module )
 757 {
 758     filter_t *p_filter;
 759
 760     p_filter = vlc_object_create( p_this, VLC_OBJECT_FILTER );
 761     vlc_object_attach( p_filter, p_this );
 762
 763     p_filter->pf_vout_buffer_new =
 764         (picture_t *(*)(filter_t *))video_new_buffer;
 765     p_filter->pf_vout_buffer_del =
 766         (void (*)(filter_t *, picture_t *))video_del_buffer;
 767
 768     p_filter->fmt_in = *p_fmt_in;
 769     p_filter->fmt_out = *p_fmt_in;
 770     p_filter->fmt_out.i_codec = p_fmt_out->i_chroma;
 771     p_filter->fmt_out.video = *p_fmt_out;
 772     p_filter->p_module = module_Need( p_filter, "video filter2",
 773                                       psz_module, 0 );
 774
 775     if( !p_filter->p_module )
 776     {
 777         msg_Dbg( p_filter, "no video filter found" );
 778         DeleteFilter( p_filter );
 779         return NULL;
 780     }
 781
 782     return p_filter;
 783 }
 784
 785 static void DeleteFilter( filter_t * p_filter )
 786 {
 787     vlc_object_detach( p_filter );
 788
 789     if( p_filter->p_module ) module_Unneed( p_filter, p_filter->p_module );
 790
 791     es_format_Clean( &p_filter->fmt_in );
 792     es_format_Clean( &p_filter->fmt_out );
 793
 794     vlc_object_destroy( p_filter );
 795     p_filter = NULL;
 796 }