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}