summaryrefslogtreecommitdiff
path: root/grc/python/Connection.py
blob: 341dd2d82109b62fad59f8fd80d4f86ec0769ec7 (plain)
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
"""
Copyright 2008-2011 Free Software Foundation, Inc.
This file is part of GNU Radio

GNU Radio Companion 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.

GNU Radio Companion 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
"""

import Constants
from .. base.Element import Element
from .. base.Connection import Connection as _Connection
from .. gui.Connection import Connection as _GUIConnection

class Connection(_Connection, _GUIConnection):

	def __init__(self, **kwargs):
		_Connection.__init__(self, **kwargs)
		_GUIConnection.__init__(self)

	def is_msg(self):
		return self.get_source().get_type() == self.get_sink().get_type() == 'msg'

	def is_message(self):
		return self.get_source().get_type() == self.get_sink().get_type() == 'message'

	def validate(self):
		"""
		Validate the connections.
		The ports must match in io size.
		"""
		Element.validate(self)
		source_size = Constants.TYPE_TO_SIZEOF[self.get_source().get_type()] * self.get_source().get_vlen()
		sink_size = Constants.TYPE_TO_SIZEOF[self.get_sink().get_type()] * self.get_sink().get_vlen()
		if source_size != sink_size:
			self.add_error_message('Source IO size "%s" does not match sink IO size "%s".'%(source_size, sink_size))