Public-Mirror/LunaTranslator
1
0
Fork 0
mirror of https://github.com/HIllya51/LunaTranslator.git synced 2024-12-29 16:44:13 +08:00
Code Issues Projects Releases Wiki Activity

multi screen

Browse Source
This commit is contained in:
恍兮惚兮 2024-05-26 20:53:10 +08:00
parent 9335e8be40
commit 1926586d0f
2 changed files with 58 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
LunaTranslator/LunaTranslator/gui/rangeselect.py
View File

@ -1,4 +1,12 @@
from PyQt5.QtWidgets import QMenu, QApplication, QMainWindow, QLabel, QAction, QDialog from PyQt5.QtWidgets import (
QMenu,
QApplication,
QMainWindow,
QLabel,
QAction,
QDialog,
QDesktopWidget,
)
from PyQt5.QtGui import QPainter, QPen, QColor from PyQt5.QtGui import QPainter, QPen, QColor
from PyQt5.QtCore import Qt, QPoint, QRect, QEvent from PyQt5.QtCore import Qt, QPoint, QRect, QEvent
from myutils.config import _TR from myutils.config import _TR
@ -69,6 +77,19 @@ class rangeadjust(Mainw):
), ),
] ]
def setGeometry(self, x, y, w, h):
if QDesktopWidget().screenCount() > 1:
windows.MoveWindow(int(self.winId()), x, y, w, h, True)
else:
super().setGeometry(x, y, w, h)
def geometry(self):
if QDesktopWidget().screenCount() > 1:
rect = windows.GetWindowRect(int(self.winId()))
return QRect(rect[0], rect[1], rect[2] - rect[0], rect[3] - rect[1])
else:
return super().geometry()
def moveEvent(self, e): def moveEvent(self, e):
if self._rect: if self._rect:
self._rect = self.rectoffset(self.geometry()) self._rect = self.rectoffset(self.geometry())
@ -125,6 +146,8 @@ class rangeselct(QMainWindow):
self.setMouseTracking(True) self.setMouseTracking(True)
self.start_point = QPoint() self.start_point = QPoint()
self.end_point = QPoint() self.end_point = QPoint()
self.__start = None
self.__end = None
self.startauto = False self.startauto = False
self.clickrelease = False self.clickrelease = False
self.rectlabel.setStyleSheet( self.rectlabel.setStyleSheet(
@ -172,27 +195,37 @@ class rangeselct(QMainWindow):
self.start_point = event.pos() self.start_point = event.pos()
self.end_point = self.start_point self.end_point = self.start_point
self.is_drawing = True self.is_drawing = True
self.__start = self.__end = windows.GetCursorPos()
def mouseMoveEvent(self, event): def mouseMoveEvent(self, event):
if self.startauto and self.is_drawing == False: if self.startauto and self.is_drawing == False:
self.is_drawing = True self.is_drawing = True
self.end_point = self.start_point = event.pos() self.end_point = self.start_point = event.pos()
self.__start = self.__end = windows.GetCursorPos()
self.startauto = False self.startauto = False
if self.is_drawing: if self.is_drawing:
self.end_point = event.pos() self.end_point = event.pos()
self.__end = windows.GetCursorPos()
self.update() self.update()
def getRange(self): def getRange(self):
start_point = self.mapToGlobal(self.start_point) if QDesktopWidget().screenCount() > 1:
end_point = self.mapToGlobal(self.end_point) x1, y1, x2, y2 = (
x1, y1, x2, y2 = ( self.__start.x,
start_point.x(), self.__start.y,
start_point.y(), self.__end.x,
end_point.x(), self.__end.y,
end_point.y(), )
) else:
start_point = self.mapToGlobal(self.start_point)
end_point = self.mapToGlobal(self.end_point)
x1, y1, x2, y2 = (
start_point.x(),
start_point.y(),
end_point.x(),
end_point.y(),
)
x1, x2 = min(x1, x2), max(x1, x2) x1, x2 = min(x1, x2), max(x1, x2)
y1, y2 = min(y1, y2), max(y1, y2) y1, y2 = min(y1, y2), max(y1, y2)

19
LunaTranslator/LunaTranslator/myutils/ocrutil.py
View File

@ -1,7 +1,7 @@
import windows import windows
import os, importlib import os, importlib
from myutils.config import globalconfig, _TR from myutils.config import globalconfig, _TR
from PyQt5.QtWidgets import QApplication from PyQt5.QtWidgets import QApplication, QDesktopWidget
from PyQt5.QtGui import QImage from PyQt5.QtGui import QImage
from PyQt5.QtCore import QByteArray, QBuffer from PyQt5.QtCore import QByteArray, QBuffer
from myutils.commonbase import ArgsEmptyExc from myutils.commonbase import ArgsEmptyExc
@ -83,9 +83,20 @@ def imageCut(hwnd, x1, y1, x2, y2, viscompare=True, rawimage=False) -> QImage:
except: except:
print_exc() print_exc()
else: else:
pix = screen.grabWindow( if QDesktopWidget().screenCount() > 1:
QApplication.desktop().winId(), x1, y1, x2 - x1, y2 - y1 desktop = QApplication.primaryScreen().virtualGeometry()
) pix = screen.grabWindow(
QApplication.desktop().winId(),
desktop.x(),
desktop.y(),
desktop.width(),
desktop.height(),
)
pix = pix.copy(x1, y1, x2 - x1, y2 - y1)
else:
pix = screen.grabWindow(
QApplication.desktop().winId(), x1, y1, x2 - x1, y2 - y1
)
image = pix.toImage() image = pix.toImage()
if rawimage: if rawimage: