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