check_duplicate.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_MAU_CHECK_DUPLICATE_H_
#define BACKENDS_TOFINO_BF_P4C_MAU_CHECK_DUPLICATE_H_
 
#include "ir/dump.h"
 
using namespace P4;
 
class CheckTableNameDuplicate : public MauInspector {
    std::set<cstring> names;
    profile_t init_apply(const IR::Node *root) override {
        names.clear();
        return MauInspector::init_apply(root);
    }
    bool preorder(const IR::MAU::Table *t) override {
        auto name = t->name;
        if (t->is_placed()) name = t->unique_id().build_name();
        if (names.count(name)) BUG("Multiple tables named '%s'", name);
        names.insert(name);
        return true;
    }
};
 
class CheckDuplicateAttached : public Inspector {
    std::map<cstring, const IR::MAU::AttachedMemory *> attached;
 
 public:
    const char *pass = "";
    bool ok = true;
    profile_t init_apply(const IR::Node *root) {
        attached.clear();
        return Inspector::init_apply(root);
    }
    bool preorder(const IR::MAU::AttachedMemory *at) {
        if (attached.count(at->name)) {
            LOG1("Duplicated attached table " << at->name << " after " << pass);
            ok = false;
        }
        attached[at->name] = at;
        return true;
    }
    void end_apply(const IR::Node *root) {
        if (!ok && LOGGING(3)) {
            std::cout << "----------  After " << pass << "  ----------" << std::endl;
            dump(root);
        }
        BUG_CHECK(ok, "abort after %s", pass);
    }
};
 
#endif /* BACKENDS_TOFINO_BF_P4C_MAU_CHECK_DUPLICATE_H_ */