import time import numpy as np from robomaster import robot ep_robot = robot.Robot() ep_robot.initialize(conn_type="ap") ep_gimbal = ep_robot.gimbal ep_sensor = ep_robot.sensor ep_robot.set_robot_mode(mode='free') dd=[] aa=[] def sub_data_handler(sub_info): d = sub_info dd.append(d[0]) # print(d) def sub_angle_handler(angle_info): pitch_angle, yaw_angle, pitch_ground_angle, yaw_ground_angle = angle_info a=-yaw_ground_angle/180*np.pi aa.append(a) # print("gimbal angle: pitch_angle:{0}, yaw_angle:{1}, pitch_ground_angle:{2}, yaw_ground_angle:{3}".format( # pitch_angle, yaw_angle, pitch_ground_angle, yaw_ground_angle)) ep_gimbal.moveto(pitch=0, yaw=-180).wait_for_completed() ep_sensor.sub_distance(freq=10, callback=sub_data_handler) ep_gimbal.sub_angle(freq=10, callback=sub_angle_handler) ep_gimbal.moveto(pitch=0, yaw=180, yaw_speed=10).wait_for_completed() ep_gimbal.unsub_angle() ep_sensor.unsub_distance() ep_gimbal.moveto(pitch=0, yaw=0).wait_for_completed() ll=min(len(aa), len(dd)) ll import matplotlib.pyplot as plt plt.polar(aa[:ll], dd[:ll], 'b+') ax = plt.gca() # gets the active axis ax.set_rmax(700) ax.set_aspect(1) plt.savefig('polar.png', dpi=300) plt.show()
網頁
▼
沒有留言:
張貼留言
HTML 編輯器