Commit ad654b78 authored by Michael Büsch's avatar Michael Büsch

fup_wire: Add property to get regular segments

Signed-off-by: Michael Büsch's avatarMichael Buesch <m@bues.ch>
parent 102b3590
......@@ -131,12 +131,20 @@ class FupWire(FupBaseClass):
self.grid.removeWire(self)
class DrawInfo(object):
__slots__ = ("segStart", "segments", "segDirect")
__slots__ = ("segStart", # First segment
"segments", # Regular segments (list)
"segDirect") # Direct connection segment
def __init__(self, segStart, segments, segDirect):
self.segStart = segStart
self.segments = segments
self.segDirect = segDirect
@property
def allRegularSegments(self):
return itertools.chain((self.segStart,),
self.segments)
def draw(self, painter):
if self.outConn is None:
return # Only inputs. Do not draw.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment