X-Git-Url: http://research.m1stereo.tv/gitweb?a=blobdiff_plain;f=src%2Fmodules%2Favformat%2Fproducer_avformat.c;h=2e39825a5d7372d8942c84747c5cfb9526cde01b;hb=6a9365cbc88b77e48d6f249808e39d32388c7094;hp=bace227bfdf1d44a433afd56d8b4b100870044a2;hpb=573abb05e2b070c67daa9f97981e4c707ba96014;p=melted

diff --git a/src/modules/avformat/producer_avformat.c b/src/modules/avformat/producer_avformat.c
index bace227..2e39825 100644
--- a/src/modules/avformat/producer_avformat.c
+++ b/src/modules/avformat/producer_avformat.c
@@ -41,6 +41,21 @@ static int producer_get_frame( mlt_producer this, mlt_frame_ptr frame, int index
 static int avformat_initialised = 0;
 static pthread_mutex_t avformat_mutex;
 
+void *av_malloc( unsigned int size )
+{
+	return mlt_pool_alloc( size );
+}
+
+void *av_realloc( void *ptr, unsigned int size )
+{
+	return mlt_pool_realloc( ptr, size );
+}
+
+void av_free( void *ptr )
+{
+	return mlt_pool_release( ptr );
+}
+
 /** Constructor for libavformat.
 */
 
@@ -72,8 +87,8 @@ mlt_producer producer_avformat_init( char *file )
 			// Initialise avformat if necessary
 			if ( avformat_initialised == 0 )
 			{
-				pthread_mutex_init( &avformat_mutex, NULL );
 				avformat_initialised = 1;
+				pthread_mutex_init( &avformat_mutex, NULL );
 				av_register_all( );
 			}
 
@@ -309,7 +324,7 @@ static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_form
 	int ignore = 0;
 
 	// Current time calcs
-	double current_time = 0;
+	double current_time = mlt_properties_get_double( properties, "current_time" );
 
 	// We may want to use the source fps if available
 	double source_fps = mlt_properties_get_double( properties, "source_fps" );
@@ -358,7 +373,7 @@ static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_form
 			av_seek_frame( context, -1, real_timecode * 1000000.0 );
 	
 			// Remove the cached info relating to the previous position
-			mlt_properties_set_double( properties, "current_time", 0 );
+			mlt_properties_set_double( properties, "current_time", real_timecode );
 			mlt_properties_set_data( properties, "current_image", NULL, 0, NULL, NULL );
 		}
 	}
@@ -395,13 +410,16 @@ static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_form
 			// We only deal with video from the selected video_index
 			if ( ret >= 0 && pkt.stream_index == index && pkt.size > 0 )
 			{
-				current_time = ( double )pkt.pts / 1000000.0;
-
 				// Decode the image
 				ret = avcodec_decode_video( codec_context, &frame, &got_picture, pkt.data, pkt.size );
 
 				if ( got_picture )
 				{
+					if ( pkt.pts != AV_NOPTS_VALUE && pkt.pts != 0  )
+						current_time = ( double )pkt.pts / 1000000.0;
+					else
+						current_time = real_timecode;
+
 					// Handle ignore
 					if ( current_time < real_timecode )
 					{
@@ -410,9 +428,10 @@ static int producer_get_image( mlt_frame frame, uint8_t **buffer, mlt_image_form
 					}
 					else if ( current_time >= real_timecode )
 					{
+						//current_time = real_timecode;
 						ignore = 0;
 					}
-					else if ( got_picture && ignore -- )
+					else if ( ignore -- )
 					{
 						got_picture = 0;
 					}
@@ -554,34 +573,8 @@ static void producer_set_up_video( mlt_producer this, mlt_frame frame )
 			// If we don't have a codec and we can't initialise it, we can't do much more...
 			if ( codec != NULL && avcodec_open( codec_context, codec ) >= 0 )
 			{
-				double aspect_ratio = 0;
-				double source_fps = 0;
-
-				// Set aspect ratio
-				if ( codec_context->sample_aspect_ratio.num == 0 ) 
-					aspect_ratio = 0;
-				else
-					aspect_ratio = av_q2d( codec_context->sample_aspect_ratio ) * codec_context->width / codec_context->height;
-
-				// XXX: This assumes square pixels!
-        		if (aspect_ratio <= 0.0)
-					aspect_ratio = ( double )codec_context->width / ( double )codec_context->height;
-
-				mlt_properties_set_double( properties, "aspect_ratio", aspect_ratio );
-				fprintf( stderr, "AVFORMAT: sample aspect %f computed display aspect %f\n", av_q2d( codec_context->sample_aspect_ratio ), aspect_ratio );
-
-				// Determine the fps
-				source_fps = ( double )codec_context->frame_rate / codec_context->frame_rate_base;
-
-				// We'll use fps if it's available
-				if ( source_fps > 0 && source_fps < 30 )
-					mlt_properties_set_double( properties, "source_fps", source_fps );
-
 				// Now store the codec with its destructor
 				mlt_properties_set_data( properties, "video_codec", codec_context, 0, producer_codec_close, NULL );
-
-				// Set to the real timecode
-				av_seek_frame( context, -1, 0 );
 			}
 			else
 			{
@@ -593,6 +586,27 @@ static void producer_set_up_video( mlt_producer this, mlt_frame frame )
 		// No codec, no show...
 		if ( codec != NULL )
 		{
+			double aspect_ratio = 1;
+			double source_fps = 0;
+
+			// Set aspect ratio
+			if ( codec_context->sample_aspect_ratio.num > 0 )
+				aspect_ratio = av_q2d( codec_context->sample_aspect_ratio );
+
+			mlt_properties_set_double( properties, "aspect_ratio", aspect_ratio );
+			//fprintf( stderr, "AVFORMAT: sample aspect %f %dx%d\n", av_q2d( codec_context->sample_aspect_ratio ), codec_context->width, codec_context->height );
+
+			// Determine the fps
+			source_fps = ( double )codec_context->frame_rate / codec_context->frame_rate_base;
+
+			// We'll use fps if it's available
+			if ( source_fps > 0 && source_fps < 30 )
+				mlt_properties_set_double( properties, "source_fps", source_fps );
+			
+			// Set the width and height
+			mlt_properties_set_int( frame_properties, "width", codec_context->width );
+			mlt_properties_set_int( frame_properties, "height", codec_context->height );
+
 			mlt_frame_push_get_image( frame, producer_get_image );
 			mlt_properties_set_data( frame_properties, "avformat_producer", this, 0, NULL, NULL );
 		}