AliPHOSMemoryWatcher moved to STEER and renamed as AliMemoryWatcher
[u/mrichter/AliRoot.git] / STEER / AliMemoryWatcher.cxx
CommitLineData
d60522e4 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15/* $Id$ */
16//_________________________________________________________________________
00399209 17//Basic Memory Leak utility.
18// You can use this tiny class to *see* if your program is leaking.
19// Usage:
ac8cd48d 20// AliMemoryWatcher memwatcher;
00399209 21// some program loop on events here {
22// if ( nevents % x == 0 )
23// {
24// // take a sample every x events
25// memwatcher.watch(nevents);
26// }
27// }
28// TFile f("out.root","RECREATE");
29// memwatcher.write();
30// f.Close();
31// In the output root file you'll get 3 graphs representing
32// the evolAliPHOSon, as a function of the number of events, of :
33// - VSIZE is the virtual size (in KBytes) of your program, that is sort of
34// the total memory used
35// - RSSIZE is the resident size (in KBytes), that is, the part of your
36// program which is really in physical memory.
37// - TIME is an estimate of time per event (really it's the time elasped
38// between two calls to watch method)
39// WARNING: this is far from a bulletproof memory report (it's basically
40// using UNIX command ps -h -p [PID] -o vsize,rssize to do its job).
41// It has only been tested on Linux so far.
42// But by fitting the VSIZE by a pol1 under ROOT, you'll see right away
43// by how much your program is leaking.
d60522e4 44//*-- Author: Laurent Aphecetche(SUBATECH)
d60522e4 45// --- std system ---
b01af8c2 46class assert ;
d60522e4 47// --- AliRoot header files ---
ac8cd48d 48#include "AliMemoryWatcher.h"
d60522e4 49// --- ROOT system ---
50#include "TSystem.h"
51#include "TGraph.h"
52#include "TH2.h"
53#include "TStopwatch.h"
dfe7f75b 54
ac8cd48d 55ClassImp(AliMemoryWatcher)
dfe7f75b 56
d60522e4 57//_____________________________________________________________________________
ac8cd48d 58AliMemoryWatcher::AliMemoryWatcher(UInt_t maxsize)
d60522e4 59{
b01af8c2 60 //ctor
d60522e4 61 fMAXSIZE=maxsize;
62 fPID = gSystem->GetPid();
63 sprintf(fCmd,"ps -h -p %d -o vsize,rssize",fPID);
b01af8c2 64 fX = new Int_t[fMAXSIZE];
65 fVSIZE = new Int_t[fMAXSIZE];
66 fRSSIZE = new Int_t[fMAXSIZE];
67 fTIME = new Double_t[fMAXSIZE];
d60522e4 68 fSize=0;
69 fDisabled=false;
70 fTimer=0;
71}
d60522e4 72//_____________________________________________________________________________
ac8cd48d 73AliMemoryWatcher::AliMemoryWatcher(AliMemoryWatcher& mw)
00399209 74{
75 //copy ctor
76 fMAXSIZE = mw.fMAXSIZE ;
77 fPID = mw.fPID ;
78 strcpy(fCmd, mw.fCmd) ;
79 fX = new Int_t[fMAXSIZE];
80 fVSIZE = new Int_t[fMAXSIZE];
81 fRSSIZE = new Int_t[fMAXSIZE];
82 fTIME = new Double_t[fMAXSIZE];
83 fSize=0;
84 fDisabled=false;
85 fTimer=0;
86}
87//_____________________________________________________________________________
ac8cd48d 88AliMemoryWatcher::~AliMemoryWatcher()
d60522e4 89{
2a6e66b9 90 // dtor
d60522e4 91 delete[] fVSIZE;
92 delete[] fRSSIZE;
93 delete[] fX;
94 delete[] fTIME;
95 delete fTimer;
96}
d60522e4 97//_____________________________________________________________________________
ac8cd48d 98void AliMemoryWatcher::Watch(Int_t x)
d60522e4 99{
b01af8c2 100 // Sets the point where CPU parameters have to be monitored
d60522e4 101 if ( !fDisabled && fSize < fMAXSIZE ) {
d60522e4 102 if ( fSize==0 ) {
103 assert(fTimer==0);
104 fTimer = new TStopwatch;
105 fTimer->Start(true);
106 fTimer->Stop();
107 }
b01af8c2 108 static Int_t vsize, rssize;
d60522e4 109 static FILE* pipe = 0;
d60522e4 110 pipe = popen(fCmd,"r");
d60522e4 111 if ( pipe ) {
112
113 fscanf(pipe,"%d %d",&vsize,&rssize);
114
115 fX[fSize] = x ;
116 fVSIZE[fSize] = vsize ;
117 fRSSIZE[fSize] = rssize ;
118 fTIME[fSize] = fTimer->CpuTime();
119 fSize++;
120 }
d60522e4 121 assert(pclose(pipe)!=-1);
d60522e4 122 fTimer->Start(true);
d60522e4 123 }
124 else {
125 fDisabled=true;
21cd0c07 126 Error("watch", "I'm full !" ) ;
d60522e4 127 }
128}
d60522e4 129//_____________________________________________________________________________
130TGraph*
ac8cd48d 131AliMemoryWatcher::GraphVSIZE(void)
d60522e4 132{
b01af8c2 133 // Fills the graph with the virtual memory sized used
d60522e4 134 TGraph* g = 0;
00399209 135 if ( Size() )
d60522e4 136 {
00399209 137 g = new TGraph(Size());
b01af8c2 138 Int_t i ;
139 for (i=0; i < g->GetN(); i++ ) {
d60522e4 140 g->SetPoint(i,X(i),VSIZE(i));
141 }
142 }
143 return g;
144}
d60522e4 145//_____________________________________________________________________________
146TGraph*
ac8cd48d 147AliMemoryWatcher::GraphRSSIZE(void)
d60522e4 148{
b01af8c2 149 // Fills the graph with the real memory sized used
d60522e4 150 TGraph* g = 0;
00399209 151 if ( Size() )
d60522e4 152 {
00399209 153 g = new TGraph(Size());
b01af8c2 154 Int_t i ;
155 for (i=0; i < g->GetN(); i++ ) {
d60522e4 156 g->SetPoint(i,X(i),RSSIZE(i));
157 }
158 }
159 return g;
160}
d60522e4 161//_____________________________________________________________________________
162TGraph*
ac8cd48d 163AliMemoryWatcher::GraphTIME(void)
d60522e4 164{
b01af8c2 165 // Fills the raph with the used CPU time
d60522e4 166 TGraph* g = 0;
00399209 167 if ( Size() )
d60522e4 168 {
00399209 169 g = new TGraph(Size());
b01af8c2 170 Int_t i ;
171 for (i=0; i < g->GetN(); i++ ) {
d60522e4 172 g->SetPoint(i,X(i),TIME(i));
173 }
174 }
175 return g;
176}
d60522e4 177//_____________________________________________________________________________
178TH2*
ac8cd48d 179AliMemoryWatcher::Frame(void) const
d60522e4 180{
b01af8c2 181 //creates the frame histo in which the graphs will be plotted
182 Double_t xmin=1E30;
183 Double_t xmax=0;
184 Double_t ymin=1;
185 Double_t ymax=0;
186 UInt_t i ;
00399209 187 for (i=0; i < Size() ; i++ ) {
d60522e4 188 if ( X(i) < xmin ) xmin = X(i);
189 if ( X(i) > xmax ) xmax = X(i);
b01af8c2 190 Double_t y = VSIZE(i)+RSSIZE(i);
d60522e4 191 if ( y > ymax ) ymax = y;
d60522e4 192 if ( VSIZE(i) < ymin ) ymin = VSIZE(i);
193 if ( RSSIZE(i) < ymin ) ymin = RSSIZE(i);
194 }
d60522e4 195 TH2F* h = new TH2F("frame","",10,xmin,xmax,10,ymin*0.8,ymax*1.2);
d60522e4 196 return h;
197}
d60522e4 198//_____________________________________________________________________________
199void
ac8cd48d 200AliMemoryWatcher::Write(void)
d60522e4 201{
b01af8c2 202 // Stores the graphs in a file
203 if ( GraphVSIZE() ) GraphVSIZE()->Write("VSIZE",TObject::kOverwrite);
00399209 204 if ( GraphRSSIZE() ) GraphRSSIZE() ->Write("RSSIZE",TObject::kOverwrite);
b01af8c2 205 if ( GraphTIME() ) GraphTIME()->Write("TIME",TObject::kOverwrite);
4ed9ed00 206}