action_source_tracker.h

 
#ifndef BF_P4C_PHV_ACTION_SOURCE_TRACKER_H_
#define BF_P4C_PHV_ACTION_SOURCE_TRACKER_H_
 
#include <optional>
 
#include "backends/tofino/bf-p4c/mau/action_analysis.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "ir/ir.h"
 
namespace PHV {
 
struct SourceOp {
    enum class OpType { move, bitwise, whole_container };
    OpType t = OpType::move;
    bool ad_or_const = false;
    std::optional<PHV::FieldSlice> phv_src = std::nullopt;
 
    SourceOp slice(int start, int len) const;
};
 
std::ostream &operator<<(std::ostream &out, const SourceOp &src);
 
using ActionClassifiedSources = ordered_map<const IR::MAU::Action *, safe_vector<SourceOp>>;
 
class ActionSourceTracker : public Inspector {
    const PhvInfo &phv;
    const ReductionOrInfo &red_info;
 
    ordered_map<const PHV::Field *, ordered_map<le_bitrange, ActionClassifiedSources>> sources;
 
 private:
    void clear() { sources.clear(); };
 
    void add_sources(const IR::MAU::Action *act,
                     const ActionAnalysis::FieldActionsMap &instructions);
 
    profile_t init_apply(const IR::Node *root) override {
        clear();
        return Inspector::init_apply(root);
    }
 
    bool preorder(const IR::MAU::Action *act) override;
 
    void end_apply() override;
 
 public:
    explicit ActionSourceTracker(const PhvInfo &phv, const ReductionOrInfo &ri)
        : phv(phv), red_info(ri) {}
 
    ActionClassifiedSources get_sources(const PHV::FieldSlice &fs) const;
 
    friend std::ostream &operator<<(std::ostream &out, const ActionSourceTracker &tracker);
};
 
std::ostream &operator<<(std::ostream &out, const ActionClassifiedSources &sources);
std::ostream &operator<<(std::ostream &out, const ActionSourceTracker &tracker);
 
}  // namespace PHV
 
#endif /* BF_P4C_PHV_ACTION_SOURCE_TRACKER_H_ */