p4tools/modules/testgen/core/target.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_TARGET_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_TARGET_H_
 
#include <string>
 
#include "backends/p4tools/common/compiler/compiler_target.h"
#include "ir/ir.h"
#include "ir/solver.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/core/small_step/cmd_stepper.h"
#include "backends/p4tools/modules/testgen/core/small_step/expr_stepper.h"
#include "backends/p4tools/modules/testgen/core/symbolic_executor/symbolic_executor.h"
#include "backends/p4tools/modules/testgen/lib/execution_state.h"
#include "backends/p4tools/modules/testgen/lib/test_backend.h"
 
namespace P4::P4Tools::P4Testgen {
 
class TestgenTarget : public CompilerTarget {
 public:
    static const TestgenTarget &get();

    static const ProgramInfo *produceProgramInfo(const CompilerResult &compilerResult);

    static TestBackEnd *getTestBackend(const ProgramInfo &programInfo,
                                       const TestBackendConfiguration &testBackendConfiguration,
                                       SymbolicExecutor &symbex);

    static CmdStepper *getCmdStepper(ExecutionState &state, AbstractSolver &solver,
                                     const ProgramInfo &programInfo);

    static ExprStepper *getExprStepper(ExecutionState &state, AbstractSolver &solver,
                                       const ProgramInfo &programInfo);
 
 protected:
    [[nodiscard]] const ProgramInfo *produceProgramInfoImpl(
        const CompilerResult &compilerResult) const;

    virtual const ProgramInfo *produceProgramInfoImpl(
        const CompilerResult &compilerResult, const IR::Declaration_Instance *mainDecl) const = 0;

    virtual TestBackEnd *getTestBackendImpl(
        const ProgramInfo &programInfo, const TestBackendConfiguration &testBackendConfiguration,
        SymbolicExecutor &symbex) const = 0;

    virtual CmdStepper *getCmdStepperImpl(ExecutionState &state, AbstractSolver &solver,
                                          const ProgramInfo &programInfo) const = 0;

    virtual ExprStepper *getExprStepperImpl(ExecutionState &state, AbstractSolver &solver,
                                            const ProgramInfo &programInfo) const = 0;
 
    explicit TestgenTarget(const std::string &deviceName, const std::string &archName);
 
    CompilerResultOrError runCompilerImpl(const CompilerOptions &options,
                                          const IR::P4Program *program) const override;
 
    [[nodiscard]] ICompileContext *makeContext() const override;
};
 
}  // namespace P4::P4Tools::P4Testgen
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_TARGET_H_ */