X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Ffilm.cc;h=3f92100802ade784a839bef4a790a939fa547a2b;hb=7561918187d51eadcd59904a71c1eda30cc6ab31;hp=cc5de0e7985f09bdaf8c6f0fca7c32a1a1ecdfcf;hpb=563de0fe1f15ca087abe45c2d259b02fcc5c0f81;p=dcpomatic.git

diff --git a/src/lib/film.cc b/src/lib/film.cc
index cc5de0e79..3f9210080 100644
--- a/src/lib/film.cc
+++ b/src/lib/film.cc
@@ -40,7 +40,6 @@
 #include "ab_transcode_job.h"
 #include "transcode_job.h"
 #include "scp_dcp_job.h"
-#include "copy_from_dvd_job.h"
 #include "make_dcp_job.h"
 #include "log.h"
 #include "options.h"
@@ -52,6 +51,9 @@
 #include "check_hashes_job.h"
 #include "version.h"
 #include "ui_signaller.h"
+#include "video_decoder.h"
+#include "audio_decoder.h"
+#include "external_audio_decoder.h"
 
 using std::string;
 using std::stringstream;
@@ -64,11 +66,15 @@ using std::ofstream;
 using std::setfill;
 using std::min;
 using std::make_pair;
+using std::cout;
 using boost::shared_ptr;
 using boost::lexical_cast;
 using boost::to_upper_copy;
 using boost::ends_with;
 using boost::starts_with;
+using boost::optional;
+
+int const Film::state_version = 1;
 
 /** Construct a Film object in a given directory, reading any metadata
  *  file that exists in that directory.  An exception will be thrown if
@@ -79,24 +85,20 @@ using boost::starts_with;
  */
 
 Film::Film (string d, bool must_exist)
-	: _use_dci_name (false)
+	: _use_dci_name (true)
 	, _dcp_content_type (0)
 	, _format (0)
 	, _scaler (Scaler::from_id ("bicubic"))
 	, _dcp_trim_start (0)
 	, _dcp_trim_end (0)
 	, _dcp_ab (false)
-	, _use_source_audio (true)
-	, _audio_stream (-1)
+	, _use_content_audio (true)
 	, _audio_gain (0)
 	, _audio_delay (0)
 	, _still_duration (10)
-	, _subtitle_stream (-1)
 	, _with_subtitles (false)
 	, _subtitle_offset (0)
 	, _subtitle_scale (1)
-	, _audio_sample_rate (0)
-	, _has_subtitles (false)
 	, _frames_per_second (0)
 	, _dirty (false)
 {
@@ -128,6 +130,8 @@ Film::Film (string d, bool must_exist)
 		}
 	}
 
+	_external_audio_stream = ExternalAudioStream::create ();
+
 	read_metadata ();
 
 	_log = new FileLog (file ("log"));
@@ -148,9 +152,9 @@ Film::Film (Film const & o)
 	, _dcp_trim_start    (o._dcp_trim_start)
 	, _dcp_trim_end      (o._dcp_trim_end)
 	, _dcp_ab            (o._dcp_ab)
-	, _use_source_audio  (o._use_source_audio)
-	, _audio_stream      (o._audio_stream)
+	, _content_audio_stream (o._content_audio_stream)
 	, _external_audio    (o._external_audio)
+	, _use_content_audio (o._use_content_audio)
 	, _audio_gain        (o._audio_gain)
 	, _audio_delay       (o._audio_delay)
 	, _still_duration    (o._still_duration)
@@ -168,10 +172,9 @@ Film::Film (Film const & o)
 	, _thumbs            (o._thumbs)
 	, _size              (o._size)
 	, _length            (o._length)
-	, _audio_sample_rate (o._audio_sample_rate)
 	, _content_digest    (o._content_digest)
-	, _has_subtitles     (o._has_subtitles)
-	, _audio_streams     (o._audio_streams)
+	, _content_audio_streams (o._content_audio_streams)
+	, _external_audio_stream (o._external_audio_stream)
 	, _subtitle_streams  (o._subtitle_streams)
 	, _frames_per_second (o._frames_per_second)
 	, _dirty             (o._dirty)
@@ -263,10 +266,12 @@ Film::make_dcp (bool transcode)
 	o->ratio = format()->ratio_as_float (shared_from_this ());
 	if (dcp_length ()) {
 		o->video_decode_range = make_pair (dcp_trim_start(), dcp_trim_start() + dcp_length().get());
-		o->audio_decode_range = make_pair (
-			video_frames_to_audio_frames (o->video_decode_range.get().first, audio_sample_rate(), frames_per_second()),
-			video_frames_to_audio_frames (o->video_decode_range.get().second, audio_sample_rate(), frames_per_second())
-			);
+		if (audio_stream()) {
+			o->audio_decode_range = make_pair (
+				video_frames_to_audio_frames (o->video_decode_range.get().first, audio_stream()->sample_rate(), frames_per_second()),
+				video_frames_to_audio_frames (o->video_decode_range.get().second, audio_stream()->sample_rate(), frames_per_second())
+				);
+		}
 			
 	}
 	o->decode_subtitles = with_subtitles ();
@@ -331,13 +336,6 @@ Film::send_dcp_to_tms ()
 	JobManager::instance()->add (j);
 }
 
-void
-Film::copy_from_dvd ()
-{
-	shared_ptr<Job> j (new CopyFromDVDJob (shared_from_this(), shared_ptr<Job> ()));
-	JobManager::instance()->add (j);
-}
-
 /** Count the number of frames that have been encoded for this film.
  *  @return frame count.
  */
@@ -400,6 +398,8 @@ Film::write_metadata () const
 		throw CreateFileError (m);
 	}
 
+	f << "version " << state_version << "\n";
+
 	/* User stuff */
 	f << "name " << _name << "\n";
 	f << "use_dci_name " << _use_dci_name << "\n";
@@ -421,15 +421,19 @@ Film::write_metadata () const
 	f << "dcp_trim_start " << _dcp_trim_start << "\n";
 	f << "dcp_trim_end " << _dcp_trim_end << "\n";
 	f << "dcp_ab " << (_dcp_ab ? "1" : "0") << "\n";
-	f << "use_source_audio " << (_use_source_audio ? "1" : "0") << "\n";
-	f << "selected_audio_stream " << _audio_stream << "\n";
+	if (_content_audio_stream) {
+		f << "selected_content_audio_stream " << _content_audio_stream->to_string() << "\n";
+	}
 	for (vector<string>::const_iterator i = _external_audio.begin(); i != _external_audio.end(); ++i) {
 		f << "external_audio " << *i << "\n";
 	}
+	f << "use_content_audio " << (_use_content_audio ? "1" : "0") << "\n";
 	f << "audio_gain " << _audio_gain << "\n";
 	f << "audio_delay " << _audio_delay << "\n";
 	f << "still_duration " << _still_duration << "\n";
-	f << "selected_subtitle_stream " << _subtitle_stream << "\n";
+	if (_subtitle_stream) {
+		f << "selected_subtitle_stream " << _subtitle_stream->to_string() << "\n";
+	}
 	f << "with_subtitles " << _with_subtitles << "\n";
 	f << "subtitle_offset " << _subtitle_offset << "\n";
 	f << "subtitle_scale " << _subtitle_scale << "\n";
@@ -450,16 +454,16 @@ Film::write_metadata () const
 	f << "width " << _size.width << "\n";
 	f << "height " << _size.height << "\n";
 	f << "length " << _length.get_value_or(0) << "\n";
-	f << "audio_sample_rate " << _audio_sample_rate << "\n";
 	f << "content_digest " << _content_digest << "\n";
-	f << "has_subtitles " << _has_subtitles << "\n";
 
-	for (vector<AudioStream>::const_iterator i = _audio_streams.begin(); i != _audio_streams.end(); ++i) {
-		f << "audio_stream " << i->to_string () << "\n";
+	for (vector<shared_ptr<AudioStream> >::const_iterator i = _content_audio_streams.begin(); i != _content_audio_streams.end(); ++i) {
+		f << "content_audio_stream " << (*i)->to_string () << "\n";
 	}
 
-	for (vector<SubtitleStream>::const_iterator i = _subtitle_streams.begin(); i != _subtitle_streams.end(); ++i) {
-		f << "subtitle_stream " << i->to_string () << "\n";
+	f << "external_audio_stream " << _external_audio_stream->to_string() << "\n";
+
+	for (vector<shared_ptr<SubtitleStream> >::const_iterator i = _subtitle_streams.begin(); i != _subtitle_streams.end(); ++i) {
+		f << "subtitle_stream " << (*i)->to_string () << "\n";
 	}
 
 	f << "frames_per_second " << _frames_per_second << "\n";
@@ -475,15 +479,33 @@ Film::read_metadata ()
 
 	_external_audio.clear ();
 	_thumbs.clear ();
-	_audio_streams.clear ();
+	_content_audio_streams.clear ();
 	_subtitle_streams.clear ();
+
+	boost::optional<int> version;
+
+	/* Backward compatibility things */
+	boost::optional<int> audio_sample_rate;
+	boost::optional<int> audio_stream_index;
+	boost::optional<int> subtitle_stream_index;
 	
 	ifstream f (file ("metadata").c_str());
 	multimap<string, string> kv = read_key_value (f);
+
+	/* We need version before anything else */
+	multimap<string, string>::iterator v = kv.find ("version");
+	if (v != kv.end ()) {
+		version = atoi (v->second.c_str());
+	}
+	
 	for (multimap<string, string>::const_iterator i = kv.begin(); i != kv.end(); ++i) {
 		string const k = i->first;
 		string const v = i->second;
 
+		if (k == "audio_sample_rate") {
+			audio_sample_rate = atoi (v.c_str());
+		}
+
 		/* User-specified stuff */
 		if (k == "name") {
 			_name = v;
@@ -513,12 +535,16 @@ Film::read_metadata ()
 			_dcp_trim_end = atoi (v.c_str ());
 		} else if (k == "dcp_ab") {
 			_dcp_ab = (v == "1");
-		} else if (k == "use_source_audio") {
-			_use_source_audio = (v == "1");
-		} else if (k == "selected_audio_stream") {
-			_audio_stream = atoi (v.c_str ());
+		} else if (k == "selected_content_audio_stream" || (!version && k == "selected_audio_stream")) {
+			if (!version) {
+				audio_stream_index = atoi (v.c_str ());
+			} else {
+				_content_audio_stream = audio_stream_factory (v, version);
+			}
 		} else if (k == "external_audio") {
 			_external_audio.push_back (v);
+		} else if (k == "use_content_audio") {
+			_use_content_audio = (v == "1");
 		} else if (k == "audio_gain") {
 			_audio_gain = atof (v.c_str ());
 		} else if (k == "audio_delay") {
@@ -526,7 +552,11 @@ Film::read_metadata ()
 		} else if (k == "still_duration") {
 			_still_duration = atoi (v.c_str ());
 		} else if (k == "selected_subtitle_stream") {
-			_subtitle_stream = atoi (v.c_str ());
+			if (!version) {
+				subtitle_stream_index = atoi (v.c_str ());
+			} else {
+				_subtitle_stream = subtitle_stream_factory (v, version);
+			}
 		} else if (k == "with_subtitles") {
 			_with_subtitles = (v == "1");
 		} else if (k == "subtitle_offset") {
@@ -565,20 +595,37 @@ Film::read_metadata ()
 			if (vv) {
 				_length = vv;
 			}
-		} else if (k == "audio_sample_rate") {
-			_audio_sample_rate = atoi (v.c_str ());
 		} else if (k == "content_digest") {
 			_content_digest = v;
-		} else if (k == "has_subtitles") {
-			_has_subtitles = (v == "1");
-		} else if (k == "audio_stream") {
-			_audio_streams.push_back (AudioStream (v));
+		} else if (k == "content_audio_stream" || (!version && k == "audio_stream")) {
+			_content_audio_streams.push_back (audio_stream_factory (v, version));
+		} else if (k == "external_audio_stream") {
+			_external_audio_stream = audio_stream_factory (v, version);
 		} else if (k == "subtitle_stream") {
-			_subtitle_streams.push_back (SubtitleStream (v));
+			_subtitle_streams.push_back (subtitle_stream_factory (v, version));
 		} else if (k == "frames_per_second") {
 			_frames_per_second = atof (v.c_str ());
 		}
 	}
+
+	if (!version) {
+		if (audio_sample_rate) {
+			/* version < 1 didn't specify sample rate in the audio streams, so fill it in here */
+			for (vector<shared_ptr<AudioStream> >::iterator i = _content_audio_streams.begin(); i != _content_audio_streams.end(); ++i) {
+				(*i)->set_sample_rate (audio_sample_rate.get());
+			}
+		}
+
+		/* also the selected stream was specified as an index */
+		if (audio_stream_index && audio_stream_index.get() >= 0 && audio_stream_index.get() < (int) _content_audio_streams.size()) {
+			_content_audio_stream = _content_audio_streams[audio_stream_index.get()];
+		}
+
+		/* similarly the subtitle */
+		if (subtitle_stream_index && subtitle_stream_index.get() >= 0 && subtitle_stream_index.get() < (int) _subtitle_streams.size()) {
+			_subtitle_stream = _subtitle_streams[subtitle_stream_index.get()];
+		}
+	}
 		
 	_dirty = false;
 }
@@ -691,15 +738,12 @@ Film::content_path () const
 ContentType
 Film::content_type () const
 {
-#if BOOST_FILESYSTEM_VERSION == 3
-	string ext = boost::filesystem::path(_content).extension().string();
-#else
-	string ext = boost::filesystem::path(_content).extension();
-#endif
+	if (boost::filesystem::is_directory (_content)) {
+		/* Directory of images, we assume */
+		return VIDEO;
+	}
 
-	transform (ext.begin(), ext.end(), ext.begin(), ::tolower);
-	
-	if (ext == ".tif" || ext == ".tiff" || ext == ".jpg" || ext == ".jpeg" || ext == ".png") {
+	if (still_image_file (_content)) {
 		return STILL;
 	}
 
@@ -710,8 +754,12 @@ Film::content_type () const
 int
 Film::target_audio_sample_rate () const
 {
+	if (!audio_stream()) {
+		return 0;
+	}
+	
 	/* Resample to a DCI-approved sample rate */
-	double t = dcp_audio_sample_rate (audio_sample_rate());
+	double t = dcp_audio_sample_rate (audio_stream()->sample_rate());
 
 	DCPFrameRate dfr = dcp_frame_rate (frames_per_second ());
 
@@ -892,6 +940,10 @@ Film::set_content (string c)
 		_content = c;
 	}
 
+	/* Reset streams here in case the new content doesn't have one or the other */
+	_content_audio_stream = shared_ptr<AudioStream> ();
+	_subtitle_stream = shared_ptr<SubtitleStream> ();
+
 	/* Create a temporary decoder so that we can get information
 	   about the content.
 	*/
@@ -900,16 +952,21 @@ Film::set_content (string c)
 		shared_ptr<Options> o (new Options ("", "", ""));
 		o->out_size = Size (1024, 1024);
 		
-		shared_ptr<Decoder> d = decoder_factory (shared_from_this(), o, 0);
+		pair<shared_ptr<VideoDecoder>, shared_ptr<AudioDecoder> > d = decoder_factory (shared_from_this(), o, 0);
 		
-		set_size (d->native_size ());
-		set_frames_per_second (d->frames_per_second ());
-		set_audio_sample_rate (d->audio_sample_rate ());
-		set_has_subtitles (d->has_subtitles ());
-		set_audio_streams (d->audio_streams ());
-		set_subtitle_streams (d->subtitle_streams ());
-		set_audio_stream (audio_streams().empty() ? -1 : 0);
-		set_subtitle_stream (subtitle_streams().empty() ? -1 : 0);
+		set_size (d.first->native_size ());
+		set_frames_per_second (d.first->frames_per_second ());
+		set_subtitle_streams (d.first->subtitle_streams ());
+		set_content_audio_streams (d.second->audio_streams ());
+
+		/* Start off with the first audio and subtitle streams */
+		if (!d.second->audio_streams().empty()) {
+			set_content_audio_stream (d.second->audio_streams().front());
+		}
+		
+		if (!d.first->subtitle_streams().empty()) {
+			set_subtitle_stream (d.first->subtitle_streams().front());
+		}
 		
 		{
 			boost::mutex::scoped_lock lm (_state_mutex);
@@ -1069,33 +1126,42 @@ Film::set_dcp_ab (bool a)
 }
 
 void
-Film::set_use_source_audio (bool s)
+Film::set_content_audio_stream (shared_ptr<AudioStream> s)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
-		_use_source_audio = s;
+		_content_audio_stream = s;
 	}
-	signal_changed (USE_SOURCE_AUDIO);
+	signal_changed (CONTENT_AUDIO_STREAM);
 }
 
 void
-Film::set_audio_stream (int s)
+Film::set_external_audio (vector<string> a)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
-		_audio_stream = s;
+		_external_audio = a;
 	}
-	signal_changed (AUDIO_STREAM);
+
+	shared_ptr<Options> o (new Options ("", "", ""));
+	o->decode_audio = true;
+	shared_ptr<ExternalAudioDecoder> decoder (new ExternalAudioDecoder (shared_from_this(), o, 0));
+	if (decoder->audio_stream()) {
+		_external_audio_stream = decoder->audio_stream ();
+	}
+	
+	signal_changed (EXTERNAL_AUDIO);
 }
 
 void
-Film::set_external_audio (vector<string> a)
+Film::set_use_content_audio (bool e)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
-		_external_audio = a;
+		_use_content_audio = e;
 	}
-	signal_changed (EXTERNAL_AUDIO);
+
+	signal_changed (USE_CONTENT_AUDIO);
 }
 
 void
@@ -1129,7 +1195,7 @@ Film::set_still_duration (int d)
 }
 
 void
-Film::set_subtitle_stream (int s)
+Film::set_subtitle_stream (shared_ptr<SubtitleStream> s)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
@@ -1278,16 +1344,6 @@ Film::unset_length ()
 	signal_changed (LENGTH);
 }	
 
-void
-Film::set_audio_sample_rate (int r)
-{
-	{
-		boost::mutex::scoped_lock lm (_state_mutex);
-		_audio_sample_rate = r;
-	}
-	signal_changed (AUDIO_SAMPLE_RATE);
-}
-
 void
 Film::set_content_digest (string d)
 {
@@ -1299,33 +1355,23 @@ Film::set_content_digest (string d)
 }
 
 void
-Film::set_has_subtitles (bool s)
+Film::set_content_audio_streams (vector<shared_ptr<AudioStream> > s)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
-		_has_subtitles = s;
+		_content_audio_streams = s;
 	}
-	signal_changed (HAS_SUBTITLES);
+	signal_changed (CONTENT_AUDIO_STREAMS);
 }
 
 void
-Film::set_audio_streams (vector<AudioStream> s)
-{
-	{
-		boost::mutex::scoped_lock lm (_state_mutex);
-		_audio_streams = s;
-	}
-	_dirty = true;
-}
-
-void
-Film::set_subtitle_streams (vector<SubtitleStream> s)
+Film::set_subtitle_streams (vector<shared_ptr<SubtitleStream> > s)
 {
 	{
 		boost::mutex::scoped_lock lm (_state_mutex);
 		_subtitle_streams = s;
 	}
-	_dirty = true;
+	signal_changed (SUBTITLE_STREAMS);
 }
 
 void
@@ -1354,24 +1400,12 @@ Film::signal_changed (Property p)
 int
 Film::audio_channels () const
 {
-	boost::mutex::scoped_lock lm (_state_mutex);
-	
-	if (_use_source_audio) {
-		if (_audio_stream >= 0) {
-			return _audio_streams[_audio_stream].channels ();
-		}
-	} else {
-		int last_filled = -1;
-		for (size_t i = 0; i < _external_audio.size(); ++i) {
-			if (!_external_audio[i].empty()) {
-				last_filled = i;
-			}
-		}
-		
-		return last_filled + 1;
+	shared_ptr<AudioStream> s = audio_stream ();
+	if (!s) {
+		return 0;
 	}
 
-	return 0;
+	return s->channels ();
 }
 
 void
@@ -1380,3 +1414,12 @@ Film::set_dci_date_today ()
 	_dci_date = boost::gregorian::day_clock::local_day ();
 }
 
+boost::shared_ptr<AudioStream>
+Film::audio_stream () const
+{
+	if (use_content_audio()) {
+		return _content_audio_stream;
+	}
+
+	return _external_audio_stream;
+}