desugar_varbit_extract.h

 
#ifndef BF_P4C_MIDEND_DESUGAR_VARBIT_EXTRACT_H_
#define BF_P4C_MIDEND_DESUGAR_VARBIT_EXTRACT_H_
 
#include "backends/tofino/bf-p4c/common/utils.h"
#include "backends/tofino/bf-p4c/midend/type_checker.h"
#include "frontends/common/resolveReferences/referenceMap.h"
#include "frontends/p4/cloner.h"
#include "ir/ir.h"
 
namespace BFN {
 
class AnnotateVarbitExtractStates : public Transform {
    IR::Node *preorder(IR::ParserState *state) override {
        for (const auto *component : state->components) {
            const auto *statement = component->to<IR::MethodCallStatement>();
            if (!statement) continue;
            const IR::MethodCallExpression *call = statement->methodCall;
            if (!call) continue;
            const auto *method = call->method->to<IR::Member>();
            if (!method) continue;
 
            if (method->member == "extract" && call->arguments->size() == 2) {
                IR::Annotations *annotations = state->annotations->clone();
                annotations->add(new IR::Annotation(IR::ID("dontmerge"), {}));
                state->annotations = annotations;
                break;
            }
        }
        return state;
    }
};
 
class CheckVarbitAccess : public PassManager {
 public:
    CheckVarbitAccess();
};
 
class DesugarVarbitExtract : public PassManager {
 public:
    explicit DesugarVarbitExtract(P4::ReferenceMap *refMap, P4::TypeMap *typeMap);
};
 
}  // namespace BFN
 
#endif /* BF_P4C_MIDEND_DESUGAR_VARBIT_EXTRACT_H_ */