X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=test%2Fatmos_test.cc;h=17073682520a945705f136a3244fad2f74a13c5d;hb=a6c4b4fa16d9c6597e362044b875f3d6df80753f;hp=fe64ed6a92d856ae4bc54d650dd8e85fc1305fea;hpb=5d937606cf96d163be98920fa5619114a28eec9d;p=dcpomatic.git

diff --git a/test/atmos_test.cc b/test/atmos_test.cc
index fe64ed6a9..170736825 100644
--- a/test/atmos_test.cc
+++ b/test/atmos_test.cc
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2020 Carl Hetherington <cth@carlh.net>
+    Copyright (C) 2020-2021 Carl Hetherington <cth@carlh.net>
 
     This file is part of DCP-o-matic.
 
@@ -25,66 +25,122 @@
 #include "lib/dcp_content.h"
 #include "lib/film.h"
 #include "test.h"
+#include <dcp/atmos_asset.h>
+#include <dcp/cpl.h>
+#include <dcp/dcp.h>
+#include <dcp/reel.h>
+#include <dcp/reel_atmos_asset.h>
 #include <boost/test/unit_test.hpp>
-#include <iostream>
 
 
+using std::make_shared;
+using std::shared_ptr;
+using std::shared_ptr;
 using std::string;
 using std::vector;
 using boost::optional;
-using boost::shared_ptr;
 
 
 BOOST_AUTO_TEST_CASE (atmos_passthrough_test)
 {
-	shared_ptr<Film> film = new_test_film2 ("atmos_passthrough_test");
-	boost::filesystem::path ref = TestPaths::private_data() / "atmos_asset.mxf";
-	shared_ptr<Content> content = content_factory (TestPaths::private_data() / "atmos_asset.mxf").front();
-	film->examine_and_add_content (content);
-	BOOST_REQUIRE (!wait_for_jobs());
+	Cleanup cl;
 
-	film->make_dcp ();
-	BOOST_REQUIRE (!wait_for_jobs());
+	auto film = new_test_film2 (
+		"atmos_passthrough_test",
+		content_factory(TestPaths::private_data() / "atmos_asset.mxf"),
+		&cl
+		);
+
+	make_and_verify_dcp (film, {dcp::VerificationNote::Code::MISSING_CPL_METADATA});
 
+	auto ref = TestPaths::private_data() / "atmos_asset.mxf";
 	BOOST_REQUIRE (mxf_atmos_files_same(ref, dcp_file(film, "atmos"), true));
+
+	cl.run ();
 }
 
 
 BOOST_AUTO_TEST_CASE (atmos_encrypted_passthrough_test)
 {
-	shared_ptr<Film> film = new_test_film2 ("atmos_encrypted_passthrough_test");
-	boost::filesystem::path ref = TestPaths::private_data() / "atmos_asset.mxf";
-	shared_ptr<Content> content = content_factory (TestPaths::private_data() / "atmos_asset.mxf").front();
-	film->examine_and_add_content (content);
-	BOOST_REQUIRE (!wait_for_jobs());
+	Cleanup cl;
+
+	auto ref = TestPaths::private_data() / "atmos_asset.mxf";
+	auto content = content_factory(TestPaths::private_data() / "atmos_asset.mxf");
+	auto film = new_test_film2 ("atmos_encrypted_passthrough_test", content, &cl);
 
 	film->set_encrypted (true);
 	film->_key = dcp::Key ("4fac12927eb122af1c2781aa91f3a4cc");
-	film->make_dcp ();
-	BOOST_REQUIRE (!wait_for_jobs());
+	make_and_verify_dcp (film, { dcp::VerificationNote::Code::MISSING_CPL_METADATA });
 
 	BOOST_REQUIRE (!mxf_atmos_files_same(ref, dcp_file(film, "atmos")));
 
-	dcp::EncryptedKDM kdm = film->make_kdm (
-		Config::instance()->decryption_chain()->leaf(),
-		vector<string>(),
-		dcp_file(film, "cpl"),
-		dcp::LocalTime(),
-		dcp::LocalTime(),
-		dcp::MODIFIED_TRANSITIONAL_1,
-		false,
-		optional<int>()
-		);
+	auto signer = Config::instance()->signer_chain();
+	BOOST_REQUIRE(signer->valid());
 
-	shared_ptr<Film> film2 = new_test_film2 ("atmos_encrypted_passthrough_test2");
-	shared_ptr<DCPContent> content2 (new DCPContent(film->dir(film->dcp_name())));
-	content2->add_kdm (kdm);
-	film2->examine_and_add_content (content2);
-	BOOST_REQUIRE (!wait_for_jobs());
+	auto const decrypted_kdm = film->make_kdm(dcp_file(film, "cpl"), dcp::LocalTime(), dcp::LocalTime());
+	auto const kdm = decrypted_kdm.encrypt(signer, Config::instance()->decryption_chain()->leaf(), {}, dcp::Formulation::MODIFIED_TRANSITIONAL_1, false, {});
 
-	film2->make_dcp ();
-	BOOST_REQUIRE (!wait_for_jobs());
+	auto content2 = make_shared<DCPContent>(film->dir(film->dcp_name()));
+	content2->add_kdm (kdm);
+	auto film2 = new_test_film2 ("atmos_encrypted_passthrough_test2", {content2}, &cl);
+	make_and_verify_dcp (film2, { dcp::VerificationNote::Code::MISSING_CPL_METADATA });
 
 	BOOST_CHECK (mxf_atmos_files_same(ref, dcp_file(film2, "atmos"), true));
+
+	cl.run ();
+}
+
+
+BOOST_AUTO_TEST_CASE (atmos_trim_test)
+{
+	Cleanup cl;
+
+	auto ref = TestPaths::private_data() / "atmos_asset.mxf";
+	auto content = content_factory(TestPaths::private_data() / "atmos_asset.mxf");
+	auto film = new_test_film2 ("atmos_trim_test", content, &cl);
+
+	content[0]->set_trim_start(film, dcpomatic::ContentTime::from_seconds(1));
+
+	/* Just check that the encode runs; I'm not sure how to test the MXF */
+	make_and_verify_dcp (film, { dcp::VerificationNote::Code::MISSING_CPL_METADATA });
+}
+
+
+BOOST_AUTO_TEST_CASE(atmos_replace_test)
+{
+	auto const frames = 240;
+
+	auto check = [](shared_ptr<const Film> film, int data) {
+		dcp::DCP dcp(film->dir(film->dcp_name()));
+		dcp.read();
+		BOOST_REQUIRE_EQUAL(dcp.cpls().size(), 1U);
+		BOOST_REQUIRE_EQUAL(dcp.cpls()[0]->reels().size(), 1U);
+		BOOST_REQUIRE(dcp.cpls()[0]->reels()[0]->atmos());
+		auto asset = dcp.cpls()[0]->reels()[0]->atmos()->asset();
+		BOOST_REQUIRE(asset);
+		auto reader = asset->start_read();
+		for (int i = 0; i < frames; ++i) {
+			auto frame = reader->get_frame(i);
+			BOOST_REQUIRE(frame);
+			for (int j = 0; j < frame->size(); ++j) {
+				BOOST_REQUIRE_EQUAL(frame->data()[j], data);
+			}
+		}
+	};
+
+	auto atmos_0 = content_factory("test/data/atmos_0.mxf");
+	auto ov = new_test_film2("atmos_merge_test_ov", atmos_0);
+	make_and_verify_dcp(ov, { dcp::VerificationNote::Code::MISSING_CPL_METADATA });
+	// atmos_0.mxf should contain all zeros for its data
+	check(ov, 0);
+
+	auto atmos_1 = content_factory("test/data/atmos_1.mxf");
+	auto ov_content = std::make_shared<DCPContent>(boost::filesystem::path("build/test/atmos_merge_test_ov") / ov->dcp_name());
+	auto vf = new_test_film2("atmos_merge_test_vf", { ov_content, atmos_1.front() });
+	ov_content->set_reference_video(true);
+	atmos_1.front()->set_position(vf, dcpomatic::DCPTime());
+	make_and_verify_dcp(vf, { dcp::VerificationNote::Code::MISSING_CPL_METADATA, dcp::VerificationNote::Code::EXTERNAL_ASSET });
+	// atmos_1.mxf should contain all ones for its data, and it should have replaced atmos_0 in this DCP
+	check(vf, 1);
 }