• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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