coverage.h

#ifndef MIDEND_COVERAGE_H_
#define MIDEND_COVERAGE_H_
 
#include <set>
#include <vector>
 
#include "ir/ir.h"
#include "ir/visitor.h"
#include "lib/source_file.h"
 
 
namespace P4::Coverage {
 
struct SourceIdCmp {
    bool operator()(const IR::Node *s1, const IR::Node *s2) const;
};
 
struct CoverageOptions {
    bool coverStatements = false;
    bool coverTableEntries = false;
    bool coverActions = false;
 
    bool onlyCoveringTests = false;
};
 
using CoverageSet = std::set<const IR::Node *, SourceIdCmp>;
 
class CollectNodes : public Inspector {
    CoverageSet coverableNodes;
 
    CoverageOptions coverageOptions;
 
    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::Entry *entry) override;
 
    bool preorder(const IR::P4Action *act) override;
 
 public:
    explicit CollectNodes(CoverageOptions coverageOptions);
 
    const CoverageSet &getCoverableNodes();
};
 
void printCoverageReport(const CoverageSet &all, const CoverageSet &visited);
 
void logCoverage(const CoverageSet &all, const CoverageSet &visited, const CoverageSet &new_);
 
}  // namespace P4::Coverage
 
#endif /* MIDEND_COVERAGE_H_ */