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