rewrite_flexible_header.h

 
#ifndef BF_P4C_MIDEND_REWRITE_FLEXIBLE_HEADER_H_
#define BF_P4C_MIDEND_REWRITE_FLEXIBLE_HEADER_H_
 
#include "frontends/p4/typeMap.h"
#include "ir/ir.h"
 
namespace P4 {
class TypeMap;
class ReferenceMap;
}  // namespace P4
 
namespace BFN {
 
struct RewriteHeader : public Transform {
 public:
    RewriteHeader() {}
    const IR::Node *postorder(IR::Type_Struct *st) override {
        if (st->getAnnotation("flexible"_cs) == nullptr) return st;
 
        IR::IndexedVector<IR::StructField> fields;
        for (auto f : st->fields) {
            fields.push_back(new IR::StructField(
                f->name, {new IR::Annotation(IR::ID("flexible"), {})}, f->type));
        }
 
        return new IR::Type_Struct(st->name, st->annotations, fields);
    }
};
 
class RewriteFlexibleStruct : public PassManager {
 public:
    explicit RewriteFlexibleStruct(P4::ReferenceMap *refMap, P4::TypeMap *typeMap) {
        addPasses({
            new RewriteHeader(),
            new P4::ClearTypeMap(typeMap),
            new BFN::TypeChecking(refMap, typeMap, true),
        });
    }
};
 
}  // namespace BFN
 
#endif /* BF_P4C_MIDEND_REWRITE_FLEXIBLE_HEADER_H_ */