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