121 lines
5.0 KiB
C++
121 lines
5.0 KiB
C++
// Copyright 2017 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "packager/base/logging.h"
|
|
#include "packager/base/strings/string_number_conversions.h"
|
|
#include "packager/media/formats/mp2t/ac3_header.h"
|
|
|
|
namespace {
|
|
|
|
const char kValidPartialAc3Frame[] =
|
|
"0B772770554043E106F575F080821010415C7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF"
|
|
"9F3E7CF9F3EFF9D5F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3"
|
|
"E7CF9F3E7CF9F3E7CF9F3E7CF9F3E3FE757CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F"
|
|
"3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF8CBFC4912248000000000F1B6DB"
|
|
"6DB6DE3C78F1DDDDDDDC00000000000000000000000000EEEEEEF1E3C6DB6DB6DB7CF9AD6B"
|
|
"5AD6B5AD6B5AD6B5AD6B5AD6B4000000078DB6DB6DB6F1E3C78EEEEEEEE000000000000000"
|
|
"0000000000077777778F1E36DB6DB6DBE7CD6B5AD6B5AD6B5AD6B5AD6B5AD6B5A600000000"
|
|
"0003C6DB6DB6DB78F1E3C77777777000000000000000000000000003BBBBBBC78F1B6DB6DB"
|
|
"6DF3E6B5AD6B5AD6B5AD6B5AD6B5AD6B5AD00000001E36DB6DB6DBC78F1E3BBBBBBB800000"
|
|
"000000000000000000001DDDDDDE3C78DB6DB6DB6F9F35AD6B5AD6B5AD6B5AD6B5AD6B5AD6"
|
|
"9800000000000F1B6DB6DB6DE3C78F1DDD";
|
|
|
|
const char kValidPartialAc3FrameSixChannels[] =
|
|
"0B77A3B35E40EBF8403EFF9DF0C3F8430FE1FC155755DF3E7CFA33E7CF9F3E7CF9F3E7CF9F"
|
|
"3ECDFF3ABE7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7CF9F3E7"
|
|
"CF9F3E7CF9F3E7CF9F3E7C31F3E7CF9F3E7C7FCEAF9F3E7CF9F3";
|
|
|
|
} // anonymous namespace
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
namespace mp2t {
|
|
|
|
class Ac3HeaderTest : public testing::Test {
|
|
public:
|
|
void SetUp() override {
|
|
ASSERT_TRUE(base::HexStringToBytes(kValidPartialAc3Frame, &ac3_frame_));
|
|
ASSERT_TRUE(base::HexStringToBytes(kValidPartialAc3FrameSixChannels,
|
|
&ac3_frame_six_channels_));
|
|
}
|
|
|
|
protected:
|
|
std::vector<uint8_t> ac3_frame_;
|
|
std::vector<uint8_t> ac3_frame_six_channels_;
|
|
};
|
|
|
|
TEST_F(Ac3HeaderTest, ParseSuccess) {
|
|
const size_t kExpectedFrameSize(836);
|
|
const size_t kExpectedHeaderSize(0);
|
|
const uint8_t kExpectedObjectType(0);
|
|
const uint32_t kExpectedSamplingFrequency(44100);
|
|
const uint8_t kExpectedNumChannels(2);
|
|
const uint8_t kExpectedAudioSpecificConfig[] = {0x50, 0x11, 0x40};
|
|
|
|
Ac3Header ac3_header;
|
|
ASSERT_TRUE(ac3_header.Parse(ac3_frame_.data(), ac3_frame_.size()));
|
|
EXPECT_EQ(kExpectedFrameSize, ac3_header.GetFrameSize());
|
|
EXPECT_EQ(kExpectedHeaderSize, ac3_header.GetHeaderSize());
|
|
EXPECT_EQ(kExpectedObjectType, ac3_header.GetObjectType());
|
|
EXPECT_EQ(kExpectedSamplingFrequency, ac3_header.GetSamplingFrequency());
|
|
EXPECT_EQ(kExpectedNumChannels, ac3_header.GetNumChannels());
|
|
std::vector<uint8_t> audio_specific_config;
|
|
ac3_header.GetAudioSpecificConfig(&audio_specific_config);
|
|
EXPECT_EQ(arraysize(kExpectedAudioSpecificConfig),
|
|
audio_specific_config.size());
|
|
EXPECT_EQ(std::vector<uint8_t>(std::begin(kExpectedAudioSpecificConfig),
|
|
std::end(kExpectedAudioSpecificConfig)),
|
|
audio_specific_config);
|
|
}
|
|
|
|
TEST_F(Ac3HeaderTest, ParseMultiChannelSuccess) {
|
|
const size_t kExpectedFrameSize(1950);
|
|
const size_t kExpectedHeaderSize(0);
|
|
const uint8_t kExpectedObjectType(0);
|
|
const uint32_t kExpectedSamplingFrequency(44100);
|
|
const uint8_t kExpectedNumChannels(6);
|
|
const uint8_t kExpectedAudioSpecificConfig[] = {0x50, 0x3D, 0xE0};
|
|
|
|
Ac3Header ac3_header;
|
|
ASSERT_TRUE(ac3_header.Parse(ac3_frame_six_channels_.data(),
|
|
ac3_frame_six_channels_.size()));
|
|
EXPECT_EQ(kExpectedFrameSize, ac3_header.GetFrameSize());
|
|
EXPECT_EQ(kExpectedHeaderSize, ac3_header.GetHeaderSize());
|
|
EXPECT_EQ(kExpectedObjectType, ac3_header.GetObjectType());
|
|
EXPECT_EQ(kExpectedSamplingFrequency, ac3_header.GetSamplingFrequency());
|
|
EXPECT_EQ(kExpectedNumChannels, ac3_header.GetNumChannels());
|
|
std::vector<uint8_t> audio_specific_config;
|
|
ac3_header.GetAudioSpecificConfig(&audio_specific_config);
|
|
EXPECT_EQ(arraysize(kExpectedAudioSpecificConfig),
|
|
audio_specific_config.size());
|
|
EXPECT_EQ(std::vector<uint8_t>(std::begin(kExpectedAudioSpecificConfig),
|
|
std::end(kExpectedAudioSpecificConfig)),
|
|
audio_specific_config);
|
|
}
|
|
|
|
TEST_F(Ac3HeaderTest, ParseVariousDataSize) {
|
|
Ac3Header ac3_header;
|
|
|
|
// Parse succeeds as long as the full metadata is provided.
|
|
EXPECT_TRUE(ac3_header.Parse(ac3_frame_.data(), ac3_frame_.size() - 1));
|
|
const size_t frame_size = ac3_header.GetFrameSize();
|
|
const size_t header_size = ac3_header.GetHeaderSize();
|
|
|
|
EXPECT_TRUE(ac3_header.Parse(ac3_frame_.data(), 100));
|
|
EXPECT_EQ(frame_size, ac3_header.GetFrameSize());
|
|
EXPECT_EQ(header_size, ac3_header.GetHeaderSize());
|
|
|
|
// Parse fails if there is not enough data (no full metadata).
|
|
EXPECT_FALSE(ac3_header.Parse(ac3_frame_.data(), 1));
|
|
EXPECT_FALSE(ac3_header.Parse(ac3_frame_.data(), 5));
|
|
}
|
|
|
|
} // Namespace mp2t
|
|
} // namespace media
|
|
} // namespace shaka
|