pna/shared_table_stepper.h

/*
 * SPDX-FileCopyrightText: 2023 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_SHARED_TABLE_STEPPER_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_SHARED_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/pna/shared_expr_stepper.h"
#include "backends/p4tools/modules/testgen/targets/pna/test_spec.h"
 
namespace P4::P4Tools::P4Testgen::Pna {
 
class SharedPnaTableStepper : public TableStepper {
 protected:
    bool checkForActionProfile();

    bool checkForActionSelector();

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

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

    enum class TableImplementation { standard, selector, profile, constant, skip };

    struct SharedPnaProperties {
        const PnaDpdkActionProfile *actionProfile = nullptr;

        const PnaDpdkActionSelector *actionSelector = nullptr;

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

        bool addProfileToState = false;

        TableImplementation implementaton = TableImplementation::standard;
    } SharedPnaProperties;
 
    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 SharedPnaTableStepper(SharedPnaExprStepper *stepper, const IR::P4Table *table);
};
 
}  // namespace P4::P4Tools::P4Testgen::Pna
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_SHARED_TABLE_STEPPER_H_ */