externalObjectsMetric.h

/*
Counts the number of external structure and function declarations
and uses. When encountering an extern structure declaration, its
method names are collected into "externFunctions". Encountering
an extern method is counted as an extern structure use and as an extern
function call too.
*/
 
#ifndef FRONTENDS_P4_METRICS_EXTERNALOBJECTSMETRIC_H_
#define FRONTENDS_P4_METRICS_EXTERNALOBJECTSMETRIC_H_
 
#include <set>
 
#include "frontends/p4/metrics/metricsStructure.h"
#include "ir/ir.h"
 
namespace P4 {
 
class ExternalObjectsMetricPass : public Inspector {
 private:
    ExternMetrics &metrics;
    std::set<cstring> externFunctions;                   // Standalone extern functions.
    std::set<cstring> externTypeNames;                   // Type names of extern structures.
    std::map<cstring, std::set<cstring>> externMethods;  // Tracks methods per extern.
 
 public:
    explicit ExternalObjectsMetricPass(Metrics &metricsRef) : metrics(metricsRef.externMetrics) {
        setName("ExternalObjectsMetricPass");
    }

    void postorder(const IR::Type_Extern *node) override;
    void postorder(const IR::Declaration_Instance *node) override;
    void postorder(const IR::Member *node) override;
    void postorder(const IR::Method *node) override;
    void postorder(const IR::MethodCallExpression *node) override;
    void postorder(const IR::P4Program * /*node*/) override;
};
 
}  // namespace P4
 
#endif /* FRONTENDS_P4_METRICS_EXTERNALOBJECTSMETRIC_H_ */