privacy_mode: win10 magnifier

Signed-off-by: fufesou <shuanglongchen@yeah.net>

libs/scrap/examples/capture_mag.rs

@@ -0,0 +1,105 @@
+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);
+    }
+}

libs/scrap/examples/screenshot.rs

@@ -46,8 +46,7 @@ fn record(i: usize) {
                 }
             }
         };
-
-        println!("Captured! Saving...");
+        println!("Captured data len: {}, Saving...", buffer.len());
 
         // Flip the BGRA image into a RGBA image.
 
@@ -96,8 +95,7 @@ fn record(i: usize) {
                 }
             }
         };
-
-        println!("Captured! Saving...");
+        println!("Captured data len: {}, Saving...", buffer.len());
 
         let mut frame = Default::default();
         i420_to_rgb(w, h, &buffer, &mut frame);