[vlc-devel] [PATCH] stats module
Ilkka Ollakka
ileoo at videolan.org
Sat Feb 8 17:53:01 CET 2014
Prints dts, segment number, length, difference to previous dts and md5
of block.
with prefix-option you can give prefix to print in stats-line, so you can
add stats module multiple times with different prefixes. Output to file
is given with output parameter.
---
modules/stream_out/Modules.am | 2 +
modules/stream_out/stats.c | 228 ++++++++++++++++++++++++++++++++++++++++++
2 files changed, 230 insertions(+)
create mode 100644 modules/stream_out/stats.c
diff --git a/modules/stream_out/Modules.am b/modules/stream_out/Modules.am
index 2340c64..b581152 100644
--- a/modules/stream_out/Modules.am
+++ b/modules/stream_out/Modules.am
@@ -1,5 +1,6 @@
SOURCES_stream_out_dummy = dummy.c
SOURCES_stream_out_delay = delay.c
+SOURCES_stream_out_stats = stats.c
SOURCES_stream_out_description = description.c
SOURCES_stream_out_standard = standard.c
SOURCES_stream_out_duplicate = duplicate.c
@@ -24,6 +25,7 @@ libstream_out_transcode_plugin_la_CFLAGS = $(AM_CFLAGS)
stream_out_LTLIBRARIES += \
libstream_out_dummy_plugin.la \
libstream_out_delay_plugin.la \
+ libstream_out_stats_plugin.la \
libstream_out_description_plugin.la \
libstream_out_standard_plugin.la \
libstream_out_duplicate_plugin.la \
diff --git a/modules/stream_out/stats.c b/modules/stream_out/stats.c
new file mode 100644
index 0000000..5414f2d
--- /dev/null
+++ b/modules/stream_out/stats.c
@@ -0,0 +1,228 @@
+/*****************************************************************************
+ * delay.c: delay a stream
+ *****************************************************************************
+ * Copyright © 2014 VLC authors and VideoLAN
+ *
+ * Authors: Ilkka Ollakka <ileoo at videolan dot org>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ *****************************************************************************/
+
+/*****************************************************************************
+ * Preamble
+ *****************************************************************************/
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <vlc_common.h>
+#include <vlc_plugin.h>
+#include <vlc_sout.h>
+#include <vlc_block.h>
+#include <vlc_md5.h>
+#include <vlc_fs.h>
+
+/*****************************************************************************
+ * Module descriptor
+ *****************************************************************************/
+#define OUTPUT_TEXT N_("Output file")
+#define OUTPUT_LONGTEXT N_( \
+ "Writes stats to file instead of stdout" )
+#define PREFIX_TEXT N_("Prefix to show on output line")
+
+static int Open ( vlc_object_t * );
+static void Close ( vlc_object_t * );
+
+#define SOUT_CFG_PREFIX "sout-stats-"
+
+vlc_module_begin()
+ set_shortname( N_("Stats"))
+ set_description( N_("Writes statistic info about stream"))
+ set_capability( "sout stream", 0 )
+ add_shortcut( "stats" )
+ set_category( CAT_SOUT )
+ set_subcategory( SUBCAT_SOUT_STREAM )
+ set_callbacks( Open, Close )
+ add_string( SOUT_CFG_PREFIX "output", "", OUTPUT_TEXT,OUTPUT_LONGTEXT, false );
+ add_string( SOUT_CFG_PREFIX "prefix", "", PREFIX_TEXT,PREFIX_TEXT, false );
+vlc_module_end()
+
+
+/*****************************************************************************
+ * Local prototypes
+ *****************************************************************************/
+static const char *ppsz_sout_options[] = {
+ "output", "prefix", NULL
+};
+
+static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
+static int Del ( sout_stream_t *, sout_stream_id_t * );
+static int Send ( sout_stream_t *, sout_stream_id_t *, block_t * );
+
+struct sout_stream_sys_t
+{
+ FILE *output;
+ char *prefix;
+};
+
+struct sout_stream_id_t
+{
+ int id;
+ uint64_t segment_number;
+ void *next_id;
+ char *type;
+ mtime_t previous_dts;
+};
+
+/*****************************************************************************
+ * Open:
+ *****************************************************************************/
+static int Open( vlc_object_t *p_this )
+{
+ sout_stream_t *p_stream = (sout_stream_t*)p_this;
+ sout_stream_sys_t *p_sys;
+ char *outputFile;
+
+ p_sys = calloc( 1, sizeof( sout_stream_sys_t ) );
+ if( !p_sys )
+ return VLC_ENOMEM;
+
+
+ config_ChainParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options,
+ p_stream->p_cfg );
+
+
+ outputFile = var_GetNonEmptyString( p_stream, SOUT_CFG_PREFIX "output" );
+
+ if( outputFile )
+ {
+ p_sys->output = vlc_fopen( outputFile, "wt" );
+ if( !p_sys->output )
+ {
+ msg_Err( p_stream, "Unable to open file '%s' for writing", outputFile );
+ free( p_sys );
+ free( outputFile );
+ return VLC_EGENERIC;
+ }
+ }
+ free( outputFile );
+ p_sys->prefix = var_GetString( p_stream, SOUT_CFG_PREFIX "prefix" );
+
+ p_stream->p_sys = p_sys;
+
+ p_stream->pf_add = Add;
+ p_stream->pf_del = Del;
+ p_stream->pf_send = Send;
+
+
+ return VLC_SUCCESS;
+}
+
+/*****************************************************************************
+ * Close:
+ *****************************************************************************/
+static void Close( vlc_object_t * p_this )
+{
+ sout_stream_t *p_stream = (sout_stream_t*)p_this;
+ sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
+
+ if( p_sys->output )
+ fclose( p_sys->output );
+
+ free( p_sys->prefix );
+ free( p_sys );
+}
+
+static sout_stream_id_t * Add( sout_stream_t *p_stream, es_format_t *p_fmt )
+{
+ sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
+ sout_stream_id_t *id;
+
+ id = calloc( 1, sizeof( sout_stream_id_t ) );
+ if( !id )
+ return NULL;
+
+ id->id = p_fmt->i_id;
+ switch( p_fmt->i_cat)
+ {
+ case VIDEO_ES:
+ id->type = strdup("Video");
+ break;
+ case AUDIO_ES:
+ id->type = strdup("Audio");
+ break;
+ case SPU_ES:
+ id->type = strdup("Spu");
+ break;
+ default:
+ id->type = strdup("Data");
+ break;
+ }
+
+ msg_Dbg( p_stream, "%s: Adding track type:%s id:%d", p_sys->prefix, id->type, id->id);
+
+ if( p_stream->p_next )
+ id->next_id = sout_StreamIdAdd( p_stream->p_next, p_fmt );
+
+ return id;
+}
+
+static int Del( sout_stream_t *p_stream, sout_stream_id_t *id )
+{
+ sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
+
+ msg_Dbg( p_stream, "%s: Removing track type:%s id:%d", p_sys->prefix, id->type, id->id );
+ if( id->next_id ) sout_StreamIdDel( p_stream->p_next, id->next_id );
+ free( id->type );
+ free( id );
+
+ return VLC_SUCCESS;
+}
+
+static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
+ block_t *p_buffer )
+{
+ sout_stream_sys_t *p_sys = (sout_stream_sys_t *)p_stream->p_sys;
+ struct md5_s hash;
+
+ block_t *p_block = p_buffer;
+ while ( p_block != NULL )
+ {
+ InitMD5( &hash );
+ AddMD5( &hash, p_block->p_buffer, p_block->i_buffer );
+ EndMD5( &hash );
+ char *outputhash = psz_md5_hash( &hash );
+ if( p_sys->output )
+ {
+ fprintf( p_sys->output, "%s: track:%d type:%s segment_number:%"PRIu64" dts_difference:%"PRId64" length:%"PRId64" md5:%16s\n",
+ p_sys->prefix, id->id, id->type, ++id->segment_number, p_block->i_dts - id->previous_dts,
+ p_block->i_length, outputhash );
+
+ } else {
+ msg_Dbg( p_stream, "%s: track:%d type:%s segment_number:%"PRIu64" dts_difference:%"PRId64" length:%"PRId64" md5:%16s",
+ p_sys->prefix, id->id, id->type, ++id->segment_number, p_block->i_dts - id->previous_dts,
+ p_block->i_length, outputhash );
+ }
+ free( outputhash );
+ id->previous_dts = p_block->i_dts;
+ p_block = p_block->p_next;
+ }
+
+ if( p_stream->p_next )
+ return sout_StreamIdSend( p_stream->p_next, id->next_id, p_buffer );
+ else
+ block_Release( p_buffer );
+ return VLC_SUCCESS;
+}
--
1.8.4.5
More information about the vlc-devel
mailing list