PDFOffice_Windows_exe/PDF Office/ViewModels/PropertyPanel/AnnotPanel/CustomCreateDialogViewModel.cs

using ComPDFKit.PDFAnnotation;
using ComPDFKitViewer.AnnotEvent;
using ComPDFKitViewer.PdfViewer;
using Microsoft.Win32;
using PDF_Office.Model;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Services.Dialogs;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Input;
using System.Windows.Media;
using System.Windows.Media.Imaging;

namespace PDF_Office.ViewModels.PropertyPanel.AnnotPanel
{
    class CustomCreateDialogViewModel : BindableBase, IDialogAware
    {
        public string Title => ".";

        public event Action<IDialogResult> RequestClose;
        public DelegateCommand CancelCommand { get; set; }

        public DelegateCommand CreateCommnad { get; set; }

        public DelegateCommand UpDataDynamicCommnad { get; set; }
        public DelegateCommand OpenImageCommnad { get; set; }

        public DelegateCommand<object> KeyDown { get; set; }

        private BitmapSource textImageSource;
        public BitmapSource TextImageSource
        {
            get { return textImageSource; }
            set
            {
                SetProperty(ref textImageSource, value);
            }
        }
        private BitmapSource imagePreviewSource;
        public BitmapSource ImagePreviewSource
        {
            get { return imagePreviewSource; }
            set
            {
                SetProperty(ref imagePreviewSource, value);
            }
        }
        private string stampText = "Stamp Text";

        public string StampText
        {
            get { return stampText; }
            set
            {
                SetProperty(ref stampText, value);
            }
        }
        private bool isCheckedDate;

        public bool IsCheckedDate
        {
            get { return isCheckedDate; }
            set
            {
                SetProperty(ref isCheckedDate, value);
                UpDataStamp();
            }
        }
        private bool isCheckedTime;

        public bool IsCheckedTime
        {
            get { return isCheckedTime; }
            set
            {
                SetProperty(ref isCheckedTime, value);
                UpDataStamp();
            }
        }

        private Visibility showImageButton = Visibility.Visible;

        public Visibility ShowImageButton
        {
            get { return showImageButton; }
            set
            {
                SetProperty(ref showImageButton, value);
            }
        }


        private C_TEXTSTAMP_SHAPE shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RECT;

        public C_TEXTSTAMP_SHAPE Shape
        {
            get { return shape; }
            set { shape = value; }
        }

        private C_TEXTSTAMP_COLOR color = C_TEXTSTAMP_COLOR.TEXTSTAMP_RED;

        public C_TEXTSTAMP_COLOR Color
        {
            get { return color; }
            set { color = value; }
        }

        public string SaveToPath { get; private set; }
        public string StampTextDate { get; private set; }
        public int StampWidth { get; private set; }
        public int StampHeight { get; private set; }
        public StampType Type { get; private set; }

        public void SetStampStyle(int index)
        {
            switch (index)
            {
                case 1:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_WHITE;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_NONE;
                    break;
                case 2:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_WHITE;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RECT;
                    break;
                case 3:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_GREEN;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RECT;
                    break;
                case 4:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_RED;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RECT;
                    break;
                case 5:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_BLUE;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RECT;
                    break;
                case 6:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_GREEN;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_LEFT_TRIANGLE;
                    break;
                case 7:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_RED;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_LEFT_TRIANGLE;
                    break;
                case 8:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_BLUE;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_LEFT_TRIANGLE;
                    break;
                case 9:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_GREEN;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RIGHT_TRIANGLE;
                    break;
                case 10:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_RED;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RIGHT_TRIANGLE;
                    break;
                case 11:
                    Color = C_TEXTSTAMP_COLOR.TEXTSTAMP_GREEN;
                    Shape = C_TEXTSTAMP_SHAPE.TEXTSTAMP_RIGHT_TRIANGLE;
                    break;
                default:
                    break;
            }
            UpDataStamp();
        }

        public CustomCreateDialogViewModel()
        {
            CancelCommand = new DelegateCommand(Cancel);
            CreateCommnad = new DelegateCommand(Create);
            UpDataDynamicCommnad = new DelegateCommand(UpDataStamp);
            KeyDown = new DelegateCommand<object>(PreviewKeyDown);
            OpenImageCommnad = new DelegateCommand(OpenImage);
        }
        private void Cancel()
        {
            RequestClose.Invoke(new DialogResult(ButtonResult.Cancel));
        }

        private void Create()
        {
            if (Type == StampType.TEXT_STAMP)
            {
                SaveImageToPath();
            }
            DialogParameters valuePairs = new DialogParameters();
            valuePairs.Add(ParameterNames.DataModel, this);
            RequestClose.Invoke(new DialogResult(ButtonResult.OK, valuePairs));
        }

        private void SaveImageToPath()
        {
            string path = App.CachePath.CustomStampPath;
            string name = Guid.NewGuid().ToString();
            if (!string.IsNullOrEmpty(path))
            {
                BitmapEncoder encoder = new PngBitmapEncoder();
                encoder.Frames.Add(BitmapFrame.Create(TextImageSource));
                path = System.IO.Path.Combine(path, name);
                using (FileStream stream = new FileStream(path, FileMode.Create))
                {
                    encoder.Save(stream);
                }
                SaveToPath = path;
            }
            else
            {
                SaveToPath = "";
            }
        }

        public bool CanCloseDialog()
        {
            return true;
        }

        public void OnDialogClosed()
        {
            return;
        }
        private void PreviewKeyDown(object e)
        {
            var args = e as KeyEventArgs;
            if (args == null)
            {
                return;
            }
            if (args.Key == Key.Enter)
            {
                StampText = (args.OriginalSource as TextBox).Text;
                UpDataStamp();
            }
        }
        public void UpDataStamp()
        {
            string date = "";
            string dateType = "";
            if (IsCheckedDate)
            {
                dateType = "yyyy-MM-dd";
            }
            if (IsCheckedTime)
            {
                dateType = dateType + " HH:mm:ss";
            }
            if (!String.IsNullOrEmpty(dateType))
            {
                date = DateTime.Now.ToString(dateType);
            }

            var bytes = CPDFStampAnnotation.GetTempTextStampImage(StampText, date,
            Shape, Color, out int stampWidth, out int stampHeight, out int width, out int height);
            if (bytes.Length > 0)
            {
                PixelFormat fmt = PixelFormats.Bgra32;
                BitmapSource bps = BitmapSource.Create(width, height, 96, 96, fmt, null, bytes, (width * fmt.BitsPerPixel + 7) / 8);
                TextImageSource = bps;
                StampTextDate = date;
                StampWidth = stampWidth;
                StampHeight = stampHeight;
                Type = StampType.TEXT_STAMP;
            }
            else
            {
                TextImageSource = null;
            }
        }

        public void OpenImage()
        {
            OpenFileDialog openFile = new OpenFileDialog();
            openFile.Filter = "All Image Files(*.bmp;*.gif;*.jpeg;*.jpg;*.png;*.tiff)|*.bmp;*.gif;*.jpeg;*.jpg;*.png;*.tiff|(*.bmp)|*.bmp|" +
                "(*.gif)|*.gif|" +
                "(*.jpeg)|*.jpeg|" +
                "(*.jpg)|*.jpg|" +
                "(*.png)|*.png|" +
                "(*.tiff)|*.tiff";
            if (openFile.ShowDialog() == false)
            {
                return;
            }
            string path = App.CachePath.CustomStampPath;
            string name = Guid.NewGuid().ToString();
            if (!string.IsNullOrEmpty(path))
            {
                BitmapImage image = new BitmapImage(new Uri(openFile.FileName));
                double scale = Math.Min((double)600 / image.PixelWidth, (double)600 / image.PixelHeight);
                scale = Math.Min(scale, 1);
                string ext = Path.GetExtension(openFile.FileName);
                if (ext.ToUpper() == ".PNG")
                {
                    BitmapEncoder encoder = new PngBitmapEncoder();
                    var targetBitmap = new TransformedBitmap(image, new ScaleTransform(scale, scale));
                    encoder.Frames.Add(BitmapFrame.Create(targetBitmap));
                    path = System.IO.Path.Combine(path, name);
                    using (FileStream stream = new FileStream(path, FileMode.Create))
                    {
                        encoder.Save(stream);
                    }
                    if (!string.IsNullOrEmpty(SaveToPath))
                    {
                        App.CachePath.AddToDeleteFiles(SaveToPath);
                    }
                    SaveToPath = path;
                    ImagePreviewSource = targetBitmap;
                    StampWidth = targetBitmap.PixelWidth;
                    StampHeight = targetBitmap.PixelHeight;
                    Type = StampType.IMAGE_STAMP;
                    ShowImageButton = Visibility.Collapsed;
                }
                else
                {
                    BitmapEncoder encoder = new JpegBitmapEncoder();
                    TransformedBitmap targetBitmap = new TransformedBitmap(image, new ScaleTransform(scale, scale));
                    encoder.Frames.Add(BitmapFrame.Create(targetBitmap));
                    path = System.IO.Path.Combine(path, name);
                    using (FileStream stream = new FileStream(path, FileMode.Create))
                    {
                        encoder.Save(stream);
                    }
                    if (!string.IsNullOrEmpty(SaveToPath))
                    {
                        App.CachePath.AddToDeleteFiles(SaveToPath);
                    }
                    SaveToPath = path;
                    ImagePreviewSource = targetBitmap;
                    StampWidth = targetBitmap.PixelWidth;
                    StampHeight = targetBitmap.PixelHeight;
                    Type = StampType.IMAGE_STAMP;
                    ShowImageButton = Visibility.Collapsed;
                }
            }
            else
            {
                SaveToPath = "";
            }
        }
        public void OnDialogOpened(IDialogParameters parameters)
        {
            UpDataStamp();
            return;
        }
    }
}