98 lines
3.1 KiB
Rust
98 lines
3.1 KiB
Rust
use cctk::{
|
|
cosmic_protocols::screencopy::v1::client::zcosmic_screencopy_session_v1,
|
|
screencopy::{BufferInfo, ScreencopyHandler, ScreencopyState},
|
|
wayland_client::{protocol::wl_shm, Connection, QueueHandle, WEnum},
|
|
};
|
|
|
|
use super::{AppData, Buffer, Capture, CaptureSource, Event};
|
|
|
|
impl ScreencopyHandler for AppData {
|
|
fn screencopy_state(&mut self) -> &mut ScreencopyState {
|
|
&mut self.screencopy_state
|
|
}
|
|
|
|
fn init_done(
|
|
&mut self,
|
|
conn: &Connection,
|
|
qh: &QueueHandle<Self>,
|
|
session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
|
|
buffer_infos: &[BufferInfo],
|
|
) {
|
|
let capture = Capture::for_session(session).unwrap();
|
|
if !capture.running() {
|
|
session.destroy();
|
|
return;
|
|
}
|
|
|
|
let buffer_info = buffer_infos
|
|
.iter()
|
|
.find(|x| {
|
|
x.type_ == WEnum::Value(zcosmic_screencopy_session_v1::BufferType::WlShm)
|
|
&& x.format == wl_shm::Format::Abgr8888.into()
|
|
})
|
|
.unwrap();
|
|
|
|
// XXX fix in compositor
|
|
if buffer_info.width == 0 || buffer_info.height == 0 || buffer_info.stride == 0 {
|
|
session.destroy();
|
|
return;
|
|
}
|
|
|
|
let mut buffer = capture.buffer.lock().unwrap();
|
|
// Create new buffer if none, or different format
|
|
if !buffer
|
|
.as_ref()
|
|
.map_or(false, |x| &x.buffer_info == buffer_info)
|
|
{
|
|
*buffer = Some(Buffer::new(buffer_info.clone(), &self.shm_state, qh));
|
|
}
|
|
let buffer = buffer.as_ref().unwrap();
|
|
|
|
session.attach_buffer(&buffer.buffer, None, 0); // XXX age?
|
|
if capture.first_frame() {
|
|
session.commit(zcosmic_screencopy_session_v1::Options::empty());
|
|
} else {
|
|
session.commit(zcosmic_screencopy_session_v1::Options::OnDamage);
|
|
}
|
|
conn.flush().unwrap();
|
|
}
|
|
|
|
fn ready(
|
|
&mut self,
|
|
_conn: &Connection,
|
|
_qh: &QueueHandle<Self>,
|
|
session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
|
|
) {
|
|
let capture = Capture::for_session(session).unwrap();
|
|
if !capture.running() {
|
|
session.destroy();
|
|
return;
|
|
}
|
|
|
|
let mut buffer = capture.buffer.lock().unwrap();
|
|
let image = unsafe { buffer.as_mut().unwrap().to_image() };
|
|
let event = match &capture.source {
|
|
CaptureSource::Toplevel(toplevel) => Event::ToplevelCapture(toplevel.clone(), image),
|
|
CaptureSource::Workspace(workspace, _) => {
|
|
Event::WorkspaceCapture(workspace.clone(), image)
|
|
}
|
|
};
|
|
self.send_event(event);
|
|
session.destroy();
|
|
|
|
// Capture again on damage
|
|
capture.capture(&self.screencopy_state.screencopy_manager, &self.qh);
|
|
}
|
|
|
|
fn failed(
|
|
&mut self,
|
|
_conn: &Connection,
|
|
_qh: &QueueHandle<Self>,
|
|
session: &zcosmic_screencopy_session_v1::ZcosmicScreencopySessionV1,
|
|
_reason: WEnum<zcosmic_screencopy_session_v1::FailureReason>,
|
|
) {
|
|
// TODO
|
|
println!("Failed");
|
|
session.destroy();
|
|
}
|
|
}
|