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(VideoCaptureInput* input, | 24 FrameGeneratorCapturer* FrameGeneratorCapturer::Create(size_t width, |
25 size_t width, | |
26 size_t height, | 25 size_t height, |
27 int target_fps, | 26 int target_fps, |
28 Clock* clock) { | 27 Clock* clock) { |
29 FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 28 FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( |
30 clock, input, FrameGenerator::CreateChromaGenerator(width, height), | 29 clock, FrameGenerator::CreateChromaGenerator(width, height), target_fps); |
31 target_fps); | |
32 if (!capturer->Init()) { | 30 if (!capturer->Init()) { |
33 delete capturer; | 31 delete capturer; |
34 return NULL; | 32 return NULL; |
35 } | 33 } |
36 | 34 |
37 return capturer; | 35 return capturer; |
38 } | 36 } |
39 | 37 |
40 FrameGeneratorCapturer* FrameGeneratorCapturer::CreateFromYuvFile( | 38 FrameGeneratorCapturer* FrameGeneratorCapturer::CreateFromYuvFile( |
41 VideoCaptureInput* input, | |
42 const std::string& file_name, | 39 const std::string& file_name, |
43 size_t width, | 40 size_t width, |
44 size_t height, | 41 size_t height, |
45 int target_fps, | 42 int target_fps, |
46 Clock* clock) { | 43 Clock* clock) { |
47 FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( | 44 FrameGeneratorCapturer* capturer = new FrameGeneratorCapturer( |
48 clock, input, | 45 clock, FrameGenerator::CreateFromYuvFile( |
49 FrameGenerator::CreateFromYuvFile(std::vector<std::string>(1, file_name), | 46 std::vector<std::string>(1, file_name), width, height, 1), |
50 width, height, 1), | |
51 target_fps); | 47 target_fps); |
52 if (!capturer->Init()) { | 48 if (!capturer->Init()) { |
53 delete capturer; | 49 delete capturer; |
54 return NULL; | 50 return NULL; |
55 } | 51 } |
56 | 52 |
57 return capturer; | 53 return capturer; |
58 } | 54 } |
59 | 55 |
60 FrameGeneratorCapturer::FrameGeneratorCapturer(Clock* clock, | 56 FrameGeneratorCapturer::FrameGeneratorCapturer(Clock* clock, |
61 VideoCaptureInput* input, | |
62 FrameGenerator* frame_generator, | 57 FrameGenerator* frame_generator, |
63 int target_fps) | 58 int target_fps) |
64 : VideoCapturer(input), | 59 : clock_(clock), |
65 clock_(clock), | |
66 sending_(false), | 60 sending_(false), |
| 61 sink_(nullptr), |
67 tick_(EventTimerWrapper::Create()), | 62 tick_(EventTimerWrapper::Create()), |
68 thread_(FrameGeneratorCapturer::Run, this, "FrameGeneratorCapturer"), | 63 thread_(FrameGeneratorCapturer::Run, this, "FrameGeneratorCapturer"), |
69 frame_generator_(frame_generator), | 64 frame_generator_(frame_generator), |
70 target_fps_(target_fps), | 65 target_fps_(target_fps), |
71 first_frame_capture_time_(-1) { | 66 first_frame_capture_time_(-1) { |
72 assert(input != NULL); | 67 RTC_DCHECK(frame_generator); |
73 assert(frame_generator != NULL); | 68 RTC_DCHECK_GT(target_fps, 0); |
74 assert(target_fps > 0); | |
75 } | 69 } |
76 | 70 |
77 FrameGeneratorCapturer::~FrameGeneratorCapturer() { | 71 FrameGeneratorCapturer::~FrameGeneratorCapturer() { |
78 Stop(); | 72 Stop(); |
79 | 73 |
80 thread_.Stop(); | 74 thread_.Stop(); |
81 } | 75 } |
82 | 76 |
83 void FrameGeneratorCapturer::SetFakeRotation(VideoRotation rotation) { | 77 void FrameGeneratorCapturer::SetFakeRotation(VideoRotation rotation) { |
84 rtc::CritScope cs(&lock_); | 78 rtc::CritScope cs(&lock_); |
(...skipping 21 matching lines...) Expand all Loading... |
106 void FrameGeneratorCapturer::InsertFrame() { | 100 void FrameGeneratorCapturer::InsertFrame() { |
107 { | 101 { |
108 rtc::CritScope cs(&lock_); | 102 rtc::CritScope cs(&lock_); |
109 if (sending_) { | 103 if (sending_) { |
110 VideoFrame* frame = frame_generator_->NextFrame(); | 104 VideoFrame* frame = frame_generator_->NextFrame(); |
111 frame->set_ntp_time_ms(clock_->CurrentNtpInMilliseconds()); | 105 frame->set_ntp_time_ms(clock_->CurrentNtpInMilliseconds()); |
112 frame->set_rotation(fake_rotation_); | 106 frame->set_rotation(fake_rotation_); |
113 if (first_frame_capture_time_ == -1) { | 107 if (first_frame_capture_time_ == -1) { |
114 first_frame_capture_time_ = frame->ntp_time_ms(); | 108 first_frame_capture_time_ = frame->ntp_time_ms(); |
115 } | 109 } |
116 input_->IncomingCapturedFrame(*frame); | 110 if (sink_) |
| 111 sink_->OnFrame(*frame); |
117 } | 112 } |
118 } | 113 } |
119 tick_->Wait(WEBRTC_EVENT_INFINITE); | 114 tick_->Wait(WEBRTC_EVENT_INFINITE); |
120 } | 115 } |
121 | 116 |
122 void FrameGeneratorCapturer::Start() { | 117 void FrameGeneratorCapturer::Start() { |
123 rtc::CritScope cs(&lock_); | 118 rtc::CritScope cs(&lock_); |
124 sending_ = true; | 119 sending_ = true; |
125 } | 120 } |
126 | 121 |
127 void FrameGeneratorCapturer::Stop() { | 122 void FrameGeneratorCapturer::Stop() { |
128 rtc::CritScope cs(&lock_); | 123 rtc::CritScope cs(&lock_); |
129 sending_ = false; | 124 sending_ = false; |
130 } | 125 } |
131 | 126 |
| 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 |
132 void FrameGeneratorCapturer::ForceFrame() { | 142 void FrameGeneratorCapturer::ForceFrame() { |
133 tick_->Set(); | 143 tick_->Set(); |
134 } | 144 } |
135 } // test | 145 } // test |
136 } // webrtc | 146 } // webrtc |
OLD | NEW |