MemoryMonitor.h

Go to the documentation of this file.

#ifndef MemoryMonitor_h
#define MemoryMonitor_h

#include "marlin/Processor.h"
#include "lcio.h"

#include "stdlib.h"
#include "stdio.h"
#include "string.h"

using namespace lcio ;
using namespace marlin ;


class MemoryMonitor : public Processor {
        
public:
    
    virtual Processor*  newProcessor() { return new MemoryMonitor ; }
    
    MemoryMonitor() ;
    
    // Initialisation - run at the beginning to start histograms, etc.
    virtual void init() ;
    
    // Called at the beginning of every run
    virtual void processRunHeader( LCRunHeader* run ) ;
    
    // Run over each event - the main algorithm
    virtual void processEvent( LCEvent * evt ) ;
    
  // Run at the end of each event
    virtual void check( LCEvent * evt ) ;
    
    // Called at the very end for cleanup, histogram saving, etc.
    virtual void end() ;
    
    
protected:
    
  int _howOften=1;
    
  // Run and event counters
  int _eventNumber=-1;
  int _runNumber=-1;
  
  int parseLine(char* line){
    int i = strlen(line);
    while (*line < '0' || *line > '9') line++;
    line[i-3] = '\0';
    i = atoi(line);
    return i;
  };
  
  //Used only on linux machines
  int getValue(){ //Note: this value is in KB!
    FILE* file = fopen("/proc/self/status", "r");
    int result = -1;
    char line[128];
    
    
    while (fgets(line, 128, file) != NULL){
      if (strncmp(line, "VmRSS:", 6) == 0){
        result = parseLine(line);
        break;
      }
    }
    fclose(file);
    return result;
  };

    
} ;

#endif