#coding:utf-8

import cv2
from splitter import ImageSplitter as IS
from tegaki_nn import TegakiNN
from data_extention import DataExtention as DE
import os

path = './images/test.png'
patterns = 500
epoch = 8
splitter = IS(path)
extention = DE()
images = splitter.split()
nn = TegakiNN(epoch,patterns)
results = []
n=0

for image in images:
    n+=1
    r,src = cv2.threshold(image,0,255,cv2.THRESH_BINARY)
    cv2.imshow('',src)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    cv2.imwrite('./images/{}.png'.format(n),src)
    ret = nn.predict(src)
    if ret == 1:
        results.append('o')
    else:
        results.append('x')

for i in range(len(results)):
    print('Image {} - [{}]'.format(i+1,results[i]))

test_files = os.listdir('./images/test/')
dirs = os.listdir('./images/test/'+test_files[1])
f = dirs[0]
src = cv2.imread('./images/test/'+test_files[1]+'/'+f,1)
src = cv2.cvtColor(src,cv2.COLOR_BGR2GRAY)
src = cv2.bitwise_not(src)
r,src = cv2.threshold(src,0,255,cv2.THRESH_BINARY)

ret = nn.predict(src)