13b565e013
Allows cleaning up some of the relationship between `screencopy.rs` and `capture.rs`, and groups variables that go together and are set to `Some`/`None` together. Should help for adding double buffering.
85 lines
2.4 KiB
Rust
85 lines
2.4 KiB
Rust
use cctk::{
|
|
cosmic_protocols::{
|
|
screencopy::v1::client::{zcosmic_screencopy_manager_v1, zcosmic_screencopy_session_v1},
|
|
toplevel_info::v1::client::zcosmic_toplevel_handle_v1,
|
|
workspace::v1::client::zcosmic_workspace_handle_v1,
|
|
},
|
|
wayland_client::{protocol::wl_output, Proxy, QueueHandle},
|
|
};
|
|
use cosmic::cctk;
|
|
|
|
use std::sync::{
|
|
atomic::{AtomicBool, Ordering},
|
|
Arc, Mutex,
|
|
};
|
|
|
|
use super::{AppData, ScreencopySession, SessionData};
|
|
|
|
#[derive(Clone, Hash, PartialEq, Eq)]
|
|
pub enum CaptureSource {
|
|
Toplevel(zcosmic_toplevel_handle_v1::ZcosmicToplevelHandleV1),
|
|
Workspace(
|
|
zcosmic_workspace_handle_v1::ZcosmicWorkspaceHandleV1,
|
|
wl_output::WlOutput,
|
|
),
|
|
}
|
|
|
|
#[derive(Clone, Debug, Default)]
|
|
pub struct CaptureFilter {
|
|
pub workspaces_on_outputs: Vec<wl_output::WlOutput>,
|
|
pub toplevels_on_workspaces: Vec<zcosmic_workspace_handle_v1::ZcosmicWorkspaceHandleV1>,
|
|
}
|
|
|
|
pub struct Capture {
|
|
pub source: CaptureSource,
|
|
first_frame: AtomicBool,
|
|
pub session: Mutex<Option<ScreencopySession>>,
|
|
}
|
|
|
|
impl Capture {
|
|
pub fn new(source: CaptureSource) -> Arc<Capture> {
|
|
Arc::new(Capture {
|
|
source,
|
|
first_frame: AtomicBool::new(true),
|
|
session: Mutex::new(None),
|
|
})
|
|
}
|
|
|
|
// Returns `None` if capture is destroyed
|
|
// (or if `session` wasn't created with `SessionData`)
|
|
pub fn for_session(
|
|
session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
|
|
) -> Option<Arc<Self>> {
|
|
session.data::<SessionData>()?.capture.upgrade()
|
|
}
|
|
|
|
pub fn running(&self) -> bool {
|
|
self.session.lock().unwrap().is_some()
|
|
}
|
|
|
|
pub fn unset_first_frame(&self) {
|
|
self.first_frame.store(false, Ordering::SeqCst);
|
|
}
|
|
|
|
pub fn first_frame(&self) -> bool {
|
|
self.first_frame.load(Ordering::SeqCst)
|
|
}
|
|
|
|
// Start capturing frames
|
|
pub fn start(
|
|
self: &Arc<Self>,
|
|
manager: &zcosmic_screencopy_manager_v1::ZcosmicScreencopyManagerV1,
|
|
qh: &QueueHandle<AppData>,
|
|
) {
|
|
let mut session = self.session.lock().unwrap();
|
|
if session.is_none() {
|
|
self.first_frame.store(true, Ordering::SeqCst);
|
|
*session = Some(ScreencopySession::new(self, manager, qh));
|
|
}
|
|
}
|
|
|
|
// Stop capturing. Can be started again with `start`
|
|
pub fn stop(&self) {
|
|
self.session.lock().unwrap().take();
|
|
}
|
|
}
|