tracy_full 1.1.0

use std::{
	ffi::{CStr, CString},
	marker::PhantomData,
};

use crate::color::Color;

/// Mark a zone in the current scope.
#[macro_export]
macro_rules! zone {
	() => {
		let loc = $crate::get_location!();
		let _zone = $crate::zone::zone(loc, true);
	};

	($name:literal $(,)?) => {
		let loc = $crate::get_location!($name);
		let _zone = $crate::zone::zone(loc, true);
	};

	($color:expr $(,)?) => {
		let loc = $crate::get_location!($color);
		let _zone = $crate::zone::zone(loc, true);
	};

	($name:literal, $enabled:expr $(,)?) => {
		let loc = $crate::get_location!($name);
		let _zone = $crate::zone::zone(loc, $enabled);
	};

	($color:expr, $enabled:expr $(,)?) => {
		let loc = $crate::get_location!($color);
		let _zone = $crate::zone::zone(loc, $enabled);
	};

	($name:literal, $color:expr, $enabled:expr $(,)?) => {
		let loc = $crate::get_location!($name, $color);
		let _zone = $crate::zone::zone(loc, $enabled);
	};
}

/// Mark a zone in the current scope, sampling the callstack.
#[macro_export]
macro_rules! zone_sample {
	($depth:literal) => {
		let _zone = $crate::zone::zone_sample($crate::get_location!(), $crate::clamp_callstack_depth($depth), true);
	};

	($name:literal, $depth:literal, $enabled:expr $(,)?) => {
		let _zone = $crate::zone::zone_sample(
			$crate::get_location!($name),
			$crate::clamp_callstack_depth($depth),
			$enabled,
		);
	};

	($color:expr, $depth:literal, $enabled:expr $(,)?) => {
		let _zone = $crate::zone::zone_sample(
			$crate::get_location!($color),
			$crate::clamp_callstack_depth($depth),
			$enabled,
		);
	};

	($name:literal, $color:expr, $depth:literal, $enabled:expr $(,)?) => {
		let _zone = $crate::zone::zone_sample(
			$crate::get_location!($name, $color),
			$crate::clamp_callstack_depth($depth),
			$enabled,
		);
	};
}

/// Create a zone.
#[inline(always)]
pub fn zone(loc: &'static ZoneLocation, active: bool) -> Zone {
	#[cfg(feature = "enable")]
	unsafe {
		Zone {
			unsend: PhantomData,
			ctx: sys::___tracy_emit_zone_begin(&loc.loc, active as _),
		}
	}

	#[cfg(not(feature = "enable"))]
	Zone {
		unsend: PhantomData,
		ctx: (),
	}
}

/// Create a callstack sampled zone.
#[inline(always)]
#[cfg(feature = "enable")]
pub fn zone_sample(loc: &'static ZoneLocation, depth: u32, active: bool) -> Zone {
	#[cfg(feature = "enable")]
	unsafe {
		Zone {
			unsend: PhantomData,
			ctx: sys::___tracy_emit_zone_begin_callstack(&loc.loc, depth as _, active as _),
		}
	}

	#[cfg(not(feature = "enable"))]
	Zone {
		unsend: PhantomData,
		ctx: (),
	}
}

pub struct Zone {
	unsend: PhantomData<*mut ()>,
	#[cfg(feature = "enable")]
	ctx: sys::___tracy_c_zone_context,
	#[cfg(not(feature = "enable"))]
	ctx: (),
}

#[cfg(feature = "enable")]
impl Drop for Zone {
	#[inline(always)]
	fn drop(&mut self) {
		unsafe {
			sys::___tracy_emit_zone_end(self.ctx);
		}
	}
}

#[doc(hidden)]
#[cfg(feature = "unstable")]
pub const fn get_function_name_from_local_type<T, const TY: usize>() -> [u8; std::any::type_name::<T>().len() - (TY + 1)]
where
	[(); std::any::type_name::<T>().len() - (TY + 1)]:,
{
	let mut name = [0; std::any::type_name::<T>().len() - (TY + 1)];
	unsafe {
		std::ptr::copy_nonoverlapping(std::any::type_name::<T>().as_ptr(), name.as_mut_ptr(), name.len() - 1);
		name
	}
}

pub struct ZoneLocation {
	#[cfg(feature = "enable")]
	loc: sys::___tracy_source_location_data,
	#[cfg(all(feature = "enable", not(feature = "unstable")))]
	function: CString,
	#[cfg(not(feature = "enable"))]
	pub loc: (),
}

unsafe impl Send for ZoneLocation {}
unsafe impl Sync for ZoneLocation {}

#[cfg(all(feature = "enable", feature = "unstable"))]
impl ZoneLocation {
	pub const fn from_function_file_line(function: &'static CStr, file: &'static CStr, line: u32) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: std::ptr::null(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: Color::none().to_u32(),
			},
		}
	}

	pub const fn from_name_function_file_line(
		name: &'static CStr, function: &'static CStr, file: &'static CStr, line: u32,
	) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: name.as_ptr(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: Color::none().to_u32(),
			},
		}
	}

	pub const fn from_function_file_line_color(
		function: &'static CStr, file: &'static CStr, line: u32, color: Color,
	) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: std::ptr::null(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: color.to_u32(),
			},
		}
	}

	pub const fn from_name_function_file_line_color(
		name: &'static CStr, function: &'static CStr, file: &'static CStr, line: u32, color: Color,
	) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: name.as_ptr(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: color.to_u32(),
			},
		}
	}
}

#[cfg(all(feature = "enable", not(feature = "unstable")))]
impl ZoneLocation {
	pub fn from_function_file_line(function: CString, file: &'static CStr, line: u32) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: std::ptr::null(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: Color::none().to_u32(),
			},
			function,
		}
	}

	pub fn from_name_function_file_line(
		name: &'static CStr, function: CString, file: &'static CStr, line: u32,
	) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: name.as_ptr(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: Color::none().to_u32(),
			},
			function,
		}
	}

	pub fn from_function_file_line_color(function: CString, file: &'static CStr, line: u32, color: Color) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: std::ptr::null(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: color.to_u32(),
			},
			function,
		}
	}

	pub fn from_name_function_file_line_color(
		name: &'static CStr, function: CString, file: &'static CStr, line: u32, color: Color,
	) -> Self {
		Self {
			loc: sys::___tracy_source_location_data {
				name: name.as_ptr(),
				function: function.as_ptr(),
				file: file.as_ptr(),
				line,
				color: color.to_u32(),
			},
			function,
		}
	}
}

/// Get a zone location.
#[cfg(all(feature = "enable", feature = "unstable"))]
#[macro_export]
macro_rules! get_location {
	() => {{
		{
			struct S;
			static FUNCTION: &[u8] = &$crate::zone::get_function_name_from_local_type::<S, 1>();
			static LOC: $crate::zone::ZoneLocation = $crate::zone::ZoneLocation::from_function_file_line(
				unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(FUNCTION) },
				unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
				line!(),
			);
			&LOC
		}
	}};

	($name:literal $(,)?) => {{
		struct S;
		static FUNCTION: &[u8] = &$crate::zone::get_function_name_from_local_type::<S, 1>();
		static LOC: $crate::zone::ZoneLocation = $crate::zone::ZoneLocation::from_name_function_file_line(
			$crate::c_str!($name),
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(FUNCTION) },
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
			line!(),
		);
		&LOC
	}};

	($color:expr $(,)?) => {{
		struct S;
		static FUNCTION: &[u8] = &$crate::zone::get_function_name_from_local_type::<S, 1>();
		static LOC: $crate::zone::ZoneLocation = $crate::zone::ZoneLocation::from_function_file_line_color(
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(FUNCTION) },
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
			line!(),
			$color,
		);
		&LOC
	}};

	($name:literal, $color:expr $(,)?) => {{
		struct S;
		static FUNCTION: &[u8] = &$crate::zone::get_function_name_from_local_type::<S, 1>();
		static LOC: $crate::zone::ZoneLocation = $crate::zone::ZoneLocation::from_name_function_file_line_color(
			$crate::c_str!($name),
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(FUNCTION) },
			unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
			line!(),
			$color,
		);
		&LOC
	}};
}

/// Get a zone location.
#[cfg(all(feature = "enable", not(feature = "unstable")))]
#[macro_export]
macro_rules! get_location {
	() => {{
		struct S;
		static LOC: $crate::once_cell::sync::Lazy<$crate::zone::ZoneLocation> =
			$crate::once_cell::sync::Lazy::new(|| {
				let name = ::std::any::type_name::<S>();
				let name = name[0..name.len() - 3].as_bytes().to_owned();
				$crate::zone::ZoneLocation::from_function_file_line(
					unsafe { ::std::ffi::CString::from_vec_unchecked(name.into()) },
					unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
					line!(),
				)
			});

		&*LOC
	}};

	($name:literal $(,)?) => {{
		struct S;
		static LOC: $crate::once_cell::sync::Lazy<$crate::zone::ZoneLocation> =
			$crate::once_cell::sync::Lazy::new(|| {
				let name = ::std::any::type_name::<S>();
				let name = name[0..name.len() - 3].as_bytes().to_owned();
				$crate::zone::ZoneLocation::from_name_function_file_line(
					$crate::c_str!($name),
					unsafe { ::std::ffi::CString::from_vec_unchecked(name.into()) },
					unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
					line!(),
				)
			});

		&*LOC
	}};

	($color:expr $(,)?) => {{
		struct S;
		static LOC: $crate::once_cell::sync::Lazy<$crate::zone::ZoneLocation> =
			$crate::once_cell::sync::Lazy::new(|| {
				let name = ::std::any::type_name::<S>();
				let name = name[0..name.len() - 3].as_bytes().to_owned();
				$crate::zone::ZoneLocation::from_function_file_line_color(
					unsafe { ::std::ffi::CString::from_vec_unchecked(name.into()) },
					unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
					line!(),
					$color,
				)
			});

		&*LOC
	}};

	($name:literal, $color:expr $(,)?) => {{
		struct S;
		static FN_NAME: $crate::once_cell::sync::Lazy<$crate::zone::ZoneLocation> =
			$crate::once_cell::sync::Lazy::new(|| {
				let name = ::std::any::type_name::<S>();
				let name = name[0..name.len() - 3].as_bytes().to_owned();
				$crate::zone::ZoneLocation::from_name_function_file_line_color(
					$crate::c_str!($name),
					unsafe { ::std::ffi::CString::from_vec_unchecked(name.into()) },
					unsafe { ::std::ffi::CStr::from_bytes_with_nul_unchecked(concat!(file!(), "\0").as_bytes()) },
					line!(),
					$color,
				)
			});

		&*LOC
	}};
}

/// Get a zone location.
#[cfg(not(feature = "enable"))]
#[macro_export]
macro_rules! get_location {
	() => {{
		&$crate::zone::ZoneLocation { loc: () }
	}};

	($name:literal) => {{
		&$crate::zone::ZoneLocation { loc: () }
	}};

	($color:expr) => {{
		&$crate::zone::ZoneLocation { loc: () }
	}};

	($name:literal, $color:expr $(,)?) => {{
		$crate::zone::ZoneLocation { loc: () }
	}};
}