1 /*
2 * Copyright (c) 2021-2024 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 #include "dbuffer_manager.h"
17 #include <buffer_handle_utils.h>
18 #include "distributed_hardware_log.h"
19
20 namespace OHOS {
21 namespace DistributedHardware {
AcquireBuffer()22 std::shared_ptr<DImageBuffer> DBufferManager::AcquireBuffer()
23 {
24 std::unique_lock<std::mutex> l(lock_);
25
26 if (!idleList_.empty()) {
27 auto it = idleList_.begin();
28 busyList_.splice(busyList_.begin(), idleList_, it);
29 DHLOGD("Acquire buffer success, index = %{public}d", (*it)->GetIndex());
30 return *it;
31 }
32 return nullptr;
33 }
34
AddBuffer(std::shared_ptr<DImageBuffer> & buffer)35 RetCode DBufferManager::AddBuffer(std::shared_ptr<DImageBuffer>& buffer)
36 {
37 std::unique_lock<std::mutex> l(lock_);
38 if (idleList_.size() + busyList_.size() >= BUFFER_QUEUE_SIZE) {
39 DHLOGI("Buffer list is full, cannot add buffer.");
40 return RC_ERROR;
41 }
42 idleList_.emplace_back(buffer);
43
44 return RC_OK;
45 }
46
RemoveBuffer(std::shared_ptr<DImageBuffer> & buffer)47 RetCode DBufferManager::RemoveBuffer(std::shared_ptr<DImageBuffer>& buffer)
48 {
49 std::unique_lock<std::mutex> l(lock_);
50
51 auto it = std::find(busyList_.begin(), busyList_.end(), buffer);
52 if (it == busyList_.end()) {
53 DHLOGE("Busy list is empty, cannot remove buffer.");
54 return RC_ERROR;
55 }
56 busyList_.erase(it);
57
58 return RC_OK;
59 }
60
NotifyStop(bool state)61 void DBufferManager::NotifyStop(bool state)
62 {
63 streamStop_ = state;
64 }
65
SurfaceBufferToDImageBuffer(const OHOS::sptr<OHOS::SurfaceBuffer> & surfaceBuffer,const std::shared_ptr<DImageBuffer> & buffer)66 RetCode DBufferManager::SurfaceBufferToDImageBuffer(const OHOS::sptr<OHOS::SurfaceBuffer> &surfaceBuffer,
67 const std::shared_ptr<DImageBuffer> &buffer)
68 {
69 if (surfaceBuffer == nullptr) {
70 DHLOGE("Convert surface buffer failed, surfaceBuffer is null.");
71 return RC_ERROR;
72 }
73
74 BufferHandle *bufHandle = surfaceBuffer->GetBufferHandle();
75 if (bufHandle == nullptr) {
76 DHLOGE("Convert surface buffer failed, BufferHandle is null.");
77 return RC_ERROR;
78 }
79 if ((bufHandle->size <= 0) || (bufHandle->width <= 0) || (bufHandle->height <= 0) || (bufHandle->usage <= 0)) {
80 DHLOGE("Convert surface buffer failed, BufferHandle is invalid.");
81 return RC_ERROR;
82 }
83
84 CHECK_AND_RETURN_RET_LOG(buffer == nullptr, RC_ERROR, "buffer is nullptr");
85 buffer->SetPhyAddress(bufHandle->phyAddr);
86 buffer->SetFileDescriptor(bufHandle->fd);
87 buffer->SetStride(bufHandle->stride);
88 buffer->SetWidth(bufHandle->width);
89 buffer->SetHeight(bufHandle->height);
90 buffer->SetFormat(PixelFormatToDCameraFormat(static_cast<PixelFormat>(bufHandle->format)));
91 buffer->SetUsage(CameraUsageToGrallocUsage(bufHandle->usage));
92 buffer->SetSize(static_cast<uint32_t>(bufHandle->size));
93 buffer->SetBufferHandle(bufHandle);
94
95 return RC_OK;
96 }
97
CameraUsageToGrallocUsage(const uint64_t cameraUsage)98 uint64_t DBufferManager::CameraUsageToGrallocUsage(const uint64_t cameraUsage)
99 {
100 uint64_t grallocUsage = 0;
101 uint64_t test = 1;
102 const uint32_t BYTE = 8;
103 for (uint32_t i = 0; i < sizeof(cameraUsage) * BYTE; i++) {
104 switch (cameraUsage & (test << i)) {
105 case CAMERA_USAGE_SW_READ_OFTEN:
106 grallocUsage |= BUFFER_USAGE_CPU_READ;
107 break;
108 case CAMERA_USAGE_SW_WRITE_OFTEN:
109 grallocUsage |= BUFFER_USAGE_CPU_WRITE;
110 break;
111 case CAMERA_USAGE_MEM_DMA:
112 grallocUsage |= BUFFER_USAGE_MEM_DMA;
113 break;
114 default:
115 break;
116 }
117 }
118
119 return grallocUsage;
120 }
121
PixelFormatToDCameraFormat(const PixelFormat format)122 uint32_t DBufferManager::PixelFormatToDCameraFormat(const PixelFormat format)
123 {
124 uint32_t cameraFormat = OHOS_CAMERA_FORMAT_INVALID;
125 switch (format) {
126 case PIXEL_FMT_RGBA_8888:
127 cameraFormat = OHOS_CAMERA_FORMAT_RGBA_8888;
128 break;
129 case PIXEL_FMT_YCBCR_420_SP:
130 cameraFormat = OHOS_CAMERA_FORMAT_YCBCR_420_888;
131 break;
132 case PIXEL_FMT_YCRCB_420_SP:
133 cameraFormat = OHOS_CAMERA_FORMAT_YCRCB_420_SP;
134 break;
135 default:
136 cameraFormat = OHOS_CAMERA_FORMAT_INVALID;
137 break;
138 }
139
140 return cameraFormat;
141 }
142
DImageBufferToDCameraBuffer(const std::shared_ptr<DImageBuffer> & imageBuffer,DCameraBuffer & buffer)143 RetCode DBufferManager::DImageBufferToDCameraBuffer(const std::shared_ptr<DImageBuffer> &imageBuffer,
144 DCameraBuffer &buffer)
145 {
146 CHECK_AND_RETURN_RET_LOG(imageBuffer == nullptr, RC_ERROR, "imageBuffer is nullptr");
147 BufferHandle *bufHandle = imageBuffer->GetBufferHandle();
148 if (bufHandle == nullptr) {
149 DHLOGE("Convert image surface buffer failed, BufferHandle is null.");
150 return RC_ERROR;
151 }
152 if ((bufHandle->size <= 0) || (bufHandle->width <= 0) || (bufHandle->height <= 0) || (bufHandle->usage <= 0)) {
153 DHLOGE("Convert image surface buffer failed, BufferHandle is invalid.");
154 return RC_ERROR;
155 }
156 buffer.index_ = imageBuffer->GetIndex();
157 buffer.size_ = imageBuffer->GetSize();
158 buffer.bufferHandle_ = sptr<NativeBuffer>(new NativeBuffer());
159 buffer.bufferHandle_->SetBufferHandle(bufHandle);
160 return RC_OK;
161 }
162 } // namespace DistributedHardware
163 } // namespace OHOS
164