mirror of
https://github.com/blender/blender
synced 2026-09-29 04:37:17 +03:00
ba82476ade
This adds the Integer Math node to shader nodes. It's the same node that can already be used in Geometry Nodes and the Compositor. While creating the regression test, I noticed that the integer math implementation was a bit broken: * The integer modulo (`%`) operator on the GPU is undefined for negative divisors (the added regression test showed that the behavior is not consistent across platforms). * Use integer power function instead of using float implementation with casts. The MaterialX implementation is omitted for now. I didn't look into it in detail yet but for the Boolean Math node (#162798) it seemed like some more work is needed there to support this type. Pull Request: https://projects.blender.org/blender/blender/pulls/163805
118 lines
3.1 KiB
C
118 lines
3.1 KiB
C
/* SPDX-FileCopyrightText: 2026 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 */
|
|
|
|
#pragma once
|
|
|
|
#include "kernel/svm/node_types.h"
|
|
#include "kernel/svm/util.h"
|
|
|
|
CCL_NAMESPACE_BEGIN
|
|
|
|
ccl_device int svm_integer_math_divide_floor(const int a, const int b)
|
|
{
|
|
const int d = a / b;
|
|
const int r = a % b;
|
|
return (r != 0) ? (d - int((a < 0) != (b < 0))) : d;
|
|
}
|
|
|
|
ccl_device int svm_integer_math_gcd(int a, int b)
|
|
{
|
|
a = abs(a);
|
|
b = abs(b);
|
|
while (b != 0) {
|
|
const int t = b;
|
|
b = a % b;
|
|
a = t;
|
|
}
|
|
return a;
|
|
}
|
|
|
|
ccl_device int svm_integer_math_power(int base, int exponent)
|
|
{
|
|
if (exponent < 0) {
|
|
if (base == 1 || base == -1) {
|
|
return (base < 0 && (exponent & 1) != 0) ? -1 : 1;
|
|
}
|
|
return 0;
|
|
}
|
|
int result = 1;
|
|
while (exponent != 0) {
|
|
if ((exponent & 1) != 0) {
|
|
result *= base;
|
|
}
|
|
exponent >>= 1;
|
|
base *= base;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
ccl_device int svm_integer_math(const NodeIntegerMathType type,
|
|
const int a,
|
|
const int b,
|
|
const int c)
|
|
{
|
|
switch (type) {
|
|
case NODE_INTEGER_MATH_ADD:
|
|
return a + b;
|
|
case NODE_INTEGER_MATH_SUBTRACT:
|
|
return a - b;
|
|
case NODE_INTEGER_MATH_MULTIPLY:
|
|
return a * b;
|
|
case NODE_INTEGER_MATH_DIVIDE:
|
|
return (b != 0) ? (a / b) : 0;
|
|
case NODE_INTEGER_MATH_MULTIPLY_ADD:
|
|
return a * b + c;
|
|
case NODE_INTEGER_MATH_POWER:
|
|
return svm_integer_math_power(a, b);
|
|
case NODE_INTEGER_MATH_FLOORED_MODULO:
|
|
return (b != 0) ? (((a % b) + b) % b) : 0;
|
|
case NODE_INTEGER_MATH_ABSOLUTE:
|
|
return abs(a);
|
|
case NODE_INTEGER_MATH_MINIMUM:
|
|
return min(a, b);
|
|
case NODE_INTEGER_MATH_MAXIMUM:
|
|
return max(a, b);
|
|
case NODE_INTEGER_MATH_GCD:
|
|
return svm_integer_math_gcd(a, b);
|
|
case NODE_INTEGER_MATH_LCM: {
|
|
const int gcd = svm_integer_math_gcd(a, b);
|
|
return (gcd != 0) ? abs(a / gcd * b) : 0;
|
|
}
|
|
case NODE_INTEGER_MATH_NEGATE:
|
|
return -a;
|
|
case NODE_INTEGER_MATH_SIGN:
|
|
return int(0 < a) - int(a < 0);
|
|
case NODE_INTEGER_MATH_DIVIDE_FLOOR:
|
|
return (b != 0) ? svm_integer_math_divide_floor(a, b) : 0;
|
|
case NODE_INTEGER_MATH_DIVIDE_CEIL:
|
|
return (b != 0) ? -svm_integer_math_divide_floor(a, -b) : 0;
|
|
case NODE_INTEGER_MATH_DIVIDE_ROUND: {
|
|
const int abs_b = abs(b);
|
|
const int sign_b = int(0 < b) - int(b < 0);
|
|
if (a >= 0) {
|
|
return ((abs_b != 0) ? ((2 * a + abs_b) / (2 * abs_b)) : 0) * sign_b;
|
|
}
|
|
return -((abs_b != 0) ? ((2 * -a + abs_b) / (2 * abs_b)) : 0) * sign_b;
|
|
}
|
|
case NODE_INTEGER_MATH_MODULO:
|
|
return (b != 0) ? (a % b) : 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
ccl_device_noinline void svm_node_integer_math(
|
|
ccl_private float *ccl_restrict stack, const ccl_global SVMNodeIntegerMath &ccl_restrict node)
|
|
{
|
|
const int a = stack_load(stack, node.value1);
|
|
const int b = stack_load(stack, node.value2);
|
|
const int c = stack_load(stack, node.value3);
|
|
const int result = svm_integer_math(node.math_type, a, b, c);
|
|
|
|
if (stack_valid(node.result_offset)) {
|
|
stack_store_int(stack, node.result_offset, result);
|
|
}
|
|
}
|
|
|
|
CCL_NAMESPACE_END
|