| /* |
| * Copyright (c) 2011 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #ifndef MEDIA_BASE_FAKE_VIDEO_RENDERER_H_ |
| #define MEDIA_BASE_FAKE_VIDEO_RENDERER_H_ |
| |
| #include <stdint.h> |
| |
| #include "api/video/video_frame.h" |
| #include "api/video/video_frame_buffer.h" |
| #include "api/video/video_rotation.h" |
| #include "api/video/video_sink_interface.h" |
| #include "rtc_base/critical_section.h" |
| #include "rtc_base/scoped_ref_ptr.h" |
| |
| namespace cricket { |
| |
| // Faked video renderer that has a callback for actions on rendering. |
| class FakeVideoRenderer : public rtc::VideoSinkInterface<webrtc::VideoFrame> { |
| public: |
| FakeVideoRenderer(); |
| |
| void OnFrame(const webrtc::VideoFrame& frame) override; |
| |
| int errors() const { return errors_; } |
| int width() const { |
| rtc::CritScope cs(&crit_); |
| return width_; |
| } |
| int height() const { |
| rtc::CritScope cs(&crit_); |
| return height_; |
| } |
| webrtc::VideoRotation rotation() const { |
| rtc::CritScope cs(&crit_); |
| return rotation_; |
| } |
| |
| int64_t timestamp_us() const { |
| rtc::CritScope cs(&crit_); |
| return timestamp_us_; |
| } |
| int num_rendered_frames() const { |
| rtc::CritScope cs(&crit_); |
| return num_rendered_frames_; |
| } |
| bool black_frame() const { |
| rtc::CritScope cs(&crit_); |
| return black_frame_; |
| } |
| |
| private: |
| static bool CheckFrameColorYuv(uint8_t y_min, |
| uint8_t y_max, |
| uint8_t u_min, |
| uint8_t u_max, |
| uint8_t v_min, |
| uint8_t v_max, |
| const webrtc::VideoFrame* frame) { |
| if (!frame || !frame->video_frame_buffer()) { |
| return false; |
| } |
| rtc::scoped_refptr<const webrtc::I420BufferInterface> i420_buffer = |
| frame->video_frame_buffer()->ToI420(); |
| // Y |
| int y_width = frame->width(); |
| int y_height = frame->height(); |
| const uint8_t* y_plane = i420_buffer->DataY(); |
| const uint8_t* y_pos = y_plane; |
| int32_t y_pitch = i420_buffer->StrideY(); |
| for (int i = 0; i < y_height; ++i) { |
| for (int j = 0; j < y_width; ++j) { |
| uint8_t y_value = *(y_pos + j); |
| if (y_value < y_min || y_value > y_max) { |
| return false; |
| } |
| } |
| y_pos += y_pitch; |
| } |
| // U and V |
| int chroma_width = i420_buffer->ChromaWidth(); |
| int chroma_height = i420_buffer->ChromaHeight(); |
| const uint8_t* u_plane = i420_buffer->DataU(); |
| const uint8_t* v_plane = i420_buffer->DataV(); |
| const uint8_t* u_pos = u_plane; |
| const uint8_t* v_pos = v_plane; |
| int32_t u_pitch = i420_buffer->StrideU(); |
| int32_t v_pitch = i420_buffer->StrideV(); |
| for (int i = 0; i < chroma_height; ++i) { |
| for (int j = 0; j < chroma_width; ++j) { |
| uint8_t u_value = *(u_pos + j); |
| if (u_value < u_min || u_value > u_max) { |
| return false; |
| } |
| uint8_t v_value = *(v_pos + j); |
| if (v_value < v_min || v_value > v_max) { |
| return false; |
| } |
| } |
| u_pos += u_pitch; |
| v_pos += v_pitch; |
| } |
| return true; |
| } |
| |
| int errors_ = 0; |
| int width_ = 0; |
| int height_ = 0; |
| webrtc::VideoRotation rotation_ = webrtc::kVideoRotation_0; |
| int64_t timestamp_us_ = 0; |
| int num_rendered_frames_ = 0; |
| bool black_frame_ = false; |
| rtc::CriticalSection crit_; |
| }; |
| |
| } // namespace cricket |
| |
| #endif // MEDIA_BASE_FAKE_VIDEO_RENDERER_H_ |