1 /*
2 * Copyright 2014, 2015 Red Hat.
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 * on the rights to use, copy, modify, merge, publish, distribute, sub
8 * license, and/or sell copies of the Software, and to permit persons to whom
9 * the 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 NON-INFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
19 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
20 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
21 * USE OR OTHER DEALINGS IN THE SOFTWARE.
22 */
23
24 #include "util/u_memory.h"
25 #include "util/u_inlines.h"
26 #include "virgl_context.h"
27 #include "virgl_encode.h"
28 #include "virgl_protocol.h"
29 #include "virgl_resource.h"
30
31 struct virgl_query {
32 uint32_t handle;
33 struct virgl_resource *buf;
34
35 unsigned index;
36 unsigned type;
37 unsigned result_size;
38 unsigned result_gotten_sent;
39 };
40 #define VIRGL_QUERY_OCCLUSION_COUNTER 0
41 #define VIRGL_QUERY_OCCLUSION_PREDICATE 1
42 #define VIRGL_QUERY_TIMESTAMP 2
43 #define VIRGL_QUERY_TIMESTAMP_DISJOINT 3
44 #define VIRGL_QUERY_TIME_ELAPSED 4
45 #define VIRGL_QUERY_PRIMITIVES_GENERATED 5
46 #define VIRGL_QUERY_PRIMITIVES_EMITTED 6
47 #define VIRGL_QUERY_SO_STATISTICS 7
48 #define VIRGL_QUERY_SO_OVERFLOW_PREDICATE 8
49 #define VIRGL_QUERY_GPU_FINISHED 9
50 #define VIRGL_QUERY_PIPELINE_STATISTICS 10
51
52 static const int pquery_map[] =
53 {
54 VIRGL_QUERY_OCCLUSION_COUNTER,
55 VIRGL_QUERY_OCCLUSION_PREDICATE,
56 -1,
57 VIRGL_QUERY_TIMESTAMP,
58 VIRGL_QUERY_TIMESTAMP_DISJOINT,
59 VIRGL_QUERY_TIME_ELAPSED,
60 VIRGL_QUERY_PRIMITIVES_GENERATED,
61 VIRGL_QUERY_PRIMITIVES_EMITTED,
62 VIRGL_QUERY_SO_STATISTICS,
63 VIRGL_QUERY_SO_OVERFLOW_PREDICATE,
64 -1,
65 VIRGL_QUERY_GPU_FINISHED,
66 VIRGL_QUERY_PIPELINE_STATISTICS,
67 };
68
pipe_to_virgl_query(enum pipe_query_type ptype)69 static int pipe_to_virgl_query(enum pipe_query_type ptype)
70 {
71 return pquery_map[ptype];
72 }
73
virgl_query(struct pipe_query * q)74 static inline struct virgl_query *virgl_query(struct pipe_query *q)
75 {
76 return (struct virgl_query *)q;
77 }
78
virgl_render_condition(struct pipe_context * ctx,struct pipe_query * q,boolean condition,enum pipe_render_cond_flag mode)79 static void virgl_render_condition(struct pipe_context *ctx,
80 struct pipe_query *q,
81 boolean condition,
82 enum pipe_render_cond_flag mode)
83 {
84 struct virgl_context *vctx = virgl_context(ctx);
85 struct virgl_query *query = virgl_query(q);
86 uint32_t handle = 0;
87 if (q)
88 handle = query->handle;
89 virgl_encoder_render_condition(vctx, handle, condition, mode);
90 }
91
virgl_create_query(struct pipe_context * ctx,unsigned query_type,unsigned index)92 static struct pipe_query *virgl_create_query(struct pipe_context *ctx,
93 unsigned query_type, unsigned index)
94 {
95 struct virgl_context *vctx = virgl_context(ctx);
96 struct virgl_query *query;
97 uint32_t handle;
98
99 query = CALLOC_STRUCT(virgl_query);
100 if (!query)
101 return NULL;
102
103 query->buf = (struct virgl_resource *)pipe_buffer_create(ctx->screen, PIPE_BIND_CUSTOM,
104 PIPE_USAGE_STAGING, sizeof(struct virgl_host_query_state));
105 if (!query->buf) {
106 FREE(query);
107 return NULL;
108 }
109
110 handle = virgl_object_assign_handle();
111 query->type = pipe_to_virgl_query(query_type);
112 query->index = index;
113 query->handle = handle;
114 query->buf->clean = FALSE;
115 virgl_encoder_create_query(vctx, handle, query->type, index, query->buf, 0);
116
117 return (struct pipe_query *)query;
118 }
119
virgl_destroy_query(struct pipe_context * ctx,struct pipe_query * q)120 static void virgl_destroy_query(struct pipe_context *ctx,
121 struct pipe_query *q)
122 {
123 struct virgl_context *vctx = virgl_context(ctx);
124 struct virgl_query *query = virgl_query(q);
125
126 virgl_encode_delete_object(vctx, query->handle, VIRGL_OBJECT_QUERY);
127
128 pipe_resource_reference((struct pipe_resource **)&query->buf, NULL);
129 FREE(query);
130 }
131
virgl_begin_query(struct pipe_context * ctx,struct pipe_query * q)132 static boolean virgl_begin_query(struct pipe_context *ctx,
133 struct pipe_query *q)
134 {
135 struct virgl_context *vctx = virgl_context(ctx);
136 struct virgl_query *query = virgl_query(q);
137
138 query->buf->clean = FALSE;
139 virgl_encoder_begin_query(vctx, query->handle);
140 return true;
141 }
142
virgl_end_query(struct pipe_context * ctx,struct pipe_query * q)143 static bool virgl_end_query(struct pipe_context *ctx,
144 struct pipe_query *q)
145 {
146 struct virgl_context *vctx = virgl_context(ctx);
147 struct virgl_query *query = virgl_query(q);
148 struct pipe_box box;
149
150 uint32_t qs = VIRGL_QUERY_STATE_WAIT_HOST;
151 u_box_1d(0, 4, &box);
152 virgl_transfer_inline_write(ctx, &query->buf->u.b, 0, PIPE_TRANSFER_WRITE,
153 &box, &qs, 0, 0);
154
155
156 virgl_encoder_end_query(vctx, query->handle);
157 return true;
158 }
159
virgl_get_query_result(struct pipe_context * ctx,struct pipe_query * q,boolean wait,union pipe_query_result * result)160 static boolean virgl_get_query_result(struct pipe_context *ctx,
161 struct pipe_query *q,
162 boolean wait,
163 union pipe_query_result *result)
164 {
165 struct virgl_context *vctx = virgl_context(ctx);
166 struct virgl_query *query = virgl_query(q);
167 struct pipe_transfer *transfer;
168 struct virgl_host_query_state *host_state;
169
170 /* ask host for query result */
171 if (!query->result_gotten_sent) {
172 query->result_gotten_sent = 1;
173 virgl_encoder_get_query_result(vctx, query->handle, 0);
174 ctx->flush(ctx, NULL, 0);
175 }
176
177 /* do we have to flush? */
178 /* now we can do the transfer to get the result back? */
179 remap:
180 host_state = pipe_buffer_map(ctx, &query->buf->u.b,
181 PIPE_TRANSFER_READ, &transfer);
182
183 if (host_state->query_state != VIRGL_QUERY_STATE_DONE) {
184 pipe_buffer_unmap(ctx, transfer);
185 if (wait)
186 goto remap;
187 else
188 return FALSE;
189 }
190
191 if (query->type == PIPE_QUERY_TIMESTAMP || query->type == PIPE_QUERY_TIME_ELAPSED)
192 result->u64 = host_state->result;
193 else
194 result->u64 = (uint32_t)host_state->result;
195
196 pipe_buffer_unmap(ctx, transfer);
197 query->result_gotten_sent = 0;
198 return TRUE;
199 }
200
201 static void
virgl_set_active_query_state(struct pipe_context * pipe,boolean enable)202 virgl_set_active_query_state(struct pipe_context *pipe, boolean enable)
203 {
204 }
205
virgl_init_query_functions(struct virgl_context * vctx)206 void virgl_init_query_functions(struct virgl_context *vctx)
207 {
208 vctx->base.render_condition = virgl_render_condition;
209 vctx->base.create_query = virgl_create_query;
210 vctx->base.destroy_query = virgl_destroy_query;
211 vctx->base.begin_query = virgl_begin_query;
212 vctx->base.end_query = virgl_end_query;
213 vctx->base.get_query_result = virgl_get_query_result;
214 vctx->base.set_active_query_state = virgl_set_active_query_state;
215 }
216