263 lines
6.0 KiB
C++
263 lines
6.0 KiB
C++
/*
|
||
|
||
* Copyright (C) 2019 ~ 2019 Deepin Technology Co., Ltd.
|
||
|
||
*
|
||
|
||
* Author: wangcong <wangcong@uniontech.com>
|
||
|
||
*
|
||
|
||
* Maintainer: wangcong <wangcong@uniontech.com>
|
||
|
||
*
|
||
|
||
* This program is free software: you can redistribute it and/or modify
|
||
|
||
* it under the terms of the GNU General Public License as published by
|
||
|
||
* the Free Software Foundation, either version 3 of the License, or
|
||
|
||
* any later version.
|
||
|
||
*
|
||
|
||
* This program is distributed in the hope that it will be useful,
|
||
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
||
* GNU General Public License for more details.
|
||
|
||
*
|
||
|
||
* You should have received a copy of the GNU General Public License
|
||
|
||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
||
*/
|
||
#ifndef TESSOCRUTILS_H
|
||
#define TESSOCRUTILS_H
|
||
//本项目的文件
|
||
|
||
//dtk的类
|
||
|
||
//qt的类
|
||
#include <QDebug>
|
||
#include <QException>
|
||
#include <QFileInfo>
|
||
#include <QImage>
|
||
//其他库
|
||
#include <tesseract/baseapi.h>
|
||
#include <leptonica/allheaders.h>
|
||
|
||
|
||
/**
|
||
* @brief 返回结果字符串的类型
|
||
*/
|
||
enum ResultType{
|
||
RESULT_STRING = 1, //纯字符串结果
|
||
RESULT_HTML = 2, //HTML文本
|
||
UNKNOWN_TYPE = -1 //XML文本
|
||
};
|
||
|
||
/**
|
||
* @brief 返回识别过程中的错误码
|
||
*/
|
||
enum ErrorCode{
|
||
OK = 1, //成功
|
||
UNKNOWN = -1, //未知错误
|
||
OCR_P_NULL = 101, //文件路径为空
|
||
OCR_RT_NULL = 102, //结果字符串类型不存在
|
||
OCR_INI_F = 103, //ocr三方库初始化失败
|
||
OCR_LI_F = 104, //OCR加载图片失败
|
||
OCR_RI_F = 105 //OCR识别图片失败
|
||
};
|
||
|
||
/**
|
||
* @brief 现今支持的语言包类型
|
||
*/
|
||
enum Languages{
|
||
UNKNOWN_LAN = -1, //未知语言
|
||
CHI_SIM=1, //简体中文
|
||
CHI_TRA=2, //繁体中文
|
||
ENG=3 //英文
|
||
};
|
||
|
||
/**
|
||
* @brief ocr识别的返回结果
|
||
*/
|
||
struct RecognitionResult{
|
||
|
||
/**
|
||
* @brief 识别是否成功
|
||
*/
|
||
bool flag;
|
||
|
||
/**
|
||
* @brief 识别返回的消息，如果识别失败发生的错误通过此字段返回
|
||
*/
|
||
QString message;
|
||
|
||
/**
|
||
* @brief 返回的错误码
|
||
*/
|
||
ErrorCode errorCode;
|
||
|
||
/**
|
||
* @brief 返回字符串结果的类型
|
||
*/
|
||
ResultType resultType;
|
||
|
||
/**
|
||
* @brief 返回的字符串结果
|
||
*/
|
||
QString result;
|
||
|
||
RecognitionResult(){
|
||
flag = false;
|
||
message.clear();
|
||
errorCode = ErrorCode::UNKNOWN;
|
||
resultType = ResultType::RESULT_STRING;
|
||
result.clear();
|
||
}
|
||
};
|
||
|
||
/**
|
||
* @brief ocr接口工具
|
||
* 使用方法: *
|
||
* 1.获取识别结果 TessOcrUtils::instance()->getRecogitionResult(t_image);
|
||
*/
|
||
class TessOcrUtils
|
||
{
|
||
public:
|
||
TessOcrUtils();
|
||
~TessOcrUtils();
|
||
static TessOcrUtils *instance();
|
||
|
||
/**
|
||
* @brief 传入待识别图片的路径和想得到的返回结果类型，获取识别结果
|
||
* @param 图片路径
|
||
* @param 返回的字符串结果类型
|
||
* @return resultType类型的字符串结果
|
||
*/
|
||
RecognitionResult getRecogitionResult(const QString &imagePath,const ResultType &resultType);
|
||
|
||
/**
|
||
* @brief 传入待识别图片的路径，获取纯字符串的识别结果
|
||
* @param 图片路径
|
||
* @return 识别的字符串结果
|
||
*/
|
||
RecognitionResult getRecogitionResult(const QString &imagePath);
|
||
|
||
/**
|
||
* @brief 传入待识别图片和想得到的返回结果类型，获取识别结果
|
||
* @param 图片路径
|
||
* @param 返回的字符串结果类型
|
||
* @return resultType类型的字符串结果
|
||
*/
|
||
RecognitionResult getRecogitionResult(QImage *image,const ResultType &resultType);
|
||
|
||
/**
|
||
* @brief 传入待识别图片，获取纯字符串的识别结果
|
||
* @param 图片路径
|
||
* @return 识别的字符串结果
|
||
*/
|
||
RecognitionResult getRecogitionResult(QImage *image);
|
||
|
||
/**
|
||
* @brief 是否正在识别中
|
||
* @return true：正在识别中 false：识别结束
|
||
*/
|
||
bool isRunning();
|
||
|
||
private :
|
||
|
||
/**
|
||
* @brief 设置当前使用的语言包
|
||
* @param 需要使用的语言包
|
||
* @return 是否成功
|
||
*/
|
||
//bool setLanguages(const QList<Languages> langs);
|
||
|
||
/**
|
||
* @brief 设置当前使用的语言包路径
|
||
* @param 语言包路径
|
||
* @return
|
||
*/
|
||
bool setLanguagesPath(const QString langsPath);
|
||
|
||
/**
|
||
* @brief 判断指定的结果类型是否存在
|
||
* @param 指定的类型
|
||
* @return 存在返回true
|
||
*/
|
||
bool isExistsResultType(ResultType resultType);
|
||
|
||
/**
|
||
* @brief 判断指定的语言包类型是否存在
|
||
* @param 指定的类型
|
||
* @return 存在返回true
|
||
*/
|
||
bool isExistsLanguage(Languages lang);
|
||
|
||
/**
|
||
* @brief 根据传入的枚举，获取相应的语言包字符串
|
||
* @param 语言包的枚举
|
||
* @return 语言包的字符串
|
||
*/
|
||
QString getLangStr(Languages lang);
|
||
|
||
/**
|
||
* @brief 获取系统当前的语言
|
||
* @return
|
||
*/
|
||
Languages getSystemLang();
|
||
|
||
/**
|
||
* @brief 获取当前使用的语言包
|
||
* @param 需要使用的语言包
|
||
* @return
|
||
*/
|
||
QString getLanguages();
|
||
|
||
/**
|
||
* @brief 获取纯字符串的识别结果
|
||
* @param 需识别的图片
|
||
* @param 返回的字符串结果类型
|
||
* @return 返回识别结果
|
||
*/
|
||
RecognitionResult getRecognizeResult(Pix * image,ResultType resultType);
|
||
|
||
/**
|
||
* @brief 设置返回结果,内部使用
|
||
* @param 错误码
|
||
* @param 错误消息
|
||
* @param 结果类型
|
||
*/
|
||
void setResult(ErrorCode errCode, const QString errMessage,const ResultType resultType,RecognitionResult &result);
|
||
|
||
/**
|
||
* @brief 语言包路径
|
||
*/
|
||
QString m_sTessdataPath;
|
||
|
||
/**
|
||
* @brief 语言包
|
||
*/
|
||
QString m_sLangs;
|
||
|
||
/**
|
||
* @brief 是否正在识别中
|
||
*/
|
||
bool m_isRunning;
|
||
|
||
static TessOcrUtils *m_tessOcrUtils;
|
||
|
||
static tesseract::TessBaseAPI *t_Tesseract;
|
||
|
||
};
|
||
|
||
#endif // TESSOCRUTILS_H
|