xcsp3_rust/constraints/
xall_different.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/14 22:50
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
44    use crate::data_structs::xint_val_var::xcsp3_core::XVarVal;
45    use crate::errors::xcsp3error::xcsp3_core::Xcsp3Error;
46    use crate::utils::utils_functions::xcsp3_utils::list_to_vec_var_val;
47    use crate::variables::xdomain::xcsp3_core::XDomainInteger;
48    use crate::variables::xvariable_set::xcsp3_core::XVariableSet;
49    use std::collections::HashMap;
50    use std::fmt::{Display, Formatter};
51
52    // #[derive(Clone)]
53    pub struct XAllDifferent<'a> {
54        scope: Vec<XVarVal>,
55        map: HashMap<String, &'a XDomainInteger>,
56        set: &'a XVariableSet,
57        // scope_vec_var: Vec<(String, &'a XDomainInteger)>,
58    }
59
60    impl Display for XAllDifferent<'_> {
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            write!(f, "XAllDifferent: list =  {}", ret)
69        }
70    }
71
72    impl XConstraintTrait for XAllDifferent<'_> {
73        fn get_scope_string(&self) -> &Vec<XVarVal> {
74            &self.scope
75        }
76
77        fn get_scope(&mut self) -> Vec<(&String, &XDomainInteger)> {
78            for e in &self.scope {
79                if let XVarVal::IntVar(s) = e {
80                    if !&self.map.contains_key(s) {
81                        if let Ok(vec) = self.set.construct_scope(&[s]) {
82                            for (vs, vv) in vec.into_iter() {
83                                self.map.insert(vs, vv);
84                            }
85                        }
86                    }
87                }
88            }
89            let mut scope_vec_var: Vec<(&String, &XDomainInteger)> = vec![];
90            for e in self.map.iter() {
91                scope_vec_var.push((e.0, e.1))
92            }
93            scope_vec_var
94        }
95    }
96
97    impl<'a> XAllDifferent<'a> {
98        pub fn from_str_vec(scope_vec_str: Vec<XVarVal>, set: &'a XVariableSet) -> Self {
99            XAllDifferent::new(scope_vec_str, set)
100        }
101
102        pub fn from_str(list: &str, set: &'a XVariableSet) -> Result<Self, Xcsp3Error> {
103            match list_to_vec_var_val(list) {
104                Ok(scope_vec_str) => Ok(XAllDifferent::new(scope_vec_str, set)),
105                Err(e) => Err(e),
106            }
107        }
108        pub fn new(scope: Vec<XVarVal>, set: &'a XVariableSet) -> Self {
109            XAllDifferent {
110                scope,
111                map: Default::default(),
112                set,
113            }
114        }
115    }
116}