ComputerVisionBoggleSolver/MainProgram.py at master · Jordan-Jarvis/ComputerVisionBoggleSolver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import numpy as np
import cv2
import operator
#from matplotlib import pyplot as plt
import joblib
from dlxsudoku import Sudoku
import complexWorker as cw
import multiprocessing as mp
import os
import ThreadWorker as tw
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

import tensorflow as tf
import tensorflow.keras as keras
def main():

	gameType = int(input("For boggle solver, type 0, for Sudoku solver type 1: "))

	number_of_processes = 7
	if (gameType == 0):
		number_of_processes = 1 # temporary fix for boggle solver
	InputQueue = mp.Queue()
	FinalImages = mp.Queue()
	ProcessedDigits= mp.Queue()
	processes = []
	temp = 1
	for i in range(number_of_processes):
		if (gameType == 1):
			sudokuSolve = mp.Process(target=tw.AttemptSudokuSolve, args=(InputQueue,ProcessedDigits,FinalImages))
			processes.append(sudokuSolve)
			sudokuSolve.start()
			boardType = 'sudoku'
		else:
			boggleSolve = mp.Process(target=tw.AttemptBoggleSolve, args=(InputQueue,ProcessedDigits,FinalImages))
			processes.append(boggleSolve)
			boggleSolve.start()
			boardType = 'boggle'

	p = mp.Process(target=tw.ShowFinalImage, args=(ProcessedDigits, FinalImages))
	p.start()


	boardFound = 1
	cap = cv2.VideoCapture(0)
	ret, frame = cap.read()
	centerX = frame.shape[1]/2
	centerY = frame.shape[0]/2
	x1 = int(centerX - 170)
	y1 = int(centerY - 170)
	x2 = int(centerX + 170)
	y2 = int(centerY + 170)
	startPoint = (x1,y1)
	endPoint = (x2,y2)
	font = cv2.FONT_HERSHEY_SIMPLEX
	org = (50, 50)
	fontScale = 1
	color = (255, 0, 0)
	thickness = 2
	result = np.zeros(81)
	img = frame
	cropped = frame
	result_old = np.zeros(81)
	result_really_old = result_old
	while(True):
		ret, frame = cap.read()
		if not np.sum(frame) == 0:
			break
	while(True):
		ret, frame = cap.read()
		gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
		frame = cv2.rectangle(frame, startPoint, endPoint, (255,0,0), 10)
		frame = cv2.putText(frame, 'Align ' + boardType + ' board with square', org, font,
		fontScale, color, thickness, cv2.LINE_AA)
		gray = gray[y1:y2, x1:x2]
		cv2.imshow('Original',frame)
		if cv2.waitKey(1) & 0xFF == ord('q'):
			break
		cv2.imshow('Original2',gray)
		if cv2.waitKey(1) & 0xFF == ord('q'):
			break

		if (InputQueue.qsize() < 10):
			InputQueue.put(gray)
	p.terminate()
	for pr in processes:
		pr.terminate()

	cap.release()
	cv2.destroyAllWindows()

if __name__ == '__main__':
	main()