Utils/src/MuonTools.cc

#include "MitPhysics/Utils/interface/MuonTools.h"
using namespace mithep;

MuonTools::MuonTools() :
  fpion_em_etaEmi(0),
  fpion_had_etaEmi(0),
  fpion_had_etaTmi(0),
  fpion_em_etaB(0),
  fpion_had_etaB(0),
  fpion_ho_etaB(0),
  fpion_had_etaTpl(0),
  fpion_em_etaEpl(0),
  fpion_had_etaEpl(0),
  fmuon_em_etaEmi(0),
  fmuon_had_etaEmi(0),
  fmuon_had_etaTmi(0),
  fmuon_em_etaB(0),
  fmuon_had_etaB(0),
  fmuon_ho_etaB(0),
  fmuon_had_etaTpl(0),
  fmuon_em_etaEpl(0),
  fmuon_had_etaEpl(0) {}
MuonTools::~MuonTools() {
  delete fpion_em_etaEmi;
  delete fpion_had_etaEmi;
  delete fpion_had_etaTmi;
  delete fpion_em_etaB;
  delete fpion_had_etaB;
  delete fpion_ho_etaB;
  delete fpion_had_etaTpl;
  delete fpion_em_etaEpl;
  delete fpion_had_etaEpl;
  delete fmuon_em_etaEmi;
  delete fmuon_had_etaEmi;
  delete fmuon_had_etaTmi;
  delete fmuon_em_etaB;
  delete fmuon_had_etaB;
  delete fmuon_ho_etaB;
  delete fmuon_had_etaTpl;
  delete fmuon_em_etaEpl;
  delete fmuon_had_etaEpl;
}

double MuonTools::getCaloCompatability(mithep::Muon* iMuon,bool iEMSpecial, bool iCorrectedHCAL) {
  if(fpion_em_etaEmi == 0) {
    TFile* fPion_templates = new TFile("$CMSSW_BASE/src/MitPhysics/Init/PionCaloTemplate.root","READ");
    TFile* fMuon_templates = new TFile("$CMSSW_BASE/src/MitPhysics/Init/MuonCaloTemplate.root","READ");
    fpion_em_etaEmi  = (TH2D*) fPion_templates->Get("em_etaEmi");
    fpion_had_etaEmi = (TH2D*) fPion_templates->Get("had_etaEmi");
    fpion_had_etaTmi = (TH2D*) fPion_templates->Get("had_etaTmi");
    fpion_em_etaB    = (TH2D*) fPion_templates->Get("em_etaB")   ;
    fpion_had_etaB   = (TH2D*) fPion_templates->Get("had_etaB")  ;
    fpion_ho_etaB    = (TH2D*) fPion_templates->Get("ho_etaB")   ;
    fpion_had_etaTpl = (TH2D*) fPion_templates->Get("had_etaTpl");
    fpion_em_etaEpl  = (TH2D*) fPion_templates->Get("em_etaEpl") ;
    fpion_had_etaEpl = (TH2D*) fPion_templates->Get("had_etaEpl");
    fmuon_em_etaEmi  = (TH2D*) fMuon_templates->Get("em_etaEmi") ;
    fmuon_had_etaEmi = (TH2D*) fMuon_templates->Get("had_etaEmi");
    fmuon_had_etaTmi = (TH2D*) fMuon_templates->Get("had_etaTmi");
    fmuon_em_etaB    = (TH2D*) fMuon_templates->Get("em_etaB")   ;
    fmuon_had_etaB   = (TH2D*) fMuon_templates->Get("had_etaB")  ;
    fmuon_ho_etaB    = (TH2D*) fMuon_templates->Get("ho_etaB")   ;
    fmuon_had_etaTpl = (TH2D*) fMuon_templates->Get("had_etaTpl");
    fmuon_em_etaEpl  = (TH2D*) fMuon_templates->Get("em_etaEpl");
    fmuon_had_etaEpl = (TH2D*) fMuon_templates->Get("had_etaEpl");

    fpion_em_etaEmi ->SetDirectory(0);
    fpion_had_etaEmi->SetDirectory(0);
    fpion_had_etaTmi->SetDirectory(0);
    fpion_em_etaB   ->SetDirectory(0); 
    fpion_had_etaB  ->SetDirectory(0); 
    fpion_ho_etaB   ->SetDirectory(0); 
    fpion_had_etaTpl->SetDirectory(0); 
    fpion_em_etaEpl ->SetDirectory(0);  
    fpion_had_etaEpl->SetDirectory(0);  
    fmuon_em_etaEmi ->SetDirectory(0);  
    fmuon_had_etaEmi->SetDirectory(0);  
    fmuon_had_etaTmi->SetDirectory(0);  
    fmuon_em_etaB   ->SetDirectory(0);  
    fmuon_had_etaB  ->SetDirectory(0);  
    fmuon_ho_etaB   ->SetDirectory(0);  
    fmuon_had_etaTpl->SetDirectory(0);  
    fmuon_em_etaEpl ->SetDirectory(0);  
    fmuon_had_etaEpl->SetDirectory(0);  
    fPion_templates->Close();
    fMuon_templates->Close();
    delete fPion_templates;
    delete fMuon_templates;
  }
  double lEta = -1.; double lP = -1;
  double lEM  = -5.;      double lHad = 0;      double lHO = 0;
  lEta = iMuon->Eta();
  lP   = iMuon->P(); 
  if(lP >= 2000.) lP = 1999.9;
  if(!iEMSpecial || iMuon->EmEnergy() != 0.) lEM  = iMuon->EmEnergy();
  lHad = iMuon->HadEnergy();
  lHO  = iMuon->HoEnergy();
  if(lP < 0. )           return 0.5; 
  if(fabs(lEta) >  2.5 ) return 0.5; 
  TH2D* lTMuonHad = NULL;
  TH2D* lTPionHad = NULL;
  TH2D* lTMuonHo  = NULL;
  TH2D* lTPionHo  = NULL;
  TH2D* lTMuonEm  = NULL;
  TH2D* lTPionEm  = NULL;
  
  if(fabs(lEta) >=  1.27) {
    if(iCorrectedHCAL) lHad *= 1.8/2.2;
    if(lEta > 0) {
      lTPionHad = fpion_had_etaEpl;
      lTMuonHad = fmuon_had_etaEpl;
    } else {
      lTPionHad = fpion_had_etaEmi;
      lTMuonHad = fmuon_had_etaEmi;
    }
  }
  if(fabs(lEta) <  1.27  && fabs(lEta) >=  1.1 ) {
    if(iCorrectedHCAL)    lHad *= (1.8/(-2.2*fabs(lEta)+5.5));
    if(lEta > 0) {
      lTPionHad  = fpion_had_etaTpl;
      lTMuonHad  = fmuon_had_etaTpl;
    } else {
      lTPionHad  = fpion_had_etaTmi;
      lTMuonHad  = fmuon_had_etaTmi;
    }
  }
  if(fabs(lEta) <  1.1) {
    if(iCorrectedHCAL)    lHad *= sin(2*atan(exp(iMuon->Eta())));
    lTPionHad  = fpion_had_etaB;
    lTMuonHad  = fmuon_had_etaB;
  }
  if(lEta >  1.479  ) {
    lTPionEm  = fpion_em_etaEpl;
    lTMuonEm  = fmuon_em_etaEpl;
  }
  if(fabs(lEta) <=  1.479) {
    lTPionEm  = fpion_em_etaB;
    lTMuonEm  = fmuon_em_etaB;
  }
  if(lEta < -1.479 ) {
    lTPionEm  = fpion_em_etaEmi;
    lTMuonEm  = fmuon_em_etaEmi;
  }
  if(fabs(lEta) < 1.28) {
    lTPionHo  = fpion_ho_etaB;
    lTMuonHo  = fmuon_ho_etaB;
  }
  
  double lPBX = 1.;     double lPSX = 1.; 
  double lPBY = 1.;     double lPSY = 1.; 
  double lPBZ = 1.;     double lPSZ = 1.; 
  if(!overflow(lTPionEm, lP,lEM))  lPBX =  lTPionEm ->GetBinContent(lTPionEm ->GetXaxis()->FindBin(lP),lTPionEm ->GetYaxis()->FindBin(lEM) );
  if(!overflow(lTPionHad,lP,lHad)) lPBY =  lTPionHad->GetBinContent(lTPionHad->GetXaxis()->FindBin(lP),lTPionHad->GetYaxis()->FindBin(lHad));
  if(!overflow(lTPionHo, lP,lHO))  lPBZ =  lTPionHo ->GetBinContent(lTPionHo ->GetXaxis()->FindBin(lP),lTPionHo ->GetYaxis()->FindBin(lHO) );
  if(!overflow(lTMuonEm, lP,lEM )) lPSX =  lTMuonEm ->GetBinContent(lTMuonEm ->GetXaxis()->FindBin(lP),lTMuonEm ->GetYaxis()->FindBin(lEM) );
  if(!overflow(lTMuonHad,lP,lHad)) lPSY =  lTMuonHad->GetBinContent(lTMuonHad->GetXaxis()->FindBin(lP),lTMuonHad->GetYaxis()->FindBin(lHad));
  if(!overflow(lTMuonHo ,lP,lHO))  lPSZ =  lTMuonHo ->GetBinContent(lTMuonHo ->GetXaxis()->FindBin(lP),lTMuonHo ->GetYaxis()->FindBin(lHO) );
  
  if(lPSX == 0. || lPBX == 0. || (lEM <= 0. && !iEMSpecial)) {lPSX = 1.; lPBX = 1.;} 
  if(lPSY == 0. || lPBY == 0. || lHad == 0.) {lPSY = 1.; lPBY = 1.;}
  if(lPSZ == 0. || lPBZ == 0. || lHO  == 0.) {lPSZ = 1.; lPBZ = 1.;}
  if((lPSX*lPSY*lPSZ+lPBX*lPBY*lPBZ) > 0.) return lPSX*lPSY*lPSZ/(lPSX*lPSY*lPSZ+lPBX*lPBY*lPBZ);
  return 0.5;
}

bool MuonTools::isGood(mithep::Muon *iMuon,selection iSelection) {
  double lVal = 0;
  switch(iSelection) {
  case AllArbitrated:
    if(iMuon->StandaloneTrk() != 0 || iMuon->GlobalTrk()!= 0)  return true;
    if(iMuon->NSegments() > 0) return true;
    return false;
    break;
  case PromptTight: 
    return promptTight(-1,iMuon);
    break;
   case TMOneStationLoose:
     return TMOneStation(iMuon,99999,999999);
     break;
  case TMOneStationTight:
    return TMOneStation(iMuon);
    break;
  case TMLastStationLoose: 
    return TMLastStation(iMuon,999999,999999);
    break;
  case TMLastStationTight: 
    return TMLastStation(iMuon);
    break;
  case TM2DCompatibilityLoose: 
    lVal             = 1.2*getSegmentCompatability(iMuon); 
    if(lVal/1.2 == 0.5) return false;
    lVal += 0.8*getCaloCompatability(iMuon,true,true);
    if(lVal > 0.7) return true;
    return false;
    break;
  case TM2DCompatibilityTight: 
    lVal             = 1.2*getSegmentCompatability(iMuon); 
    if(lVal/1.2 == 0.5) return false;
    lVal += 0.8*getCaloCompatability(iMuon,true,true);
    if(lVal > 1.0) return true;
    return false;
    break;
  default:
    return false;
  }
  return false;
}
Revision:	1.4
Committed:	Tue Nov 11 21:21:25 2008 UTC (16 years, 5 months ago) by ceballos
Content type:	text/plain
Branch:	MAIN
Changes since 1.3:	+2 -0 lines
Log Message:	fixing a memory leak
#	User	Rev	Content
1	pharris	1.2	#include "MitPhysics/Utils/interface/MuonTools.h"
2	pharris	1.1	using namespace mithep;
3
4			MuonTools::MuonTools() :
5			fpion_em_etaEmi(0),
6			fpion_had_etaEmi(0),
7			fpion_had_etaTmi(0),
8			fpion_em_etaB(0),
9			fpion_had_etaB(0),
10			fpion_ho_etaB(0),
11			fpion_had_etaTpl(0),
12			fpion_em_etaEpl(0),
13			fpion_had_etaEpl(0),
14			fmuon_em_etaEmi(0),
15			fmuon_had_etaEmi(0),
16			fmuon_had_etaTmi(0),
17			fmuon_em_etaB(0),
18			fmuon_had_etaB(0),
19			fmuon_ho_etaB(0),
20			fmuon_had_etaTpl(0),
21			fmuon_em_etaEpl(0),
22			fmuon_had_etaEpl(0) {}
23			MuonTools::~MuonTools() {
24			delete fpion_em_etaEmi;
25			delete fpion_had_etaEmi;
26			delete fpion_had_etaTmi;
27			delete fpion_em_etaB;
28			delete fpion_had_etaB;
29			delete fpion_ho_etaB;
30			delete fpion_had_etaTpl;
31			delete fpion_em_etaEpl;
32			delete fpion_had_etaEpl;
33			delete fmuon_em_etaEmi;
34			delete fmuon_had_etaEmi;
35			delete fmuon_had_etaTmi;
36			delete fmuon_em_etaB;
37			delete fmuon_had_etaB;
38			delete fmuon_ho_etaB;
39			delete fmuon_had_etaTpl;
40			delete fmuon_em_etaEpl;
41			delete fmuon_had_etaEpl;
42			}
43
44			double MuonTools::getCaloCompatability(mithep::Muon* iMuon,bool iEMSpecial, bool iCorrectedHCAL) {
45			if(fpion_em_etaEmi == 0) {
46	pharris	1.2	TFile* fPion_templates = new TFile("$CMSSW_BASE/src/MitPhysics/Init/PionCaloTemplate.root","READ");
47			TFile* fMuon_templates = new TFile("$CMSSW_BASE/src/MitPhysics/Init/MuonCaloTemplate.root","READ");
48	pharris	1.1	fpion_em_etaEmi = (TH2D*) fPion_templates->Get("em_etaEmi");
49			fpion_had_etaEmi = (TH2D*) fPion_templates->Get("had_etaEmi");
50			fpion_had_etaTmi = (TH2D*) fPion_templates->Get("had_etaTmi");
51			fpion_em_etaB = (TH2D*) fPion_templates->Get("em_etaB") ;
52			fpion_had_etaB = (TH2D*) fPion_templates->Get("had_etaB") ;
53			fpion_ho_etaB = (TH2D*) fPion_templates->Get("ho_etaB") ;
54			fpion_had_etaTpl = (TH2D*) fPion_templates->Get("had_etaTpl");
55			fpion_em_etaEpl = (TH2D*) fPion_templates->Get("em_etaEpl") ;
56			fpion_had_etaEpl = (TH2D*) fPion_templates->Get("had_etaEpl");
57			fmuon_em_etaEmi = (TH2D*) fMuon_templates->Get("em_etaEmi") ;
58			fmuon_had_etaEmi = (TH2D*) fMuon_templates->Get("had_etaEmi");
59			fmuon_had_etaTmi = (TH2D*) fMuon_templates->Get("had_etaTmi");
60			fmuon_em_etaB = (TH2D*) fMuon_templates->Get("em_etaB") ;
61			fmuon_had_etaB = (TH2D*) fMuon_templates->Get("had_etaB") ;
62			fmuon_ho_etaB = (TH2D*) fMuon_templates->Get("ho_etaB") ;
63			fmuon_had_etaTpl = (TH2D*) fMuon_templates->Get("had_etaTpl");
64			fmuon_em_etaEpl = (TH2D*) fMuon_templates->Get("em_etaEpl");
65			fmuon_had_etaEpl = (TH2D*) fMuon_templates->Get("had_etaEpl");
66	pharris	1.3
67			fpion_em_etaEmi ->SetDirectory(0);
68			fpion_had_etaEmi->SetDirectory(0);
69			fpion_had_etaTmi->SetDirectory(0);
70			fpion_em_etaB ->SetDirectory(0);
71			fpion_had_etaB ->SetDirectory(0);
72			fpion_ho_etaB ->SetDirectory(0);
73			fpion_had_etaTpl->SetDirectory(0);
74			fpion_em_etaEpl ->SetDirectory(0);
75			fpion_had_etaEpl->SetDirectory(0);
76			fmuon_em_etaEmi ->SetDirectory(0);
77			fmuon_had_etaEmi->SetDirectory(0);
78			fmuon_had_etaTmi->SetDirectory(0);
79			fmuon_em_etaB ->SetDirectory(0);
80			fmuon_had_etaB ->SetDirectory(0);
81			fmuon_ho_etaB ->SetDirectory(0);
82			fmuon_had_etaTpl->SetDirectory(0);
83			fmuon_em_etaEpl ->SetDirectory(0);
84			fmuon_had_etaEpl->SetDirectory(0);
85			fPion_templates->Close();
86			fMuon_templates->Close();
87	ceballos	1.4	delete fPion_templates;
88			delete fMuon_templates;
89	pharris	1.1	}
90			double lEta = -1.; double lP = -1;
91			double lEM = -5.; double lHad = 0; double lHO = 0;
92			lEta = iMuon->Eta();
93			lP = iMuon->P();
94			if(lP >= 2000.) lP = 1999.9;
95			if(!iEMSpecial \|\| iMuon->EmEnergy() != 0.) lEM = iMuon->EmEnergy();
96			lHad = iMuon->HadEnergy();
97			lHO = iMuon->HoEnergy();
98			if(lP < 0. ) return 0.5;
99			if(fabs(lEta) > 2.5 ) return 0.5;
100			TH2D* lTMuonHad = NULL;
101			TH2D* lTPionHad = NULL;
102			TH2D* lTMuonHo = NULL;
103			TH2D* lTPionHo = NULL;
104			TH2D* lTMuonEm = NULL;
105			TH2D* lTPionEm = NULL;
106
107			if(fabs(lEta) >= 1.27) {
108			if(iCorrectedHCAL) lHad *= 1.8/2.2;
109			if(lEta > 0) {
110			lTPionHad = fpion_had_etaEpl;
111			lTMuonHad = fmuon_had_etaEpl;
112			} else {
113			lTPionHad = fpion_had_etaEmi;
114			lTMuonHad = fmuon_had_etaEmi;
115			}
116			}
117			if(fabs(lEta) < 1.27 && fabs(lEta) >= 1.1 ) {
118			if(iCorrectedHCAL) lHad = (1.8/(-2.2fabs(lEta)+5.5));
119			if(lEta > 0) {
120			lTPionHad = fpion_had_etaTpl;
121			lTMuonHad = fmuon_had_etaTpl;
122			} else {
123			lTPionHad = fpion_had_etaTmi;
124			lTMuonHad = fmuon_had_etaTmi;
125			}
126			}
127			if(fabs(lEta) < 1.1) {
128			if(iCorrectedHCAL) lHad = sin(2atan(exp(iMuon->Eta())));
129			lTPionHad = fpion_had_etaB;
130			lTMuonHad = fmuon_had_etaB;
131			}
132			if(lEta > 1.479 ) {
133			lTPionEm = fpion_em_etaEpl;
134			lTMuonEm = fmuon_em_etaEpl;
135			}
136			if(fabs(lEta) <= 1.479) {
137			lTPionEm = fpion_em_etaB;
138			lTMuonEm = fmuon_em_etaB;
139			}
140			if(lEta < -1.479 ) {
141			lTPionEm = fpion_em_etaEmi;
142			lTMuonEm = fmuon_em_etaEmi;
143			}
144			if(fabs(lEta) < 1.28) {
145			lTPionHo = fpion_ho_etaB;
146			lTMuonHo = fmuon_ho_etaB;
147			}
148
149			double lPBX = 1.; double lPSX = 1.;
150			double lPBY = 1.; double lPSY = 1.;
151			double lPBZ = 1.; double lPSZ = 1.;
152			if(!overflow(lTPionEm, lP,lEM)) lPBX = lTPionEm ->GetBinContent(lTPionEm ->GetXaxis()->FindBin(lP),lTPionEm ->GetYaxis()->FindBin(lEM) );
153			if(!overflow(lTPionHad,lP,lHad)) lPBY = lTPionHad->GetBinContent(lTPionHad->GetXaxis()->FindBin(lP),lTPionHad->GetYaxis()->FindBin(lHad));
154			if(!overflow(lTPionHo, lP,lHO)) lPBZ = lTPionHo ->GetBinContent(lTPionHo ->GetXaxis()->FindBin(lP),lTPionHo ->GetYaxis()->FindBin(lHO) );
155			if(!overflow(lTMuonEm, lP,lEM )) lPSX = lTMuonEm ->GetBinContent(lTMuonEm ->GetXaxis()->FindBin(lP),lTMuonEm ->GetYaxis()->FindBin(lEM) );
156			if(!overflow(lTMuonHad,lP,lHad)) lPSY = lTMuonHad->GetBinContent(lTMuonHad->GetXaxis()->FindBin(lP),lTMuonHad->GetYaxis()->FindBin(lHad));
157			if(!overflow(lTMuonHo ,lP,lHO)) lPSZ = lTMuonHo ->GetBinContent(lTMuonHo ->GetXaxis()->FindBin(lP),lTMuonHo ->GetYaxis()->FindBin(lHO) );
158
159			if(lPSX == 0. \|\| lPBX == 0. \|\| (lEM <= 0. && !iEMSpecial)) {lPSX = 1.; lPBX = 1.;}
160			if(lPSY == 0. \|\| lPBY == 0. \|\| lHad == 0.) {lPSY = 1.; lPBY = 1.;}
161			if(lPSZ == 0. \|\| lPBZ == 0. \|\| lHO == 0.) {lPSZ = 1.; lPBZ = 1.;}
162			if((lPSXlPSYlPSZ+lPBXlPBYlPBZ) > 0.) return lPSXlPSYlPSZ/(lPSXlPSYlPSZ+lPBXlPBYlPBZ);
163			return 0.5;
164			}
165
166			bool MuonTools::isGood(mithep::Muon *iMuon,selection iSelection) {
167			double lVal = 0;
168			switch(iSelection) {
169			case AllArbitrated:
170			if(iMuon->StandaloneTrk() != 0 \|\| iMuon->GlobalTrk()!= 0) return true;
171			if(iMuon->NSegments() > 0) return true;
172			return false;
173			break;
174			case PromptTight:
175			return promptTight(-1,iMuon);
176			break;
177			case TMOneStationLoose:
178			return TMOneStation(iMuon,99999,999999);
179			break;
180			case TMOneStationTight:
181			return TMOneStation(iMuon);
182			break;
183			case TMLastStationLoose:
184			return TMLastStation(iMuon,999999,999999);
185			break;
186			case TMLastStationTight:
187			return TMLastStation(iMuon);
188			break;
189			case TM2DCompatibilityLoose:
190			lVal = 1.2*getSegmentCompatability(iMuon);
191			if(lVal/1.2 == 0.5) return false;
192			lVal += 0.8*getCaloCompatability(iMuon,true,true);
193			if(lVal > 0.7) return true;
194			return false;
195			break;
196			case TM2DCompatibilityTight:
197			lVal = 1.2*getSegmentCompatability(iMuon);
198			if(lVal/1.2 == 0.5) return false;
199			lVal += 0.8*getCaloCompatability(iMuon,true,true);
200			if(lVal > 1.0) return true;
201			return false;
202			break;
203			default:
204			return false;
205			}
206			return false;
207			}