X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fj2k_image_proxy.cc;h=31fda2510492d9285f8987c73a41acd9551a5435;hb=386e25f3b9d3fa59cbdeed458d9b3e0d21e338b8;hp=9100414e442760ce1c532026afbd5b945e367e5c;hpb=0ff4fa6058b305476e7bc60c590acb6135c49b1c;p=dcpomatic.git diff --git a/src/lib/j2k_image_proxy.cc b/src/lib/j2k_image_proxy.cc index 9100414e4..31fda2510 100644 --- a/src/lib/j2k_image_proxy.cc +++ b/src/lib/j2k_image_proxy.cc @@ -31,7 +31,6 @@ #include #include #include -#include #include #include "i18n.h" @@ -53,7 +52,8 @@ J2KImageProxy::J2KImageProxy (boost::filesystem::path path, dcp::Size size, AVPi , _size (size) , _pixel_format (pixel_format) { - + /* ::image assumes 16bpp */ + DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE); } J2KImageProxy::J2KImageProxy ( @@ -67,6 +67,8 @@ J2KImageProxy::J2KImageProxy ( , _pixel_format (pixel_format) , _forced_reduction (forced_reduction) { + /* ::image assumes 16bpp */ + DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE); memcpy (_data.data().get(), frame->j2k_data(), _data.size ()); } @@ -82,6 +84,8 @@ J2KImageProxy::J2KImageProxy ( , _pixel_format (pixel_format) , _forced_reduction (forced_reduction) { + /* ::image assumes 16bpp */ + DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE); switch (eye) { case dcp::EYE_LEFT: _data = Data (frame->left_j2k_size ()); @@ -114,7 +118,7 @@ J2KImageProxy::prepare (optional target_size) const { boost::mutex::scoped_lock lm (_mutex); - if (_decompressed && target_size == _target_size) { + if (_image && target_size == _target_size) { DCPOMATIC_ASSERT (_reduce); return *_reduce; } @@ -132,51 +136,45 @@ J2KImageProxy::prepare (optional target_size) const reduce = max (0, reduce); } - _decompressed = dcp::decompress_j2k (const_cast (_data.data().get()), _data.size (), reduce); - - if (_decompressed->precision(0) < 12) { - int const shift = 12 - _decompressed->precision (0); - for (int c = 0; c < 3; ++c) { - int* p = _decompressed->data (c); - for (int y = 0; y < _decompressed->size().height; ++y) { - for (int x = 0; x < _decompressed->size().width; ++x) { - *p++ <<= shift; - } - } - } - } - - _target_size = target_size; - _reduce = reduce; - - return reduce; -} - -pair, int> -J2KImageProxy::image (optional, optional target_size) const -{ - int const reduce = prepare (target_size); + shared_ptr decompressed = dcp::decompress_j2k (const_cast (_data.data().get()), _data.size (), reduce); + _image.reset (new Image (_pixel_format, decompressed->size(), true)); - shared_ptr image (new Image (_pixel_format, _decompressed->size(), true)); + int const shift = 16 - decompressed->precision (0); /* Copy data in whatever format (sRGB or XYZ) into our Image; I'm assuming the data is 12-bit either way. */ - int const width = _decompressed->size().width; + int const width = decompressed->size().width; int p = 0; - for (int y = 0; y < _decompressed->size().height; ++y) { - uint16_t* q = (uint16_t *) (image->data()[0] + y * image->stride()[0]); + int* decomp_0 = decompressed->data (0); + int* decomp_1 = decompressed->data (1); + int* decomp_2 = decompressed->data (2); + for (int y = 0; y < decompressed->size().height; ++y) { + uint16_t* q = (uint16_t *) (_image->data()[0] + y * _image->stride()[0]); for (int x = 0; x < width; ++x) { - for (int c = 0; c < 3; ++c) { - *q++ = _decompressed->data(c)[p] << 4; - } + *q++ = decomp_0[p] << shift; + *q++ = decomp_1[p] << shift; + *q++ = decomp_2[p] << shift; ++p; } } - return make_pair (image, reduce); + _target_size = target_size; + _reduce = reduce; + + return reduce; +} + +pair, int> +J2KImageProxy::image (optional target_size) const +{ + int const r = prepare (target_size); + /* I think this is safe without a lock on mutex. _image is guaranteed to be + set up when prepare() has happened. + */ + return make_pair (_image, r); } void @@ -217,16 +215,17 @@ J2KImageProxy::J2KImageProxy (Data data, dcp::Size size, AVPixelFormat pixel_for , _size (size) , _pixel_format (pixel_format) { - + /* ::image assumes 16bpp */ + DCPOMATIC_ASSERT (_pixel_format == AV_PIX_FMT_RGB48 || _pixel_format == AV_PIX_FMT_XYZ12LE); } size_t J2KImageProxy::memory_used () const { size_t m = _data.size(); - if (_decompressed) { + if (_image) { /* 3 components, 16-bits per pixel */ - m += 3 * 2 * _decompressed->size().width * _decompressed->size().height; + m += 3 * 2 * _image->size().width * _image->size().height; } return m; }