non_mocha_dark_fields.h

 
#ifndef P4C_BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_NON_MOCHA_DARK_FIELDS_H_
#define P4C_BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_NON_MOCHA_DARK_FIELDS_H_
 
#include "backends/tofino/bf-p4c/common/field_defuse.h"
#include "backends/tofino/bf-p4c/mau/action_analysis.h"
#include "backends/tofino/bf-p4c/phv/phv.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "backends/tofino/bf-p4c/phv/pragma/phv_pragmas.h"
#include "ir/ir.h"
 
class NonMochaDarkFields : public Inspector {
 public:
    using FieldMap = std::map<const int, std::map<const IR::MAU::Table *, PHV::FieldUse>>;
 
 private:
    static const PHV::FieldUse READ;
    static const PHV::FieldUse WRITE;
 
    const PhvInfo &phv;
    const PhvUse &uses;
    const FieldDefUse &defuse;
    const ReductionOrInfo &red_info;
 
    const ordered_set<const PHV::Field *> &noInitFields;
    const ordered_set<const PHV::Field *> &notParsedFields;
    const ordered_set<const PHV::Field *> &notDeparsedFields;
 
    FieldMap nonDark;
    FieldMap nonMocha;
 
    profile_t init_apply(const IR::Node *root) override;
    bool preorder(const IR::MAU::Action *act) override;
    void end_apply() override;
 
 public:
    NonMochaDarkFields(const PhvInfo &phv, const PhvUse &uses, const FieldDefUse &defuse,
                       const ReductionOrInfo &ri, const PHV::Pragmas &pragmas)
        : phv(phv),
          uses(uses),
          defuse(defuse),
          red_info(ri),
          noInitFields(pragmas.pa_no_init().getFields()),
          notParsedFields(pragmas.pa_deparser_zero().getNotDeparsedFields()),
          notDeparsedFields(pragmas.pa_deparser_zero().getNotDeparsedFields()) {}
 
    PHV::FieldUse isNotMocha(const PHV::Field *field, const IR::MAU::Table *tbl = nullptr) const;
 
    PHV::FieldUse isNotDark(const PHV::Field *field, const IR::MAU::Table *tbl = nullptr) const;
 
    const FieldMap &getNonMocha() const { return nonMocha; }
 
    const FieldMap &getNonDark() const { return nonDark; }
};
 
#endif /* P4C_BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_NON_MOCHA_DARK_FIELDS_H_ */