commit ccac023ddc38e2405d7161296ddb50f8f76237b8
author Ian Elliott <ian@lunarg.com> Fri Aug 07 14:11:14 2015 -0600
committer Ian Elliott <ian@lunarg.com> Tue Aug 11 16:55:28 2015 -0600
tree 2a79fd97f941433a8515fe204e02c615e9ec3be0
parent 7fe115db80cf5ffdfa5d08ac90b1aadedd59ca26

layers: mem_tracker: Change to new WSI query functions.

layers/screenshot.cpp

diff --git a/layers/screenshot.cpp b/layers/screenshot.cpp
index 7f8ce4a..a9313bb 100644
--- a/layers/screenshot.cpp
+++ b/layers/screenshot.cpp
@@ -426,14 +426,13 @@
     return result;
 }
 
-VK_LAYER_EXPORT VkResult VKAPI vkGetSwapChainInfoWSI(
+VK_LAYER_EXPORT VkResult VKAPI vkGetSwapChainImagesWSI(
     VkDevice                device,
     VkSwapChainWSI          swapChain,
-    VkSwapChainInfoTypeWSI  infoType,
-    size_t                 *pDataSize,
-    void                   *pData)
+    uint32_t*               pCount,
+    VkImage*                pSwapChainImages)
 {
-    VkResult result = get_dispatch_table(screenshot_device_table_map, device)->GetSwapChainInfoWSI(device, swapChain, infoType, pDataSize, pData);
+    VkResult result = get_dispatch_table(screenshot_device_table_map, device)->GetSwapChainImagesWSI(device, swapChain, pCount, pSwapChainImages);
 
     loader_platform_thread_lock_mutex(&globalLock);
     if (screenshotEnvQueried && screenshotFrames.empty()) {
@@ -443,20 +442,19 @@
     }
 
     if (result == VK_SUCCESS &&
-        pData &&
+        pSwapChainImages &&
         !swapchainMap.empty() && swapchainMap.find(swapChain.handle) != swapchainMap.end())
-    {   
-        VkSwapChainImagePropertiesWSI *swapChainImageInfo = (VkSwapChainImagePropertiesWSI *)pData;
+    {
         int i;
 
-        for (i=0; i<*pDataSize/sizeof(VkSwapChainImagePropertiesWSI); i++,swapChainImageInfo++)
+        for (i=0; i<*pCount; i++)
         {
             // Create a mapping for an image to a device, image extent, and format
             ImageMapStruct *imageMapElem = new ImageMapStruct;
             imageMapElem->device =  swapchainMap[swapChain.handle]->device;
             imageMapElem->imageExtent = swapchainMap[swapChain.handle]->imageExtent;
             imageMapElem->format = swapchainMap[swapChain.handle]->format;
-            imageMap.insert(make_pair(swapChainImageInfo->image.handle, imageMapElem));
+            imageMap.insert(make_pair(pSwapChainImages[i].handle, imageMapElem));
         }
 
         // Add list of images to swapchain to image map
@@ -465,10 +463,9 @@
         {
             VkImage *imageList = new VkImage[i];
             swapchainMapElem->imageList = imageList;
-            VkSwapChainImagePropertiesWSI *swapChainImageInfo = (VkSwapChainImagePropertiesWSI *)pData;
-            for (int j=0; j<i; j++,swapChainImageInfo++)
+            for (int j=0; j<i; j++)
             {
-                swapchainMapElem->imageList[j] =  swapChainImageInfo->image.handle;
+                swapchainMapElem->imageList[j] = pSwapChainImages[j].handle;
             }
         }
 
@@ -590,8 +587,8 @@
     {
         if (!strcmp(funcName, "vkCreateSwapChainWSI"))
             return (PFN_vkVoidFunction) vkCreateSwapChainWSI;
-        if (!strcmp(funcName, "vkGetSwapChainInfoWSI"))
-            return (PFN_vkVoidFunction) vkGetSwapChainInfoWSI;
+        if (!strcmp(funcName, "vkGetSwapChainImagesWSI"))
+            return (PFN_vkVoidFunction) vkGetSwapChainImagesWSI;
         if (!strcmp(funcName, "vkQueuePresentWSI"))
             return (PFN_vkVoidFunction) vkQueuePresentWSI;
     }