2 Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 #include <libxml++/libxml++.h>
24 #include <libdcp/dcp.h>
25 #include "lib/config.h"
27 #include "lib/ui_signaller.h"
29 #include "lib/job_manager.h"
31 #include "lib/cross.h"
32 #include "lib/server_finder.h"
33 #include "lib/image.h"
34 #define BOOST_TEST_DYN_LINK
35 #define BOOST_TEST_MODULE dcpomatic_test
36 #include <boost/test/unit_test.hpp>
44 using boost::shared_ptr;
46 class TestUISignaller : public UISignaller
49 /* No wakes in tests: we call ui_idle ourselves */
62 Config::instance()->set_num_local_encoding_threads (1);
63 Config::instance()->set_server_port_base (61920);
64 Config::instance()->set_default_dci_metadata (DCIMetadata ());
65 Config::instance()->set_default_container (static_cast<Ratio*> (0));
66 Config::instance()->set_default_dcp_content_type (static_cast<DCPContentType*> (0));
68 ServerFinder::instance()->disable ();
70 ui_signaller = new TestUISignaller ();
74 BOOST_GLOBAL_FIXTURE (TestConfig);
76 boost::filesystem::path
77 test_film_dir (string name)
79 boost::filesystem::path p;
87 new_test_film (string name)
89 boost::filesystem::path p = test_film_dir (name);
90 if (boost::filesystem::exists (p)) {
91 boost::filesystem::remove_all (p);
94 shared_ptr<Film> f = shared_ptr<Film> (new Film (p.string()));
100 check_file (string ref, string check)
102 uintmax_t N = boost::filesystem::file_size (ref);
103 BOOST_CHECK_EQUAL (N, boost::filesystem::file_size(check));
104 FILE* ref_file = fopen (ref.c_str(), "rb");
105 BOOST_CHECK (ref_file);
106 FILE* check_file = fopen (check.c_str(), "rb");
107 BOOST_CHECK (check_file);
109 int const buffer_size = 65536;
110 uint8_t* ref_buffer = new uint8_t[buffer_size];
111 uint8_t* check_buffer = new uint8_t[buffer_size];
114 uintmax_t this_time = min (uintmax_t (buffer_size), N);
115 size_t r = fread (ref_buffer, 1, this_time, ref_file);
116 BOOST_CHECK_EQUAL (r, this_time);
117 r = fread (check_buffer, 1, this_time, check_file);
118 BOOST_CHECK_EQUAL (r, this_time);
120 BOOST_CHECK_EQUAL (memcmp (ref_buffer, check_buffer, this_time), 0);
125 delete[] check_buffer;
132 note (libdcp::NoteType t, string n)
134 if (t == libdcp::ERROR) {
140 check_dcp (string ref, string check)
142 libdcp::DCP ref_dcp (ref);
144 libdcp::DCP check_dcp (check);
147 libdcp::EqualityOptions options;
148 options.max_mean_pixel_error = 5;
149 options.max_std_dev_pixel_error = 5;
150 options.max_audio_sample_error = 255;
151 options.cpl_names_can_differ = true;
152 options.mxf_names_can_differ = true;
154 BOOST_CHECK (ref_dcp.equals (check_dcp, options, boost::bind (note, _1, _2)));
158 check_xml (xmlpp::Element* ref, xmlpp::Element* test, list<string> ignore)
160 BOOST_CHECK_EQUAL (ref->get_name (), test->get_name ());
161 BOOST_CHECK_EQUAL (ref->get_namespace_prefix (), test->get_namespace_prefix ());
163 if (find (ignore.begin(), ignore.end(), ref->get_name()) != ignore.end ()) {
167 xmlpp::Element::NodeList ref_children = ref->get_children ();
168 xmlpp::Element::NodeList test_children = test->get_children ();
169 BOOST_CHECK_EQUAL (ref_children.size (), test_children.size ());
171 xmlpp::Element::NodeList::iterator k = ref_children.begin ();
172 xmlpp::Element::NodeList::iterator l = test_children.begin ();
173 while (k != ref_children.end ()) {
175 /* XXX: should be doing xmlpp::EntityReference, xmlpp::XIncludeEnd, xmlpp::XIncludeStart */
177 xmlpp::Element* ref_el = dynamic_cast<xmlpp::Element*> (*k);
178 xmlpp::Element* test_el = dynamic_cast<xmlpp::Element*> (*l);
179 BOOST_CHECK ((ref_el && test_el) || (!ref_el && !test_el));
180 if (ref_el && test_el) {
181 check_xml (ref_el, test_el, ignore);
184 xmlpp::ContentNode* ref_cn = dynamic_cast<xmlpp::ContentNode*> (*k);
185 xmlpp::ContentNode* test_cn = dynamic_cast<xmlpp::ContentNode*> (*l);
186 BOOST_CHECK ((ref_cn && test_cn) || (!ref_cn && !test_cn));
187 if (ref_cn && test_cn) {
188 BOOST_CHECK_EQUAL (ref_cn->get_content(), test_cn->get_content ());
191 xmlpp::Attribute* ref_at = dynamic_cast<xmlpp::Attribute*> (*k);
192 xmlpp::Attribute* test_at = dynamic_cast<xmlpp::Attribute*> (*l);
193 BOOST_CHECK ((ref_at && test_at) || (!ref_at && !test_at));
194 if (ref_at && test_at) {
195 BOOST_CHECK_EQUAL (ref_at->get_name(), test_at->get_name ());
196 BOOST_CHECK_EQUAL (ref_at->get_value(), test_at->get_value ());
205 check_xml (boost::filesystem::path ref, boost::filesystem::path test, list<string> ignore)
207 xmlpp::DomParser* ref_parser = new xmlpp::DomParser (ref.string ());
208 xmlpp::Element* ref_root = ref_parser->get_document()->get_root_node ();
209 xmlpp::DomParser* test_parser = new xmlpp::DomParser (test.string ());
210 xmlpp::Element* test_root = test_parser->get_document()->get_root_node ();
212 check_xml (ref_root, test_root, ignore);
218 JobManager* jm = JobManager::instance ();
219 while (jm->work_to_do ()) {
220 ui_signaller->ui_idle ();
223 for (list<shared_ptr<Job> >::iterator i = jm->_jobs.begin(); i != jm->_jobs.end(); ++i) {
224 if ((*i)->finished_in_error ()) {
225 cerr << (*i)->error_summary () << "\n"
226 << (*i)->error_details () << "\n";
231 BOOST_CHECK (!jm->errors());
233 ui_signaller->ui_idle ();
237 write_image (shared_ptr<const Image> image, boost::filesystem::path file)
239 using namespace MagickCore;
241 Magick::Image m (image->size().width, image->size().height, "ARGB", CharPixel, (void *) image->data()[0]);
242 m.write (file.string ());