egui-treeize 0.1.1

Tree-like graph visualization for egui
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

use std::cell::RefCell;

use egui::Pos2;

use crate::{InPinId, Node, OutPinId, Treeize, wire_pins};

pub enum Effect<T> {
  /// Adds a new node to the Treeize.
  InsertNode { pos: Pos2, node: T },

  /// Removes a node from treeize.
  RemoveNode { node: NodeId },

  /// Opens/closes a node.
  OpenNode { node: NodeId, open: bool },

  /// Adds connection between two nodes.
  Connect { from: OutPinId, to: InPinId },

  /// Removes connection between two nodes.
  Disconnect { from: OutPinId, to: InPinId },

  /// Removes all connections from the output pin.
  DropOutputs { pin: OutPinId },

  /// Removes all connections to the input pin.
  DropInputs { pin: InPinId },

  /// Executes a closure with mutable reference to the Treeize.
  Closure(Box<dyn FnOnce(&mut Treeize<T>)>),
}

/// Contained for deferred execution of effects.
/// It is populated by [`TreeizeViewer`] methods and then applied to the Treeize.
pub struct Effects<T> {
  effects: Vec<Effect<T>>,
}

impl<T> Default for Effects<T> {
  #[inline]
  fn default() -> Self {
    Effects { effects: Default::default() }
  }
}

impl<T> Effects<T> {
  #[inline(always)]
  #[doc(hidden)]
  pub fn new() -> Self {
    Effects { effects: Vec::new() }
  }

  /// Returns `true` if there are no effects.
  /// Returns `false` otherwise.
  #[inline(always)]
  pub fn is_empty(&self) -> bool {
    self.effects.is_empty()
  }

  /// Inserts a new node to the Treeize.
  #[inline(always)]
  pub fn insert_node(&mut self, pos: Pos2, node: T) {
    self.effects.push(Effect::InsertNode { node, pos });
  }

  /// Removes a node from the Treeize.
  #[inline(always)]
  pub fn remove_node(&mut self, node: NodeId) {
    self.effects.push(Effect::RemoveNode { node });
  }

  /// Opens/closes a node.
  #[inline(always)]
  pub fn open_node(&mut self, node: NodeId, open: bool) {
    self.effects.push(Effect::OpenNode { node, open });
  }

  /// Connects two nodes.
  #[inline(always)]
  pub fn connect(&mut self, from: OutPinId, to: InPinId) {
    self.effects.push(Effect::Connect { from, to });
  }

  /// Disconnects two nodes.
  #[inline(always)]
  pub fn disconnect(&mut self, from: OutPinId, to: InPinId) {
    self.effects.push(Effect::Disconnect { from, to });
  }

  /// Removes all connections from the output pin.
  #[inline(always)]
  pub fn drop_inputs(&mut self, pin: InPinId) {
    self.effects.push(Effect::DropInputs { pin });
  }

  /// Removes all connections to the input pin.
  #[inline(always)]
  pub fn drop_outputs(&mut self, pin: OutPinId) {
    self.effects.push(Effect::DropOutputs { pin });
  }
}

impl<T> Treeize<T> {
  pub fn apply_effects(&mut self, effects: Effects<T>) {
    if effects.effects.is_empty() {
      return;
    }
    for effect in effects.effects {
      self.apply_effect(effect);
    }
  }

  pub fn apply_effect(&mut self, effect: Effect<T>) {
    match effect {
      Effect::InsertNode { node, pos } => {
        let idx = self.nodes.insert(Node { value: RefCell::new(node), pos, open: true });
        self.draw_order.push(idx);
      }
      Effect::RemoveNode { node } => {
        if self.nodes.contains(node) {
          self.remove_node(node);
        }
      }
      Effect::OpenNode { node, open } => {
        if self.nodes.contains(node) {
          self.nodes[node].open = open;
        }
      }
      Effect::Connect { from, to } => {
        if self.nodes.contains(from.node) && self.nodes.contains(to.node) {
          self.wires.insert(wire_pins(from, to));
        }
      }
      Effect::Disconnect { from, to } => {
        if self.nodes.contains(from.node) && self.nodes.contains(to.node) {
          self.wires.remove(&wire_pins(from, to));
        }
      }
      Effect::DropOutputs { pin } => {
        if self.nodes.contains(pin.node) {
          self.wires.drop_outputs(pin);
        }
      }
      Effect::DropInputs { pin } => {
        if self.nodes.contains(pin.node) {
          self.wires.drop_inputs(pin);
        }
      }
      Effect::Closure(f) => f(self),
    }
  }
}