83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
/*
|
||
* ====================================================
|
||
|
||
* 功能:旋转图像
|
||
* 作者:刘丁维
|
||
* 生成时间:2020/4/21
|
||
* 最近修改时间:v1.0 2020/04/21
|
||
v1.1 2020/08/12 修复文稿方向自动识别导致崩溃的BUG
|
||
v1.2 2021/10/15 开放文稿方向识别模块
|
||
v1.2.1 2022/04/25 补充切换版本的预处理代码
|
||
* 版本号:v1.2.1
|
||
|
||
* ====================================================
|
||
*/
|
||
|
||
#ifndef IMAGE_APPLY_ROTATION_H
|
||
#define IMAGE_APPLY_ROTATION_H
|
||
|
||
#include "ImageApply.h"
|
||
|
||
#ifdef _WIN32
|
||
//#define USE_HANWANG
|
||
#ifdef USE_HANWANG
|
||
#define HG_GPDF_API_BUILD
|
||
#include <hg_gpdf.h>
|
||
#endif
|
||
#endif
|
||
|
||
class GIMGPROC_LIBRARY_API CImageApplyRotation : public CImageApply
|
||
{
|
||
public:
|
||
enum class RotationType
|
||
{
|
||
Invalid, //无效
|
||
Rotate_90_clockwise, //顺时针90°
|
||
Rotate_180, //180°
|
||
Rotate_90_anti_clockwise, //逆时针90°,即270°
|
||
|
||
AutoTextOrientation //自动文稿方向识别旋转
|
||
};
|
||
|
||
public:
|
||
|
||
/*
|
||
* rotation [in]:旋转类型
|
||
* isBackTransposed [in]:true为背面180°旋转,反之亦然
|
||
* dpi [in]:当前图像的DPI,该参数在rotation为AutoTextOrientation时生效。在识别文稿方向时,会默认将图像变换为200DPI进行识别
|
||
* tessadataPath [in]:训练库文件路径,该参数在rotation为AutoTextOrientation时生效
|
||
*/
|
||
CImageApplyRotation(RotationType rotation, bool isBackTransposed = false, int dpi = 200, const char* tessdataPath = nullptr);
|
||
|
||
virtual ~CImageApplyRotation();
|
||
|
||
virtual void apply(cv::Mat& pDib, int side) override;
|
||
|
||
virtual void apply(std::vector<cv::Mat>& mats, bool isTwoSide);
|
||
|
||
bool isBackTransposed() { return m_backTranspose; }
|
||
|
||
int getDPI() { return m_dpi; }
|
||
|
||
RotationType getRotationType() { return m_rotation; }
|
||
|
||
void setBackTransposed(bool enabled) { m_backTranspose = enabled; }
|
||
|
||
void setDPI(int dpi) { m_dpi = dpi; }
|
||
|
||
void setRotationType(RotationType type) { m_rotation = type; }
|
||
|
||
private:
|
||
RotationType m_rotation;
|
||
bool m_backTranspose;
|
||
int m_dpi;
|
||
|
||
#ifdef _WIN32
|
||
#ifdef USE_HANWANG
|
||
HG_OCR* m_ocr;
|
||
#endif
|
||
#endif
|
||
};
|
||
|
||
#endif // !IMAGE_APPLY_ROTATION_H
|