LeptonSelection/src/MuonSelection.cc

#include <math.h>
#include <iostream>

#include "HiggsAnaDefs.hh"

#include "MuonSelection.h"
#include "ParseArgs.h"
#include "SelectionStatus.h"


SelectionStatus muonPreSelection( ControlFlags &ctrl,  const mithep::TMuon * mu ) { 
  if(ctrl.debug) cout << "inside muonpresel ... " << endl;
  bool ret = isMuFO(mu);
  if(ctrl.debug) cout << "iS fo? ... " << ret << endl;  
  ret &= ( fabs(mu->ip3dSig) < 4 ? true : false ); 
  if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl;  
  ret &= ( mu->pt > 5 ? true : false ); 
  if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;  
  ret &= ( fabs(mu->eta) < 2.4 ? true : false ); 
  if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;  

  SelectionStatus status;
  if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
  if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;  
  return status;
} 

bool isMuFO( const mithep::TMuon * mu ) { 
  bool isgood=true;
  float reliso = mu->pfIso03/mu->pt;

  if( mu->pt < 5  )         isgood=false;                           
  if ( fabs(mu->dz) > 0.2 ) isgood=false;   // needed to remove cosmics in HF sample

  // 

  // HF seems to not want tkquality, SS does
  if( mu->pt>20 ) { 
    if ( !((mu->typeBits & kGlobal) && mu->nSeg>0  ) ) isgood=false; //&& mu->nValidHits>0
  } else {
    if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0 
         && !( (mu->typeBits & kTracker) &&  (mu->qualityBits & kTMOneStationLoose)
               )
         )   isgood=false;
  }


  //
  // for HF MC to agree w/ data, cannot put any Tk quality bits
  // but to get to 5% closure in the SS test, we need at least Tk oneStationLoose

  /*
&& ( (mu->qualityBits & kTMOneStationLoose) ||
                                               (mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) ||
                                               (mu->qualityBits & kTMLastStationOptimizedLowPtLoose) 
                                               )
  */


  // comment for more stats for mu fake shape
  if( mu->pt < 20 ) { 
    if( reliso > 3 ) isgood=false;    
  } else { 
    if( reliso > 5 ) isgood=false;    
  }

  return isgood;
};


SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) { 

  int level=0;
  unsigned failmask=0x0;
  
  if(mu->nTkHits < 11 )  
    failmask |= (1<<level);
  level++;

  if(fabs(mu->d0) > 0.2) 
    failmask |= (1<<level);
  level++;

  if(fabs(mu->dz) > 0.1) 
    failmask |= (1<<level);
  level++;

  if(!(mu->typeBits & kTracker)) 
    failmask |= (1<<level);
  level++;

  if(!(mu->qualityBits & kTMLastStationAngTight)) 
    failmask |= (1<<level);       
  level++;

  Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt;
  if(mu->pt>20 && iso<0.1) 
    failmask |= (1<<level);

  bool passtight  = !(failmask);
  bool passloose  = !(failmask);
  return SelectionStatus( passtight | passloose );
  
}

 
//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) { 
SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) { 
  int level=0;
  unsigned failmask=0x0;

  if(mu->pt < 5) {
    failmask |= (1<<level);
  }

  level++;
  if(fabs(mu->eta) > 2.4) {
    failmask |= (1<<level);
  }

  level++;
  if(!(mu->typeBits & kGlobal)) {
    failmask |= (1<<level);
  }

  level++;
  if(mu->nTkHits          < 10) {
    failmask |= (1<<level);
  }


  level++;  
  if( (mu->trkIso03/mu->pt) > 0.7 ) {
    failmask |= (1<<level);
  }

  //  if(muon->nPixHits   < 1)     return false;
  //  if(muon->muNchi2    > 10)    return false;
  //  if(muon->nMatch     < 2)     return false;
  //  if(muon->nValidHits         < 1)     return false;
  //  if(muon->pterr/muon->pt > 0.1)   return false;
  //  if(fabs(muon->dz)       > 0.1)   return false;  
  
  SelectionStatus status;
  if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
  if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
  return status;


};


//
// Kevin's WW selection
//
SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) { 
  int level=0;
  unsigned failmask=0x0;

  // 0x1
  if(mu->pt < 5) {
    failmask |= (1<<level);
  }

  // 0x2
  level++;
  if(fabs(mu->eta) > 2.4) {
    failmask |= (1<<level);
  }

  // 0x4
  level++;
  if(mu->ptErr/mu->pt > 0.1)   {
    failmask |= (1<<level);
  }

  // 0x8
  level++;
  if( fabs(mu->dz) > 0.1 )   {
    failmask |= (1<<level);
  }


  Bool_t isGlobal  = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0);
  Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight);

  // 0x10
  level++;
  if(!isGlobal && !isTracker) {
    failmask |= (1<<level);
  }

  // 0x20
  level++;
  if( mu->nTkHits < 10 ) {
    failmask |= (1<<level);
  }

  level++;
  if(mu->nPixHits         < 1) {
    failmask |= (1<<level);
  }


  level++;
  if(fabs(mu->d0)>0.02)   { 
    failmask |= (1<<level);
  }

  /*
  if(mu->pt>20) {
    if(fabs(mu->d0)>0.02)   { 
      failmask |= (1<<level);
    }
  } else {
    if(fabs(mu->d0)>0.01)  {
    failmask |= (1<<level);
    }
  }
  */

  SelectionStatus status;
  if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
  if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
  return status;


};

Revision:	1.1
Committed:	Mon Feb 13 09:35:20 2012 UTC (13 years, 3 months ago) by khahn
Content type:	text/plain
Branch:	MAIN
Log Message:	* empty log message *
#	Content
1	#include <math.h>
2	#include <iostream>
3
4	#include "HiggsAnaDefs.hh"
5
6	#include "MuonSelection.h"
7	#include "ParseArgs.h"
8	#include "SelectionStatus.h"
9
10
11	SelectionStatus muonPreSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
12	if(ctrl.debug) cout << "inside muonpresel ... " << endl;
13	bool ret = isMuFO(mu);
14	if(ctrl.debug) cout << "iS fo? ... " << ret << endl;
15	ret &= ( fabs(mu->ip3dSig) < 4 ? true : false );
16	if(ctrl.debug) cout << "and pass IP (" << mu->ip3dSig << ") ? ... " << ret << endl;
17	ret &= ( mu->pt > 5 ? true : false );
18	if(ctrl.debug) cout << "and >5 GeV ? ... " << ret << endl;
19	ret &= ( fabs(mu->eta) < 2.4 ? true : false );
20	if(ctrl.debug) cout << "and < 2.4 eta ? ... " << ret << endl;
21
22	SelectionStatus status;
23	if( ret ) status.setStatus(SelectionStatus::PRESELECTION);
24	if(ctrl.debug) cout << "muon presel returning status : " << status.getStatus() << endl;
25	return status;
26	}
27
28	bool isMuFO( const mithep::TMuon * mu ) {
29	bool isgood=true;
30	float reliso = mu->pfIso03/mu->pt;
31
32	if( mu->pt < 5 ) isgood=false;
33	if ( fabs(mu->dz) > 0.2 ) isgood=false; // needed to remove cosmics in HF sample
34
35	//
36
37	// HF seems to not want tkquality, SS does
38	if( mu->pt>20 ) {
39	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) ) isgood=false; //&& mu->nValidHits>0
40	} else {
41	if ( !((mu->typeBits & kGlobal) && mu->nSeg>0 ) //&& //&& mu->nValidHits>0
42	&& !( (mu->typeBits & kTracker) && (mu->qualityBits & kTMOneStationLoose)
43	)
44	) isgood=false;
45	}
46
47
48	//
49	// for HF MC to agree w/ data, cannot put any Tk quality bits
50	// but to get to 5% closure in the SS test, we need at least Tk oneStationLoose
51
52	/*
53	&& ( (mu->qualityBits & kTMOneStationLoose) \|\|
54	(mu->qualityBits & kTMLastStationOptimizedBarrelLowPtLoose) \|\|
55	(mu->qualityBits & kTMLastStationOptimizedLowPtLoose)
56	)
57	*/
58
59
60	// comment for more stats for mu fake shape
61	if( mu->pt < 20 ) {
62	if( reliso > 3 ) isgood=false;
63	} else {
64	if( reliso > 5 ) isgood=false;
65	}
66
67	return isgood;
68	};
69
70
71	SelectionStatus passSoftMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
72
73	int level=0;
74	unsigned failmask=0x0;
75
76	if(mu->nTkHits < 11 )
77	failmask \|= (1<<level);
78	level++;
79
80	if(fabs(mu->d0) > 0.2)
81	failmask \|= (1<<level);
82	level++;
83
84	if(fabs(mu->dz) > 0.1)
85	failmask \|= (1<<level);
86	level++;
87
88	if(!(mu->typeBits & kTracker))
89	failmask \|= (1<<level);
90	level++;
91
92	if(!(mu->qualityBits & kTMLastStationAngTight))
93	failmask \|= (1<<level);
94	level++;
95
96	Double_t iso = (mu->trkIso03 + mu->emIso03 + mu->hadIso03)/mu->pt;
97	if(mu->pt>20 && iso<0.1)
98	failmask \|= (1<<level);
99
100	bool passtight = !(failmask);
101	bool passloose = !(failmask);
102	return SelectionStatus( passtight \| passloose );
103
104	}
105
106
107	//unsigned passMuonSelectionZZ( const mithep::TMuon * mu ) {
108	SelectionStatus passMuonSelectionZZ( ControlFlags &ctrl, const mithep::TMuon * mu ) {
109	int level=0;
110	unsigned failmask=0x0;
111
112	if(mu->pt < 5) {
113	failmask \|= (1<<level);
114	}
115
116	level++;
117	if(fabs(mu->eta) > 2.4) {
118	failmask \|= (1<<level);
119	}
120
121	level++;
122	if(!(mu->typeBits & kGlobal)) {
123	failmask \|= (1<<level);
124	}
125
126	level++;
127	if(mu->nTkHits < 10) {
128	failmask \|= (1<<level);
129	}
130
131
132	level++;
133	if( (mu->trkIso03/mu->pt) > 0.7 ) {
134	failmask \|= (1<<level);
135	}
136
137	// if(muon->nPixHits < 1) return false;
138	// if(muon->muNchi2 > 10) return false;
139	// if(muon->nMatch < 2) return false;
140	// if(muon->nValidHits < 1) return false;
141	// if(muon->pterr/muon->pt > 0.1) return false;
142	// if(fabs(muon->dz) > 0.1) return false;
143
144	SelectionStatus status;
145	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
146	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
147	return status;
148
149
150
151	};
152
153
154
155
156	//
157	// Kevin's WW selection
158	//
159	SelectionStatus passMuonSelection( ControlFlags &ctrl, const mithep::TMuon * mu ) {
160	int level=0;
161	unsigned failmask=0x0;
162
163	// 0x1
164	if(mu->pt < 5) {
165	failmask \|= (1<<level);
166	}
167
168	// 0x2
169	level++;
170	if(fabs(mu->eta) > 2.4) {
171	failmask \|= (1<<level);
172	}
173
174	// 0x4
175	level++;
176	if(mu->ptErr/mu->pt > 0.1) {
177	failmask \|= (1<<level);
178	}
179
180	// 0x8
181	level++;
182	if( fabs(mu->dz) > 0.1 ) {
183	failmask \|= (1<<level);
184	}
185
186
187	Bool_t isGlobal = (mu->typeBits & kGlobal) && (mu->muNchi2 < 10) && (mu->nMatch > 1) && (mu->nValidHits > 0);
188	Bool_t isTracker = (mu->typeBits & kTracker) && (mu->qualityBits & kTMLastStationTight);
189
190	// 0x10
191	level++;
192	if(!isGlobal && !isTracker) {
193	failmask \|= (1<<level);
194	}
195
196	// 0x20
197	level++;
198	if( mu->nTkHits < 10 ) {
199	failmask \|= (1<<level);
200	}
201
202	level++;
203	if(mu->nPixHits < 1) {
204	failmask \|= (1<<level);
205	}
206
207
208	level++;
209	if(fabs(mu->d0)>0.02) {
210	failmask \|= (1<<level);
211	}
212
213	/*
214	if(mu->pt>20) {
215	if(fabs(mu->d0)>0.02) {
216	failmask \|= (1<<level);
217	}
218	} else {
219	if(fabs(mu->d0)>0.01) {
220	failmask \|= (1<<level);
221	}
222	}
223	*/
224
225	SelectionStatus status;
226	if( !failmask ) status.setStatus(SelectionStatus::LOOSEID);
227	if( !failmask ) status.setStatus(SelectionStatus::TIGHTID);
228	return status;
229
230
231	};
232