X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=test%2Fclient_server_test.cc;h=1f77f7f907cc718a0bcf61dd32b41881c9dd34c3;hp=54bbd520f8dd9c8d37ca2d1b30efd00cf2bca2f4;hb=3828baf56467224f5d44049bf1e7a7ed11f43a05;hpb=2e504b33eb9f38cac629ad31b7c107fb0cf5efda diff --git a/test/client_server_test.cc b/test/client_server_test.cc index 54bbd520f..1f77f7f90 100644 --- a/test/client_server_test.cc +++ b/test/client_server_test.cc @@ -1,49 +1,64 @@ /* - Copyright (C) 2012 Carl Hetherington + Copyright (C) 2012-2014 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ -#include -#include -#include "lib/server.h" +/** @file test/client_server_test.cc + * @brief Test the server class. + * + * Create a test image and then encode it using the standard mechanism + * and also using a EncodeServer object running on localhost. Compare the resulting + * encoded data to check that they are the same. + */ + +#include "lib/encode_server.h" #include "lib/image.h" #include "lib/cross.h" -#include "lib/dcp_video_frame.h" +#include "lib/dcp_video.h" +#include "lib/player_video.h" +#include "lib/raw_image_proxy.h" +#include "lib/j2k_image_proxy.h" +#include "lib/encode_server_description.h" +#include "lib/file_log.h" +#include +#include using std::list; using boost::shared_ptr; using boost::thread; +using boost::optional; +using dcp::Data; void -do_remote_encode (shared_ptr frame, ServerDescription description, shared_ptr locally_encoded) +do_remote_encode (shared_ptr frame, EncodeServerDescription description, Data locally_encoded) { - shared_ptr remotely_encoded; - BOOST_CHECK_NO_THROW (remotely_encoded = frame->encode_remotely (description)); - BOOST_CHECK (remotely_encoded); - - BOOST_CHECK_EQUAL (locally_encoded->size(), remotely_encoded->size()); - BOOST_CHECK_EQUAL (memcmp (locally_encoded->data(), remotely_encoded->data(), locally_encoded->size()), 0); + Data remotely_encoded; + BOOST_CHECK_NO_THROW (remotely_encoded = frame->encode_remotely (description, 60)); + + BOOST_CHECK_EQUAL (locally_encoded.size(), remotely_encoded.size()); + BOOST_CHECK_EQUAL (memcmp (locally_encoded.data().get(), remotely_encoded.data().get(), locally_encoded.size()), 0); } -BOOST_AUTO_TEST_CASE (client_server_test) +BOOST_AUTO_TEST_CASE (client_server_test_rgb) { - shared_ptr image (new Image (PIX_FMT_RGB24, dcp::Size (1998, 1080), true)); + shared_ptr image (new Image (AV_PIX_FMT_RGB24, dcp::Size (1998, 1080), true)); uint8_t* p = image->data()[0]; - + for (int y = 0; y < 1080; ++y) { uint8_t* q = p; for (int x = 0; x < 1998; ++x) { @@ -54,7 +69,7 @@ BOOST_AUTO_TEST_CASE (client_server_test) p += image->stride()[0]; } - shared_ptr sub_image (new Image (PIX_FMT_RGBA, dcp::Size (100, 200), true)); + shared_ptr sub_image (new Image (AV_PIX_FMT_RGBA, dcp::Size (100, 200), true)); p = sub_image->data()[0]; for (int y = 0; y < 200; ++y) { uint8_t* q = p; @@ -67,16 +82,108 @@ BOOST_AUTO_TEST_CASE (client_server_test) p += sub_image->stride()[0]; } -// shared_ptr subtitle (new Subtitle (Position (50, 60), sub_image)); + shared_ptr log (new FileLog ("build/test/client_server_test_rgb.log")); - shared_ptr log (new FileLog ("build/test/client_server_test.log")); + shared_ptr pvf ( + new PlayerVideo ( + shared_ptr (new RawImageProxy (image)), + DCPTime (), + Crop (), + optional (), + dcp::Size (1998, 1080), + dcp::Size (1998, 1080), + EYES_BOTH, + PART_WHOLE, + ColourConversion () + ) + ); + + pvf->set_subtitle (PositionImage (sub_image, Position (50, 60))); - shared_ptr frame ( - new DCPVideoFrame ( - image, + shared_ptr frame ( + new DCPVideo ( + pvf, 0, + 24, + 200000000, + RESOLUTION_2K, + log + ) + ); + + Data locally_encoded = frame->encode_locally (boost::bind (&Log::dcp_log, log.get(), _1, _2)); + + EncodeServer* server = new EncodeServer (log, true, 2); + + new thread (boost::bind (&EncodeServer::run, server)); + + /* Let the server get itself ready */ + dcpomatic_sleep (1); + + EncodeServerDescription description ("localhost", 2); + + list threads; + for (int i = 0; i < 8; ++i) { + threads.push_back (new thread (boost::bind (do_remote_encode, frame, description, locally_encoded))); + } + + for (list::iterator i = threads.begin(); i != threads.end(); ++i) { + (*i)->join (); + } + + for (list::iterator i = threads.begin(); i != threads.end(); ++i) { + delete *i; + } + + delete server; +} + +BOOST_AUTO_TEST_CASE (client_server_test_yuv) +{ + shared_ptr image (new Image (AV_PIX_FMT_YUV420P, dcp::Size (1998, 1080), true)); + + for (int i = 0; i < image->planes(); ++i) { + uint8_t* p = image->data()[i]; + for (int j = 0; j < image->line_size()[i]; ++j) { + *p++ = j % 256; + } + } + + shared_ptr sub_image (new Image (AV_PIX_FMT_RGBA, dcp::Size (100, 200), true)); + uint8_t* p = sub_image->data()[0]; + for (int y = 0; y < 200; ++y) { + uint8_t* q = p; + for (int x = 0; x < 100; ++x) { + *q++ = y % 256; + *q++ = x % 256; + *q++ = (x + y) % 256; + *q++ = 1; + } + p += sub_image->stride()[0]; + } + + shared_ptr log (new FileLog ("build/test/client_server_test_yuv.log")); + + shared_ptr pvf ( + new PlayerVideo ( + shared_ptr (new RawImageProxy (image)), + DCPTime (), + Crop (), + optional (), + dcp::Size (1998, 1080), + dcp::Size (1998, 1080), EYES_BOTH, - ColourConversion (), + PART_WHOLE, + ColourConversion () + ) + ); + + pvf->set_subtitle (PositionImage (sub_image, Position (50, 60))); + + shared_ptr frame ( + new DCPVideo ( + pvf, + 0, 24, 200000000, RESOLUTION_2K, @@ -84,17 +191,16 @@ BOOST_AUTO_TEST_CASE (client_server_test) ) ); - shared_ptr locally_encoded = frame->encode_locally (); - BOOST_ASSERT (locally_encoded); - - Server* server = new Server (log, true); + Data locally_encoded = frame->encode_locally (boost::bind (&Log::dcp_log, log.get(), _1, _2)); + + EncodeServer* server = new EncodeServer (log, true, 2); - new thread (boost::bind (&Server::run, server, 2)); + new thread (boost::bind (&EncodeServer::run, server)); /* Let the server get itself ready */ dcpomatic_sleep (1); - ServerDescription description ("localhost", 2); + EncodeServerDescription description ("localhost", 2); list threads; for (int i = 0; i < 8; ++i) { @@ -108,5 +214,98 @@ BOOST_AUTO_TEST_CASE (client_server_test) for (list::iterator i = threads.begin(); i != threads.end(); ++i) { delete *i; } + + delete server; } +BOOST_AUTO_TEST_CASE (client_server_test_j2k) +{ + shared_ptr image (new Image (AV_PIX_FMT_YUV420P, dcp::Size (1998, 1080), true)); + + for (int i = 0; i < image->planes(); ++i) { + uint8_t* p = image->data()[i]; + for (int j = 0; j < image->line_size()[i]; ++j) { + *p++ = j % 256; + } + } + + shared_ptr log (new FileLog ("build/test/client_server_test_j2k.log")); + + shared_ptr raw_pvf ( + new PlayerVideo ( + shared_ptr (new RawImageProxy (image)), + DCPTime (), + Crop (), + optional (), + dcp::Size (1998, 1080), + dcp::Size (1998, 1080), + EYES_BOTH, + PART_WHOLE, + ColourConversion () + ) + ); + + shared_ptr raw_frame ( + new DCPVideo ( + raw_pvf, + 0, + 24, + 200000000, + RESOLUTION_2K, + log + ) + ); + + Data raw_locally_encoded = raw_frame->encode_locally (boost::bind (&Log::dcp_log, log.get(), _1, _2)); + + shared_ptr j2k_pvf ( + new PlayerVideo ( + shared_ptr (new J2KImageProxy (raw_locally_encoded, dcp::Size (1998, 1080))), + DCPTime (), + Crop (), + optional (), + dcp::Size (1998, 1080), + dcp::Size (1998, 1080), + EYES_BOTH, + PART_WHOLE, + PresetColourConversion::all().front().conversion + ) + ); + + shared_ptr j2k_frame ( + new DCPVideo ( + j2k_pvf, + 0, + 24, + 200000000, + RESOLUTION_2K, + log + ) + ); + + Data j2k_locally_encoded = j2k_frame->encode_locally (boost::bind (&Log::dcp_log, log.get(), _1, _2)); + + EncodeServer* server = new EncodeServer (log, true, 2); + + new thread (boost::bind (&EncodeServer::run, server)); + + /* Let the server get itself ready */ + dcpomatic_sleep (1); + + EncodeServerDescription description ("localhost", 2); + + list threads; + for (int i = 0; i < 8; ++i) { + threads.push_back (new thread (boost::bind (do_remote_encode, j2k_frame, description, j2k_locally_encoded))); + } + + for (list::iterator i = threads.begin(); i != threads.end(); ++i) { + (*i)->join (); + } + + for (list::iterator i = threads.begin(); i != threads.end(); ++i) { + delete *i; + } + + delete server; +}