AliReducedGeneratedParticle.h

Go to the documentation of this file.

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

#include <TObject.h>

class TVector3;

namespace HighPtTracks {

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

  Int_t GetID() const { return fParticleID; }
  Int_t GetPdgCode() const { return fPDGCode; }
  Double_t GetParticleEnergy() const { return fEnergy; }

  Double_t Pt() const;
  Double_t Eta() const;
  Double_t Phi() const;

  void FillMomentumVector(TVector3 &) const;

  void GetMomentumVector(Double_t &px, Double_t &py, Double_t &pz) const {
    px = fPVec[0];
    py = fPVec[1];
    pz = fPVec[2];
  }

  void Set(Int_t id, Int_t pdgcode, Double_t px, Double_t py, Double_t pz, Double_t energy){
    fParticleID = id;
    fPDGCode = pdgcode;
    fPVec[0] = px;
    fPVec[1] = py;
    fPVec[2] = px;
    fEnergy = energy;
  }
  void SetID(Int_t id) { fParticleID = id; }
  void SetPDGCode(Int_t pdgcode) { fPDGCode = pdgcode; }
  void SetPVector(Double_t px, Double_t py, Double_t pz){
    fPVec[0] = px;
    fPVec[1] = py;
    fPVec[2] = px;
  }
  void SetEnergy(Double_t energy) { fEnergy = energy; }

protected:
  Int_t                       fParticleID;          
  Int_t                       fPDGCode;             
  Double_t                    fPVec[3];             
  Double_t                    fEnergy;              

  ClassDef(AliReducedGeneratedParticle, 1);
};

} /* namespace HighPtTracks */

#endif /* ALIREDUCEDGENERATEDPARTICLE_H */