[ViewVC] Diff of: cvsroot/UserCode/MitHzz4l/Angles/src/Angles.cc

Comparing UserCode/MitHzz4l/Angles/src/Angles.cc (file contents):
Revision 1.1 by khahn, Thu Sep 8 13:33:18 2011 UTC vs.
Revision 1.6 by dkralph, Mon Dec 17 12:35:18 2012 UTC

-–
+#include "Angles.h"
+#include <iostream>
-+
+#include "Angles.h"
-+
+#include "EventData.h"
-+
-+
+void fillAngles(Angles &a,
-+
+                TLorentzVector lep1, int lep1_q,
-+
+                TLorentzVector lep2, int lep2_q,
-+
+                TLorentzVector lep3, int lep3_q,
-+
+                TLorentzVector lep4, int lep4_q
-+
+                ) {
-+
-+
+  LabVectors l;
-+
-+
+  if( lep1_q > 0 ) {
-+
+    l.vecl1p = lep1;
-+
+    l.vecl1m = lep2;
-+
+  } else {
-+
+    l.vecl1p = lep2;
-+
+    l.vecl1m = lep1;
-+
+  }
-+
-+
+  if( lep3_q > 0 ) {
-+
+    l.vecl2p = lep3;
-+
+    l.vecl2m = lep4;
-+
+  } else {
-+
+    l.vecl2p = lep4;
-+
+    l.vecl2m = lep3;
-+
+  }
-+
-+
+  l.vecz1 = (l.vecl1p+l.vecl1m);
-+
+  l.vecz2 = (l.vecl2p+l.vecl2m);
-+
+  l.vec4l = (l.vecz1+l.vecz2);
-+
-+
+  getAngles( l, a );
-+
+}
-+
+void fillAngles( EventData &data, Angles &a ) {
-+
-+
+  LabVectors l;
-+
-+
+  if( data.Z1leptons[0].charge > 0 ) {
-+
+    l.vecl1p = data.Z1leptons[0].vec;
-+
+    l.vecl1m = data.Z1leptons[1].vec;
-+
+  } else {
-+
+    l.vecl1p = data.Z1leptons[1].vec;
-+
+    l.vecl1m = data.Z1leptons[0].vec;
-+
+  }
-+
-+
+  if( data.Z2leptons[0].charge > 0 ) {
-+
+    l.vecl2p = data.Z2leptons[0].vec;
-+
+    l.vecl2m = data.Z2leptons[1].vec;
-+
+  } else {
-+
+    l.vecl2p = data.Z2leptons[1].vec;
-+
+    l.vecl2m = data.Z2leptons[0].vec;
-+
+  }
-+
-+
+  l.vecz1 = (l.vecl1p+l.vecl1m);
-+
+  l.vecz2 = (l.vecl2p+l.vecl2m);
-+
+  l.vec4l = (l.vecz1+l.vecz2);
-+
-+
+  getAngles( l, a );
-+
+}
+void getAngles( LabVectors &l,
+                Angles &a ) {
-+
+  // cout << "ANGLES z1: " << setw(12) << l.vecz1.Pt() << setw(12) << l.vecz1.Eta() << setw(12) << l.vecz1.Phi() << endl;
-+
+  // cout << "ANGLES z2: " << setw(12) << l.vecz2.Pt() << setw(12) << l.vecz2.Eta() << setw(12) << l.vecz2.Phi() << endl;
-+
+  // cout << "ANGLES 4l: " << setw(12) << l.vec4l.Pt() << setw(12) << l.vec4l.Eta() << setw(12) << l.vec4l.Phi() << endl;
-+
+  // cout << "ANGLES l1p:" << setw(12) << l.vecl1p.Pt() << setw(12) << l.vecl1p.Eta() << setw(12) << l.vecl1p.Phi() << endl;
-+
+  // cout << "ANGLES l1m:" << setw(12) << l.vecl1m.Pt() << setw(12) << l.vecl1m.Eta() << setw(12) << l.vecl1m.Phi() << endl;
-+
+  // cout << "ANGLES l2p:" << setw(12) << l.vecl2p.Pt() << setw(12) << l.vecl2p.Eta() << setw(12) << l.vecl2p.Phi() << endl;
-+
+  // cout << "ANGLES l2m:" << setw(12) << l.vecl2m.Pt() << setw(12) << l.vecl2m.Eta() << setw(12) << l.vecl2m.Phi() << endl;
+  //
+  // define the frames
+  // --------------------------------------------------------------------------------------------------
+  TVector3 vecpp_in_Z1coords_Xframe  = vecpp.Transform(rotation);
+  a.Phi1 = vecpp_in_Z1coords_Xframe.Phi();
-<
+  a.mz1 = (l.vecl1m+l.vecl1p).M();
-<
+  a.mz2 = (l.vecl2m+l.vecl2p).M();
-<
+  a.m4l = (l.vec4l).M();
->
+  //  a.mz1 = (l.vecl1m+l.vecl1p).M();
->
+  //  a.mz2 = (l.vecl2m+l.vecl2p).M();
->
+  //  a.m4l = (l.vec4l).M();
+};
-+
+void computeMelaAngles(TLorentzVector p4M11, int Z1_lept1Id,
-+
+                       TLorentzVector p4M12, int Z1_lept2Id,
-+
+                       TLorentzVector p4M21, int Z2_lept1Id,
-+
+                       TLorentzVector p4M22, int Z2_lept2Id,
-+
+                       float& costhetastar,
-+
+                       float& costheta1,
-+
+                       float& costheta2,
-+
+                       float& Phi,
-+
+                       float& Phi1){
-+
-+
+  //build Z 4-vectors
-+
+  TLorentzVector p4Z1 = p4M11 + p4M12;
-+
+  TLorentzVector p4Z2 = p4M21 + p4M22;
-+
-+
+  // Sort Z1 leptons so that:
-+
+  if ( (Z1_lept1Id*Z1_lept2Id<0 && Z1_lept1Id<0) || // for OS pairs: lep1 must be the negative one
-+
+       (Z1_lept1Id*Z1_lept2Id>0 && p4M11.Phi()<=p4M12.Phi()) //for SS pairs: use random deterministic convention
-+
+       ) {
-+
+    swap(p4M11, p4M12);
-+
+  }
-+
-+
+  // Same for Z2 leptons
-+
+  if ( (Z2_lept1Id*Z2_lept2Id<0 && Z2_lept1Id<0) ||
-+
+       (Z2_lept1Id*Z2_lept2Id>0 && p4M21.Phi()<=p4M22.Phi())
-+
+       ) {
-+
+    swap(p4M21, p4M22);
-+
+  }
-+
-+
-+
+  //build H 4-vectors
-+
+  TLorentzVector p4H = p4Z1 + p4Z2;
-+
-+
+  // -----------------------------------
-+
-+
+  //// costhetastar
-+
+  TVector3 boostX = -(p4H.BoostVector());
-+
+  TLorentzVector thep4Z1inXFrame( p4Z1 );
-+
+  TLorentzVector thep4Z2inXFrame( p4Z2 );
-+
+  thep4Z1inXFrame.Boost( boostX );
-+
+  thep4Z2inXFrame.Boost( boostX );
-+
+  TVector3 theZ1X_p3 = TVector3( thep4Z1inXFrame.X(), thep4Z1inXFrame.Y(), thep4Z1inXFrame.Z() );
-+
+  TVector3 theZ2X_p3 = TVector3( thep4Z2inXFrame.X(), thep4Z2inXFrame.Y(), thep4Z2inXFrame.Z() );
-+
+  costhetastar = theZ1X_p3.CosTheta();
-+
-+
+  //// --------------------------- costheta1
-+
+  TVector3 boostV1 = -(p4Z1.BoostVector());
-+
+  TLorentzVector p4M11_BV1( p4M11 );
-+
+  TLorentzVector p4M12_BV1( p4M12 );
-+
+  TLorentzVector p4M21_BV1( p4M21 );
-+
+  TLorentzVector p4M22_BV1( p4M22 );
-+
+  p4M11_BV1.Boost( boostV1 );
-+
+  p4M12_BV1.Boost( boostV1 );
-+
+  p4M21_BV1.Boost( boostV1 );
-+
+  p4M22_BV1.Boost( boostV1 );
-+
-+
+  TLorentzVector p4V2_BV1 = p4M21_BV1 + p4M22_BV1;
-+
+  //// costheta1
-+
+  costheta1 = -p4V2_BV1.Vect().Dot( p4M11_BV1.Vect() )/p4V2_BV1.Vect().Mag()/p4M11_BV1.Vect().Mag();
-+
-+
+  //// --------------------------- costheta2
-+
+  TVector3 boostV2 = -(p4Z2.BoostVector());
-+
+  TLorentzVector p4M11_BV2( p4M11 );
-+
+  TLorentzVector p4M12_BV2( p4M12 );
-+
+  TLorentzVector p4M21_BV2( p4M21 );
-+
+  TLorentzVector p4M22_BV2( p4M22 );
-+
+  p4M11_BV2.Boost( boostV2 );
-+
+  p4M12_BV2.Boost( boostV2 );
-+
+  p4M21_BV2.Boost( boostV2 );
-+
+  p4M22_BV2.Boost( boostV2 );
-+
-+
+  TLorentzVector p4V1_BV2 = p4M11_BV2 + p4M12_BV2;
-+
+  //// costheta2
-+
+  costheta2 = -p4V1_BV2.Vect().Dot( p4M21_BV2.Vect() )/p4V1_BV2.Vect().Mag()/p4M21_BV2.Vect().Mag();
-+
-+
+  //// --------------------------- Phi and Phi1 (old phistar1 - azimuthal production angle)
-+
+  //    TVector3 boostX = -(thep4H.BoostVector());
-+
+  TLorentzVector p4M11_BX( p4M11 );
-+
+  TLorentzVector p4M12_BX( p4M12 );
-+
+  TLorentzVector p4M21_BX( p4M21 );
-+
+  TLorentzVector p4M22_BX( p4M22 );
-+
-+
+  p4M11_BX.Boost( boostX );
-+
+  p4M12_BX.Boost( boostX );
-+
+  p4M21_BX.Boost( boostX );
-+
+  p4M22_BX.Boost( boostX );
-+
-+
+  TVector3 tmp1 = p4M11_BX.Vect().Cross( p4M12_BX.Vect() );
-+
+  TVector3 tmp2 = p4M21_BX.Vect().Cross( p4M22_BX.Vect() );
-+
-+
+  TVector3 normal1_BX( tmp1.X()/tmp1.Mag(), tmp1.Y()/tmp1.Mag(), tmp1.Z()/tmp1.Mag() );
-+
+  TVector3 normal2_BX( tmp2.X()/tmp2.Mag(), tmp2.Y()/tmp2.Mag(), tmp2.Z()/tmp2.Mag() );
-+
-+
+  //// Phi
-+
+  TLorentzVector p4Z1_BX = p4M11_BX + p4M12_BX;
-+
+  float tmpSgnPhi = p4Z1_BX.Vect().Dot( normal1_BX.Cross( normal2_BX) );
-+
+  float sgnPhi = tmpSgnPhi/fabs(tmpSgnPhi);
-+
+  Phi = sgnPhi * acos( -1.*normal1_BX.Dot( normal2_BX) );
-+
-+
-+
+  //////////////
-+
-+
+  TVector3 beamAxis(0,0,1);
-+
+  TVector3 tmp3 = (p4M11_BX + p4M12_BX).Vect();
-+
-+
+  TVector3 p3V1_BX( tmp3.X()/tmp3.Mag(), tmp3.Y()/tmp3.Mag(), tmp3.Z()/tmp3.Mag() );
-+
+  TVector3 tmp4 = beamAxis.Cross( p3V1_BX );
-+
+  TVector3 normalSC_BX( tmp4.X()/tmp4.Mag(), tmp4.Y()/tmp4.Mag(), tmp4.Z()/tmp4.Mag() );
-+
-+
+  //// Phi1
-+
+  float tmpSgnPhi1 = p4Z1_BX.Vect().Dot( normal1_BX.Cross( normalSC_BX) );
-+
+  float sgnPhi1 = tmpSgnPhi1/fabs(tmpSgnPhi1);
-+
+  Phi1 = sgnPhi1 * acos( normal1_BX.Dot( normalSC_BX) );
-+
-+
+}
-+
-+
+//----------------------------------------------------------------------------------------
-+
+void Angles::dump()
-+
+{
-+
+  cout << "angles: "
-+
+     << setw(12) << costheta1 << setw(12) << costheta2 << setw(12) << Phi
-+
+     << setw(12) << costhetastar << setw(12) << Phi1
-+
+     << endl;
-+
+}
-+

Diff Legend

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

Comparing UserCode/MitHzz4l/Angles/src/Angles.cc (file contents): Revision 1.1 by khahn, Thu Sep 8 13:33:18 2011 UTC vs. Revision 1.6 by dkralph, Mon Dec 17 12:35:18 2012 UTC

Diff Legend

Comparing UserCode/MitHzz4l/Angles/src/Angles.cc (file contents):
Revision 1.1 by khahn, Thu Sep 8 13:33:18 2011 UTC vs.
Revision 1.6 by dkralph, Mon Dec 17 12:35:18 2012 UTC