RivetAnalysis_JO_MEfromFile.py

Go to the documentation of this file.

import os
 
theApp.EvtMax = -1
 
import AthenaPoolCnvSvc.ReadAthenaPool
svcMgr.EventSelector.InputCollections = ["!TESTSAMPLE!"]
from PyUtils.MetaReader import read_metadata
filename = svcMgr.EventSelector.InputCollections[0]
#Below are some example
systWeights=None
 
 
def safeFileName(name):
 name=name.strip()
 name= name.replace(".","p").replace(" ","_")
 name= name.replace("pyoda",".yoda")
 name= name.replace(":","_")
 return name
 
 
metadata = read_metadata(filename, None, 'full')[filename]
if '/Generation/Parameters' in metadata:
    genpars=metadata['/Generation/Parameters']
    if 'HepMCWeightNames' in genpars:
        systWeights=genpars['HepMCWeightNames']
    else:
        print('HepMCWeightName not found in /Generation/Parameters:')
        print(genpars)
else:
    print('/Generation/Parameters not found in metadata:')
    print(metadata)
 
 
from AthenaCommon.AlgSequence import AlgSequence
job = AlgSequence()
 
from Rivet_i.Rivet_iConf import Rivet_i
 
 
from AthenaCommon.AppMgr import ServiceMgr as svcMgr
from GaudiSvc.GaudiSvcConf import THistSvc
 
svcMgr += THistSvc()
analyses= "!RIVETANALYSIS!"
if systWeights==None: systWeights={'Nominal': 0}
for i in systWeights:
    systName=safeFileName(i)
 
    print('weight name:',i,', output name',systName)
 
    rivet = Rivet_i(systName)
    for analysis in analyses.split(","):
      rivet.Analyses +=[analysis]
    
    rivet.RunName = ""
    if i!="Nominal" : rivet.WeightName=i
    rivet.HistoFile = systName
    rivet.DoRootHistos = False
    rivet.AnalysisPath = os.environ['PWD']
    rivet.CrossSection = 1.0
 
    job += rivet