BKE_node_tree_zones.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include <iosfwd>
 
#include "DNA_node_types.h"
 
#include "BLI_vector.hh"
 
namespace blender::bke {
 
class bNodeTreeZones;
 
class bNodeTreeZone {
 public:
  bNodeTreeZones *owner = nullptr;
  int index = -1;
  int depth = -1;
  const bNode *input_node = nullptr;
  const bNode *output_node = nullptr;
  bNodeTreeZone *parent_zone = nullptr;
  Vector<bNodeTreeZone *> child_zones;
  Vector<const bNode *> child_nodes;
  Vector<const bNodeLink *> border_links;
 
  bool contains_node_recursively(const bNode &node) const;
  bool contains_zone_recursively(const bNodeTreeZone &other_zone) const;
 
  friend std::ostream &operator<<(std::ostream &stream, const bNodeTreeZone &zone);
};
 
class bNodeTreeZones {
 public:
  Vector<std::unique_ptr<bNodeTreeZone>> zones;
  Vector<bNodeTreeZone *> root_zones;
  Vector<const bNode *> nodes_outside_zones;
  Map<int, int> zone_by_node_id;
 
  const bNodeTreeZone *get_zone_by_socket(const bNodeSocket &socket) const;
 
  const bNodeTreeZone *get_zone_by_node(const int32_t node_id) const;
 
  Vector<const bNodeTreeZone *> get_zone_stack_for_node(const int32_t node_id) const;
 
  friend std::ostream &operator<<(std::ostream &stream, const bNodeTreeZones &zones);
};
 
const bNodeTreeZones *get_tree_zones(const bNodeTree &tree);
 
}  // namespace blender::bke
 
inline const blender::bke::bNodeTreeZones *bNodeTree::zones() const
{
  return blender::bke::get_tree_zones(*this);
}