mocha.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_MOCHA_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_MOCHA_H_
 
#include "backends/tofino/bf-p4c/mau/action_analysis.h"
#include "backends/tofino/bf-p4c/phv/analysis/non_mocha_dark_fields.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "ir/ir.h"
 
class CollectMochaCandidates : public Inspector {
 private:
    PhvInfo &phv;
    const PhvUse &uses;
    const ReductionOrInfo &red_info;
    const NonMochaDarkFields &nonMochaDark;
    bool pov_on_mocha = false;
 
    size_t mochaCount;
    size_t mochaSize;
 
    profile_t init_apply(const IR::Node *root) override;
    bool preorder(const IR::MAU::Action *act) override;
    void end_apply() override;
 
 public:
    CollectMochaCandidates(PhvInfo &p, const PhvUse &u, const ReductionOrInfo &ri,
                           const NonMochaDarkFields &nmd)
        : phv(p), uses(u), red_info(ri), nonMochaDark(nmd), mochaCount(0), mochaSize(0) {}
 
    static bool isPacketField(const PHV::Field *f) {
        return (f && !f->metadata && !f->pov && !f->bridged && !f->overlayable &&
                !f->is_intrinsic());
    }
};
 
#endif /* BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_MOCHA_H_ */