/*
    Copyright (C) 2012-2018 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/>.

*/

#include "filter_dialog.h"
#include "video_panel.h"
#include "wx_util.h"
#include "content_colour_conversion_dialog.h"
#include "content_widget.h"
#include "content_panel.h"
#include "static_text.h"
#include "check_box.h"
#include "dcpomatic_button.h"
#include "lib/filter.h"
#include "lib/ffmpeg_content.h"
#include "lib/colour_conversion.h"
#include "lib/config.h"
#include "lib/util.h"
#include "lib/ratio.h"
#include "lib/frame_rate_change.h"
#include "lib/dcp_content.h"
#include "lib/video_content.h"
#include <wx/spinctrl.h>
#include <boost/foreach.hpp>
#include <boost/unordered_set.hpp>
#include <boost/functional/hash.hpp>
#include <set>
#include <iostream>

using std::vector;
using std::string;
using std::pair;
using std::cout;
using std::list;
using std::set;
using boost::shared_ptr;
using boost::dynamic_pointer_cast;
using boost::bind;
using boost::optional;
#if BOOST_VERSION >= 106100
using namespace boost::placeholders;
#endif

static VideoContentScale
index_to_scale (int n)
{
	vector<VideoContentScale> scales = VideoContentScale::all ();
	DCPOMATIC_ASSERT (n >= 0);
	DCPOMATIC_ASSERT (n < int (scales.size ()));
	return scales[n];
}

static int
scale_to_index (VideoContentScale scale)
{
	vector<VideoContentScale> scales = VideoContentScale::all ();
	for (size_t i = 0; i < scales.size(); ++i) {
		if (scales[i] == scale) {
			return i;
		}
	}

	DCPOMATIC_ASSERT (false);
}

VideoPanel::VideoPanel (ContentPanel* p)
	: ContentSubPanel (p, _("Video"))
{
	_reference = new CheckBox (this, _("Use this DCP's video as OV and make VF"));
	_reference_note = new StaticText (this, wxT(""));
	_reference_note->Wrap (200);
	wxFont font = _reference_note->GetFont();
	font.SetStyle(wxFONTSTYLE_ITALIC);
	font.SetPointSize(font.GetPointSize() - 1);
	_reference_note->SetFont(font);

	_type_label = create_label (this, _("Type"), true);
	_frame_type = new ContentChoice<VideoContent, VideoFrameType> (
		this,
		new wxChoice (this, wxID_ANY),
		VideoContentProperty::FRAME_TYPE,
		&Content::video,
		boost::mem_fn (&VideoContent::frame_type),
		boost::mem_fn (&VideoContent::set_frame_type),
		&caster<int, VideoFrameType>,
		&caster<VideoFrameType, int>
		);

	_crop_label = create_label (this, _("Crop"), true);

#ifdef __WXGTK3__
	int const crop_width = 128;
#else
	int const crop_width = 56;
#endif

	_left_crop_label = create_label (this, _("Left"), true);
	_left_crop = new ContentSpinCtrl<VideoContent> (
		this,
		new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (crop_width, -1)),
		VideoContentProperty::CROP,
		&Content::video,
		boost::mem_fn (&VideoContent::left_crop),
		boost::mem_fn (&VideoContent::set_left_crop)
		);

	_right_crop_label = create_label (this, _("Right"), true);
	_right_crop = new ContentSpinCtrl<VideoContent> (
		this,
		new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (crop_width, -1)),
		VideoContentProperty::CROP,
		&Content::video,
		boost::mem_fn (&VideoContent::right_crop),
		boost::mem_fn (&VideoContent::set_right_crop)
		);

	_top_crop_label = create_label (this, _("Top"), true);
	_top_crop = new ContentSpinCtrl<VideoContent> (
		this,
		new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (crop_width, -1)),
		VideoContentProperty::CROP,
		&Content::video,
		boost::mem_fn (&VideoContent::top_crop),
		boost::mem_fn (&VideoContent::set_top_crop)
		);

	_bottom_crop_label = create_label (this, _("Bottom"), true);
	_bottom_crop = new ContentSpinCtrl<VideoContent> (
		this,
		new wxSpinCtrl (this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize (crop_width, -1)),
		VideoContentProperty::CROP,
		&Content::video,
		boost::mem_fn (&VideoContent::bottom_crop),
		boost::mem_fn (&VideoContent::set_bottom_crop)
		);

	_fade_in_label = create_label (this, _("Fade in"), true);
	_fade_in = new Timecode<ContentTime> (this);

	_fade_out_label = create_label (this, _("Fade out"), true);
	_fade_out = new Timecode<ContentTime> (this);

	_scale_to_label = create_label (this, _("Scale to"), true);
	_scale = new ContentChoice<VideoContent, VideoContentScale> (
		this,
		new wxChoice (this, wxID_ANY),
		VideoContentProperty::SCALE,
		&Content::video,
		boost::mem_fn (&VideoContent::scale),
		boost::mem_fn (&VideoContent::set_scale),
		&index_to_scale,
		&scale_to_index
		);

	wxClientDC dc (this);
	wxSize size = dc.GetTextExtent (wxT ("A quite long name"));
#ifdef __WXGTK3__
	size.SetWidth (size.GetWidth() + 64);
#endif
	size.SetHeight (-1);

	_filters_label = create_label (this, _("Filters"), true);
	_filters = new StaticText (this, _("None"), wxDefaultPosition, size);
	_filters_button = new Button (this, _("Edit..."));

	_colour_conversion_label = create_label (this, _("Colour conversion"), true);
	_colour_conversion = new wxChoice (this, wxID_ANY, wxDefaultPosition, size);
	_colour_conversion->Append (_("None"));
	BOOST_FOREACH (PresetColourConversion const & i, PresetColourConversion::all()) {
		_colour_conversion->Append (std_to_wx (i.name));
	}

	/// TRANSLATORS: translate the word "Custom" here; do not include the "Colour|" prefix
	_colour_conversion->Append (S_("Colour|Custom"));
	_edit_colour_conversion_button = new Button (this, _("Edit..."));

	_description = new StaticText (this, wxT ("\n \n \n \n \n"), wxDefaultPosition, wxDefaultSize);
	_description->SetFont(font);

	_left_crop->wrapped()->SetRange (0, 4096);
	_top_crop->wrapped()->SetRange (0, 4096);
	_right_crop->wrapped()->SetRange (0, 4096);
	_bottom_crop->wrapped()->SetRange (0, 4096);

	_scale->wrapped()->Clear ();
	BOOST_FOREACH (VideoContentScale const & i, VideoContentScale::all ()) {
		_scale->wrapped()->Append (std_to_wx (i.name ()));
	}

	_frame_type->wrapped()->Append (_("2D"));
	_frame_type->wrapped()->Append (_("3D"));
	_frame_type->wrapped()->Append (_("3D left/right"));
	_frame_type->wrapped()->Append (_("3D top/bottom"));
	_frame_type->wrapped()->Append (_("3D alternate"));
	_frame_type->wrapped()->Append (_("3D left only"));
	_frame_type->wrapped()->Append (_("3D right only"));

	content_selection_changed ();

	_fade_in->Changed.connect (boost::bind (&VideoPanel::fade_in_changed, this));
	_fade_out->Changed.connect (boost::bind (&VideoPanel::fade_out_changed, this));

	_reference->Bind                     (wxEVT_CHECKBOX, boost::bind (&VideoPanel::reference_clicked, this));
	_filters_button->Bind                (wxEVT_BUTTON,   boost::bind (&VideoPanel::edit_filters_clicked, this));
	_colour_conversion->Bind             (wxEVT_CHOICE,   boost::bind (&VideoPanel::colour_conversion_changed, this));
	_edit_colour_conversion_button->Bind (wxEVT_BUTTON,   boost::bind (&VideoPanel::edit_colour_conversion_clicked, this));

	add_to_grid ();
}

void
VideoPanel::add_to_grid ()
{
	bool const full = Config::instance()->interface_complexity() == Config::INTERFACE_FULL;

	int r = 0;

	_reference->Show (full);
	_reference_note->Show (full);

	if (full) {
		wxBoxSizer* reference_sizer = new wxBoxSizer (wxVERTICAL);
		reference_sizer->Add (_reference, 0);
		reference_sizer->Add (_reference_note, 0);
		_grid->Add (reference_sizer, wxGBPosition(r, 0), wxGBSpan(1, 3));
		++r;
	}

	add_label_to_sizer (_grid, _type_label, true, wxGBPosition(r, 0));
	_frame_type->add (_grid, wxGBPosition(r, 1), wxGBSpan(1, 2));
	++r;

	int cr = 0;
	wxGridBagSizer* crop = new wxGridBagSizer (DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP);
	add_label_to_sizer (crop, _left_crop_label, true, wxGBPosition (cr, 0));
	_left_crop->add (crop, wxGBPosition (cr, 1));
#ifdef __WXGTK3__
	++cr;
	add_label_to_sizer (crop, _right_crop_label, true, wxGBPosition (cr, 0));
	_right_crop->add (crop, wxGBPosition (cr, 1));
#else
	add_label_to_sizer (crop, _right_crop_label, true, wxGBPosition (cr, 2));
	_right_crop->add (crop, wxGBPosition (cr, 3));
#endif
	++cr;
	add_label_to_sizer (crop, _top_crop_label, true, wxGBPosition (cr, 0));
	_top_crop->add (crop, wxGBPosition (cr, 1));
#ifdef __WXGTK3__
	++cr;
	add_label_to_sizer (crop, _bottom_crop_label, true, wxGBPosition (cr, 0));
	_bottom_crop->add (crop, wxGBPosition (cr, 1));
#else
	add_label_to_sizer (crop, _bottom_crop_label, true, wxGBPosition (cr, 2));
	_bottom_crop->add (crop, wxGBPosition (cr, 3));
#endif
	add_label_to_sizer (_grid, _crop_label, true, wxGBPosition(r, 0));
	_grid->Add (crop, wxGBPosition(r, 1));
	++r;

	_scale_to_label->Show (full);
	_scale->show (full);
	_filters_label->Show (full);
	_filters->Show (full);
	_filters_button->Show (full);
	_colour_conversion_label->Show (full);
	_colour_conversion->Show (full);
	_edit_colour_conversion_button->Show (full);

	add_label_to_sizer (_grid, _fade_in_label, true, wxGBPosition (r, 0));
	_grid->Add (_fade_in, wxGBPosition (r, 1), wxGBSpan (1, 3));
	++r;

	add_label_to_sizer (_grid, _fade_out_label, true, wxGBPosition (r, 0));
	_grid->Add (_fade_out, wxGBPosition (r, 1), wxGBSpan (1, 3));
	++r;

	if (full) {
		add_label_to_sizer (_grid, _scale_to_label, true, wxGBPosition (r, 0));
		_scale->add (_grid, wxGBPosition (r, 1), wxGBSpan (1, 2));
		++r;

		add_label_to_sizer (_grid, _filters_label, true, wxGBPosition (r, 0));
		{
			wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
			s->Add (_filters, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM | wxRIGHT, 6);
			s->Add (_filters_button, 0, wxALIGN_CENTER_VERTICAL);
			_grid->Add (s, wxGBPosition (r, 1), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
		}
		++r;

		add_label_to_sizer (_grid, _colour_conversion_label, true, wxGBPosition(r, 0));
		{
			wxSizer* s = new wxBoxSizer (wxHORIZONTAL);
			s->Add (_colour_conversion, 1, wxALIGN_CENTER_VERTICAL | wxTOP | wxBOTTOM | wxRIGHT, 6);
			s->Add (_edit_colour_conversion_button, 0, wxALIGN_CENTER_VERTICAL);
			_grid->Add (s, wxGBPosition (r, 1), wxDefaultSpan, wxALIGN_CENTER_VERTICAL);
		}
		++r;
	}

	_grid->Add (_description, wxGBPosition (r, 0), wxGBSpan (1, 4), wxEXPAND | wxALIGN_CENTER_VERTICAL, 6);
	++r;
}


void
VideoPanel::film_changed (Film::Property property)
{
	switch (property) {
	case Film::VIDEO_FRAME_RATE:
	case Film::CONTAINER:
	case Film::RESOLUTION:
		setup_description ();
		setup_sensitivity ();
		break;
	case Film::REEL_TYPE:
	case Film::INTEROP:
		setup_sensitivity ();
		break;
	default:
		break;
	}
}

std::size_t
hash_value (boost::optional<ColourConversion> const & c)
{
	boost::hash<string> hasher;
	if (!c) {
		return hasher ("none");
	}
	return hasher (c->identifier());
}


void
VideoPanel::film_content_changed (int property)
{
	ContentList vc = _parent->selected_video ();
	shared_ptr<Content> vcs;
	shared_ptr<FFmpegContent> fcs;
	if (!vc.empty ()) {
		vcs = vc.front ();
		fcs = dynamic_pointer_cast<FFmpegContent> (vcs);
	}

	if (property == ContentProperty::VIDEO_FRAME_RATE ||
	    property == VideoContentProperty::FRAME_TYPE ||
	    property == VideoContentProperty::CROP ||
	    property == VideoContentProperty::SCALE) {
		setup_description ();
	} else if (property == VideoContentProperty::COLOUR_CONVERSION) {
		boost::unordered_set<optional<ColourConversion> > check;
		BOOST_FOREACH (shared_ptr<const Content> i, vc) {
			check.insert (i->video->colour_conversion());
		}

		/* Remove any "Many" entry that we might have added previously.  There should
		 * be entries for each preset plus one for "None" and one for "Custom".
		 */
		vector<PresetColourConversion> cc = PresetColourConversion::all ();
		if (_colour_conversion->GetCount() > cc.size() + 2) {
			_colour_conversion->Delete (_colour_conversion->GetCount() - 1);
		}

		if (check.size() == 1) {
			if (vcs && vcs->video->colour_conversion ()) {
				optional<size_t> preset = vcs->video->colour_conversion().get().preset ();
				if (preset) {
					checked_set (_colour_conversion, preset.get() + 1);
				} else {
					checked_set (_colour_conversion, cc.size() + 1);
				}
			} else {
				checked_set (_colour_conversion, 0);
			}
		} else if (check.size() > 1) {
			/* Add a "many" entry and select it as an indication that multiple different
			 * colour conversions are present in the selection.
			 */
			_colour_conversion->Append (_("Many"));
			checked_set (_colour_conversion, _colour_conversion->GetCount() - 1);
		}

		setup_sensitivity ();

	} else if (property == FFmpegContentProperty::FILTERS) {
		if (fcs) {
			string p = Filter::ffmpeg_string (fcs->filters ());
			if (p.empty ()) {
				checked_set (_filters, _("None"));
			} else {
				if (p.length() > 25) {
					p = p.substr (0, 25) + "...";
				}
				checked_set (_filters, p);
			}
		}
	} else if (property == VideoContentProperty::FADE_IN) {
		set<Frame> check;
		BOOST_FOREACH (shared_ptr<const Content> i, vc) {
			check.insert (i->video->fade_in ());
		}

		if (check.size() == 1) {
			_fade_in->set (
				ContentTime::from_frames (vc.front()->video->fade_in(), vc.front()->active_video_frame_rate(_parent->film())),
				vc.front()->active_video_frame_rate(_parent->film())
				);
		} else {
			_fade_in->clear ();
		}
	} else if (property == VideoContentProperty::FADE_OUT) {
		set<Frame> check;
		BOOST_FOREACH (shared_ptr<const Content> i, vc) {
			check.insert (i->video->fade_out ());
		}

		if (check.size() == 1) {
			_fade_out->set (
				ContentTime::from_frames (vc.front()->video->fade_out(), vc.front()->active_video_frame_rate(_parent->film())),
				vc.front()->active_video_frame_rate(_parent->film())
				);
		} else {
			_fade_out->clear ();
		}
	} else if (property == DCPContentProperty::REFERENCE_VIDEO) {
		if (vc.size() == 1) {
			shared_ptr<DCPContent> dcp = dynamic_pointer_cast<DCPContent> (vc.front ());
			checked_set (_reference, dcp ? dcp->reference_video () : false);
		} else {
			checked_set (_reference, false);
		}

		setup_sensitivity ();
	}
}

/** Called when the `Edit filters' button has been clicked */
void
VideoPanel::edit_filters_clicked ()
{
	FFmpegContentList c = _parent->selected_ffmpeg ();
	if (c.size() != 1) {
		return;
	}

	FilterDialog* d = new FilterDialog (this, c.front()->filters());
	d->ActiveChanged.connect (bind (&FFmpegContent::set_filters, c.front(), _1));
	d->ShowModal ();
	d->Destroy ();
}

void
VideoPanel::setup_description ()
{
	ContentList vc = _parent->selected_video ();
	if (vc.empty ()) {
		checked_set (_description, wxT (""));
		return;
	} else if (vc.size() > 1) {
		checked_set (_description, _("Multiple content selected"));
		return;
	}

	string d = vc.front()->video->processing_description (_parent->film());
	size_t lines = count (d.begin(), d.end(), '\n');

	for (int i = lines; i < 6; ++i) {
		d += "\n ";
	}

	checked_set (_description, d);
	_sizer->Layout ();
}

void
VideoPanel::colour_conversion_changed ()
{
	ContentList vc = _parent->selected_video ();

	int const s = _colour_conversion->GetSelection ();
	vector<PresetColourConversion> all = PresetColourConversion::all ();

	if (s == int(all.size() + 1)) {
		edit_colour_conversion_clicked ();
	} else {
		BOOST_FOREACH (shared_ptr<Content> i, _parent->selected_video()) {
			if (s == 0) {
				i->video->unset_colour_conversion ();
			} else if (s != int(all.size() + 2)) {
				i->video->set_colour_conversion (all[s - 1].conversion);
			}
		}
	}
}

void
VideoPanel::edit_colour_conversion_clicked ()
{
	ContentList vc = _parent->selected_video ();

	ContentColourConversionDialog* d = new ContentColourConversionDialog (this, vc.front()->video->yuv ());
	d->set (vc.front()->video->colour_conversion().get_value_or (PresetColourConversion::all().front().conversion));
	if (d->ShowModal() == wxID_OK) {
		BOOST_FOREACH (shared_ptr<Content> i, vc) {
			i->video->set_colour_conversion (d->get ());
		}
	} else {
		/* Reset the colour conversion choice */
		film_content_changed (VideoContentProperty::COLOUR_CONVERSION);
	}
	d->Destroy ();
}

void
VideoPanel::content_selection_changed ()
{
	ContentList video_sel = _parent->selected_video ();

	_frame_type->set_content (video_sel);
	_left_crop->set_content (video_sel);
	_right_crop->set_content (video_sel);
	_top_crop->set_content (video_sel);
	_bottom_crop->set_content (video_sel);
	_scale->set_content (video_sel);

	film_content_changed (ContentProperty::VIDEO_FRAME_RATE);
	film_content_changed (VideoContentProperty::CROP);
	film_content_changed (VideoContentProperty::COLOUR_CONVERSION);
	film_content_changed (VideoContentProperty::FADE_IN);
	film_content_changed (VideoContentProperty::FADE_OUT);
	film_content_changed (FFmpegContentProperty::FILTERS);
	film_content_changed (DCPContentProperty::REFERENCE_VIDEO);

	setup_sensitivity ();
}

void
VideoPanel::setup_sensitivity ()
{
	ContentList sel = _parent->selected ();

	shared_ptr<DCPContent> dcp;
	if (sel.size() == 1) {
		dcp = dynamic_pointer_cast<DCPContent> (sel.front ());
	}

	string why_not;
	bool const can_reference = dcp && dcp->can_reference_video (_parent->film(), why_not);
	setup_refer_button (_reference, _reference_note, dcp, can_reference, why_not);

	if (_reference->GetValue ()) {
		_frame_type->wrapped()->Enable (false);
		_left_crop->wrapped()->Enable (false);
		_right_crop->wrapped()->Enable (false);
		_top_crop->wrapped()->Enable (false);
		_bottom_crop->wrapped()->Enable (false);
		_fade_in->Enable (false);
		_fade_out->Enable (false);
		_scale->wrapped()->Enable (false);
		_description->Enable (false);
		_filters->Enable (false);
		_filters_button->Enable (false);
		_colour_conversion->Enable (false);
	} else {
		ContentList video_sel = _parent->selected_video ();
		FFmpegContentList ffmpeg_sel = _parent->selected_ffmpeg ();
		bool const single = video_sel.size() == 1;

		_frame_type->wrapped()->Enable (true);
		_left_crop->wrapped()->Enable (true);
		_right_crop->wrapped()->Enable (true);
		_top_crop->wrapped()->Enable (true);
		_bottom_crop->wrapped()->Enable (true);
		_fade_in->Enable (!video_sel.empty ());
		_fade_out->Enable (!video_sel.empty ());
		_scale->wrapped()->Enable (true);
		_description->Enable (true);
		_filters->Enable (true);
		_filters_button->Enable (single && !ffmpeg_sel.empty ());
		_colour_conversion->Enable (!video_sel.empty());
	}

	ContentList vc = _parent->selected_video ();
	shared_ptr<Content> vcs;
	if (!vc.empty ()) {
		vcs = vc.front ();
	}

	if (vcs && vcs->video->colour_conversion ()) {
		_edit_colour_conversion_button->Enable (!vcs->video->colour_conversion().get().preset());
	} else {
		_edit_colour_conversion_button->Enable (false);
	}
}

void
VideoPanel::fade_in_changed ()
{
	BOOST_FOREACH (shared_ptr<Content> i, _parent->selected_video ()) {
		double const vfr = i->active_video_frame_rate (_parent->film());
		i->video->set_fade_in (_fade_in->get(vfr).frames_round(vfr));
	}
}

void
VideoPanel::fade_out_changed ()
{
	BOOST_FOREACH (shared_ptr<Content> i, _parent->selected_video ()) {
		double const vfr = i->active_video_frame_rate (_parent->film());
		i->video->set_fade_out (_fade_out->get(vfr).frames_round(vfr));
	}
}

void
VideoPanel::reference_clicked ()
{
	ContentList c = _parent->selected ();
	if (c.size() != 1) {
		return;
	}

	shared_ptr<DCPContent> d = dynamic_pointer_cast<DCPContent> (c.front ());
	if (!d) {
		return;
	}

	d->set_reference_video (_reference->GetValue ());
}