18660ef51a
Description: 增加Dbus接口，可以传递打开Image的名称 Log: no Change-Id: I964a0f28504f953322fd103ba5075e1ed582a2b0
453 lines
14 KiB
C++
453 lines
14 KiB
C++
#include "mainwidget.h"
|
||
#include "view/imageview.h"
|
||
#include "loadingwidget.h"
|
||
|
||
#include <QtCore/QVariant>
|
||
#include <QtWidgets/QApplication>
|
||
#include <QtWidgets/QGridLayout>
|
||
#include <QtWidgets/QHBoxLayout>
|
||
#include <QTextBlock>
|
||
#include <QStandardPaths>
|
||
#include <QFileDialog>
|
||
#include <QFileInfo>
|
||
#include <QThread>
|
||
#include <QMutexLocker>
|
||
#include <QSplitter>
|
||
#include <QTimer>
|
||
|
||
#include <DGuiApplicationHelper>
|
||
#include <DMainWindow>
|
||
#include <DTitlebar>
|
||
#include <DMessageManager>
|
||
#include <DToolButton>
|
||
|
||
#define App (static_cast<QApplication*>(QCoreApplication::instance()))
|
||
MainWidget::MainWidget(QWidget *parent) :
|
||
DWidget(parent)
|
||
{
|
||
setupUi(this);
|
||
DGuiApplicationHelper::ColorType themeType = DGuiApplicationHelper::instance()->themeType();
|
||
setIcons(themeType);
|
||
|
||
setupConnect();
|
||
}
|
||
|
||
MainWidget::~MainWidget()
|
||
{
|
||
// m_mainGridLayout->addLayout(m_buttonHorizontalLayout, 1, 0, 1, 1);
|
||
if (m_mainGridLayout && m_buttonHorizontalLayout) {
|
||
m_mainGridLayout->removeItem(m_buttonHorizontalLayout);
|
||
}
|
||
if (m_imageview) {
|
||
m_imageview->deleteLater();
|
||
m_imageview = nullptr;
|
||
}
|
||
if (m_mainGridLayout) {
|
||
m_mainGridLayout->deleteLater();
|
||
m_mainGridLayout = nullptr;
|
||
}
|
||
if (m_horizontalLayout) {
|
||
m_horizontalLayout->deleteLater();
|
||
m_horizontalLayout = nullptr;
|
||
}
|
||
if (m_plainTextEdit) {
|
||
m_plainTextEdit->deleteLater();
|
||
m_plainTextEdit = nullptr;
|
||
}
|
||
if (m_buttonHorizontalLayout) {
|
||
m_buttonHorizontalLayout->deleteLater();
|
||
m_buttonHorizontalLayout = nullptr;
|
||
}
|
||
if (m_tipHorizontalLayout) {
|
||
m_tipHorizontalLayout->deleteLater();
|
||
m_tipHorizontalLayout = nullptr;
|
||
}
|
||
if (m_tiplabel) {
|
||
m_tiplabel->deleteLater();
|
||
m_tiplabel = nullptr;
|
||
}
|
||
if (m_copyBtn) {
|
||
m_copyBtn->deleteLater();
|
||
m_copyBtn = nullptr;
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->deleteLater();
|
||
m_exportBtn = nullptr;
|
||
}
|
||
if (m_tipIconLabel) {
|
||
m_tipIconLabel->deleteLater();
|
||
m_tipIconLabel = nullptr;
|
||
}
|
||
|
||
}
|
||
|
||
void MainWidget::setupUi(QWidget *Widget)
|
||
{
|
||
|
||
QPalette pal;
|
||
pal.setColor(QPalette::Background, QColor(255, 255, 255, 179));
|
||
setAutoFillBackground(true);
|
||
setPalette(pal);
|
||
DMainWindow *mainWindow = static_cast<DMainWindow *>(this->parent());
|
||
if (mainWindow) {
|
||
mainWindow->titlebar()->setMenuVisible(false);
|
||
}
|
||
m_mainGridLayout = new QGridLayout(Widget);
|
||
m_mainGridLayout->setSpacing(6);
|
||
m_mainGridLayout->setContentsMargins(0, 0, 0, 6);
|
||
m_mainGridLayout->setObjectName(QStringLiteral("gridLayout"));
|
||
|
||
m_horizontalLayout = new QHBoxLayout(Widget);
|
||
m_horizontalLayout->setSpacing(0);
|
||
m_horizontalLayout->setObjectName(QStringLiteral("horizontalLayout"));
|
||
|
||
m_plainTextEdit = new ResultTextView(Widget);
|
||
m_plainTextEdit->setReadOnly(true);
|
||
|
||
m_plainTextEdit->setObjectName(QStringLiteral("plainTextEdit"));
|
||
|
||
|
||
if (!m_imageview) {
|
||
m_imageview = new ImageView();
|
||
}
|
||
|
||
m_resultWidget = new DStackedWidget(this);
|
||
m_resultWidget->setFocusPolicy(Qt::NoFocus);
|
||
m_resultWidget->setMinimumWidth(220);
|
||
|
||
|
||
m_loadingOcr = new loadingWidget(this);
|
||
m_loadingOcr->setAlignment(Qt::AlignCenter);
|
||
m_resultWidget->addWidget(m_loadingOcr);
|
||
|
||
m_resultWidget->addWidget(m_plainTextEdit);
|
||
|
||
m_noResult = new DLabel(this);
|
||
m_noResult->setAlignment(Qt::AlignCenter);
|
||
m_noResult->setText(tr("No text recognized"));
|
||
m_resultWidget->addWidget(m_noResult);
|
||
|
||
connect(m_loadingOcr, &loadingWidget::sigChangeSize, [ = ] {
|
||
loadingUi();
|
||
});
|
||
QSplitter *mainSplitter = new QSplitter(Qt::Horizontal); //新建水平分割器
|
||
mainSplitter->setHandleWidth(0);//分割线的宽度
|
||
mainSplitter->setChildrenCollapsible(false);//不允许把分割出的子窗口拖小到0，最小值被限定为sizeHint或maxSize/minSize
|
||
mainSplitter->addWidget(m_imageview);//把ui中拖出的各个控件拿走，放到分割器里面
|
||
mainSplitter->addWidget(m_resultWidget);
|
||
QList<int> list;
|
||
list << 600 << 250;
|
||
mainSplitter->setSizes(list);
|
||
// m_horizontalLayout->addWidget(m_imageview);
|
||
// m_horizontalLayout->addWidget(m_plainTextEdit);
|
||
m_horizontalLayout->addWidget(mainSplitter);
|
||
m_horizontalLayout->setStretch(0, 1);
|
||
|
||
m_mainGridLayout->addLayout(m_horizontalLayout, 0, 0, 1, 1);
|
||
m_mainGridLayout->setColumnStretch(0, 1);
|
||
|
||
m_buttonHorizontalLayout = new QHBoxLayout(Widget);
|
||
m_buttonHorizontalLayout->setContentsMargins(20, 0, 59, 0); //表示控件与窗体的左右边距
|
||
// m_buttonHorizontalLayout->setSpacing(30);
|
||
|
||
|
||
m_tipHorizontalLayout = new QHBoxLayout(Widget);
|
||
|
||
m_tipIconLabel = new DLabel(Widget);
|
||
m_tipIconLabel->setObjectName(QStringLiteral("tipIconLabel"));
|
||
|
||
m_tipHorizontalLayout->addWidget(m_tipIconLabel);
|
||
|
||
m_tiplabel = new DLabel(Widget);
|
||
m_tiplabel->setObjectName(QStringLiteral("tiplabel"));
|
||
|
||
m_tipHorizontalLayout->setSpacing(4);
|
||
m_tipHorizontalLayout->addWidget(m_tiplabel);
|
||
|
||
m_buttonHorizontalLayout->addLayout(m_tipHorizontalLayout);
|
||
QSpacerItem *horizontalSpacer = new QSpacerItem(159, 20, QSizePolicy::Expanding, QSizePolicy::Minimum);
|
||
m_buttonHorizontalLayout->addItem(horizontalSpacer);
|
||
|
||
m_copyBtn = new DToolButton(Widget);
|
||
m_copyBtn->setObjectName(QStringLiteral("Copy text"));
|
||
m_copyBtn->setMaximumSize(QSize(40, 40));
|
||
m_copyBtn->setToolTip(tr("Copy text"));
|
||
|
||
m_buttonHorizontalLayout->addWidget(m_copyBtn);
|
||
m_buttonHorizontalLayout->setSpacing(20);
|
||
m_exportBtn = new DToolButton(Widget);
|
||
m_exportBtn->setObjectName(QStringLiteral("Save as TXT"));
|
||
m_exportBtn->setMaximumSize(QSize(40, 40));
|
||
m_exportBtn->setToolTip(tr("Save as TXT"));
|
||
|
||
m_buttonHorizontalLayout->addWidget(m_exportBtn);
|
||
|
||
|
||
m_mainGridLayout->addLayout(m_buttonHorizontalLayout, 1, 0, 1, 1);
|
||
|
||
|
||
retranslateUi(Widget);
|
||
|
||
QMetaObject::connectSlotsByName(Widget);
|
||
m_pwidget = new QWidget(this);
|
||
m_pwidget->setFocusPolicy(Qt::NoFocus);
|
||
m_pwidget->setAttribute(Qt::WA_TransparentForMouseEvents);
|
||
m_pwidget->setFixedSize(this->width(), this->height() - 23);
|
||
m_pwidget->move(0, 0);
|
||
|
||
}
|
||
|
||
void MainWidget::setupConnect()
|
||
{
|
||
connect(DGuiApplicationHelper::instance(), &DGuiApplicationHelper::paletteTypeChanged, this, &MainWidget::setIcons);
|
||
connect(m_exportBtn, &DIconButton::clicked, this, &MainWidget::slotExport);
|
||
connect(m_copyBtn, &DIconButton::clicked, this, &MainWidget::slotCopy);
|
||
connect(this, &MainWidget::sigResult, this, [ = ](const QString & result) {
|
||
loadString(result);
|
||
deleteLoadingUi();
|
||
});
|
||
}
|
||
|
||
void MainWidget::retranslateUi(QWidget *Widget)
|
||
{
|
||
Widget->setWindowTitle(QApplication::translate("Widget", "Widget", nullptr));
|
||
m_tiplabel->setText(QApplication::translate("Widget", "Tips: The clearer the image is, the more accurate the text is", nullptr));
|
||
// m_copyBtn->setText(QApplication::translate("Widget", "Copy", nullptr));
|
||
// m_exportBtn->setText(QApplication::translate("Widget", "Export", nullptr));
|
||
}
|
||
|
||
void MainWidget::createLoadingUi()
|
||
{
|
||
m_isLoading = true;
|
||
m_loadingWidget = new TextLoadWidget(this);
|
||
m_loadingWidget->resize(40, 40);
|
||
m_loadingWidget->show();
|
||
m_loadingTip = new DLabel(tr("Recognizing"), this);
|
||
m_loadingTip->show();
|
||
if (m_imageview) {
|
||
m_imageview->setForegroundBrush(QColor(0, 0, 0, 77)); //设置场景的前景色，类似于遮罩
|
||
}
|
||
//识别时
|
||
if (m_copyBtn) {
|
||
m_copyBtn->setEnabled(false);
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->setEnabled(false);
|
||
}
|
||
}
|
||
|
||
void MainWidget::deleteLoadingUi()
|
||
{
|
||
m_isLoading = false;
|
||
if (m_loadingWidget) {
|
||
m_loadingWidget->deleteLater();
|
||
m_loadingWidget = nullptr;
|
||
}
|
||
if (m_loadingTip) {
|
||
m_loadingTip->deleteLater();
|
||
m_loadingTip = nullptr;
|
||
}
|
||
m_imageview->setForegroundBrush(QColor(0, 0, 0, 0)); //设置场景的前景色，类似于遮罩
|
||
}
|
||
|
||
void MainWidget::loadingUi()
|
||
{
|
||
if (m_loadingWidget && m_loadingTip && m_resultWidget) {
|
||
int x = this->width() - m_resultWidget->width() / 2;
|
||
int y = this->height() / 2 - 50;
|
||
m_loadingWidget->move(x, y);
|
||
m_loadingTip->move(x - 20, y + 24);
|
||
}
|
||
if (m_pwidget) {
|
||
m_pwidget->setFixedSize(this->width(), this->height() - 48);
|
||
m_pwidget->move(0, 0);
|
||
}
|
||
}
|
||
|
||
void MainWidget::openImage(const QString &path)
|
||
{
|
||
if (m_imageview) {
|
||
QImage img(path);
|
||
m_imgName = path;
|
||
openImage(img);
|
||
}
|
||
}
|
||
|
||
void MainWidget::openImage(const QImage &img,const QString & name)
|
||
{
|
||
createLoadingUi();
|
||
if (m_imageview) {
|
||
m_imageview->openFilterImage(img);
|
||
QTimer::singleShot(100, [ = ] {
|
||
m_imageview->fitWindow();
|
||
});
|
||
m_imgName = name;
|
||
}
|
||
if (m_currentImg) {
|
||
delete m_currentImg;
|
||
m_currentImg = nullptr;
|
||
}
|
||
m_currentImg = new QImage(img);
|
||
if (!m_loadImagethread) {
|
||
m_loadImagethread = QThread::create([ = ]() {
|
||
QMutexLocker locker(&m_mutex);
|
||
m_result = TessOcrUtils::instance()->getRecogitionResult(m_currentImg, ResultType::RESULT_STRING);
|
||
emit sigResult(m_result.result);
|
||
});
|
||
}
|
||
connect(m_loadImagethread, &QThread::finished, m_loadImagethread, &QObject::deleteLater);
|
||
m_loadImagethread->start();
|
||
}
|
||
|
||
void MainWidget::loadHtml(const QString &html)
|
||
{
|
||
if (!html.isEmpty()) {
|
||
m_resultWidget->setCurrentWidget(m_plainTextEdit);
|
||
m_plainTextEdit->appendHtml(html);
|
||
} else {
|
||
resultEmpty();
|
||
}
|
||
}
|
||
|
||
void MainWidget::loadString(const QString &string)
|
||
{
|
||
if (!string.isEmpty()) {
|
||
m_resultWidget->setCurrentWidget(m_plainTextEdit);
|
||
m_plainTextEdit->appendPlainText(string);
|
||
//新增识别完成按钮恢复
|
||
if (m_copyBtn) {
|
||
m_copyBtn->setEnabled(true);
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->setEnabled(true);
|
||
}
|
||
} else {
|
||
resultEmpty();
|
||
}
|
||
}
|
||
|
||
void MainWidget::resultEmpty()
|
||
{
|
||
m_resultWidget->setCurrentWidget(m_noResult);
|
||
//新增如果未识别到，按钮置灰
|
||
if (m_copyBtn) {
|
||
m_copyBtn->setEnabled(false);
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->setEnabled(false);
|
||
}
|
||
}
|
||
|
||
void MainWidget::resizeEvent(QResizeEvent *event)
|
||
{
|
||
return DWidget::resizeEvent(event);
|
||
}
|
||
|
||
|
||
void MainWidget::slotCopy()
|
||
{
|
||
//选中内容则复制，未选中内容则不复制
|
||
if (!m_plainTextEdit->textCursor().selectedText().isEmpty()) {
|
||
m_plainTextEdit->copy();
|
||
} else {
|
||
QTextDocument *document = m_plainTextEdit->document();
|
||
DPlainTextEdit *tempTextEdit = new DPlainTextEdit(this);
|
||
tempTextEdit->setDocument(document);
|
||
tempTextEdit->selectAll();
|
||
tempTextEdit->copy();
|
||
tempTextEdit->deleteLater();
|
||
tempTextEdit = nullptr;
|
||
}
|
||
|
||
QIcon icon(":/assets/icon_toast_sucess_new.svg");
|
||
DFloatingMessage *pDFloatingMessage = new DFloatingMessage(DFloatingMessage::MessageType::TransientType, m_pwidget);
|
||
pDFloatingMessage->setBlurBackgroundEnabled(true);
|
||
pDFloatingMessage->setMessage(tr("Copied"));
|
||
pDFloatingMessage->setIcon(icon);
|
||
pDFloatingMessage->raise();
|
||
DMessageManager::instance()->sendMessage(m_pwidget, pDFloatingMessage);
|
||
|
||
}
|
||
|
||
void MainWidget::slotExport()
|
||
{
|
||
QStringList list = QStandardPaths::standardLocations(QStandardPaths::DownloadLocation);
|
||
QString download = "";
|
||
if (list.size() > 0) {
|
||
download = list.at(0);
|
||
} else {
|
||
QStringList home = QStandardPaths::standardLocations(QStandardPaths::HomeLocation);
|
||
if (home.size() > 0) {
|
||
download = home.at(0) + "/Downloads";
|
||
}
|
||
}
|
||
|
||
QString fileName;
|
||
if (!m_imgName.isEmpty()) {
|
||
fileName = QFileInfo(m_imgName).completeBaseName();
|
||
} else {
|
||
fileName = tr("Results");
|
||
}
|
||
|
||
QString file_path = QFileDialog::getSaveFileName(this, "save as", download + "/" + fileName, "*.txt");
|
||
qDebug() << file_path;
|
||
|
||
QString path = file_path + ".txt";
|
||
QFile file(path);
|
||
if (file.open(QFile::WriteOnly | QFile::Text)) {
|
||
QTextStream out(&file);
|
||
out << m_plainTextEdit->document()->toPlainText();
|
||
}
|
||
}
|
||
|
||
void MainWidget::setIcons(DGuiApplicationHelper::ColorType themeType)
|
||
{
|
||
if (themeType == DGuiApplicationHelper::DarkType) {
|
||
if (m_tipIconLabel) {
|
||
m_tipIconLabel->setPixmap(QPixmap(":/assets/tip_dark.svg"));
|
||
m_tipIconLabel->setFixedSize(QSize(14, 14));
|
||
}
|
||
|
||
if (m_copyBtn) {
|
||
m_copyBtn->setIcon(QIcon(":/assets/copy_dark.svg"));
|
||
m_copyBtn->setIconSize(QSize(36, 36));
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->setIcon(QIcon(":/assets/download_dark.svg"));
|
||
m_exportBtn->setIconSize(QSize(36, 36));
|
||
}
|
||
|
||
App->setWindowIcon(QIcon(":/assets/appicon_dark.svg"));
|
||
DMainWindow *mainWindow = static_cast<DMainWindow *>(this->parent());
|
||
if (mainWindow) {
|
||
mainWindow->titlebar()->setIcon(QIcon(":/assets/appicon_dark.svg"));
|
||
}
|
||
if (m_isLoading && m_imageview) {
|
||
m_imageview->setForegroundBrush(QColor(0, 0, 0, 150)); //设置场景的前景色，类似于遮罩
|
||
}
|
||
} else {
|
||
if (m_tipIconLabel) {
|
||
m_tipIconLabel->setPixmap(QPixmap(":/assets/tip_light.svg"));
|
||
m_tipIconLabel->setFixedSize(QSize(14, 14));
|
||
}
|
||
|
||
if (m_copyBtn) {
|
||
m_copyBtn->setIcon(QIcon(":/assets/copy_light.svg"));
|
||
m_copyBtn->setIconSize(QSize(36, 36));
|
||
}
|
||
if (m_exportBtn) {
|
||
m_exportBtn->setIcon(QIcon(":/assets/download_light.svg"));
|
||
m_exportBtn->setIconSize(QSize(36, 36));
|
||
}
|
||
|
||
App->setWindowIcon(QIcon(":/assets/appicon_light.svg"));
|
||
DMainWindow *mainWindow = static_cast<DMainWindow *>(this->parent());
|
||
if (mainWindow) {
|
||
mainWindow->titlebar()->setIcon(QIcon(":/assets/appicon_light.svg"));
|
||
}
|
||
if (m_isLoading && m_imageview) {
|
||
m_imageview->setForegroundBrush(QColor(255, 255, 255, 150)); //设置场景的前景色，类似于遮罩
|
||
}
|
||
}
|
||
} // setupUi
|