-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
12 changed files
with
754 additions
and
69 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,113 @@ | ||
| using System.Buffers; | ||
| using Core.Window; | ||
| using PluginCore; | ||
| using Silk.NET.Direct3D11; | ||
| using Silk.NET.DXGI; | ||
|
|
||
| namespace KitopiaBenchmark; | ||
|
|
||
| public static class CaptureTool | ||
| { | ||
| public static unsafe byte[] GetBytesSpan(MappedSubresource mappedSubresource, OutputDesc1 outputDesc,ref ScreenCaptureInfo screenCaptureInfo) | ||
| { | ||
| var sizeX = outputDesc.DesktopCoordinates.Size.X; | ||
| int startX = Math.Clamp(screenCaptureInfo.X, 0, outputDesc.DesktopCoordinates.Size.X - 1); | ||
| int startY = Math.Clamp(screenCaptureInfo.Y, 0, outputDesc.DesktopCoordinates.Size.Y - 1); | ||
| int endX = Math.Clamp(screenCaptureInfo.X+screenCaptureInfo.Width, 0, outputDesc.DesktopCoordinates.Size.X); | ||
| int endY = Math.Clamp(screenCaptureInfo.Y+screenCaptureInfo.Height, 0, outputDesc.DesktopCoordinates.Size.Y); | ||
| if (screenCaptureInfo.ScreenCaptureType==ScreenCaptureType.窗口) | ||
| { | ||
| sizeX=(screenCaptureInfo.WindowInfo.Rect.Width + 3) & ~3; | ||
| startX = 0; | ||
| startY = 0; | ||
| endX = (screenCaptureInfo.WindowInfo.Rect.Width + 3) & ~3; | ||
| endY = (int)screenCaptureInfo.WindowInfo.Rect.Height; | ||
| } | ||
|
|
||
| screenCaptureInfo.Height = endY - startY; | ||
| screenCaptureInfo.Width = endX - startX; | ||
| screenCaptureInfo.X = startX; | ||
| screenCaptureInfo.Y = startY; | ||
| // 结果数组：区域宽 * 区域高 * 4（RGBA） | ||
| int regionWidth = endX - startX; | ||
| int regionHeight = endY - startY; | ||
| byte[] result = new byte[regionWidth * regionHeight * 4]; | ||
| if (!outputDesc.ColorSpace.ToString().EndsWith("2020")) | ||
| { | ||
| var span = new ReadOnlySpan<uint>(mappedSubresource.PData, | ||
| (int)mappedSubresource.DepthPitch / 4); | ||
|
|
||
| for (int y = startY; y < endY; y++) | ||
| { | ||
| for (int x = startX; x < endX; x++) | ||
| { | ||
| int sourceIndex = (y * sizeX + x) * 4; | ||
| int targetIndex = ((y - startY) * regionWidth + (x - startX)) * 4; | ||
|
|
||
| // 读取原始像素并复制到结果 | ||
| uint value = span[sourceIndex / 4]; | ||
| result[targetIndex+ 2] = (byte)(value & 0xFF); // R | ||
| result[targetIndex + 1] = (byte)((value >> 8) & 0xFF); // G | ||
| result[targetIndex ] = (byte)((value >> 16) & 0xFF); // B | ||
| result[targetIndex + 3] = (byte)((value >> 24) & 0xFF); // A | ||
| } | ||
| } | ||
| } | ||
| else | ||
| { | ||
| var matrix = ColorSpaceCtr.CtrColorSpace([ | ||
| outputDesc.RedPrimary[0], | ||
| outputDesc.RedPrimary[1], | ||
| outputDesc.GreenPrimary[0], | ||
| outputDesc.GreenPrimary[1], | ||
| outputDesc.BluePrimary[0], | ||
| outputDesc.BluePrimary[1], | ||
| outputDesc.WhitePoint[0], | ||
| outputDesc.WhitePoint[1] | ||
| ], | ||
| [ | ||
| .640f, .330f, .300f, .600f, .150f, .060f, .3127f, | ||
| .3290f | ||
| ] | ||
| ); | ||
| var span = new ReadOnlySpan<Half>(mappedSubresource.PData, | ||
| (int)mappedSubresource.DepthPitch / 2).ToArray(); | ||
| // ReadOnlyMemory<Half> readOnlyMemory = new ReadOnlyMemory<Half>(span); | ||
| Parallel.For(startY, endY, y => | ||
| //for (int y = startY; y < endY; y++) | ||
| { | ||
|
|
||
| int yOffset = y * sizeX; | ||
| int targetYOffset = (y - startY) * regionWidth; | ||
|
|
||
| for (int x = startX; x < endX; x++) | ||
| { | ||
| int sourceIndex = (yOffset + x) * 4; | ||
| int targetIndex = (targetYOffset + (x - startX)) * 4; | ||
|
|
||
| // 读取并归一化 RGBA 值 | ||
|
|
||
| float r = float.Log(1 + (float)span[sourceIndex]) / 1.749199854809259f; | ||
| float g = float.Log(1 + (float)span[sourceIndex+1]) / 1.749199854809259f; | ||
| float b = float.Log(1 + (float)span[sourceIndex+2]) / 1.749199854809259f; | ||
|
|
||
| // 应用色彩转换矩阵 | ||
| float bt2020R = matrix[0, 0] * r + matrix[0, 1] * g + matrix[0, 2] * b; | ||
| float bt2020G = matrix[1, 0] * r + matrix[1, 1] * g + matrix[1, 2] * b; | ||
| float bt2020B = matrix[2, 0] * r + matrix[2, 1] * g + matrix[2, 2] * b; | ||
|
|
||
| // 转换并填充结果 | ||
|
|
||
| result[targetIndex ] = (byte)Math.Clamp(bt2020B * 255,0,255); | ||
| result[targetIndex + 1] = (byte)Math.Clamp(bt2020G * 255,0,255); | ||
| result[targetIndex+ 2] = (byte)Math.Clamp(bt2020R * 255,0,255); | ||
| result[targetIndex + 3] = 255; // Alpha 固定为 255 | ||
| } | ||
| } | ||
| ); | ||
|
|
||
| } | ||
|
|
||
| return result; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,56 @@ | ||
| using System.Buffers; | ||
| using BenchmarkDotNet.Attributes; | ||
| using Core.Window; | ||
| using PluginCore; | ||
|
|
||
| namespace KitopiaBenchmark; | ||
| [MemoryDiagnoser] | ||
| public class ScreenCapture | ||
| { | ||
| private ScreenCaptureByWGC _screenCaptureByWgc; | ||
| private ScreenCaptureByWGCWithoutPool _screenCaptureByWgcWithoutPool; | ||
| [GlobalSetup] | ||
| public void Setup() | ||
| { | ||
| _screenCaptureByWgc = new ScreenCaptureByWGC(); | ||
| _screenCaptureByWgcWithoutPool = new ScreenCaptureByWGCWithoutPool(); | ||
| } | ||
| [Benchmark] | ||
| public void WithoutPoolButDontSetNull() | ||
| { | ||
| var captureAllScreenBytes = _screenCaptureByWgcWithoutPool.CaptureAllScreenBytes(); | ||
| while (captureAllScreenBytes.TryPop(out var result)) | ||
| { | ||
| //result.Bytes = null; | ||
| // ArrayPool<byte>.Shared.Return(result.Bytes); | ||
| } | ||
| } | ||
| [Benchmark] | ||
| public void WithoutPool() | ||
| { | ||
| var captureAllScreenBytes = _screenCaptureByWgcWithoutPool.CaptureAllScreenBytes(); | ||
| while (captureAllScreenBytes.TryPop(out var result)) | ||
| { | ||
| result.Bytes = null; | ||
| // ArrayPool<byte>.Shared.Return(result.Bytes); | ||
| } | ||
| } | ||
| [Benchmark] | ||
| public void PoolButDontReturn() | ||
| { | ||
| var captureAllScreenBytes = _screenCaptureByWgc.CaptureAllScreenBytes(); | ||
| while (captureAllScreenBytes.TryPop(out var result)) | ||
| { | ||
| // ArrayPool<byte>.Shared.Return(result.Bytes); | ||
| } | ||
| } | ||
| [Benchmark] | ||
| public void Pool() | ||
| { | ||
| var captureAllScreenBytes = _screenCaptureByWgc.CaptureAllScreenBytes(); | ||
| while (captureAllScreenBytes.TryPop(out var result)) | ||
| { | ||
| ArrayPool<byte>.Shared.Return(result.Bytes); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.