Updated cs_CZ translation from Tomáš Begeni.
[dcpomatic.git] / src / lib / dcp_subtitle_decoder.cc
index e3c06378b8eabd86216d8f08394e1db57152fab6..711dc77f263e6cf7d2d64ba40bc877f33a6a761d 100644 (file)
@@ -1,50 +1,82 @@
 /*
-    Copyright (C) 2014 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2014-2021 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    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.
 
-    This program is distributed in the hope that it will be useful,
+    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 this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
-#include "dcp_subtitle_decoder.h"
+
 #include "dcp_subtitle_content.h"
-#include <dcp/interop_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 <dcp/load_font_node.h>
 
+
+using std::dynamic_pointer_cast;
 using std::list;
-using std::cout;
-using boost::shared_ptr;
+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)
-       : SubtitleDecoder (content)
+DCPSubtitleDecoder::DCPSubtitleDecoder (shared_ptr<const Film> film, shared_ptr<const DCPSubtitleContent> content)
+       : Decoder (film)
 {
-       shared_ptr<dcp::SubtitleContent> c (load (content->path (0)));
-       _subtitles = c->subtitles ();
+       /* 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 accurate)
 {
-       SubtitleDecoder::seek (time, 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().to_seconds()) < time) {
+       auto i = _subtitles.begin ();
+       while (i != _subtitles.end() && ContentTime::from_seconds ((*_next)->in().as_seconds()) < time) {
                ++i;
        }
+
+       update_position();
 }
 
+
 bool
 DCPSubtitleDecoder::pass ()
 {
@@ -52,32 +84,78 @@ DCPSubtitleDecoder::pass ()
                return true;
        }
 
-       list<dcp::SubtitleString> s;
-       s.push_back (*_next);
-       text_subtitle (s);
-       ++_next;
+       /* 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 emit*() call otherwise the
+          TextDecoder will assume there is nothing else at the
+          time of emitting the first.
+       */
+
+       vector<dcp::SubtitleString> s;
+       vector<dcp::SubtitleImage> i;
+       auto const p = content_time_period (*_next);
+
+       while (_next != _subtitles.end () && content_time_period (*_next) == p) {
+               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;
+                       }
+               }
+       }
+
+       only_text()->emit_plain(p, s, _subtitle_standard);
+
+       update_position();
 
        return false;
 }
 
-list<ContentTimePeriod>
-DCPSubtitleDecoder::subtitles_during (ContentTimePeriod p, bool starting) const
+
+ContentTimePeriod
+DCPSubtitleDecoder::content_time_period (shared_ptr<const dcp::Subtitle> s) const
 {
-       /* XXX: inefficient */
+       return {
+               ContentTime::from_seconds(s->in().as_seconds()),
+               ContentTime::from_seconds(s->out().as_seconds())
+       };
+}
 
-       list<ContentTimePeriod> d;
 
-       for (list<dcp::SubtitleString>::const_iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) {
-               ContentTimePeriod period (
-                       ContentTime::from_seconds (i->in().to_seconds ()),
-                       ContentTime::from_seconds (i->out().to_seconds ())
-                       );
-               
-               if ((starting && p.contains (period.from)) || (!starting && p.overlaps (period))) {
-                       d.push_back (period);
-               }
+/** @return time of first subtitle, if there is one */
+optional<ContentTime>
+DCPSubtitleDecoder::first () const
+{
+       if (_subtitles.empty()) {
+               return {};
        }
 
-       return d;
+       return ContentTime::from_seconds(_subtitles[0]->in().as_seconds());
+}
+
+
+void
+DCPSubtitleDecoder::update_position()
+{
+       if (_next != _subtitles.end()) {
+               only_text()->maybe_set_position(
+                       ContentTime::from_seconds((*_next)->in().as_seconds())
+                       );
+       }
 }