Merge branch 'development' of
[Mograsim.git] / LogicUI / src / era / mi / gui / LogicUICanvas.java
1 package era.mi.gui;
2
3 import java.util.function.Consumer;
4
5 import org.eclipse.swt.SWT;
6 import org.eclipse.swt.widgets.Composite;
7 import org.eclipse.swt.widgets.Event;
8
9 import era.mi.gui.model.ViewModel;
10 import era.mi.gui.model.components.GUIComponent;
11 import era.mi.gui.model.wires.GUIWire;
12 import era.mi.gui.model.wires.Pin;
13 import net.haspamelodica.swt.helper.gcs.GeneralGC;
14 import net.haspamelodica.swt.helper.swtobjectwrappers.Point;
15 import net.haspamelodica.swt.helper.swtobjectwrappers.Rectangle;
16 import net.haspamelodica.swt.helper.zoomablecanvas.ZoomableCanvas;
17
18 /**
19  * Simulation visualizer canvas.
20  * 
21  * @author Daniel Kirschten
22  */
23 public class LogicUICanvas extends ZoomableCanvas
24 {
25         private final ViewModel model;
26
27         public LogicUICanvas(Composite parent, int style, ViewModel model)
28         {
29                 super(parent, style);
30
31                 this.model = model;
32
33                 Consumer<Object> redrawConsumer = o -> redrawThreadsafe();
34                 Consumer<Pin> pinAddedListener = p ->
35                 {
36                         p.addPinMovedListener(redrawConsumer);
37                         redrawThreadsafe();
38                 };
39                 Consumer<Pin> pinRemovedListener = p ->
40                 {
41                         p.removePinMovedListener(redrawConsumer);
42                         redrawThreadsafe();
43                 };
44                 Consumer<? super GUIComponent> componentAddedListener = c ->
45                 {
46                         c.addComponentLookChangedListener(redrawConsumer);
47                         c.addComponentMovedListener(redrawConsumer);
48                         c.addPinAddedListener(pinAddedListener);
49                         c.addPinRemovedListener(pinRemovedListener);
50                         redrawThreadsafe();
51                 };
52                 model.addComponentAddedListener(componentAddedListener);
53                 model.getComponents().forEach(componentAddedListener);
54                 model.addComponentRemovedListener(c ->
55                 {
56                         c.removeComponentLookChangedListener(redrawConsumer);
57                         c.removeComponentMovedListener(redrawConsumer);
58                         c.removePinAddedListener(pinAddedListener);
59                         c.removePinRemovedListener(pinRemovedListener);
60                         redrawThreadsafe();
61                 });
62                 Consumer<? super GUIWire> wireAddedListener = w ->
63                 {
64                         w.addWireLookChangedListener(redrawConsumer);
65                         redrawThreadsafe();
66                 };
67                 model.addWireAddedListener(wireAddedListener);
68                 model.getWires().forEach(wireAddedListener);
69                 model.addWireRemovedListener(w ->
70                 {
71                         w.removeWireLookChangedListener(redrawConsumer);
72                         redrawThreadsafe();
73                 });
74
75                 addZoomedRenderer(gc ->
76                 {
77                         Rectangle visibleRegion = new Rectangle(offX, offY, gW / zoom, gH / zoom);
78                         model.getComponents().forEach(c -> drawComponent(gc, c, visibleRegion));
79                 });
80                 addZoomedRenderer(gc -> model.getWires().forEach(w -> w.render(gc)));
81                 addListener(SWT.MouseDown, this::mouseDown);
82         }
83
84         private void drawComponent(GeneralGC gc, GUIComponent component, Rectangle visibleRegion)
85         {
86                 component.render(gc, visibleRegion);
87                 gc.setBackground(getDisplay().getSystemColor(SWT.COLOR_CYAN));
88                 for (Pin p : component.getPins())
89                 {
90                         Point pos = p.getPos();
91                         gc.fillOval(pos.x - 1, pos.y - 1, 2, 2);
92                 }
93         }
94
95         private void mouseDown(Event e)
96         {
97                 if (e.button == 1)
98                 {
99                         Point click = displayToWorldCoords(e.x, e.y);
100                         for (GUIComponent component : model.getComponents())
101                                 if (component.getBounds().contains(click) && component.clicked(click.x, click.y))
102                                 {
103                                         redraw();
104                                         break;
105                                 }
106                 }
107         }
108 }