extents: add types

* lint code
This commit is contained in:
Rotzbua 2023-02-07 15:26:58 +01:00 committed by Florian Festi
parent 8db417a134
commit 64dca171b4
1 changed files with 16 additions and 16 deletions

View File

@ -1,34 +1,34 @@
class Extents: class Extents:
__slots__ = "xmin ymin xmax ymax".split() __slots__ = "xmin ymin xmax ymax".split()
def __init__(self,xmin=float('inf'),ymin=float('inf'),xmax=float('-inf'),ymax=float('-inf')) -> None: def __init__(self, xmin: float = float('inf'), ymin: float = float('inf'), xmax: float = float('-inf'), ymax: float = float('-inf')) -> None:
self.xmin = xmin self.xmin = xmin
self.ymin = ymin self.ymin = ymin
self.xmax = xmax self.xmax = xmax
self.ymax = ymax self.ymax = ymax
def add(self,x,y): def add(self, x: float, y: float) -> None:
self.xmin = min(self.xmin,x) self.xmin = min(self.xmin, x)
self.xmax = max(self.xmax,x) self.xmax = max(self.xmax, x)
self.ymin = min(self.ymin,y) self.ymin = min(self.ymin, y)
self.ymax = max(self.ymax,y) self.ymax = max(self.ymax, y)
def extend(self,l): def extend(self, l) -> None:
for x,y in l: for x, y in l:
self.add(x,y) self.add(x, y)
def __add__(self,extent): def __add__(self, extent):
#todo: why can this happen? # todo: why can this happen?
if extent == 0: if extent == 0:
return Extents(self.xmin,self.ymin,self.xmax,self.ymax) return Extents(self.xmin, self.ymin, self.xmax, self.ymax)
return Extents( return Extents(
min(self.xmin,extent.xmin),min(self.ymin,extent.ymin), min(self.xmin, extent.xmin), min(self.ymin, extent.ymin),
max(self.xmax,extent.xmax),max(self.ymax,extent.ymax) max(self.xmax, extent.xmax), max(self.ymax, extent.ymax)
) )
def __radd__(self,extent): def __radd__(self, extent):
if extent == 0: if extent == 0:
return Extents(self.xmin,self.ymin,self.xmax,self.ymax) return Extents(self.xmin, self.ymin, self.xmax, self.ymax)
return self.__add__(extent) return self.__add__(extent)
@property @property