Issues #19 Support of screenshots for windows with hardware accelerated rendering

AlexanderPro · AlexanderPro · commit 19b2fc37f00b · 2025-05-01T13:27:52.000+03:00
diff --git a/SmartContextMenu/Forms/MainForm.cs b/SmartContextMenu/Forms/MainForm.cs
@@ -499,7 +499,12 @@ private void MenuItemClick(Window window, Settings.MenuItem menuItem)
                 case MenuItemName.SaveScreenshot:
                     {
                         var manager = new LanguageManager(_settings.LanguageName);
-                        var bitmap = WindowUtils.PrintWindow(window.Handle);
+                        var result = WindowUtils.PrintWindow(window.Handle, out var bitmap);
+                        if (!result || !WindowUtils.IsCorrectScreenshot(window.Handle, bitmap))
+                        {
+                            WindowUtils.CaptureWindow(window.Handle, false, out bitmap);
+                        }
+
                         var dialog = new SaveFileDialog
                         {
                             OverwritePrompt = true,
@@ -531,7 +536,12 @@ private void MenuItemClick(Window window, Settings.MenuItem menuItem)
 
                 case MenuItemName.CopyScreenshot:
                     {
-                        var bitmap = WindowUtils.PrintWindow(window.Handle);
+                        var result = WindowUtils.PrintWindow(window.Handle, out var bitmap);
+                        if (!result || !WindowUtils.IsCorrectScreenshot(window.Handle, bitmap))
+                        {
+                            WindowUtils.CaptureWindow(window.Handle, false, out bitmap);
+                        }
+
                         Clipboard.SetImage(bitmap);
                     }
                     break;
diff --git a/SmartContextMenu/Native/Constants.cs b/SmartContextMenu/Native/Constants.cs
@@ -143,5 +143,10 @@ static class Constants
         public const uint EVENT_SYSTEM_FOREGROUND = 3;
         public const uint EVENT_OBJECT_DESTROY = 0x8001;
         public const int OBJID_WINDOW = 0;
+
+        public const int CURSOR_SHOWING = 0x00000001;
+
+        public const int DI_COMPAT = 0x0004;
+        public const int DI_NORMAL = 0x0003;
     }
 }
diff --git a/SmartContextMenu/Native/Enums/CopyPixelOperations.cs b/SmartContextMenu/Native/Enums/CopyPixelOperations.cs
@@ -0,0 +1,61 @@
+﻿using System;
+
+namespace SmartContextMenu.Native.Enums
+{
+    [Flags]
+    public enum CopyPixelOperations
+    {
+        NoMirrorBitmap = -2147483648,
+
+        /// <summary>dest = BLACK, 0x00000042</summary>
+        Blackness = 66,
+
+        ///<summary>dest = (NOT src) AND (NOT dest), 0x001100A6</summary>
+        NotSourceErase = 1114278,
+
+        ///<summary>dest = (NOT source), 0x00330008</summary>
+        NotSourceCopy = 3342344,
+
+        ///<summary>dest = source AND (NOT dest), 0x00440328</summary>
+        SourceErase = 4457256,
+
+        /// <summary>dest = (NOT dest), 0x00550009</summary>
+        DestinationInvert = 5570569,
+
+        /// <summary>dest = pattern XOR dest, 0x005A0049</summary>
+        PatInvert = 5898313,
+
+        ///<summary>dest = source XOR dest, 0x00660046</summary>
+        SourceInvert = 6684742,
+
+        ///<summary>dest = source AND dest, 0x008800C6</summary>
+        SourceAnd = 8913094,
+
+        /// <summary>dest = (NOT source) OR dest, 0x00BB0226</summary>
+        MergePaint = 12255782,
+
+        ///<summary>dest = (source AND pattern), 0x00C000CA</summary>
+        MergeCopy = 12583114,
+
+        ///<summary>dest = source, 0x00CC0020</summary>
+        SourceCopy = 13369376,
+
+        /// <summary>dest = source OR dest, 0x00EE0086</summary>
+        SourcePaint = 15597702,
+
+        /// <summary>dest = pattern, 0x00F00021</summary>
+        PatCopy = 15728673,
+
+        /// <summary>dest = DPSnoo, 0x00FB0A09</summary>
+        PatPaint = 16452105,
+
+        /// <summary>dest = WHITE, 0x00FF0062</summary>
+        Whiteness = 16711778,
+
+        /// <summary>
+        /// Capture window as seen on screen.  This includes layered windows 
+        /// such as WPF windows with AllowsTransparency="true", 0x40000000
+        /// </summary>
+        CaptureBlt = 1073741824,
+    }
+}
diff --git a/SmartContextMenu/Native/Gdi32.cs b/SmartContextMenu/Native/Gdi32.cs
@@ -0,0 +1,45 @@
+using System;
+using System.Runtime.InteropServices;
+using SmartContextMenu.Native.Enums;
+
+namespace SmartContextMenu.Native
+{
+    static class Gdi32
+    {
+        [DllImport("gdi32.dll", EntryPoint = "CreateCompatibleDC", SetLastError = true)]
+        public static extern IntPtr CreateCompatibleDC([In] IntPtr hdc);
+
+        [DllImport("gdi32.dll", EntryPoint = "SelectObject")]
+        public static extern IntPtr SelectObject([In] IntPtr hdc, [In] IntPtr hgdiobj);
+
+        [DllImport("gdi32.dll", EntryPoint = "BitBlt", SetLastError = true)]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        public static extern bool BitBlt([In] IntPtr hdc, int nXDest, int nYDest, int nWidth, int nHeight, [In] IntPtr hdcSrc, int nXSrc, int nYSrc, CopyPixelOperations dwRop);
+
+        [DllImport("gdi32.dll", EntryPoint = "StretchBlt", SetLastError = true)]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        public static extern bool StretchBlt([In] IntPtr hdc, int nXDest, int nYDest, int nWidthDest, int nHeightDest, [In] IntPtr hdcSrc, int nXSrc, int nYSrc, int nWidthSource, int nHeightSource, CopyPixelOperations dwRop);
+
+        [DllImport("gdi32.dll")]
+        public static extern IntPtr CreateDIBSection(IntPtr hdc, [In] ref IntPtr pbmi, uint pila, out IntPtr ppvBits, IntPtr hSection, uint dwOffset);
+
+        [DllImport("gdi32.dll", EntryPoint = "DeleteDC")]
+        public static extern bool DeleteDC([In] IntPtr hdc);
+
+        [DllImport("gdi32.dll")]
+        public static extern int GetObject(IntPtr hgdiobj, int cbBuffer, IntPtr lpvObject);
+
+        [DllImport("gdi32.dll", EntryPoint = "DeleteObject")]
+        [return: MarshalAs(UnmanagedType.Bool)]
+        public static extern bool DeleteObject([In] IntPtr hObject);
+        
+        [DllImport("gdi32.dll")]
+        public static extern IntPtr CreateCompatibleBitmap(IntPtr hdc, int nWidth, int nHeight);
+        
+        [DllImport("gdi32.dll")]
+        internal static extern int GetDeviceCaps(IntPtr hdc, int capindex);
+        
+        [DllImport("gdi32.dll")]
+        internal static extern bool PatBlt(IntPtr hdc, int nXLeft, int nYLeft, int nWidth, int nHeight, uint dwRop);
+    }
+}
diff --git a/SmartContextMenu/Native/Structs/CURSORINFO.cs b/SmartContextMenu/Native/Structs/CURSORINFO.cs
@@ -0,0 +1,21 @@
+﻿using System;
+using System.Runtime.InteropServices;
+
+namespace SmartContextMenu.Native.Structs
+{
+    [StructLayout(LayoutKind.Sequential)]
+    struct CURSORINFO
+    {
+        public Int32 cbSize;
+        public Int32 flags;
+        public IntPtr hCursor;
+        public POINTAPI ptScreenPos;
+    }
+
+    [StructLayout(LayoutKind.Sequential)]
+    struct POINTAPI
+    {
+        public int x;
+        public int y;
+    }
+}
diff --git a/SmartContextMenu/Native/Structs/ICONINFO.cs b/SmartContextMenu/Native/Structs/ICONINFO.cs
@@ -0,0 +1,15 @@
+﻿using System;
+using System.Runtime.InteropServices;
+
+namespace SmartContextMenu.Native.Structs
+{
+    [StructLayout(LayoutKind.Sequential)]
+    struct ICONINFO
+    {
+        public bool fIcon;
+        public Int32 xHotspot;
+        public Int32 yHotspot;
+        public IntPtr hbmMask;
+        public IntPtr hbmColor;
+    }
+}
diff --git a/SmartContextMenu/Native/User32.cs b/SmartContextMenu/Native/User32.cs
@@ -95,6 +95,27 @@ static class User32
         [DllImport("user32.dll")]
         public static extern IntPtr GetDesktopWindow();
 
+        [DllImport("user32.dll")]
+        public static extern IntPtr GetWindowDC(IntPtr ptr);
+
+        [DllImport("user32.dll")]
+        public static extern IntPtr GetDC(IntPtr hWnd);
+
+        [DllImport("user32.dll")]
+        public static extern bool ReleaseDC(IntPtr hWnd, IntPtr hDc);
+
+        [DllImport("user32.dll")]
+        public static extern bool GetCursorInfo(out CURSORINFO pci);
+
+        [DllImport("user32.dll")]
+        public static extern bool GetIconInfo(IntPtr hIcon, out ICONINFO piconinfo);
+
+        [DllImport("user32.dll")]
+        public static extern bool DrawIconEx(IntPtr hdc, int xLeft, int yTop, IntPtr hIcon, int cxWidth, int cyHeight, int istepIfAniCur, IntPtr hbrFlickerFreeDraw, int diFlags);
+
+        [DllImport("user32.dll")]
+        public static extern bool DestroyIcon(IntPtr hIcon);
+
         [DllImport("user32.dll")]
         [return: MarshalAs(UnmanagedType.Bool)]
         public static extern bool SetWindowPos(IntPtr handle, IntPtr hWndInsertAfter, int x, int y, int cx, int cy, uint uFlags);
diff --git a/SmartContextMenu/Program.cs b/SmartContextMenu/Program.cs
@@ -385,7 +385,12 @@ static ICollection<Window> ProcessCommandLine(CommandLineParser сommandLinePars
                 // Copy Screenshot
                 if (сommandLineParser.HasToggle("copyscreenshot"))
                 {
-                    var bitmap = WindowUtils.PrintWindow(window.Handle);
+                    var result = WindowUtils.PrintWindow(window.Handle, out var bitmap);
+                    if (!result || !WindowUtils.IsCorrectScreenshot(window.Handle, bitmap))
+                    {
+                        WindowUtils.CaptureWindow(window.Handle, false, out bitmap);
+                    }
+
                     Clipboard.SetImage(bitmap);
                 }
 
@@ -399,7 +404,12 @@ static ICollection<Window> ProcessCommandLine(CommandLineParser сommandLinePars
                 //Save Screenshot
                 if (сommandLineParser.HasToggle("s") || сommandLineParser.HasToggle("savescreenshot"))
                 {
-                    var bitmap = WindowUtils.PrintWindow(window.Handle);
+                    var result = WindowUtils.PrintWindow(window.Handle, out var bitmap);
+                    if (!result || !WindowUtils.IsCorrectScreenshot(window.Handle, bitmap))
+                    {
+                        WindowUtils.CaptureWindow(window.Handle, false, out bitmap);
+                    }
+
                     var dialog = new SaveFileDialog
                     {
                         OverwritePrompt = true,
diff --git a/SmartContextMenu/SmartContextMenu.csproj b/SmartContextMenu/SmartContextMenu.csproj
@@ -164,6 +164,7 @@
     <Compile Include="Native\Constants.cs" />
     <Compile Include="Native\Dwmapi.cs" />
     <Compile Include="Native\Enums\AccentState.cs" />
+    <Compile Include="Native\Enums\CopyPixelOperations.cs" />
     <Compile Include="Native\Enums\DpiAwarenessContext.cs" />
     <Compile Include="Native\Enums\DwmBB.cs" />
     <Compile Include="Native\Enums\LayeredWindow.cs" />
@@ -175,13 +176,16 @@
     <Compile Include="Native\Enums\ThreadAccess.cs" />
     <Compile Include="Native\Enums\WindowCompositionAttribute.cs" />
     <Compile Include="Native\Enums\WindowShowStyle.cs" />
+    <Compile Include="Native\Gdi32.cs" />
     <Compile Include="Native\Kernel32.cs" />
     <Compile Include="Native\Ntdll.cs" />
     <Compile Include="Native\SHCore.cs" />
     <Compile Include="Native\Structs\AccentPolicy.cs" />
     <Compile Include="Native\Structs\ConsoleScreenBufferInfo.cs" />
     <Compile Include="Native\Structs\Coord.cs" />
+    <Compile Include="Native\Structs\CURSORINFO.cs" />
     <Compile Include="Native\Structs\DwmBlurBehind.cs" />
+    <Compile Include="Native\Structs\ICONINFO.cs" />
     <Compile Include="Native\Structs\KeyboardLLHookStruct.cs" />
     <Compile Include="Native\Structs\MonitorInfo.cs" />
     <Compile Include="Native\Structs\MouseLLHookStruct.cs" />
diff --git a/SmartContextMenu/Utils/WindowUtils.cs b/SmartContextMenu/Utils/WindowUtils.cs
diff --git a/SmartContextMenu/Window.cs b/SmartContextMenu/Window.cs

Original file line number	Diff line number	Diff line change
`@@ -143,5 +143,10 @@ static class Constants`
`143`	`143`	`public const uint EVENT_SYSTEM_FOREGROUND = 3;`
`144`	`144`	`public const uint EVENT_OBJECT_DESTROY = 0x8001;`
`145`	`145`	`public const int OBJID_WINDOW = 0;`
	`146`	`+`
	`147`	`+ public const int CURSOR_SHOWING = 0x00000001;`
	`148`	`+`
	`149`	`+ public const int DI_COMPAT = 0x0004;`
	`150`	`+ public const int DI_NORMAL = 0x0003;`
`146`	`151`	`}`
`147`	`152`	`}`