python_arango_ogm.utils.math_util

  1from dataclasses import dataclass, asdict
  2from enum import Enum
  3from typing import Sequence
  4from .f_range import frange
  5
  6NORMAL_RANGE = frange(0.0, 1.0, upper_inclusive=True)
  7
  8def safe_float(obj) -> bool:
  9  if isinstance(obj, float):
 10    return obj
 11
 12  try:
 13    f = float(obj)
 14  except ValueError as x:
 15    f = 0
 16
 17  return f
 18
 19def clamp(x: float, minval=0.0, maxval=1.0):
 20  return max(min(x, maxval), minval)
 21
 22def clamp_sequence(seq: Sequence, minval=0.0, maxval=1.0):
 23  return [clamp(x, minval, maxval) for x in seq]
 24
 25
 26LerpType = Enum('LerpType', 'Linear Cube RootCube RootSquare Square')
 27
 28def lerp(pct, min, max, lerp_type: LerpType = LerpType.Linear):
 29  if max <= min:
 30    raise ValueError(f"Maximum should be greater than minimum Max:{max}, Min:{min}")
 31
 32  if pct not in NORMAL_RANGE:
 33    raise ValueError(f"Percentage should be between 0.0 and 1.0:  Is {pct}")
 34
 35  return min + pct * (max - min)
 36
 37def delerp(val, min, max, lerp_type: LerpType = LerpType.Linear):
 38  if mag <= min:
 39    raise ValueError(f"Maximum should be greater than minimum Max:{max}, Min:{min}")
 40
 41  if val < min or val > max:
 42    raise ValueError(f"Value should be between minimum and maximum:  Min:{min} <= Value:{val} <= Max:{max}")
 43
 44  return val - min / max - min
 45
 46@dataclass  # (frozen=True)
 47class Vector2d():
 48  """ Attributes for graph """
 49  x: float = 0.0
 50  y: float = 0.0
 51
 52  def __init__(self, obj1=0, obj2=None):
 53    if isinstance(obj1, tuple):
 54      self.x, self.y = obj1
 55    elif isinstance(obj1, Vector2d):
 56      self.x, self.y = other.x, other.y
 57    elif obj1 is not None and obj2 is not None:
 58      self.x, self.y = obj1, obj2
 59
 60  def to_dict(self):
 61    return asdict(self)
 62
 63  def __neg__(self):
 64    return Vector2d(-self.x, -self.y)
 65
 66  def __add__(self, other):
 67    return Vector2d(self.x + other.x, self.y + other.y)
 68
 69  def __sub__(self, other):
 70    return self.__add__(-other)
 71
 72  def __mul__(self, other):
 73    return Vector2d(self.x * other.x, self.y * other.y)
 74
 75  def __truediv__(self, other):
 76    return self.__mul__(-other)
 77
 78  def __iter__(self):
 79    self.iterstate = None
 80    return self
 81
 82  def __next__(self):
 83    state = self.iterstate  # getattr(self, 'state', -1)
 84    result = None
 85    if state == None:
 86      result = self.x
 87      self.iterstate = 'x'
 88    elif state == 'x':
 89      result = self.y
 90      self.iterstate = 'y'
 91    else:
 92      raise StopIteration('No more items (x & y only)')
 93    return result
 94
 95@dataclass
 96class Vector3d(Vector2d):
 97  z: float = 0.0
 98
 99@dataclass
100class Rect2d():
101  origin: Vector2d
102  size: Vector2d
103
104  def __init__(self, a=None, b=None, c=None, d=None):
105    if None not in [a, b, c, d]:
106      self.origin = Vector2d(a, b)
107      self.size = Vector2d(c, d)
108    else:
109      if isinstance(a, Vector2d):
110        self.origin = a
111      elif isinstance(a, tuple):
112        self.origin = Vector2d(a)
113
114      if isinstance(b, Vector2d):
115        self.size = b
116      elif isinstance(a, tuple):
117        self.size = Vector2d(b)
118
119  @property
120  def right(self):
121    return self.origin.x + self.size.x
122
123  @property
124  def bottom(self):
125    return self.origin.y + self.size.y
126
127  def __iter__(self):
128    self.iterstate = 0
129    self.iteritems = [self.origin.x, self.origin.y, self.size.x, self.size.y]
130    return self
131
132  def __next__(self):
133    state = self.iterstate  # getattr(self, 'state', -1)
134    if state > len(self.iteritems) - 1:
135      raise StopIteration('No more items (x,y,w,h) only')
136
137    self.iterstate += 1
138    return self.iteritems[state]
139
140  def __add__(self, other):
141    result = None
142    if isinstance(other, Rect2d):
143      result = Rect2d(self.origin + other.origin, self.size + other.size)
144    elif isinstance(other, Vector2d):
145      result = Rect2d(self.origin + other, self.size + other)
146    return result
147
148  def __sub__(self, other):
149    return self.__add__(other)
150
151  def __mul__(self, other):
152    result = None
153    if isinstance(other, Rect2d):
154      result = Rect2d(self.origin * other.origin, self.size * other.size)
155    elif isinstance(other, Vector2d):
156      result = Rect2d(self.origin * other, self.size * other)
157    return result
158
159  def __truediv__(self, other):
160    return self.__mul__(-other)
161
162  def __neg__(self):
163    return Rect2d(-self.origin, -self.size)