|
@@ -25,6 +25,8 @@ import traceback
|
|
|
from lstrobot_planning.msg import JointTrajectoryPoint_ex,JointTrajectory_ex
|
|
|
import json
|
|
|
import termios
|
|
|
+from decorator_time import decorator_time
|
|
|
+import check
|
|
|
|
|
|
pi = np.pi
|
|
|
|
|
@@ -36,6 +38,7 @@ class MoveIt_Control:
|
|
|
self.hf_fail=[]
|
|
|
# Init ros config
|
|
|
moveit_commander.roscpp_initialize(sys.argv)
|
|
|
+ self.robot = moveit_commander.RobotCommander()
|
|
|
|
|
|
self.arm = moveit_commander.MoveGroupCommander('manipulator')
|
|
|
self.arm.set_goal_joint_tolerance(0.0001)
|
|
@@ -88,7 +91,8 @@ class MoveIt_Control:
|
|
|
# 打印位姿信息
|
|
|
#rospy.lohome_poseinfo("Current Pose: {}".format(pose))
|
|
|
return pose
|
|
|
-
|
|
|
+
|
|
|
+ #TODO move_p_path_constraints
|
|
|
def create_path_constraints2(self,start_point,end_point,r):#创建路径约束
|
|
|
#计算起点指向终点的向量
|
|
|
vector = np.array([end_point[0]- start_point[0], end_point[1]- start_point[1], end_point[2]- start_point[2]])
|
|
@@ -130,11 +134,12 @@ class MoveIt_Control:
|
|
|
# constraints.orientation_constraints.append(orientation_constraint)
|
|
|
return constraints
|
|
|
|
|
|
- def move_p_flexible(self,point,points,trajectory,trajectory_with_type,v=1,a=1):
|
|
|
- self.arm.set_max_velocity_scaling_factor(v)
|
|
|
+ #TODO move_p_flexible
|
|
|
+ def move_p_flexible(self,point,points,trajectory,trajectory_with_type,a=1,v=1):
|
|
|
self.arm.set_max_acceleration_scaling_factor(a)
|
|
|
- rrr=0.05#初始允许半径
|
|
|
-
|
|
|
+ self.arm.set_max_velocity_scaling_factor(v)
|
|
|
+ rrr=0.09#初始允许半径 9cm
|
|
|
+ er=0
|
|
|
if trajectory:
|
|
|
#起点位置设置为规划组最后一个点
|
|
|
state = self.arm.get_current_state()
|
|
@@ -157,20 +162,24 @@ class MoveIt_Control:
|
|
|
traj = self.arm.plan()#调用plan进行规划
|
|
|
error=traj[3]
|
|
|
if error.val == moveit_msgs.msg._MoveItErrorCodes.MoveItErrorCodes.SUCCESS:
|
|
|
- rospy.loginfo("本次移动OK")
|
|
|
trajj = traj[1] #取出规划的信息
|
|
|
- trajj_with_type = mark_the_traj(trajj,"during-p",welding_sequence)
|
|
|
- trajj_with_type.points[-len(trajj.joint_trajectory.points)].type = "start"
|
|
|
- trajj_with_type.points[-1].type = "end"
|
|
|
- trajectory_with_type.append(trajj_with_type)
|
|
|
- points.append(point)
|
|
|
- trajectory.append(trajj)
|
|
|
- break
|
|
|
+ rospy.loginfo("正在检查移动轨迹有效性...")
|
|
|
+ error_c, limit_margin=check.check_joint_positions(trajj)
|
|
|
+ if not error_c:
|
|
|
+ rospy.loginfo("本次移动OK")
|
|
|
+ rospy.loginfo("*******************")
|
|
|
+ trajj_with_type = mark_the_traj(trajj,"during-p",welding_sequence)
|
|
|
+ trajj_with_type.points[-len(trajj.joint_trajectory.points)].type = "start"
|
|
|
+ trajj_with_type.points[-1].type = "end"
|
|
|
+ trajectory_with_type.append(trajj_with_type)
|
|
|
+ points.append(point)
|
|
|
+ trajectory.append(trajj)
|
|
|
+ break
|
|
|
else:
|
|
|
- rospy.loginfo(error)
|
|
|
+ rospy.loginfo("check failed! 移动轨迹无效")
|
|
|
rospy.loginfo("移动规划失败-开始第{}次重新规划".format(i+1))
|
|
|
self.arm.clear_path_constraints()
|
|
|
- rrr=rrr+0.2#每次增加10厘米
|
|
|
+ rrr=rrr+0.2#每次增加20厘米半径
|
|
|
rospy.loginfo("R值:{}".format(rrr))
|
|
|
if trajectory:
|
|
|
path_constraints = self.create_path_constraints2(points[-1],point,rrr)#将最后一个点和输入的点(焊缝的起始点)做圆柱轨迹约束
|
|
@@ -181,37 +190,38 @@ class MoveIt_Control:
|
|
|
rospy.loginfo("所有移动规划尝试失败 取消路径约束")
|
|
|
self.arm.clear_path_constraints()
|
|
|
if(i==(b-1)):
|
|
|
- rospy.loginfo("所有移动规划尝试失败 程序退出")
|
|
|
- rospy.set_param("sign_control",0)
|
|
|
- sys.exit(0)
|
|
|
+ rospy.loginfo("所有移动规划尝试失败 焊缝起点不可达")
|
|
|
+ er=1
|
|
|
+ break
|
|
|
#清除约束
|
|
|
self.arm.clear_path_constraints()
|
|
|
|
|
|
- return points,trajectory,trajectory_with_type
|
|
|
+ return points,trajectory,trajectory_with_type,er
|
|
|
|
|
|
-
|
|
|
- def move_pl(self,point,points,trajectory,trajectory_with_type,v=1,a=1):#焊缝的规划 只规划当前的点到上一个点
|
|
|
+ #TODO move_pl
|
|
|
+ def move_pl(self,point,points,trajectory,trajectory_with_type,a=1,v=1):#焊缝的规划 只规划当前的点到上一个点
|
|
|
|
|
|
- self.arm.set_max_velocity_scaling_factor(v)
|
|
|
self.arm.set_max_acceleration_scaling_factor(a)
|
|
|
+ self.arm.set_max_velocity_scaling_factor(v)
|
|
|
|
|
|
#设定约束
|
|
|
if trajectory:
|
|
|
- #起点位置设置为规划组最后一个点
|
|
|
+ #起点位置设置为规划组最后一个点
|
|
|
state = self.arm.get_current_state()
|
|
|
state.joint_state.position = trajectory[-1].joint_trajectory.points[-1].positions
|
|
|
self.arm.set_start_state(state)
|
|
|
-
|
|
|
+
|
|
|
self.arm.set_pose_target(point, self.end_effector_link)#设定目标点为传入的点
|
|
|
-
|
|
|
+
|
|
|
# 创建路径约束
|
|
|
path_constraints = self.create_path_constraints(points[-1],point)#将最后一个点和输入的点(焊缝的起始点)做圆柱轨迹约束
|
|
|
self.arm.set_path_constraints(path_constraints)#设定约束
|
|
|
-
|
|
|
+
|
|
|
traj = self.arm.plan()#调用plan进行规划
|
|
|
error=traj[3]
|
|
|
if error.val == moveit_msgs.msg._MoveItErrorCodes.MoveItErrorCodes.SUCCESS:
|
|
|
rospy.loginfo("本次焊缝规划 OK")
|
|
|
+ rospy.loginfo("*******************")
|
|
|
trajj = traj[1] #取出规划的信息
|
|
|
trajj_with_type = mark_the_traj(trajj,"during-l",welding_sequence)
|
|
|
trajj_with_type.points[-len(trajj.joint_trajectory.points)].type = "start"
|
|
@@ -224,17 +234,78 @@ class MoveIt_Control:
|
|
|
points.pop()#将本条焊缝的起点从规划中删除
|
|
|
trajectory.pop()
|
|
|
trajectory_with_type.pop()
|
|
|
- self.hf_fail.append(welding_sequence[self.hf_num])
|
|
|
+ self.hf_fail.append(welding_sequence[self.hf_num])#将焊缝添加到失败列表
|
|
|
#清除约束
|
|
|
self.arm.clear_path_constraints()
|
|
|
self.hf_num=self.hf_num+1#焊缝计数
|
|
|
return points,trajectory,trajectory_with_type
|
|
|
|
|
|
+ #TODO move_pl_check
|
|
|
+ def move_pl_check(self,point,points,trajectory,trajectory_with_type,a=1,v=1):
|
|
|
+ self.arm.set_max_acceleration_scaling_factor(a)
|
|
|
+ self.arm.set_max_velocity_scaling_factor(v)
|
|
|
+
|
|
|
+ #重新规划起点次数
|
|
|
+ a=10
|
|
|
+ for j in range(a):
|
|
|
+ if trajectory:
|
|
|
+ state=self.arm.get_current_state()
|
|
|
+ state.joint_state.position = trajectory[-1].joint_trajectory.points[-1].positions
|
|
|
+ self.arm.set_start_state(state)
|
|
|
+
|
|
|
+ self.arm.set_pose_target(point, self.end_effector_link)
|
|
|
+
|
|
|
+ path_constraints = self.create_path_constraints(points[-1],point)
|
|
|
+ self.arm.set_path_constraints(path_constraints)
|
|
|
+
|
|
|
+ #当前起点规划次数
|
|
|
+ b=10
|
|
|
+ for i in range(b):
|
|
|
+ traj = self.arm.plan()
|
|
|
+ error=traj[3]
|
|
|
+ if error.val == moveit_msgs.msg._MoveItErrorCodes.MoveItErrorCodes.SUCCESS:
|
|
|
+ trajj = traj[1] #取出规划的信息
|
|
|
+ rospy.loginfo("正在检查焊缝轨迹有效性...")
|
|
|
+ error_c,limit_margin=check.check_joint_positions(trajj)
|
|
|
+ if not error_c:
|
|
|
+ rospy.loginfo("本次焊缝规划OK")
|
|
|
+ rospy.loginfo("*******************")
|
|
|
+ break
|
|
|
+ if not limit_margin or i==b-1:
|
|
|
+ prepoint=points.pop()
|
|
|
+ trajectory.pop()
|
|
|
+ trajectory_with_type.pop()
|
|
|
+ #清除约束
|
|
|
+ self.arm.clear_path_constraints()
|
|
|
+ points,trajectory,trajectory_with_type,er1= self.move_p_flexible(prepoint,points,trajectory,trajectory_with_type)
|
|
|
+ break
|
|
|
+ rospy.loginfo("The {}-{}th replanning".format(j,i+1))
|
|
|
+ if not error_c:
|
|
|
+ break
|
|
|
+ if error_c:
|
|
|
+ rospy.loginfo("本次焊缝规划失败")
|
|
|
+ rospy.loginfo("*******************")
|
|
|
+ self.hf_fail.append(welding_sequence[self.hf_num])
|
|
|
+
|
|
|
+ self.arm.clear_path_constraints()
|
|
|
+ self.hf_num=self.hf_num+1#焊缝计数
|
|
|
+
|
|
|
+ trajj_with_type = mark_the_traj(trajj,"during-l",welding_sequence)
|
|
|
+ trajj_with_type.points[-len(trajj.joint_trajectory.points)].type = "start"
|
|
|
+ trajj_with_type.points[-1].type = "end"
|
|
|
+
|
|
|
+ points.append(point)
|
|
|
+ trajectory.append(trajj)
|
|
|
+ trajectory_with_type.append(trajj_with_type)
|
|
|
+
|
|
|
+ return points,trajectory,trajectory_with_type
|
|
|
+
|
|
|
+ #TODO move_pl_path_constraints
|
|
|
def create_path_constraints(self,start_point,end_point):
|
|
|
#计算起点指向终点的向量
|
|
|
vector = np.array([end_point[0]- start_point[0], end_point[1]- start_point[1], end_point[2]- start_point[2]])
|
|
|
height = np.linalg.norm(vector)+0.002
|
|
|
- radius = 0.001
|
|
|
+ radius = 0.01
|
|
|
|
|
|
constraints = Constraints()
|
|
|
|
|
@@ -271,7 +342,9 @@ class MoveIt_Control:
|
|
|
constraints.position_constraints.append(position_constraint)
|
|
|
|
|
|
return constraints
|
|
|
-
|
|
|
+
|
|
|
+ #TODO go_home
|
|
|
+ @decorator_time
|
|
|
def go_home(self,a=1,v=1):
|
|
|
rospy.loginfo("go_home start")
|
|
|
self.arm.set_max_acceleration_scaling_factor(a)
|
|
@@ -282,6 +355,7 @@ class MoveIt_Control:
|
|
|
rospy.loginfo("go_home end")
|
|
|
# rospy.sleep(1)
|
|
|
|
|
|
+ #TODO go_home_justplan
|
|
|
def go_home_justplan(self,trajectory,trajectory_with_type,a=1,v=1):
|
|
|
self.arm.set_max_acceleration_scaling_factor(a)
|
|
|
self.arm.set_max_velocity_scaling_factor(v)
|
|
@@ -296,10 +370,12 @@ class MoveIt_Control:
|
|
|
trajectory_with_type.append(traj_with_type)
|
|
|
return trajectory,trajectory_with_type
|
|
|
|
|
|
+ #TODO path_planning
|
|
|
+ @decorator_time
|
|
|
def path_planning(self,folder_path,gohome=True):
|
|
|
file_path_result = os.path.join(folder_path, 'result.txt')
|
|
|
all_data = process_welding_data(file_path_result)
|
|
|
-
|
|
|
+ err=0
|
|
|
points,trajectory,trajectory_with_type = [],[],[]
|
|
|
for i in range(len(all_data)):
|
|
|
rospy.loginfo("共读取到%d条焊缝,开始规划第%d条",len(all_data),i+1)
|
|
@@ -309,12 +385,19 @@ class MoveIt_Control:
|
|
|
q2 = all_data[i][3]
|
|
|
|
|
|
point11 = [start_point[0]/1000, start_point[1]/1000, start_point[2]/1000, q1[0],q1[1],q1[2],q1[3]]
|
|
|
- points,trajectory,trajectory_with_type = self.move_p_flexible(point11,points,trajectory,trajectory_with_type)
|
|
|
+ points,trajectory,trajectory_with_type,err = self.move_p_flexible(point11,points,trajectory,trajectory_with_type)
|
|
|
+ if err==1:#焊缝起点不可达 则跳过此次焊缝和移动的规划 并将焊缝添加到失败列表
|
|
|
+ self.hf_fail.append(welding_sequence[self.hf_num])
|
|
|
+ self.hf_num=self.hf_num+1#焊缝计数
|
|
|
+ continue
|
|
|
+
|
|
|
point22 = [end_point[0]/1000, end_point[1]/1000, end_point[2]/1000, q2[0],q2[1],q2[2],q2[3]]
|
|
|
- points,trajectory,trajectory_with_type = self.move_pl(point22,points,trajectory,trajectory_with_type, v=v, a=a)
|
|
|
+ points,trajectory,trajectory_with_type = self.move_pl_check(point22,points,trajectory,trajectory_with_type,v=speed_v)
|
|
|
+
|
|
|
rospy.loginfo("第%d条焊缝规划完毕",i+1)
|
|
|
+ rospy.loginfo("*******************")
|
|
|
if gohome:
|
|
|
- points,trajectory,trajectory_with_type = self.move_p_flexible(self.home_pose,points,trajectory,trajectory_with_type)
|
|
|
+ points,trajectory,trajectory_with_type,err= self.move_p_flexible(self.home_pose,points,trajectory,trajectory_with_type,v=speed_v)
|
|
|
trajectory,trajectory_with_type = self.go_home_justplan(trajectory,trajectory_with_type)
|
|
|
traj_merge= merge_trajectories(trajectory)
|
|
|
trajectory_with_type_merge= merge_trajectories_with_type(trajectory_with_type)
|
|
@@ -339,6 +422,7 @@ def process_welding_data(filename):
|
|
|
|
|
|
return all_data
|
|
|
|
|
|
+#TODO mark_the_traj
|
|
|
def mark_the_traj(traj,TYPE,SEQUENCE):
|
|
|
traj_with_type = JointTrajectory_ex()
|
|
|
traj_with_type.header = traj.joint_trajectory.header
|
|
@@ -355,6 +439,7 @@ def mark_the_traj(traj,TYPE,SEQUENCE):
|
|
|
]
|
|
|
return traj_with_type
|
|
|
|
|
|
+#TODO merge_trajectories
|
|
|
def merge_trajectories(trajectories):
|
|
|
if not trajectories:
|
|
|
return None
|
|
@@ -383,6 +468,7 @@ def merge_trajectories(trajectories):
|
|
|
|
|
|
return merged_trajectory
|
|
|
|
|
|
+#TODO merge_trajectories_with_type
|
|
|
def merge_trajectories_with_type(trajectory_with_type):
|
|
|
if not trajectory_with_type:
|
|
|
return None
|
|
@@ -458,37 +544,71 @@ def ROS2PY_msgs(msgs, m_sr):
|
|
|
# for i in range(len(py_msgs.point.type)):
|
|
|
# moveit_server.move_j(py_msgs.point.xyz_list[i])
|
|
|
# #input("任意键继续")
|
|
|
-
|
|
|
+
|
|
|
+def get_valid_input(factor, default):
|
|
|
+ while True:
|
|
|
+ user_input=input(factor)
|
|
|
+ if user_input=="":
|
|
|
+ return default
|
|
|
+ try:
|
|
|
+ value=float(user_input)
|
|
|
+ if 0<value<=1:
|
|
|
+ return value
|
|
|
+ else:
|
|
|
+ rospy.logerr("缩放因子必须在(0, 1]范围内,请重新输入!")
|
|
|
+ except ValueError:
|
|
|
+ rospy.logerr("输入值无效,请根据提示重新输入!")
|
|
|
+
|
|
|
def get_user_input():
|
|
|
"""获取用户输入的速度和加速度缩放因子,并在控制台中打印信息和错误提示"""
|
|
|
DEFUALT_V = 1.0
|
|
|
DEFUALT_A = 1.0
|
|
|
-
|
|
|
- def get_valid_input(factor, default):
|
|
|
- while True:
|
|
|
- user_input=input(factor)
|
|
|
- if user_input=="":
|
|
|
- return default
|
|
|
- try:
|
|
|
- value=float(user_input)
|
|
|
- if 0<=value<=1:
|
|
|
- return value
|
|
|
- else:
|
|
|
- rospy.logerr("缩放因子必须在[0, 1]范围内,请重新输入!")
|
|
|
- except ValueError:
|
|
|
- rospy.logerr("输入值无效,请根据提示重新输入!")
|
|
|
-
|
|
|
+
|
|
|
try:
|
|
|
termios.tcflush(sys.stdin, termios.TCIFLUSH) #清空输入缓存
|
|
|
rospy.loginfo("输入缓存区已清空!")
|
|
|
|
|
|
- v=get_valid_input("请输入速度缩放因子(0-1, 默认为1): ", DEFUALT_V)
|
|
|
- a=get_valid_input("请输入加速度缩放因子(0-1, 默认为1): ", DEFUALT_A)
|
|
|
+ vv=get_valid_input("请输入速度缩放因子(0-1, 默认为1): ", DEFUALT_V)
|
|
|
+ #a=get_valid_input("请输入加速度缩放因子(0-1, 默认为1): ", DEFUALT_A)
|
|
|
|
|
|
- rospy.loginfo(f"用户输入的速度缩放因子为{v}, 加速度缩放因子为{a}")
|
|
|
- return v, a
|
|
|
+ #rospy.loginfo(f"用户输入的速度缩放因子为{vv}")
|
|
|
+ return vv
|
|
|
except Exception as e:
|
|
|
rospy.logerr(f"发生错误:{e}")
|
|
|
+
|
|
|
+def ROS2_msgs(msgs, m_sr):
|
|
|
+ for i in range(len(msgs.points)):
|
|
|
+ py_msgs.point.xyz_list.append(msgs.points[i].positions)
|
|
|
+
|
|
|
+ f_p = os.path.join(folder_path, 'outtt.txt')
|
|
|
+ with open(f_p,'w') as file:
|
|
|
+ for point in py_msgs.point.xyz_list:
|
|
|
+ file.write(' '.join(str(value) for value in point)+ "\n")
|
|
|
+
|
|
|
+# def write_ee(msgs, m_sr):
|
|
|
+# for i in range(len(msgs.points)):
|
|
|
+# py_msgs.point.xyz_list.append(msgs.points[i].positions)
|
|
|
+
|
|
|
+# f_p1 = os.path.join(folder_path, 'ee_pos.txt')
|
|
|
+# with open(f_p1,'w') as file:
|
|
|
+# for point in py_msgs.point.xyz_list:
|
|
|
+# #更新机械臂状态
|
|
|
+# state=m_sr.arm.get_current_state()
|
|
|
+# #state.joint_state.position=point
|
|
|
+
|
|
|
+# #获取末端执行器的位姿
|
|
|
+# fk_result=m_sr.arm.get_current_pose(moveit_server.end_effector_link)
|
|
|
+
|
|
|
+# #获取位置信息
|
|
|
+# position=fk_result.pose.position
|
|
|
+# orientation=fk_result.pose.orientation
|
|
|
+
|
|
|
+# #格式化数据
|
|
|
+# pose_data = f"Position: x={position.x}, y={position.y}, z={position.z}\n"
|
|
|
+# pose_data += f"Orientation (Quaternion): x={orientation.x}, y={orientation.y}, z={orientation.z}, w={orientation.w}\n"
|
|
|
+# pose_data += "-" * 50 + "\n"
|
|
|
+
|
|
|
+# file.write(pose_data)
|
|
|
|
|
|
if __name__ =="__main__":
|
|
|
# from redis_publisher import Publisher
|
|
@@ -496,15 +616,17 @@ if __name__ =="__main__":
|
|
|
|
|
|
folder_path = rospy.get_param("folder_path")
|
|
|
rospy.init_node('moveit_control_server', anonymous=False)
|
|
|
- moveit_server = MoveIt_Control(is_use_gripper=False)
|
|
|
-
|
|
|
+ moveit_server = MoveIt_Control(is_use_gripper=False)
|
|
|
welding_sequence = rospy.get_param('welding_sequence')
|
|
|
- v,a=get_user_input()
|
|
|
+
|
|
|
+
|
|
|
+ speed_v=get_user_input()
|
|
|
+ rospy.loginfo(f"用户输入的速度缩放因子为{speed_v}")
|
|
|
+
|
|
|
trajectory,trajectory_merge,trajectory_with_type_merge = moveit_server.path_planning(folder_path)
|
|
|
+ ROS2_msgs(trajectory_with_type_merge,moveit_server)
|
|
|
+ #write_ee(trajectory_with_type_merge, moveit_server)
|
|
|
|
|
|
- # plan_result = ROS2PY_msgs(trajectory_with_type_merge, moveit_server)
|
|
|
- # publisher.pub_plan_result(plan_result)
|
|
|
-
|
|
|
moveit_server.arm.execute(trajectory_merge)
|
|
|
|
|
|
rospy.set_param("sign_control",0)
|