- Use UnitArray to store track and emcal information.

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

diff --git a/JETAN/AliJetUnitArray.h b/JETAN/AliJetUnitArray.h
index cc0f0d0856a583c4b72d8b79d7e61b9c0399663d..20e810c2f45dcfe577d6ec35a1f89a900c8670e8 100644 (file)
--- a/JETAN/AliJetUnitArray.h
+++ b/JETAN/AliJetUnitArray.h
@@ -1,19 +1,15 @@
-#ifndef ALIUA1JETFINDERUNIT_H
-#define ALIUA1JETFINDERUNIT_H
+#ifndef ALIJETUNITARRAY_H
+#define ALIJETUNITARRAY_H
 
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  *  *  * See cxx source for full Copyright notice     */
 
-/* $Id$ */
-
-//======================================================================
-// Revised Version for JETAN - 08/11/2006
-// Functions added
-// Author: magali.estienne@ires.in2p3.fr
-//======================================================================
+//------------------------------------------------------------------
 //  Unit used by UA1 algorithm
-//
-//*-- Author: Sarah Blyth (LBL/UCT)
+//  Authors: Sarah Blyth (LBL/UCT)
+//           Magali Estienne (IReS) (new version for JETAN)
+//------------------------------------------------------------------
+
 
 #include <TObject.h>
 #include "AliJetFinderTypes.h"
@@ -22,6 +18,7 @@ class AliJetUnitArray : public TObject
 {
  public:
   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);
   ~AliJetUnitArray();
 
   // Setter
@@ -31,6 +28,7 @@ class AliJetUnitArray : public TObject
   void SetUnitDeta(Float_t deta)      {fUnitDeta = deta;} 
   void SetUnitDphi(Float_t dphi)      {fUnitDphi = dphi;}
   void SetUnitID(Int_t id)            {fUnitID = id;}
+  void SetUnitTrackID(Int_t esdid)    {fUnitTrackID = esdid;}
   void SetUnitEntries(Int_t num)      {fUnitNum = num;}
   void SetUnitClusterID(Int_t id)     {fUnitClusterID = id;}
   void SetUnitFlag(AliJetFinderUnitFlagType_t flag)    
@@ -49,6 +47,8 @@ class AliJetUnitArray : public TObject
   { 
     fUnitDetectorFlag = detectorflag; 
   } 
+  void SetUnitPxPyPz(Double_t *pxyz)  {fUnitPx = pxyz[0]; fUnitPy = pxyz[1]; fUnitPz = pxyz[2];}
+  void SetUnitMass(Float_t mass) {fUnitMass = mass;}
 
   // Getter
   Float_t GetUnitEnergy() const            {return fUnitEnergy;}
@@ -57,8 +57,12 @@ class AliJetUnitArray : public TObject
   Float_t GetUnitDeta() const              {return fUnitDeta;}
   Float_t GetUnitDphi() const              {return fUnitDphi;}         
   Int_t   GetUnitID() const                {return fUnitID;}
+  Int_t   GetUnitTrackID() const             {return fUnitTrackID;}
   Int_t   GetUnitEntries() const           {return fUnitNum;}
   Int_t   GetUnitClusterID() const         {return fUnitClusterID;}
+  Float_t GetUnitMass() const              {return fUnitMass;}
+  Bool_t  GetUnitPxPyPz(Double_t* p) const {p[0]=fUnitPx; p[1]=fUnitPy; p[2]=fUnitPz; return kTRUE;}
+
   AliJetFinderUnitFlagType_t GetUnitFlag() const     
   {
          return fUnitFlag;
@@ -76,25 +80,31 @@ class AliJetUnitArray : public TObject
          return fUnitDetectorFlag;
   }
 
+
   Bool_t operator>  ( AliJetUnitArray unit1) const;
   Bool_t operator<  ( AliJetUnitArray unit1) const;
   Bool_t operator== ( AliJetUnitArray unit1) const;
 
  protected:
-  Float_t         fUnitEnergy;        // Energy of the unit 
-  Float_t         fUnitEta;           // Eta of the unit
-  Float_t         fUnitPhi;           // Phi of the unit
-  Float_t         fUnitDeta;          // Delta Eta of the unit
-  Float_t         fUnitDphi;          // Delta Phi of the unit
-  Int_t           fUnitID;            // ID of the unit
-  Int_t           fUnitNum;           // number of units
-  Int_t           fUnitClusterID;     // ID of the unit
-  AliJetFinderUnitFlagType_t         fUnitFlag; //Flag of the unit
-  AliJetFinderUnitCutFlagType_t      fUnitCutFlag; //Flag of the unit
-  AliJetFinderUnitSignalFlagType_t   fUnitSignalFlag; //Flag of the unit
+  Float_t         fUnitEnergy;                          // Energy of the unit 
+  Float_t         fUnitEta;                             // Eta of the unit
+  Float_t         fUnitPhi;                             // Phi of the unit
+  Float_t         fUnitDeta;                            // Delta Eta of the unit
+  Float_t         fUnitDphi;                            // Delta Phi of the unit
+  Int_t           fUnitID;                              // ID of the unit
+  Int_t           fUnitTrackID;                         // ID of the unit
+  Int_t           fUnitNum;                             // number of units
+  Int_t           fUnitClusterID;                       // ID of the unit
+  AliJetFinderUnitFlagType_t         fUnitFlag;         // Flag of the unit
+  AliJetFinderUnitCutFlagType_t      fUnitCutFlag;      // Flag of the unit
+  AliJetFinderUnitSignalFlagType_t   fUnitSignalFlag;   // Flag of the unit
   AliJetFinderUnitDetectorFlagType_t fUnitDetectorFlag; // Detector flag of the unit
+  Float_t         fUnitPx;                              // Px of charged track
+  Float_t         fUnitPy;                              // Py of charged track
+  Float_t         fUnitPz;                              // Pz of charged track
+  Float_t         fUnitMass;                            // Mass of charged particle
 
-  ClassDef(AliJetUnitArray,3)
+  ClassDef(AliJetUnitArray,1)
 };
 
 #endif