[u/mrichter/AliRoot.git] / PWG / EMCAL / AliParticleContainer.h

#ifndef AliParticleContainer_H
#define AliParticleContainer_H

//
// container with name, TClonesArray and cuts for particles
//

class AliVEvent;
class AliVParticle;

#include "AliEmcalContainer.h"

class AliParticleContainer : public AliEmcalContainer {
 public:
  AliParticleContainer();
  AliParticleContainer(const char *name); 
  virtual ~AliParticleContainer(){;}

  void  SetParticlePtCut(Double_t cut)                    { fParticlePtCut = cut ; }
  void  SetParticleEtaLimits(Double_t min, Double_t max)  { fParticleMaxEta = max ; fParticleMinEta = min ; }
  void  SetParticlePhiLimits(Double_t min, Double_t max)  { fParticleMaxPhi = max ; fParticleMinPhi = min ; }
  void  SetTrackBitMap(UInt_t m)                          { fTrackBitMap     = m ; }
  void  SetMCTrackBitMap(UInt_t m)                        { fMCTrackBitMap   = m ; }
  void  SetMinMCLabel(Int_t s)                            { fMinMCLabel      = s ; }

  Double_t                    GetParticlePtCut()                        const   { return fParticlePtCut; }
  Double_t                    GetParticleEtaMin()                       const   { return fParticleMinEta; }
  Double_t                    GetParticleEtaMax()                       const   { return fParticleMaxEta; }
  Double_t                    GetParticlePhiMin()                       const   { return fParticleMinPhi; }
  Double_t                    GetParticlePhiMax()                       const   { return fParticleMaxPhi; }
  AliVParticle               *GetLeadingParticle(const char* opt="")         ;
  AliVParticle               *GetParticle(Int_t i)                      const;
  AliVParticle               *GetAcceptParticle(Int_t i)                const;
  AliVParticle               *GetParticleWithLabel(Int_t lab)           const;
  AliVParticle               *GetAcceptParticleWithLabel(Int_t lab)     const;
  AliVParticle               *GetNextAcceptParticle(Int_t i=-1)              ;
  AliVParticle               *GetNextParticle(Int_t i=-1)                    ;
  void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
  Bool_t                      AcceptParticle(AliVParticle         *vp)  const;
  Int_t                       GetNParticles()                           const   {return GetNEntries();}
  void                        SetClassName(const char *clname);

 protected:
  Double_t                    fParticlePtCut;                 // cut on particle pt
  Double_t                    fParticleMinEta;                // cut on particle eta
  Double_t                    fParticleMaxEta;                // cut on particle eta
  Double_t                    fParticleMinPhi;                // cut on particle phi
  Double_t                    fParticleMaxPhi;                // cut on particle phi
  UInt_t                      fTrackBitMap;                   // bit map of accepted tracks (non MC)
  UInt_t                      fMCTrackBitMap;                 // bit map of accepted MC tracks
  Int_t                       fMinMCLabel;                    // minimum MC label value for the tracks/clusters being considered MC particles

 private:
  AliParticleContainer(const AliParticleContainer& obj); // copy constructor
  AliParticleContainer& operator=(const AliParticleContainer& other); // assignment

  ClassDef(AliParticleContainer,1);

};

#endif
Commit	Line	Data
e58333e0	1	#ifndef AliParticleContainer_H
	2	#define AliParticleContainer_H
	3
	4	//
	5	// container with name, TClonesArray and cuts for particles
	6	//
	7
e58333e0	8	class AliVEvent;
e58333e0	9	class AliVParticle;
e58333e0	10
e58333e0	11	#include "AliEmcalContainer.h"
e58333e0	12
e58333e0	13	class AliParticleContainer : public AliEmcalContainer {
	14	public:
	15	AliParticleContainer();
	16	AliParticleContainer(const char *name);
6421eeb0	17	virtual ~AliParticleContainer(){;}
e58333e0	18
ef46ebe5	19	void SetParticlePtCut(Double_t cut) { fParticlePtCut = cut ; }
	20	void SetParticleEtaLimits(Double_t min, Double_t max) { fParticleMaxEta = max ; fParticleMinEta = min ; }
	21	void SetParticlePhiLimits(Double_t min, Double_t max) { fParticleMaxPhi = max ; fParticleMinPhi = min ; }
e58333e0	22	void SetTrackBitMap(UInt_t m) { fTrackBitMap = m ; }
e58333e0	23	void SetMCTrackBitMap(UInt_t m) { fMCTrackBitMap = m ; }
6421eeb0	24	void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; }
e58333e0	25
ef46ebe5	26	Double_t GetParticlePtCut() const { return fParticlePtCut; }
	27	Double_t GetParticleEtaMin() const { return fParticleMinEta; }
	28	Double_t GetParticleEtaMax() const { return fParticleMaxEta; }
	29	Double_t GetParticlePhiMin() const { return fParticleMinPhi; }
	30	Double_t GetParticlePhiMax() const { return fParticleMaxPhi; }
	31	AliVParticle GetLeadingParticle(const char opt="") ;
	32	AliVParticle *GetParticle(Int_t i) const;
	33	AliVParticle *GetAcceptParticle(Int_t i) const;
	34	AliVParticle *GetParticleWithLabel(Int_t lab) const;
	35	AliVParticle *GetAcceptParticleWithLabel(Int_t lab) const;
	36	AliVParticle *GetNextAcceptParticle(Int_t i=-1) ;
	37	AliVParticle *GetNextParticle(Int_t i=-1) ;
6421eeb0	38	void GetMomentum(TLorentzVector &mom, Int_t i) const;
ef46ebe5	39	Bool_t AcceptParticle(AliVParticle *vp) const;
ef46ebe5	40	Int_t GetNParticles() const {return GetNEntries();}
b6f970ad	41	void SetClassName(const char *clname);
e58333e0	42
	43	protected:
	44	Double_t fParticlePtCut; // cut on particle pt
	45	Double_t fParticleMinEta; // cut on particle eta
	46	Double_t fParticleMaxEta; // cut on particle eta
	47	Double_t fParticleMinPhi; // cut on particle phi
	48	Double_t fParticleMaxPhi; // cut on particle phi
	49	UInt_t fTrackBitMap; // bit map of accepted tracks (non MC)
	50	UInt_t fMCTrackBitMap; // bit map of accepted MC tracks
	51	Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles
	52
	53	private:
	54	AliParticleContainer(const AliParticleContainer& obj); // copy constructor
	55	AliParticleContainer& operator=(const AliParticleContainer& other); // assignment
	56
	57	ClassDef(AliParticleContainer,1);
	58
	59	};
	60
	61	#endif
	62