]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliMemoryWatcher.cxx
Additional/optional data members to investigate the reconstruction algorithms (M...
[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 ;
7406fd3c 47#include <malloc.h>
d60522e4 48// --- AliRoot header files ---
ac8cd48d 49#include "AliMemoryWatcher.h"
d60522e4 50// --- ROOT system ---
51#include "TSystem.h"
52#include "TGraph.h"
53#include "TH2.h"
54#include "TStopwatch.h"
dfe7f75b 55
ac8cd48d 56ClassImp(AliMemoryWatcher)
dfe7f75b 57
d60522e4 58//_____________________________________________________________________________
ac8cd48d 59AliMemoryWatcher::AliMemoryWatcher(UInt_t maxsize)
d60522e4 60{
b01af8c2 61 //ctor
d60522e4 62 fMAXSIZE=maxsize;
7406fd3c 63 fUseMallinfo = true;
d60522e4 64 fPID = gSystem->GetPid();
65 sprintf(fCmd,"ps -h -p %d -o vsize,rssize",fPID);
b01af8c2 66 fX = new Int_t[fMAXSIZE];
67 fVSIZE = new Int_t[fMAXSIZE];
68 fRSSIZE = new Int_t[fMAXSIZE];
69 fTIME = new Double_t[fMAXSIZE];
d60522e4 70 fSize=0;
71 fDisabled=false;
72 fTimer=0;
73}
d60522e4 74//_____________________________________________________________________________
d1898505 75AliMemoryWatcher::AliMemoryWatcher(const AliMemoryWatcher& mw):
76 TObject(mw)
00399209 77{
78 //copy ctor
79 fMAXSIZE = mw.fMAXSIZE ;
7406fd3c 80 fUseMallinfo = mw.fUseMallinfo;
00399209 81 fPID = mw.fPID ;
82 strcpy(fCmd, mw.fCmd) ;
83 fX = new Int_t[fMAXSIZE];
84 fVSIZE = new Int_t[fMAXSIZE];
85 fRSSIZE = new Int_t[fMAXSIZE];
86 fTIME = new Double_t[fMAXSIZE];
87 fSize=0;
88 fDisabled=false;
89 fTimer=0;
90}
91//_____________________________________________________________________________
ac8cd48d 92AliMemoryWatcher::~AliMemoryWatcher()
d60522e4 93{
2a6e66b9 94 // dtor
d60522e4 95 delete[] fVSIZE;
96 delete[] fRSSIZE;
97 delete[] fX;
98 delete[] fTIME;
99 delete fTimer;
100}
d60522e4 101//_____________________________________________________________________________
ac8cd48d 102void AliMemoryWatcher::Watch(Int_t x)
d60522e4 103{
b01af8c2 104 // Sets the point where CPU parameters have to be monitored
d60522e4 105 if ( !fDisabled && fSize < fMAXSIZE ) {
d60522e4 106 if ( fSize==0 ) {
107 assert(fTimer==0);
108 fTimer = new TStopwatch;
109 fTimer->Start(true);
110 fTimer->Stop();
111 }
7406fd3c 112 if ( fUseMallinfo ) {
113 static struct mallinfo meminfo;
114 meminfo = mallinfo();
d60522e4 115 fX[fSize] = x ;
7406fd3c 116 fVSIZE[fSize] = (meminfo.hblkhd + meminfo.uordblks) / 1024;
117 fRSSIZE[fSize] = meminfo.uordblks / 1024;
d60522e4 118 fTIME[fSize] = fTimer->CpuTime();
119 fSize++;
7406fd3c 120 } else {
121 static Int_t vsize, rssize;
122 static FILE* pipe = 0;
123 pipe = popen(fCmd,"r");
124 if ( pipe ) {
125
126 fscanf(pipe,"%d %d",&vsize,&rssize);
127
128 fX[fSize] = x ;
129 fVSIZE[fSize] = vsize ;
130 fRSSIZE[fSize] = rssize ;
131 fTIME[fSize] = fTimer->CpuTime();
132 fSize++;
133 }
134 assert(pclose(pipe)!=-1);
d60522e4 135 }
d60522e4 136 fTimer->Start(true);
d60522e4 137 }
138 else {
139 fDisabled=true;
21cd0c07 140 Error("watch", "I'm full !" ) ;
d60522e4 141 }
142}
d60522e4 143//_____________________________________________________________________________
144TGraph*
ac8cd48d 145AliMemoryWatcher::GraphVSIZE(void)
d60522e4 146{
b01af8c2 147 // Fills the graph with the virtual memory sized used
d60522e4 148 TGraph* g = 0;
00399209 149 if ( Size() )
d60522e4 150 {
00399209 151 g = new TGraph(Size());
b01af8c2 152 Int_t i ;
153 for (i=0; i < g->GetN(); i++ ) {
d60522e4 154 g->SetPoint(i,X(i),VSIZE(i));
155 }
156 }
157 return g;
158}
d60522e4 159//_____________________________________________________________________________
160TGraph*
ac8cd48d 161AliMemoryWatcher::GraphRSSIZE(void)
d60522e4 162{
b01af8c2 163 // Fills the graph with the real memory sized used
d60522e4 164 TGraph* g = 0;
00399209 165 if ( Size() )
d60522e4 166 {
00399209 167 g = new TGraph(Size());
b01af8c2 168 Int_t i ;
169 for (i=0; i < g->GetN(); i++ ) {
d60522e4 170 g->SetPoint(i,X(i),RSSIZE(i));
171 }
172 }
173 return g;
174}
d60522e4 175//_____________________________________________________________________________
176TGraph*
ac8cd48d 177AliMemoryWatcher::GraphTIME(void)
d60522e4 178{
b01af8c2 179 // Fills the raph with the used CPU time
d60522e4 180 TGraph* g = 0;
00399209 181 if ( Size() )
d60522e4 182 {
00399209 183 g = new TGraph(Size());
b01af8c2 184 Int_t i ;
185 for (i=0; i < g->GetN(); i++ ) {
d60522e4 186 g->SetPoint(i,X(i),TIME(i));
187 }
188 }
189 return g;
190}
d60522e4 191//_____________________________________________________________________________
192TH2*
ac8cd48d 193AliMemoryWatcher::Frame(void) const
d60522e4 194{
b01af8c2 195 //creates the frame histo in which the graphs will be plotted
196 Double_t xmin=1E30;
197 Double_t xmax=0;
198 Double_t ymin=1;
199 Double_t ymax=0;
200 UInt_t i ;
00399209 201 for (i=0; i < Size() ; i++ ) {
d60522e4 202 if ( X(i) < xmin ) xmin = X(i);
203 if ( X(i) > xmax ) xmax = X(i);
b01af8c2 204 Double_t y = VSIZE(i)+RSSIZE(i);
d60522e4 205 if ( y > ymax ) ymax = y;
d60522e4 206 if ( VSIZE(i) < ymin ) ymin = VSIZE(i);
207 if ( RSSIZE(i) < ymin ) ymin = RSSIZE(i);
208 }
d60522e4 209 TH2F* h = new TH2F("frame","",10,xmin,xmax,10,ymin*0.8,ymax*1.2);
d60522e4 210 return h;
211}
d60522e4 212//_____________________________________________________________________________
6c4904c2 213Int_t
214AliMemoryWatcher::Write(const char *, Int_t, Int_t)
d60522e4 215{
b01af8c2 216 // Stores the graphs in a file
217 if ( GraphVSIZE() ) GraphVSIZE()->Write("VSIZE",TObject::kOverwrite);
00399209 218 if ( GraphRSSIZE() ) GraphRSSIZE() ->Write("RSSIZE",TObject::kOverwrite);
b01af8c2 219 if ( GraphTIME() ) GraphTIME()->Write("TIME",TObject::kOverwrite);
6c4904c2 220 return 0;
4ed9ed00 221}