[vlc-devel] [PATCH] packetizer: use a custom flush callback
Steve Lhomme
robux4 at videolabs.io
Wed Nov 18 10:05:58 CET 2015
--
needed as many don't handle BLOCK_FLAG_CORRUPTED or BLOCK_FLAG_DISCONTINUITY
---
modules/packetizer/avparser.c | 9 +++++++++
modules/packetizer/copy.c | 13 +++++++++++++
modules/packetizer/h264.c | 9 +++++++++
modules/packetizer/hevc.c | 9 +++++++++
modules/packetizer/mpeg4video.c | 9 +++++++++
modules/packetizer/mpegvideo.c | 9 +++++++++
modules/packetizer/packetizer_helper.h | 9 +++++++++
modules/packetizer/vc1.c | 9 +++++++++
8 files changed, 76 insertions(+)
diff --git a/modules/packetizer/avparser.c b/modules/packetizer/avparser.c
index 450b450..65dda5f 100644
--- a/modules/packetizer/avparser.c
+++ b/modules/packetizer/avparser.c
@@ -58,6 +58,7 @@ struct decoder_sys_t
};
static block_t * Packetize( decoder_t *, block_t ** );
+static int Flush ( decoder_t * );
/*****************************************************************************
* OpenPacketizer: probe the packetizer and return score
@@ -110,6 +111,7 @@ int OpenPacketizer( vlc_object_t *p_this )
return VLC_ENOMEM;
}
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
p_sys->p_parser_ctx = p_ctx;
p_sys->p_codec_ctx = p_codec_ctx;
p_sys->i_offset = 0;
@@ -130,6 +132,13 @@ void ClosePacketizer( vlc_object_t *p_this )
free( p_dec->p_sys );
}
+static int Flush ( decoder_t *p_dec )
+{
+ /* reopen as there's no clean way to flush avparser */
+ ClosePacketizer( VLC_OBJECT( p_dec ) );
+ return OpenPacketizer( VLC_OBJECT( p_dec ) );
+}
+
/*****************************************************************************
* Packetize: packetize a frame
*****************************************************************************/
diff --git a/modules/packetizer/copy.c b/modules/packetizer/copy.c
index bc3a623..57bc903 100644
--- a/modules/packetizer/copy.c
+++ b/modules/packetizer/copy.c
@@ -61,6 +61,7 @@ struct decoder_sys_t
static block_t *Packetize ( decoder_t *, block_t ** );
static block_t *PacketizeSub( decoder_t *, block_t ** );
+static int Flush( decoder_t * );
static void ParseWMV3( decoder_t *, block_t * );
@@ -87,6 +88,7 @@ static int Open( vlc_object_t *p_this )
p_dec->pf_packetize = PacketizeSub;
else
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
/* Create the output format */
es_format_Copy( &p_dec->fmt_out, &p_dec->fmt_in );
@@ -134,6 +136,17 @@ static void Close( vlc_object_t *p_this )
free( p_dec->p_sys );
}
+static int Flush( decoder_t *p_dec )
+{
+ block_t *p_ret = p_dec->p_sys->p_block;
+ if (p_ret)
+ {
+ block_Release( p_ret );
+ p_dec->p_sys->p_block = NULL;
+ }
+ return VLC_SUCCESS;
+}
+
/*****************************************************************************
* Packetize: packetize an unit (here copy a complete block )
*****************************************************************************/
diff --git a/modules/packetizer/h264.c b/modules/packetizer/h264.c
index 7047465..bc2a797 100644
--- a/modules/packetizer/h264.c
+++ b/modules/packetizer/h264.c
@@ -149,6 +149,7 @@ struct decoder_sys_t
static block_t *Packetize( decoder_t *, block_t ** );
static block_t *PacketizeAVC1( decoder_t *, block_t ** );
+static int Flush( decoder_t * );
static block_t *GetCc( decoder_t *p_dec, bool pb_present[4] );
static void PacketizeReset( void *p_private, bool b_broken );
@@ -354,6 +355,7 @@ static int Open( vlc_object_t *p_this )
/* CC are the same for H264/AVC in T35 sections (ETSI TS 101 154) */
p_dec->pf_get_cc = GetCc;
+ p_dec->pf_flush = Flush;
/* */
p_sys->i_cc_pts = VLC_TS_INVALID;
@@ -394,6 +396,13 @@ static void Close( vlc_object_t *p_this )
free( p_sys );
}
+static int Flush( decoder_t *p_dec )
+{
+ decoder_sys_t *p_sys = p_dec->p_sys;
+
+ return packetizer_Flush( &p_sys->packetizer );
+}
+
/****************************************************************************
* Packetize: the whole thing
* Search for the startcodes 3 or more bytes
diff --git a/modules/packetizer/hevc.c b/modules/packetizer/hevc.c
index a905f8f..93977e2 100644
--- a/modules/packetizer/hevc.c
+++ b/modules/packetizer/hevc.c
@@ -57,6 +57,7 @@ vlc_module_end ()
* Local prototypes
****************************************************************************/
static block_t *Packetize(decoder_t *, block_t **);
+static int Flush( decoder_t * );
static void PacketizeReset(void *p_private, bool b_broken);
static block_t *PacketizeParse(void *p_private, bool *pb_ts_used, block_t *);
static int PacketizeValidate(void *p_private, block_t *);
@@ -132,6 +133,7 @@ static int Open(vlc_object_t *p_this)
/* Set callback */
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
return VLC_SUCCESS;
@@ -159,6 +161,13 @@ static block_t *Packetize(decoder_t *p_dec, block_t **pp_block)
return packetizer_Packetize(&p_sys->packetizer, pp_block);
}
+static int Flush( decoder_t *p_dec )
+{
+ decoder_sys_t *p_sys = p_dec->p_sys;
+
+ return packetizer_Flush( &p_sys->packetizer );
+}
+
/****************************************************************************
* Packetizer Helpers
****************************************************************************/
diff --git a/modules/packetizer/mpeg4video.c b/modules/packetizer/mpeg4video.c
index bdf503c..888625e 100644
--- a/modules/packetizer/mpeg4video.c
+++ b/modules/packetizer/mpeg4video.c
@@ -91,6 +91,7 @@ struct decoder_sys_t
};
static block_t *Packetize( decoder_t *, block_t ** );
+static int Flush( decoder_t * );
static void PacketizeReset( void *p_private, bool b_broken );
static block_t *PacketizeParse( void *p_private, bool *pb_ts_used, block_t * );
@@ -175,6 +176,7 @@ static int Open( vlc_object_t *p_this )
/* Set callback */
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
return VLC_SUCCESS;
}
@@ -203,6 +205,13 @@ static block_t *Packetize( decoder_t *p_dec, block_t **pp_block )
return packetizer_Packetize( &p_sys->packetizer, pp_block );
}
+static int Flush( decoder_t *p_dec )
+{
+ decoder_sys_t *p_sys = p_dec->p_sys;
+
+ return packetizer_Flush( &p_sys->packetizer );
+}
+
/*****************************************************************************
* Helpers:
*****************************************************************************/
diff --git a/modules/packetizer/mpegvideo.c b/modules/packetizer/mpegvideo.c
index f9b12f0..564d154 100644
--- a/modules/packetizer/mpegvideo.c
+++ b/modules/packetizer/mpegvideo.c
@@ -136,6 +136,7 @@ struct decoder_sys_t
static block_t *Packetize( decoder_t *, block_t ** );
static block_t *GetCc( decoder_t *p_dec, bool pb_present[4] );
+static int Flush( decoder_t * );
static void PacketizeReset( void *p_private, bool b_broken );
static block_t *PacketizeParse( void *p_private, bool *pb_ts_used, block_t * );
@@ -209,6 +210,7 @@ static int Open( vlc_object_t *p_this )
cc_Init( &p_sys->cc );
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
p_dec->pf_get_cc = GetCc;
return VLC_SUCCESS;
@@ -251,6 +253,13 @@ static block_t *Packetize( decoder_t *p_dec, block_t **pp_block )
return packetizer_Packetize( &p_sys->packetizer, pp_block );
}
+static int Flush( decoder_t *p_dec )
+{
+ decoder_sys_t *p_sys = p_dec->p_sys;
+
+ return packetizer_Flush( &p_sys->packetizer );
+}
+
/*****************************************************************************
* GetCc:
*****************************************************************************/
diff --git a/modules/packetizer/packetizer_helper.h b/modules/packetizer/packetizer_helper.h
index a6cc663..bd0e2d6 100644
--- a/modules/packetizer/packetizer_helper.h
+++ b/modules/packetizer/packetizer_helper.h
@@ -93,6 +93,15 @@ static inline void packetizer_Clean( packetizer_t *p_pack )
block_BytestreamRelease( &p_pack->bytestream );
}
+static inline int packetizer_Flush( packetizer_t *p_pack )
+{
+ p_pack->i_state = STATE_NOSYNC;
+ block_BytestreamEmpty( &p_pack->bytestream );
+ p_pack->i_offset = 0;
+ p_pack->pf_reset( p_pack->p_private, false );
+ return VLC_SUCCESS;
+}
+
static inline block_t *packetizer_Packetize( packetizer_t *p_pack, block_t **pp_block )
{
if( !pp_block || !*pp_block )
diff --git a/modules/packetizer/vc1.c b/modules/packetizer/vc1.c
index 51c90f3..d5b3fdb 100644
--- a/modules/packetizer/vc1.c
+++ b/modules/packetizer/vc1.c
@@ -120,6 +120,7 @@ typedef enum
} idu_type_t;
static block_t *Packetize( decoder_t *p_dec, block_t **pp_block );
+static int Flush( decoder_t * );
static void PacketizeReset( void *p_private, bool b_broken );
static block_t *PacketizeParse( void *p_private, bool *pb_ts_used, block_t * );
@@ -144,6 +145,7 @@ static int Open( vlc_object_t *p_this )
return VLC_EGENERIC;
p_dec->pf_packetize = Packetize;
+ p_dec->pf_flush = Flush;
p_dec->pf_get_cc = GetCc;
/* Create the output format */
@@ -257,6 +259,13 @@ static block_t *Packetize( decoder_t *p_dec, block_t **pp_block )
return p_au;
}
+static int Flush( decoder_t *p_dec )
+{
+ decoder_sys_t *p_sys = p_dec->p_sys;
+
+ return packetizer_Flush( &p_sys->packetizer );
+}
+
static void PacketizeReset( void *p_private, bool b_broken )
{
decoder_t *p_dec = p_private;
--
2.6.2
More information about the vlc-devel
mailing list