author | Andrey Skvortsov <andrej.skvortzov@gmail.com> |
Wed, 03 Oct 2018 14:19:41 +0300 | |
changeset 2425 | 68e7da937162 |
parent 1881 | 091005ec69c4 |
child 2432 | dbc065a2f7a5 |
permissions | -rw-r--r-- |
814 | 1 |
#!/usr/bin/env python |
2 |
# -*- coding: utf-8 -*- |
|
3 |
||
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
4 |
# This file is part of Beremiz, a Integrated Development Environment for |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
5 |
# programming IEC 61131-3 automates supporting plcopen standard and CanFestival. |
814 | 6 |
# |
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
7 |
# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD |
814 | 8 |
# |
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
9 |
# See COPYING file for copyrights details. |
814 | 10 |
# |
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
11 |
# This program is free software; you can redistribute it and/or |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
12 |
# modify it under the terms of the GNU General Public License |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
13 |
# as published by the Free Software Foundation; either version 2 |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
14 |
# of the License, or (at your option) any later version. |
814 | 15 |
# |
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
16 |
# This program is distributed in the hope that it will be useful, |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
17 |
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
18 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
19 |
# GNU General Public License for more details. |
814 | 20 |
# |
1571
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
21 |
# You should have received a copy of the GNU General Public License |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
22 |
# along with this program; if not, write to the Free Software |
486f94a8032c
fix license notices in source files and license files under GPLv2+
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1176
diff
changeset
|
23 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
814 | 24 |
|
1881
091005ec69c4
fix pylint py3k conversion warning: "(no-absolute-import) import missing `from __future__ import absolute_import`"
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1872
diff
changeset
|
25 |
|
091005ec69c4
fix pylint py3k conversion warning: "(no-absolute-import) import missing `from __future__ import absolute_import`"
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1872
diff
changeset
|
26 |
from __future__ import absolute_import |
814 | 27 |
import wx |
28 |
||
1176
f4b434672204
Moved and rewrote DebugViewer and DebusDataConsumer classes
Laurent Bessard
parents:
1133
diff
changeset
|
29 |
from graphics.GraphicCommons import * |
814 | 30 |
from plcopen.structures import * |
31 |
||
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
32 |
|
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
33 |
# ------------------------------------------------------------------------------- |
814 | 34 |
# Function Block Diagram Block |
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
35 |
# ------------------------------------------------------------------------------- |
814 | 36 |
|
37 |
||
38 |
def TestConnectorName(name, block_type): |
|
39 |
return name in ["OUT", "MN", "MX"] or name.startswith("IN") and (block_type, name) != ("EXPT", "IN2") |
|
40 |
||
1736
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
41 |
|
814 | 42 |
class FBD_Block(Graphic_Element): |
1736
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
43 |
""" |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
44 |
Class that implements the graphic representation of a function block |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
45 |
""" |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
46 |
|
814 | 47 |
# Create a new block |
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
48 |
def __init__(self, parent, type, name, id=None, extension=0, inputs=None, connectors=None, executionControl=False, executionOrder=0): |
814 | 49 |
Graphic_Element.__init__(self, parent) |
50 |
self.Type = None |
|
51 |
self.Description = None |
|
52 |
self.Extension = None |
|
53 |
self.ExecutionControl = False |
|
54 |
self.Id = id |
|
55 |
self.SetName(name) |
|
56 |
self.SetExecutionOrder(executionOrder) |
|
57 |
self.Inputs = [] |
|
58 |
self.Outputs = [] |
|
59 |
self.Colour = wx.BLACK |
|
60 |
self.Pen = MiterPen(wx.BLACK) |
|
61 |
self.SetType(type, extension, inputs, connectors, executionControl) |
|
62 |
self.Highlights = {} |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
63 |
|
814 | 64 |
# Make a clone of this FBD_Block |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
65 |
def Clone(self, parent, id=None, name="", pos=None): |
814 | 66 |
if self.Name != "" and name == "": |
67 |
name = self.Name |
|
68 |
block = FBD_Block(parent, self.Type, name, id, self.Extension) |
|
69 |
block.SetSize(self.Size[0], self.Size[1]) |
|
70 |
if pos is not None: |
|
71 |
block.SetPosition(pos.x, pos.y) |
|
72 |
else: |
|
73 |
block.SetPosition(self.Pos.x, self.Pos.y) |
|
74 |
block.Inputs = [input.Clone(block) for input in self.Inputs] |
|
75 |
block.Outputs = [output.Clone(block) for output in self.Outputs] |
|
76 |
return block |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
77 |
|
814 | 78 |
def GetConnectorTranslation(self, element): |
79 |
return dict(zip(self.Inputs + self.Outputs, element.Inputs + element.Outputs)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
80 |
|
814 | 81 |
def Flush(self): |
82 |
for input in self.Inputs: |
|
83 |
input.Flush() |
|
84 |
self.Inputs = [] |
|
85 |
for output in self.Outputs: |
|
86 |
output.Flush() |
|
87 |
self.Outputs = [] |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
88 |
|
814 | 89 |
# Returns the RedrawRect |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
90 |
def GetRedrawRect(self, movex=0, movey=0): |
814 | 91 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
92 |
if movex != 0 or movey != 0: |
|
93 |
for input in self.Inputs: |
|
94 |
if input.IsConnected(): |
|
95 |
rect = rect.Union(input.GetConnectedRedrawRect(movex, movey)) |
|
96 |
for output in self.Outputs: |
|
97 |
if output.IsConnected(): |
|
98 |
rect = rect.Union(output.GetConnectedRedrawRect(movex, movey)) |
|
99 |
return rect |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
100 |
|
814 | 101 |
# Delete this block by calling the appropriate method |
102 |
def Delete(self): |
|
103 |
self.Parent.DeleteBlock(self) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
104 |
|
814 | 105 |
# Unconnect all inputs and outputs |
106 |
def Clean(self): |
|
107 |
for input in self.Inputs: |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
108 |
input.UnConnect(delete=True) |
814 | 109 |
for output in self.Outputs: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
110 |
output.UnConnect(delete=True) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
111 |
|
814 | 112 |
# Refresh the size of text for name |
113 |
def RefreshNameSize(self): |
|
114 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
115 |
|
814 | 116 |
# Refresh the size of text for execution order |
117 |
def RefreshExecutionOrderSize(self): |
|
118 |
self.ExecutionOrderSize = self.Parent.GetTextExtent(str(self.ExecutionOrder)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
119 |
|
814 | 120 |
# Returns if the point given is in the bounding box |
121 |
def HitTest(self, pt, connectors=True): |
|
122 |
if self.Name != "": |
|
123 |
test_text = self.GetTextBoundingBox().InsideXY(pt.x, pt.y) |
|
124 |
else: |
|
125 |
test_text = False |
|
126 |
test_block = self.GetBlockBoundingBox(connectors).InsideXY(pt.x, pt.y) |
|
127 |
return test_text or test_block |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
128 |
|
814 | 129 |
# Returns the bounding box of the name outside the block |
130 |
def GetTextBoundingBox(self): |
|
131 |
# Calculate the size of the name outside the block |
|
132 |
text_width, text_height = self.NameSize |
|
133 |
return wx.Rect(self.Pos.x + (self.Size[0] - text_width) / 2, |
|
134 |
self.Pos.y - (text_height + 2), |
|
135 |
text_width, |
|
136 |
text_height) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
137 |
|
814 | 138 |
# Returns the bounding box of function block without name outside |
139 |
def GetBlockBoundingBox(self, connectors=True): |
|
140 |
bbx_x, bbx_y = self.Pos.x, self.Pos.y |
|
141 |
bbx_width, bbx_height = self.Size |
|
142 |
if connectors: |
|
143 |
bbx_x -= min(1, len(self.Inputs)) * CONNECTOR_SIZE |
|
144 |
bbx_width += (min(1, len(self.Inputs)) + min(1, len(self.Outputs))) * CONNECTOR_SIZE |
|
145 |
if self.ExecutionOrder != 0: |
|
146 |
bbx_x = min(bbx_x, self.Pos.x + self.Size[0] - self.ExecutionOrderSize[0]) |
|
147 |
bbx_width = max(bbx_width, bbx_width + self.Pos.x + self.ExecutionOrderSize[0] - bbx_x - self.Size[0]) |
|
148 |
bbx_height = bbx_height + (self.ExecutionOrderSize[1] + 2) |
|
149 |
return wx.Rect(bbx_x, bbx_y, bbx_width + 1, bbx_height + 1) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
150 |
|
814 | 151 |
# Refresh the block bounding box |
152 |
def RefreshBoundingBox(self): |
|
153 |
self.BoundingBox = self.GetBlockBoundingBox() |
|
154 |
if self.Name != "": |
|
155 |
self.BoundingBox.Union(self.GetTextBoundingBox()) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
156 |
|
814 | 157 |
# Refresh the positions of the block connectors |
158 |
def RefreshConnectors(self): |
|
159 |
scaling = self.Parent.GetScaling() |
|
160 |
# Calculate the size for the connector lines |
|
161 |
lines = max(len(self.Inputs), len(self.Outputs)) |
|
162 |
if lines > 0: |
|
163 |
linesize = max((self.Size[1] - BLOCK_LINE_SIZE) / lines, BLOCK_LINE_SIZE) |
|
164 |
# Update inputs and outputs positions |
|
165 |
position = BLOCK_LINE_SIZE + linesize / 2 |
|
166 |
for i in xrange(lines): |
|
167 |
if scaling is not None: |
|
168 |
ypos = round_scaling(self.Pos.y + position, scaling[1]) - self.Pos.y |
|
169 |
else: |
|
170 |
ypos = position |
|
171 |
if i < len(self.Inputs): |
|
172 |
self.Inputs[i].SetPosition(wx.Point(0, ypos)) |
|
173 |
if i < len(self.Outputs): |
|
174 |
self.Outputs[i].SetPosition(wx.Point(self.Size[0], ypos)) |
|
175 |
position += linesize |
|
176 |
self.RefreshConnected() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
177 |
|
814 | 178 |
# Refresh the positions of wires connected to inputs and outputs |
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
179 |
def RefreshConnected(self, exclude=None): |
814 | 180 |
for input in self.Inputs: |
181 |
input.MoveConnected(exclude) |
|
182 |
for output in self.Outputs: |
|
183 |
output.MoveConnected(exclude) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
184 |
|
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
185 |
# Returns the block connector that starts with the point given if it exists |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
186 |
def GetConnector(self, position, output_name=None, input_name=None): |
1130
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
187 |
if input_name is not None: |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
188 |
# Test each input connector |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
189 |
for input in self.Inputs: |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
190 |
if input_name == input.GetName(): |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
191 |
return input |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
192 |
if output_name is not None: |
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
193 |
# Test each output connector |
814 | 194 |
for output in self.Outputs: |
1130
f96e0254f0ce
Fixed loading of Blocks in Viewer when block interface have changed
Laurent Bessard
parents:
1054
diff
changeset
|
195 |
if output_name == output.GetName(): |
814 | 196 |
return output |
1133
d81d99fd1932
Fixed bug in loading block in Viewer when block interface has changed
Laurent Bessard
parents:
1130
diff
changeset
|
197 |
if input_name is None and output_name is None: |
d81d99fd1932
Fixed bug in loading block in Viewer when block interface has changed
Laurent Bessard
parents:
1130
diff
changeset
|
198 |
return self.FindNearestConnector(position, self.Inputs + self.Outputs) |
d81d99fd1932
Fixed bug in loading block in Viewer when block interface has changed
Laurent Bessard
parents:
1130
diff
changeset
|
199 |
return None |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
200 |
|
814 | 201 |
def GetInputTypes(self): |
202 |
return tuple([input.GetType(True) for input in self.Inputs if input.GetName() != "EN"]) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
203 |
|
814 | 204 |
def SetOutputValues(self, values): |
205 |
for output in self.Outputs: |
|
1872
866fb3ab8778
fix pylint error "(undefined-variable) Undefined variable 'X'"
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1852
diff
changeset
|
206 |
output.SetValue(values.get(output.getName(), None)) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
207 |
|
814 | 208 |
def GetConnectionResultType(self, connector, connectortype): |
209 |
if not TestConnectorName(connector.GetName(), self.Type): |
|
210 |
return connectortype |
|
211 |
resulttype = connectortype |
|
212 |
for input in self.Inputs: |
|
213 |
if input != connector and input.GetType(True) == "ANY" and TestConnectorName(input.GetName(), self.Type): |
|
214 |
inputtype = input.GetConnectedType() |
|
215 |
if resulttype is None or inputtype is not None and self.IsOfType(inputtype, resulttype): |
|
216 |
resulttype = inputtype |
|
217 |
for output in self.Outputs: |
|
218 |
if output != connector and output.GetType(True) == "ANY" and TestConnectorName(output.GetName(), self.Type): |
|
219 |
outputtype = output.GetConnectedType() |
|
220 |
if resulttype is None or outputtype is not None and self.IsOfType(outputtype, resulttype): |
|
221 |
resulttype = outputtype |
|
222 |
return resulttype |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
223 |
|
814 | 224 |
# Returns all the block connectors |
225 |
def GetConnectors(self): |
|
1739
ec153828ded2
clean-up: fix PEP8 E203 whitespace before ':' and whitespace before ','
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1736
diff
changeset
|
226 |
return {"inputs": self.Inputs, "outputs": self.Outputs} |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
227 |
|
814 | 228 |
# Test if point given is on one of the block connectors |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
229 |
def TestConnector(self, pt, direction=None, exclude=True): |
814 | 230 |
# Test each input connector |
231 |
for input in self.Inputs: |
|
232 |
if input.TestPoint(pt, direction, exclude): |
|
233 |
return input |
|
234 |
# Test each output connector |
|
235 |
for output in self.Outputs: |
|
236 |
if output.TestPoint(pt, direction, exclude): |
|
237 |
return output |
|
238 |
return None |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
239 |
|
814 | 240 |
# Changes the block type |
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
241 |
def SetType(self, type, extension, inputs=None, connectors=None, executionControl=False): |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
242 |
if type != self.Type or self.Extension != extension or executionControl != self.ExecutionControl: |
814 | 243 |
if type != self.Type: |
244 |
self.Type = type |
|
245 |
self.TypeSize = self.Parent.GetTextExtent(self.Type) |
|
246 |
self.Extension = extension |
|
247 |
self.ExecutionControl = executionControl |
|
248 |
# Find the block definition from type given and create the corresponding |
|
249 |
# inputs and outputs |
|
250 |
blocktype = self.Parent.GetBlockType(type, inputs) |
|
251 |
if blocktype: |
|
252 |
self.Colour = wx.BLACK |
|
253 |
inputs = [input for input in blocktype["inputs"]] |
|
254 |
outputs = [output for output in blocktype["outputs"]] |
|
255 |
if blocktype["extensible"]: |
|
256 |
start = int(inputs[-1][0].replace("IN", "")) |
|
1847
6198190bc121
explicitly mark unused variables found by pylint with _ or dummy
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1782
diff
changeset
|
257 |
for dummy in xrange(self.Extension - len(blocktype["inputs"])): |
814 | 258 |
start += 1 |
1734
750eeb7230a1
clean-up: fix some PEP8 E228 missing whitespace around modulo operator
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1730
diff
changeset
|
259 |
inputs.append(("IN%d" % start, inputs[-1][1], inputs[-1][2])) |
814 | 260 |
comment = blocktype["comment"] |
261 |
self.Description = _(comment) + blocktype.get("usage", "") |
|
262 |
else: |
|
263 |
self.Colour = wx.RED |
|
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
264 |
connectors = {} if connectors is None else connectors |
814 | 265 |
inputs = connectors.get("inputs", []) |
266 |
outputs = connectors.get("outputs", []) |
|
267 |
self.Description = None |
|
268 |
if self.ExecutionControl: |
|
1740
b789b695b5c6
clean-up: fix PEP8 E231 missing whitespace after ':' or ','
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1739
diff
changeset
|
269 |
inputs.insert(0, ("EN", "BOOL", "none")) |
1761
8c98bad90b8d
clean-up: fix PEP8 E203 whitespace before ','
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1753
diff
changeset
|
270 |
outputs.insert(0, ("ENO", "BOOL", "none")) |
814 | 271 |
self.Pen = MiterPen(self.Colour) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
272 |
|
814 | 273 |
# Extract the inputs properties and create or modify the corresponding connector |
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
274 |
input_connectors = [] |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
275 |
for input_name, input_type, input_modifier in inputs: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
276 |
connector = Connector(self, input_name, input_type, wx.Point(0, 0), WEST, onlyone=True) |
814 | 277 |
if input_modifier == "negated": |
278 |
connector.SetNegated(True) |
|
279 |
elif input_modifier != "none": |
|
280 |
connector.SetEdge(input_modifier) |
|
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
281 |
for input in self.Inputs: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
282 |
if input.GetName() == input_name: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
283 |
wires = input.GetWires()[:] |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
284 |
input.UnConnect() |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
285 |
for wire in wires: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
286 |
connector.Connect(wire) |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
287 |
break |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
288 |
input_connectors.append(connector) |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
289 |
for input in self.Inputs: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
290 |
input.UnConnect(delete=True) |
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
291 |
self.Inputs = input_connectors |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
292 |
|
814 | 293 |
# Extract the outputs properties and create or modify the corresponding connector |
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
294 |
output_connectors = [] |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
295 |
for output_name, output_type, output_modifier in outputs: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
296 |
connector = Connector(self, output_name, output_type, wx.Point(0, 0), EAST) |
814 | 297 |
if output_modifier == "negated": |
298 |
connector.SetNegated(True) |
|
299 |
elif output_modifier != "none": |
|
300 |
connector.SetEdge(output_modifier) |
|
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
301 |
for output in self.Outputs: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
302 |
if output.GetName() == output_name: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
303 |
wires = output.GetWires()[:] |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
304 |
output.UnConnect() |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
305 |
for wire in wires: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
306 |
connector.Connect(wire) |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
307 |
break |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
308 |
output_connectors.append(connector) |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
309 |
for output in self.Outputs: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
310 |
output.UnConnect(delete=True) |
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
311 |
self.Outputs = output_connectors |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
312 |
|
814 | 313 |
self.RefreshMinSize() |
314 |
self.RefreshConnectors() |
|
1054
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
315 |
for output in self.Outputs: |
ef514eaacd8c
Fixed connections of block when changing block type
Laurent Bessard
parents:
873
diff
changeset
|
316 |
output.RefreshWires() |
814 | 317 |
self.RefreshBoundingBox() |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
318 |
|
814 | 319 |
# Returns the block type |
320 |
def GetType(self): |
|
321 |
return self.Type |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
322 |
|
814 | 323 |
# Changes the block name |
324 |
def SetName(self, name): |
|
325 |
self.Name = name |
|
326 |
self.RefreshNameSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
327 |
|
814 | 328 |
# Returs the block name |
329 |
def GetName(self): |
|
330 |
return self.Name |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
331 |
|
814 | 332 |
# Changes the extension name |
333 |
def SetExtension(self, extension): |
|
334 |
self.Extension = extension |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
335 |
|
814 | 336 |
# Returs the extension name |
337 |
def GetExtension(self): |
|
338 |
return self.Extension |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
339 |
|
814 | 340 |
# Changes the execution order |
341 |
def SetExecutionOrder(self, executionOrder): |
|
342 |
self.ExecutionOrder = executionOrder |
|
343 |
self.RefreshExecutionOrderSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
344 |
|
814 | 345 |
# Returs the execution order |
346 |
def GetExecutionOrder(self): |
|
347 |
return self.ExecutionOrder |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
348 |
|
814 | 349 |
# Returs the execution order |
350 |
def GetExecutionControl(self): |
|
351 |
return self.ExecutionControl |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
352 |
|
814 | 353 |
# Refresh the block minimum size |
354 |
def RefreshMinSize(self): |
|
355 |
# Calculate the inputs maximum width |
|
356 |
max_input = 0 |
|
357 |
for input in self.Inputs: |
|
1847
6198190bc121
explicitly mark unused variables found by pylint with _ or dummy
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1782
diff
changeset
|
358 |
w, _h = input.GetNameSize() |
814 | 359 |
max_input = max(max_input, w) |
360 |
# Calculate the outputs maximum width |
|
361 |
max_output = 0 |
|
362 |
for output in self.Outputs: |
|
1847
6198190bc121
explicitly mark unused variables found by pylint with _ or dummy
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1782
diff
changeset
|
363 |
w, _h = output.GetNameSize() |
814 | 364 |
max_output = max(max_output, w) |
365 |
width = max(self.TypeSize[0] + 10, max_input + max_output + 15) |
|
366 |
height = (max(len(self.Inputs), len(self.Outputs)) + 1) * BLOCK_LINE_SIZE |
|
367 |
self.MinSize = width, height |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
368 |
|
814 | 369 |
# Returns the block minimum size |
370 |
def GetMinSize(self): |
|
371 |
return self.MinSize |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
372 |
|
814 | 373 |
# Changes the negated property of the connector handled |
374 |
def SetConnectorNegated(self, negated): |
|
375 |
handle_type, handle = self.Handle |
|
376 |
if handle_type == HANDLE_CONNECTOR: |
|
377 |
handle.SetNegated(negated) |
|
378 |
self.RefreshModel(False) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
379 |
|
814 | 380 |
# Changes the edge property of the connector handled |
381 |
def SetConnectorEdge(self, edge): |
|
382 |
handle_type, handle = self.Handle |
|
383 |
if handle_type == HANDLE_CONNECTOR: |
|
384 |
handle.SetEdge(edge) |
|
385 |
self.RefreshModel(False) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
386 |
|
1753
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
387 |
# # Method called when a Motion event have been generated |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
388 |
# def OnMotion(self, event, dc, scaling): |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
389 |
# if not event.Dragging(): |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
390 |
# pos = event.GetLogicalPosition(dc) |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
391 |
# for input in self.Inputs: |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
392 |
# rect = input.GetRedrawRect() |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
393 |
# if rect.InsideXY(pos.x, pos.y): |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
394 |
# print "Find input" |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
395 |
# tip = wx.TipWindow(self.Parent, "Test") |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
396 |
# tip.SetBoundingRect(rect) |
19f19c66b67e
clean-up: fix most PEP8 E266 too many leading '#' for block comment
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1744
diff
changeset
|
397 |
# return Graphic_Element.OnMotion(self, event, dc, scaling) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
398 |
|
814 | 399 |
# Method called when a LeftDClick event have been generated |
400 |
def OnLeftDClick(self, event, dc, scaling): |
|
401 |
# Edit the block properties |
|
402 |
self.Parent.EditBlockContent(self) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
403 |
|
814 | 404 |
# Method called when a RightUp event have been generated |
405 |
def OnRightUp(self, event, dc, scaling): |
|
406 |
pos = GetScaledEventPosition(event, dc, scaling) |
|
407 |
# Popup the menu with special items for a block and a connector if one is handled |
|
408 |
connector = self.TestConnector(pos, exclude=False) |
|
409 |
if connector: |
|
410 |
self.Handle = (HANDLE_CONNECTOR, connector) |
|
411 |
self.Parent.PopupBlockMenu(connector) |
|
412 |
else: |
|
413 |
self.Parent.PopupBlockMenu() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
414 |
|
814 | 415 |
# Refreshes the block model |
416 |
def RefreshModel(self, move=True): |
|
417 |
self.Parent.RefreshBlockModel(self) |
|
418 |
# If block has moved, refresh the model of wires connected to outputs |
|
419 |
if move: |
|
420 |
for output in self.Outputs: |
|
421 |
output.RefreshWires() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
422 |
|
814 | 423 |
def GetToolTipValue(self): |
424 |
return self.Description |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
425 |
|
814 | 426 |
# Adds an highlight to the block |
1739
ec153828ded2
clean-up: fix PEP8 E203 whitespace before ':' and whitespace before ','
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1736
diff
changeset
|
427 |
def AddHighlight(self, infos, start, end, highlight_type): |
814 | 428 |
if infos[0] in ["type", "name"] and start[0] == 0 and end[0] == 0: |
429 |
highlights = self.Highlights.setdefault(infos[0], []) |
|
430 |
AddHighlight(highlights, (start, end, highlight_type)) |
|
431 |
elif infos[0] == "input" and infos[1] < len(self.Inputs): |
|
432 |
self.Inputs[infos[1]].AddHighlight(infos[2:], start, end, highlight_type) |
|
433 |
elif infos[0] == "output" and infos[1] < len(self.Outputs): |
|
434 |
self.Outputs[infos[1]].AddHighlight(infos[2:], start, end, highlight_type) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
435 |
|
814 | 436 |
# Removes an highlight from the block |
437 |
def RemoveHighlight(self, infos, start, end, highlight_type): |
|
438 |
if infos[0] in ["type", "name"]: |
|
439 |
highlights = self.Highlights.get(infos[0], []) |
|
440 |
if RemoveHighlight(highlights, (start, end, highlight_type)) and len(highlights) == 0: |
|
441 |
self.Highlights.pop(infos[0]) |
|
442 |
elif infos[0] == "input" and infos[1] < len(self.Inputs): |
|
443 |
self.Inputs[infos[1]].RemoveHighlight(infos[2:], start, end, highlight_type) |
|
444 |
elif infos[0] == "output" and infos[1] < len(self.Outputs): |
|
445 |
self.Outputs[infos[1]].RemoveHighlight(infos[2:], start, end, highlight_type) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
446 |
|
814 | 447 |
# Removes all the highlights of one particular type from the block |
448 |
def ClearHighlight(self, highlight_type=None): |
|
449 |
if highlight_type is None: |
|
450 |
self.Highlights = {} |
|
451 |
else: |
|
452 |
highlight_items = self.Highlights.items() |
|
453 |
for name, highlights in highlight_items: |
|
454 |
highlights = ClearHighlights(highlights, highlight_type) |
|
455 |
if len(highlights) == 0: |
|
456 |
self.Highlights.pop(name) |
|
457 |
for input in self.Inputs: |
|
458 |
input.ClearHighlights(highlight_type) |
|
459 |
for output in self.Outputs: |
|
460 |
output.ClearHighlights(highlight_type) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
461 |
|
814 | 462 |
# Draws block |
463 |
def Draw(self, dc): |
|
464 |
Graphic_Element.Draw(self, dc) |
|
465 |
dc.SetPen(self.Pen) |
|
466 |
dc.SetBrush(wx.WHITE_BRUSH) |
|
467 |
dc.SetTextForeground(self.Colour) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
468 |
|
814 | 469 |
if getattr(dc, "printing", False): |
470 |
name_size = dc.GetTextExtent(self.Name) |
|
471 |
type_size = dc.GetTextExtent(self.Type) |
|
472 |
executionorder_size = dc.GetTextExtent(str(self.ExecutionOrder)) |
|
473 |
else: |
|
474 |
name_size = self.NameSize |
|
475 |
type_size = self.TypeSize |
|
476 |
executionorder_size = self.ExecutionOrderSize |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
477 |
|
814 | 478 |
# Draw a rectangle with the block size |
479 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
480 |
# Draw block name and block type |
|
481 |
name_pos = (self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
|
482 |
self.Pos.y - (name_size[1] + 2)) |
|
483 |
type_pos = (self.Pos.x + (self.Size[0] - type_size[0]) / 2, |
|
484 |
self.Pos.y + 5) |
|
485 |
dc.DrawText(self.Name, name_pos[0], name_pos[1]) |
|
486 |
dc.DrawText(self.Type, type_pos[0], type_pos[1]) |
|
487 |
# Draw inputs and outputs connectors |
|
488 |
for input in self.Inputs: |
|
489 |
input.Draw(dc) |
|
490 |
for output in self.Outputs: |
|
491 |
output.Draw(dc) |
|
492 |
if self.ExecutionOrder != 0: |
|
493 |
# Draw block execution order |
|
494 |
dc.DrawText(str(self.ExecutionOrder), self.Pos.x + self.Size[0] - executionorder_size[0], |
|
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
495 |
self.Pos.y + self.Size[1] + 2) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
496 |
|
814 | 497 |
if not getattr(dc, "printing", False): |
498 |
DrawHighlightedText(dc, self.Name, self.Highlights.get("name", []), name_pos[0], name_pos[1]) |
|
499 |
DrawHighlightedText(dc, self.Type, self.Highlights.get("type", []), type_pos[0], type_pos[1]) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
500 |
|
814 | 501 |
|
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
502 |
# ------------------------------------------------------------------------------- |
814 | 503 |
# Function Block Diagram Variable |
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
504 |
# ------------------------------------------------------------------------------- |
814 | 505 |
|
506 |
||
507 |
class FBD_Variable(Graphic_Element): |
|
1736
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
508 |
""" |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
509 |
Class that implements the graphic representation of a variable |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
510 |
""" |
814 | 511 |
|
512 |
# Create a new variable |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
513 |
def __init__(self, parent, type, name, value_type, id=None, executionOrder=0): |
814 | 514 |
Graphic_Element.__init__(self, parent) |
515 |
self.Type = None |
|
516 |
self.ValueType = None |
|
517 |
self.Id = id |
|
518 |
self.SetName(name) |
|
519 |
self.SetExecutionOrder(executionOrder) |
|
520 |
self.Input = None |
|
521 |
self.Output = None |
|
522 |
self.SetType(type, value_type) |
|
523 |
self.Highlights = [] |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
524 |
|
814 | 525 |
# Make a clone of this FBD_Variable |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
526 |
def Clone(self, parent, id=None, pos=None): |
814 | 527 |
variable = FBD_Variable(parent, self.Type, self.Name, self.ValueType, id) |
528 |
variable.SetSize(self.Size[0], self.Size[1]) |
|
529 |
if pos is not None: |
|
530 |
variable.SetPosition(pos.x, pos.y) |
|
531 |
else: |
|
532 |
variable.SetPosition(self.Pos.x, self.Pos.y) |
|
533 |
if self.Input: |
|
534 |
variable.Input = self.Input.Clone(variable) |
|
535 |
if self.Output: |
|
536 |
variable.Output = self.Output.Clone(variable) |
|
537 |
return variable |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
538 |
|
814 | 539 |
def GetConnectorTranslation(self, element): |
540 |
connectors = {} |
|
541 |
if self.Input is not None: |
|
542 |
connectors[self.Input] = element.Input |
|
543 |
if self.Output is not None: |
|
544 |
connectors[self.Output] = element.Output |
|
545 |
return connectors |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
546 |
|
814 | 547 |
def Flush(self): |
548 |
if self.Input is not None: |
|
549 |
self.Input.Flush() |
|
550 |
self.Input = None |
|
551 |
if self.Output is not None: |
|
552 |
self.Output.Flush() |
|
553 |
self.Output = None |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
554 |
|
814 | 555 |
# Returns the RedrawRect |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
556 |
def GetRedrawRect(self, movex=0, movey=0): |
814 | 557 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
558 |
if movex != 0 or movey != 0: |
|
559 |
if self.Input and self.Input.IsConnected(): |
|
560 |
rect = rect.Union(self.Input.GetConnectedRedrawRect(movex, movey)) |
|
561 |
if self.Output and self.Output.IsConnected(): |
|
562 |
rect = rect.Union(self.Output.GetConnectedRedrawRect(movex, movey)) |
|
563 |
return rect |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
564 |
|
814 | 565 |
# Unconnect connector |
566 |
def Clean(self): |
|
567 |
if self.Input: |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
568 |
self.Input.UnConnect(delete=True) |
814 | 569 |
if self.Output: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
570 |
self.Output.UnConnect(delete=True) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
571 |
|
814 | 572 |
# Delete this variable by calling the appropriate method |
573 |
def Delete(self): |
|
574 |
self.Parent.DeleteVariable(self) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
575 |
|
814 | 576 |
# Refresh the size of text for name |
577 |
def RefreshNameSize(self): |
|
578 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
579 |
|
814 | 580 |
# Refresh the size of text for execution order |
581 |
def RefreshExecutionOrderSize(self): |
|
582 |
self.ExecutionOrderSize = self.Parent.GetTextExtent(str(self.ExecutionOrder)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
583 |
|
814 | 584 |
# Refresh the variable bounding box |
585 |
def RefreshBoundingBox(self): |
|
586 |
if self.Type in (OUTPUT, INOUT): |
|
587 |
bbx_x = self.Pos.x - CONNECTOR_SIZE |
|
588 |
else: |
|
589 |
bbx_x = self.Pos.x |
|
590 |
if self.Type == INOUT: |
|
591 |
bbx_width = self.Size[0] + 2 * CONNECTOR_SIZE |
|
592 |
else: |
|
593 |
bbx_width = self.Size[0] + CONNECTOR_SIZE |
|
594 |
bbx_x = min(bbx_x, self.Pos.x + (self.Size[0] - self.NameSize[0]) / 2) |
|
595 |
bbx_width = max(bbx_width, self.NameSize[0]) |
|
596 |
bbx_height = self.Size[1] |
|
597 |
if self.ExecutionOrder != 0: |
|
598 |
bbx_x = min(bbx_x, self.Pos.x + self.Size[0] - self.ExecutionOrderSize[0]) |
|
599 |
bbx_width = max(bbx_width, bbx_width + self.Pos.x + self.ExecutionOrderSize[0] - bbx_x - self.Size[0]) |
|
600 |
bbx_height = bbx_height + (self.ExecutionOrderSize[1] + 2) |
|
601 |
self.BoundingBox = wx.Rect(bbx_x, self.Pos.y, bbx_width + 1, bbx_height + 1) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
602 |
|
814 | 603 |
# Refresh the position of the variable connector |
604 |
def RefreshConnectors(self): |
|
605 |
scaling = self.Parent.GetScaling() |
|
606 |
if scaling is not None: |
|
607 |
position = round_scaling(self.Pos.y + self.Size[1] / 2, scaling[1]) - self.Pos.y |
|
608 |
else: |
|
609 |
position = self.Size[1] / 2 |
|
610 |
if self.Input: |
|
611 |
self.Input.SetPosition(wx.Point(0, position)) |
|
612 |
if self.Output: |
|
613 |
self.Output.SetPosition(wx.Point(self.Size[0], position)) |
|
614 |
self.RefreshConnected() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
615 |
|
814 | 616 |
# Refresh the position of wires connected to connector |
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
617 |
def RefreshConnected(self, exclude=None): |
814 | 618 |
if self.Input: |
619 |
self.Input.MoveConnected(exclude) |
|
620 |
if self.Output: |
|
621 |
self.Output.MoveConnected(exclude) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
622 |
|
814 | 623 |
# Test if point given is on the variable connector |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
624 |
def TestConnector(self, pt, direction=None, exclude=True): |
814 | 625 |
if self.Input and self.Input.TestPoint(pt, direction, exclude): |
626 |
return self.Input |
|
627 |
if self.Output and self.Output.TestPoint(pt, direction, exclude): |
|
628 |
return self.Output |
|
629 |
return None |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
630 |
|
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
631 |
# Returns the block connector that starts with the point given if it exists |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
632 |
def GetConnector(self, position, name=None): |
814 | 633 |
# if a name is given |
634 |
if name is not None: |
|
635 |
# Test input and output connector if they exists |
|
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
636 |
# if self.Input and name == self.Input.GetName(): |
814 | 637 |
# return self.Input |
638 |
if self.Output and name == self.Output.GetName(): |
|
639 |
return self.Output |
|
640 |
connectors = [] |
|
641 |
# Test input connector if it exists |
|
642 |
if self.Input: |
|
643 |
connectors.append(self.Input) |
|
644 |
# Test output connector if it exists |
|
645 |
if self.Output: |
|
646 |
connectors.append(self.Output) |
|
647 |
return self.FindNearestConnector(position, connectors) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
648 |
|
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
649 |
# Returns all the block connectors |
814 | 650 |
def GetConnectors(self): |
651 |
connectors = {"inputs": [], "outputs": []} |
|
652 |
if self.Input: |
|
653 |
connectors["inputs"].append(self.Input) |
|
654 |
if self.Output: |
|
655 |
connectors["outputs"].append(self.Output) |
|
656 |
return connectors |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
657 |
|
814 | 658 |
# Changes the negated property of the variable connector if handled |
659 |
def SetConnectorNegated(self, negated): |
|
660 |
handle_type, handle = self.Handle |
|
661 |
if handle_type == HANDLE_CONNECTOR: |
|
662 |
handle.SetNegated(negated) |
|
663 |
self.RefreshModel(False) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
664 |
|
814 | 665 |
# Changes the variable type |
666 |
def SetType(self, type, value_type): |
|
667 |
if type != self.Type: |
|
668 |
self.Type = type |
|
669 |
# Create an input or output connector according to variable type |
|
670 |
if self.Type != INPUT: |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
671 |
if self.Input is None: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
672 |
self.Input = Connector(self, "", value_type, wx.Point(0, 0), WEST, onlyone=True) |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
673 |
elif self.Input: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
674 |
self.Input.UnConnect(delete=True) |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
675 |
self.Input = None |
814 | 676 |
if self.Type != OUTPUT: |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
677 |
if self.Output is None: |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
678 |
self.Output = Connector(self, "", value_type, wx.Point(0, 0), EAST) |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
679 |
elif self.Output: |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
680 |
self.Output.UnConnect(delete=True) |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
681 |
self.Output = None |
814 | 682 |
self.RefreshConnectors() |
873
8737f1554494
Fix bug Variable and Continuation BoundingBox not refreshed when using Type modification shortcuts
Laurent Bessard
parents:
857
diff
changeset
|
683 |
self.RefreshBoundingBox() |
814 | 684 |
elif value_type != self.ValueType: |
685 |
if self.Input: |
|
686 |
self.Input.SetType(value_type) |
|
687 |
if self.Output: |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
688 |
self.Output.SetType(value_type) |
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
689 |
|
814 | 690 |
# Returns the variable type |
691 |
def GetType(self): |
|
692 |
return self.Type |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
693 |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
694 |
# Returns the variable value type |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
695 |
def GetValueType(self): |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
696 |
return self.ValueType |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
697 |
|
814 | 698 |
# Changes the variable name |
699 |
def SetName(self, name): |
|
700 |
self.Name = name |
|
701 |
self.RefreshNameSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
702 |
|
814 | 703 |
# Returns the variable name |
704 |
def GetName(self): |
|
705 |
return self.Name |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
706 |
|
814 | 707 |
# Changes the execution order |
708 |
def SetExecutionOrder(self, executionOrder): |
|
709 |
self.ExecutionOrder = executionOrder |
|
710 |
self.RefreshExecutionOrderSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
711 |
|
814 | 712 |
# Returs the execution order |
713 |
def GetExecutionOrder(self): |
|
714 |
return self.ExecutionOrder |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
715 |
|
814 | 716 |
# Returns the variable minimum size |
717 |
def GetMinSize(self): |
|
718 |
return self.NameSize[0] + 10, self.NameSize[1] + 10 |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
719 |
|
852
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
720 |
# Set size of the variable to the minimum size |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
721 |
def SetBestSize(self, scaling): |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
722 |
if self.Type == INPUT: |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
723 |
return Graphic_Element.SetBestSize(self, scaling, x_factor=1.) |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
724 |
elif self.Type == OUTPUT: |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
725 |
return Graphic_Element.SetBestSize(self, scaling, x_factor=0.) |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
726 |
else: |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
727 |
return Graphic_Element.SetBestSize(self, scaling) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
728 |
|
814 | 729 |
# Method called when a LeftDClick event have been generated |
730 |
def OnLeftDClick(self, event, dc, scaling): |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
731 |
if event.ControlDown(): |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
732 |
# Change variable type |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
733 |
types = [INPUT, OUTPUT, INOUT] |
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
734 |
self.Parent.ChangeVariableType( |
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
735 |
self, types[(types.index(self.Type) + 1) % len(types)]) |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
736 |
else: |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
737 |
# Edit the variable properties |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
738 |
self.Parent.EditVariableContent(self) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
739 |
|
814 | 740 |
# Method called when a RightUp event have been generated |
741 |
def OnRightUp(self, event, dc, scaling): |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
742 |
self.Parent.PopupVariableMenu() |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
743 |
|
814 | 744 |
# Refreshes the variable model |
745 |
def RefreshModel(self, move=True): |
|
746 |
self.Parent.RefreshVariableModel(self) |
|
747 |
# If variable has moved and variable is not of type OUTPUT, refresh the model |
|
748 |
# of wires connected to output connector |
|
749 |
if move and self.Type != OUTPUT: |
|
750 |
if self.Output: |
|
751 |
self.Output.RefreshWires() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
752 |
|
814 | 753 |
# Adds an highlight to the variable |
754 |
def AddHighlight(self, infos, start, end, highlight_type): |
|
755 |
if infos[0] == "expression" and start[0] == 0 and end[0] == 0: |
|
756 |
AddHighlight(self.Highlights, (start, end, highlight_type)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
757 |
|
814 | 758 |
# Removes an highlight from the variable |
759 |
def RemoveHighlight(self, infos, start, end, highlight_type): |
|
760 |
if infos[0] == "expression": |
|
761 |
RemoveHighlight(self.Highlights, (start, end, highlight_type)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
762 |
|
814 | 763 |
# Removes all the highlights of one particular type from the variable |
764 |
def ClearHighlight(self, highlight_type=None): |
|
765 |
ClearHighlights(self.Highlights, highlight_type) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
766 |
|
814 | 767 |
# Draws variable |
768 |
def Draw(self, dc): |
|
769 |
Graphic_Element.Draw(self, dc) |
|
770 |
dc.SetPen(MiterPen(wx.BLACK)) |
|
771 |
dc.SetBrush(wx.WHITE_BRUSH) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
772 |
|
814 | 773 |
if getattr(dc, "printing", False): |
774 |
name_size = dc.GetTextExtent(self.Name) |
|
775 |
executionorder_size = dc.GetTextExtent(str(self.ExecutionOrder)) |
|
776 |
else: |
|
777 |
name_size = self.NameSize |
|
778 |
executionorder_size = self.ExecutionOrderSize |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
779 |
|
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
780 |
text_pos = (self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
814 | 781 |
self.Pos.y + (self.Size[1] - name_size[1]) / 2) |
782 |
# Draw a rectangle with the variable size |
|
783 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
784 |
# Draw variable name |
|
785 |
dc.DrawText(self.Name, text_pos[0], text_pos[1]) |
|
786 |
# Draw connectors |
|
787 |
if self.Input: |
|
788 |
self.Input.Draw(dc) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
789 |
if self.Output: |
814 | 790 |
self.Output.Draw(dc) |
791 |
if self.ExecutionOrder != 0: |
|
792 |
# Draw variable execution order |
|
793 |
dc.DrawText(str(self.ExecutionOrder), self.Pos.x + self.Size[0] - executionorder_size[0], |
|
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
794 |
self.Pos.y + self.Size[1] + 2) |
814 | 795 |
if not getattr(dc, "printing", False): |
796 |
DrawHighlightedText(dc, self.Name, self.Highlights, text_pos[0], text_pos[1]) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
797 |
|
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
798 |
|
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
799 |
# ------------------------------------------------------------------------------- |
814 | 800 |
# Function Block Diagram Connector |
1782
5b6ad7a7fd9d
clean-up: fix PEP8 E265 block comment should start with '# '
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1768
diff
changeset
|
801 |
# ------------------------------------------------------------------------------- |
814 | 802 |
|
803 |
||
804 |
class FBD_Connector(Graphic_Element): |
|
1736
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
805 |
""" |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
806 |
Class that implements the graphic representation of a connection |
7e61baa047f0
clean-up: fix PEP8 E302 expected 2 blank lines, found 1
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1734
diff
changeset
|
807 |
""" |
814 | 808 |
|
809 |
# Create a new connection |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
810 |
def __init__(self, parent, type, name, id=None): |
814 | 811 |
Graphic_Element.__init__(self, parent) |
812 |
self.Type = type |
|
813 |
self.Id = id |
|
814 |
self.SetName(name) |
|
815 |
self.Pos = wx.Point(0, 0) |
|
816 |
self.Size = wx.Size(0, 0) |
|
817 |
self.Highlights = [] |
|
818 |
# Create an input or output connector according to connection type |
|
819 |
if self.Type == CONNECTOR: |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
820 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), WEST, onlyone=True) |
814 | 821 |
else: |
822 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), EAST) |
|
823 |
self.RefreshConnectors() |
|
824 |
self.RefreshNameSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
825 |
|
814 | 826 |
def Flush(self): |
827 |
if self.Connector: |
|
828 |
self.Connector.Flush() |
|
829 |
self.Connector = None |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
830 |
|
814 | 831 |
# Returns the RedrawRect |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
832 |
def GetRedrawRect(self, movex=0, movey=0): |
814 | 833 |
rect = Graphic_Element.GetRedrawRect(self, movex, movey) |
834 |
if movex != 0 or movey != 0: |
|
835 |
if self.Connector and self.Connector.IsConnected(): |
|
836 |
rect = rect.Union(self.Connector.GetConnectedRedrawRect(movex, movey)) |
|
837 |
return rect |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
838 |
|
814 | 839 |
# Make a clone of this FBD_Connector |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
840 |
def Clone(self, parent, id=None, pos=None): |
814 | 841 |
connection = FBD_Connector(parent, self.Type, self.Name, id) |
842 |
connection.SetSize(self.Size[0], self.Size[1]) |
|
843 |
if pos is not None: |
|
844 |
connection.SetPosition(pos.x, pos.y) |
|
845 |
else: |
|
846 |
connection.SetPosition(self.Pos.x, self.Pos.y) |
|
847 |
connection.Connector = self.Connector.Clone(connection) |
|
848 |
return connection |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
849 |
|
814 | 850 |
def GetConnectorTranslation(self, element): |
1739
ec153828ded2
clean-up: fix PEP8 E203 whitespace before ':' and whitespace before ','
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1736
diff
changeset
|
851 |
return {self.Connector: element.Connector} |
814 | 852 |
|
853 |
# Unconnect connector |
|
854 |
def Clean(self): |
|
855 |
if self.Connector: |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
856 |
self.Connector.UnConnect(delete=True) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
857 |
|
814 | 858 |
# Delete this connection by calling the appropriate method |
859 |
def Delete(self): |
|
860 |
self.Parent.DeleteConnection(self) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
861 |
|
814 | 862 |
# Refresh the size of text for name |
863 |
def RefreshNameSize(self): |
|
864 |
self.NameSize = self.Parent.GetTextExtent(self.Name) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
865 |
|
814 | 866 |
# Refresh the connection bounding box |
867 |
def RefreshBoundingBox(self): |
|
868 |
if self.Type == CONNECTOR: |
|
869 |
bbx_x = self.Pos.x - CONNECTOR_SIZE |
|
870 |
else: |
|
871 |
bbx_x = self.Pos.x |
|
872 |
bbx_width = self.Size[0] + CONNECTOR_SIZE |
|
873 |
self.BoundingBox = wx.Rect(bbx_x, self.Pos.y, bbx_width, self.Size[1]) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
874 |
|
814 | 875 |
# Refresh the position of the connection connector |
876 |
def RefreshConnectors(self): |
|
877 |
scaling = self.Parent.GetScaling() |
|
878 |
if scaling is not None: |
|
879 |
position = round_scaling(self.Pos.y + self.Size[1] / 2, scaling[1]) - self.Pos.y |
|
880 |
else: |
|
881 |
position = self.Size[1] / 2 |
|
882 |
if self.Type == CONNECTOR: |
|
883 |
self.Connector.SetPosition(wx.Point(0, position)) |
|
884 |
else: |
|
885 |
self.Connector.SetPosition(wx.Point(self.Size[0], position)) |
|
886 |
self.RefreshConnected() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
887 |
|
814 | 888 |
# Refresh the position of wires connected to connector |
1852
70c1cc354a8f
fix pylint warning '(dangerous-default-value) Dangerous default value {} as argument'
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1847
diff
changeset
|
889 |
def RefreshConnected(self, exclude=None): |
814 | 890 |
if self.Connector: |
891 |
self.Connector.MoveConnected(exclude) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
892 |
|
814 | 893 |
# Test if point given is on the connection connector |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
894 |
def TestConnector(self, pt, direction=None, exclude=True): |
814 | 895 |
if self.Connector and self.Connector.TestPoint(pt, direction, exclude): |
896 |
return self.Connector |
|
897 |
return None |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
898 |
|
814 | 899 |
# Returns the connection connector |
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
900 |
def GetConnector(self, position=None, name=None): |
814 | 901 |
return self.Connector |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
902 |
|
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
903 |
# Returns all the block connectors |
814 | 904 |
def GetConnectors(self): |
905 |
connectors = {"inputs": [], "outputs": []} |
|
906 |
if self.Type == CONNECTOR: |
|
907 |
connectors["inputs"].append(self.Connector) |
|
908 |
else: |
|
909 |
connectors["outputs"].append(self.Connector) |
|
910 |
return connectors |
|
1605
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
911 |
|
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
912 |
def SpreadCurrent(self): |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
913 |
if self.Type == CONNECTOR: |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
914 |
continuations = self.Parent.GetContinuationByName(self.Name) |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
915 |
if continuations is not None: |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
916 |
value = self.Connector.ReceivingCurrent() |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
917 |
for cont in continuations: |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
918 |
cont.Connector.SpreadCurrent(value) |
0b6b60241230
fix issue that in Debug mode LD instance doesn't show correctly its state after
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1571
diff
changeset
|
919 |
|
814 | 920 |
# Changes the variable type |
921 |
def SetType(self, type): |
|
922 |
if type != self.Type: |
|
923 |
self.Type = type |
|
924 |
self.Clean() |
|
925 |
# Create an input or output connector according to connection type |
|
926 |
if self.Type == CONNECTOR: |
|
1744
69dfdb26f600
clean-up: fix PEP8 E251 unexpected spaces around keyword / parameter equals
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1740
diff
changeset
|
927 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), WEST, onlyone=True) |
814 | 928 |
else: |
929 |
self.Connector = Connector(self, "", "ANY", wx.Point(0, 0), EAST) |
|
930 |
self.RefreshConnectors() |
|
873
8737f1554494
Fix bug Variable and Continuation BoundingBox not refreshed when using Type modification shortcuts
Laurent Bessard
parents:
857
diff
changeset
|
931 |
self.RefreshBoundingBox() |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
932 |
|
814 | 933 |
# Returns the connection type |
934 |
def GetType(self): |
|
935 |
return self.Type |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
936 |
|
814 | 937 |
def GetConnectionResultType(self, connector, connectortype): |
938 |
if self.Type == CONTINUATION: |
|
939 |
connector = self.Parent.GetConnectorByName(self.Name) |
|
940 |
if connector is not None: |
|
941 |
return connector.Connector.GetConnectedType() |
|
942 |
return connectortype |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
943 |
|
814 | 944 |
# Changes the connection name |
945 |
def SetName(self, name): |
|
946 |
self.Name = name |
|
947 |
self.RefreshNameSize() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
948 |
|
814 | 949 |
# Returns the connection name |
950 |
def GetName(self): |
|
951 |
return self.Name |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
952 |
|
852
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
953 |
# Set size of the variable to the minimum size |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
954 |
def SetBestSize(self, scaling): |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
955 |
if self.Type == CONTINUATION: |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
956 |
return Graphic_Element.SetBestSize(self, scaling, x_factor=1.) |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
957 |
else: |
1009f956d2ee
Fix support for adjusting block size to block minimum size and to Viewer scaling
Laurent Bessard
parents:
814
diff
changeset
|
958 |
return Graphic_Element.SetBestSize(self, scaling, x_factor=0.) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
959 |
|
814 | 960 |
# Returns the connection minimum size |
961 |
def GetMinSize(self): |
|
962 |
text_width, text_height = self.NameSize |
|
963 |
if text_height % 2 == 1: |
|
964 |
text_height += 1 |
|
965 |
return text_width + text_height + 20, text_height + 10 |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
966 |
|
814 | 967 |
# Method called when a LeftDClick event have been generated |
968 |
def OnLeftDClick(self, event, dc, scaling): |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
969 |
if event.ControlDown(): |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
970 |
# Change connection type |
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
971 |
if self.Type == CONNECTOR: |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
972 |
self.Parent.ChangeConnectionType(self, CONTINUATION) |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
973 |
else: |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
974 |
self.Parent.ChangeConnectionType(self, CONNECTOR) |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
975 |
else: |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
976 |
# Edit the connection properties |
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
977 |
self.Parent.EditConnectionContent(self) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
978 |
|
814 | 979 |
# Method called when a RightUp event have been generated |
980 |
def OnRightUp(self, event, dc, scaling): |
|
981 |
# Popup the default menu |
|
857
9695969796d0
Adding support for quickly changing variable and connection type
Laurent Bessard
parents:
852
diff
changeset
|
982 |
self.Parent.PopupConnectionMenu() |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
983 |
|
814 | 984 |
# Refreshes the connection model |
985 |
def RefreshModel(self, move=True): |
|
986 |
self.Parent.RefreshConnectionModel(self) |
|
987 |
# If connection has moved and connection is of type CONTINUATION, refresh |
|
988 |
# the model of wires connected to connector |
|
989 |
if move and self.Type == CONTINUATION: |
|
990 |
if self.Connector: |
|
991 |
self.Connector.RefreshWires() |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
992 |
|
814 | 993 |
# Adds an highlight to the connection |
994 |
def AddHighlight(self, infos, start, end, highlight_type): |
|
995 |
if infos[0] == "name" and start[0] == 0 and end[0] == 0: |
|
996 |
AddHighlight(self.Highlights, (start, end, highlight_type)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
997 |
|
814 | 998 |
# Removes an highlight from the connection |
999 |
def RemoveHighlight(self, infos, start, end, highlight_type): |
|
1000 |
if infos[0] == "name": |
|
1001 |
RemoveHighlight(self.Highlights, (start, end, highlight_type)) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1002 |
|
814 | 1003 |
# Removes all the highlights of one particular type from the connection |
1004 |
def ClearHighlight(self, highlight_type=None): |
|
1005 |
ClearHighlights(self.Highlights, highlight_type) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1006 |
|
814 | 1007 |
# Draws connection |
1008 |
def Draw(self, dc): |
|
1009 |
Graphic_Element.Draw(self, dc) |
|
1010 |
dc.SetPen(MiterPen(wx.BLACK)) |
|
1011 |
dc.SetBrush(wx.WHITE_BRUSH) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1012 |
|
814 | 1013 |
if getattr(dc, "printing", False): |
1014 |
name_size = dc.GetTextExtent(self.Name) |
|
1015 |
else: |
|
1016 |
name_size = self.NameSize |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1017 |
|
814 | 1018 |
# Draw a rectangle with the connection size with arrows inside |
1019 |
dc.DrawRectangle(self.Pos.x, self.Pos.y, self.Size[0] + 1, self.Size[1] + 1) |
|
1020 |
arrowsize = min(self.Size[1] / 2, (self.Size[0] - name_size[0] - 10) / 2) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1021 |
dc.DrawLine(self.Pos.x, self.Pos.y, self.Pos.x + arrowsize, |
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
1022 |
self.Pos.y + self.Size[1] / 2) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1023 |
dc.DrawLine(self.Pos.x + arrowsize, self.Pos.y + self.Size[1] / 2, |
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
1024 |
self.Pos.x, self.Pos.y + self.Size[1]) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1025 |
dc.DrawLine(self.Pos.x + self.Size[0] - arrowsize, self.Pos.y, |
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
1026 |
self.Pos.x + self.Size[0], self.Pos.y + self.Size[1] / 2) |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1027 |
dc.DrawLine(self.Pos.x + self.Size[0], self.Pos.y + self.Size[1] / 2, |
1768
691083b5682a
clean-up: fix PEP8 E128 continuation line under-indented for visual indent
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1761
diff
changeset
|
1028 |
self.Pos.x + self.Size[0] - arrowsize, self.Pos.y + self.Size[1]) |
814 | 1029 |
# Draw connection name |
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1030 |
text_pos = (self.Pos.x + (self.Size[0] - name_size[0]) / 2, |
814 | 1031 |
self.Pos.y + (self.Size[1] - name_size[1]) / 2) |
1032 |
dc.DrawText(self.Name, text_pos[0], text_pos[1]) |
|
1033 |
# Draw connector |
|
1034 |
if self.Connector: |
|
1035 |
self.Connector.Draw(dc) |
|
1730
64d8f52bc8c8
clean-up for PEP8: fix W291 trailing whitespace
Andrey Skvortsov <andrej.skvortzov@gmail.com>
parents:
1605
diff
changeset
|
1036 |
|
814 | 1037 |
if not getattr(dc, "printing", False): |
1038 |
DrawHighlightedText(dc, self.Name, self.Highlights, text_pos[0], text_pos[1]) |