1 /*
2  * Copyright (C) 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 <iostream>
17 #include <cstdio>
18 #include <atomic>
19 #include <fstream>
20 #include <thread>
21 #include <mutex>
22 #include <queue>
23 #include <string>
24 
25 #include "gtest/gtest.h"
26 #include "hdrcodec_inner_sample.h"
27 #include "native_averrors.h"
28 #include "avcodec_info.h"
29 
30 using namespace std;
31 using namespace OHOS;
32 using namespace OHOS::MediaAVCodec;
33 using namespace testing::ext;
34 namespace OHOS {
35 class HDRFuncInnerNdkTest : public testing::Test {
36 public:
37     // SetUpTestCase: Called before all test cases
38     static void SetUpTestCase(void);
39     // TearDownTestCase: Called after all test case
40     static void TearDownTestCase(void);
41     // SetUp: Called before each test cases
42     void SetUp(void);
43     // TearDown: Called after each test cases
44     void TearDown(void);
45 };
46 
SetUpTestCase()47 void HDRFuncInnerNdkTest::SetUpTestCase()
48 {
49 }
TearDownTestCase()50 void HDRFuncInnerNdkTest::TearDownTestCase() {}
SetUp()51 void HDRFuncInnerNdkTest::SetUp()
52 {
53 }
TearDown()54 void HDRFuncInnerNdkTest::TearDown()
55 {
56 }
57 } // namespace OHOS
58 
59 namespace {
60 /**
61  * @tc.number    : HDR_INNER_FUNC_0010
62  * @tc.name      : decode PQ HDRVivid by display mode, and then set HEVC_PROFILE_MAIN10 to encode
63  * @tc.desc      : function test
64  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0010, TestSize.Level1)65 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0010, TestSize.Level1)
66 {
67     if (!access("/system/lib64/media/", 0)) {
68         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
69         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
70         sample->INP_DIR = "/data/test/media/pq_vivid.h265";
71         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
72         ASSERT_EQ(AV_ERR_OK, sample->Configure());
73         ASSERT_EQ(AV_ERR_OK, sample->Start());
74         sample->WaitForEos();
75         ASSERT_EQ(0, sample->errorCount);
76     }
77 }
78 /**
79  * @tc.number    : HDR_INNER_FUNC_0020
80  * @tc.name      : decode HLG HDRVivid by display mode, and then set HEVC_PROFILE_MAIN10 to encode
81  * @tc.desc      : function test
82  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0020, TestSize.Level1)83 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0020, TestSize.Level1)
84 {
85     if (!access("/system/lib64/media/", 0)) {
86         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
87         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
88         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
89         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
90         ASSERT_EQ(AV_ERR_OK, sample->Configure());
91         ASSERT_EQ(AV_ERR_OK, sample->Start());
92         sample->WaitForEos();
93         ASSERT_EQ(0, sample->errorCount);
94     }
95 }
96 
97 /**
98  * @tc.number    : HDR_INNER_FUNC_0030
99  * @tc.name      : decode PQ HDRVivid by display mode, and then set HEVC_PROFILE_MAIN to encode
100  * @tc.desc      : function test
101  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0030, TestSize.Level2)102 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0030, TestSize.Level2)
103 {
104     if (!access("/system/lib64/media/", 0)) {
105         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
106         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN;
107         sample->INP_DIR = "/data/test/media/pq_vivid.h265";
108         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
109         ASSERT_EQ(AV_ERR_OK, sample->Configure());
110         ASSERT_EQ(AV_ERR_OK, sample->Start());
111         sample->WaitForEos();
112         EXPECT_LE(0, sample->errorCount);
113     }
114 }
115 
116 /**
117  * @tc.number    : HDR_INNER_FUNC_0040
118  * @tc.name      : decode HLG HDRVivid by display mode, and then set HEVC_PROFILE_MAIN to encode
119  * @tc.desc      : function test
120  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0040, TestSize.Level2)121 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0040, TestSize.Level2)
122 {
123     if (!access("/system/lib64/media/", 0)) {
124         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
125         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN;
126         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
127         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
128         ASSERT_EQ(AV_ERR_OK, sample->Configure());
129         ASSERT_EQ(AV_ERR_OK, sample->Start());
130         sample->WaitForEos();
131         EXPECT_LE(0, sample->errorCount);
132     }
133 }
134 
135 /**
136  * @tc.number    : HDR_INNER_FUNC_0050
137  * @tc.name      : decode and encode HDRVivid repeat start-stop 5 times before EOS
138  * @tc.desc      : function test
139  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0050, TestSize.Level2)140 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0050, TestSize.Level2)
141 {
142     if (!access("/system/lib64/media/", 0)) {
143         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
144         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
145         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
146         sample->REPEAT_START_STOP_BEFORE_EOS = 5;
147         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
148         ASSERT_EQ(AV_ERR_OK, sample->Configure());
149         ASSERT_EQ(AV_ERR_OK, sample->Start());
150         sample->WaitForEos();
151         ASSERT_EQ(0, sample->errorCount);
152     }
153 }
154 
155 /**
156  * @tc.number    : HDR_INNER_FUNC_0060
157  * @tc.name      : decode and encode HDRVivid repeat start-flush-stop 5 times before EOS
158  * @tc.desc      : function test
159  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0060, TestSize.Level2)160 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0060, TestSize.Level2)
161 {
162     if (!access("/system/lib64/media/", 0)) {
163         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
164         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
165         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
166         sample->REPEAT_START_FLUSH_STOP_BEFORE_EOS = 5;
167         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
168         ASSERT_EQ(AV_ERR_OK, sample->Configure());
169         ASSERT_EQ(AV_ERR_OK, sample->Start());
170         sample->WaitForEos();
171         ASSERT_EQ(0, sample->errorCount);
172     }
173 }
174 
175 /**
176  * @tc.number    : HDR_INNER_FUNC_0070
177  * @tc.name      : decode and encode HDRVivid repeat start-flush-start 5 times before EOS
178  * @tc.desc      : function test
179  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0070, TestSize.Level2)180 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0070, TestSize.Level2)
181 {
182     if (!access("/system/lib64/media/", 0)) {
183         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
184         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
185         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
186         sample->REPEAT_START_FLUSH_BEFORE_EOS = 5;
187         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
188         ASSERT_EQ(AV_ERR_OK, sample->Configure());
189         ASSERT_EQ(AV_ERR_OK, sample->Start());
190         sample->WaitForEos();
191         ASSERT_EQ(0, sample->errorCount);
192     }
193 }
194 
195 /**
196  * @tc.number    : HDR_INNER_FUNC_0080
197  * @tc.name      : decode and encode different resolution with same codec
198  * @tc.desc      : function test
199  */
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0080, TestSize.Level1)200 HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0080, TestSize.Level1)
201 {
202     if (!access("/system/lib64/media/", 0)) {
203         shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
204         sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
205         sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
206         ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
207         ASSERT_EQ(AV_ERR_OK, sample->Configure());
208         ASSERT_EQ(AV_ERR_OK, sample->Start());
209         sample->WaitForEos();
210         sample->DEFAULT_WIDTH = 1920;
211         sample->DEFAULT_HEIGHT = 1080;
212         sample->INP_DIR = "/data/test/media/hlg_vivid_1080p.h265";
213         ASSERT_EQ(AV_ERR_OK, sample->ReConfigure());
214         ASSERT_EQ(AV_ERR_OK, sample->Start());
215         sample->WaitForEos();
216         ASSERT_EQ(0, sample->errorCount);
217     }
218 }
219 } // namespace