[PATCH 5/5] d3d9: Enumerate wined3d outputs in d3d9_init().

[Zhiyi Zhang]

Signed-off-by: Zhiyi Zhang <zzhang(a)codeweavers.com>
---
 dlls/d3d9/d3d9_private.h |  2 ++
 dlls/d3d9/directx.c      | 50 +++++++++++++++++++++++++++++-----------
 2 files changed, 38 insertions(+), 14 deletions(-)

diff --git a/dlls/d3d9/d3d9_private.h b/dlls/d3d9/d3d9_private.h
index bc9a3b98b2e..bfd5ba1ff86 100644
--- a/dlls/d3d9/d3d9_private.h
+++ b/dlls/d3d9/d3d9_private.h
@@ -66,6 +66,8 @@ struct d3d9
     IDirect3D9Ex IDirect3D9Ex_iface;
     LONG refcount;
     struct wined3d *wined3d;
+    struct wined3d_output **wined3d_outputs;
+    UINT wined3d_output_count;
     BOOL extended;
 };
 
diff --git a/dlls/d3d9/directx.c b/dlls/d3d9/directx.c
index a0b6d142113..2a0a560867a 100644
--- a/dlls/d3d9/directx.c
+++ b/dlls/d3d9/directx.c
@@ -85,6 +85,7 @@ static ULONG WINAPI d3d9_Release(IDirect3D9Ex *iface)
         wined3d_decref(d3d9->wined3d);
         wined3d_mutex_unlock();
 
+        heap_free(d3d9->wined3d_outputs);
         heap_free(d3d9);
     }
 
@@ -108,22 +109,10 @@ static HRESULT WINAPI d3d9_RegisterSoftwareDevice(IDirect3D9Ex *iface, void *ini
 static UINT WINAPI d3d9_GetAdapterCount(IDirect3D9Ex *iface)
 {
     struct d3d9 *d3d9 = impl_from_IDirect3D9Ex(iface);
-    struct wined3d_adapter *wined3d_adapter;
-    UINT adapter_idx, adapter_count;
-    UINT output_count = 0;
 
     TRACE("iface %p.\n", iface);
 
-    wined3d_mutex_lock();
-    adapter_count = wined3d_get_adapter_count(d3d9->wined3d);
-    for (adapter_idx = 0; adapter_idx < adapter_count; ++adapter_idx)
-    {
-        wined3d_enum_adapters(d3d9->wined3d, adapter_idx, &wined3d_adapter);
-        output_count += wined3d_adapter_get_output_count(wined3d_adapter);
-    }
-    wined3d_mutex_unlock();
-
-    return output_count;
+    return d3d9->wined3d_output_count;
 }
 
 static HRESULT WINAPI d3d9_GetAdapterIdentifier(IDirect3D9Ex *iface, UINT adapter,
@@ -601,6 +590,8 @@ BOOL d3d9_init(struct d3d9 *d3d9, BOOL extended)
             | WINED3D_SRGB_READ_WRITE_CONTROL | WINED3D_LEGACY_UNBOUND_RESOURCE_COLOR
             | WINED3D_NO_PRIMITIVE_RESTART | WINED3D_LEGACY_CUBEMAP_FILTERING
             | WINED3D_NORMALIZED_DEPTH_BIAS;
+    UINT adapter_idx, output_idx, adapter_count, output_count = 0;
+    struct wined3d_adapter *wined3d_adapter;
 
     if (!extended)
         flags |= WINED3D_VIDMEM_ACCOUNTING;
@@ -612,9 +603,40 @@ BOOL d3d9_init(struct d3d9 *d3d9, BOOL extended)
 
     wined3d_mutex_lock();
     d3d9->wined3d = wined3d_create(flags);
-    wined3d_mutex_unlock();
     if (!d3d9->wined3d)
+    {
+        wined3d_mutex_unlock();
         return FALSE;
+    }
+
+    adapter_count = wined3d_get_adapter_count(d3d9->wined3d);
+    for (adapter_idx = 0; adapter_idx < adapter_count; ++adapter_idx)
+    {
+        wined3d_enum_adapters(d3d9->wined3d, adapter_idx, &wined3d_adapter);
+        output_count += wined3d_adapter_get_output_count(wined3d_adapter);
+    }
+
+    d3d9->wined3d_outputs = heap_calloc(output_count, sizeof(*d3d9->wined3d_outputs));
+    if (!d3d9->wined3d_outputs)
+    {
+        wined3d_decref(d3d9->wined3d);
+        wined3d_mutex_unlock();
+        return FALSE;
+    }
+
+    d3d9->wined3d_output_count = 0;
+    for (adapter_idx = 0; adapter_idx < adapter_count; ++adapter_idx)
+    {
+        wined3d_enum_adapters(d3d9->wined3d, adapter_idx, &wined3d_adapter);
+        output_count = wined3d_adapter_get_output_count(wined3d_adapter);
+        for (output_idx = 0; output_idx < output_count; ++output_idx)
+        {
+            wined3d_adapter_enum_outputs(wined3d_adapter, output_idx,
+                    &d3d9->wined3d_outputs[d3d9->wined3d_output_count++]);
+        }
+    }
+
+    wined3d_mutex_unlock();
     d3d9->extended = extended;
 
     return TRUE;
-- 
2.20.1