Re-work again so that there's just one encoder; various tweaks to still-image-with...
[dcpomatic.git] / src / lib / film.h
1 /*
2     Copyright (C) 2012 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 /** @file  src/film.h
21  *  @brief A representation of a piece of video (with sound), including naming,
22  *  the source content file, and how it should be presented in a DCP.
23  */
24
25 #ifndef DVDOMATIC_FILM_H
26 #define DVDOMATIC_FILM_H
27
28 #include <string>
29 #include <vector>
30 #include <inttypes.h>
31 #include <boost/thread/mutex.hpp>
32 #include <boost/thread.hpp>
33 #include <boost/signals2.hpp>
34 #include <boost/enable_shared_from_this.hpp>
35 extern "C" {
36 #include <libavcodec/avcodec.h>
37 }
38 #include "dcp_content_type.h"
39 #include "util.h"
40 #include "stream.h"
41
42 class Format;
43 class Job;
44 class Filter;
45 class Log;
46 class ExamineContentJob;
47 class ExternalAudioStream;
48
49 /** @class Film
50  *  @brief A representation of a video with sound.
51  *
52  *  A representation of a piece of video (with sound), including naming,
53  *  the source content file, and how it should be presented in a DCP.
54  */
55 class Film : public boost::enable_shared_from_this<Film>
56 {
57 public:
58         Film (std::string d, bool must_exist = true);
59         Film (Film const &);
60         ~Film ();
61
62         std::string j2k_dir () const;
63         std::vector<std::string> audio_files () const;
64
65         void examine_content ();
66         void send_dcp_to_tms ();
67
68         void make_dcp (bool);
69
70         /** @return Logger.
71          *  It is safe to call this from any thread.
72          */
73         Log* log () const {
74                 return _log;
75         }
76
77         int encoded_frames () const;
78         
79         std::string file (std::string f) const;
80         std::string dir (std::string d) const;
81
82         std::string content_path () const;
83         ContentType content_type () const;
84         
85         int target_audio_sample_rate () const;
86         
87         void write_metadata () const;
88         void read_metadata ();
89
90         Size cropped_size (Size) const;
91         boost::optional<int> dcp_length () const;
92         std::string dci_name () const;
93         std::string dcp_name () const;
94
95         bool dirty () const {
96                 return _dirty;
97         }
98
99         int audio_channels () const;
100
101         void set_dci_date_today ();
102
103         enum Property {
104                 NONE,
105                 NAME,
106                 USE_DCI_NAME,
107                 CONTENT,
108                 TRUST_CONTENT_HEADER,
109                 DCP_CONTENT_TYPE,
110                 FORMAT,
111                 CROP,
112                 FILTERS,
113                 SCALER,
114                 DCP_TRIM_START,
115                 DCP_TRIM_END,
116                 DCP_AB,
117                 CONTENT_AUDIO_STREAM,
118                 EXTERNAL_AUDIO,
119                 USE_CONTENT_AUDIO,
120                 AUDIO_GAIN,
121                 AUDIO_DELAY,
122                 STILL_DURATION,
123                 SUBTITLE_STREAM,
124                 WITH_SUBTITLES,
125                 SUBTITLE_OFFSET,
126                 SUBTITLE_SCALE,
127                 DCI_METADATA,
128                 SIZE,
129                 LENGTH,
130                 CONTENT_AUDIO_STREAMS,
131                 SUBTITLE_STREAMS,
132                 FRAMES_PER_SECOND,
133         };
134
135
136         /* GET */
137
138         std::string directory () const {
139                 boost::mutex::scoped_lock lm (_directory_mutex);
140                 return _directory;
141         }
142
143         std::string name () const {
144                 boost::mutex::scoped_lock lm (_state_mutex);
145                 return _name;
146         }
147
148         bool use_dci_name () const {
149                 boost::mutex::scoped_lock lm (_state_mutex);
150                 return _use_dci_name;
151         }
152
153         std::string content () const {
154                 boost::mutex::scoped_lock lm (_state_mutex);
155                 return _content;
156         }
157
158         bool trust_content_header () const {
159                 boost::mutex::scoped_lock lm (_state_mutex);
160                 return _trust_content_header;
161         }
162
163         DCPContentType const * dcp_content_type () const {
164                 boost::mutex::scoped_lock lm (_state_mutex);
165                 return _dcp_content_type;
166         }
167
168         Format const * format () const {
169                 boost::mutex::scoped_lock lm (_state_mutex);
170                 return _format;
171         }
172
173         Crop crop () const {
174                 boost::mutex::scoped_lock lm (_state_mutex);
175                 return _crop;
176         }
177
178         std::vector<Filter const *> filters () const {
179                 boost::mutex::scoped_lock lm (_state_mutex);
180                 return _filters;
181         }
182
183         Scaler const * scaler () const {
184                 boost::mutex::scoped_lock lm (_state_mutex);
185                 return _scaler;
186         }
187
188         SourceFrame dcp_trim_start () const {
189                 boost::mutex::scoped_lock lm (_state_mutex);
190                 return _dcp_trim_start;
191         }
192
193         SourceFrame dcp_trim_end () const {
194                 boost::mutex::scoped_lock lm (_state_mutex);
195                 return _dcp_trim_end;
196         }
197         
198         bool dcp_ab () const {
199                 boost::mutex::scoped_lock lm (_state_mutex);
200                 return _dcp_ab;
201         }
202
203         boost::shared_ptr<AudioStream> content_audio_stream () const {
204                 boost::mutex::scoped_lock lm (_state_mutex);
205                 return _content_audio_stream;
206         }
207
208         std::vector<std::string> external_audio () const {
209                 boost::mutex::scoped_lock lm (_state_mutex);
210                 return _external_audio;
211         }
212
213         bool use_content_audio () const {
214                 boost::mutex::scoped_lock lm (_state_mutex);
215                 return _use_content_audio;
216         }
217         
218         float audio_gain () const {
219                 boost::mutex::scoped_lock lm (_state_mutex);
220                 return _audio_gain;
221         }
222
223         int audio_delay () const {
224                 boost::mutex::scoped_lock lm (_state_mutex);
225                 return _audio_delay;
226         }
227
228         int still_duration () const {
229                 boost::mutex::scoped_lock lm (_state_mutex);
230                 return _still_duration;
231         }
232
233         boost::shared_ptr<SubtitleStream> subtitle_stream () const {
234                 boost::mutex::scoped_lock lm (_state_mutex);
235                 return _subtitle_stream;
236         }
237
238         bool with_subtitles () const {
239                 boost::mutex::scoped_lock lm (_state_mutex);
240                 return _with_subtitles;
241         }
242
243         int subtitle_offset () const {
244                 boost::mutex::scoped_lock lm (_state_mutex);
245                 return _subtitle_offset;
246         }
247
248         float subtitle_scale () const {
249                 boost::mutex::scoped_lock lm (_state_mutex);
250                 return _subtitle_scale;
251         }
252
253         std::string audio_language () const {
254                 boost::mutex::scoped_lock lm (_state_mutex);
255                 return _audio_language;
256         }
257         
258         std::string subtitle_language () const {
259                 boost::mutex::scoped_lock lm (_state_mutex);
260                 return _subtitle_language;
261         }
262         
263         std::string territory () const {
264                 boost::mutex::scoped_lock lm (_state_mutex);
265                 return _territory;
266         }
267         
268         std::string rating () const {
269                 boost::mutex::scoped_lock lm (_state_mutex);
270                 return _rating;
271         }
272         
273         std::string studio () const {
274                 boost::mutex::scoped_lock lm (_state_mutex);
275                 return _studio;
276         }
277         
278         std::string facility () const {
279                 boost::mutex::scoped_lock lm (_state_mutex);
280                 return _facility;
281         }
282         
283         std::string package_type () const {
284                 boost::mutex::scoped_lock lm (_state_mutex);
285                 return _package_type;
286         }
287
288         Size size () const {
289                 boost::mutex::scoped_lock lm (_state_mutex);
290                 return _size;
291         }
292
293         boost::optional<SourceFrame> length () const {
294                 boost::mutex::scoped_lock lm (_state_mutex);
295                 return _length;
296         }
297         
298         std::string content_digest () const {
299                 boost::mutex::scoped_lock lm (_state_mutex);
300                 return _content_digest;
301         }
302         
303         std::vector<boost::shared_ptr<AudioStream> > content_audio_streams () const {
304                 boost::mutex::scoped_lock lm (_state_mutex);
305                 return _content_audio_streams;
306         }
307
308         std::vector<boost::shared_ptr<SubtitleStream> > subtitle_streams () const {
309                 boost::mutex::scoped_lock lm (_state_mutex);
310                 return _subtitle_streams;
311         }
312         
313         float frames_per_second () const {
314                 boost::mutex::scoped_lock lm (_state_mutex);
315                 if (content_type() == STILL) {
316                         return 24;
317                 }
318                 
319                 return _frames_per_second;
320         }
321
322         boost::shared_ptr<AudioStream> audio_stream () const;
323
324         
325         /* SET */
326
327         void set_directory (std::string);
328         void set_name (std::string);
329         void set_use_dci_name (bool);
330         void set_content (std::string);
331         void set_trust_content_header (bool);
332         void set_dcp_content_type (DCPContentType const *);
333         void set_format (Format const *);
334         void set_crop (Crop);
335         void set_left_crop (int);
336         void set_right_crop (int);
337         void set_top_crop (int);
338         void set_bottom_crop (int);
339         void set_filters (std::vector<Filter const *>);
340         void set_scaler (Scaler const *);
341         void set_dcp_trim_start (int);
342         void set_dcp_trim_end (int);
343         void set_dcp_ab (bool);
344         void set_content_audio_stream (boost::shared_ptr<AudioStream>);
345         void set_external_audio (std::vector<std::string>);
346         void set_use_content_audio (bool);
347         void set_audio_gain (float);
348         void set_audio_delay (int);
349         void set_still_duration (int);
350         void set_subtitle_stream (boost::shared_ptr<SubtitleStream>);
351         void set_with_subtitles (bool);
352         void set_subtitle_offset (int);
353         void set_subtitle_scale (float);
354         void set_audio_language (std::string);
355         void set_subtitle_language (std::string);
356         void set_territory (std::string);
357         void set_rating (std::string);
358         void set_studio (std::string);
359         void set_facility (std::string);
360         void set_package_type (std::string);
361         void set_size (Size);
362         void set_length (SourceFrame);
363         void unset_length ();
364         void set_content_digest (std::string);
365         void set_content_audio_streams (std::vector<boost::shared_ptr<AudioStream> >);
366         void set_subtitle_streams (std::vector<boost::shared_ptr<SubtitleStream> >);
367         void set_frames_per_second (float);
368
369         /** Emitted when some property has changed */
370         mutable boost::signals2::signal<void (Property)> Changed;
371
372         /** Current version number of the state file */
373         static int const state_version;
374
375 private:
376         
377         /** Log to write to */
378         Log* _log;
379
380         /** Any running ExamineContentJob, or 0 */
381         boost::shared_ptr<ExamineContentJob> _examine_content_job;
382
383         /** The date that we should use in a DCI name */
384         boost::gregorian::date _dci_date;
385
386         void signal_changed (Property);
387         void examine_content_finished ();
388
389         /** Complete path to directory containing the film metadata;
390          *  must not be relative.
391          */
392         std::string _directory;
393         /** Mutex for _directory */
394         mutable boost::mutex _directory_mutex;
395         
396         /** Name for DVD-o-matic */
397         std::string _name;
398         /** True if a auto-generated DCI-compliant name should be used for our DCP */
399         bool _use_dci_name;
400         /** File or directory containing content; may be relative to our directory
401          *  or an absolute path.
402          */
403         std::string _content;
404         bool _trust_content_header;
405         /** The type of content that this Film represents (feature, trailer etc.) */
406         DCPContentType const * _dcp_content_type;
407         /** The format to present this Film in (flat, scope, etc.) */
408         Format const * _format;
409         /** The crop to apply to the source */
410         Crop _crop;
411         /** Video filters that should be used when generating DCPs */
412         std::vector<Filter const *> _filters;
413         /** Scaler algorithm to use */
414         Scaler const * _scaler;
415         /** Frames to trim off the start of the DCP */
416         int _dcp_trim_start;
417         /** Frames to trim off the end of the DCP */
418         int _dcp_trim_end;
419         /** true to create an A/B comparison DCP, where the left half of the image
420             is the video without any filters or post-processing, and the right half
421             has the specified filters and post-processing.
422         */
423         bool _dcp_ab;
424         /** The audio stream to use from our content */
425         boost::shared_ptr<AudioStream> _content_audio_stream;
426         /** List of filenames of external audio files, in channel order
427             (L, R, C, Lfe, Ls, Rs)
428         */
429         std::vector<std::string> _external_audio;
430         /** true to use audio from our content file; false to use external audio */
431         bool _use_content_audio;
432         /** Gain to apply to audio in dB */
433         float _audio_gain;
434         /** Delay to apply to audio (positive moves audio later) in milliseconds */
435         int _audio_delay;
436         /** Duration to make still-sourced films (in seconds) */
437         int _still_duration;
438         boost::shared_ptr<SubtitleStream> _subtitle_stream;
439         /** True if subtitles should be shown for this film */
440         bool _with_subtitles;
441         /** y offset for placing subtitles, in source pixels; +ve is further down
442             the frame, -ve is further up.
443         */
444         int _subtitle_offset;
445         /** scale factor to apply to subtitles */
446         float _subtitle_scale;
447
448         /* DCI naming stuff */
449         std::string _audio_language;
450         std::string _subtitle_language;
451         std::string _territory;
452         std::string _rating;
453         std::string _studio;
454         std::string _facility;
455         std::string _package_type;
456
457         /* Data which are cached to speed things up */
458
459         /** Size, in pixels, of the source (ignoring cropping) */
460         Size _size;
461         /** Actual length of the source (in video frames) from examining it */
462         boost::optional<SourceFrame> _length;
463         /** MD5 digest of our content file */
464         std::string _content_digest;
465         /** The audio streams in our content */
466         std::vector<boost::shared_ptr<AudioStream> > _content_audio_streams;
467         /** A stream to represent possible external audio (will always exist) */
468         boost::shared_ptr<AudioStream> _external_audio_stream;
469         /** the subtitle streams that we can use */
470         std::vector<boost::shared_ptr<SubtitleStream> > _subtitle_streams;
471         /** Frames per second of the source */
472         float _frames_per_second;
473
474         /** true if our state has changed since we last saved it */
475         mutable bool _dirty;
476
477         /** Mutex for all state except _directory */
478         mutable boost::mutex _state_mutex;
479
480         friend class paths_test;
481 };
482
483 #endif