From 6e8ef072f31f29fc75f643e3e8fa1a80592762ba Mon Sep 17 00:00:00 2001 From: uquzq <ventsi@DESKTOP-J41TDPJ> Date: Sat, 22 Jun 2024 15:35:35 +0200 Subject: [PATCH] Improve readability of code --- .../forward_kinematics_node_2dof_1.py | 12 ++++++------ .../forward_kinematics_node_2dof_2.py | 6 +++--- .../simple_robot_2dof/reference_node_2dof.py | 4 ++-- 3 files changed, 11 insertions(+), 11 deletions(-) diff --git a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_1.py b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_1.py index 7a835aa..ee5e6dd 100644 --- a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_1.py +++ b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_1.py @@ -43,7 +43,7 @@ class Forward_Kinematics_Node_2Dof(Node): h_fraction = self.phi * h # Define the vector - vector = np.array([-1.0, 0.0, 0.0, 1.0]) + start_vector = np.array([-1.0, 0.0, 0.0, 1.0]) # Define the matrix intermediate_rotation = np.array([ @@ -54,18 +54,18 @@ class Forward_Kinematics_Node_2Dof(Node): ]) # Calculate the intermediate vectors - intermediate_vector = np.matmul(intermediate_rotation, vector) + intermediate_vector = np.matmul(intermediate_rotation, start_vector) self.construct_path(intermediate_vector) - def construct_path(self, vector_path): + def construct_path(self, vector): # RViz pose = PoseStamped() pose.header.frame_id = "map" pose.header.stamp = self.get_clock().now().to_msg() - pose.pose.position.x = vector_path[0] - pose.pose.position.y = vector_path[1] - pose.pose.position.z = vector_path[2] + pose.pose.position.x = vector[0] + pose.pose.position.y = vector[1] + pose.pose.position.z = vector[2] pose.pose.orientation.w = 1.0 #If the path is completed clear it diff --git a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_2.py b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_2.py index ee5f2aa..4b4688d 100644 --- a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_2.py +++ b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/forward_kinematics_node_2dof_2.py @@ -50,7 +50,7 @@ class Forward_Kinematics_Node_2Dof(Node): theta_fraction = self.phi * theta # Define the vector - vector = np.array([-1.0, 0.0, 0.0, 1.0]) + start_vector = np.array([-1.0, 0.0, 0.0, 1.0]) # define the matrix intermediate_rotation = np.array([ @@ -67,10 +67,10 @@ class Forward_Kinematics_Node_2Dof(Node): ]) # calculate the intermediate vectors - rotated_intermediate_vector = np.matmul(intermediate_rotation, vector) + rotated_intermediate_vector = np.matmul(intermediate_rotation, start_vector) # calculate the end vector - self.end_vector = np.matmul(rotation, vector) + self.end_vector = np.matmul(rotation, start_vector) # construct the path of the rotation self.construct_path(rotated_intermediate_vector) diff --git a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/reference_node_2dof.py b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/reference_node_2dof.py index 680fca0..ae03518 100644 --- a/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/reference_node_2dof.py +++ b/ros2_ws/src/simple_robot_2dof/simple_robot_2dof/reference_node_2dof.py @@ -6,7 +6,7 @@ from geometry_msgs.msg import Point, PoseStamped import time -class Reference_Node_1DoF(Node): +class Reference_Node_2DoF(Node): def __init__(self): super().__init__('reference_node_1dof') @@ -38,7 +38,7 @@ class Reference_Node_1DoF(Node): def main(args = None): rclpy.init(args = args) - node = Reference_Node_1DoF() + node = Reference_Node_2DoF() rclpy.spin(node) node.destroy_node() rclpy.shutdown() -- GitLab