/* GStreamer Editing Services
 * Copyright (C) 2010 Brandon Lewis <brandon.lewis@collabora.co.uk>
 *
 * 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 <ges/ges.h>
#include <gst/check/gstcheck.h>

GST_START_TEST (test_test_source_basic)
{
  GESTimelineTestSource *source;

  ges_init ();

  source = ges_timeline_test_source_new ();
  fail_unless (source != NULL);

  g_object_unref (source);
}

GST_END_TEST;

#define gnl_object_check(gnlobj, start, duration, mstart, mduration, priority, active) { \
  guint64 pstart, pdur, pmstart, pmdur, pprio, pact;			\
  g_object_get (gnlobj, "start", &pstart, "duration", &pdur,		\
		"media-start", &pmstart, "media-duration", &pmdur,	\
		"priority", &pprio, "active", &pact,			\
		NULL);							\
  assert_equals_uint64 (pstart, start);					\
  assert_equals_uint64 (pdur, duration);					\
  assert_equals_uint64 (pmstart, mstart);					\
  assert_equals_uint64 (pmdur, mduration);					\
  assert_equals_int (pprio, priority);					\
  assert_equals_int (pact, active);					\
  }


GST_START_TEST (test_test_source_properties)
{
  GESTrack *track;
  GESTrackObject *trackobject;
  GESTimelineObject *object;

  ges_init ();

  track = ges_track_new (GES_TRACK_TYPE_AUDIO, GST_CAPS_ANY);
  fail_unless (track != NULL);

  object = (GESTimelineObject *)
      ges_timeline_test_source_new ();
  fail_unless (object != NULL);

  /* Set some properties */
  g_object_set (object, "start", (guint64) 42, "duration", (guint64) 51,
      "in-point", (guint64) 12, NULL);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_START (object), 42);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_DURATION (object), 51);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_INPOINT (object), 12);

  trackobject = ges_timeline_object_create_track_object (object, track);
  ges_timeline_object_add_track_object (object, trackobject);
  fail_unless (trackobject != NULL);
  fail_unless (ges_track_object_set_track (trackobject, track));

  /* Check that trackobject has the same properties */
  assert_equals_uint64 (GES_TRACK_OBJECT_START (trackobject), 42);
  assert_equals_uint64 (GES_TRACK_OBJECT_DURATION (trackobject), 51);
  assert_equals_uint64 (GES_TRACK_OBJECT_INPOINT (trackobject), 12);

  /* And let's also check that it propagated correctly to GNonLin */
  gnl_object_check (ges_track_object_get_gnlobject (trackobject), 42, 51, 12,
      51, 0, TRUE);

  /* Change more properties, see if they propagate */
  g_object_set (object, "start", (guint64) 420, "duration", (guint64) 510,
      "in-point", (guint64) 120, NULL);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_START (object), 420);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_DURATION (object), 510);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_INPOINT (object), 120);
  assert_equals_uint64 (GES_TRACK_OBJECT_START (trackobject), 420);
  assert_equals_uint64 (GES_TRACK_OBJECT_DURATION (trackobject), 510);
  assert_equals_uint64 (GES_TRACK_OBJECT_INPOINT (trackobject), 120);

  /* And let's also check that it propagated correctly to GNonLin */
  gnl_object_check (ges_track_object_get_gnlobject (trackobject), 420, 510, 120,
      510, 0, TRUE);

  /* Test mute support */
  g_object_set (object, "mute", TRUE, NULL);
  gnl_object_check (ges_track_object_get_gnlobject (trackobject), 420, 510, 120,
      510, 0, FALSE);
  g_object_set (object, "mute", FALSE, NULL);
  gnl_object_check (ges_track_object_get_gnlobject (trackobject), 420, 510, 120,
      510, 0, TRUE);

  ges_timeline_object_release_track_object (object, trackobject);
  g_object_unref (object);
}

GST_END_TEST;

GST_START_TEST (test_test_source_in_layer)
{
  GESTimeline *timeline;
  GESTimelineLayer *layer;
  GESTrack *a, *v;
  GESTrackObject *trobj;
  GESTimelineTestSource *source;
  GESVideoTestPattern ptrn;
  gdouble freq, volume;

  ges_init ();

  timeline = ges_timeline_new ();
  layer = (GESTimelineLayer *) ges_simple_timeline_layer_new ();
  a = ges_track_audio_raw_new ();
  v = ges_track_video_raw_new ();

  ges_timeline_add_track (timeline, a);
  ges_timeline_add_track (timeline, v);
  ges_timeline_add_layer (timeline, layer);

  source = ges_timeline_test_source_new_for_nick ((gchar *) "red");
  g_object_get (source, "vpattern", &ptrn, NULL);
  assert_equals_int (ptrn, GES_VIDEO_TEST_PATTERN_RED);

  g_object_set (source, "duration", (guint64) GST_SECOND, NULL);

  ges_simple_timeline_layer_add_object ((GESSimpleTimelineLayer *) layer,
      (GESTimelineObject *) source, 0);

  /* specifically test the vpattern property */
  g_object_set (source, "vpattern", (gint) GES_VIDEO_TEST_PATTERN_WHITE, NULL);
  g_object_get (source, "vpattern", &ptrn, NULL);
  assert_equals_int (ptrn, GES_VIDEO_TEST_PATTERN_WHITE);

  trobj =
      ges_timeline_object_find_track_object (GES_TIMELINE_OBJECT (source), v,
      GES_TYPE_TRACK_VIDEO_TEST_SOURCE);

  ptrn = (ges_track_video_test_source_get_pattern ((GESTrackVideoTestSource *)
          trobj));
  assert_equals_int (ptrn, GES_VIDEO_TEST_PATTERN_WHITE);
  g_object_unref (trobj);

  /* test audio properties as well */

  trobj = ges_timeline_object_find_track_object (GES_TIMELINE_OBJECT (source),
      a, GES_TYPE_TRACK_AUDIO_TEST_SOURCE);
  g_assert (GES_IS_TRACK_AUDIO_TEST_SOURCE (trobj));
  assert_equals_float (ges_timeline_test_source_get_frequency (source), 440);
  assert_equals_float (ges_timeline_test_source_get_volume (source), 0);

  g_object_get (source, "freq", &freq, "volume", &volume, NULL);
  assert_equals_float (freq, 440);
  assert_equals_float (volume, 0);


  freq =
      ges_track_audio_test_source_get_freq (GES_TRACK_AUDIO_TEST_SOURCE
      (trobj));
  volume =
      ges_track_audio_test_source_get_volume (GES_TRACK_AUDIO_TEST_SOURCE
      (trobj));
  g_assert (freq == 440);
  g_assert (volume == 0);


  g_object_set (source, "freq", (gdouble) 2000, "volume", (gdouble) 0.5, NULL);

  g_object_get (source, "freq", &freq, "volume", &volume, NULL);
  assert_equals_float (freq, 2000);
  assert_equals_float (volume, 0.5);

  freq =
      ges_track_audio_test_source_get_freq (GES_TRACK_AUDIO_TEST_SOURCE
      (trobj));
  volume =
      ges_track_audio_test_source_get_volume (GES_TRACK_AUDIO_TEST_SOURCE
      (trobj));
  g_assert (freq == 2000);
  g_assert (volume == 0.5);

  g_object_unref (trobj);

  ges_timeline_layer_remove_object (layer, (GESTimelineObject *) source);

  GST_DEBUG ("removing the layer");

  g_object_unref (timeline);
}

GST_END_TEST;

static gint
find_composition_func (GstElement * element)
{
  GstElementFactory *fac = gst_element_get_factory (element);
  const gchar *name = gst_plugin_feature_get_name (GST_PLUGIN_FEATURE (fac));

  if (g_strcmp0 (name, "gnlcomposition") == 0)
    return 0;

  return 1;
}

static GstElement *
find_composition (GESTrack * track)
{
  GstIterator *it = gst_bin_iterate_recurse (GST_BIN (track));
  GstElement *ret =
      gst_iterator_find_custom (it, (GCompareFunc) find_composition_func, NULL);

  gst_iterator_free (it);

  return ret;
}

#define gap_object_check(gnlobj, start, duration, priority)  \
{                                                            \
  guint64 pstart, pdur, pprio;                               \
  g_object_get (gnlobj, "start", &pstart, "duration", &pdur, \
    "priority", &pprio, NULL);                               \
  assert_equals_uint64 (pstart, start);                      \
  assert_equals_uint64 (pdur, duration);                     \
  assert_equals_int (pprio, priority);                       \
}
GST_START_TEST (test_gap_filling_basic)
{
  GESTrack *track;
  GESTrackObject *trackobject, *trackobject1, *trackobject2;
  /*GESTimelineLayer *layer; */
  GESTimelineObject *object, *object1, *object2;
  GstElement *gnlsrc, *gnlsrc1, *gap = NULL;
  GstElement *composition;
  GList *tmp;

  ges_init ();

  track = ges_track_audio_raw_new ();
  fail_unless (track != NULL);

  composition = find_composition (track);
  fail_unless (composition != NULL);

  object = GES_TIMELINE_OBJECT (ges_timeline_test_source_new ());
  fail_unless (object != NULL);

  /* Set some properties */
  g_object_set (object, "start", (guint64) 0, "duration", (guint64) 5, NULL);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_START (object), 0);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_DURATION (object), 5);

  trackobject = ges_timeline_object_create_track_object (object, track);
  ges_timeline_object_add_track_object (object, trackobject);
  fail_unless (ges_track_add_object (track, trackobject));
  fail_unless (trackobject != NULL);
  gnlsrc = ges_track_object_get_gnlobject (trackobject);
  fail_unless (gnlsrc != NULL);

  /* Check that trackobject has the same properties */
  assert_equals_uint64 (GES_TRACK_OBJECT_START (trackobject), 0);
  assert_equals_uint64 (GES_TRACK_OBJECT_DURATION (trackobject), 5);

  /* Check no gap were wrongly added */
  assert_equals_int (g_list_length (GST_BIN_CHILDREN (composition)), 1);

  object1 = GES_TIMELINE_OBJECT (ges_timeline_test_source_new ());
  fail_unless (object1 != NULL);

  g_object_set (object1, "start", (guint64) 15, "duration", (guint64) 5, NULL);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_START (object1), 15);
  assert_equals_uint64 (GES_TIMELINE_OBJECT_DURATION (object1), 5);

  trackobject1 = ges_timeline_object_create_track_object (object1, track);
  ges_timeline_object_add_track_object (object1, trackobject1);
  fail_unless (ges_track_add_object (track, trackobject1));
  fail_unless (trackobject1 != NULL);
  gnlsrc1 = ges_track_object_get_gnlobject (trackobject1);
  fail_unless (gnlsrc1 != NULL);

  /* Check that trackobject1 has the same properties */
  assert_equals_uint64 (GES_TRACK_OBJECT_START (trackobject1), 15);
  assert_equals_uint64 (GES_TRACK_OBJECT_DURATION (trackobject1), 5);

  /* Check the gap as properly been added */
  assert_equals_int (g_list_length (GST_BIN_CHILDREN (composition)), 3);

  for (tmp = GST_BIN_CHILDREN (composition); tmp; tmp = tmp->next) {
    GstElement *tmp_gnlobj = GST_ELEMENT (tmp->data);

    if (tmp_gnlobj != gnlsrc && tmp_gnlobj != gnlsrc1) {
      gap = tmp_gnlobj;
    }
  }
  fail_unless (gap != NULL);
  gap_object_check (gap, 5, 10, 0)

      object2 = GES_TIMELINE_OBJECT (ges_timeline_test_source_new ());
  fail_unless (object2 != NULL);
  g_object_set (object2, "start", (guint64) 35, "duration", (guint64) 5, NULL);
  trackobject2 = ges_timeline_object_create_track_object (object2, track);
  ges_timeline_object_add_track_object (object2, trackobject2);
  fail_unless (ges_track_add_object (track, trackobject2));
  fail_unless (trackobject2 != NULL);
  assert_equals_uint64 (GES_TRACK_OBJECT_START (trackobject2), 35);
  assert_equals_uint64 (GES_TRACK_OBJECT_DURATION (trackobject2), 5);
  assert_equals_int (g_list_length (GST_BIN_CHILDREN (composition)), 5);

  gst_object_unref (track);
}

GST_END_TEST;

static Suite *
ges_suite (void)
{
  Suite *s = suite_create ("ges-backgroundsource");
  TCase *tc_chain = tcase_create ("backgroundsource");

  suite_add_tcase (s, tc_chain);

  tcase_add_test (tc_chain, test_test_source_basic);
  tcase_add_test (tc_chain, test_test_source_properties);
  tcase_add_test (tc_chain, test_test_source_in_layer);
  tcase_add_test (tc_chain, test_gap_filling_basic);

  return s;
}

int
main (int argc, char **argv)
{
  int nf;

  Suite *s = ges_suite ();
  SRunner *sr = srunner_create (s);

  gst_check_init (&argc, &argv);

  srunner_run_all (sr, CK_NORMAL);
  nf = srunner_ntests_failed (sr);
  srunner_free (sr);

  return nf;
}