removeExits.h

/*
 * Copyright 2017 VMware, Inc.
 * SPDX-FileCopyrightText: 2017 VMware, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef MIDEND_REMOVEEXITS_H_
#define MIDEND_REMOVEEXITS_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/removeReturns.h"
 
namespace P4 {

class DoRemoveExits : public DoRemoveReturns, public ResolutionContext {
    TypeMap *typeMap;
    // In this class "Return" (inherited from RemoveReturns) should be read as "Exit"
    std::set<const IR::Node *> callsExit;  // actions, tables
    void callExit(const IR::Node *node);
 
 public:
    explicit DoRemoveExits(TypeMap *typeMap) : DoRemoveReturns("hasExited"_cs), typeMap(typeMap) {
        visitDagOnce = false;
        CHECK_NULL(typeMap);
        setName("DoRemoveExits");
    }
 
    const IR::Node *preorder(IR::ExitStatement *action) override;
    const IR::Node *preorder(IR::P4Table *table) override;
 
    const IR::Node *preorder(IR::BlockStatement *statement) override;
    const IR::Node *preorder(IR::IfStatement *statement) override;
    const IR::Node *preorder(IR::SwitchStatement *statement) override;
    const IR::Node *preorder(IR::BaseAssignmentStatement *statement) override;
    const IR::Node *preorder(IR::MethodCallStatement *statement) override;
 
    const IR::Node *preorder(IR::P4Action *action) override;
    const IR::Node *preorder(IR::P4Control *control) override;
};
 
class RemoveExits : public PassManager {
 public:
    explicit RemoveExits(TypeMap *typeMap, TypeChecking *typeChecking = nullptr) {
        if (!typeChecking) typeChecking = new TypeChecking(nullptr, typeMap);
        passes.push_back(typeChecking);
        passes.push_back(new DoRemoveExits(typeMap));
        setName("RemoveExits");
    }
};
 
}  // namespace P4
 
#endif /* MIDEND_REMOVEEXITS_H_ */