1 /* 2 * Copyright (c) 2024 Huawei Device Co., Ltd. 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 16 #ifndef RENDER_SERVICE_BASE_PIPELINE_RS_SURFACE_BUFFER_CALLBACK_MANAGER_H 17 #define RENDER_SERVICE_BASE_PIPELINE_RS_SURFACE_BUFFER_CALLBACK_MANAGER_H 18 19 #include <functional> 20 #include <map> 21 #include <mutex> 22 #include <shared_mutex> 23 #include <utility> 24 #include <vector> 25 26 #include "refbase.h" 27 28 namespace OHOS { 29 namespace Rosen { 30 31 class RSISurfaceBufferCallback; 32 33 class RSB_EXPORT RSSurfaceBufferCallbackManager { 34 public: 35 void RegisterSurfaceBufferCallback(pid_t pid, uint64_t uid, 36 sptr<RSISurfaceBufferCallback> callback); 37 void UnregisterSurfaceBufferCallback(pid_t pid); 38 void UnregisterSurfaceBufferCallback(pid_t pid, uint64_t uid); 39 40 std::function<void(pid_t, uint64_t, uint32_t)> GetSurfaceBufferOpItemCallback() const; 41 void SetRunPolicy(std::function<void(std::function<void()>)> runPolicy); 42 43 static RSSurfaceBufferCallbackManager& Instance(); 44 private: 45 RSSurfaceBufferCallbackManager() = default; 46 ~RSSurfaceBufferCallbackManager() noexcept = default; 47 48 RSSurfaceBufferCallbackManager(const RSSurfaceBufferCallbackManager&) = delete; 49 RSSurfaceBufferCallbackManager(RSSurfaceBufferCallbackManager&&) = delete; 50 RSSurfaceBufferCallbackManager& operator=(const RSSurfaceBufferCallbackManager&) = delete; 51 RSSurfaceBufferCallbackManager& operator=(RSSurfaceBufferCallbackManager&&) = delete; 52 53 sptr<RSISurfaceBufferCallback> GetSurfaceBufferCallback(pid_t pid, uint64_t uid) const; 54 size_t GetSurfaceBufferCallbackSize() const; 55 56 void EnqueueSurfaceBufferId(pid_t pid, uint64_t uid, uint32_t surfaceBufferId); 57 void OnSurfaceBufferOpItemDestruct(pid_t pid, uint64_t uid, uint32_t surfaceBufferId); 58 void RunSurfaceBufferCallback(); 59 60 std::map<std::pair<pid_t, uint64_t>, sptr<RSISurfaceBufferCallback>> 61 surfaceBufferCallbacks_; 62 std::map<std::pair<pid_t, uint64_t>, std::vector<uint32_t>> stagingSurfaceBufferIds_; 63 mutable std::shared_mutex registerSurfaceBufferCallbackMutex_; 64 std::mutex surfaceBufferOpItemMutex_; 65 std::function<void(std::function<void()>)> runPolicy_ = [](auto task) { 66 std::invoke(task); 67 }; 68 69 friend class RSDrawFrame; 70 friend class RSRenderThread; 71 }; 72 } // namespace Rosen 73 } // namespace OHOS 74 75 #endif // RENDER_SERVICE_BASE_PIPELINE_RS_SURFACE_BUFFER_CALLBACK_MANAGER_H 76