Supporters update.
[dcpomatic.git] / test / atmos_test.cc
index e5846f20e5f3927dd723914c01f710ea0de1d124..9fdddc979306c79bb451437abe0bcf5490547ad0 100644 (file)
@@ -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.
 
 #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 }, false);
+       // atmos_1.mxf should contain all ones for its data, and it should have replaced atmos_0 in this DCP
+       check(vf, 1);
 }