2 Copyright (C) 2013-2014 Carl Hetherington <cth@carlh.net>
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 #include "argb_frame.h"
22 #include "xyz_frame.h"
23 #include "gamma_lut.h"
25 #include "colour_matrix.h"
30 using boost::shared_ptr;
33 #define DCI_COEFFICIENT (48.0 / 52.37)
35 /** Convert an openjpeg XYZ image to RGBA.
36 * @param xyz_frame Frame in XYZ.
37 * @param lut_in Input Gamma LUT to use.
38 * @param lut_out Output Gamma LUT to use.
43 boost::shared_ptr<const XYZFrame> xyz_frame,
44 boost::shared_ptr<const GammaLUT> lut_in,
45 boost::shared_ptr<const GammaLUT> lut_out
48 int const max_colour = pow (2, lut_out->bit_depth()) - 1;
58 int* xyz_x = xyz_frame->data (0);
59 int* xyz_y = xyz_frame->data (1);
60 int* xyz_z = xyz_frame->data (2);
62 shared_ptr<ARGBFrame> argb_frame (new ARGBFrame (xyz_frame->size ()));
64 uint8_t* argb = argb_frame->data ();
66 for (int y = 0; y < xyz_frame->size().height; ++y) {
67 uint8_t* argb_line = argb;
68 for (int x = 0; x < xyz_frame->size().width; ++x) {
70 assert (*xyz_x >= 0 && *xyz_y >= 0 && *xyz_z >= 0 && *xyz_x < 4096 && *xyz_y < 4096 && *xyz_z < 4096);
73 s.x = lut_in->lut()[*xyz_x++];
74 s.y = lut_in->lut()[*xyz_y++];
75 s.z = lut_in->lut()[*xyz_z++];
78 s.x /= DCI_COEFFICIENT;
79 s.y /= DCI_COEFFICIENT;
80 s.z /= DCI_COEFFICIENT;
83 d.r = ((s.x * colour_matrix::xyz_to_rgb[0][0]) + (s.y * colour_matrix::xyz_to_rgb[0][1]) + (s.z * colour_matrix::xyz_to_rgb[0][2]));
84 d.g = ((s.x * colour_matrix::xyz_to_rgb[1][0]) + (s.y * colour_matrix::xyz_to_rgb[1][1]) + (s.z * colour_matrix::xyz_to_rgb[1][2]));
85 d.b = ((s.x * colour_matrix::xyz_to_rgb[2][0]) + (s.y * colour_matrix::xyz_to_rgb[2][1]) + (s.z * colour_matrix::xyz_to_rgb[2][2]));
97 *argb_line++ = lut_out->lut()[(int) (d.b * max_colour)] * 0xff;
98 *argb_line++ = lut_out->lut()[(int) (d.g * max_colour)] * 0xff;
99 *argb_line++ = lut_out->lut()[(int) (d.r * max_colour)] * 0xff;
103 argb += argb_frame->stride ();
109 /** Convert an openjpeg XYZ image to RGB.
110 * @param xyz_frame Frame in XYZ.
111 * @param lut_in Input Gamma LUT to use.
112 * @param lut_out Output Gamma LUT to use.
113 * @param buffer Buffer to write RGB data to; will be written
114 * as one byte R, one byte G, one byte B, one byte R etc. with
115 * no padding at line ends.
119 boost::shared_ptr<const XYZFrame> xyz_frame,
120 boost::shared_ptr<const GammaLUT> lut_in,
121 boost::shared_ptr<const GammaLUT> lut_out,
125 int const max_colour = pow (2, lut_out->bit_depth()) - 1;
135 int* xyz_x = xyz_frame->data (0);
136 int* xyz_y = xyz_frame->data (1);
137 int* xyz_z = xyz_frame->data (2);
139 for (int y = 0; y < xyz_frame->size().height; ++y) {
140 uint8_t* buffer_line = buffer;
141 for (int x = 0; x < xyz_frame->size().width; ++x) {
143 assert (*xyz_x >= 0 && *xyz_y >= 0 && *xyz_z >= 0 && *xyz_x < 4096 && *xyz_y < 4096 && *xyz_z < 4096);
146 s.x = lut_in->lut()[*xyz_x++];
147 s.y = lut_in->lut()[*xyz_y++];
148 s.z = lut_in->lut()[*xyz_z++];
151 s.x /= DCI_COEFFICIENT;
152 s.y /= DCI_COEFFICIENT;
153 s.z /= DCI_COEFFICIENT;
156 d.r = ((s.x * colour_matrix::xyz_to_rgb[0][0]) + (s.y * colour_matrix::xyz_to_rgb[0][1]) + (s.z * colour_matrix::xyz_to_rgb[0][2]));
157 d.g = ((s.x * colour_matrix::xyz_to_rgb[1][0]) + (s.y * colour_matrix::xyz_to_rgb[1][1]) + (s.z * colour_matrix::xyz_to_rgb[1][2]));
158 d.b = ((s.x * colour_matrix::xyz_to_rgb[2][0]) + (s.y * colour_matrix::xyz_to_rgb[2][1]) + (s.z * colour_matrix::xyz_to_rgb[2][2]));
160 d.r = min (d.r, 1.0);
161 d.r = max (d.r, 0.0);
163 d.g = min (d.g, 1.0);
164 d.g = max (d.g, 0.0);
166 d.b = min (d.b, 1.0);
167 d.b = max (d.b, 0.0);
170 *buffer_line++ = lut_out->lut()[(int) (d.r * max_colour)] * 0xff;
171 *buffer_line++ = lut_out->lut()[(int) (d.g * max_colour)] * 0xff;
172 *buffer_line++ = lut_out->lut()[(int) (d.b * max_colour)] * 0xff;
175 buffer += xyz_frame->size().width * 3;
179 shared_ptr<dcp::XYZFrame>
181 boost::shared_ptr<const Image> rgb,
182 boost::shared_ptr<const GammaLUT> lut_in,
183 boost::shared_ptr<const GammaLUT> lut_out,
184 double const colour_matrix[3][3]
187 assert (lut_in->bit_depth() == 12);
188 assert (lut_out->bit_depth() == 16);
190 shared_ptr<XYZFrame> xyz (new XYZFrame (rgb->size ()));
201 for (int y = 0; y < rgb->size().height; ++y) {
202 uint8_t* p = rgb->data()[0] + y * rgb->stride()[0];
203 for (int x = 0; x < rgb->size().width; ++x) {
205 /* In gamma LUT (converting 8-bit input to 12-bit) */
206 s.r = lut_in->lut()[*p++ << 4];
207 s.g = lut_in->lut()[*p++ << 4];
208 s.b = lut_in->lut()[*p++ << 4];
210 /* RGB to XYZ Matrix */
211 d.x = ((s.r * colour_matrix[0][0]) +
212 (s.g * colour_matrix[0][1]) +
213 (s.b * colour_matrix[0][2]));
215 d.y = ((s.r * colour_matrix[1][0]) +
216 (s.g * colour_matrix[1][1]) +
217 (s.b * colour_matrix[1][2]));
219 d.z = ((s.r * colour_matrix[2][0]) +
220 (s.g * colour_matrix[2][1]) +
221 (s.b * colour_matrix[2][2]));
224 d.x = d.x * DCI_COEFFICIENT * 65535;
225 d.y = d.y * DCI_COEFFICIENT * 65535;
226 d.z = d.z * DCI_COEFFICIENT * 65535;
228 assert (d.x >= 0 && d.x < 65536);
229 assert (d.y >= 0 && d.y < 65536);
230 assert (d.z >= 0 && d.z < 65536);
233 xyz->data(0)[jn] = lut_out->lut()[(int) d.x] * 4096;
234 xyz->data(1)[jn] = lut_out->lut()[(int) d.y] * 4096;
235 xyz->data(2)[jn] = lut_out->lut()[(int) d.z] * 4096;