59 lines
1.6 KiB
Rust
59 lines
1.6 KiB
Rust
use {
|
|
crate::{
|
|
format::XRGB8888,
|
|
render::RenderError,
|
|
state::State,
|
|
video::{
|
|
drm::DrmError,
|
|
gbm::{GbmBo, GbmError, GBM_BO_USE_LINEAR, GBM_BO_USE_RENDERING},
|
|
ModifiedFormat, INVALID_MODIFIER,
|
|
},
|
|
},
|
|
std::{ops::Deref, rc::Rc},
|
|
thiserror::Error,
|
|
uapi::OwnedFd,
|
|
};
|
|
|
|
#[derive(Debug, Error)]
|
|
pub enum ScreenshooterError {
|
|
#[error("There is no render context")]
|
|
NoRenderContext,
|
|
#[error("Display is empty")]
|
|
EmptyDisplay,
|
|
#[error(transparent)]
|
|
GbmError(#[from] GbmError),
|
|
#[error(transparent)]
|
|
RenderError(#[from] RenderError),
|
|
#[error(transparent)]
|
|
DrmError(#[from] DrmError),
|
|
}
|
|
|
|
pub struct Screenshot {
|
|
pub drm: Rc<OwnedFd>,
|
|
pub bo: GbmBo,
|
|
}
|
|
|
|
pub fn take_screenshot(state: &State) -> Result<Screenshot, ScreenshooterError> {
|
|
let ctx = match state.render_ctx.get() {
|
|
Some(ctx) => ctx,
|
|
_ => return Err(ScreenshooterError::NoRenderContext),
|
|
};
|
|
let extents = state.root.extents.get();
|
|
if extents.is_empty() {
|
|
return Err(ScreenshooterError::EmptyDisplay);
|
|
}
|
|
let format = ModifiedFormat {
|
|
format: XRGB8888,
|
|
modifier: INVALID_MODIFIER,
|
|
};
|
|
let bo = ctx.gbm.create_bo(
|
|
extents.width(),
|
|
extents.height(),
|
|
&format,
|
|
GBM_BO_USE_RENDERING | GBM_BO_USE_LINEAR,
|
|
)?;
|
|
let fb = ctx.dmabuf_fb(bo.dmabuf())?;
|
|
fb.render(state.root.deref(), state, Some(state.root.extents.get()));
|
|
let drm = ctx.gbm.drm.dup_render()?.fd().clone();
|
|
Ok(Screenshot { drm, bo })
|
|
}
|