/*

 * Copyright (C) 2004 Benjamin Otte <otte@gnome.org>

 *

 * 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.

 */

#include <stdlib.h>

#include <gst/gst.h>

#define IDENTITY_COUNT (1000)

#define BUFFER_COUNT (1000)

#define SRC_ELEMENT "fakesrc"

#define SINK_ELEMENT "fakesink"

gint

main (gint argc, gchar * argv[])

{

  GstMessage *msg;

  GstElement *pipeline, *src, *sink, *current, *last;

  guint i, buffers = BUFFER_COUNT, identities = IDENTITY_COUNT;

  GstClockTime start, end;

  const gchar *src_name = SRC_ELEMENT, *sink_name = SINK_ELEMENT;

  gst_init (&argc, &argv);

  if (argc > 1)

    identities = atoi (argv[1]);

  if (argc > 2)

    buffers = atoi (argv[2]);

  if (argc > 3)

    src_name = argv[3];

  if (argc > 4)

    sink_name = argv[4];

  g_print

      ("*** benchmarking this pipeline: %s num-buffers=%u ! %u * identity ! %s\n",

      src_name, buffers, identities, sink_name);

  start = gst_util_get_timestamp ();

  pipeline = gst_element_factory_make ("pipeline", NULL);

  g_assert (pipeline);

  src = gst_element_factory_make (src_name, NULL);

  if (!src) {

    g_print ("no element named \"%s\" found, aborting...\n", src_name);

    return 1;

  }

  g_object_set (src, "num-buffers", buffers, NULL);

  sink = gst_element_factory_make (sink_name, NULL);

  if (!sink) {

    g_print ("no element named \"%s\" found, aborting...\n", sink_name);

    return 1;

  }

  last = src;

  gst_bin_add_many (GST_BIN (pipeline), src, sink, NULL);

  for (i = 0; i < identities; i++) {

    current = gst_element_factory_make ("identity", NULL);

    g_assert (current);

    /* shut this element up (no g_strdup_printf please) */

    g_object_set (current, "silent", TRUE, NULL);

    gst_bin_add (GST_BIN (pipeline), current);

    if (!gst_element_link (last, current))

      g_assert_not_reached ();

    last = current;

  }

  if (!gst_element_link (last, sink))

    g_assert_not_reached ();

  end = gst_util_get_timestamp ();

  g_print ("%" GST_TIME_FORMAT " - creating %u identity elements\n",

      GST_TIME_ARGS (end - start), identities);

  start = gst_util_get_timestamp ();

  if (gst_element_set_state (pipeline,

          GST_STATE_PLAYING) == GST_STATE_CHANGE_FAILURE)

    g_assert_not_reached ();

  if (gst_element_get_state (pipeline, NULL, NULL,

          GST_CLOCK_TIME_NONE) == GST_STATE_CHANGE_FAILURE)

    g_assert_not_reached ();

  end = gst_util_get_timestamp ();

  g_print ("%" GST_TIME_FORMAT " - setting pipeline to playing\n",

      GST_TIME_ARGS (end - start));

  start = gst_util_get_timestamp ();

  msg = gst_bus_poll (gst_element_get_bus (pipeline),

      GST_MESSAGE_EOS | GST_MESSAGE_ERROR, -1);

  end = gst_util_get_timestamp ();

  gst_message_unref (msg);

  g_print ("%" GST_TIME_FORMAT " - putting %u buffers through\n",

      GST_TIME_ARGS (end - start), buffers);

  start = gst_util_get_timestamp ();

  if (gst_element_set_state (pipeline,

          GST_STATE_NULL) != GST_STATE_CHANGE_SUCCESS)

    g_assert_not_reached ();

  end = gst_util_get_timestamp ();

  g_print ("%" GST_TIME_FORMAT " - setting pipeline to NULL\n",

      GST_TIME_ARGS (end - start));

  start = gst_util_get_timestamp ();

  g_object_unref (pipeline);

  end = gst_util_get_timestamp ();

  g_print ("%" GST_TIME_FORMAT " - unreffing pipeline\n",

      GST_TIME_ARGS (end - start));

  return 0;

}