Welcome to mirror list, hosted at ThFree Co, Russian Federation.

git.blender.org/blender.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/tests/gtests/functions/FN_multi_function_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tests/gtests/functions/FN_multi_function_test.cc')
-rw-r--r--tests/gtests/functions/FN_multi_function_test.cc385
1 files changed, 0 insertions, 385 deletions
diff --git a/tests/gtests/functions/FN_multi_function_test.cc b/tests/gtests/functions/FN_multi_function_test.cc
deleted file mode 100644
index 8b5fb060c36..00000000000
--- a/tests/gtests/functions/FN_multi_function_test.cc
+++ /dev/null
@@ -1,385 +0,0 @@
-/* Apache License, Version 2.0 */
-
-#include "testing/testing.h"
-
-#include "FN_multi_function.hh"
-#include "FN_multi_function_builder.hh"
-
-namespace blender::fn {
-
-class AddFunction : public MultiFunction {
- public:
- AddFunction()
- {
- MFSignatureBuilder builder = this->get_builder("Add");
- builder.single_input<int>("A");
- builder.single_input<int>("B");
- builder.single_output<int>("Result");
- }
-
- void call(IndexMask mask, MFParams params, MFContext UNUSED(context)) const override
- {
- VSpan<int> a = params.readonly_single_input<int>(0, "A");
- VSpan<int> b = params.readonly_single_input<int>(1, "B");
- MutableSpan<int> result = params.uninitialized_single_output<int>(2, "Result");
-
- for (int64_t i : mask) {
- 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);
-
- MFParamsBuilder params(fn, 3);
- 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());
-
- MFContextBuilder context;
-
- fn.call({0, 2}, params, context);
-
- EXPECT_EQ(output[0], 14);
- EXPECT_EQ(output[1], -1);
- EXPECT_EQ(output[2], 36);
-}
-
-class AddPrefixFunction : public MultiFunction {
- public:
- AddPrefixFunction()
- {
- MFSignatureBuilder builder = this->get_builder("Add Prefix");
- builder.single_input<std::string>("Prefix");
- builder.single_mutable<std::string>("Strings");
- }
-
- void call(IndexMask mask, MFParams params, MFContext UNUSED(context)) const override
- {
- VSpan<std::string> prefixes = params.readonly_single_input<std::string>(0, "Prefix");
- MutableSpan<std::string> strings = params.single_mutable<std::string>(1, "Strings");
-
- for (int64_t i : mask) {
- strings[i] = prefixes[i] + strings[i];
- }
- }
-};
-
-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";
-
- MFParamsBuilder params(fn, strings.size());
- params.add_readonly_single_input(&prefix);
- params.add_single_mutable(strings.as_mutable_span());
-
- MFContextBuilder context;
-
- 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");
-}
-
-class CreateRangeFunction : public MultiFunction {
- public:
- CreateRangeFunction()
- {
- MFSignatureBuilder builder = this->get_builder("Create Range");
- builder.single_input<uint>("Size");
- builder.vector_output<uint>("Range");
- }
-
- void call(IndexMask mask, MFParams params, MFContext UNUSED(context)) const override
- {
- VSpan<uint> sizes = params.readonly_single_input<uint>(0, "Size");
- GVectorArrayRef<uint> ranges = params.vector_output<uint>(1, "Range");
-
- for (int64_t i : mask) {
- uint size = sizes[i];
- for (uint j : IndexRange(size)) {
- ranges.append(i, j);
- }
- }
- }
-};
-
-TEST(multi_function, CreateRangeFunction)
-{
- CreateRangeFunction fn;
-
- GVectorArray ranges(CPPType::get<int32_t>(), 5);
- GVectorArrayRef<uint> ranges_ref(ranges);
- Array<uint> sizes = {3, 0, 6, 1, 4};
-
- MFParamsBuilder params(fn, ranges.size());
- params.add_readonly_single_input(sizes.as_span());
- params.add_vector_output(ranges);
-
- MFContextBuilder context;
-
- fn.call({0, 1, 2, 3}, params, context);
-
- EXPECT_EQ(ranges_ref[0].size(), 3);
- EXPECT_EQ(ranges_ref[1].size(), 0);
- EXPECT_EQ(ranges_ref[2].size(), 6);
- EXPECT_EQ(ranges_ref[3].size(), 1);
- EXPECT_EQ(ranges_ref[4].size(), 0);
-
- 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);
-}
-
-class GenericAppendFunction : public MultiFunction {
- public:
- GenericAppendFunction(const CPPType &type)
- {
- MFSignatureBuilder builder = this->get_builder("Append");
- builder.vector_mutable("Vector", type);
- builder.single_input("Value", type);
- }
-
- void call(IndexMask mask, MFParams params, MFContext UNUSED(context)) const override
- {
- GVectorArray &vectors = params.vector_mutable(0, "Vector");
- GVSpan values = params.readonly_single_input(1, "Value");
-
- for (int64_t i : mask) {
- vectors.append(i, values[i]);
- }
- }
-};
-
-TEST(multi_function, GenericAppendFunction)
-{
- GenericAppendFunction fn(CPPType::get<int32_t>());
-
- GVectorArray vectors(CPPType::get<int32_t>(), 4);
- GVectorArrayRef<int> vectors_ref(vectors);
- vectors_ref.append(0, 1);
- vectors_ref.append(0, 2);
- vectors_ref.append(2, 6);
- Array<int> values = {5, 7, 3, 1};
-
- MFParamsBuilder params(fn, vectors.size());
- params.add_vector_mutable(vectors);
- params.add_readonly_single_input(values.as_span());
-
- MFContextBuilder context;
-
- fn.call(IndexRange(vectors.size()), params, context);
-
- EXPECT_EQ(vectors_ref[0].size(), 3);
- EXPECT_EQ(vectors_ref[1].size(), 1);
- EXPECT_EQ(vectors_ref[2].size(), 2);
- EXPECT_EQ(vectors_ref[3].size(), 1);
-
- 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);
-}
-
-TEST(multi_function, CustomMF_SI_SO)
-{
- CustomMF_SI_SO<std::string, uint> fn("strlen",
- [](const std::string &str) { return str.size(); });
-
- Array<std::string> strings = {"hello", "world", "test", "another test"};
- Array<uint> sizes(strings.size(), 0);
-
- MFParamsBuilder params(fn, strings.size());
- params.add_readonly_single_input(strings.as_span());
- params.add_uninitialized_single_output(sizes.as_mutable_span());
-
- MFContextBuilder context;
-
- fn.call(IndexRange(strings.size()), params, context);
-
- EXPECT_EQ(sizes[0], 5);
- EXPECT_EQ(sizes[1], 5);
- EXPECT_EQ(sizes[2], 4);
- EXPECT_EQ(sizes[3], 12);
-}
-
-TEST(multi_function, CustomMF_SI_SI_SO)
-{
- CustomMF_SI_SI_SO<int, int, int> fn("mul", [](int a, int b) { return a * b; });
-
- Array<int> values_a = {4, 6, 8, 9};
- int value_b = 10;
- Array<int> outputs(values_a.size(), -1);
-
- MFParamsBuilder params(fn, values_a.size());
- params.add_readonly_single_input(values_a.as_span());
- params.add_readonly_single_input(&value_b);
- params.add_uninitialized_single_output(outputs.as_mutable_span());
-
- MFContextBuilder context;
-
- fn.call({0, 1, 3}, params, context);
-
- EXPECT_EQ(outputs[0], 40);
- EXPECT_EQ(outputs[1], 60);
- EXPECT_EQ(outputs[2], -1);
- EXPECT_EQ(outputs[3], 90);
-}
-
-TEST(multi_function, CustomMF_SI_SI_SI_SO)
-{
- CustomMF_SI_SI_SI_SO<int, std::string, bool, uint> fn{
- "custom",
- [](int a, const std::string &b, bool c) { return (uint)((uint)a + b.size() + (uint)c); }};
-
- Array<int> values_a = {5, 7, 3, 8};
- Array<std::string> values_b = {"hello", "world", "another", "test"};
- Array<bool> values_c = {true, false, false, true};
- Array<uint> outputs(values_a.size(), 0);
-
- MFParamsBuilder params(fn, values_a.size());
- params.add_readonly_single_input(values_a.as_span());
- params.add_readonly_single_input(values_b.as_span());
- params.add_readonly_single_input(values_c.as_span());
- params.add_uninitialized_single_output(outputs.as_mutable_span());
-
- MFContextBuilder context;
-
- fn.call({1, 2, 3}, params, context);
-
- EXPECT_EQ(outputs[0], 0);
- EXPECT_EQ(outputs[1], 12);
- EXPECT_EQ(outputs[2], 10);
- EXPECT_EQ(outputs[3], 13);
-}
-
-TEST(multi_function, CustomMF_SM)
-{
- CustomMF_SM<std::string> fn("AddSuffix", [](std::string &value) { value += " test"; });
-
- Array<std::string> values = {"a", "b", "c", "d", "e"};
-
- MFParamsBuilder params(fn, values.size());
- params.add_single_mutable(values.as_mutable_span());
-
- MFContextBuilder context;
-
- fn.call({1, 2, 3}, params, context);
-
- EXPECT_EQ(values[0], "a");
- EXPECT_EQ(values[1], "b test");
- EXPECT_EQ(values[2], "c test");
- EXPECT_EQ(values[3], "d test");
- EXPECT_EQ(values[4], "e");
-}
-
-TEST(multi_function, CustomMF_Constant)
-{
- CustomMF_Constant<int> fn{42};
-
- Array<int> outputs(4, 0);
-
- MFParamsBuilder params(fn, outputs.size());
- params.add_uninitialized_single_output(outputs.as_mutable_span());
-
- MFContextBuilder context;
-
- 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);
-}
-
-TEST(multi_function, CustomMF_GenericConstant)
-{
- int value = 42;
- CustomMF_GenericConstant fn{CPPType::get<int32_t>(), (const void *)&value};
- EXPECT_EQ(fn.param_name(0), "42");
-
- Array<int> outputs(4, 0);
-
- MFParamsBuilder params(fn, outputs.size());
- params.add_uninitialized_single_output(outputs.as_mutable_span());
-
- MFContextBuilder context;
-
- 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))};
- EXPECT_EQ(fn.param_name(0), "[3, 4, 5, 6, ]");
-
- GVectorArray g_vector_array{CPPType::get<int32_t>(), 4};
- GVectorArrayRef<int> vector_array = g_vector_array;
-
- MFParamsBuilder params(fn, g_vector_array.size());
- params.add_vector_output(g_vector_array);
-
- MFContextBuilder context;
-
- 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);
- for (int i = 1; i < 4; i++) {
- EXPECT_EQ(vector_array[i][0], 3);
- EXPECT_EQ(vector_array[i][1], 4);
- EXPECT_EQ(vector_array[i][2], 5);
- EXPECT_EQ(vector_array[i][3], 6);
- }
-}
-
-TEST(multi_function, CustomMF_Convert)
-{
- CustomMF_Convert<float, int> fn;
-
- Array<float> inputs = {5.4f, 7.1f, 9.0f};
- Array<int> outputs(inputs.size(), 0);
-
- MFParamsBuilder params(fn, inputs.size());
- params.add_readonly_single_input(inputs.as_span());
- params.add_uninitialized_single_output(outputs.as_mutable_span());
-
- MFContextBuilder context;
- fn.call({0, 2}, params, context);
-
- EXPECT_EQ(outputs[0], 5);
- EXPECT_EQ(outputs[1], 0);
- EXPECT_EQ(outputs[2], 9);
-}
-
-} // namespace blender::fn
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-11 19:04:37 +0300