New class for dE/dx analysis (comparison with Bethe-Bloch, check of response function...

[u/mrichter/AliRoot.git] / JETAN / AliJetUnitArray.cxx

diff --git a/JETAN/AliJetUnitArray.cxx b/JETAN/AliJetUnitArray.cxx
index 353a7e6bbcbb186d63625d40ba4ce217d30756f8..5b55049f268597cd5343cb8b3fc692e9f815e89a 100644 (file)
--- a/JETAN/AliJetUnitArray.cxx
+++ b/JETAN/AliJetUnitArray.cxx
@@ -21,37 +21,230 @@
 // --
 //*-- Author: Sarah Blyth (LBL/UCT)
 // --
-// Revised Version for JETAN - 30/03/2006
+// Revised Version for JETAN 
 // -- Magali Estienne (IReS)
 
-#include "AliJetUnitArray.h"
+//#include <vector>
 
+#include <TVector3.h>
+#include <TLorentzVector.h>
+#include <TClonesArray.h>
 
-ClassImp(AliJetUnitArray)
+#include "AliJetUnitArray.h"
 
+ClassImp(AliJetUnitArray)
 
 AliJetUnitArray::AliJetUnitArray():
-    fUnitEnergy(0.),
-    fUnitEta(0.),
-    fUnitPhi(0.),
-    fUnitDeta(0.),
-    fUnitDphi(0.),
-    fUnitID(0),
-    fUnitNum(0),
-    fUnitClusterID(0),
-    fUnitFlag(kOutJet),
-    fUnitCutFlag(kPtSmaller),
-    fUnitSignalFlag(kGood),
-    fUnitDetectorFlag(kAll)
+  fUnitEnergy(0.0),
+  fUnitEta(0.0),
+  fUnitPhi(0.0),
+  fUnitDeta(0.),
+  fUnitDphi(0.),
+  fUnitID(0),
+  fUnitTrackID(0),
+  fUnitNum(0),
+  fUnitClusterID(0),
+  fUnitFlag(kOutJet),
+  fUnitCutFlag(kPtSmaller),
+  fUnitCutFlag2(kPtSmaller),
+  fUnitSignalFlag(kBad), 
+  fUnitDetectorFlag(kTpc),
+  fUnitPx(0.),
+  fUnitPy(0.),
+  fUnitPz(0.),
+  fUnitMass(0.),
+  fV(0),
+  fVc(0),
+  fVn(0),
+  fVet(0)
 {
   // Default constructor
 }  
 
+AliJetUnitArray::AliJetUnitArray(Int_t absId, Int_t esdId, Float_t eta, Float_t phi, Float_t en, Float_t Deta, Float_t Dphi, AliJetFinderUnitDetectorFlagType_t det, AliJetFinderUnitFlagType_t inout, AliJetFinderUnitCutFlagType_t cut, AliJetFinderUnitCutFlagType_t cut2, AliJetFinderUnitSignalFlagType_t signal,Float_t mass, Int_t clusId):
+  fUnitEnergy(en),
+  fUnitEta(eta),
+  fUnitPhi(phi),
+  fUnitDeta(Deta),
+  fUnitDphi(Dphi),
+  fUnitID(absId),
+  fUnitTrackID(esdId),
+  fUnitNum(0),
+  fUnitClusterID(clusId),
+  fUnitFlag(inout),
+  fUnitCutFlag(cut),
+  fUnitCutFlag2(cut2),
+  fUnitSignalFlag(signal), 
+  fUnitDetectorFlag(det),
+  fUnitPx(0.),
+  fUnitPy(0.),
+  fUnitPz(0.),
+  fUnitMass(mass),
+  fV(0),
+  fVc(0),
+  fVn(0),
+  fVet(0)
+{
+  //abs ID (in a eta,phi grid, track ID in ESD, eta, phi, energy, px, py, pz, Deta, Dphi, detector flag, in/out jet, mass
+
+  // Constructor 2
+}
+
+AliJetUnitArray::AliJetUnitArray(Int_t absId, Int_t esdId, Float_t eta, Float_t phi, Float_t en, Float_t px, Float_t py, Float_t pz, Float_t Deta, Float_t Dphi, AliJetFinderUnitDetectorFlagType_t det, AliJetFinderUnitFlagType_t inout, AliJetFinderUnitCutFlagType_t cut, Float_t mass, Int_t clusId):
+  fUnitEnergy(en),
+  fUnitEta(eta),
+  fUnitPhi(phi),
+  fUnitDeta(Deta),
+  fUnitDphi(Dphi),
+  fUnitID(absId),
+  fUnitTrackID(esdId),
+  fUnitNum(0),
+  fUnitClusterID(clusId),
+  fUnitFlag(inout),
+  fUnitCutFlag(cut),
+  fUnitCutFlag2(kPtSmaller),
+  fUnitSignalFlag(kBad),
+  fUnitDetectorFlag(det),
+  fUnitPx(px),
+  fUnitPy(py),
+  fUnitPz(pz),
+  fUnitMass(mass),
+  fV(0),
+  fVc(0),
+  fVn(0),
+  fVet(0)
+{
+  // Constructor 2
+}
+ 
+//------------------------------------------------------------------------
 AliJetUnitArray::~AliJetUnitArray()
 {
   // Destructor 
 }
-       
+
+//------------------------------------------------------------------------
+void AliJetUnitArray::SetUnitSignalFlagC(Bool_t init, AliJetFinderUnitSignalFlagType_t flag)
+{
+  // Set signal flag of the charged particle
+  if(init){
+    if(!fVc.empty())
+      fVc.clear();
+  }
+  else fVc.push_back(flag);
+}
+
+//------------------------------------------------------------------------
+void AliJetUnitArray::SetUnitSignalFlagN(Bool_t init, AliJetFinderUnitSignalFlagType_t flag)
+{
+  // Set signal flag of the neutral cell
+  if(init){
+    if(!fVn.empty())
+      fVn.clear();
+  }
+  else fVn.push_back(flag);
+}
+
+//------------------------------------------------------------------------
+void AliJetUnitArray::SetUnitEtN(Bool_t init, Float_t et)
+{
+  // Set transverse energy of the neutral cell
+  if(init){
+    if(!fVet.empty())
+      fVet.clear();
+  }
+  else fVet.push_back(et);
+}
+
+
+//------------------------------------------------------------------------
+void AliJetUnitArray::SetUnitPxPyPz(Bool_t init, vector<Float_t> v3)
+{
+  // Set momentum components of the charged particle
+  if(init)
+    {
+      if(!fV.empty()){
+       fV.clear();
+      }
+    }
+  else{
+    fV.push_back(v3);
+  }
+}
+
+//------------------------------------------------------------------------
+Bool_t AliJetUnitArray::GetUnitSignalFlagC(Int_t ind, AliJetFinderUnitSignalFlagType_t &flagc)
+{
+  // Get signal flag of the charged particle
+  if(ind <= (Int_t)fVc.size())
+    {
+      flagc = (AliJetFinderUnitSignalFlagType_t)fVc[ind];
+      return kTRUE;
+    }
+  else return kFALSE;
+}
+
+//------------------------------------------------------------------------
+Bool_t AliJetUnitArray::GetUnitSignalFlagN(Int_t ind, AliJetFinderUnitSignalFlagType_t &flagn)
+{
+  // Get signal flag of the neutral cell
+  if(ind <= (Int_t)fVn.size())
+    {
+      flagn = (AliJetFinderUnitSignalFlagType_t)fVn[ind];
+      return kTRUE;
+    }
+  else return kFALSE;
+}
+
+//------------------------------------------------------------------------
+Bool_t AliJetUnitArray::GetUnitEtN(Int_t ind, Float_t &et)
+{
+  // Get transverse energy of the neutral cell
+  if(ind <= (Int_t)fVet.size())
+    {
+      et = (Float_t)fVet[ind];
+      return kTRUE;
+    }
+  else return kFALSE;
+}
+
+//------------------------------------------------------------------------
+Bool_t AliJetUnitArray::GetUnitPxPyPz(Int_t ind, Float_t &px, Float_t &py, Float_t &pz)
+{
+  // Get momentum components of the charged particle
+  if(ind <= (Int_t)fV.size())
+    {
+      px = (Float_t)fV[ind][0];
+      py = (Float_t)fV[ind][1];
+      pz = (Float_t)fV[ind][2];
+      return kTRUE;
+    }
+  else return kFALSE;
+}
+
+//------------------------------------------------------------------------
+Bool_t AliJetUnitArray::GetUnitPxPyPzE(Int_t ind, Float_t &px, Float_t &py, Float_t &pz, Float_t &en)
+{
+// Get 4-momentum components of the charged particle
+  if(ind <= (Int_t)fV.size())
+    {
+      px = (Float_t)fV[ind][0];
+      py = (Float_t)fV[ind][1];
+      pz = (Float_t)fV[ind][2];
+      en = TMath::Sqrt(px*px+py*py+pz*pz);
+      return kTRUE;
+    }
+  else return kFALSE;
+}
+
+//------------------------------------------------------------------------
+Float_t  AliJetUnitArray::EtaToTheta(Float_t arg) const
+{
+  // Eta to theta transformation
+  return 2.*atan(exp(-arg));
+}
+
+//------------------------------------------------------------------------
 Bool_t AliJetUnitArray::operator>(AliJetUnitArray unit) const
 {
   // Greater than operator used by sort
@@ -61,6 +254,7 @@ Bool_t AliJetUnitArray::operator>(AliJetUnitArray unit) const
     return kFALSE;
 }
 
+//------------------------------------------------------------------------
 Bool_t AliJetUnitArray::operator<( AliJetUnitArray unit) const
 {
   // Less than operator used by sort
@@ -70,6 +264,7 @@ Bool_t AliJetUnitArray::operator<( AliJetUnitArray unit) const
     return kFALSE;
 }
 
+//------------------------------------------------------------------------
 Bool_t AliJetUnitArray::operator==( AliJetUnitArray unit) const
 {
   // equality operator used by sort