import time
import cv2
import subprocess as sp
# 推流
import ffmpeg
import numpy as np

if __name__== "__main__":


    # with open(str(cv2.__file__),"r") as f:
    #     print (f.read())

    # cap = cv2.VideoCapture("rtmp://live.play.t-aaron.com/live/THSAs")
    # # Get video information
    # fps = int(cap.get(cv2.CAP_PROP_FPS))
    # print(fps)
    # width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    # print(width)
    # height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    # print(height)
    # fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
    # print(fourcc)
    # # print(cv2.getBuildInformation())
    # # cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
    # # cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)
    # # cap.set(cv2.CAP_PROP_FPS, 60)
    # # cap.setExceptionMode(True)
    # print(cap.getExceptionMode())


    # ffmpeg command
    # command = ['ffmpeg',
    #            '-y', # 不经过确认，输出时直接覆盖同名文件。
    #            '-f', 'rawvideo',
    #            '-vcodec','rawvideo',
    #            '-pix_fmt', 'bgr24',
    #             '-s', "{}x{}".format(1920, 1080),
    #            # '-s', "{}x{}".format(1280, 720),
    #            '-i', '-', # 指定输入文件
    #            '-c:v', 'libx264', # 指定视频编码器
    #            '-pix_fmt', 'yuv420p',
    #             '-g', '5',
    #             "-an",
    #             '-b:v', '3000k',
    #            '-preset', 'ultrafast', # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,
    #                                 # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。
    #            '-f', 'flv',
    #            "rtmp://live.push.t-aaron.com/live/THSAk"]
    kwargs = {'format': 'rawvideo',
              'vcodec': 'rawvideo',
              'pix_fmt': 'bgr24',
              's': '{}x{}'.format(int(1920), int(1080))}
    out = {
        'g': str(25),
        'b:v': '6000k',
        # '-bufsize': '3000k',
        'tune': 'zerolatency', # 加速编码速度
        'c:v': 'libx264',  # 指定视频编码器
        'sc_threshold': '0',
        'pix_fmt': 'yuv420p',
        'flvflags': 'no_duration_filesize',
        'preset': 'ultrafast',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,
        # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。
        'format': 'flv'}
    # 管道配置
    process2 = (
        ffmpeg
            .input('pipe:', **kwargs)
            .output('rtmp://live.push.t-aaron.com/live/THSAk', **out)
            .global_args('-y', '-an')
            .overwrite_output()
            .run_async(pipe_stdin=True)
    )
    # 管道配置
    # p = sp.Popen(command, stdin=sp.PIPE, shell=False)

    process = (
        ffmpeg
            .input('rtmp://221.226.114.142:19350/rlive/stream_4?sign=PZIGmXvw')
            .output('pipe:', format='rawvideo', pix_fmt='bgr24')
            .overwrite_output()
            .global_args('-an')
            .run_async(pipe_stdout=True)
    )
    # while(cap.isOpened()):
    while True:
        start =time.time()
        # ret, frame = cap.read()
        # print(cap.grab())
        in_bytes = process.stdout.read(1920 * 1080 * 3)
        # print("aaaaaaaaaaaaaaa", time.time()-start)
        # print("aaaaaaaaaaaaaa", time.time()-start)
        # start =time.time()
        # a,b = cap.retrieve()
        if not in_bytes:
            print("Opening camera is failed")
            break
        frame = (np.frombuffer(in_bytes, np.uint8).reshape([1080, 1920, 3]))
        print("ccccccccccccccc", time.time()-start)
        # process2.stdin.write(frame.tostring())
        # print("bbbbbbbbbbbbbb", time.time()-start)