import os
import shutil
from libtiff import TIFF    # pip install libtiff
from scipy import misc
import random


def tif2png(_src_path, _dst_path):
    """
    Usage:
        formatting `tif/tiff` files to `jpg/png` files
    :param _src_path:
    :param _dst_path:
    :return:
    """
    tif = TIFF.open(_src_path, mode='r')
    image = tif.read_image()
    misc.imsave(_dst_path, image)


def data_split(src_list):
    """
    Usage:
        randomly spliting dataset
    :param src_list:
    :return:
    """
    counter_list = random.sample(range(0, len(src_list)), 550)

    return counter_list


if __name__ == '__main__':
    src_dir = '../Dataset/train_dataset/CVC-EndoSceneStill/CVC-612/test_split/masks_tif'
    dst_dir = '../Dataset/train_dataset/CVC-EndoSceneStill/CVC-612/test_split/masks'

    os.makedirs(dst_dir, exist_ok=True)
    for img_name in os.listdir(src_dir):
        tif2png(os.path.join(src_dir, img_name),
                os.path.join(dst_dir, img_name.replace('.tif', '.png')))