"""

DeepSlide

General helper methods used in other functions.

Authors: Jason Wei, Behnaz Abdollahi, Saeed Hassanpour

"""

import datetime

from pathlib import Path

from typing import (Dict, List)

# Valid image extensions.

IMAGE_EXTS = [".jpg", ".jpeg", ".png", ".svs", ".tif", ".tiff"]

def get_classes(folder: Path) -> List[str]:

    """

    Find the classes for classification.

    Args:

        folder: Folder containing the subfolders named by class.

    Returns:

        A list of strings corresponding to the class names.

    """

    return sorted([f.name for f in folder.iterdir() if

                   ((folder.joinpath(f.name).is_dir()) and (".DS_Store" not in f.name))], key=str)

def get_log_csv_name(log_folder: Path) -> Path:

    """

    Find the name of the CSV file for logging.

    Args:

        log_folder: Folder to save logging CSV file in.

    Returns:

        The path including the filename of the logging CSV file with date information.

    """

    now = datetime.datetime.now()

    return log_folder.joinpath(f"log_{now.month}{now.day}{now.year}"

                               f"_{now.hour}{now.minute}{now.second}.csv")

def get_image_names(folder: Path) -> List[Path]:

    """

    Find the names and paths of all of the images in a folder.

    Args:

        folder: Folder containing images (assume folder only contains images).

    Returns:

        A list of the names with paths of the images in a folder.

    """

    return sorted([Path(f.name) for f in folder.iterdir() if

                   ((folder.joinpath(f.name).is_file()) and (".DS_Store" not in f.name) and (f.suffix.casefold() in IMAGE_EXTS))], key=str)

def get_image_paths(folder: Path) -> List[Path]:

    """

    Find the full paths of the images in a folder.

    Args:

        folder: Folder containing images (assume folder only contains images).

    Returns:

        A list of the full paths to the images in the folder.

    """

    return sorted([folder.joinpath(f.name) for f in folder.iterdir() if

                   ((folder.joinpath(f.name).is_file()) and (".DS_Store" not in f.name) and (f.suffix.casefold() in IMAGE_EXTS))], key=str)

def get_subfolder_paths(folder: Path) -> List[Path]:

    """

    Find the paths of subfolders.

    Args:

        folder: Folder to look for subfolders in.

    Returns:

        A list containing the paths of the subfolders.

    """

    return sorted([folder.joinpath(f.name) for f in folder.iterdir() if

                   ((folder.joinpath(f.name).is_dir()) and (".DS_Store" not in f.name))], key=str)

def get_all_image_paths(master_folder: Path) -> List[Path]:

    """

    Finds all image paths in subfolders.

    Args:

        master_folder: Root folder containing subfolders.

    Returns:

        A list of the paths to the images found in the folder.

    """

    all_paths = []

    subfolders = get_subfolder_paths(folder=master_folder)

    if len(subfolders) > 1:

        for subfolder in subfolders:

            all_paths += get_image_paths(folder=subfolder)

    else:

        all_paths = get_image_paths(folder=master_folder)

    return all_paths

def get_csv_paths(folder: Path) -> List[Path]:

    """

    Find the CSV files contained in a folder.

    Args:

        folder: Folder to search for CSV files.

    Returns:

        A list of the paths to the CSV files in the folder.

    """

    return sorted([folder.joinpath(f.name) for f in folder.iterdir() if (

                (folder.joinpath(f.name).is_file()) and ("csv" in f.name) and (".DS_Store" not in f.name))],

                  key=str)

def create_labels(csv_path: Path) -> Dict[str, str]:

    """

    Read the labels from a CSV file.

    Args:

        csv_path: Path to the CSV file.

    Returns:

        A dictionary mapping string filenames to string labels.

    """

    with csv_path.open(mode="r") as lines_open:

        lines = lines_open.readlines()[1:]

        file_to_gt_label = {}

        for line in lines:

            if len(line) > 3:

                pieces = line[:-1].split(",")

                file = pieces[0]

                gt_label = pieces[1]

                file_to_gt_label[file] = gt_label

    return file_to_gt_label