summaryrefslogtreecommitdiff
path: root/chromium/media/base/android/test_data_factory.h
blob: b186d75e27662781babf5e79509238ab0a6a3cd0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_BASE_ANDROID_TEST_DATA_FACTORY_H_
#define MEDIA_BASE_ANDROID_TEST_DATA_FACTORY_H_

#include <stdint.h>
#include <vector>
#include "base/memory/ref_counted.h"
#include "base/time/time.h"
#include "media/base/android/demuxer_stream_player_params.h"

namespace media {

// TestDataFactory is used by MediaCodecDecoder unit test and MediaCodecPlayer
// unit test to simulate the audio or video access unit stream.
class TestDataFactory {
 public:
  // These methods return corresponding demuxer configs.
  static DemuxerConfigs CreateAudioConfigs(AudioCodec audio_codec,
                                           const base::TimeDelta& duration);
  static DemuxerConfigs CreateVideoConfigs(VideoCodec video_codec,
                                           const base::TimeDelta& duration,
                                           const gfx::Size& video_size);

  // Constructor calls |LoadPackets| to load packets from files.
  // Parameters:
  //   file_name_template: the sprintf format string used to generate a file
  //                       name for the packet in the form e.g. "h264-AxB-%d"
  //                       The |%d| will be replaced by 0, 1, 2, 3.
  //   duration: after the last AU exceeds duration the factory generates EOS
  //             unit and stops.
  //   frame_period: PTS increment between units.
  TestDataFactory(const char* file_name_template,
                  const base::TimeDelta& duration,
                  const base::TimeDelta& frame_period);
  virtual ~TestDataFactory();

  // Returns demuxer configuration for this factory.
  virtual DemuxerConfigs GetConfigs() const = 0;

  // Populates next chunk and the corresponding delay and returns true if
  // duration is not exceeded, otherwise returns false.
  // Default implementation repeatedly uses |packet_| array in order 0-1-2-3
  // and monotonically increases timestamps from 0 to |duration_|.
  // The first unit to exceed |duration_| becomes EOS. The delay is set to 0.
  virtual bool CreateChunk(DemuxerData* chunk, base::TimeDelta* delay);

  // In starvation mode we do not add EOS at the end.
  void SetStarvationMode(bool value) { starvation_mode_ = value; }

  base::TimeDelta last_pts() const { return last_pts_; }

 protected:
  // Called by constructor to load packets from files referred by
  // |file_name_template|.
  virtual void LoadPackets(const char* file_name_template);

  // Used to modify the generated access unit by a subclass.
  virtual void ModifyAccessUnit(int index_in_chunk, AccessUnit* unit) {}

  base::TimeDelta duration_;
  base::TimeDelta frame_period_;
  std::vector<uint8_t> packet_[4];
  base::TimeDelta regular_pts_;  // monotonically increasing PTS
  base::TimeDelta last_pts_;     // subclass can modify PTS, maintains the last
  bool starvation_mode_;         // true means no EOS at the end
};

}  // namespace media

#endif  // MEDIA_BASE_ANDROID_TEST_DATA_FACTORY_H_