X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Fseek_zero_test.cc;h=7abe7fa2623f290ebae98409cbf689551c971391;hb=c008066160d85b9ec9e5485375d7baaa5d27bda2;hp=d298a772bf91c8f2f887367b7b04c58d46c44a31;hpb=e194f0003b60b2607da0822485c56cd8267e78dc;p=dcpomatic.git

diff --git a/test/seek_zero_test.cc b/test/seek_zero_test.cc
index d298a772b..7abe7fa26 100644
--- a/test/seek_zero_test.cc
+++ b/test/seek_zero_test.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2013-2014 Carl Hetherington <cth@carlh.net>
 
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
@@ -28,10 +28,15 @@
 #include "lib/ratio.h"
 #include "lib/dcp_content_type.h"
 #include "lib/ffmpeg_decoder.h"
+#include "lib/ffmpeg_audio_stream.h"
+#include "lib/content_video.h"
 #include "test.h"
 
 using std::cout;
+using std::list;
 using boost::shared_ptr;
+using boost::dynamic_pointer_cast;
+using boost::optional;
 
 BOOST_AUTO_TEST_CASE (seek_zero_test)
 {
@@ -40,19 +45,22 @@ BOOST_AUTO_TEST_CASE (seek_zero_test)
 	film->set_container (Ratio::from_id ("185"));
 	film->set_dcp_content_type (DCPContentType::from_pretty_name ("Test"));
 	shared_ptr<FFmpegContent> content (new FFmpegContent (film, "test/data/count300bd48.m2ts"));
-	content->set_ratio (Ratio::from_id ("185"));
-	film->examine_and_add_content (content);
+	film->examine_and_add_content (content, true);
 	wait_for_jobs ();
+	content->set_scale (VideoContentScale (Ratio::from_id ("185")));
+	
+	/* Work out the first video frame index that we will be given, taking into account
+	 * the difference between first video and first audio.
+	 */
+	ContentTime video_delay = content->first_video().get() - content->audio_stream()->first_audio.get();
+	if (video_delay < ContentTime ()) {
+		video_delay = ContentTime ();
+	}
 
-	FFmpegDecoder decoder (film, content, true, false);
-	shared_ptr<Decoded> a = decoder.get ();
-	cout << a->content_time << "\n";
-	decoder.seek (0, true);
-	shared_ptr<Decoded> b = decoder.get ();
-	cout << b->content_time << "\n";
-
-	/* a will be after no seek, and b after a seek to zero, which should
-	   have the same effect.
-	*/
-	BOOST_CHECK_EQUAL (a->content_time, b->content_time);
+	VideoFrame const first_frame = video_delay.round_up (content->video_frame_rate ()).frames (content->video_frame_rate ());
+
+	FFmpegDecoder decoder (content, film->log());
+	list<ContentVideo> a = decoder.get_video (first_frame, true);
+	BOOST_CHECK (a.size() == 1);
+	BOOST_CHECK_EQUAL (a.front().frame, first_frame);
 }