frame_generator_for_test.cc revision 1320f92c476a1ad9d19dba2a48c72b75566198e9 
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
12namespace chromecast {
13namespace media {
14
15FrameGeneratorForTest::FrameSpec::FrameSpec()
16    : has_config(false),
17      is_eos(false),
18      has_decrypt_config(false),
19      size(0) {
20}
21
22FrameGeneratorForTest::FrameSpec::~FrameSpec() {
23}
24
25FrameGeneratorForTest::FrameGeneratorForTest(
26    const std::vector<FrameSpec> frame_specs)
27  : frame_specs_(frame_specs),
28    frame_idx_(0),
29    total_buffer_size_(0) {
30}
31
32FrameGeneratorForTest::~FrameGeneratorForTest() {
33}
34
35bool FrameGeneratorForTest::HasDecoderConfig() const {
36  if (frame_idx_ >= frame_specs_.size())
37    return false;
38
39  return frame_specs_[frame_idx_].has_config;
40}
41
42scoped_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
107size_t FrameGeneratorForTest::RemainingFrameCount() const {
108  size_t count = frame_specs_.size() - frame_idx_;
109  return count;
110}
111
112}  // namespace media
113}  // namespace chromecast
114