修改命名空间

2026-02-21 16:31:24 +08:00
parent 97c0b18dc7
commit 2ad3d0fde0
188 changed files with 783 additions and 2710 deletions
--- a/ShrlAlgoToolkit.RevitAddins/Common/Assists/WindowsThumbnailProvider.cs
+++ b/ShrlAlgoToolkit.RevitAddins/Common/Assists/WindowsThumbnailProvider.cs
@@ -0,0 +1,188 @@
+using System.Drawing;
+using System.Drawing.Imaging;
+using System.IO;
+using System.Runtime.InteropServices;
+
+using ShrlAlgoToolkit.RevitAddins.Common.Assists;
+using ShrlAlgoToolkit;
+using ShrlAlgoToolkit.RevitAddins;
+
+namespace ShrlAlgoToolkit.RevitAddins.Common.Assists;
+
+/// <summary>
+///     获取Windows缩略图
+/// </summary>
+public class WindowsThumbnailProvider
+{
+	private const string ShellItem2Guid = "7E9FB0D3-919F-4307-AB2E-9B1860310C93";
+
+	public static Bitmap CreateAlphaBitmap(Bitmap srcBitmap, PixelFormat targetPixelFormat)
+	{
+		var result = new Bitmap(srcBitmap.Width, srcBitmap.Height, targetPixelFormat);
+
+		var bmpBounds = new System.Drawing.Rectangle(0, 0, srcBitmap.Width, srcBitmap.Height);
+
+		var srcData = srcBitmap.LockBits(bmpBounds, ImageLockMode.ReadOnly, srcBitmap.PixelFormat);
+
+		var isAlplaBitmap = false;
+
+		try
+		{
+			for (var y = 0; y <= srcData.Height - 1; y++)
+			{
+				for (var x = 0; x <= srcData.Width - 1; x++)
+				{
+					var pixelColor = System.Drawing.Color.FromArgb(Marshal.ReadInt32(srcData.Scan0, (srcData.Stride * y) + (4 * x)));
+
+					if ((pixelColor.A > 0) & (pixelColor.A < 255))
+					{
+						isAlplaBitmap = true;
+					}
+
+					result.SetPixel(x, y, pixelColor);
+				}
+			}
+		}
+		finally
+		{
+			srcBitmap.UnlockBits(srcData);
+		}
+
+		return isAlplaBitmap ? result : srcBitmap;
+	}
+
+	public static Bitmap GetBitmapFromHBitmap(IntPtr nativeHBitmap)
+	{
+		var bmp = Image.FromHbitmap(nativeHBitmap);
+
+		return Image.GetPixelFormatSize(bmp.PixelFormat) < 32 ? bmp : CreateAlphaBitmap(bmp, PixelFormat.Format32bppArgb);
+	}
+
+	public static Bitmap GetThumbnail(string fileName, int width, int height, Assists.ThumbnailOptions options)
+	{
+		var hBitmap = GetHBitmap(Path.GetFullPath(fileName), width, height, options);
+
+		try
+		{
+			// return a System.Drawing.Bitmap from the hBitmap
+			return GetBitmapFromHBitmap(hBitmap);
+		}
+		finally
+		{
+			// delete HBitmap to avoid memory leaks
+			DeleteObject(hBitmap);
+		}
+	}
+
+	[DllImport("gdi32.dll")]
+	[return: MarshalAs(UnmanagedType.Bool)]
+	internal static extern bool DeleteObject(IntPtr hObject);
+
+	[DllImport("shell32.dll", CharSet = CharSet.Unicode, SetLastError = true)]
+	internal static extern int SHCreateItemFromParsingName(
+		[MarshalAs(UnmanagedType.LPWStr)] string path,
+		// The following parameter is not used - binding context.
+		IntPtr pbc,
+		ref Guid riid,
+		[MarshalAs(UnmanagedType.Interface)] out Assists.WindowsThumbnailProvider.IShellItem shellItem
+	);
+
+	private static IntPtr GetHBitmap(string fileName, int width, int height, Assists.ThumbnailOptions options)
+	{
+		var shellItem2Guid = new Guid(ShellItem2Guid);
+		var retCode = SHCreateItemFromParsingName(fileName, IntPtr.Zero, ref shellItem2Guid, out var nativeShellItem);
+
+		if (retCode != 0)
+		{
+			throw Marshal.GetExceptionForHR(retCode);
+		}
+
+		var nativeSize = new Assists.WindowsThumbnailProvider.NativeSize { Width = width, Height = height };
+
+		var hr = ((Assists.WindowsThumbnailProvider.IShellItemImageFactory)nativeShellItem).GetImage(nativeSize, options, out var hBitmap);
+
+		Marshal.ReleaseComObject(nativeShellItem);
+
+		return hr == Assists.WindowsThumbnailProvider.HResult.Ok ? hBitmap : throw Marshal.GetExceptionForHR((int)hr);
+	}
+
+	internal enum HResult
+	{
+		Ok = 0x0000,
+		False = 0x0001,
+		InvalidArguments = unchecked((int)0x80070057),
+		OutOfMemory = unchecked((int)0x8007000E),
+		NoInterface = unchecked((int)0x80004002),
+		Fail = unchecked((int)0x80004005),
+		ElementNotFound = unchecked((int)0x80070490),
+		TypeElementNotFound = unchecked((int)0x8002802B),
+		NoObject = unchecked((int)0x800401E5),
+		Win32ErrorCanceled = 1223,
+		Canceled = unchecked((int)0x800704C7),
+		ResourceInUse = unchecked((int)0x800700AA),
+		AccessDenied = unchecked((int)0x80030005)
+	}
+
+	[ComImport]
+	[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
+	[Guid("43826d1e-e718-42ee-bc55-a1e261c37bfe")]
+	internal interface IShellItem
+	{
+		void BindToHandler(IntPtr pbc, [MarshalAs(UnmanagedType.LPStruct)] Guid bhid, [MarshalAs(UnmanagedType.LPStruct)] Guid riid, out IntPtr ppv);
+
+		void Compare(Assists.WindowsThumbnailProvider.IShellItem psi, uint hint, out int piOrder);
+
+		void GetAttributes(uint sfgaoMask, out uint psfgaoAttribs);
+
+		void GetDisplayName(Assists.WindowsThumbnailProvider.SIGDN sigdnName, out IntPtr ppszName);
+
+		void GetParent(out Assists.WindowsThumbnailProvider.IShellItem ppsi);
+	}
+
+	[ComImport]
+	[Guid("bcc18b79-ba16-442f-80c4-8a59c30c463b")]
+	[InterfaceType(ComInterfaceType.InterfaceIsIUnknown)]
+	internal interface IShellItemImageFactory
+	{
+		[PreserveSig]
+        Assists.WindowsThumbnailProvider.HResult GetImage([In] [MarshalAs(UnmanagedType.Struct)] Assists.WindowsThumbnailProvider.NativeSize size, [In] Assists.ThumbnailOptions flags, [Out] out IntPtr phbm);
+	}
+
+	[StructLayout(LayoutKind.Sequential)]
+	internal struct NativeSize
+	{
+		private int width;
+		private int height;
+
+		public int Width
+		{
+			set => width = value;
+		}
+
+		public int Height
+		{
+			set => height = value;
+		}
+	}
+
+	[StructLayout(LayoutKind.Sequential)]
+	public struct RGBQUAD
+	{
+		public byte rgbBlue;
+		public byte rgbGreen;
+		public byte rgbRed;
+		public byte rgbReserved;
+	}
+
+	internal enum SIGDN : uint
+	{
+		NORMALDISPLAY = 0,
+		PARENTRELATIVEPARSING = 0x80018001,
+		PARENTRELATIVEFORADDRESSBAR = 0x8001c001,
+		DESKTOPABSOLUTEPARSING = 0x80028000,
+		PARENTRELATIVEEDITING = 0x80031001,
+		DESKTOPABSOLUTEEDITING = 0x8004c000,
+		FILESYSPATH = 0x80058000,
+		URL = 0x80068000
+	}
+}