selector_update.h

 
#ifndef BF_P4C_MAU_SELECTOR_UPDATE_H_
#define BF_P4C_MAU_SELECTOR_UPDATE_H_
 
#include "ir/pass_manager.h"
#include "lib/ordered_map.h"
#include "mau_visitor.h"
 
using namespace P4;

 
class AddSelectorSalu : public PassManager {
    // map associating selectors with the stateful alu that updates them
    ordered_map<const IR::MAU::Selector *, const IR::MAU::StatefulAlu *> sel2salu;
 
    class FindSelectorSalu : public MauInspector {
        AddSelectorSalu &self;
        bool preorder(const IR::MAU::StatefulAlu *salu) {
            if (salu->selector) self.sel2salu[salu->selector] = salu;
            return false;
        }
        bool preorder(const IR::Expression *) { return false; }  // prune all expressions
     public:
        explicit FindSelectorSalu(AddSelectorSalu &self) : self(self) {}
    };
 
    class AddSaluIfNeeded : public MauModifier {
        AddSelectorSalu &self;
        bool preorder(IR::MAU::Table *);
        bool preorder(IR::Expression *) { return false; }  // prune all expressions
     public:
        explicit AddSaluIfNeeded(AddSelectorSalu &self) : self(self) {}
    };
 
 public:
    AddSelectorSalu() : PassManager({new FindSelectorSalu(*this), new AddSaluIfNeeded(*this)}) {}
};
 
#endif /* BF_P4C_MAU_SELECTOR_UPDATE_H_ */