1// Copyright 2018 The SwiftShader Authors. All Rights Reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15#ifndef VK_QUERY_POOL_HPP_
16#define VK_QUERY_POOL_HPP_
17
18#include "VkObject.hpp"
19
20#include "marl/event.h"
21#include "marl/waitgroup.h"
22
23#include <atomic>
24#include <condition_variable>
25#include <mutex>
26
27namespace vk {
28
29class Query
30{
31public:
32 Query(VkQueryType type);
33
34 enum State
35 {
36 UNAVAILABLE,
37 ACTIVE,
38 FINISHED
39 };
40
41 struct Data
42 {
43 State state; // The current query state.
44 int64_t value; // The current query value.
45 };
46
47 // reset() sets the state of the Query to UNAVAILABLE, sets the type to
48 // INVALID_TYPE and clears the query value.
49 // reset() must not be called while the query is in the ACTIVE state.
50 void reset();
51
52 // start() begins a query task which is closed with a call to finish().
53 // Query tasks can be nested.
54 // start() sets the state to ACTIVE.
55 void start();
56
57 // finish() ends a query task begun with a call to start().
58 // Once all query tasks are complete the query will transition to the
59 // FINISHED state.
60 // finish() must only be called when in the ACTIVE state.
61 void finish();
62
63 // wait() blocks until the query reaches the FINISHED state.
64 void wait();
65
66 // getData() returns the current query state and value.
67 Data getData() const;
68
69 // getType() returns the type of query.
70 VkQueryType getType() const;
71
72 // set() replaces the current query value with val.
73 void set(int64_t val);
74
75 // add() adds val to the current query value.
76 void add(int64_t val);
77
78private:
79 marl::WaitGroup wg;
80 marl::Event finished;
81 std::atomic<State> state;
82 std::atomic<VkQueryType> type;
83 std::atomic<int64_t> value;
84};
85
86class QueryPool : public Object<QueryPool, VkQueryPool>
87{
88public:
89 QueryPool(const VkQueryPoolCreateInfo *pCreateInfo, void *mem);
90 void destroy(const VkAllocationCallbacks *pAllocator);
91
92 static size_t ComputeRequiredAllocationSize(const VkQueryPoolCreateInfo *pCreateInfo);
93
94 VkResult getResults(uint32_t firstQuery, uint32_t queryCount, size_t dataSize,
95 void *pData, VkDeviceSize stride, VkQueryResultFlags flags) const;
96 void begin(uint32_t query, VkQueryControlFlags flags);
97 void end(uint32_t query);
98 void reset(uint32_t firstQuery, uint32_t queryCount);
99
100 void writeTimestamp(uint32_t query);
101
102 inline Query *getQuery(uint32_t query) const { return &pool[query]; }
103 inline VkQueryType getType() const { return type; }
104
105private:
106 Query *const pool;
107 const VkQueryType type;
108 const uint32_t count;
109};
110
111static inline QueryPool *Cast(VkQueryPool object)
112{
113 return QueryPool::Cast(vkObject: object);
114}
115
116} // namespace vk
117
118#endif // VK_QUERY_POOL_HPP_
119

source code of flutter_engine/third_party/swiftshader/src/Vulkan/VkQueryPool.hpp