Update to test/data.
[dcpomatic.git] / src / lib / text_caption_file_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 "text_caption_file_decoder.h"
22 #include "text_caption_file_content.h"
23 #include "caption_content.h"
24 #include "caption_decoder.h"
25 #include <dcp/subtitle_string.h>
26 #include <boost/foreach.hpp>
27 #include <iostream>
28
29 using std::list;
30 using std::vector;
31 using std::string;
32 using std::cout;
33 using std::max;
34 using boost::shared_ptr;
35 using boost::optional;
36 using boost::dynamic_pointer_cast;
37
38 TextCaptionFileDecoder::TextCaptionFileDecoder (shared_ptr<const TextCaptionFileContent> content, shared_ptr<Log> log)
39         : TextCaptionFile (content)
40         , _next (0)
41 {
42         ContentTime first;
43         if (!_subtitles.empty()) {
44                 first = content_time_period(_subtitles[0]).from;
45         }
46         caption.push_back (shared_ptr<CaptionDecoder> (new CaptionDecoder (this, content->only_caption(), log, first)));
47 }
48
49 void
50 TextCaptionFileDecoder::seek (ContentTime time, bool accurate)
51 {
52         /* It's worth back-tracking a little here as decoding is cheap and it's nice if we don't miss
53            too many subtitles when seeking.
54         */
55         time -= ContentTime::from_seconds (5);
56         if (time < ContentTime()) {
57                 time = ContentTime();
58         }
59
60         Decoder::seek (time, accurate);
61
62         _next = 0;
63         while (_next < _subtitles.size() && ContentTime::from_seconds (_subtitles[_next].from.all_as_seconds ()) < time) {
64                 ++_next;
65         }
66 }
67
68 bool
69 TextCaptionFileDecoder::pass ()
70 {
71         if (_next >= _subtitles.size ()) {
72                 return true;
73         }
74
75         ContentTimePeriod const p = content_time_period (_subtitles[_next]);
76         only_caption()->emit_plain (p, _subtitles[_next]);
77
78         ++_next;
79         return false;
80 }
81
82 ContentTimePeriod
83 TextCaptionFileDecoder::content_time_period (sub::Subtitle s) const
84 {
85         return ContentTimePeriod (
86                 ContentTime::from_seconds (s.from.all_as_seconds()),
87                 ContentTime::from_seconds (s.to.all_as_seconds())
88                 );
89 }