diff options
| author | Carl Hetherington <cth@carlh.net> | 2013-07-12 21:00:30 +0100 |
|---|---|---|
| committer | Carl Hetherington <cth@carlh.net> | 2013-07-12 21:00:30 +0100 |
| commit | 414c44166031003b29d1b0b2fd2c9748a63260b4 (patch) | |
| tree | f9d955142f6c982ce1f6db0b71d0a4c10bdbfea5 /src | |
| parent | f28303e1f00075841ac00fe4bb18ca37cbf1a7af (diff) | |
Move XYZ/RGB function to its own source file.
Diffstat (limited to 'src')
| -rw-r--r-- | src/picture_frame.cc | 1 | ||||
| -rw-r--r-- | src/rgb_xyz.cc | 109 | ||||
| -rw-r--r-- | src/rgb_xyz.h | 32 | ||||
| -rw-r--r-- | src/util.cc | 80 | ||||
| -rw-r--r-- | src/util.h | 1 | ||||
| -rw-r--r-- | src/wscript | 2 |
6 files changed, 144 insertions, 81 deletions
diff --git a/src/picture_frame.cc b/src/picture_frame.cc index d0784245..07a2b074 100644 --- a/src/picture_frame.cc +++ b/src/picture_frame.cc @@ -26,6 +26,7 @@ #include "lut.h" #include "util.h" #include "gamma_lut.h" +#include "rgb_xyz.h" #define DCI_GAMMA 2.6 diff --git a/src/rgb_xyz.cc b/src/rgb_xyz.cc new file mode 100644 index 00000000..6be7180f --- /dev/null +++ b/src/rgb_xyz.cc @@ -0,0 +1,109 @@ +/* + Copyright (C) 2013 Carl Hetherington <cth@carlh.net> + + This program 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, + 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. + +*/ + +#include "rgb_xyz.h" +#include "argb_frame.h" +#include "xyz_frame.h" +#include "gamma_lut.h" + +using std::min; +using std::max; +using boost::shared_ptr; +using namespace libdcp; + +/** Convert an openjpeg XYZ image to RGB. + * @param xyz_frame Frame in XYZ. + * @return RGB image. + */ +shared_ptr<ARGBFrame> +libdcp::xyz_to_rgb (shared_ptr<const XYZFrame> xyz_frame, shared_ptr<const GammaLUT> lut_in, shared_ptr<const GammaLUT> lut_out) +{ + float const dci_coefficient = 48.0 / 52.37; + + /* sRGB color matrix for XYZ -> RGB. This is the same as the one used by the Fraunhofer + EasyDCP player, I think. + */ + + float const colour_matrix[3][3] = { + { 3.24096989631653, -1.5373831987381, -0.498610764741898 }, + { -0.96924364566803, 1.87596750259399, 0.0415550582110882 }, + { 0.0556300804018974, -0.203976958990097, 1.05697154998779 } + }; + + int const max_colour = pow (2, lut_out->bit_depth()) - 1; + + struct { + double x, y, z; + } s; + + struct { + double r, g, b; + } d; + + int* xyz_x = xyz_frame->data (0); + int* xyz_y = xyz_frame->data (1); + int* xyz_z = xyz_frame->data (2); + + shared_ptr<ARGBFrame> argb_frame (new ARGBFrame (xyz_frame->size ())); + + uint8_t* argb = argb_frame->data (); + + for (int y = 0; y < xyz_frame->size().height; ++y) { + uint8_t* argb_line = argb; + for (int x = 0; x < xyz_frame->size().width; ++x) { + + assert (*xyz_x >= 0 && *xyz_y >= 0 && *xyz_z >= 0 && *xyz_x < 4096 && *xyz_x < 4096 && *xyz_z < 4096); + + /* In gamma LUT */ + s.x = lut_in->lut()[*xyz_x++]; + s.y = lut_in->lut()[*xyz_y++]; + s.z = lut_in->lut()[*xyz_z++]; + + /* DCI companding */ + s.x /= dci_coefficient; + s.y /= dci_coefficient; + s.z /= dci_coefficient; + + /* XYZ to RGB */ + d.r = ((s.x * colour_matrix[0][0]) + (s.y * colour_matrix[0][1]) + (s.z * colour_matrix[0][2])); + d.g = ((s.x * colour_matrix[1][0]) + (s.y * colour_matrix[1][1]) + (s.z * colour_matrix[1][2])); + d.b = ((s.x * colour_matrix[2][0]) + (s.y * colour_matrix[2][1]) + (s.z * colour_matrix[2][2])); + + d.r = min (d.r, 1.0); + d.r = max (d.r, 0.0); + + d.g = min (d.g, 1.0); + d.g = max (d.g, 0.0); + + d.b = min (d.b, 1.0); + d.b = max (d.b, 0.0); + + /* Out gamma LUT */ + *argb_line++ = lut_out->lut()[(int) (d.b * max_colour)] * 0xff; + *argb_line++ = lut_out->lut()[(int) (d.g * max_colour)] * 0xff; + *argb_line++ = lut_out->lut()[(int) (d.r * max_colour)] * 0xff; + *argb_line++ = 0xff; + } + + argb += argb_frame->stride (); + } + + return argb_frame; +} + diff --git a/src/rgb_xyz.h b/src/rgb_xyz.h new file mode 100644 index 00000000..123e055a --- /dev/null +++ b/src/rgb_xyz.h @@ -0,0 +1,32 @@ +/* + Copyright (C) 2013 Carl Hetherington <cth@carlh.net> + + This program 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, + 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. + +*/ + +#include <boost/shared_ptr.hpp> + +namespace libdcp { + +class ARGBFrame; +class XYZFrame; +class GammaLUT; + +extern boost::shared_ptr<ARGBFrame> xyz_to_rgb ( + boost::shared_ptr<const XYZFrame>, boost::shared_ptr<const GammaLUT>, boost::shared_ptr<const GammaLUT> + ); + +} diff --git a/src/util.cc b/src/util.cc index 24dc37fb..848667fc 100644 --- a/src/util.cc +++ b/src/util.cc @@ -207,86 +207,6 @@ libdcp::decompress_j2k (uint8_t* data, int64_t size, int reduce) return shared_ptr<XYZFrame> (new XYZFrame (image)); } -/** Convert an openjpeg XYZ image to RGB. - * @param xyz_frame Frame in XYZ. - * @return RGB image. - */ -shared_ptr<ARGBFrame> -libdcp::xyz_to_rgb (shared_ptr<const XYZFrame> xyz_frame, shared_ptr<const GammaLUT> lut_in, shared_ptr<const GammaLUT> lut_out) -{ - float const dci_coefficient = 48.0 / 52.37; - - /* sRGB color matrix for XYZ -> RGB. This is the same as the one used by the Fraunhofer - EasyDCP player, I think. - */ - - float const colour_matrix[3][3] = { - { 3.24096989631653, -1.5373831987381, -0.498610764741898 }, - { -0.96924364566803, 1.87596750259399, 0.0415550582110882 }, - { 0.0556300804018974, -0.203976958990097, 1.05697154998779 } - }; - - int const max_colour = pow (2, lut_out->bit_depth()) - 1; - - struct { - double x, y, z; - } s; - - struct { - double r, g, b; - } d; - - int* xyz_x = xyz_frame->data (0); - int* xyz_y = xyz_frame->data (1); - int* xyz_z = xyz_frame->data (2); - - shared_ptr<ARGBFrame> argb_frame (new ARGBFrame (xyz_frame->size ())); - - uint8_t* argb = argb_frame->data (); - - for (int y = 0; y < xyz_frame->size().height; ++y) { - uint8_t* argb_line = argb; - for (int x = 0; x < xyz_frame->size().width; ++x) { - - assert (*xyz_x >= 0 && *xyz_y >= 0 && *xyz_z >= 0 && *xyz_x < 4096 && *xyz_x < 4096 && *xyz_z < 4096); - - /* In gamma LUT */ - s.x = lut_in->lut()[*xyz_x++]; - s.y = lut_in->lut()[*xyz_y++]; - s.z = lut_in->lut()[*xyz_z++]; - - /* DCI companding */ - s.x /= dci_coefficient; - s.y /= dci_coefficient; - s.z /= dci_coefficient; - - /* XYZ to RGB */ - d.r = ((s.x * colour_matrix[0][0]) + (s.y * colour_matrix[0][1]) + (s.z * colour_matrix[0][2])); - d.g = ((s.x * colour_matrix[1][0]) + (s.y * colour_matrix[1][1]) + (s.z * colour_matrix[1][2])); - d.b = ((s.x * colour_matrix[2][0]) + (s.y * colour_matrix[2][1]) + (s.z * colour_matrix[2][2])); - - d.r = min (d.r, 1.0); - d.r = max (d.r, 0.0); - - d.g = min (d.g, 1.0); - d.g = max (d.g, 0.0); - - d.b = min (d.b, 1.0); - d.b = max (d.b, 0.0); - - /* Out gamma LUT */ - *argb_line++ = lut_out->lut()[(int) (d.b * max_colour)] * 0xff; - *argb_line++ = lut_out->lut()[(int) (d.g * max_colour)] * 0xff; - *argb_line++ = lut_out->lut()[(int) (d.r * max_colour)] * 0xff; - *argb_line++ = 0xff; - } - - argb += argb_frame->stride (); - } - - return argb_frame; -} - /** @param s A string. * @return true if the string contains only space, newline or tab characters, or is empty. */ @@ -65,7 +65,6 @@ extern std::string content_kind_to_string (ContentKind kind); extern ContentKind content_kind_from_string (std::string kind); extern bool empty_or_white_space (std::string s); extern boost::shared_ptr<XYZFrame> decompress_j2k (uint8_t* data, int64_t size, int reduce); -extern boost::shared_ptr<ARGBFrame> xyz_to_rgb (boost::shared_ptr<const XYZFrame>, boost::shared_ptr<const GammaLUT>, boost::shared_ptr<const GammaLUT>); extern void init (); diff --git a/src/wscript b/src/wscript index 922ee1e9..3b23443c 100644 --- a/src/wscript +++ b/src/wscript @@ -25,6 +25,7 @@ def build(bld): picture_frame.cc rec709_linearised_gamma_lut.cc reel.cc + rgb_xyz.cc sound_asset.cc sound_frame.cc srgb_linearised_gamma_lut.cc @@ -53,6 +54,7 @@ def build(bld): mxf_asset.h picture_asset.h picture_frame.h + rgb_xyz.h rec709_linearised_gamma_lut.h reel.h argb_frame.h |
