Exemplo n.º 1
0
 def load_all(self, flight_dir="", image_dir=""):
     self.flight_dir = flight_dir
     self.image_dir = image_dir
     data, flight_format = flight_loader.load(flight_dir)
     #data = flight_data.load('aura', flight_dir)
     self.interp = flight_interp.FlightInterpolate()
     self.interp.build(data)
     #self.load_gps()
     #self.load_filter()
     self.load_events()
     self.load_images()
    recal_file = None
data = flight_loader.load(loader, path, recal_file)
print "imu records:", len(data['imu'])
print "gps records:", len(data['gps'])
if 'air' in data:
    print "airdata records:", len(data['air'])
print "filter records:", len(data['filter'])
if 'pilot' in data:
    print "pilot records:", len(data['pilot'])
if 'act' in data:
    print "act records:", len(data['act'])
if len(data['imu']) == 0 and len(data['gps']) == 0:
    print "not enough data loaded to continue."
    quit()

interp = flight_interp.FlightInterpolate()
interp.build(data)

# set approximate camera orienation (front, down, and rear supported)
cam_facing = 'down'

# resample movie data
movie = np.array(movie, dtype=float)
movie_interp = []
x = movie[:, 0]
movie_spl_roll = interpolate.interp1d(x,
                                      movie[:, 2],
                                      bounds_error=False,
                                      fill_value=0.0)
movie_spl_pitch = interpolate.interp1d(x,
                                       movie[:, 3],