X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fdcp_video.cc;h=e1666ea756107b74a2a88a6b51dc17ce628049ff;hb=ef578569611e2fa0e91bdd5a726fc0c251540e6d;hp=7987f92b4b5b59f1b06d54f639e00edaa197f3b6;hpb=53ca2dda0e9138a251adc186a745f7943fc1abc1;p=dcpomatic.git diff --git a/src/lib/dcp_video.cc b/src/lib/dcp_video.cc index 7987f92b4..e1666ea75 100644 --- a/src/lib/dcp_video.cc +++ b/src/lib/dcp_video.cc @@ -1,23 +1,24 @@ /* - Copyright (C) 2012-2015 Carl Hetherington - Taken from code Copyright (C) 2010-2011 Terrence Meiczinger + Copyright (C) 2012-2021 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 . */ + /** @file src/dcp_video_frame.cc * @brief A single frame of video destined for a DCP. * @@ -28,40 +29,46 @@ * of images that require encoding. */ -#include "dcp_video.h" + +#include "compose.hpp" #include "config.h" -#include "exceptions.h" -#include "server_description.h" +#include "cross.h" +#include "dcp_video.h" +#include "dcpomatic_log.h" #include "dcpomatic_socket.h" +#include "encode_server_description.h" +#include "exceptions.h" #include "image.h" #include "log.h" -#include "cross.h" #include "player_video.h" -#include "raw_convert.h" -#include "compose.hpp" +#include "rng.h" +#include "warnings.h" #include +#include #include #include #include -#include +DCPOMATIC_DISABLE_WARNINGS #include +DCPOMATIC_ENABLE_WARNINGS #include #include #include #include #include -#define LOG_GENERAL(...) _log->log (String::compose (__VA_ARGS__), LogEntry::TYPE_GENERAL); -#define LOG_DEBUG_ENCODE(...) _log->log (String::compose (__VA_ARGS__), LogEntry::TYPE_DEBUG_ENCODE); -#define LOG_TIMING(...) _log->log (String::compose (__VA_ARGS__), LogEntry::TYPE_TIMING); - #include "i18n.h" -using std::string; using std::cout; -using boost::shared_ptr; +using std::make_shared; +using std::shared_ptr; +using std::string; using dcp::Size; -using dcp::Data; +using dcp::ArrayData; +using dcp::raw_convert; +#if BOOST_VERSION >= 106100 +using namespace boost::placeholders; +#endif #define DCI_COEFFICENT (48.0 / 52.37) @@ -69,29 +76,26 @@ using dcp::Data; * @param frame Input frame. * @param index Index of the frame within the DCP. * @param bw J2K bandwidth to use (see Config::j2k_bandwidth ()) - * @param l Log to write to. */ DCPVideo::DCPVideo ( - shared_ptr frame, int index, int dcp_fps, int bw, Resolution r, shared_ptr l + shared_ptr frame, int index, int dcp_fps, int bw, Resolution r ) : _frame (frame) , _index (index) , _frames_per_second (dcp_fps) , _j2k_bandwidth (bw) , _resolution (r) - , _log (l) { } -DCPVideo::DCPVideo (shared_ptr frame, shared_ptr node, shared_ptr log) +DCPVideo::DCPVideo (shared_ptr frame, shared_ptr node) : _frame (frame) - , _log (log) { _index = node->number_child ("Index"); _frames_per_second = node->number_child ("FramesPerSecond"); _j2k_bandwidth = node->number_child ("J2KBandwidth"); - _resolution = Resolution (node->optional_number_child("Resolution").get_value_or (RESOLUTION_2K)); + _resolution = Resolution (node->optional_number_child("Resolution").get_value_or(static_cast(Resolution::TWO_K))); } shared_ptr @@ -99,7 +103,7 @@ DCPVideo::convert_to_xyz (shared_ptr frame, dcp::NoteHandler { shared_ptr xyz; - shared_ptr image = frame->image (note); + auto image = frame->image (bind(&PlayerVideo::keep_xyz_or_rgb, _1), VideoRange::FULL, true, false); if (frame->colour_conversion()) { xyz = dcp::rgb_to_xyz ( image->data()[0], @@ -109,7 +113,7 @@ DCPVideo::convert_to_xyz (shared_ptr frame, dcp::NoteHandler note ); } else { - xyz = dcp::xyz_to_xyz (image->data()[0], image->size(), image->stride()[0]); + xyz = make_shared(image->data()[0], image->size(), image->stride()[0]); } return xyz; @@ -118,27 +122,63 @@ DCPVideo::convert_to_xyz (shared_ptr frame, dcp::NoteHandler /** J2K-encode this frame on the local host. * @return Encoded data. */ -Data -DCPVideo::encode_locally (dcp::NoteHandler note) +ArrayData +DCPVideo::encode_locally () { - shared_ptr xyz = convert_to_xyz (_frame, note); - - Data enc = compress_j2k ( - convert_to_xyz (_frame, note), - _j2k_bandwidth, - _frames_per_second, - _frame->eyes() == EYES_LEFT || _frame->eyes() == EYES_RIGHT, - _resolution == RESOLUTION_4K + auto const comment = Config::instance()->dcp_j2k_comment(); + + ArrayData enc = {}; + int const minimum_size = Config::instance()->minimum_frame_size(); + LOG_GENERAL ("Using minimum frame size %1", minimum_size); + + auto xyz = convert_to_xyz (_frame, boost::bind(&Log::dcp_log, dcpomatic_log.get(), _1, _2)); + int noise_amount = 2; + while (true) { + enc = dcp::compress_j2k ( + xyz, + _j2k_bandwidth, + _frames_per_second, + _frame->eyes() == Eyes::LEFT || _frame->eyes() == Eyes::RIGHT, + _resolution == Resolution::FOUR_K, + comment.empty() ? "libdcp" : comment ); + if (enc.size() >= minimum_size) { + break; + } + + LOG_GENERAL (N_("Frame %1 encoded size was small (%2); adding noise at level %3"), _index, enc.size(), noise_amount); + + /* The JPEG2000 is too low-bitrate for some decoders DSS200 so add some noise + * and try again. This is slow but hopefully won't happen too often. We have to do + * convert_to_xyz() again because compress_j2k() corrupts its xyz parameter. + */ + + xyz = convert_to_xyz (_frame, boost::bind(&Log::dcp_log, dcpomatic_log.get(), _1, _2)); + auto size = xyz->size (); + auto pixels = size.width * size.height; + dcpomatic::RNG rng(42); + for (auto c = 0; c < 3; ++c) { + auto p = xyz->data(c); + for (auto i = 0; i < pixels; ++i) { + *p = std::min(4095, std::max(0, *p + (rng.get() % noise_amount))); + ++p; + } + } + + ++noise_amount; + /* Something's gone badly wrong if this much noise doesn't help */ + DCPOMATIC_ASSERT (noise_amount < 16); + } + switch (_frame->eyes()) { - case EYES_BOTH: + case Eyes::BOTH: LOG_DEBUG_ENCODE (N_("Finished locally-encoded frame %1 for mono"), _index); break; - case EYES_LEFT: + case Eyes::LEFT: LOG_DEBUG_ENCODE (N_("Finished locally-encoded frame %1 for L"), _index); break; - case EYES_RIGHT: + case Eyes::RIGHT: LOG_DEBUG_ENCODE (N_("Finished locally-encoded frame %1 for R"), _index); break; default: @@ -150,45 +190,54 @@ DCPVideo::encode_locally (dcp::NoteHandler note) /** Send this frame to a remote server for J2K encoding, then read the result. * @param serv Server to send to. + * @param timeout timeout in seconds. * @return Encoded data. */ -Data -DCPVideo::encode_remotely (ServerDescription serv, int timeout) +ArrayData +DCPVideo::encode_remotely (EncodeServerDescription serv, int timeout) { boost::asio::io_service io_service; boost::asio::ip::tcp::resolver resolver (io_service); - boost::asio::ip::tcp::resolver::query query (serv.host_name(), raw_convert (Config::instance()->server_port_base ())); + boost::asio::ip::tcp::resolver::query query (serv.host_name(), raw_convert (ENCODE_FRAME_PORT)); boost::asio::ip::tcp::resolver::iterator endpoint_iterator = resolver.resolve (query); - shared_ptr socket (new Socket (timeout)); + auto socket = make_shared(timeout); socket->connect (*endpoint_iterator); /* Collect all XML metadata */ xmlpp::Document doc; - xmlpp::Element* root = doc.create_root_node ("EncodingRequest"); + auto root = doc.create_root_node ("EncodingRequest"); root->add_child("Version")->add_child_text (raw_convert (SERVER_LINK_VERSION)); add_metadata (root); LOG_DEBUG_ENCODE (N_("Sending frame %1 to remote"), _index); - /* Send XML metadata */ - string xml = doc.write_to_string ("UTF-8"); - socket->write (xml.length() + 1); - socket->write ((uint8_t *) xml.c_str(), xml.length() + 1); + { + Socket::WriteDigestScope ds (socket); + + /* Send XML metadata */ + auto xml = doc.write_to_string ("UTF-8"); + socket->write (xml.length() + 1); + socket->write ((uint8_t *) xml.c_str(), xml.length() + 1); - /* Send binary data */ - LOG_TIMING("start-remote-send thread=%1", boost::this_thread::get_id()); - _frame->send_binary (socket); + /* Send binary data */ + LOG_TIMING("start-remote-send thread=%1", thread_id ()); + _frame->write_to_socket (socket); + } /* Read the response (JPEG2000-encoded data); this blocks until the data is ready and sent back. */ - LOG_TIMING("start-remote-encode thread=%1", boost::this_thread::get_id ()); - Data e (socket->read_uint32 ()); - LOG_TIMING("start-remote-receive thread=%1", boost::this_thread::get_id ()); - socket->read (e.data().get(), e.size()); - LOG_TIMING("finish-remote-receive thread=%1", boost::this_thread::get_id ()); + Socket::ReadDigestScope ds (socket); + LOG_TIMING("start-remote-encode thread=%1", thread_id ()); + ArrayData e (socket->read_uint32 ()); + LOG_TIMING("start-remote-receive thread=%1", thread_id ()); + socket->read (e.data(), e.size()); + LOG_TIMING("finish-remote-receive thread=%1", thread_id ()); + if (!ds.check()) { + throw NetworkError ("Checksums do not match"); + } LOG_DEBUG_ENCODE (N_("Finished remotely-encoded frame %1"), _index);