ort 1.14.0-alpha.0

use tracing::{debug, error};

use super::{error::OrtResult, ortsys, sys, AllocatorType, MemType};

#[derive(Debug)]
pub(crate) struct MemoryInfo {
	pub ptr: *mut sys::OrtMemoryInfo
}

impl MemoryInfo {
	#[tracing::instrument]
	pub fn new(allocator: AllocatorType, memory_type: MemType) -> OrtResult<Self> {
		debug!("Creating new OrtMemoryInfo.");

		let mut memory_info_ptr: *mut sys::OrtMemoryInfo = std::ptr::null_mut();
		ortsys![
			unsafe CreateCpuMemoryInfo(allocator.into(), memory_type.into(), &mut memory_info_ptr);
			nonNull(memory_info_ptr)
		];
		Ok(Self { ptr: memory_info_ptr })
	}
}

impl Drop for MemoryInfo {
	#[tracing::instrument]
	fn drop(&mut self) {
		if self.ptr.is_null() {
			error!("OrtMemoryInfo pointer is null, not dropping.");
		} else {
			debug!("Dropping OrtMemoryInfo");
			ortsys![unsafe ReleaseMemoryInfo(self.ptr)];
		}

		self.ptr = std::ptr::null_mut();
	}
}

#[cfg(test)]
mod tests {
	use test_log::test;

	use super::*;

	#[test]
	fn create_memory_info() {
		let memory_info = MemoryInfo::new(AllocatorType::Arena, MemType::Default).unwrap();
		std::mem::drop(memory_info);
	}
}