STEER/AliESDCaloCluster.h

   1 #ifndef ALIESDCALOCLUSTER_H
   2 #define ALIESDCALOCLUSTER_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7 /* $Log $ */
   8
   9 //-------------------------------------------------------------------------
  10 //                          Class AliESDCaloCluster
  11 //   This is the class to deal with during the physics analysis of data
  12 //
  13 //   New container for calorimeter clusters, which are the effective
  14 //   "tracks" for calorimeter detectors.  Can be used by PHOS and EMCAL
  15 //
  16 //     J.L. Klay (LLNL)
  17 //-------------------------------------------------------------------------
  18
  19 #include <TObject.h>
  20 #include "AliPID.h"
  21
  22 class TLorentzVector;
  23
  24 class AliESDCaloCluster : public TObject {
  25
  26 public:
  27
  28   AliESDCaloCluster();
  29   AliESDCaloCluster(const AliESDCaloCluster& clus);
  30   AliESDCaloCluster & operator=(const AliESDCaloCluster& source);
  31   virtual ~AliESDCaloCluster();
  32
  33   void SetID(Int_t id) {fID = id;}
  34   Int_t GetID() const {return fID;}
  35
  36   enum ClusterType {kPseudoCluster, kClusterv1};//Two types of clusters stored
  37                                                 //in EMCAL.
  38   void SetClusterType(Int_t type) { fClusterType = type; }
  39   Int_t GetClusterType() const {return fClusterType; }
  40
  41   void SetEMCAL(Bool_t emc) { fEMCALCluster = emc;}
  42   Bool_t IsEMCAL() const {return fEMCALCluster;}
  43
  44   void SetPHOS(Bool_t phos) { fPHOSCluster = phos;}
  45   Bool_t IsPHOS() const {return fPHOSCluster;}
  46
  47   void SetGlobalPosition(const Float_t *pos) {
  48     fGlobalPos[0] = pos[0]; fGlobalPos[1] = pos[1]; fGlobalPos[2] = pos[2];
  49   }
  50   void GetGlobalPosition(Float_t *pos) const {
  51     pos[0] = fGlobalPos[0]; pos[1] = fGlobalPos[1]; pos[2] = fGlobalPos[2];
  52   }
  53
  54   void SetClusterEnergy(Float_t ene) { fEnergy = ene;}
  55   Float_t GetClusterEnergy() const   { return fEnergy;}
  56
  57   void SetClusterDisp(Float_t disp)  { fDispersion = disp; }
  58   Float_t GetClusterDisp() const     { return fDispersion; }
  59
  60   void SetClusterChi2(Float_t chi2)  { fChi2 = chi2; }
  61   Float_t GetClusterChi2() const     { return fChi2; }
  62
  63   void SetPid(const Float_t *p);
  64   Float_t *GetPid() {return fPID;}
  65
  66   void SetPrimaryIndex(Int_t primary)     { fPrimaryIndex = primary; }
  67   Int_t GetPrimaryIndex() const           { return fPrimaryIndex; }
  68
  69   void SetM20(Float_t m20)                { fM20 = m20; }
  70   Float_t GetM20() const                  { return fM20; }
  71
  72   void SetM02(Float_t m02)                { fM02 = m02; }
  73   Float_t GetM02() const                  { return fM02; }
  74
  75   void SetM11(Float_t m11)                { fM11 = m11; }
  76   Float_t GetM11() const                  { return fM11; }
  77
  78   void SetNExMax(UShort_t nExMax)         { fNExMax = nExMax; }
  79   UShort_t GetNExMax() const              { return fNExMax; }
  80
  81   void SetEmcCpvDistance(Float_t dEmcCpv) { fEmcCpvDistance = dEmcCpv; }
  82   Float_t GetEmcCpvDistance() const       { return fEmcCpvDistance; }
  83
  84   void SetNumberOfPrimaries(Int_t nprim)      { fNumberOfPrimaries = nprim; }
  85   Int_t GetNumberOfPrimaries() const         { return fNumberOfPrimaries; }
  86
  87   void SetListOfPrimaries(UShort_t *prim)       { fListOfPrimaries = prim;}
  88   UShort_t *GetListOfPrimaries() const          { return fListOfPrimaries;}
  89
  90   void SetNumberOfDigits(Int_t ndig)      { fNumberOfDigits = ndig; }
  91   Int_t GetNumberOfDigits() const         { return fNumberOfDigits; }
  92
  93   void SetDigitAmplitude(UShort_t *adc)   { fDigitAmplitude = adc;}
  94   UShort_t *GetDigitAmplitude() const     { return fDigitAmplitude;}
  95
  96   void SetDigitTime(UShort_t *time)       { fDigitTime = time;}
  97   UShort_t *GetDigitTime() const          { return fDigitTime;}
  98
  99   void SetDigitIndex(UShort_t *digit)     { fDigitIndex = digit;}
 100   UShort_t *GetDigitIndex() const         { return fDigitIndex; }
 101
 102   void GetMomentum(TLorentzVector& p);
 103
 104 protected:
 105
 106   Int_t     fID;               // Unique Id of the cluster
 107   Int_t     fClusterType;      // Flag for different clustering versions
 108   Bool_t    fEMCALCluster;     // Is this is an EMCAL cluster?
 109   Bool_t    fPHOSCluster;      // Is this is a PHOS cluster?
 110   Float_t   fGlobalPos[3];     // position in global coordinate system
 111   Float_t   fEnergy;           // energy measured by calorimeter
 112   Float_t   fDispersion;       // cluster dispersion, for shape analysis
 113   Float_t   fChi2;             // chi2 of cluster fit
 114   Float_t   fPID[AliPID::kSPECIESN]; //"detector response probabilities" (for the PID)
 115   Int_t     fPrimaryIndex;     // primary track number associated with this cluster
 116   Float_t   fM20;              // 2-nd moment along the main eigen axis
 117   Float_t   fM02;              // 2-nd moment along the second eigen axis
 118   Float_t   fM11;              // 2-nd mixed moment Mxy
 119   UShort_t  fNExMax ;          // number of (Ex-)maxima before unfolding
 120   Float_t   fEmcCpvDistance;   // the distance from PHOS EMC rec.point to the closest CPV rec.point
 121
 122   Int_t     fNumberOfPrimaries;   // number of primaries that generated the cluster
 123   UShort_t*   fListOfPrimaries;   //[ fNumberOfPrimaries] list of primaries that generated the cluster
 124
 125   Int_t     fNumberOfDigits;   // number of calorimeter digits in cluster
 126                                // Very important! The streamer needs to
 127                                // know how big these arrays are for
 128                                // each event that is written out:
 129   UShort_t* fDigitAmplitude;   //[fNumberOfDigits] digit energy (integer units)
 130   UShort_t* fDigitTime;        //[fNumberOfDigits] time of this digit (integer units)
 131   UShort_t* fDigitIndex;       //[fNumberOfDigits] calorimeter digit index
 132
 133   ClassDef(AliESDCaloCluster,1)  //ESDCaloCluster
 134 };
 135
 136 #endif
 137