89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
from .joycon import JoyCon
|
|
|
|
|
|
# Preferably, this class gets merged into the
|
|
# parent class if approved by the original author
|
|
class PythonicJoyCon(JoyCon):
|
|
"""
|
|
A wrapper class for the JoyCon parent class.
|
|
This creates a more pythonic interface by
|
|
* using properties instead of requiring java-style getters and setters,
|
|
* bundles related xy/xyz data in tuples
|
|
* bundles the multiple measurements of the
|
|
gyroscope and accelerometer into a list
|
|
* Adds the option to invert the y and z axis of the left joycon
|
|
to make it match the right joycon. This is enabled by default
|
|
"""
|
|
|
|
def __init__(self, *a, invert_left_ime_yz=True, **kw):
|
|
super().__init__(*a, **kw)
|
|
self._ime_yz_coeff = -1 if invert_left_ime_yz and self.is_left() else 1
|
|
|
|
is_charging = property(JoyCon.get_battery_charging)
|
|
battery_level = property(JoyCon.get_battery_level)
|
|
|
|
r = property(JoyCon.get_button_r)
|
|
zr = property(JoyCon.get_button_zr)
|
|
plus = property(JoyCon.get_button_plus)
|
|
a = property(JoyCon.get_button_a)
|
|
b = property(JoyCon.get_button_b)
|
|
x = property(JoyCon.get_button_x)
|
|
y = property(JoyCon.get_button_y)
|
|
stick_r_btn = property(JoyCon.get_button_r_stick)
|
|
home = property(JoyCon.get_button_home)
|
|
right_sr = property(JoyCon.get_button_right_sr)
|
|
right_sl = property(JoyCon.get_button_right_sl)
|
|
|
|
l = property(JoyCon.get_button_l) # noqa: E741
|
|
zl = property(JoyCon.get_button_zl)
|
|
minus = property(JoyCon.get_button_minus)
|
|
stick_l_btn = property(JoyCon.get_button_l_stick)
|
|
up = property(JoyCon.get_button_up)
|
|
down = property(JoyCon.get_button_down)
|
|
left = property(JoyCon.get_button_left)
|
|
right = property(JoyCon.get_button_right)
|
|
capture = property(JoyCon.get_button_capture)
|
|
left_sr = property(JoyCon.get_button_left_sr)
|
|
left_sl = property(JoyCon.get_button_left_sl)
|
|
|
|
disconnect = JoyCon.disconnect_device
|
|
|
|
@property
|
|
def stick_l(self):
|
|
return (
|
|
self.get_stick_left_horizontal(),
|
|
self.get_stick_left_vertical(),
|
|
)
|
|
|
|
@property
|
|
def stick_r(self):
|
|
return (
|
|
self.get_stick_right_horizontal(),
|
|
self.get_stick_right_vertical(),
|
|
)
|
|
|
|
@property
|
|
def accel(self):
|
|
c = self._ime_yz_coeff
|
|
return [
|
|
(
|
|
self.get_accel_x(i),
|
|
self.get_accel_y(i) * c,
|
|
self.get_accel_z(i) * c,
|
|
)
|
|
for i in range(3)
|
|
]
|
|
|
|
@property
|
|
def accel_in_g(self):
|
|
c = 4.0 / 0x4000
|
|
c2 = c * self._ime_yz_coeff
|
|
return [
|
|
(
|
|
self.get_accel_x(i) * c,
|
|
self.get_accel_y(i) * c2,
|
|
self.get_accel_z(i) * c2,
|
|
)
|
|
for i in range(3)
|
|
]
|