2 Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
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.
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.
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.
21 #include <libavcodec/avcodec.h>
22 #include <libavformat/avformat.h>
23 #include <libswscale/swscale.h>
24 #include <libpostproc/postprocess.h>
27 #include "ffmpeg_content.h"
28 #include "exceptions.h"
33 using std::stringstream;
34 using boost::shared_ptr;
36 boost::mutex FFmpeg::_mutex;
38 FFmpeg::FFmpeg (boost::shared_ptr<const FFmpegContent> c)
51 boost::mutex::scoped_lock lm (_mutex);
53 for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
54 AVCodecContext* context = _format_context->streams[i]->codec;
55 if (context->codec_type == AVMEDIA_TYPE_VIDEO || context->codec_type == AVMEDIA_TYPE_AUDIO) {
56 avcodec_close (context);
62 avformat_close_input (&_format_context);
66 FFmpeg::setup_general ()
70 if (avformat_open_input (&_format_context, _ffmpeg_content->file().string().c_str(), 0, 0) < 0) {
71 throw OpenFileError (_ffmpeg_content->file().string ());
74 if (avformat_find_stream_info (_format_context, 0) < 0) {
75 throw DecodeError (_("could not find stream information"));
78 /* Find video stream */
80 for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
81 AVStream* s = _format_context->streams[i];
82 if (s->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
87 if (_video_stream < 0) {
88 throw DecodeError (N_("could not find video stream"));
91 _frame = avcodec_alloc_frame ();
93 throw DecodeError (N_("could not allocate frame"));
98 FFmpeg::setup_video ()
100 boost::mutex::scoped_lock lm (_mutex);
102 AVCodecContext* context = _format_context->streams[_video_stream]->codec;
103 AVCodec* codec = avcodec_find_decoder (context->codec_id);
106 throw DecodeError (_("could not find video decoder"));
109 if (avcodec_open2 (context, codec, 0) < 0) {
110 throw DecodeError (N_("could not open video decoder"));
115 FFmpeg::setup_audio ()
117 boost::mutex::scoped_lock lm (_mutex);
119 for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
120 AVCodecContext* context = _format_context->streams[i]->codec;
121 if (context->codec_type != AVMEDIA_TYPE_AUDIO) {
125 AVCodec* codec = avcodec_find_decoder (context->codec_id);
127 throw DecodeError (_("could not find audio decoder"));
130 if (avcodec_open2 (context, codec, 0) < 0) {
131 throw DecodeError (N_("could not open audio decoder"));
138 FFmpeg::video_codec_context () const
140 return _format_context->streams[_video_stream]->codec;
144 FFmpeg::audio_codec_context () const
146 return _format_context->streams[_ffmpeg_content->audio_stream()->id]->codec;