ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/VHbb/interface/VHbbNameSpace.h
(Generate patch)

Comparing UserCode/VHbb/interface/VHbbNameSpace.h (file contents):
Revision 1.1 by bortigno, Fri Dec 14 15:59:45 2012 UTC vs.
Revision 1.7 by nmohr, Thu Mar 21 17:25:17 2013 UTC

# Line 1 | Line 1
1 <
1 > #include "TLorentzVector.h"
2 > #include "TVector3.h"
3 > #include "TVector2.h"
4   #include "TMath.h"
5 + /*#if !defined(__CINT__) && !defined(__MAKECINT__)
6 + #include "CondFormats/JetMETObjects/interface/JetCorrectorParameters.h"
7 + #include "CondFormats/JetMETObjects/interface/JetCorrectionUncertainty.h"
8 + #endif*/
9  
10   namespace VHbb {
11 +  
12  
13    double deltaPhi(double phi1,double phi2)
14    {
# Line 15 | Line 22 | namespace VHbb {
22      {
23        double deta = eta1 - eta2;
24        double dphi = deltaPhi(phi1, phi2);
25 <      return std::sqrt(deta*deta + dphi*dphi);
25 >      return TMath::Sqrt(deta*deta + dphi*dphi);
26 >    }
27 >
28 >
29 >  double Hmass( double V_eta,double V_phi,double V_pt,
30 >                double hJet1_eta,double hJet1_phi,double hJet1_pt,
31 >                double hJet2_eta,double hJet2_phi,double hJet2_pt ){
32 >    
33 >    TVector3 V(1,1,1);
34 >    V.SetPtEtaPhi(V_pt,V_eta,V_phi);
35 >    
36 >    TVector3 H1(1,1,1);
37 >    H1.SetPtEtaPhi(hJet1_pt,hJet1_eta,hJet1_phi);
38 >    H1.SetMag(1/sin(H1.Theta()));
39 >    
40 >    TVector3 H2(1,1,1);
41 >    H2.SetPtEtaPhi(hJet2_pt,hJet2_eta,hJet2_phi);
42 >    H2.SetMag(1/sin(H2.Theta()));
43 >    
44 >    TVector3 n1(H1);
45 >    TVector3 n2(H2);
46 >    
47 >    double det= n1.Px() * n2.Py() - n2.Px() * n1.Py();
48 >    
49 >    H1.SetMag( (  - n2.Py() * V.Px() + n2.Px() * V.Py() )  / (sin(n1.Theta()) *det ) );
50 >    H2.SetMag( ( + n1.Py() * V.Px() - n1.Px() * V.Py() )  / (sin(n2.Theta())  *det ) );
51 >    
52 >    double mass=TMath::Sqrt( TMath::Power( (H1.Mag()+H2.Mag()),2 ) - TMath::Power(( ( H1+H2 ).Mag()),2) );
53 >    
54 >    return mass;
55 >    
56 >  }
57 >  
58 >  double Hmass_comb(double hJet1_eta,double hJet1_phi,double hJet1_pt, double hJet1_mass,
59 >                    double hJet2_eta,double hJet2_phi,double hJet2_pt, double hJet2_mass){
60 >
61 >    TLorentzVector H1, H2;
62 >    H1.SetPtEtaPhiM(hJet1_pt,hJet1_eta,hJet1_phi, hJet1_mass);;
63 >    H2.SetPtEtaPhiM(hJet2_pt,hJet2_eta,hJet2_phi, hJet2_mass);
64 >
65 >    return (H1 + H2).M();
66 >
67 >  }
68 >
69 >  double Hmass_3j(double h_eta,double h_phi,double h_pt, double h_mass,
70 >                  double aJet_eta,double aJet_phi,double aJet_pt, double aJet_mass){
71 >
72 >    TLorentzVector H, H3;
73 >    H.SetPtEtaPhiM( h_pt,h_eta,h_phi, h_mass);;
74 >    H3.SetPtEtaPhiM(aJet_pt,aJet_eta,aJet_phi, aJet_mass);
75 >
76 >    return (H + H3).M();
77 >
78 >
79 >  }
80 >  
81 >  double ANGLELZ(double pt, double eta, double phi, double mass, double pt2, double eta2, double phi2, double mass2){
82 >  TLorentzVector m1, m2, msum;
83 >  m1.SetPtEtaPhiM(pt, eta, phi, mass);
84 >  m2.SetPtEtaPhiM(pt2, eta2, phi2, mass2);
85 >  msum = m1 + m2;
86 >
87 >  TVector3 bZ =  msum.BoostVector();
88 >
89 >  m1.Boost(-bZ);  
90 >  m2.Boost(-bZ);  
91 >
92 >  TVector3 b1;
93 >
94 >
95 >  if((int) (pt) % 2 == 0)
96 >    b1 =  m1.BoostVector();
97 >  else
98 >    b1 =  m2.BoostVector();
99 >
100 > double cosTheta = b1.Dot(msum.BoostVector()) / (b1.Mag()*msum.BoostVector().Mag());
101 > return(cosTheta);
102 >  }
103 >
104 >
105 >   double ANGLEHB(double pt, double eta, double phi, double e, double pt2, double eta2, double phi2, double e2){
106 > TLorentzVector m1, m2, msum;
107 > m1.SetPtEtaPhiE(pt, eta, phi, e);
108 > m2.SetPtEtaPhiE(pt2, eta2, phi2, e2);
109 > msum = m1 + m2;
110 >
111 > TVector3 bZ =  msum.BoostVector();
112 >
113 > m1.Boost(-bZ);
114 > m2.Boost(-bZ);  
115 >
116 > TVector3 b1;
117 >
118 >  if((int) (pt) % 2 == 0)
119 >    b1 =  m1.BoostVector();
120 >  else
121 >    b1 =  m2.BoostVector();
122 >
123 > double cosTheta = b1.Dot(msum.BoostVector()) / (b1.Mag()*msum.BoostVector().Mag());
124 > return(cosTheta);
125 >   }
126 >
127 >   double metCorSysShift(double met, double metphi, int Nvtx, int EVENT_run)
128 > {
129 >    double metx = met * cos(metphi);
130 >    double mety = met * sin(metphi);
131 >    double px = 0.0, py = 0.0;
132 >    if (EVENT_run!=1) {
133 >        //pfMEtSysShiftCorrParameters_2012runAplusBvsNvtx_data
134 >        px = +1.68804e-01 + 3.37139e-01*Nvtx;
135 >        py = -1.72555e-01 - 1.79594e-01*Nvtx;
136 >    } else {
137 >        //pfMEtSysShiftCorrParameters_2012runAplusBvsNvtx_mc
138 >        px = +2.22335e-02 - 6.59183e-02*Nvtx;
139 >        py = +1.52720e-01 - 1.28052e-01*Nvtx;
140 >    }
141 >    metx -= px;
142 >    mety -= py;
143 >    return std::sqrt(metx*metx + mety*mety);
144 > }
145 >
146 >    double metphiCorSysShift(double met, double metphi, int Nvtx, int EVENT_run)
147 > {
148 >    double metx = met * cos(metphi);
149 >    double mety = met * sin(metphi);
150 >    double px = 0.0, py = 0.0;
151 >    if (EVENT_run!=1) {
152 >
153 >        //pfMEtSysShiftCorrParameters_2012runAplusBvsNvtx_data
154 >        px = +1.68804e-01 + 3.37139e-01*Nvtx;
155 >        py = -1.72555e-01 - 1.79594e-01*Nvtx;
156 >    } else {
157 >        //pfMEtSysShiftCorrParameters_2012runAplusBvsNvtx_mc
158 >        px = +2.22335e-02 - 6.59183e-02*Nvtx;
159 >        py = +1.52720e-01 - 1.28052e-01*Nvtx;
160      }
161 +    metx -= px;
162 +    mety -= py;
163 +    if (metx == 0.0 && mety == 0.0)
164 +        return 0.0;
165 +
166 +    double phi1 = std::atan2(mety,metx);
167 +    double phi2 = std::atan2(mety,metx)-2.0*M_PI;
168 +    if (std::abs(phi1-metphi) < std::abs(phi2-metphi)+0.5*M_PI)
169 +        return phi1;
170 +    else
171 +        return phi2;
172 + }
173 +
174 + TVector2 metType1Reg(double met, double metphi, double corr1, double corr2, double pt1, double eta1, double phi1, double e1, double pt2, double eta2, double phi2, double e2)
175 + {
176 +    double metx = met * cos(metphi);
177 +    double mety = met * sin(metphi);
178 +    TLorentzVector j1;
179 +    TLorentzVector j2;
180 +    j1.SetPtEtaPhiE(pt1,eta1,phi1, e1 );
181 +    j2.SetPtEtaPhiE(pt2,eta2,phi2, e2 );
182 +    metx += j1.Px()*(1-corr1);
183 +    metx += j2.Px()*(1-corr2);
184 +    mety += j1.Py()*(1-corr1);
185 +    mety += j2.Py()*(1-corr2);
186 +    TVector2 corrMET(metx, mety);
187 +     return corrMET;
188 + }
189 +
190 + double metType1Phi(double met, double metphi, double corr1, double corr2, double pt1, double eta1, double phi1, double e1, double pt2, double eta2, double phi2, double e2){
191 +    return metType1Reg(met, metphi, corr1, corr2, pt1, eta1, phi1, e1, pt2, eta2, phi2, e2).Phi();
192 +
193 + }
194 + double metType1Et(double met, double metphi, double corr1, double corr2, double pt1, double eta1, double phi1, double e1, double pt2, double eta2, double phi2, double e2){
195 +    return metType1Reg(met, metphi, corr1, corr2, pt1, eta1, phi1, e1, pt2, eta2, phi2, e2).Mod();
196 +
197 + }
198 +
199 +
200 +   double met_MPF(double met, double metphi, double pt, double phi)
201 + {
202 +    return 1.+met*pt*std::cos( deltaPhi(metphi,phi) ) / (pt*pt);
203 +
204 + }
205 +
206 + double resolutionBias(double eta)
207 + {
208 + // return 0;//Nominal!
209 +  if(eta< 0.5) return 0.052;
210 +  if(eta< 1.1) return 0.057;
211 +  if(eta< 1.7) return 0.096;
212 +  if(eta< 2.3) return 0.134;
213 +  if(eta< 5) return 0.28;
214 +  return 0;
215 + }
216 +
217 + double evalJERBias( double ptreco, double ptgen, double eta1){
218 +  double eta = fabs(eta1);
219 +  double cor =1;  
220 +  if ((fabs(ptreco - ptgen)/ ptreco)<0.5) { //Limit the effect to the core
221 +     cor = (ptreco +resolutionBias(eta) *(ptreco-ptgen))/ptreco;  
222 +  }
223 +  if (ptgen > 0.) return ptreco*cor;
224 +  else return ptreco;
225 + }
226 +
227 + double evalEt( double pt, double eta, double phi, double e){
228 +  TLorentzVector j;
229 +  j.SetPtEtaPhiE(pt,eta,phi, e );
230 +  return j.Et();
231 +
232 + }
233 +
234 + double evalMt( double pt, double eta, double phi, double e){
235 +  TLorentzVector j;
236 +  j.SetPtEtaPhiE(pt,eta,phi, e );
237 +  return j.Mt();
238 +
239 + }
240 + /*double evalJECUnc( double pt, double eta){
241 + // Total uncertainty for reference
242 + JetCorrectionUncertainty *total = new JetCorrectionUncertainty("/shome/nmohr/CMSSW_5_2_6_patch1/src/UserCode/VHbb/data/START53_V15MC_Uncertainty_AK5PFchs.txt");
243 +
244 + total->setJetPt(pt);
245 + total->setJetEta(eta);
246 + double uncert =  total->getUncertainty(true);
247 + delete total;
248 + return uncert;
249 + }*/
250 +
251 + double ptWeightDY( double lheV_pt)
252 + {
253 +    double SF = 1.;
254 +    if (50. < lheV_pt && lheV_pt < 100.){
255 +        SF = 0.873885+0.00175853*lheV_pt;
256 +    }
257 +    else if (lheV_pt > 100){
258 +        SF = 1.10651-0.000705265*lheV_pt;
259 +    }
260 +    return SF;
261 + }
262 +
263 +
264 +
265   }
266  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines