import cv2 as cv
import mahotas as mht
import os
import glob
from scipy.spatial import distance
from operator import itemgetter
import numpy as np

def chi2_distance(histA, histB, eps = 1e-10):
		d = 0.5 * np.sum([((a - b) ** 2) / (a + b + eps)
			for (a, b) in zip(histA, histB)])
		return d

dir = os.getcwd()
os.chdir("query")
slika_ime=raw_input("Vnesete go imeto na slikata od query zaedno so prefiksot: ")
img=cv.imread(slika_ime, 0)
otsus_th = cv.threshold(img,0,255,cv.THRESH_BINARY+cv.THRESH_OTSU)
zernike_main = mht.features.zernike_moments(otsus_th[1],1200)

zernike = []
os.chdir(dir)
os.chdir("database")
for image in glob.glob("*.jpg"):
    img=cv.imread(image, 0)
    otsus_th = cv.threshold(img,0,255,cv.THRESH_BINARY+cv.THRESH_OTSU)
    pom1 = mht.features.zernike_moments(otsus_th[1],1200)
    pom2 = chi2_distance(pom1, zernike_main)
    item = (pom2, image)
    zernike.append(item)

lista = sorted(zernike, key=itemgetter(0))

for item1 in lista:
    print(item1[1])