pna/backend/metadata/metadata.h

#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_BACKEND_METADATA_METADATA_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_BACKEND_METADATA_METADATA_H_
 
#include <cstddef>
#include <filesystem>
#include <fstream>
#include <optional>
#include <string>
#include <utility>
#include <vector>
 
#include <inja/inja.hpp>
 
#include "ir/ir.h"
#include "lib/cstring.h"
 
#include "backends/p4tools/modules/testgen/lib/test_framework.h"
#include "backends/p4tools/modules/testgen/lib/test_spec.h"
 
namespace P4::P4Tools::P4Testgen::Pna {

class Metadata : public TestFramework {
    std::ofstream metadataFile;
 
 public:
    ~Metadata() override = default;
    Metadata(const Metadata &) = delete;
    Metadata(Metadata &&) = delete;
    Metadata &operator=(const Metadata &) = delete;
    Metadata &operator=(Metadata &&) = delete;
 
    explicit Metadata(const TestBackendConfiguration &testBackendConfiguration);

    void writeTestToFile(const TestSpec *spec, cstring selectedBranches, size_t testId,
                         float currentCoverage) override;
 
 private:
    void emitPreamble(const std::string &preamble);

    void emitTestcase(const TestSpec *testSpec, cstring selectedBranches, size_t testId,
                      const std::string &testCase, float currentCoverage);

    static void computeTraceData(const TestSpec *testSpec, inja::json &dataJson);

    static std::string getTestCaseTemplate();

    static inja::json getSend(const TestSpec *testSpec);

    static inja::json getVerify(const TestSpec *testSpec);

    static std::vector<std::pair<size_t, size_t>> getIgnoreMasks(const IR::Constant *mask);
};
 
}  // namespace P4::P4Tools::P4Testgen::Pna
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_PNA_BACKEND_METADATA_METADATA_H_ */