2022-02-10 23:07:11 +01:00
|
|
|
/* SPDX-License-Identifier: Apache-2.0 */
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
#include "testing/testing.h"
|
|
|
|
|
|
|
|
#include "FN_multi_function.hh"
|
2020-06-22 15:50:08 +02:00
|
|
|
#include "FN_multi_function_builder.hh"
|
2021-08-20 13:14:39 +02:00
|
|
|
#include "FN_multi_function_test_common.hh"
|
2020-06-16 16:35:57 +02:00
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
namespace blender::fn::multi_function::tests {
|
2020-08-05 17:18:56 +02:00
|
|
|
namespace {
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
class AddFunction : public MultiFunction {
|
|
|
|
public:
|
|
|
|
AddFunction()
|
|
|
|
{
|
2023-01-07 18:00:37 +01:00
|
|
|
static Signature signature = []() {
|
|
|
|
Signature signature;
|
|
|
|
SignatureBuilder builder("Add", signature);
|
|
|
|
builder.single_input<int>("A");
|
|
|
|
builder.single_input<int>("B");
|
|
|
|
builder.single_output<int>("Result");
|
|
|
|
return signature;
|
|
|
|
}();
|
2021-03-22 11:57:24 +01:00
|
|
|
this->set_signature(&signature);
|
|
|
|
}
|
|
|
|
|
2023-01-14 15:42:52 +01:00
|
|
|
void call(IndexMask mask, Params params, Context /*context*/) const override
|
2020-06-16 16:35:57 +02:00
|
|
|
{
|
2021-03-21 19:31:24 +01:00
|
|
|
const VArray<int> &a = params.readonly_single_input<int>(0, "A");
|
|
|
|
const VArray<int> &b = params.readonly_single_input<int>(1, "B");
|
2020-06-16 16:35:57 +02:00
|
|
|
MutableSpan<int> result = params.uninitialized_single_output<int>(2, "Result");
|
|
|
|
|
2020-07-20 12:16:20 +02:00
|
|
|
for (int64_t i : mask) {
|
2020-06-16 16:35:57 +02:00
|
|
|
result[i] = a[i] + b[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
TEST(multi_function, AddFunction)
|
|
|
|
{
|
|
|
|
AddFunction fn;
|
|
|
|
|
|
|
|
Array<int> input1 = {4, 5, 6};
|
|
|
|
Array<int> input2 = {10, 20, 30};
|
|
|
|
Array<int> output(3, -1);
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, 3);
|
2020-06-16 16:35:57 +02:00
|
|
|
params.add_readonly_single_input(input1.as_span());
|
|
|
|
params.add_readonly_single_input(input2.as_span());
|
|
|
|
params.add_uninitialized_single_output(output.as_mutable_span());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
fn.call({0, 2}, params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(output[0], 14);
|
|
|
|
EXPECT_EQ(output[1], -1);
|
|
|
|
EXPECT_EQ(output[2], 36);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(multi_function, AddPrefixFunction)
|
|
|
|
{
|
|
|
|
AddPrefixFunction fn;
|
|
|
|
|
|
|
|
Array<std::string> strings = {
|
|
|
|
"Hello",
|
|
|
|
"World",
|
|
|
|
"This is a test",
|
|
|
|
"Another much longer string to trigger an allocation",
|
|
|
|
};
|
|
|
|
|
|
|
|
std::string prefix = "AB";
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, strings.size());
|
2020-06-16 16:35:57 +02:00
|
|
|
params.add_readonly_single_input(&prefix);
|
|
|
|
params.add_single_mutable(strings.as_mutable_span());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
fn.call({0, 2, 3}, params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(strings[0], "ABHello");
|
|
|
|
EXPECT_EQ(strings[1], "World");
|
|
|
|
EXPECT_EQ(strings[2], "ABThis is a test");
|
|
|
|
EXPECT_EQ(strings[3], "ABAnother much longer string to trigger an allocation");
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(multi_function, CreateRangeFunction)
|
|
|
|
{
|
|
|
|
CreateRangeFunction fn;
|
|
|
|
|
2021-08-20 13:14:39 +02:00
|
|
|
GVectorArray ranges(CPPType::get<int>(), 5);
|
|
|
|
GVectorArray_TypedMutableRef<int> ranges_ref{ranges};
|
|
|
|
Array<int> sizes = {3, 0, 6, 1, 4};
|
2020-06-16 16:35:57 +02:00
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, ranges.size());
|
2020-06-16 16:35:57 +02:00
|
|
|
params.add_readonly_single_input(sizes.as_span());
|
|
|
|
params.add_vector_output(ranges);
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
fn.call({0, 1, 2, 3}, params, context);
|
|
|
|
|
2021-03-21 19:31:24 +01:00
|
|
|
EXPECT_EQ(ranges[0].size(), 3);
|
|
|
|
EXPECT_EQ(ranges[1].size(), 0);
|
|
|
|
EXPECT_EQ(ranges[2].size(), 6);
|
|
|
|
EXPECT_EQ(ranges[3].size(), 1);
|
|
|
|
EXPECT_EQ(ranges[4].size(), 0);
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
EXPECT_EQ(ranges_ref[0][0], 0);
|
|
|
|
EXPECT_EQ(ranges_ref[0][1], 1);
|
|
|
|
EXPECT_EQ(ranges_ref[0][2], 2);
|
|
|
|
EXPECT_EQ(ranges_ref[2][0], 0);
|
|
|
|
EXPECT_EQ(ranges_ref[2][1], 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(multi_function, GenericAppendFunction)
|
|
|
|
{
|
2020-07-17 14:15:06 +02:00
|
|
|
GenericAppendFunction fn(CPPType::get<int32_t>());
|
2020-06-16 16:35:57 +02:00
|
|
|
|
2020-07-17 14:15:06 +02:00
|
|
|
GVectorArray vectors(CPPType::get<int32_t>(), 4);
|
2021-03-21 19:31:24 +01:00
|
|
|
GVectorArray_TypedMutableRef<int> vectors_ref{vectors};
|
2020-06-16 16:35:57 +02:00
|
|
|
vectors_ref.append(0, 1);
|
|
|
|
vectors_ref.append(0, 2);
|
|
|
|
vectors_ref.append(2, 6);
|
|
|
|
Array<int> values = {5, 7, 3, 1};
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, vectors.size());
|
2020-06-16 16:35:57 +02:00
|
|
|
params.add_vector_mutable(vectors);
|
|
|
|
params.add_readonly_single_input(values.as_span());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
fn.call(IndexRange(vectors.size()), params, context);
|
|
|
|
|
2021-03-21 19:31:24 +01:00
|
|
|
EXPECT_EQ(vectors[0].size(), 3);
|
|
|
|
EXPECT_EQ(vectors[1].size(), 1);
|
|
|
|
EXPECT_EQ(vectors[2].size(), 2);
|
|
|
|
EXPECT_EQ(vectors[3].size(), 1);
|
2020-06-16 16:35:57 +02:00
|
|
|
|
|
|
|
EXPECT_EQ(vectors_ref[0][0], 1);
|
|
|
|
EXPECT_EQ(vectors_ref[0][1], 2);
|
|
|
|
EXPECT_EQ(vectors_ref[0][2], 5);
|
|
|
|
EXPECT_EQ(vectors_ref[1][0], 7);
|
|
|
|
EXPECT_EQ(vectors_ref[2][0], 6);
|
|
|
|
EXPECT_EQ(vectors_ref[2][1], 3);
|
|
|
|
EXPECT_EQ(vectors_ref[3][0], 1);
|
|
|
|
}
|
|
|
|
|
2020-06-30 17:59:33 +02:00
|
|
|
TEST(multi_function, CustomMF_Constant)
|
|
|
|
{
|
|
|
|
CustomMF_Constant<int> fn{42};
|
|
|
|
|
|
|
|
Array<int> outputs(4, 0);
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, outputs.size());
|
2020-06-30 17:59:33 +02:00
|
|
|
params.add_uninitialized_single_output(outputs.as_mutable_span());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-06-30 17:59:33 +02:00
|
|
|
|
|
|
|
fn.call({0, 2, 3}, params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(outputs[0], 42);
|
|
|
|
EXPECT_EQ(outputs[1], 0);
|
|
|
|
EXPECT_EQ(outputs[2], 42);
|
|
|
|
EXPECT_EQ(outputs[3], 42);
|
|
|
|
}
|
|
|
|
|
2020-07-07 19:34:35 +02:00
|
|
|
TEST(multi_function, CustomMF_GenericConstant)
|
|
|
|
{
|
|
|
|
int value = 42;
|
2021-09-11 13:05:20 +02:00
|
|
|
CustomMF_GenericConstant fn{CPPType::get<int32_t>(), (const void *)&value, false};
|
2020-07-07 19:34:35 +02:00
|
|
|
|
|
|
|
Array<int> outputs(4, 0);
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, outputs.size());
|
2020-07-07 19:34:35 +02:00
|
|
|
params.add_uninitialized_single_output(outputs.as_mutable_span());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-07-07 19:34:35 +02:00
|
|
|
|
|
|
|
fn.call({0, 1, 2}, params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(outputs[0], 42);
|
|
|
|
EXPECT_EQ(outputs[1], 42);
|
|
|
|
EXPECT_EQ(outputs[2], 42);
|
|
|
|
EXPECT_EQ(outputs[3], 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(multi_function, CustomMF_GenericConstantArray)
|
|
|
|
{
|
|
|
|
std::array<int, 4> values = {3, 4, 5, 6};
|
|
|
|
CustomMF_GenericConstantArray fn{GSpan(Span(values))};
|
|
|
|
|
2021-03-21 19:31:24 +01:00
|
|
|
GVectorArray vector_array{CPPType::get<int32_t>(), 4};
|
|
|
|
GVectorArray_TypedMutableRef<int> vector_array_ref{vector_array};
|
2020-07-07 19:34:35 +02:00
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, vector_array.size());
|
2021-03-21 19:31:24 +01:00
|
|
|
params.add_vector_output(vector_array);
|
2020-07-07 19:34:35 +02:00
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2020-07-07 19:34:35 +02:00
|
|
|
|
|
|
|
fn.call({1, 2, 3}, params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(vector_array[0].size(), 0);
|
|
|
|
EXPECT_EQ(vector_array[1].size(), 4);
|
|
|
|
EXPECT_EQ(vector_array[2].size(), 4);
|
|
|
|
EXPECT_EQ(vector_array[3].size(), 4);
|
2020-07-20 12:16:20 +02:00
|
|
|
for (int i = 1; i < 4; i++) {
|
2021-03-21 19:31:24 +01:00
|
|
|
EXPECT_EQ(vector_array_ref[i][0], 3);
|
|
|
|
EXPECT_EQ(vector_array_ref[i][1], 4);
|
|
|
|
EXPECT_EQ(vector_array_ref[i][2], 5);
|
|
|
|
EXPECT_EQ(vector_array_ref[i][3], 6);
|
2020-07-07 19:34:35 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-09-14 14:52:44 +02:00
|
|
|
TEST(multi_function, IgnoredOutputs)
|
|
|
|
{
|
|
|
|
OptionalOutputsFunction fn;
|
|
|
|
{
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, 10);
|
2021-09-14 14:52:44 +02:00
|
|
|
params.add_ignored_single_output("Out 1");
|
|
|
|
params.add_ignored_single_output("Out 2");
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2021-09-14 14:52:44 +02:00
|
|
|
fn.call(IndexRange(10), params, context);
|
|
|
|
}
|
|
|
|
{
|
|
|
|
Array<int> results_1(10);
|
|
|
|
Array<std::string> results_2(10, NoInitialization());
|
|
|
|
|
2023-01-07 17:32:28 +01:00
|
|
|
ParamsBuilder params(fn, 10);
|
2021-09-14 14:52:44 +02:00
|
|
|
params.add_uninitialized_single_output(results_1.as_mutable_span(), "Out 1");
|
|
|
|
params.add_uninitialized_single_output(results_2.as_mutable_span(), "Out 2");
|
2023-01-07 17:32:28 +01:00
|
|
|
ContextBuilder context;
|
2021-09-14 14:52:44 +02:00
|
|
|
fn.call(IndexRange(10), params, context);
|
|
|
|
|
|
|
|
EXPECT_EQ(results_1[0], 5);
|
|
|
|
EXPECT_EQ(results_1[3], 5);
|
|
|
|
EXPECT_EQ(results_1[9], 5);
|
|
|
|
EXPECT_EQ(results_2[0], "hello, this is a long string");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-08-05 17:18:56 +02:00
|
|
|
} // namespace
|
2023-01-07 17:32:28 +01:00
|
|
|
} // namespace blender::fn::multi_function::tests
|