Added s property for rne
@@ -500,6 +500,24 @@ class ET(BaseET):
def __add__(self, other: "ET") -> "rtb.ETS":
return self.__mul__(other)
+ @property
+ def s(self) -> NDArray[np.float64]: # pragma: nocover
+ if self.axis[1] == "x":
+ if self.axis[0] == "R":
+ return np.array([0, 0, 0, 1, 0, 0])
+ else:
+ return np.array([1, 0, 0, 0, 0, 0])
+ elif self.axis[1] == "y":
+ if self.axis[0] == "R":
+ return np.array([0, 0, 0, 0, 1, 0])
+ else:
+ return np.array([0, 1, 0, 0, 0, 0])
+ else:
+ if self.axis[0] == "R":
+ return np.array([0, 0, 0, 0, 0, 1])
+ else:
+ return np.array([0, 0, 1, 0, 0, 0])
+
@classmethod
def Rx(
cls, eta: Union[float, Sym, None] = None, unit: str = "rad", **kwargs
