author | lbessard |
Tue, 01 Jul 2008 10:33:29 +0200 | |
changeset 218 | 1b8e9bb83f25 |
parent 213 | 4931959ea256 |
child 231 | fc2d6cbb8b39 |
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 |
# |
|
58 | 7 |
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD |
0 | 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 |
|
58 | 19 |
#General Public License for more details. |
0 | 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 |
import wx |
|
26 |
||
27 |
from GraphicCommons import * |
|
28 |
from plcopen.structures import * |
|
99
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
29 |
from plcopen.structures import IsOfType |
0 | 30 |
|
31 |
#------------------------------------------------------------------------------- |
|
32 |
# Function Block Diagram Block |
|
33 |
#------------------------------------------------------------------------------- |
|
34 |
||
35 |
""" |
|
36 |
Class that implements the graphic representation of a function block |
|
37 |
""" |
|
38 |
||
39 |
class FBD_Block(Graphic_Element): |
|
40 |
||
41 |
# Create a new block |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
42 |
def __init__(self, parent, type, name, id = None, extension = 0, inputs = None, connectors = {}, executionOrder = 0): |
0 | 43 |
Graphic_Element.__init__(self, parent) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
44 |
self.Type = None |
27 | 45 |
self.Extension = None |
0 | 46 |
self.Id = id |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
47 |
self.SetName(name) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
48 |
self.SetExecutionOrder(executionOrder) |
2 | 49 |
self.Inputs = [] |
50 |
self.Outputs = [] |
|
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
51 |
self.Colour = wx.BLACK |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
52 |
self.Pen = wx.BLACK_PEN |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
53 |
self.SetType(type, extension, inputs, connectors) |
0 | 54 |
|
112 | 55 |
# Make a clone of this FBD_Block |
162 | 56 |
def Clone(self, parent, id = None, name = "", pos = None): |
144 | 57 |
if self.Name != "" and name == "": |
58 |
name = self.Name |
|
162 | 59 |
block = FBD_Block(parent, self.Type, name, id, self.Extension) |
112 | 60 |
block.SetSize(self.Size[0], self.Size[1]) |
61 |
if pos is not None: |
|
62 |
block.SetPosition(pos.x, pos.y) |
|
63 |
block.Inputs = [input.Clone(block) for input in self.Inputs] |
|
64 |
block.Outputs = [output.Clone(block) for output in self.Outputs] |
|
65 |
return block |
|
66 |
||
0 | 67 |
# Destructor |
68 |
def __del__(self): |
|
69 |
self.Inputs = [] |
|
70 |
self.Outputs = [] |
|
71 |
||
144 | 72 |
# Returns the RedrawRect |
73 |
def GetRedrawRect(self, movex = 0, movey = 0): |
|
74 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
|
75 |
if movex != 0 or movey != 0: |
|
76 |
for input in self.Inputs: |
|
77 |
if input.IsConnected(): |
|
78 |
rect = rect.Union(input.GetConnectedRedrawRect(movex, movey)) |
|
79 |
for output in self.Outputs: |
|
80 |
if output.IsConnected(): |
|
81 |
rect = rect.Union(output.GetConnectedRedrawRect(movex, movey)) |
|
82 |
return rect |
|
83 |
||
0 | 84 |
# Delete this block by calling the appropriate method |
85 |
def Delete(self): |
|
86 |
self.Parent.DeleteBlock(self) |
|
87 |
||
88 |
# Unconnect all inputs and outputs |
|
89 |
def Clean(self): |
|
90 |
for input in self.Inputs: |
|
2 | 91 |
input.UnConnect(delete = True) |
92 |
for output in self.Outputs: |
|
93 |
output.UnConnect(delete = True) |
|
0 | 94 |
|
42 | 95 |
# Refresh the size of text for name |
96 |
def RefreshNameSize(self): |
|
165 | 97 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
42 | 98 |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
99 |
# Refresh the size of text for execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
100 |
def RefreshExecutionOrderSize(self): |
165 | 101 |
self.ExecutionOrderSize = self.Parent.GetTextExtent(str(self.ExecutionOrder)) |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
102 |
|
0 | 103 |
# Refresh the block bounding box |
104 |
def RefreshBoundingBox(self): |
|
105 |
# Calculate the size of the name outside the block |
|
42 | 106 |
text_width, text_height = self.NameSize |
0 | 107 |
# Calculate the bounding box size |
108 |
bbx_x = self.Pos.x - max(min(1, len(self.Inputs)) * CONNECTOR_SIZE, (text_width - self.Size[0]) / 2) |
|
109 |
bbx_width = self.Size[0] + 1 + (min(1, len(self.Inputs)) + min(1, len(self.Outputs))) * CONNECTOR_SIZE |
|
110 |
if self.Name != "": |
|
111 |
bbx_y = self.Pos.y - (text_height + 2) |
|
112 |
bbx_height = self.Size[1] + (text_height + 2) |
|
113 |
else: |
|
114 |
bbx_y = self.Pos.y |
|
115 |
bbx_height = self.Size[1] |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
116 |
if self.ExecutionOrder != 0: |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
117 |
bbx_x = min(bbx_x, self.Pos.x + self.Size[0] - self.ExecutionOrderSize[0]) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
118 |
bbx_width = max(bbx_width, bbx_width + self.Pos.x + self.ExecutionOrderSize[0] - bbx_x - self.Size[0]) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
119 |
bbx_height = bbx_height + (self.ExecutionOrderSize[1] + 2) |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
120 |
self.BoundingBox = wx.Rect(bbx_x, bbx_y, bbx_width + 1, bbx_height + 1) |
0 | 121 |
|
122 |
# Refresh the positions of the block connectors |
|
123 |
def RefreshConnectors(self): |
|
145 | 124 |
scaling = self.Parent.GetScaling() |
0 | 125 |
# Calculate the size for the connector lines |
126 |
lines = max(len(self.Inputs), len(self.Outputs)) |
|
103 | 127 |
if lines > 0: |
128 |
linesize = max((self.Size[1] - BLOCK_LINE_SIZE) / lines, BLOCK_LINE_SIZE) |
|
145 | 129 |
# Update inputs and outputs positions |
103 | 130 |
position = BLOCK_LINE_SIZE + linesize / 2 |
145 | 131 |
for i in xrange(lines): |
132 |
if scaling is not None: |
|
133 |
ypos = round(float(self.Pos.y + position) / float(scaling[1])) * scaling[1] - self.Pos.y |
|
134 |
else: |
|
135 |
ypos = position |
|
136 |
if i < len(self.Inputs): |
|
137 |
self.Inputs[i].SetPosition(wx.Point(0, ypos)) |
|
138 |
if i < len(self.Outputs): |
|
139 |
self.Outputs[i].SetPosition(wx.Point(self.Size[0], ypos)) |
|
103 | 140 |
position += linesize |
0 | 141 |
self.RefreshConnected() |
142 |
||
143 |
# Refresh the positions of wires connected to inputs and outputs |
|
144 |
def RefreshConnected(self, exclude = []): |
|
145 |
for input in self.Inputs: |
|
146 |
input.MoveConnected(exclude) |
|
147 |
for output in self.Outputs: |
|
148 |
output.MoveConnected(exclude) |
|
149 |
||
150 |
# Returns the block connector that starts with the point given if it exists |
|
27 | 151 |
def GetConnector(self, position, name = None): |
152 |
# if a name is given |
|
153 |
if name: |
|
154 |
# Test each input and output connector |
|
155 |
for input in self.Inputs: |
|
156 |
if name == input.GetName(): |
|
157 |
return input |
|
158 |
for output in self.Outputs: |
|
159 |
if name == output.GetName(): |
|
160 |
return output |
|
0 | 161 |
# Test each input connector |
162 |
for input in self.Inputs: |
|
163 |
input_pos = input.GetRelPosition() |
|
164 |
if position.x == self.Pos.x + input_pos.x and position.y == self.Pos.y + input_pos.y: |
|
165 |
return input |
|
166 |
# Test each output connector |
|
167 |
for output in self.Outputs: |
|
168 |
output_pos = output.GetRelPosition() |
|
169 |
if position.x == self.Pos.x + output_pos.x and position.y == self.Pos.y + output_pos.y: |
|
170 |
return output |
|
171 |
return None |
|
172 |
||
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
173 |
def GetInputTypes(self): |
112 | 174 |
return tuple([input.GetType(True) for input in self.Inputs]) |
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
175 |
|
102
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
176 |
def SetOutputValues(self, values): |
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
177 |
for output in self.Outputs: |
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
178 |
output.SetValue(values.get(ouput.getName(), None)) |
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
179 |
|
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
180 |
def GetConnectionResultType(self, connector, connectortype): |
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
181 |
resulttype = connectortype |
99
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
182 |
for input in self.Inputs: |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
183 |
name = input.GetName() |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
184 |
if input != connector and (name.startswith("IN") or name in ["MN", "MX"]): |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
185 |
inputtype = input.GetConnectedType() |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
186 |
if resulttype is None or inputtype is not None and IsOfType(inputtype, resulttype): |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
187 |
resulttype = inputtype |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
188 |
for output in self.Outputs: |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
189 |
name = output.GetName() |
144 | 190 |
if output != connector and name == "OUT" and not IsEndType(output.GetType()): |
99
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
191 |
outputtype = output.GetConnectedType() |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
192 |
if resulttype is None or outputtype is not None and IsOfType(outputtype, resulttype): |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
193 |
resulttype = outputtype |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
194 |
return resulttype |
2b18a72dcaf0
Added support for standard functions type compatibility check
lbessard
parents:
90
diff
changeset
|
195 |
|
145 | 196 |
# Returns all the block connectors |
0 | 197 |
def GetConnectors(self): |
198 |
return {"inputs" : self.Inputs, "outputs" : self.Outputs} |
|
199 |
||
200 |
# Test if point given is on one of the block connectors |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
201 |
def TestConnector(self, pt, exclude = True): |
0 | 202 |
# Test each input connector |
203 |
for input in self.Inputs: |
|
204 |
if input.TestPoint(pt, exclude): |
|
205 |
return input |
|
206 |
# Test each output connector |
|
207 |
for output in self.Outputs: |
|
208 |
if output.TestPoint(pt, exclude): |
|
209 |
return output |
|
210 |
return None |
|
211 |
||
2 | 212 |
# Changes the block type |
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
213 |
def SetType(self, type, extension, inputs = None, connectors = {}): |
27 | 214 |
if type != self.Type or self.Extension != extension: |
42 | 215 |
if type != self.Type: |
216 |
self.Type = type |
|
165 | 217 |
self.TypeSize = self.Parent.GetTextExtent(self.Type) |
27 | 218 |
self.Extension = extension |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
219 |
# Find the block definition from type given and create the corresponding |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
220 |
# inputs and outputs |
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
221 |
blocktype = GetBlockType(type, inputs) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
222 |
if blocktype: |
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
223 |
self.Colour = wx.BLACK |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
224 |
inputs = [input for input in blocktype["inputs"]] |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
225 |
outputs = [output for output in blocktype["outputs"]] |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
226 |
if blocktype["extensible"]: |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
227 |
start = int(inputs[-1][0].replace("IN", "")) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
228 |
for i in xrange(self.Extension - len(blocktype["inputs"])): |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
229 |
start += 1 |
27 | 230 |
inputs.append(("IN%d"%start, inputs[-1][1], inputs[-1][2])) |
16 | 231 |
else: |
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
232 |
self.Colour = wx.RED |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
233 |
if "inputs" in connectors: |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
234 |
inputs = connectors["inputs"] |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
235 |
else: |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
236 |
inputs = [] |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
237 |
if "outputs" in connectors: |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
238 |
outputs = connectors["outputs"] |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
239 |
else: |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
240 |
outputs = [] |
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
241 |
self.Pen = wx.Pen(self.Colour) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
242 |
self.Clean() |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
243 |
# Extract the inputs properties and create the corresponding connector |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
244 |
self.Inputs = [] |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
245 |
for input_name, input_type, input_modifier in inputs: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
246 |
connector = Connector(self, input_name, input_type, wx.Point(0, 0), WEST, onlyone = True) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
247 |
if input_modifier == "negated": |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
248 |
connector.SetNegated(True) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
249 |
elif input_modifier != "none": |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
250 |
connector.SetEdge(input_modifier) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
251 |
self.Inputs.append(connector) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
252 |
# Extract the outputs properties and create the corresponding connector |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
253 |
self.Outputs = [] |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
254 |
for output_name, output_type, output_modifier in outputs: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
255 |
connector = Connector(self, output_name, output_type, wx.Point(0, 0), EAST) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
256 |
if output_modifier == "negated": |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
257 |
connector.SetNegated(True) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
258 |
elif output_modifier != "none": |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
259 |
connector.SetEdge(output_modifier) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
260 |
self.Outputs.append(connector) |
42 | 261 |
self.RefreshMinSize() |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
262 |
self.RefreshConnectors() |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
263 |
self.RefreshBoundingBox() |
2 | 264 |
|
0 | 265 |
# Returns the block type |
266 |
def GetType(self): |
|
267 |
return self.Type |
|
268 |
||
269 |
# Changes the block name |
|
270 |
def SetName(self, name): |
|
271 |
self.Name = name |
|
42 | 272 |
self.RefreshNameSize() |
0 | 273 |
|
274 |
# Returs the block name |
|
275 |
def GetName(self): |
|
276 |
return self.Name |
|
277 |
||
2 | 278 |
# Changes the extension name |
279 |
def SetExtension(self, extension): |
|
280 |
self.Extension = extension |
|
281 |
||
282 |
# Returs the extension name |
|
283 |
def GetExtension(self): |
|
284 |
return self.Extension |
|
285 |
||
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
286 |
# Changes the execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
287 |
def SetExecutionOrder(self, executionOrder): |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
288 |
self.ExecutionOrder = executionOrder |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
289 |
self.RefreshExecutionOrderSize() |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
290 |
|
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
291 |
# Returs the execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
292 |
def GetExecutionOrder(self): |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
293 |
return self.ExecutionOrder |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
294 |
|
42 | 295 |
# Refresh the block minimum size |
296 |
def RefreshMinSize(self): |
|
0 | 297 |
# Calculate the inputs maximum width |
298 |
max_input = 0 |
|
299 |
for input in self.Inputs: |
|
42 | 300 |
w, h = input.GetNameSize() |
0 | 301 |
max_input = max(max_input, w) |
302 |
# Calculate the outputs maximum width |
|
303 |
max_output = 0 |
|
304 |
for output in self.Outputs: |
|
42 | 305 |
w, h = output.GetNameSize() |
0 | 306 |
max_output = max(max_output, w) |
42 | 307 |
width = max(self.TypeSize[0] + 10, max_input + max_output + 15) |
0 | 308 |
height = (max(len(self.Inputs), len(self.Outputs)) + 1) * BLOCK_LINE_SIZE |
42 | 309 |
self.MinSize = width, height |
310 |
||
311 |
# Returns the block minimum size |
|
312 |
def GetMinSize(self): |
|
313 |
return self.MinSize |
|
0 | 314 |
|
315 |
# Changes the negated property of the connector handled |
|
316 |
def SetConnectorNegated(self, negated): |
|
317 |
handle_type, handle = self.Handle |
|
318 |
if handle_type == HANDLE_CONNECTOR: |
|
319 |
handle.SetNegated(negated) |
|
320 |
self.RefreshModel(False) |
|
321 |
||
322 |
# Changes the edge property of the connector handled |
|
323 |
def SetConnectorEdge(self, edge): |
|
324 |
handle_type, handle = self.Handle |
|
325 |
if handle_type == HANDLE_CONNECTOR: |
|
326 |
handle.SetEdge(edge) |
|
327 |
self.RefreshModel(False) |
|
328 |
||
2 | 329 |
# Method called when a LeftDClick event have been generated |
27 | 330 |
def OnLeftDClick(self, event, dc, scaling): |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
331 |
# Edit the block properties |
2 | 332 |
self.Parent.EditBlockContent(self) |
333 |
||
0 | 334 |
# Method called when a RightUp event have been generated |
27 | 335 |
def OnRightUp(self, event, dc, scaling): |
336 |
pos = GetScaledEventPosition(event, dc, scaling) |
|
0 | 337 |
# Popup the menu with special items for a block and a connector if one is handled |
338 |
connector = self.TestConnector(pos, False) |
|
339 |
if connector: |
|
340 |
self.Handle = (HANDLE_CONNECTOR, connector) |
|
341 |
self.Parent.PopupBlockMenu(connector) |
|
342 |
else: |
|
343 |
self.Parent.PopupBlockMenu() |
|
344 |
||
345 |
# Refreshes the block model |
|
346 |
def RefreshModel(self, move=True): |
|
347 |
self.Parent.RefreshBlockModel(self) |
|
348 |
# If block has moved, refresh the model of wires connected to outputs |
|
349 |
if move: |
|
350 |
for output in self.Outputs: |
|
351 |
output.RefreshWires() |
|
352 |
||
353 |
# Draws block |
|
354 |
def Draw(self, dc): |
|
144 | 355 |
Graphic_Element.Draw(self, dc) |
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
356 |
dc.SetPen(self.Pen) |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
357 |
dc.SetBrush(wx.WHITE_BRUSH) |
90
2245e8776086
Adding support support for using PLCOpenEditor with Beremiz
lbessard
parents:
64
diff
changeset
|
358 |
dc.SetTextForeground(self.Colour) |
213 | 359 |
|
360 |
if getattr(dc, "printing", False): |
|
361 |
name_size = dc.GetTextExtent(self.Name) |
|
362 |
type_size = dc.GetTextExtent(self.Type) |
|
363 |
executionorder_size = dc.GetTextExtent(str(self.ExecutionOrder)) |
|
364 |
else: |
|
365 |
name_size = self.NameSize |
|
366 |
type_size = self.TypeSize |
|
367 |
executionorder_size = self.ExecutionOrderSize |
|
368 |
||
0 | 369 |
# Draw a rectangle with the block size |
370 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
371 |
# Draw block name and block type |
|
213 | 372 |
dc.DrawText(self.Name, self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
373 |
self.Pos.y - (name_size[1] + 2)) |
|
374 |
dc.DrawText(self.Type, self.Pos.x + (self.Size[0] - type_size[0]) / 2, |
|
0 | 375 |
self.Pos.y + 5) |
376 |
# Draw inputs and outputs connectors |
|
377 |
for input in self.Inputs: |
|
378 |
input.Draw(dc) |
|
379 |
for output in self.Outputs: |
|
380 |
output.Draw(dc) |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
381 |
if self.ExecutionOrder != 0: |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
382 |
# Draw block execution order |
213 | 383 |
dc.DrawText(str(self.ExecutionOrder), self.Pos.x + self.Size[0] - executionorder_size[0], |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
384 |
self.Pos.y + self.Size[1] + 2) |
140
06d28f03f6f4
Adding highlighting on group or element when mouse is over
lbessard
parents:
118
diff
changeset
|
385 |
dc.SetTextForeground(wx.BLACK) |
06d28f03f6f4
Adding highlighting on group or element when mouse is over
lbessard
parents:
118
diff
changeset
|
386 |
|
0 | 387 |
|
388 |
#------------------------------------------------------------------------------- |
|
389 |
# Function Block Diagram Variable |
|
390 |
#------------------------------------------------------------------------------- |
|
391 |
||
392 |
""" |
|
393 |
Class that implements the graphic representation of a variable |
|
394 |
""" |
|
395 |
||
396 |
class FBD_Variable(Graphic_Element): |
|
397 |
||
398 |
# Create a new variable |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
399 |
def __init__(self, parent, type, name, value_type, id = None, executionOrder = 0): |
0 | 400 |
Graphic_Element.__init__(self, parent) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
401 |
self.Type = None |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
402 |
self.ValueType = None |
0 | 403 |
self.Id = id |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
404 |
self.SetName(name) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
405 |
self.SetExecutionOrder(executionOrder) |
0 | 406 |
self.Input = None |
407 |
self.Output = None |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
408 |
self.SetType(type, value_type) |
0 | 409 |
|
112 | 410 |
# Make a clone of this FBD_Variable |
162 | 411 |
def Clone(self, parent, id = None, pos = None): |
412 |
variable = FBD_Variable(parent, self.Type, self.Name, self.ValueType, id) |
|
112 | 413 |
variable.SetSize(self.Size[0], self.Size[1]) |
414 |
if pos is not None: |
|
415 |
variable.SetPosition(pos.x, pos.y) |
|
144 | 416 |
if self.Input: |
417 |
variable.Input = self.Input.Clone(variable) |
|
418 |
if self.Output: |
|
419 |
variable.Output = self.Output.Clone(variable) |
|
112 | 420 |
return variable |
421 |
||
0 | 422 |
# Destructor |
423 |
def __del__(self): |
|
424 |
self.Input = None |
|
425 |
self.Output = None |
|
426 |
||
144 | 427 |
# Returns the RedrawRect |
428 |
def GetRedrawRect(self, movex = 0, movey = 0): |
|
429 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
|
430 |
if movex != 0 or movey != 0: |
|
431 |
if self.Input and self.Input.IsConnected(): |
|
432 |
rect = rect.Union(self.Input.GetConnectedRedrawRect(movex, movey)) |
|
433 |
if self.Output and self.Output.IsConnected(): |
|
434 |
rect = rect.Union(self.Output.GetConnectedRedrawRect(movex, movey)) |
|
435 |
return rect |
|
436 |
||
0 | 437 |
# Unconnect connector |
438 |
def Clean(self): |
|
439 |
if self.Input: |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
440 |
self.Input.UnConnect(delete = True) |
0 | 441 |
if self.Output: |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
442 |
self.Output.UnConnect(delete = True) |
0 | 443 |
|
444 |
# Delete this variable by calling the appropriate method |
|
445 |
def Delete(self): |
|
446 |
self.Parent.DeleteVariable(self) |
|
447 |
||
42 | 448 |
# Refresh the size of text for name |
449 |
def RefreshNameSize(self): |
|
165 | 450 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
42 | 451 |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
452 |
# Refresh the size of text for execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
453 |
def RefreshExecutionOrderSize(self): |
165 | 454 |
self.ExecutionOrderSize = self.Parent.GetTextExtent(str(self.ExecutionOrder)) |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
455 |
|
0 | 456 |
# Refresh the variable bounding box |
457 |
def RefreshBoundingBox(self): |
|
458 |
if self.Type in (OUTPUT, INOUT): |
|
459 |
bbx_x = self.Pos.x - CONNECTOR_SIZE |
|
460 |
else: |
|
461 |
bbx_x = self.Pos.x |
|
462 |
if self.Type == INOUT: |
|
463 |
bbx_width = self.Size[0] + 2 * CONNECTOR_SIZE |
|
464 |
else: |
|
465 |
bbx_width = self.Size[0] + CONNECTOR_SIZE |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
466 |
bbx_height = self.Size[1] |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
467 |
if self.ExecutionOrder != 0: |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
468 |
bbx_x = min(bbx_x, self.Pos.x + self.Size[0] - self.ExecutionOrderSize[0]) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
469 |
bbx_width = max(bbx_width, bbx_width + self.Pos.x + self.ExecutionOrderSize[0] - bbx_x - self.Size[0]) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
470 |
bbx_height = bbx_height + (self.ExecutionOrderSize[1] + 2) |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
471 |
self.BoundingBox = wx.Rect(bbx_x, self.Pos.y, bbx_width + 1, bbx_height + 1) |
0 | 472 |
|
473 |
# Refresh the position of the variable connector |
|
474 |
def RefreshConnectors(self): |
|
145 | 475 |
scaling = self.Parent.GetScaling() |
476 |
if scaling is not None: |
|
477 |
position = round(float(self.Pos.y + self.Size[1] / 2) / float(scaling[1])) * scaling[1] - self.Pos.y |
|
478 |
else: |
|
479 |
position = self.Size[1] / 2 |
|
0 | 480 |
if self.Input: |
145 | 481 |
self.Input.SetPosition(wx.Point(0, position)) |
0 | 482 |
if self.Output: |
145 | 483 |
self.Output.SetPosition(wx.Point(self.Size[0], position)) |
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
484 |
self.RefreshConnected() |
0 | 485 |
|
486 |
# Refresh the position of wires connected to connector |
|
487 |
def RefreshConnected(self, exclude = []): |
|
488 |
if self.Input: |
|
489 |
self.Input.MoveConnected(exclude) |
|
490 |
if self.Output: |
|
491 |
self.Output.MoveConnected(exclude) |
|
492 |
||
493 |
# Test if point given is on the variable connector |
|
494 |
def TestConnector(self, pt, exclude=True): |
|
495 |
if self.Input and self.Input.TestPoint(pt, exclude): |
|
496 |
return self.Input |
|
497 |
if self.Output and self.Output.TestPoint(pt, exclude): |
|
498 |
return self.Output |
|
499 |
return None |
|
500 |
||
501 |
# Returns the block connector that starts with the point given if it exists |
|
27 | 502 |
def GetConnector(self, position, name = None): |
503 |
# if a name is given |
|
504 |
if name: |
|
505 |
# Test input and output connector if they exists |
|
506 |
if self.Input and name == self.Input.GetName(): |
|
507 |
return self.Input |
|
508 |
if self.Output and name == self.Output.GetName(): |
|
509 |
return self.Output |
|
0 | 510 |
# Test input connector if it exists |
511 |
if self.Input: |
|
512 |
input_pos = self.Input.GetRelPosition() |
|
513 |
if position.x == self.Pos.x + input_pos.x and position.y == self.Pos.y + input_pos.y: |
|
514 |
return self.Input |
|
515 |
# Test output connector if it exists |
|
516 |
if self.Output: |
|
517 |
output_pos = self.Output.GetRelPosition() |
|
518 |
if position.x == self.Pos.x + output_pos.x and position.y == self.Pos.y + output_pos.y: |
|
519 |
return self.Output |
|
520 |
return None |
|
521 |
||
522 |
# Returns all the block connectors |
|
523 |
def GetConnectors(self): |
|
524 |
return {"input" : self.Input, "output" : self.Output} |
|
525 |
||
526 |
# Changes the negated property of the variable connector if handled |
|
527 |
def SetConnectorNegated(self, negated): |
|
528 |
handle_type, handle = self.Handle |
|
529 |
if handle_type == HANDLE_CONNECTOR: |
|
530 |
handle.SetNegated(negated) |
|
531 |
self.RefreshModel(False) |
|
532 |
||
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
533 |
# Changes the variable type |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
534 |
def SetType(self, type, value_type): |
32
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
535 |
if type != self.Type: |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
536 |
self.Type = type |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
537 |
self.Clean() |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
538 |
self.Input = None |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
539 |
self.Output = None |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
540 |
# Create an input or output connector according to variable type |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
541 |
if self.Type != INPUT: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
542 |
self.Input = Connector(self, "", value_type, wx.Point(0, 0), WEST, onlyone = True) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
543 |
if self.Type != OUTPUT: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
544 |
self.Output = Connector(self, "", value_type, wx.Point(0, 0), EAST) |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
545 |
self.RefreshConnectors() |
32
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
546 |
elif value_type != self.ValueType: |
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
547 |
if self.Input: |
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
548 |
self.Input.SetType(value_type) |
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
549 |
if self.Output: |
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
550 |
self.Output.SetType(value_type) |
cf9efccff009
FBD_Variable don't remove wire when just expression changed
lbessard
parents:
28
diff
changeset
|
551 |
self.RefreshConnectors() |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
552 |
|
0 | 553 |
# Returns the variable type |
554 |
def GetType(self): |
|
555 |
return self.Type |
|
556 |
||
557 |
# Changes the variable name |
|
558 |
def SetName(self, name): |
|
559 |
self.Name = name |
|
42 | 560 |
self.RefreshNameSize() |
0 | 561 |
|
562 |
# Returns the variable name |
|
563 |
def GetName(self): |
|
564 |
return self.Name |
|
565 |
||
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
566 |
# Changes the execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
567 |
def SetExecutionOrder(self, executionOrder): |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
568 |
self.ExecutionOrder = executionOrder |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
569 |
self.RefreshExecutionOrderSize() |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
570 |
|
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
571 |
# Returs the execution order |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
572 |
def GetExecutionOrder(self): |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
573 |
return self.ExecutionOrder |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
574 |
|
0 | 575 |
# Returns the variable minimum size |
576 |
def GetMinSize(self): |
|
42 | 577 |
return self.NameSize[0] + 10, self.NameSize[1] + 10 |
0 | 578 |
|
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
579 |
# Method called when a LeftDClick event have been generated |
27 | 580 |
def OnLeftDClick(self, event, dc, scaling): |
3
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
581 |
# Edit the variable properties |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
582 |
self.Parent.EditVariableContent(self) |
86ccc89d7b0b
FBD Blocks and Variables can now be modified and wires can't be unconnected on both sides
lbessard
parents:
2
diff
changeset
|
583 |
|
0 | 584 |
# Method called when a RightUp event have been generated |
27 | 585 |
def OnRightUp(self, event, dc, scaling): |
102
85875dcb7754
Adding edit user's POU by double click on block instance
lbessard
parents:
99
diff
changeset
|
586 |
self.Parent.PopupDefaultMenu() |
0 | 587 |
|
588 |
# Refreshes the variable model |
|
589 |
def RefreshModel(self, move=True): |
|
590 |
self.Parent.RefreshVariableModel(self) |
|
591 |
# If variable has moved and variable is not of type OUTPUT, refresh the model |
|
592 |
# of wires connected to output connector |
|
593 |
if move and self.Type != OUTPUT: |
|
594 |
if self.Output: |
|
595 |
self.Output.RefreshWires() |
|
596 |
||
597 |
# Draws variable |
|
598 |
def Draw(self, dc): |
|
144 | 599 |
Graphic_Element.Draw(self, dc) |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
600 |
dc.SetPen(wx.BLACK_PEN) |
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
601 |
dc.SetBrush(wx.WHITE_BRUSH) |
213 | 602 |
|
603 |
if getattr(dc, "printing", False): |
|
604 |
name_size = dc.GetTextExtent(self.Name) |
|
605 |
executionorder_size = dc.GetTextExtent(str(self.ExecutionOrder)) |
|
606 |
else: |
|
607 |
name_size = self.NameSize |
|
608 |
executionorder_size = self.ExecutionOrderSize |
|
609 |
||
0 | 610 |
# Draw a rectangle with the variable size |
611 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
612 |
# Draw variable name |
|
213 | 613 |
dc.DrawText(self.Name, self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
614 |
self.Pos.y + (self.Size[1] - name_size[1]) / 2) |
|
0 | 615 |
# Draw connectors |
616 |
if self.Input: |
|
617 |
self.Input.Draw(dc) |
|
618 |
if self.Output: |
|
619 |
self.Output.Draw(dc) |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
620 |
if self.ExecutionOrder != 0: |
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
621 |
# Draw variable execution order |
213 | 622 |
dc.DrawText(str(self.ExecutionOrder), self.Pos.x + self.Size[0] - executionorder_size[0], |
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
623 |
self.Pos.y + self.Size[1] + 2) |
0 | 624 |
|
625 |
||
626 |
#------------------------------------------------------------------------------- |
|
627 |
# Function Block Diagram Connector |
|
628 |
#------------------------------------------------------------------------------- |
|
629 |
||
630 |
""" |
|
631 |
Class that implements the graphic representation of a connection |
|
632 |
""" |
|
633 |
||
634 |
class FBD_Connector(Graphic_Element): |
|
635 |
||
636 |
# Create a new connection |
|
637 |
def __init__(self, parent, type, name, id = None): |
|
638 |
Graphic_Element.__init__(self, parent) |
|
639 |
self.Type = type |
|
640 |
self.Id = id |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
641 |
self.SetName(name) |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
642 |
self.Pos = wx.Point(0, 0) |
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
643 |
self.Size = wx.Size(0, 0) |
0 | 644 |
# Create an input or output connector according to connection type |
645 |
if self.Type == CONNECTOR: |
|
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
646 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), WEST, onlyone = True) |
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
647 |
else: |
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
648 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), EAST) |
0 | 649 |
self.RefreshConnectors() |
42 | 650 |
self.RefreshNameSize() |
0 | 651 |
|
652 |
# Destructor |
|
653 |
def __del__(self): |
|
654 |
self.Connector = None |
|
655 |
||
144 | 656 |
# Returns the RedrawRect |
657 |
def GetRedrawRect(self, movex = 0, movey = 0): |
|
658 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
|
659 |
if movex != 0 or movey != 0: |
|
660 |
if self.Connector and self.Connector.IsConnected(): |
|
661 |
rect = rect.Union(self.Connector.GetConnectedRedrawRect(movex, movey)) |
|
662 |
return rect |
|
663 |
||
112 | 664 |
# Make a clone of this FBD_Connector |
162 | 665 |
def Clone(self, parent, id = None, pos = None): |
666 |
connection = FBD_Connector(parent, self.Type, self.Name, id) |
|
112 | 667 |
connection.SetSize(self.Size[0], self.Size[1]) |
668 |
if pos is not None: |
|
669 |
connection.SetPosition(pos.x, pos.y) |
|
670 |
connection.Connector = self.Connector.Clone(connection) |
|
671 |
return connection |
|
672 |
||
0 | 673 |
# Unconnect connector |
674 |
def Clean(self): |
|
675 |
if self.Connector: |
|
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
676 |
self.Connector.UnConnect(delete = True) |
0 | 677 |
|
678 |
# Delete this connection by calling the appropriate method |
|
679 |
def Delete(self): |
|
680 |
self.Parent.DeleteConnection(self) |
|
681 |
||
42 | 682 |
# Refresh the size of text for name |
683 |
def RefreshNameSize(self): |
|
165 | 684 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
42 | 685 |
|
0 | 686 |
# Refresh the connection bounding box |
687 |
def RefreshBoundingBox(self): |
|
688 |
if self.Type == CONNECTOR: |
|
689 |
bbx_x = self.Pos.x - CONNECTOR_SIZE |
|
690 |
else: |
|
691 |
bbx_x = self.Pos.x |
|
692 |
bbx_width = self.Size[0] + CONNECTOR_SIZE |
|
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
693 |
self.BoundingBox = wx.Rect(bbx_x, self.Pos.y, bbx_width, self.Size[1]) |
0 | 694 |
|
695 |
# Refresh the position of the connection connector |
|
696 |
def RefreshConnectors(self): |
|
145 | 697 |
scaling = self.Parent.GetScaling() |
698 |
if scaling is not None: |
|
699 |
position = round(float(self.Pos.y + self.Size[1] / 2) / float(scaling[1])) * scaling[1] - self.Pos.y |
|
700 |
else: |
|
701 |
position = self.Size[1] / 2 |
|
0 | 702 |
if self.Type == CONNECTOR: |
145 | 703 |
self.Connector.SetPosition(wx.Point(0, position)) |
704 |
else: |
|
705 |
self.Connector.SetPosition(wx.Point(self.Size[0], position)) |
|
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
706 |
self.RefreshConnected() |
0 | 707 |
|
708 |
# Refresh the position of wires connected to connector |
|
709 |
def RefreshConnected(self, exclude = []): |
|
710 |
if self.Connector: |
|
711 |
self.Connector.MoveConnected(exclude) |
|
712 |
||
713 |
# Test if point given is on the connection connector |
|
714 |
def TestConnector(self, pt, exclude=True): |
|
715 |
if self.Connector and self.Connector.TestPoint(pt, exclude): |
|
716 |
return self.Connector |
|
717 |
return None |
|
718 |
||
719 |
# Returns the connection connector |
|
27 | 720 |
def GetConnector(self, position = None, name = None): |
0 | 721 |
return self.Connector |
722 |
||
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
723 |
# Changes the variable type |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
724 |
def SetType(self, type): |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
725 |
if type != self.Type: |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
726 |
self.Type = type |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
727 |
self.Clean() |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
728 |
# Create an input or output connector according to connection type |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
729 |
if self.Type == CONNECTOR: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
730 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), WEST, onlyone = True) |
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
731 |
else: |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
732 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), EAST) |
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
733 |
self.RefreshConnectors() |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
734 |
|
0 | 735 |
# Returns the connection type |
736 |
def GetType(self): |
|
737 |
return self.Type |
|
738 |
||
145 | 739 |
def GetConnectionResultType(self, connector, connectortype): |
740 |
return connectortype |
|
741 |
||
0 | 742 |
# Changes the connection name |
743 |
def SetName(self, name): |
|
744 |
self.Name = name |
|
42 | 745 |
self.RefreshNameSize() |
0 | 746 |
|
747 |
# Returns the connection name |
|
748 |
def GetName(self): |
|
749 |
return self.Name |
|
750 |
||
751 |
# Returns the connection minimum size |
|
752 |
def GetMinSize(self): |
|
42 | 753 |
text_width, text_height = self.NameSize |
0 | 754 |
if text_height % 2 == 1: |
755 |
text_height += 1 |
|
756 |
return text_width + text_height + 20, text_height + 10 |
|
757 |
||
28
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
758 |
# Method called when a LeftDClick event have been generated |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
759 |
def OnLeftDClick(self, event, dc, scaling): |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
760 |
# Edit the connection properties |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
761 |
self.Parent.EditConnectionContent(self) |
fc23e1f415d8
Adding support for concurrent overriden standard function
lbessard
parents:
27
diff
changeset
|
762 |
|
0 | 763 |
# Method called when a RightUp event have been generated |
27 | 764 |
def OnRightUp(self, event, dc, scaling): |
0 | 765 |
# Popup the default menu |
766 |
self.Parent.PopupDefaultMenu() |
|
767 |
||
768 |
# Refreshes the connection model |
|
769 |
def RefreshModel(self, move=True): |
|
770 |
self.Parent.RefreshConnectionModel(self) |
|
771 |
# If connection has moved and connection is of type CONTINUATION, refresh |
|
772 |
# the model of wires connected to connector |
|
773 |
if move and self.Type == CONTINUATION: |
|
774 |
if self.Connector: |
|
775 |
self.Connector.RefreshWires() |
|
776 |
||
777 |
# Draws connection |
|
778 |
def Draw(self, dc): |
|
144 | 779 |
Graphic_Element.Draw(self, dc) |
64
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
780 |
dc.SetPen(wx.BLACK_PEN) |
dd6f693e46a1
Cleaning code for using only wxPython 2.6 class naming
lbessard
parents:
58
diff
changeset
|
781 |
dc.SetBrush(wx.WHITE_BRUSH) |
213 | 782 |
|
783 |
if getattr(dc, "printing", False): |
|
784 |
name_size = dc.GetTextExtent(self.Name) |
|
785 |
else: |
|
786 |
name_size = self.NameSize |
|
787 |
||
0 | 788 |
# Draw a rectangle with the connection size with arrows in |
789 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
213 | 790 |
arrowsize = min(self.Size[1] / 2, (self.Size[0] - name_size[0] - 10) / 2) |
0 | 791 |
dc.DrawLine(self.Pos.x, self.Pos.y, self.Pos.x + arrowsize, |
792 |
self.Pos.y + self.Size[1] / 2) |
|
793 |
dc.DrawLine(self.Pos.x + arrowsize, self.Pos.y + self.Size[1] / 2, |
|
794 |
self.Pos.x, self.Pos.y + self.Size[1]) |
|
795 |
dc.DrawLine(self.Pos.x + self.Size[0] - arrowsize, self.Pos.y, |
|
796 |
self.Pos.x + self.Size[0], self.Pos.y + self.Size[1] / 2) |
|
797 |
dc.DrawLine(self.Pos.x + self.Size[0], self.Pos.y + self.Size[1] / 2, |
|
798 |
self.Pos.x + self.Size[0] - arrowsize, self.Pos.y + self.Size[1]) |
|
118
0c53d6a36013
Add support for defining execution order in FBD networks (related ST code not generated yet)
lbessard
parents:
112
diff
changeset
|
799 |
# Draw connection name |
213 | 800 |
dc.DrawText(self.Name, self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
801 |
self.Pos.y + (self.Size[1] - name_size[1]) / 2) |
|
0 | 802 |
# Draw connector |
803 |
if self.Connector: |
|
804 |
self.Connector.Draw(dc) |
|
140
06d28f03f6f4
Adding highlighting on group or element when mouse is over
lbessard
parents:
118
diff
changeset
|
805 |