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
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170

use std::mem;

// use imlib2::Drawable;
use libc;
use x11::xlib as x;

use {Atom, Display, Drawable, GetDisplay, Image, Rectangle, X11Error};

/// A wrapper around a window handle.
#[derive(Copy, Clone, PartialEq, Eq)]
pub struct Window {
    pub(super) display: *mut x::Display,
    pub(super) inner: x::Window,
}

/// Window Attributes
pub struct WindowAttributes {
    pub(super) display: *mut x::Display,
    pub(self) inner: *mut x::XWindowAttributes,
}

impl Window {
    /// Create a new window
    pub fn create(
        display: &Display,
        parent: Option<Window>,
        location: Rectangle,
    ) -> Result<Window, X11Error> {
        let parent = match parent {
            Some(parent) => parent,
            None => display.get_default_root_window()?,
        };
        let visual = display.default_visual(0);
        let window = unsafe {
            x::XCreateWindow(
                display.as_raw(),
                parent.as_raw(),
                location.x as i32,
                location.y as i32,
                location.width,
                location.height,
                0,
                0,
                0,
                visual.as_raw(),
                0,
                0 as *mut x::XSetWindowAttributes,
            )
        };
        Ok(Window {
            display: display.as_raw(),
            inner: window,
        })
    }

    /// Create a new Window instance from an existing ID
    pub fn create_from_handle(display: &Display, id: u64) -> Result<Window, X11Error> {
        Ok(Window {
            display: display.as_raw(),
            inner: id,
        })
    }

    /// Get window attributes.
    pub fn get_attributes(&self) -> Result<WindowAttributes, X11Error> {
        let attr = unsafe {
            libc::malloc(mem::size_of::<x::XWindowAttributes>()) as *mut x::XWindowAttributes
        };
        let result = unsafe { x::XGetWindowAttributes(self.display, self.inner, attr) };
        match result {
            0 => Err(X11Error::GetAttributesError),
            _ => Ok(WindowAttributes {
                display: self.display,
                inner: attr,
            }),
        }
    }

    /// Get the raw window handle
    pub fn as_raw(&self) -> x::Window {
        self.inner
    }

    /// Get image
    pub fn get_image(&self) -> Result<Image, X11Error> {
        let attr = self.get_attributes()?;
        Drawable::get_image(
            self,
            Rectangle::new(
                attr.get_x(),
                attr.get_y(),
                attr.get_width(),
                attr.get_height(),
            ),
        )
    }

    /// Change window property
    // TODO: make it more general
    pub fn change_property(&self, key: &Atom, val: &Atom) {
        use std::mem::transmute;
        let v = val.as_raw();
        unsafe {
            x::XChangeProperty(
                self.display,
                self.inner,
                key.as_raw(),
                x::XA_ATOM,
                32,
                x::PropModeReplace,
                transmute(&v),
                1,
            );
        }
    }
}

impl GetDisplay for Window {
    fn get_display(&self) -> *mut x::Display {
        self.display
    }
}

impl Drawable for Window {
    fn as_drawable(&self) -> x::Drawable {
        self.inner
    }
}

// impl AsRef<Drawable> for Window {
//     fn as_ref(&self) -> &Drawable {
//         &self.inner
//     }
// }

impl WindowAttributes {
    /// Gets the width of the window
    pub fn get_x(&self) -> i32 {
        unsafe { (*self.inner).x as i32 }
    }

    /// Gets the height of the window
    pub fn get_y(&self) -> i32 {
        unsafe { (*self.inner).y as i32 }
    }

    /// Gets the width of the window
    pub fn get_width(&self) -> u32 {
        unsafe { (*self.inner).width as u32 }
    }

    /// Gets the height of the window
    pub fn get_height(&self) -> u32 {
        unsafe { (*self.inner).height as u32 }
    }

    /// Get the root window of this window
    pub fn get_root(&self) -> Window {
        Window {
            display: self.display,
            inner: unsafe { (*self.inner).root },
        }
    }
}

impl Drop for WindowAttributes {
    fn drop(&mut self) {
        unsafe { libc::free(self.inner as *mut libc::c_void) };
    }
}