X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Ftest.cc;h=081a13a8cd93e07de09e5f25034dfa3c6e8c5ed5;hb=ba7516b01ddef143d58b32ac14350dcd15079641;hp=699dea6c3a223bb3947064372c2fe6fc7901b48e;hpb=dd879e5254c1e005e60ccd135a37f70cb2fe3a75;p=dcpomatic.git

diff --git a/test/test.cc b/test/test.cc
index 699dea6c3..081a13a8c 100644
--- a/test/test.cc
+++ b/test/test.cc
@@ -32,15 +32,26 @@
 #include "lib/encode_server_finder.h"
 #include "lib/image.h"
 #include "lib/ratio.h"
+#include "lib/dcp_content_type.h"
 #include "lib/log_entry.h"
 #include <dcp/dcp.h>
+#include <dcp/cpl.h>
+#include <dcp/reel.h>
+#include <dcp/reel_picture_asset.h>
+#include <dcp/mono_picture_frame.h>
+#include <dcp/mono_picture_asset.h>
+#include <dcp/openjpeg_image.h>
 #include <asdcp/AS_DCP.h>
 #include <sndfile.h>
 #include <libxml++/libxml++.h>
 #include <Magick++.h>
+extern "C" {
+#include <libavformat/avformat.h>
+}
 #define BOOST_TEST_DYN_LINK
 #define BOOST_TEST_MODULE dcpomatic_test
 #include <boost/test/unit_test.hpp>
+#include <boost/algorithm/string.hpp>
 #include <list>
 #include <vector>
 #include <iostream>
@@ -54,6 +65,7 @@ using std::list;
 using std::abs;
 using boost::shared_ptr;
 using boost::scoped_array;
+using boost::dynamic_pointer_cast;
 
 boost::filesystem::path private_data = boost::filesystem::path ("..") / boost::filesystem::path ("dcpomatic-test-private");
 
@@ -82,11 +94,18 @@ struct TestConfig
 		Config::instance()->set_default_audio_delay (0);
 		Config::instance()->set_default_j2k_bandwidth (100000000);
 		Config::instance()->set_default_interop (false);
+		Config::instance()->set_default_still_length (10);
 		Config::instance()->set_log_types (LogEntry::TYPE_GENERAL | LogEntry::TYPE_WARNING | LogEntry::TYPE_ERROR);
+		Config::instance()->set_automatic_audio_analysis (false);
 
 		EncodeServerFinder::instance()->stop ();
 
 		signal_manager = new TestSignalManager ();
+
+		char* env_private = getenv("DCPOMATIC_TEST_PRIVATE");
+		if (env_private) {
+			private_data = env_private;
+		}
 	}
 
 	~TestConfig ()
@@ -120,6 +139,21 @@ new_test_film (string name)
 	return film;
 }
 
+shared_ptr<Film>
+new_test_film2 (string name)
+{
+	boost::filesystem::path p = test_film_dir (name);
+	if (boost::filesystem::exists (p)) {
+		boost::filesystem::remove_all (p);
+	}
+
+	shared_ptr<Film> film = shared_ptr<Film> (new Film (p));
+	film->set_dcp_content_type (DCPContentType::from_isdcf_name ("TST"));
+	film->set_container (Ratio::from_id ("185"));
+	film->write_metadata ();
+	return film;
+}
+
 void
 check_wav_file (boost::filesystem::path ref, boost::filesystem::path check)
 {
@@ -283,7 +317,7 @@ check_xml (xmlpp::Element* ref, xmlpp::Element* test, list<string> ignore)
 
 	xmlpp::Element::NodeList ref_children = ref->get_children ();
 	xmlpp::Element::NodeList test_children = test->get_children ();
-	BOOST_CHECK_MESSAGE (
+	BOOST_REQUIRE_MESSAGE (
 		ref_children.size() == test_children.size(),
 		ref->get_name() << " has " << ref_children.size() << " or " << test_children.size() << " children"
 		);
@@ -387,3 +421,64 @@ write_image (shared_ptr<const Image> image, boost::filesystem::path file, string
 	Magick::Image m (image->size().width, image->size().height, format.c_str(), CharPixel, (void *) image->data()[0]);
 	m.write (file.string ());
 }
+
+static
+void
+check_ffmpeg_stream (boost::filesystem::path ref, boost::filesystem::path check, string stream)
+{
+	FILE* ref_file = popen(string("ffmpeg -loglevel error -i " + ref.string() + " -map 0:" + stream + " -f md5 -").c_str(), "r");
+	BOOST_REQUIRE (ref_file);
+	char ref_md5[64];
+	fscanf (ref_file, "%63s", ref_md5);
+	pclose (ref_file);
+
+	FILE* check_file = popen(string("ffmpeg -loglevel error -i " + check.string() + " -map 0:" + stream + " -f md5 -").c_str(), "r");
+	BOOST_REQUIRE (check_file);
+	char check_md5[64];
+	fscanf (check_file, "%63s", check_md5);
+	pclose (check_file);
+
+	BOOST_REQUIRE_EQUAL (strcmp(ref_md5, check_md5), 0);
+}
+
+void
+check_ffmpeg (boost::filesystem::path ref, boost::filesystem::path check)
+{
+	check_ffmpeg_stream (ref, check, "v");
+	check_ffmpeg_stream (ref, check, "a");
+}
+
+void
+check_one_frame (boost::filesystem::path dcp_dir, int64_t index, boost::filesystem::path ref)
+{
+	dcp::DCP dcp (dcp_dir);
+	dcp.read ();
+	shared_ptr<dcp::MonoPictureAsset> asset = dynamic_pointer_cast<dcp::MonoPictureAsset> (dcp.cpls().front()->reels().front()->main_picture()->asset());
+	BOOST_REQUIRE (asset);
+	shared_ptr<const dcp::MonoPictureFrame> frame = asset->start_read()->get_frame(index);
+
+	boost::uintmax_t const ref_size = boost::filesystem::file_size(ref);
+	BOOST_CHECK_EQUAL (frame->j2k_size(), ref_size);
+
+	FILE* ref_file = fopen_boost(ref, "rb");
+	BOOST_REQUIRE (ref_file);
+
+	uint8_t* ref_data = new uint8_t[ref_size];
+	fread (ref_data, ref_size, 1, ref_file);
+	fclose (ref_file);
+
+	BOOST_CHECK (memcmp(ref_data, frame->j2k_data(), ref_size) == 0);
+	delete[] ref_data;
+}
+
+boost::filesystem::path
+dcp_file (shared_ptr<const Film> film, string prefix)
+{
+	boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (film->dir(film->dcp_name()));
+	while (i != boost::filesystem::directory_iterator() && !boost::algorithm::starts_with (i->path().leaf().string(), prefix)) {
+		++i;
+	}
+
+	BOOST_REQUIRE (i != boost::filesystem::directory_iterator());
+	return i->path();
+}