1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_UI_THUMBNAILS_BACKGROUND_THUMBNAIL_VIDEO_CAPTURER_H_
#define CHROME_BROWSER_UI_THUMBNAILS_BACKGROUND_THUMBNAIL_VIDEO_CAPTURER_H_
#include <stdint.h>
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/sequence_checker.h"
#include "base/time/time.h"
#include "chrome/browser/ui/thumbnails/background_thumbnail_capturer.h"
#include "components/viz/host/client_frame_sink_video_capturer.h"
#include "third_party/skia/include/core/SkBitmap.h"
namespace content {
class WebContents;
} // namespace content
// A thumbnail capturer using viz::ClientFrameSinkVideoCapturer. Gets a
// sequence of frames in the same way as streaming a tab.
class BackgroundThumbnailVideoCapturer
: public BackgroundThumbnailCapturer,
public viz::mojom::FrameSinkVideoConsumer {
public:
// Client receives `SkBitmap` frames and `uin64_t` unique IDs for each
// frame. IDs are globally unique for a given browser process and are
// used for TRACE_EVENT_FLOW_* macros
using GotFrameCallback =
base::RepeatingCallback<void(const SkBitmap&, uint64_t)>;
BackgroundThumbnailVideoCapturer(content::WebContents* contents,
GotFrameCallback got_frame_callback);
~BackgroundThumbnailVideoCapturer() override;
// BackgroundThumbnailCapturer:
void Start(const ThumbnailCaptureInfo& capture_info) override;
void Stop() override;
private:
// viz::mojom::FrameSinkVideoConsumer:
void OnFrameCaptured(
::media::mojom::VideoBufferHandlePtr data,
::media::mojom::VideoFrameInfoPtr info,
const gfx::Rect& content_rect,
mojo::PendingRemote<::viz::mojom::FrameSinkVideoConsumerFrameCallbacks>
callbacks) override;
void OnNewSubCaptureTargetVersion(
uint32_t sub_capture_target_version) override;
void OnFrameWithEmptyRegionCapture() override;
void OnStopped() override;
void OnLog(const std::string& /*message*/) override;
const raw_ptr<content::WebContents> contents_;
GotFrameCallback got_frame_callback_;
ThumbnailCaptureInfo capture_info_;
std::unique_ptr<viz::ClientFrameSinkVideoCapturer> video_capturer_;
// Tracked for metrics and tracing
base::TimeTicks start_time_;
int num_received_frames_ = 0;
uint64_t cur_capture_num_ = 0;
SEQUENCE_CHECKER(sequence_checker_);
};
#endif // CHROME_BROWSER_UI_THUMBNAILS_BACKGROUND_THUMBNAIL_VIDEO_CAPTURER_H_
|
