p4_refers_to_parser.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_P4_REFERS_TO_PARSER_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_P4_REFERS_TO_PARSER_H_
 
#include "backends/p4tools/common/lib/variables.h"
#include "ir/ir.h"
#include "ir/visitor.h"
 
namespace P4::P4Tools::P4Testgen::Bmv2 {
 
class RefersToParser : public Inspector {
 private:
    ConstraintsVector restrictionsVector;

    using RefersToBuiltinMap = std::map<cstring, std::map<cstring, IR::SymbolicVariable>>;
    static const RefersToBuiltinMap REFERS_TO_BUILTIN_MAP;

    static cstring assembleKeyReference(const IR::Vector<IR::AnnotationToken> &annotationList,
                                        size_t offset);

    static const IR::SymbolicVariable *lookUpBuiltinKey(
        const IR::Annotation &refersAnno, const IR::Vector<IR::AnnotationToken> &annotationList);

    static const IR::SymbolicVariable *lookUpKeyInTable(const IR::P4Table &srcTable,
                                                        cstring keyReference);

    static const IR::SymbolicVariable *getReferencedKey(const IR::P4Control &ctrlContext,
                                                        const IR::Annotation &refersAnno);
 
    bool preorder(const IR::P4Table *table) override;
 
 public:
    RefersToParser();

    [[nodiscard]] ConstraintsVector getRestrictionsVector() const;
};
 
}  // namespace P4::P4Tools::P4Testgen::Bmv2
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_P4_REFERS_TO_PARSER_H_ */