pa_container_type.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_CONTAINER_TYPE_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_CONTAINER_TYPE_H_
 
#include <optional>
 
#include "backends/tofino/bf-p4c/phv/phv.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "ir/ir.h"
 
using namespace P4;
 
class PragmaContainerType : public Inspector {
    PhvInfo &phv_i;
 
    ordered_map<const PHV::Field *, PHV::Kind> fields;
 
    profile_t init_apply(const IR::Node *root) override {
        profile_t rv = Inspector::init_apply(root);
        fields.clear();
        return rv;
    }
 
    bool add_constraint(const IR::BFN::Pipe *pipe, const IR::Expression *expr, cstring field_name,
                        PHV::Kind kind);
 
    bool preorder(const IR::BFN::Pipe *pipe) override;
 
 public:
    explicit PragmaContainerType(PhvInfo &phv) : phv_i(phv) {}
 
    const ordered_map<const PHV::Field *, PHV::Kind> &getFields() const { return fields; }
 
    std::optional<PHV::Kind> required_kind(const PHV::Field *f) const;
 
    static const char *name;
    static const char *description;
    static const char *help;
};
 
std::ostream &operator<<(std::ostream &out, const PragmaContainerType &pa_ct);
 
#endif /* BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_CONTAINER_TYPE_H_ */