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
|
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 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()
{
// 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);
}
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()
{
duplicatedOutput.Dispose();
device.Dispose();
}
}
}
|
