!3444 新增hdrcodec inner接口、能力查询inner接口用例

Merge pull request !3444 from dwl/master
This commit is contained in:
openharmony_ci 2024-09-24 08:41:07 +00:00 committed by Gitee
commit f9737e7834
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
7 changed files with 1570 additions and 2 deletions

View File

@ -20,7 +20,10 @@ module_output_path = "av_codec/moduletest"
ohos_unittest("capability_module_test") {
module_out_path = module_output_path
sources = [ "./src/cap_test.cpp" ]
sources = [
"./src/InnerAPI/cap_test.cpp",
"./src/cap_test.cpp",
]
include_dirs = [
"Avbuffer/include",

View File

@ -0,0 +1,676 @@
/*
* Copyright (C) 2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <string>
#include <securec.h>
#include "gtest/gtest.h"
#include "native_avcodec_videoencoder.h"
#include "native_averrors.h"
#include "native_avcodec_base.h"
#include "avcodec_codec_name.h"
#include "native_avcapability.h"
#include "avcodec_info.h"
#include "avcodec_list.h"
#include "meta/meta_key.h"
#include "meta/format.h"
#include "avcodec_errors.h"
#include "av_common.h"
#include <typeinfo>
namespace {
OH_AVCodec *venc_ = NULL;
constexpr uint32_t CODEC_NAME_SIZE = 128;
char g_codecName[CODEC_NAME_SIZE] = {};
char g_codecNameHEVC[CODEC_NAME_SIZE] = {};
OH_AVFormat *format;
} // namespace
namespace OHOS {
namespace Media {
class HwCapabilityInnerNdkTest : public testing::Test {
public:
static void SetUpTestCase();
static void TearDownTestCase();
void SetUp() override;
void TearDown() override;
void InputFunc();
void OutputFunc();
void Release();
int32_t Stop();
};
} // namespace Media
} // namespace OHOS
using namespace std;
using namespace OHOS;
using namespace OHOS::Media;
using namespace testing::ext;
using namespace OHOS::MediaAVCodec;
void HwCapabilityInnerNdkTest::SetUpTestCase()
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
if (codeclist != nullptr) {
CapabilityData *cap = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
if (cap != nullptr) {
const std::string &name = cap->codecName;
const char *tmpCodecName = name.data();
if (memcpy_s(g_codecName, sizeof(g_codecName), tmpCodecName, strlen(tmpCodecName)) != 0)
cout << "memcpy failed" << endl;
cout << "codecname: " << g_codecName << endl;
}
CapabilityData *cap_hevc = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
if (cap_hevc != nullptr) {
const std::string &name = cap_hevc->codecName;
const char *tmpCodecNameHevc = name.data();
if (memcpy_s(g_codecNameHEVC, sizeof(g_codecNameHEVC), tmpCodecNameHevc, strlen(tmpCodecNameHevc)) != 0)
cout << "memcpy failed" << endl;
cout << "codecname_hevc: " << g_codecNameHEVC << endl;
}
}
}
void HwCapabilityInnerNdkTest::TearDownTestCase() {}
void HwCapabilityInnerNdkTest::SetUp() {}
void HwCapabilityInnerNdkTest::TearDown()
{
if (venc_ != NULL) {
OH_VideoEncoder_Destroy(venc_);
venc_ = nullptr;
}
if (format != nullptr) {
OH_AVFormat_Destroy(format);
format = nullptr;
}
}
namespace {
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0010
* @tc.name : IsFeatureSupported para error
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0010, TestSize.Level2)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(static_cast<AVCapabilityFeature>(4)));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0011
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0011, TestSize.Level2)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0012
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0012, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0013
* @tc.name : LTR
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0013, TestSize.Level2)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0014
* @tc.name : LTR
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0014, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0015
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0015, TestSize.Level2)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_SOFTWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0016
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0016, TestSize.Level2)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0017
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0017, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0018
* @tc.name : GetFeatureProperties para error
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0018, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION, codecInfo->GetFeatureProperties(static_cast<AVCapabilityFeature>(4),
featureFormat));
ASSERT_EQ(AVCS_ERR_INVALID_VAL, codecInfo->GetFeatureProperties(static_cast<AVCapabilityFeature>(-1),
featureFormat));
ASSERT_EQ(AVCS_ERR_INVALID_VAL, codecInfo->GetFeatureProperties(static_cast<AVCapabilityFeature>(100),
featureFormat));
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY,
featureFormat));
} else {
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0019
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0019, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0020
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0020, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY,
featureFormat));
} else {
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0021
* @tc.name : LTR能力值
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0021, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0022
* @tc.name : LTR的能力值
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0022, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE,
featureFormat));
int ltrnum = 0;
EXPECT_EQ(featureFormat.GetIntValue(OH_FEATURE_PROPERTY_KEY_VIDEO_ENCODER_MAX_LTR_FRAME_COUNT, ltrnum), true);
EXPECT_EQ(ltrnum, 10);
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0023
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0023, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_SOFTWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0024
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0024, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0025
* @tc.name :
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0025, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_AVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY,
featureFormat));
} else {
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0026
* @tc.name : LTRH265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0026, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
} else {
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0027
* @tc.name : LTRH265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0027, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0028
* @tc.name : H265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0028, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
} else {
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0029
* @tc.name : H265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0029, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
} else {
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_LOW_LATENCY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0030
* @tc.name : H265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0030, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(true, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
} else {
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0031
* @tc.name : H265
* @tc.desc : function test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0031, TestSize.Level1)
{
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(false, codecInfo->IsFeatureSupported(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0032
* @tc.name : ,H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0032, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY,
featureFormat));
} else {
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0033
* @tc.name : ,H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0033, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0034
* @tc.name : H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0034, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0035
* @tc.name : H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0035, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
if (!access("/system/lib64/media/", 0)) {
ASSERT_EQ(AVCS_ERR_OK, codecInfo->GetFeatureProperties(
AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
} else {
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0036
* @tc.name : LTR能力值H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0036, TestSize.Level2)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
false, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_TEMPORAL_SCALABILITY, featureFormat));
}
/**
* @tc.number : VIDEO_TEMPORAL_ENCODE_INNER_API_0037
* @tc.name : LTR的能力值H265
* @tc.desc : api test
*/
HWTEST_F(HwCapabilityInnerNdkTest, VIDEO_TEMPORAL_ENCODE_INNER_API_0037, TestSize.Level1)
{
Format featureFormat;
auto codeclist = AVCodecListFactory::CreateAVCodecList();
ASSERT_NE(nullptr, codeclist);
if (!access("/system/lib64/media/", 0)) {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_OK,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE, featureFormat));
int ltrnum = 0;
EXPECT_EQ(featureFormat.GetIntValue(OH_FEATURE_PROPERTY_KEY_VIDEO_ENCODER_MAX_LTR_FRAME_COUNT, ltrnum), true);
EXPECT_EQ(ltrnum, 10);
} else {
CapabilityData *capabilityData = codeclist->GetCapability(string(CodecMimeType::VIDEO_HEVC),
true, AVCodecCategory::AVCODEC_HARDWARE);
ASSERT_NE(nullptr, capabilityData);
std::shared_ptr<AVCodecInfo> codecInfo = std::make_shared<AVCodecInfo>(capabilityData);
ASSERT_EQ(AVCS_ERR_INVALID_OPERATION,
codecInfo->GetFeatureProperties(AVCapabilityFeature::VIDEO_ENCODER_LONG_TERM_REFERENCE, featureFormat));
}
}
} // namespace

View File

@ -58,6 +58,8 @@ ohos_unittest("hdrcodec_native_module_test") {
if (av_codec_support_codec) {
sources = [
"./src/InnerAPI/func_test.cpp",
"./src/InnerAPI/hdrcodec_inner_ndk_sample.cpp",
"./src/func_test.cpp",
"./src/hdrcodec_ndk_sample.cpp",
]

View File

@ -0,0 +1,138 @@
/*
* Copyright (C) 2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef HDRCODEC_SAMPLE_H
#define HDRCODEC_SAMPLE_H
#include <iostream>
#include <cstdio>
#include <unistd.h>
#include <atomic>
#include <fstream>
#include <thread>
#include <mutex>
#include <queue>
#include <string>
#include <unordered_map>
#include "securec.h"
#include "avcodec_video_encoder.h"
#include "nocopyable.h"
#include "buffer/avsharedmemory.h"
#include "meta/format.h"
#include "avcodec_errors.h"
#include "surface/window.h"
#include "media_description.h"
#include "av_common.h"
#include "avcodec_common.h"
#include "external_window.h"
#include "avcodec_video_decoder.h"
namespace OHOS {
namespace MediaAVCodec {
class InnerSignal {
public:
std::mutex inMutex_;
std::mutex outMutex_;
std::condition_variable inCond_;
std::condition_variable outCond_;
std::queue<uint32_t> inIdxQueue_;
std::queue<uint32_t> outIdxQueue_;
std::queue<std::shared_ptr<AVSharedMemory>> inBufferQueue_;
std::queue<std::shared_ptr<AVSharedMemory>> outBufferQueue_;
};
class HdrEncInnerCallback : public AVCodecCallback, public NoCopyable {
public:
explicit HdrEncInnerCallback(std::shared_ptr<InnerSignal> signal);
~HdrEncInnerCallback() = default;
void OnError(AVCodecErrorType errorType, int32_t errorCode) override;
void OnOutputFormatChanged(const Format& format) override;
void OnInputBufferAvailable(uint32_t index, std::shared_ptr<AVSharedMemory> buffer) override;
void OnOutputBufferAvailable(uint32_t index, AVCodecBufferInfo info, AVCodecBufferFlag flag,
std::shared_ptr<AVSharedMemory> buffer) override;
private:
std::shared_ptr<InnerSignal> encInnersignal_;
};
class HdrDecInnerCallback : public AVCodecCallback, public NoCopyable {
public:
explicit HdrDecInnerCallback(std::shared_ptr<InnerSignal> signal);
~HdrDecInnerCallback() = default;
void OnError(AVCodecErrorType errorType, int32_t errorCode) override;
void OnOutputFormatChanged(const Format& format) override;
void OnInputBufferAvailable(uint32_t index, std::shared_ptr<AVSharedMemory> buffer) override;
void OnOutputBufferAvailable(uint32_t index, AVCodecBufferInfo info, AVCodecBufferFlag flag,
std::shared_ptr<AVSharedMemory> buffer) override;
private:
std::shared_ptr<InnerSignal> decInnersignal_;
};
class HDRCodecInnderNdkSample : public NoCopyable {
public:
HDRCodecInnderNdkSample() = default;
~HDRCodecInnderNdkSample();
int32_t CreateCodec();
int32_t Configure();
int32_t Start();
void WaitForEos();
void ReleaseInFile();
void StopInloop();
void InputFunc();
void FlushBuffer();
void SwitchInputFile();
int32_t ReConfigure();
int32_t RepeatCall();
int32_t DecSetCallback();
int32_t EncSetCallback();
int32_t RepeatCallStartFlush();
int32_t RepeatCallStartStop();
int32_t RepeatCallStartFlushStop();
void Release();
int32_t SendData(std::shared_ptr<AVCodecVideoDecoder> codec,
uint32_t index, std::shared_ptr<AVSharedMemory> buffer);
const char *INP_DIR = "/data/test/media/1920_1080_10_30Mb.h264";
bool needEncode = false;
bool needTransCode = false;
uint32_t DEFAULT_WIDTH = 3840;
uint32_t DEFAULT_HEIGHT = 2160;
double DEFAULT_FRAME_RATE = 30.0;
uint32_t REPEAT_START_STOP_BEFORE_EOS = 0; // 1200 测试用例
uint32_t REPEAT_START_FLUSH_BEFORE_EOS = 0; // 1300 测试用例
uint32_t REPEAT_START_FLUSH_STOP_BEFORE_EOS = 0;
uint32_t frameCount_ = 0;
uint32_t repeat_time = 0;
uint32_t frameCountDec = 0;
uint32_t frameCountEnc = 0;
int32_t DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
bool needOutputFile;
std::unique_ptr<std::thread> inputLoop_;
std::shared_ptr<AVCodecVideoDecoder> vdec_;
std::shared_ptr<AVCodecVideoEncoder> venc_;
uint32_t errorCount = 0;
private:
OHNativeWindow *window;
std::shared_ptr<HdrEncInnerCallback> encCb_;
std::shared_ptr<HdrDecInnerCallback> decCb_;
std::shared_ptr<InnerSignal> signal_;
};
} // namespace Media
} // namespace OHOS
#endif

View File

@ -0,0 +1,219 @@
/*
* Copyright (C) 2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <iostream>
#include <cstdio>
#include <atomic>
#include <fstream>
#include <thread>
#include <mutex>
#include <queue>
#include <string>
#include "gtest/gtest.h"
#include "hdrcodec_inner_sample.h"
#include "native_averrors.h"
#include "avcodec_info.h"
using namespace std;
using namespace OHOS;
using namespace OHOS::MediaAVCodec;
using namespace testing::ext;
namespace OHOS {
class HDRFuncInnerNdkTest : public testing::Test {
public:
// SetUpTestCase: Called before all test cases
static void SetUpTestCase(void);
// TearDownTestCase: Called after all test case
static void TearDownTestCase(void);
// SetUp: Called before each test cases
void SetUp(void);
// TearDown: Called after each test cases
void TearDown(void);
};
void HDRFuncInnerNdkTest::SetUpTestCase()
{
}
void HDRFuncInnerNdkTest::TearDownTestCase() {}
void HDRFuncInnerNdkTest::SetUp()
{
}
void HDRFuncInnerNdkTest::TearDown()
{
}
} // namespace OHOS
namespace {
/**
* @tc.number : HDR_INNER_FUNC_0010
* @tc.name : decode PQ HDRVivid by display mode, and then set HEVC_PROFILE_MAIN10 to encode
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0010, TestSize.Level1)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/pq_vivid.h265";
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0020
* @tc.name : decode HLG HDRVivid by display mode, and then set HEVC_PROFILE_MAIN10 to encode
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0020, TestSize.Level1)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0030
* @tc.name : decode PQ HDRVivid by display mode, and then set HEVC_PROFILE_MAIN to encode
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0030, TestSize.Level2)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN;
sample->INP_DIR = "/data/test/media/pq_vivid.h265";
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
EXPECT_LE(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0040
* @tc.name : decode HLG HDRVivid by display mode, and then set HEVC_PROFILE_MAIN to encode
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0040, TestSize.Level2)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
EXPECT_LE(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0050
* @tc.name : decode and encode HDRVivid repeat start-stop 5 times before EOS
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0050, TestSize.Level2)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
sample->REPEAT_START_STOP_BEFORE_EOS = 5;
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0060
* @tc.name : decode and encode HDRVivid repeat start-flush-stop 5 times before EOS
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0060, TestSize.Level2)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
sample->REPEAT_START_FLUSH_STOP_BEFORE_EOS = 5;
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0070
* @tc.name : decode and encode HDRVivid repeat start-flush-start 5 times before EOS
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0070, TestSize.Level2)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
sample->REPEAT_START_FLUSH_BEFORE_EOS = 5;
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
/**
* @tc.number : HDR_INNER_FUNC_0080
* @tc.name : decode and encode different resolution with same codec
* @tc.desc : function test
*/
HWTEST_F(HDRFuncInnerNdkTest, HDR_INNER_FUNC_0080, TestSize.Level1)
{
if (!access("/system/lib64/media/", 0)) {
shared_ptr<HDRCodecInnderNdkSample> sample = make_shared<HDRCodecInnderNdkSample>();
sample->DEFAULT_PROFILE = HEVC_PROFILE_MAIN_10;
sample->INP_DIR = "/data/test/media/hlg_vivid_4k.h265";
ASSERT_EQ(AV_ERR_OK, sample->CreateCodec());
ASSERT_EQ(AV_ERR_OK, sample->Configure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
sample->DEFAULT_WIDTH = 1920;
sample->DEFAULT_HEIGHT = 1080;
sample->INP_DIR = "/data/test/media/hlg_vivid_1080p.h265";
ASSERT_EQ(AV_ERR_OK, sample->ReConfigure());
ASSERT_EQ(AV_ERR_OK, sample->Start());
sample->WaitForEos();
ASSERT_EQ(0, sample->errorCount);
}
}
} // namespace

View File

@ -0,0 +1,518 @@
/*
* Copyright (C) 2024 Huawei Device Co., Ltd.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "hdrcodec_inner_sample.h"
#include <arpa/inet.h>
#include <sys/time.h>
#include <utility>
#include <memory>
#include "native_avcodec_base.h"
#include "surface/window.h"
#include "native_window.h"
#include "avcodec_errors.h"
#include "avcodec_common.h"
using namespace OHOS;
using namespace OHOS::MediaAVCodec;
using namespace std;
namespace {
constexpr int64_t NANOS_IN_SECOND = 1000000000L;
constexpr int64_t NANOS_IN_MICRO = 1000L;
std::shared_ptr<std::ifstream> inFile_;
std::condition_variable g_cv;
std::atomic<bool> g_isRunning = true;
HDRCodecInnderNdkSample* hdr_sample = nullptr;
int64_t GetSystemTimeUs()
{
struct timespec now;
(void)clock_gettime(CLOCK_BOOTTIME, &now);
int64_t nanoTime = static_cast<int64_t>(now.tv_sec) * NANOS_IN_SECOND + now.tv_nsec;
return nanoTime / NANOS_IN_MICRO;
}
static void clearIntqueue(std::queue<uint32_t> &q)
{
std::queue<uint32_t> empty;
swap(empty, q);
}
}
HdrDecInnerCallback::HdrDecInnerCallback(std::shared_ptr<InnerSignal> signal) : decInnersignal_(signal) {}
HdrEncInnerCallback::HdrEncInnerCallback(std::shared_ptr<InnerSignal> signal) : encInnersignal_(signal) {}
HDRCodecInnderNdkSample::~HDRCodecInnderNdkSample()
{
Release();
}
void HdrDecInnerCallback::OnOutputFormatChanged(const Format& format)
{
cout << "Format Changed" << endl;
}
void HdrDecInnerCallback::OnInputBufferAvailable(uint32_t index, std::shared_ptr<AVSharedMemory> buffer)
{
if (decInnersignal_ == nullptr) {
cout << "buffer is null 1" << endl;
}
unique_lock<mutex> lock(decInnersignal_->inMutex_);
decInnersignal_->inIdxQueue_.push(index);
decInnersignal_->inBufferQueue_.push(buffer);
decInnersignal_->inCond_.notify_all();
}
void HdrDecInnerCallback::OnOutputBufferAvailable(uint32_t index, AVCodecBufferInfo info, AVCodecBufferFlag flag,
std::shared_ptr<AVSharedMemory> buffer)
{
hdr_sample->vdec_->ReleaseOutputBuffer(index, true);
if (flag & AVCODEC_BUFFER_FLAG_EOS) {
hdr_sample->venc_->NotifyEos();
} else {
hdr_sample->frameCountDec++;
}
}
void HdrDecInnerCallback::OnError(AVCodecErrorType errorType, int32_t errorCode)
{
hdr_sample->errorCount++;
cout << "Dec Error errorCode:" << errorCode << endl;
}
void HdrEncInnerCallback::OnError(AVCodecErrorType errorType, int32_t errorCode)
{
hdr_sample->errorCount++;
cout << "Enc Error errorCode:" << errorCode << endl;
}
void HdrEncInnerCallback::OnOutputFormatChanged(const Format& format)
{
cout << "Format Changed" << endl;
}
void HdrEncInnerCallback::OnInputBufferAvailable(uint32_t index, std::shared_ptr<AVSharedMemory> buffer)
{
(void)index;
(void)buffer;
}
void HdrEncInnerCallback::OnOutputBufferAvailable(uint32_t index, AVCodecBufferInfo info,
AVCodecBufferFlag flag, std::shared_ptr<AVSharedMemory> buffer)
{
hdr_sample->venc_->ReleaseOutputBuffer(index);
if (flag & AVCODEC_BUFFER_FLAG_EOS) {
g_isRunning.store(false);
g_cv.notify_all();
} else {
hdr_sample->frameCountEnc++;
}
}
int32_t HDRCodecInnderNdkSample::CreateCodec()
{
if (signal_ == nullptr) {
signal_ = make_shared<InnerSignal>();
}
if (signal_ == nullptr) {
return AVCS_ERR_UNKNOWN;
}
vdec_ = VideoDecoderFactory::CreateByMime(OH_AVCODEC_MIMETYPE_VIDEO_HEVC);
if (vdec_ == nullptr) {
return AVCS_ERR_UNKNOWN;
}
venc_ = VideoEncoderFactory::CreateByMime(OH_AVCODEC_MIMETYPE_VIDEO_HEVC);
if (venc_ == nullptr) {
return AVCS_ERR_UNKNOWN;
}
hdr_sample = this;
return AVCS_ERR_OK;
}
void HDRCodecInnderNdkSample::FlushBuffer()
{
unique_lock<mutex> decInLock(signal_->inMutex_);
clearIntqueue(signal_->inIdxQueue_);
std::queue<std::shared_ptr<AVSharedMemory>> empty;
swap(empty, signal_->inBufferQueue_);
signal_->inCond_.notify_all();
inFile_->clear();
inFile_->seekg(0, ios::beg);
decInLock.unlock();
}
int32_t HDRCodecInnderNdkSample::RepeatCall()
{
if (REPEAT_START_FLUSH_BEFORE_EOS > 0) {
return RepeatCallStartFlush();
}
if (REPEAT_START_STOP_BEFORE_EOS > 0) {
return RepeatCallStartStop();
}
if (REPEAT_START_FLUSH_STOP_BEFORE_EOS > 0) {
return RepeatCallStartFlushStop();
}
return 0;
}
void HDRCodecInnderNdkSample::InputFunc()
{
bool flag = true;
while (flag) {
if (!g_isRunning.load()) {
flag = false;
break;
}
int32_t ret = RepeatCall();
if (ret != 0) {
cout << "repeat call failed, errcode " << ret << endl;
errorCount++;
g_isRunning.store(false);
g_cv.notify_all();
flag = false;
break;
}
uint32_t index;
unique_lock<mutex> lock(signal_->inMutex_);
signal_->inCond_.wait(lock, [this]() {
if (!g_isRunning.load()) {
return true;
}
return signal_->inIdxQueue_.size() > 0;
});
if (!g_isRunning.load()) {
flag = false;
break;
}
index = signal_->inIdxQueue_.front();
auto buffer = signal_->inBufferQueue_.front();
signal_->inIdxQueue_.pop();
signal_->inBufferQueue_.pop();
lock.unlock();
if (SendData(vdec_, index, buffer) == 1)
break;
}
}
int32_t HDRCodecInnderNdkSample::Configure()
{
Format format;
(void)format.PutIntValue(MediaDescriptionKey::MD_KEY_WIDTH, DEFAULT_WIDTH);
(void)format.PutIntValue(MediaDescriptionKey::MD_KEY_HEIGHT, DEFAULT_HEIGHT);
(void)format.PutIntValue(MediaDescriptionKey::MD_KEY_PIXEL_FORMAT, static_cast<int32_t>(VideoPixelFormat::NV12));
(void)format.PutDoubleValue(MediaDescriptionKey::MD_KEY_FRAME_RATE, DEFAULT_FRAME_RATE);
int ret = vdec_->Configure(format);
if (ret != AVCS_ERR_OK) {
return ret;
}
(void)format.PutIntValue(MediaDescriptionKey::MD_KEY_PROFILE, DEFAULT_PROFILE);
ret = venc_->Configure(format);
if (ret != AVCS_ERR_OK) {
return ret;
}
sptr<Surface> surface = venc_->CreateInputSurface();
if (surface == nullptr) {
cout << "CreateInputSurface is fail" << endl;
return AVCS_ERR_UNKNOWN;
}
window = CreateNativeWindowFromSurface(&surface);
if (window == nullptr) {
cout << "CreateNativeWindowFromSurface is fail" << endl;
return AVCS_ERR_UNKNOWN;
}
ret = vdec_->SetOutputSurface(window->surface);
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = EncSetCallback();
if (ret != AVCS_ERR_OK) {
return ret;
}
return DecSetCallback();
}
int32_t HDRCodecInnderNdkSample::DecSetCallback()
{
if (signal_ == nullptr) {
cout << "Failed to new HdrInnerSignal" << endl;
return AVCS_ERR_UNKNOWN;
}
decCb_ = make_shared<HdrDecInnerCallback>(signal_);
return vdec_->SetCallback(decCb_);
}
int32_t HDRCodecInnderNdkSample::EncSetCallback()
{
if (signal_ == nullptr) {
cout << "Failed to new HdrInnerSignal" << endl;
return AVCS_ERR_UNKNOWN;
}
encCb_ = make_shared<HdrEncInnerCallback>(signal_);
return venc_->SetCallback(encCb_);
}
int32_t HDRCodecInnderNdkSample::ReConfigure()
{
int32_t ret = vdec_->Reset();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = venc_->Reset();
if (ret != AVCS_ERR_OK) {
return ret;
}
FlushBuffer();
Format format;
format.PutIntValue(MediaDescriptionKey::MD_KEY_WIDTH, DEFAULT_WIDTH);
format.PutIntValue(MediaDescriptionKey::MD_KEY_HEIGHT, DEFAULT_HEIGHT);
format.PutIntValue(MediaDescriptionKey::MD_KEY_PIXEL_FORMAT, static_cast<int32_t>(VideoPixelFormat::NV12));
format.PutDoubleValue(MediaDescriptionKey::MD_KEY_FRAME_RATE, DEFAULT_FRAME_RATE);
ret = vdec_->Configure(format);
if (ret != AVCS_ERR_OK) {
return ret;
}
format.PutIntValue(MediaDescriptionKey::MD_KEY_PROFILE, DEFAULT_PROFILE);
ret = venc_->Configure(format);
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->SetOutputSurface(window->surface);
if (ret != AVCS_ERR_OK) {
return ret;
}
return ret;
}
int32_t HDRCodecInnderNdkSample::Start()
{
int32_t ret = 0;
inFile_ = make_unique<ifstream>();
inFile_->open(INP_DIR, ios::in | ios::binary);
if (!inFile_->is_open()) {
(void)vdec_->Release();
(void)venc_->Release();
inFile_->close();
inFile_.reset();
inFile_ = nullptr;
return AVCS_ERR_UNKNOWN;
}
g_isRunning.store(true);
ret = venc_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
inputLoop_ = make_unique<thread>(&HDRCodecInnderNdkSample::InputFunc, this);
if (inputLoop_ == nullptr) {
g_isRunning.store(false);
(void)vdec_->Stop();
ReleaseInFile();
return AVCS_ERR_UNKNOWN;
}
return 0;
}
void HDRCodecInnderNdkSample::StopInloop()
{
if (inputLoop_ != nullptr && inputLoop_->joinable()) {
unique_lock<mutex> lock(signal_->inMutex_);
clearIntqueue(signal_->inIdxQueue_);
g_isRunning.store(false);
signal_->inCond_.notify_all();
lock.unlock();
inputLoop_->join();
inputLoop_.reset();
}
}
void HDRCodecInnderNdkSample::ReleaseInFile()
{
if (inFile_ != nullptr) {
if (inFile_->is_open()) {
inFile_->close();
}
inFile_.reset();
inFile_ = nullptr;
}
}
void HDRCodecInnderNdkSample::WaitForEos()
{
std::mutex mtx;
unique_lock<mutex> lock(mtx);
g_cv.wait(lock, []() {
return !(g_isRunning.load());
});
inputLoop_->join();
vdec_->Stop();
venc_->Stop();
}
int32_t HDRCodecInnderNdkSample::SendData(std::shared_ptr<AVCodecVideoDecoder> codec,
uint32_t index, std::shared_ptr<AVSharedMemory> buffer)
{
uint32_t bufferSize = 0;
int32_t result = 0;
AVCodecBufferInfo info;
AVCodecBufferFlag flag;
static bool isFirstFrame = true;
(void)inFile_->read(reinterpret_cast<char *>(&bufferSize), sizeof(uint32_t));
if (inFile_->eof()) {
flag = AVCODEC_BUFFER_FLAG_EOS;
info.offset = 0;
codec->QueueInputBuffer(index, info, flag);
return 1;
}
if (isFirstFrame) {
flag = AVCODEC_BUFFER_FLAG_CODEC_DATA;
isFirstFrame = false;
} else {
flag = AVCODEC_BUFFER_FLAG_NONE;
}
int32_t size = buffer->GetSize();
uint8_t *avBuffer = buffer->GetBase();
if (avBuffer == nullptr) {
return 0;
}
uint8_t *fileBuffer = new uint8_t[bufferSize];
if (fileBuffer == nullptr) {
cout << "Fatal: no memory" << endl;
delete[] fileBuffer;
return 0;
}
(void)inFile_->read(reinterpret_cast<char *>(fileBuffer), bufferSize);
if (memcpy_s(avBuffer, size, fileBuffer, bufferSize) != EOK) {
delete[] fileBuffer;
cout << "Fatal: memcpy fail" << endl;
return 0;
}
delete[] fileBuffer;
info.presentationTimeUs = GetSystemTimeUs();
info.size = bufferSize;
info.offset = 0;
result = codec->QueueInputBuffer(index, info, flag);
if (result != AVCS_ERR_OK) {
cout << "push input data failed,error:" << result << endl;
}
return 0;
}
int32_t HDRCodecInnderNdkSample::RepeatCallStartFlush()
{
int32_t ret = 0;
REPEAT_START_FLUSH_BEFORE_EOS--;
ret = venc_->Flush();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Flush();
if (ret != AVCS_ERR_OK) {
return ret;
}
FlushBuffer();
ret = venc_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
return 0;
}
int32_t HDRCodecInnderNdkSample::RepeatCallStartStop()
{
int32_t ret = 0;
REPEAT_START_STOP_BEFORE_EOS--;
ret = vdec_->Stop();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = venc_->Stop();
if (ret != AVCS_ERR_OK) {
return ret;
}
FlushBuffer();
ret = venc_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
return 0;
}
int32_t HDRCodecInnderNdkSample::RepeatCallStartFlushStop()
{
int32_t ret = 0;
REPEAT_START_FLUSH_STOP_BEFORE_EOS--;
ret = venc_->Flush();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Flush();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Stop();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = venc_->Stop();
if (ret != AVCS_ERR_OK) {
return ret;
}
FlushBuffer();
ret = venc_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
ret = vdec_->Start();
if (ret != AVCS_ERR_OK) {
return ret;
}
return 0;
}
void HDRCodecInnderNdkSample::Release()
{
if (vdec_ != nullptr) {
vdec_->Release();
vdec_ = nullptr;
}
if (venc_ != nullptr) {
venc_->Release();
venc_ = nullptr;
}
if (signal_ != nullptr) {
signal_ = nullptr;
}
if (hdr_sample != nullptr) {
hdr_sample = nullptr;
}
if (window != nullptr) {
OH_NativeWindow_DestroyNativeWindow(window);
window = nullptr;
}
}

View File

@ -243,6 +243,7 @@ static int32_t RepeatCallStartFlushStop(HDRCodecNdkSample *sample)
HDRCodecNdkSample::~HDRCodecNdkSample()
{
Release();
}
int32_t HDRCodecNdkSample::CreateCodec()
@ -480,6 +481,17 @@ void HDRCodecNdkSample::WaitForEos()
int32_t HDRCodecNdkSample::Release()
{
delete decSignal;
if (decSignal != nullptr) {
delete decSignal;
decSignal = nullptr;
}
if (vdec_ != nullptr) {
OH_VideoDecoder_Destroy(vdec_);
vdec_ = nullptr;
}
if (venc_ != nullptr) {
OH_VideoEncoder_Destroy(venc_);
venc_ = nullptr;
}
return 0;
}