def test_load_intrinsics(tmp_dir):
match_pattern = "*.time"
rename_set = RenameSet(tmp_dir, match_pattern)
# load_intrinsics
assert "world.intrinsics" not in os.listdir(tmp_dir)
rename_set.load_intrinsics()
assert "world.intrinsics" in os.listdir(tmp_dir)
def test_load_intrinsics(tmp_dir):
match_pattern = "*.time"
rename_set = RenameSet(tmp_dir, match_pattern)
# load_intrinsics
assert "world.intrinsics" not in os.listdir(tmp_dir)
rename_set.load_intrinsics()
assert "world.intrinsics" in os.listdir(tmp_dir)
def test_rewrite_timestamps(tmp_dir):
match_pattern = "*.time"
# Rewrite file end with time to npy timestamp file
rewrite_time = RenameSet(tmp_dir, match_pattern, ['time'])
rewrite_time.rewrite_time("_timestamps.npy")
# Since we didn't rename the file, so the file name will stay the same
assert all(elem in os.listdir(tmp_dir) for elem in [
"audio_00010000_timestamps.npy", "Pupil Cam1 ID0_timestamps.npy",
"Pupil Cam1 ID2_timestamps.npy"])
def test_rewrite_timestamps(tmp_dir):
match_pattern = "*.time"
# Rewrite file end with time to npy timestamp file
rewrite_time = RenameSet(tmp_dir, match_pattern, ['time'])
rewrite_time.rewrite_time("_timestamps.npy")
# Since we didn't rename the file, so the file name will stay the same
assert all(elem in os.listdir(tmp_dir) for elem in [
"audio_00010000_timestamps.npy", "Pupil Cam1 ID0_timestamps.npy",
"Pupil Cam1 ID2_timestamps.npy"
])
def test_rename_file(tmp_dir):
match_pattern = "*.time"
rename_set = RenameSet(tmp_dir, match_pattern)
# Rename to eye0
rename_set.rename("Pupil Cam(0|1) ID0", "eye0")
assert set(["eye0.mp4", "eye0_001.mp4"]).issubset(set(os.listdir(tmp_dir)))
assert not any(
elem in os.listdir(tmp_dir)
for elem in ["Pupil Cam1 ID0.mp4", "Pupil Cam1 ID0_001.mp4"])
# Rename to world
rename_set.rename("Pupil Cam(0|1) ID2", "world")
assert set(["world.mp4",
"world_001.mp4"]).issubset(set(os.listdir(tmp_dir)))
# Rename audio
rename_set.rename("audio_[0-9]+", "audio")
assert "audio.mp4" in os.listdir(tmp_dir)
def test_rename_file(tmp_dir):
match_pattern = "*.time"
rename_set = RenameSet(tmp_dir, match_pattern)
# Rename to eye0
rename_set.rename("Pupil Cam(0|1) ID0", "eye0")
assert set(["eye0.mp4", "eye0_001.mp4"]).issubset(set(os.listdir(tmp_dir)))
assert not any(
elem in os.listdir(tmp_dir) for elem in
["Pupil Cam1 ID0.mp4", "Pupil Cam1 ID0_001.mp4"])
# Rename to world
rename_set.rename("Pupil Cam(0|1) ID2", "world")
assert set(["world.mp4", "world_001.mp4"]).issubset(set(os.listdir(tmp_dir)))
# Rename audio
rename_set.rename("audio_[0-9]+", "audio")
assert "audio.mp4" in os.listdir(tmp_dir)
def convert_pupil_mobile_recording_to_v094(rec_dir):
logger.info("Converting Pupil Mobile recording to v0.9.4 format")
# convert time files and rename corresponding videos
match_pattern = "*.time"
rename_set = RenameSet(rec_dir, match_pattern)
rename_set.load_intrinsics()
rename_set.rename("Pupil Cam([0-3]) ID0", "eye0")
rename_set.rename("Pupil Cam([0-3]) ID1", "eye1")
rename_set.rename("Pupil Cam([0-2]) ID2", "world")
# Rewrite .time file to .npy file
rewrite_time = RenameSet(rec_dir, match_pattern, ["time"])
rewrite_time.rewrite_time("_timestamps.npy")
pupil_data_loc = os.path.join(rec_dir, "pupil_data")
if not os.path.exists(pupil_data_loc):
logger.info('Creating "pupil_data"')
fm.save_object(
{"pupil_positions": [], "gaze_positions": [], "notifications": []},
pupil_data_loc,
)