X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fffmpeg.cc;h=5fc33348923c85d584af16fb980c6f5f50ad2dd1;hb=8aeb741ccbe2edb528e98a431bf55459a6836a9b;hp=a39de391af92abafbe3d9e39f355752875a13f72;hpb=46cd0fe7b5b514f0d9456b25f670679cc584a218;p=dcpomatic.git

diff --git a/src/lib/ffmpeg.cc b/src/lib/ffmpeg.cc
index a39de391a..5fc333489 100644
--- a/src/lib/ffmpeg.cc
+++ b/src/lib/ffmpeg.cc
@@ -26,17 +26,23 @@ extern "C" {
 #include "ffmpeg.h"
 #include "ffmpeg_content.h"
 #include "exceptions.h"
+#include "util.h"
 
 #include "i18n.h"
 
 using std::string;
+using std::cout;
 using std::stringstream;
 using boost::shared_ptr;
+using boost::lexical_cast;
 
 boost::mutex FFmpeg::_mutex;
 
 FFmpeg::FFmpeg (boost::shared_ptr<const FFmpegContent> c)
 	: _ffmpeg_content (c)
+	, _avio_buffer (0)
+	, _avio_buffer_size (4096)
+	, _avio_context (0)
 	, _format_context (0)
 	, _frame (0)
 	, _video_stream (-1)
@@ -57,18 +63,43 @@ FFmpeg::~FFmpeg ()
 		}
 	}
 
-	av_free (_frame);
+	av_frame_free (&_frame);
 	
 	avformat_close_input (&_format_context);
 }
 
+static int
+avio_read_wrapper (void* data, uint8_t* buffer, int amount)
+{
+	return reinterpret_cast<FFmpeg*>(data)->avio_read (buffer, amount);
+}
+
+static int64_t
+avio_seek_wrapper (void* data, int64_t offset, int whence)
+{
+	return reinterpret_cast<FFmpeg*>(data)->avio_seek (offset, whence);
+}
+
 void
 FFmpeg::setup_general ()
 {
 	av_register_all ();
 
-	if (avformat_open_input (&_format_context, _ffmpeg_content->file().string().c_str(), 0, 0) < 0) {
-		throw OpenFileError (_ffmpeg_content->file().string ());
+	_file_group.set_paths (_ffmpeg_content->paths ());
+	_avio_buffer = static_cast<uint8_t*> (wrapped_av_malloc (_avio_buffer_size));
+	_avio_context = avio_alloc_context (_avio_buffer, _avio_buffer_size, 0, this, avio_read_wrapper, 0, avio_seek_wrapper);
+	_format_context = avformat_alloc_context ();
+	_format_context->pb = _avio_context;
+	
+	AVDictionary* options = 0;
+	/* These durations are in microseconds, and represent how far into the content file
+	   we will look for streams.
+	*/
+	av_dict_set (&options, "analyzeduration", lexical_cast<string> (5 * 60 * 1e6).c_str(), 0);
+	av_dict_set (&options, "probesize", lexical_cast<string> (5 * 60 * 1e6).c_str(), 0);
+	
+	if (avformat_open_input (&_format_context, 0, 0, &options) < 0) {
+		throw OpenFileError (_ffmpeg_content->path(0).string ());
 	}
 
 	if (avformat_find_stream_info (_format_context, 0) < 0) {
@@ -88,7 +119,25 @@ FFmpeg::setup_general ()
 		throw DecodeError (N_("could not find video stream"));
 	}
 
-	_frame = avcodec_alloc_frame ();
+	/* Hack: if the AVStreams have zero IDs, put some in.  We
+	   use the IDs so that we can cope with VOBs, in which streams
+	   move about in index but remain with the same ID in different
+	   VOBs.  However, some files have all-zero IDs, hence this hack.
+	*/
+	   
+	uint32_t i = 0;
+	while (i < _format_context->nb_streams && _format_context->streams[i]->id == 0) {
+		++i;
+	}
+
+	if (i == _format_context->nb_streams) {
+		/* Put in our own IDs */
+		for (uint32_t i = 0; i < _format_context->nb_streams; ++i) {
+			_format_context->streams[i]->id = i;
+		}
+	}
+
+	_frame = av_frame_alloc ();
 	if (_frame == 0) {
 		throw DecodeError (N_("could not allocate frame"));
 	}
@@ -98,7 +147,8 @@ void
 FFmpeg::setup_video ()
 {
 	boost::mutex::scoped_lock lm (_mutex);
-	
+
+	assert (_video_stream >= 0);
 	AVCodecContext* context = _format_context->streams[_video_stream]->codec;
 	AVCodec* codec = avcodec_find_decoder (context->codec_id);
 
@@ -143,5 +193,25 @@ FFmpeg::video_codec_context () const
 AVCodecContext *
 FFmpeg::audio_codec_context () const
 {
-	return _format_context->streams[_ffmpeg_content->audio_stream()->id]->codec;
+	if (!_ffmpeg_content->audio_stream ()) {
+		return 0;
+	}
+	
+	return _ffmpeg_content->audio_stream()->stream(_format_context)->codec;
+}
+
+int
+FFmpeg::avio_read (uint8_t* buffer, int const amount)
+{
+	return _file_group.read (buffer, amount);
+}
+
+int64_t
+FFmpeg::avio_seek (int64_t const pos, int whence)
+{
+	if (whence == AVSEEK_SIZE) {
+		return _file_group.length ();
+	}
+	
+	return _file_group.seek (pos, whence);
 }