backends/tofino/bf-p4c/logging/pass_manager.h

 
#ifndef _BACKENDS_TOFINO_BF_P4C_LOGGING_PASS_MANAGER_H_
#define _BACKENDS_TOFINO_BF_P4C_LOGGING_PASS_MANAGER_H_
 
#include "backends/tofino/bf-p4c/bf-p4c-options.h"
#include "backends/tofino/bf-p4c/logging/filelog.h"
#include "ir/ir.h"
#include "ir/pass_manager.h"
#include "lib/cstring.h"
 
namespace Logging {
 
class PassManager : public ::PassManager {
 protected:
    cstring _logFilePrefix;
    Logging::Mode _logMode;
    Logging::FileLog *_logFile;
 
    static int invocationCount;
 
 public:
    explicit PassManager(cstring logFilePrefix, Logging::Mode logMode = Logging::Mode::CREATE)
        : _logFilePrefix(logFilePrefix), _logMode(logMode), _logFile(nullptr) {}
 
    static cstring getNewLogFileName(const cstring &prefix) {
        return prefix + std::to_string(invocationCount) + ".log";
    }
 
 protected:
    profile_t init_apply(const IR::Node *root) override {
        if (BackendOptions().verbose == 0) {
            return ::PassManager::init_apply(root);
        }
 
        auto pipe = root->to<IR::BFN::Pipe>();
        if (pipe) {
            if (_logMode == Logging::Mode::CREATE) {
                ++invocationCount;
                auto logFileName = getNewLogFileName(_logFilePrefix);
                _logFile = new Logging::FileLog(pipe->canon_id(), logFileName, _logMode);
            } else {
                auto logFileName = _logFilePrefix + ".log";
                _logFile = new Logging::FileLog(pipe->canon_id(), logFileName, _logMode);
            }
        }
 
        return ::PassManager::init_apply(root);
    }
 
    void end_apply() override {
        Logging::FileLog::close(_logFile);
        ::PassManager::end_apply();
    }
};
}  // end namespace Logging
 
#endif  // _BACKENDS_TOFINO_BF_P4C_LOGGING_PASS_MANAGER_H_