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)