1use crate::*;
3
4pub mod byte;
5pub mod float;
6pub use byte::*;
7pub use float::*;
8
9#[inline]
10pub unsafe fn lean_alloc_sarray(elem_size: c_uint, size: usize, capacity: usize) -> lean_obj_res {
11 let o = lean_alloc_object(
12 core::mem::size_of::<lean_sarray_object>() + (elem_size as usize) * capacity,
13 ) as *mut lean_sarray_object;
14 lean_set_st_header(o as *mut _, LeanScalarArray as u32, elem_size);
15 (raw_field!(o, lean_sarray_object, m_size) as *mut usize).write(size);
16 (raw_field!(o, lean_sarray_object, m_capacity) as *mut usize).write(capacity);
17 o as *mut _
18}
19
20#[inline(always)]
21pub unsafe fn lean_sarray_elem_size(o: *const lean_object) -> c_uint {
22 debug_assert!(lean_is_sarray(o));
23 lean_ptr_other(o)
24}
25
26#[inline(always)]
27pub unsafe fn lean_sarray_capacity(o: *const lean_object) -> usize {
28 *raw_field!(lean_to_sarray(o as *mut _), lean_sarray_object, m_capacity)
29}
30
31#[inline]
32pub unsafe fn lean_sarray_byte_size(o: *const lean_object) -> usize {
33 size_of::<lean_sarray_object>() + lean_sarray_elem_size(o) as usize * lean_sarray_capacity(o)
34}
35
36#[inline(always)]
37pub unsafe fn lean_sarray_size(o: *const lean_object) -> usize {
38 *raw_field!(lean_to_sarray(o as *mut _), lean_sarray_object, m_size)
39}
40
41#[inline(always)]
42pub unsafe fn lean_sarray_data_byte_size(o: *const lean_object) -> usize {
43 size_of::<lean_sarray_object>() + lean_sarray_elem_size(o) as usize * lean_sarray_size(o)
44}
45
46#[inline(always)]
47pub unsafe fn lean_sarray_set_size(o: u_lean_obj_arg, sz: usize) {
48 debug_assert!(lean_is_exclusive(o));
49 debug_assert!(sz <= lean_sarray_capacity(o));
50 (raw_field!(lean_to_sarray(o), lean_sarray_object, m_size) as *mut usize).write(sz);
51}
52
53#[inline(always)]
54pub unsafe fn lean_sarray_cptr(o: *mut lean_object) -> *mut u8 {
55 raw_field!(lean_to_sarray(o), lean_sarray_object, m_data) as *mut _
56}