mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-04 23:46:43 +00:00
ba183af479
Split out the benchmarking code from the unit test and make it a bit more useful.
154 lines
4.7 KiB
C
154 lines
4.7 KiB
C
/* GStreamer video format conversion benchmark
|
|
* Copyright (C) 2014 Wim Taymans <wim.taymans@gmail.com>
|
|
* Copyright (C) 2019 Tim-Philipp Müller <tim centricular com>
|
|
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/video/video.h>
|
|
|
|
#define DEFAULT_WIDTH 1920
|
|
#define DEFAULT_HEIGHT 1080
|
|
|
|
#define DEFAULT_DURATION 2.0
|
|
|
|
static gint
|
|
get_num_formats (void)
|
|
{
|
|
gint num_formats = 100;
|
|
|
|
while (gst_video_format_to_string (num_formats) == NULL)
|
|
--num_formats;
|
|
GST_INFO ("number of known video formats: %d", num_formats);
|
|
return num_formats + 1;
|
|
}
|
|
|
|
static void
|
|
do_benchmark_conversions (guint width, guint height, const gchar * in_format,
|
|
const gchar * out_format, gdouble max_duration)
|
|
{
|
|
const gchar *infmt_str, *outfmt_str;
|
|
GstVideoFormat infmt, outfmt;
|
|
GTimer *timer;
|
|
gint num_formats;
|
|
|
|
timer = g_timer_new ();
|
|
|
|
num_formats = get_num_formats ();
|
|
|
|
for (infmt = GST_VIDEO_FORMAT_I420; infmt < num_formats; infmt++) {
|
|
GstVideoInfo ininfo;
|
|
GstVideoFrame inframe;
|
|
GstBuffer *inbuffer;
|
|
|
|
infmt_str = gst_video_format_to_string (infmt);
|
|
if (in_format != NULL && !g_str_equal (in_format, infmt_str))
|
|
continue;
|
|
|
|
gst_video_info_set_format (&ininfo, infmt, width, height);
|
|
inbuffer = gst_buffer_new_and_alloc (ininfo.size);
|
|
gst_buffer_memset (inbuffer, 0, 0, -1);
|
|
gst_video_frame_map (&inframe, &ininfo, inbuffer, GST_MAP_READ);
|
|
|
|
for (outfmt = GST_VIDEO_FORMAT_I420; outfmt < num_formats; outfmt++) {
|
|
GstVideoInfo outinfo;
|
|
GstVideoFrame outframe;
|
|
GstBuffer *outbuffer;
|
|
GstVideoConverter *convert;
|
|
gdouble elapsed, convert_sec;
|
|
gint count;
|
|
|
|
outfmt_str = gst_video_format_to_string (outfmt);
|
|
if (out_format != NULL && !g_str_equal (out_format, outfmt_str))
|
|
continue;
|
|
|
|
/* Or maybe we should allocate more buffers to minimise cache effects? */
|
|
gst_video_info_set_format (&outinfo, outfmt, width, height);
|
|
outbuffer = gst_buffer_new_and_alloc (outinfo.size);
|
|
gst_video_frame_map (&outframe, &outinfo, outbuffer, GST_MAP_WRITE);
|
|
|
|
convert = gst_video_converter_new (&ininfo, &outinfo, NULL);
|
|
/* warmup */
|
|
gst_video_converter_frame (convert, &inframe, &outframe);
|
|
|
|
count = 0;
|
|
g_timer_start (timer);
|
|
while (TRUE) {
|
|
gst_video_converter_frame (convert, &inframe, &outframe);
|
|
|
|
count++;
|
|
elapsed = g_timer_elapsed (timer, NULL);
|
|
if (elapsed >= max_duration)
|
|
break;
|
|
}
|
|
|
|
convert_sec = count / elapsed;
|
|
|
|
gst_println ("%8.1f conversions/sec %s -> %s @ %ux%u, %d/%.5f",
|
|
convert_sec, infmt_str, outfmt_str, width, height, count, elapsed);
|
|
|
|
gst_video_converter_free (convert);
|
|
|
|
gst_video_frame_unmap (&outframe);
|
|
gst_buffer_unref (outbuffer);
|
|
}
|
|
gst_video_frame_unmap (&inframe);
|
|
gst_buffer_unref (inbuffer);
|
|
}
|
|
|
|
g_timer_destroy (timer);
|
|
}
|
|
|
|
int
|
|
main (int argc, char **argv)
|
|
{
|
|
GError *err = NULL;
|
|
gint width = DEFAULT_WIDTH;
|
|
gint height = DEFAULT_HEIGHT;
|
|
gdouble max_dur = DEFAULT_DURATION;
|
|
gchar *from_fmt = NULL;
|
|
gchar *to_fmt = NULL;
|
|
GOptionContext *ctx;
|
|
GOptionEntry options[] = {
|
|
{"width", 'w', 0, G_OPTION_ARG_INT, &width, "Width", NULL},
|
|
{"height", 'h', 0, G_OPTION_ARG_INT, &height, "Height", NULL},
|
|
{"from-format", 'f', 0, G_OPTION_ARG_STRING, &from_fmt, "From Format",
|
|
NULL},
|
|
{"to-format", 't', 0, G_OPTION_ARG_STRING, &to_fmt, "To Format", NULL},
|
|
{"duration", 'd', 0, G_OPTION_ARG_DOUBLE, &max_dur,
|
|
"Benchmark duration for each run (in seconds)", NULL},
|
|
{NULL}
|
|
};
|
|
|
|
ctx = g_option_context_new ("");
|
|
g_option_context_add_main_entries (ctx, options, NULL);
|
|
g_option_context_add_group (ctx, gst_init_get_option_group ());
|
|
if (!g_option_context_parse (ctx, &argc, &argv, &err)) {
|
|
g_print ("Error initializing: %s\n", GST_STR_NULL (err->message));
|
|
g_option_context_free (ctx);
|
|
g_clear_error (&err);
|
|
return 1;
|
|
}
|
|
g_option_context_free (ctx);
|
|
|
|
do_benchmark_conversions (width, height, from_fmt, to_fmt, max_dur);
|
|
return 0;
|
|
}
|