From b7fac7a7d1f07ee36029084cc29b83b5a52f726c Mon Sep 17 00:00:00 2001
From: =?utf8?q?Kristian=20H=C3=B8gsberg?= <kristian.h.kristensen@intel.com>
Date: Sun, 17 May 2015 19:25:28 -0700
Subject: [PATCH] vk: Implement allocation count query

---
 src/vulkan/device.c | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/vulkan/device.c b/src/vulkan/device.c
index c76a8f73c4e..6faa0b04eb4 100644
--- a/src/vulkan/device.c
+++ b/src/vulkan/device.c
@@ -999,6 +999,18 @@ fill_memory_requirements(
    }
 }
 
+static uint32_t
+get_allocation_count(VkObjectType objType)
+{
+   switch (objType) {
+   case VK_OBJECT_TYPE_BUFFER:
+   case VK_OBJECT_TYPE_IMAGE:
+      return 1;
+   default:
+      return 0;
+   }
+}
+
 VkResult anv_GetObjectInfo(
     VkDevice                                    _device,
     VkObjectType                                objType,
@@ -1008,6 +1020,7 @@ VkResult anv_GetObjectInfo(
     void*                                       pData)
 {
    VkMemoryRequirements memory_requirements;
+   uint32_t count;
 
    switch (infoType) {
    case VK_OBJECT_INFO_TYPE_MEMORY_REQUIREMENTS:
@@ -1018,10 +1031,16 @@ VkResult anv_GetObjectInfo(
       fill_memory_requirements(objType, object, &memory_requirements);
       memcpy(pData, &memory_requirements,
              MIN2(*pDataSize, sizeof(memory_requirements)));
-      *pDataSize = sizeof(memory_requirements);
       return VK_SUCCESS;
 
    case VK_OBJECT_INFO_TYPE_MEMORY_ALLOCATION_COUNT:
+      *pDataSize = sizeof(count);
+      if (pData == NULL)
+         return VK_SUCCESS;
+
+      count = get_allocation_count(objType);
+      return VK_SUCCESS;
+
    default:
       return VK_UNSUPPORTED;
    }
-- 
2.30.2