8bf1a0caf5a88660a46c6659723c9b8faea01d16
[dcpomatic.git] / src / lib / j2k_encoder_remote_backend.cc
1 /*
2     Copyright (C) 2021 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21
22 #include "config.h"
23 #include "cross.h"
24 #include "dcp_video.h"
25 #include "dcpomatic_log.h"
26 #include "dcpomatic_socket.h"
27 #include "j2k_encoder_remote_backend.h"
28 #include "player_video.h"
29 #include "warnings.h"
30 #include <dcp/raw_convert.h>
31 DCPOMATIC_DISABLE_WARNINGS
32 #include <libxml++/libxml++.h>
33 DCPOMATIC_ENABLE_WARNINGS
34 #include <boost/asio.hpp>
35
36
37 #include "i18n.h"
38
39
40 using std::make_shared;
41 using std::shared_ptr;
42 using std::string;
43 using std::unique_ptr;
44 using boost::optional;
45 using dcp::raw_convert;
46
47
48 J2KEncoderRemoteBackend::J2KEncoderRemoteBackend (J2KEncoderRemoteBackend&& other)
49         : _server (other._server)
50         , _backoff (other._backoff)
51 {
52
53 }
54
55
56 optional<dcp::ArrayData>
57 J2KEncoderRemoteBackend::encode (DCPVideo video)
58 {
59         try {
60                 boost::asio::io_service io_service;
61                 boost::asio::ip::tcp::resolver resolver (io_service);
62                 boost::asio::ip::tcp::resolver::query query (_server.host_name(), raw_convert<string>(ENCODE_FRAME_PORT));
63                 auto endpoint_iterator = resolver.resolve (query);
64
65                 auto socket = make_shared<Socket>(_timeout);
66
67                 socket->connect (*endpoint_iterator);
68
69                 /* Collect all XML metadata */
70                 xmlpp::Document doc;
71                 auto root = doc.create_root_node ("EncodingRequest");
72                 root->add_child("Version")->add_child_text(raw_convert<string>(SERVER_LINK_VERSION));
73                 video.add_metadata (root);
74
75                 LOG_DEBUG_ENCODE (N_("Sending frame %1 to remote"), video.index());
76
77                 {
78                         Socket::WriteDigestScope ds (socket);
79
80                         /* Send XML metadata */
81                         auto xml = doc.write_to_string ("UTF-8");
82                         socket->write (xml.length() + 1);
83                         socket->write (reinterpret_cast<uint8_t const*>(xml.c_str()), xml.bytes() + 1);
84
85                         /* Send binary data */
86                         LOG_TIMING("start-remote-send thread=%1", thread_id());
87                         video.frame()->write_to_socket(socket);
88                 }
89
90                 /* Read the response (JPEG2000-encoded data); this blocks until the data
91                    is ready and sent back.
92                 */
93                 Socket::ReadDigestScope ds (socket);
94                 LOG_TIMING("start-remote-encode thread=%1", thread_id());
95                 dcp::ArrayData enc(socket->read_uint32());
96                 LOG_TIMING("start-remote-receive thread=%1", thread_id());
97                 socket->read (enc.data(), enc.size());
98                 LOG_TIMING("finish-remote-receive thread=%1", thread_id());
99                 if (!ds.check()) {
100                         throw NetworkError ("Checksums do not match");
101                 }
102
103                 LOG_DEBUG_ENCODE (N_("Finished remotely-encoded frame %1"), video.index());
104
105                 _backoff = 0;
106                 return enc;
107
108         } catch (std::exception& e) {
109                 if (_backoff < 60) {
110                         /* back off more */
111                         _backoff += 10;
112                 }
113                 LOG_ERROR (
114                         N_("Remote encode of %1 on %2 failed (%3); thread sleeping for %4s"),
115                         video.index(), _server.host_name(), e.what(), _backoff
116                         );
117                 return {};
118         }
119
120 }
121