import os

import pandas as pd

import numpy as np

from matplotlib import pyplot as plt

def cifar10_test_dataset(dir_dataset):

    files = os.listdir(dir_dataset)

    Y = []

    X = []

    for iFile in files:

        if 'Other' in iFile:

            y = 0

        else:

            y = iFile.split('_')[-2][-1]

    return X, Y

def classification_report_csv(report, dir_out):

    report_data = []

    lines = report.split('\n')

    for line in lines[2:-3]:

        row = {}

        row_data = line.split('      ')

        row['class'] = row_data[0]

        row['precision'] = float(row_data[1])

        row['recall'] = float(row_data[2])

        row['f1_score'] = float(row_data[3])

        row['support'] = float(row_data[4])

        report_data.append(row)

    dataframe = pd.DataFrame.from_dict(report_data)

    dataframe.to_xlsx(dir_out + 'classification_report.xlsx', index = False)

def plot_image(x, norm_intensity=False):

    # channel first

    x = np.transpose(x, (1, 2, 0))

    if norm_intensity:

        x = x / 255.0

    plt.imshow(x)

    plt.axis('off')

    plt.show()