amaranth/node_editor/node_stats.py

   1 # SPDX-FileCopyrightText: 2010-2022 Blender Foundation
   2 #
   3 # SPDX-License-Identifier: GPL-2.0-or-later
   4
   5 """
   6 Nodes Stats
   7
   8 Display the number of selected and total nodes on the compositor. On the
   9 Compositing Nodes Editor.
  10 """
  11
  12 import bpy
  13
  14
  15 def node_stats(self, context):
  16     if context.scene.node_tree:
  17         tree_type = context.space_data.tree_type
  18         nodes = context.scene.node_tree.nodes
  19         nodes_total = len(nodes.keys())
  20         nodes_selected = 0
  21         for n in nodes:
  22             if n.select:
  23                 nodes_selected = nodes_selected + 1
  24
  25         if tree_type == 'CompositorNodeTree':
  26             layout = self.layout
  27             row = layout.row(align=True)
  28             row.label(text="Nodes: %s/%s" % (nodes_selected, str(nodes_total)))
  29
  30
  31 def register():
  32     bpy.types.NODE_HT_header.append(node_stats)
  33
  34
  35 def unregister():
  36     bpy.types.NODE_HT_header.remove(node_stats)