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 // A fake media source that generates video and audio frames to a cast 6 // sender. 7 // This class can transcode a WebM file using FFmpeg. It can also 8 // generate an animation and audio of fixed frequency. 9 10 #ifndef MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ 11 #define MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ 12 13 #include <queue> 14 15 #include "base/files/file_path.h" 16 #include "base/files/memory_mapped_file.h" 17 #include "base/memory/ref_counted.h" 18 #include "base/memory/weak_ptr.h" 19 #include "base/single_thread_task_runner.h" 20 #include "base/time/tick_clock.h" 21 #include "media/audio/audio_parameters.h" 22 #include "media/cast/cast_config.h" 23 #include "media/filters/audio_renderer_algorithm.h" 24 #include "media/filters/ffmpeg_demuxer.h" 25 26 struct AVCodecContext; 27 struct AVFormatContext; 28 29 namespace media { 30 31 class AudioBus; 32 class AudioFifo; 33 class AudioTimestampHelper; 34 class FFmpegGlue; 35 class InMemoryUrlProtocol; 36 class MultiChannelResampler; 37 38 namespace cast { 39 40 class AudioFrameInput; 41 class VideoFrameInput; 42 class TestAudioBusFactory; 43 44 class FakeMediaSource { 45 public: 46 // |task_runner| is to schedule decoding tasks. 47 // |clock| is used by this source but is not owned. 48 // |video_config| is the desired video config. 49 FakeMediaSource(scoped_refptr<base::SingleThreadTaskRunner> task_runner, 50 base::TickClock* clock, 51 const VideoSenderConfig& video_config); 52 ~FakeMediaSource(); 53 54 // Transcode this file as the source of video and audio frames. 55 // If |override_fps| is non zero then the file is played at the desired rate. 56 void SetSourceFile(const base::FilePath& video_file, int override_fps); 57 58 void Start(scoped_refptr<AudioFrameInput> audio_frame_input, 59 scoped_refptr<VideoFrameInput> video_frame_input); 60 get_video_config()61 const VideoSenderConfig& get_video_config() const { return video_config_; } 62 63 private: is_transcoding_audio()64 bool is_transcoding_audio() const { return audio_stream_index_ >= 0; } is_transcoding_video()65 bool is_transcoding_video() const { return video_stream_index_ >= 0; } 66 67 void SendNextFrame(); 68 void SendNextFakeFrame(); 69 70 // Return true if a frame was sent. 71 bool SendNextTranscodedVideo(base::TimeDelta elapsed_time); 72 73 // Return true if a frame was sent. 74 bool SendNextTranscodedAudio(base::TimeDelta elapsed_time); 75 76 // Helper methods to compute timestamps for the frame number specified. 77 base::TimeDelta VideoFrameTime(int frame_number); 78 79 base::TimeDelta ScaleTimestamp(base::TimeDelta timestamp); 80 81 base::TimeDelta AudioFrameTime(int frame_number); 82 83 // Go to the beginning of the stream. 84 void Rewind(); 85 86 // Call FFmpeg to fetch one packet. 87 ScopedAVPacket DemuxOnePacket(bool* audio); 88 89 void DecodeAudio(ScopedAVPacket packet); 90 void DecodeVideo(ScopedAVPacket packet); 91 void Decode(bool decode_audio); 92 93 void ProvideData(int frame_delay, media::AudioBus* output_bus); 94 95 AVStream* av_audio_stream(); 96 AVStream* av_video_stream(); 97 AVCodecContext* av_audio_context(); 98 AVCodecContext* av_video_context(); 99 100 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; 101 VideoSenderConfig video_config_; 102 scoped_refptr<AudioFrameInput> audio_frame_input_; 103 scoped_refptr<VideoFrameInput> video_frame_input_; 104 uint8 synthetic_count_; 105 base::TickClock* const clock_; // Not owned by this class. 106 107 // Time when the stream starts. 108 base::TimeTicks start_time_; 109 110 // The following three members are used only for fake frames. 111 int audio_frame_count_; // Each audio frame is exactly 10ms. 112 int video_frame_count_; 113 scoped_ptr<TestAudioBusFactory> audio_bus_factory_; 114 115 // NOTE: Weak pointers must be invalidated before all other member variables. 116 base::WeakPtrFactory<FakeMediaSource> weak_factory_; 117 118 base::MemoryMappedFile file_data_; 119 scoped_ptr<InMemoryUrlProtocol> protocol_; 120 scoped_ptr<FFmpegGlue> glue_; 121 AVFormatContext* av_format_context_; 122 123 int audio_stream_index_; 124 AudioParameters audio_params_; 125 double playback_rate_; 126 127 int video_stream_index_; 128 int video_frame_rate_numerator_; 129 int video_frame_rate_denominator_; 130 131 // These are used for audio resampling. 132 scoped_ptr<media::MultiChannelResampler> audio_resampler_; 133 scoped_ptr<media::AudioFifo> audio_fifo_; 134 scoped_ptr<media::AudioBus> audio_fifo_input_bus_; 135 media::AudioRendererAlgorithm audio_algo_; 136 137 // Track the timestamp of audio sent to the receiver. 138 scoped_ptr<media::AudioTimestampHelper> audio_sent_ts_; 139 140 std::queue<scoped_refptr<VideoFrame> > video_frame_queue_; 141 int64 video_first_pts_; 142 bool video_first_pts_set_; 143 144 std::queue<AudioBus*> audio_bus_queue_; 145 146 DISALLOW_COPY_AND_ASSIGN(FakeMediaSource); 147 }; 148 149 } // namespace cast 150 } // namespace media 151 152 #endif // MEDIA_CAST_TEST_FAKE_MEDIA_SOURCE_H_ 153