migrations/egui
1
0
Fork 0
mirror of https://github.com/emilk/egui.git synced 2026-08-30 13:20:05 -04:00
Code Activity

eframe: capture a screenshot using Frame::request_screenshot

Browse Source 
Co-authored-by: Emil Ernerfeldt <emil.ernerfeldt@gmail.com>
This commit is contained in:
amfaber
2023-03-29 16:34:22 +02:00
committed by GitHub
parent 74d43bfa17
commit 870264b005
11 changed files with 384 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
examples/screenshot/src/main.rs
View File

@@ -1,13 +1,12 @@
#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")] // hide console window on Windows in release
use eframe::{
egui::{self, ColorImage},
glow::{self, HasContext},
};
use itertools::Itertools as _;
use eframe::egui::{self, ColorImage};
fn main() -> Result<(), eframe::Error> {
let options = eframe::NativeOptions::default();
let options = eframe::NativeOptions {
renderer: eframe::Renderer::Wgpu,
..Default::default()
};
eframe::run_native(
"Take screenshots and display with eframe/egui",
options,
@@ -18,13 +17,13 @@ fn main() -> Result<(), eframe::Error> {
#[derive(Default)]
struct MyApp {
continuously_take_screenshots: bool,
take_screenshot: bool,
texture: Option<egui::TextureHandle>,
screenshot: Option<ColorImage>,
save_to_file: bool,
}
impl eframe::App for MyApp {
fn update(&mut self, ctx: &egui::Context, _frame: &mut eframe::Frame) {
fn update(&mut self, ctx: &egui::Context, frame: &mut eframe::Frame) {
egui::CentralPanel::default().show(ctx, |ui| {
if let Some(screenshot) = self.screenshot.take() {
self.texture = Some(ui.ctx().load_texture(
@@ -40,6 +39,11 @@ impl eframe::App for MyApp {
"continuously take screenshots",
);
if ui.button("save to 'top_left.png'").clicked() {
self.save_to_file = true;
frame.request_screenshot();
}
ui.with_layout(egui::Layout::top_down(egui::Align::RIGHT), |ui| {
if self.continuously_take_screenshots {
if ui
@@ -50,8 +54,9 @@ impl eframe::App for MyApp {
} else {
ctx.set_visuals(egui::Visuals::light());
};
frame.request_screenshot();
} else if ui.button("take screenshot!").clicked() {
self.take_screenshot = true;
frame.request_screenshot();
}
});
});
@@ -66,43 +71,24 @@ impl eframe::App for MyApp {
});
}
#[allow(unsafe_code)]
fn post_rendering(&mut self, screen_size_px: [u32; 2], frame: &eframe::Frame) {
if !self.take_screenshot && !self.continuously_take_screenshots {
return;
}
self.take_screenshot = false;
if let Some(gl) = frame.gl() {
let [w, h] = screen_size_px;
let mut buf = vec![0u8; w as usize * h as usize * 4];
let pixels = glow::PixelPackData::Slice(&mut buf[..]);
unsafe {
gl.read_pixels(
0,
0,
w as i32,
h as i32,
glow::RGBA,
glow::UNSIGNED_BYTE,
pixels,
);
fn post_rendering(&mut self, _window_size: [u32; 2], frame: &eframe::Frame) {
if let Some(screenshot) = frame.screenshot() {
if self.save_to_file {
let pixels_per_point = frame.info().native_pixels_per_point;
let region =
egui::Rect::from_two_pos(egui::Pos2::ZERO, egui::Pos2 { x: 100., y: 100. });
let top_left_corner = screenshot.region(&region, pixels_per_point);
image::save_buffer(
"top_left.png",
top_left_corner.as_raw(),
top_left_corner.width() as u32,
top_left_corner.height() as u32,
image::ColorType::Rgba8,
)
.unwrap();
self.save_to_file = false;
}
// Flip vertically:
let mut rows: Vec<Vec<u8>> = buf
.into_iter()
.chunks(w as usize * 4)
.into_iter()
.map(|chunk| chunk.collect())
.collect();
rows.reverse();
let buf: Vec<u8> = rows.into_iter().flatten().collect();
self.screenshot = Some(ColorImage::from_rgba_unmultiplied(
[screen_size_px[0] as usize, screen_size_px[1] as usize],
&buf[..],
));
self.screenshot = Some(screenshot);
}
}
}