git.sesse.net Git - vlc/blob - modules/video_output/vmem.c

   1 /*****************************************************************************
   2  * vmem.c: memory video driver for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2008 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Sam 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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27
  28 #ifdef HAVE_CONFIG_H
  29 # include "config.h"
  30 #endif
  31
  32 #include <vlc_common.h>
  33 #include <vlc_plugin.h>
  34 #include <vlc_vout.h>
  35
  36 /*****************************************************************************
  37  * Local prototypes
  38  *****************************************************************************/
  39 static int  Create    ( vlc_object_t * );
  40 static void Destroy   ( vlc_object_t * );
  41
  42 static int  Init          ( vout_thread_t * );
  43 static void End           ( vout_thread_t * );
  44 static int  LockPicture   ( vout_thread_t *, picture_t * );
  45 static int  UnlockPicture ( vout_thread_t *, picture_t * );
  46
  47 /*****************************************************************************
  48  * Module descriptor
  49  *****************************************************************************/
  50 #define T_WIDTH N_( "Width" )
  51 #define LT_WIDTH N_( "Video memory buffer width." )
  52
  53 #define T_HEIGHT N_( "Height" )
  54 #define LT_HEIGHT N_( "Video memory buffer height." )
  55
  56 #define T_PITCH N_( "Pitch" )
  57 #define LT_PITCH N_( "Video memory buffer pitch in bytes." )
  58
  59 #define T_CHROMA N_( "Chroma" )
  60 #define LT_CHROMA N_( "Output chroma for the memory image as a 4-character " \
  61                       "string, eg. \"RV32\"." )
  62
  63 #define T_LOCK N_( "Lock function" )
  64 #define LT_LOCK N_( "Address of the locking callback function. This " \
  65                     "function must fill in valid plane memory address " \
  66                     "information for use by the video renderer." )
  67
  68 #define T_UNLOCK N_( "Unlock function" )
  69 #define LT_UNLOCK N_( "Address of the unlocking callback function" )
  70
  71 #define T_DATA N_( "Callback data" )
  72 #define LT_DATA N_( "Data for the locking and unlocking functions" )
  73
  74 vlc_module_begin ()
  75     set_description( N_( "Video memory output" ) )
  76     set_shortname( N_("Video memory") )
  77
  78     set_category( CAT_VIDEO )
  79     set_subcategory( SUBCAT_VIDEO_VOUT )
  80     set_capability( "video output", 0 )
  81
  82     add_integer( "vmem-width", 320, NULL, T_WIDTH, LT_WIDTH, false )
  83     add_integer( "vmem-height", 200, NULL, T_HEIGHT, LT_HEIGHT, false )
  84     add_integer( "vmem-pitch", 640, NULL, T_PITCH, LT_PITCH, false )
  85     add_string( "vmem-chroma", "RV16", NULL, T_CHROMA, LT_CHROMA, true )
  86     add_string( "vmem-lock", "0", NULL, T_LOCK, LT_LOCK, true )
  87     add_string( "vmem-unlock", "0", NULL, T_UNLOCK, LT_UNLOCK, true )
  88     add_string( "vmem-data", "0", NULL, T_DATA, LT_DATA, true )
  89
  90     set_callbacks( Create, Destroy )
  91 vlc_module_end ()
  92
  93 /*****************************************************************************
  94  * vout_sys_t: video output descriptor
  95  *****************************************************************************/
  96 struct vout_sys_t
  97 {
  98     int i_width, i_height, i_pitch;
  99
 100     void (*pf_lock) (void *, void **);
 101     void (*pf_unlock) (void *);
 102     void *p_data;
 103 };
 104
 105 /*****************************************************************************
 106  * Create: allocates video thread
 107  *****************************************************************************
 108  * This function allocates and initializes a vout method.
 109  *****************************************************************************/
 110 static int Create( vlc_object_t *p_this )
 111 {
 112     vout_thread_t *p_vout = ( vout_thread_t * )p_this;
 113
 114     /* Allocate instance and initialize some members */
 115     p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
 116     if( ! p_vout->p_sys )
 117         return VLC_ENOMEM;
 118
 119     p_vout->pf_init = Init;
 120     p_vout->pf_end = End;
 121     p_vout->pf_manage = NULL;
 122     p_vout->pf_render = NULL;
 123     p_vout->pf_display = NULL;
 124
 125     return VLC_SUCCESS;
 126 }
 127
 128 /*****************************************************************************
 129  * Init: initialize video thread
 130  *****************************************************************************/
 131 static int Init( vout_thread_t *p_vout )
 132 {
 133     int i_index;
 134     picture_t *p_pic;
 135     char *psz_chroma, *psz_tmp;
 136     int i_width, i_height, i_pitch, i_chroma;
 137
 138     i_width = var_CreateGetInteger( p_vout, "vmem-width" );
 139     i_height = var_CreateGetInteger( p_vout, "vmem-height" );
 140     i_pitch = var_CreateGetInteger( p_vout, "vmem-pitch" );
 141
 142     psz_chroma = var_CreateGetString( p_vout, "vmem-chroma" );
 143     if( psz_chroma )
 144     {
 145         if( strlen( psz_chroma ) < 4 )
 146         {
 147             msg_Err( p_vout, "vmem-chroma should be 4 characters long" );
 148             free( psz_chroma );
 149             return VLC_EGENERIC;
 150         }
 151         i_chroma = vlc_fourcc_GetCodec( VIDEO_ES,
 152                                         VLC_FOURCC( psz_chroma[0], psz_chroma[1],
 153                                                     psz_chroma[2], psz_chroma[3] ) );
 154         free( psz_chroma );
 155     }
 156     else
 157     {
 158         msg_Err( p_vout, "Cannot find chroma information." );
 159         return VLC_EGENERIC;
 160     }
 161
 162     psz_tmp = var_CreateGetString( p_vout, "vmem-lock" );
 163     p_vout->p_sys->pf_lock = (void (*) (void *, void **))(intptr_t)atoll( psz_tmp );
 164     free( psz_tmp );
 165
 166     psz_tmp = var_CreateGetString( p_vout, "vmem-unlock" );
 167     p_vout->p_sys->pf_unlock = (void (*) (void *))(intptr_t)atoll( psz_tmp );
 168     free( psz_tmp );
 169
 170     psz_tmp = var_CreateGetString( p_vout, "vmem-data" );
 171     p_vout->p_sys->p_data = (void *)(intptr_t)atoll( psz_tmp );
 172     free( psz_tmp );
 173
 174     if( !p_vout->p_sys->pf_lock || !p_vout->p_sys->pf_unlock )
 175     {
 176         msg_Err( p_vout, "Invalid lock or unlock callbacks" );
 177         return VLC_EGENERIC;
 178     }
 179
 180     I_OUTPUTPICTURES = 0;
 181
 182     /* Initialize the output structure */
 183     p_vout->output.i_chroma = i_chroma;
 184     p_vout->output.pf_setpalette = NULL;
 185     p_vout->output.i_width = i_width;
 186     p_vout->output.i_height = i_height;
 187     p_vout->output.i_aspect = p_vout->output.i_width
 188                                * VOUT_ASPECT_FACTOR / p_vout->output.i_height;
 189
 190     /* Define the bitmasks */
 191     switch( i_chroma )
 192     {
 193       case VLC_CODEC_RGB15:
 194         p_vout->output.i_rmask = 0x001f;
 195         p_vout->output.i_gmask = 0x03e0;
 196         p_vout->output.i_bmask = 0x7c00;
 197         break;
 198
 199       case VLC_CODEC_RGB16:
 200         p_vout->output.i_rmask = 0x001f;
 201         p_vout->output.i_gmask = 0x07e0;
 202         p_vout->output.i_bmask = 0xf800;
 203         break;
 204
 205       case VLC_CODEC_RGB24:
 206         p_vout->output.i_rmask = 0xff0000;
 207         p_vout->output.i_gmask = 0x00ff00;
 208         p_vout->output.i_bmask = 0x0000ff;
 209         break;
 210
 211       case VLC_CODEC_RGB32:
 212         p_vout->output.i_rmask = 0xff0000;
 213         p_vout->output.i_gmask = 0x00ff00;
 214         p_vout->output.i_bmask = 0x0000ff;
 215         break;
 216     }
 217
 218     /* Try to initialize 1 direct buffer */
 219     p_pic = NULL;
 220
 221     /* Find an empty picture slot */
 222     for( i_index = 0 ; i_index < VOUT_MAX_PICTURES ; i_index++ )
 223     {
 224         if( p_vout->p_picture[ i_index ].i_status == FREE_PICTURE )
 225         {
 226             p_pic = p_vout->p_picture + i_index;
 227             break;
 228         }
 229     }
 230
 231     /* Allocate the picture */
 232     if( p_pic == NULL )
 233     {
 234         return VLC_SUCCESS;
 235     }
 236
 237     vout_InitPicture( VLC_OBJECT(p_vout), p_pic, p_vout->output.i_chroma,
 238                       p_vout->output.i_width, p_vout->output.i_height,
 239                       p_vout->output.i_aspect );
 240
 241     p_pic->p->i_pitch = i_pitch;
 242
 243     p_pic->pf_lock = LockPicture;
 244     p_pic->pf_unlock = UnlockPicture;
 245
 246     p_pic->i_status = DESTROYED_PICTURE;
 247     p_pic->i_type   = DIRECT_PICTURE;
 248
 249     PP_OUTPUTPICTURE[ I_OUTPUTPICTURES ] = p_pic;
 250
 251     I_OUTPUTPICTURES++;
 252
 253     return VLC_SUCCESS;
 254 }
 255
 256 /*****************************************************************************
 257  * End: terminate video thread output method
 258  *****************************************************************************/
 259 static void End( vout_thread_t *p_vout )
 260 {
 261     (void)p_vout;
 262 }
 263
 264 /*****************************************************************************
 265  * Destroy: destroy video thread
 266  *****************************************************************************
 267  * Terminate an output method created by Create
 268  *****************************************************************************/
 269 static void Destroy( vlc_object_t *p_this )
 270 {
 271     vout_thread_t *p_vout = ( vout_thread_t * )p_this;
 272
 273     /* Destroy structure */
 274     free( p_vout->p_sys );
 275 }
 276
 277 /*****************************************************************************
 278  * LockPicture: lock a picture
 279  *****************************************************************************
 280  * This function locks the picture and prepares it for direct pixel access.
 281  *****************************************************************************/
 282 static int LockPicture( vout_thread_t *p_vout, picture_t *p_pic )
 283 {
 284     int i_index;
 285     void *planes[p_pic->i_planes];
 286
 287     p_vout->p_sys->pf_lock( p_vout->p_sys->p_data, planes );
 288
 289     for( i_index = 0; i_index < p_pic->i_planes; i_index++ )
 290     {
 291         p_pic->p[i_index].p_pixels = planes[i_index];
 292     }
 293
 294     return VLC_SUCCESS;
 295 }
 296
 297 /*****************************************************************************
 298  * UnlockPicture: unlock a picture
 299  *****************************************************************************
 300  * This function unlocks a previously locked picture.
 301  *****************************************************************************/
 302 static int UnlockPicture( vout_thread_t *p_vout, picture_t *p_pic )
 303 {
 304     p_vout->p_sys->pf_unlock( p_vout->p_sys->p_data );
 305
 306     (void)p_pic;
 307
 308     return VLC_SUCCESS;
 309 }
 310