1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
gpu / command_buffer / service / client_service_map.h [blame]
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef GPU_COMMAND_BUFFER_SERVICE_CLIENT_SERVICE_MAP_H_
#define GPU_COMMAND_BUFFER_SERVICE_CLIENT_SERVICE_MAP_H_
#include <limits>
#include <unordered_map>
#include <vector>
#include "base/check.h"
namespace gpu {
namespace gles2 {
template <typename ClientType, typename ServiceType>
class ClientServiceMap {
public:
ClientServiceMap()
: ClientServiceMap(std::numeric_limits<ServiceType>::max()) {}
explicit ClientServiceMap(ServiceType invalid_service_id)
: invalid_service_id_(invalid_service_id),
client_to_service_array_(kInitialFlatArraySize, invalid_service_id),
client_to_service_map_() {}
void SetIDMapping(ClientType client_id, ServiceType service_id) {
DCHECK(service_id != invalid_service_id());
if (client_id < kMaxFlatArraySize) {
if (client_id >= client_to_service_array_.size()) {
// Resize the array to the next power of 2 above client_id
size_t new_size = client_to_service_array_.size();
while (client_id >= new_size) {
new_size *= 2;
}
DCHECK(new_size <= kMaxFlatArraySize);
client_to_service_array_.resize(new_size, invalid_service_id());
}
DCHECK(client_to_service_array_[client_id] == invalid_service_id());
client_to_service_array_[client_id] = service_id;
} else {
DCHECK(client_to_service_map_.count(client_id) == 0);
client_to_service_map_[client_id] = service_id;
}
}
void RemoveClientID(ClientType client_id) {
if (client_id < kMaxFlatArraySize) {
if (client_id < client_to_service_array_.size()) {
client_to_service_array_[client_id] = invalid_service_id();
}
} else {
client_to_service_map_.erase(client_id);
}
}
void Clear() {
client_to_service_array_.clear();
client_to_service_array_.resize(kInitialFlatArraySize,
invalid_service_id());
client_to_service_map_.clear();
}
ALWAYS_INLINE bool GetServiceID(ClientType client_id,
ServiceType* service_id) const {
DCHECK(service_id);
if (client_id >= kMaxFlatArraySize) {
return GetServiceIDFromMap(client_id, service_id);
}
if (client_id < client_to_service_array_.size() &&
client_to_service_array_[client_id] != invalid_service_id()) {
*service_id = client_to_service_array_[client_id];
return true;
}
if (client_id == 0) {
*service_id = ServiceType{};
return true;
}
return false;
}
ALWAYS_INLINE bool HasClientID(ClientType client_id) const {
if (client_id >= kMaxFlatArraySize) {
return GetServiceIDFromMap(client_id, nullptr);
}
return client_id == 0 ||
(client_id < client_to_service_array_.size() &&
client_to_service_array_[client_id] != invalid_service_id());
}
ServiceType GetServiceIDOrInvalid(ClientType client_id) {
ServiceType service_id;
if (GetServiceID(client_id, &service_id)) {
return service_id;
}
return invalid_service_id();
}
bool GetClientID(ServiceType service_id, ClientType* client_id) const {
DCHECK(service_id != invalid_service_id());
for (size_t client_id_idx = 0;
client_id_idx < client_to_service_array_.size(); client_id_idx++) {
if (client_to_service_array_[client_id_idx] == service_id) {
if (client_id) {
*client_id = client_id_idx;
}
return true;
}
}
for (const auto& mapping : client_to_service_map_) {
if (mapping.second == service_id) {
if (client_id) {
*client_id = mapping.first;
}
return true;
}
}
if (service_id == 0) {
*client_id = 0;
}
return false;
}
const ServiceType& invalid_service_id() const { return invalid_service_id_; }
template <typename FunctionType>
void ForEach(FunctionType func) const {
for (size_t client_id_idx = 0;
client_id_idx < client_to_service_array_.size(); client_id_idx++) {
if (client_to_service_array_[client_id_idx] != invalid_service_id()) {
func(client_id_idx, client_to_service_array_[client_id_idx]);
}
}
for (const auto& mapping : client_to_service_map_) {
func(mapping.first, mapping.second);
}
}
// Start with 32 maximum elements in the map, which can grow.
static constexpr size_t kInitialFlatArraySize = 0x20;
// Experimental testing suggests that 16k is a reasonable upper limit.
static constexpr size_t kMaxFlatArraySize = 0x4000;
private:
bool GetServiceIDFromMap(ClientType client_id,
ServiceType* service_id) const {
DCHECK(client_id >= kMaxFlatArraySize);
auto iter = client_to_service_map_.find(client_id);
if (iter != client_to_service_map_.end()) {
if (service_id) {
*service_id = iter->second;
}
return true;
}
return false;
}
ServiceType invalid_service_id_;
std::vector<ServiceType> client_to_service_array_;
std::unordered_map<ClientType, ServiceType> client_to_service_map_;
};
} // namespace gles2
} // namespace gpu
#endif // GPU_COMMAND_BUFFER_SERVICE_CLIENT_SERVICE_MAP_H_