pa_no_overlay.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_NO_OVERLAY_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_NO_OVERLAY_H_
 
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "ir/ir.h"
 
using namespace P4;
 
class PragmaNoOverlay : public Inspector {
    PhvInfo &phv_i;
 
    ordered_set<const PHV::Field *> no_overlay;
    SymBitMatrix mutually_inclusive;
 
    profile_t init_apply(const IR::Node *root) override {
        profile_t rv = Inspector::init_apply(root);
        no_overlay.clear();
        mutually_inclusive.clear();
        return rv;
    }
 
    bool preorder(const IR::BFN::Pipe *pipe) override;
    bool preorder(const IR::MAU::Instruction *inst) override;
 
 public:
    explicit PragmaNoOverlay(PhvInfo &phv) : phv_i(phv) {}
 
    friend std::ostream &operator<<(std::ostream &out, const PragmaNoOverlay &pa_no);
    static const char *name;
    static const char *description;
    static const char *help;
 
    const ordered_set<const PHV::Field *> &get_no_overlay_fields() const { return no_overlay; }
 
    bool can_overlay(const PHV::Field *a, const PHV::Field *b) const {
        return !no_overlay.count(a) && !no_overlay.count(b) && !mutually_inclusive(a->id, b->id);
    }
};
 
std::ostream &operator<<(std::ostream &out, const PragmaNoOverlay &pa_no);
 
#endif /* BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_NO_OVERLAY_H_ */