+void
+check_xml (xmlpp::Element* ref, xmlpp::Element* test, list<string> ignore)
+{
+ BOOST_CHECK_EQUAL (ref->get_name (), test->get_name ());
+ BOOST_CHECK_EQUAL (ref->get_namespace_prefix (), test->get_namespace_prefix ());
+
+ if (find (ignore.begin(), ignore.end(), ref->get_name()) != ignore.end ()) {
+ return;
+ }
+
+ auto ref_children = ref->get_children ();
+ auto test_children = test->get_children ();
+ BOOST_REQUIRE_MESSAGE (
+ ref_children.size() == test_children.size(),
+ ref->get_name() << " has " << ref_children.size() << " or " << test_children.size() << " children"
+ );
+
+ auto k = ref_children.begin ();
+ auto l = test_children.begin ();
+ while (k != ref_children.end ()) {
+
+ /* XXX: should be doing xmlpp::EntityReference, xmlpp::XIncludeEnd, xmlpp::XIncludeStart */
+
+ auto ref_el = dynamic_cast<xmlpp::Element*>(*k);
+ auto test_el = dynamic_cast<xmlpp::Element*>(*l);
+ BOOST_CHECK ((ref_el && test_el) || (!ref_el && !test_el));
+ if (ref_el && test_el) {
+ check_xml (ref_el, test_el, ignore);
+ }
+
+ auto ref_cn = dynamic_cast<xmlpp::ContentNode*>(*k);
+ auto test_cn = dynamic_cast<xmlpp::ContentNode*>(*l);
+ BOOST_CHECK ((ref_cn && test_cn) || (!ref_cn && !test_cn));
+ if (ref_cn && test_cn) {
+ BOOST_CHECK_EQUAL (ref_cn->get_content(), test_cn->get_content ());
+ }
+
+ ++k;
+ ++l;
+ }
+
+ auto ref_attributes = ref->get_attributes ();
+ auto test_attributes = test->get_attributes ();
+ BOOST_CHECK_EQUAL (ref_attributes.size(), test_attributes.size ());
+
+ auto m = ref_attributes.begin();
+ auto n = test_attributes.begin();
+ while (m != ref_attributes.end ()) {
+ BOOST_CHECK_EQUAL ((*m)->get_name(), (*n)->get_name());
+ BOOST_CHECK_EQUAL ((*m)->get_value(), (*n)->get_value());
+
+ ++m;
+ ++n;
+ }
+}
+
+void
+check_xml (boost::filesystem::path ref, boost::filesystem::path test, list<string> ignore)
+{
+ auto ref_parser = new xmlpp::DomParser(ref.string());
+ auto ref_root = ref_parser->get_document()->get_root_node();
+ auto test_parser = new xmlpp::DomParser(test.string());
+ auto test_root = test_parser->get_document()->get_root_node();
+
+ check_xml (ref_root, test_root, ignore);
+}
+
+bool
+wait_for_jobs ()
+{
+ auto jm = JobManager::instance ();
+ while (jm->work_to_do()) {
+ while (signal_manager->ui_idle()) {}
+ dcpomatic_sleep_seconds (1);
+ }
+
+ if (jm->errors ()) {
+ int N = 0;
+ for (auto i: jm->_jobs) {
+ if (i->finished_in_error()) {
+ ++N;
+ }
+ }
+ cerr << N << " errors.\n";
+
+ for (auto i: jm->_jobs) {
+ if (i->finished_in_error()) {
+ cerr << i->name() << ":\n"
+ << "\tsummary: " << i->error_summary() << "\n"
+ << "\tdetails: " << i->error_details() << "\n";
+ }
+ }
+ }
+
+ while (signal_manager->ui_idle ()) {}
+
+ if (jm->errors ()) {
+ JobManager::drop ();
+ return true;
+ }
+
+ return false;
+}
+
+
+class Memory
+{
+public:
+ Memory () {}
+
+ ~Memory ()
+ {
+ free (data);
+ }
+
+ Memory (Memory const&) = delete;
+ Memory& operator= (Memory const&) = delete;
+
+ uint8_t* data = nullptr;
+ size_t size = 0;
+};
+
+
+static void
+png_write_data (png_structp png_ptr, png_bytep data, png_size_t length)
+{
+ auto 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, char const * message)
+{
+ throw EncodeError (String::compose("Error during PNG write: %1", message));
+}
+
+
+void
+write_image (shared_ptr<const Image> image, boost::filesystem::path file)
+{
+ int png_color_type = PNG_COLOR_TYPE_RGB;
+ int bits_per_pixel = 0;
+ switch (image->pixel_format()) {
+ case AV_PIX_FMT_RGB24:
+ bits_per_pixel = 8;
+ break;
+ case AV_PIX_FMT_XYZ12LE:
+ 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);
+
+ auto const width = image->size().width;
+ auto const height = image->size().height;
+ auto const stride = image->stride()[0];
+
+ vector<vector<uint8_t>> data_to_write(height);
+ for (int y = 0; y < height; ++y) {
+ data_to_write[y].resize(stride);
+ }
+
+ switch (image->pixel_format()) {
+ case AV_PIX_FMT_RGB24:
+ for (int y = 0; y < height; ++y) {
+ memcpy(data_to_write[y].data(), image->data()[0] + y * stride, stride);
+ }
+ break;
+ case AV_PIX_FMT_XYZ12LE:
+ /* 16-bit pixel values must be written MSB first */
+ for (int y = 0; y < height; ++y) {
+ data_to_write[y].resize(stride);
+ uint8_t* original = image->data()[0] + y * stride;
+ for (int x = 0; x < width * 3; ++x) {
+ data_to_write[y][x * 2] = original[1];
+ data_to_write[y][x * 2 + 1] = original[0];
+ original += 2;
+ }
+ }
+ break;
+ default:
+ break;
+ }
+
+ png_byte ** row_pointers = reinterpret_cast<png_byte**>(png_malloc(png_ptr, image->size().height * sizeof(png_byte *)));
+ for (int y = 0; y < height; ++y) {
+ row_pointers[y] = reinterpret_cast<png_byte*>(data_to_write[y].data());
+ }
+
+ png_write_info (png_ptr, info_ptr);
+ png_write_image (png_ptr, row_pointers);
+ png_write_end (png_ptr, info_ptr);
+
+ png_destroy_write_struct (&png_ptr, &info_ptr);
+ png_free (png_ptr, row_pointers);
+
+ 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);
+}
+
+void
+check_one_frame (boost::filesystem::path dcp_dir, int64_t index, boost::filesystem::path ref)
+{
+ dcp::DCP dcp (dcp_dir);
+ dcp.read ();
+ auto asset = dynamic_pointer_cast<dcp::MonoPictureAsset> (dcp.cpls().front()->reels().front()->main_picture()->asset());
+ BOOST_REQUIRE (asset);
+ auto frame = asset->start_read()->get_frame(index);
+ auto ref_frame (new dcp::MonoPictureFrame (ref));
+
+ auto image = frame->xyz_image ();
+ auto 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::filesystem::path
+dcp_file (shared_ptr<const Film> film, string prefix)
+{
+ auto i = boost::filesystem::recursive_directory_iterator(film->dir(film->dcp_name()));
+ while (i != boost::filesystem::recursive_directory_iterator() && !boost::algorithm::starts_with(i->path().leaf().string(), prefix)) {
+ ++i;
+ }
+
+ BOOST_REQUIRE_MESSAGE(i != boost::filesystem::recursive_directory_iterator(), "Could not find file with prefix " << prefix);
+ return i->path();
+}
+
+boost::filesystem::path
+subtitle_file (shared_ptr<Film> film)
+{
+ for (auto i: boost::filesystem::recursive_directory_iterator(film->directory().get() / film->dcp_name(false))) {
+ if (boost::algorithm::starts_with(i.path().leaf().string(), "sub_")) {
+ return i.path();
+ }
+ }
+
+ BOOST_REQUIRE (false);
+ /* Remove warning */
+ return boost::filesystem::path("/");
+}
+
+
+void
+make_random_file (boost::filesystem::path path, size_t size)
+{
+ dcp::File random_file(path, "wb");
+ BOOST_REQUIRE (random_file);
+
+ boost::random::mt19937 rng(1);
+ boost::random::uniform_int_distribution<uint64_t> dist(0);
+
+ while (size > 0) {
+ auto this_time = std::min(size, size_t(8));
+ uint64_t random = dist(rng);
+ random_file.write(&random, this_time, 1);
+ size -= this_time;
+ }
+}
+
+
+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, dcp::Standard t)
+{
+ switch (t) {
+ case Standard::INTEROP:
+ s << "interop";
+ break;
+ case Standard::SMPTE:
+ s << "smpte";
+ break;
+ }
+ return s;
+}
+
+std::ostream&
+operator<< (std::ostream&s, VideoFrameType f)
+{
+ s << video_frame_type_to_string(f);
+ return s;
+}
+
+
+void
+Cleanup::add (boost::filesystem::path path)
+{
+ _paths.push_back (path);
+}
+
+
+void
+Cleanup::run ()
+{
+ boost::system::error_code ec;
+ for (auto i: _paths) {
+ boost::filesystem::remove_all (i, ec);
+ }
+}
+
+
+void stage (string, boost::optional<boost::filesystem::path>) {}
+void progress (float) {}
+
+
+void
+make_and_verify_dcp (shared_ptr<Film> film, vector<dcp::VerificationNote::Code> ignore)
+{
+ film->write_metadata ();
+ make_dcp (film, TranscodeJob::ChangedBehaviour::IGNORE);
+ BOOST_REQUIRE (!wait_for_jobs());
+ auto notes = dcp::verify ({film->dir(film->dcp_name())}, &stage, &progress, TestPaths::xsd());
+ bool ok = true;
+ for (auto i: notes) {
+ if (find(ignore.begin(), ignore.end(), i.code()) == ignore.end()) {
+ std::cout << "\t" << dcp::note_to_string(i) << "\n";
+ ok = false;
+ }
+ }
+ BOOST_CHECK(ok);
+}
+
+
+void
+check_int_close (int a, int b, int d)
+{
+ BOOST_CHECK_MESSAGE (std::abs(a - b) < d, a << " differs from " << b << " by more than " << d);
+}
+
+
+void
+check_int_close (std::pair<int, int> a, std::pair<int, int> b, int d)
+{
+ check_int_close (a.first, b.first, d);
+ check_int_close (a.second, b.second, d);
+}
+
+
+ConfigRestorer::~ConfigRestorer()
+{
+ setup_test_config();
+}
+
+
+boost::filesystem::path
+find_file (boost::filesystem::path dir, string filename_part)
+{
+ boost::optional<boost::filesystem::path> found;
+ for (auto i: boost::filesystem::directory_iterator(dir)) {
+ if (i.path().filename().string().find(filename_part) != string::npos) {
+ BOOST_REQUIRE (!found);
+ found = i;
+ }
+ }
+ BOOST_REQUIRE (found);
+ return *found;
+}