ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/kiesel/TreeWriter/treeWriter.cc
(Generate patch)

Comparing UserCode/kiesel/TreeWriter/treeWriter.cc (file contents):
Revision 1.19 by kiesel, Tue Apr 23 12:35:48 2013 UTC vs.
Revision 1.28 by kiesel, Sun Apr 28 10:33:56 2013 UTC

# Line 70 | Line 70 | std::vector<Type>* getVectorFromMap( map
70          return vec;
71   }
72  
73 + float deltaPhi( float phi1, float phi2) {
74 +        float result = phi1 - phi2;
75 +        while (result > M_PI) result -= 2*M_PI;
76 +        while (result <= -M_PI) result += 2*M_PI;
77 +        return result;
78 + }
79 +
80   // useful functions
81   float deltaR( const TLorentzVector& v1, const TLorentzVector& v2 ) {
82          // deltaR  = sqrt ( deltaEta^2 + deltaPhi^2 )
83 <        return sqrt(pow(v1.Eta() - v2.Eta(), 2) + pow(v1.Phi() - v2.Phi(), 2) );
83 >        return sqrt(pow(v1.Eta() - v2.Eta(), 2) + pow(deltaPhi(v1.Phi(),v2.Phi()), 2) );
84   }
85  
86   float effectiveAreaElectron( float eta ) {
# Line 253 | Line 260 | void TreeWriter::Loop() {
260          tree->Branch("ht", &ht, "ht/F");
261          tree->Branch("nVertex", &nVertex, "nVertex/I");
262          tree->Branch("pu_weight", &pu_weight, "pu_weight/F");
263 <
263 >        tree->Branch("genElectron", &genElectron);
264 >        tree->Branch("genPhoton", &genPhoton);
265  
266          for (unsigned long jentry=0; jentry < processNEvents; ++jentry) {
267                  if ( loggingVerbosity>1 || jentry%reportEvery==0 )
# Line 265 | Line 273 | void TreeWriter::Loop() {
273                  jet.clear();
274                  electron.clear();
275                  muon.clear();
276 +                genElectron.clear();
277 +                genPhoton.clear();
278                  ht = 0;
279  
280                  // weights
# Line 288 | Line 298 | void TreeWriter::Loop() {
298  
299                  for(std::vector<susy::Photon>::iterator it = photonVector.begin();
300                                  it != photonVector.end(); ++it ) {
301 <                        if( !(it->isEB() || it->isEE()) && skim )
301 >                        if( !(it->isEE() || it->isEB()) && it->momentum.Pt()<20 && it->isEBEtaGap() && it->isEBPhiGap() && it->isEERingGap() && it->isEEDeeGap() && it->isEBEEGap() && skim )
302                                  continue;
303                          tree::Photon thisphoton;
294                        thisphoton.pt = getPtFromMatchedJet( *it, jetVector, loggingVerbosity );
304  
305                          thisphoton.chargedIso = chargedHadronIso_corrected(*it, event->rho25);
306                          thisphoton.neutralIso = neutralHadronIso_corrected(*it, event->rho25);
307                          thisphoton.photonIso = photonIso_corrected(*it, event->rho25);
308  
309 <                        bool loose_photon_barrel = thisphoton.pt>20
301 <                                && it->isEB()
309 >                        bool loose_photon_barrel = it->isEB()
310                                  && it->passelectronveto
311                                  && it->hadTowOverEm<0.05
312                                  && it->sigmaIetaIeta<0.012
313                                  && thisphoton.chargedIso<2.6
314                                  && thisphoton.neutralIso<3.5+0.04*thisphoton.pt
315                                  && thisphoton.photonIso<1.3+0.005*thisphoton.pt;
316 <                        bool loose_photon_endcap = thisphoton.pt > 20
317 <                                && it->isEE()
316 >
317 >                        bool loose_photon_endcap = it->isEE()
318                                  && it->passelectronveto
319                                  && it->hadTowOverEm<0.05
320                                  && it->sigmaIetaIeta<0.034
321                                  && thisphoton.chargedIso<2.3
322                                  && thisphoton.neutralIso<2.9+0.04*thisphoton.pt;
323 <                        if(!(loose_photon_endcap || loose_photon_barrel || thisphoton.pt > 75 ) && skim )
323 >
324 >                        thisphoton.ptJet = getPtFromMatchedJet( *it, jetVector, loggingVerbosity );
325 >                        if(!(loose_photon_endcap || loose_photon_barrel || thisphoton.ptJet > 75 ) && skim )
326                                  continue;
327 +                        thisphoton.pt = it->momentum.Pt();
328                          thisphoton.eta = it->momentum.Eta();
329                          thisphoton.phi = it->momentum.Phi();
330                          thisphoton.r9 = it->r9;
# Line 349 | Line 360 | void TreeWriter::Loop() {
360  
361                          // Calculate HT.
362                          // The definiton differs from the saved jet, since trigger is described better
363 <                        if( std::abs( corrP4.Eta() ) < 3 && corrP4.Pt > 40 )
363 >                        if( std::abs( corrP4.Eta() ) < 3 && corrP4.Pt() > 40 )
364                                  ht += thisjet.pt;
365  
366                          if(std::abs(corrP4.Eta()) > 2.6 && skim ) continue;
# Line 381 | Line 392 | void TreeWriter::Loop() {
392                  if( loggingVerbosity > 1 )
393                          std::cout << "Found " << jet.size() << " jets" << std::endl;
394  
395 +                if( ht < 450 && skim)
396 +                        continue;
397 +
398 +
399  
400                  // met
401                  std::map<TString, susy::MET>::iterator met_it = event->metMap.find("pfMet");
# Line 407 | Line 422 | void TreeWriter::Loop() {
422                          // use veto electrons
423                          if( it->momentum.Pt() < 20  || it->momentum.Pt() > 1e6 )
424                                  continue; // spike rejection
425 <                        float iso = ( it->chargedHadronIso + max(it->neutralHadronIso+it->photonIso
411 <                                                                                                                - effectiveAreaElectron(it->momentum.Eta())*event->rho25, (float)0. )
425 >                        float iso = ( it->chargedHadronIso + max(it->neutralHadronIso+it->photonIso - effectiveAreaElectron(it->momentum.Eta())*event->rho25, (float)0. )
426                                                  ) / it->momentum.Pt();
427                          float d0 = d0correction( *it, *event );
428                          float dZ = std::abs( dZcorrection( *it, *event ) );
# Line 463 | Line 477 | void TreeWriter::Loop() {
477                  // vertices
478                  nVertex = event->vertices.size();
479  
480 <                if( ht < 450 && skim)
481 <                        continue;
482 <
480 >                tree::Particle thisGenParticle;
481 >                for( std::vector<susy::Particle>::iterator it = event->genParticles.begin(); it != event->genParticles.end(); ++it ) {
482 >                        if( it->momentum.Pt() < 20 ) continue;
483 >                        thisGenParticle.pt = it->momentum.Pt();
484 >                        thisGenParticle.eta = it->momentum.Eta();
485 >                        thisGenParticle.phi = it->momentum.Phi();
486 >                        switch( std::abs(it->pdgId) ) {
487 >                                case 22: // photon
488 >                                        genPhoton.push_back( thisGenParticle );
489 >                                        break;
490 >                                case 11: // electron
491 >                                        // Demand a W boson as mother particle of electron
492 >                                        if( abs(event->genParticles[it->motherIndex].pdgId) == 24 )
493 >                                                genElectron.push_back( thisGenParticle );
494 >                                        break;
495 >                        }
496 >                }
497  
498                  tree->Fill();
499          } // for jentry

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines