| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "chromecast/media/cma/base/frame_generator_for_test.h" | |
| 6 | |
| 7 #include "chromecast/media/cma/base/decoder_buffer_adapter.h" | |
| 8 #include "chromecast/media/cma/base/decoder_buffer_base.h" | |
| 9 #include "media/base/decoder_buffer.h" | |
| 10 #include "media/base/decrypt_config.h" | |
| 11 | |
| 12 namespace chromecast { | |
| 13 namespace media { | |
| 14 | |
| 15 FrameGeneratorForTest::FrameSpec::FrameSpec() | |
| 16 : has_config(false), | |
| 17 is_eos(false), | |
| 18 has_decrypt_config(false), | |
| 19 size(0) { | |
| 20 } | |
| 21 | |
| 22 FrameGeneratorForTest::FrameSpec::~FrameSpec() { | |
| 23 } | |
| 24 | |
| 25 FrameGeneratorForTest::FrameGeneratorForTest( | |
| 26 const std::vector<FrameSpec> frame_specs) | |
| 27 : frame_specs_(frame_specs), | |
| 28 frame_idx_(0), | |
| 29 total_buffer_size_(0) { | |
| 30 } | |
| 31 | |
| 32 FrameGeneratorForTest::~FrameGeneratorForTest() { | |
| 33 } | |
| 34 | |
| 35 bool FrameGeneratorForTest::HasDecoderConfig() const { | |
| 36 if (frame_idx_ >= frame_specs_.size()) | |
| 37 return false; | |
| 38 | |
| 39 return frame_specs_[frame_idx_].has_config; | |
| 40 } | |
| 41 | |
| 42 scoped_refptr<DecoderBufferBase> FrameGeneratorForTest::Generate() { | |
| 43 if (frame_idx_ >= frame_specs_.size()) | |
| 44 return scoped_refptr<DecoderBufferBase>(); | |
| 45 | |
| 46 const FrameSpec& frame_spec = frame_specs_[frame_idx_]; | |
| 47 frame_idx_++; | |
| 48 | |
| 49 if (frame_spec.is_eos) { | |
| 50 return scoped_refptr<DecoderBufferBase>( | |
| 51 new DecoderBufferAdapter(::media::DecoderBuffer::CreateEOSBuffer())); | |
| 52 } | |
| 53 | |
| 54 scoped_refptr< ::media::DecoderBuffer> buffer( | |
| 55 new ::media::DecoderBuffer(frame_spec.size)); | |
| 56 | |
| 57 // Timestamp. | |
| 58 buffer->set_timestamp(frame_spec.timestamp); | |
| 59 | |
| 60 // Generate the frame data. | |
| 61 for (size_t k = 0; k < frame_spec.size; k++) { | |
| 62 buffer->writable_data()[k] = total_buffer_size_ & 0xff; | |
| 63 total_buffer_size_++; | |
| 64 } | |
| 65 | |
| 66 // Generate the decrypt configuration. | |
| 67 if (frame_spec.has_decrypt_config) { | |
| 68 uint32 frame_size = buffer->data_size(); | |
| 69 uint32 chunk_size = 1; | |
| 70 std::vector< ::media::SubsampleEntry> subsamples; | |
| 71 while (frame_size > 0) { | |
| 72 ::media::SubsampleEntry subsample; | |
| 73 subsample.clear_bytes = chunk_size; | |
| 74 if (subsample.clear_bytes > frame_size) | |
| 75 subsample.clear_bytes = frame_size; | |
| 76 frame_size -= subsample.clear_bytes; | |
| 77 chunk_size <<= 1; | |
| 78 | |
| 79 subsample.cypher_bytes = chunk_size; | |
| 80 if (subsample.cypher_bytes > frame_size) | |
| 81 subsample.cypher_bytes = frame_size; | |
| 82 frame_size -= subsample.cypher_bytes; | |
| 83 chunk_size <<= 1; | |
| 84 | |
| 85 subsamples.push_back(subsample); | |
| 86 } | |
| 87 | |
| 88 char key_id[] = { | |
| 89 0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, | |
| 90 0x8, 0x9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf }; | |
| 91 | |
| 92 char iv[] = { | |
| 93 0x0, 0x2, 0x1, 0x3, 0x5, 0x4, 0x7, 0x6, | |
| 94 0x9, 0x8, 0xb, 0xa, 0xd, 0xc, 0xf, 0xe }; | |
| 95 | |
| 96 scoped_ptr< ::media::DecryptConfig> decrypt_config( | |
| 97 new ::media::DecryptConfig( | |
| 98 std::string(key_id, arraysize(key_id)), | |
| 99 std::string(iv, arraysize(iv)), | |
| 100 subsamples)); | |
| 101 buffer->set_decrypt_config(decrypt_config.Pass()); | |
| 102 } | |
| 103 | |
| 104 return scoped_refptr<DecoderBufferBase>(new DecoderBufferAdapter(buffer)); | |
| 105 } | |
| 106 | |
| 107 size_t FrameGeneratorForTest::RemainingFrameCount() const { | |
| 108 size_t count = frame_specs_.size() - frame_idx_; | |
| 109 return count; | |
| 110 } | |
| 111 | |
| 112 } // namespace media | |
| 113 } // namespace chromecast | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 609383004:
Chromecast: adds test frame-segmenter code to support CMA unit tests. (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@cma-decrypt-context