xcsp3_rust/constraints/
xall_different_except.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    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::{
49        list_to_vec_var_val, list_with_bracket_comma_to_values,
50    };
51    use crate::variables::xdomain::xcsp3_core::XDomainInteger;
52    use crate::variables::xvariable_set::xcsp3_core::XVariableSet;
53
54    // #[derive(Clone)]
55    pub struct XAllDifferentExcept<'a> {
56        map: HashMap<String, &'a XDomainInteger>,
57        scope: Vec<XVarVal>,
58        set: &'a XVariableSet,
59        except: Vec<XVarVal>,
60    }
61
62    impl Display for XAllDifferentExcept<'_> {
63        fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
64            let mut ret = String::default();
65            for e in self.scope.iter() {
66                ret.push('(');
67                ret.push_str(&e.to_string());
68                ret.push_str("), ")
69            }
70            ret.push_str("except = (");
71            for (i, e) in self.except.iter().enumerate() {
72                ret.push_str(&e.to_string());
73                if i != self.except.len() - 1 {
74                    ret.push_str(", ");
75                }
76            }
77            write!(f, "XAllDifferentExcept: list =  {})", ret)
78        }
79    }
80
81    impl XConstraintTrait for XAllDifferentExcept<'_> {
82        fn get_scope_string(&self) -> &Vec<XVarVal> {
83            &self.scope
84        }
85
86        fn get_scope(&mut self) -> Vec<(&String, &XDomainInteger)> {
87            for e in &self.scope {
88                if let XVarVal::IntVar(s) = e {
89                    if !self.map.contains_key(s) {
90                        if let Ok(vec) = self.set.construct_scope(&[s]) {
91                            for (vs, vv) in vec.into_iter() {
92                                self.map.insert(vs, vv);
93                            }
94                        }
95                    }
96                }
97            }
98            let mut scope_vec_var: Vec<(&String, &XDomainInteger)> = vec![];
99            for e in self.map.iter() {
100                scope_vec_var.push((e.0, e.1))
101            }
102            scope_vec_var
103        }
104    }
105
106    impl<'a> XAllDifferentExcept<'a> {
107        pub fn from_str(
108            list: &str,
109            except_str: &str,
110            set: &'a XVariableSet,
111        ) -> Result<Self, Xcsp3Error> {
112            match list_to_vec_var_val(list) {
113                Ok(scope) => match list_with_bracket_comma_to_values(except_str) {
114                    Ok(except) => Ok(XAllDifferentExcept::new(scope, set, except)),
115                    Err(e) => Err(e),
116                },
117                Err(e) => Err(e),
118            }
119        }
120
121        pub fn new(scope: Vec<XVarVal>, set: &'a XVariableSet, except: Vec<XVarVal>) -> Self {
122            XAllDifferentExcept {
123                map: Default::default(),
124                scope,
125                except,
126                set,
127            }
128        }
129
130        /// return the except vec of the XAllDifferentExcept constraint
131        pub fn get_except(&self) -> &Vec<XVarVal> {
132            &self.except
133        }
134    }
135}
xcsp3_rust/constraints/xall_different_except.rs

xcsp3_rust/constraints/
xall_different_except.rs
