From 9fae21c65551ed1e6f0fbda541d3c3e29219afae Mon Sep 17 00:00:00 2001
From: Isorokin <sorokinilya2024@yandex.ru>
Date: Mon, 28 Oct 2024 22:33:02 +0300
Subject: [PATCH] =?UTF-8?q?P-=D1=80=D0=B5=D0=B3=D1=83=D0=BB=D1=8F=D1=82?=
 =?UTF-8?q?=D0=BE=D1=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 launch/step_5.launch.py | 26 +++++++++++++++++++
 lesson_05/step_5.py     | 55 +++++++++++++++++++++++++++++++++++++++++
 setup.py                |  1 +
 3 files changed, 82 insertions(+)
 create mode 100644 launch/step_5.launch.py
 create mode 100644 lesson_05/step_5.py

diff --git a/launch/step_5.launch.py b/launch/step_5.launch.py
new file mode 100644
index 0000000..148ca98
--- /dev/null
+++ b/launch/step_5.launch.py
@@ -0,0 +1,26 @@
+from launch import LaunchDescription
+from launch_ros.actions import Node
+
+
+def generate_launch_description():
+    return LaunchDescription([
+        Node(
+            package='turtlesim',
+            namespace='',
+            executable='turtlesim_node',
+            name='sim'
+        ),
+        Node(
+            package='lesson_05',
+            namespace='',
+            executable='step_5',
+            name='square'
+        ),
+        Node(
+            package='rqt_plot',
+            namespace='',
+            executable='rqt_plot',
+            name='rqt_plot'
+        )
+
+    ])
diff --git a/lesson_05/step_5.py b/lesson_05/step_5.py
new file mode 100644
index 0000000..52be862
--- /dev/null
+++ b/lesson_05/step_5.py
@@ -0,0 +1,55 @@
+import rclpy
+from rclpy.node import Node
+
+from geometry_msgs.msg import Twist
+from turtlesim.msg import Pose
+import math
+
+
+def angle_diff(a1, a2):
+    a = a1-a2
+    return (a+math.pi)%(2*math.pi)-math.pi
+
+
+class RurMover(Node):
+    def __init__(self):
+        super().__init__('minimal_publisher')
+        self.publisher_twist = self.create_publisher(Twist, 'turtle1/cmd_vel', 10)
+        self.pose_subscriber = self.create_subscription(Pose, 'turtle1/pose', self.pose_callback, 10)
+        self.desired_heading = - math.pi / 2
+        self.SIDE_TIME = 5.0
+        self.LINEAR_SPEED = 0.5
+        self.timer = self.create_timer(self.SIDE_TIME, self.pub_timer_callback)
+        self.kp = 100
+
+    def pose_callback(self, msg):
+        self.get_logger().info('Current heading: "%f"' % msg.theta)
+        self.get_logger().info('Setpoint: "%f"' % self.desired_heading)
+        msg_pub = Twist()
+        msg_pub.linear.x = self.LINEAR_SPEED
+        error = angle_diff(self.desired_heading, msg.theta)
+        if error > 0:
+            self.get_logger().info('Turn left')
+        msg_pub.angular.z = self.kp * error
+        if error < 0:
+            self.get_logger().info('Turn right')
+        msg_pub.angular.z = self.kp * error
+        self.publisher_twist.publish(msg_pub)
+
+    def pub_timer_callback(self):
+        self.desired_heading += math.pi / 2
+        self.desired_heading %= 2 * math.pi
+        self.get_logger().info('New angle setpoint: "%f"' % self.desired_heading)
+
+def main(args=None):
+    rclpy.init(args=args)
+
+    mover = RurMover()
+
+    rclpy.spin(mover)
+
+    mover.destroy_node()
+    rclpy.shutdown()
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 0e83277..223a33c 100644
--- a/setup.py
+++ b/setup.py
@@ -27,6 +27,7 @@ setup(
             'step_2 = lesson_05.step_2:main',
             'step_3 = lesson_05.step_3:main',
             'step_4 = lesson_05.step_4:main',
+            'step_5 = lesson_05.step_5:main',
         ],
     },
 )
-- 
GitLab