import sys
from pathlib import Path
import csv

import pupil_labs.neon_recording as nr
from tqdm import tqdm

recording_path = Path(sys.argv[1])
recording = nr.load(recording_path)

output_file_path = recording_path / 'exports' / 'world_timestamps_and_pts.csv'
with output_file_path.open('w') as csv_file:
    writer = csv.writer(csv_file)
    writer.writerow(['timestamp [ns]', 'pts'])
    for frame in tqdm(recording.scene):
        writer.writerow([frame.ts * 1e9, frame.pts])

print(f"Wrote {output_file_path.absolute()}")