src/gallium/drivers/nvfx/nvfx_query.c

   1 #include "pipe/p_context.h"
   2
   3 #include "nvfx_context.h"
   4
   5 struct nvfx_query {
   6         struct list_head list;
   7         struct nouveau_resource *object;
   8         unsigned type;
   9         boolean ready;
  10         uint64_t result;
  11 };
  12
  13 static INLINE struct nvfx_query *
  14 nvfx_query(struct pipe_query *pipe)
  15 {
  16         return (struct nvfx_query *)pipe;
  17 }
  18
  19 static struct pipe_query *
  20 nvfx_query_create(struct pipe_context *pipe, unsigned query_type)
  21 {
  22         struct nvfx_query *q;
  23
  24         q = CALLOC(1, sizeof(struct nvfx_query));
  25         q->type = query_type;
  26
  27         assert(q->type == PIPE_QUERY_OCCLUSION_COUNTER);
  28
  29         return (struct pipe_query *)q;
  30 }
  31
  32 static void
  33 nvfx_query_destroy(struct pipe_context *pipe, struct pipe_query *pq)
  34 {
  35         struct nvfx_query *q = nvfx_query(pq);
  36
  37         if (q->object)
  38         {
  39                 nouveau_resource_free(&q->object);
  40                 LIST_DEL(&q->list);
  41         }
  42         FREE(q);
  43 }
  44
  45 static void
  46 nvfx_query_begin(struct pipe_context *pipe, struct pipe_query *pq)
  47 {
  48         struct nvfx_context *nvfx = nvfx_context(pipe);
  49         struct nvfx_query *q = nvfx_query(pq);
  50         struct nvfx_screen *screen = nvfx->screen;
  51         struct nouveau_channel *chan = screen->base.channel;
  52         struct nouveau_grobj *eng3d = screen->eng3d;
  53         uint64_t tmp;
  54
  55         /* Happens when end_query() is called, then another begin_query()
  56          * without querying the result in-between.  For now we'll wait for
  57          * the existing query to notify completion, but it could be better.
  58          */
  59         if (q->object)
  60                 pipe->get_query_result(pipe, pq, 1, &tmp);
  61
  62         while (nouveau_resource_alloc(nvfx->screen->query_heap, 1, NULL, &q->object))
  63         {
  64                 struct nvfx_query* oldestq;
  65                 assert(!LIST_IS_EMPTY(&nvfx->screen->query_list));
  66                 oldestq = LIST_ENTRY(struct nvfx_query, nvfx->screen->query_list.next, list);
  67                 pipe->get_query_result(pipe, (struct pipe_query*)oldestq, 1, &tmp);
  68         }
  69
  70         LIST_ADDTAIL(&q->list, &nvfx->screen->query_list);
  71
  72         nouveau_notifier_reset(nvfx->screen->query, q->object->start);
  73
  74         BEGIN_RING(chan, eng3d, NV34TCL_QUERY_RESET, 1);
  75         OUT_RING  (chan, 1);
  76         BEGIN_RING(chan, eng3d, NV34TCL_QUERY_UNK17CC, 1);
  77         OUT_RING  (chan, 1);
  78
  79         q->ready = FALSE;
  80 }
  81
  82 static void
  83 nvfx_query_end(struct pipe_context *pipe, struct pipe_query *pq)
  84 {
  85         struct nvfx_context *nvfx = nvfx_context(pipe);
  86         struct nvfx_screen *screen = nvfx->screen;
  87         struct nouveau_channel *chan = screen->base.channel;
  88         struct nouveau_grobj *eng3d = screen->eng3d;
  89         struct nvfx_query *q = nvfx_query(pq);
  90
  91         BEGIN_RING(chan, eng3d, NV34TCL_QUERY_GET, 1);
  92         OUT_RING  (chan, (0x01 << NV34TCL_QUERY_GET_UNK24_SHIFT) |
  93                    ((q->object->start * 32) << NV34TCL_QUERY_GET_OFFSET_SHIFT));
  94         FIRE_RING(chan);
  95 }
  96
  97 static boolean
  98 nvfx_query_result(struct pipe_context *pipe, struct pipe_query *pq,
  99                   boolean wait, uint64_t *result)
 100 {
 101         struct nvfx_context *nvfx = nvfx_context(pipe);
 102         struct nvfx_query *q = nvfx_query(pq);
 103
 104         if (!q->ready) {
 105                 unsigned status;
 106
 107                 status = nouveau_notifier_status(nvfx->screen->query,
 108                                                  q->object->start);
 109                 if (status != NV_NOTIFY_STATE_STATUS_COMPLETED) {
 110                         if (wait == FALSE)
 111                                 return FALSE;
 112
 113                         nouveau_notifier_wait_status(nvfx->screen->query,
 114                                         q->object->start,
 115                                         NV_NOTIFY_STATE_STATUS_COMPLETED, 0);
 116                 }
 117
 118                 q->result = nouveau_notifier_return_val(nvfx->screen->query,
 119                                                         q->object->start);
 120                 q->ready = TRUE;
 121                 nouveau_resource_free(&q->object);
 122                 LIST_DEL(&q->list);
 123         }
 124
 125         *result = q->result;
 126         return TRUE;
 127 }
 128
 129 void
 130 nvfx_init_query_functions(struct nvfx_context *nvfx)
 131 {
 132         nvfx->pipe.create_query = nvfx_query_create;
 133         nvfx->pipe.destroy_query = nvfx_query_destroy;
 134         nvfx->pipe.begin_query = nvfx_query_begin;
 135         nvfx->pipe.end_query = nvfx_query_end;
 136         nvfx->pipe.get_query_result = nvfx_query_result;
 137 }