From ddc7c8b323385ad6c14882bbc878f423f623d665 Mon Sep 17 00:00:00 2001
From: Rodrigo Braz Monteiro <zeratul@cellosoft.com>
Date: Fri, 24 Feb 2006 05:58:45 +0000
Subject: [PATCH] We're getting there...

Originally committed to SVN as r144.
---
 core/video_provider_lavc.cpp | 98 ++++++++++++++----------------------
 core/video_provider_lavc.h   |  7 ++-
 2 files changed, 44 insertions(+), 61 deletions(-)

diff --git a/core/video_provider_lavc.cpp b/core/video_provider_lavc.cpp
index ecaf97967..4bb726a95 100644
--- a/core/video_provider_lavc.cpp
+++ b/core/video_provider_lavc.cpp
@@ -39,6 +39,8 @@
 #ifdef USE_LAVC
 #include <wx/wxprec.h>
 #include "video_provider_lavc.h"
+#include "utils.h"
+#include "vfr.h"
 
 
 ///////////////
@@ -111,7 +113,7 @@ void LAVCVideoProvider::LoadVideo(wxString filename) {
 		if (!codec) throw _T("Could not find suitable video decoder");
 
 		// Enable truncation
-		if (codec->capabilities & CODEC_CAP_TRUNCATED) codecContext->flags |= CODEC_FLAG_TRUNCATED;
+		//if (codec->capabilities & CODEC_CAP_TRUNCATED) codecContext->flags |= CODEC_FLAG_TRUNCATED;
 
 		// Open codec
 		result = avcodec_open(codecContext,codec);
@@ -119,6 +121,9 @@ void LAVCVideoProvider::LoadVideo(wxString filename) {
 
 		// Allocate frame
 		frame = avcodec_alloc_frame();
+
+		// Set frame
+		frameNumber = -1;
 	}
 
 	// Catch errors
@@ -155,64 +160,24 @@ void LAVCVideoProvider::Close() {
 //////////////////
 // Get next frame
 void LAVCVideoProvider::GetNextFrame() {
-	static AVPacket packet;
-	static bool firstTime = true;
-	static int bytesRemaining = 0;
-	static uint8_t *rawdata;
-	int decoded;
-	int frameFinished;
+	// Read packet
+	AVPacket packet;
+	while (av_read_frame(formatContext, &packet)>=0) {
+		// Check if packet is part of video stream
+		if(packet.stream_index == vidStream) {
+			// Decode frame
+			int frameFinished;
+			avcodec_decode_video(codecContext, frame, &frameFinished, packet.data, packet.size);
 
-	// First call
-	if (firstTime) {
-		firstTime = false;
-		packet.data = NULL;
-	}
-
-	// Start processing packets
-	bool run = true;
-	while (run) {
-		// Current packet has more bytes
-		while (bytesRemaining > 0) {
-			// Decode
-			decoded = avcodec_decode_video(codecContext,frame,&frameFinished,rawdata,bytesRemaining);
-
-			// Error?
-			if (decoded < 0) throw _T("Error reading frame");
-
-			// Update counts
-			bytesRemaining -= decoded;
-			rawdata += decoded;
-
-			// Finished?
-			if (frameFinished) return;
-		}
-
-		// Get a packet
-		do {
-			// Free packet
-			if (packet.data != NULL) av_free_packet(&packet);
-			
-			// Get new
-			int result = av_read_packet(formatContext,&packet);
-			if (result < 0) {
-				run = false;
-				break;
+			// Success?
+			if(frameFinished) {
+				// Free packet
+				av_free_packet(&packet);
+				return;
 			}
-		} while (packet.stream_index != vidStream);
-
-		// Get packet data
-		if (run) {
-			bytesRemaining = packet.size;
-			rawdata = packet.data;
 		}
-	}
-
-	// End of last pack
-	decoded = avcodec_decode_video(codecContext,frame,&frameFinished,rawdata,bytesRemaining);
-	if (packet.data != NULL) av_free_packet(&packet);
-
-	// Frame finished?
-	if (!frameFinished) throw _T("Unable to finish decoding frame");
+    }
+	av_free_packet(&packet);
 }
 
 
@@ -254,14 +219,29 @@ void LAVCVideoProvider::RefreshSubtitles() {
 /////////////
 // Get frame
 wxBitmap LAVCVideoProvider::GetFrame(int n) {
+	// Return stored frame
+	n = MID(0,n,GetFrameCount()-1);
+	if (n == frameNumber) return curFrame;
+
+	// Seek if needed
+	if (n != frameNumber+1) {
+		av_seek_frame(formatContext,vidStream,n,0);
+	}
+
 	// Get frame
 	GetNextFrame();
 
 	// Bitmap
-	wxBitmap bmp = AVFrameToWX(frame);
+	wxBitmap bmp;
+	if (frame) bmp = AVFrameToWX(frame);
+	else bmp = wxBitmap(GetWidth(),GetHeight());
+
+	// Set current frame
+	curFrame = bmp;
+	frameNumber = n;
 
 	// Return
-	return bmp;
+	return curFrame;
 }
 
 
@@ -274,7 +254,7 @@ void LAVCVideoProvider::GetFloatFrame(float* Buffer, int n) {
 ////////////////
 // Get position
 int LAVCVideoProvider::GetPosition() {
-	return 0;
+	return frameNumber;
 }
 
 
diff --git a/core/video_provider_lavc.h b/core/video_provider_lavc.h
index 26fc4a796..28f89063b 100644
--- a/core/video_provider_lavc.h
+++ b/core/video_provider_lavc.h
@@ -40,8 +40,8 @@
 // Headers
 #ifdef USE_LAVC
 #define EMULATE_INTTYPES
-#include <lavc/avcodec.h>
-#include <lavc/avformat.h>
+#include <ffmpeg/avcodec.h>
+#include <ffmpeg/avformat.h>
 #include "video_provider.h"
 
 
@@ -56,6 +56,9 @@ private:
 	AVFrame *frame;
 	int vidStream;
 
+	int frameNumber;
+	wxBitmap curFrame;
+
 	uint8_t *buffer;
 	int bufferSize;