/* GStreamer unit tests for subparse

 *

 * Copyright (C) 2006 Tim-Philipp Müller <tim centricular net>

 *

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

 */

#ifdef HAVE_CONFIG_H

# include <config.h>

#endif

#include <gst/gst_global.h>

#include <gst/check/gstcheck.h>

#include <string.h>

#ifndef EMULATOR

#define assert_failed *(get_assert_failed())

#define fp_std_log *(get_fp_std_log())

#define xmlfile *(get_xmlfile())

#endif

#define LOG_FILE "c:\\logs\\subparse_log1.txt"

#include "std_log_result.h"

#define LOG_FILENAME_LINE __FILE__, __LINE__

#include "libgstreamer_wsd_solution.h" 

#if EMULATOR

GET_GLOBAL_VAR_FROM_TLS(buffers,gstcheck,GList*)

#define buffers (*GET_GSTREAMER_WSD_VAR_NAME(buffers,gstcheck,g)())

#else 

extern GList *buffers;

#endif

void create_xml(int result)

{

    if(result)

        assert_failed = 1;

    testResultXml(xmlfile);

    close_log_file();

}

static GstStaticPadTemplate sinktemplate = GST_STATIC_PAD_TEMPLATE ("sink",

    GST_PAD_SINK,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("text/plain; text/x-pango-markup")

    );

static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("ANY")

    );

static GstElement *subparse;

static GstPad *mysrcpad, *mysinkpad;

static GstBuffer *

buffer_from_static_string (const gchar * s)

{

  GstBuffer *buf;

  buf = gst_buffer_new ();

  GST_BUFFER_DATA (buf) = (guint8 *) s;

  GST_BUFFER_SIZE (buf) = strlen (s);

  GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_READONLY);

  return buf;

}

typedef struct

{

  const gchar *in;

  GstClockTime from_ts;

  GstClockTime to_ts;

  const gchar *out;

} SubParseInputChunk;

static SubParseInputChunk srt_input[] = {

  {

        "1\n00:00:01,000 --> 00:00:02,000\nOne\n\n",

      1 * GST_SECOND, 2 * GST_SECOND, "One"}, {

        "2\n00:00:02,000 --> 00:00:03,000\nTwo\n\n",

      2 * GST_SECOND, 3 * GST_SECOND, "Two"}, {

        "3\n00:00:03,000 --> 00:00:04,000\nThree\n\n",

      3 * GST_SECOND, 4 * GST_SECOND, "Three"}, {

        "4\n00:00:04,000 --> 00:00:05,000\nFour\n\n",

      4 * GST_SECOND, 5 * GST_SECOND, "Four"}, {

        "5\n00:00:05,000 --> 00:00:06,000\nFive\n\n",

      5 * GST_SECOND, 6 * GST_SECOND, "Five"}, {

        /* markup should be preserved */

        "6\n00:00:06,000 --> 00:00:07,000\n<i>Six</i>\n\n",

      6 * GST_SECOND, 7 * GST_SECOND, "<i>Six</i>"}, {

        /* open markup tags should be closed */

        "7\n00:00:07,000 --> 00:00:08,000\n<i>Seven\n\n",

      7 * GST_SECOND, 8 * GST_SECOND, "<i>Seven</i>"}, {

        /* open markup tags should be closed (II) */

        "8\n00:00:08,000 --> 00:00:09,000\n<b><i>Eight\n\n",

      8 * GST_SECOND, 9 * GST_SECOND, "<b><i>Eight</i></b>"}, {

        /* broken markup should be fixed */

        "9\n00:00:09,000 --> 00:00:10,000\n</b>\n\n",

      9 * GST_SECOND, 10 * GST_SECOND, ""}, {

        "10\n00:00:10,000 --> 00:00:11,000\n</b></i>\n\n",

      10 * GST_SECOND, 11 * GST_SECOND, ""}, {

        "11\n00:00:11,000 --> 00:00:12,000\n<i>xyz</b></i>\n\n",

      11 * GST_SECOND, 12 * GST_SECOND, "<i>xyz</i>"}, {

        "12\n00:00:12,000 --> 00:00:13,000\n<i>xyz</b>\n\n",

      12 * GST_SECOND, 13 * GST_SECOND, "<i>xyz</i>"}, {

        /* skip a few chunk numbers here, the numbers shouldn't matter */

        "24\n00:01:00,000 --> 00:02:00,000\nYep, still here\n\n",

      60 * GST_SECOND, 120 * GST_SECOND, "Yep, still here"}, {

        /* make sure stuff is escaped properly, but allowed markup stays intact */

        "25\n00:03:00,000 --> 00:04:00,000\ngave <i>Rock & Roll</i> to\n\n",

      180 * GST_SECOND, 240 * GST_SECOND, "gave <i>Rock &amp; Roll</i> to"}, {

        "26\n00:04:00,000 --> 00:05:00,000\n<i>Rock & Roll</i>\n\n",

      240 * GST_SECOND, 300 * GST_SECOND, "<i>Rock &amp; Roll</i>"}, {

        "27\n00:06:00,000 --> 00:08:00,000\nRock & Roll\n\n",

      360 * GST_SECOND, 480 * GST_SECOND, "Rock & Roll"}, {

        "28\n00:10:00,000 --> 00:11:00,000\n"

        "<font \"#0000FF\"><joj>This is </xxx>in blue but <5</font>\n\n",

      600 * GST_SECOND, 660 * GST_SECOND, "This is in blue but <5"}

};

/* starts with chunk number 0 (not exactly according to spec) */

static SubParseInputChunk srt_input0[] = {

  {

        "0\n00:00:01,000 --> 00:00:02,000\nOne\n\n",

      1 * GST_SECOND, 2 * GST_SECOND, "One"}, {

        "1\n00:00:02,000 --> 00:00:03,000\nTwo\n\n",

      2 * GST_SECOND, 3 * GST_SECOND, "Two"}, {

        "2\n00:00:03,000 --> 00:00:04,000\nThree\n\n",

      3 * GST_SECOND, 4 * GST_SECOND, "Three"}

};

static void

setup_subparse (void)

{

  subparse = gst_check_setup_element ("subparse");

  mysrcpad = gst_check_setup_src_pad (subparse, &srctemplate, NULL);

  mysinkpad = gst_check_setup_sink_pad (subparse, &sinktemplate, NULL);

  gst_pad_set_active (mysrcpad, TRUE);

  gst_pad_set_active (mysinkpad, TRUE);

  fail_unless_equals_int (gst_element_set_state (subparse, GST_STATE_PLAYING),

  GST_STATE_CHANGE_SUCCESS);

}

static void

teardown_subparse (void)

{

  GST_DEBUG ("cleaning up");

  g_list_foreach (buffers, (GFunc) gst_mini_object_unref, NULL);

  g_list_free (buffers);

  buffers = NULL;

  gst_pad_set_active (mysrcpad, FALSE);

  gst_pad_set_active (mysinkpad, FALSE);

  gst_check_teardown_sink_pad (subparse);

  gst_check_teardown_src_pad (subparse);

  gst_check_teardown_element (subparse);

  subparse = NULL;

  mysrcpad = NULL;

  mysinkpad = NULL;

}

static void

test_srt_do_test (SubParseInputChunk * input, guint start_idx, guint num)

{

  guint n;

  GST_LOG ("srt test: start_idx = %u, num = %u", start_idx, num);

  setup_subparse ();

  for (n = start_idx; n < start_idx + num; ++n) {

    GstBuffer *buf;

    buf = buffer_from_static_string (input[n].in);

    fail_unless_equals_int (gst_pad_push (mysrcpad, buf), GST_FLOW_OK);

  }

  gst_pad_push_event (mysrcpad, gst_event_new_eos ());

  fail_unless_equals_int (g_list_length (buffers), num);

  for (n = start_idx; n < start_idx + num; ++n) {

    const GstStructure *buffer_caps_struct;

    GstBuffer *buf;

    gchar *out;

    guint out_size;

    buf = g_list_nth_data (buffers, n - start_idx);

    fail_unless (buf != NULL);

    fail_unless (GST_BUFFER_TIMESTAMP_IS_VALID (buf), NULL);

    fail_unless (GST_BUFFER_DURATION_IS_VALID (buf), NULL);

    fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buf), input[n].from_ts);

    fail_unless_equals_uint64 (GST_BUFFER_DURATION (buf),

        input[n].to_ts - input[n].from_ts);

    out = (gchar *) GST_BUFFER_DATA (buf);

    out_size = GST_BUFFER_SIZE (buf);

    /* shouldn't have trailing newline characters */

    fail_if (out_size > 0 && out[out_size - 1] == '\n');

    /* shouldn't include NUL-terminator in data size */

    fail_if (out_size > 0 && out[out_size - 1] == '\0');

    /* but should still have a  NUL-terminator behind the declared data */

    fail_unless_equals_int (out[out_size], '\0');

    /* make sure out string matches expected string */

    fail_unless_equals_string (out, input[n].out);

    /* check caps */

    fail_unless (GST_BUFFER_CAPS (buf) != NULL);

    buffer_caps_struct = gst_caps_get_structure (GST_BUFFER_CAPS (buf), 0);

    fail_unless_equals_string (gst_structure_get_name (buffer_caps_struct),

        "text/x-pango-markup");

  }

  teardown_subparse ();

}

void test_srt()

{

  test_srt_do_test (srt_input, 0, G_N_ELEMENTS (srt_input));

	xmlfile = "subparse_test_srt";

	std_log(LOG_FILENAME_LINE, "Test Started test_srt");

  /* make sure everything works fine if we don't start with chunk 1 */

  test_srt_do_test (srt_input, 1, G_N_ELEMENTS (srt_input) - 1);

  test_srt_do_test (srt_input, 2, G_N_ELEMENTS (srt_input) - 2);

  test_srt_do_test (srt_input, 3, G_N_ELEMENTS (srt_input) - 3);

  test_srt_do_test (srt_input, 4, G_N_ELEMENTS (srt_input) - 4);

  /* try with empty input, immediate EOS */

  test_srt_do_test (srt_input, 5, G_N_ELEMENTS (srt_input) - 5);

  /* try with chunk number 0 (which is not exactly according to spec) */

  test_srt_do_test (srt_input0, 0, G_N_ELEMENTS (srt_input0));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

static void

do_test (SubParseInputChunk * input, guint num, const gchar * media_type)

{

  guint n;

  setup_subparse ();

  for (n = 0; n < num; ++n) {

    GstBuffer *buf;

    buf = buffer_from_static_string (input[n].in);

    fail_unless_equals_int (gst_pad_push (mysrcpad, buf), GST_FLOW_OK);

  }

  gst_pad_push_event (mysrcpad, gst_event_new_eos ());

  fail_unless_equals_int (g_list_length (buffers), num);

  for (n = 0; n < num; ++n) {

    const GstStructure *buffer_caps_struct;

    GstBuffer *buf;

    gchar *out;

    guint out_size;

    buf = g_list_nth_data (buffers, n);

    fail_unless (buf != NULL);

    fail_unless (GST_BUFFER_TIMESTAMP_IS_VALID (buf), NULL);

    fail_unless (GST_BUFFER_DURATION_IS_VALID (buf), NULL);

    fail_unless_equals_uint64 (GST_BUFFER_TIMESTAMP (buf), input[n].from_ts);

    fail_unless_equals_uint64 (GST_BUFFER_DURATION (buf),

        input[n].to_ts - input[n].from_ts);

    out = (gchar *) GST_BUFFER_DATA (buf);

    out_size = GST_BUFFER_SIZE (buf);

    /* shouldn't have trailing newline characters */

    fail_if (out_size > 0 && out[out_size - 1] == '\n');

    /* shouldn't include NUL-terminator in data size */

    fail_if (out_size > 0 && out[out_size - 1] == '\0');

    /* but should still have a  NUL-terminator behind the declared data */

    fail_unless_equals_int (out[out_size], '\0');

    /* make sure out string matches expected string */

    fail_unless_equals_string (out, input[n].out);

    /* check caps */

    fail_unless (GST_BUFFER_CAPS (buf) != NULL);

    buffer_caps_struct = gst_caps_get_structure (GST_BUFFER_CAPS (buf), 0);

    fail_unless_equals_string (gst_structure_get_name (buffer_caps_struct),

        media_type);

  }

  teardown_subparse ();

}

static void

test_tmplayer_do_test (SubParseInputChunk * input, guint num)

{

  do_test (input, num, "text/plain");

}

static void

test_microdvd_do_test (SubParseInputChunk * input, guint num)

{

  do_test (input, num, "text/x-pango-markup");

}

void test_tmplayer_multiline()

{

  static SubParseInputChunk tmplayer_multiline_input[] = {

    {

          "00:00:10,1=This is the Earth at a time\n"

          "00:00:10,2=when the dinosaurs roamed...\n" "00:00:13,1=\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "00:00:14,1=a lush and fertile planet.\n" "00:00:16,1=\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_multiline";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_multiline");

  test_tmplayer_do_test (tmplayer_multiline_input,

      G_N_ELEMENTS (tmplayer_multiline_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_multiline_with_bogus_lines()

{

  static SubParseInputChunk tmplayer_multiline_b_input[] = {

    {

          "00:00:10,1=This is the Earth at a time\n"

          "Yooboo wabahablablahuguug bogus line hello test 1-2-3-4\n"

          "00:00:10,2=when the dinosaurs roamed...\n" "00:00:13,1=\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "00:00:14,1=a lush and fertile planet.\n" "00:00:16,1=\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_multiline_with_bogus_lines";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_multiline_with_bogus_lines");

  test_tmplayer_do_test (tmplayer_multiline_b_input,

      G_N_ELEMENTS (tmplayer_multiline_b_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_style1()

{

  static SubParseInputChunk tmplayer_style1_input[] = {

    {

          "00:00:10:This is the Earth at a time|when the dinosaurs roamed...\n"

          "00:00:13:\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "00:00:14:a lush and fertile planet.\n" "00:00:16:\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_style1";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_style1");

  test_tmplayer_do_test (tmplayer_style1_input,

      G_N_ELEMENTS (tmplayer_style1_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_style2()

{

  static SubParseInputChunk tmplayer_style2_input[] = {

    {

          "00:00:10=This is the Earth at a time|when the dinosaurs roamed...\n"

          "00:00:13=\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "00:00:14=a lush and fertile planet.\n" "00:00:16=\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_style2";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_style2");

  test_tmplayer_do_test (tmplayer_style2_input,

      G_N_ELEMENTS (tmplayer_style2_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_style3()

{

  static SubParseInputChunk tmplayer_style3_input[] = {

    {

          "0:00:10:This is the Earth at a time|when the dinosaurs roamed...\n"

          "0:00:13:\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "0:00:14:a lush and fertile planet.\n" "0:00:16:\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_style3";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_style3");

  test_tmplayer_do_test (tmplayer_style3_input,

      G_N_ELEMENTS (tmplayer_style3_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_style4()

{

  static SubParseInputChunk tmplayer_style4_input[] = {

    {

          "0:00:10=This is the Earth at a time|when the dinosaurs roamed...\n"

          "0:00:13=\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "0:00:14=a lush and fertile planet.\n" "0:00:16=\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_style4";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_style4");

  test_tmplayer_do_test (tmplayer_style4_input,

      G_N_ELEMENTS (tmplayer_style4_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_tmplayer_style4_with_bogus_lines()

{

  static SubParseInputChunk tmplayer_style4b_input[] = {

    {

          "0:00:10=This is the Earth at a time|when the dinosaurs roamed...\n"

          "# This is a bogus line with a comment and should just be skipped\n"

          "0:00:13=\n",

          10 * GST_SECOND, 13 * GST_SECOND,

        "This is the Earth at a time\nwhen the dinosaurs roamed..."}, {

          "0:00:14=a lush and fertile planet.\n"

          "                                                            \n"

          "0:00:16=\n",

          14 * GST_SECOND, 16 * GST_SECOND,

        "a lush and fertile planet."}

  };

	xmlfile = "subparse_test_tmplayer_style4_with_bogus_lines";

	std_log(LOG_FILENAME_LINE, "Test Started test_tmplayer_style4_with_bogus_lines");

  test_tmplayer_do_test (tmplayer_style4b_input,

      G_N_ELEMENTS (tmplayer_style4b_input));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_microdvd_with_italics()

{

  static SubParseInputChunk microdvd_italics[] = {

    {

          "{1}{1}25.000 movie info: XVID  608x256 25.0fps 699.0 MB|"

          "/SubEdit b.4060(http://subedit.com.pl)/\n"

          "{100}{200}/italics/|not italics\n",

          4 * GST_SECOND, 8 * GST_SECOND,

        "<span style=\"italic\">italics</span>\n" "<span>not italics</span>"}

  };

	xmlfile = "subparse_test_microdvd_with_italics";

	std_log(LOG_FILENAME_LINE, "Test Started test_microdvd_with_italics");

  test_microdvd_do_test (microdvd_italics, G_N_ELEMENTS (microdvd_italics));

std_log(LOG_FILENAME_LINE, "Test Successful");

create_xml(0);

}

void test_microdvd_with_fps()

{

  static SubParseInputChunk microdvd_input[] = {

    {

          "{1}{1}12.500\n{100}{200}- Hi, Eddie.|- Hiya, Scotty.\n",

          8 * GST_SECOND, 16 * GST_SECOND,