xcsp3_rust/constraints/
xinstantiation.rs

1/*=============================================================================
2* parser for CSP instances represented in XCSP3 Format
3*
4* Copyright (c) 2023 xcsp.org (contact @ xcsp.org)
5*
6* Permission is hereby granted, free of charge, to any person obtaining a copy
7* of this software and associated documentation files (the "Software"), to deal
8* in the Software without restriction, including without limitation the rights
9* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10* copies of the Software, and to permit persons to whom the Software is
11* furnished to do so, subject to the following conditions:
12*
13* The above copyright notice and this permission notice shall be included in
14* all copies or substantial portions of the Software.
15*
16* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22* THE SOFTWARE.
23*=============================================================================
24*/
25
26/*
27* <p>@project_name: xcsp3-rust
28* </p>
29* <p>@author: luhan zhen
30* </p>
31* <p>@date:  2023/7/15 14:07
32* </p>
33* <p>@email: zhenlh20@mails.jlu.edu.cn
34* </p>
35* <p>@version: 1.0
36* </p>
37 * <p>@description: 1.0
38* </p>
39 */
40
41pub mod xcsp3_core {
42    use crate::constraints::xconstraint_trait::xcsp3_core::XConstraintTrait;
43    use crate::data_structs::xint_val_var::xcsp3_core::XVarVal;
44    use crate::errors::xcsp3error::xcsp3_core::Xcsp3Error;
45    use std::collections::HashMap;
46    use std::fmt::{Display, Formatter};
47
48    use crate::utils::utils_functions::xcsp3_utils::{list_to_values, list_to_vec_var_val};
49    use crate::variables::xdomain::xcsp3_core::XDomainInteger;
50    use crate::variables::xvariable_set::xcsp3_core::XVariableSet;
51
52    // #[derive(Clone)]
53    pub struct XInstantiation<'a> {
54        scope: Vec<XVarVal>,
55        map: HashMap<String, &'a XDomainInteger>,
56        set: &'a XVariableSet,
57        values: Vec<i32>,
58    }
59
60    impl Display for XInstantiation<'_> {
61        fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
62            let mut ret = String::default();
63            for e in self.scope.iter() {
64                ret.push('(');
65                ret.push_str(&e.to_string());
66                ret.push_str("), ")
67            }
68            ret.push_str(&format!("values = {:?}", self.values));
69            write!(f, "XInstantiation: list =  {}", ret)
70        }
71    }
72
73    impl XConstraintTrait for XInstantiation<'_> {
74        fn get_scope_string(&self) -> &Vec<XVarVal> {
75            &self.scope
76        }
77
78        fn get_scope(&mut self) -> Vec<(&String, &XDomainInteger)> {
79            for e in &self.scope {
80                if let XVarVal::IntVar(s) = e {
81                    if !self.map.contains_key(s) {
82                        if let Ok(vec) = self.set.construct_scope(&[s]) {
83                            for (vs, vv) in vec.into_iter() {
84                                self.map.insert(vs, vv);
85                            }
86                        }
87                    }
88                }
89            }
90            let mut scope_vec_var: Vec<(&String, &XDomainInteger)> = vec![];
91            for e in self.map.iter() {
92                scope_vec_var.push((e.0, e.1))
93            }
94            scope_vec_var
95        }
96    }
97
98    impl<'a> XInstantiation<'a> {
99        pub fn from_str(
100            list: &str,
101            values_str: &str,
102            set: &'a XVariableSet,
103        ) -> Result<Self, Xcsp3Error> {
104            match list_to_vec_var_val(list) {
105                Ok(scope_vec_str) => match list_to_values(values_str) {
106                    Ok(values) => Ok(XInstantiation::new(scope_vec_str, set, values)),
107                    Err(e) => Err(e),
108                },
109                Err(e) => Err(e),
110            }
111        }
112
113        pub fn new(scope: Vec<XVarVal>, set: &'a XVariableSet, values: Vec<i32>) -> Self {
114            XInstantiation {
115                scope,
116                map: Default::default(),
117                set,
118                values,
119            }
120        }
121        pub fn get_values(&self) -> &Vec<i32> {
122            &self.values
123        }
124    }
125}