58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
// Copyright (c) 2019 GitHub, Inc.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
#ifndef SHELL_BROWSER_OSR_OSR_VIDEO_CONSUMER_H_
|
|
#define SHELL_BROWSER_OSR_OSR_VIDEO_CONSUMER_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "base/callback.h"
|
|
#include "base/memory/weak_ptr.h"
|
|
#include "components/viz/host/client_frame_sink_video_capturer.h"
|
|
#include "media/capture/mojom/video_capture_types.mojom.h"
|
|
|
|
namespace electron {
|
|
|
|
class OffScreenRenderWidgetHostView;
|
|
|
|
typedef base::RepeatingCallback<void(const gfx::Rect&, const SkBitmap&)>
|
|
OnPaintCallback;
|
|
|
|
class OffScreenVideoConsumer : public viz::mojom::FrameSinkVideoConsumer {
|
|
public:
|
|
OffScreenVideoConsumer(OffScreenRenderWidgetHostView* view,
|
|
OnPaintCallback callback);
|
|
~OffScreenVideoConsumer() override;
|
|
|
|
void SetActive(bool active);
|
|
void SetFrameRate(int frame_rate);
|
|
void SizeChanged();
|
|
|
|
private:
|
|
// viz::mojom::FrameSinkVideoConsumer implementation.
|
|
void OnFrameCaptured(
|
|
base::ReadOnlySharedMemoryRegion data,
|
|
::media::mojom::VideoFrameInfoPtr info,
|
|
const gfx::Rect& content_rect,
|
|
mojo::PendingRemote<viz::mojom::FrameSinkVideoConsumerFrameCallbacks>
|
|
callbacks) override;
|
|
void OnStopped() override;
|
|
void OnLog(const std::string& message) override;
|
|
|
|
bool CheckContentRect(const gfx::Rect& content_rect);
|
|
|
|
OnPaintCallback callback_;
|
|
|
|
OffScreenRenderWidgetHostView* view_;
|
|
std::unique_ptr<viz::ClientFrameSinkVideoCapturer> video_capturer_;
|
|
|
|
base::WeakPtrFactory<OffScreenVideoConsumer> weak_ptr_factory_{this};
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(OffScreenVideoConsumer);
|
|
};
|
|
|
|
} // namespace electron
|
|
|
|
#endif // SHELL_BROWSER_OSR_OSR_VIDEO_CONSUMER_H_
|