import random from PySide6.QtGui import QFont from PySide6.QtWidgets import QApplication, QMainWindow, QGraphicsLineItem, QMessageBox, QGraphicsScene import data from graph_24 import * from main_ui import Ui_MainWindow class MainWindow(QMainWindow, Ui_MainWindow): def __init__(self): super().__init__() self.drawn_tree_id = set() # 已经绘制了树的节点 self.seq = 1 # 在有树的情况下，添加节点的序号，以确定新加的节点位置 self.trees_root = [] # 保存所有树的根节点 self.selected_id = [] # 选中的节点ID self.adds_id = set() # 已经用过的ID self.value_nodes = [] # 数值节点 # self.operator_nodes = [] self.selected_ops = None # 选择的运算符 self.selected_value = None # 当前选择的卡片值 self.r_count = 0 # 记录所有答案点击次数 self.offset = 0 # 语法树遍历时随深度平移长度，避免节点重合 self.TreeNode = [] self.Tree_4_nums = None self.setupUi(self) self.init_ui() self.init_slot() self.pushButton_q.clicked.connect(self.question_card) # 出题 self.pushButton_ex.clicked.connect(self.change_card) # 换牌 self.pushButton_autosv.clicked.connect(self.AutoTest_9_10) # 自动求解 self.pushButton_dtree.clicked.connect(self.draw_tree) # 绘制语法树 self.pushButton_ast.clicked.connect(self.calculate_button) # 生成语法树 self.pushButton_allan.clicked.connect(self.AutoTest) # 自动求解 self.clear_button.clicked.connect(self.secen_clear) # 清空画布 self.pushButton_add.clicked.connect(lambda: self.select_ops('+')) self.pushButton_sub.clicked.connect(lambda: self.select_ops('-')) self.pushButton_mul.clicked.connect(lambda: self.select_ops('*')) self.pushButton_div.clicked.connect(lambda: self.select_ops('/')) self.add_val_node.clicked.connect(self.add_value_node) # 添加数值节点 self.add_ops_node.clicked.connect(self.add_operator_node) # 添加运算符节点 self.ex_node_value.clicked.connect(self.exchange_value_node) # 修改数值节点 self.ex_node_ops.clicked.connect(self.exchange_ops_node) # 修改运算符节点 self.ex_node_self.clicked.connect(self.exchange_two_nodes) # 交换两个节点 self.pushButton_exp.clicked.connect(self.agent_exp) # 生成表达式 self.pushButton_check_te.clicked.connect(self.check_tree_and_expression) # 验证语法树和表达式是否匹配 def secen_clear(self): self.scene.clear() self.selected_id = [] # 选中的节点ID self.value_nodes = [] # 数值节点 self.TreeNode = [] # 所有节点 self.adds_id = set() # 已经用过的ID self.trees_root = [] # 保存所有树的根节点 self.drawn_tree_id = set() # 已经绘制了树的节点 self.selected_value = None # 当前选择的卡片值. self.selected_ops = None # 选择的运算符 # 清空运算符显示 self.lb_ops_selected.setText("") # 运算符显示 self.p1_selected.setStyleSheet("border-image: none;") # 清空已选牌面 self.pushButton_ast.setEnabled(True) self.pushButton_allan.setEnabled(True) self.pushButton_dtree.setEnabled(True) self.pushButton_autosv.setEnabled(True) def select_ops(self, ops): self.lb_ops_selected.setText(ops) self.selected_ops = ops def add_value_node(self): # self.scene.clear() if self.selected_value is not None: # 选择了卡片值 self.value_nodes.append(self.selected_value) # 添加到节点数值列表 for i in range(len(self.adds_id) + 1): if i not in self.adds_id: value = self.selected_value nodet = RoundedRectangleItem(150 * self.seq, 0, str(value), i) self.seq += 1 # 序号加1 nodet.selected.connect(lambda id: self.selected_node(id)) # 连接信号 nodet.unselected.connect(lambda id: self.unselected_node(id)) node = data.Node([i, 'Value', None, None, None, value, None]) self.adds_id.add(i) # 添加全局id self.TreeNode.append(node) self.scene.addItem(nodet) else: QMessageBox.warning(self, '警告', '请先选择一张牌') def judge_node_conn(self): # 判断节点是否已经连接 for id in self.selected_id: for node in self.TreeNode: if node.LeftNodeID == id or node.RightNodeID == id: return True def add_operator_node(self): # 添加运算符节点 if self.selected_ops is not None: if len(self.selected_id) == 2: # 判断已选节点是否已经被连接 if self.judge_node_conn(): QMessageBox.warning(self, '警告', '节点已经连接') return for i in range(len(self.adds_id) + 1): # 确保id唯一 if i not in self.adds_id: node = data.Node([i, 'Operator', self.selected_ops, None, None, None, None]) self.TreeNode.append(node) self.adds_id.add(i) # 全局ID node.LeftNodeID = int(self.selected_id[1]) node.RightNodeID = int(self.selected_id[0]) self.flash_node(self.selected_id[0]) self.flash_node(self.selected_id[1]) self.selected_id = [] # 清空选中的节点 self.trees_root.append(node) # 保存根节点 # 从根节点列表中删去作为子节点连接的节点 node_to_remove = [] for root in self.trees_root: if root.NodeID == node.LeftNodeID or root.NodeID == node.RightNodeID: node_to_remove.append(root) for root in node_to_remove: self.trees_root.remove(root) print("已用ID", self.adds_id) self.scene.clear() for i, root in enumerate(self.trees_root): if self.has_both_operators(self.TreeNode): self.offset = 35 self.ShowTree(root, self.TreeNode, i * 300, 0) self.offset = 0 self.seq = 1 # 重置序号 for noed in self.TreeNode: print("运算符节点", noed) # exps = data.get_expression(self.TreeNode) # print("表达式", exps) else: QMessageBox.warning(self, '警告', '请先选择两个节点') else: QMessageBox.warning(self, '警告', '请先选择一个运算符') # 根据节点ID，找到对应的节点 def find_node(self, id): for node in self.TreeNode: if node.NodeID == id: return node def exchange_value_node(self): # 修改节点值 if len(self.selected_id) == 1: node = self.find_node(self.selected_id[0]) if node.NodeType == 'Operator': QMessageBox.warning(self, '警告', '请选择数值节点') return if self.selected_value: if self.TreeNode: for node in self.TreeNode: if node.NodeID == self.selected_id[0]: node.FaceValue = str(self.selected_value) for nodet in self.scene.items(): if isinstance(nodet, RoundedRectangleItem): if nodet.id == self.selected_id[0]: nodet.text = str(self.selected_value) nodet.update() else: QMessageBox.warning(self, '警告', '请先选择一张牌') else: QMessageBox.warning(self, '警告', '请先选择一个节点') def exchange_ops_node(self): # 修改节点运算符 if len(self.selected_id) == 1: node = self.find_node(self.selected_id[0]) if node.NodeType == 'Value': QMessageBox.warning(self, '警告', '请选择运算符节点') return if self.selected_ops: if self.TreeNode: for node in self.TreeNode: if node.NodeID == self.selected_id[0]: node.Ops = self.selected_ops for nodet in self.scene.items(): if isinstance(nodet, RoundedRectangleItem): if nodet.id == self.selected_id[0]: nodet.text = str(self.selected_ops) nodet.update() else: QMessageBox.warning(self, '警告', '请先选择一个运算符') else: QMessageBox.warning(self, '警告', '请先选择一个节点') def exchange_two_nodes(self): # 交换选中两个节点FaceValue的值 if len(self.selected_id) == 2: if self.TreeNode: node_1 = None node_2 = None for node in self.TreeNode: if node.NodeID == self.selected_id[0]: if node.NodeType == 'Value': node_1 = node else: QMessageBox.warning(self, '警告', '请选择两个数值节点') return elif node.NodeID == self.selected_id[1]: if node.NodeType == 'Value': node_2 = node else: QMessageBox.warning(self, '警告', '请选择两个数值节点') return if node_1 and node_2: node_1.FaceValue, node_2.FaceValue = node_2.FaceValue, node_1.FaceValue for item in self.scene.items(): if isinstance(item, RoundedRectangleItem): if item.id == self.selected_id[0]: item.text = str(node_1.FaceValue) item.update() elif item.id == self.selected_id[1]: item.text = str(node_2.FaceValue) item.update() else: QMessageBox.warning(self, '警告', '请选择两个节点') def selected_node(self, id): if len(self.selected_id) < 2: self.selected_id.append(id) else: QMessageBox.warning(self, '警告', '不能选择超过3个节点') for nodet in self.scene.items(): # 遍历场景中的所有图元 if isinstance(nodet, RoundedRectangleItem): # 如果是圆角矩形 if nodet.id == id: # 如果是当前选中的节点 nodet.isSelected = False # 取消选中 nodet.update() # 更新图元 def unselected_node(self, id): self.selected_id.remove(id) print(self.selected_id) def flash_node(self, id): # 还原节点为未选中状态 for nodet in self.scene.items(): # 遍历场景中的所有图元 if isinstance(nodet, RoundedRectangleItem): # 如果是圆角矩形 if nodet.id == id: # 如果是当前选中的节点 nodet.isSelected = False # 取消选中 nodet.update() # 更新图元 def init_ui(self): self.scene = QGraphicsScene() self.graphicsView.setScene(self.scene) def init_slot(self): self.pushButton.clicked.connect(self.on_calc) def load_image(self, path, Tree_4_nums, index): def current_selection(value, path): self.selected_value = str(value) self.p1_selected.setStyleSheet(f"border-image: url({path});") self.pushButton_ast.setEnabled(False) self.pushButton_allan.setEnabled(False) self.pushButton_dtree.setEnabled(False) self.pushButton_autosv.setEnabled(False) # print(value) if index == 0: self.p1.setStyleSheet(f"border-image: url({path});") self.p1.clicked.connect(lambda: current_selection(Tree_4_nums[index], path)) elif index == 1: self.p2.setStyleSheet(f"border-image: url({path});") self.p2.clicked.connect(lambda: current_selection(Tree_4_nums[index], path)) elif index == 2: self.p3.setStyleSheet(f"border-image: url({path});") self.p3.clicked.connect(lambda: current_selection(Tree_4_nums[index], path)) elif index == 3: self.p4.setStyleSheet(f"border-image: url({path});") self.p4.clicked.connect(lambda: current_selection(Tree_4_nums[index], path)) def show_card(self): # 原始数组 original_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13] face_card = ['Club', 'Diamond', 'Heart', 'Spade'] # 从原始数组中随机选择4个元素组成新列表 self.Tree_4_nums = random.sample(original_list, 4) # 从花色中随机选择一个 for index, num in enumerate(self.Tree_4_nums): card_face = random.choice(face_card) if num == 11: card_path = f"card/{card_face}/{card_face}j.png" elif num == 12: card_path = f"card/{card_face}/{card_face}q.png" elif num == 13: card_path = f"card/{card_face}/{card_face}k.png" else: card_path = f"card/{card_face}/{card_face}{num}.png" self.load_image(card_path, self.Tree_4_nums, index) # return self.Tree_4_nums def question_card(self): # 显示题目卡片 self.secen_clear() # 清空画布 self.lineEdit.clear() self.label_result.setText(" =? ") self.label_TF.setText("") self.show_card() # 显示卡片 # 换牌 def change_card(self): self.show_card() # 显示卡片 def AutoTest(self): if self.Tree_4_nums is None: QMessageBox.warning(self, '警告', '请先出题') return else: results = data.get_Aexps(self.Tree_4_nums) if results == "无法得出24点": self.lineEdit.clear() self.lineEdit.setText("无法得出24点") return else: if self.r_count < len(results): self.lineEdit.clear() self.lineEdit.setText(f"{results[self.r_count]}") self.on_calc() # 自动计算 self.draw_tree() # 绘制语法树 self.r_count += 1 else: QMessageBox.information(self, '提示', '已经是最后一个表达式了') self.r_count = 0 def AutoTest_9_10(self): if self.Tree_4_nums is None: QMessageBox.warning(self, '警告', '请先出题') return else: aexp = data.gen_exp(self.Tree_4_nums) if aexp == "无法得出24点": self.lineEdit.clear() self.lineEdit.setText("无法得出24点") else: self.lineEdit.clear() self.lineEdit.setText(f"{aexp}") self.on_calc() # 自动计算 self.draw_tree() # 绘制语法树 def calculate_button(self): # 计算生成语法树 self.scene.clear() aexp = self.lineEdit.text() try: exp_ls = data.get_postfix_expression(aexp) # 获取后缀表达式列表 self.TreeNode = data.ExpressionAnalyse(exp_ls) # 将后缀表达式转化为节点树 result = data.calculate(self.TreeNode[-1], self.TreeNode) self.label_result.setText(f"={round(result, 2)}") except: QMessageBox.warning(self, '警告', '表达式错误！') return if self.TreeNode: # 如果节点树不为空 root = self.TreeNode[-1] result = data.calculate(root, self.TreeNode) # 计算表达式的值 self.label_result.setText(f"={round(result, 2)}") self.result_ToF(int(result)) # 判断结果是否为24 if self.has_both_operators(self.TreeNode): self.offset = 35 self.ShowTree(root, self.TreeNode) self.offset = 0 def has_both_operators(self, TreeNode): # 判断是否同时有两个运算符 ops = data.OPS def find_op_by_id(id): # 根据ID查找运算符 for node in TreeNode: if node.NodeID == id and node.NodeType == "Operator": return node.Ops for node in TreeNode: # 遍历节点树 if node.NodeType == "Operator": opl = find_op_by_id(node.LeftNodeID) opr = find_op_by_id(node.RightNodeID) if opl in ops and opr in ops: return True def check_tree_and_expression(self): # 检查节点树和表达式是否一致 if self.TreeNode is not None: aexp = self.lineEdit.text() if aexp == '': QMessageBox.warning(self, '警告', '请输入表达式') return try: exp_ls = data.get_postfix_expression(aexp) # 获取后缀表达式列表 TreeNode2 = data.ExpressionAnalyse(exp_ls) # 将后缀表达式转化为节点树 if self.check_tree(self.TreeNode, TreeNode2): QMessageBox.information(self, '提示', '节点树与表达式一致') else: QMessageBox.warning(self, '警告', '节点树与表达式不一致') except Exception as e: QMessageBox.warning(self, '警告', '表达式错误') return else: QMessageBox.warning(self, '警告', '当前无语法树') # 检查两个节点树是否一致的函数 def check_tree(self, TreeNode1, TreeNode2): if len(TreeNode1) != len(TreeNode2): return False for i in range(len(TreeNode1)): if TreeNode1[i].NodeID != TreeNode2[i].NodeID: return False if TreeNode1[i].NodeType != TreeNode2[i].NodeType: return False if TreeNode1[i].Ops != TreeNode2[i].Ops: return False if TreeNode1[i].FaceValue != TreeNode2[i].FaceValue: return False if TreeNode1[i].LeftNodeID != TreeNode2[i].LeftNodeID: return False if TreeNode1[i].RightNodeID != TreeNode2[i].RightNodeID: return False return True def draw_tree(self): self.scene.clear() aexp = self.lineEdit.text() if aexp == "": QMessageBox.warning(self, '警告', '请输入表达式') return try: exp_ls = data.get_postfix_expression(aexp) # 获取后缀表达式列表 self.TreeNode = data.ExpressionAnalyse(exp_ls) # 将后缀表达式转化为节点树 root = self.TreeNode[-1] if self.has_both_operators(self.TreeNode): self.offset = 35 self.ShowTree(root, self.TreeNode, depth=0) self.offset = 0 except Exception as e: QMessageBox.warning(self, '警告', '表达式错误') def ShowTree(self, root, TreeNode, startx=0, starty=0, depth=0): self.selected_id = [] # 清空选中的节点 self.drawn_tree_id = set() # 清空已绘制的节点ID # 绘制根节点 if root.NodeType == 'Value': nds_root = RoundedRectangleItem(startx, starty, root.FaceValue, root.NodeID) nds_root.selected.connect(lambda id: self.selected_node(id)) nds_root.unselected.connect(lambda id: self.unselected_node(id)) self.drawn_tree_id.add(root.NodeID) self.adds_id.add(root.NodeID) else: nds_root = RoundedRectangleItem(startx, starty, root.Ops, root.NodeID) nds_root.selected.connect(lambda id: self.selected_node(id)) nds_root.unselected.connect(lambda id: self.unselected_node(id)) self.drawn_tree_id.add(root.NodeID) self.adds_id.add(root.NodeID) nds_root.setZValue(1) self.scene.addItem(nds_root) # 计算节点间的水平偏移量 horizontal_offset = 100 - depth * self.offset # 计算左子树和右子树的位置 left_x, left_y = startx - horizontal_offset, starty + 100 right_x, right_y = startx + horizontal_offset, starty + 100 # 绘制左子树 if root.LeftNodeID is not None: left_child = next(node for node in TreeNode if node.NodeID == root.LeftNodeID) self.ShowTree(left_child, TreeNode, left_x, left_y, depth + 1) # 绘制左子树与根节点之间的连线 line_left = QGraphicsLineItem(startx + 35, starty + 35, left_x + 35, left_y + 35) line_left.setZValue(-1) line_left.setPen(QPen(Qt.red, 3)) self.scene.addItem(line_left) # 绘制右子树 if root.RightNodeID is not None: right_child = next(node for node in TreeNode if node.NodeID == root.RightNodeID) self.ShowTree(right_child, TreeNode, right_x, right_y, depth + 1) # 绘制右子树与根节点之间的连线 line_right = QGraphicsLineItem(startx + 35, starty + 35, right_x + 35, right_y + 35) line_right.setZValue(-1) line_right.setPen(QPen(Qt.red, 3)) self.scene.addItem(line_right) def on_calc(self): aexp = self.lineEdit.text() if aexp == '': QMessageBox.warning(self, '警告', '请输入表达式') return try: result = eval(aexp) self.label_result.setText(f"={int(result)}") # 设置字体样式 self.result_ToF(result) except Exception as e: QMessageBox.warning(self, '警告', '表达式错误') return def result_ToF(self, result): font = QFont('Microsoft YaHei', 16) font.setBold(True) # 将字体设置为粗体 self.label_TF.setText('√' if result == 24 else '×') # 将字体样式应用到 QLabel 上 self.label_TF.setFont(font) # 将文本颜色设置为红色 self.label_TF.setStyleSheet("color: red;background-color:rgb(219, 205, 166)") def agent_exp(self): if len(self.TreeNode) > 0: exp_str = data.get_expression(self.TreeNode) if exp_str is not None: self.label_result.clear() # 清空显示结果 self.lineEdit.clear() # 清空显示表达式 self.lineEdit.setText(exp_str) # 显示表达式 else: QMessageBox.warning(self, '警告', '无法生成表达式') else: QMessageBox.warning(self, '警告', '当前无语法树') if __name__ == '__main__': app = QApplication() window = MainWindow() window.show() app.exec()