7 from AthenaCommon.AthArgumentParser
import AthArgumentParser
8 athArgsParser = AthArgumentParser()
9 athArgsParser.add_argument(
"--data-type", action =
"store", dest =
"data_type",
11 help =
"Type of input to run over. Valid options are 'data', 'mc', 'afii'")
12 athArgs = athArgsParser.parse_args()
14 dataType = athArgs.data_type
15 if not dataType
in [
"data",
"mc",
"afii"] :
16 raise Exception (
"invalid data type: " + dataType)
18 print(
"Running on data type: " + dataType)
20 inputfile = {
"data":
'ASG_TEST_FILE_DATA',
21 "mc":
'ASG_TEST_FILE_MC',
22 "afii":
'ASG_TEST_FILE_MC_AFII'}
24 jetContainer =
"AntiKt4EMTopoJets"
27 import AthenaPoolCnvSvc.ReadAthenaPool
29 testFile = os.getenv ( inputfile[dataType] )
30 svcMgr.EventSelector.InputCollections = [testFile]
32 from JetAnalysisAlgorithms.JetAnalysisAlgorithmsTest
import makeSequence
33 algSeq = makeSequence (dataType, jetContainer)
40 ServiceMgr += CfgMgr.THistSvc()
41 ServiceMgr.THistSvc.Output += [
42 "ANALYSIS DATAFILE='JetAnalysisAlgorithmsTestEMTopo." + dataType +
".hist.root' OPT='RECREATE'"
46 include(
"AthAnalysisBaseComps/SuppressLogging.py" )