mirror of
https://github.com/wmjordan/PDFPatcher.git
synced 2026-10-18 13:42:28 +08:00
WMJ
246 lines
8.9 KiB
C#
246 lines
8.9 KiB
C#
using System;
|
|
using System.ComponentModel;
|
|
using System.IO;
|
|
using System.Windows.Forms;
|
|
using PDFPatcher.Common;
|
|
using PDFPatcher.Model;
|
|
|
|
namespace PDFPatcher.Functions
|
|
{
|
|
[ToolboxItem(false)]
|
|
public partial class OcrControl : FunctionControl, IResettableControl
|
|
{
|
|
OcrOptions _options;
|
|
|
|
public override string FunctionName => "识别图像文本";
|
|
|
|
public override System.Drawing.Bitmap IconImage => Properties.Resources.Ocr;
|
|
|
|
public override Button DefaultButton => _ExportBookmarkButton;
|
|
|
|
public OcrControl() {
|
|
InitializeComponent();
|
|
this.OnFirstLoad(OnLoad);
|
|
}
|
|
|
|
void OnLoad() {
|
|
_BookmarkControl.FileDialog.Filter = Constants.FileExtensions.XmlFilter + "|" + Constants.FileExtensions.TxtFilter + "|" + Constants.FileExtensions.XmlOrTxtFilter;
|
|
|
|
AppContext.MainForm.SetTooltip(_SourceFileControl.FileList, "需要识别文本的 PDF 源文件路径");
|
|
AppContext.MainForm.SetTooltip(_BookmarkControl.FileList, "指定识别文本后生成的信息文件或文本文件路径，如路径为空则不输出文件");
|
|
AppContext.MainForm.SetTooltip(_ExportBookmarkButton, "点击此按钮导出识别后的文本到文件");
|
|
AppContext.MainForm.SetTooltip(_ImportOcrResultButton, "点击此按钮，将识别后的文本内容写入到目标 PDF 文件。");
|
|
AppContext.MainForm.SetTooltip(_PageRangeBox, Messages.PageRanges);
|
|
AppContext.MainForm.SetTooltip(_DetectColumnsBox, "允许将距离较远的文本合并为同一行文本");
|
|
AppContext.MainForm.SetTooltip(_DetectContentPunctuationsBox, "将三个以上的连续标点替换成“ .... ”");
|
|
AppContext.MainForm.SetTooltip(_CompressWhiteSpaceBox, "将三个以上连续出现的空格压缩成两个空格");
|
|
AppContext.MainForm.SetTooltip(_OrientBox, "自动检测页面横竖置放方向");
|
|
AppContext.MainForm.SetTooltip(_StretchBox, "自动纠直倾斜的页面");
|
|
AppContext.MainForm.SetTooltip(_OutputOriginalOcrResultBox, "保存原始的未经过优化合并的识别结果（可用于写入 PDF 文档）");
|
|
|
|
var lb = _OcrLangBox.Items;
|
|
if (Processor.ModiOcr.ModiInstalled) {
|
|
foreach (var item in Constants.Ocr.LangIDs) {
|
|
if (Processor.ModiOcr.IsLanguageInstalled(item)) {
|
|
lb.Add(ValueHelper.MapValue(item, Constants.Ocr.LangIDs, Constants.Ocr.LangNames));
|
|
}
|
|
}
|
|
}
|
|
if (lb.Count == 0) {
|
|
lb.Add("无");
|
|
}
|
|
_ExportBookmarkButton.Enabled = Processor.ModiOcr.ModiInstalled;
|
|
if (_ExportBookmarkButton.Enabled == false) {
|
|
AppContext.MainForm.SetTooltip(_OcrLangBox, "当前系统尚未安装识别引擎，请先安装微软 Office 文字识别引擎，再重新启动程序。");
|
|
}
|
|
Reload();
|
|
|
|
var d = _BookmarkControl.FileDialog;
|
|
d.CheckFileExists = false;
|
|
d.CheckPathExists = false;
|
|
|
|
if (d is SaveFileDialog sd) {
|
|
sd.OverwritePrompt = false;
|
|
}
|
|
}
|
|
|
|
public override void SetupCommand(ToolStripItem item) {
|
|
switch (item.Name) {
|
|
case Commands.SaveBookmark:
|
|
item.Text = "写入PDF文件(&Q)";
|
|
item.ToolTipText = "将识别结果写入 PDF 文件";
|
|
EnableCommand(item, true, true);
|
|
break;
|
|
}
|
|
base.SetupCommand(item);
|
|
}
|
|
|
|
public override void ExecuteCommand(string commandName, params string[] parameters) {
|
|
switch (commandName) {
|
|
case Commands.SaveBookmark:
|
|
_ImportOcrResultButton.PerformClick();
|
|
return;
|
|
}
|
|
base.ExecuteCommand(commandName, parameters);
|
|
}
|
|
|
|
public void Reset() {
|
|
AppContext.Ocr = new OcrOptions();
|
|
Reload();
|
|
}
|
|
|
|
public void Reload() {
|
|
_options = AppContext.Ocr;
|
|
_CompressWhiteSpaceBox.Checked = _options.CompressWhiteSpaces;
|
|
_ConvertToMonoColorBox.Checked = !_options.PreserveColor;
|
|
_DetectColumnsBox.Checked = _options.DetectColumns;
|
|
_DetectContentPunctuationsBox.Checked = _options.DetectContentPunctuations;
|
|
var i = Array.IndexOf(Constants.Ocr.LangIDs, _options.OcrLangID);
|
|
_OcrLangBox.Select(i > 0 ? i : 0);
|
|
_OrientBox.Checked = _options.OrientPage;
|
|
_RemoveSpaceBetweenChineseBox.Checked = _options.RemoveWhiteSpacesBetweenChineseCharacters;
|
|
_SaveOcredImageBox.Checked = !String.IsNullOrEmpty(_options.SaveOcredImagePath);
|
|
_StretchBox.Checked = _options.StretchPage;
|
|
_OutputOriginalOcrResultBox.Checked = _options.OutputOriginalOcrResult;
|
|
_PrintOcrResultBox.Checked = _options.PrintOcrResult;
|
|
|
|
_WritingDirectionBox.Select((int)_options.WritingDirection);
|
|
_QuantitiveFactorBox.SetValue(_options.QuantitativeFactor);
|
|
}
|
|
|
|
void Button_Click(object sender, EventArgs e) {
|
|
if (File.Exists(_SourceFileControl.FirstFile) == false) {
|
|
FormHelper.ErrorBox(Messages.SourceFileNotFound);
|
|
return;
|
|
}
|
|
if (sender == _ImportOcrResultButton) {
|
|
if (FileHelper.IsPathValid(_TargetFileControl.Text) == false) {
|
|
FormHelper.ErrorBox(Messages.TargetFileNameInvalid);
|
|
return;
|
|
}
|
|
if (_BookmarkControl.Text.Length == 0) {
|
|
FormHelper.ErrorBox("请指定识别结果文件。");
|
|
return;
|
|
}
|
|
}
|
|
|
|
AppContext.SourceFiles = _SourceFileControl.Files;
|
|
AppContext.BookmarkFile = _BookmarkControl.Text;
|
|
AppContext.TargetFile = _TargetFileControl.Text;
|
|
if (_SourceFileControl.Files.Length == 1) {
|
|
_SourceFileControl.FileList.AddHistoryItem();
|
|
if (_BookmarkControl.Text.Length > 0) {
|
|
_BookmarkControl.FileList.AddHistoryItem();
|
|
}
|
|
}
|
|
if (sender == _ImportOcrResultButton) {
|
|
_TargetFileControl.FileList.AddHistoryItem();
|
|
}
|
|
|
|
AppContext.MainForm.ResetWorker();
|
|
|
|
SyncOptions();
|
|
|
|
var worker = AppContext.MainForm.GetWorker();
|
|
if (sender != _ImportOcrResultButton) {
|
|
worker.DoWork += OcrExport;
|
|
worker.RunWorkerAsync(new object[] {
|
|
AppContext.SourceFiles,
|
|
AppContext.BookmarkFile,
|
|
_options
|
|
});
|
|
}
|
|
else {
|
|
worker.DoWork += ImportOcr;
|
|
worker.RunWorkerAsync(new object[] {
|
|
AppContext.SourceFiles,
|
|
AppContext.BookmarkFile,
|
|
AppContext.TargetFile
|
|
});
|
|
}
|
|
worker.RunWorkerCompleted += Worker_Completed;
|
|
}
|
|
|
|
void Worker_Completed(object sender, RunWorkerCompletedEventArgs e) {
|
|
var worker = (BackgroundWorker)sender;
|
|
worker.RunWorkerCompleted -= Worker_Completed;
|
|
worker.DoWork -= OcrExport;
|
|
worker.DoWork -= ImportOcr;
|
|
}
|
|
|
|
void SyncOptions() {
|
|
_options.CompressWhiteSpaces = _CompressWhiteSpaceBox.Checked;
|
|
_options.PreserveColor = !_ConvertToMonoColorBox.Checked;
|
|
_options.DetectColumns = _DetectColumnsBox.Checked;
|
|
_options.DetectContentPunctuations = _DetectContentPunctuationsBox.Checked;
|
|
_options.PageRanges = _PageRangeBox.Text;
|
|
_options.OcrLangID = ValueHelper.MapValue(_OcrLangBox.Text, Constants.Ocr.LangNames, Constants.Ocr.LangIDs, -1);
|
|
_options.OrientPage = _OrientBox.Checked;
|
|
_options.OutputOriginalOcrResult = _OutputOriginalOcrResultBox.Checked;
|
|
_options.QuantitativeFactor = (float)_QuantitiveFactorBox.Value;
|
|
_options.PrintOcrResult = _PrintOcrResultBox.Checked;
|
|
_options.RemoveWhiteSpacesBetweenChineseCharacters = _RemoveSpaceBetweenChineseBox.Checked;
|
|
_options.StretchPage = _StretchBox.Checked;
|
|
// _options.SaveOcredImagePath = String.IsNullOrEmpty (this._BookmarkControl.Text) ? null : Common.FileHelper.CombinePath (Path.GetDirectoryName (this._BookmarkControl.Text), Path.GetFileNameWithoutExtension (_BookmarkControl.Text) + Constants.FileExtensions.Tif);
|
|
_options.WritingDirection = (WritingDirection)_WritingDirectionBox.SelectedIndex;
|
|
}
|
|
|
|
void OcrExport(object sender, DoWorkEventArgs e) {
|
|
var a = e.Argument as object[];
|
|
var files = a[0] as string[];
|
|
var b = a[1] as string;
|
|
var options = a[2] as OcrOptions;
|
|
if (files.Length > 1) {
|
|
var p = Path.GetDirectoryName(b);
|
|
var ext = Path.GetExtension(b);
|
|
foreach (var file in files) {
|
|
Processor.Worker.Ocr(file, FileHelper.CombinePath(p, Path.GetFileNameWithoutExtension(file) + ext), options);
|
|
if (AppContext.Abort) {
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
Processor.Worker.Ocr(files[0], b, options);
|
|
}
|
|
}
|
|
|
|
void ImportOcr(object sender, DoWorkEventArgs e) {
|
|
var a = e.Argument as object[];
|
|
var files = a[0] as string[];
|
|
var b = a[1] as string;
|
|
var target = a[2] as string;
|
|
if (files.Length > 1) {
|
|
var p = Path.GetDirectoryName(b);
|
|
var ext = Path.GetExtension(b);
|
|
foreach (var file in files) {
|
|
Processor.Worker.ImportOcr(file, FileHelper.CombinePath(p, Path.GetFileNameWithoutExtension(file) + ext), target);
|
|
if (AppContext.Abort) {
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
Processor.Worker.ImportOcr(files[0], b, target);
|
|
}
|
|
}
|
|
|
|
void _ImportLink_LinkClicked(object sender, LinkLabelLinkClickedEventArgs e) {
|
|
AppContext.MainForm.SelectFunctionList(Function.Patcher);
|
|
}
|
|
|
|
void ControlEvent(object sender, EventArgs e) {
|
|
if (sender == _WritingDirectionBox) {
|
|
_DetectColumnsBox.Enabled = _WritingDirectionBox.SelectedIndex != 0;
|
|
}
|
|
else if (sender == _OutputOriginalOcrResultBox) {
|
|
_DetectColumnsBox.Enabled
|
|
= _DetectContentPunctuationsBox.Enabled
|
|
= _CompressWhiteSpaceBox.Enabled
|
|
= _RemoveSpaceBetweenChineseBox.Enabled
|
|
= !_OutputOriginalOcrResultBox.Checked;
|
|
}
|
|
}
|
|
}
|
|
}
|