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 = vk_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    vk_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    if (!pool)
  91       return;
  92
  93    anv_gem_munmap(pool->bo.map, pool->bo.size);
  94    anv_gem_close(device, pool->bo.gem_handle);
  95    vk_free2(&device->alloc, pAllocator, pool);
  96 }
  97
  98 VkResult anv_GetQueryPoolResults(
  99     VkDevice                                    _device,
 100     VkQueryPool                                 queryPool,
 101     uint32_t                                    firstQuery,
 102     uint32_t                                    queryCount,
 103     size_t                                      dataSize,
 104     void*                                       pData,
 105     VkDeviceSize                                stride,
 106     VkQueryResultFlags                          flags)
 107 {
 108    ANV_FROM_HANDLE(anv_device, device, _device);
 109    ANV_FROM_HANDLE(anv_query_pool, pool, queryPool);
 110    int64_t timeout = INT64_MAX;
 111    uint64_t result;
 112    int ret;
 113
 114    assert(pool->type == VK_QUERY_TYPE_OCCLUSION ||
 115           pool->type == VK_QUERY_TYPE_TIMESTAMP);
 116
 117    if (pData == NULL)
 118       return VK_SUCCESS;
 119
 120    if (flags & VK_QUERY_RESULT_WAIT_BIT) {
 121       ret = anv_gem_wait(device, pool->bo.gem_handle, &timeout);
 122       if (ret == -1) {
 123          /* We don't know the real error. */
 124          return vk_errorf(VK_ERROR_OUT_OF_DEVICE_MEMORY,
 125                           "gem_wait failed %m");
 126       }
 127    }
 128
 129    void *data_end = pData + dataSize;
 130    struct anv_query_pool_slot *slot = pool->bo.map;
 131
 132    for (uint32_t i = 0; i < queryCount; i++) {
 133       switch (pool->type) {
 134       case VK_QUERY_TYPE_OCCLUSION: {
 135          result = slot[firstQuery + i].end - slot[firstQuery + i].begin;
 136          break;
 137       }
 138       case VK_QUERY_TYPE_PIPELINE_STATISTICS:
 139          unreachable("pipeline stats not supported");
 140       case VK_QUERY_TYPE_TIMESTAMP: {
 141          result = slot[firstQuery + i].begin;
 142          break;
 143       }
 144       default:
 145          unreachable("invalid pool type");
 146       }
 147
 148       if (flags & VK_QUERY_RESULT_64_BIT) {
 149          uint64_t *dst = pData;
 150          dst[0] = result;
 151          if (flags & VK_QUERY_RESULT_WITH_AVAILABILITY_BIT)
 152             dst[1] = slot[firstQuery + i].available;
 153       } else {
 154          uint32_t *dst = pData;
 155          if (result > UINT32_MAX)
 156             result = UINT32_MAX;
 157          dst[0] = result;
 158          if (flags & VK_QUERY_RESULT_WITH_AVAILABILITY_BIT)
 159             dst[1] = slot[firstQuery + i].available;
 160       }
 161
 162       pData += stride;
 163       if (pData >= data_end)
 164          break;
 165    }
 166
 167    return VK_SUCCESS;
 168 }
 169
 170 void anv_CmdResetQueryPool(
 171     VkCommandBuffer                             commandBuffer,
 172     VkQueryPool                                 queryPool,
 173     uint32_t                                    firstQuery,
 174     uint32_t                                    queryCount)
 175 {
 176    ANV_FROM_HANDLE(anv_query_pool, pool, queryPool);
 177
 178    for (uint32_t i = 0; i < queryCount; i++) {
 179       switch (pool->type) {
 180       case VK_QUERY_TYPE_OCCLUSION:
 181       case VK_QUERY_TYPE_TIMESTAMP: {
 182          struct anv_query_pool_slot *slot = pool->bo.map;
 183          slot[firstQuery + i].available = 0;
 184          break;
 185       }
 186       default:
 187          assert(!"Invalid query type");
 188       }
 189    }
 190 }