def get_merged_data(): # this is not needed unless TP location is an issue! # cdata, h5file = get_cleaned_data(DATA_FILE) cdata, h5file = get_uncleaned_data(DATA_FILE) naksu_raw = h5file.root.ramppi11.naksu_data.read() merged = [] for d in foreach(cdata, ["session_id", "lap"]): naksu_x, naksu_y, lap_d = get_interp_naksu(d, naksu_raw) naksu_d = np.rec.fromarrays((naksu_x, naksu_y), names="naksu_x,naksu_y") merged.append(colstack(lap_d, naksu_d)) merged = rowstack(merged) return merged
def get_tp_referenced(data): refx = data['scenecam_x'] - data['naksu_x'] refy = data['scenecam_y'] - data['naksu_y'] rec = np.rec.fromarrays((refx, refy), names='tp_referenced_x, tp_referenced_y') data = colstack(data, rec) return data