AliReducedJetParticle.h

Go to the documentation of this file.

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

#include <TObjArray.h>
#include "AliReducedMatchedTrack.h"

class TLorentzVector;

namespace HighPtTracks {

class AliReducedJetParticle : public TObject {
public:
  AliReducedJetParticle();
  AliReducedJetParticle(double px, double py, double pz, double e, int pdgcode);
  AliReducedJetParticle(const AliReducedJetParticle &ref);
  AliReducedJetParticle &operator=(const AliReducedJetParticle &ref);
  virtual ~AliReducedJetParticle();

  void FillLorentzVector(TLorentzVector &ref) const;

  double GetDistanceToJetMainAxis() const { return fDr; }

  int GetPdgCode() const { return fPdgCode; }

  bool IsReconstructed() const { return fMatchedTrackContainer->GetEntries() > 0; }

  int GetNumberOfMatchedTracks() const { return fMatchedTrackContainer->GetEntries(); }

  double GetDeltaPt(int itrk = 0) const;

  unsigned short GetNumberOfTPCTrackReferences() const { return fNTPCTrackReferences; }

  TObjArray *GetMatchedTracks() const { return fMatchedTrackContainer; }

  AliReducedMatchedTrack *GetMatchedTrack(int itrk){
    if(itrk >= fMatchedTrackContainer->GetEntries()) return NULL;
    return static_cast<AliReducedMatchedTrack *>(fMatchedTrackContainer->At(itrk));
  }

  void SetKine(double px, double py, double pz, double e){
    fPx = px;
    fPy = py;
    fPz = pz;
    fE = e;
  }

  void SetDistanceToMainJetAxis(double dr) { fDr = dr; }

  void SetPdgCode(int pdg) { fPdgCode = pdg; }

  void AddMatchedTrack(AliReducedMatchedTrack *trk);

  void SetNumberOfTPCtrackReferences(unsigned short nref) { fNTPCTrackReferences = nref; }

private:
  double      fPx;              
  double      fPy;              
  double      fPz;              
  double      fE;               
  double      fDr;              
  int       fPdgCode;           
  unsigned short fNTPCTrackReferences;  
  TObjArray *fMatchedTrackContainer;    

  ClassDef(AliReducedJetParticle, 3);
};

} /* namespace HighPtTracks */

#endif /* ALIREDUCEDJETPARTICLE_H */