1 /*
2 * Copyright (c) 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 #include "stream_customer.h"
17 #include "video_key_info.h"
18
StreamCustomer()19 StreamCustomer::StreamCustomer() {}
~StreamCustomer()20 StreamCustomer::~StreamCustomer() {}
21
CamFrame(const std::function<void (const unsigned char *,uint32_t)> callback)22 void StreamCustomer::CamFrame(const std::function<void(const unsigned char *, uint32_t)> callback)
23 {
24 CAMERA_LOGD("test:enter CamFrame thread ++ ");
25 OHOS::Rect damage;
26 int32_t flushFence = 0;
27 int64_t timestamp = 0;
28 constexpr uint32_t delayTime = 12000;
29
30 do {
31 OHOS::sptr<OHOS::SurfaceBuffer> buff = nullptr;
32 consumer_->AcquireBuffer(buff, flushFence, timestamp, damage);
33 if (buff != nullptr) {
34 void* addr = buff->GetVirAddr();
35 int32_t gotSize = 0;
36 buff->GetExtraData()->ExtraGet(OHOS::Camera::dataSize, gotSize);
37 CAMERA_LOGD("OHOS::Camera::dataSize: %{public}d", gotSize);
38 uint32_t size = gotSize > 0 ? gotSize : buff->GetSize();
39 if (callback != nullptr) {
40 callback(static_cast<const unsigned char*>(addr), size);
41 }
42 auto ret = consumer_->ReleaseBuffer(buff, flushFence);
43 if (!ret) {
44 CAMERA_LOGE("ReleaseBuffer fail: %{public}d", ret);
45 }
46 }
47 usleep(delayTime);
48 } while (camFrameExit_ == 0);
49
50 CAMERA_LOGD("test:Exiting CamFrame thread -- ");
51 }
52
CreateProducer()53 OHOS::sptr<OHOS::IBufferProducer> StreamCustomer::CreateProducer()
54 {
55 consumer_ = OHOS::IConsumerSurface::Create();
56 if (consumer_ == nullptr) {
57 return nullptr;
58 }
59 OHOS::sptr<OHOS::IBufferConsumerListener> listener = new TestBuffersConsumerListener();
60 CHECK_IF_PTR_NULL_RETURN_VALUE(listener, nullptr);
61 consumer_->RegisterConsumerListener(listener);
62
63 auto producer = consumer_->GetProducer();
64 if (producer == nullptr) {
65 return nullptr;
66 }
67
68 CAMERA_LOGI("test, create a buffer queue producer");
69 return producer;
70 }
71
ReceiveFrameOn(const std::function<void (const unsigned char *,uint32_t)> callback)72 OHOS::Camera::RetCode StreamCustomer::ReceiveFrameOn(
73 const std::function<void(const unsigned char *, uint32_t)> callback)
74 {
75 CAMERA_LOGD("test:ReceiveFrameOn enter");
76
77 if (camFrameExit_ == 1) {
78 camFrameExit_ = 0;
79 previewThreadId_ = new (std::nothrow) std::thread(&StreamCustomer::CamFrame, this, callback);
80 if (previewThreadId_ == nullptr) {
81 CAMERA_LOGE("test:ReceiveFrameOn failed");
82 return OHOS::Camera::RC_ERROR;
83 }
84 } else {
85 CAMERA_LOGI("test:ReceiveFrameOn loop thread is running");
86 }
87 CAMERA_LOGD("test:ReceiveFrameOn exit");
88
89 return OHOS::Camera::RC_OK;
90 }
91
ReceiveFrameOff()92 void StreamCustomer::ReceiveFrameOff()
93 {
94 CAMERA_LOGD("test:ReceiveFrameOff enter");
95
96 if (camFrameExit_ == 0) {
97 camFrameExit_ = 1;
98 if (previewThreadId_ != nullptr) {
99 previewThreadId_->join();
100 delete previewThreadId_;
101 previewThreadId_ = nullptr;
102 }
103 }
104
105 CAMERA_LOGD("test:ReceiveFrameOff exit");
106 }
107