1 /*
2 * Copyright (c) 2021-2021 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 #if !defined(OHOS_LITE) && defined(VIDEO_SUPPORT)
17 #include "common/surface_memory.h"
18 #include "plugin/plugin_memory.h"
19 #include <utility>
20 #include "inner_api/common/log.h"
21 #include "surface_allocator.h"
22 #include "cpp_ext/type_cast_ext.h"
23 #include "osal/task/autolock.h"
24
25 namespace {
26 constexpr OHOS::HiviewDFX::HiLogLabel LABEL = { LOG_CORE, LOG_DOMAIN_FOUNDATION, "SurfaceMemory" };
27 }
28
29 namespace OHOS {
30 namespace Media {
31 namespace Plugins {
SurfaceMemory(size_t capacity,std::shared_ptr<Allocator> allocator,size_t align)32 SurfaceMemory::SurfaceMemory(size_t capacity, std::shared_ptr<Allocator> allocator, size_t align)
33 : Memory(capacity, std::move(allocator), align, MemoryType::SURFACE_MEMORY, false),
34 fence_(-1),
35 stride_(0)
36 {
37 MEDIA_LOG_DD("SurfaceMemory ctor.");
38 if (this->allocator != nullptr && this->allocator->GetMemoryType() == MemoryType::SURFACE_MEMORY) {
39 surfaceAllocator_ = ReinterpretPointerCast<SurfaceAllocator>(this->allocator);
40 AllocSurfaceBuffer();
41 }
42 }
43
SurfaceMemory(sptr<SurfaceBuffer> surfaceBuffer,int32_t surfaceCapacity)44 SurfaceMemory::SurfaceMemory(sptr<SurfaceBuffer> surfaceBuffer, int32_t surfaceCapacity)
45 : Memory(surfaceCapacity, nullptr, 1, MemoryType::SURFACE_MEMORY, false), // align 1
46 surfaceBuffer_(surfaceBuffer), fence_(-1), stride_(0)
47 {
48 }
49
~SurfaceMemory()50 SurfaceMemory::~SurfaceMemory()
51 {
52 MEDIA_LOG_DD("SurfaceMemory dtor.");
53 ReleaseSurfaceBuffer();
54 }
55
AllocSurfaceBuffer()56 void SurfaceMemory::AllocSurfaceBuffer()
57 {
58 if (surfaceAllocator_ == nullptr || surfaceBuffer_ != nullptr) {
59 MEDIA_LOG_E("No need to allocate surface buffer.");
60 return;
61 }
62 surfaceBuffer_ = surfaceAllocator_->AllocSurfaceBuffer();
63 if (surfaceBuffer_ != nullptr) {
64 auto bufferHandle = surfaceBuffer_->GetBufferHandle();
65 if (bufferHandle != nullptr) {
66 stride_ = bufferHandle->stride;
67 }
68 fence_ = -1;
69 } else {
70 // Surface often obtain buffer failed, but doesn't cause any problem.
71 MEDIA_LOG_DD("AllocSurfaceBuffer failed.");
72 }
73 }
74
GetSurfaceBuffer()75 sptr<SurfaceBuffer> SurfaceMemory::GetSurfaceBuffer()
76 {
77 OHOS::Media::AutoLock l(memMutex_);
78 if (!surfaceBuffer_ || needRender_) {
79 // request surface buffer again when old buffer flush to nullptr
80 surfaceBuffer_ = nullptr;
81 AllocSurfaceBuffer();
82 needRender_ = false;
83 }
84 return surfaceBuffer_;
85 }
86
ReleaseSurfaceBuffer()87 void SurfaceMemory::ReleaseSurfaceBuffer()
88 {
89 OHOS::Media::AutoLock l(memMutex_);
90 if (surfaceBuffer_ != nullptr && surfaceAllocator_) {
91 surfaceAllocator_->ReleaseSurfaceBuffer(surfaceBuffer_, needRender_);
92 }
93 }
94
GetFlushFence()95 int32_t SurfaceMemory::GetFlushFence()
96 {
97 OHOS::Media::AutoLock l(memMutex_);
98 return fence_;
99 }
100
GetBufferHandle()101 BufferHandle *SurfaceMemory::GetBufferHandle()
102 {
103 OHOS::Media::AutoLock l(memMutex_);
104 if (surfaceBuffer_) {
105 return surfaceBuffer_->GetBufferHandle();
106 }
107 return nullptr;
108 }
109
SetNeedRender(bool needRender)110 void SurfaceMemory::SetNeedRender(bool needRender)
111 {
112 OHOS::Media::AutoLock l(memMutex_);
113 needRender_ = needRender;
114 }
115
GetSurfaceBufferStride()116 uint32_t SurfaceMemory::GetSurfaceBufferStride()
117 {
118 OHOS::Media::AutoLock l(memMutex_);
119 return stride_;
120 }
121
GetRealAddr() const122 uint8_t* SurfaceMemory::GetRealAddr() const
123 {
124 OHOS::Media::AutoLock l(memMutex_);
125 if (surfaceBuffer_) {
126 return static_cast<uint8_t *>(surfaceBuffer_->GetVirAddr());
127 }
128 return nullptr;
129 }
130 } // namespace Plugins
131 } // namespace Media
132 } // namespace OHOS
133 #endif