ad9c00b63da4962a0d3a3447c567f363721fc862
[dcpomatic.git] / src / lib / text_subtitle_decoder.cc
1 /*
2     Copyright (C) 2014-2016 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 "text_subtitle_decoder.h"
22 #include "text_subtitle_content.h"
23 #include "subtitle_content.h"
24 #include <dcp/subtitle_string.h>
25 #include <boost/foreach.hpp>
26 #include <iostream>
27
28 using std::list;
29 using std::vector;
30 using std::string;
31 using std::cout;
32 using std::max;
33 using boost::shared_ptr;
34 using boost::optional;
35 using boost::dynamic_pointer_cast;
36
37 TextSubtitleDecoder::TextSubtitleDecoder (shared_ptr<const TextSubtitleContent> content)
38         : TextSubtitle (content)
39         , _next (0)
40 {
41         subtitle.reset (
42                 new SubtitleDecoder (
43                         this,
44                         content->subtitle,
45                         bind (&TextSubtitleDecoder::image_subtitles_during, this, _1, _2),
46                         bind (&TextSubtitleDecoder::text_subtitles_during, this, _1, _2)
47                         )
48                 );
49 }
50
51 void
52 TextSubtitleDecoder::seek (ContentTime time, bool accurate)
53 {
54         subtitle->seek (time, accurate);
55
56         _next = 0;
57         while (_next < _subtitles.size() && ContentTime::from_seconds (_subtitles[_next].from.all_as_seconds ()) < time) {
58                 ++_next;
59         }
60 }
61
62 bool
63 TextSubtitleDecoder::pass (PassReason, bool)
64 {
65         if (_next >= _subtitles.size ()) {
66                 return true;
67         }
68
69         subtitle->give_text (content_time_period (_subtitles[_next]), _subtitles[_next]);
70
71         ++_next;
72         return false;
73 }
74
75 list<ContentTimePeriod>
76 TextSubtitleDecoder::image_subtitles_during (ContentTimePeriod, bool) const
77 {
78         return list<ContentTimePeriod> ();
79 }
80
81 list<ContentTimePeriod>
82 TextSubtitleDecoder::text_subtitles_during (ContentTimePeriod p, bool starting) const
83 {
84         /* XXX: inefficient */
85
86         list<ContentTimePeriod> d;
87
88         for (vector<sub::Subtitle>::const_iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) {
89                 ContentTimePeriod t = content_time_period (*i);
90                 if ((starting && p.contains (t.from)) || (!starting && p.overlaps (t))) {
91                         d.push_back (t);
92                 }
93         }
94
95         return d;
96 }
97
98 ContentTimePeriod
99 TextSubtitleDecoder::content_time_period (sub::Subtitle s) const
100 {
101         return ContentTimePeriod (
102                 ContentTime::from_seconds (s.from.all_as_seconds()),
103                 ContentTime::from_seconds (s.to.all_as_seconds())
104                 );
105 }
106
107 void
108 TextSubtitleDecoder::reset ()
109 {
110         subtitle->reset ();
111 }