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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#![allow(non_upper_case_globals)]
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]
include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn compress_array() {
let nx = 100;
let ny = 100;
let nz = 100;
let mut array: Vec<f32> = vec![0.0; nx * ny * nz];
for i in 0..nx {
for j in 0..ny {
for k in 0..nz {
let x = 2.0 * (i as f32) / (nx as f32);
let y = 2.0 * (j as f32) / (ny as f32);
let z = 2.0 * (k as f32) / (nz as f32);
array[i + nx * (j + ny * k)] = (-(x * x + y * y + z * z)).exp();
}
}
}
let bufsize = 1024 + array.len() * std::mem::size_of::<f32>();
let mut buffer: Vec<u8> = vec![0; bufsize];
let fpz = unsafe {
fpzip_write_to_buffer(buffer.as_mut_ptr() as *mut std::ffi::c_void, bufsize as u64)
};
unsafe {
(*fpz).type_ = FPZIP_TYPE_FLOAT as i32;
(*fpz).prec = 16;
(*fpz).nx = nx as i32;
(*fpz).ny = ny as i32;
(*fpz).nz = nz as i32;
(*fpz).nf = 1;
}
let stat = unsafe { fpzip_write_header(fpz) };
if stat == 0 {
assert!(false)
};
let outbytes = unsafe { fpzip_write(fpz, array.as_ptr() as *const std::ffi::c_void) };
if outbytes == 0 {
assert!(false)
};
unsafe { fpzip_write_close(fpz) };
assert!(true);
}
}