225 lines
6.3 KiB
Python
225 lines
6.3 KiB
Python
import colorsys
|
|
|
|
class StructureBase(object):
|
|
@classmethod
|
|
def _attrs(cls):
|
|
a = []
|
|
i = cls()
|
|
for attr in dir(i):
|
|
if not attr.startswith('_') and not callable(getattr(i, attr)):
|
|
a.append(attr)
|
|
return a
|
|
|
|
def _dict(self):
|
|
d = {}
|
|
attrs = self._attrs()
|
|
for a in attrs:
|
|
d[a] = getattr(self, a)
|
|
return d
|
|
|
|
def _dict_items(self):
|
|
for a in reversed(self._attrs()):
|
|
yield a, getattr(self,a)
|
|
|
|
def _dict_string(self):
|
|
return ', '.join('{}={}'.format(key, val) for key, val in self._dict_items())
|
|
|
|
def __repr__(self):
|
|
return unicode(self)
|
|
|
|
def __str__(self):
|
|
return unicode(self).encode('utf-8')
|
|
|
|
def __unicode__(self):
|
|
return u'{} [{}]'.format(self.__class__.__name__, self._dict_string())
|
|
|
|
def __eq__(self, other):
|
|
eq = True
|
|
for attr in self._attrs():
|
|
val = getattr(self, attr)
|
|
if hasattr(other, attr):
|
|
eq &= getattr(other, attr) == val
|
|
else:
|
|
return False
|
|
return eq
|
|
|
|
def __getitem__(self, index):
|
|
return list(self)[index]
|
|
|
|
def __len__(self):
|
|
return len(self._attrs())
|
|
|
|
def __ne__(self, other):
|
|
return not(self == other)
|
|
|
|
@classmethod
|
|
def is_like(cls, obj):
|
|
"""Checks if the object is like the current class
|
|
If it has everything a duck has, then it's good."""
|
|
for key in cls._attrs():
|
|
if not hasattr(obj, key):
|
|
return False
|
|
return True
|
|
|
|
@classmethod
|
|
def from_value(cls, value):
|
|
try:
|
|
return cls(*value)
|
|
except TypeError:
|
|
return cls()
|
|
|
|
class Position(StructureBase):
|
|
def __init__(self, x=0, y=0):
|
|
self.x = x
|
|
self.y = y
|
|
|
|
def __add__(self, other):
|
|
other = Position.from_value(other)
|
|
return Position(self.x+other.x, self.y+other.y)
|
|
|
|
def __sub__(self, other):
|
|
other = Position.from_value(other)
|
|
return Position(self.x-other.x, self.y-other.y)
|
|
|
|
def __mul__(self, value):
|
|
if isinstance(value, (int, long, float)):
|
|
return Position(self.x*value, self.y*value)
|
|
raise Exception("Cannot multiply position by {}.".format(value.__class__.__name__))
|
|
|
|
def __iter__(self):
|
|
yield self.x
|
|
yield self.y
|
|
|
|
|
|
class Size(StructureBase):
|
|
def __init__(self, width=0, height=0):
|
|
self.width = width
|
|
self.height = height
|
|
|
|
def __add__(self, other):
|
|
other = Size.from_value(other)
|
|
return Size(self.width+other.width, self.height+other.height)
|
|
|
|
def __sub__(self, other):
|
|
other = Size.from_value(other)
|
|
return Size(self.width-other.width, self.height-other.height)
|
|
|
|
def __iter__(self):
|
|
yield self.width
|
|
yield self.height
|
|
|
|
|
|
class Color(StructureBase):
|
|
def __init__(self, r=0, g=0, b=0, a=1):
|
|
self.r = float(r)
|
|
self.g = float(g)
|
|
self.b = float(b)
|
|
self.a = float(a)
|
|
|
|
def __add__(self, other):
|
|
"""Blend two colors using self as the background and other as the foreground"""
|
|
other = Color.from_value(other)
|
|
r = Color()
|
|
r.a = 1 - (1-other.a)*(1-self.a)
|
|
r_a_inv = self.a*(1-other.a)/r.a
|
|
o_factor = other.a/r.a
|
|
r.r = other.r*o_factor + self.r*r_a_inv
|
|
r.g = other.g*o_factor + self.g*r_a_inv
|
|
r.b = other.b*o_factor + self.b*r_a_inv
|
|
return r
|
|
|
|
def brightness(self, value):
|
|
h, s, v = colorsys.rgb_to_hsv(self.r, self.g, self.b)
|
|
v = value
|
|
self.r, self.g, self.b = colorsys.hsv_to_rgb(h, s, v)
|
|
|
|
def hue(self, value):
|
|
h, s, v = colorsys.rgb_to_hsv(self.r, self.g, self.b)
|
|
h = value
|
|
self.r, self.g, self.b = colorsys.hsv_to_rgb(h, s, v)
|
|
|
|
def saturation(self, value):
|
|
h, s, v = colorsys.rgb_to_hsv(self.r, self.g, self.b)
|
|
s = value
|
|
self.r, self.g, self.b = colorsys.hsv_to_rgb(h, s, v)
|
|
|
|
def from_hsv(self, h, s, v):
|
|
self.r, self.g, self.b = colorsys.hsv_to_rgb(h, s, v)
|
|
return self
|
|
|
|
def __iter__(self):
|
|
yield self.r
|
|
yield self.g
|
|
yield self.b
|
|
yield self.a
|
|
|
|
@classmethod
|
|
def hex_to_rgba(cls, hex_val):
|
|
"""Converts a hex string in the format 0xFFAABB/CC or an integer to rgba"""
|
|
if isinstance(hex_val, (long, int)):
|
|
hexstring = "{0:x}".format(abs(hex_val))
|
|
elif isinstance(hex_val, basestring):
|
|
hexstring = hexval[2:]
|
|
else:
|
|
return None
|
|
if len(hexstring) % 2 != 0:
|
|
hexstring = '0' + hexstring
|
|
if len(hexstring) != 6 and len(hexstring) != 8:
|
|
return None
|
|
ba = []
|
|
for i in xrange(0, len(hexstring), 2):
|
|
ba.append(int(hexstring[i:i+2], 16))
|
|
c = cls(*ba)
|
|
return c
|
|
|
|
@classmethod
|
|
def from_value(cls, value):
|
|
try:
|
|
return cls(*value)
|
|
except TypeError:
|
|
c = cls.hex_to_rgba(value)
|
|
if c is not None:
|
|
return c
|
|
return cls()
|
|
|
|
class Rectangle(StructureBase):
|
|
def __init__(self, position=None, size=None):
|
|
""" A rectangle object with coordinates and size.
|
|
position: can take the first two values of an array or a Position object
|
|
size: can take the first two values of an array or a Size object
|
|
"""
|
|
self._position = None
|
|
self._size = None
|
|
self.position = position
|
|
self.size = size
|
|
|
|
@property
|
|
def position(self):
|
|
return self._position
|
|
|
|
@position.setter
|
|
def position(self, position):
|
|
self._position = Position.from_value(position)
|
|
|
|
def __iter__(self):
|
|
yield list(self.position)
|
|
yield list(self.size)
|
|
|
|
@property
|
|
def size(self):
|
|
return self._size
|
|
|
|
@size.setter
|
|
def size(self, size):
|
|
self._size = Size.from_value(size)
|
|
|
|
def intersects_with(self, position):
|
|
"""Checks if the position is within the bounds of the rectangle including the edges"""
|
|
pos = Position.from_value(position)
|
|
if pos.x < self.position.x + self.size.width and\
|
|
pos.x >= self.position.x:
|
|
if pos.y < self.position.y + self.size.height and\
|
|
pos.y >= self.position.y:
|
|
return True
|
|
return False
|
|
|