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'}
25 import AthenaPoolCnvSvc.ReadAthenaPool
27 testFile = os.getenv ( inputfile[dataType] )
28 svcMgr.EventSelector.InputCollections = [testFile]
30 from MuonAnalysisAlgorithms.MuonAnalysisConfigTest
import makeSequence
31 algSeq = makeSequence (dataType)
38 ServiceMgr += CfgMgr.THistSvc()
39 ServiceMgr.THistSvc.Output += [
40 "ANALYSIS DATAFILE='MuonAnalysisAlgorithmsConfigTest." + dataType +
".hist.root' OPT='RECREATE'"
44 include(
"AthAnalysisBaseComps/SuppressLogging.py" )