1 // Copyright 2019 The Android Open Source Project
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 #include "ExternalObjectManager.h"
15
16 #include <utility>
17
18 using android::base::ManagedDescriptor;
19
20 namespace gfxstream {
21
sMapping()22 static ExternalObjectManager* sMapping() {
23 static ExternalObjectManager* s = new ExternalObjectManager;
24 return s;
25 }
26
27 // static
get()28 ExternalObjectManager* ExternalObjectManager::get() { return sMapping(); }
29
addMapping(uint32_t ctxId,uint64_t blobId,void * addr,uint32_t caching)30 void ExternalObjectManager::addMapping(uint32_t ctxId, uint64_t blobId, void* addr,
31 uint32_t caching) {
32 struct HostMemInfo info = {
33 .addr = addr,
34 .caching = caching,
35 };
36
37 auto key = std::make_pair(ctxId, blobId);
38 std::lock_guard<std::mutex> lock(mLock);
39 mHostMemInfos.insert(std::make_pair(key, info));
40 }
41
removeMapping(uint32_t ctxId,uint64_t blobId)42 std::optional<HostMemInfo> ExternalObjectManager::removeMapping(uint32_t ctxId, uint64_t blobId) {
43 auto key = std::make_pair(ctxId, blobId);
44 std::lock_guard<std::mutex> lock(mLock);
45 auto found = mHostMemInfos.find(key);
46 if (found != mHostMemInfos.end()) {
47 std::optional<HostMemInfo> ret = found->second;
48 mHostMemInfos.erase(found);
49 return ret;
50 }
51
52 return std::nullopt;
53 }
54
addBlobDescriptorInfo(uint32_t ctxId,uint64_t blobId,ManagedDescriptor descriptor,uint32_t handleType,uint32_t caching,std::optional<VulkanInfo> vulkanInfoOpt)55 void ExternalObjectManager::addBlobDescriptorInfo(uint32_t ctxId, uint64_t blobId,
56 ManagedDescriptor descriptor, uint32_t handleType,
57 uint32_t caching,
58 std::optional<VulkanInfo> vulkanInfoOpt) {
59 struct BlobDescriptorInfo info = {
60 .descriptor = std::move(descriptor),
61 .handleType = handleType,
62 .caching = caching,
63 .vulkanInfoOpt = vulkanInfoOpt,
64 };
65
66 auto key = std::make_pair(ctxId, blobId);
67 std::lock_guard<std::mutex> lock(mLock);
68 mBlobDescriptorInfos.insert(std::make_pair(key, std::move(info)));
69 }
70
removeBlobDescriptorInfo(uint32_t ctxId,uint64_t blobId)71 std::optional<BlobDescriptorInfo> ExternalObjectManager::removeBlobDescriptorInfo(uint32_t ctxId,
72 uint64_t blobId) {
73 auto key = std::make_pair(ctxId, blobId);
74 std::lock_guard<std::mutex> lock(mLock);
75 auto found = mBlobDescriptorInfos.find(key);
76 if (found != mBlobDescriptorInfos.end()) {
77 std::optional<BlobDescriptorInfo> ret = std::move(found->second);
78 mBlobDescriptorInfos.erase(found);
79 return ret;
80 }
81
82 return std::nullopt;
83 }
84
addSyncDescriptorInfo(uint32_t ctxId,uint64_t syncId,ManagedDescriptor descriptor,uint32_t handleType)85 void ExternalObjectManager::addSyncDescriptorInfo(uint32_t ctxId, uint64_t syncId,
86 ManagedDescriptor descriptor,
87 uint32_t handleType) {
88 struct SyncDescriptorInfo info = {
89 .descriptor = std::move(descriptor),
90 .handleType = handleType,
91 };
92
93 auto key = std::make_pair(ctxId, syncId);
94 std::lock_guard<std::mutex> lock(mLock);
95 mSyncDescriptorInfos.insert(std::make_pair(key, std::move(info)));
96 }
97
removeSyncDescriptorInfo(uint32_t ctxId,uint64_t syncId)98 std::optional<SyncDescriptorInfo> ExternalObjectManager::removeSyncDescriptorInfo(uint32_t ctxId,
99 uint64_t syncId) {
100 auto key = std::make_pair(ctxId, syncId);
101 std::lock_guard<std::mutex> lock(mLock);
102 auto found = mSyncDescriptorInfos.find(key);
103 if (found != mSyncDescriptorInfos.end()) {
104 std::optional<SyncDescriptorInfo> ret = std::move(found->second);
105 mSyncDescriptorInfos.erase(found);
106 return ret;
107 }
108
109 return std::nullopt;
110 }
111
112 } // namespace gfxstream
113