213 lines
7.7 KiB
QML
213 lines
7.7 KiB
QML
// =========================================
|
||
// =============== 截图管理器 ===============
|
||
// =========================================
|
||
|
||
import QtQuick 2.15
|
||
import QtQuick.Window 2.15
|
||
|
||
Item {
|
||
id: ssWinRoot
|
||
|
||
// ==================== 【接口】 ====================
|
||
|
||
// 开始一次截图。传入回调函数。
|
||
function screenshot(callback) {
|
||
// 重复调用验证
|
||
if(running) {
|
||
qmlapp.popup.simple(errorTitle, errorRepeat)
|
||
return
|
||
}
|
||
running = true
|
||
// 获取所有屏幕的截图图像
|
||
const grabList = getGrabList()
|
||
// 遍历截图列表，收集覆盖窗口属性 argds
|
||
let argds = []
|
||
for(let i in grabList) {
|
||
const g = grabList[i] // 截图属性
|
||
// 合法性检查
|
||
if(g.imgID.startsWith("[")) {
|
||
qmlapp.popup.message(errorTitle,
|
||
qsTr("显示器： %1\n错误信息： %2").arg(g.screenName).arg(g.imgID), "error")
|
||
callback()
|
||
running = false
|
||
return
|
||
}
|
||
const screen = Qt.application.screens[i] // 获取对应编号的屏幕
|
||
if(screen.name !== g.screenName) {
|
||
qmlapp.popup.message(errorTitle,
|
||
qsTr("屏幕设备名称不相同：\n%1\n%2").arg(screen.name).arg(g.screenName), "error")
|
||
callback()
|
||
running = false
|
||
return
|
||
}
|
||
const argd = {
|
||
imgID: g.imgID,
|
||
screenName: screen.name,
|
||
screen: screen, // 为Window设定所属屏幕属性
|
||
screenRatio: screen.devicePixelRatio, // 屏幕缩放比
|
||
x: screen.virtualX,
|
||
y: screen.virtualY,
|
||
width: screen.width,
|
||
height: screen.height,
|
||
screenshotEnd: ssWinRoot.ssEnd // 关闭函数
|
||
}
|
||
argds.push(argd)
|
||
}
|
||
// 记录回调
|
||
lastCallback = callback
|
||
if(winDict === undefined) winDict = {}
|
||
// 生成覆盖窗口
|
||
for(let a in argds) {
|
||
const obj = ssWinComp.createObject(this, argds[a])
|
||
winDict[argds[a].imgID] = obj
|
||
}
|
||
// 注册esc事件监听
|
||
qmlapp.pubSub.subscribeGroup("<<esc>>", ssWinRoot, "ssEsc", "ssEsc")
|
||
}
|
||
|
||
// 重复上一次截图区域
|
||
function reScreenshot(callback) {
|
||
// 重复调用验证
|
||
if(running) {
|
||
qmlapp.popup.simple(errorTitle, errorRepeat)
|
||
return
|
||
}
|
||
running = true
|
||
if(!lastClipArgd) {
|
||
qmlapp.popup.simple(qsTr("尚未记录截图区域"), "")
|
||
callback()
|
||
running = false
|
||
return
|
||
}
|
||
// 获取所有屏幕的截图图像
|
||
const grabList = getGrabList()
|
||
let errorMsg = "" // 缓存报错信息
|
||
// 在截图列表中，寻找上一次截图所在的屏幕
|
||
for(let i in grabList) {
|
||
const g = grabList[i] // 截图属性
|
||
// 合法性检查
|
||
if(g.imgID.startsWith("[")) {
|
||
errorMsg = g.imgID // 记录失败，跳过本轮
|
||
continue
|
||
}
|
||
// 找到对应屏幕
|
||
if(lastClipArgd.screenName === g.screenName) {
|
||
// 向py汇报，获取裁剪后的imgID
|
||
const clipImgID = imageConn.getClipImgID(g.imgID,
|
||
lastClipArgd.clipX, lastClipArgd.clipY, lastClipArgd.clipW, lastClipArgd.clipH)
|
||
// 成功 调用回调
|
||
runLastCallback(clipImgID)
|
||
running = false
|
||
return
|
||
}
|
||
}
|
||
// 失败，未找到相同屏幕
|
||
lastClipArgd = undefined
|
||
if(!errorMsg) errorMsg = qsTr("未找到匹配的屏幕")
|
||
qmlapp.popup.simple(qsTr("重复截图失败"), errorMsg)
|
||
runLastCallback()
|
||
running = false
|
||
}
|
||
|
||
// 【同步】获取指定区域的截图ID，失败返回 "[Error]..."
|
||
function getScreenshot(rect, screen=0) { // screen 屏幕编号
|
||
// 无需验证 running
|
||
// 获取所有屏幕的截图图像
|
||
const grabList = getGrabList()
|
||
// 参数检查
|
||
if(!Number.isInteger(screen) || screen < 0 || screen >= grabList.length)
|
||
return `[Error] Invalid screen=${screen}: must be an integer (0~${grabList.length-1})`
|
||
if(rect.length != 4) // 不检查内部是否合法，getClipImgID会检查
|
||
return `[Error] Invalid rect=${rect}: must be integers [x,y,w,h]`
|
||
const grab = grabList[screen]
|
||
const imgID = grab.imgID
|
||
if(imgID.startsWith("[")) // 获取截图数据失败
|
||
return imgID
|
||
let [x, y, w, h] = rect
|
||
// 补充缺省宽高
|
||
if(w <= 0) w = grab.width - x
|
||
if(h <= 0) h = grab.height - y
|
||
// 返回裁剪后的imgID
|
||
return imageConn.getClipImgID(imgID, x, y, w, h)
|
||
}
|
||
|
||
// =================================================
|
||
|
||
// 获取所有屏幕的截图图像
|
||
function getGrabList() {
|
||
/*
|
||
返回列表(不为空)，每项为：\n
|
||
{
|
||
"imgID": 图片ID 或 报错信息 "[Error]开头" ,
|
||
"screenName": 显示器名称 ,
|
||
"width": 截图宽度 ,
|
||
"height": 截图高度 ,
|
||
}
|
||
*/
|
||
|
||
// 截图前等待时间
|
||
let wait = 0
|
||
// 需要隐藏主窗口
|
||
if(qmlapp.globalConfigs.getValue("screenshot.hideWindow")) {
|
||
// 若主窗口可见，则记录位置，避免linux隐藏→显示主窗口后位置偏移。
|
||
if(qmlapp.mainWin.getVisibility())
|
||
qmlapp.mainWin.saveGeometry()
|
||
qmlapp.mainWin.setVisibility(false)
|
||
wait = qmlapp.globalConfigs.getValue("screenshot.hideWindowTime")
|
||
}
|
||
// 获取所有屏幕的截图
|
||
const grabList = imageConn.getScreenshot(wait)
|
||
return grabList
|
||
}
|
||
|
||
// Esc退出截图
|
||
function ssEsc() {
|
||
const argd = {clipX: -1, clipY: -1, clipW: -1, clipH: -1}
|
||
ssEnd(argd)
|
||
}
|
||
|
||
// 截图窗口操作完毕的回调
|
||
function ssEnd(argd) {
|
||
// 注销esc事件监听
|
||
qmlapp.pubSub.unsubscribeGroup("ssEsc")
|
||
// 关闭所有覆盖窗口
|
||
for (let key in winDict) {
|
||
winDict[key].destroy()
|
||
}
|
||
winDict = {}
|
||
// 检测是否有效。如果无效，可能是按Esc退出了截图流程
|
||
if(argd.clipX<0 || argd.clipY<0 || argd.clipW<1 || argd.clipH<1 || !argd.imgID) {
|
||
runLastCallback()
|
||
lastClipArgd = undefined
|
||
running = false
|
||
return
|
||
}
|
||
// 记录当前截图信息
|
||
lastClipArgd = argd
|
||
// 向py汇报，获取裁剪后的imgID
|
||
const clipImgID = imageConn.getClipImgID(argd.imgID, argd.clipX, argd.clipY, argd.clipW, argd.clipH)
|
||
// 调用回调
|
||
runLastCallback(clipImgID)
|
||
running = false
|
||
}
|
||
|
||
// 调用上级回调
|
||
function runLastCallback(clipImgID) {
|
||
if (lastCallback && typeof lastCallback === "function") {
|
||
lastCallback(clipImgID)
|
||
}
|
||
}
|
||
|
||
property string errorTitle: qsTr("截图失败")
|
||
property string errorRepeat: qsTr("上次截图操作未结束，不能进行新的截图！")
|
||
property bool running: false // 当前是否正在截图
|
||
property var lastClipArgd: undefined // 最后一次截图的信息
|
||
property var lastCallback: undefined // 截图完毕的回调，得到 clipImgID
|
||
property var winDict: {} // 存放当前已打开的窗口
|
||
property QtObject imageConn: qmlapp.imageManager.imageConnector
|
||
|
||
Component {
|
||
id: ssWinComp
|
||
ScreenshotWindowComp { }
|
||
}
|
||
} |