autocommit 2022-04-13 21:01:32 CEST

src/screenshoter.rs

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