fpzip_sys/
lib.rs

1#![allow(non_upper_case_globals)]
2#![allow(non_camel_case_types)]
3#![allow(non_snake_case)]
4#![allow(deref_nullptr)]
5
6include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
7
8#[cfg(test)]
9mod tests {
10    use super::*;
11
12    #[test]
13    fn compress_array() {
14        let nx = 100;
15        let ny = 100;
16        let nz = 100;
17
18        let mut array: Vec<f32> = vec![0.0; nx * ny * nz];
19
20        for i in 0..nx {
21            for j in 0..ny {
22                for k in 0..nz {
23                    let x = 2.0 * (i as f32) / (nx as f32);
24                    let y = 2.0 * (j as f32) / (ny as f32);
25                    let z = 2.0 * (k as f32) / (nz as f32);
26                    array[i + nx * (j + ny * k)] = (-(x * x + y * y + z * z)).exp();
27                }
28            }
29        }
30
31        /* allocate buffer for compressed data */
32        let bufsize = 1024 + array.len() * std::mem::size_of::<f32>();
33        let mut buffer: Vec<u8> = vec![0; bufsize];
34
35        /* compress to memory */
36        let fpz =
37            unsafe { fpzip_write_to_buffer(buffer.as_mut_ptr() as *mut std::ffi::c_void, bufsize) };
38
39        unsafe {
40            (*fpz).type_ = FPZIP_TYPE_FLOAT as i32;
41            (*fpz).prec = 16;
42            (*fpz).nx = nx as i32;
43            (*fpz).ny = ny as i32;
44            (*fpz).nz = nz as i32;
45            (*fpz).nf = 1;
46        }
47
48        let stat = unsafe { fpzip_write_header(fpz) };
49
50        if stat == 0 {
51            assert!(false)
52        };
53
54        let outbytes = unsafe { fpzip_write(fpz, array.as_ptr() as *const std::ffi::c_void) };
55
56        if outbytes == 0 {
57            assert!(false)
58        };
59
60        unsafe { fpzip_write_close(fpz) };
61
62        assert!(true);
63    }
64}