rewriter.hpp

Go to the documentation of this file.

#ifndef __REWRITER_HPP_DEFINED__
#define __REWRITER_HPP_DEFINED__

#include <ilasynth/ast.hpp>
#include <ilasynth/common.hpp>
#include <unordered_map>

namespace ilasynth {
// A function object that rewrites the AST according to the result
// of the synthesis.
class Rewriter {
public:
  // Define types.
protected:
  // the map between old and new nodes.
  rwmap_t rwmap;

  // find the rewritten args.
  void getNewArgs(const Node* n, nptr_vec_t& args);

  // get the nodes' replacement.
  nptr_t getRepl(const Node* n) const;

  // do the actual rewriting.
  void doRewrite(const Node* n);

public:
  // Constructor.
  Rewriter();

  // Destructor.
  ~Rewriter();

  // This is used by depthFirstVisit.
  void operator()(const Node* n);

  // do the rewrite.
  nptr_t rewrite(const Node* n);

  // add a node to be rewritten.
  void addRewrite(const Node* n, const nptr_t& nprime);
};
} // namespace ilasynth
#endif