From 50da081be987f0bd13b28b5f61e21713b9396230 Mon Sep 17 00:00:00 2001
From: Ellpeck <me@ellpeck.de>
Date: Tue, 11 Jul 2023 12:11:12 +0200
Subject: [PATCH] Added WithRenderTargets, a multi-target version of
 WithRenderTarget

---
 CHANGELOG.md                          |  4 ++++
 MLEM/Extensions/GraphicsExtensions.cs | 29 ++++++++++++++++++++++++---
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index a497667..90c3afa 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -13,6 +13,10 @@ Jump to version:
 
 ## 6.3.0 (In Development)
 
+### MLEM
+Additions
+- Added GraphicsExtensions.WithRenderTargets, a multi-target version of WithRenderTarget
+
 ## 6.2.0
 
 ### MLEM
diff --git a/MLEM/Extensions/GraphicsExtensions.cs b/MLEM/Extensions/GraphicsExtensions.cs
index 4c1b230..57dd9b8 100644
--- a/MLEM/Extensions/GraphicsExtensions.cs
+++ b/MLEM/Extensions/GraphicsExtensions.cs
@@ -69,11 +69,22 @@ namespace MLEM.Extensions {
         /// </summary>
         /// <param name="device">The graphics device</param>
         /// <param name="target">The render target to apply</param>
-        /// <returns></returns>
+        /// <returns>The render target context, to be used in a <c>using</c> statement</returns>
         public static TargetContext WithRenderTarget(this GraphicsDevice device, RenderTarget2D target) {
             return new TargetContext(device, target);
         }
 
+        /// <summary>
+        /// Starts a new <see cref="TargetContext"/> using the specified render target bindings.
+        /// The returned context automatically disposes when used in a <c>using</c> statement, which causes any previously applied render targets to be reapplied automatically.
+        /// </summary>
+        /// <param name="device">The graphics device</param>
+        /// <param name="targets">The render targets to apply</param>
+        /// <returns>The render target context, to be used in a <c>using</c> statement</returns>
+        public static TargetContext WithRenderTargets(this GraphicsDevice device, params RenderTargetBinding[] targets) {
+            return new TargetContext(device, targets);
+        }
+
         /// <summary>
         /// Represents a context in which a <see cref="RenderTarget2D"/> is applied.
         /// This class should be used with <see cref="GraphicsExtensions.WithRenderTarget"/>.
@@ -88,7 +99,20 @@ namespace MLEM.Extensions {
             /// </summary>
             /// <param name="device">The graphics device to apply the target on</param>
             /// <param name="target">The target to apply</param>
-            public TargetContext(GraphicsDevice device, RenderTarget2D target) {
+            public TargetContext(GraphicsDevice device, RenderTarget2D target) : this(device) {
+                device.SetRenderTarget(target);
+            }
+
+            /// <summary>
+            /// Creates a new target context with the given settings.
+            /// </summary>
+            /// <param name="device">The graphics device to apply the target on</param>
+            /// <param name="targets">The targets to apply</param>
+            public TargetContext(GraphicsDevice device, RenderTargetBinding[] targets) : this(device) {
+                device.SetRenderTargets(targets);
+            }
+
+            private TargetContext(GraphicsDevice device) {
                 this.device = device;
 #if FNA
                 // RenderTargetCount doesn't exist in FNA but we still want the optimization in MG
@@ -96,7 +120,6 @@ namespace MLEM.Extensions {
 #else
                 this.lastTargets = device.RenderTargetCount <= 0 ? null : device.GetRenderTargets();
 #endif
-                device.SetRenderTarget(target);
             }
 
             /// <summary>