targets/bmv2/table_stepper.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TABLE_STEPPER_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TABLE_STEPPER_H_
 
#include <vector>
 
#include "ir/ir.h"
 
#include "backends/p4tools/modules/testgen/core/small_step/table_stepper.h"
#include "backends/p4tools/modules/testgen/lib/execution_state.h"
#include "backends/p4tools/modules/testgen/lib/test_spec.h"
#include "backends/p4tools/modules/testgen/targets/bmv2/expr_stepper.h"
#include "backends/p4tools/modules/testgen/targets/bmv2/test_spec.h"
 
namespace P4::P4Tools::P4Testgen::Bmv2 {
 
class Bmv2V1ModelTableStepper : public TableStepper {
 private:
    enum class TableImplementation { standard, selector, profile, constant, skip };

    struct Bmv2V1ModelProperties {
        const Bmv2V1ModelActionProfile *actionProfile = nullptr;

        const Bmv2V1ModelActionSelector *actionSelector = nullptr;

        std::vector<const IR::Expression *> actionSelectorKeys;

        bool addProfileToState = false;

        TableImplementation implementaton = TableImplementation::standard;
    } bmv2V1ModelProperties;

    bool checkForActionProfile();

    bool checkForActionSelector();

    void evalTableActionProfile(const std::vector<const IR::ActionListElement *> &tableActionList);

    void evalTableActionSelector(const std::vector<const IR::ActionListElement *> &tableActionList);
 
 protected:
    const IR::Expression *computeTargetMatchType(const TableUtils::KeyProperties &keyProperties,
                                                 TableMatchMap *matches,
                                                 const IR::Expression *hitCondition) override;
 
    void checkTargetProperties(
        const std::vector<const IR::ActionListElement *> &tableActionList) override;
 
    void evalTargetTable(
        const std::vector<const IR::ActionListElement *> &tableActionList) override;
 
 public:
    explicit Bmv2V1ModelTableStepper(Bmv2V1ModelExprStepper *stepper, const IR::P4Table *table);
};
 
}  // namespace P4::P4Tools::P4Testgen::Bmv2
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TABLE_STEPPER_H_ */