git.sesse.net Git - vlc/blob - plugins/filter/deinterlace.c

   1 /*****************************************************************************
   2  * deinterlace.c : deinterlacer plugin for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2000, 2001 VideoLAN
   5  * $Id: deinterlace.c,v 1.2 2002/01/02 14:37:42 sam Exp $
   6  *
   7  * Authors: Samuel Hocevar <sam@zoy.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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include <errno.h>
  28 #include <stdlib.h>                                      /* malloc(), free() */
  29 #include <string.h>
  30
  31 #include <videolan/vlc.h>
  32
  33 #include "video.h"
  34 #include "video_output.h"
  35
  36 #include "filter_common.h"
  37
  38 #define DEINTERLACE_MODE_BOB     1
  39 #define DEINTERLACE_MODE_BLEND   2
  40
  41 /*****************************************************************************
  42  * Capabilities defined in the other files.
  43  *****************************************************************************/
  44 static void vout_getfunctions( function_list_t * p_function_list );
  45
  46 static void *memblend( void *, const void *, const void *, size_t );
  47
  48 /*****************************************************************************
  49  * Build configuration tree.
  50  *****************************************************************************/
  51 MODULE_CONFIG_START
  52 MODULE_CONFIG_STOP
  53
  54 MODULE_INIT_START
  55     SET_DESCRIPTION( "deinterlacing module" )
  56     /* Capability score set to 0 because we don't want to be spawned
  57      * as a video output unless explicitly requested to */
  58     ADD_CAPABILITY( VOUT, 0 )
  59     ADD_SHORTCUT( "deinterlace" )
  60 MODULE_INIT_STOP
  61
  62 MODULE_ACTIVATE_START
  63     vout_getfunctions( &p_module->p_functions->vout );
  64 MODULE_ACTIVATE_STOP
  65
  66 MODULE_DEACTIVATE_START
  67 MODULE_DEACTIVATE_STOP
  68
  69 /*****************************************************************************
  70  * vout_sys_t: Deinterlace video output method descriptor
  71  *****************************************************************************
  72  * This structure is part of the video output thread descriptor.
  73  * It describes the Deinterlace specific properties of an output thread.
  74  *****************************************************************************/
  75 typedef struct vout_sys_s
  76 {
  77     int i_mode;
  78     struct vout_thread_s *p_vout;
  79
  80 } vout_sys_t;
  81
  82 /*****************************************************************************
  83  * Local prototypes
  84  *****************************************************************************/
  85 static int  vout_Probe     ( probedata_t *p_data );
  86 static int  vout_Create    ( struct vout_thread_s * );
  87 static int  vout_Init      ( struct vout_thread_s * );
  88 static void vout_End       ( struct vout_thread_s * );
  89 static void vout_Destroy   ( struct vout_thread_s * );
  90 static int  vout_Manage    ( struct vout_thread_s * );
  91 static void vout_Display   ( struct vout_thread_s *, struct picture_s * );
  92
  93 /*****************************************************************************
  94  * Functions exported as capabilities. They are declared as static so that
  95  * we don't pollute the namespace too much.
  96  *****************************************************************************/
  97 static void vout_getfunctions( function_list_t * p_function_list )
  98 {
  99     p_function_list->pf_probe = vout_Probe;
 100     p_function_list->functions.vout.pf_create     = vout_Create;
 101     p_function_list->functions.vout.pf_init       = vout_Init;
 102     p_function_list->functions.vout.pf_end        = vout_End;
 103     p_function_list->functions.vout.pf_destroy    = vout_Destroy;
 104     p_function_list->functions.vout.pf_manage     = vout_Manage;
 105     p_function_list->functions.vout.pf_display    = vout_Display;
 106     p_function_list->functions.vout.pf_setpalette = NULL;
 107 }
 108
 109 /*****************************************************************************
 110  * intf_Probe: return a score
 111  *****************************************************************************/
 112 static int vout_Probe( probedata_t *p_data )
 113 {
 114     return( 0 );
 115 }
 116
 117 /*****************************************************************************
 118  * vout_Create: allocates Deinterlace video thread output method
 119  *****************************************************************************
 120  * This function allocates and initializes a Deinterlace vout method.
 121  *****************************************************************************/
 122 static int vout_Create( vout_thread_t *p_vout )
 123 {
 124     char *psz_method;
 125
 126     /* Allocate structure */
 127     p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
 128     if( p_vout->p_sys == NULL )
 129     {
 130         intf_ErrMsg("error: %s", strerror(ENOMEM) );
 131         return( 1 );
 132     }
 133
 134     /* Look what method was requested */
 135     psz_method = main_GetPszVariable( VOUT_FILTER_VAR, "" );
 136
 137     while( *psz_method && *psz_method != ':' )
 138     {
 139         psz_method++;
 140     }
 141
 142     if( !strcmp( psz_method, ":bob" ) )
 143     {
 144         p_vout->p_sys->i_mode = DEINTERLACE_MODE_BOB;
 145     }
 146     else if( !strcmp( psz_method, ":blend" ) )
 147     {
 148         p_vout->p_sys->i_mode = DEINTERLACE_MODE_BLEND;
 149     }
 150     else
 151     {
 152         intf_ErrMsg( "filter error: no valid deinterlace mode provided, "
 153                      "using deinterlace:bob" );
 154         p_vout->p_sys->i_mode = DEINTERLACE_MODE_BOB;
 155     }
 156
 157     return( 0 );
 158 }
 159
 160 /*****************************************************************************
 161  * vout_Init: initialize Deinterlace video thread output method
 162  *****************************************************************************/
 163 static int vout_Init( vout_thread_t *p_vout )
 164 {
 165     int i_index;
 166     char *psz_filter;
 167     picture_t *p_pic;
 168
 169     I_OUTPUTPICTURES = 0;
 170
 171     /* Initialize the output structure, full of directbuffers since we want
 172      * the decoder to output directly to our structures. */
 173     switch( p_vout->render.i_chroma )
 174     {
 175         case FOURCC_I420:
 176         case FOURCC_IYUV:
 177         case FOURCC_YV12:
 178         case FOURCC_I422:
 179             p_vout->output.i_chroma = p_vout->render.i_chroma;
 180             p_vout->output.i_width  = p_vout->render.i_width;
 181             p_vout->output.i_height = p_vout->render.i_height;
 182             p_vout->output.i_aspect = p_vout->render.i_aspect;
 183             break;
 184
 185         default:
 186             return( 0 ); /* unknown chroma */
 187             break;
 188     }
 189
 190     /* Try to open the real video output, with half the height our images */
 191     psz_filter = main_GetPszVariable( VOUT_FILTER_VAR, "" );
 192     main_PutPszVariable( VOUT_FILTER_VAR, "" );
 193
 194     intf_WarnMsg( 1, "filter: spawning the real video output" );
 195
 196     switch( p_vout->render.i_chroma )
 197     {
 198     case FOURCC_I420:
 199     case FOURCC_IYUV:
 200     case FOURCC_YV12:
 201         switch( p_vout->p_sys->i_mode )
 202         {
 203         case DEINTERLACE_MODE_BOB:
 204             p_vout->p_sys->p_vout =
 205                 vout_CreateThread( NULL,
 206                        p_vout->output.i_width, p_vout->output.i_height / 2,
 207                        p_vout->output.i_chroma, p_vout->output.i_aspect );
 208             break;
 209
 210         case DEINTERLACE_MODE_BLEND:
 211             p_vout->p_sys->p_vout =
 212                 vout_CreateThread( NULL,
 213                        p_vout->output.i_width, p_vout->output.i_height,
 214                        p_vout->output.i_chroma, p_vout->output.i_aspect );
 215             break;
 216         }
 217         break;
 218
 219     case FOURCC_I422:
 220         p_vout->p_sys->p_vout =
 221             vout_CreateThread( NULL,
 222                        p_vout->output.i_width, p_vout->output.i_height,
 223                        FOURCC_I420, p_vout->output.i_aspect );
 224         break;
 225
 226     default:
 227         break;
 228     }
 229
 230     /* Everything failed */
 231     if( p_vout->p_sys->p_vout == NULL )
 232     {
 233         intf_ErrMsg( "filter error: can't open vout, aborting" );
 234
 235         return( 0 );
 236     }
 237
 238     main_PutPszVariable( VOUT_FILTER_VAR, psz_filter );
 239
 240     ALLOCATE_DIRECTBUFFERS( VOUT_MAX_PICTURES );
 241
 242     return( 0 );
 243 }
 244
 245 /*****************************************************************************
 246  * vout_End: terminate Deinterlace video thread output method
 247  *****************************************************************************/
 248 static void vout_End( vout_thread_t *p_vout )
 249 {
 250     int i_index;
 251
 252     /* Free the fake output buffers we allocated */
 253     for( i_index = I_OUTPUTPICTURES ; i_index ; )
 254     {
 255         i_index--;
 256         free( PP_OUTPUTPICTURE[ i_index ]->planes[ 0 ].p_data );
 257     }
 258 }
 259
 260 /*****************************************************************************
 261  * vout_Destroy: destroy Deinterlace video thread output method
 262  *****************************************************************************
 263  * Terminate an output method created by DeinterlaceCreateOutputMethod
 264  *****************************************************************************/
 265 static void vout_Destroy( vout_thread_t *p_vout )
 266 {
 267     vout_DestroyThread( p_vout->p_sys->p_vout, NULL );
 268
 269     free( p_vout->p_sys );
 270 }
 271
 272 /*****************************************************************************
 273  * vout_Manage: handle Deinterlace events
 274  *****************************************************************************
 275  * This function should be called regularly by video output thread. It manages
 276  * console events. It returns a non null value on error.
 277  *****************************************************************************/
 278 static int vout_Manage( vout_thread_t *p_vout )
 279 {
 280     return( 0 );
 281 }
 282
 283 /*****************************************************************************
 284  * vout_Display: displays previously rendered output
 285  *****************************************************************************
 286  * This function send the currently rendered image to Deinterlace image,
 287  * waits until it is displayed and switch the two rendering buffers, preparing
 288  * next frame.
 289  *****************************************************************************/
 290 static void vout_Display( vout_thread_t *p_vout, picture_t *p_pic )
 291 {
 292     picture_t *p_outpic;
 293     int i_index, i_field;
 294
 295     for( i_field = 0 ; i_field < 2 ; i_field++ )
 296     {
 297         /* Get a structure from the video_output. */
 298         while( ( p_outpic = vout_CreatePicture( p_vout->p_sys->p_vout,
 299                                                 0, 0, 0 ) )
 300                   == NULL )
 301         {
 302             if( p_vout->b_die || p_vout->b_error )
 303             {
 304                 return;
 305             }
 306             msleep( VOUT_OUTMEM_SLEEP );
 307         }
 308
 309         /* XXX: completely arbitrary values ! */
 310         vout_DatePicture( p_vout->p_sys->p_vout, p_outpic,
 311                           mdate() + (mtime_t)(50000 + i_field * 20000) );
 312
 313         /* Copy image and skip lines */
 314         for( i_index = 0 ; i_index < p_pic->i_planes ; i_index++ )
 315         {
 316             pixel_data_t *p_in, *p_out_end, *p_out;
 317             int i_increment;
 318
 319             p_in = p_pic->planes[ i_index ].p_data
 320                        + i_field * p_pic->planes[ i_index ].i_line_bytes;
 321
 322             p_out = p_outpic->planes[ i_index ].p_data;
 323             p_out_end = p_out + p_outpic->planes[ i_index ].i_bytes;
 324
 325             switch( p_vout->render.i_chroma )
 326             {
 327             case FOURCC_I420:
 328             case FOURCC_IYUV:
 329             case FOURCC_YV12:
 330
 331                 switch( p_vout->p_sys->i_mode )
 332                 {
 333                 case DEINTERLACE_MODE_BOB:
 334                     for( ; p_out < p_out_end ; )
 335                     {
 336                         FAST_MEMCPY( p_out, p_in,
 337                                      p_pic->planes[ i_index ].i_line_bytes );
 338
 339                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 340                         p_in += 2 * p_pic->planes[ i_index ].i_line_bytes;
 341                     }
 342                     break;
 343
 344                 case DEINTERLACE_MODE_BLEND:
 345                     if( i_index != Y_PLANE )
 346                     {
 347                         for( ; p_out < p_out_end ; )
 348                         {
 349                             FAST_MEMCPY( p_out, p_in,
 350                                      p_pic->planes[ i_index ].i_line_bytes );
 351
 352                             p_out += p_pic->planes[ i_index ].i_line_bytes;
 353
 354                             FAST_MEMCPY( p_out, p_in,
 355                                      p_pic->planes[ i_index ].i_line_bytes );
 356
 357                             p_out += p_pic->planes[ i_index ].i_line_bytes;
 358                             p_in += 2 * p_pic->planes[ i_index ].i_line_bytes;
 359                         }
 360                         break;
 361                     }
 362
 363                     if( i_field == 0 )
 364                     {
 365                         FAST_MEMCPY( p_out, p_in,
 366                                      p_pic->planes[ i_index ].i_line_bytes );
 367                         p_in += 2 * p_pic->planes[ i_index ].i_line_bytes;
 368                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 369                     }
 370
 371                     for( ; p_out < p_out_end ; )
 372                     {
 373                         FAST_MEMCPY( p_out, p_in,
 374                                      p_pic->planes[ i_index ].i_line_bytes );
 375
 376                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 377
 378                         memblend( p_out, p_in, p_in + 2 * p_pic->planes[ i_index ].i_line_bytes, p_pic->planes[ i_index ].i_line_bytes );
 379
 380                         p_in += 2 * p_pic->planes[ i_index ].i_line_bytes;
 381                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 382                     }
 383                     break;
 384                 }
 385                 break;
 386
 387             case FOURCC_I422:
 388
 389                 i_increment = 2 * p_pic->planes[ i_index ].i_line_bytes;
 390
 391                 if( i_index == Y_PLANE )
 392                 {
 393                     for( ; p_out < p_out_end ; )
 394                     {
 395                         FAST_MEMCPY( p_out, p_in,
 396                                  p_pic->planes[ i_index ].i_line_bytes );
 397                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 398                         FAST_MEMCPY( p_out, p_in,
 399                                  p_pic->planes[ i_index ].i_line_bytes );
 400                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 401                         p_in += i_increment;
 402                     }
 403                 }
 404                 else
 405                 {
 406                     for( ; p_out < p_out_end ; )
 407                     {
 408                         FAST_MEMCPY( p_out, p_in,
 409                                  p_pic->planes[ i_index ].i_line_bytes );
 410                         p_out += p_pic->planes[ i_index ].i_line_bytes;
 411                         p_in += i_increment;
 412                     }
 413                 }
 414                 break;
 415
 416             default:
 417                 break;
 418             }
 419         }
 420
 421         vout_DisplayPicture( p_vout->p_sys->p_vout, p_outpic );
 422     }
 423 }
 424
 425 static void *memblend( void *p_dest, const void *p_s1,
 426                        const void *p_s2, size_t i_bytes )
 427 {
 428     u8* p_end = (u8*)p_dest + i_bytes - 8;
 429
 430     while( (u8*)p_dest < p_end )
 431     {
 432         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 433         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 434         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 435         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 436         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 437         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 438         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 439         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 440     }
 441
 442     p_end += 8;
 443
 444     while( (u8*)p_dest < p_end )
 445     {
 446         *(u8*)p_dest++ = ( (u16)(*(u8*)p_s1++) + (u16)(*(u8*)p_s2++) ) >> 1;
 447     }
 448
 449     return p_dest;
 450 }
 451