src/mesa/state_tracker/st_cb_queryobj.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2007 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28
  29 /**
  30  * glBegin/EndQuery interface to pipe
  31  *
  32  * \author Brian Paul
  33  */
  34
  35
  36 #include "main/imports.h"
  37 #include "main/context.h"
  38
  39 #include "pipe/p_context.h"
  40 #include "pipe/p_defines.h"
  41 #include "pipe/p_screen.h"
  42 #include "st_context.h"
  43 #include "st_cb_queryobj.h"
  44 #include "st_cb_bitmap.h"
  45
  46
  47 static struct gl_query_object *
  48 st_NewQueryObject(struct gl_context *ctx, GLuint id)
  49 {
  50    struct st_query_object *stq = ST_CALLOC_STRUCT(st_query_object);
  51    if (stq) {
  52       stq->base.Id = id;
  53       stq->base.Ready = GL_TRUE;
  54       stq->pq = NULL;
  55       stq->type = PIPE_QUERY_TYPES; /* an invalid value */
  56       return &stq->base;
  57    }
  58    return NULL;
  59 }
  60
  61
  62
  63 static void
  64 st_DeleteQuery(struct gl_context *ctx, struct gl_query_object *q)
  65 {
  66    struct pipe_context *pipe = st_context(ctx)->pipe;
  67    struct st_query_object *stq = st_query_object(q);
  68
  69    if (stq->pq) {
  70       pipe->destroy_query(pipe, stq->pq);
  71       stq->pq = NULL;
  72    }
  73
  74    if (stq->pq_begin) {
  75       pipe->destroy_query(pipe, stq->pq_begin);
  76       stq->pq_begin = NULL;
  77    }
  78
  79    free(stq);
  80 }
  81
  82
  83 static void
  84 st_BeginQuery(struct gl_context *ctx, struct gl_query_object *q)
  85 {
  86    struct st_context *st = st_context(ctx);
  87    struct pipe_context *pipe = st->pipe;
  88    struct st_query_object *stq = st_query_object(q);
  89    unsigned type;
  90
  91    st_flush_bitmap_cache(st_context(ctx));
  92
  93    /* convert GL query type to Gallium query type */
  94    switch (q->Target) {
  95    case GL_ANY_SAMPLES_PASSED:
  96    case GL_ANY_SAMPLES_PASSED_CONSERVATIVE:
  97       /* fall-through */
  98    case GL_SAMPLES_PASSED_ARB:
  99       type = PIPE_QUERY_OCCLUSION_COUNTER;
 100       break;
 101    case GL_PRIMITIVES_GENERATED:
 102       type = PIPE_QUERY_PRIMITIVES_GENERATED;
 103       break;
 104    case GL_TRANSFORM_FEEDBACK_PRIMITIVES_WRITTEN:
 105       type = PIPE_QUERY_PRIMITIVES_EMITTED;
 106       break;
 107    case GL_TIME_ELAPSED:
 108       if (st->has_time_elapsed)
 109          type = PIPE_QUERY_TIME_ELAPSED;
 110       else
 111          type = PIPE_QUERY_TIMESTAMP;
 112       break;
 113    case GL_VERTICES_SUBMITTED_ARB:
 114    case GL_PRIMITIVES_SUBMITTED_ARB:
 115    case GL_VERTEX_SHADER_INVOCATIONS_ARB:
 116    case GL_TESS_CONTROL_SHADER_PATCHES_ARB:
 117    case GL_TESS_EVALUATION_SHADER_INVOCATIONS_ARB:
 118    case GL_GEOMETRY_SHADER_INVOCATIONS:
 119    case GL_GEOMETRY_SHADER_PRIMITIVES_EMITTED_ARB:
 120    case GL_FRAGMENT_SHADER_INVOCATIONS_ARB:
 121    case GL_COMPUTE_SHADER_INVOCATIONS_ARB:
 122    case GL_CLIPPING_INPUT_PRIMITIVES_ARB:
 123    case GL_CLIPPING_OUTPUT_PRIMITIVES_ARB:
 124       type = PIPE_QUERY_PIPELINE_STATISTICS;
 125       break;
 126    default:
 127       assert(0 && "unexpected query target in st_BeginQuery()");
 128       return;
 129    }
 130
 131    if (stq->type != type) {
 132       /* free old query of different type */
 133       if (stq->pq) {
 134          pipe->destroy_query(pipe, stq->pq);
 135          stq->pq = NULL;
 136       }
 137       if (stq->pq_begin) {
 138          pipe->destroy_query(pipe, stq->pq_begin);
 139          stq->pq_begin = NULL;
 140       }
 141       stq->type = PIPE_QUERY_TYPES; /* an invalid value */
 142    }
 143
 144    if (q->Target == GL_TIME_ELAPSED &&
 145        type == PIPE_QUERY_TIMESTAMP) {
 146       /* Determine time elapsed by emitting two timestamp queries. */
 147       if (!stq->pq_begin) {
 148          stq->pq_begin = pipe->create_query(pipe, type, 0);
 149          stq->type = type;
 150       }
 151       pipe->end_query(pipe, stq->pq_begin);
 152    } else {
 153       if (!stq->pq) {
 154          stq->pq = pipe->create_query(pipe, type, q->Stream);
 155          stq->type = type;
 156       }
 157       if (stq->pq) {
 158          pipe->begin_query(pipe, stq->pq);
 159       }
 160       else {
 161          _mesa_error(ctx, GL_OUT_OF_MEMORY, "glBeginQuery");
 162          return;
 163       }
 164    }
 165    assert(stq->type == type);
 166 }
 167
 168
 169 static void
 170 st_EndQuery(struct gl_context *ctx, struct gl_query_object *q)
 171 {
 172    struct pipe_context *pipe = st_context(ctx)->pipe;
 173    struct st_query_object *stq = st_query_object(q);
 174
 175    st_flush_bitmap_cache(st_context(ctx));
 176
 177    if ((q->Target == GL_TIMESTAMP ||
 178         q->Target == GL_TIME_ELAPSED) &&
 179        !stq->pq) {
 180       stq->pq = pipe->create_query(pipe, PIPE_QUERY_TIMESTAMP, 0);
 181       stq->type = PIPE_QUERY_TIMESTAMP;
 182    }
 183
 184    if (stq->pq)
 185       pipe->end_query(pipe, stq->pq);
 186 }
 187
 188
 189 static boolean
 190 get_query_result(struct pipe_context *pipe,
 191                  struct st_query_object *stq,
 192                  boolean wait)
 193 {
 194    union pipe_query_result data;
 195
 196    if (!stq->pq) {
 197       /* Only needed in case we failed to allocate the gallium query earlier.
 198        * Return TRUE so we don't spin on this forever.
 199        */
 200       return TRUE;
 201    }
 202
 203    if (!pipe->get_query_result(pipe, stq->pq, wait, &data))
 204       return FALSE;
 205
 206    switch (stq->base.Target) {
 207    case GL_VERTICES_SUBMITTED_ARB:
 208       stq->base.Result = data.pipeline_statistics.ia_vertices;
 209       break;
 210    case GL_PRIMITIVES_SUBMITTED_ARB:
 211       stq->base.Result = data.pipeline_statistics.ia_primitives;
 212       break;
 213    case GL_VERTEX_SHADER_INVOCATIONS_ARB:
 214       stq->base.Result = data.pipeline_statistics.vs_invocations;
 215       break;
 216    case GL_TESS_CONTROL_SHADER_PATCHES_ARB:
 217       stq->base.Result = data.pipeline_statistics.hs_invocations;
 218       break;
 219    case GL_TESS_EVALUATION_SHADER_INVOCATIONS_ARB:
 220       stq->base.Result = data.pipeline_statistics.ds_invocations;
 221       break;
 222    case GL_GEOMETRY_SHADER_INVOCATIONS:
 223       stq->base.Result = data.pipeline_statistics.gs_invocations;
 224       break;
 225    case GL_GEOMETRY_SHADER_PRIMITIVES_EMITTED_ARB:
 226       stq->base.Result = data.pipeline_statistics.gs_primitives;
 227       break;
 228    case GL_FRAGMENT_SHADER_INVOCATIONS_ARB:
 229       stq->base.Result = data.pipeline_statistics.ps_invocations;
 230       break;
 231    case GL_COMPUTE_SHADER_INVOCATIONS_ARB:
 232       stq->base.Result = data.pipeline_statistics.cs_invocations;
 233       break;
 234    case GL_CLIPPING_INPUT_PRIMITIVES_ARB:
 235       stq->base.Result = data.pipeline_statistics.c_invocations;
 236       break;
 237    case GL_CLIPPING_OUTPUT_PRIMITIVES_ARB:
 238       stq->base.Result = data.pipeline_statistics.c_primitives;
 239       break;
 240    default:
 241       stq->base.Result = data.u64;
 242       break;
 243    }
 244
 245    if (stq->base.Target == GL_TIME_ELAPSED &&
 246        stq->type == PIPE_QUERY_TIMESTAMP) {
 247       /* Calculate the elapsed time from the two timestamp queries */
 248       GLuint64EXT Result0 = 0;
 249       assert(stq->pq_begin);
 250       pipe->get_query_result(pipe, stq->pq_begin, TRUE, (void *)&Result0);
 251       stq->base.Result -= Result0;
 252    } else {
 253       assert(!stq->pq_begin);
 254    }
 255
 256    return TRUE;
 257 }
 258
 259
 260 static void
 261 st_WaitQuery(struct gl_context *ctx, struct gl_query_object *q)
 262 {
 263    struct pipe_context *pipe = st_context(ctx)->pipe;
 264    struct st_query_object *stq = st_query_object(q);
 265
 266    /* this function should only be called if we don't have a ready result */
 267    assert(!stq->base.Ready);
 268
 269    while (!stq->base.Ready &&
 270           !get_query_result(pipe, stq, TRUE))
 271    {
 272       /* nothing */
 273    }
 274
 275    q->Ready = GL_TRUE;
 276 }
 277
 278
 279 static void
 280 st_CheckQuery(struct gl_context *ctx, struct gl_query_object *q)
 281 {
 282    struct pipe_context *pipe = st_context(ctx)->pipe;
 283    struct st_query_object *stq = st_query_object(q);
 284    assert(!q->Ready);   /* we should not get called if Ready is TRUE */
 285    q->Ready = get_query_result(pipe, stq, FALSE);
 286 }
 287
 288
 289 static uint64_t
 290 st_GetTimestamp(struct gl_context *ctx)
 291 {
 292    struct pipe_context *pipe = st_context(ctx)->pipe;
 293    struct pipe_screen *screen = pipe->screen;
 294
 295    /* Prefer the per-screen function */
 296    if (screen->get_timestamp) {
 297       return screen->get_timestamp(screen);
 298    }
 299    else {
 300       /* Fall back to the per-context function */
 301       assert(pipe->get_timestamp);
 302       return pipe->get_timestamp(pipe);
 303    }
 304 }
 305
 306
 307 void st_init_query_functions(struct dd_function_table *functions)
 308 {
 309    functions->NewQueryObject = st_NewQueryObject;
 310    functions->DeleteQuery = st_DeleteQuery;
 311    functions->BeginQuery = st_BeginQuery;
 312    functions->EndQuery = st_EndQuery;
 313    functions->WaitQuery = st_WaitQuery;
 314    functions->CheckQuery = st_CheckQuery;
 315    functions->GetTimestamp = st_GetTimestamp;
 316 }