1 /*
2 * Copyright (c) 2025 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 <gtest/gtest.h>
17 #include <cmath>
18 #include <memory>
19 #include "hpae_loudness_gain_node.h"
20 #include "test_case_common.h"
21 #include "audio_errors.h"
22
23 using namespace OHOS;
24 using namespace AudioStandard;
25 using namespace HPAE;
26 using namespace testing::ext;
27 using namespace testing;
28
29 class HpaeLoudnessGainNodeTest : public testing::Test {
30 public:
31 void SetUp();
32 void TearDown();
33 };
34
SetUp()35 void HpaeLoudnessGainNodeTest::SetUp()
36 {}
37
TearDown()38 void HpaeLoudnessGainNodeTest::TearDown()
39 {}
40
41 namespace {
42
43 constexpr uint32_t TEST_ID = 1234;
44 constexpr uint32_t TEST_FRAMELEN = 960;
45 constexpr int TIMES = 5;
46 constexpr float LOUDNESS_GAIN_VALUE = 10.0f;
47
48 HWTEST_F(HpaeLoudnessGainNodeTest, testLoudnessGainNode, TestSize.Level0)
49 {
50 HpaeNodeInfo nodeInfo;
51 nodeInfo.nodeId = TEST_ID;
52 nodeInfo.frameLen = TEST_FRAMELEN;
53 nodeInfo.samplingRate = SAMPLE_RATE_48000;
54 nodeInfo.channels = STEREO;
55 nodeInfo.format = SAMPLE_F32LE;
56 std::shared_ptr<HpaeLoudnessGainNode> hpaeLoudnessGainNode = std::make_shared<HpaeLoudnessGainNode>(nodeInfo);
57
58 std::vector<HpaePcmBuffer*> inputs;
59 PcmBufferInfo pcmBufferInfo(STEREO, TEST_FRAMELEN, SAMPLE_RATE_48000);
60 HpaePcmBuffer hpaePcmBuffer(pcmBufferInfo);
61 inputs.emplace_back(&hpaePcmBuffer);
62 for (int32_t i = 0; i < TIMES; i++) {
63 hpaeLoudnessGainNode->SignalProcess(inputs);
64 }
65 EXPECT_EQ(hpaeLoudnessGainNode->SetLoudnessGain(0.0f), SUCCESS);
66 EXPECT_FLOAT_EQ(hpaeLoudnessGainNode->GetLoudnessGain(), 0.0f);
67 for (int32_t i = 0; i < TIMES; i++) {
68 hpaeLoudnessGainNode->SignalProcess(inputs);
69 }
70 EXPECT_EQ(hpaeLoudnessGainNode->SetLoudnessGain(LOUDNESS_GAIN_VALUE), SUCCESS);
71 EXPECT_FLOAT_EQ(hpaeLoudnessGainNode->GetLoudnessGain(), LOUDNESS_GAIN_VALUE);
72 for (int32_t i = 0; i < TIMES; i++) {
73 hpaeLoudnessGainNode->SignalProcess(inputs);
74 }
75 EXPECT_EQ(hpaeLoudnessGainNode->SetLoudnessGain(0.0f), SUCCESS);
76 EXPECT_FLOAT_EQ(hpaeLoudnessGainNode->GetLoudnessGain(), 0.0f);
77 for (int32_t i = 0; i < TIMES; i++) {
78 hpaeLoudnessGainNode->SignalProcess(inputs);
79 }
80 std::vector<HpaePcmBuffer*> inputs1;
81 PcmBufferInfo pcmBufferInfo1(CHANNEL_6, TEST_FRAMELEN, SAMPLE_RATE_48000);
82 HpaePcmBuffer hpaePcmBuffer1(pcmBufferInfo1);
83 inputs1.emplace_back(&hpaePcmBuffer1);
84 EXPECT_EQ(hpaeLoudnessGainNode->SetLoudnessGain(LOUDNESS_GAIN_VALUE), SUCCESS);
85 for (int32_t i = 0; i < TIMES; i++) {
86 hpaeLoudnessGainNode->SignalProcess(inputs1);
87 }
88 EXPECT_FLOAT_EQ(hpaeLoudnessGainNode->GetLoudnessGain(), LOUDNESS_GAIN_VALUE);
89 }
90
91 }