targets/bmv2/program_info.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_
 
#include <cstddef>
 
#include "control-plane/p4RuntimeSerializer.h"
#include "ir/ir.h"
#include "lib/cstring.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/targets/bmv2/compiler_result.h"
 
namespace P4::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 P4::P4Tools::P4Testgen::Bmv2
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_PROGRAM_INFO_H_ */