00001 import numpy as np 00002 from generic import get_header_time 00003 from starmac_tools.timeseries import quaternion_to_eulerypr 00004 00005 def process_imu(bag, topic, output_var): 00006 """ 00007 """ 00008 output_var.t = get_header_time(bag, topic) 00009 output_var.ori_quat = np.array(tuple(bag._data[topic+'/orientation/%s' % ax] for ax in 'wxyz')).T 00010 output_var.ori_ypr = np.degrees(np.array(quaternion_to_eulerypr(output_var.ori_quat.T)).T)