using System;
using System.Drawing;
using System.Drawing.Imaging;
using System.IO;
using System.Runtime.InteropServices.ComTypes;
using System.Windows;
using System.Windows.Media.Imaging;

namespace AIStudio.Wpf.DiagramDesigner
{
    public static class BitmapHelper
    {

        /// <summary>
        /// 截图转换成bitmap
        /// </summary>
        /// <param name="element"></param>
        /// <param name="width">默认控件宽度</param>
        /// <param name="height">默认控件高度</param>
        /// <param name="x">默认0</param>
        /// <param name="y">默认0</param>
        /// <returns></returns>
        public static Bitmap ToBitmap(this FrameworkElement element, int width = 0, int height = 0, int x = 0, int y = 0)
        {
            if (width == 0) width = (int)element.ActualWidth;
            if (height == 0) height = (int)element.ActualHeight;

            var rtb = new RenderTargetBitmap(width, height, x, y, System.Windows.Media.PixelFormats.Default);
            rtb.Render(element);
            var bit = BitmapSourceToBitmap(rtb);

            ////测试代码
            //DirectoryInfo d = new DirectoryInfo(System.IO.Path.Combine(Directory.GetCurrentDirectory(), "Cache"));
            //if (!d.Exists) d.Create();
            //bit.Save(System.IO.Path.Combine(d.FullName, "控件截图.png"));

            return bit;
        }


        public static Bitmap ToBitmap(this System.Windows.Media.VisualBrush brush, Rect rect)
        {
            var visual = new System.Windows.Media.DrawingVisual();
            using (var dc = visual.RenderOpen())
            {
                dc.DrawRectangle(brush, null, rect);
            }

            var rtb = new RenderTargetBitmap((int)rect.Width, (int)rect.Height, 96, 96, System.Windows.Media.PixelFormats.Default);
            rtb.Render(visual);

            var bit = BitmapSourceToBitmap(rtb);

            ////测试代码
            //DirectoryInfo d = new DirectoryInfo(System.IO.Path.Combine(Directory.GetCurrentDirectory(), "Cache"));
            //if (!d.Exists) d.Create();
            //bit.Save(System.IO.Path.Combine(d.FullName, "控件截图.png"));

            return bit;
        }


        /// <summary>
        /// BitmapSource转Bitmap
        /// </summary>
        /// <param name="source"></param>
        /// <returns></returns>
        public static Bitmap BitmapSourceToBitmap(this BitmapSource source)
        {
            return BitmapSourceToBitmap(source, source.PixelWidth, source.PixelHeight);
        }

        /// <summary>
        /// Convert BitmapSource to Bitmap
        /// </summary>
        /// <param name="source"></param>
        /// <returns></returns>
        public static Bitmap BitmapSourceToBitmap(this BitmapSource source, int width, int height)
        {
            Bitmap bmp = null;
            try
            {
                PixelFormat format = PixelFormat.Format24bppRgb;
                /*set the translate type according to the in param(source)*/
                switch (source.Format.ToString())
                {
                    case "Rgb24":
                    case "Bgr24": format = PixelFormat.Format24bppRgb; break;
                    case "Bgra32": format = PixelFormat.Format32bppPArgb; break;
                    case "Bgr32": format = PixelFormat.Format32bppRgb; break;
                    case "Pbgra32": format = PixelFormat.Format32bppArgb; break;
                }
                bmp = new Bitmap(width, height, format);
                BitmapData data = bmp.LockBits(new System.Drawing.Rectangle(System.Drawing.Point.Empty, bmp.Size),
                    ImageLockMode.WriteOnly,
                    format);
                source.CopyPixels(Int32Rect.Empty, data.Scan0, data.Height * data.Stride, data.Stride);
                bmp.UnlockBits(data);
            }
            catch
            {
                if (bmp != null)
                {
                    bmp.Dispose();
                    bmp = null;
                }
            }

            return bmp;
        }

        public static System.Windows.Media.ImageSource ToBitmapSource(this Bitmap bitmap, int width, int height)
        {
            MemoryStream stream = new MemoryStream();
            bitmap.Save(stream, ImageFormat.Bmp);
            stream.Position = 0;
            BitmapImage bitmapImage = new BitmapImage();
            bitmapImage.BeginInit();
            bitmapImage.CreateOptions = BitmapCreateOptions.IgnoreColorProfile;
            bitmapImage.CacheOption = BitmapCacheOption.OnLoad;
            bitmapImage.DecodePixelWidth = width;//需要缩略图的解码宽度
            bitmapImage.DecodePixelHeight = height;//缩略图的解码高度
            bitmapImage.StreamSource = stream;
            bitmapImage.EndInit();
            return bitmapImage;
        }

        public static string ToBase64String(this System.Windows.Media.Brush brush)
        {
            try
            {
                if (brush is System.Windows.Media.VisualBrush visualBrush)
                {
                    var size = ((UIElement)visualBrush.Visual).DesiredSize;
                    var image = visualBrush.ToBitmap(new Rect(size));
                    MemoryStream stream = new MemoryStream();
                    image.Save(stream, ImageFormat.Bmp);
                    stream.Position = 0;
                    var bytearray = stream.ToArray();
                    return Convert.ToBase64String(bytearray);
                }
                else if (brush is System.Windows.Media.ImageBrush imageBrush)
                {
                    var bitmap = (imageBrush.ImageSource as BitmapImage);
                    byte[] bytearray = null;
                    Stream smarket = bitmap.StreamSource; 
                    if (smarket != null && smarket.Length > 0)
                    {
                        //设置当前位置
                        smarket.Position = 0;
                        using (BinaryReader br = new BinaryReader(smarket))
                        {
                            bytearray = br.ReadBytes((int)smarket.Length);
                        }
                    }
                    return Convert.ToBase64String(bytearray);
                }
                else
                {
                    return null;
                }
            }
            catch (Exception ex)
            {
                return null;
            }
        }

        public static System.Windows.Media.Brush ToBrush(this string base64String, int width, int height)
        {
            if (string.IsNullOrEmpty(base64String))
                return null;

            try
            {
                var byteArray = Convert.FromBase64String(base64String);
                BitmapImage bitmapImage = null;
                bitmapImage = new BitmapImage();
                bitmapImage.BeginInit();
                bitmapImage.CreateOptions = BitmapCreateOptions.IgnoreColorProfile;
                bitmapImage.CacheOption = BitmapCacheOption.OnLoad;
                bitmapImage.DecodePixelWidth = width;//需要缩略图的解码宽度
                bitmapImage.DecodePixelHeight = height;//缩略图的解码高度
                bitmapImage.StreamSource = new MemoryStream(byteArray);
                bitmapImage.EndInit();

                return new System.Windows.Media.ImageBrush(bitmapImage) { Stretch = System.Windows.Media.Stretch.Uniform };
            }
            catch
            {
                return null;
            }
        }

    }
}