param_binding.h

 
#ifndef _MIDEND_PARAM_BINDING_H_
#define _MIDEND_PARAM_BINDING_H_
 
#include <map>
 
#include "ir/ir.h"
 
namespace P4 {
class TypeMap;
}  // namespace P4
 
class ParamBinding : public Inspector {
    const P4::TypeMap *typeMap;
    bool isV1;
    std::map<const IR::Type *, const IR::InstanceRef *> by_type;
    std::map<const IR::Parameter *, const IR::InstanceRef *> by_param;
    std::map<const IR::Declaration_Variable *, const IR::InstanceRef *> by_declvar;
 
 public:
    explicit ParamBinding(const P4::TypeMap *typeMap, bool isV1 = false)
        : typeMap(typeMap), isV1(isV1) {}
 
    void bind(const IR::Parameter *param);
 
    void bind(const IR::Declaration_Variable *var);
 
    const IR::InstanceRef *get(const IR::Parameter *param) const {
        return by_param.count(param) ? by_param.at(param) : nullptr;
    }
 
    const IR::InstanceRef *get(const IR::Declaration_Variable *var) const {
        return by_declvar.count(var) ? by_declvar.at(var) : nullptr;
    }
 
    void postorder(const IR::Parameter *param);
 
    void postorder(const IR::Declaration_Variable *var);
};
 
#endif /* _MIDEND_PARAM_BINDING_H_ */