src/lib/ffmpeg_examiner.cc

   1 /*
   2     Copyright (C) 2013-2014 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 extern "C" {
  21 #include <libavcodec/avcodec.h>
  22 #include <libavformat/avformat.h>
  23 }
  24 #include "ffmpeg_examiner.h"
  25 #include "ffmpeg_content.h"
  26 #include "ffmpeg_audio_stream.h"
  27 #include "ffmpeg_subtitle_stream.h"
  28
  29 #include "i18n.h"
  30
  31 using std::string;
  32 using std::cout;
  33 using std::max;
  34 using std::stringstream;
  35 using boost::shared_ptr;
  36 using boost::optional;
  37
  38 FFmpegExaminer::FFmpegExaminer (shared_ptr<const FFmpegContent> c)
  39         : FFmpeg (c)
  40 {
  41         /* Find audio and subtitle streams */
  42
  43         for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
  44                 AVStream* s = _format_context->streams[i];
  45                 if (s->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
  46
  47                         /* This is a hack; sometimes it seems that _audio_codec_context->channel_layout isn't set up,
  48                            so bodge it here.  No idea why we should have to do this.
  49                         */
  50
  51                         if (s->codec->channel_layout == 0) {
  52                                 s->codec->channel_layout = av_get_default_channel_layout (s->codec->channels);
  53                         }
  54
  55                         _audio_streams.push_back (
  56                                 shared_ptr<FFmpegAudioStream> (
  57                                         new FFmpegAudioStream (audio_stream_name (s), s->id, s->codec->sample_rate, s->codec->channels)
  58                                         )
  59                                 );
  60
  61                 } else if (s->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
  62                         _subtitle_streams.push_back (shared_ptr<FFmpegSubtitleStream> (new FFmpegSubtitleStream (subtitle_stream_name (s), s->id)));
  63                 }
  64         }
  65
  66         /* Run through until we find the first audio (for each stream) and video */
  67
  68         while (1) {
  69                 int r = av_read_frame (_format_context, &_packet);
  70                 if (r < 0) {
  71                         break;
  72                 }
  73
  74                 AVCodecContext* context = _format_context->streams[_packet.stream_index]->codec;
  75
  76                 if (_packet.stream_index == _video_stream) {
  77                         video_packet (context);
  78                 }
  79
  80                 for (size_t i = 0; i < _audio_streams.size(); ++i) {
  81                         if (_audio_streams[i]->uses_index (_format_context, _packet.stream_index)) {
  82                                 audio_packet (context, _audio_streams[i]);
  83                         }
  84                 }
  85
  86                 for (size_t i = 0; i < _subtitle_streams.size(); ++i) {
  87                         if (_subtitle_streams[i]->uses_index (_format_context, _packet.stream_index)) {
  88                                 subtitle_packet (context, _subtitle_streams[i]);
  89                         }
  90                 }
  91
  92                 av_free_packet (&_packet);
  93         }
  94 }
  95
  96 void
  97 FFmpegExaminer::video_packet (AVCodecContext* context)
  98 {
  99         if (_first_video) {
 100                 return;
 101         }
 102
 103         int frame_finished;
 104         if (avcodec_decode_video2 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
 105                 _first_video = frame_time (_format_context->streams[_video_stream]);
 106         }
 107 }
 108
 109 void
 110 FFmpegExaminer::audio_packet (AVCodecContext* context, shared_ptr<FFmpegAudioStream> stream)
 111 {
 112         if (stream->first_audio) {
 113                 return;
 114         }
 115
 116         int frame_finished;
 117         if (avcodec_decode_audio4 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
 118                 stream->first_audio = frame_time (stream->stream (_format_context));
 119         }
 120 }
 121
 122 void
 123 FFmpegExaminer::subtitle_packet (AVCodecContext* context, shared_ptr<FFmpegSubtitleStream> stream)
 124 {
 125
 126 }
 127
 128 optional<ContentTime>
 129 FFmpegExaminer::frame_time (AVStream* s) const
 130 {
 131         optional<ContentTime> t;
 132
 133         int64_t const bet = av_frame_get_best_effort_timestamp (_frame);
 134         if (bet != AV_NOPTS_VALUE) {
 135                 t = ContentTime::from_seconds (bet * av_q2d (s->time_base));
 136         }
 137
 138         return t;
 139 }
 140
 141 float
 142 FFmpegExaminer::video_frame_rate () const
 143 {
 144         /* This use of r_frame_rate is debateable; there's a few different
 145          * frame rates in the format context, but this one seems to be the most
 146          * reliable.
 147          */
 148         return av_q2d (av_stream_get_r_frame_rate (_format_context->streams[_video_stream]));
 149 }
 150
 151 dcp::Size
 152 FFmpegExaminer::video_size () const
 153 {
 154         return dcp::Size (video_codec_context()->width, video_codec_context()->height);
 155 }
 156
 157 /** @return Length according to our content's header */
 158 ContentTime
 159 FFmpegExaminer::video_length () const
 160 {
 161         ContentTime const length = ContentTime::from_seconds (double (_format_context->duration) / AV_TIME_BASE);
 162         return ContentTime (max (int64_t (1), length.get ()));
 163 }
 164
 165 string
 166 FFmpegExaminer::audio_stream_name (AVStream* s) const
 167 {
 168         stringstream n;
 169
 170         n << stream_name (s);
 171
 172         if (!n.str().empty()) {
 173                 n << "; ";
 174         }
 175
 176         n << s->codec->channels << " channels";
 177
 178         return n.str ();
 179 }
 180
 181 string
 182 FFmpegExaminer::subtitle_stream_name (AVStream* s) const
 183 {
 184         stringstream n;
 185
 186         n << stream_name (s);
 187
 188         if (n.str().empty()) {
 189                 n << _("unknown");
 190         }
 191
 192         return n.str ();
 193 }
 194
 195 string
 196 FFmpegExaminer::stream_name (AVStream* s) const
 197 {
 198         stringstream n;
 199
 200         if (s->metadata) {
 201                 AVDictionaryEntry const * lang = av_dict_get (s->metadata, "language", 0, 0);
 202                 if (lang) {
 203                         n << lang->value;
 204                 }
 205
 206                 AVDictionaryEntry const * title = av_dict_get (s->metadata, "title", 0, 0);
 207                 if (title) {
 208                         if (!n.str().empty()) {
 209                                 n << " ";
 210                         }
 211                         n << title->value;
 212                 }
 213         }
 214
 215         return n.str ();
 216 }