src/lib/ffmpeg_examiner.cc

   1 /*
   2     Copyright (C) 2013 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
  27 using std::string;
  28 using std::cout;
  29 using std::max;
  30 using std::stringstream;
  31 using boost::shared_ptr;
  32 using boost::optional;
  33
  34 FFmpegExaminer::FFmpegExaminer (shared_ptr<const FFmpegContent> c)
  35         : FFmpeg (c)
  36 {
  37         /* Find audio and subtitle streams */
  38
  39         for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
  40                 AVStream* s = _format_context->streams[i];
  41                 if (s->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
  42
  43                         /* This is a hack; sometimes it seems that _audio_codec_context->channel_layout isn't set up,
  44                            so bodge it here.  No idea why we should have to do this.
  45                         */
  46
  47                         if (s->codec->channel_layout == 0) {
  48                                 s->codec->channel_layout = av_get_default_channel_layout (s->codec->channels);
  49                         }
  50
  51                         _audio_streams.push_back (
  52                                 shared_ptr<FFmpegAudioStream> (
  53                                         new FFmpegAudioStream (stream_name (s), i, s->codec->sample_rate, s->codec->channels)
  54                                         )
  55                                 );
  56
  57                 } else if (s->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
  58                         _subtitle_streams.push_back (shared_ptr<FFmpegSubtitleStream> (new FFmpegSubtitleStream (stream_name (s), i)));
  59                 }
  60         }
  61
  62         /* Run through until we find the first audio (for each stream) and video */
  63
  64         while (1) {
  65                 int r = av_read_frame (_format_context, &_packet);
  66                 if (r < 0) {
  67                         break;
  68                 }
  69
  70                 int frame_finished;
  71                 avcodec_get_frame_defaults (_frame);
  72
  73                 AVCodecContext* context = _format_context->streams[_packet.stream_index]->codec;
  74
  75                 if (_packet.stream_index == _video_stream && !_first_video) {
  76                         if (avcodec_decode_video2 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
  77                                 _first_video = frame_time (_video_stream);
  78                         }
  79                 } else {
  80                         for (size_t i = 0; i < _audio_streams.size(); ++i) {
  81                                 if (_packet.stream_index == _audio_streams[i]->id && !_audio_streams[i]->first_audio) {
  82                                         if (avcodec_decode_audio4 (context, _frame, &frame_finished, &_packet) >= 0 && frame_finished) {
  83                                                 _audio_streams[i]->first_audio = frame_time (_audio_streams[i]->id);
  84                                         }
  85                                 }
  86                         }
  87                 }
  88
  89                 bool have_all_audio = true;
  90                 size_t i = 0;
  91                 while (i < _audio_streams.size() && have_all_audio) {
  92                         have_all_audio = _audio_streams[i]->first_audio;
  93                         ++i;
  94                 }
  95
  96                 av_free_packet (&_packet);
  97
  98                 if (_first_video && have_all_audio) {
  99                         break;
 100                 }
 101         }
 102 }
 103
 104 optional<double>
 105 FFmpegExaminer::frame_time (int stream) const
 106 {
 107         optional<double> t;
 108
 109         int64_t const bet = av_frame_get_best_effort_timestamp (_frame);
 110         if (bet != AV_NOPTS_VALUE) {
 111                 t = bet * av_q2d (_format_context->streams[stream]->time_base);
 112         }
 113
 114         return t;
 115 }
 116
 117 float
 118 FFmpegExaminer::video_frame_rate () const
 119 {
 120         AVStream* s = _format_context->streams[_video_stream];
 121
 122         if (s->avg_frame_rate.num && s->avg_frame_rate.den) {
 123                 return av_q2d (s->avg_frame_rate);
 124         }
 125
 126         return av_q2d (s->r_frame_rate);
 127 }
 128
 129 libdcp::Size
 130 FFmpegExaminer::video_size () const
 131 {
 132         return libdcp::Size (video_codec_context()->width, video_codec_context()->height);
 133 }
 134
 135 /** @return Length (in video frames) according to our content's header */
 136 VideoContent::Frame
 137 FFmpegExaminer::video_length () const
 138 {
 139         VideoContent::Frame const length = (double (_format_context->duration) / AV_TIME_BASE) * video_frame_rate();
 140         return max (1, length);
 141 }
 142
 143 string
 144 FFmpegExaminer::stream_name (AVStream* s) const
 145 {
 146         stringstream n;
 147
 148         if (s->metadata) {
 149                 AVDictionaryEntry const * lang = av_dict_get (s->metadata, "language", 0, 0);
 150                 if (lang) {
 151                         n << lang->value;
 152                 }
 153
 154                 AVDictionaryEntry const * title = av_dict_get (s->metadata, "title", 0, 0);
 155                 if (title) {
 156                         if (!n.str().empty()) {
 157                                 n << " ";
 158                         }
 159                         n << title->value;
 160                 }
 161         }
 162
 163         if (!n.str().empty()) {
 164                 n << "; ";
 165         }
 166
 167         n << s->codec->channels << " channels";
 168
 169         return n.str ();
 170 }