X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fdcp_subtitle_decoder.cc;h=711dc77f263e6cf7d2d64ba40bc877f33a6a761d;hb=a75cac36969942d1fcb43dadfdf60434c840af54;hp=9db3254019675061cc0f8095abb9f89604b111f1;hpb=de2af791bdfdcd653752cba970e59efc7bf810c7;p=dcpomatic.git

diff --git a/src/lib/dcp_subtitle_decoder.cc b/src/lib/dcp_subtitle_decoder.cc
index 9db325401..711dc77f2 100644
--- a/src/lib/dcp_subtitle_decoder.cc
+++ b/src/lib/dcp_subtitle_decoder.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2014-2016 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2014-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -18,66 +18,144 @@
 
 */
 
-#include "dcp_subtitle_decoder.h"
+
 #include "dcp_subtitle_content.h"
+#include "dcp_subtitle_decoder.h"
+#include "film.h"
+#include "font.h"
+#include "text_content.h"
+#include "util.h"
 #include <dcp/interop_subtitle_asset.h>
-#include <iostream>
+#include <dcp/load_font_node.h>
+
 
+using std::dynamic_pointer_cast;
 using std::list;
-using std::cout;
-using boost::shared_ptr;
-using boost::bind;
+using std::make_shared;
+using std::shared_ptr;
+using std::string;
+using std::vector;
+using boost::optional;
+using namespace dcpomatic;
 
-DCPSubtitleDecoder::DCPSubtitleDecoder (shared_ptr<const DCPSubtitleContent> content, shared_ptr<Log> log)
-{
-	subtitle.reset (new SubtitleDecoder (this, content->subtitle, log));
 
-	shared_ptr<dcp::SubtitleAsset> c (load (content->path (0)));
-	_subtitles = c->subtitles ();
+DCPSubtitleDecoder::DCPSubtitleDecoder (shared_ptr<const Film> film, shared_ptr<const DCPSubtitleContent> content)
+	: Decoder (film)
+{
+	/* Load the XML or MXF file */
+	_asset = load(content->path(0));
+	_asset->fix_empty_font_ids();
+	_subtitles = _asset->subtitles();
 	_next = _subtitles.begin ();
+
+	_subtitle_standard = _asset->subtitle_standard();
+
+	text.push_back (make_shared<TextDecoder>(this, content->only_text()));
+	update_position();
+
+	FontIDAllocator font_id_allocator;
+
+	for (auto node: _asset->load_font_nodes()) {
+		_font_id_allocator.add_font(0, _asset->id(), node->id);
+	}
+	_font_id_allocator.allocate();
 }
 
+
 void
-DCPSubtitleDecoder::seek (ContentTime time, bool)
+DCPSubtitleDecoder::seek (ContentTime time, bool accurate)
 {
+	Decoder::seek (time, accurate);
+
 	_next = _subtitles.begin ();
-	list<dcp::SubtitleString>::const_iterator i = _subtitles.begin ();
-	while (i != _subtitles.end() && ContentTime::from_seconds (_next->in().as_seconds()) < time) {
+	auto i = _subtitles.begin ();
+	while (i != _subtitles.end() && ContentTime::from_seconds ((*_next)->in().as_seconds()) < time) {
 		++i;
 	}
+
+	update_position();
 }
 
-void
+
+bool
 DCPSubtitleDecoder::pass ()
 {
 	if (_next == _subtitles.end ()) {
-		return;
+		return true;
 	}
 
 	/* Gather all subtitles with the same time period that are next
 	   on the list.  We must emit all subtitles for the same time
-	   period with the same text_subtitle() call otherwise the
-	   SubtitleDecoder will assume there is nothing else at the
-	   time of emit the first.
+	   period with the same emit*() call otherwise the
+	   TextDecoder will assume there is nothing else at the
+	   time of emitting the first.
 	*/
 
-	list<dcp::SubtitleString> s;
-	ContentTimePeriod const p = content_time_period (*_next);
+	vector<dcp::SubtitleString> s;
+	vector<dcp::SubtitleImage> i;
+	auto const p = content_time_period (*_next);
 
 	while (_next != _subtitles.end () && content_time_period (*_next) == p) {
-		s.push_back (*_next);
-		++_next;
+		auto ns = dynamic_pointer_cast<const dcp::SubtitleString>(*_next);
+		if (ns) {
+			dcp::SubtitleString ns_copy = *ns;
+			if (ns_copy.font()) {
+				ns_copy.set_font(_font_id_allocator.font_id(0, _asset->id(), ns_copy.font().get()));
+			} else {
+				ns_copy.set_font(_font_id_allocator.default_font_id());
+			}
+			s.push_back(ns_copy);
+			++_next;
+		} else {
+			/* XXX: perhaps these image subs should also be collected together like the string ones are;
+			   this would need to be done both here and in DCPDecoder.
+			*/
+
+			auto ni = dynamic_pointer_cast<const dcp::SubtitleImage>(*_next);
+			if (ni) {
+				emit_subtitle_image (p, *ni, film()->frame_size(), only_text());
+				++_next;
+			}
+		}
 	}
 
-	subtitle->emit_text (p, s);
-	subtitle->set_position (p.from);
+	only_text()->emit_plain(p, s, _subtitle_standard);
+
+	update_position();
+
+	return false;
 }
 
+
 ContentTimePeriod
-DCPSubtitleDecoder::content_time_period (dcp::SubtitleString s) const
+DCPSubtitleDecoder::content_time_period (shared_ptr<const dcp::Subtitle> s) const
+{
+	return {
+		ContentTime::from_seconds(s->in().as_seconds()),
+		ContentTime::from_seconds(s->out().as_seconds())
+	};
+}
+
+
+/** @return time of first subtitle, if there is one */
+optional<ContentTime>
+DCPSubtitleDecoder::first () const
+{
+	if (_subtitles.empty()) {
+		return {};
+	}
+
+	return ContentTime::from_seconds(_subtitles[0]->in().as_seconds());
+}
+
+
+void
+DCPSubtitleDecoder::update_position()
 {
-	return ContentTimePeriod (
-		ContentTime::from_seconds (s.in().as_seconds ()),
-		ContentTime::from_seconds (s.out().as_seconds ())
-		);
+	if (_next != _subtitles.end()) {
+		only_text()->maybe_set_position(
+			ContentTime::from_seconds((*_next)->in().as_seconds())
+			);
+	}
 }
+