b873222c112e3d06c77bb989475f2a69f4bc7bbe
[dcpomatic.git] / src / lib / ffmpeg_examiner.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 #include "ffmpeg.h"
21 #include "video_examiner.h"
22 #include <boost/optional.hpp>
23
24 class FFmpegAudioStream;
25 class FFmpegSubtitleStream;
26
27 class FFmpegExaminer : public FFmpeg, public VideoExaminer
28 {
29 public:
30         FFmpegExaminer (boost::shared_ptr<const FFmpegContent>, boost::shared_ptr<Job> job = boost::shared_ptr<Job> ());
31         
32         boost::optional<float> video_frame_rate () const;
33         dcp::Size video_size () const;
34         ContentTime video_length () const;
35         boost::optional<float> sample_aspect_ratio () const;
36
37         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
38                 return _subtitle_streams;
39         }
40         
41         std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
42                 return _audio_streams;
43         }
44
45         boost::optional<ContentTime> first_video () const {
46                 return _first_video;
47         }
48         
49 private:
50         void video_packet (AVCodecContext *);
51         void audio_packet (AVCodecContext *, boost::shared_ptr<FFmpegAudioStream>);
52         void subtitle_packet (AVCodecContext *, boost::shared_ptr<FFmpegSubtitleStream>);
53         
54         std::string stream_name (AVStream* s) const;
55         std::string audio_stream_name (AVStream* s) const;
56         std::string subtitle_stream_name (AVStream* s) const;
57         boost::optional<ContentTime> frame_time (AVStream* s) const;
58         
59         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
60         std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
61         boost::optional<ContentTime> _first_video;
62         /** Video length, either obtained from the header or derived by running
63          *  through the whole file.
64          */
65         ContentTime _video_length;
66         bool _need_video_length;
67 };