#!/usr/bin/python
# -*- coding: utf-8 -*-
#
"""
  Nome: File.py
  Autor: Alessandro dos Santos Ferreira ( santosferreira.alessandro@gmail.com )

  Descricão: TODO
"""

import cv2
#import numpy as np
import os
from skimage.util import img_as_float
from scipy import ndimage

class File(object):

    @staticmethod
    def get_filename(filepath):
        path, name = os.path.split(filepath)
        
        return name
    
    @staticmethod
    def get_path():
        path, name = os.path.split(filepath)
        
        return path    
        
    @staticmethod
    def open_image(filepath):
        image = cv2.imread(filepath)
            
        #image = np.flipud(image)
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        
        #return img_as_float(image)
        return image

    @staticmethod
    def save_image(image, dataset, directory, filename, idx, ext = '.tif'):
        
        for root, dirs, files in os.walk(dataset):
            for d in dirs:
                filepath = dataset + '/' + d + '/' + filename + '_%05d' % idx + ext
                if(os.path.isfile(filepath)):
                    os.remove(filepath)

        dir_path = dataset + '/' + directory
        filepath = dir_path + '/' + filename + '_%05d' % idx + ext
        if not os.path.exists(dir_path):
            os.makedirs(dir_path)
        cv2.imwrite(filepath, cv2.cvtColor(image, cv2.COLOR_RGB2BGR))

        return filepath

    @staticmethod
    def list_dirs(dataset):
        return [name for name in os.listdir(dataset) 
                    if os.path.isdir(os.path.join(dataset, name))]