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

    This file is part of libdcp.

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

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

/* If you are using an installed libdcp, these #includes would need to be changed to
#include <dcp/dcp.h>
#include <dcp/picture_asset.h>
... etc. ...
*/

#include "dcp.h"
#include "cpl.h"
#include "reel.h"
#include "reel_picture_asset.h"
#include "mono_j2k_picture_frame.h"
#include "mono_j2k_picture_asset.h"
#include "mono_j2k_picture_asset_reader.h"
#include "stereo_j2k_picture_asset.h"
#include "sound_asset.h"
#include "text_asset.h"
#include "openjpeg_image.h"
#include "colour_conversion.h"
#include "rgb_xyz.h"
/* This DISABLE/ENABLE pair is just to ignore some warnings from Magick++.h; they
 * can be removed.
 */
LIBDCP_DISABLE_WARNINGS
#include <Magick++.h>
LIBDCP_ENABLE_WARNINGS

/** @file examples/read_dcp.cc
 *  @brief Shows how to read a DCP.
 */

int
main ()
{
	/* Unless libdcp has been installed, you need to pass a path containing the "tags" folder here */
	dcp::init(boost::filesystem::path("."));
	Magick::InitializeMagick(nullptr);

	/* Create a DCP, specifying where our existing data is */
	dcp::DCP dcp("/home/carl/Test_DCP");
	/* Read the DCP to find out about it */
	dcp.read();

	if (dcp.all_encrypted()) {
		std::cout << "DCP is encrypted.\n";
	} else if (dcp.any_encrypted()) {
		std::cout << "DCP is partially encrypted.\n";
	} else {
		std::cout << "DCP is not encrypted.\n";
	}

	std::cout << "DCP has " << dcp.cpls().size() << " CPLs.\n";
	auto assets = dcp.assets();
	std::cout << "DCP has " << assets.size() << " assets.\n";
	for (auto i: assets) {
		if (std::dynamic_pointer_cast<dcp::MonoJ2KPictureAsset>(i)) {
			std::cout << "2D picture\n";
		} else if (std::dynamic_pointer_cast<dcp::StereoJ2KPictureAsset>(i)) {
			std::cout << "3D picture\n";
		} else if (std::dynamic_pointer_cast<dcp::SoundAsset>(i)) {
			std::cout << "Sound\n";
		} else if (std::dynamic_pointer_cast<dcp::TextAsset>(i)) {
			std::cout << "Subtitle/caption\n";
		} else if (std::dynamic_pointer_cast<dcp::CPL>(i)) {
			std::cout << "CPL\n";
		}
		std::cout << "\t" << i->file()->filename().string() << "\n";
	}

	/* Take the first CPL */
	auto cpl = dcp.cpls().front();

	/* Get the picture asset in the first reel */
	auto picture_asset = std::dynamic_pointer_cast<dcp::MonoJ2KPictureAsset>(
		cpl->reels()[0]->main_picture()->asset()
		);

	/* Get a reader for it */
	auto picture_asset_reader = picture_asset->start_read();

	/* Get the 1000th frame of it */
	auto picture_frame_j2k = picture_asset_reader->get_frame(999);

	/* Get the frame as an XYZ image */
	auto picture_image_xyz = picture_frame_j2k->xyz_image();

	/* Convert to ARGB */
	std::vector<uint8_t> rgba(picture_image_xyz->size().width * picture_image_xyz->size().height * 4);
	dcp::xyz_to_rgba(picture_image_xyz, dcp::ColourConversion::srgb_to_xyz(), rgba.data(), picture_image_xyz->size().width * 4);

	Magick::Image image(picture_image_xyz->size().width, picture_image_xyz->size().height, "BGRA", Magick::CharPixel, rgba.data());
	image.write("frame.png");

	return 0;
}