MoonWorksGraphicsTests/CopyTexture/CopyTextureGame.cs

using System.Runtime.InteropServices;
using MoonWorks;
using MoonWorks.Graphics;
using MoonWorks.Math.Float;

namespace MoonWorks.Test
{
	class CopyTextureGame : Game
	{
		private GraphicsPipeline pipeline;
		private Buffer vertexBuffer;
		private Buffer indexBuffer;
		private Texture originalTexture;
		private Texture textureCopy;
		private Texture textureSmallCopy;
		private Sampler sampler;

		public unsafe CopyTextureGame() : base(TestUtils.GetStandardWindowCreateInfo(), TestUtils.GetStandardFrameLimiterSettings(), 60, true)
		{
			// Load the shaders
			ShaderModule vertShaderModule = new ShaderModule(GraphicsDevice, TestUtils.GetShaderPath("TexturedQuad.vert"));
			ShaderModule fragShaderModule = new ShaderModule(GraphicsDevice, TestUtils.GetShaderPath("TexturedQuad.frag"));

			// Create the graphics pipeline
			GraphicsPipelineCreateInfo pipelineCreateInfo = TestUtils.GetStandardGraphicsPipelineCreateInfo(
				MainWindow.SwapchainFormat,
				vertShaderModule,
				fragShaderModule
			);
			pipelineCreateInfo.AttachmentInfo.ColorAttachmentDescriptions[0].BlendState = ColorAttachmentBlendState.AlphaBlend;
			pipelineCreateInfo.VertexInputState = VertexInputState.CreateSingleBinding<PositionTextureVertex>();
			pipelineCreateInfo.FragmentShaderInfo.SamplerBindingCount = 1;
			pipeline = new GraphicsPipeline(GraphicsDevice, pipelineCreateInfo);

			// Create sampler
			sampler = new Sampler(GraphicsDevice, SamplerCreateInfo.PointClamp);

			// Create and populate the GPU resources
			vertexBuffer = Buffer.Create<PositionTextureVertex>(GraphicsDevice, BufferUsageFlags.Vertex, 12);
			indexBuffer = Buffer.Create<ushort>(GraphicsDevice, BufferUsageFlags.Index, 12);

			CommandBuffer cmdbuf = GraphicsDevice.AcquireCommandBuffer();

			cmdbuf.SetBufferData(
				vertexBuffer,
				new PositionTextureVertex[]
				{
					new PositionTextureVertex(new Vector3(-1f, 0f, 0), new Vector2(0, 0)),
					new PositionTextureVertex(new Vector3( 0f, 0f, 0), new Vector2(1, 0)),
					new PositionTextureVertex(new Vector3( 0f, 1f, 0), new Vector2(1, 1)),
					new PositionTextureVertex(new Vector3(-1f, 1f, 0), new Vector2(0, 1)),

					new PositionTextureVertex(new Vector3(0f, 0f, 0), new Vector2(0, 0)),
					new PositionTextureVertex(new Vector3(1f, 0f, 0), new Vector2(1, 0)),
					new PositionTextureVertex(new Vector3(1f, 1f, 0), new Vector2(1, 1)),
					new PositionTextureVertex(new Vector3(0f, 1f, 0), new Vector2(0, 1)),

					new PositionTextureVertex(new Vector3(-0.5f, -1f, 0), new Vector2(0, 0)),
					new PositionTextureVertex(new Vector3( 0.5f, -1f, 0), new Vector2(1, 0)),
					new PositionTextureVertex(new Vector3( 0.5f,  0f, 0), new Vector2(1, 1)),
					new PositionTextureVertex(new Vector3(-0.5f,  0f, 0), new Vector2(0, 1))
				}
			);
			cmdbuf.SetBufferData(
				indexBuffer,
				new ushort[]
				{
					0, 1, 2,
					0, 2, 3,
				}
			);

			// Load the texture. Storing the texture bytes so we can compare them.
			var fileStream = new System.IO.FileStream(TestUtils.GetTexturePath("ravioli.png"), System.IO.FileMode.Open, System.IO.FileAccess.Read);
			var fileLength = fileStream.Length;
			var fileBuffer = NativeMemory.Alloc((nuint) fileLength);
			var fileSpan = new System.Span<byte>(fileBuffer, (int) fileLength);
			fileStream.ReadExactly(fileSpan);

			var pixels = RefreshCS.Refresh.Refresh_Image_Load(
				(nint) fileBuffer,
				(int) fileLength,
				out var width,
				out var height,
				out var byteCount
			);

			NativeMemory.Free(fileBuffer);

			TextureCreateInfo textureCreateInfo = new TextureCreateInfo();
			textureCreateInfo.Width = (uint) width;
			textureCreateInfo.Height = (uint) height;
			textureCreateInfo.Depth = 1;
			textureCreateInfo.Format = TextureFormat.R8G8B8A8;
			textureCreateInfo.IsCube = false;
			textureCreateInfo.LevelCount = 1;
			textureCreateInfo.UsageFlags = TextureUsageFlags.Sampler;

			originalTexture = new Texture(GraphicsDevice, textureCreateInfo);
			cmdbuf.SetTextureData(originalTexture, pixels, (uint) byteCount);

			// Create a 1:1 copy of the texture
			textureCopy = new Texture(GraphicsDevice, textureCreateInfo);
			cmdbuf.CopyTextureToTexture(
				new TextureSlice(originalTexture),
				new TextureSlice(textureCopy),
				Filter.Linear
			);

			// Create a half-sized copy of this texture
			textureCreateInfo.Width /= 2;
			textureCreateInfo.Height /= 2;
			textureSmallCopy = new Texture(GraphicsDevice, textureCreateInfo);
			cmdbuf.CopyTextureToTexture(
				new TextureSlice(originalTexture),
				new TextureSlice(
					textureSmallCopy,
					new Rect(
						(int) textureCreateInfo.Width,
						(int) textureCreateInfo.Height
					)
				),
				Filter.Linear
			);

			// Copy the texture to a buffer
			Buffer compareBuffer = Buffer.Create<byte>(GraphicsDevice, 0, (uint) byteCount);
			cmdbuf.CopyTextureToBuffer(new TextureSlice(originalTexture), compareBuffer);

			var fence = GraphicsDevice.SubmitAndAcquireFence(cmdbuf);
			GraphicsDevice.WaitForFences(fence);
			GraphicsDevice.ReleaseFence(fence);

			// Compare the original bytes to the copied bytes.
			var copiedBytes = NativeMemory.Alloc((nuint) byteCount);
			var copiedSpan = new System.Span<byte>(copiedBytes, byteCount);
			compareBuffer.GetData(copiedSpan);

			var originalSpan = new System.Span<byte>((void*) pixels, byteCount);

			if (System.MemoryExtensions.SequenceEqual(originalSpan, copiedSpan))
			{
				Logger.LogError("SUCCESS! Original texture bytes and the bytes from CopyTextureToBuffer match!");

			}
			else
			{
				Logger.LogError("FAIL! Original texture bytes do not match bytes from CopyTextureToBuffer!");
			}

			RefreshCS.Refresh.Refresh_Image_Free(pixels);
		}

		protected override void Update(System.TimeSpan delta) { }

		protected override void Draw(double alpha)
		{
			CommandBuffer cmdbuf = GraphicsDevice.AcquireCommandBuffer();
			Texture? backbuffer = cmdbuf.AcquireSwapchainTexture(MainWindow);
			if (backbuffer != null)
			{
				cmdbuf.BeginRenderPass(new ColorAttachmentInfo(backbuffer, Color.Black));
				cmdbuf.BindGraphicsPipeline(pipeline);
				cmdbuf.BindVertexBuffers(vertexBuffer);
				cmdbuf.BindIndexBuffer(indexBuffer, IndexElementSize.Sixteen);
				cmdbuf.BindFragmentSamplers(new TextureSamplerBinding(originalTexture, sampler));
				cmdbuf.DrawIndexedPrimitives(0, 0, 2, 0, 0);
				cmdbuf.BindFragmentSamplers(new TextureSamplerBinding(textureCopy, sampler));
				cmdbuf.DrawIndexedPrimitives(4, 0, 2, 0, 0);
				cmdbuf.BindFragmentSamplers(new TextureSamplerBinding(textureSmallCopy, sampler));
				cmdbuf.DrawIndexedPrimitives(8, 0, 2, 0, 0);
				cmdbuf.EndRenderPass();
			}
			GraphicsDevice.Submit(cmdbuf);
		}

		public static void Main(string[] args)
		{
			CopyTextureGame game = new CopyTextureGame();
			game.Run();
		}
	}
}