diff options
author | Cristian Cadar <cristic@cs.stanford.edu> | 2012-10-19 09:19:20 +0000 |
---|---|---|
committer | Cristian Cadar <cristic@cs.stanford.edu> | 2012-10-19 09:19:20 +0000 |
commit | 01817eb65cfc8f40ae5200b6dd238c86f7e8c915 (patch) | |
tree | e42264ffa12ea30031a30928ae8f5d3de8848eb9 /include | |
parent | e2a2fceee17dbf7323b2dac00feb2293365fcc34 (diff) | |
download | klee-01817eb65cfc8f40ae5200b6dd238c86f7e8c915.tar.gz |
Added missing header file (part of the last patch).
git-svn-id: https://llvm.org/svn/llvm-project/klee/trunk@166277 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'include')
-rw-r--r-- | include/klee/util/ArrayExprHash.h | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/include/klee/util/ArrayExprHash.h b/include/klee/util/ArrayExprHash.h new file mode 100644 index 00000000..9a527bcd --- /dev/null +++ b/include/klee/util/ArrayExprHash.h @@ -0,0 +1,135 @@ +//===-- ArrayExprHash.h --------------------------------------------*- C++ -*-===// +// +// The KLEE Symbolic Virtual Machine +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + +#ifndef __UTIL_ARRAYEXPRHASH_H__ +#define __UTIL_ARRAYEXPRHASH_H__ + +#include "klee/Expr.h" +#include "klee/TimerStatIncrementer.h" +#include "SolverStats.h" + +#include <map> +#include <tr1/unordered_map> + +namespace klee { + +struct ArrayHashFn { + unsigned operator()(const Array* array) const { + return(array ? array->hash() : 0); + } +}; + +struct ArrayCmpFn { + bool operator()(const Array* array1, const Array* array2) const { + return(array1 == array2); + } +}; + +struct UpdateNodeHashFn { + unsigned operator()(const UpdateNode* un) const { + return(un ? un->hash() : 0); + } +}; + +struct UpdateNodeCmpFn { + bool operator()(const UpdateNode* un1, const UpdateNode* un2) const { + return(un1 == un2); + } +}; + +template<class T> +class ArrayExprHash { +public: + + ArrayExprHash() {}; + // Note: Extend the class and overload the destructor if the objects of type T + // that are to be hashed need to be explicitly destroyed + // As an example, see class STPArrayExprHash + virtual ~ArrayExprHash() {}; + + bool lookupArrayExpr(const Array* array, T& exp) const; + void hashArrayExpr(const Array* array, T& exp); + + bool lookupUpdateNodeExpr(const UpdateNode* un, T& exp) const; + void hashUpdateNodeExpr(const UpdateNode* un, T& exp); + +protected: + typedef std::tr1::unordered_map<const Array*, T, ArrayHashFn, ArrayCmpFn> ArrayHash; + typedef typename ArrayHash::iterator ArrayHashIter; + typedef typename ArrayHash::const_iterator ArrayHashConstIter; + + typedef std::tr1::unordered_map<const UpdateNode*, T, UpdateNodeHashFn, UpdateNodeCmpFn> UpdateNodeHash; + typedef typename UpdateNodeHash::iterator UpdateNodeHashIter; + typedef typename UpdateNodeHash::const_iterator UpdateNodeHashConstIter; + + ArrayHash _array_hash; + UpdateNodeHash _update_node_hash; +}; + + +template<class T> +bool ArrayExprHash<T>::lookupArrayExpr(const Array* array, T& exp) const { + bool res = false; + +#ifdef DEBUG + TimerStatIncrementer t(stats::arrayHashTime); +#endif + + assert(array); + ArrayHashConstIter it = _array_hash.find(array); + if (it != _array_hash.end()) { + exp = it->second; + res = true; + } + return(res); +} + +template<class T> +void ArrayExprHash<T>::hashArrayExpr(const Array* array, T& exp) { + +#ifdef DEBUG + TimerStatIncrementer t(stats::arrayHashTime); +#endif + + assert(array); + _array_hash[array] = exp; +} + +template<class T> +bool ArrayExprHash<T>::lookupUpdateNodeExpr(const UpdateNode* un, T& exp) const +{ + bool res = false; + +#ifdef DEBUG + TimerStatIncrementer t(stats::arrayHashTime); +#endif + + assert(un); + UpdateNodeHashConstIter it = _update_node_hash.find(un); + if (it != _update_node_hash.end()) { + exp = it->second; + res = true; + } + return(res); +} + +template<class T> +void ArrayExprHash<T>::hashUpdateNodeExpr(const UpdateNode* un, T& exp) +{ +#ifdef DEBUG + TimerStatIncrementer t(stats::arrayHashTime); +#endif + + assert(un); + _update_node_hash[un] = exp; +} + +} + +#endif \ No newline at end of file |