xcsp3_rust/variables/
xvariable_type.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 18:35
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::errors::xcsp3error::xcsp3_core::Xcsp3Error;
43    use crate::variables::xdomain::xcsp3_core::XDomainInteger;
44    use crate::variables::xvariable_array::xcsp3_core::XVariableArray;
45    use crate::variables::xvariable_int::xcsp3_core::XVariableInt;
46    use std::fmt::{Display, Formatter};
47    // use crate::variables::xvariable_trait::xcsp3_core::XVariableTrait;
48    use crate::variables::xvariable_tree::xcsp3_core::XVariableTree;
49
50    #[derive(Clone)]
51    pub enum XVariableType {
52        XVariableNone(Xcsp3Error),
53        XVariableArray(XVariableArray),
54        XVariableInt(XVariableInt),
55        XVariableTree(XVariableTree),
56    }
57
58    impl XVariableType {
59        pub fn new_int(id: &str, domain: XDomainInteger) -> XVariableType {
60            XVariableType::XVariableInt(XVariableInt::new(id.to_string(), domain))
61        }
62
63        pub fn new_array(id: &str, sizes: &str, domain: XDomainInteger) -> XVariableType {
64            match XVariableArray::new(id, sizes, domain) {
65                Ok(array) => XVariableType::XVariableArray(array),
66                Err(e) => XVariableType::XVariableNone(e),
67            }
68        }
69
70        pub fn new_tree(
71            id: &str,
72            sizes: &str,
73            domain_for: Vec<&String>,
74            domain_value: Vec<&String>,
75        ) -> Result<Self, Xcsp3Error> {
76            match XVariableTree::new(id, sizes, domain_for, domain_value) {
77                Ok(t) => Ok(XVariableType::XVariableTree(t)),
78                Err(e) => Err(e),
79            }
80        }
81
82        pub fn get_id(&self) -> String {
83            match self {
84                XVariableType::XVariableArray(v) => v.id.clone(),
85                XVariableType::XVariableInt(v) => v.id.clone(),
86                XVariableType::XVariableTree(v) => v.id.clone(),
87                _ => String::default(),
88            }
89        }
90    }
91
92    impl Display for XVariableType {
93        fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
94            write!(
95                f,
96                "{}",
97                match self {
98                    XVariableType::XVariableArray(a) => a.to_string(),
99                    XVariableType::XVariableInt(i) => i.to_string(),
100                    XVariableType::XVariableTree(t) => t.to_string(),
101                    _ => {
102                        String::from(
103                            "XVariableNone: there must be an error when parse this variable.",
104                        )
105                    }
106                }
107            )
108        }
109    }
110}