addMissingIds.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef CONTROL_PLANE_ADDMISSINGIDS_H_
#define CONTROL_PLANE_ADDMISSINGIDS_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/typeChecking/typeChecker.h"
#include "ir/ir.h"
#include "p4RuntimeSymbolTable.h"
 
namespace P4 {
 
class MissingIdAssigner : public Transform {
    ReferenceMap *refMap;

    TypeMap *typeMap;

    static constexpr int ID_BIT_WIDTH = 32;

    const ControlPlaneAPI::P4RuntimeSymbolTable *symbols = nullptr;

    const ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface &archBuilder;
 
    const IR::P4Program *preorder(IR::P4Program *program) override;
    const IR::Property *postorder(IR::Property *property) override;
    const IR::P4Table *postorder(IR::P4Table *table) override;
    const IR::Type_Header *postorder(IR::Type_Header *hdr) override;
    const IR::P4ValueSet *postorder(IR::P4ValueSet *valueSet) override;
    const IR::P4Action *postorder(IR::P4Action *action) override;
 
 public:
    explicit MissingIdAssigner(
        ReferenceMap *refMap, TypeMap *typeMap,
        const ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface &archBuilder);
 
    explicit MissingIdAssigner(
        ReferenceMap *refMap, TypeMap *typeMap,
        const ControlPlaneAPI::P4RuntimeSymbolTable *symbols,
        const ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface &archBuilder);
};

class AddMissingIdAnnotations final : public PassManager {
 public:
    AddMissingIdAnnotations(ReferenceMap *refMap, TypeMap *typeMap,
                            const ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface *archBuilder);
};
 
}  // namespace P4
 
#endif /* CONTROL_PLANE_ADDMISSINGIDS_H_ */