gstreamer/tests/lat.c

218 lines
5.6 KiB
C
Raw Normal View History

#include <gst/gst.h>
#include <stdlib.h>
#include <string.h>
/* FIXME: WTF does this do? */
static guint64 max = 0, min = -1, total = 0;
static guint count = 0;
static guint print_del = 1;
static guint iterations = 0;
static guint mhz = 0;
void handoff_src(GstElement *src, GstBuffer *buf, gpointer user_data) {
gst_trace_read_tsc(&GST_BUFFER_TIMESTAMP(buf));
}
void handoff_sink(GstElement *sink, GstBuffer *buf, gpointer user_data) {
guint64 end, d, avg;
guint avg_ns;
gst_trace_read_tsc(&end);
d = end - GST_BUFFER_TIMESTAMP(buf);
if (d > max) max = d;
if (d < min) min = d;
total += d;
count++;
avg = total/count;
avg_ns = (guint)(1000.0*(double)avg/(double)mhz);
if ((count % print_del) == 0) {
g_print("%07d:%08lld min:%08lld max:%08lld avg:%08lld avg-s:0.%09d\r",
count, d, min, max, avg, avg_ns);
}
}
GstElement *identity_add(GstPipeline *pipeline, GstElement *first, int count) {
GstElement *last, *ident;
int i;
char buf[20];
last = first;
for (i=0; i<count; i++) {
snprintf(buf, 20, "identity_%03d", i);
ident = gst_element_factory_make("identity",buf);
g_return_val_if_fail(ident != NULL,NULL);
g_object_set(G_OBJECT(ident),"silent",TRUE,NULL);
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(ident));
gst_pad_connect(gst_element_get_pad(last,"src"),
gst_element_get_pad(ident,"sink"));
last = ident;
}
return last;
}
GstElement *fakesrc() {
GstElement *src;
src = gst_element_factory_make("fakesrc","src");
g_return_val_if_fail(src != NULL,NULL);
g_object_set(G_OBJECT(src),"silent",TRUE,NULL);
g_object_set(G_OBJECT(src),"num_buffers",iterations,NULL);
g_signal_connect(G_OBJECT(src),
"handoff",G_CALLBACK(handoff_src),NULL);
return src;
}
GstElement *fakesink() {
GstElement *sink;
sink = gst_element_factory_make("fakesink","fakesink");
g_return_val_if_fail(sink != NULL,NULL);
g_object_set(G_OBJECT(sink),"silent",TRUE,NULL);
g_signal_connect(G_OBJECT(sink),
"handoff",G_CALLBACK(handoff_sink),NULL);
return sink;
}
GstPipeline *simple(int argc, int argi, char *argv[]) {
GstPipeline *pipeline;
GstElement *last, *src, *sink;
int idents;
if ((argc - argi) < 1) {
fprintf(stderr, "bad params");
return NULL;
}
idents = atoi(argv[argi]);
if ((argc - argi) == 2) {
gst_scheduler_factory_set_default_name (argv[argi+1]);
}
pipeline = GST_PIPELINE(gst_pipeline_new("pipeline"));
g_return_val_if_fail(pipeline != NULL,NULL);
src = fakesrc();
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(src));
last = identity_add(pipeline, src, idents);
sink = fakesink();
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(sink));
gst_pad_connect(gst_element_get_pad(last,"src"),
gst_element_get_pad(sink,"sink"));
return pipeline;
}
GstPipeline *queue(int argc, int argi, char *argv[]) {
GstPipeline *pipeline;
GstElement *last, *src, *sink, *src_thr, *src_q, *sink_q, *sink_thr;
int idents;
if ((argc - argi) < 1) {
fprintf(stderr, "bad params");
return NULL;
}
idents = atoi(argv[argi]);
if ((argc - argi) == 2) {
gst_scheduler_factory_set_default_name (argv[argi+1]);
}
pipeline = GST_PIPELINE(gst_pipeline_new("pipeline"));
g_return_val_if_fail(pipeline != NULL,NULL);
src_thr = GST_ELEMENT(gst_thread_new("src_thread"));
g_return_val_if_fail(src_thr != NULL,NULL);
src = fakesrc();
g_return_val_if_fail(src != NULL,NULL);
gst_bin_add(GST_BIN(src_thr),GST_ELEMENT(src));
src_q = gst_element_factory_make("queue","src_q");
g_return_val_if_fail(src_q != NULL,NULL);
gst_bin_add(GST_BIN(src_thr),GST_ELEMENT(src_q));
gst_pad_connect(gst_element_get_pad(src,"src"),
gst_element_get_pad(src_q,"sink"));
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(src_thr));
last = identity_add(pipeline, src_q, idents);
sink_q = gst_element_factory_make("queue","sink_q");
g_return_val_if_fail(sink_q != NULL,NULL);
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(sink_q));
gst_pad_connect(gst_element_get_pad(last,"src"),
gst_element_get_pad(sink_q,"sink"));
sink_thr = GST_ELEMENT(gst_thread_new("sink_thread"));
g_return_val_if_fail(sink_thr != NULL,NULL);
sink = fakesink();
g_return_val_if_fail(sink != NULL,NULL);
gst_bin_add(GST_BIN(sink_thr),GST_ELEMENT(sink));
gst_bin_add(GST_BIN(pipeline),GST_ELEMENT(sink_thr));
gst_pad_connect(gst_element_get_pad(sink_q,"src"),
gst_element_get_pad(sink,"sink"));
return pipeline;
}
struct test {
char *name;
char *params;
GstPipeline *(*func)(int argc, int argi, char *argv[]);
};
static struct test tests[] = {
{"simple", "ident_count [scheduler_name]", simple},
{"queue", "ident_count [scheduler_name]", queue},
{NULL, NULL, NULL}
};
int main(int argc,char *argv[]) {
GstPipeline *pipeline;
int i;
char *name;
gst_init(&argc,&argv);
if (argc < 3) {
fprintf(stderr, "usage: %s iterations print_del mhz test_name [test_params...]\n",
argv[0]);
for (i=0; tests[i].name; i++) {
fprintf(stderr, " %s %s\n", tests[i].name, tests[i].params);
}
exit(1);
} else {
iterations = atoi(argv[1]);
print_del = atoi(argv[2]);
mhz = atoi(argv[3]);
name = argv[4];
}
pipeline = NULL;
for (i=0; tests[i].name && !pipeline; i++) {
if (!strcmp(name, tests[i].name)) {
pipeline = tests[i].func(argc,5,argv);
}
}
g_return_val_if_fail(pipeline != NULL, -1);
/*xmlSaveFile("lat.gst", gst_xml_write(GST_ELEMENT(pipeline))); */
gst_element_set_state(GST_ELEMENT(pipeline),GST_STATE_PLAYING);
while (count < iterations) {
gst_bin_iterate(GST_BIN(pipeline));
}
g_print("\n");
return 0;
}