Differential D7618 Diff 24385 tests/gtests/blenlib/BLI_task_graph_test.cc

Changeset View

Standalone View

tests/gtests/blenlib/BLI_task_graph_test.cc

This file was added.

				/* Apache License, Version 2.0 */

				#include "testing/testing.h"

				#include "MEM_guardedalloc.h"

				extern "C" {
				#include "BLI_task.h"
				};

				struct TaskData {
				int value;
				int store;
				};

				static void TaskData_increase_value(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->value += 1;
				}
				static void TaskData_decrease_value(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->value -= 1;
				}
				static void TaskData_multiply_by_two_value(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->value *= 2;
				}

				static void TaskData_multiply_by_two_store(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->store *= 2;
				}

				static void TaskData_store_value(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->store = data->value;
				}

				static void TaskData_square_value(void *taskdata)
				{
				TaskData data = (TaskData )taskdata;
				data->value *= data->value;
				}

				/* Sequential Test for using `BLI_task_graph` */
				TEST(task, GraphSequential)
				{
				TaskData data = {0};
				TaskGraph *graph = BLI_task_graph_create();

				/* 0 => 1 */
				TaskNode *node_a = BLI_task_graph_node_create(graph, TaskData_increase_value, &data, NULL);
				/* 1 => 2 */
				TaskNode *node_b = BLI_task_graph_node_create(
				graph, TaskData_multiply_by_two_value, &data, NULL);
				/* 2 => 1 */
				TaskNode *node_c = BLI_task_graph_node_create(graph, TaskData_decrease_value, &data, NULL);
				/* 2 => 1 */
				TaskNode *node_d = BLI_task_graph_node_create(graph, TaskData_square_value, &data, NULL);
				/* 1 => 1 */
				TaskNode *node_e = BLI_task_graph_node_create(graph, TaskData_increase_value, &data, NULL);
				/* 1 => 2 */
				const int answer = 2;

				BLI_task_graph_edge_create(node_a, node_b);
				BLI_task_graph_edge_create(node_b, node_c);
				BLI_task_graph_edge_create(node_c, node_d);
				BLI_task_graph_edge_create(node_d, node_e);

				EXPECT_TRUE(BLI_task_graph_node_push_work(node_a));
				BLI_task_graph_work_and_wait(graph);

				EXPECT_EQ(answer, data.value);
				BLI_task_graph_free(graph);
				}

				TEST(task, GraphSplit)
				{
				TaskData data = {1};

				TaskGraph *graph = BLI_task_graph_create();
				TaskNode *node_a = BLI_task_graph_node_create(graph, TaskData_increase_value, &data, NULL);
				TaskNode *node_b = BLI_task_graph_node_create(graph, TaskData_store_value, &data, NULL);
				TaskNode *node_c = BLI_task_graph_node_create(graph, TaskData_increase_value, &data, NULL);
				TaskNode *node_d = BLI_task_graph_node_create(
				graph, TaskData_multiply_by_two_store, &data, NULL);
				BLI_task_graph_edge_create(node_a, node_b);
				BLI_task_graph_edge_create(node_b, node_c);
				BLI_task_graph_edge_create(node_b, node_d);
				EXPECT_TRUE(BLI_task_graph_node_push_work(node_a));
				BLI_task_graph_work_and_wait(graph);

				EXPECT_EQ(3, data.value);
				EXPECT_EQ(4, data.store);
				BLI_task_graph_free(graph);
				}