2323bd2691
This extracts the construction of the `SocketValueInferencer` out of `SocketUsageInferencer`. This leads to better separation of concerns and gives the caller more flexibility. In the future, I especially want to get information about which group input values were required to determine the usage of other group inputs. This might help with caching the inferenced values. Pull Request: https://projects.blender.org/blender/blender/pulls/147352
109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
/* SPDX-FileCopyrightText: 2025 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include "BLI_generic_pointer.hh"
|
|
#include "BLI_resource_scope.hh"
|
|
|
|
#include "BKE_compute_context_cache_fwd.hh"
|
|
|
|
#include "DNA_material_types.h"
|
|
#include "NOD_node_in_compute_context.hh"
|
|
|
|
struct bNodeTree;
|
|
|
|
namespace blender::nodes {
|
|
|
|
/**
|
|
* During socket usage inferencing, some socket values are computed. This class represents such a
|
|
* computed value. Not all possible values can be presented here, only "basic" once (like int, but
|
|
* not int-field). A value can also be unknown if it can't be determined statically.
|
|
*/
|
|
class InferenceValue {
|
|
private:
|
|
/**
|
|
* Non-owning pointer to a value of type #bNodeSocketType.base_cpp_type of the corresponding
|
|
* socket. If this is null, the value is assumed to be unknown (aka, it can't be determined
|
|
* statically).
|
|
*/
|
|
const void *value_ = nullptr;
|
|
|
|
InferenceValue(const void *value) : value_(value) {}
|
|
|
|
public:
|
|
static InferenceValue from_primitive(const void *value)
|
|
{
|
|
BLI_assert(value != nullptr);
|
|
return InferenceValue(value);
|
|
}
|
|
|
|
static InferenceValue Unknown()
|
|
{
|
|
return InferenceValue(nullptr);
|
|
}
|
|
|
|
bool is_unknown() const
|
|
{
|
|
return value_ == nullptr;
|
|
}
|
|
|
|
bool is_primitive_value() const
|
|
{
|
|
return !this->is_unknown();
|
|
}
|
|
|
|
const void *get_primitive_ptr() const
|
|
{
|
|
BLI_assert(this->is_primitive_value());
|
|
return value_;
|
|
}
|
|
|
|
template<typename T> T get_primitive() const
|
|
{
|
|
BLI_assert(this->is_primitive_value());
|
|
return *static_cast<const T *>(this->value_);
|
|
}
|
|
|
|
template<typename T> std::optional<T> get_if_primitive() const
|
|
{
|
|
if (!this->is_primitive_value()) {
|
|
return std::nullopt;
|
|
}
|
|
return this->get_primitive<T>();
|
|
}
|
|
};
|
|
|
|
class SocketValueInferencerImpl;
|
|
|
|
class SocketValueInferencer {
|
|
private:
|
|
SocketValueInferencerImpl &impl_;
|
|
|
|
public:
|
|
SocketValueInferencer(
|
|
const bNodeTree &tree,
|
|
ResourceScope &scope,
|
|
bke::ComputeContextCache &compute_context_cache,
|
|
FunctionRef<InferenceValue(int group_input_i)> group_input_value_fn = nullptr,
|
|
std::optional<Span<bool>> top_level_ignored_inputs = std::nullopt);
|
|
|
|
InferenceValue get_socket_value(const SocketInContext &socket);
|
|
};
|
|
|
|
namespace switch_node_inference_utils {
|
|
|
|
bool is_socket_selected__switch(const SocketInContext &socket, const InferenceValue &condition);
|
|
bool is_socket_selected__index_switch(const SocketInContext &socket,
|
|
const InferenceValue &condition);
|
|
bool is_socket_selected__menu_switch(const SocketInContext &socket,
|
|
const InferenceValue &condition);
|
|
bool is_socket_selected__mix_node(const SocketInContext &socket, const InferenceValue &condition);
|
|
bool is_socket_selected__shader_mix_node(const SocketInContext &socket,
|
|
const InferenceValue &condition);
|
|
|
|
} // namespace switch_node_inference_utils
|
|
|
|
} // namespace blender::nodes
|