class GUIBasic(QWidget):
def __init__(self):
super().__init__()
self.model = None
def init_ui(self):
self.resize(1200, 680)
self.pic_list = ['maze7_1', 'maze7_2', 'maze7_3', 'maze10_1', 'maze10_2', 'maze10_3', 'maze11_1']
self.timer = QBasicTimer()
widget1 = QWidget(parent=self)
widget1.setGeometry(QRect(30, 50, 800, 500))
table_area = QGroupBox(parent=widget1) # 图形显示区域
table_area.setGeometry(QRect(widget1.x(), widget1.y(), widget1.width(), widget1.height()))
self.Plot = DrawUI(width=3, height=3, dpi=100)
gridlayout1 = QGridLayout(table_area) # 继承容器groupBox
gridlayout1.addWidget(self.Plot, 0, 1)
pic_choose_label = QLabel(self)
pic_choose_label.move(table_area.x() + table_area.width(), table_area.y() + 80)
pic_choose_label.setText("Choose Maze：")
pic_choose_label.setFont(QFont("Fixed", 10))
self.pic_choose_combo = QComboBox(self)
self.pic_choose_combo.move(pic_choose_label.geometry().x() + pic_choose_label.geometry().width() + 30,
pic_choose_label.geometry().y() - 8)
self.pic_choose_combo.resize(100, self.pic_choose_combo.geometry().height())
self.pic_choose_combo.addItems(self.pic_list)
self.pic_choose_combo.currentIndexChanged.connect(self.pic_change)
self.pic_choose_combo.setFont(QFont("Fixed", 10))
self.pic_change()
middle_x = (
pic_choose_label.geometry().x() + self.pic_choose_combo.geometry().x() + self.pic_choose_combo.geometry().width()) / 2
self.playing_index = -1
self.problem_solving = False
self.solve_problem_button = QPushButton(parent=self)
self.solve_problem_button.setText("Train Now")
self.solve_problem_button.move(pic_choose_label.geometry().x() + pic_choose_label.geometry().width() + 30,
self.pic_choose_combo.y() + self.pic_choose_combo.height() + 100)
self.solve_problem_button.pressed.connect(self.solve_button_pressed)
self.solve_problem_button.setFont(QFont("Fixed", 10))
self.solve_test = QLabel(parent=self) # 解答过程中的信息显示
self.solve_test.setText("Training...")
self.solve_test.resize(400, self.solve_test.height())
self.solve_test.setFont(QFont("Fixed", 9))
self.solve_test.move(table_area.x() + table_area.width() - 20,
self.solve_problem_button.geometry().y() + self.solve_problem_button.geometry().height() + 20)
self.solve_test.setHidden(True)
speed_choose_label = QLabel(self)
speed_choose_label.move(table_area.x() + table_area.width(), self.solve_test.geometry().y() + 40)
speed_choose_label.setText("Play Speed：")
speed_choose_label.setFont(QFont("Fixed", 10))
self.play_speed_combo = QComboBox(self)
self.play_speed_combo.move(speed_choose_label.geometry().x() + speed_choose_label.geometry().width() + 30,
speed_choose_label.geometry().y() - 2)
self.play_speed_combo.addItems(["High", "Middle", "Low"])
self.play_speed_combo.setFont(QFont("Fixed", 10))
play_button = QPushButton(self)
play_button.setText("Play Result Now!")
play_button.move(pic_choose_label.geometry().x() + pic_choose_label.geometry().width() + 30,
self.play_speed_combo.geometry().y() + self.play_speed_combo.geometry().height() + 40)
play_button.pressed.connect(self.play_button_pressed)
play_button.setFont(QFont("Fixed", 10))
def pic_change(self):
self.timer.stop()
current_text = self.pic_choose_combo.currentText()
maze = Mazes[current_text]
my_maze = Maze(maze_map=np.array(maze), period=2)
self.model = QTableModel(my_maze)
# try:
# self.model.load_table('./Saved_QTable/' + current_text + '.npy')
# except:
# QMessageBox.information(self, "提示", "没找到Q表保存文件", QMessageBox.Ok | QMessageBox.Close,
# QMessageBox.Close)
self.model.play_game((0, 0), 0)
self.Plot.draw_root(self.model.my_maze, (0, 0), 1, 0, False)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period - 1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
def play_button_pressed(self):
if self.model == None:
QMessageBox.information(self, "Tip", "Please choose the Maze first.", QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.model.play_game((0, 0), 0)
speed_text = self.play_speed_combo.currentText()
self.playing_index = 0
if speed_text == "High":
self.timer.start(100, self)
elif speed_text == "Middle":
self.timer.start(500, self)
else:
self.timer.start(1500, self)
def timerEvent(self, event):
if event.timerId() == self.timer.timerId():
period = self.model.my_maze.period
if period != 0 and (self.playing_index % period) >= period / 2:
fire_flag = True
else:
fire_flag = False
self.Plot.draw_qtable(self.model, self.playing_index % period if period != 0 else 0, fire_flag)
self.Plot.draw_root(self.model.my_maze, (0, 0), self.playing_index, period, fire_flag)
self.playing_index = self.playing_index + 1
if self.playing_index >= len(self.model.my_maze.visited) + 2:
self.playing_index = 0
# print("up",self.playing_index)
else:
super(GUIBasic, self).timerEvent(event)
def solve_button_pressed(self):
if self.problem_solving:
return
if type(self.model) == type(None):
QMessageBox.information(self, "Tip", "Please choose the Maze first.", QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.problem_solving = True
self.playing_index = -1
self.solve_test.setHidden(False)
self.timer.stop()
self.repaint()
start_time = time.time()
# path = "tangrams\\" + self.parent().pic_choose_combo.currentText() + ".png"
self.model.train(output_line=self.solve_test, main_ui=self)
end_time = time.time()
QMessageBox.information(self, "Tip", "Training finished，spend：%.3f s" % (end_time - start_time),
QMessageBox.Ok | QMessageBox.Close, QMessageBox.Close)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period - 1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
self.problem_solving = False
self.solve_test.setHidden(True)
class Ui_basic(QWidget):
def __init__(self):
super().__init__()
def initUI(self):
self.resize(1200, 700)
self.pic_list = ["tangram%d" % i for i in range(1, 20 + 1)]
self.pic_area = QLabel(parent=self) #图形显示区域
self.pic_area.resize(800, 600)
self.pic_area.move(30, 50)
self.current_img = cv2.imread("tangrams\\tangram1.png",
cv2.COLOR_BGRA2BGR)
self.showImage(self.pic_area, self.current_img)
pic_choose_label = QLabel(self)
pic_choose_label.move(
self.pic_area.geometry().x() + self.pic_area.geometry().width() +
30,
self.pic_area.geometry().y() + 20)
pic_choose_label.setText("选择图片：")
self.pic_choose_combo = QComboBox(self)
self.pic_choose_combo.move(
pic_choose_label.geometry().x() +
pic_choose_label.geometry().width() + 30,
pic_choose_label.geometry().y())
self.pic_choose_combo.resize(150,
self.pic_choose_combo.geometry().height())
self.pic_choose_combo.addItems(self.pic_list)
self.pic_choose_combo.currentIndexChanged.connect(self.pic_change)
middle_x = (pic_choose_label.geometry().x() +
self.pic_choose_combo.geometry().x() +
self.pic_choose_combo.geometry().width()) / 2
self.solver_class = Solve_frame(
self,
size=((middle_x - pic_choose_label.geometry().x()) * 2, 600),
pic_area=self.pic_area)
self.solver_class.move(
pic_choose_label.geometry().x(),
pic_choose_label.geometry().y() +
pic_choose_label.geometry().height() + 40)
self.solver_class.solve_problem_button.pressed.connect(
lambda: self.solver_class.solve_button_pressed(
path="tangrams\\" + self.pic_choose_combo.currentText(
) + ".png"))
def pic_change(self):
self.solver_class.solver = None
self.solver_class.playing_index = -1
self.solver_class.timer.stop()
current_text = self.pic_choose_combo.currentText()
self.current_img = cv2.imread("tangrams\\" + current_text + ".png",
cv2.COLOR_BGRA2BGR)
self.showImage(self.pic_area, self.current_img)
#print(current_text)
def showImage(self, qlabel, img):
size = (int(qlabel.width()), int(qlabel.height()))
shrink = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
# cv2.imshow('img', shrink)
shrink = cv2.cvtColor(shrink, cv2.COLOR_BGR2RGB)
self.QtImg = QtGui.QImage(shrink.data, shrink.shape[1],
shrink.shape[0], shrink.shape[1] * 3,
QtGui.QImage.Format_RGB888)
qlabel.setPixmap(QtGui.QPixmap.fromImage(self.QtImg))
class Ui_userDefine(QWidget):
def __init__(self):
super().__init__()
self.model = None
self.playing_index = -1
self.problem_solving = False
def initUI(self):
self.resize(1200, 700)
self.timer = QBasicTimer()
widget1 = QWidget(parent=self)
widget1.setGeometry(QRect(30, 50, 800, 500))
table_area = QGroupBox(parent=widget1) #图形显示区域
table_area.setGeometry(
QRect(widget1.x(), widget1.y(), widget1.width(), widget1.height()))
self.Plot = Draw_ui(width=3, height=3, dpi=100)
gridlayout1 = QGridLayout(table_area) # 继承容器groupBox
gridlayout1.addWidget(self.Plot, 0, 1)
pic_choose_label = QLabel(self)
pic_choose_label.move(table_area.x() + table_area.width() + 30,
table_area.y() + 20)
pic_choose_label.setText("输入迷宫：")
pic_choose_label2 = QLabel(self)
pic_choose_label2.move(
pic_choose_label.geometry().x(),
pic_choose_label.y() + pic_choose_label.height() + 20)
pic_choose_label2.setText("（1表示空格，0表示墙, 2表示陷阱, 3表示火）")
self.maze_input = QPlainTextEdit(parent=self)
self.maze_input.setGeometry(
QRect(pic_choose_label2.x(),
pic_choose_label2.y() + pic_choose_label2.height() + 20, 300,
200))
self.maze_input.setPlainText(
'1, 0, 1, 1, 1, 1, 1,\r\n1, 1, 3, 0, 0, 1, 0,\r\n0, 0, 0, 1, 1, 1, 0,\r\n1, 1, 1, 1, 0, 0, 1,\r\n1, 0, 0, 0, 1, 1, 1,\r\n1, 0, 1, 1, 1, 2, 2,\r\n1, 1, 1, 0, 1, 1, 1,'
)
period_label = QLabel(parent=self)
period_label.setText('火焰周期:')
period_label.move(self.maze_input.x(),
self.maze_input.height() + self.maze_input.y() + 10)
self.period_input = QSpinBox(parent=self)
self.period_input.setValue(1)
self.period_input.move(period_label.x() + period_label.width() + 15,
period_label.y())
period_label2 = QLabel(parent=self)
period_label2.setText('*2')
period_label2.move(
self.period_input.x() + self.period_input.width() - 40,
self.period_input.y())
maze_input_button = QPushButton(parent=self)
maze_input_button.move(
self.period_input.x() - 50,
self.period_input.y() + self.period_input.height() + 10)
maze_input_button.setText('确认输入')
maze_input_button.pressed.connect(self.pic_change)
middle_x = self.maze_input.geometry().x(
) + self.maze_input.geometry().width() / 2
train_epoch_label = QLabel(parent=self)
train_epoch_label.setText('训练次数上限:')
train_epoch_label.move(
self.maze_input.x(),
maze_input_button.height() + maze_input_button.y() + 40)
self.epoch_input = QSpinBox(parent=self)
self.epoch_input.move(
train_epoch_label.x() + train_epoch_label.width() + 15,
train_epoch_label.y())
self.epoch_input.setValue(30)
train_epoch_label2 = QLabel(parent=self)
train_epoch_label2.setText('*1000')
train_epoch_label2.move(
self.epoch_input.x() + self.epoch_input.width() - 40,
self.epoch_input.y())
self.solve_problem_button = QPushButton(parent=self)
self.solve_problem_button.setText("训练")
self.solve_problem_button.move(
maze_input_button.x(),
train_epoch_label.y() + train_epoch_label.height() + 10)
self.solve_problem_button.pressed.connect(self.solve_button_pressed)
self.solve_test = QLabel(parent=self) # 解答过程中的信息显示
self.solve_test.setText("正在训练。。。")
self.solve_test.resize(250, self.solve_test.height())
self.solve_test.move(
middle_x - self.solve_test.geometry().width() / 2,
self.solve_problem_button.y() +
self.solve_problem_button.height() + 10)
self.solve_test.setHidden(True)
speed_choose_label = QLabel(self)
speed_choose_label.move(
train_epoch_label.x(),
self.solve_test.y() + self.solve_test.height() + 10)
speed_choose_label.setText("播放速度：")
self.play_speed_combo = QComboBox(self)
self.play_speed_combo.move(
speed_choose_label.geometry().x() +
speed_choose_label.geometry().width() + 30,
speed_choose_label.geometry().y())
self.play_speed_combo.addItems(["高速", "中速", "慢速"])
play_button = QPushButton(self)
play_button.setText("播放走迷宫过程")
play_button.move(
speed_choose_label.x() + 40,
self.play_speed_combo.geometry().y() +
self.play_speed_combo.geometry().height() + 10)
play_button.pressed.connect(self.play_button_pressed)
def pic_change(self):
self.timer.stop()
current_text = self.maze_input.toPlainText()
rows = current_text.split('\n')
maze_map = []
try:
for row in rows:
row_sp = row.split(',')
row_list = []
for c in row_sp:
c = c.strip()
if len(c) == 0:
continue
else:
row_list.append(int(c))
maze_map.append(row_list)
except:
QMessageBox.information(self, "提示", "无法解析输入",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
maze_len = len(maze_map[0])
for i in range(1, len(maze_map)):
if len(maze_map[i]) != maze_len:
QMessageBox.information(self, "提示", "错误，每一行的列数应当相等",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
my_maze = Maze(maze_map=np.array(maze_map),
period=self.period_input.value() * 2)
self.model = QTableModel(my_maze)
#self.model.play_game((0, 0), 0)
self.Plot.draw_root(self.model.my_maze, (0, 0), 1, 0, False)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period -
1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
def play_button_pressed(self):
if self.model == None:
QMessageBox.information(self, "提示", "请先输入迷宫",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.model.play_game((0, 0), 0)
speed_text = self.play_speed_combo.currentText()
self.playing_index = 0
if speed_text == "高速":
self.timer.start(150, self)
elif speed_text == "中速":
self.timer.start(500, self)
else:
self.timer.start(1500, self)
def timerEvent(self, event):
if event.timerId() == self.timer.timerId():
period = self.model.my_maze.period
if period != 0 and (self.playing_index % period) >= period / 2:
fire_flag = True
else:
fire_flag = False
self.Plot.draw_qtable(
self.model, self.playing_index % period if period != 0 else 0,
fire_flag)
self.Plot.draw_root(self.model.my_maze, (0, 0), self.playing_index,
period, fire_flag)
self.playing_index = self.playing_index + 1
if self.playing_index >= len(self.model.my_maze.visited) + 2:
self.playing_index = 0
# print("up",self.playing_index)
else:
super(Ui_userDefine, self).timerEvent(event)
def solve_button_pressed(self):
if self.problem_solving:
return
if type(self.model) == type(None):
QMessageBox.information(self, "提示", "请先输入迷宫",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.problem_solving = True
self.playing_index = -1
self.solve_test.setHidden(False)
self.timer.stop()
self.repaint()
train_epoch = self.epoch_input.value() * 1000
start_time = time.time()
#path = "tangrams\\" + self.parent().pic_choose_combo.currentText() + ".png"
self.model.train(output_line=self.solve_test,
main_ui=self,
epoch_N=train_epoch)
end_time = time.time()
QMessageBox.information(self, "提示",
"完成训练，用时：%.3f s" % (end_time - start_time),
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period -
1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
self.problem_solving = False
self.solve_test.setHidden(True)
class Solve_frame(QWidget):
def __init__(self, parent, size, pic_area):
super().__init__()
self.playing_index = -1
self.solver = None
self.pic_area = pic_area
self.img_path = "tangrams\\tangram1.png"
self.problem_solving = False
self.setParent(parent)
self.resize(*size)
self.solve_problem_button = QPushButton(parent=self)
self.solve_problem_button.setText("解答")
middle_x = self.geometry().width() / 2
self.solve_problem_button.move(
middle_x - self.solve_problem_button.geometry().width() / 2, 0)
#self.solve_problem_button.pressed.connect(self.solve_button_pressed)
self.solve_test = QLabel(parent=self) #解答过程中的信息显示
self.solve_test.setText("正在解答。。。")
self.solve_test.resize(250, self.solve_test.height())
self.solve_test.move(
middle_x - self.solve_test.geometry().width() / 2,
self.solve_problem_button.geometry().y() +
self.solve_problem_button.geometry().height() + 20)
self.solve_test.setHidden(True)
speed_choose_label = QLabel(self)
speed_choose_label.move(0, self.solve_test.geometry().y() + 40)
speed_choose_label.setText("播放速度：")
self.play_speed_combo = QComboBox(self)
self.play_speed_combo.move(
speed_choose_label.geometry().x() +
speed_choose_label.geometry().width() + 30,
speed_choose_label.geometry().y())
self.play_speed_combo.addItems(["高速", "中速", "慢速"])
play_button = QPushButton(self)
play_button.setText("播放解答过程")
play_button.move(
middle_x - play_button.geometry().width() / 2,
self.play_speed_combo.geometry().y() +
self.play_speed_combo.geometry().height() + 40)
play_button.pressed.connect(self.play_button_pressed)
self.timer = QBasicTimer()
def solve_button_pressed(self, path, pieces_num=[2, 1, 1, 1, 2]):
if self.problem_solving:
return
if type(path) != str and type(path) != numpy.ndarray:
QMessageBox.information(self, "警告", "还未形成图像！", QMessageBox.Ok)
return
self.problem_solving = True
self.playing_index = -1
self.solve_test.setHidden(False)
self.timer.stop()
self.repaint()
start_time = time.time()
#path = "tangrams\\" + self.parent().pic_choose_combo.currentText() + ".png"
solver = Solver(path,
pieces_num=pieces_num,
output_line=self.solve_test)
solver.solve_dfs2()
end_time = time.time()
QMessageBox.information(
self, "提示",
"完成解答，用时：%.3f s，经%d步" % (end_time - start_time, len(solver.trace)),
QMessageBox.Ok | QMessageBox.Close, QMessageBox.Close)
self.showImage(self.pic_area, img=solver.graph.current_graph)
self.solver = solver
self.problem_solving = False
self.solve_test.setHidden(True)
def showImage(self, qlabel, img):
size = (int(qlabel.width()), int(qlabel.height()))
shrink = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
# cv2.imshow('img', shrink)
shrink = cv2.cvtColor(shrink, cv2.COLOR_BGR2RGB)
self.QtImg = QtGui.QImage(shrink.data, shrink.shape[1],
shrink.shape[0], shrink.shape[1] * 3,
QtGui.QImage.Format_RGB888)
qlabel.setPixmap(QtGui.QPixmap.fromImage(self.QtImg))
def play_button_pressed(self):
if self.solver == None:
QMessageBox.information(self, "提示", "未完成解答，请先按解答按钮并等待",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
speed_text = self.play_speed_combo.currentText()
self.playing_index = 0
if speed_text == "高速":
self.timer.start(300, self)
elif speed_text == "中速":
self.timer.start(800, self)
else:
self.timer.start(1500, self)
def timerEvent(self, event):
if event.timerId() == self.timer.timerId():
if self.playing_index < len(
self.solver.trace) and self.solver != None:
node_num = self.solver.trace[self.playing_index]
node = self.solver.open_all_list[node_num]
self.playing_index += 1
graph = copy.deepcopy(self.solver.graph.init_color_img)
self.solver.draw_current_graph(graph=graph, node=node)
self.showImage(self.pic_area, graph)
# print("up",self.playing_index)
else:
super(Solve_frame, self).timerEvent(event)
class UIUserDefine(QWidget):
def __init__(self):
super().__init__()
self.model = None
self.playing_index = -1
self.problem_solving = False
def init_ui(self):
self.resize(1200, 680)
self.timer = QBasicTimer()
widget1 = QWidget(parent=self)
widget1.setGeometry(QRect(30, 50, 800, 500))
table_area = QGroupBox(parent=widget1) # 图形显示区域
table_area.setGeometry(
QRect(widget1.x(), widget1.y(), widget1.width(), widget1.height()))
self.Plot = DrawUI(width=3, height=3, dpi=100)
gridlayout1 = QGridLayout(table_area) # 继承容器groupBox
gridlayout1.addWidget(self.Plot, 0, 1)
pic_choose_label = QLabel(self)
pic_choose_label.move(table_area.x() + table_area.width() + 30,
table_area.y() + 20)
pic_choose_label.setText("Input Maze：")
pic_choose_label.setFont(QFont("Fixed", 10))
pic_choose_label2 = QLabel(self)
pic_choose_label2.move(
pic_choose_label.geometry().x(),
pic_choose_label.y() + pic_choose_label.height() + 20)
pic_choose_label2.setText(
"（1 is space，0 is wall, 2 is grass, 3 is light）")
pic_choose_label2.setFont(QFont("Fixed", 10))
self.maze_input = QPlainTextEdit(parent=self)
self.maze_input.setGeometry(
QRect(pic_choose_label2.x(),
pic_choose_label2.y() + pic_choose_label2.height() + 20, 250,
150))
self.maze_input.setPlainText(
'1, 0, 1, 1, 1, 1, 1,\r\n1, 1, 3, 0, 0, 1, 0,\r\n0, 0, 0, 1, 1, 1, 0,\r\n1, 1, 1, 1, 0, 0, 1,\r\n1, 0, 0, 0, 1, 1, 1,\r\n1, 0, 1, 1, 1, 2, 2,\r\n1, 1, 1, 0, 1, 1, 1,'
)
period_label = QLabel(parent=self)
period_label.setText('Light Period:')
period_label.setFont(QFont("Fixed", 10))
period_label.move(self.maze_input.x(),
self.maze_input.height() + self.maze_input.y() + 10)
self.period_input = QSpinBox(parent=self)
self.period_input.setValue(1)
self.period_input.setFont(QFont("Fixed", 10))
self.period_input.move(period_label.x() + period_label.width() + 15,
period_label.y() - 2)
period_label2 = QLabel(parent=self)
period_label2.setText('*2')
period_label2.setFont(QFont("Fixed", 10))
period_label2.move(
self.period_input.x() + self.period_input.width() - 40,
self.period_input.y() + 2)
maze_input_button = QPushButton(parent=self)
maze_input_button.move(
period_label.x() + period_label.width() + 15,
self.period_input.y() + self.period_input.height() + 10)
maze_input_button.setText('Done')
maze_input_button.setFont(QFont("Fixed", 10))
maze_input_button.pressed.connect(self.pic_change)
middle_x = self.maze_input.geometry().x(
) + self.maze_input.geometry().width() / 2
train_epoch_label = QLabel(parent=self)
train_epoch_label.setText('Training Count:')
train_epoch_label.setFont(QFont("Fixed", 10))
train_epoch_label.move(
self.maze_input.x(),
maze_input_button.height() + maze_input_button.y() + 40)
self.epoch_input = QSpinBox(parent=self)
self.epoch_input.move(
train_epoch_label.x() + train_epoch_label.width() + 15,
train_epoch_label.y() - 2)
self.epoch_input.setValue(30)
self.epoch_input.setFont(QFont("Fixed", 10))
train_epoch_label2 = QLabel(parent=self)
train_epoch_label2.setText('*1000')
train_epoch_label2.setFont(QFont("Fixed", 10))
train_epoch_label2.move(
self.epoch_input.x() + self.epoch_input.width() - 40,
self.epoch_input.y() + 2)
self.solve_problem_button = QPushButton(parent=self)
self.solve_problem_button.setText("Train Now")
self.solve_problem_button.setFont(QFont("Fixed", 10))
self.solve_problem_button.move(
period_label.x() + period_label.width() + 15,
train_epoch_label.y() + train_epoch_label.height() + 10)
self.solve_problem_button.pressed.connect(self.solve_button_pressed)
self.solve_test = QLabel(parent=self) # 解答过程中的信息显示
self.solve_test.setText("Training...")
self.solve_test.setFont(QFont("Fixed", 10))
self.solve_test.resize(250, self.solve_test.height())
self.solve_test.move(
middle_x - self.solve_test.geometry().width() / 2,
self.solve_problem_button.y() +
self.solve_problem_button.height() + 10)
self.solve_test.setHidden(True)
speed_choose_label = QLabel(self)
speed_choose_label.move(
train_epoch_label.x(),
self.solve_test.y() + self.solve_test.height() + 10)
speed_choose_label.setText("Play Speed：")
speed_choose_label.setFont(QFont("Fixed", 10))
self.play_speed_combo = QComboBox(self)
self.play_speed_combo.move(
period_label.x() + period_label.width() + 15,
speed_choose_label.geometry().y() - 2)
self.play_speed_combo.addItems(["High", "Middle", "Low"])
self.play_speed_combo.setFont(QFont("Fixed", 10))
play_button = QPushButton(self)
play_button.setText("Play Result Now!")
play_button.setFont(QFont("Fixed", 10))
play_button.move(
period_label.x() + period_label.width() + 15,
self.play_speed_combo.geometry().y() +
self.play_speed_combo.geometry().height() + 10)
play_button.pressed.connect(self.play_button_pressed)
def pic_change(self):
self.timer.stop()
current_text = self.maze_input.toPlainText()
rows = current_text.split('\n')
maze_map = []
try:
for row in rows:
row_sp = row.split(',')
row_list = []
for c in row_sp:
c = c.strip()
if len(c) == 0:
continue
else:
row_list.append(int(c))
maze_map.append(row_list)
except:
QMessageBox.information(self, "Warning", "Invalid Input.",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
maze_len = len(maze_map[0])
for i in range(1, len(maze_map)):
if len(maze_map[i]) != maze_len:
QMessageBox.information(
self, "Warning",
"Error，each row should have the same number of columns.",
QMessageBox.Ok | QMessageBox.Close, QMessageBox.Close)
return
my_maze = Maze(maze_map=np.array(maze_map),
period=self.period_input.value() * 2)
self.model = QTableModel(my_maze)
# self.model.play_game((0, 0), 0)
self.Plot.draw_root(self.model.my_maze, (0, 0), 1, 0, False)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period -
1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
def play_button_pressed(self):
if self.model == None:
QMessageBox.information(self, "Tip",
"Please input the maze first.",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.model.play_game((0, 0), 0)
speed_text = self.play_speed_combo.currentText()
self.playing_index = 0
if speed_text == "High":
self.timer.start(100, self)
elif speed_text == "Middle":
self.timer.start(500, self)
else:
self.timer.start(1500, self)
def timerEvent(self, event):
if event.timerId() == self.timer.timerId():
period = self.model.my_maze.period
if period != 0 and (self.playing_index % period) >= period / 2:
fire_flag = True
else:
fire_flag = False
self.Plot.draw_qtable(
self.model, self.playing_index % period if period != 0 else 0,
fire_flag)
self.Plot.draw_root(self.model.my_maze, (0, 0), self.playing_index,
period, fire_flag)
self.playing_index = self.playing_index + 1
if self.playing_index >= len(self.model.my_maze.visited) + 2:
self.playing_index = 0
# print("up",self.playing_index)
else:
super(UIUserDefine, self).timerEvent(event)
def solve_button_pressed(self):
if self.problem_solving:
return
if type(self.model) == type(None):
QMessageBox.information(self, "Tip",
"Please input the maze first.",
QMessageBox.Ok | QMessageBox.Close,
QMessageBox.Close)
return
self.problem_solving = True
self.playing_index = -1
self.solve_test.setHidden(False)
self.timer.stop()
self.repaint()
train_epoch = self.epoch_input.value() * 1000
start_time = time.time()
# path = "tangrams\\" + self.parent().pic_choose_combo.currentText() + ".png"
self.model.train(output_line=self.solve_test,
main_ui=self,
epoch_N=train_epoch)
end_time = time.time()
QMessageBox.information(
self, "Tip",
"Training finished，spend：%.3f s" % (end_time - start_time),
QMessageBox.Ok | QMessageBox.Close, QMessageBox.Close)
self.Plot.draw_qtable(qtable_model=self.model,
time_=self.model.my_maze.period -
1 if self.model.my_maze.period != 0 else 0,
fire_flag=True)
self.problem_solving = False
self.solve_test.setHidden(True)
