Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
makeARTScripts.py
Go to the documentation of this file.
1 import os
2 
3 makeDataDAODs=True
4 makeMCDAODs=True
5 makeTruthDAODs=True
6 makeTrains=True
7 
8 formatList = ["PHYSVAL","PHYS","PHYSLITE",
9  "LLP1","HIGG1D1","HIGG1D2",
10  "JETM1","JETM2","JETM3","JETM4","JETM5","JETM6","JETM10","JETM11","JETM12","JETM14",
11  "IDTR2",
12  "EGAM1","EGAM2","EGAM3","EGAM4","EGAM5","EGAM7","EGAM8","EGAM9","EGAM10",
13  "FTAG1","FTAG2","FTAG3",
14  "BPHY1","BPHY2","BPHY3","BPHY4","BPHY5","BPHY6","BPHY10","BPHY12","BPHY15","BPHY16","BPHY18","BPHY21","BPHY22",
15  "BPHY23","BPHY24",
16  "STDM7","STDM13",
17  "TRIG8","TRIG9",
18  "MUON1"
19 ]
20 
21 truthFormatList = ["TRUTH0", "TRUTH1", "TRUTH3"]
22 
23 trainList = [
24  ["EGAM1","EGAM2","EGAM3","EGAM4","EGAM5","EGAM7","EGAM8","EGAM9","EGAM10","JETM1","JETM3","JETM4","JETM6","FTAG1","FTAG2","FTAG3","IDTR2","TRIG8","TRIG9","LLP1","STDM7","STDM13","HIGG1D1","MUON1"]
25 ]
26 
27 
28 # Files
29 com_dir = "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/CampaignInputs/"
30 mc20File = com_dir+"mc20/AOD/mc20_13TeV.410470.PhPy8EG_A14_ttbar_hdamp258p75_nonallhad.recon.AOD.e6337_s3681_r13145/1000events.AOD.27121237._002005.pool.root.1"
31 mc23File = com_dir+"mc23/AOD/mc23_13p6TeV.601229.PhPy8EG_A14_ttbar_hdamp258p75_SingleLep.recon.AOD.e8514_s4162_r14622/1000events.AOD.33799166._000073.pool.root.1"
32 truthFile = com_dir+"mc23/EVNT/mc23_13p6TeV.601229.PhPy8EG_A14_ttbar_hdamp258p75_SingleLep.evgen.EVNT.e8514/EVNT.32288062._002040.pool.root.1"
33 data18File = com_dir+"data18/AOD/data18_13TeV.00357772.physics_Main.merge.AOD.r13286_p4910/1000events.AOD.27655096._000455.pool.root.1"
34 data22File = com_dir+"data22/AOD/data22_13p6TeV.00431906.physics_Main.merge.AOD.r13928_p5279/1000events.AOD.30220215._001367.pool.root.1"
35 data23File = com_dir+"data23/AOD/data23_13p6TeV.00453713.physics_Main.recon.AOD.f1357/2012events.data23_13p6TeV.00453713.physics_Main.recon.AOD.f1357._lb1416._0006.1"
36 data23CosFile = com_dir+"data23_cos/AOD/data23_cos.00459152.physics_CosmicMuons.merge.AOD.f1383_m2195/data23_cos.00459152.physics_CosmicMuons.merge.AOD.f1383_m2195._lb0124-lb0126._0001.1"
37 
38 
39 def generateText(formatName,label,inputFile,isTruth,isMC,nEvents):
40  add_str = ""
41  outputFileName = "test_"+label+formatName+add_str+".sh"
42  outputFile = open(outputFileName,"w")
43  outputFile.write("#!/bin/sh"+"\n")
44  outputFile.write("\n")
45  outputFile.write("# art-include: main/Athena"+"\n")
46  if (formatName.find("EGAM")!=-1 or formatName.find("JETM")!=-1 or formatName.find("FTAG")!=-1 or formatName.find("IDTR")!=-1 or formatName.find("TRIG")!=-1 or (formatName.find("PHYS")!=-1 and formatName.find("PHYSLITE")==-1)):
47  outputFile.write("# art-include: main/Athena"+"\n")
48  outputFile.write("# art-description: DAOD building "+formatName+" "+label+"\n")
49  outputFile.write("# art-type: grid"+"\n")
50  outputFile.write("# art-output: *.pool.root"+"\n")
51  outputFile.write("# art-output: checkFile*.txt"+"\n")
52  outputFile.write("# art-output: checkxAOD*.txt"+"\n")
53  outputFile.write("# art-output: checkIndexRefs*.txt"+"\n")
54  outputFile.write("\n")
55  outputFile.write("set -e"+"\n")
56  outputFile.write("\n")
57  if (not isTruth):
58  outputFile.write("Derivation_tf.py \\\n")
59  outputFile.write("--inputAODFile "+inputFile+" \\\n")
60  outputFile.write("--outputDAODFile art.pool.root \\\n")
61  outputFile.write("--formats "+formatName+" \\\n")
62  outputFile.write("--maxEvents "+nEvents+" \\\n")
63  if isTruth:
64  outputFile.write("Derivation_tf.py \\\n")
65  outputFile.write("--inputEVNTFile "+inputFile+" \\\n")
66  outputFile.write("--outputDAODFile art.pool.root \\\n")
67  outputFile.write("--formats "+formatName+" \\\n")
68  outputFile.write("--maxEvents "+nEvents+"\n")
69  outputFile.write("\n")
70  outputFile.write("echo \"art-result: $? reco\""+"\n")
71  outputFile.write("\n")
72  outputFile.write("checkFile.py DAOD_"+formatName+".art.pool.root > checkFile_"+formatName+".txt"+"\n")
73  outputFile.write("\n")
74  outputFile.write("echo \"art-result: $? checkfile\""+'\n')
75  outputFile.write("\n")
76  outputFile.write("checkxAOD.py DAOD_"+formatName+".art.pool.root > checkxAOD_"+formatName+".txt"+"\n")
77  outputFile.write("\n")
78  outputFile.write("echo \"art-result: $? checkxAOD\""+'\n')
79  outputFile.write("\n")
80  outputFile.write("checkIndexRefs.py DAOD_"+formatName+".art.pool.root > checkIndexRefs_"+formatName+".txt 2>&1"+"\n")
81  outputFile.write("\n")
82  outputFile.write("echo \"art-result: $? checkIndexRefs\""+'\n')
83  outputFile.close()
84  os.system("chmod +x "+outputFileName)
85 
86 def generateTrains(formatList,label,inputFile,isMC,nEvents):
87  add_str = ""
88  outputFileName = "test_"+label+"_".join(formatList)+add_str+".sh"
89  outputFile = open(outputFileName,"w")
90  outputFile.write("#!/bin/sh"+"\n")
91  outputFile.write("\n")
92  outputFile.write("# art-include: main/Athena"+"\n")
93  outputFile.write("# art-description: DAOD building "+" ".join(formatList)+" "+label+"\n")
94  outputFile.write("# art-type: grid"+"\n")
95  outputFile.write("# art-output: *.pool.root"+"\n")
96  outputFile.write("# art-output: checkFile*.txt"+"\n")
97  outputFile.write("# art-output: checkxAOD*.txt"+"\n")
98  outputFile.write("# art-output: checkIndexRefs*.txt"+"\n")
99  outputFile.write("\n")
100  outputFile.write("set -e"+"\n")
101  outputFile.write("\n")
102  outputFile.write("Derivation_tf.py \\\n")
103  outputFile.write("--inputAODFile "+inputFile+" \\\n")
104  outputFile.write("--outputDAODFile art.pool.root \\\n")
105  outputFile.write("--formats "+" ".join(formatList)+" \\\n")
106  outputFile.write("--maxEvents "+nEvents+" \\\n")
107  outputFile.write("\n")
108  outputFile.write("echo \"art-result: $? reco\""+"\n")
109  for formatname in formatList:
110  outputFile.write("\n")
111  outputFile.write("checkFile.py DAOD_"+formatname+".art.pool.root > checkFile_"+formatname+".txt"+"\n")
112  outputFile.write("\n")
113  outputFile.write("echo \"art-result: $? checkfile\""+'\n')
114  outputFile.write("\n")
115  outputFile.write("checkxAOD.py DAOD_"+formatname+".art.pool.root > checkxAOD_"+formatname+".txt"+"\n")
116  outputFile.write("\n")
117  outputFile.write("echo \"art-result: $? checkxAOD\""+'\n')
118  outputFile.write("\n")
119  outputFile.write("checkIndexRefs.py DAOD_"+formatname+".art.pool.root > checkIndexRefs_"+formatname+".txt 2>&1"+"\n")
120  outputFile.write("\n")
121  outputFile.write("echo \"art-result: $? checkIndexRefs\""+'\n')
122  outputFile.close()
123  os.system("chmod +x "+outputFileName)
124 
125 if (makeDataDAODs or makeMCDAODs):
126  for formatName in formatList:
127  if makeDataDAODs:
128  generateText(formatName,"data18",data18File,False,False,"-1")
129  generateText(formatName,"data22",data22File,False,False,"-1")
130  generateText(formatName,"data23",data23File,False,False,"-1")
131  if makeMCDAODs:
132  generateText(formatName,"mc20",mc20File,False,True,"-1")
133  generateText(formatName,"mc23",mc23File,False,True,"-1")
134  generateText("NCB1","data23cos",data23CosFile,False,False,"-1")
135 if makeTruthDAODs:
136  for formatName in truthFormatList:
137  generateText(formatName,"mc23",truthFile,True,False,"1000")
138 
139 if makeTrains:
140  for train in trainList:
141  if makeDataDAODs:
142  generateTrains(train,"data18",data18File,False,"-1")
143  generateTrains(train,"data22",data22File,False,"-1")
144  generateTrains(train,"data23",data23File,False,"-1")
145  if makeMCDAODs:
146  generateTrains(train,"mc20",mc20File,True,"-1")
147  generateTrains(train,"mc23",mc23File,True,"-1")
makeARTScripts.generateTrains
def generateTrains(formatList, label, inputFile, isMC, nEvents)
Definition: makeARTScripts.py:86
TCS::join
std::string join(const std::vector< std::string > &v, const char c=',')
Definition: Trigger/TrigT1/L1Topo/L1TopoCommon/Root/StringUtils.cxx:10
makeARTScripts.generateText
def generateText(formatName, label, inputFile, isTruth, isMC, nEvents)
Definition: makeARTScripts.py:39
Trk::open
@ open
Definition: BinningType.h:40
Generated on Sat May 10 2025 21:13:25 for ATLAS Offline Software by   doxygen 1.8.18