modules/packetizer/copy.c

   1 /*****************************************************************************
   2  * copy.c
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002, 2006 VLC authors and VideoLAN
   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 it
  11  * under the terms of the GNU Lesser General Public License as published by
  12  * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public License
  21  * along with this program; if not, write to the Free Software Foundation,
  22  * 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 typedef struct
  57 {
  58     block_t *p_block;
  59     void (*pf_parse)( decoder_t *, block_t * );
  60 } decoder_sys_t;
  61
  62 static block_t *Packetize   ( decoder_t *, block_t ** );
  63 static block_t *PacketizeSub( decoder_t *, block_t ** );
  64 static void Flush( decoder_t * );
  65
  66 static void ParseWMV3( decoder_t *, block_t * );
  67
  68 /*****************************************************************************
  69  * Open: probe the packetizer and return score
  70  *****************************************************************************
  71  * Tries to launch a decoder and return score so that the interface is able
  72  * to choose.
  73  *****************************************************************************/
  74 static int Open( vlc_object_t *p_this )
  75 {
  76     decoder_t     *p_dec = (decoder_t*)p_this;
  77     decoder_sys_t *p_sys;
  78
  79     if( p_dec->fmt_in.i_cat != AUDIO_ES &&
  80         p_dec->fmt_in.i_cat != VIDEO_ES &&
  81         p_dec->fmt_in.i_cat != SPU_ES )
  82     {
  83         msg_Err( p_dec, "invalid ES type" );
  84         return VLC_EGENERIC;
  85     }
  86
  87     p_dec->p_sys = p_sys = malloc( sizeof(*p_sys) );
  88     if (unlikely(p_sys == NULL))
  89         return VLC_ENOMEM;
  90
  91     p_sys->p_block    = NULL;
  92     switch( p_dec->fmt_in.i_codec )
  93     {
  94     case VLC_CODEC_WMV3:
  95         p_sys->pf_parse = ParseWMV3;
  96         break;
  97     default:
  98         p_sys->pf_parse = NULL;
  99         break;
 100     }
 101
 102     vlc_fourcc_t fcc = p_dec->fmt_in.i_codec;
 103     /* Fix the value of the fourcc for audio */
 104     if( p_dec->fmt_in.i_cat == AUDIO_ES )
 105     {
 106         fcc = vlc_fourcc_GetCodecAudio( p_dec->fmt_in.i_codec,
 107                                                      p_dec->fmt_in.audio.i_bitspersample );
 108         if( !fcc )
 109         {
 110             msg_Err( p_dec, "unknown raw audio sample size" );
 111             free( p_sys );
 112             return VLC_EGENERIC;
 113         }
 114     }
 115
 116     /* Create the output format */
 117     es_format_Copy( &p_dec->fmt_out, &p_dec->fmt_in );
 118     p_dec->fmt_out.i_codec = fcc;
 119     if( p_dec->fmt_in.i_cat == SPU_ES )
 120         p_dec->pf_packetize = PacketizeSub;
 121     else
 122         p_dec->pf_packetize = Packetize;
 123     p_dec->pf_flush = Flush;
 124     p_dec->pf_get_cc = NULL;
 125
 126     return VLC_SUCCESS;
 127 }
 128
 129 /*****************************************************************************
 130  * Close:
 131  *****************************************************************************/
 132 static void Close( vlc_object_t *p_this )
 133 {
 134     decoder_t     *p_dec = (decoder_t*)p_this;
 135     decoder_sys_t *p_sys = p_dec->p_sys;
 136
 137     if( p_sys->p_block )
 138     {
 139         block_ChainRelease( p_sys->p_block );
 140     }
 141
 142     free( p_dec->p_sys );
 143 }
 144
 145 static void Flush( decoder_t *p_dec )
 146 {
 147     decoder_sys_t *p_sys = p_dec->p_sys;
 148     block_t *p_ret = p_sys->p_block;
 149     if ( p_ret )
 150     {
 151         block_Release( p_ret );
 152         p_sys->p_block = NULL;
 153     }
 154 }
 155
 156 /*****************************************************************************
 157  * Packetize: packetize an unit (here copy a complete block )
 158  *****************************************************************************/
 159 static block_t *Packetize ( decoder_t *p_dec, block_t **pp_block )
 160 {
 161     block_t *p_block;
 162     decoder_sys_t *p_sys = p_dec->p_sys;
 163     block_t *p_ret = p_sys->p_block;
 164
 165     if( pp_block == NULL || *pp_block == NULL )
 166         return NULL;
 167     if( (*pp_block)->i_flags&(BLOCK_FLAG_CORRUPTED) )
 168     {
 169         block_Release( *pp_block );
 170         return NULL;
 171     }
 172
 173     p_block = *pp_block;
 174     *pp_block = NULL;
 175
 176     if( p_block->i_dts == VLC_TICK_INVALID )
 177     {
 178         p_block->i_dts = p_block->i_pts;
 179     }
 180
 181     if( p_block->i_dts == VLC_TICK_INVALID )
 182     {
 183         msg_Dbg( p_dec, "need valid dts" );
 184         block_Release( p_block );
 185         return NULL;
 186     }
 187
 188     if( p_ret != NULL && p_block->i_pts > p_ret->i_pts )
 189     {
 190         if (p_dec->fmt_in.i_codec != VLC_CODEC_OPUS)
 191             p_ret->i_length = p_block->i_pts - p_ret->i_pts;
 192     }
 193     p_sys->p_block = p_block;
 194
 195     if( p_ret && p_sys->pf_parse )
 196         p_sys->pf_parse( p_dec, p_ret );
 197     return p_ret;
 198 }
 199
 200 /*****************************************************************************
 201  * PacketizeSub: packetize an unit (here copy a complete block )
 202  *****************************************************************************/
 203 static block_t *PacketizeSub( decoder_t *p_dec, block_t **pp_block )
 204 {
 205     block_t *p_block;
 206
 207     if( pp_block == NULL || *pp_block == NULL )
 208         return NULL;
 209     if( (*pp_block)->i_flags&(BLOCK_FLAG_CORRUPTED) )
 210     {
 211         block_Release( *pp_block );
 212         return NULL;
 213     }
 214
 215     p_block = *pp_block;
 216     *pp_block = NULL;
 217
 218     if( p_block->i_dts == VLC_TICK_INVALID )
 219     {
 220         p_block->i_dts = p_block->i_pts;
 221     }
 222
 223     if( p_block->i_dts == VLC_TICK_INVALID )
 224     {
 225         msg_Dbg( p_dec, "need valid dts" );
 226         block_Release( p_block );
 227         return NULL;
 228     }
 229
 230     return p_block;
 231 }
 232
 233 /* Parse WMV3 packet and extract frame type information */
 234 static void ParseWMV3( decoder_t *p_dec, block_t *p_block )
 235 {
 236     bs_t s;
 237
 238     /* Parse Sequence header */
 239     bs_init( &s, p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra );
 240     if( bs_read( &s, 2 ) == 3 )
 241         return;
 242     bs_skip( &s, 22 );
 243     const bool b_range_reduction = bs_read( &s, 1 );
 244     const bool b_has_frames = bs_read( &s, 3 ) > 0;
 245     bs_skip( &s, 2 );
 246     const bool b_frame_interpolation = bs_read( &s, 1 );
 247     if( bs_eof( &s ) )
 248         return;
 249
 250     /* Parse frame type */
 251     bs_init( &s, p_block->p_buffer, p_block->i_buffer );
 252     bs_skip( &s, b_frame_interpolation +
 253                  2 +
 254                  b_range_reduction );
 255
 256     p_block->i_flags &= ~BLOCK_FLAG_TYPE_MASK;
 257     if( bs_read( &s, 1 ) )
 258         p_block->i_flags |= BLOCK_FLAG_TYPE_P;
 259     else if( !b_has_frames || bs_read( &s, 1 ) )
 260         p_block->i_flags |= BLOCK_FLAG_TYPE_I;
 261     else
 262         p_block->i_flags |= BLOCK_FLAG_TYPE_B;
 263 }
 264