mirror of
https://github.com/SHC-ASTRA/rover-ros2.git
synced 2026-04-20 20:01:15 -05:00
Compare commits
11 Commits
main
...
ea36ce6ef4
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ea36ce6ef4 | ||
|
|
3f795bf8ed | ||
|
|
b257dc7556 | ||
|
|
45825189a5 | ||
|
|
a17060ceda | ||
|
|
ff4a58e6ed | ||
|
|
120891c8e5 | ||
|
|
178d5001d6 | ||
|
|
684c2994ec | ||
|
|
bfa50e3a25 | ||
|
|
90fbbac813 |
@@ -1,14 +1,21 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
from launch import LaunchDescription
|
from launch import LaunchDescription
|
||||||
from launch.actions import DeclareLaunchArgument, OpaqueFunction, Shutdown
|
from launch.actions import (
|
||||||
|
DeclareLaunchArgument,
|
||||||
|
OpaqueFunction,
|
||||||
|
Shutdown,
|
||||||
|
IncludeLaunchDescription,
|
||||||
|
)
|
||||||
from launch.substitutions import (
|
from launch.substitutions import (
|
||||||
LaunchConfiguration,
|
LaunchConfiguration,
|
||||||
ThisLaunchFileDir,
|
ThisLaunchFileDir,
|
||||||
PathJoinSubstitution,
|
PathJoinSubstitution,
|
||||||
)
|
)
|
||||||
from launch_ros.actions import Node
|
from launch.launch_description_sources import PythonLaunchDescriptionSource
|
||||||
from launch.conditions import IfCondition
|
from launch.conditions import IfCondition
|
||||||
|
from launch_ros.actions import Node
|
||||||
|
from launch_ros.substitutions import FindPackageShare
|
||||||
|
|
||||||
|
|
||||||
# Prevent making __pycache__ directories
|
# Prevent making __pycache__ directories
|
||||||
@@ -41,7 +48,10 @@ def launch_setup(context, *args, **kwargs):
|
|||||||
executable="core", # change as needed
|
executable="core", # change as needed
|
||||||
name="core",
|
name="core",
|
||||||
output="both",
|
output="both",
|
||||||
parameters=[{"launch_mode": mode}],
|
parameters=[
|
||||||
|
{"launch_mode": mode},
|
||||||
|
{"use_ros2_control": LaunchConfiguration("use_ros2_control", default=False)},
|
||||||
|
],
|
||||||
on_exit=Shutdown(),
|
on_exit=Shutdown(),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -135,4 +145,46 @@ def generate_launch_description():
|
|||||||
description="Launch mode: arm, core, bio, anchor, or ptz",
|
description="Launch mode: arm, core, bio, anchor, or ptz",
|
||||||
)
|
)
|
||||||
|
|
||||||
return LaunchDescription([declare_arg, OpaqueFunction(function=launch_setup)])
|
ros2_control_arg = DeclareLaunchArgument(
|
||||||
|
"use_ros2_control",
|
||||||
|
default_value="false",
|
||||||
|
description="Whether to use DiffDriveController for driving instead of direct Twist",
|
||||||
|
)
|
||||||
|
|
||||||
|
rsp = IncludeLaunchDescription(
|
||||||
|
PythonLaunchDescriptionSource(
|
||||||
|
PathJoinSubstitution(
|
||||||
|
[
|
||||||
|
FindPackageShare("core_description"),
|
||||||
|
"launch",
|
||||||
|
"robot_state_publisher.launch.py",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
),
|
||||||
|
condition=IfCondition(LaunchConfiguration("use_ros2_control")),
|
||||||
|
launch_arguments={("hardware_mode", "physical")},
|
||||||
|
)
|
||||||
|
|
||||||
|
controllers = IncludeLaunchDescription(
|
||||||
|
PythonLaunchDescriptionSource(
|
||||||
|
PathJoinSubstitution(
|
||||||
|
[
|
||||||
|
FindPackageShare("core_description"),
|
||||||
|
"launch",
|
||||||
|
"spawn_controllers.launch.py",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
),
|
||||||
|
condition=IfCondition(LaunchConfiguration("use_ros2_control")),
|
||||||
|
launch_arguments={("hardware_mode", "physical")},
|
||||||
|
)
|
||||||
|
|
||||||
|
return LaunchDescription(
|
||||||
|
[
|
||||||
|
declare_arg,
|
||||||
|
ros2_control_arg,
|
||||||
|
rsp,
|
||||||
|
controllers,
|
||||||
|
OpaqueFunction(function=launch_setup),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|||||||
@@ -2,14 +2,20 @@
|
|||||||
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
|
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
|
||||||
<package format="3">
|
<package format="3">
|
||||||
<name>anchor_pkg</name>
|
<name>anchor_pkg</name>
|
||||||
<version>0.0.0</version>
|
<version>1.0.0</version>
|
||||||
<description>TODO: Package description</description>
|
<description>ASTRA VicCAN driver package, using python-can and pyserial.</description>
|
||||||
<maintainer email="tristanmcginnis26@gmail.com">tristan</maintainer>
|
<maintainer email="tristanmcginnis26@gmail.com">tristan</maintainer>
|
||||||
<license>AGPL-3.0-only</license>
|
<license>AGPL-3.0-only</license>
|
||||||
|
|
||||||
<depend>rclpy</depend>
|
<depend>rclpy</depend>
|
||||||
<depend>common_interfaces</depend>
|
|
||||||
<depend>python3-serial</depend>
|
<exec_depend>common_interfaces</exec_depend>
|
||||||
|
<exec_depend>python3-serial</exec_depend>
|
||||||
|
|
||||||
|
<exec_depend>core_pkg</exec_depend>
|
||||||
|
<exec_depend>arm_pkg</exec_depend>
|
||||||
|
<exec_depend>bio_pkg</exec_depend>
|
||||||
|
<exec_depend>core_description</exec_depend>
|
||||||
|
|
||||||
<build_depend>black</build_depend>
|
<build_depend>black</build_depend>
|
||||||
|
|
||||||
|
|||||||
@@ -2,3 +2,5 @@
|
|||||||
script_dir=$base/lib/anchor_pkg
|
script_dir=$base/lib/anchor_pkg
|
||||||
[install]
|
[install]
|
||||||
install_scripts=$base/lib/anchor_pkg
|
install_scripts=$base/lib/anchor_pkg
|
||||||
|
[build_scripts]
|
||||||
|
executable= /usr/bin/env python3
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ package_name = "anchor_pkg"
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name=package_name,
|
name=package_name,
|
||||||
version="0.0.0",
|
version="1.0.0",
|
||||||
packages=find_packages(exclude=["test"]),
|
packages=find_packages(exclude=["test"]),
|
||||||
data_files=[
|
data_files=[
|
||||||
("share/ament_index/resource_index/packages", ["resource/" + package_name]),
|
("share/ament_index/resource_index/packages", ["resource/" + package_name]),
|
||||||
@@ -17,8 +17,8 @@ setup(
|
|||||||
zip_safe=True,
|
zip_safe=True,
|
||||||
maintainer="tristan",
|
maintainer="tristan",
|
||||||
maintainer_email="tristanmcginnis26@gmail.com",
|
maintainer_email="tristanmcginnis26@gmail.com",
|
||||||
description="Anchor node used to run all modules through a single modules MCU/Computer. Commands to all modules will be relayed through CAN",
|
description="ASTRA VicCAN driver package, using python-can and pyserial.",
|
||||||
license="All Rights Reserved",
|
license="AGPL-3.0-only",
|
||||||
entry_points={
|
entry_points={
|
||||||
"console_scripts": ["anchor = anchor_pkg.anchor_node:main"],
|
"console_scripts": ["anchor = anchor_pkg.anchor_node:main"],
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
import sys
|
import sys
|
||||||
import threading
|
|
||||||
import signal
|
import signal
|
||||||
import math
|
import math
|
||||||
from warnings import deprecated
|
from warnings import deprecated
|
||||||
@@ -26,8 +25,6 @@ control_qos = qos.QoSProfile(
|
|||||||
# liveliness_lease_duration=Duration(seconds=5),
|
# liveliness_lease_duration=Duration(seconds=5),
|
||||||
)
|
)
|
||||||
|
|
||||||
thread = None
|
|
||||||
|
|
||||||
|
|
||||||
class ArmNode(Node):
|
class ArmNode(Node):
|
||||||
"""Relay between Anchor and Basestation/Headless/Moveit2 for Arm related topics."""
|
"""Relay between Anchor and Basestation/Headless/Moveit2 for Arm related topics."""
|
||||||
|
|||||||
@@ -3,14 +3,15 @@
|
|||||||
<package format="3">
|
<package format="3">
|
||||||
<name>arm_pkg</name>
|
<name>arm_pkg</name>
|
||||||
<version>1.0.0</version>
|
<version>1.0.0</version>
|
||||||
<description>Core arm package which handles ROS2 commnuication.</description>
|
<description>Relays topics related to Arm between VicCAN (through Anchor) and basestation.</description>
|
||||||
<maintainer email="tristanmcginnis26@gmail.com">tristan</maintainer>
|
<maintainer email="ds0196@uah.edu">David Sharpe</maintainer>
|
||||||
<license>AGPL-3.0-only</license>
|
<license>AGPL-3.0-only</license>
|
||||||
|
|
||||||
<depend>rclpy</depend>
|
<depend>rclpy</depend>
|
||||||
<depend>common_interfaces</depend>
|
|
||||||
<depend>python3-numpy</depend>
|
<exec_depend>common_interfaces</exec_depend>
|
||||||
<depend>astra_msgs</depend>
|
<exec_depend>python3-numpy</exec_depend>
|
||||||
|
<exec_depend>astra_msgs</exec_depend>
|
||||||
|
|
||||||
<test_depend>ament_copyright</test_depend>
|
<test_depend>ament_copyright</test_depend>
|
||||||
<test_depend>ament_flake8</test_depend>
|
<test_depend>ament_flake8</test_depend>
|
||||||
|
|||||||
@@ -12,9 +12,9 @@ setup(
|
|||||||
],
|
],
|
||||||
install_requires=["setuptools"],
|
install_requires=["setuptools"],
|
||||||
zip_safe=True,
|
zip_safe=True,
|
||||||
maintainer="David",
|
maintainer="David Sharpe",
|
||||||
maintainer_email="ds0196@uah.edu",
|
maintainer_email="ds0196@uah.edu",
|
||||||
description="Relays topics related to the arm between VicCAN (through Anchor) and basestation.",
|
description="Relays topics related to Arm between VicCAN (through Anchor) and basestation.",
|
||||||
license="AGPL-3.0-only",
|
license="AGPL-3.0-only",
|
||||||
entry_points={
|
entry_points={
|
||||||
"console_scripts": ["arm = arm_pkg.arm_node:main"],
|
"console_scripts": ["arm = arm_pkg.arm_node:main"],
|
||||||
|
|||||||
Submodule src/astra_descriptions updated: c36bd8233d...04c2061443
Submodule src/astra_msgs updated: 2264a2cb67...93dc0e0919
@@ -2,14 +2,16 @@
|
|||||||
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
|
<?xml-model href="http://download.ros.org/schema/package_format3.xsd" schematypens="http://www.w3.org/2001/XMLSchema"?>
|
||||||
<package format="3">
|
<package format="3">
|
||||||
<name>bio_pkg</name>
|
<name>bio_pkg</name>
|
||||||
<version>0.0.0</version>
|
<version>1.0.0</version>
|
||||||
<description>TODO: Package description</description>
|
<description>Biosensor package to handle command interpretation and embedded interfacing.</description>
|
||||||
<maintainer email="tristanmcginnis26@gmail.com">tristan</maintainer>
|
<maintainer email="ds0196@uah.edu">David Sharpe</maintainer>
|
||||||
<license>AGPL-3.0-only</license>
|
<license>AGPL-3.0-only</license>
|
||||||
|
|
||||||
<depend>rclpy</depend>
|
<depend>rclpy</depend>
|
||||||
<depend>common_interfaces</depend>
|
|
||||||
<depend>astra_msgs</depend>
|
<exec_depend>common_interfaces</exec_depend>
|
||||||
|
|
||||||
|
<exec_depend>astra_msgs</exec_depend>
|
||||||
|
|
||||||
<test_depend>ament_copyright</test_depend>
|
<test_depend>ament_copyright</test_depend>
|
||||||
<test_depend>ament_flake8</test_depend>
|
<test_depend>ament_flake8</test_depend>
|
||||||
|
|||||||
@@ -2,3 +2,5 @@
|
|||||||
script_dir=$base/lib/bio_pkg
|
script_dir=$base/lib/bio_pkg
|
||||||
[install]
|
[install]
|
||||||
install_scripts=$base/lib/bio_pkg
|
install_scripts=$base/lib/bio_pkg
|
||||||
|
[build_scripts]
|
||||||
|
executable= /usr/bin/env python3
|
||||||
|
|||||||
@@ -14,8 +14,8 @@ setup(
|
|||||||
zip_safe=True,
|
zip_safe=True,
|
||||||
maintainer="tristan",
|
maintainer="tristan",
|
||||||
maintainer_email="tristanmcginnis26@gmail.com",
|
maintainer_email="tristanmcginnis26@gmail.com",
|
||||||
description="TODO: Package description",
|
description="Relays topics related to Biosensor between VicCAN (through Anchor) and basestation.",
|
||||||
license="TODO: License declaration",
|
license="AGPL-3.0-only",
|
||||||
entry_points={
|
entry_points={
|
||||||
"console_scripts": ["bio = bio_pkg.bio_node:main"],
|
"console_scripts": ["bio = bio_pkg.bio_node:main"],
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,112 +0,0 @@
|
|||||||
import rclpy
|
|
||||||
from rclpy.node import Node
|
|
||||||
|
|
||||||
import pygame
|
|
||||||
|
|
||||||
import time
|
|
||||||
|
|
||||||
import serial
|
|
||||||
import sys
|
|
||||||
import threading
|
|
||||||
import glob
|
|
||||||
import os
|
|
||||||
|
|
||||||
import importlib
|
|
||||||
from std_msgs.msg import String
|
|
||||||
from astra_msgs.msg import CoreControl
|
|
||||||
|
|
||||||
|
|
||||||
os.environ["SDL_VIDEODRIVER"] = "dummy" # Prevents pygame from trying to open a display
|
|
||||||
os.environ["SDL_AUDIODRIVER"] = (
|
|
||||||
"dummy" # Force pygame to use a dummy audio driver before pygame.init()
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
max_speed = 90 # Max speed as a duty cycle percentage (1-100)
|
|
||||||
|
|
||||||
|
|
||||||
class Headless(Node):
|
|
||||||
def __init__(self):
|
|
||||||
# Initialize pygame first
|
|
||||||
pygame.init()
|
|
||||||
pygame.joystick.init()
|
|
||||||
|
|
||||||
# Wait for a gamepad to be connected
|
|
||||||
self.gamepad = None
|
|
||||||
print("Waiting for gamepad connection...")
|
|
||||||
while pygame.joystick.get_count() == 0:
|
|
||||||
# Process any pygame events to keep it responsive
|
|
||||||
for event in pygame.event.get():
|
|
||||||
if event.type == pygame.QUIT:
|
|
||||||
pygame.quit()
|
|
||||||
sys.exit(0)
|
|
||||||
time.sleep(1.0) # Check every second
|
|
||||||
print("No gamepad found. Waiting...")
|
|
||||||
|
|
||||||
# Initialize the gamepad
|
|
||||||
self.gamepad = pygame.joystick.Joystick(0)
|
|
||||||
self.gamepad.init()
|
|
||||||
print(f"Gamepad Found: {self.gamepad.get_name()}")
|
|
||||||
|
|
||||||
# Now initialize the ROS2 node
|
|
||||||
super().__init__("core_headless")
|
|
||||||
self.create_timer(0.15, self.send_controls)
|
|
||||||
self.publisher = self.create_publisher(CoreControl, "/core/control", 10)
|
|
||||||
self.lastMsg = (
|
|
||||||
String()
|
|
||||||
) # Used to ignore sending controls repeatedly when they do not change
|
|
||||||
|
|
||||||
def run(self):
|
|
||||||
# This thread makes all the update processes run in the background
|
|
||||||
thread = threading.Thread(target=rclpy.spin, args={self}, daemon=True)
|
|
||||||
thread.start()
|
|
||||||
|
|
||||||
try:
|
|
||||||
while rclpy.ok():
|
|
||||||
self.send_controls()
|
|
||||||
time.sleep(0.1) # Small delay to avoid CPU hogging
|
|
||||||
except KeyboardInterrupt:
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
def send_controls(self):
|
|
||||||
for event in pygame.event.get():
|
|
||||||
if event.type == pygame.QUIT:
|
|
||||||
pygame.quit()
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
# Check if controller is still connected
|
|
||||||
if pygame.joystick.get_count() == 0:
|
|
||||||
print("Gamepad disconnected. Exiting...")
|
|
||||||
# Send one last zero control message
|
|
||||||
input = CoreControl()
|
|
||||||
input.left_stick = 0
|
|
||||||
input.right_stick = 0
|
|
||||||
input.max_speed = 0
|
|
||||||
self.publisher.publish(input)
|
|
||||||
self.get_logger().info("Final stop command sent. Shutting down.")
|
|
||||||
# Clean up
|
|
||||||
pygame.quit()
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
input = CoreControl()
|
|
||||||
input.max_speed = max_speed
|
|
||||||
input.right_stick = -1 * round(self.gamepad.get_axis(4), 2) # right y-axis
|
|
||||||
if self.gamepad.get_axis(5) > 0:
|
|
||||||
input.left_stick = input.right_stick
|
|
||||||
else:
|
|
||||||
input.left_stick = -1 * round(self.gamepad.get_axis(1), 2) # left y-axis
|
|
||||||
|
|
||||||
output = f"L: {input.left_stick}, R: {input.right_stick}, M: {max_speed}"
|
|
||||||
self.get_logger().info(f"[Ctrl] {output}")
|
|
||||||
self.publisher.publish(input)
|
|
||||||
|
|
||||||
|
|
||||||
def main(args=None):
|
|
||||||
rclpy.init(args=args)
|
|
||||||
node = Headless()
|
|
||||||
rclpy.spin(node)
|
|
||||||
rclpy.shutdown()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
main()
|
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -3,15 +3,17 @@
|
|||||||
<package format="3">
|
<package format="3">
|
||||||
<name>core_pkg</name>
|
<name>core_pkg</name>
|
||||||
<version>1.0.0</version>
|
<version>1.0.0</version>
|
||||||
<description>Core rover control package to handle command interpretation and embedded interfacing.</description>
|
<description>Relays topics related to Core between VicCAN (through Anchor) and basestation.</description>
|
||||||
<maintainer email="tristanmcginnis26@gmail.com">tristan</maintainer>
|
<maintainer email="ds0196@uah.edu">David Sharpe</maintainer>
|
||||||
<license>AGPL-3.0-only</license>
|
<license>AGPL-3.0-only</license>
|
||||||
|
|
||||||
<depend>rclpy</depend>
|
<depend>rclpy</depend>
|
||||||
<depend>common_interfaces</depend>
|
|
||||||
<depend>python3-scipy</depend>
|
<exec_depend>common_interfaces</exec_depend>
|
||||||
<depend>python-crccheck-pip</depend>
|
<exec_depend>python3-scipy</exec_depend>
|
||||||
<depend>astra_msgs</depend>
|
<exec_depend>python-crccheck-pip</exec_depend>
|
||||||
|
|
||||||
|
<exec_depend>astra_msgs</exec_depend>
|
||||||
|
|
||||||
<test_depend>ament_copyright</test_depend>
|
<test_depend>ament_copyright</test_depend>
|
||||||
<test_depend>ament_flake8</test_depend>
|
<test_depend>ament_flake8</test_depend>
|
||||||
|
|||||||
@@ -2,3 +2,5 @@
|
|||||||
script_dir=$base/lib/core_pkg
|
script_dir=$base/lib/core_pkg
|
||||||
[install]
|
[install]
|
||||||
install_scripts=$base/lib/core_pkg
|
install_scripts=$base/lib/core_pkg
|
||||||
|
[build_scripts]
|
||||||
|
executable= /usr/bin/env python3
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ package_name = "core_pkg"
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name=package_name,
|
name=package_name,
|
||||||
version="0.0.0",
|
version="1.0.0",
|
||||||
packages=find_packages(exclude=["test"]),
|
packages=find_packages(exclude=["test"]),
|
||||||
data_files=[
|
data_files=[
|
||||||
("share/ament_index/resource_index/packages", ["resource/" + package_name]),
|
("share/ament_index/resource_index/packages", ["resource/" + package_name]),
|
||||||
@@ -12,14 +12,13 @@ setup(
|
|||||||
],
|
],
|
||||||
install_requires=["setuptools"],
|
install_requires=["setuptools"],
|
||||||
zip_safe=True,
|
zip_safe=True,
|
||||||
maintainer="tristan",
|
maintainer="David Sharpe",
|
||||||
maintainer_email="tristanmcginnis26@gmail.com",
|
maintainer_email="ds0196@uah.edu",
|
||||||
description="Core rover control package to handle command interpretation and embedded interfacing.",
|
description="Relays topics related to Core between VicCAN (through Anchor) and basestation.",
|
||||||
license="All Rights Reserved",
|
license="AGPL-3.0-only",
|
||||||
entry_points={
|
entry_points={
|
||||||
"console_scripts": [
|
"console_scripts": [
|
||||||
"core = core_pkg.core_node:main",
|
"core = core_pkg.core_node:main",
|
||||||
"headless = core_pkg.core_headless:main",
|
|
||||||
"ptz = core_pkg.core_ptz:main",
|
"ptz = core_pkg.core_ptz:main",
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -3,14 +3,15 @@
|
|||||||
<package format="3">
|
<package format="3">
|
||||||
<name>headless_pkg</name>
|
<name>headless_pkg</name>
|
||||||
<version>1.0.0</version>
|
<version>1.0.0</version>
|
||||||
<description>Headless rover control package to handle command interpretation and embedded interfacing.</description>
|
<description>Provides headless rover control, similar to Basestation.</description>
|
||||||
<maintainer email="ds0196@uah.edu">David Sharpe</maintainer>
|
<maintainer email="ds0196@uah.edu">David Sharpe</maintainer>
|
||||||
<license>AGPL-3.0-only</license>
|
<license>AGPL-3.0-only</license>
|
||||||
|
|
||||||
<depend>rclpy</depend>
|
<depend>rclpy</depend>
|
||||||
<depend>common_interfaces</depend>
|
|
||||||
<depend>python3-pygame</depend>
|
<exec_depend>common_interfaces</exec_depend>
|
||||||
<depend>astra_msgs</depend>
|
<exec_depend>python3-pygame</exec_depend>
|
||||||
|
<exec_depend>astra_msgs</exec_depend>
|
||||||
|
|
||||||
<test_depend>ament_copyright</test_depend>
|
<test_depend>ament_copyright</test_depend>
|
||||||
<test_depend>ament_flake8</test_depend>
|
<test_depend>ament_flake8</test_depend>
|
||||||
|
|||||||
@@ -2,3 +2,5 @@
|
|||||||
script_dir=$base/lib/headless_pkg
|
script_dir=$base/lib/headless_pkg
|
||||||
[install]
|
[install]
|
||||||
install_scripts=$base/lib/headless_pkg
|
install_scripts=$base/lib/headless_pkg
|
||||||
|
[build_scripts]
|
||||||
|
executable= /usr/bin/env python3
|
||||||
|
|||||||
@@ -14,11 +14,9 @@ setup(
|
|||||||
zip_safe=True,
|
zip_safe=True,
|
||||||
maintainer="David Sharpe",
|
maintainer="David Sharpe",
|
||||||
maintainer_email="ds0196@uah.edu",
|
maintainer_email="ds0196@uah.edu",
|
||||||
description="Headless rover control package to handle command interpretation and embedded interfacing.",
|
description="Provides headless rover control, similar to Basestation.",
|
||||||
license="All Rights Reserved",
|
license="AGPL-3.0-only",
|
||||||
entry_points={
|
entry_points={
|
||||||
"console_scripts": [
|
"console_scripts": ["headless_full = src.headless_node:main"],
|
||||||
"headless_full = src.headless_node:main",
|
|
||||||
],
|
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -138,6 +138,11 @@ class Headless(Node):
|
|||||||
self.get_parameter("use_old_topics").get_parameter_value().bool_value
|
self.get_parameter("use_old_topics").get_parameter_value().bool_value
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self.declare_parameter("use_cmd_vel", False)
|
||||||
|
self.use_cmd_vel = (
|
||||||
|
self.get_parameter("use_cmd_vel").get_parameter_value().bool_value
|
||||||
|
)
|
||||||
|
|
||||||
self.declare_parameter("use_bio", False)
|
self.declare_parameter("use_bio", False)
|
||||||
self.use_bio = self.get_parameter("use_bio").get_parameter_value().bool_value
|
self.use_bio = self.get_parameter("use_bio").get_parameter_value().bool_value
|
||||||
|
|
||||||
@@ -145,7 +150,6 @@ class Headless(Node):
|
|||||||
self.use_arm_ik = (
|
self.use_arm_ik = (
|
||||||
self.get_parameter("use_arm_ik").get_parameter_value().bool_value
|
self.get_parameter("use_arm_ik").get_parameter_value().bool_value
|
||||||
)
|
)
|
||||||
|
|
||||||
# NOTE: only applicable if use_old_topics == True
|
# NOTE: only applicable if use_old_topics == True
|
||||||
self.declare_parameter("use_new_arm_manual_scheme", True)
|
self.declare_parameter("use_new_arm_manual_scheme", True)
|
||||||
self.use_new_arm_manual_scheme = (
|
self.use_new_arm_manual_scheme = (
|
||||||
@@ -155,6 +159,13 @@ class Headless(Node):
|
|||||||
)
|
)
|
||||||
|
|
||||||
# Check parameter validity
|
# Check parameter validity
|
||||||
|
if self.use_cmd_vel:
|
||||||
|
self.get_logger().info("Using cmd_vel for core control")
|
||||||
|
global CORE_MODE
|
||||||
|
CORE_MODE = "twist"
|
||||||
|
else:
|
||||||
|
self.get_logger().info("Using astra_msgs/CoreControl for core control")
|
||||||
|
|
||||||
if self.use_arm_ik and self.use_old_topics:
|
if self.use_arm_ik and self.use_old_topics:
|
||||||
self.get_logger().fatal("Old topics do not support arm IK control.")
|
self.get_logger().fatal("Old topics do not support arm IK control.")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
@@ -184,6 +195,9 @@ class Headless(Node):
|
|||||||
self.core_twist_pub_ = self.create_publisher(
|
self.core_twist_pub_ = self.create_publisher(
|
||||||
Twist, "/core/twist", qos_profile=control_qos
|
Twist, "/core/twist", qos_profile=control_qos
|
||||||
)
|
)
|
||||||
|
self.core_cmd_vel_pub_ = self.create_publisher(
|
||||||
|
TwistStamped, "/diff_controller/cmd_vel", qos_profile=control_qos
|
||||||
|
)
|
||||||
self.core_state_pub_ = self.create_publisher(
|
self.core_state_pub_ = self.create_publisher(
|
||||||
CoreCtrlState, "/core/control/state", qos_profile=control_qos
|
CoreCtrlState, "/core/control/state", qos_profile=control_qos
|
||||||
)
|
)
|
||||||
@@ -231,13 +245,19 @@ class Headless(Node):
|
|||||||
# Rumble when node is ready (returns False if rumble not supported)
|
# Rumble when node is ready (returns False if rumble not supported)
|
||||||
self.gamepad.rumble(0.7, 0.8, 150)
|
self.gamepad.rumble(0.7, 0.8, 150)
|
||||||
|
|
||||||
|
# Added so you can tell when it starts running after changing the constant logging to debug from info
|
||||||
|
self.get_logger().info("Defaulting to Core mode. Ready.")
|
||||||
|
|
||||||
def stop_all(self):
|
def stop_all(self):
|
||||||
if self.use_old_topics:
|
if self.use_old_topics:
|
||||||
self.core_publisher.publish(CORE_STOP_MSG)
|
self.core_publisher.publish(CORE_STOP_MSG)
|
||||||
self.arm_publisher.publish(ARM_STOP_MSG)
|
self.arm_publisher.publish(ARM_STOP_MSG)
|
||||||
self.bio_publisher.publish(BIO_STOP_MSG)
|
self.bio_publisher.publish(BIO_STOP_MSG)
|
||||||
else:
|
else:
|
||||||
self.core_twist_pub_.publish(CORE_STOP_TWIST_MSG)
|
if self.use_cmd_vel:
|
||||||
|
self.core_cmd_vel_pub_.publish(self.core_cmd_vel_stop_msg())
|
||||||
|
else:
|
||||||
|
self.core_twist_pub_.publish(CORE_STOP_TWIST_MSG)
|
||||||
if self.use_arm_ik:
|
if self.use_arm_ik:
|
||||||
self.arm_ik_twist_publisher.publish(self.arm_ik_twist_stop_msg())
|
self.arm_ik_twist_publisher.publish(self.arm_ik_twist_stop_msg())
|
||||||
else:
|
else:
|
||||||
@@ -332,18 +352,28 @@ class Headless(Node):
|
|||||||
self.core_publisher.publish(input)
|
self.core_publisher.publish(input)
|
||||||
|
|
||||||
else: # New topics
|
else: # New topics
|
||||||
input = Twist()
|
twist = Twist()
|
||||||
|
|
||||||
# Forward/back and Turn
|
# Forward/back and Turn
|
||||||
input.linear.x = -1.0 * left_stick_y
|
twist.linear.x = -1.0 * left_stick_y
|
||||||
input.angular.z = -1.0 * copysign(
|
twist.angular.z = -1.0 * copysign(
|
||||||
right_stick_x**2, right_stick_x
|
right_stick_x**2, right_stick_x
|
||||||
) # Exponent for finer control (curve)
|
) # Exponent for finer control (curve)
|
||||||
|
|
||||||
|
# This kinda looks dumb being seperate from the following block, but this
|
||||||
|
# maintains the separation between modifying the control message and sending it
|
||||||
|
if self.use_cmd_vel:
|
||||||
|
twist.linear.x *= 1.5
|
||||||
|
twist.angular.z *= 0.5
|
||||||
|
|
||||||
# Publish
|
# Publish
|
||||||
self.core_twist_pub_.publish(input)
|
if self.use_cmd_vel:
|
||||||
self.get_logger().info(
|
header = Header(stamp=self.get_clock().now().to_msg())
|
||||||
f"[Core Ctrl] Linear: {round(input.linear.x, 2)}, Angular: {round(input.angular.z, 2)}"
|
self.core_cmd_vel_pub_.publish(TwistStamped(header=header, twist=twist))
|
||||||
|
else:
|
||||||
|
self.core_twist_pub_.publish(twist)
|
||||||
|
self.get_logger().debug(
|
||||||
|
f"[Core Ctrl] Linear: {round(twist.linear.x, 2)}, Angular: {round(twist.angular.z, 2)}"
|
||||||
)
|
)
|
||||||
|
|
||||||
# Brake mode
|
# Brake mode
|
||||||
@@ -643,6 +673,11 @@ class Headless(Node):
|
|||||||
else:
|
else:
|
||||||
pass # TODO: implement new bio control topics
|
pass # TODO: implement new bio control topics
|
||||||
|
|
||||||
|
def core_cmd_vel_stop_msg(self):
|
||||||
|
return TwistStamped(
|
||||||
|
header=Header(frame_id="base_link", stamp=self.get_clock().now().to_msg())
|
||||||
|
)
|
||||||
|
|
||||||
def arm_manual_stop_msg(self):
|
def arm_manual_stop_msg(self):
|
||||||
return JointJog(
|
return JointJog(
|
||||||
header=Header(frame_id="base_link", stamp=self.get_clock().now().to_msg()),
|
header=Header(frame_id="base_link", stamp=self.get_clock().now().to_msg()),
|
||||||
|
|||||||
Reference in New Issue
Block a user