| 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 // A fake media source that generates video and audio frames to a cast | 5 // A fake media source that generates video and audio frames to a cast |
| 6 // sender. | 6 // sender. |
| 7 // This class can transcode a WebM file using FFmpeg. It can also | 7 // This class can transcode a WebM file using FFmpeg. It can also |
| 8 // generate an animation and audio of fixed frequency. | 8 // generate an animation and audio of fixed frequency. |
| 9 | 9 |
| 10 #ifndef MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ | 10 #ifndef MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ |
| 11 #define MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ | 11 #define MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ |
| 12 | 12 |
| 13 #include <queue> | 13 #include <queue> |
| 14 | 14 |
| 15 #include "base/files/file_path.h" | 15 #include "base/files/file_path.h" |
| 16 #include "base/files/memory_mapped_file.h" | 16 #include "base/files/memory_mapped_file.h" |
| 17 #include "base/memory/ref_counted.h" | 17 #include "base/memory/ref_counted.h" |
| 18 #include "base/memory/weak_ptr.h" | 18 #include "base/memory/weak_ptr.h" |
| 19 #include "base/single_thread_task_runner.h" | 19 #include "base/single_thread_task_runner.h" |
| 20 #include "base/time/tick_clock.h" | 20 #include "base/time/tick_clock.h" |
| 21 #include "media/audio/audio_parameters.h" | 21 #include "media/audio/audio_parameters.h" |
| 22 #include "media/base/audio_converter.h" |
| 22 #include "media/cast/cast_config.h" | 23 #include "media/cast/cast_config.h" |
| 23 #include "media/filters/audio_renderer_algorithm.h" | 24 #include "media/filters/audio_renderer_algorithm.h" |
| 24 #include "media/filters/ffmpeg_demuxer.h" | 25 #include "media/filters/ffmpeg_demuxer.h" |
| 25 | 26 |
| 26 struct AVCodecContext; | 27 struct AVCodecContext; |
| 27 struct AVFormatContext; | 28 struct AVFormatContext; |
| 28 | 29 |
| 29 namespace media { | 30 namespace media { |
| 30 | 31 |
| 31 class AudioBus; | 32 class AudioBus; |
| 33 class AudioConverter; |
| 32 class AudioFifo; | 34 class AudioFifo; |
| 33 class AudioTimestampHelper; | 35 class AudioTimestampHelper; |
| 34 class FFmpegGlue; | 36 class FFmpegGlue; |
| 35 class InMemoryUrlProtocol; | 37 class InMemoryUrlProtocol; |
| 36 class MultiChannelResampler; | |
| 37 | 38 |
| 38 namespace cast { | 39 namespace cast { |
| 39 | 40 |
| 40 class AudioFrameInput; | 41 class AudioFrameInput; |
| 41 class VideoFrameInput; | 42 class VideoFrameInput; |
| 42 class TestAudioBusFactory; | 43 class TestAudioBusFactory; |
| 43 | 44 |
| 44 class FakeMediaSource { | 45 class FakeMediaSource : public media::AudioConverter::InputCallback { |
| 45 public: | 46 public: |
| 46 // |task_runner| is to schedule decoding tasks. | 47 // |task_runner| is to schedule decoding tasks. |
| 47 // |clock| is used by this source but is not owned. | 48 // |clock| is used by this source but is not owned. |
| 49 // |audio_config| is the desired audio config. |
| 48 // |video_config| is the desired video config. | 50 // |video_config| is the desired video config. |
| 49 // |keep_frames| is true if all VideoFrames are saved in a queue. | 51 // |keep_frames| is true if all VideoFrames are saved in a queue. |
| 50 FakeMediaSource(scoped_refptr<base::SingleThreadTaskRunner> task_runner, | 52 FakeMediaSource(scoped_refptr<base::SingleThreadTaskRunner> task_runner, |
| 51 base::TickClock* clock, | 53 base::TickClock* clock, |
| 54 const AudioSenderConfig& audio_config, |
| 52 const VideoSenderConfig& video_config, | 55 const VideoSenderConfig& video_config, |
| 53 bool keep_frames); | 56 bool keep_frames); |
| 54 ~FakeMediaSource(); | 57 ~FakeMediaSource() override; |
| 55 | 58 |
| 56 // Transcode this file as the source of video and audio frames. | 59 // Transcode this file as the source of video and audio frames. |
| 57 // If |override_fps| is non zero then the file is played at the desired rate. | 60 // If |override_fps| is non zero then the file is played at the desired rate. |
| 58 void SetSourceFile(const base::FilePath& video_file, int override_fps); | 61 void SetSourceFile(const base::FilePath& video_file, int override_fps); |
| 59 | 62 |
| 60 // Set to true to randomly change the frame size at random points in time. | 63 // Set to true to randomly change the frame size at random points in time. |
| 61 // Only applies when SetSourceFile() is not used. | 64 // Only applies when SetSourceFile() is not used. |
| 62 void SetVariableFrameSizeMode(bool enabled); | 65 void SetVariableFrameSizeMode(bool enabled); |
| 63 | 66 |
| 64 void Start(scoped_refptr<AudioFrameInput> audio_frame_input, | 67 void Start(scoped_refptr<AudioFrameInput> audio_frame_input, |
| (...skipping 28 matching lines...) Expand all Loading... |
| 93 // Go to the beginning of the stream. | 96 // Go to the beginning of the stream. |
| 94 void Rewind(); | 97 void Rewind(); |
| 95 | 98 |
| 96 // Call FFmpeg to fetch one packet. | 99 // Call FFmpeg to fetch one packet. |
| 97 ScopedAVPacket DemuxOnePacket(bool* audio); | 100 ScopedAVPacket DemuxOnePacket(bool* audio); |
| 98 | 101 |
| 99 void DecodeAudio(ScopedAVPacket packet); | 102 void DecodeAudio(ScopedAVPacket packet); |
| 100 void DecodeVideo(ScopedAVPacket packet); | 103 void DecodeVideo(ScopedAVPacket packet); |
| 101 void Decode(bool decode_audio); | 104 void Decode(bool decode_audio); |
| 102 | 105 |
| 103 void ProvideData(int frame_delay, media::AudioBus* output_bus); | 106 // media::AudioConverter::InputCallback implementation. |
| 107 double ProvideInput(media::AudioBus* output_bus, base::TimeDelta buffer_delay) |
| 108 override; |
| 104 | 109 |
| 105 AVStream* av_audio_stream(); | 110 AVStream* av_audio_stream(); |
| 106 AVStream* av_video_stream(); | 111 AVStream* av_video_stream(); |
| 107 AVCodecContext* av_audio_context(); | 112 AVCodecContext* av_audio_context(); |
| 108 AVCodecContext* av_video_context(); | 113 AVCodecContext* av_video_context(); |
| 109 | 114 |
| 110 const scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 115 const scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 116 const media::AudioParameters output_audio_params_; |
| 111 const VideoSenderConfig video_config_; | 117 const VideoSenderConfig video_config_; |
| 112 const bool keep_frames_; | 118 const bool keep_frames_; |
| 113 bool variable_frame_size_mode_; | 119 bool variable_frame_size_mode_; |
| 114 gfx::Size current_frame_size_; | 120 gfx::Size current_frame_size_; |
| 115 base::TimeTicks next_frame_size_change_time_; | 121 base::TimeTicks next_frame_size_change_time_; |
| 116 scoped_refptr<AudioFrameInput> audio_frame_input_; | 122 scoped_refptr<AudioFrameInput> audio_frame_input_; |
| 117 scoped_refptr<VideoFrameInput> video_frame_input_; | 123 scoped_refptr<VideoFrameInput> video_frame_input_; |
| 118 uint8 synthetic_count_; | 124 uint8 synthetic_count_; |
| 119 base::TickClock* const clock_; // Not owned by this class. | 125 base::TickClock* const clock_; // Not owned by this class. |
| 120 | 126 |
| 121 // Time when the stream starts. | 127 // Time when the stream starts. |
| 122 base::TimeTicks start_time_; | 128 base::TimeTicks start_time_; |
| 123 | 129 |
| 124 // The following three members are used only for fake frames. | 130 // The following three members are used only for fake frames. |
| 125 int audio_frame_count_; // Each audio frame is exactly 10ms. | 131 int audio_frame_count_; // Each audio frame is exactly 10ms. |
| 126 int video_frame_count_; | 132 int video_frame_count_; |
| 127 scoped_ptr<TestAudioBusFactory> audio_bus_factory_; | 133 scoped_ptr<TestAudioBusFactory> audio_bus_factory_; |
| 128 | 134 |
| 129 base::MemoryMappedFile file_data_; | 135 base::MemoryMappedFile file_data_; |
| 130 scoped_ptr<InMemoryUrlProtocol> protocol_; | 136 scoped_ptr<InMemoryUrlProtocol> protocol_; |
| 131 scoped_ptr<FFmpegGlue> glue_; | 137 scoped_ptr<FFmpegGlue> glue_; |
| 132 AVFormatContext* av_format_context_; | 138 AVFormatContext* av_format_context_; |
| 133 | 139 |
| 134 int audio_stream_index_; | 140 int audio_stream_index_; |
| 135 AudioParameters audio_params_; | 141 AudioParameters source_audio_params_; |
| 136 double playback_rate_; | 142 double playback_rate_; |
| 137 | 143 |
| 138 int video_stream_index_; | 144 int video_stream_index_; |
| 139 int video_frame_rate_numerator_; | 145 int video_frame_rate_numerator_; |
| 140 int video_frame_rate_denominator_; | 146 int video_frame_rate_denominator_; |
| 141 | 147 |
| 142 // These are used for audio resampling. | 148 // These are used for audio resampling. |
| 143 scoped_ptr<media::MultiChannelResampler> audio_resampler_; | 149 scoped_ptr<media::AudioConverter> audio_converter_; |
| 144 scoped_ptr<media::AudioFifo> audio_fifo_; | 150 scoped_ptr<media::AudioFifo> audio_fifo_; |
| 145 scoped_ptr<media::AudioBus> audio_fifo_input_bus_; | 151 scoped_ptr<media::AudioBus> audio_fifo_input_bus_; |
| 146 media::AudioRendererAlgorithm audio_algo_; | 152 media::AudioRendererAlgorithm audio_algo_; |
| 147 | 153 |
| 148 // Track the timestamp of audio sent to the receiver. | 154 // Track the timestamp of audio sent to the receiver. |
| 149 scoped_ptr<media::AudioTimestampHelper> audio_sent_ts_; | 155 scoped_ptr<media::AudioTimestampHelper> audio_sent_ts_; |
| 150 | 156 |
| 151 std::queue<scoped_refptr<VideoFrame> > video_frame_queue_; | 157 std::queue<scoped_refptr<VideoFrame> > video_frame_queue_; |
| 152 std::queue<scoped_refptr<VideoFrame> > inserted_video_frame_queue_; | 158 std::queue<scoped_refptr<VideoFrame> > inserted_video_frame_queue_; |
| 153 int64 video_first_pts_; | 159 int64 video_first_pts_; |
| 154 bool video_first_pts_set_; | 160 bool video_first_pts_set_; |
| 155 base::TimeDelta last_video_frame_timestamp_; | 161 base::TimeDelta last_video_frame_timestamp_; |
| 156 | 162 |
| 157 std::queue<AudioBus*> audio_bus_queue_; | 163 std::queue<AudioBus*> audio_bus_queue_; |
| 158 | 164 |
| 159 // NOTE: Weak pointers must be invalidated before all other member variables. | 165 // NOTE: Weak pointers must be invalidated before all other member variables. |
| 160 base::WeakPtrFactory<FakeMediaSource> weak_factory_; | 166 base::WeakPtrFactory<FakeMediaSource> weak_factory_; |
| 161 | 167 |
| 162 DISALLOW_COPY_AND_ASSIGN(FakeMediaSource); | 168 DISALLOW_COPY_AND_ASSIGN(FakeMediaSource); |
| 163 }; | 169 }; |
| 164 | 170 |
| 165 } // namespace cast | 171 } // namespace cast |
| 166 } // namespace media | 172 } // namespace media |
| 167 | 173 |
| 168 #endif // MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ | 174 #endif // MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ |
| OLD | NEW |