Changeset 264 in svn for trunk/Utilities/ExRootAnalysis/src/BlockClasses.cc

Timestamp:

Feb 11, 2009, 10:22:30 AM (16 years ago)

Author:

Xavier Rouby

Message:

first test 2.0

File:

• trunk/Utilities/ExRootAnalysis/src/BlockClasses.cc (modified) (4 diffs)

trunk/Utilities/ExRootAnalysis/src/BlockClasses.cc

@@ -11 +11 @@
  *  to preserve mother-dautherlinks between particles.
  *
- *  $Date: 2009-02-02 11:32:01 $
- *  $Revision: 1.7 $
+ *  $Date: 2009-02-11 09:19:14 $
+ *  $Revision: 1.8 $
  *
  *  
@@ -22 +22 @@
 #include "BlockCompare.h"
 #include "TLorentzVector.h"
+#include <cmath>
 
 TCompare *TRootGenParticle::fgCompare = 0;
@@ -38 +39 @@
 TCompare *TRootRomanPotHits::fgCompare =0;
 TCompare *TRootTrigger::fgCompare =0;
+//TCompare *D_CaloTower::fgCompare=0;
+TCompare *D_Track::fgCompare=0;
 
 void TRootParticle::Set(const TLorentzVector& momentum) {
@@ -61 +64 @@
 }
 
+void TRootParticle::SetEtaPhiEET(const float eta, const float phi, const float e, const float et) {
+   Eta = eta; Phi = phi; E = e; PT = et;
+   //float theta = 2* atan(exp(-Eta);
+   Px = PT*cos(Phi);
+   Py = PT*sin(Phi);
+}
 
+
+void TRootGenParticle::setFractions() {
+  switch(abs(PID)) {
+    default:  _Fem = 0; _Fhad=1; break;
+    case(pE):
+    case(pGAMMA):
+    case(pPI0):
+        _Fem = 1; _Fhad=0; break;
+    case(pNU1):
+    case(pNU2):
+    case(pNU3):
+    case(pMU):
+        _Fem =0; _Fhad=0; break;
+    case(pK0S):
+    case(pLAMBDA):
+        _Fem=0.3; _Fhad=0.7; break;
+  }
+  _initialised=true;
+}
+
+    float Eta, Phi; // (eta,phi) at the beginning of the track
+    float EtaOuter, PhiOuter; // (eta,phi) at the end of the track
+    float PT, E, Px, Py, Pz; 
+
+D_Track::D_Track() :
+        Eta(UNDEFINED), Phi(UNDEFINED), EtaOuter(UNDEFINED), PhiOuter(UNDEFINED), 
+        PT(UNDEFINED), E(UNDEFINED), Px(UNDEFINED), Py(UNDEFINED), Pz(UNDEFINED) {}
+
+D_Track::D_Track(const float inEta, const float inPhi, const float outEta, const float outPhi, const float pt) :
+        Eta(inEta), Phi(inPhi), EtaOuter(outEta), PhiOuter(outPhi), 
+        PT(pt) {
+        TLorentzVector p; p.SetPtEtaPhiE(PT,Eta,Phi,PT*cosh(Eta));
+        Px = p.Px(); Py = p.Py(); Pz = p.Pz();  E=p.E();
+}
+
+D_Track::D_Track(const D_Track& track) :
+        Eta(track.Eta), Phi(track.Phi), 
+        EtaOuter(track.EtaOuter), PhiOuter(track.PhiOuter), 
+        PT(track.PT), E(track.E), Px(track.Px), Py(track.Py), Pz(track.Pz) {}
+
+D_Track& D_Track::operator=(const D_Track& track) {
+  if(this == &track) return *this;
+        Eta = track.Eta;             Phi = track.Phi; 
+        EtaOuter = track.EtaOuter;   PhiOuter = track.PhiOuter; 
+        PT = track.PT; E = track.E;  Px = track.Px;  Py= track.Py; Pz =track.Pz; 
+  return *this;
+}
+
+void D_Track::Set(const float inEta, const float inPhi, const float outEta, const float outPhi, const float pt) {
+        Eta= inEta; Phi=inPhi;  EtaOuter=outEta; PhiOuter=outPhi; PT=pt; 
+        TLorentzVector p; p.SetPtEtaPhiE(PT,Eta,Phi,PT*cosh(Eta));
+        Px = p.Px(); Py = p.Py(); Pz = p.Pz();  E=p.E();
+}
+
+const TLorentzVector D_Track::GetFourVector() const {
+        TLorentzVector v;
+        v.SetPtEtaPhiE(PT,Eta,Phi,PT*cosh(Eta));
+        return TLorentzVector(v);
+}
+
+TRootCalo::TRootCalo() :
+        E_em(UNDEFINED), E_had(UNDEFINED), ET(UNDEFINED) {}
+        //Eta(UNDEFINED), Phi(UNDEFINED), E_em(UNDEFINED), E_had(UNDEFINED), E(UNDEFINED), ET(UNDEFINED) {}
+
+TRootCalo::TRootCalo(const TRootCalo& cal) {
+        E_em=cal.E_em; E_had=cal.E_had;  ET=cal.ET;
+        //Eta =cal.Eta;  Phi=cal.Phi; E_em=cal.E_em; E_had=cal.E_had;  E=cal.E;   ET=cal.ET;
+}
+
+TRootCalo& TRootCalo::operator=(const TRootCalo& cal) {
+   if(this==&cal) return *this;
+        Eta =cal.Eta;  Phi=cal.Phi; E_em=cal.E_em; E_had=cal.E_had;  E=cal.E;   ET=cal.ET;
+   return *this;
+}
+
+void TRootCalo::set(const D_CaloTower& cal) {
+        Eta =cal.getEta(); Phi=cal.getPhi(); 
+        E_em=cal.getEem(); E_had=cal.getEhad();
+        E   =cal.getE();   ET   =cal.getET();
+        EtaCalo = cal.getEta();
+        PhiCalo = cal.getPhi();
+}
+
+
+TRootTracks::TRootTracks(const D_Track& track) :
+   E(track.getE()), Px(track.getPx()), Py(track.getPy()), Pz(track.getPz()),
+   Eta(track.getEta()), Phi(track.getPhi()),
+   /*Etaout(track.getEtaOuter()),*/ EtaCalo(track.getEtaOuter()), /*Phiout(track.getPhiOuter()),*/ PhiCalo(track.getPhiOuter()) {}
+