# 獲取視頻信息
you-get -i https://www.bilibili.com/video/BV1f4411M7QC
# 下載視頻
you-get --format=flv -o E:\Desktop\output https://www.bilibili.com/video/BV1f4411M7QC

    def cut_video(cls, origin_file_path, to_file_path, start, end):
        """
        視頻剪輯
        :param origin_file_path: 原視頻文件路徑
        :param to_file_path: 保存路徑
        :param start: 起始時間點
        :param end: 結束時間點
        """
        clip = VideoFileClip(origin_file_path).subclip(start, end)
        clip.write_videofile(to_file_path)
    def cut_audio(cls, origin_file_path, to_file_path, start, end):
        """
        音頻剪輯
        :param origin_file_path: 原視頻文件路徑
        :param to_file_path: 保存路徑
        :param start: 起始時間點
        :param end: 結束時間點
        """
        clip = AudioFileClip(origin_file_path).subclip(start, end)
        clip.write_audiofile(to_file_path)
    def get_audio_from_video(cls, video_file_path, to_file_path):
        """
        音頻提取
        :param video_file_path: 視頻文件路徑
        :param to_file_path: 音頻文件路徑
        """
        video = VideoFileClip(video_file_path)
        video.audio.write_audiofile(to_file_path)

    def split(cls, from_file_path, to_folder_path, frames=0):
        """
        視頻按幀讀取并保存
        :param from_file_path: 視頻路徑
        :param to_folder_path: 保存路徑
        :param frames: 保存幀數(張數)，為0則保存所有幀
        """
        vc = cv2.VideoCapture(from_file_path)    # cv2打開視頻文件
        frames_count = vc.get(7)    # 獲取視頻總幀數
        c = 0
        if vc.isOpened():
            ret, frame = vc.read()          # 按幀讀取視頻
        else:
            ret = False
        while ret:
            if 0  frames == c:
                break
            ret, frame = vc.read()  # 讀取每一視頻幀，并保存至圖片中
            cv2.imwrite(os.path.join(to_folder_path, '{}.jpg'.format(c)), frame)
            c += 1
            if c == frames_count - 1:
                break
            print('第 {} 張圖片存放成功！'.format(c))

    def binary_option_cv2(cls, from_file_path, to_file_path):
        """
        圖片二值化并保存(使用cv2)
        :param from_file_path: 原圖路徑
        :param to_file_path: 二值化圖路徑
        """
        img = cv2.imread(from_file_path)
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        new_gray = np.uint8((255 * (gray / 255.0) ** 1.4))
        dst = cv2.adaptiveThreshold(new_gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 15, 1)
        cv2.medianBlur(dst, 5)
        cv2.imwrite(to_file_path, dst)
    def binary_option_baidu(cls, from_file_path, to_file_path):
        """
        圖片二值化并保存(使用百度人像分割)
        :param from_file_path: 原圖路徑
        :param to_file_path: 二值化圖路徑
        """
        def get_file_content(filePath):
            with open(filePath, 'rb') as fp:
                return fp.read()
        height, width, bgr = cv2.imread(from_file_path).shape
        image = get_file_content(from_file_path)
        cls.client.bodySeg(image)
        res = cls.client.bodySeg(image)
        labelmap = base64.b64decode(res['labelmap'])
        labelimg = np.frombuffer(labelmap, np.uint8)  # 轉化為np數組 0-255
        labelimg = cv2.imdecode(labelimg, 1)
        labelimg = cv2.resize(labelimg, (width, height), interpolation=cv2.INTER_NEAREST)
        img_new = np.where(labelimg == 1, 255, labelimg)  # 將 1 轉化為 255
        cv2.imwrite(to_file_path, img_new)

    def joint(cls, origin_folder, word_cloud_folder, to_file_path):
        """
        批量拼接圖片并合成視頻
        :param origin_folder: 原圖文件夾
        :param word_cloud_folder: 詞云圖片文件夾
        :param to_file_path: 保存路徑
        """
        num_list = [int(str(i).split('.')[0]) for i in os.listdir(origin_folder)]
        fps = 30  # 視頻幀率，需要根據原視頻幀率做調整
        height, width, _ = cv2.imread(os.path.join(origin_folder, '{}.jpg'.format(num_list[0]))).shape  # 視頻高度和寬度
        width = width * 2
        # 創建一個寫入操作;
        video_writer = cv2.VideoWriter(to_file_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
        for i in sorted(num_list):
            i = '{}.jpg'.format(i)
            ori_jpg = os.path.join(origin_folder, str(i))
            word_jpg = os.path.join(word_cloud_folder, str(i))
            # com_jpg = os.path.join(Composite_path,str(i))
            ori_arr = cv2.imread(ori_jpg)
            word_arr = cv2.imread(word_jpg)
            # 利用 Numpy 進行拼接
            com_arr = np.hstack((ori_arr, word_arr))
            video_writer.write(com_arr)  # 將每一幀畫面寫入視頻流中
            print("{}寫入視頻流成功".format(ori_jpg))

    def set_audio_for_video(cls, video_file_path, audio_file_path, to_file_path):
        """
        音視頻合并
        :param video_file_path: 視頻文件路徑
        :param audio_file_path: 音頻文件路徑
        :param to_file_path: 保存路徑
        """
        video = VideoFileClip(video_file_path)
        audio = AudioFileClip(audio_file_path)
        new_video = video.set_audio(audio)
        new_video.write_videofile(to_file_path)