MemoryMonitorProcessor.cc

Go to the documentation of this file.

// -- marlinmt headers
#include <marlinmt/Processor.h>
#include <marlinmt/PluginManager.h>

// -- std headers
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#ifdef __APPLE__
#include <sys/sysctl.h>
#include <mach/mach.h>
#else
#include <sys/sysinfo.h>
#endif

namespace marlinmt {

  class MemoryMonitorProcessor : public Processor {
  public:
    MemoryMonitorProcessor() ;

    // from Processor
    void init() override ;
    void processEvent( EventStore * evt ) override ;

  protected:
    UIntParameter _howOften {*this, "howOften",
              "Print event number every N events", 1 } ;

    unsigned int      _eventNumber {0} ;
  };

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  MemoryMonitorProcessor::MemoryMonitorProcessor() :
    Processor("MemoryMonitor") {
    // modify processor description
    setDescription( "Simple processor to print out the memory consumption at defined intervals" ) ;
    // It doesn't make sense to create clones of this processor in MT environement
    // So we don't duplicate it per thread. To make it thread-safe we thus need to lock
    // make it thread-safe by using a lock
    setRuntimeOption( Processor::ERuntimeOption::eCritical, true ) ;
    // don't duplicate opening/writing of output file
    setRuntimeOption( Processor::ERuntimeOption::eClone, false ) ;
  }

  //--------------------------------------------------------------------------

  void MemoryMonitorProcessor::init() {
    // Print the initial parameters
    printParameters() ;
  }

  //--------------------------------------------------------------------------

  void MemoryMonitorProcessor::processEvent( EventStore * /*evt*/ ) {
    if (_eventNumber % _howOften == 0) {

#ifdef __APPLE__
    struct task_basic_info t_info;
    mach_msg_type_number_t t_info_count = TASK_BASIC_INFO_COUNT;

    if (KERN_SUCCESS != task_info(mach_task_self(),
      TASK_BASIC_INFO, (task_info_t)&t_info,
                                  &t_info_count)) {
      log<ERROR>() << "Problem accessing system information from  MacOS X!"<< std::endl ;
    }
    log<MESSAGE>() << " Processed event  " << _eventNumber
    << " Resident size is: "<< t_info.resident_size<<" virtual size: "<<t_info.virtual_size
    << std::endl ;
#else
    struct sysinfo memInfo;
    sysinfo (&memInfo);
    unsigned long physMemUsed = memInfo.totalram - memInfo.freeram;
    log<MESSAGE>() << " Processed event  " << _eventNumber
                          << " Physical memory in use: " << physMemUsed
                          << std::endl ;
#endif
    }
    _eventNumber++ ;
  }

  // plugin declaration
  MARLINMT_DECLARE_PROCESSOR( MemoryMonitorProcessor )
}