diff --git a/huagao/ImageProcess/hg_gpdf.cpp b/huagao/ImageProcess/hg_gpdf.cpp deleted file mode 100644 index bf0546c3..00000000 --- a/huagao/ImageProcess/hg_gpdf.cpp +++ /dev/null @@ -1,163 +0,0 @@ -#include "hg_gpdf.h" - -#if defined(WIN32) -#include "baseapi.h" -#include "allheaders.h" -#include "renderer.h" -#else -#include -#include -#include -#endif - -//#define USE_QT -#ifdef USE_QT -#include -#include -#endif - -Pix* createPix(const unsigned char * imagedata, int width, int height, int bytes_per_pixel, int bytes_per_line, int dpi) -{ - int bpp = bytes_per_pixel * 8; - if (bpp == 0) bpp = 1; - Pix* pix = pixCreate(width, height, bpp == 24 ? 32 : bpp); - pixSetXRes(pix, dpi); - pixSetYRes(pix, dpi); - l_uint32* data = pixGetData(pix); - int wpl = pixGetWpl(pix); - switch (bpp) - { - case 1: - for (int y = 0; y < height; ++y, data += wpl, imagedata += bytes_per_line) - for (int x = 0; x < width; ++x) - if (imagedata[x / 8] & (0x80 >> (x % 8))) - CLEAR_DATA_BIT(data, x); - else - SET_DATA_BIT(data, x); - break; - - case 8: - // Greyscale just copies the bytes in the right order. - for (int y = 0; y < height; ++y, data += wpl, imagedata += bytes_per_line) - for (int x = 0; x < width; ++x) - SET_DATA_BYTE(data, x, imagedata[x]); - break; - - case 24: - // Put the colors in the correct places in the line buffer. - for (int y = 0; y < height; ++y, imagedata += bytes_per_line) - for (int x = 0; x < width; ++x, ++data) { - SET_DATA_BYTE(data, COLOR_RED, imagedata[3 * x]); - SET_DATA_BYTE(data, COLOR_GREEN, imagedata[3 * x + 1]); - SET_DATA_BYTE(data, COLOR_BLUE, imagedata[3 * x + 2]); - } - break; - - case 32: - // Maintain byte order consistency across different endianness. - for (int y = 0; y < height; ++y, imagedata += bytes_per_line, data += wpl) - for (int x = 0; x < width; ++x) - data[x] = static_cast((imagedata[x * 4] << 24) | (imagedata[x * 4 + 1] << 16) | - (imagedata[x * 4 + 2] << 8) | imagedata[x * 4 + 3]); - break; - - default: - break; - } - - pix->informat = bytes_per_pixel == 1 ? 1 : 2; - if (bytes_per_pixel == 1) - { - PIXCMAP* colormap = pixcmapCreate(8); - LEPT_FREE(colormap->array); - colormap->array = reinterpret_cast(LEPT_CALLOC(256, sizeof(RGBA_QUAD))); - colormap->n = 256; - colormap->nalloc = 256; - colormap->depth = 8; - l_uint8* ptr = reinterpret_cast(colormap->array); - for (int i = 0; i < 256; i++) - ptr[i * 4 + 0] = ptr[i * 4 + 1] = ptr[i * 4 + 2] = ptr[i * 4 + 3] = static_cast(i); - pixSetColormap(pix, colormap); - } - pixSetXRes(pix, 200); - pixSetYRes(pix, 200); - return pix; -} - -HG_OCR::HG_OCR() - : m_ptr(nullptr) -{ -} - -HG_OCR::~HG_OCR() -{ - if (m_ptr) - { - reinterpret_cast(m_ptr)->End(); - delete reinterpret_cast(m_ptr); - } -} - -int HG_OCR::init(const char* trainFile, RECOGNITION_MODE mode) -{ - if (m_ptr) delete reinterpret_cast(m_ptr); - tesseract::TessBaseAPI* api = new tesseract::TessBaseAPI(); - api->SetPageSegMode(tesseract::PSM_AUTO_OSD); - m_ptr = reinterpret_cast(api); - return api->Init(trainFile, mode == RECOGNITION_OCR ? "chi_sim" : "osd"); -} - -bool HG_OCR::ocr(const char** inputFileNames, int numOfFiles, const char* outputFileName, SAVE_MODE flag) -{ - if (m_ptr == nullptr) return false; - if (inputFileNames == nullptr || numOfFiles == 0) return false; - - tesseract::PointerVector renderers; - tesseract::TessBaseAPI *api = reinterpret_cast(m_ptr); - if (flag & SAVE_PDF) - renderers.push_back(new tesseract::TessPDFRenderer(outputFileName, api->GetDatapath(), false)); - if (flag & SAVE_TXT) - renderers.push_back(new tesseract::TessTextRenderer(outputFileName)); - -#ifdef USE_QT - QTime timer; - timer.start(); -#endif - - int numOfRenderer = renderers.length(); - for (int i = 0; i < numOfRenderer; i++) - renderers[i]->BeginDocument(""); - - for (int i = 0; i < numOfFiles; i++) - { - Pix* img = pixRead(inputFileNames[i]); - api->SetInputName(outputFileName); - api->SetImage(img); - api->Recognize(nullptr); - for (int i = 0; i < numOfRenderer; i++) - renderers[i]->AddImage(api); - pixDestroy(&img); - } - - for (int i = 0; i < numOfRenderer; i++) - renderers[i]->EndDocument(); - - renderers.clear(); - -#ifdef USE_QT - qDebug() << timer.elapsed(); -#endif - - return true; -} - -int HG_OCR::orientation(const unsigned char *data, int width, int height, int channels, int dpi) -{ - Pix* pix = createPix(data, width, height, channels, (width * channels + 3) / 4 * 4, dpi); - tesseract::TessBaseAPI *api = reinterpret_cast(m_ptr); - api->SetImage(pix); - int orient_deg = 1; - api->DetectOrientationScript(&orient_deg, nullptr, nullptr, nullptr); - pixDestroy(&pix); - return orient_deg; -} diff --git a/huagao/ImageProcess/hg_gpdf.h b/huagao/ImageProcess/hg_gpdf.h deleted file mode 100644 index 4f77d513..00000000 --- a/huagao/ImageProcess/hg_gpdf.h +++ /dev/null @@ -1,85 +0,0 @@ -/* -* ==================================================== - -* 功能：本地图片进行OCR，并生成多页PDF功能 -* 作者：刘丁维 -* 生成时间：2020/3/5 -* 最近修改时间：2020/3/5 -* 版本号：v1.0 - -* ==================================================== -*/ - -#ifndef HG_GPDF_H -#define HG_GPDF_H - -#if defined (_WIN32) - #if !defined (HG_GPDF_API_BUILD) - #define HG_GPDF_API //__declspec(dllexport) - #else - #define HG_GPDF_API //__declspec(dllimport) - #endif -#else - #define HG_GPDF_API -#endif - -class HG_GPDF_API HG_OCR -{ -public: - - //识别模式 - enum RECOGNITION_MODE - { - RECOGNITION_OSD, //文稿方向识别模式 - RECOGNITION_OCR //字符识别模式 - }; - - //字符识别保存模式 - enum SAVE_MODE - { - SAVE_PDF = 0x01, //PDF保存字符识别结果 - SAVE_TXT = 0x02, //TXT保存字符识别结果 - SAVE_PDF_TXT = 0x03 //同时采用PDF和TXT保存字符识别结果 - }; -public: - - HG_OCR(); - - ~HG_OCR(); - - /* - * 函数功能：初始化PDF生成器 - * trainFile:[in] 训练库文件路径，不含文件名，末尾无需'/'结尾 - * mode：[in] 具体选项参照 enum RECOGNITION_MODE，默认值为RECOGNITION_OCR - * 返回值：0为成功，否则为异常。异常查看打印输出 - */ - int init(const char* trainFile, RECOGNITION_MODE mode = RECOGNITION_OCR); - - /* - * 函数功能：字符识别，并生成文本 - * inputFileNames:[in] 本地待识别图片完整路径，可同时输入多个图片路径 - * numOfFiles：[in] inputFileNames长度 - * numOfFiles：[in] inputFileNames长度 - * outputFileName：[in] 识别结果保存路径，包含文件基础名字，不含文件后缀名 - * flag：[in] 字符识别保存模式，参照 enum SAVE_MODE - * 返回值：true为成功，否则为异常。异常查看打印输出 - */ - bool ocr(const char** inputFileNames, int numOfFiles, const char* outputFileName, SAVE_MODE flag = SAVE_PDF); - - /* - * 函数功能：文本方向识别 - * data:[in] 图像数据指针 - * width：[in] 图像宽度 - * height：[in] 图像高度 - * channels：[in] 图像通道 - * dpi：[in] 图片DPI - * 返回值：以顺时针进行计数，输出0,90,180,270四种结果 - */ - int orientation(const unsigned char* data, int width, int height, int channels, int dpi); - -private: - void* m_ptr; -}; - - -#endif //HG_GPDF_H