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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_CAPTURER_ANDROID_H_
#define CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_CAPTURER_ANDROID_H_
#include <jni.h>
#include <optional>
#include "base/android/scoped_java_ref.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/numerics/checked_math.h"
#include "base/time/time.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_capture_options.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"
namespace content {
// `DesktopCapturer` implementation for Android. There are a few things
// involved:
// - An instance of `DesktopCapturerAndroid` which creates a Java side object,
// `ScreenCapture`.
// - `ScreenCapture` Java object, which manages interaction with the OS.
//
// This is complicated by the following factors:
// - Screen capture may be stopped from either the C++ side or the Java side.
// - Buffers must be freed on the Java side, but must be consumed on the desktop
// capturer thread.
//
// On Android, the desktop capturer thread is created with an Android message
// pump, so we can keep everything on one thread.
class DesktopCapturerAndroid final : public webrtc::DesktopCapturer {
public:
DesktopCapturerAndroid(const webrtc::DesktopCaptureOptions& options);
DesktopCapturerAndroid(const DesktopCapturerAndroid&) = delete;
DesktopCapturerAndroid& operator=(const DesktopCapturerAndroid&) = delete;
~DesktopCapturerAndroid() override;
// DesktopCapturer:
void Start(Callback* callback) override;
void SetSharedMemoryFactory(std::unique_ptr<webrtc::SharedMemoryFactory>
shared_memory_factory) override;
void CaptureFrame() override;
bool SelectSource(SourceId id) override;
// JNI:
void OnRgbaFrameAvailable(JNIEnv* env,
const base::android::JavaRef<jobject>& release_cb,
jlong timestamp_ns,
const base::android::JavaRef<jobject>& buf,
jint unchecked_pixel_stride,
jint unchecked_row_stride,
jint unchecked_crop_left,
jint unchecked_crop_top,
jint unchecked_crop_right,
jint unchecked_crop_bottom);
void OnStop(JNIEnv* env);
private:
// `PlaneInfo` stores all info needed to process buffers received from
// Android.
struct PlaneInfo {
PlaneInfo();
~PlaneInfo();
PlaneInfo(PlaneInfo&& other);
PlaneInfo& operator=(PlaneInfo&& other);
PlaneInfo(const PlaneInfo&) = delete;
PlaneInfo& operator=(const PlaneInfo&) = delete;
// Java callback to run when this plane's buffer is no longer in use.
base::android::ScopedJavaGlobalRef<jobject> release_cb;
// Timestamp of the frame in nanoseconds.
int64_t timestamp_ns;
// Java ByteBuffer containing the plane data.
base::android::ScopedJavaGlobalRef<jobject> buf;
// The number of bytes between the start of adjacent pixels in a row.
base::CheckedNumeric<uint32_t> pixel_stride;
// The number of bytes between the start of adjacent rows of pixels.
base::CheckedNumeric<uint32_t> row_stride;
// The x-coordinate of the top-left corner of the crop rectangle.
base::CheckedNumeric<uint32_t> crop_left;
// The y-coordinate of the top-left corner of the crop rectangle.
base::CheckedNumeric<uint32_t> crop_top;
// The x-coordinate of the bottom-right corner of the crop rectangle.
base::CheckedNumeric<uint32_t> crop_right;
// The y-coordinate of the bottom-right corner of the crop rectangle.
base::CheckedNumeric<uint32_t> crop_bottom;
};
void Shutdown();
void ProcessRgbaFrame(int64_t timestamp_ns, PlaneInfo plane);
raw_ptr<Callback> callback_ = nullptr;
base::android::ScopedJavaGlobalRef<jobject> screen_capture_;
std::unique_ptr<webrtc::DesktopFrame> next_frame_;
int64_t last_frame_time_ns_ = 0;
bool finishing_ = false;
};
} // namespace content
#endif // CONTENT_BROWSER_MEDIA_CAPTURE_DESKTOP_CAPTURER_ANDROID_H_
|
