1 /* 2 * Copyright (c) 2023 Shenzhen Kaihong Digital Industry Development 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 OHOS_SHARING_VIDEO_SOURCE_SCREEN_H 17 #define OHOS_SHARING_VIDEO_SOURCE_SCREEN_H 18 19 #include <surface.h> 20 #include <thread> 21 #include "dm_common.h" 22 #include "screen_manager.h" 23 #include "sync_fence.h" 24 #include "timer.h" 25 #include "video_source_encoder.h" 26 namespace OHOS { 27 namespace Sharing { 28 constexpr uint32_t INVALID_SEQ = 0xFFFFFFFF; 29 30 class VideoSourceScreen : public std::enable_shared_from_this<VideoSourceScreen> { 31 public: 32 class ScreenGroupListener : public Rosen::ScreenManager::IScreenGroupListener { 33 public: 34 void OnChange(const std::vector<uint64_t> &screenIds, Rosen::ScreenGroupChangeEvent event) override; 35 }; 36 37 class ScreenBufferConsumerListener : public IBufferConsumerListener { 38 public: ScreenBufferConsumerListener(std::weak_ptr<VideoSourceScreen> parent)39 explicit ScreenBufferConsumerListener(std::weak_ptr<VideoSourceScreen> parent) : parent_(parent) {} 40 void OnBufferAvailable() override; 41 42 public: 43 std::weak_ptr<VideoSourceScreen> parent_; 44 }; 45 VideoSourceScreen(sptr<OHOS::Surface> encoderSurface)46 explicit VideoSourceScreen(sptr<OHOS::Surface> encoderSurface) : encoderSurface_(encoderSurface) {} 47 ~VideoSourceScreen(); 48 49 int32_t ReleaseScreenBuffer() const; 50 int32_t InitScreenSource(const VideoSourceConfigure &configure); 51 52 void StopScreenSourceCapture(); 53 void StartScreenSourceCapture(); 54 55 private: 56 void OnScreenBufferAvailable(); 57 void RemoveScreenFromGroup() const; 58 void FrameRateControlTimerWorker(); 59 60 int32_t DestroyVirtualScreen() const; 61 int32_t RegisterScreenGroupListener(); 62 int32_t UnregisterScreenGroupListener() const; 63 int32_t SetEncoderSurface(sptr<OHOS::Surface> surface); 64 65 uint64_t CreateVirtualScreen(const VideoSourceConfigure &configure); 66 67 private: 68 volatile uint32_t queueSzie_; 69 uint32_t lastEncFrameBufferSeq_ = INVALID_SEQ; 70 71 uint64_t screenId_ = SCREEN_ID_INVALID; 72 uint64_t srcScreenId_ = SCREEN_ID_INVALID; 73 74 std::mutex frameRateCtrlMutex_; 75 std::unique_ptr<std::thread> rateControlWorker_ = nullptr; 76 std::unique_ptr<OHOS::Utils::Timer> timer_ = std::make_unique<OHOS::Utils::Timer>("FrameRateControlTimer"); 77 78 OHOS::sptr<OHOS::Surface> encoderSurface_ = nullptr; 79 OHOS::sptr<OHOS::Surface> consumerSurface_ = nullptr; 80 OHOS::sptr<OHOS::Surface> producerSurface_ = nullptr; 81 OHOS::sptr<OHOS::SurfaceBuffer> lastBuffer_ = nullptr; 82 OHOS::sptr<OHOS::SurfaceBuffer> lastEncFrameBuffer_ = nullptr; 83 OHOS::sptr<ScreenGroupListener> screenGroupListener_ = nullptr; 84 OHOS::sptr<ScreenBufferConsumerListener> screenBufferListener_ = nullptr; 85 }; 86 } // namespace Sharing 87 } // namespace OHOS 88 #endif 89