| author | lbessard |
| Mon, 09 Jul 2007 11:10:14 +0200 | |
| changeset 27 | dae55dd9ee14 |
| parent 13 | 69075340d6a9 |
| child 28 | fc23e1f415d8 |
| permissions | -rw-r--r-- |
| 0 | 1 |
#!/usr/bin/env python |
2 |
# -*- coding: utf-8 -*- |
|
3 |
||
4 |
#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor |
|
5 |
#based on the plcopen standard. |
|
6 |
# |
|
7 |
#Copyright (C): Edouard TISSERANT and Laurent BESSARD |
|
8 |
# |
|
9 |
#See COPYING file for copyrights details. |
|
10 |
# |
|
11 |
#This library is free software; you can redistribute it and/or |
|
| 5 | 12 |
#modify it under the terms of the GNU General Public |
| 0 | 13 |
#License as published by the Free Software Foundation; either |
14 |
#version 2.1 of the License, or (at your option) any later version. |
|
15 |
# |
|
16 |
#This library is distributed in the hope that it will be useful, |
|
17 |
#but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
18 |
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
19 |
#Lesser General Public License for more details. |
|
20 |
# |
|
| 5 | 21 |
#You should have received a copy of the GNU General Public |
| 0 | 22 |
#License along with this library; if not, write to the Free Software |
23 |
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
|
24 |
||
25 |
from wxPython.wx import * |
|
26 |
import wx |
|
27 |
||
28 |
from plcopen.structures import * |
|
29 |
from graphics.GraphicCommons import * |
|
30 |
from graphics.FBD_Objects import * |
|
31 |
from graphics.LD_Objects import * |
|
| 27 | 32 |
from graphics.SFC_Objects import * |
33 |
||
34 |
from Dialogs import * |
|
35 |
||
36 |
SCROLLBAR_UNIT = 10 |
|
37 |
WINDOW_BORDER = 10 |
|
38 |
SCROLL_ZONE = 10 |
|
| 0 | 39 |
|
40 |
#------------------------------------------------------------------------------- |
|
41 |
# Graphic elements Viewer base class |
|
42 |
#------------------------------------------------------------------------------- |
|
43 |
||
44 |
# ID Constants for menu items |
|
45 |
[wxID_FBDVIEWERCONTEXTUALMENUITEMS0, wxID_FBDVIEWERCONTEXTUALMENUITEMS1, |
|
46 |
wxID_FBDVIEWERCONTEXTUALMENUITEMS2, wxID_FBDVIEWERCONTEXTUALMENUITEMS3, |
|
47 |
wxID_FBDVIEWERCONTEXTUALMENUITEMS5, wxID_FBDVIEWERCONTEXTUALMENUITEMS6, |
|
48 |
wxID_FBDVIEWERCONTEXTUALMENUITEMS8, wxID_FBDVIEWERCONTEXTUALMENUITEMS9, |
|
49 |
wxID_FBDVIEWERCONTEXTUALMENUITEMS11, |
|
50 |
] = [wx.NewId() for _init_coll_ContextualMenu_Items in range(9)] |
|
51 |
||
52 |
||
53 |
""" |
|
54 |
Class that implements a Viewer based on a wxScrolledWindow for drawing and |
|
55 |
manipulating graphic elements |
|
56 |
""" |
|
57 |
||
58 |
class Viewer(wx.ScrolledWindow): |
|
59 |
||
60 |
# Create Contextual Menu items |
|
61 |
def _init_coll_ContextualMenu_Items(self, parent): |
|
62 |
# Create menu items |
|
63 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS0, |
|
64 |
kind=wx.ITEM_RADIO, text=u'No Modifier') |
|
65 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS1, |
|
66 |
kind=wx.ITEM_RADIO, text=u'Negated') |
|
67 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS2, |
|
68 |
kind=wx.ITEM_RADIO, text=u'Rising Edge') |
|
69 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS3, |
|
70 |
kind=wx.ITEM_RADIO, text=u'Falling Edge') |
|
71 |
parent.AppendSeparator() |
|
72 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS5, |
|
73 |
kind=wx.ITEM_NORMAL, text=u'Add Wire Segment') |
|
74 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS6, |
|
75 |
kind=wx.ITEM_NORMAL, text=u'Delete Wire Segment') |
|
76 |
parent.AppendSeparator() |
|
77 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS8, |
|
78 |
kind=wx.ITEM_NORMAL, text=u'Add Divergence Branch') |
|
79 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS9, |
|
80 |
kind=wx.ITEM_NORMAL, text=u'Delete Divergence Branch') |
|
81 |
parent.AppendSeparator() |
|
82 |
parent.Append(help='', id=wxID_FBDVIEWERCONTEXTUALMENUITEMS11, |
|
83 |
kind=wx.ITEM_NORMAL, text=u'Delete') |
|
84 |
# Link menu event to corresponding called functions |
|
85 |
self.Bind(wx.EVT_MENU, self.OnNoModifierMenu, |
|
86 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS0) |
|
87 |
self.Bind(wx.EVT_MENU, self.OnNegatedMenu, |
|
88 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS1) |
|
89 |
self.Bind(wx.EVT_MENU, self.OnRisingEdgeMenu, |
|
90 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS2) |
|
91 |
self.Bind(wx.EVT_MENU, self.OnFallingEdgeMenu, |
|
92 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS3) |
|
93 |
self.Bind(wx.EVT_MENU, self.OnAddSegmentMenu, |
|
94 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS5) |
|
95 |
self.Bind(wx.EVT_MENU, self.OnDeleteSegmentMenu, |
|
96 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS6) |
|
97 |
self.Bind(wx.EVT_MENU, self.OnAddBranchMenu, |
|
98 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS8) |
|
99 |
self.Bind(wx.EVT_MENU, self.OnDeleteBranchMenu, |
|
100 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS9) |
|
101 |
self.Bind(wx.EVT_MENU, self.OnDeleteMenu, |
|
102 |
id=wxID_FBDVIEWERCONTEXTUALMENUITEMS11) |
|
103 |
||
104 |
# Create and initialize Contextual Menu |
|
105 |
def _init_menus(self): |
|
106 |
self.ContextualMenu = wx.Menu(title='') |
|
107 |
||
108 |
self._init_coll_ContextualMenu_Items(self.ContextualMenu) |
|
109 |
||
110 |
# Create a new Viewer |
|
111 |
def __init__(self, parent, window, controler): |
|
| 27 | 112 |
wx.ScrolledWindow.__init__(self, parent, style=wx.SUNKEN_BORDER | wx.HSCROLL | wx.VSCROLL) |
| 0 | 113 |
self._init_menus() |
114 |
# Adding a rubberband to Viewer |
|
115 |
self.rubberBand = RubberBand(drawingSurface=self) |
|
116 |
self.SetBackgroundColour(wxColour(255,255,255)) |
|
117 |
self.ResetView() |
|
118 |
self.Scaling = None |
|
119 |
#self.Scaling = (8, 8) |
|
120 |
self.DrawGrid = True |
|
121 |
self.current_id = 0 |
|
122 |
||
123 |
# Initialize Viewer mode to Selection mode |
|
124 |
self.Mode = MODE_SELECTION |
|
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
125 |
self.SavedMode = False |
| 0 | 126 |
|
127 |
self.Parent = window |
|
128 |
self.Controler = controler |
|
129 |
||
130 |
# Link Viewer event to corresponding methods |
|
131 |
EVT_PAINT(self, self.OnPaint) |
|
132 |
EVT_LEFT_DOWN(self, self.OnViewerLeftDown) |
|
133 |
EVT_LEFT_UP(self, self.OnViewerLeftUp) |
|
134 |
EVT_LEFT_DCLICK(self, self.OnViewerLeftDClick) |
|
135 |
EVT_RIGHT_UP(self, self.OnViewerRightUp) |
|
136 |
EVT_MOTION(self, self.OnViewerMotion) |
|
137 |
EVT_CHAR(self, self.OnChar) |
|
| 27 | 138 |
EVT_SCROLLWIN(self, self.OnMoveWindow) |
139 |
EVT_SIZE(self, self.OnMoveWindow) |
|
| 0 | 140 |
|
141 |
# Returns a new id |
|
142 |
def GetNewId(self): |
|
143 |
self.current_id += 1 |
|
144 |
return self.current_id |
|
145 |
||
146 |
# Destructor |
|
147 |
def __del__(self): |
|
148 |
self.ResetView() |
|
149 |
||
| 27 | 150 |
def GetLogicalDC(self): |
151 |
dc = wxClientDC(self) |
|
152 |
self.DoPrepareDC(dc) |
|
153 |
return dc |
|
154 |
||
| 0 | 155 |
#------------------------------------------------------------------------------- |
156 |
# Reset functions |
|
157 |
#------------------------------------------------------------------------------- |
|
158 |
||
159 |
# Resets Viewer lists |
|
160 |
def ResetView(self): |
|
161 |
self.Blocks = [] |
|
162 |
self.Wires = [] |
|
163 |
self.Elements = [] |
|
164 |
self.SelectedElement = None |
|
165 |
||
166 |
# Changes Viewer mode |
|
167 |
def SetMode(self, mode): |
|
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
168 |
if self.Mode != mode or mode == MODE_SELECTION: |
|
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
169 |
self.Mode = mode |
|
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
170 |
self.SavedMode = False |
|
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
171 |
else: |
|
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
172 |
self.SavedMode = True |
| 0 | 173 |
# Reset selection |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
174 |
if self.Mode != MODE_SELECTION and self.SelectedElement: |
| 0 | 175 |
self.SelectedElement.SetSelected(False) |
176 |
self.SelectedElement = None |
|
177 |
self.Refresh() |
|
178 |
||
| 27 | 179 |
# Return current drawing mode |
180 |
def GetDrawingMode(self): |
|
181 |
return self.Parent.GetDrawingMode() |
|
182 |
||
| 0 | 183 |
#------------------------------------------------------------------------------- |
184 |
# Refresh functions |
|
185 |
#------------------------------------------------------------------------------- |
|
186 |
||
187 |
# Refresh Viewer elements |
|
188 |
def RefreshView(self): |
|
189 |
self.current_id = 0 |
|
190 |
# Start by reseting Viewer |
|
191 |
self.ResetView() |
|
192 |
instance = True |
|
193 |
# List of ids of already loaded blocks |
|
194 |
ids = [] |
|
195 |
# Load Blocks until they are all loaded |
|
196 |
while instance: |
|
197 |
instance = self.Controler.GetCurrentElementEditingInstanceInfos(exclude=ids) |
|
198 |
if instance: |
|
199 |
self.loadInstance(instance, ids) |
|
| 27 | 200 |
self.RefreshScrollBar() |
| 0 | 201 |
self.Refresh() |
202 |
||
| 27 | 203 |
def RefreshScrollBar(self): |
204 |
xstart, ystart = self.GetViewStart() |
|
205 |
window_size = self.GetClientSize() |
|
206 |
maxx = maxy = 0 |
|
207 |
for element in self.Elements: |
|
208 |
posx, posy = element.GetPosition() |
|
209 |
width, height = element.GetSize() |
|
210 |
maxx = max(maxx, posx + width) |
|
211 |
maxy = max(maxy, posy + height) |
|
212 |
maxx = max(maxx + WINDOW_BORDER, xstart * SCROLLBAR_UNIT + window_size[0]) |
|
213 |
maxy = max(maxy + WINDOW_BORDER, ystart * SCROLLBAR_UNIT + window_size[1]) |
|
214 |
if self.rubberBand.IsShown(): |
|
215 |
extent = self.rubberBand.GetCurrentExtent() |
|
216 |
maxx = max(maxx, extent.x + extent.width) |
|
217 |
maxy = max(maxy, extent.y + extent.height) |
|
218 |
self.SetScrollbars(SCROLLBAR_UNIT, SCROLLBAR_UNIT, |
|
219 |
maxx / SCROLLBAR_UNIT, maxy / SCROLLBAR_UNIT, xstart, ystart) |
|
220 |
||
| 0 | 221 |
# Load instance from given informations |
222 |
def loadInstance(self, instance, ids): |
|
223 |
ids.append(instance["id"]) |
|
224 |
self.current_id = max(self.current_id, instance["id"]) |
|
225 |
if instance["type"] == "input": |
|
226 |
variable = FBD_Variable(self, INPUT, instance["name"], instance["value_type"], instance["id"]) |
|
227 |
variable.SetPosition(instance["x"], instance["y"]) |
|
228 |
variable.SetSize(instance["width"], instance["height"]) |
|
229 |
self.Blocks.append(variable) |
|
230 |
self.Elements.append(variable) |
|
231 |
connectors = variable.GetConnectors() |
|
232 |
connectors["output"].SetPosition(wxPoint(*instance["connector"]["position"])) |
|
233 |
if instance["connector"]["negated"]: |
|
234 |
connectors["output"].SetNegated(True) |
|
235 |
if instance["connector"]["edge"]: |
|
236 |
connectors["output"].SetEdge(instance["connector"]["edge"]) |
|
237 |
elif instance["type"] == "output": |
|
238 |
variable = FBD_Variable(self, OUTPUT, instance["name"], instance["value_type"], instance["id"]) |
|
239 |
variable.SetPosition(instance["x"], instance["y"]) |
|
240 |
variable.SetSize(instance["width"], instance["height"]) |
|
241 |
self.Blocks.append(variable) |
|
242 |
self.Elements.append(variable) |
|
243 |
connectors = variable.GetConnectors() |
|
244 |
connectors["input"].SetPosition(wxPoint(*instance["connector"]["position"])) |
|
245 |
if instance["connector"]["negated"]: |
|
246 |
connectors["input"].SetNegated(True) |
|
247 |
if instance["connector"]["edge"]: |
|
248 |
connectors["input"].SetEdge(instance["connector"]["edge"]) |
|
249 |
self.CreateWires(connectors["input"], instance["connector"]["links"], ids) |
|
250 |
elif instance["type"] == "inout": |
|
251 |
variable = FBD_Variable(self, INOUT, instance["name"], instance["value_type"], instance["id"]) |
|
252 |
variable.SetPosition(instance["x"], instance["y"]) |
|
253 |
variable.SetSize(instance["width"], instance["height"]) |
|
254 |
self.Blocks.append(variable) |
|
255 |
self.Elements.append(variable) |
|
256 |
connectors = variable.GetConnectors() |
|
257 |
connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) |
|
258 |
connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) |
|
259 |
if instance["connectors"]["output"]["negated"]: |
|
260 |
connectors["output"].SetNegated(True) |
|
261 |
if instance["connectors"]["output"]["edge"]: |
|
262 |
connectors["output"].SetEdge(instance["connectors"]["output"]["edge"]) |
|
263 |
if instance["connectors"]["input"]["negated"]: |
|
264 |
connectors["input"].SetNegated(True) |
|
265 |
if instance["connectors"]["input"]["edge"]: |
|
266 |
connectors["input"].SetEdge(instance["connectors"]["input"]["edge"]) |
|
267 |
self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) |
|
268 |
elif instance["type"] == "continuation": |
|
269 |
connection = FBD_Connector(self, CONTINUATION, instance["name"], instance["id"]) |
|
270 |
connection.SetPosition(instance["x"], instance["y"]) |
|
271 |
connection.SetSize(instance["width"], instance["height"]) |
|
272 |
self.Blocks.append(connection) |
|
273 |
self.Elements.append(connection) |
|
274 |
connector = connection.GetConnector() |
|
275 |
connector.SetPosition(wxPoint(*instance["connector"]["position"])) |
|
276 |
elif instance["type"] == "connection": |
|
277 |
connection = FBD_Connection(self, CONNECTOR, instance["name"], instance["id"]) |
|
278 |
connection.SetPosition(instance["x"], instance["y"]) |
|
279 |
connection.SetSize(instance["width"], instance["height"]) |
|
280 |
self.Blocks.append(connection) |
|
281 |
self.Elements.append(connection) |
|
282 |
connector = connection.GetConnector() |
|
283 |
connector.SetPosition(wxPoint(*instance["connector"]["position"])) |
|
284 |
self.CreateWires(connector, instance["connector"]["links"], ids) |
|
285 |
elif instance["type"] == "comment": |
|
286 |
comment = Comment(self, instance["content"], instance["id"]) |
|
287 |
comment.SetPosition(instance["x"], instance["y"]) |
|
288 |
comment.SetSize(instance["width"], instance["height"]) |
|
289 |
self.Elements.append(comment) |
|
290 |
elif instance["type"] == "leftPowerRail": |
|
291 |
leftpowerrail = LD_PowerRail(self, LEFTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) |
|
292 |
leftpowerrail.SetPosition(instance["x"], instance["y"]) |
|
293 |
self.Blocks.append(leftpowerrail) |
|
294 |
self.Elements.append(leftpowerrail) |
|
295 |
connectors = leftpowerrail.GetConnectors() |
|
296 |
for i, connector in enumerate(instance["connectors"]): |
|
297 |
connectors[i].SetPosition(wxPoint(*connector["position"])) |
|
298 |
elif instance["type"] == "rightPowerRail": |
|
299 |
rightpowerrail = LD_PowerRail(self, RIGHTRAIL, instance["id"], [True for i in range(len(instance["connectors"]))]) |
|
300 |
rightpowerrail.SetPosition(instance["x"], instance["y"]) |
|
301 |
self.Blocks.append(rightpowerrail) |
|
302 |
self.Elements.append(rightpowerrail) |
|
303 |
connectors = rightpowerrail.GetConnectors() |
|
304 |
for i, connector in enumerate(instance["connectors"]): |
|
305 |
connectors[i].SetPosition(wxPoint(*connector["position"])) |
|
306 |
self.CreateWires(connectors[i], connector["links"], ids) |
|
307 |
elif instance["type"] == "contact": |
|
308 |
if instance["negated"]: |
|
309 |
negated = instance["negated"] |
|
310 |
else: |
|
311 |
negated = False |
|
312 |
if instance["edge"]: |
|
313 |
edge = instance["edge"] |
|
314 |
else: |
|
315 |
edge = "none" |
|
316 |
if negated and edge == "none": |
|
317 |
contact_type = CONTACT_REVERSE |
|
318 |
elif not negated and edge == "rising": |
|
319 |
contact_type = CONTACT_RISING |
|
320 |
elif not negated and edge == "falling": |
|
321 |
contact_type = CONTACT_FALLING |
|
322 |
else: |
|
323 |
contact_type = CONTACT_NORMAL |
|
324 |
contact = LD_Contact(self, contact_type, instance["name"], instance["id"]) |
|
325 |
contact.SetPosition(instance["x"], instance["y"]) |
|
326 |
self.Blocks.append(contact) |
|
327 |
self.Elements.append(contact) |
|
328 |
connectors = contact.GetConnectors() |
|
329 |
connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) |
|
330 |
self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) |
|
331 |
connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) |
|
332 |
elif instance["type"] == "coil": |
|
333 |
if instance["negated"]: |
|
334 |
negated = instance["negated"] |
|
335 |
else: |
|
336 |
negated = False |
|
337 |
if instance["storage"]: |
|
338 |
storage = instance["storage"] |
|
339 |
else: |
|
340 |
storage = "none" |
|
341 |
if negated and storage == "none": |
|
342 |
coil_type = COIL_REVERSE |
|
343 |
elif not negated and storage == "set": |
|
344 |
coil_type = COIL_SET |
|
345 |
elif not negated and storage == "reset": |
|
346 |
coil_type = COIL_RESET |
|
347 |
else: |
|
348 |
coil_type = COIL_NORMAL |
|
349 |
coil = LD_Coil(self, coil_type, instance["name"], instance["id"]) |
|
350 |
coil.SetPosition(instance["x"], instance["y"]) |
|
351 |
self.Blocks.append(coil) |
|
352 |
self.Elements.append(coil) |
|
353 |
connectors = coil.GetConnectors() |
|
354 |
connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) |
|
355 |
self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) |
|
356 |
connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) |
|
357 |
elif instance["type"] == "step": |
|
358 |
if instance["initial"]: |
|
359 |
initial = instance["initial"] |
|
360 |
else: |
|
361 |
initial = False |
|
362 |
step = SFC_Step(self, instance["name"], initial, instance["id"]) |
|
363 |
step.SetPosition(instance["x"], instance["y"]) |
|
364 |
step.SetSize(instance["width"], instance["height"]) |
|
365 |
self.Blocks.append(step) |
|
366 |
self.Elements.append(step) |
|
367 |
if "output" in instance["connectors"]: |
|
368 |
step.AddOutput() |
|
369 |
if "action" in instance["connectors"]: |
|
370 |
step.AddAction() |
|
371 |
connectors = step.GetConnectors() |
|
372 |
if connectors["input"]: |
|
373 |
connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) |
|
374 |
self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) |
|
375 |
if connectors["output"]: |
|
376 |
connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) |
|
377 |
if connectors["action"]: |
|
378 |
connectors["action"].SetPosition(wxPoint(*instance["connectors"]["action"]["position"])) |
|
379 |
elif instance["type"] == "transition": |
|
380 |
transition = SFC_Transition(self, instance["condition_type"], instance["condition"], instance["id"]) |
|
381 |
transition.SetPosition(instance["x"], instance["y"]) |
|
382 |
self.Blocks.append(transition) |
|
383 |
self.Elements.append(transition) |
|
384 |
connectors = transition.GetConnectors() |
|
385 |
connectors["input"].SetPosition(wxPoint(*instance["connectors"]["input"]["position"])) |
|
386 |
self.CreateWires(connectors["input"], instance["connectors"]["input"]["links"], ids) |
|
387 |
connectors["output"].SetPosition(wxPoint(*instance["connectors"]["output"]["position"])) |
|
388 |
elif instance["type"] in ["selectionDivergence", "selectionConvergence", "simultaneousDivergence", "simultaneousConvergence"]: |
|
389 |
if instance["type"] == "selectionDivergence": |
|
390 |
divergence = SFC_Divergence(self, SELECTION_DIVERGENCE, |
|
391 |
len(instance["connectors"]["outputs"]), instance["id"]) |
|
392 |
elif instance["type"] == "selectionConvergence": |
|
393 |
divergence = SFC_Divergence(self, SELECTION_CONVERGENCE, |
|
394 |
len(instance["connectors"]["inputs"]), instance["id"]) |
|
395 |
elif instance["type"] == "simultaneousDivergence": |
|
396 |
divergence = SFC_Divergence(self, SIMULTANEOUS_DIVERGENCE, |
|
397 |
len(instance["connectors"]["outputs"]), instance["id"]) |
|
398 |
else: |
|
399 |
divergence = SFC_Divergence(self, SIMULTANEOUS_CONVERGENCE, |
|
400 |
len(instance["connectors"]["inputs"]), instance["id"]) |
|
401 |
divergence.SetPosition(instance["x"], instance["y"]) |
|
402 |
divergence.SetSize(instance["width"], instance["height"]) |
|
403 |
self.Blocks.append(divergence) |
|
404 |
self.Elements.append(divergence) |
|
405 |
connectors = divergence.GetConnectors() |
|
406 |
for i, input_connector in enumerate(instance["connectors"]["inputs"]): |
|
407 |
connector = connectors["inputs"][i] |
|
408 |
connector.SetPosition(wxPoint(*input_connector["position"])) |
|
409 |
self.CreateWires(connector, input_connector["links"], ids) |
|
410 |
for i, output_connector in enumerate(instance["connectors"]["outputs"]): |
|
411 |
connector = connectors["outputs"][i] |
|
412 |
connector.SetPosition(wxPoint(*output_connector["position"])) |
|
413 |
elif instance["type"] == "jump": |
|
414 |
jump = SFC_Jump(self, instance["target"], instance["id"]) |
|
415 |
jump.SetPosition(instance["x"], instance["y"]) |
|
416 |
self.Blocks.append(jump) |
|
417 |
self.Elements.append(jump) |
|
418 |
connector = jump.GetConnector() |
|
419 |
connector.SetPosition(wxPoint(*instance["connector"]["position"])) |
|
420 |
self.CreateWires(connector, instance["connector"]["links"], ids) |
|
421 |
elif instance["type"] == "actionBlock": |
|
422 |
actionBlock = SFC_ActionBlock(self, instance["actions"], instance["id"]) |
|
423 |
actionBlock.SetPosition(instance["x"], instance["y"]) |
|
424 |
actionBlock.SetSize(instance["width"], instance["height"]) |
|
425 |
self.Blocks.append(actionBlock) |
|
426 |
self.Elements.append(actionBlock) |
|
427 |
connector = actionBlock.GetConnector() |
|
428 |
connector.SetPosition(wxPoint(*instance["connector"]["position"])) |
|
429 |
self.CreateWires(connector, instance["connector"]["links"], ids) |
|
430 |
else: |
|
431 |
if instance["name"] != None: |
|
432 |
block = FBD_Block(self, instance["type"], instance["name"], instance["id"], len(instance["connectors"]["inputs"])) |
|
433 |
else: |
|
434 |
block = FBD_Block(self, instance["type"], "", instance["id"], len(instance["connectors"]["inputs"])) |
|
435 |
block.SetPosition(instance["x"], instance["y"]) |
|
436 |
block.SetSize(instance["width"], instance["height"]) |
|
437 |
self.Blocks.append(block) |
|
438 |
self.Elements.append(block) |
|
439 |
connectors = block.GetConnectors() |
|
440 |
for i, input_connector in enumerate(instance["connectors"]["inputs"]): |
|
441 |
connector = connectors["inputs"][i] |
|
442 |
connector.SetPosition(wxPoint(*input_connector["position"])) |
|
443 |
if input_connector["negated"]: |
|
444 |
connector.SetNegated(True) |
|
445 |
if input_connector["edge"]: |
|
446 |
connector.SetEdge(input_connector["edge"]) |
|
447 |
self.CreateWires(connector, input_connector["links"], ids) |
|
448 |
for i, output_connector in enumerate(instance["connectors"]["outputs"]): |
|
449 |
connector = connectors["outputs"][i] |
|
450 |
if output_connector["negated"]: |
|
451 |
connector.SetNegated(True) |
|
452 |
if output_connector["edge"]: |
|
453 |
connector.SetEdge(output_connector["edge"]) |
|
454 |
connector.SetPosition(wxPoint(*output_connector["position"])) |
|
455 |
||
456 |
def CreateWires(self, start_connector, links, ids): |
|
457 |
for link in links: |
|
458 |
refLocalId = link["refLocalId"] |
|
459 |
if refLocalId != None: |
|
460 |
if refLocalId not in ids: |
|
461 |
new_instance = self.Controler.GetCurrentElementEditingInstanceInfos(refLocalId) |
|
462 |
if new_instance: |
|
463 |
self.loadInstance(new_instance, ids) |
|
464 |
connected = self.FindElementById(refLocalId) |
|
465 |
if connected: |
|
466 |
points = link["points"] |
|
| 27 | 467 |
end_connector = connected.GetConnector(wxPoint(points[-1][0], points[-1][1]), link["formalParameter"]) |
| 0 | 468 |
if end_connector: |
469 |
wire = Wire(self) |
|
470 |
wire.SetPoints(points) |
|
471 |
start_connector.Connect((wire, 0), False) |
|
472 |
end_connector.Connect((wire, -1), False) |
|
473 |
wire.ConnectStartPoint(None, start_connector) |
|
474 |
wire.ConnectEndPoint(None, end_connector) |
|
475 |
self.Wires.append(wire) |
|
476 |
self.Elements.append(wire) |
|
477 |
||
478 |
#------------------------------------------------------------------------------- |
|
479 |
# Search Element functions |
|
480 |
#------------------------------------------------------------------------------- |
|
481 |
||
482 |
def FindBlock(self, pos): |
|
483 |
for block in self.Blocks: |
|
484 |
if block.HitTest(pos) or block.TestHandle(pos) != (0, 0): |
|
485 |
return block |
|
486 |
return None |
|
487 |
||
488 |
def FindWire(self, pos): |
|
489 |
for wire in self.Wires: |
|
490 |
if wire.HitTest(pos) or wire.TestHandle(pos) != (0, 0): |
|
491 |
return wire |
|
492 |
return None |
|
493 |
||
494 |
def FindElement(self, pos, exclude_group = False): |
|
495 |
if self.SelectedElement and not (exclude_group and isinstance(self.SelectedElement, Graphic_Group)): |
|
496 |
if self.SelectedElement.HitTest(pos) or self.SelectedElement.TestHandle(pos) != (0, 0): |
|
497 |
return self.SelectedElement |
|
498 |
for element in self.Elements: |
|
499 |
if element.HitTest(pos) or element.TestHandle(pos) != (0, 0): |
|
500 |
return element |
|
501 |
return None |
|
502 |
||
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
503 |
def FindBlockConnector(self, pos, exclude = True): |
| 0 | 504 |
for block in self.Blocks: |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
505 |
result = block.TestConnector(pos, exclude) |
| 0 | 506 |
if result: |
507 |
return result |
|
508 |
return None |
|
509 |
||
510 |
def FindElementById(self, id): |
|
511 |
for element in self.Elements: |
|
512 |
if element.GetId() == id: |
|
513 |
return element |
|
514 |
return None |
|
515 |
||
516 |
def SearchElements(self, bbox): |
|
517 |
elements = [] |
|
518 |
for element in self.Elements: |
|
519 |
element_bbox = element.GetBoundingBox() |
|
520 |
if element_bbox.x >= bbox.x and element_bbox.y >= bbox.y and element_bbox.x + element_bbox.width <= bbox.x + bbox.width and element_bbox.y + element_bbox.height <= bbox.y + bbox.height: |
|
521 |
elements.append(element) |
|
522 |
return elements |
|
523 |
||
524 |
#------------------------------------------------------------------------------- |
|
525 |
# Popup menu functions |
|
526 |
#------------------------------------------------------------------------------- |
|
527 |
||
528 |
def PopupBlockMenu(self, connector = None): |
|
|
13
69075340d6a9
Adding support for forbidding insertion of function block into function
lbessard
parents:
10
diff
changeset
|
529 |
type = self.Controler.GetCurrentElementEditingType() |
| 0 | 530 |
self.ContextualMenu.FindItemByPosition(0).Enable(connector != None) |
531 |
self.ContextualMenu.FindItemByPosition(1).Enable(connector != None) |
|
|
13
69075340d6a9
Adding support for forbidding insertion of function block into function
lbessard
parents:
10
diff
changeset
|
532 |
self.ContextualMenu.FindItemByPosition(2).Enable(connector != None and type != "function") |
|
69075340d6a9
Adding support for forbidding insertion of function block into function
lbessard
parents:
10
diff
changeset
|
533 |
self.ContextualMenu.FindItemByPosition(3).Enable(connector != None and type != "function") |
| 0 | 534 |
self.ContextualMenu.FindItemByPosition(5).Enable(False) |
535 |
self.ContextualMenu.FindItemByPosition(6).Enable(False) |
|
536 |
self.ContextualMenu.FindItemByPosition(8).Enable(False) |
|
537 |
self.ContextualMenu.FindItemByPosition(9).Enable(False) |
|
538 |
if connector: |
|
539 |
if connector.IsNegated(): |
|
540 |
self.ContextualMenu.FindItemByPosition(1).Check(True) |
|
541 |
elif connector.GetEdge() == "rising": |
|
542 |
self.ContextualMenu.FindItemByPosition(2).Check(True) |
|
543 |
elif connector.GetEdge() == "falling": |
|
544 |
self.ContextualMenu.FindItemByPosition(3).Check(True) |
|
545 |
else: |
|
546 |
self.ContextualMenu.FindItemByPosition(0).Check(True) |
|
547 |
self.PopupMenu(self.ContextualMenu) |
|
548 |
||
549 |
def PopupVariableMenu(self, connector = None): |
|
550 |
self.ContextualMenu.FindItemByPosition(0).Enable(connector != None) |
|
551 |
self.ContextualMenu.FindItemByPosition(1).Enable(connector != None) |
|
552 |
self.ContextualMenu.FindItemByPosition(2).Enable(False) |
|
553 |
self.ContextualMenu.FindItemByPosition(3).Enable(False) |
|
554 |
self.ContextualMenu.FindItemByPosition(5).Enable(False) |
|
555 |
self.ContextualMenu.FindItemByPosition(6).Enable(False) |
|
556 |
self.ContextualMenu.FindItemByPosition(8).Enable(False) |
|
557 |
self.ContextualMenu.FindItemByPosition(9).Enable(False) |
|
558 |
if connector: |
|
559 |
if connector.IsNegated(): |
|
560 |
self.ContextualMenu.FindItemByPosition(1).Check(True) |
|
561 |
else: |
|
562 |
self.ContextualMenu.FindItemByPosition(0).Check(True) |
|
563 |
self.PopupMenu(self.ContextualMenu) |
|
564 |
||
565 |
def PopupWireMenu(self): |
|
566 |
self.ContextualMenu.FindItemByPosition(0).Enable(False) |
|
567 |
self.ContextualMenu.FindItemByPosition(1).Enable(False) |
|
568 |
self.ContextualMenu.FindItemByPosition(2).Enable(False) |
|
569 |
self.ContextualMenu.FindItemByPosition(3).Enable(False) |
|
570 |
self.ContextualMenu.FindItemByPosition(5).Enable(True) |
|
571 |
self.ContextualMenu.FindItemByPosition(6).Enable(True) |
|
572 |
self.ContextualMenu.FindItemByPosition(8).Enable(False) |
|
573 |
self.ContextualMenu.FindItemByPosition(9).Enable(False) |
|
574 |
self.PopupMenu(self.ContextualMenu) |
|
575 |
||
576 |
def PopupDivergenceMenu(self, connector): |
|
577 |
self.ContextualMenu.FindItemByPosition(0).Enable(False) |
|
578 |
self.ContextualMenu.FindItemByPosition(1).Enable(False) |
|
579 |
self.ContextualMenu.FindItemByPosition(2).Enable(False) |
|
580 |
self.ContextualMenu.FindItemByPosition(3).Enable(False) |
|
581 |
self.ContextualMenu.FindItemByPosition(5).Enable(False) |
|
582 |
self.ContextualMenu.FindItemByPosition(6).Enable(False) |
|
583 |
self.ContextualMenu.FindItemByPosition(8).Enable(True) |
|
584 |
self.ContextualMenu.FindItemByPosition(9).Enable(connector) |
|
585 |
self.PopupMenu(self.ContextualMenu) |
|
586 |
||
587 |
def PopupDefaultMenu(self): |
|
588 |
self.ContextualMenu.FindItemByPosition(0).Enable(False) |
|
589 |
self.ContextualMenu.FindItemByPosition(1).Enable(False) |
|
590 |
self.ContextualMenu.FindItemByPosition(2).Enable(False) |
|
591 |
self.ContextualMenu.FindItemByPosition(3).Enable(False) |
|
592 |
self.ContextualMenu.FindItemByPosition(5).Enable(False) |
|
593 |
self.ContextualMenu.FindItemByPosition(6).Enable(False) |
|
594 |
self.ContextualMenu.FindItemByPosition(8).Enable(False) |
|
595 |
self.ContextualMenu.FindItemByPosition(9).Enable(False) |
|
596 |
self.PopupMenu(self.ContextualMenu) |
|
597 |
||
598 |
def EditCommentContent(self, comment): |
|
599 |
dialog = wxTextEntryDialog(self.Parent, "Edit comment", "Please enter comment text", comment.GetContent(), wxOK|wxCANCEL|wxTE_MULTILINE) |
|
600 |
if dialog.ShowModal() == wxID_OK: |
|
601 |
value = dialog.GetValue() |
|
602 |
comment.SetContent(value) |
|
603 |
infos = {"content" : value}
|
|
604 |
infos["width"], infos["height"] = comment.GetSize() |
|
605 |
self.Controler.SetCurrentElementEditingCommentInfos(comment.GetId(), infos) |
|
606 |
self.Refresh() |
|
607 |
dialog.Destroy() |
|
608 |
||
609 |
#------------------------------------------------------------------------------- |
|
610 |
# Menu items functions |
|
611 |
#------------------------------------------------------------------------------- |
|
612 |
||
613 |
def OnNoModifierMenu(self, event): |
|
614 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
615 |
self.SelectedElement.SetConnectorNegated(False) |
|
616 |
event.Skip() |
|
617 |
||
618 |
def OnNegatedMenu(self, event): |
|
619 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
620 |
self.SelectedElement.SetConnectorNegated(True) |
|
621 |
event.Skip() |
|
622 |
||
623 |
def OnRisingEdgeMenu(self, event): |
|
624 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
625 |
self.SelectedElement.SetConnectorEdge("rising")
|
|
626 |
event.Skip() |
|
627 |
||
628 |
def OnFallingEdgeMenu(self, event): |
|
629 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
630 |
self.SelectedElement.SetConnectorEdge("falling")
|
|
631 |
event.Skip() |
|
632 |
||
633 |
def OnAddSegmentMenu(self, event): |
|
634 |
if self.SelectedElement and self.SelectedElement in self.Wires: |
|
635 |
self.SelectedElement.AddSegment() |
|
636 |
event.Skip() |
|
637 |
||
638 |
def OnDeleteSegmentMenu(self, event): |
|
639 |
if self.SelectedElement and self.SelectedElement in self.Wires: |
|
640 |
self.SelectedElement.DeleteSegment() |
|
641 |
event.Skip() |
|
642 |
||
643 |
def OnAddBranchMenu(self, event): |
|
644 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
645 |
self.AddDivergenceBranch(self.SelectedElement) |
|
646 |
event.Skip() |
|
647 |
||
648 |
def OnDeleteBranchMenu(self, event): |
|
649 |
if self.SelectedElement and self.SelectedElement in self.Blocks: |
|
650 |
self.RemoveDivergenceBranch(self.SelectedElement) |
|
651 |
event.Skip() |
|
652 |
||
653 |
def OnDeleteMenu(self, event): |
|
654 |
if self.SelectedElement: |
|
655 |
self.SelectedElement.Delete() |
|
656 |
self.SelectedElement = None |
|
657 |
event.Skip() |
|
658 |
||
659 |
#------------------------------------------------------------------------------- |
|
660 |
# Mouse event functions |
|
661 |
#------------------------------------------------------------------------------- |
|
662 |
||
663 |
def OnViewerLeftDown(self, event): |
|
| 27 | 664 |
if self.Mode == MODE_SELECTION: |
665 |
dc = self.GetLogicalDC() |
|
666 |
pos = event.GetLogicalPosition(dc) |
|
667 |
if event.ControlDown() and self.SelectedElement: |
|
668 |
element = self.FindElement(pos, True) |
|
669 |
if element: |
|
670 |
if isinstance(self.SelectedElement, Graphic_Group): |
|
671 |
self.SelectedElement.SetSelected(False) |
|
672 |
self.SelectedElement.SelectElement(element) |
|
673 |
elif self.SelectedElement: |
|
674 |
group = Graphic_Group(self) |
|
675 |
group.SelectElement(self.SelectedElement) |
|
676 |
group.SelectElement(element) |
|
677 |
self.SelectedElement = group |
|
678 |
elements = self.SelectedElement.GetElements() |
|
679 |
if len(elements) == 0: |
|
680 |
self.SelectedElement = element |
|
681 |
elif len(elements) == 1: |
|
682 |
self.SelectedElement = elements[0] |
|
683 |
self.SelectedElement.SetSelected(True) |
|
684 |
else: |
|
685 |
element = self.FindElement(pos) |
|
686 |
if self.SelectedElement and self.SelectedElement != element: |
|
687 |
self.SelectedElement.SetSelected(False) |
|
688 |
self.SelectedElement = None |
|
689 |
self.Refresh() |
|
690 |
if element: |
|
691 |
self.SelectedElement = element |
|
692 |
self.SelectedElement.OnLeftDown(event, dc, self.Scaling) |
|
693 |
self.Refresh() |
|
694 |
else: |
|
695 |
self.rubberBand.Reset() |
|
696 |
self.rubberBand.OnLeftDown(event, dc, self.Scaling) |
|
697 |
elif self.Mode in [MODE_BLOCK, MODE_VARIABLE, MODE_CONNECTION, MODE_COMMENT, |
|
698 |
MODE_CONTACT, MODE_COIL, MODE_POWERRAIL, MODE_INITIALSTEP, |
|
699 |
MODE_STEP, MODE_TRANSITION, MODE_DIVERGENCE, MODE_JUMP, MODE_ACTION]: |
|
700 |
self.rubberBand.Reset() |
|
701 |
self.rubberBand.OnLeftDown(event, self.GetLogicalDC(), self.Scaling) |
|
702 |
elif self.Mode == MODE_WIRE: |
|
703 |
pos = GetScaledEventPosition(event, self.GetLogicalDC(), self.Scaling) |
|
704 |
connector = self.FindBlockConnector(pos) |
|
705 |
if connector: |
|
706 |
if (connector.GetDirection() == EAST): |
|
707 |
wire = Wire(self, [wxPoint(pos.x, pos.y), EAST], [wxPoint(pos.x, pos.y), WEST]) |
|
708 |
else: |
|
709 |
wire = Wire(self, [wxPoint(pos.x, pos.y), WEST], [wxPoint(pos.x, pos.y), EAST]) |
|
710 |
wire.oldPos = pos |
|
711 |
wire.Handle = (HANDLE_POINT, 0) |
|
712 |
wire.ProcessDragging(0, 0) |
|
713 |
wire.Handle = (HANDLE_POINT, 1) |
|
714 |
self.Wires.append(wire) |
|
715 |
self.Elements.append(wire) |
|
716 |
if self.SelectedElement: |
|
717 |
self.SelectedElement.SetSelected(False) |
|
718 |
self.SelectedElement = wire |
|
719 |
elif self.SelectedElement: |
|
720 |
self.SelectedElement.SetSelected(False) |
|
721 |
self.SelectedElement = None |
|
722 |
self.Refresh() |
|
| 0 | 723 |
event.Skip() |
724 |
||
725 |
def OnViewerLeftUp(self, event): |
|
| 27 | 726 |
if self.rubberBand.IsShown(): |
727 |
if self.Mode == MODE_SELECTION: |
|
728 |
elements = self.SearchElements(self.rubberBand.GetCurrentExtent()) |
|
729 |
self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) |
|
730 |
if len(elements) > 0: |
|
731 |
self.SelectedElement = Graphic_Group(self) |
|
732 |
self.SelectedElement.SetElements(elements) |
|
733 |
self.SelectedElement.SetSelected(True) |
|
734 |
self.Refresh() |
|
735 |
else: |
|
736 |
bbox = self.rubberBand.GetCurrentExtent() |
|
737 |
self.rubberBand.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) |
|
738 |
if self.Mode == MODE_BLOCK: |
|
739 |
wxCallAfter(self.AddNewBlock, bbox) |
|
740 |
elif self.Mode == MODE_VARIABLE: |
|
741 |
wxCallAfter(self.AddNewVariable, bbox) |
|
742 |
elif self.Mode == MODE_CONNECTION: |
|
743 |
wxCallAfter(self.AddNewConnection, bbox) |
|
744 |
elif self.Mode == MODE_COMMENT: |
|
745 |
wxCallAfter(self.AddNewComment, bbox) |
|
746 |
elif self.Mode == MODE_CONTACT: |
|
747 |
wxCallAfter(self.AddNewContact, bbox) |
|
748 |
elif self.Mode == MODE_COIL: |
|
749 |
wxCallAfter(self.AddNewContact, bbox) |
|
750 |
elif self.Mode == MODE_POWERRAIL: |
|
751 |
wxCallAfter(self.AddNewPowerRail, bbox) |
|
752 |
elif self.Mode == MODE_INITIALSTEP: |
|
753 |
wxCallAfter(self.AddNewInitialStep, bbox) |
|
754 |
elif self.Mode == MODE_STEP: |
|
755 |
wxCallAfter(self.AddNewStep, bbox) |
|
756 |
elif self.Mode == MODE_TRANSITION: |
|
757 |
wxCallAfter(self.AddNewTransition, bbox) |
|
758 |
elif self.Mode == MODE_DIVERGENCE: |
|
759 |
wxCallAfter(self.AddNewDivergence, bbox) |
|
760 |
elif self.Mode == MODE_JUMP: |
|
761 |
wxCallAfter(self.AddNewJump, bbox) |
|
762 |
elif self.Mode == MODE_ACTION: |
|
763 |
wxCallAfter(self.AddNewActionBlock, bbox) |
|
764 |
elif self.Mode == MODE_SELECTION and self.SelectedElement: |
|
765 |
self.SelectedElement.OnLeftUp(event, self.GetLogicalDC(), self.Scaling) |
|
766 |
wxCallAfter(self.SetCursor, wxNullCursor) |
|
767 |
self.ReleaseMouse() |
|
768 |
self.Refresh() |
|
769 |
elif self.Mode == MODE_WIRE and self.SelectedElement: |
|
770 |
dc = self.GetLogicalDC() |
|
771 |
pos = GetScaledEventPosition(event, dc, self.Scaling) |
|
772 |
connector = self.FindBlockConnector(pos, False) |
|
773 |
if connector and connector != self.SelectedElement.StartConnected: |
|
774 |
self.SelectedElement.ResetPoints() |
|
775 |
self.SelectedElement.OnMotion(event, dc, self.Scaling) |
|
776 |
self.SelectedElement.GeneratePoints() |
|
777 |
self.SelectedElement.RefreshModel() |
|
778 |
self.SelectedElement.SetSelected(True) |
|
779 |
else: |
|
780 |
self.SelectedElement.Delete() |
|
781 |
self.SelectedElement = None |
|
782 |
self.Refresh() |
|
783 |
if not self.SavedMode: |
|
784 |
wxCallAfter(self.Parent.ResetCurrentMode) |
|
| 0 | 785 |
event.Skip() |
786 |
||
787 |
def OnViewerRightUp(self, event): |
|
| 27 | 788 |
pos = event.GetPosition() |
789 |
element = self.FindElement(pos) |
|
790 |
if element: |
|
791 |
if self.SelectedElement and self.SelectedElement != element: |
|
792 |
self.SelectedElement.SetSelected(False) |
|
793 |
self.SelectedElement = element |
|
794 |
self.SelectedElement.SetSelected(True) |
|
795 |
self.SelectedElement.OnRightUp(event, self.GetLogicalDC(), self.Scaling) |
|
796 |
wxCallAfter(self.SetCursor, wxNullCursor) |
|
797 |
self.ReleaseMouse() |
|
798 |
self.Refresh() |
|
| 0 | 799 |
event.Skip() |
800 |
||
801 |
def OnViewerLeftDClick(self, event): |
|
| 27 | 802 |
if self.Mode == MODE_SELECTION and self.SelectedElement: |
803 |
self.SelectedElement.OnLeftDClick(event, self.GetLogicalDC(), self.Scaling) |
|
804 |
self.Refresh() |
|
| 0 | 805 |
event.Skip() |
806 |
||
807 |
def OnViewerMotion(self, event): |
|
| 27 | 808 |
if self.rubberBand.IsShown(): |
809 |
self.rubberBand.OnMotion(event, self.GetLogicalDC(), self.Scaling) |
|
810 |
elif self.Mode == MODE_SELECTION and self.SelectedElement: |
|
811 |
self.SelectedElement.OnMotion(event, self.GetLogicalDC(), self.Scaling) |
|
812 |
self.Refresh() |
|
813 |
elif self.Mode == MODE_WIRE and self.SelectedElement: |
|
814 |
dc = self.GetLogicalDC() |
|
815 |
pos = GetScaledEventPosition(event, dc, self.Scaling) |
|
816 |
connector = self.FindBlockConnector(pos, False) |
|
817 |
if not connector or self.SelectedElement.EndConnected == None: |
|
818 |
self.SelectedElement.ResetPoints() |
|
819 |
self.SelectedElement.OnMotion(event, dc, self.Scaling) |
|
820 |
self.SelectedElement.GeneratePoints() |
|
821 |
self.Refresh() |
|
822 |
if (event.Dragging() and self.SelectedElement) or self.rubberBand.IsShown(): |
|
823 |
position = event.GetPosition() |
|
824 |
move_window = wxPoint() |
|
825 |
window_size = self.GetClientSize() |
|
826 |
xstart, ystart = self.GetViewStart() |
|
827 |
if position.x < SCROLL_ZONE and xstart > 0: |
|
828 |
move_window.x = -1 |
|
829 |
elif position.x > window_size[0] - SCROLL_ZONE: |
|
830 |
move_window.x = 1 |
|
831 |
if position.y < SCROLL_ZONE and ystart > 0: |
|
832 |
move_window.y = -1 |
|
833 |
elif position.y > window_size[1] - SCROLL_ZONE: |
|
834 |
move_window.y = 1 |
|
835 |
if move_window.x != 0 or move_window.y != 0: |
|
836 |
self.Scroll(xstart + move_window.x, ystart + move_window.y) |
|
837 |
self.RefreshScrollBar() |
|
| 0 | 838 |
event.Skip() |
839 |
||
840 |
#------------------------------------------------------------------------------- |
|
841 |
# Keyboard event functions |
|
842 |
#------------------------------------------------------------------------------- |
|
843 |
||
844 |
def OnChar(self, event): |
|
| 27 | 845 |
keycode = event.GetKeyCode() |
846 |
if self.Scaling: |
|
847 |
scaling = self.Scaling |
|
848 |
else: |
|
849 |
scaling = (8, 8) |
|
850 |
if keycode == WXK_DELETE and self.SelectedElement: |
|
851 |
self.SelectedElement.Clean() |
|
852 |
self.SelectedElement.Delete() |
|
853 |
self.SelectedElement = None |
|
854 |
elif keycode == WXK_LEFT and self.SelectedElement: |
|
855 |
self.SelectedElement.Move(-scaling[0], 0) |
|
856 |
elif keycode == WXK_RIGHT and self.SelectedElement: |
|
857 |
self.SelectedElement.Move(scaling[0], 0) |
|
858 |
elif keycode == WXK_UP and self.SelectedElement: |
|
859 |
self.SelectedElement.Move(0, -scaling[1]) |
|
860 |
elif keycode == WXK_DOWN and self.SelectedElement: |
|
861 |
self.SelectedElement.Move(0, scaling[1]) |
|
862 |
self.Refresh() |
|
863 |
event.Skip() |
|
864 |
||
865 |
#------------------------------------------------------------------------------- |
|
866 |
# Model adding functions |
|
867 |
#------------------------------------------------------------------------------- |
|
868 |
||
869 |
def AddNewBlock(self, bbox): |
|
870 |
dialog = BlockPropertiesDialog(self.Parent) |
|
871 |
dialog.SetBlockList(self.Controler.GetBlockTypes()) |
|
872 |
dialog.SetMinBlockSize((bbox.width, bbox.height)) |
|
873 |
if dialog.ShowModal() == wxID_OK: |
|
874 |
id = self.GetNewId() |
|
875 |
values = dialog.GetValues() |
|
876 |
if "name" in values: |
|
877 |
block = FBD_Block(self, values["type"], values["name"], id, values["extension"]) |
|
878 |
else: |
|
879 |
block = FBD_Block(self, values["type"], "", id, values["extension"]) |
|
880 |
block.SetPosition(bbox.x, bbox.y) |
|
881 |
block.SetSize(values["width"], values["height"]) |
|
882 |
self.Blocks.append(block) |
|
883 |
self.Elements.append(block) |
|
884 |
self.Controler.AddCurrentElementEditingBlock(id) |
|
885 |
self.RefreshBlockModel(block) |
|
886 |
self.Refresh() |
|
887 |
dialog.Destroy() |
|
888 |
||
889 |
def AddNewVariable(self, bbox): |
|
890 |
dialog = VariablePropertiesDialog(self.Parent) |
|
891 |
dialog.SetMinVariableSize((bbox.width, bbox.height)) |
|
892 |
varlist = [] |
|
893 |
vars = self.Controler.GetCurrentElementEditingInterfaceVars() |
|
894 |
if vars: |
|
895 |
for var in vars: |
|
896 |
varlist.append((var["Name"], var["Class"], var["Type"])) |
|
897 |
returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() |
|
898 |
if returntype: |
|
899 |
varlist.append((self.Controler.GetCurrentElementEditingName(), "Output", returntype)) |
|
900 |
dialog.SetVariables(varlist) |
|
901 |
if dialog.ShowModal() == wxID_OK: |
|
902 |
id = self.GetNewId() |
|
903 |
values = dialog.GetValues() |
|
904 |
variable = FBD_Variable(self, values["type"], values["name"], values["value_type"], id) |
|
905 |
variable.SetPosition(bbox.x, bbox.y) |
|
906 |
variable.SetSize(values["width"], values["height"]) |
|
907 |
self.Blocks.append(variable) |
|
908 |
self.Elements.append(variable) |
|
909 |
self.Controler.AddCurrentElementEditingVariable(id, values["type"]) |
|
910 |
self.RefreshVariableModel(variable) |
|
911 |
self.Refresh() |
|
912 |
dialog.Destroy() |
|
913 |
||
914 |
def AddNewConnection(self, bbox): |
|
915 |
dialog = ConnectionPropertiesDialog(self.Parent) |
|
916 |
dialog.SetMinConnectionSize((bbox.width, bbox.height)) |
|
917 |
if dialog.ShowModal() == wxID_OK: |
|
918 |
id = self.GetNewId() |
|
919 |
values = dialog.GetValues() |
|
920 |
connection = FBD_Connector(self, values["type"], values["name"], id) |
|
921 |
connection.SetPosition(bbox.x, bbox.y) |
|
922 |
connection.SetSize(values["width"], values["height"]) |
|
923 |
self.Blocks.append(connection) |
|
924 |
self.Elements.append(connection) |
|
925 |
self.Controler.AddCurrentElementEditingConnection(id, values["type"]) |
|
926 |
self.RefreshConnectionModel(connection) |
|
927 |
self.Refresh() |
|
928 |
dialog.Destroy() |
|
929 |
||
930 |
def AddNewComment(self, bbox): |
|
931 |
dialog = wxTextEntryDialog(self.Parent, "Add a new comment", "Please enter comment text", "", wxOK|wxCANCEL|wxTE_MULTILINE) |
|
932 |
if dialog.ShowModal() == wxID_OK: |
|
933 |
value = dialog.GetValue() |
|
934 |
id = self.GetNewId() |
|
935 |
comment = Comment(self, value, id) |
|
936 |
comment.SetPosition(bbox.x, bbox.y) |
|
937 |
min_width, min_height = comment.GetMinSize() |
|
938 |
comment.SetSize(max(min_width,bbox.width),max(min_height,bbox.height)) |
|
939 |
self.Elements.append(comment) |
|
940 |
self.Controler.AddCurrentElementEditingComment(id) |
|
941 |
self.RefreshCommentModel(comment) |
|
942 |
self.Refresh() |
|
943 |
dialog.Destroy() |
|
944 |
||
945 |
def AddNewContact(self, bbox): |
|
946 |
dialog = LDElementDialog(self.Parent, "contact") |
|
947 |
varlist = [] |
|
948 |
vars = self.Controler.GetCurrentElementEditingInterfaceVars() |
|
949 |
if vars: |
|
950 |
for var in vars: |
|
951 |
if var["Class"] != "Output" and var["Type"] == "BOOL": |
|
952 |
varlist.append(var["Name"]) |
|
953 |
dialog.SetVariables(varlist) |
|
954 |
dialog.SetValues({"name":"","type":CONTACT_NORMAL})
|
|
955 |
dialog.SetElementSize(bbox.width, bbox.height) |
|
956 |
if dialog.ShowModal() == wxID_OK: |
|
957 |
id = self.GetNewId() |
|
958 |
values = dialog.GetValues() |
|
959 |
contact = LD_Contact(self, values["type"], values["name"], id) |
|
960 |
contact.SetPosition(bbox.x, bbox.y) |
|
961 |
contact.SetSize(values["width"], values["height"]) |
|
962 |
self.Blocks.append(contact) |
|
963 |
self.Elements.append(contact) |
|
964 |
self.Controler.AddCurrentElementEditingContact(id) |
|
965 |
self.RefreshContactModel(contact) |
|
966 |
self.Refresh() |
|
967 |
dialog.Destroy() |
|
968 |
||
969 |
def AddNewCoil(self, bbox): |
|
970 |
dialog = LDElementDialog(self.Parent, "coil") |
|
971 |
varlist = [] |
|
972 |
vars = self.Controler.GetCurrentElementEditingInterfaceVars() |
|
973 |
if vars: |
|
974 |
for var in vars: |
|
975 |
if var["Class"] != "Input" and var["Type"] == "BOOL": |
|
976 |
varlist.append(var["Name"]) |
|
977 |
returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() |
|
978 |
if returntype == "BOOL": |
|
979 |
varlist.append(self.Controler.GetCurrentElementEditingName()) |
|
980 |
dialog.SetVariables(varlist) |
|
981 |
dialog.SetValues({"name":"","type":COIL_NORMAL})
|
|
982 |
dialog.SetElementSize(bbox.width, bbox.height) |
|
983 |
if dialog.ShowModal() == wxID_OK: |
|
984 |
id = self.GetNewId() |
|
985 |
values = dialog.GetValues() |
|
986 |
coil = LD_Coil(self, values["type"], values["name"], id) |
|
987 |
coil.SetPosition(bbox.x, bbox.y) |
|
988 |
coil.SetSize(values["width"], values["height"]) |
|
989 |
self.Blocks.append(coil) |
|
990 |
self.Elements.append(coil) |
|
991 |
self.Controler.AddCurrentElementEditingCoil(id) |
|
992 |
self.RefreshCoilModel(contact) |
|
993 |
self.Refresh() |
|
994 |
dialog.Destroy() |
|
995 |
||
996 |
def AddNewPowerRail(self, bbox): |
|
997 |
dialog = LDPowerRailDialog(self.Parent) |
|
998 |
dialog.SetMinSize((bbox.width, bbox.height)) |
|
999 |
if dialog.ShowModal() == wxID_OK: |
|
1000 |
id = self.GetNewId() |
|
1001 |
values = dialog.GetValues() |
|
1002 |
powerrail = LD_PowerRail(self, values["type"], id, [True for i in xrange(values["number"])]) |
|
1003 |
powerrail.SetPosition(bbox.x, bbox.y) |
|
1004 |
powerrail.SetSize(values["width"], values["height"]) |
|
1005 |
self.Blocks.append(powerrail) |
|
1006 |
self.Elements.append(powerrail) |
|
1007 |
self.Controler.AddCurrentElementEditingPowerRail(id, values["type"]) |
|
1008 |
self.RefreshPowerRailModel(powerrail) |
|
1009 |
self.Refresh() |
|
1010 |
dialog.Destroy() |
|
1011 |
||
1012 |
def AddNewTransition(self, bbox): |
|
1013 |
dialog = TransitionContentDialog(self.Parent) |
|
1014 |
dialog.SetTransitions(self.Controler.GetCurrentElementEditingTransitions()) |
|
1015 |
if dialog.ShowModal() == wxID_OK: |
|
1016 |
id = self.GetNewId() |
|
1017 |
values = dialog.GetValues() |
|
1018 |
transition = SFC_Transition(self, values["type"], values["value"], id) |
|
1019 |
transition.SetPosition(bbox.x, bbox.y) |
|
1020 |
min_width, min_height = transition.GetMinSize() |
|
1021 |
transition.SetSize(max(bbox.width, min_width), max(bbox.height, min_height)) |
|
1022 |
self.Blocks.append(transition) |
|
1023 |
self.Elements.append(transition) |
|
1024 |
self.Controler.AddCurrentElementEditingTransition(id) |
|
1025 |
self.RefreshTransitionModel(transition) |
|
1026 |
self.Refresh() |
|
1027 |
dialog.Destroy() |
|
1028 |
||
1029 |
def AddNewDivergence(self, bbox): |
|
1030 |
dialog = DivergenceCreateDialog(self.Parent) |
|
1031 |
dialog.SetMinSize((bbox.width, bbox.height)) |
|
1032 |
if dialog.ShowModal() == wxID_OK: |
|
1033 |
id = self.GetNewId() |
|
1034 |
values = dialog.GetValues() |
|
1035 |
divergence = SFC_Divergence(self, values["type"], values["number"], id) |
|
1036 |
divergence.SetPosition(bbox.x, bbox.y) |
|
1037 |
min_width, min_height = divergence.GetMinSize() |
|
1038 |
divergence.SetSize(max(bbox.width, min_width), max(bbox.height, min_height)) |
|
1039 |
self.Blocks.append(divergence) |
|
1040 |
self.Elements.append(divergence) |
|
1041 |
self.Controler.AddCurrentElementEditingDivergence(id, values["type"]) |
|
1042 |
self.RefreshDivergenceModel(divergence) |
|
1043 |
self.Refresh() |
|
1044 |
dialog.Destroy() |
|
| 0 | 1045 |
|
1046 |
#------------------------------------------------------------------------------- |
|
1047 |
# Model update functions |
|
1048 |
#------------------------------------------------------------------------------- |
|
1049 |
||
1050 |
def RefreshBlockModel(self, block): |
|
1051 |
blockid = block.GetId() |
|
1052 |
infos = {}
|
|
1053 |
infos["type"] = block.GetType() |
|
1054 |
infos["name"] = block.GetName() |
|
1055 |
infos["x"], infos["y"] = block.GetPosition() |
|
1056 |
infos["width"], infos["height"] = block.GetSize() |
|
1057 |
infos["connectors"] = block.GetConnectors() |
|
1058 |
self.Controler.SetCurrentElementEditingBlockInfos(blockid, infos) |
|
1059 |
||
1060 |
def RefreshVariableModel(self, variable): |
|
1061 |
variableid = variable.GetId() |
|
1062 |
infos = {}
|
|
1063 |
infos["name"] = variable.GetName() |
|
1064 |
infos["x"], infos["y"] = variable.GetPosition() |
|
1065 |
infos["width"], infos["height"] = variable.GetSize() |
|
1066 |
infos["connectors"] = variable.GetConnectors() |
|
1067 |
self.Controler.SetCurrentElementEditingVariableInfos(variableid, infos) |
|
1068 |
||
1069 |
def RefreshConnectionModel(self, connection): |
|
1070 |
connectionid = connection.GetId() |
|
1071 |
infos = {}
|
|
1072 |
infos["name"] = connection.GetName() |
|
1073 |
infos["x"], infos["y"] = connection.GetPosition() |
|
1074 |
infos["width"], infos["height"] = connection.GetSize() |
|
1075 |
infos["connector"] = connection.GetConnector() |
|
1076 |
self.Controler.SetCurrentElementEditingConnectionInfos(connectionid, infos) |
|
1077 |
||
1078 |
def RefreshCommentModel(self, comment): |
|
1079 |
commentid = comment.GetId() |
|
1080 |
infos = {}
|
|
1081 |
infos["content"] = comment.GetContent() |
|
1082 |
infos["x"], infos["y"] = comment.GetPosition() |
|
1083 |
infos["width"], infos["height"] = comment.GetSize() |
|
1084 |
self.Controler.SetCurrentElementEditingCommentInfos(commentid, infos) |
|
1085 |
||
1086 |
def RefreshPowerRailModel(self, powerrail): |
|
1087 |
powerrailid = powerrail.GetId() |
|
1088 |
infos = {}
|
|
1089 |
infos["x"], infos["y"] = powerrail.GetPosition() |
|
1090 |
infos["width"], infos["height"] = powerrail.GetSize() |
|
1091 |
infos["connectors"] = powerrail.GetConnectors() |
|
1092 |
self.Controler.SetCurrentElementEditingPowerRailInfos(powerrailid, infos) |
|
1093 |
||
1094 |
def RefreshContactModel(self, contact): |
|
1095 |
contactid = contact.GetId() |
|
1096 |
infos = {}
|
|
1097 |
infos["name"] = contact.GetName() |
|
1098 |
infos["type"] = contact.GetType() |
|
1099 |
infos["x"], infos["y"] = contact.GetPosition() |
|
1100 |
infos["width"], infos["height"] = contact.GetSize() |
|
1101 |
infos["connectors"] = contact.GetConnectors() |
|
1102 |
self.Controler.SetCurrentElementEditingContactInfos(contactid, infos) |
|
1103 |
||
1104 |
def RefreshCoilModel(self, coil): |
|
1105 |
coilid = coil.GetId() |
|
1106 |
infos = {}
|
|
1107 |
infos["name"] = coil.GetName() |
|
1108 |
infos["type"] = coil.GetType() |
|
1109 |
infos["x"], infos["y"] = coil.GetPosition() |
|
1110 |
infos["width"], infos["height"] = coil.GetSize() |
|
1111 |
infos["connectors"] = coil.GetConnectors() |
|
1112 |
self.Controler.SetCurrentElementEditingCoilInfos(coilid, infos) |
|
1113 |
||
1114 |
def RefreshStepModel(self, step): |
|
1115 |
stepid = step.GetId() |
|
1116 |
infos = {}
|
|
1117 |
infos["name"] = step.GetName() |
|
1118 |
infos["initial"] = step.GetInitial() |
|
1119 |
infos["x"], infos["y"] = step.GetPosition() |
|
1120 |
infos["width"], infos["height"] = step.GetSize() |
|
1121 |
infos["connectors"] = step.GetConnectors() |
|
1122 |
self.Controler.SetCurrentElementEditingStepInfos(stepid, infos) |
|
1123 |
||
1124 |
def RefreshTransitionModel(self, transition): |
|
1125 |
transitionid = transition.GetId() |
|
1126 |
infos = {}
|
|
1127 |
infos["type"] = transition.GetType() |
|
1128 |
infos["condition"] = transition.GetCondition() |
|
1129 |
infos["x"], infos["y"] = transition.GetPosition() |
|
1130 |
infos["width"], infos["height"] = transition.GetSize() |
|
1131 |
infos["connectors"] = transition.GetConnectors() |
|
1132 |
self.Controler.SetCurrentElementEditingTransitionInfos(transitionid, infos) |
|
1133 |
||
1134 |
def RefreshDivergenceModel(self, divergence): |
|
1135 |
divergenceid = divergence.GetId() |
|
1136 |
infos = {}
|
|
1137 |
infos["x"], infos["y"] = divergence.GetPosition() |
|
1138 |
infos["width"], infos["height"] = divergence.GetSize() |
|
1139 |
infos["connectors"] = divergence.GetConnectors() |
|
1140 |
self.Controler.SetCurrentElementEditingDivergenceInfos(divergenceid, infos) |
|
1141 |
||
1142 |
def RefreshJumpModel(self, jump): |
|
1143 |
jumpid = jump.GetId() |
|
1144 |
infos = {}
|
|
1145 |
infos["target"] = jump.GetTarget() |
|
1146 |
infos["x"], infos["y"] = jump.GetPosition() |
|
1147 |
infos["width"], infos["height"] = jump.GetSize() |
|
1148 |
infos["connector"] = jump.GetConnector() |
|
1149 |
self.Controler.SetCurrentElementEditingJumpInfos(jumpid, infos) |
|
1150 |
||
1151 |
def RefreshActionBlockModel(self, actionblock): |
|
1152 |
actionblockid = actionblock.GetId() |
|
1153 |
infos = {}
|
|
1154 |
infos["actions"] = actionblock.GetActions() |
|
1155 |
infos["x"], infos["y"] = actionblock.GetPosition() |
|
1156 |
infos["width"], infos["height"] = actionblock.GetSize() |
|
1157 |
infos["connector"] = actionblock.GetConnector() |
|
1158 |
self.Controler.SetCurrentElementEditingActionBlockInfos(actionblockid, infos) |
|
1159 |
||
| 27 | 1160 |
|
1161 |
#------------------------------------------------------------------------------- |
|
1162 |
# Model delete functions |
|
1163 |
#------------------------------------------------------------------------------- |
|
1164 |
||
1165 |
||
1166 |
def DeleteBlock(self, block): |
|
1167 |
elements = [] |
|
1168 |
for output in block.GetConnectors()["outputs"]: |
|
1169 |
for element in output.GetConnectedBlocks(): |
|
1170 |
if element not in elements: |
|
1171 |
elements.append(element) |
|
1172 |
block.Clean() |
|
1173 |
self.Blocks.remove(block) |
|
1174 |
self.Elements.remove(block) |
|
1175 |
self.Controler.RemoveCurrentElementEditingInstance(block.GetId()) |
|
1176 |
for element in elements: |
|
1177 |
element.RefreshModel() |
|
1178 |
||
1179 |
def DeleteVariable(self, variable): |
|
1180 |
connectors = variable.GetConnectors() |
|
1181 |
if connectors["output"]: |
|
1182 |
elements = connectors["output"].GetConnectedBlocks() |
|
1183 |
else: |
|
1184 |
elements = [] |
|
1185 |
variable.Clean() |
|
1186 |
self.Blocks.remove(variable) |
|
1187 |
self.Elements.remove(variable) |
|
1188 |
self.Controler.RemoveCurrentElementEditingInstance(variable.GetId()) |
|
1189 |
for element in elements: |
|
1190 |
element.RefreshModel() |
|
1191 |
||
1192 |
def DeleteConnection(self, connection): |
|
1193 |
if connection.GetType() == CONTINUATION: |
|
1194 |
elements = connection.GetConnector().GetConnectedBlocks() |
|
1195 |
else: |
|
1196 |
elements = [] |
|
1197 |
connection.Clean() |
|
1198 |
self.Blocks.remove(connection) |
|
1199 |
self.Elements.remove(connection) |
|
1200 |
self.Controler.RemoveCurrentElementEditingInstance(connection.GetId()) |
|
1201 |
for element in elements: |
|
1202 |
element.RefreshModel() |
|
1203 |
||
1204 |
def DeleteComment(self, comment): |
|
1205 |
self.Elements.remove(comment) |
|
1206 |
self.Controler.RemoveCurrentElementEditingInstance(comment.GetId()) |
|
1207 |
||
1208 |
def DeleteWire(self, wire): |
|
1209 |
connected = wire.GetConnected() |
|
1210 |
wire.Clean() |
|
1211 |
self.Wires.remove(wire) |
|
1212 |
self.Elements.remove(wire) |
|
1213 |
for connector in connected: |
|
1214 |
connector.RefreshParentBlock() |
|
1215 |
||
1216 |
def DeleteContact(self, contact): |
|
1217 |
connectors = contact.GetConnectors() |
|
1218 |
elements = connectors["output"].GetConnectedBlocks() |
|
1219 |
contact.Clean() |
|
1220 |
self.Blocks.remove(contact) |
|
1221 |
self.Elements.remove(contact) |
|
1222 |
self.Controler.RemoveCurrentElementEditingInstance(contact.GetId()) |
|
1223 |
for element in elements: |
|
1224 |
element.RefreshModel() |
|
1225 |
||
1226 |
def DeleteCoil(self, coil): |
|
1227 |
connectors = coil.GetConnectors() |
|
1228 |
elements = connectors["output"].GetConnectedBlocks() |
|
1229 |
coil.Clean() |
|
1230 |
self.Blocks.remove(coil) |
|
1231 |
self.Elements.remove(coil) |
|
1232 |
self.Controler.RemoveCurrentElementEditingInstance(coil.GetId()) |
|
1233 |
for element in elements: |
|
1234 |
element.RefreshModel() |
|
1235 |
||
1236 |
def DeletePowerRail(self, powerrail): |
|
1237 |
elements = [] |
|
1238 |
if powerrail.GetType() == LEFTRAIL: |
|
1239 |
for connector in powerrail.GetConnectors(): |
|
1240 |
for element in connector.GetConnectedBlocks(): |
|
1241 |
if element not in elements: |
|
1242 |
elements.append(element) |
|
1243 |
powerrrail.Clean() |
|
1244 |
self.Blocks.remove(powerrrail) |
|
1245 |
self.Elements.remove(powerrrail) |
|
1246 |
self.Controler.RemoveCurrentElementEditingInstance(powerrrail.GetId()) |
|
1247 |
for element in elements: |
|
1248 |
element.RefreshModel() |
|
1249 |
||
1250 |
#------------------------------------------------------------------------------- |
|
1251 |
# Edit element content functions |
|
1252 |
#------------------------------------------------------------------------------- |
|
1253 |
||
1254 |
def EditBlockContent(self, block): |
|
1255 |
dialog = BlockPropertiesDialog(self.Parent) |
|
1256 |
dialog.SetBlockList(self.Controler.GetBlockTypes()) |
|
1257 |
dialog.SetMinBlockSize(block.GetSize()) |
|
1258 |
values = {"name" : block.GetName(), "type" : block.GetType()}
|
|
1259 |
values["extension"] = block.GetExtension() |
|
1260 |
dialog.SetValues(values) |
|
1261 |
if dialog.ShowModal() == wxID_OK: |
|
1262 |
values = dialog.GetValues() |
|
1263 |
if "name" in values: |
|
1264 |
block.SetName(values["name"]) |
|
1265 |
block.SetSize(values["width"], values["height"]) |
|
1266 |
block.SetType(values["type"], values["extension"]) |
|
1267 |
self.RefreshBlockModel(block) |
|
1268 |
self.Refresh() |
|
1269 |
dialog.Destroy() |
|
1270 |
||
1271 |
def EditVariableContent(self, variable): |
|
1272 |
dialog = VariablePropertiesDialog(self.Parent) |
|
1273 |
dialog.SetMinVariableSize(variable.GetSize()) |
|
1274 |
varlist = [] |
|
1275 |
vars = self.Controler.GetCurrentElementEditingInterfaceVars() |
|
1276 |
if vars: |
|
1277 |
for var in vars: |
|
1278 |
varlist.append((var["Name"], var["Class"], var["Type"])) |
|
1279 |
returntype = self.Controler.GetCurrentElementEditingInterfaceReturnType() |
|
1280 |
if returntype: |
|
1281 |
varlist.append((self.Controler.GetCurrentElementEditingName(), "Output", returntype)) |
|
1282 |
dialog.SetVariables(varlist) |
|
1283 |
values = {"name" : variable.GetName(), "type" : variable.GetType()}
|
|
1284 |
dialog.SetValues(values) |
|
1285 |
if dialog.ShowModal() == wxID_OK: |
|
1286 |
old_type = variable.GetType() |
|
1287 |
values = dialog.GetValues() |
|
1288 |
variable.SetName(values["name"]) |
|
1289 |
variable.SetType(values["type"], values["value_type"]) |
|
1290 |
variable.SetSize(values["width"], values["height"]) |
|
1291 |
if old_type != values["type"]: |
|
1292 |
id = variable.GetId() |
|
1293 |
self.Controler.RemoveCurrentElementEditingInstance(id) |
|
1294 |
self.Controler.AddCurrentElementEditingVariable(id, values["type"]) |
|
1295 |
self.RefreshVariableModel(variable) |
|
1296 |
self.Refresh() |
|
1297 |
dialog.Destroy() |
|
1298 |
||
1299 |
||
| 0 | 1300 |
#------------------------------------------------------------------------------- |
1301 |
# Editing functions |
|
1302 |
#------------------------------------------------------------------------------- |
|
1303 |
||
1304 |
def Cut(self): |
|
1305 |
pass |
|
1306 |
||
1307 |
def Copy(self): |
|
1308 |
pass |
|
1309 |
||
1310 |
def Paste(self): |
|
1311 |
pass |
|
1312 |
||
1313 |
#------------------------------------------------------------------------------- |
|
1314 |
# Drawing functions |
|
1315 |
#------------------------------------------------------------------------------- |
|
1316 |
||
| 27 | 1317 |
def OnMoveWindow(self, event): |
1318 |
self.RefreshScrollBar() |
|
1319 |
event.Skip() |
|
1320 |
||
| 0 | 1321 |
def OnPaint(self, event): |
| 27 | 1322 |
dc = self.GetLogicalDC() |
| 0 | 1323 |
dc.Clear() |
1324 |
dc.SetPen(wxPen(wxColour(230, 230, 230))) |
|
1325 |
if self.Scaling and self.DrawGrid: |
|
1326 |
width, height = dc.GetSize() |
|
1327 |
for i in xrange(1, width / self.Scaling[0] + 1): |
|
1328 |
dc.DrawLine(i * self.Scaling[0], 0, i * self.Scaling[0], height) |
|
1329 |
for i in xrange(1, height / self.Scaling[1] + 1): |
|
1330 |
dc.DrawLine(0, i * self.Scaling[1], width, i * self.Scaling[1]) |
|
1331 |
for wire in self.Wires: |
|
1332 |
if wire != self.SelectedElement: |
|
1333 |
wire.Draw(dc) |
|
1334 |
for element in self.Elements: |
|
1335 |
if element not in self.Wires and element != self.SelectedElement: |
|
1336 |
element.Draw(dc) |
|
1337 |
if self.SelectedElement: |
|
1338 |
self.SelectedElement.Draw(dc) |
|
| 27 | 1339 |
if self.rubberBand.IsShown(): |
1340 |
self.rubberBand.Draw() |
|
1341 |
event.Skip() |
|
1342 |
||
1343 |