collect_coverable_nodes.h

/*
 * SPDX-FileCopyrightText: 2023 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#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::BaseAssignmentStatement *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_*/