def test_seek_from_end(playback: PyK4APlayback):
# TODO fetch capture/data and validate time
playback.open()
playback.seek(0, origin=SeekOrigin.END)
capture = playback.get_previouse_capture()
assert capture.color is not None
with pytest.raises(EOFError):
playback.get_next_capture()
def test_seek_from_start(playback: PyK4APlayback):
# TODO fetch capture/data and validate time
playback.open()
playback.get_next_capture()
playback.seek(playback.configuration["start_timestamp_offset_usec"], origin=SeekOrigin.BEGIN)
capture = playback.get_next_capture()
assert capture.color is not None
with pytest.raises(EOFError):
playback.get_previouse_capture()
def test_seek_by_device_time(playback: PyK4APlayback):
# TODO fetch capture/data and validate time
playback.open()
playback.seek(1, origin=SeekOrigin.DEVICE_TIME
) # TODO add correct timestamp from datablock here
capture = playback.get_next_capture()
assert capture.color is not None
def test_get_next_capture(playback: PyK4APlayback):
playback.open()
capture = playback.get_next_capture()
assert capture is not None
assert capture.depth is not None
assert capture.color is not None
assert capture._calibration is not None # Issue #81
def test_get_next_capture(playback: PyK4APlayback):
playback.open()
capture = playback.get_next_capture()
assert capture is not None
assert capture.depth is not None
assert capture.color is not None
assert capture.depth_timestamp_usec == 800222
assert capture.color_timestamp_usec == 800222
assert capture.ir_timestamp_usec == 800222
assert capture._calibration is not None # Issue #81
def play(playback: PyK4APlayback):
while True:
try:
capture = playback.get_next_capture()
if capture.color is not None:
cv2.imshow(
"Color",
convert_to_bgra_if_required(
playback.configuration["color_format"], capture.color))
if capture.depth is not None:
cv2.imshow("Depth", colorize(capture.depth, (None, 5000)))
key = cv2.waitKey(10)
if key != -1:
break
except EOFError:
break
cv2.destroyAllWindows()
def capture(playback: PyK4APlayback) -> PyK4ACapture:
playback.open()
return playback.get_next_capture()