Loading [MathJax]/extensions/tex2jax.js
LSST Applications g04a91732dc+a777afbe81,g07dc498a13+7e3c5f68a2,g12483e3c20+0145ec33cd,g1409bbee79+7e3c5f68a2,g1a7e361dbc+7e3c5f68a2,g1fd858c14a+9f35e23ec3,g35bb328faa+fcb1d3bbc8,g3ad4f90e5c+0145ec33cd,g3bd4b5ce2c+cbf1bea503,g4e0f332c67+5d362be553,g53246c7159+fcb1d3bbc8,g5477a8d5ce+db04660fe6,g60b5630c4e+0145ec33cd,g623d845a50+0145ec33cd,g6f0c2978f1+3526b51a37,g75b6c65c88+d54b601591,g78460c75b0+2f9a1b4bcd,g786e29fd12+cf7ec2a62a,g7b71ed6315+fcb1d3bbc8,g8852436030+4639f750a5,g89139ef638+7e3c5f68a2,g9125e01d80+fcb1d3bbc8,g919ac25b3e+6220c5324a,g95236ca021+f7a31438ed,g989de1cb63+7e3c5f68a2,g9f33ca652e+2d6fa11d35,gaaedd4e678+7e3c5f68a2,gabe3b4be73+1e0a283bba,gb1101e3267+4a428ef779,gb4a253aaf5+0122250889,gb58c049af0+f03b321e39,gc99c83e5f0+76d20ab76d,gcf25f946ba+4639f750a5,gd6cbbdb0b4+c8606af20c,gde0f65d7ad+3d8a3b7e46,ge278dab8ac+932305ba37,gf795337580+03b96afe58,gfba249425e+fcb1d3bbc8,w.2025.08
LSST Data Management Base Package
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
_gamutFixer.cc
Go to the documentation of this file.
1/*
2 * Developed for the LSST Data Management System.
3 * This product includes software developed by the LSST Project
4 * (https://www.lsst.org).
5 * See the COPYRIGHT file at the top-level directory of this distribution
6 * for details of code ownership.
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program. If not, see <https://www.gnu.org/licenses/>.
20 */
21#include "pybind11/pybind11.h"
22#include "pybind11/numpy.h"
25
26namespace py = pybind11;
27
28namespace lsst {
29namespace cpputils {
30
31
32py::array_t<double> fixGamutOK(py::array_t<double, py::array::c_style | py::array::forcecast> & Lab_points) {
33 py::buffer_info Lab_buffer = Lab_points.request();
34 auto Lab_ptr = Lab_points.unchecked<2>();
35 py::array_t<double> result(Lab_buffer.shape);
36 py::buffer_info result_buffer = result.request();
37 auto result_ptr = result.mutable_unchecked<2>();
38 float alpha = 0.5f;
39
40 for (int pixel_number=0; pixel_number < Lab_buffer.shape[0]; pixel_number++){
41 double L = Lab_ptr(pixel_number, 0);
42 double a = Lab_ptr(pixel_number, 1);
43 double b = Lab_ptr(pixel_number, 2);
44 double esp = 0.00001;
45 float C = std::max(esp, sqrt(a*a + b*b));
46 float a_ = a/C;
47 float b_ = b/C;
48
49 details::LC cusp = details::find_cusp(a_, b_);
50 float Ld = L - cusp.L;
51 float k = 2.f * (Ld > 0 ? 1.f - cusp.L : cusp.L);
52
53 float e1 = 0.5f*k + fabs(Ld) + alpha * C/k;
54 float L0 = cusp.L + 0.5f * (details::sgn(Ld) * (e1 - sqrtf(e1 * e1 - 2.f * k * fabs(Ld))));
55
56 float t = details::find_gamut_intersection(a_, b_, L, C, L0);
57 float L_clipped = L0 * (1.f - t) + t * L;
58 float C_clipped = t * C;
59
60 result_ptr(pixel_number, 0) = L_clipped;
61 result_ptr(pixel_number, 1) = C_clipped * a_;
62 result_ptr(pixel_number, 2) = C_clipped * b_;
63 }
64 return result;
65
66}
67
69 wrappers.wrap([](auto &mod) {
70 mod.def("fixGamutOK", &fixGamutOK,"");
71 });
72}
73
74}
75}
76
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
void wrap(WrapperCallback function)
Add a set of wrappers without defining a class.
Definition python.h:369
T max(T... args)
LC find_cusp(float a, float b)
float find_gamut_intersection(float a, float b, float L1, float C1, float L0)
py::array_t< double > fixGamutOK(py::array_t< double, py::array::c_style|py::array::forcecast > &Lab_points)
void wrapFixGamut(python::WrapperCollection &wrappers)