intervalues_pyrust 0.1.0

Rust-to-python wrapper around intervalues
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
41
42
43
44
45
46
47
48
49
50
51
52
53

//! # Intervalues_pyrust
//!
//! `intervalues` brings functionality to combine valued intervals together in an efficient manner.
//! This package contains the bindings for using the Rust package in Python.
use pyo3::prelude::*;
use intervalues::{combine_intervals, Interval};
use intfloat::IntFloat;
use itertools::Itertools;
use pyo3::pyfunction;


#[pyfunction]
fn combine_intervals_int<'a>(raw_ivs: Vec<(isize, isize, isize)>) -> Vec<(isize, isize, isize)> {
    let raw_ivs: Vec<Interval<isize, isize>> = raw_ivs
        .iter()
        .map(|x| Interval::new(x.0, x.1, x.2))
        .collect();
    combine_intervals(raw_ivs)
        .to_vec()
        .iter()
        .map(|x| x.to_tuple())
        .collect_vec()
}

#[pyfunction]
fn combine_intervals_float<'a>(
    raw_ivs: Vec<(f32, f32, f32)>,
    nr_decimal: isize,
) -> Vec<(f32, f32, f32)> {
    let raw_ivs: Vec<Interval<IntFloat, IntFloat>> = raw_ivs
        .iter()
        .map(|x| {
            Interval::new(
                IntFloat::from(x.0, nr_decimal),
                IntFloat::from(x.1, nr_decimal),
                IntFloat::from(x.2, nr_decimal),
            )
        })
        .collect();
    combine_intervals(raw_ivs)
        .to_vec()
        .iter()
        .map(|x| (x.to_f32()))
        .collect_vec()
}

#[pymodule]
#[pyo3(name = "intervalues_pyrust")]
pub fn intervalues_pyrust(m: &Bound<'_, PyModule>) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(combine_intervals_int, m)?)?;
    m.add_function(wrap_pyfunction!(combine_intervals_float, m)?)?;
    Ok(())
}