|
|
@ -1,86 +1,107 @@ |
|
|
|
from datetime import datetime, date, time, timedelta |
|
|
|
from datetime import datetime, date, time, timedelta |
|
|
|
import os |
|
|
|
import os |
|
|
|
import cv2 |
|
|
|
import cv2 |
|
|
|
|
|
|
|
from wakepy import keep |
|
|
|
|
|
|
|
|
|
|
|
def five_minute_segment_from(the_time): |
|
|
|
def five_minute_segment_from(the_time): |
|
|
|
return the_time.replace(minute=(the_time.minute // 1)*1, second=0, microsecond=0) |
|
|
|
# this is mocked out too be a one minute segment instead |
|
|
|
|
|
|
|
return the_time.replace(second=0, microsecond=0) |
|
|
|
|
|
|
|
#return the_time.replace(minute=(the_time.minute // 1)*1, second=0, microsecond=0) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def path_to_video_at(time_stamp): |
|
|
|
def path_to_video_at(time_stamp): |
|
|
|
return 'videos/' + str(time_stamp) + '.mkv' |
|
|
|
return 'videos/' + str(time_stamp) + '.mkv' # mkv required for h264 encoding |
|
|
|
|
|
|
|
|
|
|
|
#delay for long delay |
|
|
|
os.environ["OPENCV_FFMPEG_CAPTURE_OPTIONS"]="hwaccel;vaapi;hw_decoders_any;vaapi,vdpau" # maybe this enables hardware accelerated ffmpeg? |
|
|
|
long_delay = timedelta(hours=1) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#delay for long delay |
|
|
|
|
|
|
|
long_delay = timedelta(minutes=1) |
|
|
|
|
|
|
|
|
|
|
|
queue = [(False, None)] * 100 #queue of recent frames |
|
|
|
|
|
|
|
last_time_stamp = five_minute_segment_from(datetime.now()) |
|
|
|
last_time_stamp = five_minute_segment_from(datetime.now()) |
|
|
|
|
|
|
|
|
|
|
|
# live webcam feed |
|
|
|
# live webcam feed |
|
|
|
camera = cv2.VideoCapture(0) |
|
|
|
camera = cv2.VideoCapture(0, cv2.CAP_FFMPEG) |
|
|
|
if not camera.isOpened(): |
|
|
|
if not camera.isOpened(): |
|
|
|
print("Cannot open camera") |
|
|
|
print("Cannot open camera") |
|
|
|
exit() |
|
|
|
exit() |
|
|
|
|
|
|
|
|
|
|
|
# video writer |
|
|
|
# video writer |
|
|
|
fourcc = cv2.VideoWriter_fourcc(*'H264') |
|
|
|
fourcc = cv2.VideoWriter_fourcc(*'H264') |
|
|
|
out = cv2.VideoWriter(path_to_video_at(last_time_stamp), fourcc, camera.get(cv2.CAP_PROP_FPS), (int(camera.get(3)),int(camera.get(4)))) |
|
|
|
video_writer = cv2.VideoWriter(path_to_video_at(last_time_stamp), |
|
|
|
|
|
|
|
fourcc, |
|
|
|
|
|
|
|
camera.get(cv2.CAP_PROP_FPS), |
|
|
|
|
|
|
|
(int(camera.get(3)),int(camera.get(4))), |
|
|
|
|
|
|
|
(cv2.VIDEOWRITER_PROP_HW_ACCELERATION, cv2.VIDEO_ACCELERATION_ANY) |
|
|
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
# video feed from previously captured video |
|
|
|
# video feed from previously captured video |
|
|
|
old_time = cv2.VideoCapture(path_to_video_at(last_time_stamp)) |
|
|
|
old_time = cv2.VideoCapture(path_to_video_at(last_time_stamp), cv2.CAP_FFMPEG) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# fullscreen window yess good |
|
|
|
cv2.namedWindow("mirror", cv2.WINDOW_NORMAL) |
|
|
|
cv2.namedWindow("mirror", cv2.WINDOW_NORMAL) |
|
|
|
cv2.setWindowProperty("mirror", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN) |
|
|
|
cv2.setWindowProperty("mirror", cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN) |
|
|
|
|
|
|
|
|
|
|
|
idx = 0 |
|
|
|
queue = [(False, None)] * 100 #queue of recent frames |
|
|
|
offset = 0 |
|
|
|
idx = 0 # current index into the frame queue |
|
|
|
longterm = False |
|
|
|
short_term_delay_frames = 20 # hoow many frames back are we going in the buffer when we're on a short term delay? |
|
|
|
while(True): |
|
|
|
longterm = False # are we shoowing a long term delay? (versus a sshort term delay) |
|
|
|
if last_time_stamp != five_minute_segment_from(datetime.now()): |
|
|
|
|
|
|
|
# find file older than our long delay |
|
|
|
with keep.presenting(): |
|
|
|
old_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay - timedelta(minutes=5))) |
|
|
|
while(True): |
|
|
|
print(f"deleting old file {old_file}") |
|
|
|
if last_time_stamp != five_minute_segment_from(datetime.now()): |
|
|
|
os.remove(old_file) |
|
|
|
# find file older than our long delay |
|
|
|
|
|
|
|
old_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay - timedelta(minutes=1))) |
|
|
|
out.release() |
|
|
|
if os.path.isfile(old_file): |
|
|
|
out = cv2.VideoWriter(path_to_video_at(last_time_stamp), |
|
|
|
print(f"deleting old file {old_file}") |
|
|
|
fourcc, |
|
|
|
os.remove(old_file) |
|
|
|
camera.get(cv2.CAP_PROP_FPS), |
|
|
|
|
|
|
|
(int(camera.get(3)), int(camera.get(4))) |
|
|
|
video_writer.release() |
|
|
|
) |
|
|
|
newly_recording_file = path_to_video_at(five_minute_segment_from(datetime.now())) |
|
|
|
if old_time.isOpened(): |
|
|
|
|
|
|
|
old_time.release() |
|
|
|
print(f"opening {newly_recording_file} too record this upcoming ssegment to") |
|
|
|
|
|
|
|
video_writer = cv2.VideoWriter(newly_recording_file, |
|
|
|
new_file = path_to_video_at(five_minute_segment_from(datetime.now() - timedelta(minutes=10))) |
|
|
|
fourcc, |
|
|
|
pint(f"opening the previously recorded file {new_file} to read from for long mirror") |
|
|
|
camera.get(cv2.CAP_PROP_FPS), |
|
|
|
old_time = cv2.VideoCapture(new_file) |
|
|
|
(int(camera.get(3)), int(camera.get(4))), |
|
|
|
|
|
|
|
(cv2.VIDEOWRITER_PROP_HW_ACCELERATION, cv2.VIDEO_ACCELERATION_ANY) |
|
|
|
last_time_stamp = five_minute_segment_from(datetime.now()) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
ret, frame = camera.read() |
|
|
|
|
|
|
|
frame = cv2.flip(frame, 1) |
|
|
|
if old_time.isOpened(): |
|
|
|
|
|
|
|
old_time.release() |
|
|
|
# record the frame to the appropriate file, and also to the short term buffer |
|
|
|
|
|
|
|
out.write(frame) |
|
|
|
new_file = path_to_video_at(five_minute_segment_from(datetime.now() - long_delay)) |
|
|
|
queue[idx] = ret, frame |
|
|
|
print(f"opening the previously recorded file {new_file} to read from for long mirror") |
|
|
|
|
|
|
|
old_time = cv2.VideoCapture(new_file, cv2.CAP_FFMPEG) |
|
|
|
prev_frame_valid, prev_frame = queue[(idx - offset) % len(queue)] |
|
|
|
|
|
|
|
old_frame_valid, old_frame = old_time.read() |
|
|
|
|
|
|
|
if not longterm and prev_frame_valid: |
|
|
|
last_time_stamp = five_minute_segment_from(datetime.now()) |
|
|
|
cv2.imshow('mirror', prev_frame) |
|
|
|
|
|
|
|
elif longterm and old_frame_valid: |
|
|
|
ret, frame = camera.read() |
|
|
|
cv2.imshow('mirror', old_frame) |
|
|
|
frame = cv2.flip(frame, 1) |
|
|
|
|
|
|
|
|
|
|
|
key = cv2.waitKey(16) # listen for keyprress for some number of ms |
|
|
|
# record the frame to the appropriate file, and also to the short term buffer |
|
|
|
if key & 0xFF == ord('q') or key == 27: # q or esc |
|
|
|
video_writer.write(frame) |
|
|
|
break |
|
|
|
queue[idx] = ret, frame |
|
|
|
if key & 0xFF == ord('a'): |
|
|
|
|
|
|
|
offset += 1 |
|
|
|
prev_frame_valid, prev_frame = queue[(idx - short_term_delay_frames) % len(queue)] |
|
|
|
if key & 0xFF == ord('d'): |
|
|
|
old_frame_valid, old_frame = old_time.read() |
|
|
|
offset -= 1 |
|
|
|
if not longterm and prev_frame_valid: |
|
|
|
if key & 0xFF == ord('z'): |
|
|
|
cv2.imshow('mirror', prev_frame) |
|
|
|
longterm = not longterm |
|
|
|
elif longterm and old_frame_valid: |
|
|
|
|
|
|
|
cv2.imshow('mirror', old_frame) |
|
|
|
idx = (idx + 1) % len(queue) |
|
|
|
|
|
|
|
|
|
|
|
key = cv2.waitKey(16) # listen for keyprress for 16 ms |
|
|
|
|
|
|
|
if key & 0xFF == ord('q') or key == 27: # q or esc |
|
|
|
|
|
|
|
break |
|
|
|
|
|
|
|
if key & 0xFF == ord('a'): |
|
|
|
|
|
|
|
short_term_delay_frames += 1 |
|
|
|
|
|
|
|
if key & 0xFF == ord('d'): |
|
|
|
|
|
|
|
short_term_delay_frames -= 1 |
|
|
|
|
|
|
|
if key & 0xFF == ord('z'): |
|
|
|
|
|
|
|
longterm = not longterm |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
idx = (idx + 1) % len(queue) |
|
|
|
|
|
|
|
|
|
|
|
# After the loop release the cap object |
|
|
|
# After the loop release the cap object |
|
|
|
camera.release() |
|
|
|
camera.release() |
|
|
|