collect_coverable_nodes.h

#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_LIB_COLLECT_COVERABLE_NODES_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_LIB_COLLECT_COVERABLE_NODES_H_
 
#include <set>
 
#include "ir/ir.h"
#include "ir/visitor.h"
#include "midend/coverage.h"
 
#include "backends/p4tools/modules/testgen/lib/execution_state.h"
 
namespace P4::P4Tools::P4Testgen {

using NodeCache = std::map<const IR::Node *, P4::Coverage::CoverageSet>;

class CoverableNodesScanner : public Inspector {
    P4::Coverage::CoverageSet coverableNodes;

    std::reference_wrapper<const ExecutionState> state;

    std::set<int> seenParserIds;

    P4::Coverage::CoverageOptions coverageOptions;

    bool preorder(const IR::ParserState *parserState) override;
    bool preorder(const IR::AssignmentStatement *stmt) override;
    bool preorder(const IR::MethodCallStatement *stmt) override;
    bool preorder(const IR::ExitStatement *stmt) override;
    bool preorder(const IR::MethodCallExpression *call) override;

    bool preorder(const IR::P4Action *act) override;
 
 public:
    explicit CoverableNodesScanner(const ExecutionState &state);

    void updateNodeCoverage(const IR::Node *node, P4::Coverage::CoverageSet &nodes);

    const P4::Coverage::CoverageSet &getCoverableNodes();
};
 
}  // namespace P4::P4Tools::P4Testgen
 
#endif /*BACKENDS_P4TOOLS_MODULES_TESTGEN_LIB_COLLECT_COVERABLE_NODES_H_*/