Loading [MathJax]/extensions/tex2jax.js
LSST Applications g04a91732dc+9666464c73,g0fba68d861+d6c6f70ffa,g1fd858c14a+94f68680cf,g208c678f98+1ca806343c,g271391ec13+ac98094cfc,g2c84ff76c0+120a924478,g2c9e612ef2+a92a2e6025,g35bb328faa+fcb1d3bbc8,g4d2262a081+7c332456db,g4e0f332c67+c58e4b632d,g53246c7159+fcb1d3bbc8,g60b5630c4e+a92a2e6025,g67b6fd64d1+9d1b2ab50a,g78460c75b0+2f9a1b4bcd,g786e29fd12+cf7ec2a62a,g7b71ed6315+fcb1d3bbc8,g8852436030+506db7da85,g89139ef638+9d1b2ab50a,g8d6b6b353c+a92a2e6025,g9125e01d80+fcb1d3bbc8,g989de1cb63+9d1b2ab50a,g9f33ca652e+d1749da127,ga2b97cdc51+a92a2e6025,gabe3b4be73+1e0a283bba,gb1101e3267+6ecbd0580e,gb58c049af0+f03b321e39,gb89ab40317+9d1b2ab50a,gb90eeb9370+384e1fc23b,gcf25f946ba+506db7da85,gd315a588df+382ef11c06,gd6cbbdb0b4+75aa4b1db4,gd9a9a58781+fcb1d3bbc8,gde0f65d7ad+9984fb7d9e,ge278dab8ac+c61fbefdff,ge410e46f29+9d1b2ab50a,ge82c20c137+e12a08b75a,gf67bdafdda+9d1b2ab50a,gfd5510ef7b+864ead200a,v29.0.0.rc2
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)