Blender V4.3
node_composite_colorcorrection.cc
Go to the documentation of this file.
1/* SPDX-FileCopyrightText: 2006 Blender Authors
2 *
3 * SPDX-License-Identifier: GPL-2.0-or-later */
4
9#include "IMB_colormanagement.hh"
10
11#include "UI_interface.hh"
12#include "UI_resources.hh"
13
14#include "GPU_material.hh"
15
16#include "COM_shader_node.hh"
17
18#include "node_composite_util.hh"
19
20/* ******************* Color Correction ********************************* */
21
22namespace blender::nodes::node_composite_colorcorrection_cc {
23
24NODE_STORAGE_FUNCS(NodeColorCorrection)
25
26static void cmp_node_colorcorrection_declare(NodeDeclarationBuilder &b)
27{
28 b.add_input<decl::Color>("Image")
29 .default_value({1.0f, 1.0f, 1.0f, 1.0f})
30 .compositor_domain_priority(0);
31 b.add_input<decl::Float>("Mask")
32 .default_value(1.0f)
33 .min(0.0f)
34 .max(1.0f)
35 .compositor_domain_priority(1);
36 b.add_output<decl::Color>("Image");
37}
38
39static void node_composit_init_colorcorrection(bNodeTree * /*ntree*/, bNode *node)
40{
41 NodeColorCorrection *n = MEM_cnew<NodeColorCorrection>(__func__);
42 n->startmidtones = 0.2f;
43 n->endmidtones = 0.7f;
44 n->master.contrast = 1.0f;
45 n->master.gain = 1.0f;
46 n->master.gamma = 1.0f;
47 n->master.lift = 0.0f;
48 n->master.saturation = 1.0f;
49 n->midtones.contrast = 1.0f;
50 n->midtones.gain = 1.0f;
51 n->midtones.gamma = 1.0f;
52 n->midtones.lift = 0.0f;
53 n->midtones.saturation = 1.0f;
54 n->shadows.contrast = 1.0f;
55 n->shadows.gain = 1.0f;
56 n->shadows.gamma = 1.0f;
57 n->shadows.lift = 0.0f;
58 n->shadows.saturation = 1.0f;
59 n->highlights.contrast = 1.0f;
60 n->highlights.gain = 1.0f;
61 n->highlights.gamma = 1.0f;
62 n->highlights.lift = 0.0f;
63 n->highlights.saturation = 1.0f;
64 node->custom1 = 7; // red + green + blue enabled
65 node->storage = n;
66}
67
68static void node_composit_buts_colorcorrection(uiLayout *layout, bContext * /*C*/, PointerRNA *ptr)
69{
70 uiLayout *row;
71
72 row = uiLayoutRow(layout, false);
73 uiItemR(row, ptr, "red", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
74 uiItemR(row, ptr, "green", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
75 uiItemR(row, ptr, "blue", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
76
77 row = uiLayoutRow(layout, false);
78 uiItemL(row, "", ICON_NONE);
79 uiItemL(row, IFACE_("Saturation"), ICON_NONE);
80 uiItemL(row, IFACE_("Contrast"), ICON_NONE);
81 uiItemL(row, IFACE_("Gamma"), ICON_NONE);
82 uiItemL(row, IFACE_("Gain"), ICON_NONE);
83 uiItemL(row, IFACE_("Lift"), ICON_NONE);
84
85 row = uiLayoutRow(layout, false);
86 uiItemL(row, IFACE_("Master"), ICON_NONE);
87 uiItemR(
88 row, ptr, "master_saturation", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
89 uiItemR(
90 row, ptr, "master_contrast", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
91 uiItemR(row, ptr, "master_gamma", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
92 uiItemR(row, ptr, "master_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
93 uiItemR(row, ptr, "master_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
94
95 row = uiLayoutRow(layout, false);
96 uiItemL(row, IFACE_("Highlights"), ICON_NONE);
97 uiItemR(row,
98 ptr,
99 "highlights_saturation",
100 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
101 "",
102 ICON_NONE);
103 uiItemR(row,
104 ptr,
105 "highlights_contrast",
106 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
107 "",
108 ICON_NONE);
109 uiItemR(
110 row, ptr, "highlights_gamma", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
111 uiItemR(
112 row, ptr, "highlights_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
113 uiItemR(
114 row, ptr, "highlights_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
115
116 row = uiLayoutRow(layout, false);
117 uiItemL(row, IFACE_("Midtones"), ICON_NONE);
118 uiItemR(row,
119 ptr,
120 "midtones_saturation",
121 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
122 "",
123 ICON_NONE);
124 uiItemR(
125 row, ptr, "midtones_contrast", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
126 uiItemR(
127 row, ptr, "midtones_gamma", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
128 uiItemR(row, ptr, "midtones_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
129 uiItemR(row, ptr, "midtones_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
130
131 row = uiLayoutRow(layout, false);
132 uiItemL(row, IFACE_("Shadows"), ICON_NONE);
133 uiItemR(row,
134 ptr,
135 "shadows_saturation",
136 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
137 "",
138 ICON_NONE);
139 uiItemR(
140 row, ptr, "shadows_contrast", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
141 uiItemR(row, ptr, "shadows_gamma", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
142 uiItemR(row, ptr, "shadows_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
143 uiItemR(row, ptr, "shadows_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, "", ICON_NONE);
144
145 row = uiLayoutRow(layout, false);
146 uiItemR(row,
147 ptr,
148 "midtones_start",
149 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
150 nullptr,
151 ICON_NONE);
152 uiItemR(
153 row, ptr, "midtones_end", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
154}
155
156static void node_composit_buts_colorcorrection_ex(uiLayout *layout,
157 bContext * /*C*/,
158 PointerRNA *ptr)
159{
160 uiLayout *row;
161
162 row = uiLayoutRow(layout, false);
163 uiItemR(row, ptr, "red", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
164 uiItemR(row, ptr, "green", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
165 uiItemR(row, ptr, "blue", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
166 row = layout;
167 uiItemL(row, IFACE_("Saturation"), ICON_NONE);
168 uiItemR(row,
169 ptr,
170 "master_saturation",
171 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
172 nullptr,
173 ICON_NONE);
174 uiItemR(row,
175 ptr,
176 "highlights_saturation",
177 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
178 nullptr,
179 ICON_NONE);
180 uiItemR(row,
181 ptr,
182 "midtones_saturation",
183 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
184 nullptr,
185 ICON_NONE);
186 uiItemR(row,
187 ptr,
188 "shadows_saturation",
189 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
190 nullptr,
191 ICON_NONE);
192
193 uiItemL(row, IFACE_("Contrast"), ICON_NONE);
194 uiItemR(row,
195 ptr,
196 "master_contrast",
197 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
198 nullptr,
199 ICON_NONE);
200 uiItemR(row,
201 ptr,
202 "highlights_contrast",
203 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
204 nullptr,
205 ICON_NONE);
206 uiItemR(row,
207 ptr,
208 "midtones_contrast",
209 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
210 nullptr,
211 ICON_NONE);
212 uiItemR(row,
213 ptr,
214 "shadows_contrast",
215 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
216 nullptr,
217 ICON_NONE);
218
219 uiItemL(row, IFACE_("Gamma"), ICON_NONE);
220 uiItemR(
221 row, ptr, "master_gamma", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
222 uiItemR(row,
223 ptr,
224 "highlights_gamma",
225 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
226 nullptr,
227 ICON_NONE);
228 uiItemR(row,
229 ptr,
230 "midtones_gamma",
231 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
232 nullptr,
233 ICON_NONE);
234 uiItemR(row,
235 ptr,
236 "shadows_gamma",
237 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
238 nullptr,
239 ICON_NONE);
240
241 uiItemL(row, IFACE_("Gain"), ICON_NONE);
242 uiItemR(
243 row, ptr, "master_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
244 uiItemR(row,
245 ptr,
246 "highlights_gain",
247 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
248 nullptr,
249 ICON_NONE);
250 uiItemR(row,
251 ptr,
252 "midtones_gain",
253 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
254 nullptr,
255 ICON_NONE);
256 uiItemR(
257 row, ptr, "shadows_gain", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
258
259 uiItemL(row, IFACE_("Lift"), ICON_NONE);
260 uiItemR(
261 row, ptr, "master_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
262 uiItemR(row,
263 ptr,
264 "highlights_lift",
265 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
266 nullptr,
267 ICON_NONE);
268 uiItemR(row,
269 ptr,
270 "midtones_lift",
271 UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER,
272 nullptr,
273 ICON_NONE);
274 uiItemR(
275 row, ptr, "shadows_lift", UI_ITEM_R_SPLIT_EMPTY_NAME | UI_ITEM_R_SLIDER, nullptr, ICON_NONE);
276
277 row = uiLayoutRow(layout, false);
278 uiItemR(row, ptr, "midtones_start", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
279 uiItemR(row, ptr, "midtones_end", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
280}
281
282using namespace blender::realtime_compositor;
283
284class ColorCorrectionShaderNode : public ShaderNode {
285 public:
286 using ShaderNode::ShaderNode;
287
288 void compile(GPUMaterial *material) override
289 {
290 GPUNodeStack *inputs = get_inputs_array();
291 GPUNodeStack *outputs = get_outputs_array();
292
293 float enabled_channels[3];
294 get_enabled_channels(enabled_channels);
295 float luminance_coefficients[3];
296 IMB_colormanagement_get_luminance_coefficients(luminance_coefficients);
297
298 const NodeColorCorrection &node_color_correction = node_storage(bnode());
299
300 GPU_stack_link(material,
301 &bnode(),
302 "node_composite_color_correction",
303 inputs,
304 outputs,
305 GPU_constant(enabled_channels),
306 GPU_uniform(&node_color_correction.startmidtones),
307 GPU_uniform(&node_color_correction.endmidtones),
308 GPU_uniform(&node_color_correction.master.saturation),
309 GPU_uniform(&node_color_correction.master.contrast),
310 GPU_uniform(&node_color_correction.master.gamma),
311 GPU_uniform(&node_color_correction.master.gain),
312 GPU_uniform(&node_color_correction.master.lift),
313 GPU_uniform(&node_color_correction.shadows.saturation),
314 GPU_uniform(&node_color_correction.shadows.contrast),
315 GPU_uniform(&node_color_correction.shadows.gamma),
316 GPU_uniform(&node_color_correction.shadows.gain),
317 GPU_uniform(&node_color_correction.shadows.lift),
318 GPU_uniform(&node_color_correction.midtones.saturation),
319 GPU_uniform(&node_color_correction.midtones.contrast),
320 GPU_uniform(&node_color_correction.midtones.gamma),
321 GPU_uniform(&node_color_correction.midtones.gain),
322 GPU_uniform(&node_color_correction.midtones.lift),
323 GPU_uniform(&node_color_correction.highlights.saturation),
324 GPU_uniform(&node_color_correction.highlights.contrast),
325 GPU_uniform(&node_color_correction.highlights.gamma),
326 GPU_uniform(&node_color_correction.highlights.gain),
327 GPU_uniform(&node_color_correction.highlights.lift),
328 GPU_constant(luminance_coefficients));
329 }
330
331 void get_enabled_channels(float enabled_channels[3])
332 {
333 for (int i = 0; i < 3; i++) {
334 enabled_channels[i] = (bnode().custom1 & (1 << i)) ? 1.0f : 0.0f;
335 }
336 }
337};
338
339static ShaderNode *get_compositor_shader_node(DNode node)
340{
341 return new ColorCorrectionShaderNode(node);
342}
343
344} // namespace blender::nodes::node_composite_colorcorrection_cc
345
346void register_node_type_cmp_colorcorrection()
347{
348 namespace file_ns = blender::nodes::node_composite_colorcorrection_cc;
349
350 static blender::bke::bNodeType ntype;
351
352 cmp_node_type_base(&ntype, CMP_NODE_COLORCORRECTION, "Color Correction", NODE_CLASS_OP_COLOR);
353 ntype.declare = file_ns::cmp_node_colorcorrection_declare;
354 ntype.draw_buttons = file_ns::node_composit_buts_colorcorrection;
355 ntype.draw_buttons_ex = file_ns::node_composit_buts_colorcorrection_ex;
356 blender::bke::node_type_size(&ntype, 400, 200, 600);
357 ntype.initfunc = file_ns::node_composit_init_colorcorrection;
358 blender::bke::node_type_storage(
359 &ntype, "NodeColorCorrection", node_free_standard_storage, node_copy_standard_storage);
360 ntype.get_compositor_shader_node = file_ns::get_compositor_shader_node;
361
362 blender::bke::node_register_type(&ntype);
363}
NODE_STORAGE_FUNCS
#define NODE_STORAGE_FUNCS(StorageT)
Definition BKE_node.hh:1799
NODE_CLASS_OP_COLOR
#define NODE_CLASS_OP_COLOR
Definition BKE_node.hh:406
IFACE_
#define IFACE_(msgid)
Definition BLT_translation.hh:41
COM_shader_node.hh
GPU_material.hh
GPU_stack_link
bool GPU_stack_link(GPUMaterial *mat, const bNode *node, const char *name, GPUNodeStack *in, GPUNodeStack *out,...)
Definition gpu_node_graph.cc:821
GPU_constant
GPUNodeLink * GPU_constant(const float *num)
Definition gpu_node_graph.cc:625
GPU_uniform
GPUNodeLink * GPU_uniform(const float *num)
Definition gpu_node_graph.cc:633
IMB_colormanagement.hh
IMB_colormanagement_get_luminance_coefficients
BLI_INLINE void IMB_colormanagement_get_luminance_coefficients(float r_rgb[3])
Definition colormanagement_inline.h:17
UI_interface.hh
uiItemL
void uiItemL(uiLayout *layout, const char *name, int icon)
Definition interface_layout.cc:3339
uiLayoutRow
uiLayout * uiLayoutRow(uiLayout *layout, bool align)
Definition interface_layout.cc:4968
uiItemR
void uiItemR(uiLayout *layout, PointerRNA *ptr, const char *propname, eUI_Item_Flag flag, const char *name, int icon)
Definition interface_layout.cc:2536
UI_ITEM_R_SPLIT_EMPTY_NAME
@ UI_ITEM_R_SPLIT_EMPTY_NAME
Definition UI_interface_c.hh:2183
UI_ITEM_R_SLIDER
@ UI_ITEM_R_SLIDER
Definition UI_interface_c.hh:2157
UI_resources.hh
blender::nodes::node_composite_colorcorrection_cc::ColorCorrectionShaderNode
Definition node_composite_colorcorrection.cc:284
blender::nodes::node_composite_colorcorrection_cc::ColorCorrectionShaderNode::compile
void compile(GPUMaterial *material) override
Definition node_composite_colorcorrection.cc:288
blender::nodes::node_composite_colorcorrection_cc::ColorCorrectionShaderNode::get_enabled_channels
void get_enabled_channels(float enabled_channels[3])
Definition node_composite_colorcorrection.cc:331
blender::realtime_compositor::ShaderNode
Definition COM_shader_node.hh:31
blender::realtime_compositor::ShaderNode::get_inputs_array
GPUNodeStack * get_inputs_array()
Definition shader_node.cc:28
blender::realtime_compositor::ShaderNode::get_outputs_array
GPUNodeStack * get_outputs_array()
Definition shader_node.cc:33
blender::realtime_compositor::ShaderNode::ShaderNode
ShaderNode(DNode node)
Definition shader_node.cc:22
blender::realtime_compositor::ShaderNode::bnode
const bNode & bnode() const
Definition shader_node.cc:62
b
local_group_size(16, 16) .push_constant(Type b
Definition compositor_morphological_distance_info.hh:16
blender::bke::node_type_size
void node_type_size(bNodeType *ntype, int width, int minwidth, int maxwidth)
Definition node.cc:4602
blender::bke::node_type_storage
void node_type_storage(bNodeType *ntype, const char *storagename, void(*freefunc)(bNode *node), void(*copyfunc)(bNodeTree *dest_ntree, bNode *dest_node, const bNode *src_node))
Definition node.cc:4632
blender::bke::node_register_type
void node_register_type(bNodeType *ntype)
Definition node.cc:1708
blender::nodes::node_composite_colorcorrection_cc
Definition node_composite_colorcorrection.cc:22
blender::nodes::node_composite_colorcorrection_cc::node_composit_buts_colorcorrection_ex
static void node_composit_buts_colorcorrection_ex(uiLayout *layout, bContext *, PointerRNA *ptr)
Definition node_composite_colorcorrection.cc:156
blender::nodes::node_composite_colorcorrection_cc::node_composit_buts_colorcorrection
static void node_composit_buts_colorcorrection(uiLayout *layout, bContext *, PointerRNA *ptr)
Definition node_composite_colorcorrection.cc:68
blender::nodes::node_composite_colorcorrection_cc::cmp_node_colorcorrection_declare
static void cmp_node_colorcorrection_declare(NodeDeclarationBuilder &b)
Definition node_composite_colorcorrection.cc:26
blender::nodes::node_composite_colorcorrection_cc::get_compositor_shader_node
static ShaderNode * get_compositor_shader_node(DNode node)
Definition node_composite_colorcorrection.cc:339
blender::nodes::node_composite_colorcorrection_cc::node_composit_init_colorcorrection
static void node_composit_init_colorcorrection(bNodeTree *, bNode *node)
Definition node_composite_colorcorrection.cc:39
blender::realtime_compositor
Definition BKE_node.hh:80
register_node_type_cmp_colorcorrection
void register_node_type_cmp_colorcorrection()
Definition node_composite_colorcorrection.cc:346
cmp_node_type_base
void cmp_node_type_base(blender::bke::bNodeType *ntype, int type, const char *name, short nclass)
Definition node_composite_util.cc:31
node_composite_util.hh
node_free_standard_storage
void node_free_standard_storage(bNode *node)
Definition node_util.cc:46
node_copy_standard_storage
void node_copy_standard_storage(bNodeTree *, bNode *dest_node, const bNode *src_node)
Definition node_util.cc:58
NodeColorCorrection::shadows
ColorCorrectionData shadows
Definition DNA_node_types.h:1048
NodeColorCorrection::midtones
ColorCorrectionData midtones
Definition DNA_node_types.h:1049
NodeColorCorrection::master
ColorCorrectionData master
Definition DNA_node_types.h:1047
NodeColorCorrection::highlights
ColorCorrectionData highlights
Definition DNA_node_types.h:1050
bNode::custom1
int16_t custom1
Definition DNA_node_types.h:400
blender::bke::bNodeType
Defines a node type.
Definition BKE_node.hh:218
blender::bke::bNodeType::initfunc
void(* initfunc)(bNodeTree *ntree, bNode *node)
Definition BKE_node.hh:267
blender::bke::bNodeType::draw_buttons_ex
void(* draw_buttons_ex)(uiLayout *, bContext *C, PointerRNA *ptr)
Definition BKE_node.hh:240
blender::bke::bNodeType::draw_buttons
void(* draw_buttons)(uiLayout *, bContext *C, PointerRNA *ptr)
Definition BKE_node.hh:238
blender::bke::bNodeType::declare
NodeDeclareFunction declare
Definition BKE_node.hh:347
blender::bke::bNodeType::get_compositor_shader_node
NodeGetCompositorShaderNodeFunction get_compositor_shader_node
Definition BKE_node.hh:328
ptr
PointerRNA * ptr
Definition wm_files.cc:4126
Generated on Thu Feb 6 2025 07:36:39 for Blender by  doxygen 1.11.0