pa_alias.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_ALIAS_H_
#define BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_ALIAS_H_
 
#include <map>
#include <optional>
 
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "backends/tofino/bf-p4c/phv/pragma/pa_no_overlay.h"
#include "backends/tofino/bf-p4c/phv/pragma/pretty_print.h"
#include "backends/tofino/bf-p4c/phv/utils/utils.h"
#include "ir/ir.h"
 
using namespace P4;
class PragmaAlias : public Inspector, public Pragma::PrettyPrint {
 public:
    enum CreatedBy { PRAGMA, COMPILER };
 
    struct AliasDestination {
        cstring field;
 
        std::optional<le_bitrange> range;
 
        CreatedBy who;
    };
 
    using AliasMap = ordered_map<cstring, AliasDestination>;
 
 private:
    const PhvInfo &phv_i;
    AliasMap aliasMap;
    PragmaNoOverlay &no_overlay;
 
    bitvec fieldsWithExpressions;
    bitvec fieldsWithAliasingSrc;
    bitvec fieldsWithAliasingDst;
 
    profile_t init_apply(const IR::Node *root) override;
 
    bool preorder(const IR::Expression *expr) override;
 
    void postorder(const IR::BFN::Pipe *pipe) override;
 
 public:
    explicit PragmaAlias(PhvInfo &phv, PragmaNoOverlay &no_ovrl)
        : phv_i(phv), no_overlay(no_ovrl) {}
    const AliasMap &getAliasMap() const { return aliasMap; }
 
    static const char *name;
    static const char *description;
    static const char *help;
 
    std::optional<std::pair<const PHV::Field *, const PHV::Field *>> mayAddAlias(
        const PHV::Field *f1, const PHV::Field *f2, bool suppressWarning = false,
        CreatedBy who = PRAGMA);
 
    bool addAlias(const PHV::Field *f1, const PHV::Field *f2, bool suppressWarning = false,
                  CreatedBy who = PRAGMA);
 
    std::string pretty_print() const override;
};
 
std::ostream &operator<<(std::ostream &out, const PragmaAlias &pa_a);
std::ostream &operator<<(std::ostream &, const PragmaAlias::AliasDestination &dest);
 
#endif /* BACKENDS_TOFINO_BF_P4C_PHV_PRAGMA_PA_ALIAS_H_ */