474 lines
18 KiB
Rust
474 lines
18 KiB
Rust
extern crate drm;
|
|
#[macro_use]
|
|
extern crate glium;
|
|
extern crate rand;
|
|
extern crate input as libinput;
|
|
extern crate image;
|
|
extern crate udev;
|
|
#[macro_use(define_roles)]
|
|
extern crate smithay;
|
|
extern crate xkbcommon;
|
|
extern crate wayland_server;
|
|
|
|
#[macro_use]
|
|
extern crate slog;
|
|
extern crate slog_async;
|
|
extern crate slog_term;
|
|
|
|
extern crate ctrlc;
|
|
|
|
mod helpers;
|
|
|
|
use drm::control::{Device as ControlDevice, ResourceInfo};
|
|
use drm::control::connector::{Info as ConnectorInfo, State as ConnectorState};
|
|
use drm::control::encoder::Info as EncoderInfo;
|
|
use drm::control::crtc;
|
|
use drm::result::Error as DrmError;
|
|
use glium::Surface;
|
|
use image::{ImageBuffer, Rgba};
|
|
use libinput::{Libinput, Device as LibinputDevice, event};
|
|
use libinput::event::keyboard::KeyboardEventTrait;
|
|
use helpers::{init_shell, GliumDrawer, MyWindowMap, Roles, SurfaceData};
|
|
use slog::{Drain, Logger};
|
|
use smithay::backend::drm::{DrmBackend, DrmDevice, DrmHandler};
|
|
use smithay::backend::graphics::GraphicsBackend;
|
|
use smithay::backend::graphics::egl::EGLGraphicsBackend;
|
|
use smithay::backend::input::{self, Event, InputBackend, InputHandler, KeyboardKeyEvent, PointerButtonEvent,
|
|
PointerAxisEvent};
|
|
use smithay::backend::libinput::{LibinputInputBackend, libinput_bind, PointerAxisEvent as LibinputPointerAxisEvent, LibinputSessionInterface};
|
|
use smithay::backend::udev::{UdevBackend, UdevHandler, udev_backend_bind};
|
|
use smithay::backend::session::{Session, SessionNotifier};
|
|
use smithay::backend::session::direct::{direct_session_bind, DirectSession};
|
|
use smithay::wayland::compositor::{CompositorToken, SubsurfaceRole, TraversalAction};
|
|
use smithay::wayland::compositor::roles::Role;
|
|
use smithay::wayland::output::{Mode, Output, PhysicalProperties};
|
|
use smithay::wayland::seat::{KeyboardHandle, PointerHandle, Seat};
|
|
use smithay::wayland::shell::ShellState;
|
|
use smithay::wayland::shm::init_shm_global;
|
|
use std::cell::RefCell;
|
|
use std::collections::HashSet;
|
|
use std::io::Error as IoError;
|
|
use std::rc::Rc;
|
|
use std::sync::Arc;
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
|
use std::time::Duration;
|
|
use xkbcommon::xkb::keysyms as xkb;
|
|
use wayland_server::{StateToken, StateProxy};
|
|
use wayland_server::protocol::{wl_output, wl_pointer};
|
|
|
|
struct LibinputInputHandler {
|
|
log: Logger,
|
|
pointer: PointerHandle,
|
|
keyboard: KeyboardHandle,
|
|
window_map: Rc<RefCell<MyWindowMap>>,
|
|
pointer_location: Rc<RefCell<(f64, f64)>>,
|
|
screen_size: (u32, u32),
|
|
serial: u32,
|
|
running: Arc<AtomicBool>,
|
|
}
|
|
|
|
impl LibinputInputHandler {
|
|
fn next_serial(&mut self) -> u32 {
|
|
self.serial += 1;
|
|
self.serial
|
|
}
|
|
}
|
|
|
|
impl InputHandler<LibinputInputBackend> for LibinputInputHandler {
|
|
fn on_seat_created(&mut self, _: &input::Seat) {
|
|
/* we just create a single static one */
|
|
}
|
|
fn on_seat_destroyed(&mut self, _: &input::Seat) {
|
|
/* we just create a single static one */
|
|
}
|
|
fn on_seat_changed(&mut self, _: &input::Seat) {
|
|
/* we just create a single static one */
|
|
}
|
|
fn on_keyboard_key(&mut self, _: &input::Seat, evt: event::keyboard::KeyboardKeyEvent) {
|
|
let keycode = evt.key();
|
|
let state = evt.state();
|
|
debug!(self.log, "key"; "keycode" => keycode, "state" => format!("{:?}", state));
|
|
|
|
let serial = self.next_serial();
|
|
self.keyboard.input(keycode, state, serial, |modifiers, keysym| {
|
|
if modifiers.ctrl && modifiers.alt && keysym == xkb::KEY_BackSpace {
|
|
self.running.store(false, Ordering::SeqCst);
|
|
false
|
|
} else {
|
|
true
|
|
}
|
|
});
|
|
}
|
|
fn on_pointer_move(&mut self, _: &input::Seat, evt: event::pointer::PointerMotionEvent) {
|
|
let (x, y) = (evt.dx(), evt.dy());
|
|
let serial = self.next_serial();
|
|
let mut location = self.pointer_location.borrow_mut();
|
|
location.0 += x;
|
|
location.1 += y;
|
|
let under = self.window_map.borrow().get_surface_under((location.0, location.1));
|
|
self.pointer.motion(
|
|
under.as_ref().map(|&(ref s, (x, y))| (s, x, y)),
|
|
serial,
|
|
evt.time(),
|
|
);
|
|
}
|
|
fn on_pointer_move_absolute(&mut self, _: &input::Seat, evt: event::pointer::PointerMotionAbsoluteEvent) {
|
|
let (x, y) = (evt.absolute_x_transformed(self.screen_size.0), evt.absolute_y_transformed(self.screen_size.1));
|
|
*self.pointer_location.borrow_mut() = (x, y);
|
|
let serial = self.next_serial();
|
|
let under = self.window_map.borrow().get_surface_under((x, y));
|
|
self.pointer.motion(
|
|
under.as_ref().map(|&(ref s, (x, y))| (s, x, y)),
|
|
serial,
|
|
evt.time(),
|
|
);
|
|
}
|
|
fn on_pointer_button(&mut self, _: &input::Seat, evt: event::pointer::PointerButtonEvent) {
|
|
let serial = self.next_serial();
|
|
let button = evt.button();
|
|
let state = match evt.state() {
|
|
input::MouseButtonState::Pressed => {
|
|
// change the keyboard focus
|
|
let under = self.window_map
|
|
.borrow_mut()
|
|
.get_surface_and_bring_to_top(*self.pointer_location.borrow());
|
|
self.keyboard
|
|
.set_focus(under.as_ref().map(|&(ref s, _)| s), serial);
|
|
wl_pointer::ButtonState::Pressed
|
|
}
|
|
input::MouseButtonState::Released => wl_pointer::ButtonState::Released,
|
|
};
|
|
self.pointer.button(button, state, serial, evt.time());
|
|
}
|
|
fn on_pointer_axis(&mut self, _: &input::Seat, evt: LibinputPointerAxisEvent) {
|
|
let axis = match evt.axis() {
|
|
input::Axis::Vertical => wayland_server::protocol::wl_pointer::Axis::VerticalScroll,
|
|
input::Axis::Horizontal => wayland_server::protocol::wl_pointer::Axis::HorizontalScroll,
|
|
};
|
|
self.pointer.axis(axis, evt.amount(), evt.time());
|
|
}
|
|
fn on_touch_down(&mut self, _: &input::Seat, _: event::touch::TouchDownEvent) {
|
|
/* not done in this example */
|
|
}
|
|
fn on_touch_motion(&mut self, _: &input::Seat, _: event::touch::TouchMotionEvent) {
|
|
/* not done in this example */
|
|
}
|
|
fn on_touch_up(&mut self, _: &input::Seat, _: event::touch::TouchUpEvent) {
|
|
/* not done in this example */
|
|
}
|
|
fn on_touch_cancel(&mut self, _: &input::Seat, _: event::touch::TouchCancelEvent) {
|
|
/* not done in this example */
|
|
}
|
|
fn on_touch_frame(&mut self, _: &input::Seat, _: event::touch::TouchFrameEvent) {
|
|
/* not done in this example */
|
|
}
|
|
fn on_input_config_changed(&mut self, _: &mut [LibinputDevice]) {
|
|
/* not done in this example */
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
// A logger facility, here we use the terminal for this example
|
|
let log = Logger::root(
|
|
slog_term::FullFormat::new(slog_term::PlainSyncDecorator::new(std::io::stdout())).build().fuse(),
|
|
o!(),
|
|
);
|
|
|
|
// Initialize the wayland server
|
|
let (mut display, mut event_loop) = wayland_server::create_display();
|
|
|
|
/*
|
|
* Initialize the compositor
|
|
*/
|
|
init_shm_global(&mut event_loop, vec![], log.clone());
|
|
|
|
let (compositor_token, shell_state_token, window_map) = init_shell(&mut event_loop, log.clone());
|
|
|
|
/*
|
|
* Initialize session on the current tty
|
|
*/
|
|
let (session, mut notifier) = DirectSession::new(None, log.clone()).unwrap();
|
|
let session = Rc::new(RefCell::new(session));
|
|
|
|
let running = Arc::new(AtomicBool::new(true));
|
|
let r = running.clone();
|
|
ctrlc::set_handler(move || {
|
|
r.store(false, Ordering::SeqCst);
|
|
}).expect("Error setting Ctrl-C handler");
|
|
|
|
let pointer_location = Rc::new(RefCell::new((0.0, 0.0)));
|
|
|
|
/*
|
|
* Initialize the udev backend
|
|
*/
|
|
let context = udev::Context::new().unwrap();
|
|
let bytes = include_bytes!("resources/cursor2.rgba");
|
|
let udev_token
|
|
= UdevBackend::new(&mut event_loop, &context, session.clone(), UdevHandlerImpl {
|
|
shell_state_token,
|
|
compositor_token,
|
|
window_map: window_map.clone(),
|
|
pointer_location: pointer_location.clone(),
|
|
pointer_image: ImageBuffer::from_raw(64, 64, bytes.to_vec()).unwrap(),
|
|
logger: log.clone(),
|
|
}, log.clone()).unwrap();
|
|
|
|
let udev_session_id = notifier.register(udev_token.clone());
|
|
|
|
let (seat_token, _) = Seat::new(&mut event_loop, session.seat().into(), log.clone());
|
|
|
|
let pointer = event_loop.state().get_mut(&seat_token).add_pointer();
|
|
let keyboard = event_loop
|
|
.state()
|
|
.get_mut(&seat_token)
|
|
.add_keyboard("", "", "", None, 1000, 500)
|
|
.expect("Failed to initialize the keyboard");
|
|
|
|
let (output_token, _output_global) = Output::new(
|
|
&mut event_loop,
|
|
"Drm".into(),
|
|
PhysicalProperties {
|
|
width: 0,
|
|
height: 0,
|
|
subpixel: wl_output::Subpixel::Unknown,
|
|
maker: "Smithay".into(),
|
|
model: "Generic DRM".into(),
|
|
},
|
|
log.clone(),
|
|
);
|
|
|
|
let (w, h) = (1920, 1080); // Hardcode full-hd res
|
|
event_loop
|
|
.state()
|
|
.get_mut(&output_token)
|
|
.change_current_state(
|
|
Some(Mode {
|
|
width: w as i32,
|
|
height: h as i32,
|
|
refresh: 60_000,
|
|
}),
|
|
None,
|
|
None,
|
|
);
|
|
event_loop
|
|
.state()
|
|
.get_mut(&output_token)
|
|
.set_preferred(Mode {
|
|
width: w as i32,
|
|
height: h as i32,
|
|
refresh: 60_000,
|
|
});
|
|
|
|
/*
|
|
* Initialize libinput backend
|
|
*/
|
|
let seat = session.seat();
|
|
let mut libinput_context = Libinput::new_from_udev::<LibinputSessionInterface<Rc<RefCell<DirectSession>>>>(session.into(), &context);
|
|
let libinput_session_id = notifier.register(libinput_context.clone());
|
|
libinput_context.udev_assign_seat(&seat).unwrap();
|
|
let mut libinput_backend = LibinputInputBackend::new(libinput_context, log.clone());
|
|
libinput_backend.set_handler(LibinputInputHandler {
|
|
log: log.clone(),
|
|
pointer,
|
|
keyboard,
|
|
window_map: window_map.clone(),
|
|
pointer_location,
|
|
screen_size: (w, h),
|
|
serial: 0,
|
|
running: running.clone(),
|
|
});
|
|
let libinput_event_source = libinput_bind(libinput_backend, &mut event_loop).unwrap();
|
|
|
|
let session_event_source = direct_session_bind(notifier, &mut event_loop, log.clone()).unwrap();
|
|
let udev_event_source = udev_backend_bind(&mut event_loop, udev_token).unwrap();
|
|
|
|
/*
|
|
* Add a listening socket
|
|
*/
|
|
let name = display.add_socket_auto().unwrap().into_string().unwrap();
|
|
println!("Listening on socket: {}", name);
|
|
|
|
while running.load(Ordering::SeqCst) {
|
|
event_loop.dispatch(Some(16));
|
|
display.flush_clients();
|
|
window_map.borrow_mut().refresh();
|
|
}
|
|
|
|
println!("Bye Bye");
|
|
|
|
let mut notifier = session_event_source.remove();
|
|
notifier.unregister(udev_session_id);
|
|
notifier.unregister(libinput_session_id);
|
|
|
|
libinput_event_source.remove();
|
|
|
|
let udev_token = udev_event_source.remove();
|
|
let udev = event_loop.state().remove(udev_token);
|
|
udev.close(event_loop.state());
|
|
}
|
|
|
|
struct UdevHandlerImpl {
|
|
shell_state_token: StateToken<ShellState<SurfaceData, Roles, (), ()>>,
|
|
compositor_token: CompositorToken<SurfaceData, Roles, ()>,
|
|
window_map: Rc<RefCell<MyWindowMap>>,
|
|
pointer_location: Rc<RefCell<(f64, f64)>>,
|
|
pointer_image: ImageBuffer<Rgba<u8>, Vec<u8>>,
|
|
logger: ::slog::Logger,
|
|
}
|
|
|
|
impl UdevHandlerImpl {
|
|
pub fn scan_connectors<'a, S: Into<StateProxy<'a>>>(&self, state: S, device: &mut DrmDevice<GliumDrawer<DrmBackend>>) {
|
|
// Get a set of all modesetting resource handles (excluding planes):
|
|
let res_handles = device.resource_handles().unwrap();
|
|
|
|
// Use first connected connector
|
|
let connector_infos: Vec<ConnectorInfo> = res_handles
|
|
.connectors()
|
|
.iter()
|
|
.map(|conn| {
|
|
ConnectorInfo::load_from_device(device, *conn).unwrap()
|
|
})
|
|
.filter(|conn| conn.connection_state() == ConnectorState::Connected)
|
|
.inspect(|conn| info!(self.logger, "Connected: {:?}", conn.connector_type()))
|
|
.collect();
|
|
|
|
let mut used_crtcs: HashSet<crtc::Handle> = HashSet::new();
|
|
|
|
let mut state = state.into();
|
|
|
|
// very naive way of finding good crtc/encoder/connector combinations. This problem is np-complete
|
|
for connector_info in connector_infos {
|
|
let encoder_infos = connector_info.encoders().iter().flat_map(|encoder_handle| EncoderInfo::load_from_device(device, *encoder_handle)).collect::<Vec<EncoderInfo>>();
|
|
for encoder_info in encoder_infos {
|
|
for crtc in res_handles.filter_crtcs(encoder_info.possible_crtcs()) {
|
|
if !used_crtcs.contains(&crtc) {
|
|
let mode = connector_info.modes()[0]; // Use first mode (usually highest resoltion, but in reality you should filter and sort and check and match with other connectors, if you use more then one.)
|
|
// create a backend
|
|
let renderer_token = device.create_backend(&mut state, crtc, mode, vec![connector_info.handle()]).unwrap();
|
|
|
|
// create cursor
|
|
{
|
|
let renderer = state.get_mut(renderer_token);
|
|
renderer.set_cursor_representation(&self.pointer_image, (2, 2)).unwrap();
|
|
}
|
|
|
|
// render first frame
|
|
{
|
|
let renderer = state.get_mut(renderer_token);
|
|
let mut frame = renderer.draw();
|
|
frame.clear_color(0.8, 0.8, 0.9, 1.0);
|
|
frame.finish().unwrap();
|
|
}
|
|
|
|
used_crtcs.insert(crtc);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl UdevHandler<GliumDrawer<DrmBackend>, DrmHandlerImpl> for UdevHandlerImpl {
|
|
fn device_added<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &mut DrmDevice<GliumDrawer<DrmBackend>>) -> Option<DrmHandlerImpl>
|
|
{
|
|
self.scan_connectors(state, device);
|
|
|
|
Some(DrmHandlerImpl {
|
|
shell_state_token: self.shell_state_token.clone(),
|
|
compositor_token: self.compositor_token.clone(),
|
|
window_map: self.window_map.clone(),
|
|
pointer_location: self.pointer_location.clone(),
|
|
logger: self.logger.clone(),
|
|
})
|
|
}
|
|
|
|
fn device_changed<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &StateToken<DrmDevice<GliumDrawer<DrmBackend>>>) {
|
|
//quick and dirt, just re-init the device
|
|
let mut state = state.into();
|
|
self.device_removed(&mut state, device);
|
|
state.with_value(device, |state, device| self.scan_connectors(state, device));
|
|
}
|
|
|
|
fn device_removed<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &StateToken<DrmDevice<GliumDrawer<DrmBackend>>>) {
|
|
state.into().with_value(device, |state, device| {
|
|
let crtcs = device.current_backends().into_iter().map(|backend| state.get(backend).crtc()).collect::<Vec<crtc::Handle>>();
|
|
let mut state: StateProxy = state.into();
|
|
for crtc in crtcs {
|
|
device.destroy_backend(&mut state, &crtc);
|
|
}
|
|
});
|
|
}
|
|
|
|
fn error<'a, S: Into<StateProxy<'a>>>(&mut self, _state: S, error: IoError) {
|
|
error!(self.logger, "{:?}", error);
|
|
}
|
|
}
|
|
|
|
pub struct DrmHandlerImpl {
|
|
shell_state_token: StateToken<ShellState<SurfaceData, Roles, (), ()>>,
|
|
compositor_token: CompositorToken<SurfaceData, Roles, ()>,
|
|
window_map: Rc<RefCell<MyWindowMap>>,
|
|
pointer_location: Rc<RefCell<(f64, f64)>>,
|
|
logger: ::slog::Logger,
|
|
}
|
|
|
|
impl DrmHandler<GliumDrawer<DrmBackend>> for DrmHandlerImpl {
|
|
fn ready<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, _device: &mut DrmDevice<GliumDrawer<DrmBackend>>,
|
|
backend: &StateToken<GliumDrawer<DrmBackend>>, _crtc: crtc::Handle, _frame: u32, _duration: Duration) {
|
|
let state = state.into();
|
|
let drawer = state.get(backend);
|
|
{
|
|
let (x, y) = *self.pointer_location.borrow();
|
|
let _ = (**drawer).set_cursor_position(x.trunc().abs() as u32, y.trunc().abs() as u32);
|
|
}
|
|
let mut frame = drawer.draw();
|
|
frame.clear_color(0.8, 0.8, 0.9, 1.0);
|
|
// redraw the frame, in a simple but inneficient way
|
|
{
|
|
let screen_dimensions = drawer.get_framebuffer_dimensions();
|
|
self.window_map
|
|
.borrow()
|
|
.with_windows_from_bottom_to_top(|toplevel_surface, initial_place| {
|
|
if let Some(wl_surface) = toplevel_surface.get_surface() {
|
|
// this surface is a root of a subsurface tree that needs to be drawn
|
|
self.compositor_token
|
|
.with_surface_tree_upward(
|
|
wl_surface,
|
|
initial_place,
|
|
|_surface, attributes, role, &(mut x, mut y)| {
|
|
if let Some((ref contents, (w, h))) = attributes.user_data.buffer {
|
|
// there is actually something to draw !
|
|
if let Ok(subdata) = Role::<SubsurfaceRole>::data(role) {
|
|
x += subdata.x;
|
|
y += subdata.y;
|
|
}
|
|
drawer.render(
|
|
&mut frame,
|
|
contents,
|
|
(w, h),
|
|
(x, y),
|
|
screen_dimensions,
|
|
);
|
|
TraversalAction::DoChildren((x, y))
|
|
} else {
|
|
// we are not display, so our children are neither
|
|
TraversalAction::SkipChildren
|
|
}
|
|
},
|
|
)
|
|
.unwrap();
|
|
}
|
|
});
|
|
}
|
|
if let Err(err) = frame.finish() {
|
|
error!(self.logger, "Error during rendering: {:?}", err);
|
|
}
|
|
}
|
|
|
|
fn error<'a, S: Into<StateProxy<'a>>>(&mut self, _state: S, _device: &mut DrmDevice<GliumDrawer<DrmBackend>>,
|
|
error: DrmError) {
|
|
error!(self.logger, "{:?}", error);
|
|
}
|
|
}
|