hierarchicalNames.h

/*
 * Copyright 2017 VMware, Inc.
 * SPDX-FileCopyrightText: 2017 VMware, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef FRONTENDS_P4_HIERARCHICALNAMES_H_
#define FRONTENDS_P4_HIERARCHICALNAMES_H_
 
#include "ir/ir.h"
#include "ir/visitor.h"
 
namespace P4 {

class HierarchicalNames : public Modifier {
    std::vector<cstring> stack;
 
    cstring getName(const IR::IDeclaration *decl) const { return decl->getName(); }
 
 public:
    HierarchicalNames() {
        setName("HierarchicalNames");
        visitDagOnce = false;
    }
    bool preorder(IR::P4Parser *parser) override {
        stack.push_back(getName(parser));
        return true;
    }
    void postorder(IR::P4Parser *) override { stack.pop_back(); }
 
    bool preorder(IR::P4Control *control) override {
        stack.push_back(getName(control));
        return true;
    }
    void postorder(IR::P4Control *) override { stack.pop_back(); }
 
    bool preorder(IR::P4Table *table) override {
        visit(table->annotations);
        return false;
    }
 
    void postorder(IR::Annotation *annotation) override;
    // Do not change name annotations on parameters
    bool preorder(IR::Parameter *) override { return false; }
};
 
}  // namespace P4
 
#endif /* FRONTENDS_P4_HIERARCHICALNAMES_H_ */