use glib::translate::*;
use std::cell::Cell;
use std::sync::atomic::{AtomicBool, Ordering};
#[cfg(target_os = "macos")]
extern "C" {
fn pthread_main_np() -> i32;
}
thread_local! {
static IS_MAIN_THREAD: Cell<bool> = Cell::new(false)
}
static INITIALIZED: AtomicBool = AtomicBool::new(false);
macro_rules! assert_initialized_main_thread {
() => {
if !crate::rt::is_initialized_main_thread() {
if crate::rt::is_initialized() {
panic!("GTK may only be used from the main thread.");
} else {
panic!("GTK has not been initialized. Call `gtk::init` first.");
}
}
};
}
macro_rules! skip_assert_initialized {
() => {};
}
#[allow(unused_macros)]
macro_rules! assert_not_initialized {
() => {
if crate::rt::is_initialized() {
panic!("This function has to be called before `gtk::init`.");
}
};
}
#[inline]
pub fn is_initialized() -> bool {
skip_assert_initialized!();
INITIALIZED.load(Ordering::Acquire)
}
#[inline]
pub fn is_initialized_main_thread() -> bool {
skip_assert_initialized!();
IS_MAIN_THREAD.with(|c| c.get())
}
pub unsafe fn set_initialized() {
skip_assert_initialized!();
if is_initialized_main_thread() {
return;
} else if is_initialized() {
panic!("Attempted to initialize GTK from two different threads.");
}
#[cfg(target_os = "macos")]
{
if pthread_main_np() == 0 {
panic!("Attempted to initialize GTK on OSX from non-main thread");
}
}
gdk::set_initialized();
INITIALIZED.store(true, Ordering::Release);
IS_MAIN_THREAD.with(|c| c.set(true));
}
#[doc(alias = "gtk_init")]
pub fn init() -> Result<(), glib::BoolError> {
skip_assert_initialized!();
if is_initialized_main_thread() {
return Ok(());
} else if is_initialized() {
panic!("Attempted to initialize GTK from two different threads.");
}
unsafe {
let argv = ::std::env::args().take(1).collect::<Vec<_>>();
if from_glib(ffi::gtk_init_check(&mut 1, &mut argv.to_glib_none().0)) {
let result: bool = from_glib(glib::ffi::g_main_context_acquire(
glib::ffi::g_main_context_default(),
));
if !result {
return Err(glib::bool_error!("Failed to acquire default main context"));
}
set_initialized();
Ok(())
} else {
Err(glib::bool_error!("Failed to initialize GTK"))
}
}
}
#[doc(alias = "gtk_main_quit")]
pub fn main_quit() {
assert_initialized_main_thread!();
unsafe {
if ffi::gtk_main_level() > 0 {
ffi::gtk_main_quit();
} else if cfg!(debug_assertions) {
panic!("Attempted to quit a GTK main loop when none is running.");
}
}
}
#[cfg(test)]
mod tests {
use crate::TEST_THREAD_WORKER;
#[test]
fn init_should_acquire_default_main_context() {
TEST_THREAD_WORKER
.push(move || {
let context = glib::MainContext::ref_thread_default();
assert!(context.is_owner());
})
.expect("Failed to schedule a test call");
while TEST_THREAD_WORKER.unprocessed() > 0 {}
}
}