modules/packetizer/copy.c

   1 /*****************************************************************************
   2  * copy.c
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002, 2006 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Eric Petit <titer@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28
  29 #ifdef HAVE_CONFIG_H
  30 # include "config.h"
  31 #endif
  32
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35 #include <vlc_codec.h>
  36 #include <vlc_block.h>
  37 #include <vlc_bits.h>
  38
  39 /*****************************************************************************
  40  * Module descriptor
  41  *****************************************************************************/
  42 static int  Open ( vlc_object_t * );
  43 static void Close( vlc_object_t * );
  44
  45 vlc_module_begin ()
  46     set_category( CAT_SOUT )
  47     set_subcategory( SUBCAT_SOUT_PACKETIZER )
  48     set_description( N_("Copy packetizer") )
  49     set_capability( "packetizer", 1 )
  50     set_callbacks( Open, Close )
  51 vlc_module_end ()
  52
  53 /*****************************************************************************
  54  * Local prototypes
  55  *****************************************************************************/
  56 struct decoder_sys_t
  57 {
  58     block_t *p_block;
  59     void (*pf_parse)( decoder_t *, block_t * );
  60 };
  61
  62 static block_t *Packetize   ( decoder_t *, block_t ** );
  63 static block_t *PacketizeSub( decoder_t *, block_t ** );
  64
  65 static void ParseWMV3( decoder_t *, block_t * );
  66
  67 /*****************************************************************************
  68  * Open: probe the packetizer and return score
  69  *****************************************************************************
  70  * Tries to launch a decoder and return score so that the interface is able
  71  * to choose.
  72  *****************************************************************************/
  73 static int Open( vlc_object_t *p_this )
  74 {
  75     decoder_t     *p_dec = (decoder_t*)p_this;
  76     decoder_sys_t *p_sys;
  77
  78     if( p_dec->fmt_in.i_cat != AUDIO_ES &&
  79         p_dec->fmt_in.i_cat != VIDEO_ES &&
  80         p_dec->fmt_in.i_cat != SPU_ES )
  81     {
  82         msg_Err( p_dec, "invalid ES type" );
  83         return VLC_EGENERIC;
  84     }
  85
  86     if( p_dec->fmt_in.i_cat == SPU_ES )
  87         p_dec->pf_packetize = PacketizeSub;
  88     else
  89         p_dec->pf_packetize = Packetize;
  90
  91     /* Create the output format */
  92     es_format_Copy( &p_dec->fmt_out, &p_dec->fmt_in );
  93
  94     /* Fix the value of the fourcc for audio */
  95     if( p_dec->fmt_in.i_cat == AUDIO_ES )
  96     {
  97         p_dec->fmt_out.i_codec = vlc_fourcc_GetCodecAudio( p_dec->fmt_in.i_codec,
  98                                                            p_dec->fmt_in.audio.i_bitspersample );
  99         if( !p_dec->fmt_out.i_codec )
 100         {
 101             msg_Err( p_dec, "unknown raw audio sample size" );
 102             return VLC_EGENERIC;
 103         }
 104     }
 105
 106     p_dec->p_sys = p_sys = malloc( sizeof(*p_sys) );
 107     p_sys->p_block    = NULL;
 108     switch( p_dec->fmt_in.i_codec )
 109     {
 110     case VLC_CODEC_WMV3:
 111         p_sys->pf_parse = ParseWMV3;
 112         break;
 113     default:
 114         p_sys->pf_parse = NULL;
 115         break;
 116     }
 117
 118     return VLC_SUCCESS;
 119 }
 120
 121 /*****************************************************************************
 122  * Close:
 123  *****************************************************************************/
 124 static void Close( vlc_object_t *p_this )
 125 {
 126     decoder_t     *p_dec = (decoder_t*)p_this;
 127
 128     if( p_dec->p_sys->p_block )
 129     {
 130         block_ChainRelease( p_dec->p_sys->p_block );
 131     }
 132
 133     es_format_Clean( &p_dec->fmt_out );
 134     free( p_dec->p_sys );
 135 }
 136
 137 /*****************************************************************************
 138  * Packetize: packetize an unit (here copy a complete block )
 139  *****************************************************************************/
 140 static block_t *Packetize ( decoder_t *p_dec, block_t **pp_block )
 141 {
 142     block_t *p_block;
 143     block_t *p_ret = p_dec->p_sys->p_block;
 144
 145     if( pp_block == NULL || *pp_block == NULL )
 146         return NULL;
 147     if( (*pp_block)->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
 148     {
 149         block_Release( *pp_block );
 150         return NULL;
 151     }
 152
 153     p_block = *pp_block;
 154     *pp_block = NULL;
 155
 156     if( p_block->i_dts <= VLC_TS_INVALID )
 157     {
 158         p_block->i_dts = p_block->i_pts;
 159     }
 160
 161     if( p_block->i_dts <= VLC_TS_INVALID )
 162     {
 163         msg_Dbg( p_dec, "need valid dts" );
 164         block_Release( p_block );
 165         return NULL;
 166     }
 167
 168     if( p_ret != NULL && p_block->i_pts > p_ret->i_pts )
 169     {
 170         p_ret->i_length = p_block->i_pts - p_ret->i_pts;
 171     }
 172     p_dec->p_sys->p_block = p_block;
 173
 174     if( p_ret && p_dec->p_sys->pf_parse )
 175         p_dec->p_sys->pf_parse( p_dec, p_ret );
 176     return p_ret;
 177 }
 178
 179 /*****************************************************************************
 180  * PacketizeSub: packetize an unit (here copy a complete block )
 181  *****************************************************************************/
 182 static block_t *PacketizeSub( decoder_t *p_dec, block_t **pp_block )
 183 {
 184     block_t *p_block;
 185
 186     if( pp_block == NULL || *pp_block == NULL )
 187         return NULL;
 188     if( (*pp_block)->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
 189     {
 190         block_Release( *pp_block );
 191         return NULL;
 192     }
 193
 194     p_block = *pp_block;
 195     *pp_block = NULL;
 196
 197     if( p_block->i_dts <= VLC_TS_INVALID )
 198     {
 199         p_block->i_dts = p_block->i_pts;
 200     }
 201
 202     if( p_block->i_dts <= VLC_TS_INVALID )
 203     {
 204         msg_Dbg( p_dec, "need valid dts" );
 205         block_Release( p_block );
 206         return NULL;
 207     }
 208
 209     return p_block;
 210 }
 211
 212 /* Parse WMV3 packet and extract frame type information */
 213 static void ParseWMV3( decoder_t *p_dec, block_t *p_block )
 214 {
 215     bs_t s;
 216
 217     /* Parse Sequence header */
 218     bs_init( &s, p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra );
 219     if( bs_read( &s, 2 ) == 3 )
 220         return;
 221     bs_skip( &s, 22 );
 222     const bool b_range_reduction = bs_read( &s, 1 );
 223     const bool b_has_frames = bs_read( &s, 3 ) > 0;
 224     bs_skip( &s, 2 );
 225     const bool b_frame_interpolation = bs_read( &s, 1 );
 226     if( bs_eof( &s ) )
 227         return;
 228
 229     /* Parse frame type */
 230     bs_init( &s, p_block->p_buffer, p_block->i_buffer );
 231     bs_skip( &s, b_frame_interpolation +
 232                  2 +
 233                  b_range_reduction );
 234
 235     p_block->i_flags &= ~BLOCK_FLAG_TYPE_MASK;
 236     if( bs_read( &s, 1 ) )
 237         p_block->i_flags |= BLOCK_FLAG_TYPE_P;
 238     else if( !b_has_frames || bs_read( &s, 1 ) )
 239         p_block->i_flags |= BLOCK_FLAG_TYPE_I;
 240     else
 241         p_block->i_flags |= BLOCK_FLAG_TYPE_B;
 242 }
 243