program_info.h

#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_
 
#include <cstddef>
#include <map>
#include <vector>
 
#include "control-plane/p4RuntimeSerializer.h"
#include "ir/ir.h"
#include "lib/cstring.h"
#include "lib/ordered_map.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/lib/continuation.h"
#include "backends/p4tools/modules/testgen/targets/bmv2/bmv2.h"
 
namespace P4Tools::P4Testgen::Bmv2 {
 
class Bmv2V1ModelProgramInfo : public ProgramInfo {
 private:
    const ordered_map<cstring, const IR::Type_Declaration *> programmableBlocks;
 
    const std::map<int, int> declIdToGress;
 
    static const IR::Type_Bits PARSER_ERR_BITS;
 
    std::vector<Continuation::Command> processDeclaration(const IR::Type_Declaration *typeDecl,
                                                          size_t blockIdx) const;
 
 public:
    Bmv2V1ModelProgramInfo(const BMv2V1ModelCompilerResult &compilerResult,
                           ordered_map<cstring, const IR::Type_Declaration *> inputBlocks,
                           std::map<int, int> declIdToGress);
 
    int getGress(const IR::Type_Declaration *) const;
 
    [[nodiscard]] P4::P4RuntimeAPI getP4RuntimeAPI() const;
 
    const IR::P4Table *getTableofDirectExtern(const IR::IDeclaration *directExternDecl) const;
 
    [[nodiscard]] const ArchSpec &getArchSpec() const override;
 
    [[nodiscard]] const ordered_map<cstring, const IR::Type_Declaration *> *getProgrammableBlocks()
        const;
 
    [[nodiscard]] const IR::PathExpression *getBlockParam(cstring blockLabel,
                                                          size_t paramIndex) const;
 
    [[nodiscard]] const IR::StateVariable &getTargetInputPortVar() const override;
 
    static const IR::Expression *getPortConstraint(
        const IR::StateVariable &portVar,
        const std::vector<std::pair<int, int>> &permittedPortRanges);
 
    [[nodiscard]] const IR::StateVariable &getTargetOutputPortVar() const override;
 
    [[nodiscard]] const IR::Expression *dropIsActive() const override;
 
    [[nodiscard]] const IR::Type_Bits *getParserErrorType() const override;
 
    [[nodiscard]] const BMv2V1ModelCompilerResult &getCompilerResult() const override;
 
    static const IR::Member *getParserParamVar(const IR::P4Parser *parser, const IR::Type *type,
                                               size_t paramIndex, cstring paramLabel);
 
    static const ArchSpec ARCH_SPEC;
 
    DECLARE_TYPEINFO(Bmv2V1ModelProgramInfo, ProgramInfo);
};
 
}  // namespace P4Tools::P4Testgen::Bmv2
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_ */