mirror of
https://github.com/QL-Win/QuickLook.git
synced 2025-09-04 03:36:04 +00:00
Code Cleanup
This commit is contained in:
ema
@@ -1,17 +1,17 @@
|
||||
// Copyright © 2017 Paddy Xu
|
||||
//
|
||||
//
|
||||
// This file is part of QuickLook program.
|
||||
//
|
||||
//
|
||||
// This program is free software: you can redistribute it and/or modify
|
||||
// it under the terms of the GNU General Public License as published by
|
||||
// the Free Software Foundation, either version 3 of the License, or
|
||||
// (at your option) any later version.
|
||||
//
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
//
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
@@ -25,212 +25,213 @@ using System.Windows.Interop;
|
||||
using System.Windows.Media;
|
||||
using System.Windows.Media.Imaging;
|
||||
|
||||
namespace QuickLook.Plugin.ArchiveViewer
|
||||
namespace QuickLook.Plugin.ArchiveViewer;
|
||||
|
||||
/// <summary>
|
||||
/// Internals are mostly from here:
|
||||
/// http://www.codeproject.com/Articles/2532/Obtaining-and-managing-file-and-folder-icons-using
|
||||
/// Caches all results.
|
||||
/// </summary>
|
||||
public static class IconManager
|
||||
{
|
||||
/// <summary>
|
||||
/// Internals are mostly from here:
|
||||
/// http://www.codeproject.com/Articles/2532/Obtaining-and-managing-file-and-folder-icons-using
|
||||
/// Caches all results.
|
||||
/// </summary>
|
||||
public static class IconManager
|
||||
private static ImageSource SmallDirIcon;
|
||||
private static ImageSource LargeDirIcon;
|
||||
private static readonly Dictionary<string, ImageSource> SmallIconCache = [];
|
||||
private static readonly Dictionary<string, ImageSource> LargeIconCache = [];
|
||||
|
||||
public static void ClearCache()
|
||||
{
|
||||
private static ImageSource SmallDirIcon;
|
||||
private static ImageSource LargeDirIcon;
|
||||
private static readonly Dictionary<string, ImageSource> SmallIconCache = new Dictionary<string, ImageSource>();
|
||||
private static readonly Dictionary<string, ImageSource> LargeIconCache = new Dictionary<string, ImageSource>();
|
||||
SmallDirIcon = LargeDirIcon = null;
|
||||
|
||||
public static void ClearCache()
|
||||
SmallIconCache.Clear();
|
||||
LargeIconCache.Clear();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Get the icon of a directory
|
||||
/// </summary>
|
||||
/// <param name="large">16x16 or 32x32 icon</param>
|
||||
/// <returns>an icon</returns>
|
||||
public static ImageSource FindIconForDir(bool large)
|
||||
{
|
||||
var icon = large ? LargeDirIcon : SmallDirIcon;
|
||||
if (icon != null)
|
||||
return icon;
|
||||
icon = IconReader.GetFolderIcon(large ? IconReader.IconSize.Large : IconReader.IconSize.Small,
|
||||
false)
|
||||
.ToImageSource();
|
||||
if (large)
|
||||
LargeDirIcon = icon;
|
||||
else
|
||||
SmallDirIcon = icon;
|
||||
return icon;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Get an icon for a given filename
|
||||
/// </summary>
|
||||
/// <param name="fileName">any filename</param>
|
||||
/// <param name="large">16x16 or 32x32 icon</param>
|
||||
/// <returns>null if path is null, otherwise - an icon</returns>
|
||||
public static ImageSource FindIconForFilename(string fileName, bool large)
|
||||
{
|
||||
var extension = Path.GetExtension(fileName);
|
||||
if (extension == null)
|
||||
return null;
|
||||
var cache = large ? LargeIconCache : SmallIconCache;
|
||||
if (cache.TryGetValue(extension, out ImageSource icon))
|
||||
return icon;
|
||||
icon = IconReader.GetFileIcon(fileName, large ? IconReader.IconSize.Large : IconReader.IconSize.Small,
|
||||
false)
|
||||
.ToImageSource();
|
||||
cache.Add(extension, icon);
|
||||
return icon;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// http://stackoverflow.com/a/6580799/1943849
|
||||
/// </summary>
|
||||
private static ImageSource ToImageSource(this Icon icon)
|
||||
{
|
||||
var imageSource = Imaging.CreateBitmapSourceFromHIcon(
|
||||
icon.Handle,
|
||||
Int32Rect.Empty,
|
||||
BitmapSizeOptions.FromEmptyOptions());
|
||||
return imageSource;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Provides static methods to read system icons for both folders and files.
|
||||
/// </summary>
|
||||
/// <example>
|
||||
/// <code>IconReader.GetFileIcon("c:\\general.xls");</code>
|
||||
/// </example>
|
||||
private static class IconReader
|
||||
{
|
||||
/// <summary>
|
||||
/// Options to specify the size of icons to return.
|
||||
/// </summary>
|
||||
public enum IconSize
|
||||
{
|
||||
SmallDirIcon = LargeDirIcon = null;
|
||||
/// <summary>
|
||||
/// Specify large icon - 32 pixels by 32 pixels.
|
||||
/// </summary>
|
||||
Large = 0,
|
||||
|
||||
SmallIconCache.Clear();
|
||||
LargeIconCache.Clear();
|
||||
/// <summary>
|
||||
/// Specify small icon - 16 pixels by 16 pixels.
|
||||
/// </summary>
|
||||
Small = 1
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Get the icon of a directory
|
||||
/// Returns the icon of a folder.
|
||||
/// </summary>
|
||||
/// <param name="large">16x16 or 32x32 icon</param>
|
||||
/// <returns>an icon</returns>
|
||||
public static ImageSource FindIconForDir(bool large)
|
||||
/// <param name="size">Large or small</param>
|
||||
/// <param name="linkOverlay">Whether to include the link icon</param>
|
||||
/// <returns>System.Drawing.Icon</returns>
|
||||
public static Icon GetFolderIcon(IconSize size, bool linkOverlay)
|
||||
{
|
||||
var icon = large ? LargeDirIcon : SmallDirIcon;
|
||||
if (icon != null)
|
||||
return icon;
|
||||
icon = IconReader.GetFolderIcon(large ? IconReader.IconSize.Large : IconReader.IconSize.Small,
|
||||
false)
|
||||
.ToImageSource();
|
||||
if (large)
|
||||
LargeDirIcon = icon;
|
||||
var shfi = new Shell32.Shfileinfo();
|
||||
var flags = Shell32.ShgfiIcon | Shell32.ShgfiUsefileattributes;
|
||||
if (linkOverlay) flags += Shell32.ShgfiLinkoverlay;
|
||||
/* Check the size specified for return. */
|
||||
if (IconSize.Small == size)
|
||||
flags += Shell32.ShgfiSmallicon;
|
||||
else
|
||||
SmallDirIcon = icon;
|
||||
flags += Shell32.ShgfiLargeicon;
|
||||
Shell32.SHGetFileInfo("placeholder",
|
||||
Shell32.FileAttributeDirectory,
|
||||
ref shfi,
|
||||
(uint)Marshal.SizeOf(shfi),
|
||||
flags);
|
||||
// Copy (clone) the returned icon to a new object, thus allowing us to clean-up properly
|
||||
var icon = (Icon)Icon.FromHandle(shfi.hIcon).Clone();
|
||||
User32.DestroyIcon(shfi.hIcon); // Cleanup
|
||||
return icon;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Get an icon for a given filename
|
||||
/// Returns an icon for a given file - indicated by the name parameter.
|
||||
/// </summary>
|
||||
/// <param name="fileName">any filename</param>
|
||||
/// <param name="large">16x16 or 32x32 icon</param>
|
||||
/// <returns>null if path is null, otherwise - an icon</returns>
|
||||
public static ImageSource FindIconForFilename(string fileName, bool large)
|
||||
/// <param name="name">Pathname for file.</param>
|
||||
/// <param name="size">Large or small</param>
|
||||
/// <param name="linkOverlay">Whether to include the link icon</param>
|
||||
/// <returns>System.Drawing.Icon</returns>
|
||||
public static Icon GetFileIcon(string name, IconSize size, bool linkOverlay)
|
||||
{
|
||||
var extension = Path.GetExtension(fileName);
|
||||
if (extension == null)
|
||||
return null;
|
||||
var cache = large ? LargeIconCache : SmallIconCache;
|
||||
ImageSource icon;
|
||||
if (cache.TryGetValue(extension, out icon))
|
||||
return icon;
|
||||
icon = IconReader.GetFileIcon(fileName, large ? IconReader.IconSize.Large : IconReader.IconSize.Small,
|
||||
false)
|
||||
.ToImageSource();
|
||||
cache.Add(extension, icon);
|
||||
var shfi = new Shell32.Shfileinfo();
|
||||
var flags = Shell32.ShgfiIcon | Shell32.ShgfiUsefileattributes;
|
||||
if (linkOverlay) flags += Shell32.ShgfiLinkoverlay;
|
||||
/* Check the size specified for return. */
|
||||
if (IconSize.Small == size)
|
||||
flags += Shell32.ShgfiSmallicon;
|
||||
else
|
||||
flags += Shell32.ShgfiLargeicon;
|
||||
Shell32.SHGetFileInfo(name,
|
||||
Shell32.FileAttributeNormal,
|
||||
ref shfi,
|
||||
(uint)Marshal.SizeOf(shfi),
|
||||
flags);
|
||||
// Copy (clone) the returned icon to a new object, thus allowing us to clean-up properly
|
||||
var icon = (Icon)Icon.FromHandle(shfi.hIcon).Clone();
|
||||
User32.DestroyIcon(shfi.hIcon); // Cleanup
|
||||
return icon;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// http://stackoverflow.com/a/6580799/1943849
|
||||
/// </summary>
|
||||
private static ImageSource ToImageSource(this Icon icon)
|
||||
{
|
||||
var imageSource = Imaging.CreateBitmapSourceFromHIcon(
|
||||
icon.Handle,
|
||||
Int32Rect.Empty,
|
||||
BitmapSizeOptions.FromEmptyOptions());
|
||||
return imageSource;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Provides static methods to read system icons for both folders and files.
|
||||
/// </summary>
|
||||
/// <example>
|
||||
/// <code>IconReader.GetFileIcon("c:\\general.xls");</code>
|
||||
/// </example>
|
||||
private static class IconReader
|
||||
{
|
||||
/// <summary>
|
||||
/// Options to specify the size of icons to return.
|
||||
/// </summary>
|
||||
public enum IconSize
|
||||
{
|
||||
/// <summary>
|
||||
/// Specify large icon - 32 pixels by 32 pixels.
|
||||
/// </summary>
|
||||
Large = 0,
|
||||
/// <summary>
|
||||
/// Specify small icon - 16 pixels by 16 pixels.
|
||||
/// </summary>
|
||||
Small = 1
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Returns the icon of a folder.
|
||||
/// </summary>
|
||||
/// <param name="size">Large or small</param>
|
||||
/// <param name="linkOverlay">Whether to include the link icon</param>
|
||||
/// <returns>System.Drawing.Icon</returns>
|
||||
public static Icon GetFolderIcon(IconSize size, bool linkOverlay)
|
||||
{
|
||||
var shfi = new Shell32.Shfileinfo();
|
||||
var flags = Shell32.ShgfiIcon | Shell32.ShgfiUsefileattributes;
|
||||
if (linkOverlay) flags += Shell32.ShgfiLinkoverlay;
|
||||
/* Check the size specified for return. */
|
||||
if (IconSize.Small == size)
|
||||
flags += Shell32.ShgfiSmallicon;
|
||||
else
|
||||
flags += Shell32.ShgfiLargeicon;
|
||||
Shell32.SHGetFileInfo("placeholder",
|
||||
Shell32.FileAttributeDirectory,
|
||||
ref shfi,
|
||||
(uint) Marshal.SizeOf(shfi),
|
||||
flags);
|
||||
// Copy (clone) the returned icon to a new object, thus allowing us to clean-up properly
|
||||
var icon = (Icon) Icon.FromHandle(shfi.hIcon).Clone();
|
||||
User32.DestroyIcon(shfi.hIcon); // Cleanup
|
||||
return icon;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Returns an icon for a given file - indicated by the name parameter.
|
||||
/// </summary>
|
||||
/// <param name="name">Pathname for file.</param>
|
||||
/// <param name="size">Large or small</param>
|
||||
/// <param name="linkOverlay">Whether to include the link icon</param>
|
||||
/// <returns>System.Drawing.Icon</returns>
|
||||
public static Icon GetFileIcon(string name, IconSize size, bool linkOverlay)
|
||||
{
|
||||
var shfi = new Shell32.Shfileinfo();
|
||||
var flags = Shell32.ShgfiIcon | Shell32.ShgfiUsefileattributes;
|
||||
if (linkOverlay) flags += Shell32.ShgfiLinkoverlay;
|
||||
/* Check the size specified for return. */
|
||||
if (IconSize.Small == size)
|
||||
flags += Shell32.ShgfiSmallicon;
|
||||
else
|
||||
flags += Shell32.ShgfiLargeicon;
|
||||
Shell32.SHGetFileInfo(name,
|
||||
Shell32.FileAttributeNormal,
|
||||
ref shfi,
|
||||
(uint) Marshal.SizeOf(shfi),
|
||||
flags);
|
||||
// Copy (clone) the returned icon to a new object, thus allowing us to clean-up properly
|
||||
var icon = (Icon) Icon.FromHandle(shfi.hIcon).Clone();
|
||||
User32.DestroyIcon(shfi.hIcon); // Cleanup
|
||||
return icon;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wraps necessary Shell32.dll structures and functions required to retrieve Icon Handles using SHGetFileInfo. Code
|
||||
/// courtesy of MSDN Cold Rooster Consulting case study.
|
||||
/// </summary>
|
||||
private static class Shell32
|
||||
{
|
||||
private const int MaxPath = 256;
|
||||
public const uint ShgfiIcon = 0x000000100; // get icon
|
||||
public const uint ShgfiLinkoverlay = 0x000008000; // put a link overlay on icon
|
||||
public const uint ShgfiLargeicon = 0x000000000; // get large icon
|
||||
public const uint ShgfiSmallicon = 0x000000001; // get small icon
|
||||
public const uint ShgfiUsefileattributes = 0x000000010; // use passed dwFileAttribute
|
||||
public const uint FileAttributeNormal = 0x00000080;
|
||||
public const uint FileAttributeDirectory = 0x00000010;
|
||||
|
||||
[DllImport("Shell32.dll")]
|
||||
public static extern IntPtr SHGetFileInfo(
|
||||
string pszPath,
|
||||
uint dwFileAttributes,
|
||||
ref Shfileinfo psfi,
|
||||
uint cbFileInfo,
|
||||
uint uFlags
|
||||
);
|
||||
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
public struct Shfileinfo
|
||||
{
|
||||
private const int Namesize = 80;
|
||||
public readonly IntPtr hIcon;
|
||||
private readonly int iIcon;
|
||||
private readonly uint dwAttributes;
|
||||
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = MaxPath)]
|
||||
private readonly string szDisplayName;
|
||||
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = Namesize)]
|
||||
private readonly string szTypeName;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wraps necessary functions imported from User32.dll. Code courtesy of MSDN Cold Rooster Consulting example.
|
||||
/// </summary>
|
||||
private static class User32
|
||||
{
|
||||
/// <summary>
|
||||
/// Provides access to function required to delete handle. This method is used internally
|
||||
/// and is not required to be called separately.
|
||||
/// </summary>
|
||||
/// <param name="hIcon">Pointer to icon handle.</param>
|
||||
/// <returns>N/A</returns>
|
||||
[DllImport("User32.dll")]
|
||||
public static extern int DestroyIcon(IntPtr hIcon);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wraps necessary Shell32.dll structures and functions required to retrieve Icon Handles using SHGetFileInfo. Code
|
||||
/// courtesy of MSDN Cold Rooster Consulting case study.
|
||||
/// </summary>
|
||||
private static class Shell32
|
||||
{
|
||||
private const int MaxPath = 256;
|
||||
public const uint ShgfiIcon = 0x000000100; // get icon
|
||||
public const uint ShgfiLinkoverlay = 0x000008000; // put a link overlay on icon
|
||||
public const uint ShgfiLargeicon = 0x000000000; // get large icon
|
||||
public const uint ShgfiSmallicon = 0x000000001; // get small icon
|
||||
public const uint ShgfiUsefileattributes = 0x000000010; // use passed dwFileAttribute
|
||||
public const uint FileAttributeNormal = 0x00000080;
|
||||
public const uint FileAttributeDirectory = 0x00000010;
|
||||
|
||||
[DllImport("Shell32.dll")]
|
||||
public static extern IntPtr SHGetFileInfo(
|
||||
string pszPath,
|
||||
uint dwFileAttributes,
|
||||
ref Shfileinfo psfi,
|
||||
uint cbFileInfo,
|
||||
uint uFlags
|
||||
);
|
||||
|
||||
[StructLayout(LayoutKind.Sequential)]
|
||||
public struct Shfileinfo
|
||||
{
|
||||
private const int Namesize = 80;
|
||||
public readonly IntPtr hIcon;
|
||||
private readonly int iIcon;
|
||||
private readonly uint dwAttributes;
|
||||
|
||||
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = MaxPath)]
|
||||
private readonly string szDisplayName;
|
||||
|
||||
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = Namesize)]
|
||||
private readonly string szTypeName;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Wraps necessary functions imported from User32.dll. Code courtesy of MSDN Cold Rooster Consulting example.
|
||||
/// </summary>
|
||||
private static class User32
|
||||
{
|
||||
/// <summary>
|
||||
/// Provides access to function required to delete handle. This method is used internally
|
||||
/// and is not required to be called separately.
|
||||
/// </summary>
|
||||
/// <param name="hIcon">Pointer to icon handle.</param>
|
||||
/// <returns>N/A</returns>
|
||||
[DllImport("User32.dll")]
|
||||
public static extern int DestroyIcon(IntPtr hIcon);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user