inode.h

/*
 * SPDX-FileCopyrightText: 2013 Barefoot Networks, Inc.
 * Copyright 2013-present Barefoot Networks, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef IR_INODE_H_
#define IR_INODE_H_
 
#include "ir/gen-tree-macro.h"
#include "lib/castable.h"
#include "lib/cstring.h"
#include "lib/exceptions.h"
#include "lib/rtti.h"
#include "lib/source_file.h"
 
namespace P4 {
class JSONGenerator;
class JSONLoader;
}  // namespace P4
 
namespace P4::IR {
 
class Node;
class Annotation;
template <class>
class Vector;

template <class, class = void>
struct has_static_type_name : std::false_type {};
 
template <class T>
struct has_static_type_name<T, std::void_t<decltype(T::static_type_name())>> : std::true_type {};
 
template <class T>
inline constexpr bool has_static_type_name_v = has_static_type_name<T>::value;
 
// node interface
class INode : public Util::IHasSourceInfo, public IHasDbPrint, public ICastable {
 public:
    virtual ~INode() {}
    virtual const Node *getNode() const = 0;
    virtual Node *getNode() = 0;
    virtual void toJSON(JSONGenerator &) const = 0;
    virtual cstring node_type_name() const = 0;
    virtual void validate() const {}
 
    // default checkedTo implementation for nodes: just fallback to generic ICastable method
    template <typename T>
    std::enable_if_t<!has_static_type_name_v<T>, const T *> checkedTo() const {
        return ICastable::checkedTo<T>();
    }
 
    // alternative checkedTo implementation that produces slightly better error message
    // due to node_type_name() / static_type_name() being available
    template <typename T>
    std::enable_if_t<has_static_type_name_v<T>, const T *> checkedTo() const {
        const auto *result = to<T>();
        BUG_CHECK(result, "Cast failed: %1% with type %2% is not a %3%.", this, node_type_name(),
                  T::static_type_name());
        return result;
    }
 
    // helpers for getting annotations
    bool hasAnnotation(cstring) const;
    const Annotation *getAnnotation(cstring) const;
    const Vector<Annotation> &getAnnotations() const;
 
    DECLARE_TYPEINFO_WITH_TYPEID(INode, NodeKind::INode);
};
 
}  // namespace P4::IR
 
#endif /* IR_INODE_H_ */