In a DCP with any subs/ccaps, make sure every reel has them (#1340).

4 files changed, 69 insertions, 13 deletions

diff --git a/src/lib/reel_writer.cc b/src/lib/reel_writer.cc
index 8ced98a11..2bd732d5b 100644
--- a/src/lib/reel_writer.cc
+++ b/src/lib/reel_writer.cc
@@ -60,6 +60,7 @@ using std::string;
 using std::cout;
 using std::exception;
 using std::map;
+using std::set;
 using std::vector;
 using boost::shared_ptr;
 using boost::optional;
@@ -616,25 +617,57 @@ ReelWriter::create_reel_text (
 	list<ReferencedReelAsset> const & refs,
 	list<shared_ptr<Font> > const& fonts,
 	int64_t duration,
-	boost::filesystem::path output_dcp
+	boost::filesystem::path output_dcp,
+	bool ensure_subtitles,
+	set<DCPTextTrack> ensure_closed_captions
 	) const
 {
 	shared_ptr<dcp::ReelSubtitleAsset> subtitle = maybe_add_text<dcp::ReelSubtitleAsset> (
 		_subtitle_asset, duration, reel, refs, fonts, film(), _period, output_dcp, _text_only
 		);
-	if (subtitle && !film()->subtitle_languages().empty()) {
-		subtitle->set_language (film()->subtitle_languages().front());
+
+	if (subtitle) {
+		/* We have a subtitle asset that we either made or are referencing */
+		if (!film()->subtitle_languages().empty()) {
+			subtitle->set_language (film()->subtitle_languages().front());
+		}
+	} else if (ensure_subtitles) {
+		/* We had no subtitle asset, but we've been asked to make sure there is one */
+		subtitle = maybe_add_text<dcp::ReelSubtitleAsset>(
+			empty_text_asset(TEXT_OPEN_SUBTITLE, optional<DCPTextTrack>()),
+			duration,
+			reel,
+			refs,
+			fonts,
+			film(),
+			_period,
+			output_dcp,
+			_text_only
+			);
 	}
 
 	for (map<DCPTextTrack, shared_ptr<dcp::SubtitleAsset> >::const_iterator i = _closed_caption_assets.begin(); i != _closed_caption_assets.end(); ++i) {
 		shared_ptr<dcp::ReelClosedCaptionAsset> a = maybe_add_text<dcp::ReelClosedCaptionAsset> (
 			i->second, duration, reel, refs, fonts, film(), _period, output_dcp, _text_only
 			);
-		if (a) {
-			a->set_annotation_text (i->first.name);
-			if (!i->first.language.empty()) {
-				a->set_language (dcp::LanguageTag(i->first.language));
-			}
+		DCPOMATIC_ASSERT (a);
+		a->set_annotation_text (i->first.name);
+		if (!i->first.language.empty()) {
+			a->set_language (dcp::LanguageTag(i->first.language));
+		}
+
+		ensure_closed_captions.erase (i->first);
+	}
+
+	/* Make empty tracks for anything we've been asked to ensure but that we haven't added */
+	BOOST_FOREACH (DCPTextTrack i, ensure_closed_captions) {
+		shared_ptr<dcp::ReelClosedCaptionAsset> a = maybe_add_text<dcp::ReelClosedCaptionAsset> (
+			empty_text_asset(TEXT_CLOSED_CAPTION, i), duration, reel, refs, fonts, film(), _period, output_dcp, _text_only
+			);
+		DCPOMATIC_ASSERT (a);
+		a->set_annotation_text (i.name);
+		if (!i.language.empty()) {
+			a->set_language (dcp::LanguageTag(i.language));
 		}
 	}
 }
@@ -666,8 +699,17 @@ ReelWriter::create_reel_markers (shared_ptr<dcp::Reel> reel) const
 }
 
 
+/** @param ensure_subtitles true to make sure the reel has a subtitle asset.
+ *  @param ensure_closed_captions make sure the reel has these closed caption tracks.
+ */
 shared_ptr<dcp::Reel>
-ReelWriter::create_reel (list<ReferencedReelAsset> const & refs, list<shared_ptr<Font> > const & fonts, boost::filesystem::path output_dcp)
+ReelWriter::create_reel (
+	list<ReferencedReelAsset> const & refs,
+	list<shared_ptr<Font> > const & fonts,
+	boost::filesystem::path output_dcp,
+	bool ensure_subtitles,
+	set<DCPTextTrack> ensure_closed_captions
+	)
 {
 	LOG_GENERAL ("create_reel for %1-%2; %3 of %4", _period.from.get(), _period.to.get(), _reel_index, _reel_count);
 
@@ -685,7 +727,7 @@ ReelWriter::create_reel (list<ReferencedReelAsset> const & refs, list<shared_ptr
 		create_reel_markers (reel);
 	}
 
-	create_reel_text (reel, refs, fonts, duration, output_dcp);
+	create_reel_text (reel, refs, fonts, duration, output_dcp, ensure_subtitles, ensure_closed_captions);
 
 	if (_atmos_asset) {
 		reel->add (shared_ptr<dcp::ReelAtmosAsset>(new dcp::ReelAtmosAsset(_atmos_asset, 0)));
diff --git a/src/lib/reel_writer.h b/src/lib/reel_writer.h
index 08b85a785..388494247 100644
--- a/src/lib/reel_writer.h
+++ b/src/lib/reel_writer.h
@@ -77,7 +77,11 @@ public:
 
 	void finish (boost::filesystem::path output_dcp);
 	boost::shared_ptr<dcp::Reel> create_reel (
-		std::list<ReferencedReelAsset> const & refs, std::list<boost::shared_ptr<dcpomatic::Font> > const & fonts, boost::filesystem::path output_dcp
+		std::list<ReferencedReelAsset> const & refs,
+		std::list<boost::shared_ptr<dcpomatic::Font> > const & fonts,
+		boost::filesystem::path output_dcp,
+		bool ensure_subtitles,
+		std::set<DCPTextTrack> ensure_closed_captions
 		);
 	void calculate_digests (boost::function<void (float)> set_progress);
 
@@ -109,7 +113,9 @@ private:
 		boost::shared_ptr<dcp::Reel> reel,
 		std::list<ReferencedReelAsset> const & refs, std::list<boost::shared_ptr<dcpomatic::Font> > const& fonts,
 		int64_t duration,
-		boost::filesystem::path output_dcp
+		boost::filesystem::path output_dcp,
+		bool ensure_subtitles,
+		std::set<DCPTextTrack> ensure_closed_captions
 		) const;
 	void create_reel_markers (boost::shared_ptr<dcp::Reel> reel) const;
 
diff --git a/src/lib/writer.cc b/src/lib/writer.cc
index 0b85a7f32..fcf084c56 100644
--- a/src/lib/writer.cc
+++ b/src/lib/writer.cc
@@ -95,6 +95,7 @@ Writer::Writer (weak_ptr<const Film> weak_film, weak_ptr<Job> j, bool text_only)
 	, _repeat_written (0)
 	, _pushed_to_disk (0)
 	, _text_only (text_only)
+	, _have_subtitles (false)
 {
 	shared_ptr<Job> job = _job.lock ();
 
@@ -590,7 +591,7 @@ Writer::finish (boost::filesystem::path output_dcp)
 	/* Add reels */
 
 	BOOST_FOREACH (ReelWriter& i, _reels) {
-		cpl->add (i.create_reel(_reel_assets, _fonts, output_dcp));
+		cpl->add (i.create_reel(_reel_assets, _fonts, output_dcp, _have_subtitles, _have_closed_captions));
 	}
 
 	/* Add metadata */
@@ -769,11 +770,13 @@ Writer::write (PlayerText text, TextType type, optional<DCPTextTrack> track, DCP
 	switch (type) {
 	case TEXT_OPEN_SUBTITLE:
 		reel = &_subtitle_reel;
+		_have_subtitles = true;
 		break;
 	case TEXT_CLOSED_CAPTION:
 		DCPOMATIC_ASSERT (track);
 		DCPOMATIC_ASSERT (_caption_reels.find(*track) != _caption_reels.end());
 		reel = &_caption_reels[*track];
+		_have_closed_captions.insert (*track);
 		break;
 	default:
 		DCPOMATIC_ASSERT (false);
diff --git a/src/lib/writer.h b/src/lib/writer.h
index 3b5cc3dc3..52cb222e7 100644
--- a/src/lib/writer.h
+++ b/src/lib/writer.h
@@ -197,4 +197,9 @@ private:
 	std::list<ReferencedReelAsset> _reel_assets;
 
 	std::list<boost::shared_ptr<dcpomatic::Font> > _fonts;
+
+	/** true if any reel has any subtitles */
+	bool _have_subtitles;
+	/** all closed caption tracks that we have on any reel */
+	std::set<DCPTextTrack> _have_closed_captions;
 };