3a182c0c7a20c4ea953c6ea02ad601c051029d60
[dcpomatic.git] / src / lib / dcp_subtitle_decoder.cc
1 /*
2     Copyright (C) 2014-2018 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "dcp_subtitle_decoder.h"
22 #include "dcp_subtitle_content.h"
23 #include <dcp/interop_subtitle_asset.h>
24 #include <iostream>
25
26 using std::list;
27 using std::cout;
28 using boost::shared_ptr;
29 using boost::dynamic_pointer_cast;
30 using boost::bind;
31
32 DCPSubtitleDecoder::DCPSubtitleDecoder (shared_ptr<const Film> film, shared_ptr<const DCPSubtitleContent> content)
33         : Decoder (film)
34 {
35         shared_ptr<dcp::SubtitleAsset> c (load (content->path (0)));
36         _subtitles = c->subtitles ();
37         _next = _subtitles.begin ();
38
39         ContentTime first;
40         if (_next != _subtitles.end()) {
41                 first = content_time_period(*_next).from;
42         }
43         text.push_back (shared_ptr<TextDecoder> (new TextDecoder (this, content->only_text(), first)));
44 }
45
46 void
47 DCPSubtitleDecoder::seek (ContentTime time, bool accurate)
48 {
49         Decoder::seek (time, accurate);
50
51         _next = _subtitles.begin ();
52         list<shared_ptr<dcp::Subtitle> >::const_iterator i = _subtitles.begin ();
53         while (i != _subtitles.end() && ContentTime::from_seconds ((*_next)->in().as_seconds()) < time) {
54                 ++i;
55         }
56 }
57
58 bool
59 DCPSubtitleDecoder::pass ()
60 {
61         if (_next == _subtitles.end ()) {
62                 return true;
63         }
64
65         /* Gather all subtitles with the same time period that are next
66            on the list.  We must emit all subtitles for the same time
67            period with the same emit*() call otherwise the
68            TextDecoder will assume there is nothing else at the
69            time of emitting the first.
70         */
71
72         list<dcp::SubtitleString> s;
73         list<dcp::SubtitleImage> i;
74         ContentTimePeriod const p = content_time_period (*_next);
75
76         while (_next != _subtitles.end () && content_time_period (*_next) == p) {
77                 shared_ptr<dcp::SubtitleString> ns = dynamic_pointer_cast<dcp::SubtitleString>(*_next);
78                 if (ns) {
79                         s.push_back (*ns);
80                         ++_next;
81                 } else {
82                         /* XXX: perhaps these image subs should also be collected together like the string ones are;
83                            this would need to be done both here and in DCPDecoder.
84                         */
85
86                         shared_ptr<dcp::SubtitleImage> ni = dynamic_pointer_cast<dcp::SubtitleImage>(*_next);
87                         if (ni) {
88                                 emit_subtitle_image (p, *ni, film()->frame_size(), only_text());
89                                 ++_next;
90                         }
91                 }
92         }
93
94         only_text()->emit_plain (p, s);
95         return false;
96 }
97
98 ContentTimePeriod
99 DCPSubtitleDecoder::content_time_period (shared_ptr<dcp::Subtitle> s) const
100 {
101         return ContentTimePeriod (
102                 ContentTime::from_seconds (s->in().as_seconds ()),
103                 ContentTime::from_seconds (s->out().as_seconds ())
104                 );
105 }