ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/dasu/UltraFastSim/generateData.cc
(Generate patch)

Comparing UserCode/dasu/UltraFastSim/generateData.cc (file contents):
Revision 1.12 by mmhl, Thu Feb 24 20:17:12 2011 UTC vs.
Revision 1.20 by dasu, Tue Jul 31 05:58:42 2012 UTC

# Line 13 | Line 13 | using namespace std;
13   #include "Pythia.h"
14   using namespace Pythia8;
15  
16 – #include "TROOT.h"
17 – #include "TFile.h"
18 –
16   #include "UltraFastSim.h"
17 < #include "bbHAnalysis.h"
18 < //#include "ZHAnalysis.h"
19 < #include "bbHAnalysis.cc"
20 < //#include "ZHAnalysis.cc"
17 > #include "UFSDataStore.h"
18 >
19 > int poisson(double mean, Rndm *rndmPtr) {
20 >  static double oldMean = -1;
21 >  static double g;
22 >  if(mean != oldMean) {
23 >    oldMean = mean;
24 >    if(mean == 0) {
25 >      g = 0;
26 >    }
27 >    else {
28 >      g = exp(-mean);
29 >    }
30 >  }    
31 >  double em = -1;
32 >  double t = 1;
33 >  do {
34 >    em++;
35 >    t *= rndmPtr->flat();
36 >  } while(t > g);
37 >  return em;
38 > }
39  
40   int main(int argc, char **argv) {
41    // Generator. Process selection. LHC initialization. Histogram.
# Line 28 | Line 43 | int main(int argc, char **argv) {
43    int nEvents = 0;
44    int runNumber = 0;
45    int meanPileupEventCount = 0;
46 <  if(argc < 3 || argc > 5)
46 >  float cmEnergy = 14000.;
47 >  bool cutOnMET = false;
48 >  if(argc < 3 || argc > 6)
49      {
50 <      cerr << "Command syntax: " << argv[0] << " ZHmmbb[...]" << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
51 <      cerr << "or              " << argv[0] << " ZHeebb[...]" << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
52 <      cerr << "or              " << argv[0] << " ZZmmbb[...]" << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
53 <      cerr << "or              " << argv[0] << " BBHmh[...] " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
54 <      cerr << "or              " << argv[0] << " BBAmh[...] " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
55 <      cerr << "or              " << argv[0] << " Zee[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
56 <      cerr << "or              " << argv[0] << " Zmm[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
57 <      cerr << "or              " << argv[0] << " Zeh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
58 <      cerr << "or              " << argv[0] << " Zmh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
59 <      cerr << "or              " << argv[0] << " Zhh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
60 <      cerr << "or              " << argv[0] << " Wen[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
61 <      cerr << "or              " << argv[0] << " Wmn[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
62 <      cerr << "or              " << argv[0] << " Ten[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
63 <      cerr << "or              " << argv[0] << " Tmn[...]   " << " <nEvents> [RandomSeed] [meanPileupEventCount]" << endl;
50 >      cerr << "Command syntax: " << argv[0] << " ZHmmbb[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
51 >      cerr << "or              " << argv[0] << " ZHeebb[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
52 >      cerr << "or              " << argv[0] << " ZHmmXX[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
53 >      cerr << "or              " << argv[0] << " ZHeeXX[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
54 >      cerr << "or              " << argv[0] << " ZZmmbb[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
55 >      cerr << "or              " << argv[0] << " ZZmmnn[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
56 >      cerr << "or              " << argv[0] << " ZZeebb[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
57 >      cerr << "or              " << argv[0] << " ZZeenn[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
58 >      cerr << "or              " << argv[0] << " BBHmh[...] " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
59 >      cerr << "or              " << argv[0] << " BBAmh[...] " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
60 >      cerr << "or              " << argv[0] << " VBFHXX[..] " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
61 >      cerr << "or              " << argv[0] << " Zee[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
62 >      cerr << "or              " << argv[0] << " Zmm[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
63 >      cerr << "or              " << argv[0] << " Zeh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
64 >      cerr << "or              " << argv[0] << " Zmh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
65 >      cerr << "or              " << argv[0] << " Zhh[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
66 >      cerr << "or              " << argv[0] << " Znn[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
67 >      cerr << "or              " << argv[0] << " Wen[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
68 >      cerr << "or              " << argv[0] << " Wmn[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
69 >      cerr << "or              " << argv[0] << " Ten[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
70 >      cerr << "or              " << argv[0] << " Tmn[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
71 >      cerr << "or              " << argv[0] << " QCD[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
72 >      cerr << "or              " << argv[0] << " QED[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
73 >      cerr << "or              " << argv[0] << " EWK[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
74 >      cerr << "or              " << argv[0] << " Top[...]   " << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
75 >      cerr << "or              " << argv[0] << " QCDMET[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
76 >      cerr << "or              " << argv[0] << " ZnnMET[...]" << " <nEvents> [RandomSeed] [meanPileupEventcount] [cmEnergy]" << endl;
77        cerr << "In above e-electron, m-muon, h-hadronic tau. n-neutrino" << endl;
78 +      cerr << "QCD, QED, EWK and Top do not specify the decays -- they are useful for generic BG production" << endl;
79 +      cerr << "QCDMET has MET > 50 GeV to make efficient to study QCD MET BG" << endl;
80 +      cerr << "ZnnMET has MET > 50 GeV to make efficient to study Znn MET BG" << endl;
81        exit(1);
82      }
83    else {
84      nEvents = atoi(argv[2]);
85      if(argc >= 4) runNumber = atoi(argv[3]);
86      if(argc >= 5) meanPileupEventCount = atoi(argv[4]);
87 +    if(argc >= 6) cmEnergy = atof(argv[5]);
88      if(strncmp(argv[1], "Zee", 3) == 0)
89        {
90          pythia.readString("WeakSingleBoson:ffbar2gmZ = on");
# Line 67 | Line 101 | int main(int argc, char **argv) {
101          pythia.readString("23:onMode = 0");       // Z decays to muons only
102          pythia.readString("23:onIfAny = 13");
103        }
104 +    else if(strncmp(argv[1], "Znn", 3) == 0)
105 +      {
106 +        pythia.readString("WeakSingleBoson:ffbar2gmZ = on");
107 +        pythia.readString("PhaseSpace:mHatMin = 60.");    
108 +        pythia.readString("PhaseSpace:mHatMax = -1");    
109 +        pythia.readString("23:onMode = 0");       // Z decays to neutrinos only
110 +        pythia.readString("23:onIfAny = 12, 14, 16");
111 +        if(strncmp(argv[1], "ZnnMET", 6) == 0) {
112 +          cutOnMET = true;
113 +        }
114 +      }
115      else if(strncmp(argv[1], "Zeh", 3) == 0)
116        {
117          pythia.readString("WeakSingleBoson:ffbar2gmZ = on");
# Line 119 | Line 164 | int main(int argc, char **argv) {
164          pythia.readString("24:onMode = 2");       // W+ decays to muon
165          pythia.readString("24:onIfAny = 13");
166        }
167 +    else if(strncmp(argv[1], "HZGamma", 7) == 0)
168 +      {
169 +        pythia.readString("HiggsSM:all = on");
170 +        pythia.readString("25:m0 = 125");
171 +        pythia.readString("25:onMode = 0");      // Higgs decays to Zgamma only
172 +        pythia.readString("25:onIfMatch = 22 23");
173 +        pythia.readString("23:onMode = 0");      // Z decays to light leptons
174 +        pythia.readString("23:onIfAny = 11 13");
175 +      }
176      else if(strncmp(argv[1], "ZHmmbb", 6) == 0)
177        {
178          pythia.readString("HiggsSM:ffbar2HZ = on");
179 <        pythia.readString("25:m0 = 120");
179 >        pythia.readString("25:m0 = 125");
180          pythia.readString("25:onMode = 0");      // Higgs decays to bbBar only
181          pythia.readString("25:onIfAny = 5");
182          pythia.readString("23:onMode = 0");      // Z decays to mu+,mu- only (for trigger)
183          pythia.readString("23:onIfAny = 13");
184        }
185 +     else if(strncmp(argv[1], "ZHmmXX", 6) == 0)
186 +       {
187 +        pythia.readString("HiggsSM:ffbar2HZ = on");
188 +        pythia.readString("25:m0 = 125");
189 +        pythia.readString("25:mayDecay = false");      // Higgs does not decay -- becomes invisible
190 +        pythia.readString("25:isVisible = false");
191 +        pythia.readString("23:onMode = 0");      // Z decays to mu+,mu- only (for trigger)
192 +        pythia.readString("23:onIfAny = 13");
193 +       }
194      else if(strncmp(argv[1], "ZHeebb", 6) == 0)
195        {
196          pythia.readString("HiggsSM:ffbar2HZ = on");
197 <        pythia.readString("25:m0 = 120");
197 >        pythia.readString("25:m0 = 125");
198          pythia.readString("25:onMode = 0");      // Higgs decays to bbBar only
199          pythia.readString("25:onIfAny = 5");
200          pythia.readString("23:onMode = 0");      // Z decays to e+,e- only (for trigger)
201          pythia.readString("23:onIfAny = 11");
202        }
203 +     else if(strncmp(argv[1], "ZHeeXX", 6) == 0)
204 +       {
205 +        pythia.readString("HiggsSM:ffbar2HZ = on");
206 +        pythia.readString("25:m0 = 125");
207 +        pythia.readString("25:mayDecay = false");      // Higgs does not decay -- becomes invisible
208 +        pythia.readString("25:isVisible = false");
209 +        pythia.readString("23:onMode = 0");      // Z decays to mu+,mu- only (for trigger)
210 +        pythia.readString("23:onIfAny = 11");
211 +       }
212      else if(strncmp(argv[1], "BB", 2) == 0)
213        {
214          pythia.readString("Higgs:useBSM = on");
# Line 167 | Line 239 | int main(int argc, char **argv) {
239           pythia.readString("23:onMode = 0");
240           pythia.readString("23:onIfAny = 5 13");   // Let the Z decay to bQuarks or muons
241         }
242 <    else
243 <      {
244 <        cerr << "Unknown process type " << argv[1] << " - aborting" << endl;
245 <        exit(1);
246 <      }
242 >     else if(strncmp(argv[1], "ZZmmnn", 6) == 0)
243 >       {
244 >         pythia.readString("WeakDoubleBoson:ffbar2gmZgmZ = on");
245 >         pythia.readString("23:onMode = 0");
246 >         pythia.readString("23:onIfAny = 13 12 14 16");   // Let the Z decay to muons or neutrinos
247 >       }
248 >     else if(strncmp(argv[1], "ZZeebb", 6) == 0)
249 >       {
250 >         pythia.readString("WeakDoubleBoson:ffbar2gmZgmZ = on");
251 >         pythia.readString("23:onMode = 0");
252 >         pythia.readString("23:onIfAny = 5 11");   // Let the Z decay to bQuarks or electrons
253 >       }
254 >     else if(strncmp(argv[1], "ZZeenn", 6) == 0)
255 >       {
256 >         pythia.readString("WeakDoubleBoson:ffbar2gmZgmZ = on");
257 >         pythia.readString("23:onMode = 0");
258 >         pythia.readString("23:onIfAny = 11 12 14 16");   // Let the Z decay to electrons or neutrinos
259 >       }
260 >     else if(strncmp(argv[1], "QCD", 3) == 0)
261 >       {
262 >         pythia.readString("HardQCD:all = on");    
263 >         pythia.readString("PhaseSpace:pTHatMin = 50.");
264 >         if(strncmp(argv[1], "QCDMET", 6) == 0) {
265 >           cutOnMET = true;
266 >         }
267 >       }
268 >     else if(strcmp(argv[1], "QED") == 0)
269 >       {
270 >         pythia.readString("PromptPhoton:all = on");
271 >         pythia.readString("PhaseSpace:mHatMin = 60.");    
272 >         pythia.readString("PhaseSpace:mHatMax = -1");
273 >       }
274 >     else if(strncmp(argv[1], "EWK", 3) == 0)
275 >       {
276 >         pythia.readString("WeakSingleBoson:all = on");
277 >         pythia.readString("PhaseSpace:mHatMin = 60.");    
278 >         pythia.readString("PhaseSpace:mHatMax = -1");    
279 >         pythia.readString("WeakDoubleBoson:all = on");
280 >       }
281 >     else if(strncmp(argv[1], "Top", 3) == 0)
282 >       {
283 >         pythia.readString("Top:all = on");
284 >       }
285 >     else if(strncmp(argv[1], "VBFHXX", 6) == 0)
286 >       {
287 >         pythia.readString("HiggsSM:ff2Hff(t:ZZ) = on");
288 >         pythia.readString("HiggsSM:ff2Hff(t:WW) = on");
289 >         pythia.readString("25:m0 = 125");
290 >         pythia.readString("25:mayDecay = false");      // Higgs does not decay -- becomes invisible
291 >         pythia.readString("25:isVisible = false");
292 >       }
293 >     else
294 >       {
295 >         cerr << "Unknown process type " << argv[1] << " - aborting" << endl;
296 >         exit(1);
297 >       }
298    }
299  
177 –  string name(argv[1]);
178 –  name += ".root";
179 –  TFile *outFile = TFile::Open(name.c_str(), "recreate");
180 –
300    // Initialize pythia
301  
302 <  pythia.init( 2212, 2212, 14000.);
302 >  pythia.init(2212, 2212, cmEnergy);
303    pythia.particleData.listAll();
304    Rndm * rndmPtr = &pythia.rndm;
305    rndmPtr->init(runNumber);
# Line 194 | Line 313 | int main(int argc, char **argv) {
313  
314    // Ultra Fast Simulator
315  
316 <  UltraFastSim ufs(rndmPtr);
317 <  bbHAnalysis bbH(outFile, &pythia, &ufs, true);
318 <  //ZHAnalysis ZH; // outFile, &pythia, &ufs, true);
319 <  //ZH.BookTree();
316 >  UltraFastSim ufs;
317 >  char jobName[256];
318 >  sprintf(jobName, "%s-PU%3.3d-%4.4d", argv[1], meanPileupEventCount, runNumber);
319 >  UFSDataStore dataStore(jobName, &ufs);
320  
321    // Begin event loop
322    for (int iEvent = 0; iEvent < nEvents; ) {
# Line 210 | Line 329 | int main(int argc, char **argv) {
329      // Add pileup
330  
331      int pileupEventCount = 0;
332 <    if(meanPileupEventCount > 0) pileupEventCount = meanPileupEventCount; // * rmdmPtr->pois(); (not available yet)
332 >    if(meanPileupEventCount > 0) pileupEventCount = poisson(meanPileupEventCount, rndmPtr);
333      for (int puEvent = 0; puEvent < pileupEventCount; ) {
334        if(!pileupPythia.next()) continue;
335        double vx = rndmPtr->gauss()*2.; // Mean beam spread is 2 mm in x-y and 75 mm in z
# Line 234 | Line 353 | int main(int argc, char **argv) {
353  
354      // Ultra fast simulation
355  
356 <    if(!ufs.run(pythia.event))
356 >    if(!ufs.run(pythia.event, rndmPtr))
357        {
358          cerr << "Ultra fast simulation failed - aborting" << endl;
359          exit(1);
360        }
361  
362 <    // Run bbHAnalysis
362 >    // If necessary, check MET and decide to store or not
363  
364 <    if(!bbH.run())
365 <      {
366 <        cerr << "bbH Analysis failed - aborting" << endl;
367 <        exit(2);
364 >    bool doStore = true;
365 >    if(cutOnMET) {
366 >      double MET = ufs.getMET().Pt();
367 >      if(MET < 50.) {
368 >        doStore = false;
369        }
370 +    }
371 +    // Store data
372  
373 <    // Run ZHAnalysis
252 <
253 <    /* if(!ZH.Run(&ufs))
373 >    if(doStore && !dataStore.run())
374        {
375 <        cerr << "ZH Analysis failed - aborting" << endl;
375 >        cerr << "Failed to store data" << endl;
376          exit(3);
377        }
378 <    */
378 >
379      // End of event loop. Statistics. Histogram. Done.
380  
381      if(!(iEvent % 100)) cout << "Processed event " << iEvent << endl;
# Line 268 | Line 388 | int main(int argc, char **argv) {
388    pythia.statistics();
389    pileupPythia.statistics();
390  
271 –  bbH.end();
272 –
273 –  outFile->cd();
274 –  outFile->Write();
275 –  outFile->Close();
276 –
391    return 0;
392  
393   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines