scap-rs 0.1.0

Modern, high-performance screen capture library for Rust. Cross-platform.
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

/// macOS 平台目标管理模块
#[cfg(target_os = "macos")]
mod mac;

/// Windows 平台目标管理模块
#[cfg(target_os = "windows")]
mod win;

use anyhow::Result;

/// Linux/FreeBSD 平台目标管理模块
#[cfg(any(target_os = "linux", target_os = "freebsd"))]
pub(crate) mod linux;

/// 窗口结构体，表示一个可捕获的窗口
#[derive(Debug, Clone)]
pub struct Window {
    pub id: u32,       // 窗口 ID
    pub title: String, // 窗口标题

    /// Windows 平台原生窗口句柄
    #[cfg(target_os = "windows")]
    pub raw_handle: windows::Win32::Foundation::HWND,

    /// macOS 平台原生窗口 ID
    #[cfg(target_os = "macos")]
    pub raw_handle: core_graphics_helmer_fork::window::CGWindowID,

    /// Linux 平台 X11 窗口句柄
    #[cfg(all(any(target_os = "linux", target_os = "freebsd"), feature = "x11"))]
    pub raw_handle: xcb::x::Window,
}

/// 显示器结构体，表示一个可捕获的显示器
#[derive(Debug, Clone)]
pub struct Display {
    pub id: u32,       // 显示器 ID
    pub title: String, // 显示器名称

    /// Windows 平台原生显示器句柄
    #[cfg(target_os = "windows")]
    pub raw_handle: windows::Win32::Graphics::Gdi::HMONITOR,

    /// macOS 平台原生显示器对象
    #[cfg(target_os = "macos")]
    pub raw_handle: core_graphics_helmer_fork::display::CGDisplay,

    /// Linux 平台 X11 显示器句柄
    #[cfg(all(any(target_os = "linux", target_os = "freebsd"), feature = "x11"))]
    pub raw_handle: xcb::x::Window,

    /// 显示器宽度（像素）
    #[cfg(any(target_os = "linux", target_os = "freebsd", target_os = "windows"))]
    pub width: u16,

    /// 显示器高度（像素）
    #[cfg(any(target_os = "linux", target_os = "freebsd", target_os = "windows"))]
    pub height: u16,

    /// 显示器 X 偏移量
    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    pub x_offset: i16,

    /// 显示器 Y 偏移量
    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    pub y_offset: i16,
}

/// 捕获目标枚举，可以是窗口或显示器
#[derive(Debug, Clone)]
pub enum Target {
    Window(Window),   // 窗口目标
    Display(Display), // 显示器目标
}

// 为 Windows 平台的 Display 和 Window 实现 Send 和 Sync trait
// 因为 HWND 和 HMONITOR 都是线程安全的
#[cfg(target_os = "windows")]
unsafe impl Send for Display {}
#[cfg(target_os = "windows")]
unsafe impl Sync for Display {}

#[cfg(target_os = "windows")]
unsafe impl Send for Window {}
#[cfg(target_os = "windows")]
unsafe impl Sync for Window {}

/// 获取所有可捕获的目标列表
/// 包括所有显示器和窗口
pub fn get_all_targets() -> Result<Vec<Target>> {
    #[cfg(target_os = "macos")]
    return mac::get_all_targets();

    #[cfg(target_os = "windows")]
    return win::get_all_targets();

    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    return linux::get_all_targets();
}

/// 获取目标的缩放因子
/// 用于处理高 DPI 显示器
#[allow(unused_variables, dead_code)]
pub fn get_scale_factor(target: &Target) -> f64 {
    #[cfg(target_os = "macos")]
    return mac::get_scale_factor(target);

    #[cfg(target_os = "windows")]
    return win::get_scale_factor(target);

    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    return 1.0;
}

/// 获取主显示器信息
/// 返回系统主显示器的详细信息
#[allow(dead_code)]
pub fn get_main_display() -> Result<Display> {
    #[cfg(target_os = "macos")]
    return mac::get_main_display();

    #[cfg(target_os = "windows")]
    return win::get_main_display();

    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    return linux::get_main_display();
}

/// 获取目标的像素尺寸
/// 返回 (宽度, 高度) 的元组
#[allow(dead_code)]
pub fn get_target_dimensions(target: &Target) -> (u64, u64) {
    #[cfg(target_os = "macos")]
    return mac::get_target_dimensions(target);

    #[cfg(target_os = "windows")]
    return win::get_target_dimensions(target);

    #[cfg(any(target_os = "linux", target_os = "freebsd"))]
    return linux::get_target_dimensions(target);
}