diff options
author | Haniel Barbosa <hanielbbarbosa@gmail.com> | 2018-08-24 20:19:14 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-08-24 20:19:14 -0500 |
commit | 7b9c2529c149a9cd046083af401cbdeadf406804 (patch) | |
tree | bbae5bbf4c9538181f01fae61f0e38bbf46dc3d2 /src/preprocessing/passes/nl_ext_purify.h | |
parent | 248f841f37b8b2d514d7308faa8f4573115f82e9 (diff) |
Refactor nlExtPurify preprocessing pass (#1963)
Diffstat (limited to 'src/preprocessing/passes/nl_ext_purify.h')
-rw-r--r-- | src/preprocessing/passes/nl_ext_purify.h | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/src/preprocessing/passes/nl_ext_purify.h b/src/preprocessing/passes/nl_ext_purify.h new file mode 100644 index 000000000..8d28b0742 --- /dev/null +++ b/src/preprocessing/passes/nl_ext_purify.h @@ -0,0 +1,57 @@ +/********************* */ +/*! \file nl_ext_purify.h + ** \verbatim + ** Top contributors (to current version): + ** Haniel Barbosa + ** This file is part of the CVC4 project. + ** Copyright (c) 2009-2018 by the authors listed in the file AUTHORS + ** in the top-level source directory) and their institutional affiliations. + ** All rights reserved. See the file COPYING in the top-level source + ** directory for licensing information.\endverbatim + ** + ** \brief The NlExtPurify preprocessing pass + ** + ** Purifies non-linear terms by replacing sums under multiplications by fresh + ** variables + **/ + +#include "cvc4_private.h" + +#ifndef __CVC4__PREPROCESSING__PASSES__NL_EXT_PURIFY_H +#define __CVC4__PREPROCESSING__PASSES__NL_EXT_PURIFY_H + +#include <unordered_map> +#include <vector> + +#include "expr/node.h" +#include "preprocessing/preprocessing_pass.h" +#include "preprocessing/preprocessing_pass_context.h" + +namespace CVC4 { +namespace preprocessing { +namespace passes { + +using NodeMap = std::unordered_map<Node, Node, NodeHashFunction>; + +class NlExtPurify : public PreprocessingPass +{ + public: + NlExtPurify(PreprocessingPassContext* preprocContext); + + protected: + PreprocessingPassResult applyInternal( + AssertionPipeline* assertionsToPreprocess) override; + + private: + Node purifyNlTerms(TNode n, + NodeMap& cache, + NodeMap& bcache, + std::vector<Node>& var_eq, + bool beneathMult = false); +}; + +} // namespace passes +} // namespace preprocessing +} // namespace CVC4 + +#endif /* __CVC4__PREPROCESSING__PASSES__NL_EXT_PURIFY_H */ |