Skip to content

Commit

Permalink
Unify Window struct across backends (#162)
Browse files Browse the repository at this point in the history
The public Window struct holds a mutable reference to `platform::Window`, which is a pattern that doesn't make sense for all backends. On Windows, the `platform::Window` struct itself just holds another (immutable) reference, and on macOS the `platform::Window` struct has to be wrapped in a `RefCell` so that a mutable reference to it can be formed.

Change the public `Window` struct to hold `platform::Window` directly, and change `platform::Window` in the macOS and X11 backends so that it simply holds a reference to another `WindowInner` struct similarly to the Windows backend. This allows us to remove the platform conditional in the declaration of the top-level `Window` struct. It also allows us to remove the `RefCell` wrapping `platform::Window` in the macOS backend and replace it with `Cell`s wrapping its individual fields.
  • Loading branch information
micahrj authored Dec 17, 2023
1 parent 937ef96 commit fdb43ea
Show file tree
Hide file tree
Showing 5 changed files with 110 additions and 103 deletions.
84 changes: 45 additions & 39 deletions src/macos/window.rs
Original file line number Diff line number Diff line change
Expand Up @@ -95,22 +95,26 @@ impl Drop for ParentHandle {
}
}

pub struct Window {
struct WindowInner {
/// Only set if we created the parent window, i.e. we are running in
/// parentless mode
ns_app: Option<id>,
ns_app: Cell<Option<id>>,
/// Only set if we created the parent window, i.e. we are running in
/// parentless mode
ns_window: Option<id>,
ns_window: Cell<Option<id>>,
/// Our subclassed NSView
ns_view: id,
close_requested: bool,
close_requested: Cell<bool>,

#[cfg(feature = "opengl")]
gl_context: Option<GlContext>,
}

impl Window {
pub struct Window<'a> {
inner: &'a WindowInner,
}

impl<'a> Window<'a> {
pub fn open_parented<P, H, B>(parent: &P, options: WindowOpenOptions, build: B) -> WindowHandle
where
P: HasRawWindowHandle,
Expand All @@ -135,19 +139,19 @@ impl Window {

let ns_view = unsafe { create_view(&options) };

let window = Window {
ns_app: None,
ns_window: None,
let window_inner = WindowInner {
ns_app: Cell::new(None),
ns_window: Cell::new(None),
ns_view,
close_requested: false,
close_requested: Cell::new(false),

#[cfg(feature = "opengl")]
gl_context: options
.gl_config
.map(|gl_config| Self::create_gl_context(None, ns_view, gl_config)),
};

let window_handle = Self::init(true, window, window_info, build);
let window_handle = Self::init(true, window_inner, window_info, build);

unsafe {
let _: id = msg_send![handle.ns_view as *mut Object, addSubview: ns_view];
Expand Down Expand Up @@ -211,19 +215,19 @@ impl Window {

let ns_view = unsafe { create_view(&options) };

let window = Window {
ns_app: Some(app),
ns_window: Some(ns_window),
let window_inner = WindowInner {
ns_app: Cell::new(Some(app)),
ns_window: Cell::new(Some(ns_window)),
ns_view,
close_requested: false,
close_requested: Cell::new(false),

#[cfg(feature = "opengl")]
gl_context: options
.gl_config
.map(|gl_config| Self::create_gl_context(Some(ns_window), ns_view, gl_config)),
};

let _ = Self::init(false, window, window_info, build);
let _ = Self::init(false, window_inner, window_info, build);

unsafe {
ns_window.setContentView_(ns_view);
Expand All @@ -236,24 +240,26 @@ impl Window {
}

fn init<H, B>(
parented: bool, mut window: Window, window_info: WindowInfo, build: B,
parented: bool, window_inner: WindowInner, window_info: WindowInfo, build: B,
) -> WindowHandle
where
H: WindowHandler + 'static,
B: FnOnce(&mut crate::Window) -> H,
B: Send + 'static,
{
let window_handler = Box::new(build(&mut crate::Window::new(&mut window)));
let mut window = crate::Window::new(Window { inner: &window_inner });
let window_handler = Box::new(build(&mut window));

let (parent_handle, window_handle) = ParentHandle::new(window.raw_window_handle());
let parent_handle = if parented { Some(parent_handle) } else { None };

let retain_count_after_build: usize = unsafe { msg_send![window.ns_view, retainCount] };
let retain_count_after_build: usize =
unsafe { msg_send![window_inner.ns_view, retainCount] };

let ns_view = window.ns_view;
let ns_view = window_inner.ns_view;

let window_state_ptr = Box::into_raw(Box::new(WindowState {
window: RefCell::new(window),
window_inner,
window_handler: RefCell::new(window_handler),
keyboard_state: KeyboardState::new(),
frame_timer: Cell::new(None),
Expand All @@ -272,27 +278,27 @@ impl Window {
}

pub fn close(&mut self) {
self.close_requested = true;
self.inner.close_requested.set(true);
}

pub fn resize(&mut self, size: Size) {
// NOTE: macOS gives you a personal rave if you pass in fractional pixels here. Even though
// the size is in fractional pixels.
let size = NSSize::new(size.width.round(), size.height.round());

unsafe { NSView::setFrameSize(self.ns_view, size) };
unsafe { NSView::setFrameSize(self.inner.ns_view, size) };
unsafe {
let _: () = msg_send![self.ns_view, setNeedsDisplay: YES];
let _: () = msg_send![self.inner.ns_view, setNeedsDisplay: YES];
}

// When using OpenGL the `NSOpenGLView` needs to be resized separately? Why? Because macOS.
#[cfg(feature = "opengl")]
if let Some(gl_context) = &self.gl_context {
if let Some(gl_context) = &self.inner.gl_context {
gl_context.resize(size);
}

// If this is a standalone window then we'll also need to resize the window itself
if let Some(ns_window) = self.ns_window {
if let Some(ns_window) = self.inner.ns_window.get() {
unsafe { NSWindow::setContentSize_(ns_window, size) };
}
}
Expand All @@ -303,7 +309,7 @@ impl Window {

#[cfg(feature = "opengl")]
pub fn gl_context(&self) -> Option<&GlContext> {
self.gl_context.as_ref()
self.inner.gl_context.as_ref()
}

#[cfg(feature = "opengl")]
Expand All @@ -318,7 +324,7 @@ impl Window {
}

pub(super) struct WindowState {
window: RefCell<Window>,
window_inner: WindowInner,
window_handler: RefCell<Box<dyn WindowHandler>>,
keyboard_state: KeyboardState,
frame_timer: Cell<Option<CFRunLoopTimer>>,
Expand All @@ -337,13 +343,13 @@ impl WindowState {
}

pub(super) fn trigger_event(&self, event: Event) -> EventStatus {
let mut window = self.window.borrow_mut();
self.window_handler.borrow_mut().on_event(&mut crate::Window::new(&mut window), event)
let mut window = crate::Window::new(Window { inner: &self.window_inner });
self.window_handler.borrow_mut().on_event(&mut window, event)
}

pub(super) fn trigger_frame(&self) {
let mut window = self.window.borrow_mut();
self.window_handler.borrow_mut().on_frame(&mut crate::Window::new(&mut window));
let mut window = crate::Window::new(Window { inner: &self.window_inner });
self.window_handler.borrow_mut().on_frame(&mut window);

let mut do_close = false;

Expand All @@ -359,14 +365,14 @@ impl WindowState {
*/

// Check if the user requested the window to close
if window.close_requested {
if self.window_inner.close_requested.get() {
do_close = true;
window.close_requested = false;
self.window_inner.close_requested.set(false);
}

if do_close {
unsafe {
let ns_window = self.window.borrow_mut().ns_window.take();
let ns_window = self.window_inner.ns_window.take();
if let Some(ns_window) = ns_window {
ns_window.close();
} else {
Expand Down Expand Up @@ -430,26 +436,26 @@ impl WindowState {
window_state.trigger_event(Event::Window(WindowEvent::WillClose));

// If in non-parented mode, we want to also quit the app altogether
let app = window_state.window.borrow_mut().ns_app.take();
let app = window_state.window_inner.ns_app.take();
if let Some(app) = app {
app.stop_(app);
}
}
}

unsafe impl HasRawWindowHandle for Window {
unsafe impl<'a> HasRawWindowHandle for Window<'a> {
fn raw_window_handle(&self) -> RawWindowHandle {
let ns_window = self.ns_window.unwrap_or(ptr::null_mut()) as *mut c_void;
let ns_window = self.inner.ns_window.get().unwrap_or(ptr::null_mut()) as *mut c_void;

let mut handle = AppKitWindowHandle::empty();
handle.ns_window = ns_window;
handle.ns_view = self.ns_view as *mut c_void;
handle.ns_view = self.inner.ns_view as *mut c_void;

RawWindowHandle::AppKit(handle)
}
}

unsafe impl HasRawDisplayHandle for Window {
unsafe impl<'a> HasRawDisplayHandle for Window<'a> {
fn raw_display_handle(&self) -> RawDisplayHandle {
RawDisplayHandle::AppKit(AppKitDisplayHandle::empty())
}
Expand Down
3 changes: 1 addition & 2 deletions src/win/drop_target.rs
Original file line number Diff line number Diff line change
Expand Up @@ -90,8 +90,7 @@ impl DropTarget {
};

unsafe {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let event = Event::Mouse(event);
let event_status =
Expand Down
24 changes: 8 additions & 16 deletions src/win/window.rs
Original file line number Diff line number Diff line change
Expand Up @@ -168,8 +168,7 @@ unsafe fn wnd_proc_inner(
) -> Option<LRESULT> {
match msg {
WM_MOUSEMOVE => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let x = (lparam & 0xFFFF) as i16 as i32;
let y = ((lparam >> 16) & 0xFFFF) as i16 as i32;
Expand All @@ -189,8 +188,7 @@ unsafe fn wnd_proc_inner(
Some(0)
}
WM_MOUSEWHEEL | WM_MOUSEHWHEEL => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let value = (wparam >> 16) as i16;
let value = value as i32;
Expand All @@ -214,8 +212,7 @@ unsafe fn wnd_proc_inner(
}
WM_LBUTTONDOWN | WM_LBUTTONUP | WM_MBUTTONDOWN | WM_MBUTTONUP | WM_RBUTTONDOWN
| WM_RBUTTONUP | WM_XBUTTONDOWN | WM_XBUTTONUP => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let mut mouse_button_counter = window_state.mouse_button_counter.get();

Expand Down Expand Up @@ -278,8 +275,7 @@ unsafe fn wnd_proc_inner(
None
}
WM_TIMER => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

if wparam == WIN_FRAME_TIMER {
window_state.handler.borrow_mut().as_mut().unwrap().on_frame(&mut window);
Expand All @@ -290,8 +286,7 @@ unsafe fn wnd_proc_inner(
WM_CLOSE => {
// Make sure to release the borrow before the DefWindowProc call
{
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

window_state
.handler
Expand All @@ -307,8 +302,7 @@ unsafe fn wnd_proc_inner(
}
WM_CHAR | WM_SYSCHAR | WM_KEYDOWN | WM_SYSKEYDOWN | WM_KEYUP | WM_SYSKEYUP
| WM_INPUTLANGCHANGE => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let opt_event =
window_state.keyboard_state.borrow_mut().process_message(hwnd, msg, wparam, lparam);
Expand All @@ -329,8 +323,7 @@ unsafe fn wnd_proc_inner(
}
}
WM_SIZE => {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

let width = (lparam & 0xFFFF) as u16 as u32;
let height = ((lparam >> 16) & 0xFFFF) as u16 as u32;
Expand Down Expand Up @@ -676,8 +669,7 @@ impl Window<'_> {
});

let handler = {
let mut window = window_state.create_window();
let mut window = crate::Window::new(&mut window);
let mut window = crate::Window::new(window_state.create_window());

build(&mut window)
};
Expand Down
9 changes: 3 additions & 6 deletions src/window.rs
Original file line number Diff line number Diff line change
Expand Up @@ -50,23 +50,20 @@ pub trait WindowHandler {
}

pub struct Window<'a> {
#[cfg(target_os = "windows")]
window: &'a mut platform::Window<'a>,
#[cfg(not(target_os = "windows"))]
window: &'a mut platform::Window,
window: platform::Window<'a>,

// so that Window is !Send on all platforms
phantom: PhantomData<*mut ()>,
}

impl<'a> Window<'a> {
#[cfg(target_os = "windows")]
pub(crate) fn new(window: &'a mut platform::Window<'a>) -> Window<'a> {
pub(crate) fn new(window: platform::Window<'a>) -> Window<'a> {
Window { window, phantom: PhantomData }
}

#[cfg(not(target_os = "windows"))]
pub(crate) fn new(window: &mut platform::Window) -> Window {
pub(crate) fn new(window: platform::Window) -> Window {
Window { window, phantom: PhantomData }
}

Expand Down
Loading

0 comments on commit fdb43ea

Please sign in to comment.