106 lines
3.1 KiB
Rust
106 lines
3.1 KiB
Rust
|
extern crate repng;
|
||
|
|
extern crate scrap;
|
||
|
|
|
||
|
|
use std::fs::File;
|
||
|
|
|
||
|
|
#[cfg(windows)]
|
||
|
|
use scrap::CapturerMag;
|
||
|
|
use scrap::{i420_to_rgb, Display};
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let n = Display::all().unwrap().len();
|
||
|
|
for i in 0..n {
|
||
|
|
#[cfg(windows)]
|
||
|
|
record(i);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn get_display(i: usize) -> Display {
|
||
|
|
Display::all().unwrap().remove(i)
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(windows)]
|
||
|
|
fn record(i: usize) {
|
||
|
|
for d in Display::all().unwrap() {
|
||
|
|
println!("{:?} {} {}", d.origin(), d.width(), d.height());
|
||
|
|
}
|
||
|
|
|
||
|
|
let display = get_display(i);
|
||
|
|
let (w, h) = (display.width(), display.height());
|
||
|
|
|
||
|
|
{
|
||
|
|
let mut capture_mag =
|
||
|
|
CapturerMag::new(display.origin(), display.width(), display.height(), false)
|
||
|
|
.expect("Couldn't begin capture.");
|
||
|
|
let wnd_cls = "";
|
||
|
|
let wnd_name = "RustDeskPrivacyWindow";
|
||
|
|
if false == capture_mag.exclude(wnd_cls, wnd_name).unwrap() {
|
||
|
|
println!("No window found for cls {} name {}", wnd_cls, wnd_name);
|
||
|
|
} else {
|
||
|
|
println!("Filter window for cls {} name {}", wnd_cls, wnd_name);
|
||
|
|
}
|
||
|
|
|
||
|
|
let frame = capture_mag.frame(0).unwrap();
|
||
|
|
println!("Capture data len: {}, Saving...", frame.len());
|
||
|
|
|
||
|
|
let mut bitflipped = Vec::with_capacity(w * h * 4);
|
||
|
|
let stride = frame.len() / h;
|
||
|
|
|
||
|
|
for y in 0..h {
|
||
|
|
for x in 0..w {
|
||
|
|
let i = stride * y + 4 * x;
|
||
|
|
bitflipped.extend_from_slice(&[frame[i + 2], frame[i + 1], frame[i], 255]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Save the image.
|
||
|
|
let name = format!("capture_mag_{}_1.png", i);
|
||
|
|
repng::encode(
|
||
|
|
File::create(name.clone()).unwrap(),
|
||
|
|
w as u32,
|
||
|
|
h as u32,
|
||
|
|
&bitflipped,
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
println!("Image saved to `{}`.", name);
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
let mut capture_mag =
|
||
|
|
CapturerMag::new(display.origin(), display.width(), display.height(), true)
|
||
|
|
.expect("Couldn't begin capture.");
|
||
|
|
let wnd_cls = "";
|
||
|
|
let wnd_title = "RustDeskPrivacyWindow";
|
||
|
|
if false == capture_mag.exclude(wnd_cls, wnd_title).unwrap() {
|
||
|
|
println!("No window found for cls {} title {}", wnd_cls, wnd_title);
|
||
|
|
} else {
|
||
|
|
println!("Filter window for cls {} title {}", wnd_cls, wnd_title);
|
||
|
|
}
|
||
|
|
|
||
|
|
let buffer = capture_mag.frame(0).unwrap();
|
||
|
|
println!("Capture data len: {}, Saving...", buffer.len());
|
||
|
|
|
||
|
|
let mut frame = Default::default();
|
||
|
|
i420_to_rgb(w, h, &buffer, &mut frame);
|
||
|
|
|
||
|
|
let mut bitflipped = Vec::with_capacity(w * h * 4);
|
||
|
|
let stride = frame.len() / h;
|
||
|
|
|
||
|
|
for y in 0..h {
|
||
|
|
for x in 0..w {
|
||
|
|
let i = stride * y + 3 * x;
|
||
|
|
bitflipped.extend_from_slice(&[frame[i], frame[i + 1], frame[i + 2], 255]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
let name = format!("capture_mag_{}_2.png", i);
|
||
|
|
repng::encode(
|
||
|
|
File::create(name.clone()).unwrap(),
|
||
|
|
w as u32,
|
||
|
|
h as u32,
|
||
|
|
&bitflipped,
|
||
|
|
)
|
||
|
|
.unwrap();
|
||
|
|
|
||
|
|
println!("Image saved to `{}`.", name);
|
||
|
|
}
|
||
|
|
}
|