BKE_node_tree_anonymous_attributes.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include <variant>
 
#include "DNA_node_types.h"
 
#include "BLI_array.hh"
#include "BLI_bit_group_vector.hh"
#include "BLI_resource_scope.hh"
#include "BLI_vector.hh"
#include "BLI_vector_set.hh"
 
#include "NOD_node_declaration.hh"
 
namespace blender::bke::anonymous_attribute_inferencing {
 
struct InputFieldSource {
  int input_index;
};
 
struct SocketFieldSource {
  const bNodeSocket *socket;
};
 
struct FieldSource {
  std::variant<InputFieldSource, SocketFieldSource> data;
  Vector<int> geometry_sources;
};
 
struct InputGeometrySource {
  int input_index;
};
 
struct SocketGeometrySource {
  const bNodeSocket *socket;
};
 
struct GeometrySource {
  std::variant<InputGeometrySource, SocketGeometrySource> data;
  Vector<int> field_sources;
};
 
struct AnonymousAttributeInferencingResult {
  Vector<FieldSource> all_field_sources;
  Vector<GeometrySource> all_geometry_sources;
 
  BitGroupVector<> propagated_fields_by_socket;
  BitGroupVector<> propagated_geometries_by_socket;
  BitGroupVector<> available_fields_by_geometry_socket;
  BitGroupVector<> required_fields_by_geometry_socket;
 
  VectorSet<int> propagated_output_geometry_indices;
  BitGroupVector<> propagate_to_output_by_geometry_socket;
 
  nodes::aal::RelationsInNode tree_relations;
};
 
Array<const nodes::aal::RelationsInNode *> get_relations_by_node(const bNodeTree &tree,
                                                                 ResourceScope &scope);
bool update_anonymous_attribute_relations(bNodeTree &tree);
 
}  // namespace blender::bke::anonymous_attribute_inferencing