From fe366f7b24779bfc6ba9f92bbcf6a7ae08edaa6c Mon Sep 17 00:00:00 2001
From: Edward Hervey <bilboed@bilboed.com>
Date: Wed, 23 Sep 2009 16:19:32 +0200
Subject: [PATCH] benchmark: New benchmark for testing contention when creating
 buffers

---
 tests/benchmarks/Makefile.am       |   3 +-
 tests/benchmarks/gstbufferstress.c | 132 +++++++++++++++++++++++++++++++++++++
 2 files changed, 134 insertions(+), 1 deletion(-)
 create mode 100644 tests/benchmarks/gstbufferstress.c

diff --git a/tests/benchmarks/Makefile.am b/tests/benchmarks/Makefile.am
index 6d07065..e18c8de 100644
--- a/tests/benchmarks/Makefile.am
+++ b/tests/benchmarks/Makefile.am
@@ -5,7 +5,8 @@ noinst_PROGRAMS = \
         init \
         mass-elements \
         gstpollstress \
-        gstclockstress
+        gstclockstress	\
+	gstbufferstress
 
 LDADD = $(GST_OBJ_LIBS)
 AM_CFLAGS = $(GST_OBJ_CFLAGS)
diff --git a/tests/benchmarks/gstbufferstress.c b/tests/benchmarks/gstbufferstress.c
new file mode 100644
index 0000000..2305ad8
--- /dev/null
+++ b/tests/benchmarks/gstbufferstress.c
@@ -0,0 +1,132 @@
+/* GStreamer
+ * Copyright (C) <2009> Edward Hervey <bilboed@bilboed.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., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#include <stdlib.h>
+#include <gst/gst.h>
+
+#define MAX_THREADS  1000
+
+static guint64 nbbuffers;
+static GMutex *mutex;
+static GCond *cond;
+static gboolean ready = FALSE;
+
+static GstClockTime
+gst_get_current_time (void)
+{
+  GTimeVal tv;
+
+  g_get_current_time (&tv);
+  return GST_TIMEVAL_TO_TIME (tv);
+}
+
+static void *
+run_test (void *user_data)
+{
+  gint threadid = GPOINTER_TO_INT (user_data);
+  guint64 nb;
+  GstBuffer *buf;
+  GstClockTime start, end;
+
+  g_mutex_lock (mutex);
+  if (!ready) {
+    g_print ("Waiting for cond thread %d\n", threadid);
+    g_cond_wait (cond, mutex);
+  }
+  g_mutex_unlock (mutex);
+
+  start = gst_get_current_time ();
+
+  for (nb = nbbuffers; nb; nb--) {
+    buf = gst_buffer_new ();
+    gst_buffer_unref (buf);
+  }
+
+  end = gst_get_current_time ();
+  g_print ("total %" GST_TIME_FORMAT " - average %" GST_TIME_FORMAT
+      "  - Thread %d\n", GST_TIME_ARGS (end - start),
+      GST_TIME_ARGS ((end - start) / nbbuffers), threadid);
+
+
+  g_thread_exit (NULL);
+  return NULL;
+}
+
+gint
+main (gint argc, gchar * argv[])
+{
+  GThread *threads[MAX_THREADS];
+  gint num_threads;
+  gint t;
+  GstBuffer *tmp;
+  GstClockTime start, end;
+
+  gst_init (&argc, &argv);
+  mutex = g_mutex_new ();
+  cond = g_cond_new ();
+
+  if (argc != 3) {
+    g_print ("usage: %s <num_threads> <nbbuffers>\n", argv[0]);
+    exit (-1);
+  }
+
+  num_threads = atoi (argv[1]);
+  nbbuffers = atoi (argv[2]);
+
+  /* Let's just make sure the GstBufferClass is loaded ... */
+  tmp = gst_buffer_new ();
+
+  printf ("main(): Creating %d threads.\n", num_threads);
+  for (t = 0; t < num_threads; t++) {
+    GError *error = NULL;
+
+    threads[t] = g_thread_create (run_test, GINT_TO_POINTER (t), TRUE, &error);
+    if (error) {
+      printf ("ERROR: g_thread_create() %s\n", error->message);
+      exit (-1);
+    }
+  }
+
+  g_print ("MAIN taking lock\n");
+  g_mutex_lock (mutex);
+  start = gst_get_current_time ();
+  g_print ("MAIN broadcasting\n");
+  ready = TRUE;
+  g_cond_broadcast (cond);
+  g_print ("MAIN RELEASING\n");
+  g_mutex_unlock (mutex);
+  /* Signal all threads to start */
+
+  for (t = 0; t < num_threads; t++) {
+    if (threads[t])
+      g_thread_join (threads[t]);
+  }
+
+  end = gst_get_current_time ();
+  g_print ("*** total %" GST_TIME_FORMAT " - average %" GST_TIME_FORMAT
+      "  - Done creating %" G_GUINT64_FORMAT " buffers\n",
+      GST_TIME_ARGS (end - start),
+      GST_TIME_ARGS ((end - start) / (num_threads * nbbuffers)),
+      num_threads * nbbuffers);
+
+
+  gst_buffer_unref (tmp);
+
+  return 0;
+}
-- 
2.7.4