In This Package:

__init__.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Check memory usage as a function of execute number
# He Miao, Jun. 2009
#
# How to use:
#   nuwa.py MDC09a.chkIBD.AdPerformance MemoryCheck $FILELIST

__all__ = ['checkMemory']
from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
import ROOT
import os
import commands

class checkMemory(GaudiAlgo):
          
    def __init__(self,name):
        GaudiAlgo.__init__(self,name)

        return

    def initialize(self):
        print "Initializing Check Memory", self.name()
        status = GaudiAlgo.initialize(self)
        if status.isFailure(): 
            return status

        self.execNumber = 0
        self.virt = {}
        self.res = {}
        self.virtMax = 0
        self.virtMin = 999
        self.resMax = 0
        self.resMin = 999

        return SUCCESS

    def execute(self):
        if self.execNumber % 1 == 0:
            print "------- Check Memory, excute: ", self.execNumber

        # Use system command to record memory use
        ps = 'ps uh ' + str(os.getpid())
        line = commands.getoutput(ps)
        word = line.split()

        self.virt[self.execNumber] = float(word[4])/(1024.*1024.)
        self.res[self.execNumber] = float(word[5])/(1024.*1024.)

        # Find maximum and minimum memory use
        if self.virtMax < self.virt[self.execNumber]:
            self.virtMax = self.virt[self.execNumber]
        if self.virtMin > self.virt[self.execNumber]:
            self.virtMin = self.virt[self.execNumber]

        if self.resMax < self.res[self.execNumber]:
            self.resMax = self.res[self.execNumber]
        if self.resMin > self.res[self.execNumber]:
            self.resMin = self.res[self.execNumber]

        self.execNumber = self.execNumber + 1

        return SUCCESS

    def finalize(self):

        self.histSvc = self.svc('ITHistSvc', 'THistSvc')
        self.hist = {}

        # Define histogram for virtual and resident memory
        self.hist["virt"] = gbl.TH2F("virt", "Virtual Memory",
            self.execNumber + 1, 0., self.execNumber, 
            100, self.virtMin*0.95, self.virtMax*1.05)
        status = self.histSvc.regHist('/fileMemory/virt', 
            self.hist["virt"])
        if status.isFailure(): return status

        self.hist["res"] = gbl.TH2F("res", "Resident Memory",
            self.execNumber + 1, 0., self.execNumber, 
            100, self.resMin*0.95, self.resMax*1.05)
        status = self.histSvc.regHist('/fileMemory/res', 
            self.hist["res"])
        if status.isFailure(): return status

        # Fill histogram
        for i in self.virt:
            self.hist["virt"].Fill(float(i),self.virt[i])
        for i in self.res:
            self.hist["res"].Fill(float(i),self.res[i])

        print "Finalizing ", self.name()
        print "Max resident memory is %s GByte" % self.resMax
        print "Max virtual memory is %s GByte" % self.virtMax
        status = GaudiAlgo.finalize(self)
        return status

def configure(argv=[]):

    import sys, getopt
    opts,args = getopt.getopt(argv,"o:")
    outFile = "memory.root"
    for opt,arg in opts:
        if opt == "-o":
            outFile = arg

    from GaudiSvc.GaudiSvcConf import THistSvc

    histsvc = THistSvc()
    histsvc.Output += ["fileMemory DATAFILE='" + outFile + "' OPT='RECREATE' TYP='ROOT' "]

    return

def run(app):
    '''Configure and add this algorithm to job'''

    app.ExtSvc += ["THistSvc"]

    chMemory = checkMemory("myCheckMemory")
    app.addAlgorithm(chMemory)
    pass

---