昨天寫自動化測試的CASE的時候,碰到一個疑難雜症,調用截圖的函數去截取一個Popup窗口,但是總是把背景程序給截下來,Popup窗口就跟看不到一樣。本來以為是同步的問題,也就是以為先截圖再點擊彈出Popup窗口了。後來加了N個Thread.Sleep來測試,發現根本不是因為這個原因,而是截圖的函數截不下來這個窗口。
這個為啥呢,只好把截圖的函數代碼翻出來看,以前是用這種方式的:
BitBlt(dcImage, 0, 0, (int)(rect.Width), (int)(rect.Height), dcScreen, (int)(rect.Left), (int)(rect.Top), TernaryRasterOperations.SRCCOPY);
憑直覺感覺應該是因為這種通過DC的方式對WPF程序支持有問題,但是又覺得奇怪就是截取其它的WPF組件和窗口都沒有問題,偏偏Popup窗口不行。
前些天聽說另外一種截屏的方法,這種方法連被遮擋的窗口都可以截,於是就Google一大把,找打了PrintWindow函數,於是就有了第二種解決方案,代碼如下:
IntPtr hdc = Native.GetWindowDC(this.Handle);
if (hdc != IntPtr.Zero)
{
IntPtr hdcMem = Native.CreateCompatibleDC(hdc);
if (hdcMem != IntPtr.Zero)
{
IntPtr hbitmap = Native.CreateCompatibleBitmap(hdc, (int)(Rect.Width), (int)(Rect.Height));
if (hbitmap != IntPtr.Zero)
{
Native.SelectObject(hdcMem, hbitmap);
Native.PrintWindow(this.Handle, hdcMem, 0);
Native.DeleteObject(hbitmap);
Bitmap bmp = Bitmap.FromHbitmap(hbitmap);
bmp.Save(sPath);
}
Native.DeleteObject(hdcMem);
}
Native.ReleaseDC(this.Handle, hdc);
}
就是拿到窗口的句柄,通過PrintWindow API來截取窗口。
但是更讓人氣憤的事情出現了,截出來的窗口中,只要是用到WPF組件的地方,全部是黑塊兒,只有MFC的窗口框架和按鈕可以正常被截取。
於是乎,就無奈的繼續分析這個問題,我記得WPF是沒有走GDI,而是通過Directx渲染的,那就是說DC的方式和PrintWindow的方式都不靠譜,但是截Directx的貌似還比較復雜。
突然想起來,平常報bug的時候都是按PrintScreen,然後再處理一下的,那應該PrintScreen按鍵是管用的,看來只能曲線救國了。但是那樣就得走剪切板了,貌似會破壞剪切板的數據,不過如果我在截取前保存一下數據,在截取後再恢復一下剪切板數據,那就沒有問題了。
於是就有了第三種解決方案(暫時還沒有加恢復剪切板數據的代碼):
const uint KEYEVENTF_EXTENDEDKEY = 0x1;
const uint KEYEVENTF_KEYUP = 0x2;
const byte VK_SNAPSHOT = 0x2C;
Native.keybd_event(VK_SNAPSHOT, 0x45, KEYEVENTF_EXTENDEDKEY, UIntPtr.Zero);
Native.keybd_event(VK_SNAPSHOT, 0x45, KEYEVENTF_EXTENDEDKEY | KEYEVENTF_KEYUP, UIntPtr.Zero);
IDataObject iObj = Clipboard.GetDataObject();
if (iObj.GetDataPresent(DataFormats.Bitmap, true))
{
Bitmap bmpScreen = iObj.GetData(DataFormats.Bitmap, true) as Bitmap;
Bitmap bmpOutput = new Bitmap((int)this.Rect.Width, (int)this.Rect.Height, System.Drawing.Imaging.PixelFormat.Format24bppRgb);
Graphics g = Graphics.FromImage(bmpOutput);
Rectangle destRectangle = new Rectangle(0, 0, (int)this.Rect.Width, (int)this.Rect.Height);
g.DrawImage(bmpScreen,destRectangle, (int)this.Rect.X, (int)this.Rect.Y, (int)this.Rect.Width, (int)this.Rect.Height, GraphicsUnit.Pixel);
bmpOutput.Save(sPath, System.Drawing.Imaging.ImageFormat.Bmp);
}
測試可用,只好先用著了
不過還有幾個問題,先寫下來,留待以後解決:
1. 針對第三種方案,既然可以按PrintScreen鍵截圖,那對應的API是什麼,總覺得發鍵盤消息沒有直接調API穩定
2. 針對WPF截圖有沒有更好的解決方案