snapshot.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:14k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * snapshot.c : snapshot plugin for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2002 the VideoLAN team
  5.  * $Id: 2fc7c2af8f89885436fb48481c887723fef1b57a $
  6.  *
  7.  * Authors: Olivier Aubert <oaubert@lisi.univ-lyon1.fr>
  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.  * This module is a pseudo video output that offers the possibility to
  25.  * keep a cache of low-res snapshots.
  26.  * The snapshot structure is defined in include/snapshot.h
  27.  * In order to access the current snapshot cache, object variables are used:
  28.  *   vout-snapshot-list-pointer : the pointer on the first element in the list
  29.  *   vout-snapshot-datasize     : size of a snapshot
  30.  *                           (also available in snapshot_t->i_datasize)
  31.  *   vout-snapshot-cache-size   : size of the cache list
  32.  *
  33.  * It is used for the moment by the CORBA module and a specialized
  34.  * python-vlc binding.
  35.  *****************************************************************************/
  36. /*****************************************************************************
  37.  * Preamble
  38.  *****************************************************************************/
  39. #ifdef HAVE_CONFIG_H
  40. # include "config.h"
  41. #endif
  42. #include <vlc_common.h>
  43. #include <vlc_plugin.h>
  44. #include <vlc_vout.h>
  45. #include <vlc_interface.h>
  46. #include <vlc_input.h>
  47. /*****************************************************************************
  48.  * Local prototypes
  49.  *****************************************************************************/
  50. static int  Create    ( vlc_object_t * );
  51. static void Destroy   ( vlc_object_t * );
  52. static int  Init      ( vout_thread_t * );
  53. static void End       ( vout_thread_t * );
  54. static void Display   ( vout_thread_t *, picture_t * );
  55. /*****************************************************************************
  56.  * Module descriptor
  57.  *****************************************************************************/
  58. #define WIDTH_TEXT N_( "Snapshot width" )
  59. #define WIDTH_LONGTEXT N_( "Width of the snapshot image." )
  60. #define HEIGHT_TEXT N_( "Snapshot height" )
  61. #define HEIGHT_LONGTEXT N_( "Height of the snapshot image." )
  62. #define CHROMA_TEXT N_( "Chroma" )
  63. #define CHROMA_LONGTEXT N_( "Output chroma for the snapshot image " 
  64.                             "(a 4 character string, like "RV32")." )
  65. #define CACHE_TEXT N_( "Cache size (number of images)" )
  66. #define CACHE_LONGTEXT N_( "Snapshot cache size (number of images to keep)." )
  67. vlc_module_begin ()
  68.     set_description( N_( "Snapshot output" ) )
  69.     set_shortname( N_("Snapshot") )
  70.     set_category( CAT_VIDEO )
  71.     set_subcategory( SUBCAT_VIDEO_VOUT )
  72.     set_capability( "video output", 1 )
  73.     add_integer( "vout-snapshot-width", 320, NULL, WIDTH_TEXT, WIDTH_LONGTEXT, false )
  74.     add_integer( "vout-snapshot-height", 200, NULL, HEIGHT_TEXT, HEIGHT_LONGTEXT, false )
  75.     add_string( "vout-snapshot-chroma", "RV32", NULL, CHROMA_TEXT, CHROMA_LONGTEXT, true )
  76.         add_deprecated_alias( "snapshot-chroma" )
  77.     add_integer( "vout-snapshot-cache-size", 50, NULL, CACHE_TEXT, CACHE_LONGTEXT, true )
  78.         add_deprecated_alias( "snapshot-cache-size" )
  79.     set_callbacks( Create, Destroy )
  80. vlc_module_end ()
  81. /*****************************************************************************
  82.  * vout_sys_t: video output descriptor
  83.  *****************************************************************************/
  84. typedef struct snapshot_t
  85. {
  86.   uint8_t *p_data;   /* Data area */
  87.   int i_width;       /* In pixels */
  88.   int i_height;      /* In pixels */
  89.   int i_datasize;    /* In bytes */
  90.   mtime_t date;      /* Presentation time */
  91. } snapshot_t;
  92. struct vout_sys_t
  93. {
  94.     snapshot_t **p_list;    /* List of available snapshots */
  95.     int i_index;            /* Index of the next available list member */
  96.     int i_size;             /* Size of the cache */
  97.     int i_datasize;         /* Size of an image */
  98.     input_thread_t *p_input;             /* The input thread */
  99. };
  100. /*****************************************************************************
  101.  * Create: allocates video thread
  102.  *****************************************************************************
  103.  * This function allocates and initializes a vout method.
  104.  *****************************************************************************/
  105. static int Create( vlc_object_t *p_this )
  106. {
  107.     vout_thread_t *p_vout = ( vout_thread_t * )p_this;
  108.     /* Allocate instance and initialize some members */
  109.     p_vout->p_sys = malloc( sizeof( vout_sys_t ) );
  110.     if( ! p_vout->p_sys )
  111.         return VLC_ENOMEM;
  112.     var_Create( p_vout, "vout-snapshot-width", VLC_VAR_INTEGER );
  113.     var_Create( p_vout, "vout-snapshot-height", VLC_VAR_INTEGER );
  114.     var_Create( p_vout, "vout-snapshot-datasize", VLC_VAR_INTEGER );
  115.     var_Create( p_vout, "vout-snapshot-cache-size", VLC_VAR_INTEGER );
  116.     var_Create( p_vout, "vout-snapshot-list-pointer", VLC_VAR_ADDRESS );
  117.     p_vout->pf_init = Init;
  118.     p_vout->pf_end = End;
  119.     p_vout->pf_manage = NULL;
  120.     p_vout->pf_render = NULL;
  121.     p_vout->pf_display = Display;
  122.     return VLC_SUCCESS;
  123. }
  124. /*****************************************************************************
  125.  * Init: initialize video thread
  126.  *****************************************************************************/
  127. static int Init( vout_thread_t *p_vout )
  128. {
  129.     int i_index;
  130.     picture_t *p_pic;
  131.     vlc_value_t val;
  132.     char* psz_chroma;
  133.     int i_chroma;
  134.     int i_width;
  135.     int i_height;
  136.     int i_datasize;
  137.     i_width  = config_GetInt( p_vout, "vout-snapshot-width" );
  138.     i_height = config_GetInt( p_vout, "vout-snapshot-height" );
  139.     psz_chroma = config_GetPsz( p_vout, "vout-snapshot-chroma" );
  140.     if( psz_chroma )
  141.     {
  142.         if( strlen( psz_chroma ) < 4 )
  143.         {
  144.             msg_Err( p_vout, "vout-snapshot-chroma should be 4 characters long" );
  145.             return VLC_EGENERIC;
  146.         }
  147.         i_chroma = VLC_FOURCC( psz_chroma[0], psz_chroma[1],
  148.                                psz_chroma[2], psz_chroma[3] );
  149.         free( psz_chroma );
  150.     }
  151.     else
  152.     {
  153.         msg_Err( p_vout, "Cannot find chroma information." );
  154.         return VLC_EGENERIC;
  155.     }
  156.     I_OUTPUTPICTURES = 0;
  157.     /* Initialize the output structure */
  158.     p_vout->output.i_chroma = i_chroma;
  159.     p_vout->output.pf_setpalette = NULL;
  160.     p_vout->output.i_width = i_width;
  161.     p_vout->output.i_height = i_height;
  162.     p_vout->output.i_aspect = p_vout->output.i_width
  163.                                * VOUT_ASPECT_FACTOR / p_vout->output.i_height;
  164.     /* Define the bitmasks */
  165.     switch( i_chroma )
  166.     {
  167.       case VLC_FOURCC( 'R','V','1','5' ):
  168.         p_vout->output.i_rmask = 0x001f;
  169.         p_vout->output.i_gmask = 0x03e0;
  170.         p_vout->output.i_bmask = 0x7c00;
  171.         break;
  172.       case VLC_FOURCC( 'R','V','1','6' ):
  173.         p_vout->output.i_rmask = 0x001f;
  174.         p_vout->output.i_gmask = 0x07e0;
  175.         p_vout->output.i_bmask = 0xf800;
  176.         break;
  177.       case VLC_FOURCC( 'R','V','2','4' ):
  178.         p_vout->output.i_rmask = 0xff0000;
  179.         p_vout->output.i_gmask = 0x00ff00;
  180.         p_vout->output.i_bmask = 0x0000ff;
  181.         break;
  182.       case VLC_FOURCC( 'R','V','3','2' ):
  183.         p_vout->output.i_rmask = 0xff0000;
  184.         p_vout->output.i_gmask = 0x00ff00;
  185.         p_vout->output.i_bmask = 0x0000ff;
  186.         break;
  187.     }
  188.     /* Try to initialize 1 direct buffer */
  189.     p_pic = NULL;
  190.     /* Find an empty picture slot */
  191.     for( i_index = 0 ; i_index < VOUT_MAX_PICTURES ; i_index++ )
  192.     {
  193.         if( p_vout->p_picture[ i_index ].i_status == FREE_PICTURE )
  194.         {
  195.             p_pic = p_vout->p_picture + i_index;
  196.             break;
  197.         }
  198.     }
  199.     /* Allocate the picture */
  200.     if( p_pic == NULL )
  201.     {
  202.         return VLC_SUCCESS;
  203.     }
  204.     vout_AllocatePicture( VLC_OBJECT(p_vout), p_pic, p_vout->output.i_chroma,
  205.                           p_vout->output.i_width, p_vout->output.i_height,
  206.                           p_vout->output.i_aspect );
  207.     if( p_pic->i_planes == 0 )
  208.     {
  209.         return VLC_EGENERIC;
  210.     }
  211.     p_pic->i_status = DESTROYED_PICTURE;
  212.     p_pic->i_type   = DIRECT_PICTURE;
  213.     PP_OUTPUTPICTURE[ I_OUTPUTPICTURES ] = p_pic;
  214.     I_OUTPUTPICTURES++;
  215.     /* Get datasize and set variables */
  216.     i_datasize = i_width * i_height * p_pic->p->i_pixel_pitch;
  217.     p_vout->p_sys->i_datasize = i_datasize;
  218.     p_vout->p_sys->i_index = 0;
  219.     p_vout->p_sys->i_size = config_GetInt( p_vout, "vout-snapshot-cache-size" );
  220.     if( p_vout->p_sys->i_size < 2 )
  221.     {
  222.         msg_Err( p_vout, "vout-snapshot-cache-size must be at least 1." );
  223.         return VLC_EGENERIC;
  224.     }
  225.     p_vout->p_sys->p_list = malloc( p_vout->p_sys->i_size * sizeof( snapshot_t * ) );
  226.     if( p_vout->p_sys->p_list == NULL )
  227.         return VLC_ENOMEM;
  228.     /* Initialize the structures for the circular buffer */
  229.     for( i_index = 0; i_index < p_vout->p_sys->i_size; i_index++ )
  230.     {
  231.         snapshot_t *p_snapshot = malloc( sizeof( snapshot_t ) );
  232.         if( p_snapshot == NULL )
  233.             return VLC_ENOMEM;
  234.         p_snapshot->i_width = i_width;
  235.         p_snapshot->i_height = i_height;
  236.         p_snapshot->i_datasize = i_datasize;
  237.         p_snapshot->date = 0;
  238.         p_snapshot->p_data = malloc( i_datasize );
  239.         if( p_snapshot->p_data == NULL )
  240.         {
  241.             free( p_snapshot );
  242.             return VLC_ENOMEM;
  243.         }
  244.         p_vout->p_sys->p_list[i_index] = p_snapshot;
  245.     }
  246.     val.i_int = i_width;
  247.     var_Set( p_vout, "vout-snapshot-width", val );
  248.     val.i_int = i_height;
  249.     var_Set( p_vout, "vout-snapshot-height", val );
  250.     val.i_int = i_datasize;
  251.     var_Set( p_vout, "vout-snapshot-datasize", val );
  252.     val.i_int = p_vout->p_sys->i_size;
  253.     var_Set( p_vout, "vout-snapshot-cache-size", val );
  254.     val.p_address = p_vout->p_sys->p_list;
  255.     var_Set( p_vout, "vout-snapshot-list-pointer", val );
  256.     /* Get the p_input pointer (to access video times) */
  257.     p_vout->p_sys->p_input = vlc_object_find( p_vout, VLC_OBJECT_INPUT,
  258.                                               FIND_PARENT );
  259.     if( !p_vout->p_sys->p_input )
  260.         return VLC_ENOOBJ;
  261.     if( var_Create( p_vout->p_sys->p_input, "vout-snapshot-id", VLC_VAR_INTEGER ) )
  262.     {
  263.         msg_Err( p_vout, "Cannot create vout-snapshot-id variable in p_input(%p).",
  264.                  p_vout->p_sys->p_input );
  265.         return VLC_EGENERIC;
  266.     }
  267.     /* Register the snapshot vout module at the input level */
  268.     val.p_address = p_vout;
  269.     if( var_Set( p_vout->p_sys->p_input, "vout-snapshot-id", val ) )
  270.     {
  271.         msg_Err( p_vout, "Cannot register vout-snapshot-id in p_input(%p).",
  272.                  p_vout->p_sys->p_input );
  273.         return VLC_EGENERIC;
  274.     }
  275.     return VLC_SUCCESS;
  276. }
  277. /*****************************************************************************
  278.  * End: terminate video thread output method
  279.  *****************************************************************************/
  280. static void End( vout_thread_t *p_vout )
  281. {
  282.     (void)p_vout;
  283. }
  284. /*****************************************************************************
  285.  * Destroy: destroy video thread
  286.  *****************************************************************************
  287.  * Terminate an output method created by Create
  288.  *****************************************************************************/
  289. static void Destroy( vlc_object_t *p_this )
  290. {
  291.     vout_thread_t *p_vout = ( vout_thread_t * )p_this;
  292.     int i_index;
  293.     var_Destroy( p_vout->p_sys->p_input, "vout-snapshot-id" );
  294.     vlc_object_release( p_vout->p_sys->p_input );
  295.     var_Destroy( p_this, "vout-snapshot-width" );
  296.     var_Destroy( p_this, "vout-snapshot-height" );
  297.     var_Destroy( p_this, "vout-snapshot-datasize" );
  298.     for( i_index = 0 ; i_index < p_vout->p_sys->i_size ; i_index++ )
  299.     {
  300.         free( p_vout->p_sys->p_list[ i_index ]->p_data );
  301.     }
  302.     free( p_vout->p_sys->p_list );
  303.     /* Destroy structure */
  304.     free( p_vout->p_sys );
  305. }
  306. /* Return the position in ms from the start of the movie */
  307. static mtime_t snapshot_GetMovietime( vout_thread_t *p_vout )
  308. {
  309.     input_thread_t *p_input = p_vout->p_sys->p_input;
  310.     if( !p_input )
  311.         return 0;
  312.     return var_GetTime( p_input, "time" ) / 1000;
  313. }
  314. /*****************************************************************************
  315.  * Display: displays previously rendered output
  316.  *****************************************************************************
  317.  * This function copies the rendered picture into our circular buffer.
  318.  *****************************************************************************/
  319. static void Display( vout_thread_t *p_vout, picture_t *p_pic )
  320. {
  321.     int i_index;
  322.     mtime_t i_date;
  323.     i_index = p_vout->p_sys->i_index;
  324.     vlc_memcpy( p_vout->p_sys->p_list[i_index]->p_data, p_pic->p->p_pixels,
  325.                 p_vout->p_sys->i_datasize );
  326.     i_date = snapshot_GetMovietime( p_vout );
  327.     p_vout->p_sys->p_list[i_index]->date = i_date;
  328.     i_index++;
  329.     if( i_index >= p_vout->p_sys->i_size )
  330.     {
  331.         i_index = 0;
  332.     }
  333.     p_vout->p_sys->i_index = i_index;
  334. }