egui-snarl 0.1.0

Node-graphs 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

use egui::{style::Spacing, Context, Id, Pos2, Rect, Vec2};

/// Node UI state.
#[derive(Clone, Copy, PartialEq)]
pub struct NodeState {
    /// Node size for this frame.
    /// It is updated to fit content.
    pub size: Vec2,
}

impl NodeState {
    pub fn load(cx: &Context, id: Id) -> Option<Self> {
        cx.data_mut(|d| d.get_temp(id))
    }

    pub fn store(&self, cx: &Context, id: Id) {
        cx.data_mut(|d| d.insert_temp(id, *self));
    }

    /// Finds node rect at specific position (excluding node frame margin).
    pub fn node_rect(&self, pos: Pos2) -> Rect {
        Rect::from_min_size(pos, self.size)
    }

    pub fn initial(spacing: &Spacing) -> Self {
        NodeState {
            // title_size: spacing.interact_size,
            // inputs_size: spacing.interact_size,
            // outputs_size: spacing.interact_size,
            size: spacing.interact_size * 3.0,
        }
    }
}

#[derive(Clone, Copy, PartialEq)]
pub struct SnarlState {
    /// Where viewport's left-top in graph's space.
    offset: Vec2,

    /// Scale of the viewport.
    scale: f32,

    target_scale: f32,
}

impl Default for SnarlState {
    fn default() -> Self {
        SnarlState {
            offset: Vec2::ZERO,
            scale: 1.0,
            target_scale: 1.0,
        }
    }
}

impl SnarlState {
    #[inline(always)]
    pub fn load(cx: &Context, id: Id) -> Option<Self> {
        cx.data_mut(|d| d.get_temp(id))
    }

    #[inline(always)]
    pub fn store(&self, cx: &Context, id: Id) {
        cx.data_mut(|d| d.insert_temp(id, *self));
    }

    #[inline(always)]
    pub fn animate(&mut self, id: Id, cx: &Context, pivot: Pos2, viewport: Rect) {
        let new_scale = cx.animate_value_with_time(id.with("zoom-scale"), self.target_scale, 0.1);

        let a = pivot + self.offset - viewport.center().to_vec2();

        self.offset += a * new_scale / self.scale - a;
        self.scale = new_scale;
    }

    #[inline(always)]
    pub fn pan(&mut self, delta: Vec2) {
        self.offset += delta;
    }

    #[inline(always)]
    pub fn scale(&self) -> f32 {
        self.scale
    }

    #[inline(always)]
    pub fn set_scale(&mut self, scale: f32) {
        self.target_scale = scale;
    }

    #[inline(always)]
    pub fn graph_center(&self) -> Pos2 {
        (self.offset / self.scale).to_pos2()
    }

    #[inline(always)]
    pub fn screen_pos_to_graph(&self, pos: Pos2, viewport: Rect) -> Pos2 {
        (pos + self.offset - viewport.center().to_vec2()) / self.scale
    }

    #[inline(always)]
    pub fn graph_pos_to_screen(&self, pos: Pos2, viewport: Rect) -> Pos2 {
        pos * self.scale - self.offset + viewport.center().to_vec2()
    }

    #[inline(always)]
    pub fn graph_vec_to_screen(&self, size: Vec2) -> Vec2 {
        size * self.scale
    }

    #[inline(always)]
    pub fn screen_vec_to_graph(&self, size: Vec2) -> Vec2 {
        size / self.scale
    }

    #[inline(always)]
    pub fn graph_value_to_screen(&self, value: f32) -> f32 {
        value * self.scale
    }

    #[inline(always)]
    pub fn screen_value_to_graph(&self, value: f32) -> f32 {
        value / self.scale
    }
}