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