b80f1f5322
Implementation of #127106. This is just a visual representation of the field/single/grid status of sockets to make the workflow more intuitive. With a visual representation for volume grid sockets, volume features should be unblocked for further development. The structure type will also be used to distinguish list sockets in the interface. Group input nodes now have a "Structure Type" option instead of the existing "Single Value Only". Usually the auto option should be enough, but in some cases where the inferencing cannot (yet) make a clear determination, it can be helpful to choose a specific type. The new visualization and the group input structure type option are hidden behind a new experimental option for now. Pull Request: https://projects.blender.org/blender/blender/pulls/134811
98 lines
3.7 KiB
C++
98 lines
3.7 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include "BLI_compute_context.hh"
|
|
#include "BLI_generic_pointer.hh"
|
|
#include "BLI_resource_scope.hh"
|
|
#include "BLI_vector_set.hh"
|
|
|
|
#include "DNA_node_types.h"
|
|
|
|
#include "BKE_idprop.hh"
|
|
|
|
struct bNodeTree;
|
|
struct bNodeTreeInterfaceSocket;
|
|
namespace blender::bke {
|
|
struct GeometrySet;
|
|
}
|
|
struct IDProperty;
|
|
namespace blender::nodes {
|
|
struct GeoNodesCallData;
|
|
namespace geo_eval_log {
|
|
class GeoNodesLog;
|
|
} // namespace geo_eval_log
|
|
} // namespace blender::nodes
|
|
|
|
namespace blender::nodes {
|
|
|
|
constexpr StringRef input_use_attribute_suffix = "_use_attribute";
|
|
constexpr StringRef input_attribute_name_suffix = "_attribute_name";
|
|
|
|
struct IDPropNameGetter {
|
|
StringRef operator()(const IDProperty *value) const
|
|
{
|
|
return StringRef(value->name);
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Use a #VectorSet to store properties for constant time lookup, to avoid slowdown with many
|
|
* inputs.
|
|
*/
|
|
using PropertiesVectorSet = CustomIDVectorSet<IDProperty *, IDPropNameGetter, 16>;
|
|
PropertiesVectorSet build_properties_vector_set(const IDProperty *properties);
|
|
|
|
std::optional<StringRef> input_attribute_name_get(const PropertiesVectorSet &properties,
|
|
const bNodeTreeInterfaceSocket &io_input);
|
|
|
|
/**
|
|
* \return Whether using an attribute to input values of this type is supported.
|
|
*/
|
|
bool socket_type_has_attribute_toggle(eNodeSocketDatatype type);
|
|
|
|
/**
|
|
* \return Whether using an attribute to input values of this type is supported, and the node
|
|
* group's input for this socket accepts a field rather than just single values.
|
|
*/
|
|
bool input_has_attribute_toggle(const bNodeTree &node_tree, const int socket_index);
|
|
|
|
bool id_property_type_matches_socket(const bNodeTreeInterfaceSocket &socket,
|
|
const IDProperty &property,
|
|
bool use_name_for_ids = false);
|
|
|
|
std::unique_ptr<IDProperty, bke::idprop::IDPropertyDeleter> id_property_create_from_socket(
|
|
const bNodeTreeInterfaceSocket &socket,
|
|
nodes::StructureType structure_type,
|
|
bool use_name_for_ids);
|
|
|
|
bke::GeometrySet execute_geometry_nodes_on_geometry(const bNodeTree &btree,
|
|
const PropertiesVectorSet &properties_set,
|
|
const ComputeContext &base_compute_context,
|
|
GeoNodesCallData &call_data,
|
|
bke::GeometrySet input_geometry);
|
|
|
|
void update_input_properties_from_node_tree(const bNodeTree &tree,
|
|
const IDProperty *old_properties,
|
|
IDProperty &properties,
|
|
bool use_name_for_ids = false);
|
|
|
|
void update_output_properties_from_node_tree(const bNodeTree &tree,
|
|
const IDProperty *old_properties,
|
|
IDProperty &properties);
|
|
|
|
/**
|
|
* Get the "base" input values that are passed into geometry nodes. In this context, "base" means
|
|
* that the retrieved input types are #bNodeSocketType::base_cpp_type (e.g. `float` for float
|
|
* sockets). If the input value can't be represented as base value, null is returned instead (e.g.
|
|
* for attribute inputs).
|
|
*/
|
|
void get_geometry_nodes_input_base_values(const bNodeTree &btree,
|
|
const PropertiesVectorSet &properties,
|
|
ResourceScope &scope,
|
|
MutableSpan<GPointer> r_values);
|
|
|
|
} // namespace blender::nodes
|