pygtk/canvas/blockitem.py

from gaphas.item import Element
from gaphas.constraint import LineConstraint, LessThanConstraint, EqualsConstraint, Constraint, _update, BalanceConstraint
from gaphas.item import Line, SW, NE, NW, SE, Element, Handle
from gaphas.util import *

from port import *

class BlockItem(Element):
    """
    This is an ASCEND 'block' in the canvas-based modeller. The block will have
    sets of input and output ports to which connector lines can be 'glued'.
    The block will also have a corresponding ASCEND MODEL type, and a name
    which will be used in ASCEND to refer to this block. Each of the ports will
    be special visual elements, but note that these are not 'handles', because
    they can not be used to resize/modify the element.
    """

    def __init__(self, width=64, height=64):

        self.ports = []
        super(BlockItem, self).__init__(width, height)

    def draw(self, context):
        #print 'Box.draw', self
        c = context.cairo
    
        phalfsize = 3
        for p in self.ports:
            c.rectangle(p.x - phalfsize, p.y - phalfsize, 2*phalfsize, 2*phalfsize)
            if p.connected_to is None:
                c.set_source_rgba(0.8,0.8,1, 0.8)
            else:
                c.set_source_rgba(1,0,0,1)
            c.fill_preserve()
            c.set_source_rgb(0.8,0.8,0)
            c.stroke()

    def glue(self,item, handle, ix, iy):
        gluerange = 10
        mindist = -1;
        minport = None
        for p in self.ports:
            dist = math.sqrt((ix-p.x)**2 + (iy-p.y)**2)
            if dist < gluerange:
                if not minport or dist<mindist:
                    mindist = dist
                    minport = p
        return mindist, minport

    def pre_update(self,context):
        #print "PRE-UPDATE BLOCK"
        pass

class DefaultBlockItem(BlockItem):
    """
    This is a 'default block' with a certain number of input and output ports
    shown depending on the values sent to __init__. It is drawn as a simple
    box with the input ports on the left and the output ports on the right.
    """

    def __init__(self, blockinstance):

        self.blockinstance = blockinstance
        inputs = len(blockinstance.blocktype.inputs)
        outputs = len(blockinstance.blocktype.outputs)
        super(DefaultBlockItem, self).__init__(64, 64)

        eq = EqualsConstraint
        bal = BalanceConstraint
        handles = self._handles
        h_nw = handles[NW]
        h_ne = handles[NE]
        h_sw = handles[SW]
        h_se = handles[SE]

        for i in range(inputs):
            p = Port(self)
            self.ports.append(p)
            self._constraints.append(eq(p.x, h_nw.x))
            self._constraints.append(bal(band=(h_nw.y, h_sw.y),v=p.y, balance=(0.5 + i)/inputs))

        for i in range(outputs):
            p = Port(self)
            self.ports.append(p)
            self._constraints.append(eq(p.x, h_ne.x))
            self._constraints.append(bal(band=(h_ne.y,h_se.y),v=p.y, balance=(0.5 + i)/outputs))

    def draw(self, context):
        # draw the box itself
        c = context.cairo
        nw = self._handles[NW]
        c.rectangle(nw.x, nw.y, self.width, self.height)
        if context.hovered:
            c.set_source_rgba(.8,.8,1, .8)
        else:
            c.set_source_rgba(1,1,1, .8)
        c.fill_preserve()
        c.set_source_rgb(0,0,0.8)
        c.stroke()

        text_center(c,self.width/2,self.height/2,self.blockinstance.name)

        # now the draw the ports using the base class
        super(DefaultBlockItem, self).draw(context)

1	jpye	1945	from gaphas.item import Element
2			from gaphas.constraint import LineConstraint, LessThanConstraint, EqualsConstraint, Constraint, _update, BalanceConstraint
3			from gaphas.item import Line, SW, NE, NW, SE, Element, Handle
4			from gaphas.util import *
5
6			from port import *
7
8			class BlockItem(Element):
9			"""
10			This is an ASCEND 'block' in the canvas-based modeller. The block will have
11			sets of input and output ports to which connector lines can be 'glued'.
12			The block will also have a corresponding ASCEND MODEL type, and a name
13			which will be used in ASCEND to refer to this block. Each of the ports will
14			be special visual elements, but note that these are not 'handles', because
15			they can not be used to resize/modify the element.
16			"""
17
18			def __init__(self, width=64, height=64):
19
20			self.ports = []
21			super(BlockItem, self).__init__(width, height)
22
23			def draw(self, context):
24			#print 'Box.draw', self
25			c = context.cairo
26
27			phalfsize = 3
28			for p in self.ports:
29			c.rectangle(p.x - phalfsize, p.y - phalfsize, 2phalfsize, 2phalfsize)
30			if p.connected_to is None:
31			c.set_source_rgba(0.8,0.8,1, 0.8)
32			else:
33			c.set_source_rgba(1,0,0,1)
34			c.fill_preserve()
35			c.set_source_rgb(0.8,0.8,0)
36			c.stroke()
37
38			def glue(self,item, handle, ix, iy):
39			gluerange = 10
40			mindist = -1;
41			minport = None
42			for p in self.ports:
43			dist = math.sqrt((ix-p.x)2 + (iy-p.y)2)
44			if dist < gluerange:
45			if not minport or dist<mindist:
46			mindist = dist
47			minport = p
48			return mindist, minport
49
50			def pre_update(self,context):
51			#print "PRE-UPDATE BLOCK"
52			pass
53
54			class DefaultBlockItem(BlockItem):
55			"""
56			This is a 'default block' with a certain number of input and output ports
57			shown depending on the values sent to __init__. It is drawn as a simple
58			box with the input ports on the left and the output ports on the right.
59			"""
60
61			def __init__(self, blockinstance):
62
63			self.blockinstance = blockinstance
64			inputs = len(blockinstance.blocktype.inputs)
65			outputs = len(blockinstance.blocktype.outputs)
66			super(DefaultBlockItem, self).__init__(64, 64)
67
68			eq = EqualsConstraint
69			bal = BalanceConstraint
70			handles = self._handles
71			h_nw = handles[NW]
72			h_ne = handles[NE]
73			h_sw = handles[SW]
74			h_se = handles[SE]
75
76			for i in range(inputs):
77			p = Port(self)
78			self.ports.append(p)
79			self._constraints.append(eq(p.x, h_nw.x))
80			self._constraints.append(bal(band=(h_nw.y, h_sw.y),v=p.y, balance=(0.5 + i)/inputs))
81
82			for i in range(outputs):
83			p = Port(self)
84			self.ports.append(p)
85			self._constraints.append(eq(p.x, h_ne.x))
86			self._constraints.append(bal(band=(h_ne.y,h_se.y),v=p.y, balance=(0.5 + i)/outputs))
87
88			def draw(self, context):
89			# draw the box itself
90			c = context.cairo
91			nw = self._handles[NW]
92			c.rectangle(nw.x, nw.y, self.width, self.height)
93			if context.hovered:
94			c.set_source_rgba(.8,.8,1, .8)
95			else:
96			c.set_source_rgba(1,1,1, .8)
97			c.fill_preserve()
98			c.set_source_rgb(0,0,0.8)
99			c.stroke()
100
101			text_center(c,self.width/2,self.height/2,self.blockinstance.name)
102
103			# now the draw the ports using the base class
104			super(DefaultBlockItem, self).draw(context)
105	jpye	1954