4e77831ac2ba0faaf409c033c01c245ba2bae607
[dcpomatic.git] / src / lib / content.cc
1 /*
2     Copyright (C) 2013-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 /** @file  src/lib/content.cc
22  *  @brief Content class.
23  */
24
25 #include "content.h"
26 #include "change_signaller.h"
27 #include "util.h"
28 #include "content_factory.h"
29 #include "video_content.h"
30 #include "audio_content.h"
31 #include "text_content.h"
32 #include "exceptions.h"
33 #include "film.h"
34 #include "job.h"
35 #include "compose.hpp"
36 #include <dcp/locale_convert.h>
37 #include <dcp/raw_convert.h>
38 #include <libcxml/cxml.h>
39 #include <libxml++/libxml++.h>
40 #include <boost/thread/mutex.hpp>
41 #include <iostream>
42
43 #include "i18n.h"
44
45 using std::string;
46 using std::list;
47 using std::cout;
48 using std::vector;
49 using std::max;
50 using std::pair;
51 using boost::shared_ptr;
52 using boost::optional;
53 using dcp::raw_convert;
54 using dcp::locale_convert;
55
56 int const ContentProperty::PATH = 400;
57 int const ContentProperty::POSITION = 401;
58 int const ContentProperty::LENGTH = 402;
59 int const ContentProperty::TRIM_START = 403;
60 int const ContentProperty::TRIM_END = 404;
61 int const ContentProperty::VIDEO_FRAME_RATE = 405;
62
63 Content::Content (shared_ptr<const Film> film)
64         : _film (film)
65         , _position (0)
66         , _trim_start (0)
67         , _trim_end (0)
68         , _change_signals_frequent (false)
69 {
70
71 }
72
73 Content::Content (shared_ptr<const Film> film, DCPTime p)
74         : _film (film)
75         , _position (p)
76         , _trim_start (0)
77         , _trim_end (0)
78         , _change_signals_frequent (false)
79 {
80
81 }
82
83 Content::Content (shared_ptr<const Film> film, boost::filesystem::path p)
84         : _film (film)
85         , _position (0)
86         , _trim_start (0)
87         , _trim_end (0)
88         , _change_signals_frequent (false)
89 {
90         add_path (p);
91 }
92
93 Content::Content (shared_ptr<const Film> film, cxml::ConstNodePtr node)
94         : _film (film)
95         , _change_signals_frequent (false)
96 {
97         list<cxml::NodePtr> path_children = node->node_children ("Path");
98         BOOST_FOREACH (cxml::NodePtr i, path_children) {
99                 _paths.push_back (i->content());
100                 optional<time_t> const mod = i->optional_number_child<time_t>("mtime");
101                 if (mod) {
102                         _last_write_times.push_back (*mod);
103                 } else if (boost::filesystem::exists(i->content())) {
104                         _last_write_times.push_back (boost::filesystem::last_write_time(i->content()));
105                 } else {
106                         _last_write_times.push_back (0);
107                 }
108         }
109         _digest = node->optional_string_child ("Digest").get_value_or ("X");
110         _position = DCPTime (node->number_child<DCPTime::Type> ("Position"));
111         _trim_start = ContentTime (node->number_child<ContentTime::Type> ("TrimStart"));
112         _trim_end = ContentTime (node->number_child<ContentTime::Type> ("TrimEnd"));
113         _video_frame_rate = node->optional_number_child<double> ("VideoFrameRate");
114 }
115
116 Content::Content (shared_ptr<const Film> film, vector<shared_ptr<Content> > c)
117         : _film (film)
118         , _position (c.front()->position ())
119         , _trim_start (c.front()->trim_start ())
120         , _trim_end (c.back()->trim_end ())
121         , _video_frame_rate (c.front()->video_frame_rate())
122         , _change_signals_frequent (false)
123 {
124         for (size_t i = 0; i < c.size(); ++i) {
125                 if (i > 0 && c[i]->trim_start() > ContentTime ()) {
126                         throw JoinError (_("Only the first piece of content to be joined can have a start trim."));
127                 }
128
129                 if (i < (c.size() - 1) && c[i]->trim_end () > ContentTime ()) {
130                         throw JoinError (_("Only the last piece of content to be joined can have an end trim."));
131                 }
132
133                 if (
134                         (_video_frame_rate && !c[i]->_video_frame_rate) ||
135                         (!_video_frame_rate && c[i]->_video_frame_rate)
136                         ) {
137                         throw JoinError (_("Content to be joined must have the same video frame rate"));
138                 }
139
140                 if (_video_frame_rate && fabs (_video_frame_rate.get() - c[i]->_video_frame_rate.get()) > VIDEO_FRAME_RATE_EPSILON) {
141                         throw JoinError (_("Content to be joined must have the same video frame rate"));
142                 }
143
144                 for (size_t j = 0; j < c[i]->number_of_paths(); ++j) {
145                         _paths.push_back (c[i]->path(j));
146                         _last_write_times.push_back (c[i]->_last_write_times[j]);
147                 }
148         }
149 }
150
151 void
152 Content::as_xml (xmlpp::Node* node, bool with_paths) const
153 {
154         boost::mutex::scoped_lock lm (_mutex);
155
156         if (with_paths) {
157                 for (size_t i = 0; i < _paths.size(); ++i) {
158                         xmlpp::Element* p = node->add_child("Path");
159                         p->add_child_text (_paths[i].string());
160                         p->set_attribute ("mtime", raw_convert<string>(_last_write_times[i]));
161                 }
162         }
163         node->add_child("Digest")->add_child_text (_digest);
164         node->add_child("Position")->add_child_text (raw_convert<string> (_position.get ()));
165         node->add_child("TrimStart")->add_child_text (raw_convert<string> (_trim_start.get ()));
166         node->add_child("TrimEnd")->add_child_text (raw_convert<string> (_trim_end.get ()));
167         if (_video_frame_rate) {
168                 node->add_child("VideoFrameRate")->add_child_text (raw_convert<string> (_video_frame_rate.get()));
169         }
170 }
171
172 void
173 Content::examine (shared_ptr<Job> job)
174 {
175         if (job) {
176                 job->sub (_("Computing digest"));
177         }
178
179         boost::mutex::scoped_lock lm (_mutex);
180         vector<boost::filesystem::path> p = _paths;
181         lm.unlock ();
182
183         /* Some content files are very big, so we use a poor man's
184            digest here: a digest of the first and last 1e6 bytes with the
185            size of the first file tacked on the end as a string.
186         */
187         string const d = digest_head_tail (p, 1000000) + raw_convert<string> (boost::filesystem::file_size (p.front ()));
188
189         lm.lock ();
190         _digest = d;
191 }
192
193 void
194 Content::signal_change (ChangeType c, int p)
195 {
196         try {
197                 if (c == CHANGE_TYPE_PENDING || c == CHANGE_TYPE_CANCELLED) {
198                         Change (c, shared_from_this(), p, _change_signals_frequent);
199                 } else {
200                         emit (boost::bind (boost::ref(Change), c, shared_from_this(), p, _change_signals_frequent));
201                 }
202         } catch (boost::bad_weak_ptr) {
203                 /* This must be during construction; never mind */
204         }
205 }
206
207 void
208 Content::set_position (DCPTime p)
209 {
210         /* video and audio content can modify its position */
211
212         if (video) {
213                 video->modify_position (p);
214         }
215
216         if (audio) {
217                 audio->modify_position (p);
218         }
219
220         ChangeSignaller<Content> cc (this, ContentProperty::POSITION);
221
222         {
223                 boost::mutex::scoped_lock lm (_mutex);
224                 if (p == _position) {
225                         cc.abort ();
226                         return;
227                 }
228
229                 _position = p;
230         }
231 }
232
233 void
234 Content::set_trim_start (ContentTime t)
235 {
236         /* video and audio content can modify its start trim */
237
238         if (video) {
239                 video->modify_trim_start (t);
240         }
241
242         if (audio) {
243                 audio->modify_trim_start (t);
244         }
245
246         ChangeSignaller<Content> cc (this, ContentProperty::TRIM_START);
247
248         {
249                 boost::mutex::scoped_lock lm (_mutex);
250                 _trim_start = t;
251         }
252 }
253
254 void
255 Content::set_trim_end (ContentTime t)
256 {
257         ChangeSignaller<Content> cc (this, ContentProperty::TRIM_END);
258
259         {
260                 boost::mutex::scoped_lock lm (_mutex);
261                 _trim_end = t;
262         }
263 }
264
265
266 shared_ptr<Content>
267 Content::clone () const
268 {
269         shared_ptr<const Film> film = _film.lock ();
270         if (!film) {
271                 return shared_ptr<Content> ();
272         }
273
274         /* This is a bit naughty, but I can't think of a compelling reason not to do it ... */
275         xmlpp::Document doc;
276         xmlpp::Node* node = doc.create_root_node ("Content");
277         as_xml (node, true);
278
279         /* notes is unused here (we assume) */
280         list<string> notes;
281         return content_factory (film, cxml::NodePtr (new cxml::Node (node)), Film::current_state_version, notes);
282 }
283
284 string
285 Content::technical_summary () const
286 {
287         return String::compose ("%1 %2 %3", path_summary(), digest(), position().seconds());
288 }
289
290 DCPTime
291 Content::length_after_trim () const
292 {
293         return max (DCPTime (), full_length() - DCPTime (trim_start() + trim_end(), film()->active_frame_rate_change (position ())));
294 }
295
296 /** @return string which changes when something about this content changes which affects
297  *  the appearance of its video.
298  */
299 string
300 Content::identifier () const
301 {
302         char buffer[256];
303         snprintf (
304                 buffer, sizeof(buffer), "%s_%" PRId64 "_%" PRId64 "_%" PRId64,
305                 Content::digest().c_str(), position().get(), trim_start().get(), trim_end().get()
306                 );
307         return buffer;
308 }
309
310 bool
311 Content::paths_valid () const
312 {
313         BOOST_FOREACH (boost::filesystem::path i, _paths) {
314                 if (!boost::filesystem::exists (i)) {
315                         return false;
316                 }
317         }
318
319         return true;
320 }
321
322 void
323 Content::set_paths (vector<boost::filesystem::path> paths)
324 {
325         ChangeSignaller<Content> cc (this, ContentProperty::PATH);
326
327         {
328                 boost::mutex::scoped_lock lm (_mutex);
329                 _paths = paths;
330                 _last_write_times.clear ();
331                 BOOST_FOREACH (boost::filesystem::path i, _paths) {
332                         _last_write_times.push_back (boost::filesystem::last_write_time(i));
333                 }
334         }
335 }
336
337 string
338 Content::path_summary () const
339 {
340         /* XXX: should handle multiple paths more gracefully */
341
342         DCPOMATIC_ASSERT (number_of_paths ());
343
344         string s = path(0).filename().string ();
345         if (number_of_paths() > 1) {
346                 s += " ...";
347         }
348
349         return s;
350 }
351
352 /** @return a list of properties that might be interesting to the user */
353 list<UserProperty>
354 Content::user_properties () const
355 {
356         list<UserProperty> p;
357         add_properties (p);
358         return p;
359 }
360
361 shared_ptr<const Film>
362 Content::film () const
363 {
364         shared_ptr<const Film> film = _film.lock ();
365         DCPOMATIC_ASSERT (film);
366         return film;
367 }
368
369 /** @return DCP times of points within this content where a reel split could occur */
370 list<DCPTime>
371 Content::reel_split_points () const
372 {
373         list<DCPTime> t;
374         /* This is only called for video content and such content has its position forced
375            to start on a frame boundary.
376         */
377         t.push_back (position());
378         return t;
379 }
380
381 void
382 Content::set_video_frame_rate (double r)
383 {
384         ChangeSignaller<Content> cc (this, ContentProperty::VIDEO_FRAME_RATE);
385
386         {
387                 boost::mutex::scoped_lock lm (_mutex);
388                 _video_frame_rate = r;
389         }
390
391         /* Make sure trim is still on a frame boundary */
392         if (video) {
393                 set_trim_start (trim_start());
394         }
395 }
396
397 void
398 Content::unset_video_frame_rate ()
399 {
400         ChangeSignaller<Content> cc (this, ContentProperty::VIDEO_FRAME_RATE);
401
402         {
403                 boost::mutex::scoped_lock lm (_mutex);
404                 _video_frame_rate = optional<double>();
405         }
406 }
407
408 double
409 Content::active_video_frame_rate () const
410 {
411         {
412                 boost::mutex::scoped_lock lm (_mutex);
413                 if (_video_frame_rate) {
414                         return _video_frame_rate.get ();
415                 }
416         }
417
418         /* No frame rate specified, so assume this content has been
419            prepared for any concurrent video content or perhaps
420            just the DCP rate.
421         */
422         shared_ptr<const Film> film = _film.lock ();
423         DCPOMATIC_ASSERT (film);
424         return film->active_frame_rate_change(position()).source;
425 }
426
427 void
428 Content::add_properties (list<UserProperty>& p) const
429 {
430         p.push_back (UserProperty (UserProperty::GENERAL, _("Filename"), path(0).string ()));
431
432         if (_video_frame_rate) {
433                 if (video) {
434                         p.push_back (
435                                 UserProperty (
436                                         UserProperty::VIDEO,
437                                         _("Frame rate"),
438                                         locale_convert<string> (_video_frame_rate.get(), 5),
439                                         _("frames per second")
440                                         )
441                                 );
442                 } else {
443                         p.push_back (
444                                 UserProperty (
445                                         UserProperty::GENERAL,
446                                         _("Prepared for video frame rate"),
447                                         locale_convert<string> (_video_frame_rate.get(), 5),
448                                         _("frames per second")
449                                         )
450                                 );
451                 }
452         }
453 }
454
455 /** Take settings from the given content if it is of the correct type */
456 void
457 Content::take_settings_from (shared_ptr<const Content> c)
458 {
459         if (video && c->video) {
460                 video->take_settings_from (c->video);
461         }
462         if (audio && c->audio) {
463                 audio->take_settings_from (c->audio);
464         }
465
466         list<shared_ptr<TextContent> >::iterator i = text.begin ();
467         list<shared_ptr<TextContent> >::const_iterator j = c->text.begin ();
468         while (i != text.end() && j != c->text.end()) {
469                 (*i)->take_settings_from (*j);
470                 ++i;
471                 ++j;
472         }
473 }
474
475 shared_ptr<TextContent>
476 Content::only_text () const
477 {
478         DCPOMATIC_ASSERT (text.size() < 2);
479         if (text.empty ()) {
480                 return shared_ptr<TextContent> ();
481         }
482         return text.front ();
483 }
484
485 shared_ptr<TextContent>
486 Content::text_of_original_type (TextType type) const
487 {
488         BOOST_FOREACH (shared_ptr<TextContent> i, text) {
489                 if (i->original_type() == type) {
490                         return i;
491                 }
492         }
493
494         return shared_ptr<TextContent> ();
495 }
496
497 void
498 Content::add_path (boost::filesystem::path p)
499 {
500         boost::mutex::scoped_lock lm (_mutex);
501         _paths.push_back (p);
502         _last_write_times.push_back (boost::filesystem::last_write_time(p));
503 }