/*
    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <cmath>
#include <boost/filesystem.hpp>
#include "KM_prng.h"
#include "dcp.h"
#include "util.h"
#include "metadata.h"
#include "types.h"
#include "exceptions.h"
#include "subtitle_asset.h"
#include "picture_asset.h"
#include "sound_asset.h"
#include "reel.h"

#define BOOST_TEST_DYN_LINK
#define BOOST_TEST_MODULE libdcp_test
#include <boost/test/unit_test.hpp>

using namespace std;
using namespace boost;

string
j2c (int)
{
	return "test/data/32x32_red_square.j2c";
}

string
wav (libdcp::Channel)
{
	return "test/data/1s_24-bit_48k_silence.wav";
}
		

BOOST_AUTO_TEST_CASE (dcp_test)
{
	Kumu::libdcp_test = true;
	
	libdcp::Metadata* t = libdcp::Metadata::instance ();
	t->issuer = "OpenDCP 0.0.25";
	t->creator = "OpenDCP 0.0.25";
	t->company_name = "OpenDCP";
	t->product_name = "OpenDCP";
	t->product_version = "0.0.25";
	t->issue_date = "2012-07-17T04:45:18+00:00";
	filesystem::remove_all ("build/test/foo");
	filesystem::create_directories ("build/test/foo");
	libdcp::DCP d ("build/test/foo", "A Test DCP", libdcp::FEATURE, 24, 24);

	shared_ptr<libdcp::PictureAsset> mp (new libdcp::PictureAsset (
						    sigc::ptr_fun (&j2c),
						    "build/test/foo",
						    "video.mxf",
						    &d.Progress,
						    24,
						    24,
						    32,
						    32
						    ));

	shared_ptr<libdcp::SoundAsset> ms (new libdcp::SoundAsset (
						  sigc::ptr_fun (&wav),
						  "build/test/foo",
						  "audio.mxf",
						  &(d.Progress),
						  24,
						  24,
						  2
						  ));

	d.add_reel (shared_ptr<libdcp::Reel> (new libdcp::Reel (mp, ms, shared_ptr<libdcp::SubtitleAsset> ())));

	d.write_xml ();
}

BOOST_AUTO_TEST_CASE (error_test)
{
	libdcp::DCP d ("build/test/bar", "A Test DCP", libdcp::TEST, 24, 24);
	vector<string> p;
	p.push_back ("frobozz");

	BOOST_CHECK_THROW (new libdcp::PictureAsset (p, "build/test/bar", "video.mxf", &d.Progress, 24, 24, 32, 32), libdcp::FileError);
	BOOST_CHECK_THROW (new libdcp::SoundAsset (p, "build/test/bar", "audio.mxf", &d.Progress, 24, 24), libdcp::FileError);
}

BOOST_AUTO_TEST_CASE (read_dcp)
{
	libdcp::DCP d ("test/ref/DCP");

	BOOST_CHECK_EQUAL (d.name(), "A Test DCP");
	BOOST_CHECK_EQUAL (d.content_kind(), libdcp::FEATURE);
	BOOST_CHECK_EQUAL (d.frames_per_second(), 24);
	BOOST_CHECK_EQUAL (d.length(), 24);
}
	
BOOST_AUTO_TEST_CASE (subtitles)
{
	libdcp::SubtitleAsset subs ("test/ref", "subs.xml");

	BOOST_CHECK_EQUAL (subs.language(), "French");

	list<shared_ptr<libdcp::Subtitle> > s = subs.subtitles_at (libdcp::Time (0, 0, 6, 1));
	BOOST_CHECK_EQUAL (s.size(), 1);
	BOOST_CHECK_EQUAL (s.front()->text(), "My jacket was Idi Amin's");
	BOOST_CHECK_EQUAL (s.front()->v_position(), 15);
	BOOST_CHECK_EQUAL (s.front()->in(), libdcp::Time (0, 0, 5, 198));
	BOOST_CHECK_EQUAL (s.front()->out(), libdcp::Time (0, 0, 7, 115));
	BOOST_CHECK_EQUAL (s.front()->font(), "Arial");
	BOOST_CHECK_EQUAL (s.front()->italic(), false);
	BOOST_CHECK_EQUAL (s.front()->color(), libdcp::Color(255, 255, 255));
	BOOST_CHECK_EQUAL (s.front()->size_in_pixels(1080), 53);
	BOOST_CHECK_EQUAL (s.front()->effect(), libdcp::BORDER);
	BOOST_CHECK_EQUAL (s.front()->effect_color(), libdcp::Color(0, 0, 0));
	BOOST_CHECK_EQUAL (s.front()->v_align(), libdcp::BOTTOM);

	s = subs.subtitles_at (libdcp::Time (0, 0, 7, 190));
	BOOST_CHECK_EQUAL (s.size(), 2);
	BOOST_CHECK_EQUAL (s.front()->text(), "My corset was H.M. The Queen's");
	BOOST_CHECK_EQUAL (s.front()->v_position(), 21);
	BOOST_CHECK_EQUAL (s.front()->in(), libdcp::Time (0, 0, 7, 177));
	BOOST_CHECK_EQUAL (s.front()->out(), libdcp::Time (0, 0, 11, 31));
	BOOST_CHECK_EQUAL (s.front()->font(), "Arial");
	BOOST_CHECK_EQUAL (s.front()->italic(), true);
	BOOST_CHECK_EQUAL (s.front()->size_in_pixels(1080), 53);
	BOOST_CHECK_EQUAL (s.front()->effect(), libdcp::BORDER);
	BOOST_CHECK_EQUAL (s.front()->effect_color(), libdcp::Color(0, 0, 0));
	BOOST_CHECK_EQUAL (s.front()->v_align(), libdcp::BOTTOM);
	BOOST_CHECK_EQUAL (s.back()->text(), "My large wonderbra");
	BOOST_CHECK_EQUAL (s.back()->v_position(), 15);
	BOOST_CHECK_EQUAL (s.back()->in(), libdcp::Time (0, 0, 7, 177));
	BOOST_CHECK_EQUAL (s.back()->out(), libdcp::Time (0, 0, 11, 31));
	BOOST_CHECK_EQUAL (s.back()->font(), "Arial");
	BOOST_CHECK_EQUAL (s.back()->italic(), true);
	BOOST_CHECK_EQUAL (s.back()->color(), libdcp::Color(255, 255, 255));
	BOOST_CHECK_EQUAL (s.back()->size_in_pixels(1080), 53);
	BOOST_CHECK_EQUAL (s.back()->effect(), libdcp::BORDER);
	BOOST_CHECK_EQUAL (s.back()->effect_color(), libdcp::Color(0, 0, 0));
	BOOST_CHECK_EQUAL (s.back()->v_align(), libdcp::BOTTOM);
	
	s = subs.subtitles_at (libdcp::Time (0, 0, 11, 95));
	BOOST_CHECK_EQUAL (s.size(), 1);
	BOOST_CHECK_EQUAL (s.front()->text(), "Once belonged to the Shah");
	BOOST_CHECK_EQUAL (s.front()->v_position(), 15);
	BOOST_CHECK_EQUAL (s.front()->in(), libdcp::Time (0, 0, 11, 94));
	BOOST_CHECK_EQUAL (s.front()->out(), libdcp::Time (0, 0, 13, 63));
	BOOST_CHECK_EQUAL (s.front()->font(), "Arial");
	BOOST_CHECK_EQUAL (s.front()->italic(), false);
	BOOST_CHECK_EQUAL (s.front()->color(), libdcp::Color(255, 255, 255));
	BOOST_CHECK_EQUAL (s.front()->size_in_pixels(1080), 53);
	BOOST_CHECK_EQUAL (s.front()->effect(), libdcp::BORDER);
	BOOST_CHECK_EQUAL (s.front()->effect_color(), libdcp::Color(0, 0, 0));
	BOOST_CHECK_EQUAL (s.front()->v_align(), libdcp::BOTTOM);

	s = subs.subtitles_at (libdcp::Time (0, 0, 14, 42));
	BOOST_CHECK_EQUAL (s.size(), 1);
	BOOST_CHECK_EQUAL (s.front()->text(), "And these are Roy Hattersley's jeans");
	BOOST_CHECK_EQUAL (s.front()->v_position(), 15);
	BOOST_CHECK_EQUAL (s.front()->in(), libdcp::Time (0, 0, 13, 104));
	BOOST_CHECK_EQUAL (s.front()->out(), libdcp::Time (0, 0, 15, 177));
	BOOST_CHECK_EQUAL (s.front()->font(), "Arial");
	BOOST_CHECK_EQUAL (s.front()->italic(), false);
	BOOST_CHECK_EQUAL (s.front()->color(), libdcp::Color(255, 255, 255));
	BOOST_CHECK_EQUAL (s.front()->size_in_pixels(1080), 53);
	BOOST_CHECK_EQUAL (s.front()->effect(), libdcp::BORDER);
	BOOST_CHECK_EQUAL (s.front()->effect_color(), libdcp::Color(0, 0, 0));
	BOOST_CHECK_EQUAL (s.front()->v_align(), libdcp::BOTTOM);
}

BOOST_AUTO_TEST_CASE (dcp_time)
{
	libdcp::Time t (977143, 24);

	BOOST_CHECK_EQUAL (t.t, 73);
	BOOST_CHECK_EQUAL (t.s, 34);
	BOOST_CHECK_EQUAL (t.m, 18);
	BOOST_CHECK_EQUAL (t.h, 11);

	libdcp::Time a (3, 2, 3, 4);
	libdcp::Time b (2, 3, 4, 5);

	libdcp::Time r = a - b;
	BOOST_CHECK_EQUAL (r.h, 0);
	BOOST_CHECK_EQUAL (r.m, 58);
	BOOST_CHECK_EQUAL (r.s, 58);
	BOOST_CHECK_EQUAL (r.t, 249);

	a = libdcp::Time (1, 58, 56, 240);
	b = libdcp::Time (1, 7, 12, 120);
	r = a + b;
	BOOST_CHECK_EQUAL (r.h, 3);
	BOOST_CHECK_EQUAL (r.m, 6);
	BOOST_CHECK_EQUAL (r.s, 9);
	BOOST_CHECK_EQUAL (r.t, 110);

	a = libdcp::Time (24, 12, 6, 3);
	b = libdcp::Time (16, 8, 4, 2);
	BOOST_CHECK_EQUAL (a / b, 1.5);
}

BOOST_AUTO_TEST_CASE (color)
{
	libdcp::Color c ("FFFF0000");

	BOOST_CHECK_EQUAL (c.r, 255);
	BOOST_CHECK_EQUAL (c.g, 0);
	BOOST_CHECK_EQUAL (c.b, 0);

	c = libdcp::Color ("FF00FF00");

	BOOST_CHECK_EQUAL (c.r, 0);
	BOOST_CHECK_EQUAL (c.g, 255);
	BOOST_CHECK_EQUAL (c.b, 0);

	c = libdcp::Color ("FF0000FF");

	BOOST_CHECK_EQUAL (c.r, 0);
	BOOST_CHECK_EQUAL (c.g, 0);
	BOOST_CHECK_EQUAL (c.b, 255);
}