def compute_frame_indices(self): """ For each frame, find the start and end indices of the time synchronized events """ frame_indices = [] for start_t, end_t in self.timestamps: start_idx = binary_search_h5_dset(self.frame_ts, start_t) end_idx = binary_search_h5_dset(self.frame_ts, end_t) frame_indices.append([start_idx, end_idx]) # assert start_idx != end_idx, "frame start_idx shoudn't be the same as end_idx" return frame_indices
def find_ts_frame_index(self, timestamp): indices = np.array(self.frame_indices) indices = indices[:, 1] _, _, ts, _ = self.get_events_by_idx(indices) idx = binary_search_h5_dset(ts, timestamp) return idx
def find_ts_index(self, timestamp): with h5py.File(self.h5_file_path, 'r') as h5_file: idx = binary_search_h5_dset(h5_file['events/ts'], timestamp) return idx