path: root/awlsim/gui/geo2d.py

# -*- coding: utf-8 -*-
#
# AWL simulator - 2D geometry
#
# Copyright 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_loader.common import *


__all__ = [ "Point2D", "Vect2D", "Inter2D", "LineSeg2D", ]


class Base2D(object):
	EPSILON = 0.000001

	__slots__ = ()

	def __eq__(self, other):
		return self is other

	def __ne__(self, other):
		return not self.__eq__(other)

	__hash__ = None

	def __bool__(self):
		return True

	__nonzero__ = __bool__ # Python 2 compat

class BaseXY2D(Base2D):
	"""2D X/Y base object
	"""

	__slots__ = ( "x", "y", )

	def __init__(self, x=0, y=0):
		self.x = x
		self.y = y

	@property
	def xInt(self):
		if isInteger(self.x):
			return self.x
		return int(round(self.x))

	@property
	def yInt(self):
		if isInteger(self.y):
			return self.y
		return int(round(self.y))

	def __eq__(self, other):
		return self is other or\
		       (other is not None and\
		        abs(self.x - other.x) < self.EPSILON and\
			abs(self.y - other.y) < self.EPSILON)

	def __bool__(self):
		return bool(self.x or self.y)

class Point2D(BaseXY2D):
	"""2D point.
	"""

	def __repr__(self):
		return "Point2D(x=%f, y=%f)" % (self.x, self.y)

class Vect2D(BaseXY2D):
	"""2D vector.
	"""

	def __repr__(self):
		return "Vect2D(x=%f, y=%f)" % (self.x, self.y)

class Inter2D(Base2D):
	"""Intersection information for two 2D line segments.
	"""

	__slots__ = ( "point", "vect", "__intersects", )

	def __init__(self, point=None, vect=None, intersects=False):
		"""point => Intersection point, or None.
		vect => Intersection vector.
			None or Vect2D(0, 0), if the intersection is only in one point.
		intersects => True, if there is an intersection.
		"""
		self.point = point
		self.vect = vect or Vect2D()
		self.__intersects = intersects

	def __eq__(self, other):
		return self is other or\
		       (other is not None and\
		        self.point == other.point and\
		        self.vect == other.vect and\
		        self.__intersects == other.__intersects)

	def __bool__(self):
		return self.intersects

	@property
	def intersects(self):
		"""Returns True, if there is an intersection.
		"""
		return self.__intersects and\
		       self.point is not None

	@property
	def lineSeg(self):
		"""Get the line segment of the intersection.
		Returns None, if self.point is None.
		"""
		if self.point is None:
			return None
		return LineSeg2D(self.pointA, self.pointB)

	@property
	def pointA(self):
		"""Get the starting point of the intersection.
		Returns None, if there is no starting point.
		"""
		return self.point

	@property
	def pointB(self):
		"""Get the end point of the intersection.
		Returns None, if there is no end point.
		"""
		if self.point is None:
			return None
		return Point2D(self.point.x + self.vect.x,
			       self.point.y + self.vect.y)

	def __repr__(self):
		return "Inter2D(point=%s, vect=%s, intersects=%s)" % (
			self.point, self.vect, self.__intersects)

class LineSeg2D(Base2D):
	"""Line segment in 2D space.
	"""

	__slots__ = ( "pointA", "pointB", )

	@classmethod
	def fromCoords(cls, x0, y0, x1, y1):
		return cls(Point2D(x0, y0), Point2D(x1, y1))

	def __init__(self, pointA, pointB):
		self.pointA = pointA
		self.pointB = pointB

	def __eq__(self, other):
		return self is other or\
		       (other is not None and\
		        self.pointA == other.pointA and\
		        self.pointB == other.pointB)

	def __bool__(self):
		"""Returns True, if the segment is of non-zero length.
		"""
		return bool(self.vect)

	@property
	def isHorizontal(self):
		"""Returns True, if the line segment is parallel to the X axis.
		"""
		return self.pointA.y == self.pointB.y

	@property
	def isVertical(self):
		"""Returns True, if the line segment is parallel to the Y axis.
		"""
		return self.pointA.x == self.pointB.x

	@property
	def slope(self):
		"""Get the slope of the line segment.
		Raises ZeroDivisionError if the line segment is vertical.
		"""
		return float(self.pointB.y - self.pointA.y) / \
			    (self.pointB.x - self.pointA.x)

	@property
	def intercept(self):
		"""Get the Y value of the Y axis crossing of this line.
		Raises ZeroDivisionError if the line segment is vertical.
		"""
		return self.pointA.y - (self.pointA.x * self.slope)

	@property
	def vect(self):
		"""Get the line segment vector.
		"""
		return Vect2D(-self.pointA.x + self.pointB.x,
			      -self.pointA.y + self.pointB.y)

	@staticmethod
	def __inRect(x, y, diaPointA, diaPointB):
		"""Check if point (x,y) is within an axis-aligned rect
		with the diagonal (diaPointA,diaPointB).
		"""
		diaMinX, diaMaxX = min(diaPointA.x, diaPointB.x),\
				   max(diaPointA.x, diaPointB.x)
		diaMinY, diaMaxY = min(diaPointA.y, diaPointB.y),\
				   max(diaPointA.y, diaPointB.y)
		return (x >= diaMinX and x <= diaMaxX and\
			y >= diaMinY and y <= diaMaxY)

	def __intersectionAligned(self, other):
		"""Get the intersection (if any) of two aligned line segments.
		'self' and 'other' must be aligned in order for this to
		return correct results.
		"""

		def find(selfPointA, selfPointB, otherPointA, otherPointB):
			for interA in (selfPointA, selfPointB):
				if not self.__inRect(interA.x, interA.y,
						     otherPointA, otherPointB):
					continue
				for interB in (otherPointA, otherPointB,
					       selfPointA, selfPointB):
					if interA == interB:
						continue
					if not self.__inRect(interB.x, interB.y,
							     selfPointA, selfPointB) or\
					   not self.__inRect(interB.x, interB.y,
							     otherPointA, otherPointB):
						continue
					return Inter2D(point=Point2D(interA.x, interA.y),
						       vect=Vect2D(interB.x - interA.x,
								   interB.y - interA.y),
						       intersects=True)
				return Inter2D(point=Point2D(interA.x, interA.y),
					       intersects=True)
			return None

		inter = find(self.pointA, self.pointB,
			     other.pointA, other.pointB)
		if inter is None:
			# Swap self and other
			inter = find(other.pointA, other.pointB,
				     self.pointA, self.pointB)
		if inter is None:
			return Inter2D()
		return inter

	def __intersectionVertical(self, other):
		"""Get the intersection of a vertical line segment 'self'
		and a non-vertical line segment 'other'.
		'self' must be vertical in order for this to
		return correct results.
		"""
		x = self.pointA.x
		y = (x - other.pointA.x) * other.slope + other.pointA.y
		return Inter2D(point=Point2D(x, y),
			       intersects=self.__inRect(x, y, self.pointA, self.pointB) and\
					  self.__inRect(x, y, other.pointA, other.pointB))

	def intersection(self, other):
		"""Get the intersection of this line segment
		with another line segment.
		Returns an Inter2D.
		"""
		try:
			selfVert, otherVert = self.isVertical, other.isVertical
			if selfVert and otherVert:
				# Both line segments are vertical.
				# If they are aligned, they might overlap.
				if self.pointA.x == other.pointA.x:
					return self.__intersectionAligned(other)
				return Inter2D()
			elif selfVert:
				# self is vertical. other is not vertical.
				return self.__intersectionVertical(other)
			elif otherVert:
				# self is not vertical. other is vertical.
				return other.__intersectionVertical(self)
			# Get the intersection of two arbitrary
			# non-vertical line segments.
			selfSlope, otherSlope = self.slope, other.slope
			selfInter, otherInter = self.intercept, other.intercept
			try:
				x = (otherInter - selfInter) / \
				    (selfSlope - otherSlope)
				y = selfSlope * x + selfInter
				assert(abs(y - (otherSlope * x + otherInter)) < self.EPSILON)
			except ZeroDivisionError:
				# The line segments are parallel.
				# If they have the same intercept they are aligned
				# and might overlap.
				if abs(selfInter - otherInter) < self.EPSILON:
					return self.__intersectionAligned(other)
				return Inter2D()
			return Inter2D(point=Point2D(x, y),
				       intersects=self.__inRect(x, y, self.pointA, self.pointB) and\
						  self.__inRect(x, y, other.pointA, other.pointB))
		except ZeroDivisionError:
			pass
		return Inter2D()

	def __repr__(self):
		return "LineSeg2D(pointA=%s, pointB=%s)" % (
			self.pointA, self.pointB)