#!/usr/bin/env python
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
from matplotlib import rc
import numpy as np
from skimage import img_as_float, color
from datetime import datetime
from skimage import filters, io, morphology, feature, data
from scipy import ndimage
from skimage import measure

airplaneFile = "F:\\Studia\\PUT\\V\\Komunikacja człowiek-komputer\\Zadanie 4- Samoloty\\Samolot\\samolot"
savePath = "C:\\Users\\Szczepan\\Desktop\\" + 'Samolociki' + datetime.now().strftime('%Y%m%d%H%M%S') +'.pdf'
fileExtension = ".jpg"
numOfImages = 20


def loadFiles(numOfImage):
    img = [[]] * numOfImage
    for i in range(0, numOfImage):
        if i < 10:
            img[i] = io.imread(airplaneFile + "0" + str(i) + fileExtension, as_grey=False)
        else:
            img[i] = io.imread(airplaneFile + str(i) + fileExtension, as_grey=False)
    return img

def plot(numOfImage):
    size = 400
    fig, axes = plt.subplots(nrows= numOfImage,  sharex=True, sharey=True, figsize=(1.61* size, size))
    fig.subplots_adjust(top=1.00, bottom=0.05, left=0.25, right=0.95)
    rc('legend', fontsize=10)
    return fig, axes

def main():
    grayscale_image = img_as_float(data.camera()[::2, ::2])
    img = color.gray2rgb(grayscale_image) #Wczytuję pliki
    fig, axes = plot(numOfImages)
    for i in range (0, numOfImages):

        im = axes[i].imshow(img)
        im.set_extent([0, 1, 0, 1])

        axes[i].yaxis.set_visible(False)

        pos = list(axes[i].get_position().bounds)
        x_text = pos[0] - 0.25
        y_text = pos[1] + pos[3]/2.
        fig.text(x_text, y_text, i, va='center', ha='left', fontsize=10)


    fig.savefig(savePath)


if _name_ == '_main_':
    main()