• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2023 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file expected 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 "hdi_common_v1_2.h"
17 #include "camera.h"
18 #include "video_key_info.h"
19 
20 namespace OHOS::Camera {
21 Test::ResultCallback Test::resultCallback_ = 0;
22 OHOS::HDI::Camera::V1_0::FlashlightStatus Test::statusCallback =
23                 static_cast<OHOS::HDI::Camera::V1_0::FlashlightStatus>(0);
GetCurrentLocalTimeStamp()24 uint64_t Test::GetCurrentLocalTimeStamp()
25 {
26     std::chrono::time_point<std::chrono::system_clock, std::chrono::milliseconds> tp =
27         std::chrono::time_point_cast<std::chrono::milliseconds>(std::chrono::system_clock::now());
28     auto tmp = std::chrono::duration_cast<std::chrono::milliseconds>(tp.time_since_epoch());
29     return static_cast<uint64_t>(tmp.count());
30 }
31 
DumpImageFile(int streamId,std::string suffix,const void * buffer,int32_t size)32 int32_t Test::DumpImageFile(int streamId, std::string suffix, const void* buffer, int32_t size)
33 {
34     if (imageDataSaveSwitch == SWITCH_OFF) {
35         return 0;
36     }
37     if (streamId < 0) {
38         CAMERA_LOGE("ivalid stream id: %{public}d", streamId);
39         return -1;
40     }
41     char mkdirCmd[PATH_MAX] = {0};
42     char path[PATH_MAX] = {0};
43     int ret = sprintf_s(mkdirCmd, sizeof(mkdirCmd) / sizeof(mkdirCmd[0]),
44         "mkdir -p /data/stream-%d", streamId);
45     if (ret < 0) {
46         return -1;
47     }
48     system(mkdirCmd);
49     ret = sprintf_s(path, sizeof(path) / sizeof(path[0]), "/data/stream-%d/%lld.%s",
50         streamId, GetCurrentLocalTimeStamp(), suffix.c_str());
51     if (ret < 0) {
52         return -1;
53     }
54 
55     int imgFd = open(path, O_RDWR | O_CREAT, 00766);
56     if (imgFd == -1) {
57         CAMERA_LOGE("open file failed, errno: %{public}s", strerror(errno));
58         return -1;
59     }
60 
61     ret = write(imgFd, buffer, size);
62     if (ret == -1) {
63         CAMERA_LOGE("write file failed, error: %{public}s", strerror(errno));
64         close(imgFd);
65         return -1;
66     }
67     close(imgFd);
68     return 0;
69 }
70 
Init()71 void Test::Init()
72 {
73     uint32_t mainVer;
74     uint32_t minVer;
75     int32_t ret;
76     if (serviceV1_2 == nullptr) {
77         serviceV1_2 = OHOS::HDI::Camera::V1_2::ICameraHost::Get("camera_service", false);
78         EXPECT_NE(serviceV1_2, nullptr);
79         CAMERA_LOGI("V1_2::ICameraHost get success");
80         ret = serviceV1_2->GetVersion(mainVer, minVer);
81         EXPECT_EQ(ret, 0);
82         CAMERA_LOGI("V1_2::ICameraHost get version success, %{public}d, %{public}d", mainVer, minVer);
83         service = static_cast<OHOS::HDI::Camera::V1_0::ICameraHost *>(serviceV1_2.GetRefPtr());
84     }
85 
86     hostCallback = new TestCameraHostCallback();
87     ret = service->SetCallback(hostCallback);
88     EXPECT_EQ(ret, 0);
89 }
90 
DefferredImageTestInit()91 int32_t Test::DefferredImageTestInit()
92 {
93     constexpr const char* serviceName = "camera_image_process_service";
94     constexpr const int userId = 100;
95     int ret = 0;
96 
97     // get ImageProcessService
98     imageProcessService_ = OHOS::HDI::Camera::V1_2::ImageProcessServiceProxy::Get(serviceName, false);
99     if (imageProcessService_ == nullptr) {
100         CAMERA_LOGE("ImageProcessServiceProxy::Get Fail, imageProcessService is nullptr");
101         printf("ImageProcessServiceProxy::Get Fail, imageProcessService is nullptr");
102         return -1;
103     }
104     imageProcessCallback_ = new OHOS::Camera::Test::TestImageProcessCallback();
105     EXPECT_NE(imageProcessCallback_, nullptr);
106     ret = imageProcessService_->CreateImageProcessSession(userId, imageProcessCallback_, imageProcessSession_);
107     EXPECT_EQ(ret, 0);
108     if (imageProcessSession_ == nullptr) {
109         CAMERA_LOGE("CreateImageProcessSession Fail, imageProcessSession is nullptr: %{public}d", ret);
110         printf("CreateImageProcessSession Fail, imageProcessSession is nullptr: %d\r\n", ret);
111         return -1;
112     }
113     return 0;
114 }
115 
Open(int cameraId)116 void Test::Open(int cameraId)
117 {
118     if (cameraDevice == nullptr) {
119         EXPECT_NE(service, nullptr);
120         service->GetCameraIds(cameraIds);
121         EXPECT_NE(cameraIds.size(), 0);
122         GetCameraMetadata(cameraId);
123         deviceCallback = new OHOS::Camera::Test::DemoCameraDeviceCallback();
124 
125         EXPECT_NE(serviceV1_2, nullptr);
126         if (DEVICE_1 == cameraId) {
127             rc = serviceV1_2->OpenCamera_V1_1(cameraIds[1], deviceCallback, cameraDeviceV1_1);
128         } else {
129             rc = serviceV1_2->OpenCamera_V1_1(cameraIds[0], deviceCallback, cameraDeviceV1_1);
130         }
131         EXPECT_EQ(rc, HDI::Camera::V1_0::NO_ERROR);
132         EXPECT_NE(cameraDeviceV1_1, nullptr);
133         cameraDevice = static_cast<OHOS::HDI::Camera::V1_0::ICameraDevice *>(cameraDeviceV1_1.GetRefPtr());
134         CAMERA_LOGI("OpenCamera V1_2 success");
135     }
136 }
137 
OpenCameraV1_2(int cameraId)138 void Test::OpenCameraV1_2(int cameraId)
139 {
140     if (cameraDevice == nullptr) {
141         EXPECT_NE(service, nullptr);
142         service->GetCameraIds(cameraIds);
143         EXPECT_NE(cameraIds.size(), 0);
144         GetCameraMetadata(cameraId);
145         deviceCallback = new OHOS::Camera::Test::DemoCameraDeviceCallback();
146 
147         EXPECT_NE(serviceV1_2, nullptr);
148         if (cameraId == DEVICE_1) {
149             rc = serviceV1_2->OpenCamera_V1_2(cameraIds[1], deviceCallback, cameraDeviceV1_2);
150         } else {
151             rc = serviceV1_2->OpenCamera_V1_2(cameraIds[0], deviceCallback, cameraDeviceV1_2);
152         }
153         EXPECT_EQ(rc, HDI::Camera::V1_0::NO_ERROR);
154         EXPECT_NE(cameraDeviceV1_2, nullptr);
155         cameraDevice = static_cast<OHOS::HDI::Camera::V1_0::ICameraDevice *>(cameraDeviceV1_2.GetRefPtr());
156         CAMERA_LOGI("OpenCamera V1_2 success");
157     }
158 }
159 
GetCameraMetadata(int cameraId)160 void Test::GetCameraMetadata(int cameraId)
161 {
162     if (DEVICE_1 == cameraId) {
163         rc = service->GetCameraAbility(cameraIds[1], abilityVec);
164     } else {
165         rc = service->GetCameraAbility(cameraIds[0], abilityVec);
166     }
167     if (rc != HDI::Camera::V1_0::NO_ERROR) {
168         CAMERA_LOGE("GetCameraAbility failed, rc = %{public}d", rc);
169     }
170     MetadataUtils::ConvertVecToMetadata(abilityVec, ability);
171     EXPECT_NE(ability, nullptr);
172 }
173 
Close()174 void Test::Close()
175 {
176     if (cameraDevice != nullptr) {
177         cameraDevice->Close();
178         cameraDevice = nullptr;
179     }
180 }
181 
DefaultPreview(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)182 void Test::DefaultPreview(
183     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
184 {
185     infos->v1_0.streamId_ = streamIdPreview;
186     infos->v1_0.width_ = previewWidth;
187     infos->v1_0.height_ = previewHeight;
188     infos->v1_0.format_ = previewFormat;
189     infos->v1_0.dataspace_ = UT_DATA_SIZE;
190     infos->v1_0.intent_ = StreamIntent::PREVIEW;
191     infos->v1_0.tunneledMode_ = UT_TUNNEL_MODE;
192 }
193 
DefaultCapture(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)194 void Test::DefaultCapture(
195     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
196 {
197     infos->v1_0.streamId_ = streamIdCapture;
198     infos->v1_0.width_ = captureWidth;
199     infos->v1_0.height_ = captureHeight;
200     infos->v1_0.format_ = snapshotFormat;
201     infos->v1_0.dataspace_ = UT_DATA_SIZE;
202     infos->v1_0.intent_ = StreamIntent::STILL_CAPTURE;
203     infos->v1_0.tunneledMode_ = UT_TUNNEL_MODE;
204 }
205 
DefaultSketch(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)206 void Test::DefaultSketch(
207     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
208 {
209     infos->v1_0.streamId_ = streamIdSketch;
210     infos->v1_0.width_ = sketchWidth;
211     infos->v1_0.height_ = sketchHeight;
212     infos->v1_0.format_ = previewFormat;
213     infos->v1_0.dataspace_ = UT_DATA_SIZE;
214     infos->v1_0.intent_ = StreamIntent::PREVIEW;
215     infos->v1_0.tunneledMode_ = UT_TUNNEL_MODE;
216 }
217 
DefaultInfosSketch(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)218 void Test::DefaultInfosSketch(
219     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
220 {
221     DefaultSketch(infos);
222     std::shared_ptr<StreamConsumer> consumer_pre = std::make_shared<StreamConsumer>();
223     infos->v1_0.bufferQueue_ = consumer_pre->CreateProducerSeq([this](void* addr, uint32_t size) {
224         DumpImageFile(streamIdSketch, "yuv", addr, size);
225     });
226     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
227     consumerMap_[StreamIntent::PREVIEW] = consumer_pre;
228 }
229 
DefaultInfosPreviewV1_2(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)230 void Test::DefaultInfosPreviewV1_2(
231     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
232 {
233     DefaultPreview(infos);
234     std::shared_ptr<StreamConsumer> consumer_pre = std::make_shared<StreamConsumer>();
235     infos->v1_0.bufferQueue_ = consumer_pre->CreateProducerSeq([this](void* addr, uint32_t size) {
236         DumpImageFile(streamIdPreview, "yuv", addr, size);
237     });
238     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
239 }
240 
DefaultInfosPreview(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)241 void Test::DefaultInfosPreview(
242     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
243 {
244     DefaultPreview(infos);
245     std::shared_ptr<StreamConsumer> consumer_pre = std::make_shared<StreamConsumer>();
246     infos->v1_0.bufferQueue_ = consumer_pre->CreateProducerSeq([this](void* addr, uint32_t size) {
247         DumpImageFile(streamIdPreview, "yuv", addr, size);
248     });
249     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
250     consumerMap_[StreamIntent::PREVIEW] = consumer_pre;
251 }
252 
DefaultInfosCapture(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)253 void Test::DefaultInfosCapture(
254     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
255 {
256     DefaultCapture(infos);
257     std::shared_ptr<StreamConsumer> consumer_capture = std::make_shared<StreamConsumer>();
258     infos->v1_0.bufferQueue_ = consumer_capture->CreateProducerSeq([this](void* addr, uint32_t size) {
259         DumpImageFile(streamIdCapture, "jpeg", addr, size);
260     });
261     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
262     consumerMap_[StreamIntent::STILL_CAPTURE] = consumer_capture;
263 }
264 
DefaultInfosVideo(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)265 void Test::DefaultInfosVideo(
266     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
267 {
268     infos->v1_0.streamId_ = streamIdVideo;
269     infos->v1_0.width_ = videoWidth;
270     infos->v1_0.height_ = videoHeight;
271     infos->v1_0.format_ = videoFormat;
272     infos->v1_0.dataspace_ = UT_DATA_SIZE;
273     infos->v1_0.intent_ = StreamIntent::VIDEO;
274     infos->v1_0.encodeType_ = ENCODE_TYPE_H265;
275     infos->v1_0.tunneledMode_ = UT_TUNNEL_MODE;
276     std::shared_ptr<StreamConsumer> consumer_video = std::make_shared<StreamConsumer>();
277     infos->v1_0.bufferQueue_ = consumer_video->CreateProducerSeq([this](void* addr, uint32_t size) {
278         DumpImageFile(streamIdVideo, "yuv", addr, size);
279     });
280     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
281     consumerMap_[StreamIntent::VIDEO] = consumer_video;
282 }
283 
DefaultInfosAnalyze(std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> & infos)284 void Test::DefaultInfosAnalyze(
285     std::shared_ptr<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1> &infos)
286 {
287     infos->v1_0.streamId_ = streamIdAnalyze;
288     infos->v1_0.width_ = analyzeWidth;
289     infos->v1_0.height_ = analyzeHeight;
290     infos->v1_0.format_ = analyzeFormat;
291     infos->v1_0.dataspace_ = UT_DATA_SIZE;
292     infos->v1_0.intent_ = StreamIntent::ANALYZE;
293     infos->v1_0.tunneledMode_ = UT_TUNNEL_MODE;
294 
295     std::shared_ptr<StreamConsumer> consumer_analyze = std::make_shared<StreamConsumer>();
296     infos->v1_0.bufferQueue_ = consumer_analyze->CreateProducerSeq([this](void* addr, uint32_t size) {
297         common_metadata_header_t *data = static_cast<common_metadata_header_t *>(addr);
298         camera_metadata_item_t entry = {};
299 
300         int ret = FindCameraMetadataItem(data, OHOS_STATISTICS_FACE_IDS, &entry);
301         if (ret == 0) {
302             for (size_t i = 0; i < entry.count; i++) {
303                 int id = entry.data.i32[i];
304                 CAMERA_LOGI("Face ids : %{public}d", id);
305             }
306         }
307 
308         ret = FindCameraMetadataItem(data, OHOS_STATISTICS_FACE_RECTANGLES, &entry);
309         if (ret == 0) {
310             for (size_t i = 0; i < entry.count; i++) {
311                 int id = entry.data.i32[i];
312                 CAMERA_LOGI("Face rectangles : %{public}d", id);
313             }
314         }
315     });
316     infos->v1_0.bufferQueue_->producer_->SetQueueSize(UT_DATA_SIZE);
317     consumerMap_[StreamIntent::ANALYZE] = consumer_analyze;
318 }
319 
StartStream(std::vector<StreamIntent> intents)320 void Test::StartStream(std::vector<StreamIntent> intents)
321 {
322     streamOperatorCallback = new TestStreamOperatorCallback();
323     uint32_t mainVersion = 1;
324     uint32_t minVersion = 0;
325     rc = cameraDeviceV1_1->GetStreamOperator_V1_1(streamOperatorCallback, streamOperator_V1_1);
326     if (rc == HDI::Camera::V1_0::NO_ERROR) {
327         rc = streamOperator_V1_1->GetVersion(mainVersion, minVersion);
328         streamOperator = static_cast<OHOS::HDI::Camera::V1_0::IStreamOperator *>(streamOperator_V1_1.GetRefPtr());
329         if (rc != HDI::Camera::V1_0::NO_ERROR) {
330             CAMERA_LOGE("StreamOperator V1_1 get version failed, rc = %{public}d", rc);
331         } else {
332             CAMERA_LOGI("StreamOperator V1_1 get version success, %{public}u, %{public}u",
333                 mainVersion, minVersion);
334         }
335         CAMERA_LOGI("GetStreamOperator success");
336     } else {
337         CAMERA_LOGE("GetStreamOperator fail, rc = %{public}d", rc);
338     }
339     streamInfoPre = std::make_shared<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1>();
340     streamInfoVideo = std::make_shared<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1>();
341     streamInfoCapture = std::make_shared<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1>();
342     streamInfoAnalyze = std::make_shared<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1>();
343 
344     for (auto& intent : intents) {
345         if (intent == StreamIntent::PREVIEW) {
346             DefaultInfosPreview(streamInfoPre);
347             streamInfos.push_back(*streamInfoPre);
348         } else if (intent == StreamIntent::VIDEO) {
349             DefaultInfosVideo(streamInfoVideo);
350             streamInfos.push_back(*streamInfoVideo);
351         } else if (intent == StreamIntent::ANALYZE) {
352             DefaultInfosAnalyze(streamInfoAnalyze);
353             streamInfos.push_back(*streamInfoAnalyze);
354         } else {
355             DefaultInfosCapture(streamInfoCapture);
356             streamInfos.push_back(*streamInfoCapture);
357         }
358     }
359 
360     rc = streamOperator_V1_1->CreateStreams_V1_1(streamInfos);
361     EXPECT_EQ(false, rc != HDI::Camera::V1_0::NO_ERROR);
362     rc = streamOperator_V1_1->CommitStreams(OperationMode::NORMAL, abilityVec);
363     EXPECT_EQ(false, rc != HDI::Camera::V1_0::NO_ERROR);
364     sleep(1);
365     std::vector<OHOS::HDI::Camera::V1_1::StreamInfo_V1_1>().swap(streamInfos);
366 }
367 
StartCapture(int streamId,int captureId,bool shutterCallback,bool isStreaming)368 void Test::StartCapture(int streamId, int captureId, bool shutterCallback, bool isStreaming)
369 {
370     captureInfo = std::make_shared<CaptureInfo>();
371     captureInfo->streamIds_ = {streamId};
372     captureInfo->captureSetting_ = abilityVec;
373     captureInfo->enableShutterCallback_ = shutterCallback;
374     if (streamOperator_V1_2 != nullptr) {
375         rc = (CamRetCode)streamOperator_V1_2->Capture(captureId, *captureInfo, isStreaming);
376     } else if (streamOperator_V1_1 != nullptr) {
377         rc = (CamRetCode)streamOperator_V1_1->Capture(captureId, *captureInfo, isStreaming);
378     } else {
379         rc = (CamRetCode)streamOperator->Capture(captureId, *captureInfo, isStreaming);
380     }
381 
382     EXPECT_EQ(true, rc == HDI::Camera::V1_0::NO_ERROR);
383     if (rc == HDI::Camera::V1_0::NO_ERROR) {
384         CAMERA_LOGI("check Capture: Capture success, %{public}d", captureId);
385     } else {
386         std::cout << "rc = " << rc << std::endl;
387         CAMERA_LOGE("check Capture: Capture fail, rc = %{public}d", rc);
388     }
389     sleep(UT_SLEEP_TIME);
390 }
391 
StopStream(std::vector<int> & captureIds,std::vector<int> & streamIds)392 void Test::StopStream(std::vector<int>& captureIds, std::vector<int>& streamIds)
393 {
394     if (sizeof(captureIds) > 0) {
395         for (auto &captureId : captureIds) {
396             if (streamOperator_V1_2 != nullptr) {
397                 rc = streamOperator_V1_2->CancelCapture(captureId);
398             } else if (streamOperator_V1_1 != nullptr) {
399                 rc = streamOperator_V1_1->CancelCapture(captureId);
400             } else {
401                 rc = streamOperator->CancelCapture(captureId);
402             }
403             EXPECT_EQ(true, rc == HDI::Camera::V1_0::NO_ERROR);
404             if (rc == HDI::Camera::V1_0::NO_ERROR) {
405                 CAMERA_LOGI("check Capture: CancelCapture success, %{public}d", captureId);
406             } else {
407                 CAMERA_LOGE("check Capture: CancelCapture fail, rc = %{public}d, captureId = %{public}d",
408                     rc, captureId);
409             }
410         }
411     }
412     if (sizeof(streamIds) > 0) {
413         if (streamOperator_V1_2 != nullptr) {
414             rc = streamOperator_V1_2->ReleaseStreams(streamIds);
415         } else if (streamOperator_V1_1 != nullptr) {
416             rc = streamOperator_V1_1->ReleaseStreams(streamIds);
417         } else {
418             rc = streamOperator->ReleaseStreams(streamIds);
419         }
420         EXPECT_EQ(true, rc == HDI::Camera::V1_0::NO_ERROR);
421         if (rc == HDI::Camera::V1_0::NO_ERROR) {
422             CAMERA_LOGI("check Capture: ReleaseStream success");
423         } else {
424             CAMERA_LOGE("check Capture: ReleaseStreams fail, rc = %{public}d", rc);
425         }
426     }
427 }
428 
CalculateFps(int64_t timestamp,int32_t streamId)429 void Test::StreamConsumer::CalculateFps(int64_t timestamp, int32_t streamId)
430 {
431     if (isFirstCalculateFps_) {
432         if ((timestamp - intervalTimestamp_) >= interval_) {
433             int64_t timeInterval = timestamp - intervalTimestamp_;
434             if (timeInterval != 0) {
435                 float fps = (int64_t)(100000000000 * timestampCount_ / timeInterval) / 100.0;
436                 CAMERA_LOGI("Calculate FPS success, streamId: %{public}d, Fps:%{public}f", streamId, fps);
437                 interval_ = ONESECOND_OF_MICROSECOND_UNIT;
438             } else {
439                 CAMERA_LOGE("Calculate FPS error timeInerval is 0");
440             }
441         }
442     } else {
443         intervalTimestamp_ = timestamp;
444         isFirstCalculateFps_ = true;
445     }
446     if ((timestamp - intervalTimestamp_) >= ONESECOND_OF_MICROSECOND_UNIT * UT_SECOND_TIMES) {
447         intervalTimestamp_ = timestamp;
448         timestampCount_ = 0;
449         interval_ = ONESECOND_OF_MICROSECOND_UNIT;
450     }
451     timestampCount_++;
452 }
453 
GetTimeStamp(int64_t * g_timestamp,uint32_t lenght,int64_t timestamp,int32_t gotSize)454 void Test::StreamConsumer::GetTimeStamp(int64_t *g_timestamp, uint32_t lenght, int64_t timestamp, int32_t gotSize)
455 {
456     if (gotSize != UT_PREVIEW_SIZE) {
457         if (g_timestamp[0] == 0) {
458             g_timestamp[0] = timestamp;
459         } else {
460             g_timestamp[1] = timestamp;
461         }
462     }
463 }
464 
CreateProducer(std::function<void (void *,uint32_t)> callback)465 OHOS::sptr<OHOS::IBufferProducer> Test::StreamConsumer::CreateProducer(std::function<void(void*, uint32_t)> callback)
466 {
467     consumer_ = OHOS::IConsumerSurface::Create();
468     if (consumer_ == nullptr) {
469         return nullptr;
470     }
471     sptr<IBufferConsumerListener> listener = new TestBufferConsumerListener();
472     consumer_->RegisterConsumerListener(listener);
473     auto producer = consumer_->GetProducer();
474     if (producer == nullptr) {
475         return nullptr;
476     }
477 
478     callback_ = callback;
479     consumerThread_ = new std::thread([this, listener] {
480         int32_t flushFence = 0;
481         int64_t timestamp = 0;
482         OHOS::Rect damage;
483         TestBufferConsumerListener* checker = static_cast<TestBufferConsumerListener*>(listener.GetRefPtr());
484         while (running_ == true) {
485             OHOS::sptr<OHOS::SurfaceBuffer> buffer = nullptr;
486             if (checker->checkBufferAvailable()) {
487                 consumer_->AcquireBuffer(buffer, flushFence, timestamp, damage);
488                 if (buffer != nullptr) {
489                     void* addr = buffer->GetVirAddr();
490                     uint32_t size = buffer->GetSize();
491 
492                     int32_t gotSize = 0;
493                     int32_t isKey = 0;
494                     int32_t streamId = 0;
495                     int32_t captureId = 0;
496                     buffer->GetExtraData()->ExtraGet(OHOS::Camera::dataSize, gotSize);
497                     buffer->GetExtraData()->ExtraGet(OHOS::Camera::isKeyFrame, isKey);
498                     buffer->GetExtraData()->ExtraGet(OHOS::Camera::timeStamp, timestamp);
499                     buffer->GetExtraData()->ExtraGet(OHOS::Camera::streamId, streamId);
500                     buffer->GetExtraData()->ExtraGet(OHOS::Camera::captureId, captureId);
501                     GetTimeStamp(g_timestamp, sizeof(g_timestamp) / sizeof(g_timestamp[0]), timestamp, gotSize);
502                     if (gotSize) {
503                         CalculateFps(timestamp, streamId);
504                         callback_(addr, gotSize);
505                     } else {
506                         callback_(addr, size);
507                     }
508 
509                     consumer_->ReleaseBuffer(buffer, -1);
510                     shotCount_--;
511                     if (shotCount_ == 0) {
512                         std::unique_lock<std::mutex> l(l_);
513                         cv_.notify_one();
514                     }
515                 }
516             }
517             if (running_ == false) {
518                 break;
519             }
520             usleep(1);
521         }
522     });
523 
524     return producer;
525 }
526 
CreateProducerSeq(std::function<void (void *,uint32_t)> callback)527 OHOS::sptr<BufferProducerSequenceable> Test::StreamConsumer::CreateProducerSeq(
528     std::function<void(void*, uint32_t)> callback)
529 {
530     OHOS::sptr<OHOS::IBufferProducer> producer = CreateProducer(callback);
531     if (producer == nullptr) {
532         return nullptr;
533     }
534 
535     return new BufferProducerSequenceable(producer);
536 }
537 
OnCaptureStarted(int32_t captureId,const std::vector<int32_t> & streamId)538 int32_t Test::TestStreamOperatorCallback::OnCaptureStarted(int32_t captureId, const std::vector<int32_t> &streamId)
539 {
540     for (auto it : streamId) {
541         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d", captureId, it);
542     }
543     return HDI::Camera::V1_0::NO_ERROR;
544 }
545 
OnCaptureEnded(int32_t captureId,const std::vector<CaptureEndedInfo> & infos)546 int32_t Test::TestStreamOperatorCallback::OnCaptureEnded(int32_t captureId, const std::vector<CaptureEndedInfo> &infos)
547 {
548     for (auto it : infos) {
549         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d, count: %{public}d", captureId, it.streamId_,
550             it.frameCount_);
551     }
552     return HDI::Camera::V1_0::NO_ERROR;
553 }
554 
OnCaptureError(int32_t captureId,const std::vector<CaptureErrorInfo> & infos)555 int32_t Test::TestStreamOperatorCallback::OnCaptureError(int32_t captureId, const std::vector<CaptureErrorInfo> &infos)
556 {
557     for (auto it : infos) {
558         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d, error: %{public}d", captureId, it.streamId_,
559             it.error_);
560     }
561     return HDI::Camera::V1_0::NO_ERROR;
562 }
563 
OnFrameShutter(int32_t captureId,const std::vector<int32_t> & streamIds,uint64_t timestamp)564 int32_t Test::TestStreamOperatorCallback::OnFrameShutter(int32_t captureId,
565     const std::vector<int32_t> &streamIds, uint64_t timestamp)
566 {
567     (void)timestamp;
568     for (auto it : streamIds) {
569         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d", captureId, it);
570     }
571     return HDI::Camera::V1_0::NO_ERROR;
572 }
573 
OnCaptureStarted(int32_t captureId,const std::vector<int32_t> & streamId)574 int32_t Test::TestStreamOperatorCallbackV1_2::OnCaptureStarted(int32_t captureId, const std::vector<int32_t> &streamId)
575 {
576     for (auto it : streamId) {
577         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d", captureId, it);
578     }
579     return HDI::Camera::V1_0::NO_ERROR;
580 }
581 
OnCaptureEnded(int32_t captureId,const std::vector<CaptureEndedInfo> & infos)582 int32_t Test::TestStreamOperatorCallbackV1_2::OnCaptureEnded(int32_t captureId,
583     const std::vector<CaptureEndedInfo> &infos)
584 {
585     for (auto it : infos) {
586         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d, count: %{public}d", captureId, it.streamId_,
587             it.frameCount_);
588     }
589     return HDI::Camera::V1_0::NO_ERROR;
590 }
591 
OnCaptureError(int32_t captureId,const std::vector<CaptureErrorInfo> & infos)592 int32_t Test::TestStreamOperatorCallbackV1_2::OnCaptureError(int32_t captureId,
593     const std::vector<CaptureErrorInfo> &infos)
594 {
595     for (auto it : infos) {
596         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d, error: %{public}d", captureId, it.streamId_,
597             it.error_);
598     }
599     return HDI::Camera::V1_0::NO_ERROR;
600 }
601 
OnFrameShutter(int32_t captureId,const std::vector<int32_t> & streamIds,uint64_t timestamp)602 int32_t Test::TestStreamOperatorCallbackV1_2::OnFrameShutter(int32_t captureId,
603     const std::vector<int32_t> &streamIds, uint64_t timestamp)
604 {
605     (void)timestamp;
606     for (auto it : streamIds) {
607         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d", captureId, it);
608     }
609     return HDI::Camera::V1_0::NO_ERROR;
610 }
611 
OnCaptureStarted_V1_2(int32_t captureId,const std::vector<HDI::Camera::V1_2::CaptureStartedInfo> & infos)612 int32_t Test::TestStreamOperatorCallbackV1_2::OnCaptureStarted_V1_2(int32_t captureId,
613     const std::vector<HDI::Camera::V1_2::CaptureStartedInfo> &infos)
614 {
615     for (auto it : infos) {
616         CAMERA_LOGE("captureId: %{public}d, streamId: %{public}d", captureId, it.streamId_);
617     }
618     return HDI::Camera::V1_0::NO_ERROR;
619 }
620 
OnError(ErrorType type,int32_t errorMsg)621 int32_t Test::DemoCameraDeviceCallback::OnError(ErrorType type, int32_t errorMsg)
622 {
623     CAMERA_LOGE("type: %{public}d, errorMsg: %{public}d", type, errorMsg);
624     return HDI::Camera::V1_0::NO_ERROR;
625 }
626 
OnResult(uint64_t timestamp,const std::vector<uint8_t> & result)627 int32_t Test::DemoCameraDeviceCallback::OnResult(uint64_t timestamp, const std::vector<uint8_t> &result)
628 {
629     MetadataUtils::ConvertVecToMetadata(result, resultMeta);
630     if (Test::resultCallback_) {
631         Test::resultCallback_(timestamp, resultMeta);
632     }
633     return HDI::Camera::V1_0::NO_ERROR;
634 }
635 
OnCameraStatus(const std::string & cameraId,CameraStatus status)636 int32_t Test::TestCameraHostCallback::OnCameraStatus(const std::string& cameraId, CameraStatus status)
637 {
638     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), status);
639     return HDI::Camera::V1_0::NO_ERROR;
640 }
641 
OnFlashlightStatus(const std::string & cameraId,FlashlightStatus status)642 int32_t Test::TestCameraHostCallback::OnFlashlightStatus(const std::string& cameraId, FlashlightStatus status)
643 {
644     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), status);
645     return HDI::Camera::V1_0::NO_ERROR;
646 }
647 
OnCameraEvent(const std::string & cameraId,CameraEvent event)648 int32_t Test::TestCameraHostCallback::OnCameraEvent(const std::string& cameraId, CameraEvent event)
649 {
650     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), event);
651     return HDI::Camera::V1_0::NO_ERROR;
652 }
653 
OnCameraStatus(const std::string & cameraId,CameraStatus status)654 int32_t Test::TestCameraHostCallbackV1_2::OnCameraStatus(const std::string& cameraId, CameraStatus status)
655 {
656     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), status);
657     return HDI::Camera::V1_0::NO_ERROR;
658 }
659 
OnFlashlightStatus(const std::string & cameraId,FlashlightStatus status)660 int32_t Test::TestCameraHostCallbackV1_2::OnFlashlightStatus(const std::string& cameraId, FlashlightStatus status)
661 {
662     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), status);
663     return HDI::Camera::V1_0::NO_ERROR;
664 }
665 
OnCameraEvent(const std::string & cameraId,CameraEvent event)666 int32_t Test::TestCameraHostCallbackV1_2::OnCameraEvent(const std::string& cameraId, CameraEvent event)
667 {
668     CAMERA_LOGE("cameraId: %{public}s, status: %{public}d", cameraId.c_str(), event);
669     return HDI::Camera::V1_0::NO_ERROR;
670 }
671 
OnFlashlightStatus_V1_2(FlashlightStatus status)672 int32_t Test::TestCameraHostCallbackV1_2::OnFlashlightStatus_V1_2(FlashlightStatus status)
673 {
674     CAMERA_LOGE("status: %{public}d", status);
675     Test::statusCallback = status;
676     return HDI::Camera::V1_0::NO_ERROR;
677 }
678 
OnProcessDone(const std::string & imageId,const OHOS::HDI::Camera::V1_2::ImageBufferInfo & buffer)679 int32_t Test::TestImageProcessCallback::OnProcessDone(const std::string& imageId,
680     const OHOS::HDI::Camera::V1_2::ImageBufferInfo& buffer)
681 {
682     CAMERA_LOGI("imageId: %{public}s", imageId.c_str());
683     coutProcessDone_++;
684     curImageId_ = imageId;
685     curImageBufferInfo_ = buffer;
686     return 0;
687 }
688 
OnStatusChanged(OHOS::HDI::Camera::V1_2::SessionStatus status)689 int32_t Test::TestImageProcessCallback::OnStatusChanged(OHOS::HDI::Camera::V1_2::SessionStatus status)
690 {
691     CAMERA_LOGI("status: %{public}d", status);
692     curStatus_ = status;
693     coutStatusChanged_++;
694     return 0;
695 }
696 
OnError(const std::string & imageId,OHOS::HDI::Camera::V1_2::ErrorCode errorCode)697 int32_t Test::TestImageProcessCallback::OnError(const std::string& imageId,
698     OHOS::HDI::Camera::V1_2::ErrorCode errorCode)
699 {
700     CAMERA_LOGI("imageId: %{public}s, errorCode: %{public}d", imageId.c_str(), errorCode);
701     curImageId_ = imageId;
702     curErrorCode_ = errorCode;
703     countError_++;
704     return 0;
705 }
706 } // OHOS::Camera