def __init__(self, exp_meta, griddy):
self.exp_meta = exp_meta
self.preStimTime = exp_meta.preStimTime
self.stimTime = exp_meta.stimTime
self.trialTypes = exp_meta.trialTypes
self.trialTypeCnt = len(self.trialTypes)
self.griddy = griddy
self.fly_traj_utils = ftu.fly_trajectory_utils(self.exp_meta)
def __init__(self, exp_meta, trial_data_all):
self.exp_meta = exp_meta
self.trial_data_all = trial_data_all
self.fly_traj_utils = ftu.fly_trajectory_utils( self.exp_meta)
self.fly_traj_utils.rotate( self.trial_data_all )
self.fly_traj_utils.calculate_velocity( self.trial_data_all )
self.FWD_VELOCITY_THRESHOLD = 1000
self.trial_data_vel_filtered = self.fly_traj_utils.filter_fwd_velocity(self.trial_data_all, self.FWD_VELOCITY_THRESHOLD, self.exp_meta.preStimTime, self.exp_meta.stimTime)
self.griddy = ftg.fly_trajectory_griddy(exp_meta, self.trial_data_vel_filtered)
self.classifier = ftc.fly_trajectory_classifier(self.exp_meta, self.griddy)
