2 * Copyright 2014, 2015 Red Hat.
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:
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
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.
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"
33 struct virgl_resource
*buf
;
38 unsigned result_gotten_sent
;
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
52 static const int pquery_map
[] =
54 VIRGL_QUERY_OCCLUSION_COUNTER
,
55 VIRGL_QUERY_OCCLUSION_PREDICATE
,
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
,
65 VIRGL_QUERY_GPU_FINISHED
,
66 VIRGL_QUERY_PIPELINE_STATISTICS
,
69 static int pipe_to_virgl_query(enum pipe_query_type ptype
)
71 return pquery_map
[ptype
];
74 static inline struct virgl_query
*virgl_query(struct pipe_query
*q
)
76 return (struct virgl_query
*)q
;
79 static void virgl_render_condition(struct pipe_context
*ctx
,
82 enum pipe_render_cond_flag mode
)
84 struct virgl_context
*vctx
= virgl_context(ctx
);
85 struct virgl_query
*query
= virgl_query(q
);
88 handle
= query
->handle
;
89 virgl_encoder_render_condition(vctx
, handle
, condition
, mode
);
92 static struct pipe_query
*virgl_create_query(struct pipe_context
*ctx
,
93 unsigned query_type
, unsigned index
)
95 struct virgl_context
*vctx
= virgl_context(ctx
);
96 struct virgl_query
*query
;
99 query
= CALLOC_STRUCT(virgl_query
);
103 query
->buf
= (struct virgl_resource
*)pipe_buffer_create(ctx
->screen
, PIPE_BIND_CUSTOM
,
104 PIPE_USAGE_STAGING
, sizeof(struct virgl_host_query_state
));
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);
117 return (struct pipe_query
*)query
;
120 static void virgl_destroy_query(struct pipe_context
*ctx
,
121 struct pipe_query
*q
)
123 struct virgl_context
*vctx
= virgl_context(ctx
);
124 struct virgl_query
*query
= virgl_query(q
);
126 virgl_encode_delete_object(vctx
, query
->handle
, VIRGL_OBJECT_QUERY
);
128 pipe_resource_reference((struct pipe_resource
**)&query
->buf
, NULL
);
132 static boolean
virgl_begin_query(struct pipe_context
*ctx
,
133 struct pipe_query
*q
)
135 struct virgl_context
*vctx
= virgl_context(ctx
);
136 struct virgl_query
*query
= virgl_query(q
);
138 query
->buf
->clean
= FALSE
;
139 virgl_encoder_begin_query(vctx
, query
->handle
);
143 static bool virgl_end_query(struct pipe_context
*ctx
,
144 struct pipe_query
*q
)
146 struct virgl_context
*vctx
= virgl_context(ctx
);
147 struct virgl_query
*query
= virgl_query(q
);
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
,
156 virgl_encoder_end_query(vctx
, query
->handle
);
160 static boolean
virgl_get_query_result(struct pipe_context
*ctx
,
161 struct pipe_query
*q
,
163 union pipe_query_result
*result
)
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
;
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);
177 /* do we have to flush? */
178 /* now we can do the transfer to get the result back? */
180 host_state
= pipe_buffer_map(ctx
, &query
->buf
->u
.b
,
181 PIPE_TRANSFER_READ
, &transfer
);
183 if (host_state
->query_state
!= VIRGL_QUERY_STATE_DONE
) {
184 pipe_buffer_unmap(ctx
, transfer
);
191 if (query
->type
== PIPE_QUERY_TIMESTAMP
|| query
->type
== PIPE_QUERY_TIME_ELAPSED
)
192 result
->u64
= host_state
->result
;
194 result
->u64
= (uint32_t)host_state
->result
;
196 pipe_buffer_unmap(ctx
, transfer
);
197 query
->result_gotten_sent
= 0;
202 virgl_set_active_query_state(struct pipe_context
*pipe
, boolean enable
)
206 void virgl_init_query_functions(struct virgl_context
*vctx
)
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
;