484 lines
17 KiB
QML
484 lines
17 KiB
QML
|
// ==============================================
|
|||
|
|
// =============== 功能页：截图OCR ===============
|
|||
|
|
// ==============================================
|
|||
|
|
|
|||
|
|
import QtQuick 2.15
|
|||
|
|
|
|||
|
|
import ".."
|
|||
|
|
import "../../Widgets"
|
|||
|
|
import "../../Widgets/ResultLayout"
|
|||
|
|
import "../../Widgets/ImageViewer"
|
|||
|
|
|
|||
|
|
TabPage {
|
|||
|
|
id: tabPage
|
|||
|
|
// 配置
|
|||
|
|
configsComp: ScreenshotOcrConfigs {}
|
|||
|
|
property string msnState: "none" // OCR任务状态， none run
|
|||
|
|
|
|||
|
|
// ========================= 【逻辑】 =========================
|
|||
|
|
|
|||
|
|
// 重复截图
|
|||
|
|
function reScreenshot() {
|
|||
|
|
qmlapp.imageManager.reScreenshot(screenshotEnd)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开始截图
|
|||
|
|
function screenshot() {
|
|||
|
|
qmlapp.imageManager.screenshot(screenshotEnd)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 截图完毕
|
|||
|
|
function screenshotEnd(clipID) {
|
|||
|
|
popMainWindow()
|
|||
|
|
if(!clipID) { // 截图取消
|
|||
|
|
tabPage.callPy("ocrImgID", undefined, undefined)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const configDict = configsComp.getValueDict()
|
|||
|
|
tabPage.callPy("ocrImgID", clipID, configDict)
|
|||
|
|
qmlapp.tab.showTabPageObj(tabPage) // 切换标签页
|
|||
|
|
imageText.showImgID(clipID) // 展示图片
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 指定区域截图。rect=[x,y,w,h] screen=屏幕编号 返回"[Success]"为成功
|
|||
|
|
function autoScreenshot(rect, screen) {
|
|||
|
|
// 获取截图
|
|||
|
|
const clipID = qmlapp.imageManager.getScreenshot(rect, screen)
|
|||
|
|
if(!clipID) {
|
|||
|
|
tabPage.callPy("ocrImgID", "[Error] Unknow", undefined)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if(clipID.startsWith("[")) {
|
|||
|
|
tabPage.callPy("ocrImgID", clipID, undefined)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 进行识别
|
|||
|
|
const configDict = configsComp.getValueDict()
|
|||
|
|
tabPage.callPy("ocrImgID", clipID, configDict)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开始粘贴
|
|||
|
|
function paste() {
|
|||
|
|
popMainWindow()
|
|||
|
|
const res = qmlapp.imageManager.getPaste()
|
|||
|
|
if(res.error) {
|
|||
|
|
const t = qsTr("获取剪贴板异常")
|
|||
|
|
qmlapp.popup.simple(t, res.error)
|
|||
|
|
tabPage.callPy("ocrImgID", `[Error] ${t} ${res.error}`, undefined)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if(res.text) {
|
|||
|
|
const t = qsTr("剪贴板中为文本")
|
|||
|
|
qmlapp.popup.simple(t, res.text)
|
|||
|
|
tabPage.callPy("ocrImgID", `[Warning] ${t}`, undefined)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
qmlapp.tab.showTabPageObj(tabPage) // 切换标签页
|
|||
|
|
if(res.imgID) { // 图片
|
|||
|
|
imageText.showImgID(res.imgID)
|
|||
|
|
const configDict = configsComp.getValueDict()
|
|||
|
|
tabPage.callPy("ocrImgID", res.imgID, configDict)
|
|||
|
|
}
|
|||
|
|
else if(res.paths) { // 地址
|
|||
|
|
ocrPaths(res.paths)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 异步扫描一批图像路径
|
|||
|
|
function ocrPaths(paths) {
|
|||
|
|
qmlapp.asynFilesLoader.run(paths,"image",false,onAddImages)
|
|||
|
|
}
|
|||
|
|
// 完毕后，对合法路径进行OCR
|
|||
|
|
function onAddImages(paths) {
|
|||
|
|
if(!paths || paths.length < 1) {
|
|||
|
|
qmlapp.popup.simple(qsTr("无有效图片"), "")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
const configDict = configsComp.getValueDict()
|
|||
|
|
const simpleType = configDict["other.simpleNotificationType"]
|
|||
|
|
qmlapp.popup.simple(qsTr("导入%1条图片路径").arg(paths.length), "", simpleType)
|
|||
|
|
imageText.showPath(paths[0])
|
|||
|
|
tabPage.callPy("ocrPaths", paths, configDict)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 停止所有任务
|
|||
|
|
function msnStop() {
|
|||
|
|
tabPage.callPy("msnStop")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭页面
|
|||
|
|
function closePage() {
|
|||
|
|
if(msnState !== "none") {
|
|||
|
|
const argd = {yesText: qsTr("依然关闭")}
|
|||
|
|
const callback = (flag)=>{
|
|||
|
|
if(flag) {
|
|||
|
|
msnStop()
|
|||
|
|
eventUnsub()
|
|||
|
|
delPage()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
qmlapp.popup.dialog("", qsTr("任务正在进行中。\n要结束任务并关闭页面吗？"), callback, "warning", argd)
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
eventUnsub()
|
|||
|
|
delPage()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 弹出主窗口
|
|||
|
|
function popMainWindow() {
|
|||
|
|
// 若主窗口已经可见，则不处理
|
|||
|
|
if(qmlapp.mainWin.getVisibility())
|
|||
|
|
return
|
|||
|
|
// 等一回合再弹，防止与收回截图窗口相冲突
|
|||
|
|
if(configsComp.getValue("action.popMainWindow")) {
|
|||
|
|
Qt.callLater(()=>{
|
|||
|
|
qmlapp.mainWin.loadGeometry(false)
|
|||
|
|
qmlapp.mainWin.setVisibility(true)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【事件管理】 =========================
|
|||
|
|
|
|||
|
|
Component.onCompleted: {
|
|||
|
|
eventSub() // 订阅事件
|
|||
|
|
}
|
|||
|
|
// 订阅事件
|
|||
|
|
function eventSub() {
|
|||
|
|
qmlapp.pubSub.subscribeGroup("<<reScreenshot>>", this, "reScreenshot", ctrlKey)
|
|||
|
|
qmlapp.pubSub.subscribeGroup("<<screenshot>>", this, "screenshot", ctrlKey)
|
|||
|
|
qmlapp.pubSub.subscribeGroup("<<paste>>", this, "paste", ctrlKey)
|
|||
|
|
qmlapp.systemTray.addMenuItem("<<screenshot>>", qsTr("屏幕截图"), screenshot)
|
|||
|
|
qmlapp.systemTray.addMenuItem("<<paste>>", qsTr("粘贴图片"), paste)
|
|||
|
|
}
|
|||
|
|
// 取消订阅事件
|
|||
|
|
function eventUnsub() {
|
|||
|
|
qmlapp.pubSub.unsubscribeGroup(ctrlKey)
|
|||
|
|
qmlapp.systemTray.delMenuItem("<<screenshot>>")
|
|||
|
|
qmlapp.systemTray.delMenuItem("<<paste>>")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【python调用qml】 =========================
|
|||
|
|
|
|||
|
|
// 设置任务状态
|
|||
|
|
function setMsnState(flag) {
|
|||
|
|
msnState = flag
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取一个OCR的返回值
|
|||
|
|
function onOcrGet(res, imgID="", imgPath="") {
|
|||
|
|
// 添加到结果
|
|||
|
|
const resText = resultsTableView.addOcrResult(res)
|
|||
|
|
if(imgID) // 图片类型
|
|||
|
|
imageText.showImgID(imgID)
|
|||
|
|
else if(imgPath) // 地址类型
|
|||
|
|
imageText.showPath(imgPath)
|
|||
|
|
imageText.showTextBoxes(res)
|
|||
|
|
// 若tabPanel面板的下标没有变化过，则切换到记录页
|
|||
|
|
if(tabPanel.indexChangeNum < 2)
|
|||
|
|
tabPanel.currentIndex = 1
|
|||
|
|
// 复制到剪贴板
|
|||
|
|
const copy = configsComp.getValue("action.copy")
|
|||
|
|
if(copy && resText!="")
|
|||
|
|
qmlapp.utilsConnector.copyText(resText)
|
|||
|
|
// 弹出通知
|
|||
|
|
showSimple(res, resText, copy)
|
|||
|
|
// 升起主窗口
|
|||
|
|
popMainWindow()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 一组OCR任务完毕
|
|||
|
|
function onOcrEnd(msg) {
|
|||
|
|
if(msg.startsWith("[Error]")) {
|
|||
|
|
qmlapp.popup.message(qsTr("截图识别任务异常"), msg, "error")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【后处理】 =========================
|
|||
|
|
|
|||
|
|
// 任务完成后发送通知
|
|||
|
|
function showSimple(res, resText, isCopy) {
|
|||
|
|
// 获取弹窗类型
|
|||
|
|
let simpleType = configsComp.getValue("other.simpleNotificationType")
|
|||
|
|
if(simpleType==="default") {
|
|||
|
|
simpleType = qmlapp.globalConfigs.getValue("window.simpleNotificationType")
|
|||
|
|
}
|
|||
|
|
const code = res.code
|
|||
|
|
const time = res.time.toFixed(2)
|
|||
|
|
let title = ""
|
|||
|
|
resText = resText.replace(/\n/g, " ") // 换行符替换空格
|
|||
|
|
if(code === 100 || code === 101) { // 成功时，不发送内部弹窗
|
|||
|
|
if(simpleType==="inside" || simpleType==="onlyInside")
|
|||
|
|
if(qmlapp.mainWin.getVisibility())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if(code === 100) {
|
|||
|
|
if(isCopy) title = qsTr("已复制到剪贴板")
|
|||
|
|
else title = qsTr("识图完成")
|
|||
|
|
}
|
|||
|
|
else if(code === 101) {
|
|||
|
|
title = qsTr("无文字")
|
|||
|
|
resText = ""
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
title = qsTr("识别失败")
|
|||
|
|
}
|
|||
|
|
title += ` - ${time}s`
|
|||
|
|
qmlapp.popup.simple(title, resText, simpleType)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【布局】 =========================
|
|||
|
|
|
|||
|
|
// 左侧栏。主区域为左右双栏且左栏隐藏时，才显示左侧栏。
|
|||
|
|
Item {
|
|||
|
|
id: leftCtrlPanel
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
// 展示条件
|
|||
|
|
visible: doubleLayout.isRow && doubleLayout.hideAB === 1
|
|||
|
|
anchors.leftMargin: visible ? size_.smallSpacing : 0
|
|||
|
|
width: visible ? size_.line * 1.5 : 0
|
|||
|
|
|
|||
|
|
Column {
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.rightMargin: -size_.smallSpacing // 负间距，增加空间利用
|
|||
|
|
spacing: size_.spacing
|
|||
|
|
|
|||
|
|
Item {
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
height: width
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
IconButton {
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
height: width
|
|||
|
|
icon_: "screenshot"
|
|||
|
|
color: theme.textColor
|
|||
|
|
toolTip: qsTr("屏幕截图")
|
|||
|
|
onClicked: tabPage.screenshot()
|
|||
|
|
}
|
|||
|
|
IconButton {
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
height: width
|
|||
|
|
icon_: "paste"
|
|||
|
|
color: theme.textColor
|
|||
|
|
toolTip: qsTr("粘贴图片")
|
|||
|
|
onClicked: tabPage.paste()
|
|||
|
|
}
|
|||
|
|
IconButton {
|
|||
|
|
visible: msnState==="run"
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
height: width
|
|||
|
|
icon_: "stop"
|
|||
|
|
color: theme.noColor
|
|||
|
|
toolTip: qsTr("停止任务")
|
|||
|
|
onClicked: tabPage.msnStop()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 主区域：可切换双栏面板
|
|||
|
|
DoubleSwitchableLayout {
|
|||
|
|
id: doubleLayout
|
|||
|
|
saveKey: "ScreenshotOCR_1"
|
|||
|
|
anchors.left: leftCtrlPanel.right
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
|
|||
|
|
// 面板A：图像展示
|
|||
|
|
itemA: Panel {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
clip: true
|
|||
|
|
// 顶部控制栏
|
|||
|
|
Item {
|
|||
|
|
id: dLeftTop
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
anchors.bottomMargin: 0
|
|||
|
|
height: size_.line * 1.5
|
|||
|
|
clip: true
|
|||
|
|
|
|||
|
|
// 靠右
|
|||
|
|
Row {
|
|||
|
|
id: dLeftTopR
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
spacing: size_.smallSpacing
|
|||
|
|
|
|||
|
|
// 显示文字
|
|||
|
|
CheckButton {
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
text_: qsTr("文字")
|
|||
|
|
toolTip: qsTr("在图片上叠加显示识别文字\n可在全局设置中设为默认关闭")
|
|||
|
|
checked: imageText.showOverlay
|
|||
|
|
enabledAnime: true
|
|||
|
|
onCheckedChanged: imageText.showOverlay = checked
|
|||
|
|
}
|
|||
|
|
IconButtonBar {
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
btnList: [
|
|||
|
|
{
|
|||
|
|
icon: "menu",
|
|||
|
|
onClicked: imageText.popupMenu,
|
|||
|
|
toolTip: tr("右键菜单"),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
icon: "save",
|
|||
|
|
onClicked: imageText.saveImage,
|
|||
|
|
toolTip: tr("保存图片"),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
icon: "full_screen",
|
|||
|
|
onClicked: imageText.imageFullFit,
|
|||
|
|
toolTip: tr("图片大小：适应窗口"),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
icon: "one_to_one",
|
|||
|
|
onClicked: imageText.imageScaleAddSub,
|
|||
|
|
toolTip: tr("图片大小：实际"),
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
// 百分比显示
|
|||
|
|
Text_ {
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
verticalAlignment: Text.AlignVCenter
|
|||
|
|
horizontalAlignment: Text.AlignRight
|
|||
|
|
text: (imageText.scale*100).toFixed(0) + "%"
|
|||
|
|
color: theme.subTextColor
|
|||
|
|
width: size_.line * 2.5
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 靠左
|
|||
|
|
Rectangle { // 背景
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
width: dLeftTopL.width
|
|||
|
|
color: theme.bgColor
|
|||
|
|
Rectangle {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
color: theme.coverColor1
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
Row { // 按钮栏
|
|||
|
|
id: dLeftTopL
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
spacing: size_.smallSpacing
|
|||
|
|
|
|||
|
|
IconButtonBar {
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
btnList: [
|
|||
|
|
{
|
|||
|
|
icon: "screenshot",
|
|||
|
|
onClicked: tabPage.screenshot,
|
|||
|
|
color: theme.textColor,
|
|||
|
|
bgColor: theme.bgColor,
|
|||
|
|
text: tr("截图"),
|
|||
|
|
toolTip: tr("屏幕截图"),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
icon: "paste",
|
|||
|
|
onClicked: tabPage.paste,
|
|||
|
|
color: theme.textColor,
|
|||
|
|
bgColor: theme.bgColor,
|
|||
|
|
text: tr("粘贴"),
|
|||
|
|
toolTip: tr("粘贴图片"),
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
// 停止任务
|
|||
|
|
Button_ {
|
|||
|
|
visible: msnState==="run"
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
text_: qsTr("停止任务")
|
|||
|
|
textColor_: theme.noColor
|
|||
|
|
onClicked: tabPage.msnStop()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 图片预览区域
|
|||
|
|
ImageWithText {
|
|||
|
|
id: imageText
|
|||
|
|
anchors.top: dLeftTop.bottom
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
anchors.topMargin: size_.smallSpacing
|
|||
|
|
|
|||
|
|
// 加载中 动态图标
|
|||
|
|
Loading {
|
|||
|
|
visible: msnState==="run"
|
|||
|
|
anchors.centerIn: parent
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 提示
|
|||
|
|
DefaultTips {
|
|||
|
|
visibleFlag: msnState
|
|||
|
|
anchors.fill: parent
|
|||
|
|
tips: qsTr("截图、拖入或粘贴图片")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 面板B：结果
|
|||
|
|
itemB: Panel {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
|
|||
|
|
TabPanel {
|
|||
|
|
id: tabPanel
|
|||
|
|
anchors.fill: parent
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
isMenuTop: doubleLayout.isRow // 左右布局时，菜单在顶部；上下布局时菜单在底部
|
|||
|
|
menuHeight: size_.line * 1.5
|
|||
|
|
|
|||
|
|
// 结果面板
|
|||
|
|
ResultsTableView {
|
|||
|
|
id: resultsTableView
|
|||
|
|
anchors.fill: parent
|
|||
|
|
visible: false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tabsModel: [
|
|||
|
|
{
|
|||
|
|
"key": "configs",
|
|||
|
|
"title": qsTr("设置"),
|
|||
|
|
"component": configsComp.panelComponent,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"key": "ocrResult",
|
|||
|
|
"title": qsTr("记录"),
|
|||
|
|
"component": resultsTableView,
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 鼠标拖入图片
|
|||
|
|
DropArea_ {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
callback: tabPage.ocrPaths
|
|||
|
|
}
|
|||
|
|
}
|