[ViewVC] Diff of: cvsroot/UserCode/MitPhysics/Utils/src/MVATools.cc

Comparing UserCode/MitPhysics/Utils/src/MVATools.cc (file contents):
Revision 1.9 by bendavid, Sat Dec 17 20:00:40 2011 UTC vs.
Revision 1.15 by mingyang, Mon Jun 11 09:04:33 2012 UTC

+#include "MitPhysics/Utils/interface/MVATools.h"
+#include "MitPhysics/Utils/interface/ElectronTools.h"
+#include "MitPhysics/Utils/interface/IsolationTools.h"
-+
+#include "MitAna/DataTree/interface/PFCandidateCol.h"
+#include "MitAna/DataTree/interface/StableData.h"
-+
+#include <TMath.h>
+#include <TFile.h>
+#include <TRandom3.h>
+#include "TMVA/Tools.h"//MVA
+#include "TMVA/Reader.h"//MVA
-+
+ClassImp(mithep::MVATools)
+using namespace mithep;
+      readers[i]->AddVariable( "ScEta", &ScEta );
-+
+    if(VariableType==10){
-+
+      readers[i]->AddVariable( "HoE", &HoE );
-+
+      readers[i]->AddVariable( "covIEtaIEta", &covIEtaIEta );
-+
+      readers[i]->AddVariable( "tIso1abs", &tIso1abs );
-+
+      readers[i]->AddVariable( "tIso3abs", &tIso3abs );
-+
+      readers[i]->AddVariable( "tIso2abs", &tIso2abs );
-+
+      readers[i]->AddVariable( "R9", &R9 );
-+
+      readers[i]->AddVariable( "absIsoEcal", &absIsoEcal );
-+
+      readers[i]->AddVariable( "absIsoHcal", &absIsoHcal );
-+
+      readers[i]->AddVariable( "NVertexes", &NVertexes );
-+
+      readers[i]->AddVariable( "ScEta", &ScEta );
-+
+      readers[i]->AddVariable( "EtaWidth", &EtaWidth );
-+
+      readers[i]->AddVariable( "PhiWidth", &PhiWidth );
-+
+    }
-+
-+
+    if(VariableType==1201){
-+
+      /*readers[i]->AddVariable( "myphoton_pfchargedisogood03", &myphoton_pfchargedisogood03);
-+
+      readers[i]->AddVariable( "myphoton_pfchargedisobad03", &myphoton_pfchargedisobad03);
-+
+      readers[i]->AddVariable( "myphoton_pfphotoniso03", &myphoton_pfphotoniso03 );
-+
+      readers[i]->AddVariable( "myphoton_sieie", &myphoton_sieie );
-+
+      readers[i]->AddVariable( "myphoton_sieip", &myphoton_sieip );
-+
+      readers[i]->AddVariable( "myphoton_etawidth", &myphoton_etawidth );
-+
+      readers[i]->AddVariable( "myphoton_phiwidth", &myphoton_phiwidth );
-+
+      readers[i]->AddVariable( "myphoton_r9", &myphoton_r9 );
-+
+      readers[i]->AddVariable( "myphoton_s4ratio", &myphoton_s4ratio );
-+
+      readers[i]->AddVariable( "myphoton_SCeta", &myphoton_SCeta );
-+
+      readers[i]->AddVariable( "event_rho", &event_rho );
-+
+      if(i==0){
-+
+        readers[i]->AddVariable( "myphoton_ESEffSigmaRR", &myphoton_ESEffSigmaRR);
-+
+        }*/
-+
+      readers[i]->AddVariable( "ph.r9", &myphoton_r9 );
-+
+      readers[i]->AddVariable( "ph.sigietaieta", &myphoton_sieie );
-+
+      readers[i]->AddVariable( "ph.scetawidth", &myphoton_etawidth );
-+
+      readers[i]->AddVariable( "ph.scphiwidth", &myphoton_phiwidth );
-+
+      readers[i]->AddVariable( "ph.idmva_CoviEtaiPhi", &myphoton_sieip );
-+
+      readers[i]->AddVariable( "ph.idmva_s4ratio", &myphoton_s4ratio );
-+
+      readers[i]->AddVariable( "ph.idmva_GammaIso", &myphoton_pfphotoniso03 );
-+
+      readers[i]->AddVariable( "ph.idmva_ChargedIso_selvtx", &myphoton_pfchargedisogood03);
-+
+      readers[i]->AddVariable( "ph.idmva_ChargedIso_worstvtx", &myphoton_pfchargedisobad03);
-+
+      readers[i]->AddVariable( "ph.sceta", &myphoton_SCeta );
-+
+      readers[i]->AddVariable( "rho", &event_rho );
-+
+      if(i==0){
-+
+        readers[i]->AddVariable( "1.00023*ph.idmva_PsEffWidthSigmaRR + 0.0913", &myphoton_ESEffSigmaRR);
-+
+      }
-+
+    }
-+
+  fReaderEndcap->BookMVA("BDT method",EndcapWeights);
+  //initilize the bool value
+  PassMVA=kFALSE;
-<
+  Float_t photon_bdt =  MVATools::GetMVAbdtValue(p,vtx,trackCol,vtxCol, _tRho, els, applyElectronVeto);
->
+  Float_t photon_bdt =  MVATools::GetMVAbdtValue_2011(p,vtx,trackCol,vtxCol, _tRho, els, applyElectronVeto);
+  if (isbarrel) {
+    if(bdt>bdtCutBarrel){
+//--------------------------------------------------------------------------------------------------
-<
+Float_t MVATools::GetMVAbdtValue(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const ElectronCol* els, Bool_t applyElectronVeto) {
->
+Float_t MVATools::GetMVAbdtValue_2012_globe(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho, const PFCandidateCol *fPFCands,const ElectronCol* els,Bool_t applyElectronVeto) {
+  //get the variables used to compute MVA variables
+  ecalIso3 = p->EcalRecHitIsoDr03();
+  RelE2x5Right=p->SCluster()->Seed()->E2x5Right()/RawEnergy;
+  RelE5x5=p->SCluster()->Seed()->E5x5()/RawEnergy;
-<
+  EtaWidth=p->SCluster()->EtaWidth();
-<
+  PhiWidth=p->SCluster()->PhiWidth();
->
+  EtaWidth=p->EtaWidth();
->
+  PhiWidth=p->PhiWidth();
+  CoviEtaiPhi=p->SCluster()->Seed()->CoviEtaiPhi();
+  CoviPhiiPhi=p->SCluster()->Seed()->CoviPhiiPhi();
+  //spectator variables
+  Pt_MVA=p->Pt();
+  ScEta_MVA=p->SCluster()->Eta();
-+
-+
+  //
-+
-+
+  isbarrel = (fabs(ScEta_MVA)<1.4442);
-+
-+
+  //variable 1201
-+
+  myphoton_pfchargedisogood03=IsolationTools::PFChargedIsolation(p,vtx,0.3,0,fPFCands);
-+
+  myphoton_pfchargedisobad03=IsolationTools::PFChargedIsolation(p,vtx,0.3,0,fPFCands,&wVtxInd,vtxCol);
-+
+  myphoton_pfphotoniso03=IsolationTools::PFGammaIsolation(p,0.3,0,fPFCands);
-+
+  myphoton_sieie=covIEtaIEta;
-+
+  myphoton_sieip=CoviEtaiPhi;
-+
+  myphoton_etawidth=EtaWidth;
-+
+  myphoton_phiwidth=PhiWidth;
-+
+  myphoton_r9=R9;
-+
+  myphoton_s4ratio=p->S4Ratio();
-+
+  myphoton_SCeta=ScEta_MVA;
-+
+  event_rho= _tRho;
-+
-+
+  myphoton_ESEffSigmaRR=-99;
-+
-+
+  if(!isbarrel){
-+
+    myphoton_ESEffSigmaRR=p->EffSigmaRR();
-+
+  }
-+
-+
+  if (isbarrel) {
-+
+    reader = fReaderBarrel;
-+
+  }
-+
+  else {
-+
+    reader = fReaderEndcap;
-+
+  }
-+
-+
+  assert(reader);
-+
-+
+  bdt = reader->EvaluateMVA("BDT method");
-+
-+
+  /* printf("HoE: %f\n",HoE);
-+
+  printf("covIEtaIEta: %f\n",covIEtaIEta);
-+
+  printf("tIso1abs: %f\n",tIso1abs);
-+
+  printf("tIso3abs: %f\n",tIso3abs);
-+
+  printf("tIso2abs: %f\n",tIso2abs);
-+
-+
+  printf("absIsoEcal: %f\n",absIsoEcal);
-+
+  printf("absIsoHcal: %f\n",absIsoHcal);
-+
+  printf("RelEMax: %f\n",RelEMax);
-+
+  printf("RelETop: %f\n",RelETop);
-+
+  printf("RelEBottom: %f\n",RelEBottom);
-+
+  printf("RelELeft: %f\n",RelELeft);
-+
+  printf("RelERight: %f\n",RelERight);
-+
+  printf("RelE2x5Max: %f\n",RelE2x5Max);
-+
+  printf("RelE2x5Top: %f\n",RelE2x5Top);
-+
+  printf("RelE2x5Bottom: %f\n",RelE2x5Bottom);
-+
+  printf("RelE2x5Left: %f\n",RelE2x5Left);
-+
+  printf("RelE2x5Right;: %f\n",RelE2x5Right);
-+
+  printf("RelE5x5: %f\n",RelE5x5);
-+
-+
+  printf("EtaWidth: %f\n",EtaWidth);
-+
+  printf("PhiWidth: %f\n",PhiWidth);
-+
+  printf("CoviEtaiPhi: %f\n",CoviEtaiPhi);
-+
+  printf("CoviPhiiPhi: %f\n",CoviPhiiPhi);
-+
-+
+  if (!isbarrel) {
-+
+    printf("RelPreshowerEnergy: %f\n",RelPreshowerEnergy);
-+
+    }*/
-+
-+
+  return bdt;
-+
+}
-+
-+
+Float_t MVATools::GetMVAbdtValue_2011(const Photon* p,const Vertex* vtx,const TrackCol* trackCol,const VertexCol* vtxCol,Double_t _tRho,const ElectronCol* els,Bool_t applyElectronVeto) {
-+
-+
+  //get the variables used to compute MVA variables
-+
+  ecalIso3 = p->EcalRecHitIsoDr03();
-+
+  ecalIso4 = p->EcalRecHitIsoDr04();
-+
+  hcalIso4 = p->HcalTowerSumEtDr04();
-+
-+
+  wVtxInd = 0;
-+
-+
+  trackIso1 = IsolationTools::CiCTrackIsolation(p,vtx, 0.3, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, NULL, NULL, (!applyElectronVeto ? els : NULL) );//Question Ming:whyfPV->At(0) instead of selected vertex using ranking method?
-+
-+
+  // track iso worst vtx
-+
+  trackIso2 = IsolationTools::CiCTrackIsolation(p,vtx, 0.4, 0.02, 0.0, 0.0, 0.1, 1.0, trackCol, &wVtxInd,vtxCol, (!applyElectronVeto ? els : NULL) );
-+
-+
+  combIso1 = ecalIso3+hcalIso4+trackIso1 - 0.17*_tRho;
-+
+  combIso2 = ecalIso4+hcalIso4+trackIso2 - 0.52*_tRho;
-+
-+
+  RawEnergy = p->SCluster()->RawEnergy();
-+
-+
+  ScEta = p->SCluster()->Eta();
-+
-+
+  //mva varialbes v1 and v2
-+
+  tIso1 = (combIso1) *50./p->Et();
-+
+  tIso3 = (trackIso1)*50./p->Et();
-+
+  tIso2 = (combIso2) *50./(p->MomVtx(vtxCol->At(wVtxInd)->Position()).Pt());
-+
+  RelIsoEcal=(ecalIso3-0.17*_tRho)/p->Et();
-+
+  RelIsoHcal=(hcalIso4-0.17*_tRho)/p->Et();
-+
-+
+  //compute mva variables for v3
-+
+  HoE = p->HadOverEm();
-+
+  covIEtaIEta = p->CoviEtaiEta();
-+
+  tIso1abs = combIso1;
-+
+  tIso3abs = trackIso1;
-+
+  tIso2abs = combIso2;
-+
+  R9 = p->R9();
-+
-+
+  absIsoEcal=(ecalIso3-0.17*_tRho);
-+
+  absIsoHcal=(hcalIso4-0.17*_tRho);
-+
+  RelEMax=p->SCluster()->Seed()->EMax()/RawEnergy;
-+
+  RelETop=p->SCluster()->Seed()->ETop()/RawEnergy;
-+
+  RelEBottom=p->SCluster()->Seed()->EBottom()/RawEnergy;
-+
+  RelELeft=p->SCluster()->Seed()->ELeft()/RawEnergy;
-+
+  RelERight=p->SCluster()->Seed()->ERight()/RawEnergy;
-+
+  RelE2x5Max=p->SCluster()->Seed()->E2x5Max()/RawEnergy;
-+
+  RelE2x5Top=p->SCluster()->Seed()->E2x5Top()/RawEnergy;
-+
+  RelE2x5Bottom=p->SCluster()->Seed()->E2x5Bottom()/RawEnergy;
-+
+  RelE2x5Left=p->SCluster()->Seed()->E2x5Left()/RawEnergy;
-+
+  RelE2x5Right=p->SCluster()->Seed()->E2x5Right()/RawEnergy;
-+
+  RelE5x5=p->SCluster()->Seed()->E5x5()/RawEnergy;
-+
-+
+  EtaWidth=p->EtaWidth();
-+
+  PhiWidth=p->PhiWidth();
-+
+  CoviEtaiPhi=p->SCluster()->Seed()->CoviEtaiPhi();
-+
+  CoviPhiiPhi=p->SCluster()->Seed()->CoviPhiiPhi();
-+
-+
+  RelPreshowerEnergy=p->SCluster()->PreshowerEnergy()/RawEnergy;
-+
+  NVertexes=vtxCol->GetEntries();
-+
+  //spectator variables
-+
+  Pt_MVA=p->Pt();
-+
+  ScEta_MVA=p->SCluster()->Eta();
-+
-+
+  //
-+
+  isbarrel = (fabs(ScEta_MVA)<1.4442);
+  if (isbarrel) {

Diff Legend

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

Comparing UserCode/MitPhysics/Utils/src/MVATools.cc (file contents): Revision 1.9 by bendavid, Sat Dec 17 20:00:40 2011 UTC vs. Revision 1.15 by mingyang, Mon Jun 11 09:04:33 2012 UTC

Diff Legend

Comparing UserCode/MitPhysics/Utils/src/MVATools.cc (file contents):
Revision 1.9 by bendavid, Sat Dec 17 20:00:40 2011 UTC vs.
Revision 1.15 by mingyang, Mon Jun 11 09:04:33 2012 UTC