with graphics

Author: birhann

components/fiveThread.py

@@ -3,7 +3,7 @@
 from PyQt5.QtWidgets import QMainWindow, QApplication
 from PyQt5.QtCore import QUrl, QThread, pyqtSignal
 from PyQt5.QtWidgets import QApplication, QFileDialog
-
+import pyqtgraph as pg
 import numpy as np
 from time import sleep
 
@@ -21,6 +21,7 @@ def __init__(self):
         self.loadButton.clicked.connect(self.chooseFile)
         self.clearButton.clicked.connect(self.clear)
         self.solveButton.clicked.connect(self.solveSudoku)
+        self.compareButton.clicked.connect(self.compareSudoku)
 
     def chooseFile(self):
         self.filePath, self.filetype = QFileDialog.getOpenFileName(
@@ -87,6 +88,15 @@ def setInfo(self, msg):
             "main: {}\n".format(msg))
         self.logScreen.ensureCursorVisible()
 
+    def compareSudoku(self):
+        con = False
+        with open('fiveThreadLog.txt', 'r') as file:
+            con = True if file.readline() else False
+        with open('tenThreadLog.txt', 'r') as file:
+            con = True if file.readline() else False
+        if con:
+            graphObject = CompareSudokuGraph(self)
+
     def startLoad(self):
         if self.threadRadioButton5.isChecked():
             self.fiveThreadObject = FiveThreadOptions(self)
@@ -121,7 +131,7 @@ def solveSudoku(self):
         if self.threadCount == 5:
             self.fiveThreadObject.solveSudoku()
         elif self.threadCount == 10:
-            pass
+            self.tenThreadObject.solveSudoku()
         else:
             self.setInfo("Unknow Error!")
 
@@ -131,6 +141,63 @@ def setClickables(self, status):
         self.clearButton.setEnabled(status)
         self.threadRadioButton5.setEnabled(status)
         self.threadRadioButton10.setEnabled(status)
+        self.compareButton.setEnabled(status)
+
+
+class CompareSudokuGraph():
+    def __init__(self, GUI):
+        self.gui = GUI
+        self.SEC_AXIS_RANGE = 7
+        self.fiveTGraphSeconds = []
+        self.tenTGraphSeconds = []
+        self.fiveTGraphCount = []
+        self.tenTGraphCount = []
+        self.getDatafromTxt()
+        self.createGraphics()
+
+    def getDatafromTxt(self):
+        with open('fiveThreadLog.txt', 'r') as file:
+            line = file.readline()
+            if line[0] == 'g':
+                self.fiveTGraphSeconds.append(line.split("_")[1])
+                self.fiveTGraphCount.append(line.split("_")[2])
+        with open('tenThreadLog.txt', 'r') as file:
+            line = file.readline()
+            if line[0] == 'g':
+                self.tenTGraphSeconds.append(float(line.split("_")[1]))
+                self.tenTGraphCount.append(float(line.split("_")[2]))
+
+    def createGraphics(self):
+        self.createAxisLabels()
+        self.createPlotWidgets()
+
+    def createAxisLabels(self):
+        self.gui.sudokuGraph.setLabel(
+            axis='left', text='Count')
+        self.gui.sudokuGraph.setLabel(
+            axis='bottom', text='Second')
+
+    def createPlotWidgets(self):
+        self.pen = pg.mkPen('r', width=3)
+        self.pen2 = pg.mkPen('b', width=3)
+        self.sudokuGW = self.gui.sudokuGraph
+
+        # five thread
+        self.secondFT = self.fiveTGraphSeconds
+        self.cellCountFT = self.fiveTGraphCount
+        self.dataLine = self.sudokuGW.plot(
+            self.secondFT, self.cellCountFT, pen=self.pen)
+
+        # ten threads
+        self.secondTT = self.tenTGraphSeconds
+        self.cellCountTT = self.tenTGraphCount
+        self.dataLine = self.sudokuGW.plot(
+            self.secondTT, self.cellCountTT, pen=self.pen2)
+
+        if len(self.fiveTGraphSeconds) > len(self.tenTGraphSeconds):
+            self.sudokuGW.setXRange(0, len(self.fiveTGraphSeconds))
+        else:
+            self.sudokuGW.setXRange(0, len(self.tenTGraphSeconds))
 
 
 class ClearCellWorker(QThread):