Hand-apply d849d411cff28ef5453085791d0b4d7cd73bd070 from master; replace all assert...
[dcpomatic.git] / src / lib / playlist.cc
1 /*
2     Copyright (C) 2013 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 <libcxml/cxml.h>
21 #include <boost/shared_ptr.hpp>
22 #include "playlist.h"
23 #include "sndfile_content.h"
24 #include "sndfile_decoder.h"
25 #include "video_content.h"
26 #include "ffmpeg_decoder.h"
27 #include "ffmpeg_content.h"
28 #include "image_decoder.h"
29 #include "content_factory.h"
30 #include "job.h"
31 #include "config.h"
32 #include "util.h"
33 #include "md5_digester.h"
34
35 #include "i18n.h"
36
37 using std::list;
38 using std::cout;
39 using std::vector;
40 using std::min;
41 using std::max;
42 using std::string;
43 using std::pair;
44 using boost::optional;
45 using boost::shared_ptr;
46 using boost::weak_ptr;
47 using boost::dynamic_pointer_cast;
48
49 Playlist::Playlist ()
50         : _sequence_video (true)
51         , _sequencing_video (false)
52 {
53
54 }
55
56 Playlist::~Playlist ()
57 {
58         _content.clear ();
59         reconnect ();
60 }
61
62 void
63 Playlist::content_changed (weak_ptr<Content> content, int property, bool frequent)
64 {
65         /* Don't respond to position changes here, as:
66            - sequencing after earlier/later changes is handled by move_earlier/move_later
67            - any other position changes will be timeline drags which should not result in content
68            being sequenced.
69         */
70         
71         if (property == ContentProperty::LENGTH || property == VideoContentProperty::VIDEO_FRAME_TYPE) {
72                 maybe_sequence_video ();
73         }
74         
75         ContentChanged (content, property, frequent);
76 }
77
78 void
79 Playlist::maybe_sequence_video ()
80 {
81         if (!_sequence_video || _sequencing_video) {
82                 return;
83         }
84         
85         _sequencing_video = true;
86         
87         ContentList cl = _content;
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                 if (vc) {
118                         t += vc->identifier ();
119                 }
120         }
121
122         MD5Digester digester;
123         digester.add (t.c_str(), t.length());
124         return digester.get ();
125 }
126
127 /** @param node <Playlist> node */
128 void
129 Playlist::set_from_xml (shared_ptr<const Film> film, cxml::ConstNodePtr node, int version, list<string>& notes)
130 {
131         list<cxml::NodePtr> c = node->node_children ("Content");
132         for (list<cxml::NodePtr>::iterator i = c.begin(); i != c.end(); ++i) {
133                 _content.push_back (content_factory (film, *i, version, notes));
134         }
135
136         sort (_content.begin(), _content.end(), ContentSorter ());
137
138         reconnect ();
139 }
140
141 /** @param node <Playlist> node */
142 void
143 Playlist::as_xml (xmlpp::Node* node)
144 {
145         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
146                 (*i)->as_xml (node->add_child ("Content"));
147         }
148 }
149
150 void
151 Playlist::add (shared_ptr<Content> c)
152 {
153         _content.push_back (c);
154         sort (_content.begin(), _content.end(), ContentSorter ());
155         reconnect ();
156         Changed ();
157 }
158
159 void
160 Playlist::remove (shared_ptr<Content> c)
161 {
162         ContentList::iterator i = _content.begin ();
163         while (i != _content.end() && *i != c) {
164                 ++i;
165         }
166         
167         if (i != _content.end ()) {
168                 _content.erase (i);
169                 Changed ();
170         }
171
172         /* This won't change order, so it does not need a sort */
173 }
174
175 void
176 Playlist::remove (ContentList c)
177 {
178         for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
179                 ContentList::iterator j = _content.begin ();
180                 while (j != _content.end() && *j != *i) {
181                         ++j;
182                 }
183         
184                 if (j != _content.end ()) {
185                         _content.erase (j);
186                 }
187         }
188
189         /* This won't change order, so it does not need a sort */
190         
191         Changed ();
192 }
193
194 class FrameRateCandidate
195 {
196 public:
197         FrameRateCandidate (float source_, int dcp_)
198                 : source (source_)
199                 , dcp (dcp_)
200         {}
201
202         float source;
203         int dcp;
204 };
205
206 int
207 Playlist::best_dcp_frame_rate () const
208 {
209         list<int> const allowed_dcp_frame_rates = Config::instance()->allowed_dcp_frame_rates ();
210
211         /* Work out what rates we could manage, including those achieved by using skip / repeat. */
212         list<FrameRateCandidate> candidates;
213
214         /* Start with the ones without skip / repeat so they will get matched in preference to skipped/repeated ones */
215         for (list<int>::const_iterator i = allowed_dcp_frame_rates.begin(); i != allowed_dcp_frame_rates.end(); ++i) {
216                 candidates.push_back (FrameRateCandidate (*i, *i));
217         }
218
219         /* Then the skip/repeat ones */
220         for (list<int>::const_iterator i = allowed_dcp_frame_rates.begin(); i != allowed_dcp_frame_rates.end(); ++i) {
221                 candidates.push_back (FrameRateCandidate (float (*i) / 2, *i));
222                 candidates.push_back (FrameRateCandidate (float (*i) * 2, *i));
223         }
224
225         /* Pick the best one */
226         float error = std::numeric_limits<float>::max ();
227         optional<FrameRateCandidate> best;
228         list<FrameRateCandidate>::iterator i = candidates.begin();
229         while (i != candidates.end()) {
230
231                 float this_error = 0;
232                 for (ContentList::const_iterator j = _content.begin(); j != _content.end(); ++j) {
233                         shared_ptr<VideoContent> vc = dynamic_pointer_cast<VideoContent> (*j);
234                         if (!vc) {
235                                 continue;
236                         }
237
238                         /* Use the largest difference between DCP and source as the "error" */
239                         this_error = max (this_error, float (fabs (i->source - vc->video_frame_rate ())));
240                 }
241
242                 if (this_error < error) {
243                         error = this_error;
244                         best = *i;
245                 }
246
247                 ++i;
248         }
249
250         if (!best) {
251                 return 24;
252         }
253         
254         return best->dcp;
255 }
256
257 DCPTime
258 Playlist::length () const
259 {
260         DCPTime len;
261         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
262                 len = max (len, (*i)->end() + DCPTime::delta ());
263         }
264
265         return len;
266 }
267
268 void
269 Playlist::reconnect ()
270 {
271         for (list<boost::signals2::connection>::iterator i = _content_connections.begin(); i != _content_connections.end(); ++i) {
272                 i->disconnect ();
273         }
274
275         _content_connections.clear ();
276                 
277         for (ContentList::iterator i = _content.begin(); i != _content.end(); ++i) {
278                 _content_connections.push_back ((*i)->Changed.connect (bind (&Playlist::content_changed, this, _1, _2, _3)));
279         }
280 }
281
282 DCPTime
283 Playlist::video_end () const
284 {
285         DCPTime end;
286         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
287                 if (dynamic_pointer_cast<const VideoContent> (*i)) {
288                         end = max (end, (*i)->end ());
289                 }
290         }
291
292         return end;
293 }
294
295 FrameRateChange
296 Playlist::active_frame_rate_change (DCPTime t, int dcp_video_frame_rate) const
297 {
298         for (ContentList::const_iterator i = _content.begin(); i != _content.end(); ++i) {
299                 shared_ptr<const VideoContent> vc = dynamic_pointer_cast<const VideoContent> (*i);
300                 if (!vc) {
301                         continue;
302                 }
303
304                 if (vc->position() >= t && t < vc->end()) {
305                         return FrameRateChange (vc->video_frame_rate(), dcp_video_frame_rate);
306                 }
307         }
308
309         return FrameRateChange (dcp_video_frame_rate, dcp_video_frame_rate);
310 }
311
312 void
313 Playlist::set_sequence_video (bool s)
314 {
315         _sequence_video = s;
316 }
317
318 bool
319 ContentSorter::operator() (shared_ptr<Content> a, shared_ptr<Content> b)
320 {
321         return a->position() < b->position();
322 }
323
324 /** @return content in an undefined order */
325 ContentList
326 Playlist::content () const
327 {
328         return _content;
329 }
330
331 void
332 Playlist::repeat (ContentList c, int n)
333 {
334         pair<DCPTime, DCPTime> range (DCPTime::max (), DCPTime ());
335         for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
336                 range.first = min (range.first, (*i)->position ());
337                 range.second = max (range.second, (*i)->position ());
338                 range.first = min (range.first, (*i)->end ());
339                 range.second = max (range.second, (*i)->end ());
340         }
341
342         DCPTime pos = range.second;
343         for (int i = 0; i < n; ++i) {
344                 for (ContentList::iterator i = c.begin(); i != c.end(); ++i) {
345                         shared_ptr<Content> copy = (*i)->clone ();
346                         copy->set_position (pos + copy->position() - range.first);
347                         _content.push_back (copy);
348                 }
349                 pos += range.second - range.first;
350         }
351
352         sort (_content.begin(), _content.end(), ContentSorter ());
353         
354         reconnect ();
355         Changed ();
356 }
357
358 void
359 Playlist::move_earlier (shared_ptr<Content> c)
360 {
361         sort (_content.begin(), _content.end(), ContentSorter ());
362         
363         ContentList::iterator previous = _content.end ();
364         ContentList::iterator i = _content.begin();
365         while (i != _content.end() && *i != c) {
366                 previous = i;
367                 ++i;
368         }
369
370         DCPOMATIC_ASSERT (i != _content.end ());
371         if (previous == _content.end ()) {
372                 return;
373         }
374
375         
376         DCPTime const p = (*previous)->position ();
377         (*previous)->set_position (p + c->length_after_trim ());
378         c->set_position (p);
379         sort (_content.begin(), _content.end(), ContentSorter ());
380 }
381
382 void
383 Playlist::move_later (shared_ptr<Content> c)
384 {
385         sort (_content.begin(), _content.end(), ContentSorter ());
386         
387         ContentList::iterator i = _content.begin();
388         while (i != _content.end() && *i != c) {
389                 ++i;
390         }
391
392         DCPOMATIC_ASSERT (i != _content.end ());
393
394         ContentList::iterator next = i;
395         ++next;
396
397         if (next == _content.end ()) {
398                 return;
399         }
400
401         (*next)->set_position (c->position ());
402         c->set_position (c->position() + (*next)->length_after_trim ());
403         sort (_content.begin(), _content.end(), ContentSorter ());
404 }