fbdb803750
This patch implements the multi-function procedure operation for the new CPU compositor, which is a concrete implementation of the PixelOperation abstraction, much like ShaderOperation, but uses the FN system to more efficiently evaluate a group of pixel-wise operations. A few changes were done to FN to support development. The multi-function builder now allows retrieving the built function. A new builder method construct_and_set_matching_fn_cb was added to allow using the SI_SO builders with non static functions. A few other SI_SO were added to. And a CPP type for float4 was added. Additionally, the Gamma, Math, Brightness, and Normal nodes were implemented as an example. The Math node implementation reused the existing GN math node implementation, so the code was moved to a common file. Reference #125968. Pull Request: https://projects.blender.org/blender/blender/pulls/126988
289 lines
10 KiB
C++
289 lines
10 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include "NOD_math_functions.hh"
|
|
|
|
namespace blender::nodes {
|
|
|
|
static const mf::MultiFunction *get_base_multi_function(const bNode &node)
|
|
{
|
|
const int mode = node.custom1;
|
|
const mf::MultiFunction *base_fn = nullptr;
|
|
|
|
try_dispatch_float_math_fl_to_fl(
|
|
mode, [&](auto devi_fn, auto function, const FloatMathOperationInfo &info) {
|
|
static auto fn = mf::build::SI1_SO<float, float>(
|
|
info.title_case_name.c_str(), function, devi_fn);
|
|
base_fn = &fn;
|
|
});
|
|
if (base_fn != nullptr) {
|
|
return base_fn;
|
|
}
|
|
|
|
try_dispatch_float_math_fl_fl_to_fl(
|
|
mode, [&](auto devi_fn, auto function, const FloatMathOperationInfo &info) {
|
|
static auto fn = mf::build::SI2_SO<float, float, float>(
|
|
info.title_case_name.c_str(), function, devi_fn);
|
|
base_fn = &fn;
|
|
});
|
|
if (base_fn != nullptr) {
|
|
return base_fn;
|
|
}
|
|
|
|
try_dispatch_float_math_fl_fl_fl_to_fl(
|
|
mode, [&](auto devi_fn, auto function, const FloatMathOperationInfo &info) {
|
|
static auto fn = mf::build::SI3_SO<float, float, float, float>(
|
|
info.title_case_name.c_str(), function, devi_fn);
|
|
base_fn = &fn;
|
|
});
|
|
if (base_fn != nullptr) {
|
|
return base_fn;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
class ClampWrapperFunction : public mf::MultiFunction {
|
|
private:
|
|
const mf::MultiFunction &fn_;
|
|
|
|
public:
|
|
ClampWrapperFunction(const mf::MultiFunction &fn) : fn_(fn)
|
|
{
|
|
this->set_signature(&fn.signature());
|
|
}
|
|
|
|
void call(const IndexMask &mask, mf::Params params, mf::Context context) const override
|
|
{
|
|
fn_.call(mask, params, context);
|
|
|
|
/* Assumes the output parameter is the last one. */
|
|
const int output_param_index = this->param_amount() - 1;
|
|
/* This has actually been initialized in the call above. */
|
|
MutableSpan<float> results = params.uninitialized_single_output<float>(output_param_index);
|
|
|
|
mask.foreach_index_optimized<int>([&](const int i) {
|
|
float &value = results[i];
|
|
CLAMP(value, 0.0f, 1.0f);
|
|
});
|
|
}
|
|
};
|
|
|
|
void node_math_build_multi_function(NodeMultiFunctionBuilder &builder)
|
|
{
|
|
const mf::MultiFunction *base_function = get_base_multi_function(builder.node());
|
|
|
|
const bool clamp_output = builder.node().custom2 != 0;
|
|
if (clamp_output) {
|
|
builder.construct_and_set_matching_fn<ClampWrapperFunction>(*base_function);
|
|
}
|
|
else {
|
|
builder.set_matching_fn(base_function);
|
|
}
|
|
}
|
|
|
|
const FloatMathOperationInfo *get_float_math_operation_info(const int operation)
|
|
{
|
|
|
|
#define RETURN_OPERATION_INFO(title_case_name, shader_name) \
|
|
{ \
|
|
static const FloatMathOperationInfo info{title_case_name, shader_name}; \
|
|
return &info; \
|
|
} \
|
|
((void)0)
|
|
|
|
switch (operation) {
|
|
case NODE_MATH_ADD:
|
|
RETURN_OPERATION_INFO("Add", "math_add");
|
|
case NODE_MATH_SUBTRACT:
|
|
RETURN_OPERATION_INFO("Subtract", "math_subtract");
|
|
case NODE_MATH_MULTIPLY:
|
|
RETURN_OPERATION_INFO("Multiply", "math_multiply");
|
|
case NODE_MATH_DIVIDE:
|
|
RETURN_OPERATION_INFO("Divide", "math_divide");
|
|
case NODE_MATH_SINE:
|
|
RETURN_OPERATION_INFO("Sine", "math_sine");
|
|
case NODE_MATH_COSINE:
|
|
RETURN_OPERATION_INFO("Cosine", "math_cosine");
|
|
case NODE_MATH_TANGENT:
|
|
RETURN_OPERATION_INFO("Tangent", "math_tangent");
|
|
case NODE_MATH_ARCSINE:
|
|
RETURN_OPERATION_INFO("Arc Sine", "math_arcsine");
|
|
case NODE_MATH_ARCCOSINE:
|
|
RETURN_OPERATION_INFO("Arc Cosine", "math_arccosine");
|
|
case NODE_MATH_ARCTANGENT:
|
|
RETURN_OPERATION_INFO("Arc Tangent", "math_arctangent");
|
|
case NODE_MATH_POWER:
|
|
RETURN_OPERATION_INFO("Power", "math_power");
|
|
case NODE_MATH_LOGARITHM:
|
|
RETURN_OPERATION_INFO("Logarithm", "math_logarithm");
|
|
case NODE_MATH_MINIMUM:
|
|
RETURN_OPERATION_INFO("Minimum", "math_minimum");
|
|
case NODE_MATH_MAXIMUM:
|
|
RETURN_OPERATION_INFO("Maximum", "math_maximum");
|
|
case NODE_MATH_ROUND:
|
|
RETURN_OPERATION_INFO("Round", "math_round");
|
|
case NODE_MATH_LESS_THAN:
|
|
RETURN_OPERATION_INFO("Less Than", "math_less_than");
|
|
case NODE_MATH_GREATER_THAN:
|
|
RETURN_OPERATION_INFO("Greater Than", "math_greater_than");
|
|
case NODE_MATH_MODULO:
|
|
RETURN_OPERATION_INFO("Modulo", "math_modulo");
|
|
case NODE_MATH_FLOORED_MODULO:
|
|
RETURN_OPERATION_INFO("Floored Modulo", "math_floored_modulo");
|
|
case NODE_MATH_ABSOLUTE:
|
|
RETURN_OPERATION_INFO("Absolute", "math_absolute");
|
|
case NODE_MATH_ARCTAN2:
|
|
RETURN_OPERATION_INFO("Arc Tangent 2", "math_arctan2");
|
|
case NODE_MATH_FLOOR:
|
|
RETURN_OPERATION_INFO("Floor", "math_floor");
|
|
case NODE_MATH_CEIL:
|
|
RETURN_OPERATION_INFO("Ceil", "math_ceil");
|
|
case NODE_MATH_FRACTION:
|
|
RETURN_OPERATION_INFO("Fraction", "math_fraction");
|
|
case NODE_MATH_SQRT:
|
|
RETURN_OPERATION_INFO("Sqrt", "math_sqrt");
|
|
case NODE_MATH_INV_SQRT:
|
|
RETURN_OPERATION_INFO("Inverse Sqrt", "math_inversesqrt");
|
|
case NODE_MATH_SIGN:
|
|
RETURN_OPERATION_INFO("Sign", "math_sign");
|
|
case NODE_MATH_EXPONENT:
|
|
RETURN_OPERATION_INFO("Exponent", "math_exponent");
|
|
case NODE_MATH_RADIANS:
|
|
RETURN_OPERATION_INFO("Radians", "math_radians");
|
|
case NODE_MATH_DEGREES:
|
|
RETURN_OPERATION_INFO("Degrees", "math_degrees");
|
|
case NODE_MATH_SINH:
|
|
RETURN_OPERATION_INFO("Hyperbolic Sine", "math_sinh");
|
|
case NODE_MATH_COSH:
|
|
RETURN_OPERATION_INFO("Hyperbolic Cosine", "math_cosh");
|
|
case NODE_MATH_TANH:
|
|
RETURN_OPERATION_INFO("Hyperbolic Tangent", "math_tanh");
|
|
case NODE_MATH_TRUNC:
|
|
RETURN_OPERATION_INFO("Truncate", "math_trunc");
|
|
case NODE_MATH_SNAP:
|
|
RETURN_OPERATION_INFO("Snap", "math_snap");
|
|
case NODE_MATH_WRAP:
|
|
RETURN_OPERATION_INFO("Wrap", "math_wrap");
|
|
case NODE_MATH_COMPARE:
|
|
RETURN_OPERATION_INFO("Compare", "math_compare");
|
|
case NODE_MATH_MULTIPLY_ADD:
|
|
RETURN_OPERATION_INFO("Multiply Add", "math_multiply_add");
|
|
case NODE_MATH_PINGPONG:
|
|
RETURN_OPERATION_INFO("Ping Pong", "math_pingpong");
|
|
case NODE_MATH_SMOOTH_MIN:
|
|
RETURN_OPERATION_INFO("Smooth Min", "math_smoothmin");
|
|
case NODE_MATH_SMOOTH_MAX:
|
|
RETURN_OPERATION_INFO("Smooth Max", "math_smoothmax");
|
|
}
|
|
|
|
#undef RETURN_OPERATION_INFO
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
const FloatMathOperationInfo *get_float_compare_operation_info(const int operation)
|
|
{
|
|
|
|
#define RETURN_OPERATION_INFO(title_case_name, shader_name) \
|
|
{ \
|
|
static const FloatMathOperationInfo info{title_case_name, shader_name}; \
|
|
return &info; \
|
|
} \
|
|
((void)0)
|
|
|
|
switch (operation) {
|
|
case NODE_COMPARE_LESS_THAN:
|
|
RETURN_OPERATION_INFO("Less Than", "math_less_than");
|
|
case NODE_COMPARE_LESS_EQUAL:
|
|
RETURN_OPERATION_INFO("Less Than or Equal", "math_less_equal");
|
|
case NODE_COMPARE_GREATER_THAN:
|
|
RETURN_OPERATION_INFO("Greater Than", "math_greater_than");
|
|
case NODE_COMPARE_GREATER_EQUAL:
|
|
RETURN_OPERATION_INFO("Greater Than or Equal", "math_greater_equal");
|
|
case NODE_COMPARE_EQUAL:
|
|
RETURN_OPERATION_INFO("Equal", "math_equal");
|
|
case NODE_COMPARE_NOT_EQUAL:
|
|
RETURN_OPERATION_INFO("Not Equal", "math_not_equal");
|
|
}
|
|
|
|
#undef RETURN_OPERATION_INFO
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
const FloatMathOperationInfo *get_float3_math_operation_info(const int operation)
|
|
{
|
|
|
|
#define RETURN_OPERATION_INFO(title_case_name, shader_name) \
|
|
{ \
|
|
static const FloatMathOperationInfo info{title_case_name, shader_name}; \
|
|
return &info; \
|
|
} \
|
|
((void)0)
|
|
|
|
switch (operation) {
|
|
case NODE_VECTOR_MATH_ADD:
|
|
RETURN_OPERATION_INFO("Add", "vector_math_add");
|
|
case NODE_VECTOR_MATH_SUBTRACT:
|
|
RETURN_OPERATION_INFO("Subtract", "vector_math_subtract");
|
|
case NODE_VECTOR_MATH_MULTIPLY:
|
|
RETURN_OPERATION_INFO("Multiply", "vector_math_multiply");
|
|
case NODE_VECTOR_MATH_DIVIDE:
|
|
RETURN_OPERATION_INFO("Divide", "vector_math_divide");
|
|
case NODE_VECTOR_MATH_CROSS_PRODUCT:
|
|
RETURN_OPERATION_INFO("Cross Product", "vector_math_cross");
|
|
case NODE_VECTOR_MATH_PROJECT:
|
|
RETURN_OPERATION_INFO("Project", "vector_math_project");
|
|
case NODE_VECTOR_MATH_REFLECT:
|
|
RETURN_OPERATION_INFO("Reflect", "vector_math_reflect");
|
|
case NODE_VECTOR_MATH_DOT_PRODUCT:
|
|
RETURN_OPERATION_INFO("Dot Product", "vector_math_dot");
|
|
case NODE_VECTOR_MATH_DISTANCE:
|
|
RETURN_OPERATION_INFO("Distance", "vector_math_distance");
|
|
case NODE_VECTOR_MATH_LENGTH:
|
|
RETURN_OPERATION_INFO("Length", "vector_math_length");
|
|
case NODE_VECTOR_MATH_SCALE:
|
|
RETURN_OPERATION_INFO("Scale", "vector_math_scale");
|
|
case NODE_VECTOR_MATH_NORMALIZE:
|
|
RETURN_OPERATION_INFO("Normalize", "vector_math_normalize");
|
|
case NODE_VECTOR_MATH_SNAP:
|
|
RETURN_OPERATION_INFO("Snap", "vector_math_snap");
|
|
case NODE_VECTOR_MATH_FLOOR:
|
|
RETURN_OPERATION_INFO("Floor", "vector_math_floor");
|
|
case NODE_VECTOR_MATH_CEIL:
|
|
RETURN_OPERATION_INFO("Ceiling", "vector_math_ceil");
|
|
case NODE_VECTOR_MATH_MODULO:
|
|
RETURN_OPERATION_INFO("Modulo", "vector_math_modulo");
|
|
case NODE_VECTOR_MATH_FRACTION:
|
|
RETURN_OPERATION_INFO("Fraction", "vector_math_fraction");
|
|
case NODE_VECTOR_MATH_ABSOLUTE:
|
|
RETURN_OPERATION_INFO("Absolute", "vector_math_absolute");
|
|
case NODE_VECTOR_MATH_MINIMUM:
|
|
RETURN_OPERATION_INFO("Minimum", "vector_math_minimum");
|
|
case NODE_VECTOR_MATH_MAXIMUM:
|
|
RETURN_OPERATION_INFO("Maximum", "vector_math_maximum");
|
|
case NODE_VECTOR_MATH_WRAP:
|
|
RETURN_OPERATION_INFO("Wrap", "vector_math_wrap");
|
|
case NODE_VECTOR_MATH_SINE:
|
|
RETURN_OPERATION_INFO("Sine", "vector_math_sine");
|
|
case NODE_VECTOR_MATH_COSINE:
|
|
RETURN_OPERATION_INFO("Cosine", "vector_math_cosine");
|
|
case NODE_VECTOR_MATH_TANGENT:
|
|
RETURN_OPERATION_INFO("Tangent", "vector_math_tangent");
|
|
case NODE_VECTOR_MATH_REFRACT:
|
|
RETURN_OPERATION_INFO("Refract", "vector_math_refract");
|
|
case NODE_VECTOR_MATH_FACEFORWARD:
|
|
RETURN_OPERATION_INFO("Faceforward", "vector_math_faceforward");
|
|
case NODE_VECTOR_MATH_MULTIPLY_ADD:
|
|
RETURN_OPERATION_INFO("Multiply Add", "vector_math_multiply_add");
|
|
}
|
|
|
|
#undef RETURN_OPERATION_INFO
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
} // namespace blender::nodes
|