pa_byte_pack.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_BYTE_PACK_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_BYTE_PACK_H_
 
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "backends/tofino/bf-p4c/phv/utils/utils.h"
#include "ir/ir.h"
#include "lib/safe_vector.h"
 
using namespace P4;
 
class PragmaBytePack : public Inspector {
 public:
    struct PackConstraint {
        bool compiler_added;
        std::optional<Util::SourceInfo> src_info;
        PHV::PackingLayout packing;
    };
 
    struct AddConstraintResult {
        ordered_set<const PHV::Field *> alignment_added;
        std::optional<cstring> error;
        bool ok() const { return !error; }
    };
 
 private:
    PhvInfo &phv_i;
    safe_vector<PackConstraint> packing_layouts_i;
 
    AddConstraintResult add_packing_constraint(const PackConstraint &packing);
 
    profile_t init_apply(const IR::Node *root) override {
        profile_t rv = Inspector::init_apply(root);
        packing_layouts_i.clear();
        return rv;
    }
 
    bool preorder(const IR::BFN::Pipe *pipe) override;
 
 public:
    explicit PragmaBytePack(PhvInfo &phv) : phv_i(phv) {}
 
    const safe_vector<PackConstraint> &packings() const { return packing_layouts_i; }
 
    AddConstraintResult add_compiler_added_packing(const PHV::PackingLayout &packing);
 
    static const char *name;
    static const char *description;
    static const char *help;
};
 
#endif /* BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_BYTE_PACK_H_ */