/*
    Copyright (C) 2016-2021 Carl Hetherington <cth@carlh.net>

    This file is part of DCP-o-matic.

    DCP-o-matic 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.

    DCP-o-matic 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 DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.

*/


/** @file  test/file_naming_test.cc
 *  @brief Test how files in DCPs are named.
 *  @ingroup feature
 */


#include "test.h"
#include "lib/config.h"
#include "lib/film.h"
#include "lib/ffmpeg_content.h"
#include "lib/dcp_content_type.h"
#include "lib/video_content.h"
#ifdef DCPOMATIC_WINDOWS
#include <boost/locale.hpp>
#endif
#include <boost/test/unit_test.hpp>
#include <boost/regex.hpp>


using std::string;
using std::shared_ptr;
using std::make_shared;


class Keep
{
public:
	Keep ()
	{
		_format = Config::instance()->dcp_asset_filename_format ();
	}

	~Keep ()
	{
		Config::instance()->set_dcp_asset_filename_format (_format);
	}

private:
	dcp::NameFormat _format;
};


BOOST_AUTO_TEST_CASE (file_naming_test)
{
	Keep k;
	Config::instance()->set_dcp_asset_filename_format (dcp::NameFormat("%c"));

	auto film = new_test_film ("file_naming_test");
	film->set_name ("file_naming_test");
	film->set_video_frame_rate (24);
	film->set_dcp_content_type (DCPContentType::from_isdcf_name ("FTR"));
	auto r = make_shared<FFmpegContent>("test/data/flat_red.png");
	film->examine_and_add_content (r);
	auto g = make_shared<FFmpegContent>("test/data/flat_green.png");
	film->examine_and_add_content (g);
	auto b = make_shared<FFmpegContent>("test/data/flat_blue.png");
	film->examine_and_add_content (b);
	BOOST_REQUIRE (!wait_for_jobs());

	r->set_position (film, dcpomatic::DCPTime::from_seconds(0));
	r->set_video_frame_rate (24);
	r->video->set_length (24);
	g->set_position (film, dcpomatic::DCPTime::from_seconds(1));
	g->set_video_frame_rate (24);
	g->video->set_length (24);
	b->set_position (film, dcpomatic::DCPTime::from_seconds(2));
	b->set_video_frame_rate (24);
	b->video->set_length (24);

	film->set_reel_type (ReelType::BY_VIDEO_CONTENT);
	film->write_metadata ();
	make_and_verify_dcp (
		film,
		{
			dcp::VerificationNote::Code::MISSING_FFMC_IN_FEATURE,
			dcp::VerificationNote::Code::MISSING_FFEC_IN_FEATURE
		});

	int got[3] = { 0, 0, 0 };
	for (auto i: boost::filesystem::directory_iterator(film->file(film->dcp_name()))) {
		if (boost::regex_match(i.path().string(), boost::regex(".*flat_red\\.png_.*\\.mxf"))) {
			++got[0];
		} else if (boost::regex_match(i.path().string(), boost::regex(".*flat_green\\.png_.*\\.mxf"))) {
			++got[1];
		} else if (boost::regex_match(i.path().string(), boost::regex(".*flat_blue\\.png_.*\\.mxf"))) {
			++got[2];
		}
	}

	for (int i = 0; i < 3; ++i) {
		BOOST_CHECK (got[i] == 2);
	}
}


BOOST_AUTO_TEST_CASE (file_naming_test2)
{
	Keep k;
	Config::instance()->set_dcp_asset_filename_format (dcp::NameFormat ("%c"));

	auto film = new_test_film ("file_naming_test2");
	film->set_name ("file_naming_test2");
	film->set_dcp_content_type (DCPContentType::from_isdcf_name ("FTR"));

#ifdef DCPOMATIC_WINDOWS
	/* This is necessary so that the UTF8 string constant below gets converted properly */
	std::locale::global(boost::locale::generator().generate(""));
	boost::filesystem::path::imbue(std::locale());
#endif

	auto r = make_shared<FFmpegContent>("test/data/flät_red.png");
	film->examine_and_add_content (r);
	auto g = make_shared<FFmpegContent>("test/data/flat_green.png");
	film->examine_and_add_content (g);
	auto b = make_shared<FFmpegContent>("test/data/flat_blue.png");
	film->examine_and_add_content (b);
	BOOST_REQUIRE (!wait_for_jobs());

	r->set_position (film, dcpomatic::DCPTime::from_seconds(0));
	r->set_video_frame_rate (24);
	r->video->set_length (24);
	g->set_position (film, dcpomatic::DCPTime::from_seconds(1));
	g->set_video_frame_rate (24);
	g->video->set_length (24);
	b->set_position (film, dcpomatic::DCPTime::from_seconds(2));
	b->set_video_frame_rate (24);
	b->video->set_length (24);

	film->set_reel_type (ReelType::BY_VIDEO_CONTENT);
	make_and_verify_dcp (
		film,
		{
			dcp::VerificationNote::Code::MISSING_FFMC_IN_FEATURE,
			dcp::VerificationNote::Code::MISSING_FFEC_IN_FEATURE
		});

	int got[3] = { 0, 0, 0 };
	for (auto i: boost::filesystem::directory_iterator (film->file(film->dcp_name()))) {
		if (boost::regex_match(i.path().string(), boost::regex(".*flat_red\\.png_.*\\.mxf"))) {
			++got[0];
		} else if (boost::regex_match(i.path().string(), boost::regex(".*flat_green\\.png_.*\\.mxf"))) {
			++got[1];
		} else if (boost::regex_match(i.path().string(), boost::regex(".*flat_blue\\.png_.*\\.mxf"))) {
			++got[2];
		}
	}

	for (int i = 0; i < 3; ++i) {
		BOOST_CHECK (got[i] == 2);
	}
}