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                 int frame_finished;
  75
  76                 AVCodecContext* context = _format_context->streams[_packet.stream_index]->codec;
  77
  78                 if (_packet.stream_index == _video_stream && !_first_video) {
  79                         if (avcodec_decode_video2 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
  80                                 _first_video = frame_time (_format_context->streams[_video_stream]);
  81                         }
  82                 } else {
  83                         for (size_t i = 0; i < _audio_streams.size(); ++i) {
  84                                 if (_audio_streams[i]->uses_index (_format_context, _packet.stream_index) && !_audio_streams[i]->first_audio) {
  85                                         if (avcodec_decode_audio4 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
  86                                                 _audio_streams[i]->first_audio = frame_time (_audio_streams[i]->stream (_format_context));
  87                                         }
  88                                 }
  89                         }
  90                 }
  91
  92                 bool have_all_audio = true;
  93                 size_t i = 0;
  94                 while (i < _audio_streams.size() && have_all_audio) {
  95                         have_all_audio = _audio_streams[i]->first_audio;
  96                         ++i;
  97                 }
  98
  99                 av_free_packet (&_packet);
 100
 101                 if (_first_video && have_all_audio) {
 102                         break;
 103                 }
 104         }
 105 }
 106
 107 optional<ContentTime>
 108 FFmpegExaminer::frame_time (AVStream* s) const
 109 {
 110         optional<ContentTime> t;
 111
 112         int64_t const bet = av_frame_get_best_effort_timestamp (_frame);
 113         if (bet != AV_NOPTS_VALUE) {
 114                 t = ContentTime::from_seconds (bet * av_q2d (s->time_base));
 115         }
 116
 117         return t;
 118 }
 119
 120 float
 121 FFmpegExaminer::video_frame_rate () const
 122 {
 123         /* This use of r_frame_rate is debateable; there's a few different
 124          * frame rates in the format context, but this one seems to be the most
 125          * reliable.
 126          */
 127         return av_q2d (av_stream_get_r_frame_rate (_format_context->streams[_video_stream]));
 128 }
 129
 130 dcp::Size
 131 FFmpegExaminer::video_size () const
 132 {
 133         return dcp::Size (video_codec_context()->width, video_codec_context()->height);
 134 }
 135
 136 /** @return Length according to our content's header */
 137 ContentTime
 138 FFmpegExaminer::video_length () const
 139 {
 140         ContentTime const length = ContentTime::from_seconds (double (_format_context->duration) / AV_TIME_BASE);
 141         return ContentTime (max (int64_t (1), length.get ()));
 142 }
 143
 144 string
 145 FFmpegExaminer::audio_stream_name (AVStream* s) const
 146 {
 147         stringstream n;
 148
 149         n << stream_name (s);
 150
 151         if (!n.str().empty()) {
 152                 n << "; ";
 153         }
 154
 155         n << s->codec->channels << " channels";
 156
 157         return n.str ();
 158 }
 159
 160 string
 161 FFmpegExaminer::subtitle_stream_name (AVStream* s) const
 162 {
 163         stringstream n;
 164
 165         n << stream_name (s);
 166
 167         if (n.str().empty()) {
 168                 n << _("unknown");
 169         }
 170
 171         return n.str ();
 172 }
 173
 174 string
 175 FFmpegExaminer::stream_name (AVStream* s) const
 176 {
 177         stringstream n;
 178
 179         if (s->metadata) {
 180                 AVDictionaryEntry const * lang = av_dict_get (s->metadata, "language", 0, 0);
 181                 if (lang) {
 182                         n << lang->value;
 183                 }
 184
 185                 AVDictionaryEntry const * title = av_dict_get (s->metadata, "title", 0, 0);
 186                 if (title) {
 187                         if (!n.str().empty()) {
 188                                 n << " ";
 189                         }
 190                         n << title->value;
 191                 }
 192         }
 193
 194         return n.str ();
 195 }