simplify.h

/*
 * SPDX-FileCopyrightText: 2013 Barefoot Networks, Inc.
 * Copyright 2013-present Barefoot Networks, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef FRONTENDS_P4_SIMPLIFY_H_
#define FRONTENDS_P4_SIMPLIFY_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/methodInstance.h"
#include "frontends/p4/typeChecking/typeChecker.h"
#include "ir/ir.h"
 
namespace P4 {

class DoSimplifyControlFlow : public Transform, public ResolutionContext {
    TypeMap *typeMap;
    bool foldInlinedFrom;
 
 public:
    explicit DoSimplifyControlFlow(TypeMap *typeMap, bool foldInlinedFrom)
        : typeMap(typeMap), foldInlinedFrom(foldInlinedFrom) {
        CHECK_NULL(typeMap);
        setName("DoSimplifyControlFlow");
        // We may want to replace the same statement with different things
        // in different places.
        visitDagOnce = false;
    }
    const IR::Node *postorder(IR::BlockStatement *statement) override;
    const IR::Node *postorder(IR::IfStatement *statement) override;
    const IR::Node *postorder(IR::EmptyStatement *statement) override;
    const IR::Node *postorder(IR::SwitchStatement *statement) override;
};

class SimplifyControlFlow : public PassRepeated {
 public:
    explicit SimplifyControlFlow(TypeMap *typeMap, bool foldInlinedFrom,
                                 TypeChecking *typeChecking = nullptr) {
        if (!typeChecking) typeChecking = new TypeChecking(nullptr, typeMap);
        passes.push_back(typeChecking);
        passes.push_back(new DoSimplifyControlFlow(typeMap, foldInlinedFrom));
        setName("SimplifyControlFlow");
    }
};
 
}  // namespace P4
 
#endif /* FRONTENDS_P4_SIMPLIFY_H_ */