dark.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_DARK_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_DARK_H_
 
#include "backends/tofino/bf-p4c/common/field_defuse.h"
#include "backends/tofino/bf-p4c/mau/mau_visitor.h"
#include "backends/tofino/bf-p4c/phv/analysis/mocha.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 CollectNonDarkUses : public MauInspector {
 private:
    const PhvInfo &phv;
    const ReductionOrInfo &red_info;
    bitvec nonDarkMauUses;
 
    profile_t init_apply(const IR::Node *root) override;
    bool preorder(const IR::MAU::Table *) override;
    bool preorder(const IR::Expression *) override;
    bool preorder(const IR::MAU::Action *) override;
    bool contextNeedsIXBar();
 
 public:
    CollectNonDarkUses(const PhvInfo &p, const ReductionOrInfo &ri) : phv(p), red_info(ri) {}
 
    bool hasNonDarkUse(const PHV::Field *f) const;
};
 
class MarkDarkCandidates : public Inspector {
 private:
    PhvInfo &phv;
    const PhvUse &uses;
    const CollectNonDarkUses &nonDarkUses;
 
    bitvec nonDarkMauUses;
 
    size_t darkCount;
    size_t darkSize;
 
    profile_t init_apply(const IR::Node *root) override;
    void end_apply() override;
 
 public:
    explicit MarkDarkCandidates(PhvInfo &p, const PhvUse &u, const CollectNonDarkUses &d)
        : phv(p), uses(u), nonDarkUses(d), darkCount(0), darkSize(0) {}
};
 
class CollectDarkCandidates : public PassManager {
 private:
    CollectNonDarkUses nonDarkUses;
 
 public:
    CollectDarkCandidates(PhvInfo &p, PhvUse &u, const ReductionOrInfo &ri) : nonDarkUses(p, ri) {
        addPasses({&nonDarkUses, new MarkDarkCandidates(p, u, nonDarkUses)});
    }
};
 
#endif /*  BACKENDS_TOFINO_BF_P4C_PHV_ANALYSIS_DARK_H_  */