a345c4597e
This patch refactors how values of unlinked sockets are provided to nodes. Previously, the GPU node stack values were initialized at construction time and linked by the node's compile methods. This meant that we needed to handle implicit conversion if the socket is linked to an unlinked node group input. Alternatively, we now insert a GPU setter node for each unlinked socket that carries the value and type of the origin socket, and let the GPU code generator do implicit conversion at the shader level. This has three advantages: - It makes it easier to add new types since we no longer have to handle those types in shader node code and it reduces code duplication. - It makes the code more inline with how we implement multi-function procedures. So refactoring is easier. - It opens the door to implement things like implicit inputs, which will be needed later for things like texture nodes.
68 lines
2.7 KiB
C++
68 lines
2.7 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. The compiler is expected to initialize the input links of the node inputs
|
|
* 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(StringRef identifier);
|
|
|
|
/* Returns the GPU node stack of the output with the given identifier. */
|
|
GPUNodeStack &get_output(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
|