simplifySelectCases.h

/*
 * Copyright 2016 VMware, Inc.
 * SPDX-FileCopyrightText: 2016 VMware, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef MIDEND_SIMPLIFYSELECTCASES_H_
#define MIDEND_SIMPLIFYSELECTCASES_H_
 
#include "frontends/p4/typeChecking/typeChecker.h"
#include "ir/ir.h"
 
namespace P4 {

class DoSimplifySelectCases : public Transform, ResolutionContext {
    const TypeMap *typeMap;
    bool requireConstants;
 
    void checkSimpleConstant(const IR::Expression *expr) const;
 
 public:
    DoSimplifySelectCases(const TypeMap *typeMap, bool requireConstants)
        : typeMap(typeMap), requireConstants(requireConstants) {
        setName("DoSimplifySelectCases");
    }
    const IR::Node *preorder(IR::SelectExpression *expression) override;
};
 
class SimplifySelectCases : public PassManager {
 public:
    SimplifySelectCases(TypeMap *typeMap, bool requireConstants,
                        TypeChecking *typeChecking = nullptr) {
        if (!typeChecking) typeChecking = new TypeChecking(nullptr, typeMap);
        passes.push_back(typeChecking);
        passes.push_back(new DoSimplifySelectCases(typeMap, requireConstants));
        setName("SimplifySelectCases");
    }
};
 
}  // namespace P4
 
#endif /* MIDEND_SIMPLIFYSELECTCASES_H_ */