PDFOffice_Windows_exe/PDF Office/ViewModels/EditTools/Bates/BatesDocumentContentViewModel.cs

using ComPDFKit.PDFDocument;
using ComPDFKit.PDFPage;
using ComPDFKitViewer;
using ComPDFKitViewer.PdfViewer;
using PDF_Master.EventAggregators;
using PDF_Master.Helper;
using PDF_Master.Model.EditTools.Bates;
using Prism.Commands;
using Prism.Events;
using Prism.Mvvm;
using Prism.Regions;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.RegularExpressions;
using System.Windows;
using System.Windows.Media.Imaging;
using System.Windows.Media;
using System.Windows.Controls;
using PDF_Master.Model;
using PDF_Master.Model.EditTools.HeaderFooter;
using PDF_Master.Model.EditTools.Background;

namespace PDF_Master.ViewModels.EditTools.Bates
{
    public class BatesDocumentContentViewModel : BindableBase, INavigationAware
    {
        public IEventAggregator eventAggregator;
        public IRegionManager regionManager;
        public CPDFViewer PDFViewer;
        public CPDFDocument Document;
        private ViewContentViewModel viewContentViewModel;
        public EnumDelete EnumDelete = EnumDelete.StatusCreate;
        public CPDFViewer pdfViewer;

        private CPDFBates bates;
        private BatesInfo batesInfo;

        public bool firstin = true;

        public string ViewerRegionName { get; set; }

        public string unicode = null;
        public string Unicode = null;

        private int _pageSize;
        public int PageRangeNumber
        {
            get { return _pageSize; }
            set { SetProperty(ref _pageSize, value); }
        }

        private int _currentPageIndex;
        public int CurrentPageIndex
        {
            get { return _currentPageIndex; }
            set
            {
                SetProperty(ref _currentPageIndex, value);

            }
        }

        private Visibility _inputIndexBoxVisible = Visibility.Collapsed;
        public Visibility InputIndexBoxVisible
        {
            get
            {
                return _inputIndexBoxVisible;
            }
            set
            {
                SetProperty(ref _inputIndexBoxVisible, value);
            }
        }

        public DelegateCommand<object> ShowInputIndexBoxCommand { set; get; }
        public DelegateCommand<object> GoToPageCommand { set; get; }

        public BatesDocumentContentViewModel(IEventAggregator eventAggregator, IRegionManager regionManager)
        {
            this.regionManager = regionManager;
            this.eventAggregator = eventAggregator;
            ShowInputIndexBoxCommand = new DelegateCommand<object>(ShowInputIndexBox);
            GoToPageCommand = new DelegateCommand<object>(GoToPage);
            Unicode = App.mainWindowViewModel.SelectedItem.Unicode;
            ViewerRegionName = RegionNames.BatesViewerRegionName;
            eventAggregator.GetEvent<SetBatesEvent>().Subscribe(SetBates, e => e.Unicode == Unicode);
            eventAggregator.GetEvent<ConfirmEditToolsBatesEvent>().Subscribe(ConfirmEditToolsBates, e => e == Unicode);
            eventAggregator.GetEvent<DeleteBatesEvent>().Subscribe(DeleteBates, e => e.Unicode == Unicode);
        }

        public void ShowInputIndexBox(object obj)
        {
            if (obj != null)
            {
                InputIndexBoxVisible = Visibility.Visible;
            }
        }

        public void GoToPage(object obj)
        {

            if (obj != null)
            {
                var args = obj as TextBox;
                if (args.Text == null)
                {
                    return;
                }
                if (int.Parse(args.Text) > 0 && int.Parse(args.Text) <= PDFViewer.Document.PageCount && Regex.Match(args.Text, "^\\d+$").Success)
                {
                    PDFViewer.GoToPage(int.Parse(args.Text) - 1);
                    InputIndexBoxVisible = Visibility.Collapsed;
                }
            }
        }

        public void ConfirmEditToolsBates(string unicode)
        {
            if (EnumDelete == EnumDelete.StatusDeleteAll)
            {
                PDFViewer.Document.GetBates().Clear();
            }
            else
            {
                if (batesInfo != null)
                {
                    CreateBates(viewContentViewModel.PDFViewer.Document);
                }
            }

            viewContentViewModel.PDFViewer.UndoManager.CanSave = true;
        }

        public void DeleteBates(EnumDeleteUnicode enumDeleteunicode)
        {
            if (enumDeleteunicode.Status == EnumDelete.StatusDeleteAll)
            {
                EnumDelete enumDelete = enumDeleteunicode.Status;
                EnumDelete = enumDelete;
                PDFViewer.Document.GetBates().Clear();
                PDFViewer.Document.ReleasePages();
                PDFViewer.ReloadDocument();
            }
        }


        public void SetBates(BatesInfoUnicode batesInfoUnicode)
        {
            EnumDelete = EnumDelete.StatusCreate;
            BatesInfo batesInfo = batesInfoUnicode.Status;
            EditToolsHelper.GetPageRange(batesInfo.PageRangeIndex, PDFViewer.Document, ref batesInfo.PageRange, batesInfo.PageRange);
            this.batesInfo = batesInfo;
            PDFViewer.InvalidChildVisual(false);
        }

        public void CreateBates(CPDFDocument document, bool IsNewDoc = false)
        {
            if (batesInfo != null)
            {
                if (bates != null) { bates.Release(); }
                bates = document.GetBates();
                for (int i = 0; i < 6; i++)
                {
                    bates.SetText(i, batesInfo.TextData[i].text);

                    bates.SetFontName(i, batesInfo.TextData[i].fontName);
                    bates.SetFontSize(i, batesInfo.TextData[i].fontSize);

                    bates.SetTextColor(i, batesInfo.TextData[i].Color);
                }
                bates.SetPageOffset(batesInfo.StarPagetNumber + 1);
                if (IsNewDoc) { bates.SetPages("0"); }
                else
                {
                    bates.SetPages(batesInfo.PageRange);
                }
                bates.SetMargin(batesInfo.margin);
                bates.Update();
                
            }
        }


        private void UndoManager_PropertyChanged(object sender, System.ComponentModel.PropertyChangedEventArgs e)
        {

        }

        private void CurrentViewer_CustomDrawHandler(object sender, CustomDrawData e)
        {
            if (e.DrawPages.Count > 0 && e.DrawContext != null)
            {
                List<int> PageIndexLists = new List<int>();
                foreach (DrawPageData drawPageData in e.DrawPages)
                {
                    if (batesInfo != null)
                    {
                        char[] enumerationSeparator = new char[] { ',' };
                        char[] rangeSeparator = new char[] { '-' };
                        if (CommonHelper.GetPagesInRange(ref PageIndexLists, batesInfo.PageRange, Document.PageCount, enumerationSeparator, rangeSeparator, true))
                        { //TODO
                            if (PageIndexLists.Contains(drawPageData.PageIndex - 1))
                            {
                                WriteableBitmap backgroundBitmap = GetBates(PDFViewer.Document, e.Zoom, drawPageData.PageIndex);
                                e.DrawContext.DrawImage(backgroundBitmap, drawPageData.PageBound);
                            }
                        }
                    }
                }
            }
            CurrentPageIndex = PDFViewer.CurrentIndex + 1;
        }

        private WriteableBitmap GetBates(CPDFDocument oldDoc, double zoom, int pageIndex)
        {
            Size pageSize = oldDoc.GetPageSize(pageIndex);

            CPDFDocument newDoc = CPDFDocument.CreateDocument();
            newDoc.InsertPage(0, pageSize.Width, pageSize.Height, null);
            CreateBates(newDoc, true);
            CPDFPage newPage = newDoc.PageAtIndex(0);
            double scale = 96.0 / 72.0;
            zoom = zoom * 1.5;
            Rect renderRect = new Rect(0, 0, (int)(pageSize.Width * scale), (int)(pageSize.Height * scale));
            byte[] imageArray = new byte[(int)(renderRect.Width * renderRect.Height * 4)];
            newPage.RenderPageBitmapWithMatrix(1 / (float)zoom, renderRect, 0x00FFFFFF, imageArray, 1, true);
            WriteableBitmap WirteBitmap = new WriteableBitmap((int)renderRect.Width, (int)renderRect.Height, 96, 96, PixelFormats.Bgra32, null);
            WirteBitmap.WritePixels(new Int32Rect(0, 0, (int)renderRect.Width, (int)renderRect.Height), imageArray, WirteBitmap.BackBufferStride, 0);
            newDoc.Release();
            return WirteBitmap;
        }

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

        public void OnNavigatedFrom(NavigationContext navigationContext)
        {
        }

        public void OnNavigatedTo(NavigationContext navigationContext)
        {
            navigationContext.Parameters.TryGetValue<CPDFViewer>(ParameterNames.PDFViewer, out pdfViewer);
            navigationContext.Parameters.TryGetValue<ViewContentViewModel>(ParameterNames.ViewContentViewModel, out viewContentViewModel);
            if (pdfViewer != null)
            {
                PDFViewer = new CPDFViewer();
                PDFViewer.InitDocument(pdfViewer.Document);
                Document = PDFViewer.Document;
                PDFViewer.CustomDrawHandler += CurrentViewer_CustomDrawHandler;
                PDFViewer.UndoManager.PropertyChanged += UndoManager_PropertyChanged;
                regionManager.AddToRegion(ViewerRegionName, PDFViewer);
                PDFViewer.SetAnnotInteraction(!PDFViewer.GetAnnotInteraction());
                PDFViewer.Load();
                PDFViewer.ChangeViewMode(ViewMode.Single);
                PDFViewer.SetMouseMode(MouseModes.PanTool);
                PageRangeNumber = PDFViewer.Document.PageCount;
                PDFViewer.Zoom(0.5);
            }
        }
    }
}