1 /* 2 * Copyright (c) 2021-2025 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 HISTREAMER_FOUNDATION_AVBUFFER_QUEUE_IMPL_H 17 #define HISTREAMER_FOUNDATION_AVBUFFER_QUEUE_IMPL_H 18 19 #include <list> 20 #include <map> 21 #include <string> 22 #include <mutex> 23 #include <condition_variable> 24 #include "buffer/avbuffer_queue.h" 25 26 namespace OHOS { 27 namespace Media { 28 29 using AVBufferState = enum AVBufferState { 30 AVBUFFER_STATE_RELEASED, 31 AVBUFFER_STATE_REQUESTED, 32 AVBUFFER_STATE_PUSHED, 33 AVBUFFER_STATE_RETURNED, 34 AVBUFFER_STATE_ACQUIRED, 35 AVBUFFER_STATE_ATTACHED, 36 }; 37 38 using AVBufferElement = struct AVBufferElement { 39 AVBufferConfig config; 40 AVBufferState state; 41 bool isDeleting; 42 std::shared_ptr<AVBuffer> buffer; 43 }; 44 45 class AVBufferQueueProducerImpl; 46 class AVBufferQueueConsumerImpl; 47 48 // 当前调试版本，错误码统一用int32_t表示，0表示返回正确，非0表示返回错误。 49 class AVBufferQueueImpl : public AVBufferQueue, public std::enable_shared_from_this<AVBufferQueueImpl> { 50 public: 51 explicit AVBufferQueueImpl(const std::string &name); 52 AVBufferQueueImpl(uint32_t size, MemoryType type, const std::string &name, bool disableAlloc = false); 53 ~AVBufferQueueImpl() override = default; 54 AVBufferQueueImpl(const AVBufferQueueImpl&) = delete; 55 AVBufferQueueImpl operator=(const AVBufferQueueImpl&) = delete; 56 57 std::shared_ptr<AVBufferQueueProducer> GetLocalProducer() override; 58 std::shared_ptr<AVBufferQueueConsumer> GetLocalConsumer() override; 59 60 sptr<AVBufferQueueProducer> GetProducer() override; 61 sptr<AVBufferQueueConsumer> GetConsumer() override; 62 GetSurfaceAsProducer()63 inline sptr<Surface> GetSurfaceAsProducer() override { return nullptr; } GetSurfaceAsConsumer()64 inline sptr<Surface> GetSurfaceAsConsumer() override { return nullptr; } 65 66 uint32_t GetQueueSize() override; 67 Status SetQueueSize(uint32_t size) override; 68 virtual Status SetLargerQueueSize(uint32_t size) override; 69 bool IsBufferInQueue(const std::shared_ptr<AVBuffer>& buffer) override; 70 Status Clear() override; 71 Status ClearBufferIf(std::function<bool(const std::shared_ptr<AVBuffer>&)> pred) override; 72 73 virtual Status RequestBuffer(std::shared_ptr<AVBuffer>& buffer, 74 const AVBufferConfig& config, int32_t timeoutMs); 75 virtual Status PushBuffer(uint64_t uniqueId, bool available); 76 virtual Status PushBuffer(const std::shared_ptr<AVBuffer>& buffer, bool available); 77 virtual Status ReturnBuffer(uint64_t uniqueId, bool available); 78 virtual Status ReturnBuffer(const std::shared_ptr<AVBuffer>& buffer, bool available); 79 80 virtual Status AttachBuffer(std::shared_ptr<AVBuffer>& buffer, bool isFilled); 81 virtual Status DetachBuffer(uint64_t uniqueId); 82 virtual Status DetachBuffer(const std::shared_ptr<AVBuffer>& buffer); 83 84 virtual Status AcquireBuffer(std::shared_ptr<AVBuffer>& buffer); 85 virtual Status ReleaseBuffer(const std::shared_ptr<AVBuffer>& buffer); 86 87 virtual Status SetBrokerListener(sptr<IBrokerListener>& listener); 88 virtual Status RemoveBrokerListener(sptr<IBrokerListener>& listener); 89 virtual Status SetProducerListener(sptr<IProducerListener>& listener); 90 virtual Status SetConsumerListener(sptr<IConsumerListener>& listener); 91 92 Status SetQueueSizeAndAttachBuffer(uint32_t size, std::shared_ptr<AVBuffer>& buffer, bool isFilled) override; 93 94 uint32_t GetFilledBufferSize() override; 95 96 protected: 97 std::string name_; 98 99 std::mutex producerCreatorMutex_; 100 std::mutex consumerCreatorMutex_; 101 102 std::vector<sptr<IBrokerListener>> brokerListeners_; 103 sptr<IProducerListener> producerListener_; 104 sptr<IConsumerListener> consumerListener_; 105 106 std::mutex producerListenerMutex_; 107 std::mutex consumerListenerMutex_; 108 std::mutex brokerListenerMutex_; 109 110 std::mutex queueMutex_; 111 112 std::weak_ptr<AVBufferQueueProducerImpl> localProducer_; 113 std::weak_ptr<AVBufferQueueConsumerImpl> localConsumer_; 114 115 wptr<AVBufferQueueProducerImpl> producer_; 116 wptr<AVBufferQueueConsumerImpl> consumer_; 117 118 private: 119 Status AttachAvailableBufferLocked(std::shared_ptr<AVBuffer>& buffer); 120 Status PushBufferOnFilled(uint64_t uniqueId, bool isFilled); 121 void SetQueueSizeBeforeAttachBufferLocked(uint32_t size); 122 uint32_t size_; 123 MemoryType memoryType_; 124 bool disableAlloc_; 125 126 std::map<uint64_t, AVBufferElement> cachedBufferMap_; 127 128 std::list<uint64_t> freeBufferList_; // 记录已分配的且处于空闲状态的buffer uniqueId，按bufferSize升序排列 129 std::list<uint64_t> dirtyBufferList_; 130 131 std::condition_variable requestCondition; 132 133 Status CheckConfig(const AVBufferConfig& config); 134 135 bool wait_for(std::unique_lock<std::mutex>& lock, int32_t timeoutMs); 136 137 uint32_t GetCachedBufferCount() const; 138 Status RequestReuseBuffer(std::shared_ptr<AVBuffer>& buffer, const AVBufferConfig& config); 139 void InsertFreeBufferInOrder(uint64_t uniqueId); 140 Status CancelBuffer(uint64_t uniqueId); 141 Status DetachBuffer(uint64_t uniqueId, bool force); 142 Status ReleaseBuffer(uint64_t uniqueId); 143 Status PopFromFreeBufferList(std::shared_ptr<AVBuffer>& buffer, const AVBufferConfig& config); 144 Status PopFromDirtyBufferList(std::shared_ptr<AVBuffer>& buffer); 145 Status AllocBuffer(std::shared_ptr<AVBuffer>& buffer, const AVBufferConfig& config); 146 147 void DeleteBuffers(uint32_t count); 148 void DeleteCachedBufferById(uint64_t uniqueId_); 149 }; 150 151 } // namespace Media 152 } // namespace OHOS 153 154 #endif // HISTREAMER_FOUNDATION_AVBUFFER_QUEUE_IMPL_H 155