534bfad50f
1. use src width/height to convert yuv 2. align dst yuv to avoid illegal memory access 3. init yuvfmt when new codec 4. move remote reset calls from empty conns judge to emtpy remote conns judge Signed-off-by: 21pages <pages21@163.com>
116 lines
2.9 KiB
Rust
116 lines
2.9 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>, 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) -> io::Result<Capturer> {
|
|
let r = display.0.recorder(false).map_err(map_err)?;
|
|
Ok(Capturer(display, r, Default::default()))
|
|
}
|
|
|
|
pub fn width(&self) -> usize {
|
|
self.0.width()
|
|
}
|
|
|
|
pub fn height(&self) -> usize {
|
|
self.0.height()
|
|
}
|
|
}
|
|
|
|
impl TraitCapturer for Capturer {
|
|
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::new(x, crate::Pixfmt::BGRA, w, h)),
|
|
PixelProvider::RGB0(w, h, x) => Ok(Frame::new(x, crate::Pixfmt::RGBA, w,h)),
|
|
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()
|
|
}
|
|
}
|