public viz::mojom::FrameSinkVideoConsumer {

public:

using FrameCaptureCallback =

base::RepeatingCallback<void(const gfx::Image&, const gfx::Rect&)>;

FrameSubscriber(content::WebContents* web_contents,

const FrameCaptureCallback& callback,

bool only_dirty);

~FrameSubscriber() override;

// disable copy

FrameSubscriber(const FrameSubscriber&) = delete;

FrameSubscriber& operator=(const FrameSubscriber&) = delete;

private:

void AttachToHost(content::RenderWidgetHost* host);

void DetachFromHost();

void RenderFrameCreated(content::RenderFrameHost* render_frame_host) override;

void RenderViewDeleted(content::RenderViewHost* host) override;

void RenderViewHostChanged(content::RenderViewHost* old_host,

content::RenderViewHost* new_host) override;

// viz::mojom::FrameSinkVideoConsumer implementation.

void OnFrameCaptured(

::media::mojom::VideoBufferHandlePtr data,

::media::mojom::VideoFrameInfoPtr info,

const gfx::Rect& content_rect,

mojo::PendingRemote<viz::mojom::FrameSinkVideoConsumerFrameCallbacks>

callbacks) override;

void OnFrameWithEmptyRegionCapture() override;

void OnStopped() override;

void OnLog(const std::string& message) override;

void Done(const gfx::Rect& damage, const SkBitmap& frame);

// Get the pixel size of render view.

gfx::Size GetRenderViewSize() const;

FrameCaptureCallback callback_;

bool only_dirty_;

content::RenderWidgetHost* host_;

std::unique_ptr<viz::ClientFrameSinkVideoCapturer> video_capturer_;

base::WeakPtrFactory<FrameSubscriber> weak_ptr_factory_{this};