ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitPhysics/Utils/src/JetIDMVA.cc
(Generate patch)

Comparing UserCode/MitPhysics/Utils/src/JetIDMVA.cc (file contents):
Revision 1.14 by bendavid, Tue May 29 17:09:02 2012 UTC vs.
Revision 1.16 by pharris, Tue Sep 25 15:39:15 2012 UTC

# Line 177 | Line 177 | void JetIDMVA::Initialize( JetIDMVA::Cut
177      fReader->AddSpectator( "jetPt"    , &fJPt1      );  
178      fReader->AddSpectator( "jetEta"   , &fJEta1     );
179    }
180 +  if (fType == kQGP) {
181 +    fReader->AddVariable( "nvtx"      , &fNVtx       );
182 +    fReader->AddVariable( "jetPt"     , &fJPt1       );  
183 +    fReader->AddVariable( "jetEta"    , &fJEta1      );
184 +    fReader->AddVariable( "jetPhi"    , &fJPhi1      );            
185 +    fReader->AddVariable( "beta"      , &fBeta      );
186 +    fReader->AddVariable( "betaStar"  , &fBetaStar  );
187 +    fReader->AddVariable( "nParticles", &fNNeutrals );
188 +    fReader->AddVariable( "nCharged"  , &fNCharged  );
189 +    fReader->AddVariable( "dRMean"    , &fDRMean    );
190 +    fReader->AddVariable( "ptD"       , &fPtD       );
191 +    fReader->AddVariable( "frac01"    , &fFrac01    );
192 +    fReader->AddVariable( "frac02"    , &fFrac02    );
193 +    fReader->AddVariable( "frac03"    , &fFrac03    );
194 +    fReader->AddVariable( "frac04"    , &fFrac04    );
195 +    fReader->AddVariable( "frac05"    , &fFrac05    );
196 +  }
197  
198    fLowPtReader->BookMVA(fLowPtMethodName  , iLowPtWeights );
199    fReader->BookMVA(fHighPtMethodName , iHighPtWeights );
# Line 236 | Line 253 | Double_t JetIDMVA::MVAValue(
253    return lMVA;
254   }
255   //--------------------------------------------------------------------------------------------------
256 + Bool_t JetIDMVA::passPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector,
257 +                        const PileupEnergyDensityCol *iPileupEnergyDensity,
258 +                        RhoUtilities::RhoType type) {
259 +  if(iJetCorrector == 0) return (iJet->Pt() > fJetPtMin);
260 +  double lPt  = correctedPt(iJet,                  iJetCorrector,iPileupEnergyDensity,type);
261 +  if(lPt < fJetPtMin)                         return false;
262 +  return true;
263 + }
264 + //--------------------------------------------------------------------------------------------------
265   Bool_t JetIDMVA::pass(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices,
266                        FactorizedJetCorrector *iJetCorrector,
267                        const PileupEnergyDensityCol *iPileupEnergyDensity,
# Line 434 | Line 460 | Double_t JetIDMVA::MVAValue(const PFJet
460  
461    return lMVA;
462   }
463 + //--------------------------------------------------------------------------------------------------
464 + Double_t* JetIDMVA::QGValue(const PFJet *iJet,const Vertex *iVertex,const VertexCol *iVertices, //Vertex here is the PV
465 +                            FactorizedJetCorrector *iJetCorrector,
466 +                            const PileupEnergyDensityCol *iPileupEnergyDensity,
467 +                            Bool_t printDebug) {
468 +
469 +  Double_t *lId = new double[3];
470 +  lId[0] = -2;
471 +  lId[1] = -2;
472 +  lId[2] = -2;
473 +  if (!fIsInitialized) {
474 +    std::cout << "Error: JetIDMVA not properly initialized.\n";
475 +    return lId;
476 +  }
477 +  if(!JetTools::passPFLooseId(iJet)) return lId;
478 +
479 +  fJPt1       = correctedPt(iJet,iJetCorrector,iPileupEnergyDensity);
480 +  if(fJPt1 < 20) return lId;
481 +
482 +  //set all input variables
483 +  fNVtx       = iVertices->GetEntries();
484 +  fJEta1      = iJet->RawMom().Eta();
485 +  fJPhi1      = iJet->RawMom().Phi();
486 +  fJD01       = JetTools::impactParameter(iJet,iVertex);  
487 +  fJDZ1       = JetTools::impactParameter(iJet,iVertex,true);
488 +  fBeta       = JetTools::Beta(iJet,iVertex,fDZCut);
489 +  fBetaStar   = JetTools::betaStar(iJet,iVertex,iVertices,fDZCut);
490 +  fNCharged   = iJet->ChargedMultiplicity();
491 +  fNNeutrals  = iJet->NeutralMultiplicity();
492 +  fNParticles = fNCharged+fNNeutrals;
493 +  fPtD        = JetTools::W(iJet,-1,0);  
494 +
495 +  fDRMean    = JetTools::dRMean(iJet,-1);
496 +  fDR2Mean   = JetTools::dR2Mean(iJet,-1);
497 +  fFrac01    = JetTools::frac  (iJet,0.1,0. ,-1);
498 +  fFrac02    = JetTools::frac  (iJet,0.2,0.1,-1);
499 +  fFrac03    = JetTools::frac  (iJet,0.3,0.2,-1);
500 +  fFrac04    = JetTools::frac  (iJet,0.4,0.3,-1);
501 +  fFrac05    = JetTools::frac  (iJet,0.5,0.4,-1);
502 +
503 +  double lMVA = 0;
504 +  lId[0] = fReader->EvaluateMulticlass( fHighPtMethodName )[0];
505 +  lId[1] = fReader->EvaluateMulticlass( fHighPtMethodName )[1];
506 +  lId[2] = fReader->EvaluateMulticlass( fHighPtMethodName )[2];
507 +  if (printDebug == kTRUE) {
508 +    std::cout << "Debug Jet MVA: "
509 +              << fNVtx      << " "
510 +              << fJPt1      << " "
511 +              << fJEta1     << " "
512 +              << fJPhi1     << " "
513 +              << fJD01      << " "
514 +              << fJDZ1      << " "
515 +              << fBeta      << " "
516 +              << fBetaStar  << " "
517 +              << fNCharged  << " "
518 +              << fNNeutrals << " "
519 +              << fDRMean    << " "
520 +              << fFrac01    << " "
521 +              << fFrac02    << " "
522 +              << fFrac03    << " "
523 +              << fFrac04    << " "
524 +              << fFrac05    << " "
525 +              << fDRMean    
526 +              << " === : === "
527 +              << lMVA << " "    
528 +              << std::endl;
529 +  }
530 +  return lId;
531 + }
532   Double_t JetIDMVA::correctedPt(const PFJet *iJet, FactorizedJetCorrector *iJetCorrector,
533                                 const PileupEnergyDensityCol *iPUEnergyDensity,
534 <                               RhoUtilities::RhoType type) {
534 >                               RhoUtilities::RhoType type,int iId) {
535    Double_t Rho = 0.0;
536    switch(type) {
537    case RhoUtilities::MIT_RHO_VORONOI_HIGH_ETA:
# Line 468 | Line 563 | Double_t JetIDMVA::correctedPt(const PFJ
563    iJetCorrector->setRho   (Rho);
564    iJetCorrector->setJetA  (iJet->JetArea());
565    iJetCorrector->setJetEMF(-99.0);    
566 <  Double_t correction = iJetCorrector->getCorrection();
566 >  Double_t correction = 1.;
567 >  if(iId < 0 || iId == 100) correction = iJetCorrector->getCorrection();
568 >  std::vector<float> lCorrections; if(iId != -1 && iId != 100) lCorrections = iJetCorrector->getSubCorrections();
569 >  if(iId > -1 && iId < int(lCorrections.size())) correction = lCorrections[iId];
570 >  if(iId == 100) {
571 >    iJetCorrector->setJetEta(rawMom.Eta());
572 >    iJetCorrector->setJetPt (rawMom.Pt());
573 >    iJetCorrector->setJetPhi(rawMom.Phi());
574 >    iJetCorrector->setJetE  (rawMom.E());
575 >    iJetCorrector->setRho   (Rho);
576 >    iJetCorrector->setJetA  (iJet->JetArea());
577 >    iJetCorrector->setJetEMF(-99.0);
578 >    lCorrections = iJetCorrector->getSubCorrections();
579 >    double correction2 = 1;
580 >    correction2 *= lCorrections[0];
581 >    correction = correction-correction2;
582 >  }
583 >
584    return rawMom.Pt()*correction;
585   }

Diff Legend

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