312 lines
9.7 KiB
QML
312 lines
9.7 KiB
QML
|
// ==============================================
|
|||
|
|
// =============== 功能页：批量OCR ===============
|
|||
|
|
// ==============================================
|
|||
|
|
|
|||
|
|
import QtQuick 2.15
|
|||
|
|
import QtQuick.Controls 2.15
|
|||
|
|
|
|||
|
|
import ".."
|
|||
|
|
import "../../Widgets"
|
|||
|
|
import "../../Widgets/ResultLayout"
|
|||
|
|
import "../../Widgets/IgnoreArea"
|
|||
|
|
|
|||
|
|
TabPage {
|
|||
|
|
id: tabPage
|
|||
|
|
|
|||
|
|
// ========================= 【逻辑】 =========================
|
|||
|
|
|
|||
|
|
property int errorNum: 0 // 异常的任务个数
|
|||
|
|
property string msnID: "" // 当前任务ID
|
|||
|
|
|
|||
|
|
Component.onCompleted: {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 异步加载一批图像路径
|
|||
|
|
function addImages(paths) {
|
|||
|
|
if(ctrlPanel.state_ !== "stop") return
|
|||
|
|
if(paths.length <= 0) return
|
|||
|
|
const isRecurrence = configsComp.getValue("mission.recurrence")
|
|||
|
|
qmlapp.asynFilesLoader.run(paths,"image",isRecurrence,onAddImages)
|
|||
|
|
}
|
|||
|
|
// 完毕后，将合法路径加入表格
|
|||
|
|
function onAddImages(paths) {
|
|||
|
|
for(let i in paths) {
|
|||
|
|
filesTableView.add({ path: paths[i], time: "", state: "" })
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 运行OCR
|
|||
|
|
function ocrStart() {
|
|||
|
|
let msnLength = filesTableView.rowCount
|
|||
|
|
if(msnLength <= 0) {
|
|||
|
|
ctrlPanel.stopFinished()
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 刷新表格
|
|||
|
|
for(let i = 0; i < msnLength; i++) {
|
|||
|
|
filesTableView.set(i, { time: "", state: qsTr("排队") })
|
|||
|
|
}
|
|||
|
|
// 刷新计数
|
|||
|
|
errorNum = 0 // 异常任务个数
|
|||
|
|
// 开始运行
|
|||
|
|
const paths = filesTableView.getColumnsValue("path")
|
|||
|
|
const argd = configsComp.getValueDict()
|
|||
|
|
msnID = tabPage.callPy("msnPaths", paths, argd)
|
|||
|
|
// 若tabPanel面板的下标没有变化过，则切换到记录页
|
|||
|
|
if(tabPanel.indexChangeNum < 2)
|
|||
|
|
tabPanel.currentIndex = 1
|
|||
|
|
ctrlPanel.runFinished(msnLength)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 停止OCR
|
|||
|
|
function ocrStop() {
|
|||
|
|
_ocrStop()
|
|||
|
|
tabPage.callPy("msnStop")
|
|||
|
|
ctrlPanel.stopFinished()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function _ocrStop() {
|
|||
|
|
msnID = "" // 清除任务ID
|
|||
|
|
// 刷新表格，清空未执行的任务的状态
|
|||
|
|
let msnLength = filesTableView.rowCount
|
|||
|
|
for(let i = 0; i < msnLength; i++) {
|
|||
|
|
const row = filesTableView.get(i)
|
|||
|
|
if(row.time === "") {
|
|||
|
|
filesTableView.setProperty(i, "state", "")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭页面
|
|||
|
|
function closePage() {
|
|||
|
|
if(ctrlPanel.state_ !== "stop") {
|
|||
|
|
const argd = { yesText: qsTr("依然关闭") }
|
|||
|
|
const callback = (flag)=>{
|
|||
|
|
if(flag) {
|
|||
|
|
ocrStop()
|
|||
|
|
delPage()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
qmlapp.popup.dialog("", qsTr("任务正在进行中。\n要结束任务并关闭页面吗？"), callback, "warning", argd)
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
delPage()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 预览
|
|||
|
|
function msnPreview(path) {
|
|||
|
|
const argd = configsComp.getValueDict()
|
|||
|
|
tabPage.callPy("msnPreview", path, argd)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【python调用qml】 =========================
|
|||
|
|
|
|||
|
|
// 准备开始一个任务
|
|||
|
|
function onOcrReady(path) {
|
|||
|
|
// 刷新表格显示
|
|||
|
|
filesTableView.setProperty(path, "state", qsTr("处理"))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取一个OCR的返回值
|
|||
|
|
function onOcrGet(path, res) {
|
|||
|
|
const time = res.time.toFixed(2)
|
|||
|
|
let state = ""
|
|||
|
|
switch(res.code){
|
|||
|
|
case 100:
|
|||
|
|
state = "√ "+res.score.toFixed(2);break
|
|||
|
|
case 101:
|
|||
|
|
state = "√ ---- ";break
|
|||
|
|
default:
|
|||
|
|
state = "× "+res.code
|
|||
|
|
errorNum++ // 异常任务数量+1
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
// 刷新表格显示
|
|||
|
|
filesTableView.set(path, { "time": time, "state": state })
|
|||
|
|
// 提取文字，添加到结果表格
|
|||
|
|
res.title = res.fileName
|
|||
|
|
resultsTableView.addOcrResult(res)
|
|||
|
|
ctrlPanel.msnStep(1) // 任务计数器步进
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 任务队列完毕
|
|||
|
|
function onOcrEnd(msg, thisMsnID) {
|
|||
|
|
// msg: [Success] [Warning] [Error]
|
|||
|
|
if(msnID !== thisMsnID) { // 返回的任务ID不等于前端展示的任务ID，则不处理
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
_ocrStop()
|
|||
|
|
// 任务成功
|
|||
|
|
if(msg.startsWith("[Success]")) {
|
|||
|
|
let errMsg = ""
|
|||
|
|
if(errorNum > 0) { // 有异常任务
|
|||
|
|
errMsg = qsTr("%1 张图片识别失败！").arg(errorNum)
|
|||
|
|
}
|
|||
|
|
const simpleType = configsComp.getValue("other.simpleNotificationType")
|
|||
|
|
qmlapp.popup.simple(qsTr("批量识别完成"), errMsg, simpleType)
|
|||
|
|
// 任务完成后续操作
|
|||
|
|
qmlapp.globalConfigs.utilsDicts.postTaskHardwareCtrl(
|
|||
|
|
configsComp.getValue("postTaskActions.system")
|
|||
|
|
)
|
|||
|
|
}
|
|||
|
|
// 任务失败
|
|||
|
|
else if(msg.startsWith("[Error]")) {
|
|||
|
|
qmlapp.popup.message(qsTr("批量识别任务异常"), msg, "error")
|
|||
|
|
}
|
|||
|
|
ctrlPanel.stopFinished()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 预览
|
|||
|
|
function onPreview(path, res) {
|
|||
|
|
ignoreArea.getPreview(res, path, "")
|
|||
|
|
}
|
|||
|
|
// 路径转文件名
|
|||
|
|
function path2name(path) {
|
|||
|
|
const parts = path.split("/")
|
|||
|
|
return parts[parts.length - 1]
|
|||
|
|
}
|
|||
|
|
// 文件表格中单击路径
|
|||
|
|
function onClickPath(index) {
|
|||
|
|
let info = filesTableView.get(index)
|
|||
|
|
let fileName = path2name(info.path)
|
|||
|
|
let res = resultsTableView.getResult(fileName)
|
|||
|
|
let data = undefined, text = undefined
|
|||
|
|
if(res) {
|
|||
|
|
if(res.source)
|
|||
|
|
data = JSON.parse(res.source)
|
|||
|
|
if(res.resText)
|
|||
|
|
text = res.resText
|
|||
|
|
}
|
|||
|
|
previewImage.show(info.path, data, text)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ========================= 【布局】 =========================
|
|||
|
|
|
|||
|
|
// 配置
|
|||
|
|
configsComp: BatchOCRConfigs {
|
|||
|
|
// 点按钮打开忽略区域
|
|||
|
|
onClickIgnoreArea: {
|
|||
|
|
if(filesTableView.rowCount > 0) {
|
|||
|
|
const path = filesTableView.get(0).path
|
|||
|
|
console.log("打开路径", path)
|
|||
|
|
ignoreArea.showPath(path)
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
ignoreArea.show()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 主区域：可切换双栏面板
|
|||
|
|
DoubleSwitchableLayout {
|
|||
|
|
id: doubleLayout
|
|||
|
|
saveKey: "BatchOCR_1"
|
|||
|
|
anchors.fill: parent
|
|||
|
|
|
|||
|
|
// 面板A：控制板+文件表格
|
|||
|
|
itemA: Panel {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
|
|||
|
|
// 上方控制板
|
|||
|
|
MissionCtrlPanel {
|
|||
|
|
id: ctrlPanel
|
|||
|
|
anchors.top: parent.top
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
height: size_.line * 2
|
|||
|
|
|
|||
|
|
onRunClicked: tabPage.ocrStart()
|
|||
|
|
onPauseClicked: {
|
|||
|
|
tabPage.callPy("msnPause")
|
|||
|
|
pauseFinished()
|
|||
|
|
}
|
|||
|
|
onResumeClicked: {
|
|||
|
|
tabPage.callPy("msnResume")
|
|||
|
|
resumeFinished()
|
|||
|
|
}
|
|||
|
|
onStopClicked: tabPage.ocrStop()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 下方文件表格
|
|||
|
|
FilesTableView {
|
|||
|
|
id: filesTableView
|
|||
|
|
anchors.top: ctrlPanel.bottom
|
|||
|
|
anchors.left: parent.left
|
|||
|
|
anchors.right: parent.right
|
|||
|
|
anchors.bottom: parent.bottom
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
anchors.topMargin: size_.smallSpacing
|
|||
|
|
headers: [
|
|||
|
|
{key: "path", title: qsTr("图片"), left: true, display:path2name,
|
|||
|
|
btn: true, onClicked:onClickPath},
|
|||
|
|
{key: "time", title: qsTr("耗时"), },
|
|||
|
|
{key: "state", title: qsTr("状态"), },
|
|||
|
|
]
|
|||
|
|
openBtnText: qsTr("打开图片")
|
|||
|
|
clearBtnText: qsTr("清空")
|
|||
|
|
defaultTips: qsTr("拖入图片或文件夹")
|
|||
|
|
fileDialogTitle: qsTr("请选择图片")
|
|||
|
|
fileDialogNameFilters: [qsTr("图片")+" (*.jpg *.jpe *.jpeg *.jfif *.png *.webp *.bmp *.tif *.tiff)"]
|
|||
|
|
isLock: ctrlPanel.state_ !== "stop"
|
|||
|
|
onAddPaths: {
|
|||
|
|
tabPage.addImages(paths)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// 面板B：文字输出 & 设置
|
|||
|
|
itemB: Panel {
|
|||
|
|
anchors.fill: parent
|
|||
|
|
|
|||
|
|
// 配置项控制板
|
|||
|
|
TabPanel {
|
|||
|
|
id: tabPanel
|
|||
|
|
anchors.fill: parent
|
|||
|
|
anchors.margins: size_.spacing
|
|||
|
|
isMenuTop: doubleLayout.isRow // 左右布局时，菜单在顶部；上下布局时菜单在底部
|
|||
|
|
menuHeight: size_.line * 2
|
|||
|
|
|
|||
|
|
// 结果面板
|
|||
|
|
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.addImages
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 预览面板
|
|||
|
|
PreviewImage {
|
|||
|
|
id: previewImage
|
|||
|
|
anchors.fill: parent
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 忽略区域编辑器
|
|||
|
|
IgnoreArea {
|
|||
|
|
id: ignoreArea
|
|||
|
|
anchors.fill: parent
|
|||
|
|
pathPreview: msnPreview
|
|||
|
|
configsComp: tabPage.configsComp
|
|||
|
|
configKey: "tbpu.ignoreArea"
|
|||
|
|
}
|
|||
|
|
}
|