bmv2/common/midend.h

/*
 * SPDX-FileCopyrightText: 2013 Barefoot Networks, Inc.
 * Copyright 2013-present Barefoot Networks, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_BMV2_COMMON_MIDEND_H_
#define BACKENDS_BMV2_COMMON_MIDEND_H_
 
#include "frontends/common/options.h"
#include "ir/ir.h"
#include "lower.h"
#include "midend/convertEnums.h"
 
namespace P4::BMV2 {

class EnumOn32Bits : public P4::ChooseEnumRepresentation {
    cstring filename;
 
    bool convert(const IR::Type_Enum *type) const override {
        if (type->srcInfo.isValid()) {
            auto sourceFile = type->srcInfo.getSourceFile();
            if (sourceFile.endsWith(filename))
                // Don't convert any of the standard enums
                return false;
        }
        return true;
    }
    unsigned enumSize(unsigned) const override { return 32; }
 
 public:
    explicit EnumOn32Bits(cstring filename) : filename(filename) {}
};
 
class MidEnd : public PassManager {
 public:
    P4::ReferenceMap refMap;
    P4::TypeMap typeMap;
    const IR::ToplevelBlock *toplevel = nullptr;
    P4::ConvertEnums::EnumMapping enumMap;
    bool isv1;
 
    explicit MidEnd(CompilerOptions &options) {
        isv1 = options.isv1();
        refMap.setIsV1(isv1);  // must be done BEFORE creating passes
    }
    const IR::ToplevelBlock *process(const IR::P4Program *&program) {
        program = program->apply(*this);
        return toplevel;
    }
};
 
}  // namespace P4::BMV2
 
#endif /* BACKENDS_BMV2_COMMON_MIDEND_H_ */