import numpy as np
from glob import glob
import os
import cv2

class Movie2imgConverter:
    def __init__(self, args):
        self.args = args

    def convert_endo_movie2img(self):
        src_dir = self.args.movie_dir
        if os.path.exists(src_dir) is False:
            print("動画のディレクトリパスがミスってます")
            raise

        movie_name_list = glob(os.path.join(src_dir, "*" + self.args.endo_movie_extend))

        if len(movie_name_list) == 0:
            print("対象のディレクトリに動画が見つかりません")
            raise

        sequence_num = 0
        for movie_name in movie_name_list:
            print("start processing {}".format(os.path.basename(movie_name)))
            sum_frames = cv2.VideoCapture(movie_name)
            out_subdir = os.path.join(self.args.out_dir, '{}seq'.format(sequence_num))