/*
    Copyright (C) 2012-2016 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  src/transcoder.cc
 *  @brief A class which takes a Film and some Options, then uses those to transcode the film.
 *
 *  A decoder is selected according to the content type, and the encoder can be specified
 *  as a parameter to the constructor.
 */

#include "transcoder.h"
#include "encoder.h"
#include "film.h"
#include "video_decoder.h"
#include "audio_decoder.h"
#include "player.h"
#include "job.h"
#include "writer.h"
#include "compose.hpp"
#include "referenced_reel_asset.h"
#include "subtitle_content.h"
#include "player_video.h"
#include <boost/signals2.hpp>
#include <boost/foreach.hpp>
#include <iostream>

#include "i18n.h"

using std::string;
using std::cout;
using std::list;
using boost::shared_ptr;
using boost::weak_ptr;
using boost::dynamic_pointer_cast;

/** Construct a transcoder.
 *  @param film Film that we are transcoding.
 *  @param job Job that this transcoder is being used in.
 */
Transcoder::Transcoder (shared_ptr<const Film> film, weak_ptr<Job> job)
	: _film (film)
	, _job (job)
	, _player (new Player (film, film->playlist ()))
	, _writer (new Writer (film, job))
	, _encoder (new Encoder (film, _writer))
	, _finishing (false)
	, _non_burnt_subtitles (false)
{
	_player->Video.connect (bind (&Transcoder::video, this, _1, _2));
	_player->Audio.connect (bind (&Transcoder::audio, this, _1, _2));
	_player->Subtitle.connect (bind (&Transcoder::subtitle, this, _1, _2));

	BOOST_FOREACH (shared_ptr<const Content> c, _film->content ()) {
		if (c->subtitle && c->subtitle->use() && !c->subtitle->burn()) {
			_non_burnt_subtitles = true;
		}
	}
}

void
Transcoder::go ()
{
	_writer->start ();
	_encoder->begin ();

	{
		shared_ptr<Job> job = _job.lock ();
		DCPOMATIC_ASSERT (job);
		job->sub (_("Encoding"));
	}

	if (_non_burnt_subtitles) {
		_writer->write (_player->get_subtitle_fonts ());
	}

	while (!_player->pass ()) {}

	BOOST_FOREACH (ReferencedReelAsset i, _player->get_reel_assets ()) {
		_writer->write (i);
	}

	_finishing = true;
	_encoder->end ();
	_writer->finish ();
}

void
Transcoder::video (shared_ptr<PlayerVideo> data, DCPTime time)
{
	if (!_film->three_d() && data->eyes() == EYES_LEFT) {
		/* Use left-eye images for both eyes */
		data->set_eyes (EYES_BOTH);
	}

	_encoder->encode (data, time);
}

void
Transcoder::audio (shared_ptr<AudioBuffers> data, DCPTime time)
{
	_writer->write (data);

	shared_ptr<Job> job = _job.lock ();
	DCPOMATIC_ASSERT (job);
	job->set_progress (float(time.get()) / _film->length().get());
}

void
Transcoder::subtitle (PlayerSubtitles data, DCPTimePeriod period)
{
	if (_non_burnt_subtitles) {
		_writer->write (data, period);
	}
}

float
Transcoder::current_encoding_rate () const
{
	return _encoder->current_encoding_rate ();
}

int
Transcoder::video_frames_enqueued () const
{
	return _encoder->video_frames_enqueued ();
}