amplrs 0.1.1

Safe Rust API for AMPL optimization modeling system
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use crate::ffi;
use crate::ampl::Ampl;

use libc::c_char;
use std::ffi::{CStr, CString};
use std::ptr;

pub struct Parameter {
    raw: *mut ffi::AMPL,
    name: String,
}

impl Parameter {
    pub fn new(ampl: &mut Ampl, name: String) -> Self {
        Parameter { raw: ampl.raw, name:name }
    }

    pub fn print(&self) {
        println!("Parameter: {}", self.name);
    }

    pub fn indexarity(&self) -> usize {
        let name = CString::new(&*self.name).unwrap();
        let mut indexarity: usize = 0;
        unsafe { ffi::AMPL_EntityGetIndexarity(self.raw, name.as_ptr(), &mut indexarity as *mut usize) };
        indexarity
    }

    pub fn num_instances(&self) -> usize {
        let name = CString::new(&*self.name).unwrap();
        let mut num_instances: usize = 0;
        unsafe { ffi::AMPL_EntityGetNumInstances(self.raw, name.as_ptr(), &mut num_instances as *mut usize) };
        num_instances
    }

    pub fn declaration(&mut self) -> String {
        let name = CString::new(&*self.name).unwrap();
        let mut value_ptr: *mut c_char = ptr::null_mut();
        unsafe {
            ffi::AMPL_EntityGetDeclaration(self.raw, name.as_ptr(), &mut value_ptr);
            if value_ptr.is_null() {
                return String::new();
            }
            let value_str = CStr::from_ptr(value_ptr).to_str().unwrap().to_string();
            ffi::AMPL_StringFree(&mut value_ptr);
            value_str
        }
    }

    /// Assign all values by position (must match the number of instances in the parameter).
    pub fn set_all_double_values(&self, values: &[f64]) {
        let name = CString::new(&*self.name).unwrap();
        unsafe {
            ffi::AMPL_ParameterSetArgsDoubleValues(self.raw, name.as_ptr(), values.len(), values.as_ptr());
        }
    }

    /// Assign values to specific instances identified by string indices.
    pub fn set_some_double_values(&self, indices: &[&str], values: &[f64]) {
        assert_eq!(indices.len(), values.len());
        let name = CString::new(&*self.name).unwrap();

        let mut tuples: Vec<*mut ffi::AMPL_TUPLE> = indices.iter().map(|&s| {
            let cs = CString::new(s).unwrap();
            let p = cs.as_ptr();
            let mut tuple: *mut ffi::AMPL_TUPLE = ptr::null_mut();
            unsafe { ffi::AMPL_TupleCreateString(&mut tuple, 1, &p) };
            std::mem::forget(cs);
            tuple
        }).collect();

        let mut vals = values.to_vec();
        unsafe {
            ffi::AMPL_ParameterSetSomeDoubleValues(
                self.raw,
                name.as_ptr(),
                tuples.len(),
                tuples.as_mut_ptr(),
                vals.as_mut_ptr(),
            );
            for t in &mut tuples {
                ffi::AMPL_TupleFree(t);
            }
        }
    }

    pub fn drop(&self) {
        let name = CString::new(&*self.name).unwrap();
        unsafe { ffi::AMPL_EntityDrop(self.raw, name.as_ptr()) };
    }

    pub fn restore(&self) {
        let name = CString::new(&*self.name).unwrap();
        unsafe { ffi::AMPL_EntityRestore(self.raw, name.as_ptr()) };
    }
}