from matplotlib import pyplot as plt
from imagen import Imagen
import cv2
import numpy as np
prueba = Imagen("rebabas.tif", blancoNegro=True)
prueba = prueba.ecualizar()
otsu1 = prueba.segmentacionOtsu()
#secciones = otsu1.secciones(ancho=2)
imgErosion = otsu1.erosion(iteraciones=10)
#seccionesO = imgErosion.secciones(ancho=2)
imgErosion2 = imgErosion.erosion(iteraciones=10)
#secciones1 = imgErosion2.secciones(ancho=2)
secciones = imgErosion2.secciones(ancho=100)
"""
plt.subplot("131"),plt.imshow(otsu1.img, cmap = 'gray')
plt.title(str(otsu1.porcentage_pixeles_blancos())+' % blancos'), plt.xticks([]), plt.yticks([])
plt.subplot("132"),plt.imshow(imgErosion.img, cmap = 'gray')
plt.title(str(imgErosion.porcentage_pixeles_blancos())+' % blancos'), plt.xticks([]), plt.yticks([])
plt.subplot("133"),plt.imshow(imgErosion2.img, cmap = 'gray')
plt.title(str(imgErosion2.porcentage_pixeles_blancos())+' % blancos'), plt.xticks([]), plt.yticks([])
plt.show()
"""
blancos = []
for seccion in secciones:
blancos.append(seccion.porcentage_pixeles_blancos())
plt.plot(blancos)
plt.title("Umbral")
plt.xlabel("Ancho de la imagen")
plt.ylabel("% Píxeles blancos")
print("[Min,max]: ["+str(min)+", "+str(max)+"] -> "+str(diferencia))
plt.text(max, 102, 'Max: '+str(max))
plt.text(min, 2, 'Min: '+str(min))
plt.text(medio, medio_y, 'Medio: '+str(medio))
plt.plot([max, min, medio], [100,0, medio_y],marker='o', color='r', ls='')
"""
plt.subplot(2, 3, 1)
#3array = [0,1,3,6,7]
array = [40, 41, 28, 29, 34]
for x in array:
data = ok_data[x]
lista = [float(i) for i in data[3:103]]
texto = data[0]
plt.plot(lista, label=texto[texto.rindex('/') + 1:])
plt.legend()
plt.title("Umbral")
count = 2
for x in array:
plt.subplot(2, 3, count)
count = count + 1
img = Imagen(ok_data[x][0], blancoNegro=True)
img = img.ecualizar()
#img = img.segmentacionOtsu()
plt.imshow(img.img, cmap='gray')
plt.title(ok_data[x][0])
plt.show()
