1 files changed, 63 insertions, 63 deletions

diff --git a/src/mesa/pipe/nv40/nv40_query.c b/src/mesa/pipe/nv40/nv40_query.c
index bcd6fe0cf4..f51b34b119 100644
--- a/src/mesa/pipe/nv40/nv40_query.c
+++ b/src/mesa/pipe/nv40/nv40_query.c
@@ -3,33 +3,57 @@
 #include "nv40_context.h"
 #include "nv40_dma.h"
 
-static uint
-nv40_query_object_find(struct nv40_context *nv40, struct pipe_query_object *q)
+/*XXX: Maybe go notifier per-query one day?  not sure if PRAMIN space is
+ *     plentiful enough however.. */
+struct nv40_query {
+	unsigned type;
+	int id;
+};
+#define nv40_query(o) ((struct nv40_query *)(o))
+
+static struct pipe_query *
+nv40_query_create(struct pipe_context *pipe, unsigned query_type)
 {
+	struct nv40_context *nv40 = nv40_context(pipe);
+	struct nv40_query *nv40query;
 	int id;
 
 	for (id = 0; id < nv40->num_query_objects; id++) {
-		if (nv40->query_objects[id] == q)
-			return id;
+		if (nv40->query_objects[id] == 0)
+			break;
 	}
+	
+	if (id == nv40->num_query_objects)
+		return NULL;
+	nv40->query_objects[id] = TRUE;
 
-	return -1;
+	nv40query = malloc(sizeof(struct nv40_query));
+	nv40query->type = query_type;
+	nv40query->id = id;
+
+	return (struct pipe_query *)nv40query;
 }
 
 static void
-nv40_query_begin(struct pipe_context *pipe, struct pipe_query_object *q)
+nv40_query_destroy(struct pipe_context *pipe, struct pipe_query *q)
 {
-	struct nv40_context *nv40 = (struct nv40_context *)pipe;
-	int id;
+	struct nv40_context *nv40 = nv40_context(pipe);
+	struct nv40_query *nv40query = nv40_query(q);
 
-	assert(q->type == PIPE_QUERY_OCCLUSION_COUNTER);
+	assert(nv40->query_objects[nv40query->id]);
+	nv40->query_objects[nv40query->id] = FALSE;
+	free(nv40query);
+}
 
-	id = nv40_query_object_find(nv40, NULL);
-	assert(id >= 0);
-	nv40->query_objects[id] = q;
+static void
+nv40_query_begin(struct pipe_context *pipe, struct pipe_query *q)
+{
+	struct nv40_context *nv40 = nv40_context(pipe);
+	struct nv40_query *nv40query = nv40_query(q);
+
+	assert(nv40query->type == PIPE_QUERY_OCCLUSION_COUNTER);
 
-	nv40->nvws->notifier_reset(nv40->query, id);
-	q->ready = 0;
+	nv40->nvws->notifier_reset(nv40->query, nv40query->id);
 
 	BEGIN_RING(curie, NV40TCL_QUERY_RESET, 1);
 	OUT_RING  (1);
@@ -38,68 +62,44 @@ nv40_query_begin(struct pipe_context *pipe, struct pipe_query_object *q)
 }
 
 static void
-nv40_query_update(struct pipe_context *pipe, struct pipe_query_object *q)
+nv40_query_end(struct pipe_context *pipe, struct pipe_query *q)
 {
 	struct nv40_context *nv40 = (struct nv40_context *)pipe;
-	int id;
+	struct nv40_query *nv40query = nv40_query(q);
 
-	id = nv40_query_object_find(nv40, q);
-	assert(id >= 0);
-
-	if (nv40->nvws->notifier_status(nv40->query, id) == 0) {
-		q->ready = 1;
-		q->count = nv40->nvws->notifier_retval(nv40->query, id);
-		nv40->query_objects[id] = NULL;
-	}
-}
-
-static void
-nv40_query_wait(struct pipe_context *pipe, struct pipe_query_object *q)
-{
-	nv40_query_update(pipe, q);
-	if (!q->ready) {
-		struct nv40_context *nv40 = (struct nv40_context *)pipe;
-		int id;
-		
-		id = nv40_query_object_find(nv40, q);
-		assert(id >= 0);
-
-		nv40->nvws->notifier_wait(nv40->query, id, 0, 0);
-		nv40_query_update(pipe, q);
-		assert(q->ready);
-	}
+	BEGIN_RING(curie, NV40TCL_QUERY_GET, 1);
+	OUT_RING  ((0x01 << NV40TCL_QUERY_GET_UNK24_SHIFT) |
+		   ((nv40query->id * 32) << NV40TCL_QUERY_GET_OFFSET_SHIFT));
+	FIRE_RING();
 }
 
-static void
-nv40_query_end(struct pipe_context *pipe, struct pipe_query_object *q)
+static boolean
+nv40_query_result(struct pipe_context *pipe, struct pipe_query *q,
+		  boolean wait, uint64_t *result)
 {
 	struct nv40_context *nv40 = (struct nv40_context *)pipe;
-	int id;
-
-	id = nv40_query_object_find(nv40, q);
-	assert(id >= 0);
+	struct nv40_query *nv40query = nv40_query(q);
+	struct nouveau_winsys *nvws = nv40->nvws;
+	boolean status;
+
+	status = nvws->notifier_status(nv40->query, nv40query->id);
+	if (status != NV_NOTIFY_STATE_STATUS_COMPLETED) {
+		if (wait == FALSE)
+			return FALSE;
+		nvws->notifier_wait(nv40->query, nv40query->id,
+				    NV_NOTIFY_STATE_STATUS_COMPLETED, 0);
+	}
 
-	BEGIN_RING(curie, NV40TCL_QUERY_GET, 1);
-	OUT_RING  ((0x01 << NV40TCL_QUERY_GET_UNK24_SHIFT) |
-		   ((id * 32) << NV40TCL_QUERY_GET_OFFSET_SHIFT));
-	FIRE_RING ();
-
-	/*XXX: Some apps spin waiting for GL_QUERY_RESULT_AVAILABLE_ARB.
-	 *     Core mesa won't ask the driver to update the query object's
-	 *     status in this case, so the app waits forever.. fix this some
-	 *     day.
-	 */
-#if 0
-	nv40_query_update(pipe, q);
-#else
-	nv40_query_wait(pipe, q);
-#endif
+	*result = nvws->notifier_retval(nv40->query, nv40query->id);
+	return TRUE;
 }
 
 void
 nv40_init_query_functions(struct nv40_context *nv40)
 {
+	nv40->pipe.create_query = nv40_query_create;
+	nv40->pipe.destroy_query = nv40_query_destroy;
 	nv40->pipe.begin_query = nv40_query_begin;
 	nv40->pipe.end_query = nv40_query_end;
-	nv40->pipe.wait_query = nv40_query_wait;
+	nv40->pipe.get_query_result = nv40_query_result;
 }