PDFOffice_Windows_exe/PDF Office/ViewModels/Tools/ScanContentViewModel.cs

using ComDocumentAIKit;
using ComPDFKit.PDFDocument;
using ComPDFKit.PDFPage;
using ComPDFKit.PDFPage.Edit;
using ComPDFKitViewer.PdfViewer;
using PDF_Office.Helper;
using PDF_Office.Model;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Regions;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Media;
using System.Windows.Media.Imaging;

namespace PDF_Office.ViewModels.Tools
{
    class ScanContentViewModel : BindableBase, INavigationAware
    {
        #region 模板与导航相关内容

        private CPDFViewer PDFViewer;

        public ViewContentViewModel viewContentViewModel;
        public bool IsNavigationTarget(NavigationContext navigationContext)
        {
            return true;
        }

        public void OnNavigatedFrom(NavigationContext navigationContext)
        {
            return;
        }

        public void OnNavigatedTo(NavigationContext navigationContext)
        {
            navigationContext.Parameters.TryGetValue<ViewContentViewModel>(ParameterNames.ViewContentViewModel, out viewContentViewModel);
            navigationContext.Parameters.TryGetValue<CPDFViewer>(ParameterNames.PDFViewer, out PDFViewer);

        }

        #endregion

        private IRegionManager regions;
        public DelegateCommand NavigationCommand { get; set; }
        public DelegateCommand EnhancedCommand { get; set; }
        public List<string> EnhancedFilePathList;
        public ScanContentViewModel(IRegionManager regionManager)
        {
            regions = regionManager;
            NavigationCommand = new DelegateCommand(BtnNavigation);
            EnhancedCommand = new DelegateCommand(EnhancedScan);
            EnhancedFilePathList = new List<string>();
        }

        /// <summary>
        ///导航到OCR的侧边栏
        /// </summary>
        private void BtnNavigation()
        {
            //替换属性面板的内容
            NavigationParameters parameters = new NavigationParameters();
            parameters.Add(ParameterNames.PDFViewer, PDFViewer);
            parameters.Add(ParameterNames.PropertyPanelContentViewModel, null);
            regions.RequestNavigate(RegionNames.PropertyRegionName, "ScanPropertyPanel", parameters);
            //显示属性面板
            viewContentViewModel.IsPropertyOpen = true;
        }
        private void EnhancedScan()
        {
            CPDFDocument CurrentDoc = PDFViewer.Document;
            string path = App.CachePath.MergeFilePath;
            string name = Guid.NewGuid().ToString();
            path = Path.Combine(path, name);
            string EnhancePath = Path.Combine(path, "Enhance");

            Directory.CreateDirectory(path);
            Directory.CreateDirectory(EnhancePath);

            CIMEngine imEngine = new CIMEngine(App.modelFolderPath);
            CErrorCode error = imEngine.SetModel();
            for (int i = 0; i < CurrentDoc.PageCount; i++)
            {
                string pageImagePath = Path.Combine(path, i.ToString());
                string pageEnhancePath = Path.Combine(EnhancePath, i.ToString() + ".png");
                try
                {
                    CPDFPage pdfPage = CurrentDoc.PageAtIndex(i);
                    float zoom = (float)(DpiHelpers.Dpi / 72D);
                    int renderWidth = (int)(pdfPage.PageSize.Width * zoom);
                    int renderHeight = (int)(pdfPage.PageSize.Height * zoom);
                    byte[] renderData = new byte[renderWidth * renderHeight * 4];
                    pdfPage.RenderPageBitmapWithMatrix(zoom, new Rect(0, 0, renderWidth, renderHeight), 0xFFFFFFFF, renderData, 0);
                    WriteableBitmap bitmap = new WriteableBitmap(renderWidth, renderHeight, DpiHelpers.Dpi, DpiHelpers.Dpi, PixelFormats.Bgra32, null);
                    bitmap.WritePixels(new Int32Rect(0, 0, renderWidth, renderHeight), renderData, bitmap.BackBufferStride, 0);
                    BitmapEncoder encoder = new PngBitmapEncoder();
                    encoder.Frames.Add(BitmapFrame.Create(bitmap));

                    using (FileStream imageStream = File.Create(pageImagePath))
                    {
                        encoder.Save(imageStream);
                        imageStream.Flush();
                    }

                    //File.Create(pageEnhancePath);
                    error = imEngine.Process(pageImagePath, pageEnhancePath);
                    EnhancedFilePathList.Add(pageEnhancePath);
                }
                catch
                {

                }
            }
        }

        private void OCREnhancedScan()
        {
            CPDFDocument EnhanceDoc = CPDFDocument.CreateDocument();
            COCREngine imEngine = new COCREngine(App.modelFolderPath);
            //CIMEngine imEngine = new CIMEngine(App.modelFolderPath);

            string path = App.CachePath.MergeFilePath;
            string name = Guid.NewGuid().ToString();
            path = Path.Combine(path, name);
            string EnhancePath = Path.Combine(path, "Enhance");

            Directory.CreateDirectory(path);
            Directory.CreateDirectory(EnhancePath);

            CPDFDocument CurrentDoc = PDFViewer.Document;
            //CurrentDoc.FilePath;

            CErrorCode error = imEngine.SetModel(COCRLanguage.Chinese_S);
            //CErrorCode error = imEngine.SetModel();
            for (int i = 0; i < 1; i++)
            {
                string pageImagePath = Path.Combine(path, i.ToString());
                string pageEnhancePath = Path.Combine(EnhancePath, i.ToString() + ".png");
                try
                {
                    CPDFPage pdfPage = CurrentDoc.PageAtIndex(i);
                    float zoom = (float)(DpiHelpers.Dpi / 72D);
                    int renderWidth = (int)(pdfPage.PageSize.Width * zoom);
                    int renderHeight = (int)(pdfPage.PageSize.Height * zoom);
                    byte[] renderData = new byte[renderWidth * renderHeight * 4];
                    pdfPage.RenderPageBitmapWithMatrix(zoom, new Rect(0, 0, renderWidth, renderHeight), 0xFFFFFFFF, renderData, 0);
                    WriteableBitmap bitmap = new WriteableBitmap(renderWidth, renderHeight, DpiHelpers.Dpi, DpiHelpers.Dpi, PixelFormats.Bgra32, null);
                    bitmap.WritePixels(new Int32Rect(0, 0, renderWidth, renderHeight), renderData, bitmap.BackBufferStride, 0);
                    BitmapEncoder encoder = new PngBitmapEncoder();
                    encoder.Frames.Add(BitmapFrame.Create(bitmap));

                    using (FileStream imageStream = File.Create(pageImagePath))
                    {
                        encoder.Save(imageStream);
                        imageStream.Flush();
                    }

                    //File.Create(pageEnhancePath);
                    //error = imEngine.Process(pageImagePath, pageEnhancePath);
                    error = imEngine.Process(pageImagePath);

                    if (imEngine.OCRResultList == null)
                        return;

                    List<KeyValuePair<Rect, string>> textRectList = new List<KeyValuePair<Rect, string>>();
                    foreach (COCRResult ocrResult in imEngine.OCRResultList)
                    {
                        List<Point> rectPoints = new List<Point>();
                        for (int j = 0; j < ocrResult.position.Length; j += 2)
                        {
                            rectPoints.Add(new Point(ocrResult.position[j], ocrResult.position[j + 1]));
                        }
                        int left = (int)rectPoints.AsEnumerable().Min(x => x.X);
                        int right = (int)rectPoints.AsEnumerable().Max(x => x.X);
                        int top = (int)rectPoints.AsEnumerable().Min(x => x.Y);
                        int bottom = (int)rectPoints.AsEnumerable().Max(x => x.Y);

                        textRectList.Add(new KeyValuePair<Rect, string>(new Rect(left, top, right - left, bottom - top), ocrResult.text));
                    }

                    CPDFEditPage editPage = pdfPage.GetEditPage();

                    editPage.BeginEdit(CPDFEditType.EditText);
                    for (int y = textRectList.Count - 1; y >= 0; y--)
                    {
                        KeyValuePair<Rect, string> textBlock = textRectList[y];
                        int fontSize = 1;
                        Rect pdfRect = DpiHelpers.GetRawRect(textBlock.Key);

                        for (; fontSize < 100; fontSize++)
                        {
                            Rect cmpRect = pdfPage.GetTextContentRect(fontSize, "Helvetica", textBlock.Value);
                            if (cmpRect.Width > pdfRect.Width || cmpRect.Height > pdfRect.Height)
                            {
                                fontSize -= 1;
                                break;
                            }
                        }

                        CPDFEditTextArea editArea = editPage.CreateNewTextArea(pdfRect, "Helvetica", fontSize, new byte[] { 0, 0, 0 });

                        bool x = editArea.SetTextAreaAlign(TextAlignType.AlignChar);
                        editArea.InsertText(textBlock.Value);

                    }
                    editPage.EndEdit();
                }
                catch (Exception e)
                {
                    throw;
                }
            }
            //error = imEngine.Process("C:/test/9.png", "C:/test/9IM.png");
            imEngine.Release();
            PDFViewer.InvalidChildVisual(true);
            return;

            //更新预览
            //int PageIndex = PDFViewer.CurrentIndex;
            //PDFViewer.InitDocument(EnhanceDoc);
            //PDFViewer.GoToPage(PageIndex);

        }

    }
}