simplifyIfStatement.h

#ifndef BACKENDS_TOFINO_BF_P4C_MIDEND_SIMPLIFYIFSTATEMENT_H_
#define BACKENDS_TOFINO_BF_P4C_MIDEND_SIMPLIFYIFSTATEMENT_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/simplify.h"
#include "frontends/p4/typeMap.h"
#include "ir/ir.h"
 
namespace P4 {

 
class ElimCallExprInIfCond : public Transform {
    ReferenceMap *refMap;
 
 public:
    ElimCallExprInIfCond(ReferenceMap *refMap, TypeMap *) : refMap(refMap) {}
    const IR::Node *preorder(IR::IfStatement *statement) override;
    const IR::Node *postorder(IR::MethodCallExpression *methodCall) override;
 
    IR::IndexedVector<IR::StatOrDecl> stack;
};

class SimplifyIfStatement : public PassManager {
 public:
    SimplifyIfStatement(ReferenceMap *refMap, TypeMap *typeMap) {
        passes.push_back(new ElimCallExprInIfCond(refMap, typeMap));
        passes.push_back(new SimplifyControlFlow(typeMap));
    }
};
 
}  // namespace P4
 
#endif /* BACKENDS_TOFINO_BF_P4C_MIDEND_SIMPLIFYIFSTATEMENT_H_ */