src/gallium/drivers/zink/zink_batch.c

   1 #include "zink_batch.h"
   2
   3 #include "zink_context.h"
   4 #include "zink_fence.h"
   5 #include "zink_framebuffer.h"
   6 #include "zink_render_pass.h"
   7 #include "zink_resource.h"
   8 #include "zink_screen.h"
   9
  10 #include "util/u_debug.h"
  11 #include "util/set.h"
  12
  13 static void
  14 reset_batch(struct zink_screen *screen, struct zink_batch *batch)
  15 {
  16    batch->descs_left = ZINK_BATCH_DESC_SIZE;
  17
  18    // cmdbuf hasn't been submitted before
  19    if (!batch->fence)
  20       return;
  21
  22    zink_fence_finish(screen, batch->fence, PIPE_TIMEOUT_INFINITE);
  23    zink_fence_reference(screen, &batch->fence, NULL);
  24
  25    zink_render_pass_reference(screen, &batch->rp, NULL);
  26    zink_framebuffer_reference(screen, &batch->fb, NULL);
  27
  28    /* unref all used resources */
  29    set_foreach(batch->resources, entry) {
  30       struct pipe_resource *pres = (struct pipe_resource *)entry->key;
  31       pipe_resource_reference(&pres, NULL);
  32    }
  33    _mesa_set_clear(batch->resources, NULL);
  34
  35    /* unref all used sampler-views */
  36    set_foreach(batch->sampler_views, entry) {
  37       struct pipe_sampler_view *pres = (struct pipe_sampler_view *)entry->key;
  38       pipe_sampler_view_reference(&pres, NULL);
  39    }
  40    _mesa_set_clear(batch->sampler_views, NULL);
  41
  42    util_dynarray_foreach(&batch->zombie_samplers, VkSampler, samp) {
  43       vkDestroySampler(screen->dev, *samp, NULL);
  44    }
  45    util_dynarray_clear(&batch->zombie_samplers);
  46
  47    if (vkResetDescriptorPool(screen->dev, batch->descpool, 0) != VK_SUCCESS)
  48       fprintf(stderr, "vkResetDescriptorPool failed\n");
  49 }
  50
  51 void
  52 zink_start_batch(struct zink_context *ctx, struct zink_batch *batch)
  53 {
  54    reset_batch(zink_screen(ctx->base.screen), batch);
  55
  56    VkCommandBufferBeginInfo cbbi = {};
  57    cbbi.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO;
  58    cbbi.flags = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT;
  59    if (vkBeginCommandBuffer(batch->cmdbuf, &cbbi) != VK_SUCCESS)
  60       debug_printf("vkBeginCommandBuffer failed\n");
  61 }
  62
  63 void
  64 zink_end_batch(struct zink_context *ctx, struct zink_batch *batch)
  65 {
  66    if (vkEndCommandBuffer(batch->cmdbuf) != VK_SUCCESS) {
  67       debug_printf("vkEndCommandBuffer failed\n");
  68       return;
  69    }
  70
  71    assert(batch->fence == NULL);
  72    batch->fence = zink_create_fence(ctx->base.screen);
  73    if (!batch->fence)
  74       return;
  75
  76    VkSubmitInfo si = {};
  77    si.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO;
  78    si.waitSemaphoreCount = 0;
  79    si.pWaitSemaphores = NULL;
  80    si.signalSemaphoreCount = 0;
  81    si.pSignalSemaphores = NULL;
  82    si.pWaitDstStageMask = NULL;
  83    si.commandBufferCount = 1;
  84    si.pCommandBuffers = &batch->cmdbuf;
  85
  86    if (vkQueueSubmit(ctx->queue, 1, &si, batch->fence->fence) != VK_SUCCESS)
  87       debug_printf("vkQueueSubmit failed\n");
  88 }
  89
  90 void
  91 zink_batch_reference_resoure(struct zink_batch *batch,
  92                              struct zink_resource *res)
  93 {
  94    struct set_entry *entry = _mesa_set_search(batch->resources, res);
  95    if (!entry) {
  96       struct pipe_resource *tmp = NULL;
  97       entry = _mesa_set_add(batch->resources, res);
  98       pipe_resource_reference(&tmp, &res->base);
  99    }
 100 }
 101
 102 void
 103 zink_batch_reference_sampler_view(struct zink_batch *batch,
 104                                   struct zink_sampler_view *sv)
 105 {
 106    struct set_entry *entry = _mesa_set_search(batch->sampler_views, sv);
 107    if (!entry) {
 108       struct pipe_sampler_view *tmp = NULL;
 109       entry = _mesa_set_add(batch->sampler_views, sv);
 110       pipe_sampler_view_reference(&tmp, &sv->base);
 111    }
 112 }