def get_all_tracks(self, filename):
events = self.get_all_particles_in_event(filename)
tracks = []
for event in events:
track = Track()
for hit in event.tracker_hits:
if (hit.magnitude() < 5):
track.add_point(hit)
tracks.append(track)
return tracks
def analysis_to_track_dict(file_path, track_map):
file_name = file_path.split('/')[-1]
camera_id = file_name.replace('_train.txt', '')
camera_id = camera_id.replace('c', '')
lines = open(file_path, 'r').readlines()
for line in lines:
words = line.strip('\n').split(',')
index = words[0]
#id = camera_id + '_' +words[1]
id = words[1]
gps = words[7].split('-')
gps_x = float(gps[0])
gps_y = float(gps[2])
time = float(words[9])
point = Point(gps_x, gps_y, time, id, camera_id)
if id not in track_map.tracks_dict:
track = Track(id, [])
track.add_point(point)
track_map.add_track(track)
else:
track_map.add_point(point)
return len(lines)
