gloss_renderer/
lib.rs

1#![doc = include_str!("../README.md")]
2
3#[macro_use]
4extern crate static_assertions;
5
6use gloss_utils::string::float2string;
7use log::log;
8use re_memory::{AccountingAllocator, MemoryUse};
9
10#[global_allocator]
11static GLOBAL: AccountingAllocator<std::alloc::System> = AccountingAllocator::new(std::alloc::System);
12
13pub mod actor;
14pub mod camera;
15pub mod components;
16pub mod config;
17pub mod forward_renderer;
18pub mod geom;
19#[cfg(feature = "with-gui")]
20pub mod gui;
21pub mod light;
22pub mod logger;
23pub mod plugin_manager;
24pub mod scene;
25pub mod viewer;
26pub mod viewer_headless;
27
28pub use logger::{gloss_setup_logger, gloss_setup_logger_from_config, gloss_setup_logger_from_config_file};
29
30pub fn print_memory_usage_info(mem_use: MemoryUse, verbosity: log::Level) {
31    if let Some(mem_resident) = mem_use.resident {
32        log!(verbosity, "MB resident total: {}", (mem_resident / (1024 * 1024)));
33    }
34    if let Some(mem_counted) = mem_use.counted {
35        log!(verbosity, "MB counted total: {}", (mem_counted / (1024 * 1024)));
36    }
37}
38
39pub fn aa_print_memory_usage_info(show_backtrace: bool, verbosity: log::Level) {
40    if let Some(tracks) = re_memory::accounting_allocator::tracking_stats() {
41        #[allow(clippy::cast_precision_loss)]
42        for cb in tracks.top_callstacks.iter() {
43            let mb_cb = cb.extant.size as f32 / (1024.0 * 1024.0);
44            if show_backtrace {
45                log!(verbosity, "MB: {} Callstack: {}", float2string(mb_cb, 1), cb.readable_backtrace);
46            } else {
47                log!(verbosity, "MB: {} Func: {}", float2string(mb_cb, 1), cb.readable_backtrace,);
48            }
49        }
50    }
51}
52
53fn set_panic_hook() {
54    cfg_if::cfg_if! {
55        if #[cfg(target_arch = "wasm32")] {
56            std::panic::set_hook(Box::new(move |info| {
57                web_sys::console::error_1(&format!("PANICKED: Will print memory usage info:").into());
58                // MemoryUse::capture().print_memory_usage_info(log::Level::Error);
59                // accounting_allocator::print_memory_usage_info(false, log::Level::Error);
60                print_memory_usage_info(MemoryUse::capture(), log::Level::Error);
61                aa_print_memory_usage_info(false, log::Level::Error);
62                console_error_panic_hook::hook(info);
63            }));
64        }else{
65            let default_panic = std::panic::take_hook();
66            std::panic::set_hook(Box::new(move |info| {
67                // println!("PANICKED: Will print memory usage info:");
68                // MemoryUse::capture().print_memory_usage_info(log::Level::Error);
69                // let enabled_backtrace = std::env::var("RUST_BACKTRACE").map_or(false, |_| true);
70                // accounting_allocator::print_memory_usage_info(enabled_backtrace, log::Level::Error);
71                default_panic(info);
72            }));
73        }
74    }
75}
gloss_renderer/lib.rs

gloss_renderer/
lib.rs
