blob: 47f1393c3f3819e851710bb84bb281a4ebd40d38 [file] [log] [blame]
/*
* Copyright (c) 2004 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_ENGINE_FAKEWEBRTCVIDEOCAPTUREMODULE_H_
#define MEDIA_ENGINE_FAKEWEBRTCVIDEOCAPTUREMODULE_H_
#include <vector>
#include "api/video/i420_buffer.h"
#include "media/base/testutils.h"
#include "media/engine/webrtcvideocapturer.h"
#include "rtc_base/task_queue_for_test.h"
// Fake class for mocking out webrtc::VideoCaptureModule.
class FakeWebRtcVideoCaptureModule : public webrtc::VideoCaptureModule {
public:
FakeWebRtcVideoCaptureModule()
: callback_(NULL), running_(false) {}
~FakeWebRtcVideoCaptureModule() {}
void RegisterCaptureDataCallback(
rtc::VideoSinkInterface<webrtc::VideoFrame>* callback) override {
callback_ = callback;
}
void DeRegisterCaptureDataCallback() override { callback_ = NULL; }
int32_t StartCapture(const webrtc::VideoCaptureCapability& cap) override {
if (running_)
return -1;
cap_ = cap;
running_ = true;
return 0;
}
int32_t StopCapture() override {
running_ = false;
return 0;
}
const char* CurrentDeviceName() const override {
return NULL; // not implemented
}
bool CaptureStarted() override { return running_; }
int32_t CaptureSettings(webrtc::VideoCaptureCapability& settings) override {
if (!running_)
return -1;
settings = cap_;
return 0;
}
int32_t SetCaptureRotation(webrtc::VideoRotation rotation) override {
return -1; // not implemented
}
bool SetApplyRotation(bool enable) override {
return true; // ignored
}
bool GetApplyRotation() override {
return true; // Rotation compensation is turned on.
}
void SendFrame(int w, int h) {
if (!running_ || !callback_)
return;
task_queue_.SendTask([this, w, h]() {
rtc::scoped_refptr<webrtc::I420Buffer> buffer =
webrtc::I420Buffer::Create(w, h);
// Initialize memory to satisfy DrMemory tests. See
// https://bugs.chromium.org/p/libyuv/issues/detail?id=377
buffer->InitializeData();
callback_->OnFrame(webrtc::VideoFrame(buffer, webrtc::kVideoRotation_0,
0 /* timestamp_us */));
});
}
const webrtc::VideoCaptureCapability& cap() const { return cap_; }
private:
rtc::test::TaskQueueForTest task_queue_{"FakeWebRtcVideoCaptureModule"};
rtc::VideoSinkInterface<webrtc::VideoFrame>* callback_;
bool running_;
webrtc::VideoCaptureCapability cap_;
};
#endif // MEDIA_ENGINE_FAKEWEBRTCVIDEOCAPTUREMODULE_H_