class PlotWrapper:
"""
PlotWrapper class for connecting ROS topics to the Plotter object
"""
def __init__(self, update_freq=30, use_degrees=True):
# Store parameters
self.use_degrees = use_degrees
# Setup the plotter window
self.plotter = Plotter()
# Define plot names
plots = [
'n _n_g', 'e _e_g', 'h _h_g', 'Va _Va_g', 'chi _chi_g', 'e n --2d'
]
#plots = [['n', 'n_g'], ['e', 'e_g'], ['h', 'h_g'],
# ['Va', 'Va_g'], ['chi', 'chi_g'], ['traj']]
# Add plots to the window
for p in plots:
self.plotter.add_plot(p)
self.tf_listener = tf.TransformListener()
# Define input vectors for easier input
self.plotter.define_input_vector('neh', ['n', 'e', 'h'])
self.plotter.define_input_vector('neh_g', ['n_g', 'e_g', 'h_g'])
# Subscribe to relevant ROS topics
rospy.Subscriber('ned_g', Vector3Stamped, self.ned_g_cb_)
rospy.Subscriber('twist', TwistStamped, self.velocity_cb_)
rospy.Subscriber('chi_g', Float32, self.chi_g_cb_)
rospy.Subscriber('Va_g', Float32, self.va_g_cb_)
self.crab = 0.0
self.wind = np.zeros(3)
# Update the plots
rate = rospy.Rate(update_freq)
while not rospy.is_shutdown():
self.tick()
self.plotter.update_plots()
rate.sleep()
def velocity_cb_(self, msg):
t = msg.header.stamp.to_sec()
twist = msg.twist
self.crab = np.arctan2(twist.linear.y, twist.linear.x)
V_g = np.array([twist.linear.x, twist.linear.y, twist.linear.z])
Va = np.linalg.norm(V_g - self.wind)
self.plotter.add_measurement('Va', Va, t)
def ned_g_cb_(self, msg):
t = msg.header.stamp.to_sec()
vector = msg.vector
self.plotter.add_vector_measurement('neh_g',
[vector.x, vector.y, -vector.z], t)
def va_g_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('Va_g', msg.data, t)
def chi_g_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('chi_g',
msg.data,
t,
rad2deg=self.use_degrees)
def pose(self):
try:
# Extract time
t = self.tf_listener.getLatestCommonTime("world_ned", "base_link")
except:
return
# Handle position measurements
position, quaternion = self.tf_listener.lookupTransform(
"world_ned", "base_link", t)
position[2] = -position[2]
t = t.to_sec()
self.plotter.add_vector_measurement('neh', position, t)
# Use ROS tf to convert to Euler angles from quaternion
euler = tf.transformations.euler_from_quaternion(quaternion)
# Add angles and angular velocities
self.plotter.add_measurement('chi',
euler[2] + self.crab,
t,
rad2deg=self.use_degrees)
def tick(self):
self.pose()
class PlotWrapper:
"""
PlotWrapper class for connecting ROS topics to the Plotter object
"""
def __init__(self, update_freq=30, use_degrees=True):
self.t0 = rospy.Time.now().to_sec()
# Store parameters
self.use_degrees = use_degrees
# Setup the plotter window
self.plotter = Plotter()
# Define plot names
plots = [
'n n_hat n_gps -l', 'e e_hat e_gps -l', 'h h_hat h_gps h_c -l',
'phi phi_hat -l', 'theta theta_hat -l', 'psi psi_hat -l',
'u u_hat -l', 'v v_hat -l', 'w w_hat -l', 'p p_hat p_gyro -l',
'q q_hat q_gyro -l', 'r r_hat r_gyro -l', 'ax', 'ay', 'az',
'Va Va_c -l', 'chi chi_hat chi_c chi_gps -l', 'p_diff p_abs -l'
]
self.wind = np.zeros(3)
self.crab = 0.0
# Add plots to the window
for p in plots:
self.plotter.add_plot(p)
# Add legends
# self.plotter.add_legend('n')
# self.plotter.add_legend('e')
# self.plotter.add_legend('h')
# self.plotter.add_legend('p')
# self.plotter.add_legend('q')
# self.plotter.add_legend('r')
# self.plotter.add_legend('Va')
# self.plotter.add_legend('chi')
self.tf_listener = tf.TransformListener()
# Define input vectors for easier input
self.plotter.define_input_vector('neh', ['n', 'e', 'h'])
self.plotter.define_input_vector('lin_velocity', ['u', 'v', 'w'])
self.plotter.define_input_vector('ang_velocity', ['p', 'q', 'r'])
self.plotter.define_input_vector('euler', ['phi', 'theta', 'psi'])
self.plotter.define_input_vector('neh_gps',
['n_gps', 'e_gps', 'h_gps'])
self.plotter.define_input_vector('ang_gyro',
['p_gyro', 'q_gyro', 'r_gyro'])
self.plotter.define_input_vector('acc', ['ax', 'ay', 'az'])
self.plotter.define_input_vector('euler_est',
['phi_hat', 'theta_hat', 'psi_hat'])
self.plotter.define_input_vector('neh_est',
['n_hat', 'e_hat', 'h_hat'])
self.plotter.define_input_vector('vb_est', ['u_hat', 'v_hat', 'w_hat'])
# Subscribe to relevant ROS topics
rospy.Subscriber('twist', TwistStamped, self.velocity_cb_)
rospy.Subscriber('imu_lpf', Imu, self.imu_cb_)
rospy.Subscriber('Va_c', Float32, self.va_cb_)
rospy.Subscriber('gps_neh', Vector3Stamped, self.gps_neh_cb_)
rospy.Subscriber('gps_chi', Float32, self.gps_chi_cb_)
rospy.Subscriber('gps_vg', Float32, self.gps_vg_cb_)
rospy.Subscriber('h_c', Float32, self.h_cb_)
rospy.Subscriber('chi_c', Float32, self.chi_cb_)
rospy.Subscriber('p_diff', Float32, self.p_diff_cb_)
rospy.Subscriber('p_static', Float32, self.p_static_cb_)
rospy.Subscriber('euler_est', Vector3Stamped, self.euler_est_cb_)
rospy.Subscriber('ned_est', Vector3Stamped, self.ned_est_cb_)
rospy.Subscriber('chi_est', Float32, self.chi_est_cb_)
rospy.Subscriber('vb_est', Vector3Stamped, self.vb_est_cb_)
self.va_c = None
self.h_c = None
self.chi_c = None
# Update the plots
rate = rospy.Rate(update_freq)
while not rospy.is_shutdown():
self.tick()
self.plotter.update_plots()
rate.sleep()
def p_diff_cb_(self, msg):
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('p_diff', msg.data, t - self.t0)
def chi_est_cb_(self, msg):
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('chi_hat',
msg.data,
t - self.t0,
rad2deg=True)
def p_static_cb_(self, msg):
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('p_abs', msg.data, t - self.t0)
def velocity_cb_(self, msg):
# Extract time
t = msg.header.stamp.to_sec()
# Handle position measurements
twist = msg.twist
self.plotter.add_vector_measurement(
'lin_velocity', [twist.linear.x, twist.linear.y, twist.linear.z],
t - self.t0)
self.plotter.add_vector_measurement(
'ang_velocity',
[twist.angular.x, twist.angular.y, twist.angular.z],
t - self.t0,
rad2deg=self.use_degrees)
self.crab = np.arctan2(twist.linear.y, twist.linear.x)
V_g = np.array([twist.linear.x, twist.linear.y, twist.linear.z])
Va = np.linalg.norm(V_g - self.wind)
self.plotter.add_measurement('Va', Va, t - self.t0)
def imu_cb_(self, msg):
t = msg.header.stamp.to_sec()
gyro = msg.angular_velocity
acc = msg.linear_acceleration
self.plotter.add_vector_measurement('acc', [acc.x, acc.y, acc.z],
t - self.t0)
self.plotter.add_vector_measurement('ang_gyro',
[gyro.x, gyro.y, gyro.z],
t - self.t0,
rad2deg=self.use_degrees)
def va_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('Va_c', msg.data, t - self.t0)
self.va_c = msg.data
def gps_neh_cb_(self, msg):
t = msg.header.stamp.to_sec()
self.plotter.add_vector_measurement(
'neh_gps', [msg.vector.x, msg.vector.y, msg.vector.z], t - self.t0)
def vb_est_cb_(self, msg):
t = msg.header.stamp.to_sec()
self.plotter.add_vector_measurement(
'vb_est', [msg.vector.x, msg.vector.y, msg.vector.z], t - self.t0)
def ned_est_cb_(self, msg):
t = msg.header.stamp.to_sec()
self.plotter.add_vector_measurement(
'neh_est', [msg.vector.x, msg.vector.y, -msg.vector.z],
t - self.t0)
def gps_chi_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('chi_gps',
msg.data,
t - self.t0,
rad2deg=self.use_degrees)
def gps_vg_cb_(self, msg):
pass
def h_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('h_c', msg.data, t - self.t0)
self.h_c = msg.data
def chi_cb_(self, msg):
# Extract time
t = rospy.Time.now().to_sec()
self.plotter.add_measurement('chi_c',
msg.data,
t - self.t0,
rad2deg=self.use_degrees)
self.chi_c = msg.data
def euler_est_cb_(self, msg):
# Extract time
t = msg.header.stamp.to_sec()
self.plotter.add_vector_measurement(
'euler_est', [msg.vector.x, msg.vector.y, msg.vector.z],
t - self.t0,
rad2deg=self.use_degrees)
def pose(self):
try:
# Extract time
t = self.tf_listener.getLatestCommonTime("world_ned", "base_link")
except:
return
# Handle position measurements
position, quaternion = self.tf_listener.lookupTransform(
"world_ned", "base_link", t)
position[2] = -position[2]
t = t.to_sec()
self.plotter.add_vector_measurement('neh', position, t - self.t0)
# Use ROS tf to convert to Euler angles from quaternion
euler = tf.transformations.euler_from_quaternion(quaternion)
# Add angles and angular velocities
self.plotter.add_vector_measurement('euler',
euler,
t - self.t0,
rad2deg=self.use_degrees)
self.plotter.add_measurement('chi',
euler[2] + self.crab,
t - self.t0,
rad2deg=self.use_degrees)
def tick(self):
self.pose()
t = rospy.Time.now().to_sec()
if self.va_c is not None:
self.plotter.add_measurement('Va_c', self.va_c, t - self.t0)
if self.h_c is not None:
self.plotter.add_measurement('h_c', self.h_c, t - self.t0)
if self.chi_c is not None:
self.plotter.add_measurement('chi_c',
self.chi_c,
t - self.t0,
rad2deg=self.use_degrees)
class OrbitPlotter:
''' Plotter wrapper for orbit analysis '''
def __init__(self, plotting_freq=1):
self.plotter = Plotter(plotting_freq, time_window=30)
self.plotter.set_plots_per_row(2)
# Define plot names
plots = self._define_plots()
# Add plots to the window
for p in plots:
self.plotter.add_plotboxes(p)
# Define state vectors for simpler input
self._define_input_vectors()
self.R_err_thresh = 0.01
self.R_thresh_reached = False
self.az_err_thresh = 0.001
self.az_thresh_reached = False
def _define_plots(self):
plots = [['y_r x_r y_t_r x_t_r -2d', 'y x y_t x_t -2d'],
['_R R_error', '_az az_error'], ['phi_c', 'psi'],
['vx vx_e -l', 'vy vy_e']]
return plots
def _define_input_vectors(self):
self.plotter.define_input_vector("state",
['x_r', 'y_r', 'psi', 'az', 'R'])
self.plotter.define_input_vector("actual_pos", ['x', 'y'])
self.plotter.define_input_vector("target_pos", ['x_t', 'y_t'])
self.plotter.define_input_vector("target_rel_pos", ['x_t_r', 'y_t_r'])
self.plotter.define_input_vector("target_vel", ['vx', 'vy'])
self.plotter.define_input_vector("target_vel_est", ['vx_e', 'vy_e'])
def update(self, state, R_c, az_err, phi_c, target_pos, target_vel,
target_vel_est, t):
x_r, y_r, psi, az, R = state
self.plotter.add_vector_measurement("state", state, t)
actual_pos = [x_r + target_pos[0], y_r + target_pos[1]]
self.plotter.add_vector_measurement("actual_pos", actual_pos, t)
self.plotter.add_measurement("phi_c", phi_c, t)
self.plotter.add_vector_measurement("target_pos", target_pos, t)
self.plotter.add_vector_measurement("target_rel_pos", [0, 0], t)
R_err = R - R_c
if not self.R_thresh_reached and abs(R_err) < self.R_err_thresh:
print("R error threshold ({0}) reached at t={1}".format(
self.R_err_thresh, t))
self.R_thresh_reached = True
if not self.az_thresh_reached and abs(az_err) < self.az_err_thresh:
print("az error threshold ({0}) reached at t={1}".format(
self.az_err_thresh, t))
self.az_thresh_reached = True
self.plotter.add_measurement("R_error", R_err, t)
self.plotter.add_measurement("az_error", az_err, t)
self.plotter.add_vector_measurement("target_vel", target_vel, t)
self.plotter.add_vector_measurement("target_vel_est", target_vel_est,
t)
self.plotter.update_plots()