selected_branches.h

/*
 * SPDX-FileCopyrightText: 2023 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SYMBOLIC_EXECUTOR_SELECTED_BRANCHES_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SYMBOLIC_EXECUTOR_SELECTED_BRANCHES_H_
 
#include <cstdint>
#include <list>
#include <string>
#include <vector>
 
#include "ir/solver.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/core/symbolic_executor/symbolic_executor.h"
#include "backends/p4tools/modules/testgen/lib/execution_state.h"
 
namespace P4::P4Tools::P4Testgen {

class SelectedBranches : public SymbolicExecutor {
 public:
    void runImpl(const Callback &callBack, ExecutionStateReference executionState) override;

    SelectedBranches(AbstractSolver &solver, const ProgramInfo &programInfo,
                     std::string selectedBranchesStr);
 
 private:
    ExecutionState *chooseBranch(const std::vector<Branch> &branches, uint64_t nextBranch);

    std::list<uint64_t> selectedBranches;
};
 
}  // namespace P4::P4Tools::P4Testgen
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SYMBOLIC_EXECUTOR_SELECTED_BRANCHES_H_ */