Use ComPtr for all D3D11 pointers

Author: 0x5bfa

src/Files.App.CsWin32/Windows.Win32.ComPtr.cs

@@ -27,6 +27,14 @@ public ComPtr(T* ptr)
 				((IUnknown*)ptr)->AddRef();
 		}
 
+		public void Attach(T* other)
+		{
+			if (_ptr is not null)
+				((IUnknown*)_ptr)->Release();
+
+			_ptr = other;
+		}
+
 		[MethodImpl(MethodImplOptions.AggressiveInlining)]
 		public readonly T* Get()
 		{

src/Files.App/Assets/FilesOpenDialog/Files.App.Launcher.exe.sha256

@@ -1 +1 @@
-83f9dfee3c40c31ab2319cc0cf94c0f754aadf483ffe4aa6de16b1413ddb83c8
+2c50ef2c68ccdf705767f11a94caf940e0339e4bd2261961b6ff807fd4c12e44

src/Files.App/ViewModels/UserControls/Previews/ShellPreviewViewModel.cs

@@ -161,61 +161,62 @@ private unsafe bool ChildWindowToXaml(nint parent, UIElement presenter)
 				D3D_DRIVER_TYPE.D3D_DRIVER_TYPE_WARP,
 			];
 
-			ID3D11Device* pD3D11Device = default;
-			ID3D11DeviceContext* pD3D11DeviceContext = default;
-
-			foreach (var driveType in driverTypes)
+			Windows.Win32.Foundation.HRESULT hr = default;
+			using ComPtr<ID3D11Device> pD3D11Device = default;
+			using ComPtr<ID3D11DeviceContext> pD3D11DeviceContext = default;
+			using ComPtr<IDXGIDevice> pDXGIDevice = default;
+			using ComPtr<IDCompositionDevice> pDCompositionDevice = default;
+			using ComPtr<IUnknown> pControlSurface = default;
+			ComPtr<IDCompositionVisual> pChildVisual = default; // Don't dispose this one, it's used by the compositor
+
+			// Create the D3D11 device
+			foreach (var driverType in driverTypes)
 			{
-				var hr = PInvoke.D3D11CreateDevice(
-					null,
-					driveType,
-					new(nint.Zero),
+				hr = PInvoke.D3D11CreateDevice(
+					null, driverType, new(nint.Zero),
 					D3D11_CREATE_DEVICE_FLAG.D3D11_CREATE_DEVICE_BGRA_SUPPORT,
-					null,
-					0,
-					7,
-					&pD3D11Device,
-					null,
-					&pD3D11DeviceContext);
+					null, /* FeatureLevels */ 0, /* SDKVersion */ 7,
+					pD3D11Device.GetAddressOf(), null,
+					pD3D11DeviceContext.GetAddressOf());
 
 				if (hr.Succeeded)
 					break;
 			}
 
-			if (pD3D11Device is null)
+			if (pD3D11Device.IsNull)
 				return false;
 
-			IDXGIDevice* pDXGIDevice = (IDXGIDevice*)pD3D11Device;
-			if (PInvoke.DCompositionCreateDevice(pDXGIDevice, typeof(IDCompositionDevice).GUID, out var compositionDevicePtr).Failed)
-				return false;
+			Guid IID_IDCompositionDevice = typeof(IDCompositionDevice).GUID;
 
-			var pDCompositionDevice = (IDCompositionDevice*)compositionDevicePtr;
-			IDCompositionVisual* pChildVisual = default;
-			IUnknown* pControlSurface = default;
+			// Create the DComp device
+			pDXGIDevice.Attach((IDXGIDevice*)pD3D11Device.Get());
+			hr = PInvoke.DCompositionCreateDevice(
+				pDXGIDevice.Get(),
+				&IID_IDCompositionDevice,
+				(void**)pDCompositionDevice.GetAddressOf());
+			if (hr.Failed)
+				return false;
 
-			pDCompositionDevice->CreateVisual(&pChildVisual);
-			pDCompositionDevice->CreateSurfaceFromHwnd(new(hwnd.DangerousGetHandle()), &pControlSurface);
-			pChildVisual->SetContent(pControlSurface);
-			if (pChildVisual is null || pControlSurface is null)
+			// Create the visual
+			hr = pDCompositionDevice.Get()->CreateVisual(pChildVisual.GetAddressOf());
+			hr = pDCompositionDevice.Get()->CreateSurfaceFromHwnd(new(hwnd.DangerousGetHandle()), pControlSurface.GetAddressOf());
+			hr = pChildVisual.Get()->SetContent(pControlSurface.Get());
+			if (pChildVisual.IsNull || pControlSurface.IsNull)
 				return false;
 
+			// Get the compositor and set the visual on it
 			var compositor = ElementCompositionPreview.GetElementVisual(presenter).Compositor;
 			outputLink = ContentExternalOutputLink.Create(compositor);
 
 			var target = outputLink.As<IDCompositionTarget>();
-			target.SetRoot((nint)pChildVisual);
+			target.SetRoot((nint)pChildVisual.Get());
 
 			outputLink.PlacementVisual.Size = new(0, 0);
 			outputLink.PlacementVisual.Scale = new(1 / (float)presenter.XamlRoot.RasterizationScale);
 			ElementCompositionPreview.SetElementChildVisual(presenter, outputLink.PlacementVisual);
 
-			pDCompositionDevice->Commit();
-
-			pControlSurface->Release();
-			pDCompositionDevice->Release();
-			pDXGIDevice->Release();
-			pD3D11Device->Release();
-			pD3D11DeviceContext->Release();
+			// Commit the all pending DComp commands
+			pDCompositionDevice.Get()->Commit();
 
 			var dwAttrib = Convert.ToUInt32(true);