基于python+opencv实现的欢乐斗地主记牌器系统源码.zip

#!/usr/bin/python # coding:utf-8 """ @author: css @software: PyCharm @file: Window.py @time: 2021/3/3 22:02:25 """ from PyQt5 import QtGui, QtWidgets from PyQt5.QtCore import * from PyQt5.QtGui import QIcon, QFont, QTextCursor from PyQt5.QtWidgets import (QWidget, QPushButton, QSystemTrayIcon, QScrollArea, QTextBrowser, QMenu, QLabel, QAction) import time import win32gui, win32con from util.screen import screen import cv2 as cv class Window(QWidget): isTopWin = True selfWnd = None height = 80 pokerLogStr = '' def __init__(self, hWnd): super().__init__() left, top, right, bot = win32gui.GetWindowRect(hWnd) self.setGeometry(0, 0, right - left, self.height) # Qt.WindowStaysOnTopHint | self.setWindowFlags(Qt.FramelessWindowHint | Qt.Tool) # 任务栏托盘 # 在系统托盘处显示图标 self.tp = QSystemTrayIcon(self) self.tp.setIcon(QIcon('icon.png')) # 设置系统托盘图标的菜单 a1 = QAction('&欢乐斗地主记牌器', self, triggered=self.showTray) a2 = QAction('&退出(Exit)', self, triggered=self.quitApp) # 直接退出可以用qApp.quit tpMenu = QMenu() tpMenu.addAction(a1) tpMenu.addAction(a2) self.tp.setObjectName("记牌器") self.tp.setContextMenu(tpMenu) # 不调用show不会显示系统托盘 self.tp.show() self.tp.activated.connect(self.act) # 信息提示 # 参数1：标题 # 参数2：内容 # 参数3：图标（0没有图标 1信息图标 2警告图标 3错误图标），0还是有一个小图标 self.tp.showMessage('tp', 'tpContent', icon=0) self.hWnd = hWnd self.setWindowTitle("CV记牌器") pokerLabel = QLabel(self) pokerLabel.setText("牌型： 王 2 A K Q J 10 9 8 7 6 5 4 3 ") # 设置标签的左边距，上边距，宽，高 pokerLabel.setGeometry(QRect(0, 0, 800, 30)) # 设置文本标签的字体和大小，粗细等 pokerLabel.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.pokerNumLabel = QLabel(self) # 设置标签的左边距，上边距，宽，高 self.pokerNumLabel.setGeometry(QRect(0, 25, 800, 30)) # 设置文本标签的字体和大小，粗细等 self.pokerNumLabel.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.handlePokerLabel = QLabel(self) # 设置标签的左边距，上边距，宽，高 self.handlePokerLabel.setGeometry(QRect(0, 50, 800, 30)) # 设置文本标签的字体和大小，粗细等 self.handlePokerLabel.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.screen = QPushButton(self) self.screen.setText("截图") self.screen.setGeometry(QRect(950, 3, 50, 20)) self.screen.clicked.connect(self.screenshot) self.status = QLabel(self) self.status.setGeometry(QRect(230, 58, 80, 20)) self.status.setText("未开始") self.status.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.status.setStyleSheet('color: orange') self.setPokerSizeLabel = QPushButton(self) self.setPokerSizeLabel.setGeometry(QRect(950, 29, 80, 20)) self.setPokerSizeLabel.setText("调整大小") # self.setPokerSizeLabel.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.setPokerSizeLabel.clicked.connect(self.setPokerSize) self.clearLogButton = QPushButton(self) self.clearLogButton.setGeometry(QRect(950, 55, 80, 20)) self.clearLogButton.setText("清空日志") # self.setPokerSizeLabel.setFont(QtGui.QFont("Roman times", 15, QtGui.QFont.Bold)) self.clearLogButton.clicked.connect(self.clearLog) line = QLabel(self) # 设置标签的左边距，上边距，宽，高 line.setGeometry(QRect(0, 27, 690, 1)) line.setAutoFillBackground(True) line.setStyleSheet('background-color: rgb(0, 0, 0)') line = QLabel(self) # 设置标签的左边距，上边距，宽，高 line.setGeometry(QRect(0, 52, 690, 1)) line.setAutoFillBackground(True) line.setStyleSheet('background-color: rgb(0, 0, 0)') line = QLabel(self) # 设置标签的左边距，上边距，宽，高 line.setGeometry(QRect(700, 0, 80, 30)) line.setAutoFillBackground(True) # line.setStyleSheet('background-color: rgb(0, 0, 0)') line.setText("出牌\n日志:") line = QLabel(self) # 设置标签的左边距，上边距，宽，高 line.setGeometry(QRect(690, 0, 1, self.height)) line.setAutoFillBackground(True) line.setStyleSheet('background-color: rgb(0, 0, 0)') for index in range(13): line = QLabel(self) # 设置标签的左边距，上边距，宽，高 line.setGeometry(QRect(index * 45 + 113, 0, 1, self.height if index < 3 else 53)) line.setAutoFillBackground(True) line.setStyleSheet('background-color: rgb(0, 0, 0)') self.log_text = QTextBrowser(self) self.log_text.move(740, 0) self.log_text.resize(200, 80) self.log_text.setText(self.pokerLogStr) def setStatus(self, status): if (status == 0): self.status.setText("未开始") sss = self.getFontLabelText("对局结束", 'red') print(sss) self.log_text.append(self.getFontLabelText("对局结束", 'red')) elif status == 1: self.status.setText("开局") self.log_text.append(self.getFontLabelText("对局开始", 'green')) elif status == 2: self.status.setText("进行中") def setPokerSize(self): left, top, right, bot = win32gui.GetWindowRect(self.hWnd) win32gui.SetWindowPos(self.hWnd, None, left, top, 1040, 629, win32con.SWP_SHOWWINDOW) self.resize(1040, 80) def clearLog(self): self.log_text.clear() def screenshot(self): img = screen(self.hWnd) cv.imwrite('screen/screen' + str(time.time()).split('.')[0] + '.png', img) def color(self, num): if num == 0: return 'black' if num == 1: return 'green' if num == 2: return '#66b1ff' if num == 3: return 'orange' if num == 4: return 'red' def getFontLabelText(self, text, color, font=18): return "<font style = 'font-size:" + str( font) + "px; font-weight: bold; color:" + color + ";'>" + text + "</font>" def setNum(self, num, handNum, outList): text = '数量：' if len(num) > 0: text += '&nbsp;' for index, value in enumerate(num): text = text + self.gText(str(value), self.color(value)) if (index == 6): text += ' ' self.pokerNumLabel.setText(text) self.setHandPokerNum(handNum) self.appendPokerLog(outList) def appendPokerLog(self, logList): list = [] pokerLogStr = '' # print(logList) if 'upotherList' in logList: pokerLogStr += '上家：' list = logList['upotherList'] if 'downotherList' in logList: pokerLogStr += '下家：' list = logList['downotherList'] if 'selfOutList' in logList: pokerLogStr += '自家：' list = logList['selfOutList'] if (len(list) > 0): list.sort(key=lambda poker: poker['x']) for item in list: for num in range(0, item['num']): pokerLogStr += item['name'] self.log_text.append(pokerLogStr) self.log_text.moveCursor(self.log_text.textCursor().End) def gText(self, text, colo