AliReducedEmcalCluster.h

Go to the documentation of this file.

#ifndef ALIREDUCEDEMCALCLUSTER_H
#define ALIREDUCEDEMCALCLUSTER_H
/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <TObject.h>

class TArrayD;
class TLorentzVector;
class TObjArray;

namespace HighPtTracks {

class AliReducedClusterParticle : public TObject {
public:
  AliReducedClusterParticle();
  AliReducedClusterParticle(Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t energy);
  virtual ~AliReducedClusterParticle();

  void FillLorentzVector(TLorentzVector &target) const;
  Int_t GetPdgCode() const { return fPdg; }

  void SetMomentum(Double_t px, Double_t py, Double_t pz){
    fPvec[0]= px;
    fPvec[1]= py;
    fPvec[2]= pz;
  }
  void SetEnergy(Double_t energy) { fEnergy = energy; }
  void SetPdgCode(Int_t pdg) { fPdg = pdg; }

protected:
  Int_t               fPdg;               
  Double_t            fPvec[3];           
  Double_t            fEnergy;            

  ClassDef(AliReducedClusterParticle, 1);
};

class AliReducedEmcalCluster: public TObject {
public:
  AliReducedEmcalCluster();
  AliReducedEmcalCluster(Int_t id, Float_t energy, Float_t eta, Float_t phi, Float_t m02, Float_t m20);
  AliReducedEmcalCluster(const AliReducedEmcalCluster &ref);
  AliReducedEmcalCluster &operator=(const AliReducedEmcalCluster &ref);
  virtual ~AliReducedEmcalCluster();
  void Copy(TObject &target) const;

  Int_t           GetClusterID() const        { return fClusterID; }
  Float_t         GetClusterEnergy() const    { return fEnergy; }
  Float_t         GetEta() const              { return fEta; }
  Float_t         GetPhi() const              { return fPhi; }
  Float_t         GetM02() const              { return fM02; }
  Float_t         GetM20() const              { return fM20; }
  void            FillCellEnergies(TArrayD &target);
  TObjArray      *GetClusterContributors() const { return fContributors; }

  void            SetClusterID(Int_t id)                                          { fClusterID = id; }
  void            SetClusterEnergy(Float_t energy)                                { fEnergy = energy; }
  void            SetClusterPosition(Float_t eta, Float_t phi)                    { fEta = eta; fPhi = phi; }
  void            SetShowerShapeParameters(Float_t m02, Float_t m20)              { fM02 = m02; fM20 = m20; }
  void            SetLeadingCellEnergies(Double_t e1, Double_t e2, Double_t e3)   {
    fCellEnergies[0] = e1;
    fCellEnergies[1] = e2;
    fCellEnergies[2] = e3;
  }
  void  Set(Int_t id, Float_t energy, Float_t eta, Float_t phi, Float_t m02, Float_t m20){
    fClusterID = id;
    fEnergy = energy;
    fEta = eta;
    fPhi = phi;
    fM02 = m02;
    fM20 = m20;
  }
  void AddTrueContributor(Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t energy);
protected:
  Int_t                     fClusterID;             
  Float_t                   fEnergy;                
  Float_t                   fEta;                   
  Float_t                   fPhi;                   
  Float_t                   fM02;                   
  Float_t                   fM20;                   
  Float_t                   fCellEnergies[3];       
  TObjArray                 *fContributors;         

  ClassDef(AliReducedEmcalCluster, 1);
};

} /* namespace HighPtTracks */

#endif /* ALIREDUCEDEMCALCLUSTER_H */