Pull ActiveSubtitles code out to a class.
[dcpomatic.git] / src / lib / active_subtitles.cc
1 /*
2     Copyright (C) 2017 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 "active_subtitles.h"
22 #include "piece.h"
23 #include "subtitle_content.h"
24 #include <boost/shared_ptr.hpp>
25 #include <boost/weak_ptr.hpp>
26
27 using std::list;
28 using std::pair;
29 using std::make_pair;
30 using boost::weak_ptr;
31 using boost::shared_ptr;
32
33 list<PlayerSubtitles>
34 ActiveSubtitles::get (DCPTime time, bool always_burn_subtitles) const
35 {
36         list<PlayerSubtitles> ps;
37
38         for (Map::const_iterator i = _data.begin(); i != _data.end(); ++i) {
39
40                 shared_ptr<Piece> piece = i->first.lock ();
41                 if (!piece) {
42                         continue;
43                 }
44
45                 if (!piece->content->subtitle->use() || (!always_burn_subtitles && !piece->content->subtitle->burn())) {
46                         continue;
47                 }
48
49                 BOOST_FOREACH (Period j, i->second) {
50                         if (j.from <= time && (!j.to || j.to.get() > time)) {
51                                 ps.push_back (j.subs);
52                         }
53                 }
54         }
55
56         return ps;
57 }
58
59 void
60 ActiveSubtitles::clear_before (DCPTime time)
61 {
62         Map updated;
63         for (Map::const_iterator i = _data.begin(); i != _data.end(); ++i) {
64                 list<Period> as;
65                 BOOST_FOREACH (Period j, i->second) {
66                         if (!j.to || j.to.get() >= time) {
67                                 as.push_back (j);
68                         }
69                 }
70                 if (!as.empty ()) {
71                         updated[i->first] = as;
72                 }
73         }
74         _data = updated;
75 }
76
77 void
78 ActiveSubtitles::add_from (weak_ptr<Piece> piece, PlayerSubtitles ps, DCPTime from)
79 {
80         if (_data.find(piece) == _data.end()) {
81                 _data[piece] = list<Period>();
82         }
83         _data[piece].push_back (Period (ps, from));
84 }
85
86 pair<PlayerSubtitles, DCPTime>
87 ActiveSubtitles::add_to (weak_ptr<Piece> piece, DCPTime to)
88 {
89         DCPOMATIC_ASSERT (_data.find(piece) != _data.end());
90
91         _data[piece].back().to = to;
92
93         BOOST_FOREACH (SubtitleString& i, _data[piece].back().subs.text) {
94                 i.set_out (dcp::Time(to.seconds(), 1000));
95         }
96
97         return make_pair (_data[piece].back().subs, _data[piece].back().from);
98 }
99
100 bool
101 ActiveSubtitles::have (weak_ptr<Piece> piece) const
102 {
103         Map::const_iterator i = _data.find(piece);
104         if (i == _data.end()) {
105                 return false;
106         }
107
108         return !i->second.empty();
109 }
110
111 void
112 ActiveSubtitles::clear ()
113 {
114         _data.clear ();
115 }