11 parser = optparse.OptionParser()
12 parser.add_option(
'-d',
'--data-type', dest =
'data_type',
13 action =
'store', type =
'string', default =
'data',
14 help =
'Type of data to run over. Valid options are data, mc, afii' )
15 parser.add_option(
'-s',
'--submission-dir', dest =
'submission_dir',
16 action =
'store', type =
'string', default =
'submitDir',
17 help =
'Submission directory for EventLoop' )
18 parser.add_option(
'-u',
'--unit-test', dest=
'unit_test',
19 action =
'store_true', default =
False,
20 help =
'Run the job in "unit test mode"' )
21 ( options, args ) = parser.parse_args()
30 dataType = options.data_type
32 inputfile = {
"data":
'ASG_TEST_FILE_DATA',
33 "mc":
'ASG_TEST_FILE_MC',
34 "afii":
'ASG_TEST_FILE_MC_AFII'}
36 jetContainer =
"AntiKt4EMTopoJets"
38 if dataType
not in [
"data",
"mc",
"afii"] :
39 raise ValueError (
"invalid data type: " + dataType)
44 sh = ROOT.SH.SampleHandler()
45 sh.setMetaString(
'nc_tree',
'CollectionTree' )
46 sample = ROOT.SH.SampleLocal (dataType)
47 sample.add (os.getenv (inputfile[dataType]))
53 job.sampleHandler( sh )
54 job.options().setDouble( ROOT.EL.Job.optMaxEvents, 500 )
56 from JetAnalysisAlgorithms.JetAnalysisAlgorithmsTest
import makeSequence
57 algSeq = makeSequence (dataType, jetContainer)
59 algSeq.addSelfToJob( job )
62 job.outputAdd( ROOT.EL.OutputStream(
'ANALYSIS' ) )
65 submitDir = options.submission_dir
69 submitDir = tempfile.mkdtemp( prefix =
'jetTest_'+dataType+
'_EMTopo_', dir = os.getcwd() )
74 driver = ROOT.EL.DirectDriver()
75 driver.submit( job, submitDir )