diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/tracy-rs/src | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/tracy-rs/src')
-rw-r--r-- | third_party/rust/tracy-rs/src/disabled.rs | 40 | ||||
-rw-r--r-- | third_party/rust/tracy-rs/src/lib.rs | 15 | ||||
-rw-r--r-- | third_party/rust/tracy-rs/src/profiler.rs | 189 |
3 files changed, 244 insertions, 0 deletions
diff --git a/third_party/rust/tracy-rs/src/disabled.rs b/third_party/rust/tracy-rs/src/disabled.rs new file mode 100644 index 0000000000..14f88b930b --- /dev/null +++ b/third_party/rust/tracy-rs/src/disabled.rs @@ -0,0 +1,40 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +// A set of empty stub macros that are defined when the enable_profiler feature +// is not enabled. This ensures that we don't include callsite definitions in +// the resulting binary. + +#[macro_export] +macro_rules! profile_scope { + ($string:expr) => {} +} + +#[macro_export] +macro_rules! tracy_frame_marker { + () => {} +} + +#[macro_export] +macro_rules! tracy_begin_frame { + ($name:expr) => {} +} + +#[macro_export] +macro_rules! tracy_end_frame { + ($name:expr) => {} +} + +#[macro_export] +macro_rules! tracy_plot { + ($name:expr, $value:expr) => {} +} + +pub unsafe fn load(_: &str) -> bool { + println!("Can't load the tracy profiler unless enable_profiler feature is enabled!"); + false +} + +pub fn register_thread_with_profiler(_: String) { +} diff --git a/third_party/rust/tracy-rs/src/lib.rs b/third_party/rust/tracy-rs/src/lib.rs new file mode 100644 index 0000000000..e444d10da5 --- /dev/null +++ b/third_party/rust/tracy-rs/src/lib.rs @@ -0,0 +1,15 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#[cfg(feature = "enable_profiler")] +pub mod profiler; + +#[cfg(feature = "enable_profiler")] +pub use profiler::{load, register_thread_with_profiler}; + +#[cfg(not(feature = "enable_profiler"))] +mod disabled; + +#[cfg(not(feature = "enable_profiler"))] +pub use disabled::{load, register_thread_with_profiler}; diff --git a/third_party/rust/tracy-rs/src/profiler.rs b/third_party/rust/tracy-rs/src/profiler.rs new file mode 100644 index 0000000000..220aa955e8 --- /dev/null +++ b/third_party/rust/tracy-rs/src/profiler.rs @@ -0,0 +1,189 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +use minidl::Library; + +// Must match `___tracy_source_location_data` struct in TracyC.h +#[repr(C)] +pub struct SourceLocation { + pub name: *const u8, + pub function: *const u8, + pub file: *const u8, + pub line: u32, + pub color: u32, +} + +// Must match `___tracy_c_zone_context` in TracyC.h +#[repr(C)] +#[derive(Copy, Clone)] +pub struct ZoneContext { + id: u32, + active: i32, +} + +// Define a set of no-op implementation functions that are called if the tracy +// shared library is not loaded. +extern "C" fn unimpl_mark_frame(_: *const u8) {} +extern "C" fn unimpl_mark_frame_start(_: *const u8) {} +extern "C" fn unimpl_mark_frame_end(_: *const u8) {} +extern "C" fn unimpl_zone_begin(_: *const SourceLocation, _: i32) -> ZoneContext { + ZoneContext { + id: 0, + active: 0, + } +} +extern "C" fn unimpl_zone_end(_: ZoneContext) {} +extern "C" fn unimpl_zone_text(_: ZoneContext, _: *const u8, _: usize) {} + +extern "C" fn unimpl_plot(_: *const u8, _: f64) {} + +// Function pointers to the tracy API functions (if loaded), or the no-op functions above. +pub static mut MARK_FRAME: extern "C" fn(name: *const u8) = unimpl_mark_frame; +pub static mut MARK_FRAME_START: extern "C" fn(name: *const u8) = unimpl_mark_frame_start; +pub static mut MARK_FRAME_END: extern "C" fn(name: *const u8) = unimpl_mark_frame_end; +pub static mut EMIT_ZONE_BEGIN: extern "C" fn(srcloc: *const SourceLocation, active: i32) -> ZoneContext = unimpl_zone_begin; +pub static mut EMIT_ZONE_END: extern "C" fn(ctx: ZoneContext) = unimpl_zone_end; +pub static mut EMIT_ZONE_TEXT: extern "C" fn(ctx: ZoneContext, txt: *const u8, size: usize) = unimpl_zone_text; +pub static mut EMIT_PLOT: extern "C" fn(name: *const u8, val: f64) = unimpl_plot; + +// Load the tracy library, and get function pointers. This is unsafe since: +// - It must not be called while other threads are calling the functions. +// - It doesn't ensure that the library is not unloaded during use. +pub unsafe fn load(path: &str) -> bool { + match Library::load(path) { + Ok(lib) => { + // If lib loading succeeds, assume we can find all the symbols required. + MARK_FRAME = lib.sym("___tracy_emit_frame_mark\0").unwrap(); + MARK_FRAME_START = lib.sym("___tracy_emit_frame_mark_start\0").unwrap(); + MARK_FRAME_END = lib.sym("___tracy_emit_frame_mark_end\0").unwrap(); + EMIT_ZONE_BEGIN = lib.sym("___tracy_emit_zone_begin\0").unwrap(); + EMIT_ZONE_END = lib.sym("___tracy_emit_zone_end\0").unwrap(); + EMIT_ZONE_TEXT = lib.sym("___tracy_emit_zone_text\0").unwrap(); + EMIT_PLOT = lib.sym("___tracy_emit_plot\0").unwrap(); + + true + } + Err(..) => { + println!("Failed to load the tracy profiling library!"); + + false + } + } +} + +/// A simple stack scope for tracing function execution time +pub struct ProfileScope { + ctx: ZoneContext, +} + +impl ProfileScope { + pub fn new(callsite: &'static SourceLocation) -> Self { + let ctx = unsafe { EMIT_ZONE_BEGIN(callsite, 1) }; + + ProfileScope { + ctx, + } + } + + pub fn text(&self, text: &[u8]) { + unsafe { EMIT_ZONE_TEXT(self.ctx, text.as_ptr(), text.len()) }; + } +} + +impl Drop for ProfileScope { + fn drop(&mut self) { + unsafe { + EMIT_ZONE_END(self.ctx) + } + } +} + +/// Define a profiling scope. +/// +/// This macro records a Tracy 'zone' starting at the point at which the macro +/// occurs, and extending to the end of the block. More precisely, the zone ends +/// when a variable declared by this macro would be dropped, so early returns +/// exit the zone. +/// +/// For example: +/// +/// fn an_operation_that_may_well_take_a_long_time() { +/// profile_scope!("an_operation_that_may_well_take_a_long_time"); +/// ... +/// } +/// +/// The first argument to `profile_scope!` is the name of the zone, and must be +/// a string literal. +/// +/// Tracy zones can also have associated 'text' values that vary from one +/// execution to the next - for example, a zone's text might record the name of +/// the file being processed. The text must be a `CStr` value. You can provide +/// zone text like this: +/// +/// fn run_reftest(test_name: &str) { +/// profile_scope!("run_reftest", text: test_name); +/// ... +/// } +#[macro_export] +macro_rules! profile_scope { + ($string:literal $(, text: $text:expr )? ) => { + const CALLSITE: $crate::profiler::SourceLocation = $crate::profiler::SourceLocation { + name: concat!($string, "\0").as_ptr(), + function: concat!(module_path!(), "\0").as_ptr(), + file: concat!(file!(), "\0").as_ptr(), + line: line!(), + color: 0xff0000ff, + }; + + let _profile_scope = $crate::profiler::ProfileScope::new(&CALLSITE); + + $( + _profile_scope.text(str::as_bytes($text)); + )? + } +} + +/// Define the main frame marker, typically placed after swap_buffers or similar. +#[macro_export] +macro_rules! tracy_frame_marker { + () => { + unsafe { + $crate::profiler::MARK_FRAME(std::ptr::null()) + } + } +} + +/// Define start of an explicit frame marker, typically used for sub-frames. +#[macro_export] +macro_rules! tracy_begin_frame { + ($name:expr) => { + unsafe { + $crate::profiler::MARK_FRAME_START(concat!($name, "\0").as_ptr()) + } + } +} + +/// Define end of an explicit frame marker, typically used for sub-frames. +#[macro_export] +macro_rules! tracy_end_frame { + ($name:expr) => { + unsafe { + $crate::profiler::MARK_FRAME_END(concat!($name, "\0").as_ptr()) + } + } +} + +#[macro_export] +macro_rules! tracy_plot { + ($name:expr, $value:expr) => { + unsafe { + $crate::profiler::EMIT_PLOT(concat!($name, "\0").as_ptr(), $value) + } + } +} + +// Provide a name for this thread to the profiler +pub fn register_thread_with_profiler(_: String) { + // TODO(gw): Add support for passing this to the tracy api +} |