bd379e1892
This patch refactors the ShaderNode class to be a concrete class that is implemented in terms of the node type gpu_fn. This is done to make it easier to reuse existing nodes in other parts of Blender. Pull Request: https://projects.blender.org/blender/blender/pulls/134210
69 lines
2.8 KiB
C++
69 lines
2.8 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include "BLI_string_ref.hh"
|
|
#include "BLI_vector.hh"
|
|
|
|
#include "DNA_node_types.h"
|
|
|
|
#include "GPU_material.hh"
|
|
|
|
#include "NOD_derived_node_tree.hh"
|
|
|
|
namespace blender::compositor {
|
|
|
|
using namespace nodes::derived_node_tree_types;
|
|
|
|
/* ------------------------------------------------------------------------------------------------
|
|
* Shader Node
|
|
*
|
|
* A shader node encapsulates a compositor node tree that is capable of being used together with
|
|
* other shader nodes to construct a Shader Operation using the GPU material compiler. A GPU node
|
|
* stack for each of the node inputs and outputs is stored and populated during construction in
|
|
* order to represent the node as a GPU node inside the GPU material graph, see GPU_material.hh for
|
|
* more information. Derived classes should implement the compile method to add the node and link
|
|
* it to the GPU material given to the method. The compiler is expected to initialize the input
|
|
* links of the node before invoking the compile method. See the discussion in
|
|
* COM_shader_operation.hh for more information. */
|
|
class ShaderNode {
|
|
private:
|
|
/* The node that this operation represents. */
|
|
DNode node_;
|
|
/* The GPU node stacks of the inputs of the node. Those are populated during construction in the
|
|
* populate_inputs method. The links of the inputs are initialized by the GPU material compiler
|
|
* prior to calling the compile method. There is an extra stack at the end to mark the end of the
|
|
* array, as this is what the GPU module functions expect. */
|
|
Vector<GPUNodeStack> inputs_;
|
|
/* The GPU node stacks of the outputs of the node. Those are populated during construction in the
|
|
* populate_outputs method. There is an extra stack at the end to mark the end of the array, as
|
|
* this is what the GPU module functions expect. */
|
|
Vector<GPUNodeStack> outputs_;
|
|
|
|
public:
|
|
/* Construct the node by populating both its inputs and outputs. */
|
|
ShaderNode(DNode node);
|
|
|
|
/* Compile the node by adding the appropriate GPU material graph nodes and linking the
|
|
* appropriate resources. */
|
|
void compile(GPUMaterial *material);
|
|
|
|
/* Returns the GPU node stack of the input with the given identifier. */
|
|
GPUNodeStack &get_input(const StringRef identifier);
|
|
|
|
/* Returns the GPU node stack of the output with the given identifier. */
|
|
GPUNodeStack &get_output(const StringRef identifier);
|
|
|
|
private:
|
|
/* Populate the inputs of the node. The input link is set to nullptr and is expected to be
|
|
* initialized by the GPU material compiler before calling the compile method. */
|
|
void populate_inputs();
|
|
/* Populate the outputs of the node. The output link is set to nullptr and is expected to be
|
|
* initialized by the compile method. */
|
|
void populate_outputs();
|
|
};
|
|
|
|
} // namespace blender::compositor
|