ProfileModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <vector>
#include <functional>
 
namespace Belle2 {
  class ProfileModule : public Module {
 
  public:
 
    ProfileModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  private:
 
    std::string m_outputFileName;
 
    std::string m_rssOutputFileName;
 
    static const int k_burnIn = 1;         
    static const int k_maxPoints = 100;    
    struct MemTime {
      MemTime(unsigned long vm = 0, unsigned long rssm = 0, double t = 0) : virtualMem(vm), rssMem(rssm), time(t) {};
      unsigned long virtualMem;  
      unsigned long rssMem;  
      double time;        
    };
 
    typedef std::function< unsigned long (MemTime const&) > MemoryExtractLambda;
 
    const MemoryExtractLambda m_extractVirtualMem = [](MemTime const& m) { return m.virtualMem;};
 
    const MemoryExtractLambda m_extractRssMem = [](MemTime const& m) { return m.rssMem;};
 
    void storeMemoryGraph(const std::string& name, const std::string& title, const std::string& xAxisName, const std::string& imgOutput,
                          const MemoryExtractLambda& lmdMemoryExtract);
 
    double m_timeOffset;      
    MemTime m_initializeInfo; 
    MemTime m_terminateInfo;  
    int m_nEvents;            
    int m_step;               
    std::vector<MemTime> m_eventInfo; 
    MemTime m_startInfo;      
    MemTime m_endInfo;        
  };
}