import os

import time

import math

import pathlib

from functools import reduce

from collections import Counter

import cv2 as cv

import numpy as np

import pandas as pd

from tqdm import tqdm

import matplotlib.pyplot as plt

def get_path(file_dir):

    path_list = []

    name_list = []

    for path in pathlib.Path(file_dir).iterdir():

        path_list.append(str(path))

        name_list.append(path.name)

    path_list = sorted(path_list, key=lambda path_: int(pathlib.Path(path_).stem))

    name_list = sorted(name_list, key=lambda path_: int(pathlib.Path(path_).stem))

    return path_list, name_list

def check_file(paths):

    if type(paths) is not list:

        paths = [paths]

    for path in paths:

        if not os.path.exists(path):

            os.makedirs(path)

    return

def list_file(dir_path):

    paths = []

    for path in pathlib.Path(dir_path).iterdir():

        paths.append(str(path))

    return paths

def crop_image(read_dir, save_dir, o_w, o_h, r_w, r_h, split=False):

    if not os.path.exists(save_dir):

        os.makedirs(save_dir)

    # Crop large images into small images

    i = 0

    file_paths_list = get_file_path(read_dir)

    for file_path in file_paths_list:

        for row in range(o_h // r_h):

            for col in range(o_w // r_w):

                img = cv.imread(file_path)

                cropped = img[row * r_h: (row + 1) * r_h, col * r_w: (col + 1) * r_w, :]

                if split:

                    save_path = os.path.join(save_dir, str(pathlib.Path(file_path).stem))

                    if not os.path.exists(save_path):

                        os.makedirs(save_path)

                    cv.imwrite(os.path.join(save_path, '{}.jpg'.format(i)), cropped)

                else:

                    cv.imwrite(os.path.join(save_dir, '{}.jpg'.format(i)), cropped)

                i += 1

    print('Cropped image is complete!')

    return