gallium: add basic support for stream output queries
[mesa.git] / src / gallium / drivers / softpipe / sp_query.c
1 /**************************************************************************
2 *
3 * Copyright 2007 Tungsten Graphics, Inc., Cedar Park, Texas.
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 TUNGSTEN GRAPHICS 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 /* Author:
29 * Keith Whitwell <keith@tungstengraphics.com>
30 */
31
32 #include "draw/draw_context.h"
33 #include "os/os_time.h"
34 #include "pipe/p_defines.h"
35 #include "util/u_memory.h"
36 #include "sp_context.h"
37 #include "sp_query.h"
38 #include "sp_state.h"
39
40 struct softpipe_query {
41 unsigned type;
42 uint64_t start;
43 uint64_t end;
44 struct pipe_query_data_so_statistics so;
45 };
46
47
48 static struct softpipe_query *softpipe_query( struct pipe_query *p )
49 {
50 return (struct softpipe_query *)p;
51 }
52
53 static struct pipe_query *
54 softpipe_create_query(struct pipe_context *pipe,
55 unsigned type)
56 {
57 struct softpipe_query* sq;
58
59 assert(type == PIPE_QUERY_OCCLUSION_COUNTER ||
60 type == PIPE_QUERY_TIME_ELAPSED ||
61 type == PIPE_QUERY_SO_STATISTICS);
62 sq = CALLOC_STRUCT( softpipe_query );
63 sq->type = type;
64
65 return (struct pipe_query *)sq;
66 }
67
68
69 static void
70 softpipe_destroy_query(struct pipe_context *pipe, struct pipe_query *q)
71 {
72 FREE(q);
73 }
74
75
76 static void
77 softpipe_begin_query(struct pipe_context *pipe, struct pipe_query *q)
78 {
79 struct softpipe_context *softpipe = softpipe_context( pipe );
80 struct softpipe_query *sq = softpipe_query(q);
81
82 switch (sq->type) {
83 case PIPE_QUERY_OCCLUSION_COUNTER:
84 sq->start = softpipe->occlusion_count;
85 break;
86 case PIPE_QUERY_TIME_ELAPSED:
87 sq->start = 1000*os_time_get();
88 break;
89 case PIPE_QUERY_SO_STATISTICS:
90 sq->so.num_primitives_written = 0;
91 sq->so.primitives_storage_needed = 0;
92 default:
93 assert(0);
94 break;
95 }
96 softpipe->active_query_count++;
97 softpipe->dirty |= SP_NEW_QUERY;
98 }
99
100
101 static void
102 softpipe_end_query(struct pipe_context *pipe, struct pipe_query *q)
103 {
104 struct softpipe_context *softpipe = softpipe_context( pipe );
105 struct softpipe_query *sq = softpipe_query(q);
106
107 softpipe->active_query_count--;
108 switch (sq->type) {
109 case PIPE_QUERY_OCCLUSION_COUNTER:
110 sq->end = softpipe->occlusion_count;
111 break;
112 case PIPE_QUERY_TIME_ELAPSED:
113 sq->end = 1000*os_time_get();
114 break;
115 case PIPE_QUERY_SO_STATISTICS:
116 sq->so.num_primitives_written =
117 softpipe->so_stats.num_primitives_written;
118 sq->so.primitives_storage_needed =
119 softpipe->so_stats.primitives_storage_needed;
120 default:
121 assert(0);
122 break;
123 }
124 softpipe->dirty |= SP_NEW_QUERY;
125 }
126
127
128 static boolean
129 softpipe_get_query_result(struct pipe_context *pipe,
130 struct pipe_query *q,
131 boolean wait,
132 void *vresult)
133 {
134 struct softpipe_query *sq = softpipe_query(q);
135 uint64_t *result = (uint64_t*)vresult;
136
137 switch (sq->type) {
138 case PIPE_QUERY_SO_STATISTICS:
139 memcpy(vresult, &sq->so,
140 sizeof(struct pipe_query_data_so_statistics));
141 break;
142 default:
143 *result = sq->end - sq->start;
144 break;
145 }
146 return TRUE;
147 }
148
149
150 /**
151 * Called by rendering function to check rendering is conditional.
152 * \return TRUE if we should render, FALSE if we should skip rendering
153 */
154 boolean
155 softpipe_check_render_cond(struct softpipe_context *sp)
156 {
157 struct pipe_context *pipe = &sp->pipe;
158 boolean b, wait;
159 uint64_t result;
160
161 if (!sp->render_cond_query) {
162 return TRUE; /* no query predicate, draw normally */
163 }
164
165 wait = (sp->render_cond_mode == PIPE_RENDER_COND_WAIT ||
166 sp->render_cond_mode == PIPE_RENDER_COND_BY_REGION_WAIT);
167
168 b = pipe->get_query_result(pipe, sp->render_cond_query, wait, &result);
169 if (b)
170 return result > 0;
171 else
172 return TRUE;
173 }
174
175
176 void softpipe_init_query_funcs(struct softpipe_context *softpipe )
177 {
178 softpipe->pipe.create_query = softpipe_create_query;
179 softpipe->pipe.destroy_query = softpipe_destroy_query;
180 softpipe->pipe.begin_query = softpipe_begin_query;
181 softpipe->pipe.end_query = softpipe_end_query;
182 softpipe->pipe.get_query_result = softpipe_get_query_result;
183 }
184
185