heffte_trace.h

/*
    -- heFFTe --
       Univ. of Tennessee, Knoxville
       @date
*/
 
#ifndef HEFFTE_TRACE_H
#define HEFFTE_TRACE_H
 
#include "heffte_utils.h"

 
namespace heffte {
 
    #ifdef Heffte_ENABLE_TRACING
    struct event {
        std::string name;
        double start_time;
        double duration;
    };

    extern std::deque<event> event_log;
    extern std::string log_filename;

    struct add_trace {
        add_trace(std::string s) : name(s), start_time(MPI_Wtime()){}
        ~add_trace(){
            double duration = MPI_Wtime() - start_time;
            event_log.push_back({std::move(name), start_time, duration});
        }

        std::string name;
        double start_time;
    };

    inline void init_tracing(std::string root_filename){
        event_log = std::deque<event>();
        log_filename = root_filename + "_" + std::to_string(mpi::world_rank()) + ".log";
    }

    inline void finalize_tracing(){
        std::ofstream ofs(log_filename);
        ofs.precision(12);
 
        for(auto const &e : event_log)
            ofs << std::setw(40) << e.name << std::setw(20) << e.start_time << std::setw(20) << e.duration << "\n";
    }
 
    #else
 
    struct add_trace{
        add_trace(std::string){}
    };
    inline void init_tracing(std::string){}
    inline void finalize_tracing(){}
 
    #endif
 
}
 
#endif        //  #ifndef HEFFTE_TRACE_H