ort-web 0.1.0+1.23

ONNX Runtime on the web 🌐 - An alternative backend for ort
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use alloc::boxed::Box;

use crate::binding;

pub(crate) struct Environment {
	pub with_telemetry: bool
}

impl Environment {
	pub fn new_sys() -> *mut ort_sys::OrtEnv {
		(Box::leak(Box::new(Self { with_telemetry: true })) as *mut Environment).cast()
	}

	pub unsafe fn cast_from_sys<'e>(ptr: *const ort_sys::OrtEnv) -> &'e Environment {
		unsafe { &*ptr.cast::<Environment>() }
	}

	pub unsafe fn cast_from_sys_mut<'e>(ptr: *mut ort_sys::OrtEnv) -> &'e mut Environment {
		unsafe { &mut *ptr.cast::<Environment>() }
	}

	pub unsafe fn consume_sys(ptr: *mut ort_sys::OrtEnv) -> Box<Environment> {
		unsafe { Box::from_raw(ptr.cast::<Environment>()) }
	}

	pub fn send_telemetry_event(&self, event: TelemetryEvent) {
		if !self.with_telemetry {
			return;
		}

		let _ = match event {
			TelemetryEvent::SessionInit => binding::track_session_init()
		};
	}
}

#[derive(Debug)]
pub enum TelemetryEvent {
	SessionInit
}