"""
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))