130 lines
3.3 KiB
Rust
130 lines
3.3 KiB
Rust
use crate::common::{x11::Frame, TraitCapturer};
|
|
use crate::wayland::{capturable::*, *};
|
|
use std::{io, sync::RwLock, time::Duration};
|
|
|
|
pub struct Capturer(Display, Box<dyn Recorder>, bool, Vec<u8>);
|
|
|
|
static mut IS_CURSOR_EMBEDDED: Option<bool> = None;
|
|
|
|
lazy_static::lazy_static! {
|
|
static ref MAP_ERR: RwLock<Option<fn(err: String)-> io::Error>> = Default::default();
|
|
}
|
|
|
|
pub fn is_cursor_embedded() -> bool {
|
|
unsafe {
|
|
if IS_CURSOR_EMBEDDED.is_none() {
|
|
init_cursor_embedded();
|
|
}
|
|
IS_CURSOR_EMBEDDED.unwrap_or(false)
|
|
}
|
|
}
|
|
|
|
unsafe fn init_cursor_embedded() {
|
|
use crate::common::wayland::pipewire::get_available_cursor_modes;
|
|
match get_available_cursor_modes() {
|
|
Ok(_modes) => {
|
|
// IS_CURSOR_EMBEDDED = Some((_modes & 0x02) > 0);
|
|
IS_CURSOR_EMBEDDED = Some(false)
|
|
}
|
|
Err(..) => {
|
|
IS_CURSOR_EMBEDDED = Some(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn set_map_err(f: fn(err: String) -> io::Error) {
|
|
*MAP_ERR.write().unwrap() = Some(f);
|
|
}
|
|
|
|
fn map_err<E: ToString>(err: E) -> io::Error {
|
|
if let Some(f) = *MAP_ERR.read().unwrap() {
|
|
f(err.to_string())
|
|
} else {
|
|
io::Error::new(io::ErrorKind::Other, err.to_string())
|
|
}
|
|
}
|
|
|
|
impl Capturer {
|
|
pub fn new(display: Display, yuv: bool) -> io::Result<Capturer> {
|
|
let r = display.0.recorder(false).map_err(map_err)?;
|
|
Ok(Capturer(display, r, yuv, Default::default()))
|
|
}
|
|
|
|
pub fn width(&self) -> usize {
|
|
self.0.width()
|
|
}
|
|
|
|
pub fn height(&self) -> usize {
|
|
self.0.height()
|
|
}
|
|
}
|
|
|
|
impl TraitCapturer for Capturer {
|
|
fn set_use_yuv(&mut self, use_yuv: bool) {
|
|
self.2 = use_yuv;
|
|
}
|
|
|
|
fn frame<'a>(&'a mut self, timeout: Duration) -> io::Result<Frame<'a>> {
|
|
match self.1.capture(timeout.as_millis() as _).map_err(map_err)? {
|
|
PixelProvider::BGR0(w, h, x) => Ok(Frame(if self.2 {
|
|
crate::common::bgra_to_i420(w as _, h as _, &x, &mut self.3);
|
|
&self.3[..]
|
|
} else {
|
|
x
|
|
})),
|
|
PixelProvider::RGB0(w, h, x) => Ok(Frame(if self.2 {
|
|
crate::common::rgba_to_i420(w as _, h as _, &x, &mut self.3);
|
|
&self.3[..]
|
|
} else {
|
|
x
|
|
})),
|
|
PixelProvider::NONE => Err(std::io::ErrorKind::WouldBlock.into()),
|
|
_ => Err(map_err("Invalid data")),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub struct Display(pipewire::PipeWireCapturable);
|
|
|
|
impl Display {
|
|
pub fn primary() -> io::Result<Display> {
|
|
let mut all = Display::all()?;
|
|
if all.is_empty() {
|
|
return Err(io::ErrorKind::NotFound.into());
|
|
}
|
|
Ok(all.remove(0))
|
|
}
|
|
|
|
pub fn all() -> io::Result<Vec<Display>> {
|
|
Ok(pipewire::get_capturables(is_cursor_embedded())
|
|
.map_err(map_err)?
|
|
.drain(..)
|
|
.map(|x| Display(x))
|
|
.collect())
|
|
}
|
|
|
|
pub fn width(&self) -> usize {
|
|
self.0.size.0
|
|
}
|
|
|
|
pub fn height(&self) -> usize {
|
|
self.0.size.1
|
|
}
|
|
|
|
pub fn origin(&self) -> (i32, i32) {
|
|
self.0.position
|
|
}
|
|
|
|
pub fn is_online(&self) -> bool {
|
|
true
|
|
}
|
|
|
|
pub fn is_primary(&self) -> bool {
|
|
false
|
|
}
|
|
|
|
pub fn name(&self) -> String {
|
|
"".to_owned()
|
|
}
|
|
}
|