X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Ftest.cc;h=cbd12e1ee363cf7d6a592703db2d7984980d2193;hb=28111007e2e6fd62f5810be780706ae1618bd33f;hp=b02fa99c95bb50a3591022f28b446eed06f486c0;hpb=7d651ec877927e0887da48ce441e5c5158e1f34f;p=dcpomatic.git

diff --git a/test/test.cc b/test/test.cc
index b02fa99c9..cbd12e1ee 100644
--- a/test/test.cc
+++ b/test/test.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2012-2017 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2012-2020 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -30,20 +30,33 @@
 #include "lib/job.h"
 #include "lib/cross.h"
 #include "lib/encode_server_finder.h"
+#include "lib/ffmpeg_image_proxy.h"
 #include "lib/image.h"
 #include "lib/ratio.h"
+#include "lib/dcp_content_type.h"
 #include "lib/log_entry.h"
+#include "lib/compose.hpp"
+#include "lib/file_log.h"
+#include "lib/dcpomatic_log.h"
+#include "test.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 <png.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>
@@ -55,10 +68,53 @@ using std::cout;
 using std::cerr;
 using std::list;
 using std::abs;
-using boost::shared_ptr;
+using std::shared_ptr;
 using boost::scoped_array;
+using std::dynamic_pointer_cast;
+#if BOOST_VERSION >= 106100
+using namespace boost::placeholders;
+#endif
+
+
+boost::filesystem::path
+TestPaths::TestPaths::private_data ()
+{
+	char* env = getenv("DCPOMATIC_TEST_PRIVATE");
+	if (env) {
+		return boost::filesystem::path(env);
+	}
+
+	return boost::filesystem::canonical(boost::filesystem::path ("..") / boost::filesystem::path ("dcpomatic-test-private"));
+}
+
+
+boost::filesystem::path TestPaths::xsd ()
+{
+	return boost::filesystem::canonical(boost::filesystem::path("..") / boost::filesystem::path("libdcp") / boost::filesystem::path("xsd"));
+}
+
 
-boost::filesystem::path private_data = boost::filesystem::path ("..") / boost::filesystem::path ("dcpomatic-test-private");
+void
+setup_test_config ()
+{
+	Config::instance()->set_master_encoding_threads (boost::thread::hardware_concurrency());
+	Config::instance()->set_server_encoding_threads (1);
+	Config::instance()->set_server_port_base (61921);
+	Config::instance()->set_default_isdcf_metadata (ISDCFMetadata ());
+	Config::instance()->set_default_container (Ratio::from_id ("185"));
+	Config::instance()->set_default_dcp_content_type (static_cast<DCPContentType*> (0));
+	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 | LogEntry::TYPE_DEBUG_THREED |
+		LogEntry::TYPE_DEBUG_ENCODE | LogEntry::TYPE_DEBUG_PLAYER |
+		LogEntry::TYPE_DISK
+		);
+	Config::instance()->set_automatic_audio_analysis (false);
+}
 
 class TestSignalManager : public SignalManager
 {
@@ -74,23 +130,17 @@ struct TestConfig
 {
 	TestConfig ()
 	{
-		dcpomatic_setup ();
+		State::override_path = "build/test/state";
+		boost::filesystem::remove_all (*State::override_path);
 
-		Config::instance()->set_master_encoding_threads (1);
-		Config::instance()->set_server_encoding_threads (1);
-		Config::instance()->set_server_port_base (61921);
-		Config::instance()->set_default_isdcf_metadata (ISDCFMetadata ());
-		Config::instance()->set_default_container (Ratio::from_id ("185"));
-		Config::instance()->set_default_dcp_content_type (static_cast<DCPContentType*> (0));
-		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);
+		dcpomatic_setup ();
+		setup_test_config ();
 
 		EncodeServerFinder::instance()->stop ();
 
 		signal_manager = new TestSignalManager ();
+
+		dcpomatic_log.reset (new FileLog("build/test/log"));
 	}
 
 	~TestConfig ()
@@ -124,6 +174,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)
 {
@@ -193,22 +258,165 @@ check_mxf_audio_file (boost::filesystem::path ref, boost::filesystem::path check
 	}
 }
 
-void
-check_image (boost::filesystem::path ref, boost::filesystem::path check)
+
+/** @return true if the files are the same, otherwise false */
+bool
+mxf_atmos_files_same (boost::filesystem::path ref, boost::filesystem::path check, bool verbose)
 {
-#ifdef DCPOMATIC_IMAGE_MAGICK
-	using namespace MagickCore;
-#else
-	using namespace MagickLib;
-#endif
+	ASDCP::ATMOS::MXFReader ref_reader;
+	BOOST_REQUIRE (!ASDCP_FAILURE(ref_reader.OpenRead(ref.string().c_str())));
+
+	ASDCP::ATMOS::AtmosDescriptor ref_desc;
+	BOOST_REQUIRE (!ASDCP_FAILURE(ref_reader.FillAtmosDescriptor(ref_desc)));
+
+	ASDCP::ATMOS::MXFReader check_reader;
+	BOOST_REQUIRE (!ASDCP_FAILURE(check_reader.OpenRead(check.string().c_str())));
 
-	Magick::Image ref_image;
-	ref_image.read (ref.string ());
-	Magick::Image check_image;
-	check_image.read (check.string ());
-	BOOST_CHECK_MESSAGE (ref_image.compare (check_image), ref << " differs from " << check);
+	ASDCP::ATMOS::AtmosDescriptor check_desc;
+	BOOST_REQUIRE (!ASDCP_FAILURE(check_reader.FillAtmosDescriptor(check_desc)));
+
+	if (ref_desc.EditRate.Numerator != check_desc.EditRate.Numerator) {
+		if (verbose) {
+			std::cout << "EditRate.Numerator differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.EditRate.Denominator != check_desc.EditRate.Denominator) {
+		if (verbose) {
+			std::cout << "EditRate.Denominator differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.ContainerDuration != check_desc.ContainerDuration) {
+		if (verbose) {
+			std::cout << "EditRate.ContainerDuration differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.FirstFrame != check_desc.FirstFrame) {
+		if (verbose) {
+			std::cout << "EditRate.FirstFrame differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.MaxChannelCount != check_desc.MaxChannelCount) {
+		if (verbose) {
+			std::cout << "EditRate.MaxChannelCount differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.MaxObjectCount != check_desc.MaxObjectCount) {
+		if (verbose) {
+			std::cout << "EditRate.MaxObjectCount differs.\n";
+		}
+		return false;
+	}
+	if (ref_desc.AtmosVersion != check_desc.AtmosVersion) {
+		if (verbose) {
+			std::cout << "EditRate.AtmosVersion differs.\n";
+		}
+		return false;
+	}
+
+	ASDCP::DCData::FrameBuffer ref_buffer (Kumu::Megabyte);
+	ASDCP::DCData::FrameBuffer check_buffer (Kumu::Megabyte);
+	for (size_t i = 0; i < ref_desc.ContainerDuration; ++i) {
+		ref_reader.ReadFrame (i, ref_buffer, 0);
+		check_reader.ReadFrame (i, check_buffer, 0);
+		if (memcmp(ref_buffer.RoData(), check_buffer.RoData(), ref_buffer.Size())) {
+			if (verbose) {
+				std::cout << "data differs.\n";
+			}
+			return false;
+		}
+	}
+
+	return true;
+}
+
+
+static
+double
+rms_error (boost::filesystem::path ref, boost::filesystem::path check)
+{
+	FFmpegImageProxy ref_proxy (ref, VIDEO_RANGE_FULL);
+	shared_ptr<Image> ref_image = ref_proxy.image().image;
+	FFmpegImageProxy check_proxy (check, VIDEO_RANGE_FULL);
+	shared_ptr<Image> check_image = check_proxy.image().image;
+
+	BOOST_REQUIRE_EQUAL (ref_image->pixel_format(), check_image->pixel_format());
+	AVPixelFormat const format = ref_image->pixel_format();
+
+	BOOST_REQUIRE (ref_image->size() == check_image->size());
+	int const width = ref_image->size().width;
+	int const height = ref_image->size().height;
+
+	double sum_square = 0;
+	switch (format) {
+		case AV_PIX_FMT_RGBA:
+		{
+			for (int y = 0; y < height; ++y) {
+				uint8_t* p = ref_image->data()[0] + y * ref_image->stride()[0];
+				uint8_t* q = check_image->data()[0] + y * check_image->stride()[0];
+				for (int x = 0; x < width; ++x) {
+					for (int c = 0; c < 4; ++c) {
+						sum_square += pow((*p++ - *q++), 2);
+					}
+				}
+			}
+			break;
+		}
+		case AV_PIX_FMT_RGB24:
+		{
+			for (int y = 0; y < height; ++y) {
+				uint8_t* p = ref_image->data()[0] + y * ref_image->stride()[0];
+				uint8_t* q = check_image->data()[0] + y * check_image->stride()[0];
+				for (int x = 0; x < width; ++x) {
+					for (int c = 0; c < 3; ++c) {
+						sum_square += pow((*p++ - *q++), 2);
+					}
+				}
+			}
+			break;
+		}
+		case AV_PIX_FMT_RGB48BE:
+		{
+			for (int y = 0; y < height; ++y) {
+				uint16_t* p = reinterpret_cast<uint16_t*>(ref_image->data()[0] + y * ref_image->stride()[0]);
+				uint16_t* q = reinterpret_cast<uint16_t*>(check_image->data()[0] + y * check_image->stride()[0]);
+				for (int x = 0; x < width; ++x) {
+					for (int c = 0; c < 3; ++c) {
+						sum_square += pow((*p++ - *q++), 2);
+					}
+				}
+			}
+			break;
+		}
+		default:
+			BOOST_REQUIRE_MESSAGE (false, "unrecognised pixel format " << format);
+	}
+
+	return sqrt(sum_square / (height * width));
+}
+
+
+BOOST_AUTO_TEST_CASE (rms_error_test)
+{
+	BOOST_CHECK_CLOSE (rms_error("test/data/check_image0.png", "test/data/check_image0.png"), 0, 0.001);
+	BOOST_CHECK_CLOSE (rms_error("test/data/check_image0.png", "test/data/check_image1.png"), 2.2778, 0.001);
+	BOOST_CHECK_CLOSE (rms_error("test/data/check_image0.png", "test/data/check_image2.png"), 59.8896, 0.001);
+	BOOST_CHECK_CLOSE (rms_error("test/data/check_image0.png", "test/data/check_image3.png"), 0.89164, 0.001);
 }
 
+
+void
+check_image (boost::filesystem::path ref, boost::filesystem::path check, double threshold)
+{
+	double const e = rms_error (ref, check);
+	BOOST_CHECK_MESSAGE (e < threshold, ref << " differs from " << check << " " << e);
+}
+
+
 void
 check_file (boost::filesystem::path ref, boost::filesystem::path check)
 {
@@ -250,11 +458,19 @@ check_file (boost::filesystem::path ref, boost::filesystem::path check)
 static void
 note (dcp::NoteType t, string n)
 {
-	if (t == dcp::DCP_ERROR) {
+	if (t == dcp::NoteType::ERROR) {
 		cerr << n << "\n";
 	}
 }
 
+
+void
+check_dcp (boost::filesystem::path ref, shared_ptr<const Film> film)
+{
+	check_dcp (ref, film->dir(film->dcp_name()));
+}
+
+
 void
 check_dcp (boost::filesystem::path ref, boost::filesystem::path check)
 {
@@ -348,7 +564,7 @@ wait_for_jobs ()
 	JobManager* jm = JobManager::instance ();
 	while (jm->work_to_do ()) {
 		while (signal_manager->ui_idle ()) {}
-		dcpomatic_sleep (1);
+		dcpomatic_sleep_seconds (1);
 	}
 
 	if (jm->errors ()) {
@@ -379,66 +595,219 @@ wait_for_jobs ()
 	return false;
 }
 
-void
-write_image (shared_ptr<const Image> image, boost::filesystem::path file, string format)
-{
-#ifdef DCPOMATIC_IMAGE_MAGICK
-		using namespace MagickCore;
-#else
-		using namespace MagickLib;
-#endif
 
-	Magick::Image m (image->size().width, image->size().height, format.c_str(), CharPixel, (void *) image->data()[0]);
-	m.write (file.string ());
-}
-
-class Reader
+class Memory
 {
 public:
-	Reader (boost::filesystem::path file)
-	{
-		format_context = avformat_alloc_context ();
-		BOOST_REQUIRE (format_context);
-		BOOST_REQUIRE (avformat_open_input (&format_context, file.string().c_str(), 0, 0) >= 0);
-		BOOST_REQUIRE (avformat_find_stream_info (format_context, 0) >= 0);
-	}
+	Memory ()
+		: data(0)
+		, size(0)
+	{}
 
-	~Reader ()
+	~Memory ()
 	{
-		avformat_close_input (&format_context);
+		free (data);
 	}
 
-	AVFormatContext* format_context;
+	uint8_t* data;
+	size_t size;
 };
 
 
+static void
+png_write_data (png_structp png_ptr, png_bytep data, png_size_t length)
+{
+	Memory* mem = reinterpret_cast<Memory*>(png_get_io_ptr(png_ptr));
+	size_t size = mem->size + length;
+
+	if (mem->data) {
+		mem->data = reinterpret_cast<uint8_t*>(realloc(mem->data, size));
+	} else {
+		mem->data = reinterpret_cast<uint8_t*>(malloc(size));
+	}
+
+	BOOST_REQUIRE (mem->data);
+
+	memcpy (mem->data + mem->size, data, length);
+	mem->size += length;
+}
+
+
+static void
+png_flush (png_structp)
+{
+
+}
+
+
+static void
+png_error_fn (png_structp png_ptr, char const * message)
+{
+	reinterpret_cast<Image*>(png_get_error_ptr(png_ptr))->png_error (message);
+}
+
+
 void
-check_ffmpeg (boost::filesystem::path ref, boost::filesystem::path check, int skip_packet_stream)
+write_image (shared_ptr<const Image> image, boost::filesystem::path file)
 {
-	Reader ref_r (ref);
-	Reader check_r (check);
+	int png_color_type = 0;
+	int bits_per_pixel = 0;
+	switch (image->pixel_format()) {
+	case AV_PIX_FMT_RGB24:
+		png_color_type = PNG_COLOR_TYPE_RGB;
+		bits_per_pixel = 8;
+		break;
+	case AV_PIX_FMT_XYZ12LE:
+		png_color_type = PNG_COLOR_TYPE_RGB;
+		bits_per_pixel = 16;
+		break;
+	default:
+		BOOST_REQUIRE_MESSAGE (false, "unexpected pixel format " << image->pixel_format());
+	}
+
+	/* error handling? */
+	png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, reinterpret_cast<void*>(const_cast<Image*>(image.get())), png_error_fn, 0);
+	BOOST_REQUIRE (png_ptr);
+
+	Memory state;
+
+	png_set_write_fn (png_ptr, &state, png_write_data, png_flush);
+
+	png_infop info_ptr = png_create_info_struct(png_ptr);
+	BOOST_REQUIRE (info_ptr);
+
+	png_set_IHDR (png_ptr, info_ptr, image->size().width, image->size().height, bits_per_pixel, png_color_type, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
+
+	png_byte ** row_pointers = reinterpret_cast<png_byte **>(png_malloc(png_ptr, image->size().height * sizeof(png_byte *)));
+	for (int i = 0; i < image->size().height; ++i) {
+		row_pointers[i] = (png_byte *) (image->data()[0] + i * image->stride()[0]);
+	}
 
-	BOOST_REQUIRE_EQUAL (ref_r.format_context->nb_streams, check_r.format_context->nb_streams);
+	png_write_info (png_ptr, info_ptr);
+	png_write_image (png_ptr, row_pointers);
+	png_write_end (png_ptr, info_ptr);
 
-	AVPacket ref_p;
-	AVPacket check_p;
+	png_destroy_write_struct (&png_ptr, &info_ptr);
+	png_free (png_ptr, row_pointers);
 
-	bool skipped = false;
+	dcp::ArrayData(state.data, state.size).write(file);
+}
+
+
+void
+check_ffmpeg (boost::filesystem::path ref, boost::filesystem::path check, int audio_tolerance)
+{
+	int const r = system (String::compose("ffcmp -t %1 %2 %3", audio_tolerance, ref.string(), check.string()).c_str());
+	BOOST_REQUIRE_EQUAL (WEXITSTATUS(r), 0);
+}
 
-	while (true) {
-		int p = av_read_frame (ref_r.format_context, &ref_p);
-		int q = av_read_frame (check_r.format_context, &check_p);
-		if (ref_p.stream_index == skip_packet_stream && check_p.stream_index == skip_packet_stream && !skipped) {
-			skipped = true;
-			continue;
+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);
+	shared_ptr<const dcp::MonoPictureFrame> ref_frame (new dcp::MonoPictureFrame (ref));
+
+	shared_ptr<dcp::OpenJPEGImage> image = frame->xyz_image ();
+	shared_ptr<dcp::OpenJPEGImage> ref_image = ref_frame->xyz_image ();
+
+	BOOST_REQUIRE (image->size() == ref_image->size());
+
+	int off = 0;
+	for (int y = 0; y < ref_image->size().height; ++y) {
+		for (int x = 0; x < ref_image->size().width; ++x) {
+			BOOST_REQUIRE_EQUAL (ref_image->data(0)[off], image->data(0)[off]);
+			BOOST_REQUIRE_EQUAL (ref_image->data(1)[off], image->data(1)[off]);
+			BOOST_REQUIRE_EQUAL (ref_image->data(2)[off], image->data(2)[off]);
+			++off;
 		}
-		BOOST_REQUIRE_EQUAL (p, q);
-		BOOST_REQUIRE (p == 0 || p == AVERROR_EOF);
-		if (p == AVERROR_EOF) {
-			break;
+	}
+}
+
+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();
+}
+
+boost::filesystem::path
+subtitle_file (shared_ptr<Film> film)
+{
+	for (
+		boost::filesystem::directory_iterator i = boost::filesystem::directory_iterator (film->directory().get() / film->dcp_name (false));
+		i != boost::filesystem::directory_iterator ();
+		++i) {
+
+		if (boost::filesystem::is_directory (i->path ())) {
+			for (
+				boost::filesystem::directory_iterator j = boost::filesystem::directory_iterator (i->path ());
+				j != boost::filesystem::directory_iterator ();
+				++j) {
+
+				if (boost::algorithm::starts_with (j->path().leaf().string(), "sub_")) {
+					return j->path();
+				}
+			}
 		}
+	}
+
+	BOOST_REQUIRE (false);
+	/* Remove warning */
+	return boost::filesystem::path("/");
+}
 
-		BOOST_REQUIRE_EQUAL (ref_p.buf->size, check_p.buf->size);
-		BOOST_REQUIRE_EQUAL (memcmp (ref_p.buf->data, check_p.buf->data, ref_p.buf->size), 0);
+void
+make_random_file (boost::filesystem::path path, size_t size)
+{
+	FILE* t = fopen_boost(path, "wb");
+	BOOST_REQUIRE (t);
+	for (size_t i = 0; i < size; ++i) {
+		uint8_t r = rand() & 0xff;
+		fwrite (&r, 1, 1, t);
 	}
+	fclose (t);
 }
+
+
+LogSwitcher::LogSwitcher (shared_ptr<Log> log)
+	: _old (dcpomatic_log)
+{
+	dcpomatic_log = log;
+}
+
+
+LogSwitcher::~LogSwitcher ()
+{
+	dcpomatic_log = _old;
+}
+
+std::ostream&
+dcp::operator<< (std::ostream& s, dcp::Size i)
+{
+	s << i.width << "x" << i.height;
+	return s;
+}
+
+std::ostream&
+dcp::operator<< (std::ostream& s, Standard t)
+{
+	switch (t) {
+	case Standard::INTEROP:
+		s << "interop";
+		break;
+	case Standard::SMPTE:
+		s << "smpte";
+		break;
+	}
+	return s;
+}
+