1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
using System;
using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using SharpDX;
using SharpDX.Direct3D11;
using SharpDX.DXGI;
using Device = SharpDX.Direct3D11.Device;
using MapFlags = SharpDX.Direct3D11.MapFlags;
namespace Tango.ScreenCapture
{
public class DirectXScreenCapture : IScreenCaptureMethod
{
private static bool _hasInstance;
private Device device;
private Output1 output1;
private Texture2DDescription textureDesc;
private OutputDuplication duplicatedOutput;
private int monitorWidth;
private int monitorHeight;
public int AdapterIndex { get; set; }
public int MonitorIndex { get; set; }
public DirectXScreenCapture()
{
if (_hasInstance)
{
throw new InvalidOperationException("An instance of the DirectX screen capture method already exists. Please dispose it before attempting to create a new one.");
}
// Create DXGI Factory1
var factory = new Factory1();
var adapter = factory.GetAdapter1(AdapterIndex);
// Create device from Adapter
device = new Device(adapter);
// Get DXGI.Output
var output = adapter.GetOutput(MonitorIndex);
output1 = output.QueryInterface<Output1>();
// Width/Height of desktop to capture
monitorWidth = ((SharpDX.Rectangle)output.Description.DesktopBounds).Width;
monitorHeight = ((SharpDX.Rectangle)output.Description.DesktopBounds).Height;
// Create Staging texture CPU-accessible
textureDesc = new Texture2DDescription
{
CpuAccessFlags = CpuAccessFlags.Read,
BindFlags = BindFlags.None,
Format = Format.B8G8R8A8_UNorm,
Width = monitorWidth,
Height = monitorHeight,
OptionFlags = ResourceOptionFlags.None,
MipLevels = 1,
ArraySize = 1,
SampleDescription = { Count = 1, Quality = 0 },
Usage = ResourceUsage.Staging
};
// Duplicate the output
duplicatedOutput = output1.DuplicateOutput(device);
_hasInstance = true;
}
public virtual Bitmap GetDesktopBitmap(CaptureRegion region)
{
var screenTexture = new Texture2D(device, textureDesc);
SharpDX.DXGI.Resource screenResource;
OutputDuplicateFrameInformation duplicateFrameInformation;
// Try to get duplicated frame within given time
duplicatedOutput.AcquireNextFrame(10000, out duplicateFrameInformation, out screenResource);
// copy resource into memory that can be accessed by the CPU
using (var screenTexture2D = screenResource.QueryInterface<Texture2D>())
device.ImmediateContext.CopyResource(screenTexture2D, screenTexture);
// Get the desktop capture texture
var mapSource = device.ImmediateContext.MapSubresource(screenTexture, 0, MapMode.Read, MapFlags.None);
// Create Drawing.Bitmap
var bitmap = new Bitmap(monitorWidth, monitorHeight, System.Drawing.Imaging.PixelFormat.Format32bppRgb);
var boundsRect = new System.Drawing.Rectangle(0, 0, monitorWidth, monitorHeight);
// Copy pixels from screen capture Texture to GDI bitmap
var mapDest = bitmap.LockBits(boundsRect, System.Drawing.Imaging.ImageLockMode.WriteOnly, bitmap.PixelFormat);
var sourcePtr = mapSource.DataPointer;
var destPtr = mapDest.Scan0;
for (int y = 0; y < region.Height; y++)
{
// Copy a single line
Utilities.CopyMemory(destPtr, sourcePtr, monitorWidth * 4);
// Advance pointers
sourcePtr = IntPtr.Add(sourcePtr, mapSource.RowPitch);
destPtr = IntPtr.Add(destPtr, mapDest.Stride);
}
// Release source and dest locks
bitmap.UnlockBits(mapDest);
device.ImmediateContext.UnmapSubresource(screenTexture, 0);
if (region.Left > 0 || region.Top > 0 || region.Width < monitorWidth || region.Height < monitorHeight)
{
var cropped = CropImage(bitmap, new System.Drawing.Rectangle(region.Left, region.Top, region.Width, region.Height));
bitmap.Dispose();
bitmap = cropped;
}
screenTexture.Dispose();
screenResource.Dispose();
duplicatedOutput.ReleaseFrame();
return bitmap;
}
private Bitmap CropImage(Bitmap img, System.Drawing.Rectangle cropArea)
{
Bitmap bmpImage = new Bitmap(img);
return bmpImage.Clone(cropArea, bmpImage.PixelFormat);
}
public void Dispose()
{
_hasInstance = false;
duplicatedOutput.Dispose();
device.Dispose();
}
}
}
|
