Bug: webrtc:5584 Change-Id: I2fd1395d35596d9002e19cc90fcda3a5d4cde9e7 Reviewed-on: https://webrtc-review.googlesource.com/16564 Commit-Queue: Steve Anton <steveanton@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#20504}
88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
/*
|
|
* 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"
|
|
|
|
class FakeWebRtcVcmFactory;
|
|
|
|
// Fake class for mocking out webrtc::VideoCaptureModule.
|
|
class FakeWebRtcVideoCaptureModule : public webrtc::VideoCaptureModule {
|
|
public:
|
|
explicit FakeWebRtcVideoCaptureModule(FakeWebRtcVcmFactory* factory)
|
|
: factory_(factory), 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_) return;
|
|
|
|
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();
|
|
if (callback_) {
|
|
callback_->OnFrame(
|
|
webrtc::VideoFrame(buffer, 0, 0, webrtc::kVideoRotation_0));
|
|
}
|
|
}
|
|
|
|
const webrtc::VideoCaptureCapability& cap() const {
|
|
return cap_;
|
|
}
|
|
|
|
private:
|
|
FakeWebRtcVcmFactory* factory_;
|
|
rtc::VideoSinkInterface<webrtc::VideoFrame>* callback_;
|
|
bool running_;
|
|
webrtc::VideoCaptureCapability cap_;
|
|
};
|
|
|
|
#endif // MEDIA_ENGINE_FAKEWEBRTCVIDEOCAPTUREMODULE_H_
|