cplayer.py

import os
import cv2
import curses
import argparse
import time

import pyximport
pyximport.install()

from painter import paint_screen, invert_chars


parser = argparse.ArgumentParser(description='ASCII Player')
parser.add_argument("--width", type=int, default = 120, help="width of the terminal window")
parser.add_argument("--fps", type=int, default = 30, help="width of the terminal window")
parser.add_argument("--show", type=bool, default = False, help="show the original video in an opencv window")
parser.add_argument("--inv", type=bool, default = False, help="invert the shades")
parser.add_argument("video", type=str, help="path to video or webcam index")
args = parser.parse_args()
width = args.width

if args.inv:
    invert_chars()

video = args.video
try:
    video = int(video)
except ValueError:
    pass

try:
    if type(video) is str and not os.path.isfile(video):
        print("failed to find video at:", args.video)

    video = cv2.VideoCapture(video)
    ok, frame = video.read()
    if not ok:
        print("could not extract frame from video")

    ratio = width/frame.shape[1]
    height = int(frame.shape[0]*ratio) // 2  # charachter height is 2 times character width
    print(frame.shape)
    print(width, height, ratio)

    curses.initscr()
    window = curses.newwin(height, width, 0, 0)

    frame_count = 0
    frames_per_ms = args.fps/1000
    start = time.perf_counter_ns()//1000000
    while True:
        ok, orig_frame = video.read()
        if not ok:
            break

        frame = cv2.resize(orig_frame, (width, height))
        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        if args.show:
            cv2.imshow("frame", orig_frame)
            cv2.waitKey(1)

        paint_screen(window, frame, width, height)

        elapsed = (time.perf_counter_ns()//1000000) - start
        supposed_frame_count = frames_per_ms * elapsed
        if frame_count > supposed_frame_count:
            time.sleep((frame_count-supposed_frame_count)*(1/frames_per_ms)/1000)
        window.refresh()
        frame_count += 1
finally:
    cv2.destroyAllWindows()
    curses.endwin()
    fps = frame_count / (((time.perf_counter_ns()//1000000) - start) / 1000)
    print("played on average at %d fps" % fps)