smithay/anvil/src/main.rs

#![warn(rust_2018_idioms)]

#[macro_use]
extern crate glium;
#[macro_use]
extern crate slog;
#[macro_use(define_roles)]
extern crate smithay;

use std::{
    cell::RefCell,
    rc::Rc,
    sync::{
        atomic::{AtomicBool, Ordering},
        Arc,
    },
};

use slog::Drain;
use smithay::reexports::{
    calloop::{generic::Generic, mio::Interest, EventLoop},
    wayland_server::Display,
};

#[macro_use]
mod shaders;
mod buffer_utils;
mod glium_drawer;
mod input_handler;
mod shell;
mod shm_load;
#[cfg(feature = "udev")]
mod udev;
mod window_map;
#[cfg(feature = "winit")]
mod winit;

static POSSIBLE_BACKENDS: &[&str] = &[
    #[cfg(feature = "winit")]
    "--winit : Run anvil as a X11 or Wayland client using winit.",
    #[cfg(feature = "udev")]
    "--tty-udev : Run anvil as a tty udev client (requires root if without logind).",
];

pub struct AnvilState {
    pub running: Arc<AtomicBool>,
}

impl Default for AnvilState {
    fn default() -> AnvilState {
        AnvilState {
            running: Arc::new(AtomicBool::new(true)),
        }
    }
}

fn main() {
    // A logger facility, here we use the terminal here
    let log = slog::Logger::root(
        slog_async::Async::default(slog_term::term_full().fuse()).fuse(),
        o!(),
    );

    let event_loop = EventLoop::<AnvilState>::new().unwrap();
    let display = Rc::new(RefCell::new(Display::new()));

    // Glue for event dispatching
    let mut wayland_event_source = Generic::from_raw_fd(display.borrow().get_poll_fd());
    wayland_event_source.set_interest(Interest::READABLE);
    let _wayland_source = event_loop.handle().insert_source(wayland_event_source, {
        let display = display.clone();
        let log = log.clone();
        move |_, state: &mut AnvilState| {
            let mut display = display.borrow_mut();
            match display.dispatch(std::time::Duration::from_millis(0), state) {
                Ok(_) => {}
                Err(e) => {
                    error!(log, "I/O error on the Wayland display: {}", e);
                    state.running.store(false, Ordering::SeqCst);
                }
            }
        }
    });

    let arg = ::std::env::args().nth(1);
    match arg.as_ref().map(|s| &s[..]) {
        #[cfg(feature = "winit")]
        Some("--winit") => {
            info!(log, "Starting anvil with winit backend");
            if let Err(()) = winit::run_winit(display, event_loop, log.clone()) {
                crit!(log, "Failed to initialize winit backend.");
            }
        }
        #[cfg(feature = "udev")]
        Some("--tty-udev") => {
            info!(log, "Starting anvil on a tty using udev");
            if let Err(()) = udev::run_udev(display, event_loop, log.clone()) {
                crit!(log, "Failed to initialize tty backend.");
            }
        }
        _ => {
            println!("USAGE: anvil --backend");
            println!();
            println!("Possible backends are:");
            for b in POSSIBLE_BACKENDS {
                println!("\t{}", b);
            }
        }
    }
}