1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# -*- coding: utf-8 -*-
#
# AWL simulator - FUP compiler - Connection
#
# Copyright 2016-2017 Michael Buesch <m@bues.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
from __future__ import division, absolute_import, print_function, unicode_literals
from awlsim.common.compat import *
from awlsim.common.xmlfactory import *
from awlsim.fupcompiler.fupcompiler_base import *
class FupCompiler_ConnFactory(XmlFactory):
def parser_open(self, tag=None):
self.inConn = False
XmlFactory.parser_open(self, tag)
def parser_beginTag(self, tag):
if not self.inConn:
if tag.name == "connection":
self.inConn = True
pos = tag.getAttrInt("pos")
dirIn = tag.getAttrInt("dir_in")
dirOut = tag.getAttrInt("dir_out")
wireId = tag.getAttrInt("wire")
conn = FupCompiler_Conn(self.elem,
pos, dirIn, dirOut, wireId)
if not self.elem.addConn(conn):
raise self.Error("Invalid connection")
return
XmlFactory.parser_beginTag(self, tag)
def parser_endTag(self, tag):
if self.inConn:
if tag.name == "connection":
self.inConn = False
return
else:
if tag.name == "connections":
self.parser_finish()
return
XmlFactory.parser_endTag(self, tag)
class FupCompiler_Conn(FupCompiler_BaseObj):
factory = FupCompiler_ConnFactory
def __init__(self, elem, pos, dirIn, dirOut, wireId):
FupCompiler_BaseObj.__init__(self)
self.elem = elem # FupCompiler_Elem
self.pos = pos # Position index
self.dirIn = bool(dirIn) # Input
self.dirOut = bool(dirOut) # Output
self.wireId = wireId # Wire ID number
self.wire = None
def getConnected(self, getOutputs=False, getInputs=False):
"""Get all other connections that are connected
via self.wire to this connection.
This excludes self.
If 'getOutputs' is True, connections with dirOut=True are returned.
If 'getInputs' is True, connections with dirIn=True are returned.
"""
for conn in self.wire.connections:
if conn is not self and\
((conn.dirOut and getOutputs) or\
(conn.dirIn and getInputs)):
yield conn
def getConnectedElems(self, viaOut=False, viaIn=False):
"""Get all elements that are connected to this connection.
If 'viaOut' is True, elements connected to the wire via OUT
connection are returned.
If 'viaIn' is True, elements connected to the wire via OUT
connection are returned.
If neither 'viaOut' nor 'viaIn' is True, no element is returned.
The element that belongs to 'self' is not returned.
"""
for conn in self.getConnected(getOutputs=viaOut, getInputs=viaIn):
yield conn.elem
|
