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
