add_always_run.h

 
#ifndef BF_P4C_MAU_ADD_ALWAYS_RUN_H_
#define BF_P4C_MAU_ADD_ALWAYS_RUN_H_
 
#include <optional>
 
#include "backends/tofino/bf-p4c/logging/pass_manager.h"
#include "backends/tofino/bf-p4c/mau/table_flow_graph.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
 
using namespace P4;
 
class AddAlwaysRun : public PassManager {
    friend class AddAlwaysRunTest;
 
 private:
    const ordered_map<gress_t, ConstraintMap> &allTablesToAdd;
 
    ordered_map<gress_t, FlowGraph> flowGraphs;
 
    std::map<gress_t, ordered_map<const UniqueId, unsigned>> globalOrderings;
 
    std::map<const UniqueId, std::optional<UniqueId>> subsequentTables;
 
    int compare(const IR::MAU::Table *t1, const IR::MAU::Table *t2) const;
    int compare(const IR::MAU::Table *t1, std::optional<UniqueId> t2) const;
 
    class PrepareToAdd : public MauInspector {
        AddAlwaysRun &self;
 
        const IR::MAU::Table *subsequentTable = nullptr;
 
        std::map<const IR::MAU::Table *, const IR::MAU::Table *> minSubsequentTables;
 
        profile_t init_apply(const IR::Node *root) override;
        bool preorder(const IR::MAU::TableSeq *) override;
        bool preorder(const IR::MAU::Table *) override;
        void end_apply(const IR::Node *root) override;
 
     public:
        explicit PrepareToAdd(AddAlwaysRun &self) : self(self) {}
    };
 
    class AddTables : public MauTransform, ControlFlowVisitor {
        AddAlwaysRun &self;
 
        //
        // Invariant: this is empty at the start and end of each visit to each gress.
        std::list<const IR::MAU::Table *> tablesToAdd;
 
        std::optional<UniqueId> subsequentTable;
 
        const IR::BFN::Pipe *preorder(IR::BFN::Pipe *) override;
        const IR::Node *preorder(IR::MAU::TableSeq *) override;
 
        AddTables *clone() const override;
        AddTables &flow_clone() override;
        void flow_merge(Visitor &v) override;
        void flow_copy(::ControlFlowVisitor &v) override;
 
        AddTables(const AddTables &) = default;
        AddTables(AddTables &&) = default;
 
     public:
        explicit AddTables(AddAlwaysRun &self) : self(self) {}
    };
 
 public:
    explicit AddAlwaysRun(const ordered_map<gress_t, ConstraintMap> &tablesToAdd);
};
 
#endif /* BF_P4C_MAU_ADD_ALWAYS_RUN_H_ */