| OLD | NEW | 
|---|
| 1 /* | 1 /* | 
| 2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 
| 3  * | 3  * | 
| 4  *  Use of this source code is governed by a BSD-style license | 4  *  Use of this source code is governed by a BSD-style license | 
| 5  *  that can be found in the LICENSE file in the root of the source | 5  *  that can be found in the LICENSE file in the root of the source | 
| 6  *  tree. An additional intellectual property rights grant can be found | 6  *  tree. An additional intellectual property rights grant can be found | 
| 7  *  in the file PATENTS.  All contributing project authors may | 7  *  in the file PATENTS.  All contributing project authors may | 
| 8  *  be found in the AUTHORS file in the root of the source tree. | 8  *  be found in the AUTHORS file in the root of the source tree. | 
| 9  */ | 9  */ | 
| 10 | 10 | 
| 11 #include "webrtc/test/frame_generator_capturer.h" | 11 #include "webrtc/test/frame_generator_capturer.h" | 
| 12 | 12 | 
| 13 #include "webrtc/base/criticalsection.h" | 13 #include "webrtc/base/criticalsection.h" | 
| 14 #include "webrtc/base/platform_thread.h" | 14 #include "webrtc/base/platform_thread.h" | 
| 15 #include "webrtc/system_wrappers/include/clock.h" | 15 #include "webrtc/system_wrappers/include/clock.h" | 
| 16 #include "webrtc/system_wrappers/include/event_wrapper.h" | 16 #include "webrtc/system_wrappers/include/event_wrapper.h" | 
| 17 #include "webrtc/system_wrappers/include/sleep.h" | 17 #include "webrtc/system_wrappers/include/sleep.h" | 
| 18 #include "webrtc/test/frame_generator.h" | 18 #include "webrtc/test/frame_generator.h" | 
| 19 #include "webrtc/video_send_stream.h" | 19 #include "webrtc/video_send_stream.h" | 
| 20 | 20 | 
| 21 namespace webrtc { | 21 namespace webrtc { | 
| 22 namespace test { | 22 namespace test { | 
| 23 | 23 | 
| 24 FrameGeneratorCapturer* FrameGeneratorCapturer::Create(size_t width, | 24 FrameGeneratorCapturer* FrameGeneratorCapturer::Create(VideoCaptureInput* input, | 
|  | 25                                                        size_t width, | 
| 25                                                        size_t height, | 26                                                        size_t height, | 
| 26                                                        int target_fps, | 27                                                        int target_fps, | 
| 27                                                        Clock* clock) { | 28                                                        Clock* clock) { | 
| 28   FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 29   FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 
| 29       clock, FrameGenerator::CreateChromaGenerator(width, height), target_fps); | 30       clock, input, FrameGenerator::CreateChromaGenerator(width, height), | 
|  | 31       target_fps); | 
| 30   if (!capturer->Init()) { | 32   if (!capturer->Init()) { | 
| 31     delete capturer; | 33     delete capturer; | 
| 32     return NULL; | 34     return NULL; | 
| 33   } | 35   } | 
| 34 | 36 | 
| 35   return capturer; | 37   return capturer; | 
| 36 } | 38 } | 
| 37 | 39 | 
| 38 FrameGeneratorCapturer* FrameGeneratorCapturer::CreateFromYuvFile( | 40 FrameGeneratorCapturer* FrameGeneratorCapturer::CreateFromYuvFile( | 
|  | 41     VideoCaptureInput* input, | 
| 39     const std::string& file_name, | 42     const std::string& file_name, | 
| 40     size_t width, | 43     size_t width, | 
| 41     size_t height, | 44     size_t height, | 
| 42     int target_fps, | 45     int target_fps, | 
| 43     Clock* clock) { | 46     Clock* clock) { | 
| 44   FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 47   FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 
| 45       clock, FrameGenerator::CreateFromYuvFile( | 48       clock, input, | 
| 46                  std::vector<std::string>(1, file_name), width, height, 1), | 49       FrameGenerator::CreateFromYuvFile(std::vector<std::string>(1, file_name), | 
|  | 50                                         width, height, 1), | 
| 47       target_fps); | 51       target_fps); | 
| 48   if (!capturer->Init()) { | 52   if (!capturer->Init()) { | 
| 49     delete capturer; | 53     delete capturer; | 
| 50     return NULL; | 54     return NULL; | 
| 51   } | 55   } | 
| 52 | 56 | 
| 53   return capturer; | 57   return capturer; | 
| 54 } | 58 } | 
| 55 | 59 | 
| 56 FrameGeneratorCapturer::FrameGeneratorCapturer(Clock* clock, | 60 FrameGeneratorCapturer::FrameGeneratorCapturer(Clock* clock, | 
|  | 61                                                VideoCaptureInput* input, | 
| 57                                                FrameGenerator* frame_generator, | 62                                                FrameGenerator* frame_generator, | 
| 58                                                int target_fps) | 63                                                int target_fps) | 
| 59     : clock_(clock), | 64     : VideoCapturer(input), | 
|  | 65       clock_(clock), | 
| 60       sending_(false), | 66       sending_(false), | 
| 61       sink_(nullptr), |  | 
| 62       tick_(EventTimerWrapper::Create()), | 67       tick_(EventTimerWrapper::Create()), | 
| 63       thread_(FrameGeneratorCapturer::Run, this, "FrameGeneratorCapturer"), | 68       thread_(FrameGeneratorCapturer::Run, this, "FrameGeneratorCapturer"), | 
| 64       frame_generator_(frame_generator), | 69       frame_generator_(frame_generator), | 
| 65       target_fps_(target_fps), | 70       target_fps_(target_fps), | 
| 66       first_frame_capture_time_(-1) { | 71       first_frame_capture_time_(-1) { | 
| 67   RTC_DCHECK(frame_generator); | 72   assert(input != NULL); | 
| 68   RTC_DCHECK_GT(target_fps, 0); | 73   assert(frame_generator != NULL); | 
|  | 74   assert(target_fps > 0); | 
| 69 } | 75 } | 
| 70 | 76 | 
| 71 FrameGeneratorCapturer::~FrameGeneratorCapturer() { | 77 FrameGeneratorCapturer::~FrameGeneratorCapturer() { | 
| 72   Stop(); | 78   Stop(); | 
| 73 | 79 | 
| 74   thread_.Stop(); | 80   thread_.Stop(); | 
| 75 } | 81 } | 
| 76 | 82 | 
| 77 void FrameGeneratorCapturer::SetFakeRotation(VideoRotation rotation) { | 83 void FrameGeneratorCapturer::SetFakeRotation(VideoRotation rotation) { | 
| 78   rtc::CritScope cs(&lock_); | 84   rtc::CritScope cs(&lock_); | 
| (...skipping 21 matching lines...) Expand all  Loading... | 
| 100 void FrameGeneratorCapturer::InsertFrame() { | 106 void FrameGeneratorCapturer::InsertFrame() { | 
| 101   { | 107   { | 
| 102     rtc::CritScope cs(&lock_); | 108     rtc::CritScope cs(&lock_); | 
| 103     if (sending_) { | 109     if (sending_) { | 
| 104       VideoFrame* frame = frame_generator_->NextFrame(); | 110       VideoFrame* frame = frame_generator_->NextFrame(); | 
| 105       frame->set_ntp_time_ms(clock_->CurrentNtpInMilliseconds()); | 111       frame->set_ntp_time_ms(clock_->CurrentNtpInMilliseconds()); | 
| 106       frame->set_rotation(fake_rotation_); | 112       frame->set_rotation(fake_rotation_); | 
| 107       if (first_frame_capture_time_ == -1) { | 113       if (first_frame_capture_time_ == -1) { | 
| 108         first_frame_capture_time_ = frame->ntp_time_ms(); | 114         first_frame_capture_time_ = frame->ntp_time_ms(); | 
| 109       } | 115       } | 
| 110       if (sink_) | 116       input_->IncomingCapturedFrame(*frame); | 
| 111         sink_->OnFrame(*frame); |  | 
| 112     } | 117     } | 
| 113   } | 118   } | 
| 114   tick_->Wait(WEBRTC_EVENT_INFINITE); | 119   tick_->Wait(WEBRTC_EVENT_INFINITE); | 
| 115 } | 120 } | 
| 116 | 121 | 
| 117 void FrameGeneratorCapturer::Start() { | 122 void FrameGeneratorCapturer::Start() { | 
| 118   rtc::CritScope cs(&lock_); | 123   rtc::CritScope cs(&lock_); | 
| 119   sending_ = true; | 124   sending_ = true; | 
| 120 } | 125 } | 
| 121 | 126 | 
| 122 void FrameGeneratorCapturer::Stop() { | 127 void FrameGeneratorCapturer::Stop() { | 
| 123   rtc::CritScope cs(&lock_); | 128   rtc::CritScope cs(&lock_); | 
| 124   sending_ = false; | 129   sending_ = false; | 
| 125 } | 130 } | 
| 126 | 131 | 
| 127 void FrameGeneratorCapturer::AddOrUpdateSink( |  | 
| 128     rtc::VideoSinkInterface<VideoFrame>* sink, |  | 
| 129     const rtc::VideoSinkWants& wants) { |  | 
| 130   rtc::CritScope cs(&lock_); |  | 
| 131   RTC_CHECK(!sink_); |  | 
| 132   sink_ = sink; |  | 
| 133 } |  | 
| 134 |  | 
| 135 void FrameGeneratorCapturer::RemoveSink( |  | 
| 136     rtc::VideoSinkInterface<VideoFrame>* sink) { |  | 
| 137   rtc::CritScope cs(&lock_); |  | 
| 138   RTC_CHECK(sink_ == sink); |  | 
| 139   sink_ = nullptr; |  | 
| 140 } |  | 
| 141 |  | 
| 142 void FrameGeneratorCapturer::ForceFrame() { | 132 void FrameGeneratorCapturer::ForceFrame() { | 
| 143   tick_->Set(); | 133   tick_->Set(); | 
| 144 } | 134 } | 
| 145 }  // test | 135 }  // test | 
| 146 }  // webrtc | 136 }  // webrtc | 
| OLD | NEW | 
|---|