src/intel/vulkan/anv_query.c

   1 /*
   2  * Copyright © 2015 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  */
  23
  24 #include <assert.h>
  25 #include <stdbool.h>
  26 #include <string.h>
  27 #include <unistd.h>
  28 #include <fcntl.h>
  29
  30 #include "anv_private.h"
  31
  32 VkResult anv_CreateQueryPool(
  33     VkDevice                                    _device,
  34     const VkQueryPoolCreateInfo*                pCreateInfo,
  35     const VkAllocationCallbacks*                pAllocator,
  36     VkQueryPool*                                pQueryPool)
  37 {
  38    ANV_FROM_HANDLE(anv_device, device, _device);
  39    struct anv_query_pool *pool;
  40    VkResult result;
  41    uint32_t slot_size;
  42    uint64_t size;
  43
  44    assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_QUERY_POOL_CREATE_INFO);
  45
  46    switch (pCreateInfo->queryType) {
  47    case VK_QUERY_TYPE_OCCLUSION:
  48    case VK_QUERY_TYPE_TIMESTAMP:
  49       break;
  50    case VK_QUERY_TYPE_PIPELINE_STATISTICS:
  51       return VK_ERROR_INCOMPATIBLE_DRIVER;
  52    default:
  53       assert(!"Invalid query type");
  54    }
  55
  56    slot_size = sizeof(struct anv_query_pool_slot);
  57    pool = anv_alloc2(&device->alloc, pAllocator, sizeof(*pool), 8,
  58                      VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
  59    if (pool == NULL)
  60       return vk_error(VK_ERROR_OUT_OF_HOST_MEMORY);
  61
  62    pool->type = pCreateInfo->queryType;
  63    pool->slots = pCreateInfo->queryCount;
  64
  65    size = pCreateInfo->queryCount * slot_size;
  66    result = anv_bo_init_new(&pool->bo, device, size);
  67    if (result != VK_SUCCESS)
  68       goto fail;
  69
  70    pool->bo.map = anv_gem_mmap(device, pool->bo.gem_handle, 0, size, 0);
  71
  72    *pQueryPool = anv_query_pool_to_handle(pool);
  73
  74    return VK_SUCCESS;
  75
  76  fail:
  77    anv_free2(&device->alloc, pAllocator, pool);
  78
  79    return result;
  80 }
  81
  82 void anv_DestroyQueryPool(
  83     VkDevice                                    _device,
  84     VkQueryPool                                 _pool,
  85     const VkAllocationCallbacks*                pAllocator)
  86 {
  87    ANV_FROM_HANDLE(anv_device, device, _device);
  88    ANV_FROM_HANDLE(anv_query_pool, pool, _pool);
  89
  90    anv_gem_munmap(pool->bo.map, pool->bo.size);
  91    anv_gem_close(device, pool->bo.gem_handle);
  92    anv_free2(&device->alloc, pAllocator, pool);
  93 }
  94
  95 VkResult anv_GetQueryPoolResults(
  96     VkDevice                                    _device,
  97     VkQueryPool                                 queryPool,
  98     uint32_t                                    firstQuery,
  99     uint32_t                                    queryCount,
 100     size_t                                      dataSize,
 101     void*                                       pData,
 102     VkDeviceSize                                stride,
 103     VkQueryResultFlags                          flags)
 104 {
 105    ANV_FROM_HANDLE(anv_device, device, _device);
 106    ANV_FROM_HANDLE(anv_query_pool, pool, queryPool);
 107    int64_t timeout = INT64_MAX;
 108    uint64_t result;
 109    int ret;
 110
 111    assert(pool->type == VK_QUERY_TYPE_OCCLUSION ||
 112           pool->type == VK_QUERY_TYPE_TIMESTAMP);
 113
 114    if (pData == NULL)
 115       return VK_SUCCESS;
 116
 117    if (flags & VK_QUERY_RESULT_WAIT_BIT) {
 118       ret = anv_gem_wait(device, pool->bo.gem_handle, &timeout);
 119       if (ret == -1) {
 120          /* We don't know the real error. */
 121          return vk_errorf(VK_ERROR_OUT_OF_DEVICE_MEMORY,
 122                           "gem_wait failed %m");
 123       }
 124    }
 125
 126    void *data_end = pData + dataSize;
 127    struct anv_query_pool_slot *slot = pool->bo.map;
 128
 129    for (uint32_t i = 0; i < queryCount; i++) {
 130       switch (pool->type) {
 131       case VK_QUERY_TYPE_OCCLUSION: {
 132          result = slot[firstQuery + i].end - slot[firstQuery + i].begin;
 133          break;
 134       }
 135       case VK_QUERY_TYPE_PIPELINE_STATISTICS:
 136          unreachable("pipeline stats not supported");
 137       case VK_QUERY_TYPE_TIMESTAMP: {
 138          result = slot[firstQuery + i].begin;
 139          break;
 140       }
 141       default:
 142          unreachable("invalid pool type");
 143       }
 144
 145       if (flags & VK_QUERY_RESULT_64_BIT) {
 146          uint64_t *dst = pData;
 147          dst[0] = result;
 148          if (flags & VK_QUERY_RESULT_WITH_AVAILABILITY_BIT)
 149             dst[1] = slot[firstQuery + i].available;
 150       } else {
 151          uint32_t *dst = pData;
 152          if (result > UINT32_MAX)
 153             result = UINT32_MAX;
 154          dst[0] = result;
 155          if (flags & VK_QUERY_RESULT_WITH_AVAILABILITY_BIT)
 156             dst[1] = slot[firstQuery + i].available;
 157       }
 158
 159       pData += stride;
 160       if (pData >= data_end)
 161          break;
 162    }
 163
 164    return VK_SUCCESS;
 165 }
 166
 167 void anv_CmdResetQueryPool(
 168     VkCommandBuffer                             commandBuffer,
 169     VkQueryPool                                 queryPool,
 170     uint32_t                                    firstQuery,
 171     uint32_t                                    queryCount)
 172 {
 173    ANV_FROM_HANDLE(anv_query_pool, pool, queryPool);
 174
 175    for (uint32_t i = 0; i < queryCount; i++) {
 176       switch (pool->type) {
 177       case VK_QUERY_TYPE_OCCLUSION:
 178       case VK_QUERY_TYPE_TIMESTAMP: {
 179          struct anv_query_pool_slot *slot = pool->bo.map;
 180          slot[firstQuery + i].available = 0;
 181          break;
 182       }
 183       default:
 184          assert(!"Invalid query type");
 185       }
 186    }
 187 }