diff options
Diffstat (limited to 'src/theory/arrays/array_info.cpp')
| -rw-r--r-- | src/theory/arrays/array_info.cpp | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/src/theory/arrays/array_info.cpp b/src/theory/arrays/array_info.cpp index bb9a9e417..32eaff355 100644 --- a/src/theory/arrays/array_info.cpp +++ b/src/theory/arrays/array_info.cpp @@ -167,6 +167,20 @@ void ArrayInfo::setRIntro1Applied(const TNode a) { } +void ArrayInfo::setModelRep(const TNode a, const TNode b) { + Assert(a.getType().isArray()); + Info* temp_info; + CNodeInfoMap::iterator it = info_map.find(a); + if(it == info_map.end()) { + temp_info = new Info(ct, bck); + temp_info->modelRep = b; + info_map[a] = temp_info; + } else { + (*it).second->modelRep = b; + } + +} + /** * Returns the information associated with TNode a */ @@ -200,6 +214,16 @@ const bool ArrayInfo::rIntro1Applied(const TNode a) const return false; } +const TNode ArrayInfo::getModelRep(const TNode a) const +{ + CNodeInfoMap::const_iterator it = info_map.find(a); + + if(it!= info_map.end()) { + return (*it).second->modelRep; + } + return TNode(); +} + const CTNodeList* ArrayInfo::getIndices(const TNode a) const{ CNodeInfoMap::const_iterator it = info_map.find(a); if(it!= info_map.end()) { |