#!/usr/bin/env python # -*- coding: utf-8 -*- ################################################################################ # Copyright 2018 ROBOTIS CO., LTD. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. ################################################################################ # Authors: Leon Jung, [AuTURBO] Kihoon Kim (https://github.com/auturbo), Gilbert, Ashe Kim import rospy, roslaunch import subprocess import os import sys from enum import Enum from std_msgs.msg import UInt8, Float64 class CoreNodeController(): def __init__(self): self.ros_package_path = os.path.dirname(os.path.realpath(__file__)) self.ros_package_path = self.ros_package_path.replace('turtlebot3_autorace_core/nodes', '') # subscribes : status returned self.sub_parking_stamped = rospy.Subscriber('/detect/parking_stamped', UInt8, self.cbParkingStamped, queue_size=1) self.sub_mode_control = rospy.Subscriber('/core/decided_mode', UInt8, self.cbReceiveMode, queue_size=1) # publishes orders self.pub_parking_order = rospy.Publisher('/detect/parking_order', UInt8, queue_size=1) self.pub_mode_return = rospy.Publisher('/core/returned_mode', UInt8, queue_size=1) self.CurrentMode = Enum('CurrentMode', 'idle lane_following parking') self.current_mode = self.CurrentMode.idle.value self.StepOfParking = Enum('StepOfParking', 'parking exit') self.current_step_parking = self.StepOfParking.parking.value self.Launcher = Enum('Launcher', 'launch_camera_ex_calib launch_detect_sign launch_detect_lane launch_control_lane launch_detect_parking launch_control_moving') self.uuid = roslaunch.rlutil.get_or_generate_uuid(None, False) self.launch_camera_launched = False self.launch_detect_lane_launched = False self.launch_detect_sign_launched = False self.launch_detect_parking_launched = False self.launch_control_lane_launched = False self.launch_control_moving_launched = False self.is_triggered = False loop_rate = rospy.Rate(10) # 10hz while not rospy.is_shutdown(): if self.is_triggered == True: self.fnControlNode() loop_rate.sleep() def cbReceiveMode(self, mode_msg): rospy.loginfo("starts the progress with %d", mode_msg.data) self.current_mode = mode_msg.data self.is_triggered = True def cbParkingStamped(self, parking_msg): rospy.loginfo("Parking Step changed from %d", self.current_step_parking) self.current_step_parking = parking_msg.data if self.current_step_parking == self.StepOfParking.exit.value: self.current_mode = self.CurrentMode.lane_following.value msg_mode_return = UInt8() msg_mode_return.data = self.current_mode self.pub_mode_return.publish(msg_mode_return) self.is_triggered = True def fnControlNode(self): # lane_following if self.current_mode == self.CurrentMode.lane_following.value: rospy.loginfo("Current step : searching sign") rospy.loginfo("Go to next step : parking") self.fnLaunch(self.Launcher.launch_camera_ex_calib.value, True) self.fnLaunch(self.Launcher.launch_detect_lane.value, True) self.fnLaunch(self.Launcher.launch_detect_sign.value, True) self.fnLaunch(self.Launcher.launch_detect_parking.value, False) self.fnLaunch(self.Launcher.launch_control_lane.value, True) self.fnLaunch(self.Launcher.launch_control_moving.value, False) # parking elif self.current_mode == self.CurrentMode.parking.value: rospy.loginfo("New trigger for parking") msg_pub_parking_order = UInt8() if self.current_step_parking == self.StepOfParking.parking.value: rospy.loginfo("Current step : parking") rospy.loginfo("Go to next step : finish parking") msg_pub_parking_order.data = self.StepOfParking.parking.value self.fnLaunch(self.Launcher.launch_camera_ex_calib.value, True) self.fnLaunch(self.Launcher.launch_detect_lane.value, True) self.fnLaunch(self.Launcher.launch_detect_sign.value, False) self.fnLaunch(self.Launcher.launch_detect_parking.value, True) self.fnLaunch(self.Launcher.launch_control_lane.value, False) self.fnLaunch(self.Launcher.launch_control_moving.value, True) elif self.current_step_parking == self.StepOfParking.exit.value: rospy.loginfo("Current step : finish parking") msg_pub_parking_order.data = self.StepOfParking.exit.value self.fnLaunch(self.Launcher.launch_camera_ex_calib.value, True) self.fnLaunch(self.Launcher.launch_detect_sign.value, True) self.fnLaunch(self.Launcher.launch_detect_lane.value, True) self.fnLaunch(self.Launcher.launch_detect_parking.value, False) self.fnLaunch(self.Launcher.launch_control_lane.value, True) self.fnLaunch(self.Launcher.launch_control_moving.value, False) rospy.sleep(2) self.pub_parking_order.publish(msg_pub_parking_order) def fnLaunch(self, launch_num, is_start): if launch_num == self.Launcher.launch_camera_ex_calib.value: if is_start == True: if self.launch_camera_launched == False: self.launch_camera = roslaunch.scriptapi.ROSLaunch() self.launch_camera = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_camera/launch/extrinsic_camera_calibration.launch"]) self.launch_camera_launched = True self.launch_camera.start() else: pass else: if self.launch_camera_launched == True: self.launch_camera_launched = False self.launch_camera.shutdown() else: pass elif launch_num == self.Launcher.launch_detect_sign.value: if is_start == True: if self.launch_detect_sign_launched == False: self.launch_detect_sign = roslaunch.scriptapi.ROSLaunch() self.launch_detect_sign = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_detect/launch/detect_sign.launch"]) self.launch_detect_sign_launched = True self.launch_detect_sign.start() else: pass else: if self.launch_detect_sign_launched == True: self.launch_detect_sign_launched = False self.launch_detect_sign.shutdown() else: pass elif launch_num == self.Launcher.launch_detect_lane.value: if is_start == True: if self.launch_detect_lane_launched == False: self.launch_detect_lane = roslaunch.scriptapi.ROSLaunch() self.launch_detect_lane = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_detect/launch/detect_lane.launch"]) self.launch_detect_lane_launched = True self.launch_detect_lane.start() else: pass else: if self.launch_detect_lane_launched == True: self.launch_detect_lane_launched = False self.launch_detect_lane.shutdown() else: pass elif launch_num == self.Launcher.launch_control_lane.value: if is_start == True: if self.launch_control_lane_launched == False: self.launch_control_lane = roslaunch.scriptapi.ROSLaunch() self.launch_control_lane = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_driving/launch/turtlebot3_autorace_control_lane.launch"]) self.launch_control_lane_launched = True self.launch_control_lane.start() else: pass else: if self.launch_control_lane_launched == True: self.launch_control_lane_launched = False self.launch_control_lane.shutdown() else: pass elif launch_num == self.Launcher.launch_detect_parking.value: if is_start == True: if self.launch_detect_parking_launched == False: self.launch_detect_parking = roslaunch.scriptapi.ROSLaunch() self.launch_detect_parking = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_detect/launch/detect_parking.launch"]) self.launch_detect_parking_launched = True self.launch_detect_parking.start() else: pass else: if self.launch_detect_parking_launched == True: self.launch_detect_parking_launched = False self.launch_detect_parking.shutdown() else: pass elif launch_num == self.Launcher.launch_control_moving.value: if is_start == True: if self.launch_control_moving_launched == False: self.launch_control_moving = roslaunch.scriptapi.ROSLaunch() self.launch_control_moving = roslaunch.parent.ROSLaunchParent(self.uuid, [self.ros_package_path + "turtlebot3_autorace_driving/launch/turtlebot3_autorace_control_moving.launch"]) self.launch_control_moving_launched = True self.launch_control_moving.start() else: pass else: if self.launch_control_moving_launched == True: self.launch_control_moving_launched = False self.launch_control_moving.shutdown() pass def main(self): rospy.spin() if __name__ == '__main__': rospy.init_node('core_node_controller') node = CoreNodeController() node.main()