Include tidying src/lib/a-j*.h
[dcpomatic.git] / src / lib / playlist.cc
1 /*
2     Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include "playlist.h"
21 #include "sndfile_content.h"
22 #include "sndfile_decoder.h"
23 #include "video_content.h"
24 #include "ffmpeg_decoder.h"
25 #include "ffmpeg_content.h"
26 #include "image_decoder.h"
27 #include "content_factory.h"
28 #include "job.h"
29 #include "config.h"
30 #include "util.h"
31 #include "md5_digester.h"
32 #include <libcxml/cxml.h>
33 #include <libxml++/libxml++.h>
34 #include <boost/shared_ptr.hpp>
35
36 #include "i18n.h"
37
38 using std::list;
39 using std::cout;
40 using std::vector;
41 using std::min;
42 using std::max;
43 using std::string;
44 using std::pair;
45 using boost::optional;
46 using boost::shared_ptr;
47 using boost::weak_ptr;
48 using boost::dynamic_pointer_cast;
49
50 Playlist::Playlist ()
51         : _sequence_video (true)
52         , _sequencing_video (false)
53 {
54
55 }
56
57 Playlist::~Playlist ()
58 {
59         _content.clear ();
60         reconnect ();
61 }
62
63 void
64 Playlist::content_changed (weak_ptr<Content> content, int property, bool frequent)
65 {
66         /* Don't respond to position changes here, as:
67            - sequencing after earlier/later changes is handled by move_earlier/move_later
68            - any other position changes will be timeline drags which should not result in content
69            being sequenced.
70         */
71
72         if (property == ContentProperty::LENGTH || property == VideoContentProperty::VIDEO_FRAME_TYPE) {
73                 maybe_sequence_video ();
74         }
75
76         ContentChanged (content, property, frequent);
77 }
78
79 void
80 Playlist::maybe_sequence_video ()
81 {
82         if (!_sequence_video || _sequencing_video) {
83                 return;
84         }
85
86         _sequencing_video = true;
87
88         DCPTime next_left;
89         DCPTime next_right;
90         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
91                 shared_ptr<VideoContent> vc = dynamic_pointer_cast<VideoContent> (*i);
92                 if (!vc) {
93                         continue;
94                 }
95
96                 if (vc->video_frame_type() == VIDEO_FRAME_TYPE_3D_RIGHT) {
97                         vc->set_position (next_right);
98                         next_right = vc->end() + DCPTime::delta ();
99                 } else {
100                         vc->set_position (next_left);
101                         next_left = vc->end() + DCPTime::delta ();
102                 }
103         }
104
105         /* This won't change order, so it does not need a sort */
106
107         _sequencing_video = false;
108 }
109
110 string
111 Playlist::video_identifier () const
112 {
113         string t;
114
115         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
116                 shared_ptr<const VideoContent> vc = dynamic_pointer_cast<const VideoContent> (*i);
117                 shared_ptr<const SubtitleContent> sc = dynamic_pointer_cast<const SubtitleContent> (*i);
118                 if (vc) {
119                         t += vc->identifier ();
120                 } else if (sc && sc->burn_subtitles ()) {
121                         t += sc->identifier ();
122                 }
123         }
124
125         MD5Digester digester;
126         digester.add (t.c_str(), t.length());
127         return digester.get ();
128 }
129
130 /** @param node <Playlist> node */
131 void
132 Playlist::set_from_xml (shared_ptr<const Film> film, cxml::ConstNodePtr node, int version, list<string>& notes)
133 {
134         list<cxml::NodePtr> c = node->node_children ("Content");
135         for (list<cxml::NodePtr>::iterator i = c.begin(); i != c.end(); ++i) {
136                 _content.push_back (content_factory (film, *i, version, notes));
137         }
138
139         sort (_content.begin(), _content.end(), ContentSorter ());
140
141         reconnect ();
142 }
143
144 /** @param node <Playlist> node */
145 void
146 Playlist::as_xml (xmlpp::Node* node)
147 {
148         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
149                 (*i)->as_xml (node->add_child ("Content"));
150         }
151 }
152
153 void
154 Playlist::add (shared_ptr<Content> c)
155 {
156         _content.push_back (c);
157         sort (_content.begin(), _content.end(), ContentSorter ());
158         reconnect ();
159         Changed ();
160 }
161
162 void
163 Playlist::remove (shared_ptr<Content> c)
164 {
165         ContentList::iterator i = _content.begin ();
166         while (i != _content.end() && *i != c) {
167                 ++i;
168         }
169
170         if (i != _content.end ()) {
171                 _content.erase (i);
172                 Changed ();
173         }
174
175         /* This won't change order, so it does not need a sort */
176 }
177
178 void
179 Playlist::remove (ContentList c)
180 {
181         for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
182                 ContentList::iterator j = _content.begin ();
183                 while (j != _content.end() && *j != *i) {
184                         ++j;
185                 }
186
187                 if (j != _content.end ()) {
188                         _content.erase (j);
189                 }
190         }
191
192         /* This won't change order, so it does not need a sort */
193
194         Changed ();
195 }
196
197 class FrameRateCandidate
198 {
199 public:
200         FrameRateCandidate (float source_, int dcp_)
201                 : source (source_)
202                 , dcp (dcp_)
203         {}
204
205         float source;
206         int dcp;
207 };
208
209 int
210 Playlist::best_dcp_frame_rate () const
211 {
212         list<int> const allowed_dcp_frame_rates = Config::instance()->allowed_dcp_frame_rates ();
213
214         /* Work out what rates we could manage, including those achieved by using skip / repeat. */
215         list<FrameRateCandidate> candidates;
216
217         /* Start with the ones without skip / repeat so they will get matched in preference to skipped/repeated ones */
218         for (list<int>::const_iterator i = allowed_dcp_frame_rates.begin(); i != allowed_dcp_frame_rates.end(); ++i) {
219                 candidates.push_back (FrameRateCandidate (*i, *i));
220         }
221
222         /* Then the skip/repeat ones */
223         for (list<int>::const_iterator i = allowed_dcp_frame_rates.begin(); i != allowed_dcp_frame_rates.end(); ++i) {
224                 candidates.push_back (FrameRateCandidate (float (*i) / 2, *i));
225                 candidates.push_back (FrameRateCandidate (float (*i) * 2, *i));
226         }
227
228         /* Pick the best one */
229         float error = std::numeric_limits<float>::max ();
230         optional<FrameRateCandidate> best;
231         list<FrameRateCandidate>::iterator i = candidates.begin();
232         while (i != candidates.end()) {
233
234                 float this_error = 0;
235                 for (ContentList::const_iterator j = _content.begin(); j != _content.end(); ++j) {
236                         shared_ptr<VideoContent> vc = dynamic_pointer_cast<VideoContent> (*j);
237                         if (!vc) {
238                                 continue;
239                         }
240
241                         /* Best error for this content; we could use the content as-is or double its rate */
242                         float best_error = min (
243                                 float (fabs (i->source - vc->video_frame_rate ())),
244                                 float (fabs (i->source - vc->video_frame_rate () * 2))
245                                 );
246
247                         /* Use the largest difference between DCP and source as the "error" */
248                         this_error = max (this_error, best_error);
249                 }
250
251                 if (this_error < error) {
252                         error = this_error;
253                         best = *i;
254                 }
255
256                 ++i;
257         }
258
259         if (!best) {
260                 return 24;
261         }
262
263         return best->dcp;
264 }
265
266 DCPTime
267 Playlist::length () const
268 {
269         DCPTime len;
270         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
271                 len = max (len, (*i)->end());
272         }
273
274         return len;
275 }
276
277 void
278 Playlist::reconnect ()
279 {
280         for (list<boost::signals2::connection>::iterator i = _content_connections.begin(); i != _content_connections.end(); ++i) {
281                 i->disconnect ();
282         }
283
284         _content_connections.clear ();
285
286         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
287                 _content_connections.push_back ((*i)->Changed.connect (bind (&Playlist::content_changed, this, _1, _2, _3)));
288         }
289 }
290
291 DCPTime
292 Playlist::video_end () const
293 {
294         DCPTime end;
295         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
296                 if (dynamic_pointer_cast<const VideoContent> (*i)) {
297                         end = max (end, (*i)->end ());
298                 }
299         }
300
301         return end;
302 }
303
304 FrameRateChange
305 Playlist::active_frame_rate_change (DCPTime t, int dcp_video_frame_rate) const
306 {
307         for (ContentList::const_reverse_iterator i = _content.rbegin(); i != _content.rend(); ++i) {
308                 shared_ptr<const VideoContent> vc = dynamic_pointer_cast<const VideoContent> (*i);
309                 if (!vc) {
310                         continue;
311                 }
312
313                 if (vc->position() <= t) {
314                         /* This is the first piece of content (going backwards...) that starts before t,
315                            so it's the active one.
316                         */
317                         return FrameRateChange (vc->video_frame_rate(), dcp_video_frame_rate);
318                 }
319         }
320
321         return FrameRateChange (dcp_video_frame_rate, dcp_video_frame_rate);
322 }
323
324 void
325 Playlist::set_sequence_video (bool s)
326 {
327         _sequence_video = s;
328 }
329
330 bool
331 ContentSorter::operator() (shared_ptr<Content> a, shared_ptr<Content> b)
332 {
333         return a->position() < b->position();
334 }
335
336 /** @return content in an undefined order */
337 ContentList
338 Playlist::content () const
339 {
340         return _content;
341 }
342
343 void
344 Playlist::repeat (ContentList c, int n)
345 {
346         pair<DCPTime, DCPTime> range (DCPTime::max (), DCPTime ());
347         for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
348                 range.first = min (range.first, (*i)->position ());
349                 range.second = max (range.second, (*i)->position ());
350                 range.first = min (range.first, (*i)->end ());
351                 range.second = max (range.second, (*i)->end ());
352         }
353
354         DCPTime pos = range.second;
355         for (int i = 0; i < n; ++i) {
356                 for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
357                         shared_ptr<Content> copy = (*i)->clone ();
358                         copy->set_position (pos + copy->position() - range.first);
359                         _content.push_back (copy);
360                 }
361                 pos += range.second - range.first;
362         }
363
364         sort (_content.begin(), _content.end(), ContentSorter ());
365
366         reconnect ();
367         Changed ();
368 }
369
370 void
371 Playlist::move_earlier (shared_ptr<Content> c)
372 {
373         sort (_content.begin(), _content.end(), ContentSorter ());
374
375         ContentList::iterator previous = _content.end ();
376         ContentList::iterator i = _content.begin();
377         while (i != _content.end() && *i != c) {
378                 previous = i;
379                 ++i;
380         }
381
382         DCPOMATIC_ASSERT (i != _content.end ());
383         if (previous == _content.end ()) {
384                 return;
385         }
386
387
388         DCPTime const p = (*previous)->position ();
389         (*previous)->set_position (p + c->length_after_trim ());
390         c->set_position (p);
391         sort (_content.begin(), _content.end(), ContentSorter ());
392 }
393
394 void
395 Playlist::move_later (shared_ptr<Content> c)
396 {
397         sort (_content.begin(), _content.end(), ContentSorter ());
398
399         ContentList::iterator i = _content.begin();
400         while (i != _content.end() && *i != c) {
401                 ++i;
402         }
403
404         DCPOMATIC_ASSERT (i != _content.end ());
405
406         ContentList::iterator next = i;
407         ++next;
408
409         if (next == _content.end ()) {
410                 return;
411         }
412
413         (*next)->set_position (c->position ());
414         c->set_position (c->position() + (*next)->length_after_trim ());
415         sort (_content.begin(), _content.end(), ContentSorter ());
416 }