2 Copyright (C) 2014-2015 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
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.
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.
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/>.
21 #include "j2k_image_proxy.h"
22 #include "dcpomatic_socket.h"
24 #include "dcpomatic_assert.h"
26 #include <dcp/raw_convert.h>
27 #include <dcp/openjpeg_image.h>
28 #include <dcp/mono_picture_frame.h>
29 #include <dcp/stereo_picture_frame.h>
30 #include <dcp/colour_conversion.h>
31 #include <dcp/rgb_xyz.h>
32 #include <dcp/j2k_transcode.h>
33 #include <libcxml/cxml.h>
34 DCPOMATIC_DISABLE_WARNINGS
35 #include <libxml++/libxml++.h>
36 DCPOMATIC_ENABLE_WARNINGS
46 using std::shared_ptr;
47 using boost::optional;
48 using std::dynamic_pointer_cast;
50 using dcp::raw_convert;
52 /** Construct a J2KImageProxy from a JPEG2000 file */
53 J2KImageProxy::J2KImageProxy (boost::filesystem::path path, dcp::Size size, AVPixelFormat pixel_format)
54 : _data (new dcp::ArrayData(path))
56 , _pixel_format (pixel_format)
59 /* ::image assumes 16bpp */
60 DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE);
64 J2KImageProxy::J2KImageProxy (
65 shared_ptr<const dcp::MonoPictureFrame> frame,
67 AVPixelFormat pixel_format,
68 optional<int> forced_reduction
72 , _pixel_format (pixel_format)
73 , _forced_reduction (forced_reduction)
76 /* ::image assumes 16bpp */
77 DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE);
81 J2KImageProxy::J2KImageProxy (
82 shared_ptr<const dcp::StereoPictureFrame> frame,
85 AVPixelFormat pixel_format,
86 optional<int> forced_reduction
88 : _data (eye == dcp::Eye::LEFT ? frame->left() : frame->right())
91 , _pixel_format (pixel_format)
92 , _forced_reduction (forced_reduction)
95 /* ::image assumes 16bpp */
96 DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE);
100 J2KImageProxy::J2KImageProxy (shared_ptr<cxml::Node> xml, shared_ptr<Socket> socket)
103 _size = dcp::Size (xml->number_child<int> ("Width"), xml->number_child<int> ("Height"));
104 if (xml->optional_number_child<int> ("Eye")) {
105 _eye = static_cast<dcp::Eye> (xml->number_child<int> ("Eye"));
107 shared_ptr<ArrayData> data(new ArrayData(xml->number_child<int>("Size")));
108 /* This only matters when we are using J2KImageProxy for the preview, which
109 will never use this constructor (which is only used for passing data to
110 encode servers). So we can put anything in here. It's a bit of a hack.
112 _pixel_format = AV_PIX_FMT_XYZ12LE;
113 socket->read (data->data(), data->size());
118 J2KImageProxy::prepare (optional<dcp::Size> target_size) const
120 boost::mutex::scoped_lock lm (_mutex);
122 if (_image && target_size == _target_size) {
123 DCPOMATIC_ASSERT (_reduce);
129 if (_forced_reduction) {
130 reduce = *_forced_reduction;
132 while (target_size && (_size.width / pow(2, reduce)) > target_size->width && (_size.height / pow(2, reduce)) > target_size->height) {
137 reduce = max (0, reduce);
141 /* XXX: should check that potentially trashing _data here doesn't matter */
142 shared_ptr<dcp::OpenJPEGImage> decompressed = dcp::decompress_j2k (const_cast<uint8_t*>(_data->data()), _data->size(), reduce);
143 _image.reset (new Image (_pixel_format, decompressed->size(), true));
145 int const shift = 16 - decompressed->precision (0);
147 /* Copy data in whatever format (sRGB or XYZ) into our Image; I'm assuming
148 the data is 12-bit either way.
151 int const width = decompressed->size().width;
154 int* decomp_0 = decompressed->data (0);
155 int* decomp_1 = decompressed->data (1);
156 int* decomp_2 = decompressed->data (2);
157 for (int y = 0; y < decompressed->size().height; ++y) {
158 uint16_t* q = (uint16_t *) (_image->data()[0] + y * _image->stride()[0]);
159 for (int x = 0; x < width; ++x) {
160 *q++ = decomp_0[p] << shift;
161 *q++ = decomp_1[p] << shift;
162 *q++ = decomp_2[p] << shift;
166 } catch (dcp::J2KDecompressionError& e) {
167 _image.reset (new Image (_pixel_format, _size, true));
168 _image->make_black ();
172 _target_size = target_size;
180 J2KImageProxy::image (optional<dcp::Size> target_size) const
182 int const r = prepare (target_size);
184 /* I think this is safe without a lock on mutex. _image is guaranteed to be
185 set up when prepare() has happened.
187 return Result (_image, r, _error);
192 J2KImageProxy::add_metadata (xmlpp::Node* node) const
194 node->add_child("Type")->add_child_text (N_("J2K"));
195 node->add_child("Width")->add_child_text (raw_convert<string> (_size.width));
196 node->add_child("Height")->add_child_text (raw_convert<string> (_size.height));
198 node->add_child("Eye")->add_child_text (raw_convert<string> (static_cast<int> (_eye.get ())));
200 node->add_child("Size")->add_child_text (raw_convert<string>(_data->size()));
204 J2KImageProxy::write_to_socket (shared_ptr<Socket> socket) const
206 socket->write (_data->data(), _data->size());
210 J2KImageProxy::same (shared_ptr<const ImageProxy> other) const
212 shared_ptr<const J2KImageProxy> jp = dynamic_pointer_cast<const J2KImageProxy> (other);
217 return *_data == *jp->_data;
220 J2KImageProxy::J2KImageProxy (ArrayData data, dcp::Size size, AVPixelFormat pixel_format)
221 : _data (new ArrayData(data))
223 , _pixel_format (pixel_format)
226 /* ::image assumes 16bpp */
227 DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE);
231 J2KImageProxy::memory_used () const
233 size_t m = _data->size();
235 /* 3 components, 16-bits per pixel */
236 m += 3 * 2 * _image->size().width * _image->size().height;