| 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 #include "webrtc/test/frame_generator.h" | 10 #include "webrtc/test/frame_generator.h" |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 57 // Ensure stride == width. | 57 // Ensure stride == width. |
| 58 rtc::scoped_refptr<I420Buffer> buffer( | 58 rtc::scoped_refptr<I420Buffer> buffer( |
| 59 I420Buffer::Create(width_, height_, width_, half_width_, half_width_)); | 59 I420Buffer::Create(width_, height_, width_, half_width_, half_width_)); |
| 60 memset(buffer->MutableDataY(), 127, y_size_); | 60 memset(buffer->MutableDataY(), 127, y_size_); |
| 61 memset(buffer->MutableDataU(), 127, uv_size_); | 61 memset(buffer->MutableDataU(), 127, uv_size_); |
| 62 memset(buffer->MutableDataV(), 127, uv_size_); | 62 memset(buffer->MutableDataV(), 127, uv_size_); |
| 63 | 63 |
| 64 for (const auto& square : squares_) | 64 for (const auto& square : squares_) |
| 65 square->Draw(buffer); | 65 square->Draw(buffer); |
| 66 | 66 |
| 67 frame_.reset(new VideoFrame(buffer, 0, 0, webrtc::kVideoRotation_0)); | 67 frame_.reset(new VideoFrame(buffer, 0, 0, webrtc::kVideoRotation_0, |
| 68 webrtc::VideoContentType::kDefault)); |
| 68 return frame_.get(); | 69 return frame_.get(); |
| 69 } | 70 } |
| 70 | 71 |
| 71 private: | 72 private: |
| 72 class Square { | 73 class Square { |
| 73 public: | 74 public: |
| 74 Square(int width, int height, int seed) | 75 Square(int width, int height, int seed) |
| 75 : random_generator_(seed), | 76 : random_generator_(seed), |
| 76 x_(random_generator_.Rand(0, width)), | 77 x_(random_generator_.Rand(0, width)), |
| 77 y_(random_generator_.Rand(0, height)), | 78 y_(random_generator_.Rand(0, height)), |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 144 for (FILE* file : files_) | 145 for (FILE* file : files_) |
| 145 fclose(file); | 146 fclose(file); |
| 146 } | 147 } |
| 147 | 148 |
| 148 VideoFrame* NextFrame() override { | 149 VideoFrame* NextFrame() override { |
| 149 if (current_display_count_ == 0) | 150 if (current_display_count_ == 0) |
| 150 ReadNextFrame(); | 151 ReadNextFrame(); |
| 151 if (++current_display_count_ >= frame_display_count_) | 152 if (++current_display_count_ >= frame_display_count_) |
| 152 current_display_count_ = 0; | 153 current_display_count_ = 0; |
| 153 | 154 |
| 154 temp_frame_.reset( | 155 temp_frame_.reset(new VideoFrame(last_read_buffer_, 0, 0, |
| 155 new VideoFrame(last_read_buffer_, 0, 0, webrtc::kVideoRotation_0)); | 156 webrtc::kVideoRotation_0, |
| 157 webrtc::VideoContentType::kDefault)); |
| 156 return temp_frame_.get(); | 158 return temp_frame_.get(); |
| 157 } | 159 } |
| 158 | 160 |
| 159 void ReadNextFrame() { | 161 void ReadNextFrame() { |
| 160 last_read_buffer_ = | 162 last_read_buffer_ = |
| 161 test::ReadI420Buffer(static_cast<int>(width_), | 163 test::ReadI420Buffer(static_cast<int>(width_), |
| 162 static_cast<int>(height_), | 164 static_cast<int>(height_), |
| 163 files_[file_index_]); | 165 files_[file_index_]); |
| 164 if (!last_read_buffer_) { | 166 if (!last_read_buffer_) { |
| 165 // No more frames to read in this file, rewind and move to next file. | 167 // No more frames to read in this file, rewind and move to next file. |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 260 (pixels_scrolled_y / 2)) + | 262 (pixels_scrolled_y / 2)) + |
| 261 (pixels_scrolled_x / 2); | 263 (pixels_scrolled_x / 2); |
| 262 int offset_v = (current_source_frame_->video_frame_buffer()->StrideV() * | 264 int offset_v = (current_source_frame_->video_frame_buffer()->StrideV() * |
| 263 (pixels_scrolled_y / 2)) + | 265 (pixels_scrolled_y / 2)) + |
| 264 (pixels_scrolled_x / 2); | 266 (pixels_scrolled_x / 2); |
| 265 | 267 |
| 266 rtc::scoped_refptr<VideoFrameBuffer> frame_buffer( | 268 rtc::scoped_refptr<VideoFrameBuffer> frame_buffer( |
| 267 current_source_frame_->video_frame_buffer()); | 269 current_source_frame_->video_frame_buffer()); |
| 268 current_frame_ = rtc::Optional<webrtc::VideoFrame>(webrtc::VideoFrame( | 270 current_frame_ = rtc::Optional<webrtc::VideoFrame>(webrtc::VideoFrame( |
| 269 new rtc::RefCountedObject<webrtc::WrappedI420Buffer>( | 271 new rtc::RefCountedObject<webrtc::WrappedI420Buffer>( |
| 270 target_width_, target_height_, | 272 target_width_, target_height_, &frame_buffer->DataY()[offset_y], |
| 271 &frame_buffer->DataY()[offset_y], frame_buffer->StrideY(), | 273 frame_buffer->StrideY(), &frame_buffer->DataU()[offset_u], |
| 272 &frame_buffer->DataU()[offset_u], frame_buffer->StrideU(), | 274 frame_buffer->StrideU(), &frame_buffer->DataV()[offset_v], |
| 273 &frame_buffer->DataV()[offset_v], frame_buffer->StrideV(), | 275 frame_buffer->StrideV(), KeepRefUntilDone(frame_buffer)), |
| 274 KeepRefUntilDone(frame_buffer)), | 276 kVideoRotation_0, webrtc::VideoContentType::kDefault, 0)); |
| 275 kVideoRotation_0, 0)); | |
| 276 } | 277 } |
| 277 | 278 |
| 278 Clock* const clock_; | 279 Clock* const clock_; |
| 279 const int64_t start_time_; | 280 const int64_t start_time_; |
| 280 const int64_t scroll_time_; | 281 const int64_t scroll_time_; |
| 281 const int64_t pause_time_; | 282 const int64_t pause_time_; |
| 282 const size_t num_frames_; | 283 const size_t num_frames_; |
| 283 const int target_width_; | 284 const int target_width_; |
| 284 const int target_height_; | 285 const int target_height_; |
| 285 | 286 |
| (...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 365 files.push_back(file); | 366 files.push_back(file); |
| 366 } | 367 } |
| 367 | 368 |
| 368 return std::unique_ptr<FrameGenerator>(new ScrollingImageFrameGenerator( | 369 return std::unique_ptr<FrameGenerator>(new ScrollingImageFrameGenerator( |
| 369 clock, files, source_width, source_height, target_width, target_height, | 370 clock, files, source_width, source_height, target_width, target_height, |
| 370 scroll_time_ms, pause_time_ms)); | 371 scroll_time_ms, pause_time_ms)); |
| 371 } | 372 } |
| 372 | 373 |
| 373 } // namespace test | 374 } // namespace test |
| 374 } // namespace webrtc | 375 } // namespace webrtc |
| OLD | NEW |