gather.c
上传用户:riyaled888
上传日期:2009-03-27
资源大小:7338k
文件大小:6k
源码类别:

多媒体

开发平台:

MultiPlatform

  1. /*****************************************************************************
  2.  * gather.c: gathering stream output module
  3.  *****************************************************************************
  4.  * Copyright (C) 2003-2004 VideoLAN
  5.  * $Id: gather.c 8162 2004-07-10 17:20:59Z fenrir $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir@via.ecp.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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #include <stdlib.h>
  27. #include <vlc/vlc.h>
  28. #include <vlc/input.h>
  29. #include <vlc/sout.h>
  30. /*****************************************************************************
  31.  * Module descriptor
  32.  *****************************************************************************/
  33. static int      Open    ( vlc_object_t * );
  34. static void     Close   ( vlc_object_t * );
  35. vlc_module_begin();
  36.     set_description( _("Gathering stream output") );
  37.     set_capability( "sout stream", 50 );
  38.     add_shortcut( "gather" );
  39.     set_callbacks( Open, Close );
  40. vlc_module_end();
  41. /*****************************************************************************
  42.  * Exported prototypes
  43.  *****************************************************************************/
  44. static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
  45. static int               Del ( sout_stream_t *, sout_stream_id_t * );
  46. static int               Send( sout_stream_t *, sout_stream_id_t *,
  47.                                block_t* );
  48. struct sout_stream_id_t
  49. {
  50.     vlc_bool_t    b_used;
  51.     es_format_t fmt;
  52.     void          *id;
  53. };
  54. struct sout_stream_sys_t
  55. {
  56.     sout_stream_t   *p_out;
  57.     int              i_id;
  58.     sout_stream_id_t **id;
  59. };
  60. /*****************************************************************************
  61.  * Open:
  62.  *****************************************************************************/
  63. static int Open( vlc_object_t *p_this )
  64. {
  65.     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
  66.     sout_stream_sys_t *p_sys;
  67.     p_stream->p_sys = p_sys = malloc( sizeof( sout_stream_sys_t ) );
  68.     p_sys->p_out    = sout_StreamNew( p_stream->p_sout, p_stream->psz_next );
  69.     if( p_sys->p_out == NULL )
  70.     {
  71.         free( p_sys );
  72.         return VLC_EGENERIC;
  73.     }
  74.     p_sys->i_id         = 0;
  75.     p_sys->id           = NULL;
  76.     p_stream->pf_add    = Add;
  77.     p_stream->pf_del    = Del;
  78.     p_stream->pf_send   = Send;
  79.     return VLC_SUCCESS;
  80. }
  81. /*****************************************************************************
  82.  * Close:
  83.  *****************************************************************************/
  84. static void Close( vlc_object_t * p_this )
  85. {
  86.     sout_stream_t     *p_stream = (sout_stream_t*)p_this;
  87.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  88.     int i;
  89.     for( i = 0; i < p_sys->i_id; i++ )
  90.     {
  91.         p_sys->p_out->pf_del( p_sys->p_out, p_sys->id[i]->id );
  92.         free( p_sys->id[i] );
  93.     }
  94.     free( p_sys->id );
  95.     sout_StreamDelete( p_sys->p_out );
  96.     free( p_sys );
  97. }
  98. /*****************************************************************************
  99.  * Add:
  100.  *****************************************************************************/
  101. static sout_stream_id_t * Add( sout_stream_t *p_stream, es_format_t *p_fmt )
  102. {
  103.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  104.     sout_stream_id_t  *id;
  105.     int i;
  106.     /* search a output compatible */
  107.     for( i = 0; i < p_sys->i_id; i++ )
  108.     {
  109.         id = p_sys->id[i];
  110.         if( !id->b_used &&
  111.             id->fmt.i_cat == p_fmt->i_cat &&
  112.             id->fmt.i_codec == p_fmt->i_codec &&
  113.             ( ( id->fmt.i_cat == AUDIO_ES &&
  114.                 id->fmt.audio.i_rate == p_fmt->audio.i_rate &&
  115.                 id->fmt.audio.i_channels == p_fmt->audio.i_channels &&
  116.                 id->fmt.audio.i_blockalign == p_fmt->audio.i_blockalign ) ||
  117.               ( id->fmt.i_cat == VIDEO_ES &&
  118.                 id->fmt.video.i_width == p_fmt->video.i_width &&
  119.                 id->fmt.video.i_height == p_fmt->video.i_height ) ) )
  120.         {
  121.             msg_Dbg( p_stream, "reusing already opened output" );
  122.             id->b_used = VLC_TRUE;
  123.             return id;
  124.         }
  125.     }
  126.     /* destroy all output of the same categorie */
  127.     for( i = 0; i < p_sys->i_id; i++ )
  128.     {
  129.         id = p_sys->id[i];
  130.         if( !id->b_used && id->fmt.i_cat == p_fmt->i_cat )
  131.         {
  132.             TAB_REMOVE( p_sys->i_id, p_sys->id, id );
  133.             p_sys->p_out->pf_del( p_sys->p_out, id );
  134.             free( id );
  135.             i = 0;
  136.             continue;
  137.         }
  138.     }
  139.     id = malloc( sizeof( sout_stream_id_t ) );
  140.     msg_Dbg( p_stream, "creating new output" );
  141.     memcpy( &id->fmt, p_fmt, sizeof( es_format_t ) );
  142.     id->fmt.i_extra = 0;
  143.     id->fmt.p_extra = NULL;
  144.     id->b_used           = VLC_TRUE;
  145.     id->id               = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
  146.     if( id->id == NULL )
  147.     {
  148.         free( id );
  149.         return NULL;
  150.     }
  151.     TAB_APPEND( p_sys->i_id, p_sys->id, id );
  152.     return id;
  153. }
  154. /*****************************************************************************
  155.  * Del:
  156.  *****************************************************************************/
  157. static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
  158. {
  159.     id->b_used = VLC_FALSE;
  160.     return VLC_SUCCESS;
  161. }
  162. /*****************************************************************************
  163.  * Send:
  164.  *****************************************************************************/
  165. static int Send( sout_stream_t *p_stream,
  166.                  sout_stream_id_t *id, block_t *p_buffer )
  167. {
  168.     sout_stream_sys_t *p_sys = p_stream->p_sys;
  169.     return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer );
  170. }