In This Package:

MergeStats.py

Go to the documentation of this file.

#!/usr/bin/env python

'''
usage example:

 nuwa.py -A -n 0 -m "StatisticSvc.MergeStats -i input1.root -i input2.root -o output.root -d name1,name2,name3"

 -i : input statistics file (as many as you want)
 -o : output statistics file (1)
 -d : Comma-separated list of names of objects that should not be added
        (First occurance is copied, the remaining are ignored)
  '''

# Load GaudiPython
from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl

from GaudiKernel.Constants import *

# Make your algorithm
class MergeAlg(GaudiAlgo):
    "Statistics Merging Algorithm"
    def __init__(self,name):
        GaudiAlgo.__init__(self,name)
        self.Input  = ['file1', 'file2']
        self.Output = 'file0'
        self.DontAdd = []
        return

    def initialize(self):
        status = GaudiAlgo.initialize(self)
        print "Init MergeAlg",self.name()
        if status.isFailure(): return status

        # Initialize services
        self.msgSvc = self.svc('IMessageSvc','MessageSvc')
        self.statsSvc = self.svc('IStatisticsSvc','StatisticsSvc')

        # Merge input paths to output path
        for path in self.Input:
            status = self.mergePaths('/'+path, '/'+self.Output )
            if status.isFailure(): break

        return status

    def execute(self):
        return SUCCESS
        
    def finalize(self):        
        print "Finalizing MergeAlg",self.name()
        status = GaudiAlgo.finalize(self)
        return status

    def mergePaths(self, inputPath, outputPath):
        hasInPath = self.statsSvc.exists(inputPath)
        if( not hasInPath ):
            self.error("Input path doesn't exist: "+inputPath)
            return FAILURE
        contents = self.statsSvc.getContents(inputPath)
        for name in contents:
            status = self.mergeObjects(inputPath + "/" + name,
                                       outputPath + "/" + name)
            if status.isFailure(): return status
        subfolders = self.statsSvc.getSubFolders(inputPath)
        for folder in subfolders:
            status = self.mergePaths(inputPath+'/'+folder,
                                     outputPath+'/'+folder)
            if status.isFailure(): return status
        return SUCCESS

    def mergeObjects(self, inputPath, outputPath):
        inputObject = self.statsSvc.get(inputPath)
        if( inputObject == None ):
            self.error("No item at "+inputPath)
            return FAILURE
        hasOutput = self.statsSvc.exists(outputPath)
        if( not hasOutput ):
            # Add item to output
            status = self.statsSvc.put(outputPath, inputObject)
            if status.isFailure(): return status
        else:
            # Combine objects if possible
            for name in self.DontAdd:
                if inputObject.GetName().find(name) > -1:
                    return SUCCESS
            outputObject = self.statsSvc.get(outputPath)
            if( outputObject == None ):
                self.error("No item at "+outputPath)
                return FAILURE
            if outputObject.IsA().InheritsFrom("TH1"):
                # Add histograms
                if( outputObject.TestBit( gbl.TH1.kIsAverage ) ):
                    # Average histograms must use TH1::Add
                    outputObject.Add( inputObject )
                else:
                    # Other histograms should use TH1::Merge
                    ilist = gbl.TList();
                    ilist.Add(inputObject)
                    outputObject.Merge(ilist)
            elif outputObject.ClassName().startswith("TParameter"):
                # Add parameters
                outputObject.SetVal( outputObject.GetVal()
                                     + inputObject.GetVal() )
            else:
                self.error("Can't add "+inputPath)
            
        return SUCCESS

    def error(self, message):
        self.msgSvc.reportMessage(self.name(), ERROR, message)
        
inputMap = {}
output = 'file0'
dontAdd = []

def configure(argv = []):
    global inputMap, output, dontAdd
    import sys,getopt

    inFiles = []
    outFile = "merge.root"
    
    opts, args = getopt.getopt(argv,"i:o:d:")
    print "opts = ",opts
    for opt, arg in opts:
        if opt == "-i":
            inFiles.append(arg)
        if opt == "-o":
            outFile = arg
        if opt == "-d":
            print arg.split(",")
            dontAdd = [name.strip() for name in arg.split(",")]

    print "inFiles: ",inFiles
    print "outFile: ",outFile
    print "dontAdd: ",dontAdd

    nPath = 1
    for filename in inFiles:
        inputMap["file%d" % nPath] = filename
        nPath += 1

    print "inputMap = ", inputMap
    from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    statssvc = StatisticsSvc()
    statssvc.Input = inputMap
    statssvc.Output = { output : outFile }

    return

def run(app):
    '''
    Configure and add an algorithm to job
    '''
    global inputMap, output, dontAdd
    app.ExtSvc += ["StatisticsSvc"]
    mergeAlg = MergeAlg("MyMerge")
    mergeAlg.Input = inputMap.keys()
    mergeAlg.Output = output
    mergeAlg.DontAdd = dontAdd
    
    app.addAlgorithm(mergeAlg)
    pass

---