"""

    A utility script for data augmentation 

    Usage: assign your path string to global variable 'path'

"""

import SimpleITK as sitk

import numpy as np

import csv

from glob import glob

import pandas as pd

import os

import matplotlib

import matplotlib.pyplot as plt

import traceback

import random

from PIL import Image

path = '/home/ubuntu/data/train-3d/'

file_list=glob(path+"*true*")

for file in file_list:

    e = np.load(file)

    file_name = file.split('/')[-1]

    # several flip and rotation cases

    np.save(path+'1_'+file_name,np.transpose(e,(0,2,1)))

    np.save(path+'2_'+file_name,np.flipud(e))

    np.save(path+'3_'+file_name,np.flipud(np.transpose(e,(0,2,1))))

    np.save(path+'4_'+file_name,np.fliplr(e))

    np.save(path+'5_'+file_name,np.transpose(np.fliplr(e),(0,2,1)))

    np.save(path+'6_'+file_name,np.flipud(np.fliplr(e)))

    np.save(path+'7_'+file_name,np.flipud(np.transpose(np.fliplr(e),(0,2,1))))

    np.save(path+'8_'+file_name,np.transpose(np.rot90(e,1,(1,2)),(0,2,1)))

    np.save(path+'9_'+file_name,np.rot90(e,1,(1,2)))

    np.save(path+'10_'+file_name,np.flipud(np.transpose(np.rot90(e,1,(1,2)),(0,2,1))))

    np.save(path+'11_'+file_name,np.flipud(np.rot90(e,1,(1,2))))