X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fffmpeg_examiner.h;h=a511a2fc1fd20360eab70dbbffaeed4f7efa19bc;hb=ba8a5a15cc27988e2bbc6acd470d8532f1d8e99f;hp=1ef7e3ba90f00f6dc04dfa2c9fe029e1a108fe87;hpb=cf8e0e9e26dbaefab768000ebc1655b34ad11105;p=dcpomatic.git

diff --git a/src/lib/ffmpeg_examiner.h b/src/lib/ffmpeg_examiner.h
index 1ef7e3ba9..a511a2fc1 100644
--- a/src/lib/ffmpeg_examiner.h
+++ b/src/lib/ffmpeg_examiner.h
@@ -1,19 +1,20 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2017 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
     the Free Software Foundation; either version 2 of the License, or
     (at your option) any later version.
 
-    This program is distributed in the hope that it will be useful,
+    DCP-o-matic is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     GNU General Public License for more details.
 
     You should have received a copy of the GNU General Public License
-    along with this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
@@ -21,23 +22,29 @@
 #include "video_examiner.h"
 #include <boost/optional.hpp>
 
+struct AVStream;
+
 class FFmpegAudioStream;
 class FFmpegSubtitleStream;
+class Job;
 
 class FFmpegExaminer : public FFmpeg, public VideoExaminer
 {
 public:
-	FFmpegExaminer (boost::shared_ptr<const FFmpegContent>);
-	
-	boost::optional<float> video_frame_rate () const;
+	FFmpegExaminer (boost::shared_ptr<const FFmpegContent>, boost::shared_ptr<Job> job = boost::shared_ptr<Job> ());
+
+	bool has_video () const;
+
+	boost::optional<double> video_frame_rate () const;
 	dcp::Size video_size () const;
-	ContentTime video_length () const;
-	boost::optional<float> sample_aspect_ratio () const;
+	Frame video_length () const;
+	boost::optional<double> sample_aspect_ratio () const;
+	bool yuv () const;
 
 	std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
 		return _subtitle_streams;
 	}
-	
+
 	std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
 		return _audio_streams;
 	}
@@ -45,18 +52,56 @@ public:
 	boost::optional<ContentTime> first_video () const {
 		return _first_video;
 	}
-	
+
+	AVColorRange color_range () const {
+		return video_codec_context()->color_range;
+	}
+
+	AVColorPrimaries color_primaries () const {
+		return video_codec_context()->color_primaries;
+	}
+
+	AVColorTransferCharacteristic color_trc () const {
+		return video_codec_context()->color_trc;
+	}
+
+	AVColorSpace colorspace () const {
+		return video_codec_context()->colorspace;
+	}
+
+	int bits_per_pixel () const;
+
 private:
 	void video_packet (AVCodecContext *);
 	void audio_packet (AVCodecContext *, boost::shared_ptr<FFmpegAudioStream>);
-	void subtitle_packet (AVCodecContext *, boost::shared_ptr<FFmpegSubtitleStream>);
-	
+
 	std::string stream_name (AVStream* s) const;
-	std::string audio_stream_name (AVStream* s) const;
 	std::string subtitle_stream_name (AVStream* s) const;
 	boost::optional<ContentTime> frame_time (AVStream* s) const;
-	
+
 	std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
 	std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
 	boost::optional<ContentTime> _first_video;
+	/** Video length, either obtained from the header or derived by running
+	 *  through the whole file.
+	 */
+	Frame _video_length;
+	bool _need_video_length;
+
+	struct SubtitleStart
+	{
+		SubtitleStart (std::string id_, bool image_, ContentTime time_)
+			: id (id_)
+			, image (image_)
+			, time (time_)
+		{}
+
+		std::string id;
+		/** true if it's an image subtitle, false for text */
+		bool image;
+		ContentTime time;
+	};
+
+	typedef std::map<boost::shared_ptr<FFmpegSubtitleStream>, boost::optional<SubtitleStart> > LastSubtitleMap;
+	LastSubtitleMap _last_subtitle_start;
 };