Look up unknown subtitle end times from the data prepared by the examiner.
[dcpomatic.git] / src / lib / ffmpeg_content.h
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 #ifndef DCPOMATIC_FFMPEG_CONTENT_H
21 #define DCPOMATIC_FFMPEG_CONTENT_H
22
23 #include "video_content.h"
24 #include "audio_content.h"
25 #include "subtitle_content.h"
26 #include "audio_mapping.h"
27 #include <boost/enable_shared_from_this.hpp>
28 #include <boost/lexical_cast.hpp>
29
30 struct AVFormatContext;
31 struct AVStream;
32
33 class Filter;
34 class FFmpegSubtitleStream;
35 class FFmpegAudioStream;
36 struct ffmpeg_pts_offset_test;
37
38 class FFmpegContentProperty : public VideoContentProperty
39 {
40 public:
41         static int const SUBTITLE_STREAMS;
42         static int const SUBTITLE_STREAM;
43         static int const AUDIO_STREAMS;
44         static int const AUDIO_STREAM;
45         static int const FILTERS;
46 };
47
48 class FFmpegContent : public VideoContent, public AudioContent, public SubtitleContent
49 {
50 public:
51         FFmpegContent (boost::shared_ptr<const Film>, boost::filesystem::path);
52         FFmpegContent (boost::shared_ptr<const Film>, cxml::ConstNodePtr, int version, std::list<std::string> &);
53         FFmpegContent (boost::shared_ptr<const Film>, std::vector<boost::shared_ptr<Content> >);
54
55         boost::shared_ptr<FFmpegContent> shared_from_this () {
56                 return boost::dynamic_pointer_cast<FFmpegContent> (Content::shared_from_this ());
57         }
58         
59         void examine (boost::shared_ptr<Job>);
60         std::string summary () const;
61         std::string technical_summary () const;
62         void as_xml (xmlpp::Node *) const;
63         DCPTime full_length () const;
64
65         std::string identifier () const;
66         
67         /* AudioContent */
68         int audio_channels () const;
69         ContentTime audio_length () const;
70         int audio_frame_rate () const;
71         AudioMapping audio_mapping () const;
72         void set_audio_mapping (AudioMapping);
73         boost::filesystem::path audio_analysis_path () const;
74
75         /* SubtitleContent */
76         bool has_subtitles () const;
77
78         void set_filters (std::vector<Filter const *> const &);
79         
80         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
81                 boost::mutex::scoped_lock lm (_mutex);
82                 return _subtitle_streams;
83         }
84
85         boost::shared_ptr<FFmpegSubtitleStream> subtitle_stream () const {
86                 boost::mutex::scoped_lock lm (_mutex);
87                 return _subtitle_stream;
88         }
89
90         std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
91                 boost::mutex::scoped_lock lm (_mutex);
92                 return _audio_streams;
93         }
94         
95         boost::shared_ptr<FFmpegAudioStream> audio_stream () const {
96                 boost::mutex::scoped_lock lm (_mutex);
97                 return _audio_stream;
98         }
99
100         std::vector<Filter const *> filters () const {
101                 boost::mutex::scoped_lock lm (_mutex);
102                 return _filters;
103         }
104
105         void set_subtitle_stream (boost::shared_ptr<FFmpegSubtitleStream>);
106         void set_audio_stream (boost::shared_ptr<FFmpegAudioStream>);
107
108         boost::optional<ContentTime> first_video () const {
109                 boost::mutex::scoped_lock lm (_mutex);
110                 return _first_video;
111         }
112
113         std::list<ContentTimePeriod> subtitles_during (ContentTimePeriod, bool starting) const;
114
115 private:
116         friend struct ffmpeg_pts_offset_test;
117         
118         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
119         boost::shared_ptr<FFmpegSubtitleStream> _subtitle_stream;
120         std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
121         boost::shared_ptr<FFmpegAudioStream> _audio_stream;
122         boost::optional<ContentTime> _first_video;
123         /** Video filters that should be used when generating DCPs */
124         std::vector<Filter const *> _filters;
125 };
126
127 #endif