test/j2k_encode_threading_test.cc

   1 #include "lib/config.h"
   2 #include "lib/content_factory.h"
   3 #include "lib/dcp_encoder.h"
   4 #include "lib/dcp_transcode_job.h"
   5 #include "lib/encode_server_description.h"
   6 #include "lib/film.h"
   7 #include "lib/j2k_encoder.h"
   8 #include "lib/job_manager.h"
   9 #include "lib/make_dcp.h"
  10 #include "lib/transcode_job.h"
  11 #include "test.h"
  12 #include <dcp/cpl.h>
  13 #include <dcp/dcp.h>
  14 #include <dcp/reel.h>
  15 #include <dcp/reel_picture_asset.h>
  16 #include <boost/test/unit_test.hpp>
  17
  18
  19 using std::dynamic_pointer_cast;
  20 using std::list;
  21
  22
  23 BOOST_AUTO_TEST_CASE(local_threads_created_and_destroyed)
  24 {
  25         auto film = new_test_film2("local_threads_created_and_destroyed", {});
  26         Writer writer(film, {});
  27         J2KEncoder encoder(film, writer);
  28
  29         encoder.remake_threads(32, 0, {});
  30         BOOST_CHECK_EQUAL(encoder._threads.size(), 32U);
  31
  32         encoder.remake_threads(9, 0, {});
  33         BOOST_CHECK_EQUAL(encoder._threads.size(), 9U);
  34
  35         encoder.end();
  36         BOOST_CHECK_EQUAL(encoder._threads.size(), 0U);
  37 }
  38
  39
  40 BOOST_AUTO_TEST_CASE(remote_threads_created_and_destroyed)
  41 {
  42         auto film = new_test_film2("remote_threads_created_and_destroyed", {});
  43         Writer writer(film, {});
  44         J2KEncoder encoder(film, writer);
  45
  46         list<EncodeServerDescription> servers = {
  47                 { "fred", 7, SERVER_LINK_VERSION },
  48                 { "jim", 2, SERVER_LINK_VERSION },
  49                 { "sheila", 14, SERVER_LINK_VERSION },
  50         };
  51
  52         encoder.remake_threads(0, 0, servers);
  53         BOOST_CHECK_EQUAL(encoder._threads.size(), 7U + 2U + 14U);
  54
  55         servers = {
  56                 { "fred", 7, SERVER_LINK_VERSION },
  57                 { "jim", 5, SERVER_LINK_VERSION },
  58                 { "sheila", 14, SERVER_LINK_VERSION },
  59         };
  60
  61         encoder.remake_threads(0, 0, servers);
  62         BOOST_CHECK_EQUAL(encoder._threads.size(), 7U + 5U + 14U);
  63
  64         servers = {
  65                 { "fred", 0, SERVER_LINK_VERSION },
  66                 { "jim", 0, SERVER_LINK_VERSION },
  67                 { "sheila", 11, SERVER_LINK_VERSION },
  68         };
  69
  70         encoder.remake_threads(0, 0, servers);
  71         BOOST_CHECK_EQUAL(encoder._threads.size(), 11U);
  72 }
  73
  74
  75 BOOST_AUTO_TEST_CASE(frames_not_lost_when_threads_disappear)
  76 {
  77         auto content = content_factory(TestPaths::private_data() / "clapperboard.mp4");
  78         auto film = new_test_film2("frames_not_lost", content);
  79         film->write_metadata();
  80         auto job = make_dcp(film, TranscodeJob::ChangedBehaviour::IGNORE);
  81         auto& encoder = dynamic_pointer_cast<DCPEncoder>(job->_encoder)->_j2k_encoder;
  82
  83         while (JobManager::instance()->work_to_do()) {
  84                 encoder.remake_threads(rand() % 8, 0, {});
  85                 dcpomatic_sleep_seconds(1);
  86         }
  87
  88         BOOST_CHECK(!JobManager::instance()->errors());
  89
  90         dcp::DCP dcp(film->dir(film->dcp_name()));
  91         dcp.read();
  92         BOOST_REQUIRE_EQUAL(dcp.cpls().size(), 1U);
  93         BOOST_REQUIRE_EQUAL(dcp.cpls()[0]->reels().size(), 1U);
  94         BOOST_REQUIRE_EQUAL(dcp.cpls()[0]->reels()[0]->main_picture()->intrinsic_duration(), 423U);
  95 }
  96