OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <vector> | 5 #include <vector> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "base/time/time.h" | 9 #include "base/time/time.h" |
10 #include "media/base/buffers.h" | 10 #include "media/base/buffers.h" |
11 #include "media/base/stream_parser_buffer.h" | 11 #include "media/base/stream_parser_buffer.h" |
12 #include "media/formats/mp2t/es_parser_adts.h" | 12 #include "media/formats/mp2t/es_parser_adts.h" |
13 #include "media/formats/mp2t/es_parser_test_base.h" | 13 #include "media/formats/mp2t/es_parser_test_base.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 | 15 |
16 namespace media { | 16 namespace media { |
17 class AudioDecoderConfig; | 17 class AudioDecoderConfig; |
18 | 18 |
19 namespace mp2t { | 19 namespace mp2t { |
20 | 20 |
21 class EsParserAdtsTest : public EsParserTestBase, | 21 class EsParserAdtsTest : public EsParserTestBase, |
22 public testing::Test { | 22 public testing::Test { |
23 public: | 23 public: |
24 EsParserAdtsTest(); | 24 EsParserAdtsTest(); |
25 virtual ~EsParserAdtsTest() {} | 25 virtual ~EsParserAdtsTest() {} |
26 | 26 |
27 protected: | 27 protected: |
28 bool Process(const std::vector<Packet>& pes_packets, bool force_timing); | 28 bool Process(const std::vector<Packet>& pes_packets, bool force_timing); |
29 | 29 |
30 std::vector<Packet> GenerateFixedSizePesPacket(size_t pes_size); | |
31 | |
32 private: | 30 private: |
33 DISALLOW_COPY_AND_ASSIGN(EsParserAdtsTest); | 31 DISALLOW_COPY_AND_ASSIGN(EsParserAdtsTest); |
34 }; | 32 }; |
35 | 33 |
36 EsParserAdtsTest::EsParserAdtsTest() { | 34 EsParserAdtsTest::EsParserAdtsTest() { |
37 } | 35 } |
38 | 36 |
39 bool EsParserAdtsTest::Process( | 37 bool EsParserAdtsTest::Process( |
40 const std::vector<Packet>& pes_packets, | 38 const std::vector<Packet>& pes_packets, |
41 bool force_timing) { | 39 bool force_timing) { |
42 EsParserAdts es_parser( | 40 EsParserAdts es_parser( |
43 base::Bind(&EsParserAdtsTest::NewAudioConfig, base::Unretained(this)), | 41 base::Bind(&EsParserAdtsTest::NewAudioConfig, base::Unretained(this)), |
44 base::Bind(&EsParserAdtsTest::EmitBuffer, base::Unretained(this)), | 42 base::Bind(&EsParserAdtsTest::EmitBuffer, base::Unretained(this)), |
45 false); | 43 false); |
46 return ProcessPesPackets(&es_parser, pes_packets, force_timing); | 44 return ProcessPesPackets(&es_parser, pes_packets, force_timing); |
47 } | 45 } |
48 | 46 |
49 std::vector<EsParserTestBase::Packet> | |
50 EsParserAdtsTest::GenerateFixedSizePesPacket(size_t pes_size) { | |
51 DCHECK_GT(stream_.size(), 0u); | |
52 std::vector<Packet> pes_packets; | |
53 | |
54 Packet cur_pes_packet; | |
55 cur_pes_packet.offset = 0; | |
56 cur_pes_packet.pts = kNoTimestamp(); | |
57 while (cur_pes_packet.offset < stream_.size()) { | |
58 pes_packets.push_back(cur_pes_packet); | |
59 cur_pes_packet.offset += pes_size; | |
60 } | |
61 ComputePacketSize(&pes_packets); | |
62 | |
63 return pes_packets; | |
64 } | |
65 | |
66 TEST_F(EsParserAdtsTest, NoInitialPts) { | 47 TEST_F(EsParserAdtsTest, NoInitialPts) { |
67 LoadStream("bear.adts"); | 48 LoadStream("bear.adts"); |
68 std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512); | 49 std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512); |
69 EXPECT_FALSE(Process(pes_packets, false)); | 50 EXPECT_FALSE(Process(pes_packets, false)); |
70 EXPECT_EQ(0u, buffer_count_); | 51 EXPECT_EQ(0u, buffer_count_); |
71 } | 52 } |
72 | 53 |
73 TEST_F(EsParserAdtsTest, SinglePts) { | 54 TEST_F(EsParserAdtsTest, SinglePts) { |
74 LoadStream("bear.adts"); | 55 LoadStream("bear.adts"); |
75 | 56 |
76 std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512); | 57 std::vector<Packet> pes_packets = GenerateFixedSizePesPacket(512); |
77 pes_packets.front().pts = base::TimeDelta::FromSeconds(10); | 58 pes_packets.front().pts = base::TimeDelta::FromSeconds(10); |
78 | 59 |
79 EXPECT_TRUE(Process(pes_packets, false)); | 60 EXPECT_TRUE(Process(pes_packets, false)); |
80 EXPECT_EQ(1u, config_count_); | 61 EXPECT_EQ(1u, config_count_); |
81 EXPECT_EQ(45u, buffer_count_); | 62 EXPECT_EQ(45u, buffer_count_); |
82 } | 63 } |
83 | 64 |
84 } // namespace mp2t | 65 } // namespace mp2t |
85 } // namespace media | 66 } // namespace media |
86 | 67 |
OLD | NEW |