Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Updates to Joystick Gremlin plugin #82

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Updates to Joystick Gremlin plugin #82

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
54 changes: 38 additions & 16 deletions joystick_diagrams/plugins/joystick_gremlin_plugin/joystick_gremlin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@

from joystick_diagrams.exceptions import JoystickDiagramsError
from joystick_diagrams.input.button import Button
from joystick_diagrams.input.axis import Axis, AxisDirection
from joystick_diagrams.input.hat import Hat, HatDirection
from joystick_diagrams.input.profile_collection import ProfileCollection

Expand Down Expand Up @@ -91,10 +92,10 @@ def create_dictionary(self) -> ProfileCollection:

match bind_type:
case "axis":
# TODO Axis Support requires DILL AxisMap to infer from identifiers to types
_logger.info(
f"AXIS binds not currently supported for {bind_type} {bind_identifier}"
)
if bind_description:
_device_obj.create_input(
Axis(self.map_axis(bind_identifier)), bind_description
)
case "button":
if bind_description:
_device_obj.create_input(
Expand All @@ -113,6 +114,24 @@ def create_dictionary(self) -> ProfileCollection:

return profile_collection

def map_axis(self, id):
match id:
case 1:
return AxisDirection.X
case 2:
return AxisDirection.Y
case 3:
return AxisDirection.Z
case 4:
return AxisDirection.RX
case 5:
return AxisDirection.RY
case 6:
return AxisDirection.RZ
case 7:
return AxisDirection.SLIDER


def extract_hats(self, hat_node: minidom.Element) -> list[tuple[Hat, str]]:
"""Extract the hat positions for a given HAT node.

Expand Down Expand Up @@ -146,19 +165,21 @@ def extract_hats(self, hat_node: minidom.Element) -> list[tuple[Hat, str]]:
_logger.debug(f"Hat Containers: {len(filtered_hat_containers)}")

if filtered_hat_containers:
hat_mappings = self.handle_hat_button_container(
hat_id, filtered_hat_containers
self.handle_hat_button_container(
hat_id, filtered_hat_containers, hat_mappings
)

if basic_containers:
hat_mappings = self.handle_virtual_button_container(hat_id, hat_containers)
self.handle_virtual_button_container(
hat_id, hat_containers, hat_mappings
)

_logger.debug(f"Hat Mappings: {hat_mappings}")
return hat_mappings

def handle_virtual_button_container(
self, hat_id, containers: minidom.NodeList[minidom.Element]
self, hat_id, containers: minidom.NodeList[minidom.Element], hat_mappings
):
hat_mappings = []
for container in containers:
# Check if we have a top level description
container_description = (
Expand All @@ -178,7 +199,7 @@ def handle_virtual_button_container(

# Skip processing if we have no descriptions
if not container_description:
print("No point continuing")
_logger.debug(f"No point continuing")
continue

virtual_buttons: minidom.Element = container.getElementsByTagName(
Expand All @@ -190,7 +211,7 @@ def handle_virtual_button_container(
continue

if len(virtual_buttons) != 1:
print("Not expected number of virtual button elements")
_logger.debug(f"Not expected number of virtual button elements")
continue

attributes = [x for x in virtual_buttons[0].attributes.keys()]
Expand All @@ -202,11 +223,10 @@ def handle_virtual_button_container(
container_description,
)
)

return hat_mappings
_logger.debug(f"Hat Mappings from Virtual Button: {hat_mappings}")

def handle_hat_button_container(
self, hat_id, hat_containers: minidom.NodeList[minidom.Element]
self, hat_id, hat_containers: minidom.NodeList[minidom.Element], hat_mappings
):
four_way_hat = 4

Expand All @@ -215,13 +235,14 @@ def handle_hat_button_container(

hat_positions = container.getElementsByTagName("action-set")

hat_mappings = []
_logger.debug(f"Hat Positions: {hat_positions}")

# Iterate each ACTION_SET, for the HAT_BUTTONS
for hat_direction_no, position in enumerate(hat_positions, 1):
# Get the description node if exists
hat_description_node = position.getElementsByTagName("description")

_logger.debug(f"Hat Description Node: {hat_description_node}")
hat_direction = hat_direction_no

if button_count == four_way_hat:
Expand All @@ -234,6 +255,7 @@ def handle_hat_button_container(
# What if multiple hat_description_nodes

hat_description = hat_description_node[0].getAttribute("description")
_logger.debug(f"Hat Description: {hat_description}")

if not hat_description:
# If we don't have a description then no point using the item
Expand All @@ -244,7 +266,7 @@ def handle_hat_button_container(
(Hat(hat_id, HatDirection[hat_position_to_string]), hat_description)
)

return hat_mappings
_logger.debug(f"Hat Mappings from Hat Button Container: {hat_mappings}")


if __name__ == "__main__":
Expand Down