X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Fffmpeg_examiner.h;h=9ff5a79e7a6d79614c2812aa99dc1e0d6aa68c5b;hp=8c31eb2c4aedadae419e7c5e41b83841f848c93b;hb=HEAD;hpb=3574212ee42b2bd924eb95d5c0f4f69ec9e0a2f0

diff --git a/src/lib/ffmpeg_examiner.h b/src/lib/ffmpeg_examiner.h
index 8c31eb2c4..45313ec18 100644
--- a/src/lib/ffmpeg_examiner.h
+++ b/src/lib/ffmpeg_examiner.h
@@ -1,61 +1,126 @@
 /*
-    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/>.
 
 */
 
-#include <boost/optional.hpp>
+
 #include "ffmpeg.h"
 #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>);
-	
-	float video_frame_rate () const;
-	dcp::Size video_size () const;
-	ContentTime video_length () const;
+	FFmpegExaminer (std::shared_ptr<const FFmpegContent>, std::shared_ptr<Job> job = std::shared_ptr<Job>());
+
+	bool has_video () const override;
+
+	boost::optional<double> video_frame_rate () const override;
+	boost::optional<dcp::Size> video_size() const override;
+	Frame video_length () const override;
+	boost::optional<double> sample_aspect_ratio () const override;
+	bool yuv () const override;
 
-	std::vector<boost::shared_ptr<FFmpegSubtitleStream> > subtitle_streams () const {
+	std::vector<std::shared_ptr<FFmpegSubtitleStream>> subtitle_streams () const {
 		return _subtitle_streams;
 	}
-	
-	std::vector<boost::shared_ptr<FFmpegAudioStream> > audio_streams () const {
+
+	std::vector<std::shared_ptr<FFmpegAudioStream>> audio_streams () const {
 		return _audio_streams;
 	}
 
-	boost::optional<ContentTime> first_video () const {
+	boost::optional<dcpomatic::ContentTime> first_video () const {
 		return _first_video;
 	}
-	
+
+	VideoRange range () const override;
+
+	PixelQuanta pixel_quanta () const override;
+
+	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;
+	}
+
+	boost::optional<int> bits_per_pixel () const;
+
+	bool has_alpha() const;
+
+	boost::optional<double> rotation () const {
+		return _rotation;
+	}
+
+	bool pulldown () const {
+		return _pulldown;
+	}
+
 private:
-	void video_packet (AVCodecContext *);
-	void audio_packet (AVCodecContext *, boost::shared_ptr<FFmpegAudioStream>);
-	void subtitle_packet (AVCodecContext *, boost::shared_ptr<FFmpegSubtitleStream>);
-	
+	bool video_packet (AVCodecContext* context, std::string& temporal_reference, AVPacket* packet);
+	void audio_packet (AVCodecContext* context, std::shared_ptr<FFmpegAudioStream>, AVPacket* packet);
+
 	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;
+	boost::optional<dcpomatic::ContentTime> frame_time (AVFrame* frame, AVStream* stream) const;
+
+	std::vector<std::shared_ptr<FFmpegSubtitleStream>> _subtitle_streams;
+	std::vector<std::shared_ptr<FFmpegAudioStream>> _audio_streams;
+	boost::optional<dcpomatic::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;
+
+	boost::optional<double> _rotation;
+	bool _pulldown;
+
+	struct SubtitleStart
+	{
+		SubtitleStart (std::string id_, bool image_, dcpomatic::ContentTime time_)
+			: id (id_)
+			, image (image_)
+			, time (time_)
+		{}
+
+		std::string id;
+		/** true if it's an image subtitle, false for text */
+		bool image;
+		dcpomatic::ContentTime time;
+	};
+
+	typedef std::map<std::shared_ptr<FFmpegSubtitleStream>, boost::optional<SubtitleStart>> LastSubtitleMap;
+	LastSubtitleMap _last_subtitle_start;
 };