griefith/test2
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Compositor: Implement Set Alpha for new CPU compositor

Browse Source 
Reference #125968.
This commit is contained in:
Omar Emara
2024-10-29 16:06:33 +03:00
parent 04ab5e1bde
commit 1dd0493c5a
1 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
source/blender/nodes/composite/nodes/node_composite_setalpha.cc
View File

@@ -54,6 +54,11 @@ static void node_composit_buts_set_alpha(uiLayout *layout, bContext * /*C*/, Poi
using namespace blender::realtime_compositor;
static CMPNodeSetAlphaMode get_mode(const bNode &node)
{
return static_cast<CMPNodeSetAlphaMode>(node_storage(node).mode);
}
class SetAlphaShaderNode : public ShaderNode {
public:
using ShaderNode::ShaderNode;
@@ -63,7 +68,7 @@ class SetAlphaShaderNode : public ShaderNode {
GPUNodeStack *inputs = get_inputs_array();
GPUNodeStack *outputs = get_outputs_array();
if (node_storage(bnode()).mode == CMP_NODE_SETALPHA_MODE_APPLY) {
if (get_mode(bnode()) == CMP_NODE_SETALPHA_MODE_APPLY) {
GPU_stack_link(material, &bnode(), "node_composite_set_alpha_apply", inputs, outputs);
return;
}
@@ -79,12 +84,24 @@ static ShaderNode *get_compositor_shader_node(DNode node)
static void node_build_multi_function(blender::nodes::NodeMultiFunctionBuilder &builder)
{
/* Not yet implemented. Return zero. */
static auto function = mf::build::SI2_SO<float4, float, float4>(
"Set Alpha",
[](const float4 & /*color*/, const float /*alpha*/) -> float4 { return float4(0.0f); },
static auto apply_function = mf::build::SI2_SO<float4, float, float4>(
"Set Alpha Apply",
[](const float4 &color, const float alpha) -> float4 { return color * alpha; },
mf::build::exec_presets::AllSpanOrSingle());
builder.set_matching_fn(function);
static auto replace_function = mf::build::SI2_SO<float4, float, float4>(
"Set Alpha Replace",
[](const float4 &color, const float alpha) -> float4 { return float4(color.xyz(), alpha); },
mf::build::exec_presets::AllSpanOrSingle());
switch (get_mode(builder.node())) {
case CMP_NODE_SETALPHA_MODE_APPLY:
builder.set_matching_fn(apply_function);
break;
case CMP_NODE_SETALPHA_MODE_REPLACE_ALPHA:
builder.set_matching_fn(replace_function);
break;
}
}
} // namespace blender::nodes::node_composite_setalpha_cc