Example OO wrapper around the Vector2 struct, by @Emtyloc
This commit is contained in:
Richard Smith
parent
c9a5ec09e3
commit
fde8354a8a
1 changed files with 260 additions and 0 deletions
260
examples/extra/vector2_extended.py
Normal file
260
examples/extra/vector2_extended.py
Normal file
|
|
@ -0,0 +1,260 @@
|
|||
# An OO wrapper around the Vector2 struct, by @Emtyloc
|
||||
|
||||
from pyray import *
|
||||
|
||||
class Vector2Ex(list):
|
||||
def __init__(self, x, y):
|
||||
super(Vector2Ex, self).__init__([x, y])
|
||||
|
||||
@property
|
||||
def x(self):
|
||||
return self[0]
|
||||
|
||||
@x.setter
|
||||
def x(self, value):
|
||||
self[0]= value
|
||||
|
||||
@property
|
||||
def y(self):
|
||||
return self[1]
|
||||
|
||||
@y.setter
|
||||
def y(self, value):
|
||||
self[1]= value
|
||||
|
||||
@staticmethod
|
||||
def to_Vec2(v: Vector2):
|
||||
"""
|
||||
Cast Vector2 to Vec2.
|
||||
"""
|
||||
return Vector2Ex(v.x, v.y)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"{self.x}, {self.y}"
|
||||
|
||||
def __eq__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
return self.x == other.x and self.y == other.y
|
||||
return False
|
||||
|
||||
def __add__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
return Vector2Ex(self.x + other.x, self.y + other.y)
|
||||
return Vector2Ex(self.x + other, self.y + other)
|
||||
|
||||
def __iadd__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
self.x += other.x
|
||||
self.y += other.y
|
||||
else:
|
||||
res = vector2_add_value(self, other)
|
||||
self.x = res.x
|
||||
self.y = res.y
|
||||
return self
|
||||
|
||||
def __radd__(self, other):
|
||||
return self + other
|
||||
|
||||
def __sub__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
return Vector2Ex(self.x - other.x, self.y - other.y)
|
||||
return Vector2Ex(self.x - other, self.y - other)
|
||||
|
||||
def __isub__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
self.x -= other.x
|
||||
self.y -= other.y
|
||||
else:
|
||||
self.x -= other
|
||||
self.y -= other
|
||||
return self
|
||||
|
||||
def __rsub__(self, other):
|
||||
return Vector2Ex(other - self.x, other - self.y)
|
||||
|
||||
def __mul__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
res = vector2_multiply(self, other)
|
||||
return self.to_Vec2(res)
|
||||
return Vector2Ex(self.x * other, self.y * other)
|
||||
|
||||
def __imul__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
res = vector2_multiply(self, other)
|
||||
else:
|
||||
res = vector2_scale(self, other)
|
||||
self.x = res.x
|
||||
self.y = res.y
|
||||
return self
|
||||
|
||||
def __truediv__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
res = vector_2divide(self, other)
|
||||
return self.to_Vec2(res)
|
||||
return Vector2Ex(self.x / other, self.y / other)
|
||||
|
||||
def __itruediv__(self, other):
|
||||
if isinstance(other, Vector2Ex):
|
||||
res = vector_2divide(self, other)
|
||||
else:
|
||||
res = vector2_scale(self, 1/other)
|
||||
self.x = res.x
|
||||
self.y = res.y
|
||||
return self
|
||||
|
||||
def __neg__(self):
|
||||
return Vector2Ex(-self.x, -self.y)
|
||||
|
||||
def __pos__(self):
|
||||
return Vector2Ex(self.x, self.y)
|
||||
|
||||
def __pow__(self, exponent):
|
||||
return Vector2Ex(self.x ** exponent, self.y ** exponent)
|
||||
|
||||
# PyRay mapped vector2 functions
|
||||
|
||||
def angle(self, vec2):
|
||||
return vector2_angle(self, vec2)
|
||||
|
||||
def clamp(self, min_vec2, max_vec2):
|
||||
res = vector2_clamp(self, min_vec2, max_vec2)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def clamp_value(self, min_val: float, max_val: float):
|
||||
res = vector2_clamp_value(self, min_val, max_val)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def distance(self, vec2):
|
||||
return vector_2distance(self, vec2)
|
||||
|
||||
def distance_sqr(self, vec2) -> float:
|
||||
return vector_2distance_sqr(self, vec2)
|
||||
|
||||
def dot_product(self, vec2) -> float:
|
||||
return vector_2dot_product(self, vec2)
|
||||
|
||||
def invert(self):
|
||||
res = vector2_invert(self)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def length(self):
|
||||
return vector2_length(self)
|
||||
|
||||
def length_sqr(self) -> float:
|
||||
return vector2_length_sqr(self)
|
||||
|
||||
def lerp(self, vec2, amount: float):
|
||||
res = vector2_lerp(self, vec2, amount)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def move_towards(self, target_vec2, max_distance: float):
|
||||
res = vector2_move_towards(self, target_vec2, max_distance)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def negate(self):
|
||||
res = vector2_negate(self)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def normalize(self):
|
||||
res = vector2_normalize(self)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def reflect(self, normal_vec2):
|
||||
res = vector2_reflect(self, normal_vec2)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def rotate(self, angle: float):
|
||||
res = vector2_rotate(self, angle)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
def transform(self, mat: Matrix):
|
||||
res = vector2_transform(self, mat)
|
||||
return self.to_Vec2(res)
|
||||
|
||||
@staticmethod
|
||||
def line_angle(start_vec2, end_vec2) -> float:
|
||||
return vector2_line_angle(start_vec2, end_vec2)
|
||||
|
||||
@staticmethod
|
||||
def one():
|
||||
return Vector2Ex(1, 1)
|
||||
|
||||
@staticmethod
|
||||
def zero():
|
||||
return Vector2Ex(0, 0)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Arithmetic ops
|
||||
v1 = Vector2Ex(5, 5)
|
||||
v2 = Vector2Ex(10, 10)
|
||||
|
||||
print(v1 + v2) # 15, 15
|
||||
print(v1 - v2) # -5, -5
|
||||
|
||||
print(v1 * v2) # 50.0, 50.0
|
||||
print(v1 / v2) # 0.5, 0.5
|
||||
|
||||
print(v1 * 2) # 10, 10
|
||||
print(v2 / 2) # 5.0, 5.0
|
||||
|
||||
v1+=v2
|
||||
print(v1) #15, 15
|
||||
v2-=v1
|
||||
print(v2) #-5, -5
|
||||
|
||||
v1/=-v2
|
||||
print(v1) #3.0, 3.0
|
||||
v2*=v1
|
||||
print(v2) #-15.0, -15.0
|
||||
|
||||
v3 = Vector2Ex(3, 5)
|
||||
print(v3 ** 2) #9, 25
|
||||
|
||||
v1 = Vector2Ex.one()
|
||||
print(v1)
|
||||
|
||||
v0 = Vector2Ex.zero()
|
||||
print(v0)
|
||||
|
||||
# Vector2 pyray methods
|
||||
v1 = Vector2Ex(3, 4)
|
||||
v2 = Vector2Ex(1, 2)
|
||||
v_min = Vector2Ex(0, 0)
|
||||
v_max = Vector2Ex(5, 5)
|
||||
|
||||
print("Angle:", v1.angle(v2))
|
||||
|
||||
print("Clamp:", v1.clamp(v_min, v_max))
|
||||
|
||||
print("Clamp value:", v1.clamp_value(1.5, 3.5))
|
||||
|
||||
print("Distance:", v1.distance(v2))
|
||||
|
||||
print("Distance Sqr:", v1.distance_sqr(v2))
|
||||
|
||||
print("Dot Product:", v1.dot_product(v2))
|
||||
|
||||
print("Invert:", v1.invert())
|
||||
|
||||
print("Length:", v1.length())
|
||||
|
||||
print("Length Sqr:", v1.length_sqr())
|
||||
|
||||
print("Lerp:", v1.lerp(v2, 0.5))
|
||||
|
||||
print("Line Angle:", Vector2Ex.line_angle(v1, v2))
|
||||
|
||||
print("Move Towards:", v1.move_towards(v2, 0.5))
|
||||
|
||||
print("Negate:", v1.negate())
|
||||
|
||||
print("Normalize:", v1.normalize())
|
||||
|
||||
print("Reflect:", v1.reflect(v2))
|
||||
|
||||
print("Rotate:", v1.rotate(45))
|
||||
|
||||
# I don't know why this not work
|
||||
# mat = Matrix2x2(1, 0, 0, 1)
|
||||
# print("Transform:", v1.transform(mat))
|
||||
Reference in a new issue