NtupleWriter14/run/AANtupleWriter14.py

##############################
#
# Author: MDT group, MPI Munich
#
#    run with: athena  -c 'jetcontainer = "Cone4H1Topo" ; dataset = "5333" ; outputdir =
#    "/.at/mdtraid06/mgroh"' AANtupleWriter14.py
#
#    for a more detailed description please see at the end of this file
#
##############################

# if values are not given as an argument
# run with default values:
#################################################################
# T.E.: this stuff is new for release 14 (mainly trigger)
#       I don't have a clue if this is correct
##########################################
## Set up trigger configuration service and metadata service is relies on, for analysis job without RecExCommon

from AthenaCommon.GlobalFlags import GlobalFlags
GlobalFlags.DetGeo.set_atlas()

## set up trigger decision tool
from TrigDecision.TrigDecisionConfig import TrigDecisionTool
tdt = TrigDecisionTool()
ToolSvc += tdt

# T.E.: seems like we don't need this:
# might be needed for TriggerConfigGetter...
from RecExConfig.RecFlags  import rec
rec.readAOD=True
rec.doWriteAOD=False
rec.doWriteESD=False

# To read files with trigger config stored as in-file meta-data,
# i.e. 14.0.40 and above: ds
# To read AOD produced with 14.0.30 you need to change ds to aod:
from TriggerJobOpts.TriggerFlags import TriggerFlags
TriggerFlags.configurationSourceList = ['ds']

# set up trigger config service
from TriggerJobOpts.TriggerConfigGetter import TriggerConfigGetter
cfg =  TriggerConfigGetter()

LoadATLASGeometry=True
DetDescrVersion="ATLAS-CSC-02-01-00"
# T.E.: end
###############################################################


sys.path.append('data')
import datatools
import datetime

#check command line and set default values
if not 'dataset' in dir():
   dataset = "default"

if not 'outputdir' in dir():
   outputdir = ""

if not 'AtlFastFlag' in dir():
   AtlFastFlag = False;
   if (dataset.lower().find('atlfast') != -1):
      AtlFastFlag = True;

if not 'doAODFix1mm' in dir():
   doAODFix1mm = False;

if not 'doEventWeight' in dir():
   doEventWeight = True;
   if (dataset.lower().find('sherpa') != -1):
      doEventWeight = False;

if not 'filenumber' in dir() or filenumber < 0:
   filenumber = 0;
   
if not 'jetcontainer' in dir():
   jetcontainer = "Cone4H1Topo"

if not 'nevents' in dir():
   nevents = -1

if not 'doTrackJets' in dir() or AtlFastFlag == True:
   doTrackJets = False

if not 'doTrackParticles' in dir() or AtlFastFlag == True:
   doTrackParticles = False

# put the names of the jetcontainers together
jetcontainername =  jetcontainer + "Jets"
namelist = ["H1Topo", "UnCalTopo", "Topo", "Tower"]
truthjetcontainername = ""

for k_string in namelist:
   
   if k_string in jetcontainername:
      
      truthjetcontainername = jetcontainername.replace(k_string,"Truth")
      break


# import the data types 
import EventKernel.ParticleDataType

#EventSelector = Service( "EventSelector" )

# load additional libraries for back navigation
from AnalysisExamples.AnalysisFlags import AnalysisFlags

include( "AthenaPoolCnvSvc/ReadAthenaPool_jobOptions.py" )
if AnalysisFlags.DoNavigation:
   include ("RecExCommon/AllDet_detDescr.py")
   ServiceMgr.EventSelector.BackNavigation = True

# the Dlls
include( "ParticleBuilderOptions/ESD_PoolCnv_jobOptions.py" )
include( "ParticleBuilderOptions/AOD_PoolCnv_jobOptions.py")
include( "ParticleBuilderOptions/McAOD_PoolCnv_jobOptions.py")
include( "EventAthenaPool/EventAthenaPool_joboptions.py" )
include( "TrackIsolationTools/TrackIsolationTool_jobOptions.py" )


if(doTrackJets):
   # configure VertexCollectionSvc
   theApp.ExtSvc += [ "VertexCollectionSvc" ]
   VertexCollectionSvc = Service( "VertexCollectionSvc" )
   theApp.TopAlg += [ "MPIHiggsAnalysis::VertexCollector" ]
   VertexCollector = Algorithm( "MPIHiggsAnalysis::VertexCollector" )
   #VertexCollector.VertexCollectionSvc = "MPIHiggsAnalysis::VertexCollector"

   include( "Cone4TrackJet_jobOptions.py" )
   include( "Cone4TruthTrackJet_jobOptions.py" )


HistogramPersistencySvc = Service( "HistogramPersistencySvc" )
theApp.Dlls += [ "RootHistCnv" ]
theApp.HistogramPersistency = "ROOT"
#HistogramPersistencySvc.OutputFile = "histos.root"


# write out a summary of the time spent
theApp.Dlls += [ "GaudiAud" ]
#theAuditorSvc = AuditorSvc()
theAuditorSvc.Auditors  += [ "ChronoAuditor", "MemStatAuditor"]

# User Analysis Dll's
theApp.Dlls   += [ "CBNT_Athena", "CBNT_Utils" ]
theApp.Dlls   += [ "AnalysisTools" ]
theApp.Dlls   += [ "AnalysisExamples" ]

# The algorithms to run
#theApp.TopAlg += [ "CBNT_Athena/AOD_ANA" ]
#AOD_ANA = Algorithm( "AOD_ANA" )

#init the dataset library 
data_lib = datatools.DataLib()


#check if joboption from default path should be included 
jo_path = ''

if dataset.find('jo:') != -1:
   dataset = dataset.replace('jo:','');
   jo_path = '/.at/mdtraid04/12.0.X_data/data_jo_files/'
   print "Include data from .py file in path: ", jo_path 


#fill the dataset library
include("data/datasetdef.py")
include("data/datasetdef_manfred.py")
include("data/datasetdef_steffen.py")
include("data/datasetdef_thies.py")
include("data/datasetdef_joerg.py")

#print data_lib

#get the dataset
if dataset == "default":
   datasetname = 'defaultfile' #will be used for the filename
   print '\nError: You have to pass a dataset through the command line'
   print '       or define a default dataset here.\n'
   sys.exit()
  
elif dataset.endswith(".py"):
   include(jo_path + dataset)
   tuple = dataset.rsplit("/",1)
   datasetname = tuple[len(tuple)-1].replace(".py","")

elif dataset.endswith(".root"):
   ServiceMgr.EventSelector.InputCollections = [ dataset ]
   tuple = dataset.rsplit("/",1)
   datasetname = tuple[len(tuple)-1].replace(".root","")

else:
   ds = data_lib.find_dataset(dataset)
   print ds
   if jo_path == '':
      datatools.include_files(ds.get_path(),ServiceMgr.EventSelector)
   else:
      include(jo_path + ds.get_long() + '.py')
   datasetname = dataset


if len(outputdir)>0 and not outputdir.endswith("/"):
   outputdir += "/"


#put together filename
filename = outputdir + datasetname + "_" + jetcontainername + ".aan.root"

if filenumber > 0:
   filename = filename.replace(jetcontainername,
                               jetcontainername+"_"+str(filenumber))

if (AtlFastFlag == True) and\
       (datasetname.lower().find('atlfast') == -1):

   filename = filename.replace("_"+jetcontainername,
                               "_atlfast_"+jetcontainername)

if doAODFix1mm == True:
   filename = filename.replace(jetcontainername,
                               jetcontainername+"_1mmFix")

if doTrackJets == True:
   filename = filename.replace(jetcontainername,
                               jetcontainername+"_TrackJets")

#print configuration
print '\n************************************************************\n'
print " Using the following configuration: "
print "   dataset:           ", dataset
print "   datasetname:       ", datasetname
print "   outputdir:         ", outputdir
print "   AtlFastFlag:       ", AtlFastFlag
print "   doAODFix1mm:       ", doAODFix1mm
print "   doEventWeight:     ", doEventWeight
print "   doTrackJets:       ", doTrackJets
print "   filenumber:        ", filenumber
print "   nevents:           ", nevents
print "   filename:          ", filename
print "   jetcontainer:      ", jetcontainername  
print "   truthjetcontainer: ", truthjetcontainername
print '\n************************************************************\n'

#do 1mm correction
#include( "ElectronPhotonAlgs/ElectronPhotonCorrector_jobOptions.py")
#include( "JetMissingEtAlgs/ParticleJet1mmCorrection_jobOptions.py")
#include( "JetMissingEtAlgs/MissingET1mmCorrection_jobOptions.py")


theApp.Dlls   += [ "AnalysisTools" ]
#theApp.Dlls   += [ "NtupleWriter14" ]
CBNTAthenaAware = True
include ("CBNT_Athena/CBNT_AthenaAware_jobOptions.py")
include ("CBNT_Athena/CBNT_EventInfo_jobOptions.py")
#CBNT_AthenaAware = Algorithm( "CBNT_AthenaAware" )
#CBNT_AthenaAware.Members += [ "AANtupleWriter14" ]
from NtupleWriter14.NtupleWriter14Conf import AANtupleWriter14
CBNT_AthenaAware += AANtupleWriter14() 


############# The properties of the AANtupleWriter14 Algorithm

#AANtupleWriter14 = Algorithm( "AANtupleWriter14" )

CBNT_AthenaAware.AANtupleWriter14.doTrigger        = True
CBNT_AthenaAware.AANtupleWriter14.doMissingEt      = True
CBNT_AthenaAware.AANtupleWriter14.doElectrons      = True
CBNT_AthenaAware.AANtupleWriter14.doPhotons        = True
CBNT_AthenaAware.AANtupleWriter14.doMuons          = True
CBNT_AthenaAware.AANtupleWriter14.doJets           = True
CBNT_AthenaAware.AANtupleWriter14.doTauJets        = True
CBNT_AthenaAware.AANtupleWriter14.doTruthJets      = True
CBNT_AthenaAware.AANtupleWriter14.doTruthParticles = True
CBNT_AthenaAware.AANtupleWriter14.doEventWeight    = doEventWeight

CBNT_AthenaAware.AANtupleWriter14.doTrackJets      = doTrackJets
CBNT_AthenaAware.AANtupleWriter14.doVertices       = doTrackJets

CBNT_AthenaAware.AANtupleWriter14.truthparticleMaximumLeptonBarcode = 100000
CBNT_AthenaAware.AANtupleWriter14.truthparticleNoPDGAbove100        = False

CBNT_AthenaAware.AANtupleWriter14.AtlFastFlag = AtlFastFlag
CBNT_AthenaAware.AANtupleWriter14.doAODFix1mm = doAODFix1mm


CBNT_AthenaAware.AANtupleWriter14.AOD_files = ServiceMgr.EventSelector.InputCollections


if AtlFastFlag == True:
    
   CBNT_AthenaAware.AANtupleWriter14.triggerDecisionContainer     = "notAvailable"
   CBNT_AthenaAware.AANtupleWriter14.photonContainer              = "AtlfastPhotonCollection"
   CBNT_AthenaAware.AANtupleWriter14.electronContainer            = "AtlfastElectronCollection"
   CBNT_AthenaAware.AANtupleWriter14.muonContainer                = ["AtlfastMuonCollection"]
   CBNT_AthenaAware.AANtupleWriter14.jetContainer                 = "AtlfastParticleJetContainer"
   CBNT_AthenaAware.AANtupleWriter14.taujetContainer              = "AtlfastTauJetContainer"
   CBNT_AthenaAware.AANtupleWriter14.missingEtContainerCollection = ["AtlfastMissingEt"]
   CBNT_AthenaAware.AANtupleWriter14.truthparticleContainer       = "SpclMC"
   CBNT_AthenaAware.AANtupleWriter14.truthjetContainer            = "notAvailable"
   CBNT_AthenaAware.AANtupleWriter14.truthmissingEtContainer      = "notAvailable"
   CBNT_AthenaAware.AANtupleWriter14.TrackParticleContainer       = "AtlfastTrackParticles"

   
else:

   CBNT_AthenaAware.AANtupleWriter14.triggerDecisionContainer     = "TrigDecision"
   CBNT_AthenaAware.AANtupleWriter14.triggerMuonLVL1Container     = "LVL1_ROI"
   CBNT_AthenaAware.AANtupleWriter14.triggerMuonLVL2Container     = "CombinedMuonFeature"
   CBNT_AthenaAware.AANtupleWriter14.triggerMuonEFContainer       = "TrigMuonEFContainer"
   CBNT_AthenaAware.AANtupleWriter14.triggerMissingEtContainer    = "TrigEFMissingET"
   CBNT_AthenaAware.AANtupleWriter14.photonContainer              = "PhotonAODCollection"
   CBNT_AthenaAware.AANtupleWriter14.electronContainer            = "ElectronAODCollection"
   CBNT_AthenaAware.AANtupleWriter14.muonContainer                = ["MuidMuonCollection",
                                                                     "StacoMuonCollection"]
   CBNT_AthenaAware.AANtupleWriter14.jetContainer                 = jetcontainername
   #CBNT_AthenaAware.AANtupleWriter14.jetContainer                 = "Cone4H1TopoParticleJets"
   CBNT_AthenaAware.AANtupleWriter14.taujetContainer              = "TauRecContainer"
   CBNT_AthenaAware.AANtupleWriter14.truthjetContainer            = truthjetcontainername
   #CBNT_AthenaAware.AANtupleWriter14.truthjetContainer           = "Cone4TruthParticleJets"
   CBNT_AthenaAware.AANtupleWriter14.missingEtContainerCollection = ["MET_Base",
                                                                              "MET_Calib",
                                                                              "MET_CorrTopo",
                                                                              "MET_LocHadTopo",
                                                                              "MET_Topo",
                                                                              "MET_TopoObj",
                                                                              "MET_MuonBoy",
                                                                              "MET_Cryo",
                                                                              "MET_CryoCone",
                                                                              "MET_RefFinal",
                                                                              "MET_RefEle",
                                                                              "MET_RefJet",
                                                                              "MET_CellOut",
                                                                              "MET_CorrTopo",
                                                                              "MET_LocHadTopo",
                                                                              #"LocHadTopoObj",
                                                                              #"TopoObj",
                                                                              "MET_Topo",
                                                                              "ObjMET_Elec",
                                                                              "ObjMET_Muon",
                                                                              "ObjMET_Jet",
                                                                              "ObjMET_MiniJet",
                                                                              "ObjMET_IdTrk",
                                                                              "ObjMET_Rest", 
                                                                              "ObjMET_Final"]
   CBNT_AthenaAware.AANtupleWriter14.truthmissingEtContainer      = "MET_Truth"
   CBNT_AthenaAware.AANtupleWriter14.truthparticleContainer       = "SpclMC"
   CBNT_AthenaAware.AANtupleWriter14.jetTrackContainer            = "Cone4TrackJets"
   CBNT_AthenaAware.AANtupleWriter14.jetTruthTrackContainer       = "Cone4TruthTrackJets"
   CBNT_AthenaAware.AANtupleWriter14.TrackParticleContainer       = "TrackParticleCandidate"

# set the file info
CBNT_AthenaAware.AANtupleWriter14.release    = "14.2.0"
CBNT_AthenaAware.AANtupleWriter14.svnversion = os.popen("svnversion -n .").read()
CBNT_AthenaAware.AANtupleWriter14.date       = datetime.datetime.now().strftime("%Y-%m-%d/%H:%M:%S")

##########################################
# setup TTree registration Service
# save ROOT histograms and Tuple
from GaudiSvc.GaudiSvcConf import THistSvc
ServiceMgr += THistSvc()
ServiceMgr.THistSvc.Output = [ "AANT DATAFILE='"+filename+"' OPT='RECREATE'" ]
from AnalysisTools.AnalysisToolsConf import AANTupleStream
topSequence += AANTupleStream()
AANTupleStream = AANTupleStream()
AANTupleStream.ExtraRefNames = [ "StreamATLFAST","Stream1" ]
AANTupleStream.OutputName = filename
AANTupleStream.WriteInputDataHeader = True
AANTupleStream.OutputLevel = WARNING

# Set output level threshold (2=DEBUG, 3=INFO, 4=WARNING, 5=ERROR, 6=FATAL )
MessageSvc = Service( "MessageSvc" )
MessageSvc.OutputLevel = 3
#MessageSvc.OutputLevel = WARNING
#MessageSvc.OutputLevel = INFO

# Number of Events to process
theApp.EvtMax = nevents

# skip events
if(filenumber > 0):
   ServiceMgr.EventSelector.SkipEvents = (filenumber-1) * theApp.EvtMax
   print 'Skip:', ServiceMgr.EventSelector.SkipEvents, 'Events'


###############################################################
#
#  1) Default:  athena AANtupleWriter14.py
#      - default values will be used (defined at the beginning of this file)
#      - you have to put 'include("<your_dataset_jo>.py")' in the 'if dataset == "default"' block 
#
#  2) dataset: athena -c 'dataset="description"' AANtupleWriter14.py
#      a) dataset="path/filename.py": joboptions will be directly included
#      b) dataset="jo:filename.py":   joboptions will be icluded from path: /.at/mdtraid04/12.0.X_data/data_jo_files/
#      c) dataset="filename.root"     the root file will directly be included
#      d) dataset="long/short descr":
#                   - e.g.:  dataset="5200_1206"
#                   - the datset will be searched in the data_lib library
#                   - both the short or long description can be specified, as defined in
#                     data/datasetdef.py
#                   - all files which contain '.root' in the name in the path defined
#                     in the dataset definition will be included
#      e) dataset="jo:5200_1206"                             
#                   - same as d) but instead of looping over all rootfiles the
#                     joboptions from the path: /.at/mdtraid04/12.0.X_data/data_jo_files/ is included
#                   - the name of the joboption is build as: <long descripion of dataset>.py
#    
#  3) outpudir: athena -c 'outputdir="mydata/"' AANtupleWriter14.py
#      - the path where the file will be written
# 
#  4) AtlFastFlag: athena -c 'AtlFastFlag=1' AANtupleWriter114py
#      - specifies if writer will be run on AtlFast containers
#      - default value is False
#      - if dataset is defined in command line an contains a string "atlfast"
#        -> flag will be set to True
#      - this will be overwritten if AtlFastFlag is set in the command line
#      - if true: ouptfilename will contain a string: '_atlfast' 
#
#  5) doEventWeight: athena -c 'doEventWeight=0' AANtupleWriter14.py 
#      - write EventWeight or not
#      - default value is True
#      - if dataset is defined in command line an contains a string "sherpa"
#        ->  flag will be set to False
#      - this will be overwritten if doEventWeight is set in the command line
#
#  6) doAODFix1mm: athena -c 'doAODFix1mm=1' AANtupleWriter14.py
#      - default: false
#      - if true 1mm bug corrections will be applied     
#      - will be included in the outputfilename
# 
#
#  7) filenumber: athena -c 'filenumber=3' AANtupleWriter14.py
#      - default value is 0
#      - if > 0: - (filenumber-1)*nevent will be skipped the nevents ar processed
#                - filenumber will be included in the outputfilename
#
#  8) jetcontainer: athena -c 'jetcontainer="Cone4H1Tower"' AANtupleWriter14.py
#     - default: "Cone4H1Topo"
#     - use this jetcontainer in analysis
#     - the appropriate truthjet container will be used
#     - will be included in the outputfilename  
#
#  9) nevents: athena -c 'nevents="100"' AANtupleWriter14.py
#     - the number of events which will be processed
#
# 10) Compose:
#     athena  -c 'dataset="5189_14_2_0"; nevents=100' AANtupleWriter14.py
#
#
#################################################################

---