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

use anyhow::{anyhow, Result};
pub use display_info::DisplayInfo;

mod image;
pub use image::Image;

#[cfg(target_os = "macos")]
mod darwin;
#[cfg(target_os = "macos")]
use darwin::*;

#[cfg(target_os = "windows")]
mod win32;
#[cfg(target_os = "windows")]
use win32::*;

#[cfg(target_os = "linux")]
mod linux;
#[cfg(target_os = "linux")]
use linux::*;

#[derive(Debug, Clone, Copy)]
pub struct Screen {
  pub display_info: DisplayInfo,
}

impl Screen {
  pub fn new(display_info: &DisplayInfo) -> Self {
    Screen {
      display_info: *display_info,
    }
  }

  pub fn all() -> Result<Vec<Screen>> {
    let screens = DisplayInfo::all()?.iter().map(Screen::new).collect();
    Ok(screens)
  }

  pub fn from_point(x: i32, y: i32) -> Result<Screen> {
    let display_info = DisplayInfo::from_point(x, y)?;
    Ok(Screen::new(&display_info))
  }

  pub fn capture(&self) -> Result<Image> {
    capture_screen(&self.display_info)
  }

  /**
   * 截取指定区域
   * 区域x,y为相对于当前屏幕的x,y坐标
   */
  pub fn capture_area(&self, x: i32, y: i32, width: u32, height: u32) -> Result<Image> {
    let display_info = self.display_info;
    let screen_x2 = display_info.x + display_info.width as i32;
    let screen_y2 = display_info.y + display_info.height as i32;

    let mut x1 = x + display_info.x;
    let mut y1 = y + display_info.y;
    let mut x2 = x1 + width as i32;
    let mut y2 = y1 + height as i32;

    // x y 必须在屏幕范围内
    if x1 < display_info.x {
      x1 = display_info.x;
    } else if x1 > screen_x2 {
      x1 = screen_x2
    }

    if y1 < display_info.y {
      y1 = display_info.y;
    } else if y1 > screen_y2 {
      y1 = screen_y2;
    }

    if x2 > screen_x2 {
      x2 = screen_x2;
    }

    if y2 > screen_y2 {
      y2 = screen_y2;
    }

    if x1 >= x2 || y1 >= y2 {
      return Err(anyhow!("Area size is invalid"));
    }

    capture_screen_area(
      &display_info,
      x1 - display_info.x,
      y1 - display_info.y,
      (x2 - x1) as u32,
      (y2 - y1) as u32,
    )
  }
}