webcam_redis.py

#! /usr/bin/env python3

import cv2
import os
import sys
import pickle
import time
import threading 
import queue 
import redis
from camera_processor import camera_processor 
from expire_processor import expire_processor 

CAMERA_INDEX = 0
CAMERA_REQUEST_VID_WIDTH = 640
CAMERA_REQUEST_VID_HEIGHT = 480

CAMERA_QUEUE_PUT_WAIT_MAX = 0.001
CAMERA_QUEUE_FULL_SLEEP_SECONDS = 0.01

CAMERA_QUEUE_SIZE = 1
#ip_camera=os.environ['IP_CAMERA']
#cap = cv2.VideoCapture('rtsp://localhost:554/video.h264')
#cap = cv2.VideoCapture("http://localhost:8090/facstream.mjpeg")
r = redis.StrictRedis(host='localhost', port=6379, db=0)

def main():

    # Queue of camera images.  Only need two spots
    camera_queue = queue.Queue(CAMERA_QUEUE_SIZE)
    # camera processor that will put camera images on the camera_queue
    camera_proc = camera_processor(camera_queue, CAMERA_QUEUE_PUT_WAIT_MAX, CAMERA_INDEX,
            CAMERA_REQUEST_VID_WIDTH, CAMERA_REQUEST_VID_HEIGHT, CAMERA_QUEUE_FULL_SLEEP_SECONDS)
    camera_proc.start_processing()

    # start thread to expire zset members
    expire_proc = expire_processor(r, 'cam1', 30, 5)
    expire_proc.start()

    n_frames = 0
    start = time.time()
    try:
        while True:
        #for i in range(300):
            f = camera_queue.get()
            camera_queue.task_done()
            
            ts = int(time.time()*1000.0)
            val = {'frame': pickle.dumps(f), 'timestamp': ts}
            r.hmset('cam1-current', val)
            r.zadd('cam1', ts, pickle.dumps(f))
            
            n_frames += 1
            if n_frames % 10 == 0:
                print("FPS: {}".format(n_frames / (time.time() - start)))
                n_frames = 0
                start = time.time()
    except KeyboardInterrupt:
        sys.exit()
    finally:
        print('stop camera thread')
        camera_proc.stop_processing()
        camera_proc.cleanup()
        print('stop redis expire thread')
        expire_proc.stop()
        expire_proc.join()


def frames2video(save_name):
    fps = 15
    #fourcc = cv2.VideoWriter_fourcc(*'FMP4')
    #fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    fourcc = 0x00000021 
    video_writer = cv2.VideoWriter(save_name, fourcc, fps, (CAMERA_REQUEST_VID_WIDTH, CAMERA_REQUEST_VID_HEIGHT), isColor=True)
    #video_writer = cv2.VideoWriter(save_name, 0x00000021, fps, (CAMERA_REQUEST_VID_WIDTH, CAMERA_REQUEST_VID_HEIGHT))
    #import subprocess
    #from PIL import Image 
    #from io import BytesIO
    #ffmpeg_cmd_str = "ffmpeg -loop 0 -f image2pipe -s 640*480 -r 20 -i- -c:v libx264 -pix_fmt yuv420p -y output.mp4"
    #ffmpeg_cmd = [
    #        'ffmpeg',
            #'-loop', '0',
    #        '-f', 'rawvideo',
    #        '-pixel_format', 'bgr24'
    #        '-s', '640*480',
    #        '-r', '20',
    #        '-an', # no audio
    #        '-i', '-', 
    #        '-c:v', 'libx264',
    #        '-pix_fmt', 'yuv420p',
    #        '-y', 'output.mp4'
    #        ]
    #print(" ".join(ffmpeg_cmd))
    obj_lst = r.zrangebyscore('cam1', 0, '+inf')
    #wc = subprocess.Popen(ffmpeg_cmd_str.split(' '), stdin=subprocess.PIPE)
    #wc = subprocess.Popen(ffmpeg_cmd, stdin=subprocess.PIPE)
    for frame in obj_lst:
        video_writer.write(pickle.loads(frame))
        #im = Image.frombuffer(data=pickle.loads(frame).tostring(), size=(640, 480), mode='RGB')
        #im.save(wc.stdin, 'PNG')
        #wc.stdin.write(pickle.loads(frame).tostring())

    #wc.stdin.close()
    #wc.wait()
    video_writer.release()


if __name__ == "__main__":
    main()
    #frames2video('video.mp4')