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         Frame 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         AVColorRange color_range () const {
  50                 return video_codec_context()->color_range;
  51         }
  52
  53         AVColorPrimaries color_primaries () const {
  54                 return video_codec_context()->color_primaries;
  55         }
  56
  57         AVColorTransferCharacteristic color_trc () const {
  58                 return video_codec_context()->color_trc;
  59         }
  60
  61         AVColorSpace colorspace () const {
  62                 return video_codec_context()->colorspace;
  63         }
  64
  65 private:
  66         void video_packet (AVCodecContext *);
  67         void audio_packet (AVCodecContext *, boost::shared_ptr<FFmpegAudioStream>);
  68         void subtitle_packet (AVCodecContext *, boost::shared_ptr<FFmpegSubtitleStream>);
  69
  70         std::string stream_name (AVStream* s) const;
  71         std::string audio_stream_name (AVStream* s) const;
  72         std::string subtitle_stream_name (AVStream* s) const;
  73         boost::optional<ContentTime> frame_time (AVStream* s) const;
  74
  75         std::vector<boost::shared_ptr<FFmpegSubtitleStream> > _subtitle_streams;
  76         std::vector<boost::shared_ptr<FFmpegAudioStream> > _audio_streams;
  77         boost::optional<ContentTime> _first_video;
  78         /** Video length, either obtained from the header or derived by running
  79          *  through the whole file.
  80          */
  81         Frame _video_length;
  82         bool _need_video_length;
  83
  84         boost::optional<ContentTime> _last_subtitle_start;
  85 };