页面视图新增缩略图模式

AIStudio.Wpf.DiagramDesigner/Helpers/BitmapHelper.cs

@@ -0,0 +1,196 @@
+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;
+            }
+        }
+
+    }
+}