mirror of
https://github.com/ShareX/ShareX.git
synced 2024-07-02 21:20:28 +12:00
2c35af0592
Merge commits from Greenshot up through bd71f47e95a46642fe6a83bdd76cc4d2a7eed7c2. Some bug fixes, but mostly structural cleanup.
255 lines
10 KiB
C#
255 lines
10 KiB
C#
/*
|
|
* Greenshot - a free and open source screenshot tool
|
|
* Copyright (C) 2007-2015 Thomas Braun, Jens Klingen, Robin Krom
|
|
*
|
|
* For more information see: http://getgreenshot.org/
|
|
* The Greenshot project is hosted on Sourceforge: http://sourceforge.net/projects/greenshot/
|
|
*
|
|
* 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 1 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/>.
|
|
*/
|
|
|
|
using System;
|
|
using System.Drawing;
|
|
using System.IO;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
|
|
namespace GreenshotPlugin.UnmanagedHelpers
|
|
{
|
|
/// <summary>
|
|
/// Description of Shell32.
|
|
/// </summary>
|
|
public static class Shell32
|
|
{
|
|
[DllImport("shell32", CharSet = CharSet.Unicode)]
|
|
public static extern int ExtractIconEx(string sFile, int iIndex, out IntPtr piLargeVersion, out IntPtr piSmallVersion, int amountIcons);
|
|
|
|
[DllImport("shell32", CharSet = CharSet.Unicode)]
|
|
internal static extern IntPtr ExtractAssociatedIcon(HandleRef hInst, StringBuilder iconPath, ref int index);
|
|
|
|
[DllImport("shell32", CharSet = CharSet.Unicode)]
|
|
private static extern IntPtr SHGetFileInfo(string pszPath, uint dwFileAttributes, ref SHFILEINFO psfi, uint cbFileInfo, uint uFlags);
|
|
|
|
#region Structs
|
|
|
|
[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
|
|
private struct SHFILEINFO
|
|
{
|
|
public readonly IntPtr hIcon;
|
|
public readonly int iIcon;
|
|
public readonly uint dwAttributes;
|
|
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = 260)]
|
|
public readonly string szDisplayName;
|
|
[MarshalAs(UnmanagedType.ByValTStr, SizeConst = 80)]
|
|
public readonly string szTypeName;
|
|
};
|
|
|
|
#endregion Structs
|
|
|
|
#region Constants
|
|
|
|
// Browsing for directory.
|
|
private const uint BIF_RETURNONLYFSDIRS = 0x0001;
|
|
private const uint BIF_DONTGOBELOWDOMAIN = 0x0002;
|
|
private const uint BIF_STATUSTEXT = 0x0004;
|
|
private const uint BIF_RETURNFSANCESTORS = 0x0008;
|
|
private const uint BIF_EDITBOX = 0x0010;
|
|
private const uint BIF_VALIDATE = 0x0020;
|
|
private const uint BIF_NEWDIALOGSTYLE = 0x0040;
|
|
private const uint BIF_USENEWUI = (BIF_NEWDIALOGSTYLE | BIF_EDITBOX);
|
|
private const uint BIF_BROWSEINCLUDEURLS = 0x0080;
|
|
private const uint BIF_BROWSEFORCOMPUTER = 0x1000;
|
|
private const uint BIF_BROWSEFORPRINTER = 0x2000;
|
|
private const uint BIF_BROWSEINCLUDEFILES = 0x4000;
|
|
private const uint BIF_SHAREABLE = 0x8000;
|
|
|
|
private const uint SHGFI_ICON = 0x000000100; // get icon
|
|
private const uint SHGFI_DISPLAYNAME = 0x000000200; // get display name
|
|
private const uint SHGFI_TYPENAME = 0x000000400; // get type name
|
|
private const uint SHGFI_ATTRIBUTES = 0x000000800; // get attributes
|
|
private const uint SHGFI_ICONLOCATION = 0x000001000; // get icon location
|
|
private const uint SHGFI_EXETYPE = 0x000002000; // return exe type
|
|
private const uint SHGFI_SYSICONINDEX = 0x000004000; // get system icon index
|
|
private const uint SHGFI_LINKOVERLAY = 0x000008000; // put a link overlay on icon
|
|
private const uint SHGFI_SELECTED = 0x000010000; // show icon in selected state
|
|
private const uint SHGFI_ATTR_SPECIFIED = 0x000020000; // get only specified attributes
|
|
private const uint SHGFI_LARGEICON = 0x000000000; // get large icon
|
|
private const uint SHGFI_SMALLICON = 0x000000001; // get small icon
|
|
private const uint SHGFI_OPENICON = 0x000000002; // get open icon
|
|
private const uint SHGFI_SHELLICONSIZE = 0x000000004; // get shell size icon
|
|
private const uint SHGFI_PIDL = 0x000000008; // pszPath is a pidl
|
|
private const uint SHGFI_USEFILEATTRIBUTES = 0x000000010; // use passed dwFileAttribute
|
|
private const uint SHGFI_ADDOVERLAYS = 0x000000020; // apply the appropriate overlays
|
|
private const uint SHGFI_OVERLAYINDEX = 0x000000040; // Get the index of the overlay
|
|
|
|
private const uint FILE_ATTRIBUTE_DIRECTORY = 0x00000010;
|
|
private const uint FILE_ATTRIBUTE_NORMAL = 0x00000080;
|
|
|
|
#endregion Constants
|
|
|
|
/// <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>
|
|
/// Options to specify whether folders should be in the open or closed state.
|
|
/// </summary>
|
|
public enum FolderType
|
|
{
|
|
/// <summary>
|
|
/// Specify open folder.
|
|
/// </summary>
|
|
Open = 0,
|
|
/// <summary>
|
|
/// Specify closed folder.
|
|
/// </summary>
|
|
Closed = 1
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns an icon for a given file extension - indicated by the name parameter.
|
|
/// See: http://msdn.microsoft.com/en-us/library/windows/desktop/bb762179(v=vs.85).aspx
|
|
/// </summary>
|
|
/// <param name="filename">Filename</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 filename, IconSize size, bool linkOverlay)
|
|
{
|
|
SHFILEINFO shfi = new SHFILEINFO();
|
|
// SHGFI_USEFILEATTRIBUTES makes it simulate, just gets the icon for the extension
|
|
uint flags = SHGFI_ICON | SHGFI_USEFILEATTRIBUTES;
|
|
|
|
if (linkOverlay)
|
|
{
|
|
flags += SHGFI_LINKOVERLAY;
|
|
}
|
|
|
|
// Check the size specified for return.
|
|
if (IconSize.Small == size)
|
|
{
|
|
flags += SHGFI_SMALLICON;
|
|
}
|
|
else
|
|
{
|
|
flags += SHGFI_LARGEICON;
|
|
}
|
|
|
|
SHGetFileInfo(Path.GetFileName(filename), FILE_ATTRIBUTE_NORMAL, ref shfi, (uint)Marshal.SizeOf(shfi), flags);
|
|
|
|
// Only return an icon if we really got one
|
|
if (shfi.hIcon != IntPtr.Zero)
|
|
{
|
|
// Copy (clone) the returned icon to a new object, thus allowing us to clean-up properly
|
|
Icon icon = (Icon)Icon.FromHandle(shfi.hIcon).Clone();
|
|
// Cleanup
|
|
User32.DestroyIcon(shfi.hIcon);
|
|
return icon;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Used to access system folder icons.
|
|
/// </summary>
|
|
/// <param name="size">Specify large or small icons.</param>
|
|
/// <param name="folderType">Specify open or closed FolderType.</param>
|
|
/// <returns>System.Drawing.Icon</returns>
|
|
public static Icon GetFolderIcon(IconSize size, FolderType folderType)
|
|
{
|
|
// Need to add size check, although errors generated at present!
|
|
uint flags = SHGFI_ICON | SHGFI_USEFILEATTRIBUTES;
|
|
|
|
if (FolderType.Open == folderType)
|
|
{
|
|
flags += SHGFI_OPENICON;
|
|
}
|
|
|
|
if (IconSize.Small == size)
|
|
{
|
|
flags += SHGFI_SMALLICON;
|
|
}
|
|
else
|
|
{
|
|
flags += SHGFI_LARGEICON;
|
|
}
|
|
|
|
// Get the folder icon
|
|
SHFILEINFO shfi = new SHFILEINFO();
|
|
SHGetFileInfo(null, FILE_ATTRIBUTE_DIRECTORY, ref shfi, (uint)Marshal.SizeOf(shfi), flags);
|
|
|
|
//Icon.FromHandle(shfi.hIcon); // Load the icon from an HICON handle
|
|
// Now clone the icon, so that it can be successfully stored in an ImageList
|
|
Icon icon = (Icon)Icon.FromHandle(shfi.hIcon).Clone();
|
|
|
|
// Cleanup
|
|
User32.DestroyIcon(shfi.hIcon);
|
|
return icon;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns an icon representation of an image contained in the specified file.
|
|
/// This function is identical to System.Drawing.Icon.ExtractAssociatedIcon, xcept this version works.
|
|
/// See: http://stackoverflow.com/questions/1842226/how-to-get-the-associated-icon-from-a-network-share-file
|
|
/// </summary>
|
|
/// <param name="filePath">The path to the file that contains an image.</param>
|
|
/// <returns>The System.Drawing.Icon representation of the image contained in the specified file.</returns>
|
|
public static Icon ExtractAssociatedIcon(String filePath)
|
|
{
|
|
int index = 0;
|
|
|
|
Uri uri;
|
|
if (filePath == null)
|
|
{
|
|
throw new ArgumentException(String.Format("'{0}' is not valid for '{1}'", "null", "filePath"), "filePath");
|
|
}
|
|
try
|
|
{
|
|
uri = new Uri(filePath);
|
|
}
|
|
catch (UriFormatException)
|
|
{
|
|
filePath = Path.GetFullPath(filePath);
|
|
uri = new Uri(filePath);
|
|
}
|
|
|
|
if (uri.IsFile)
|
|
{
|
|
if (File.Exists(filePath))
|
|
{
|
|
StringBuilder iconPath = new StringBuilder(1024);
|
|
iconPath.Append(filePath);
|
|
|
|
IntPtr handle = ExtractAssociatedIcon(new HandleRef(null, IntPtr.Zero), iconPath, ref index);
|
|
if (handle != IntPtr.Zero)
|
|
{
|
|
return Icon.FromHandle(handle);
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
} |