# Declare global variables

global panelA,panelB,img,im_canny,topframe,path,ballot_path,prg_bar

# prg_bar.start(50)

if len(path)>0 and len(ballot_path)>0:

count = len([file for file in os.listdir(ballot_path) if file.endswith('.jpg')])

for file in os.listdir(ballot_path):

if file.endswith('.jpg'):

loc_path = ballot_path+'/'+file

print('processing image: '+str(file))

if len(loc_path)>0:

ballot_paper = cv2.imread(loc_path)

# P1 = cv2.imread('img/Party_1.jpg')

img_main = ballot_paper.copy()

# -------------------<Party Voting>--------------------------------------------------------------------------

img_main = segmentation.segmentPartyVoting(img_main)

img = img_main

# Display the cropped image

# img_main = cv2.resize(img_main, (0, 0), fx=0.25, fy=0.25)

# cv2.imshow('Main', img_main)

# Display the edged image

# cannyout = cv2.resize(cannyout,(0,0),fx=0.25,fy=0.25)

# cv2.imshow('Canny',cannyout)

# ------------------</Party Voting>-------------------------------------------------------------------------

# -------------------<Preference Voting>--------------------------------------------------------------------

img_main = ballot_paper.copy()

pref_im = segmentation.segmentPreferenceVoting(img_main)

pref_gray = cv2.cvtColor(pref_im, cv2.COLOR_BGR2GRAY)

pref_gray_erode = cv2.erode(pref_gray.copy(), np.ones((3, 3), np.uint8), iterations=1)

pref_gray_erode = cv2.blur(pref_gray_erode, (3, 3), 0)

# pref_gray_erode = pref_gray

pref_cannyout = cv2.Canny(pref_gray_erode, 100, 200, 5)

pref_im2, pref_contours, pref_hierarchy = cv2.findContours(pref_cannyout.copy(), cv2.RETR_TREE,

cv2.CHAIN_APPROX_SIMPLE)

# Select the contours with convexity and four points

pref_contours = [contour for contour in pref_contours if

cv2.contourArea(contour) > 20 and cv2.contourArea(contour) < 10000 and len(

cv2.approxPolyDP(contour, 0.02 * cv2.arcLength(contour, True), True)) == 4]

pref_contours = sorted(pref_contours, key=lambda x: (cv2.contourArea(x)), reverse=True)

# print(len(pref_contours))

# cv2.drawContours(pref_im, pref_contours, -1, (0, 0, 255), 3)

rect = []

centroids = []

for cnt in pref_contours:

x, y, w, h = cv2.boundingRect(cnt)

cv2.rectangle(pref_im, (x, y), (x + w, y + h), (0, 255, 0), 3)

rect.append([x, y, w, h])

cx = x + w / 2

cy = y + h / 2

centroids.append([cx, cy])

temp = []

for cx, cy in centroids:

if [cx, cy] not in temp:

temp.append([cx, cy])

centroids = temp

centroids = sorted(centroids, key=lambda x: (x[1], x[0]))

# print(len(centroids))

# print(centroids)

height, width = pref_im.shape[:2]

blocks = [None] * 40

for i in range(0, 4):

for j in range(0, 10):

for cx, cy in centroids:

index = i * 10 + j

if cx < (j + 1) * width / 10 and cx > j * width / 10 and cy < (

i + 1) * height / 4 and cy > i * height / 4:

blocks[i * 10 + j] = [cx, cy]

voted = [i + 1 for i, val in enumerate(blocks) if val is None]

print(voted)

# cv2.imshow('im', pref_im)

# -------------------</Preference Voting>--------------------------------------------------------------------

for key,val in arr.iteritems():

val[1] = 0

for k,v in val[0].iteritems():

global arr

prg_bar.stop()

if not os.path.exists('output'):

os.makedirs('output')

with open('output/final_count.txt','w') as file:

for key,val in arr.iteritems():

file.write(str(key)+' - votes:'+str(val[1])+'\n')

for k,v in val[0].iteritems():

file.write('\t'+str(k)+': '+str(v)+'\n')

clear_arr()

prg_bar.start()

# select_image()

t = threading.Thread(target=select_image, name='select_image')

global panelA,panelB,img

# Open directory

p = selct_signs()

t2.delete(0, END)

t2.insert(0, p)

# Copy the main image

global path

path = tkFileDialog.askdirectory()

if len(path) > 0:

for file in os.listdir(path):

if file.endswith('.jpg'):

filepath = path + '/' + file

if len(filepath) > 0:

# template_ori = cv2.imread(filepath, 0)

arr[str(file)] = [dict(),0]

global ballot_path

ballot_path = tkFileDialog.askdirectory()

t1.delete(0, END)

t1.insert(0, ballot_path)