|
@@ -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,6 +134,7 @@ class MoveIt_Control:
|
|
|
# constraints.orientation_constraints.append(orientation_constraint)
|
|
|
return constraints
|
|
|
|
|
|
+ #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)
|
|
|
self.arm.set_max_velocity_scaling_factor(v)
|
|
@@ -157,39 +162,43 @@ 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
|
|
|
- else:
|
|
|
- rospy.loginfo(error)
|
|
|
- rospy.loginfo("移动规划失败-开始第{}次重新规划".format(i+1))
|
|
|
- self.arm.clear_path_constraints()
|
|
|
- rrr=rrr+0.2#每次增加20厘米半径
|
|
|
- rospy.loginfo("R值:{}".format(rrr))
|
|
|
- if trajectory:
|
|
|
- path_constraints = self.create_path_constraints2(points[-1],point,rrr)#将最后一个点和输入的点(焊缝的起始点)做圆柱轨迹约束
|
|
|
+ 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:
|
|
|
- path_constraints = self.create_path_constraints2(self.home_pose,point,rrr)#将最后一个点和起点(焊缝的起始点)做圆柱轨迹约束
|
|
|
- self.arm.set_path_constraints(path_constraints)#设定约束
|
|
|
- if(i>=(b-5)):
|
|
|
- rospy.loginfo("所有移动规划尝试失败 取消路径约束")
|
|
|
+ rospy.loginfo("check failed! 移动轨迹无效")
|
|
|
+ rospy.loginfo("移动规划失败-开始第{}次重新规划".format(i+1))
|
|
|
self.arm.clear_path_constraints()
|
|
|
- if(i==(b-1)):
|
|
|
- rospy.loginfo("所有移动规划尝试失败 焊缝起点不可达")
|
|
|
- er=1
|
|
|
- break
|
|
|
+ rrr=rrr+0.2#每次增加20厘米半径
|
|
|
+ rospy.loginfo("R值:{}".format(rrr))
|
|
|
+ if trajectory:
|
|
|
+ path_constraints = self.create_path_constraints2(points[-1],point,rrr)#将最后一个点和输入的点(焊缝的起始点)做圆柱轨迹约束
|
|
|
+ else:
|
|
|
+ path_constraints = self.create_path_constraints2(self.home_pose,point,rrr)#将最后一个点和起点(焊缝的起始点)做圆柱轨迹约束
|
|
|
+ self.arm.set_path_constraints(path_constraints)#设定约束
|
|
|
+ if(i>=(b-5)):
|
|
|
+ rospy.loginfo("所有移动规划尝试失败 取消路径约束")
|
|
|
+ self.arm.clear_path_constraints()
|
|
|
+ if(i==(b-1)):
|
|
|
+ rospy.loginfo("所有移动规划尝试失败 焊缝起点不可达")
|
|
|
+ er=1
|
|
|
+ break
|
|
|
#清除约束
|
|
|
self.arm.clear_path_constraints()
|
|
|
|
|
|
return points,trajectory,trajectory_with_type,er
|
|
|
|
|
|
-
|
|
|
+ #TODO move_pl
|
|
|
def move_pl(self,point,points,trajectory,trajectory_with_type,a=1,v=1):#焊缝的规划 只规划当前的点到上一个点
|
|
|
|
|
|
self.arm.set_max_acceleration_scaling_factor(a)
|
|
@@ -197,21 +206,22 @@ class MoveIt_Control:
|
|
|
|
|
|
#设定约束
|
|
|
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,19 @@ 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_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 +283,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 +296,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,6 +311,8 @@ 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)
|
|
@@ -309,16 +326,17 @@ 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,err = self.move_p_flexible(point11,points,trajectory,trajectory_with_type,v=speed_v)
|
|
|
+ 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)
|
|
|
+ points,trajectory,trajectory_with_type = self.move_pl(point22,points,trajectory,trajectory_with_type,v=speed_v)
|
|
|
|
|
|
rospy.loginfo("第%d条焊缝规划完毕",i+1)
|
|
|
+ rospy.loginfo("*******************")
|
|
|
if gohome:
|
|
|
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)
|
|
@@ -345,6 +363,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
|
|
@@ -361,6 +380,7 @@ def mark_the_traj(traj,TYPE,SEQUENCE):
|
|
|
]
|
|
|
return traj_with_type
|
|
|
|
|
|
+#TODO merge_trajectories
|
|
|
def merge_trajectories(trajectories):
|
|
|
if not trajectories:
|
|
|
return None
|
|
@@ -389,6 +409,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
|
|
@@ -472,10 +493,10 @@ def get_valid_input(factor, default):
|
|
|
return default
|
|
|
try:
|
|
|
value=float(user_input)
|
|
|
- if 0<=value<=1:
|
|
|
+ if 0<value<=1:
|
|
|
return value
|
|
|
else:
|
|
|
- rospy.logerr("缩放因子必须在[0, 1]范围内,请重新输入!")
|
|
|
+ rospy.logerr("缩放因子必须在(0, 1]范围内,请重新输入!")
|
|
|
except ValueError:
|
|
|
rospy.logerr("输入值无效,请根据提示重新输入!")
|
|
|
|
|
@@ -495,8 +516,10 @@ def get_user_input():
|
|
|
return vv
|
|
|
except Exception as e:
|
|
|
rospy.logerr(f"发生错误:{e}")
|
|
|
-
|
|
|
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
if __name__ =="__main__":
|
|
|
# from redis_publisher import Publisher
|
|
|
# publisher = Publisher()
|
|
@@ -506,12 +529,13 @@ if __name__ =="__main__":
|
|
|
moveit_server = MoveIt_Control(is_use_gripper=False)
|
|
|
welding_sequence = rospy.get_param('welding_sequence')
|
|
|
|
|
|
-
|
|
|
- speed_v=get_user_input()
|
|
|
- rospy.loginfo(f"用户输入的速度缩放因子为{speed_v}")
|
|
|
+ speed_v=1
|
|
|
+ # speed_v=get_user_input()
|
|
|
+ # rospy.loginfo(f"用户输入的速度缩放因子为{speed_v}")
|
|
|
|
|
|
trajectory,trajectory_merge,trajectory_with_type_merge = moveit_server.path_planning(folder_path)
|
|
|
-
|
|
|
+
|
|
|
+
|
|
|
moveit_server.arm.execute(trajectory_merge)
|
|
|
|
|
|
rospy.set_param("sign_control",0)
|