n3tael/p3he
1
0
Fork 0
forked from bedohswe/p3he
Code Pull Requests Activity

Add __iadd__ and __isub__ to GyroVector.

Browse Source
This commit is contained in:
bʰedoh₂ swé 2024-04-06 23:45:25 +05:00
parent c745c6a82b
commit 1494f28a78
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gyro.py
View File

@ -100,6 +100,17 @@ class GyroVector:
cAdd, cGyr = MobiusAddGyr(gA.cPos, gB.cPos)
return GyroVector(cAdd, gA.cRot * gB.cRot * cGyr)
def __iadd__(self, gA):
cAdd, cGyr = MobiusAddGyr(self.cPos, gA.cPos)
self.cPos = cAdd
self.cRot = self.cRot * gA.cRot * cGyr
def __isub__(self, gA):
ngA = -gA
cAdd, cGyr = MobiusAddGyr(self.cPos, ngA.cPos)
self.cPos = cAdd
self.cRot = self.cRot * ngA.cRot * cGyr
def __neg__(self):
return GyroVector(-(self.cRot * self.cPos), 1/self.cRot)