1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
use x11; use std::{io, ops}; use std::rc::Rc; pub struct Capturer(x11::Capturer); impl Capturer { pub fn new(display: Display) -> io::Result<Capturer> { x11::Capturer::new(display.0).map(Capturer) } pub fn width(&self) -> usize { self.0.display().rect().w as usize } pub fn height(&self) -> usize { self.0.display().rect().h as usize } pub fn frame<'a>(&'a mut self) -> io::Result<Frame<'a>> { Ok(Frame(self.0.frame())) } } pub struct Frame<'a>(&'a [u8]); impl<'a> ops::Deref for Frame<'a> { type Target = [u8]; fn deref(&self) -> &[u8] { self.0 } } pub struct Display(x11::Display); impl Display { pub fn primary() -> io::Result<Display> { let server = Rc::new(match x11::Server::default() { Ok(server) => server, Err(_) => return Err(io::ErrorKind::ConnectionRefused.into()) }); let mut displays = x11::Server::displays(server); let mut best = displays.next(); if best.as_ref().map(|x| x.is_default()) == Some(false) { best = displays.find(|x| x.is_default()).or(best); } match best { Some(best) => Ok(Display(best)), None => Err(io::ErrorKind::NotFound.into()) } } pub fn all() -> io::Result<Vec<Display>> { let server = Rc::new(match x11::Server::default() { Ok(server) => server, Err(_) => return Err(io::ErrorKind::ConnectionRefused.into()) }); Ok(x11::Server::displays(server).map(Display).collect()) } pub fn width(&self) -> usize { self.0.rect().w as usize } pub fn height(&self) -> usize { self.0.rect().h as usize } }