blob: e0e589e577098765867e8742cf390327caf9b72d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/********************* */
/*! \file arithvar_node_map.h
** \verbatim
** Original author: taking
** Major contributors: mdeters
** Minor contributors (to current version): dejan
** This file is part of the CVC4 prototype.
** Copyright (c) 2009-2012 New York University and The University of Iowa
** See the file COPYING in the top-level source directory for licensing
** information.\endverbatim
**
** \brief [[ Add one-line brief description here ]]
**
** [[ Add lengthier description here ]]
** \todo document this file
**/
#include "cvc4_private.h"
#ifndef __CVC4__THEORY__ARITH__ARITHVAR_NODE_MAP_H
#define __CVC4__THEORY__ARITH__ARITHVAR_NODE_MAP_H
#include "theory/arith/arithvar.h"
#include "context/context.h"
#include "context/cdlist.h"
#include "context/cdhashmap.h"
#include "context/cdo.h"
namespace CVC4 {
namespace theory {
namespace arith {
class ArithVarNodeMap {
private:
/**
* Bidirectional map between Nodes and ArithVars.
*/
NodeToArithVarMap d_nodeToArithVarMap;
ArithVarToNodeMap d_arithVarToNodeMap;
public:
typedef ArithVarToNodeMap::const_iterator var_iterator;
ArithVarNodeMap() {}
inline bool hasArithVar(TNode x) const {
return d_nodeToArithVarMap.find(x) != d_nodeToArithVarMap.end();
}
inline bool hasNode(ArithVar a) const {
return d_arithVarToNodeMap.isKey(a);
}
inline ArithVar asArithVar(TNode x) const{
Assert(hasArithVar(x));
Assert((d_nodeToArithVarMap.find(x))->second <= ARITHVAR_SENTINEL);
return (d_nodeToArithVarMap.find(x))->second;
}
inline Node asNode(ArithVar a) const{
Assert(hasNode(a));
return d_arithVarToNodeMap[a];
}
inline void setArithVar(TNode x, ArithVar a){
Assert(!hasArithVar(x));
Assert(!d_arithVarToNodeMap.isKey(a));
d_arithVarToNodeMap.set(a, x);
d_nodeToArithVarMap[x] = a;
}
inline void remove(ArithVar x){
Assert(hasNode(x));
Node node = asNode(x);
d_nodeToArithVarMap.erase(d_nodeToArithVarMap.find(node));
d_arithVarToNodeMap.remove(x);
}
var_iterator var_begin() const {
return d_arithVarToNodeMap.begin();
}
var_iterator var_end() const {
return d_arithVarToNodeMap.end();
}
};/* class ArithVarNodeMap */
}/* CVC4::theory::arith namespace */
}/* CVC4::theory namespace */
}/* CVC4 namespace */
#endif /* __CVC4__THEORY__ARITH__ARITHVAR_NODE_MAP_H */
|