ir-traversal.h

/*
 * SPDX-FileCopyrightText: 2024 Intel Corporation.
 * Copyright 2024-present Intel Corporation.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef IR_IR_TRAVERSAL_H_
#define IR_IR_TRAVERSAL_H_
#define IR_IR_TRAVERSAL_INTERNAL_ENABLE
 
#include <cstddef>
 
namespace P4::IR::Traversal {

template <typename T>
struct Assign {
    explicit Assign(const T &value) : value(value) {}
    T value;
};

struct Index {
    explicit Index(size_t value) : value(value) {}
    size_t value;
};
 
}  // namespace P4::IR::Traversal
 
// Put the internals out of the main file to increase its readability.
#include "ir-traversal-internal.h"
 
namespace P4::IR::Traversal {

template <typename Obj, typename... Selectors>
Obj *modify(Obj *obj, Selectors &&...selectors) {
    return Detail::Traverse::modify(obj, std::forward<Selectors>(selectors)...);
}

template <typename Obj, typename... Selectors>
Obj *apply(const Obj *obj, Selectors &&...selectors) {
    return Detail::Traverse::modify(obj->clone(), std::forward<Selectors>(selectors)...);
}
 
}  // namespace P4::IR::Traversal
 
#undef IR_IR_TRAVERSAL_INTERNAL_ENABLE
#endif  // IR_IR_TRAVERSAL_H_