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

#ifndef AliClusterContainer_H
#define AliClusterContainer_H

// $Id$ 

class TLorentzVector;

class AliVEvent;
class AliVCluster;

#include "AliEmcalContainer.h"

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

  void SetClusPtCut(Double_t cut)                  { fClusPtCut      = cut ; }
  void SetClusTimeCut(Double_t min, Double_t max)  { fClusTimeCutLow = min ; fClusTimeCutUp = max ; }
  void SetClusterBitMap(UInt_t m)                  { fClusterBitMap     = m ; }
  void SetMCClusterBitMap(UInt_t m)                { fMCClusterBitMap   = m ; }
  void SetMinMCLabel(Int_t s)                      { fMinMCLabel        = s ; }

  AliVCluster                *GetLeadingCluster(const char* opt="")       ;
  AliVCluster                *GetCluster(Int_t i)                    const;
  AliVCluster                *GetAcceptCluster(Int_t i)              const;
  AliVCluster                *GetClusterWithLabel(Int_t lab)         const;
  AliVCluster                *GetAcceptClusterWithLabel(Int_t lab)   const;
  AliVCluster                *GetNextAcceptCluster(Int_t i=-1)            ;
  AliVCluster                *GetNextCluster(Int_t i=-1)                  ;
  void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
  Bool_t                      AcceptCluster(AliVCluster         *vp) const;
  Int_t                       GetNClusters()                         const   {return GetNEntries();}
  void                        SetClassName(const char *clname);

 protected:
  Double_t                    fClusPtCut;                  // cut on cluster pt
  Double_t                    fClusTimeCutLow;             // low time cut for clusters
  Double_t                    fClusTimeCutUp;              // up time cut for clusters
  UInt_t                      fClusterBitMap;              // bit map of accepted clusters (non MC)
  UInt_t                      fMCClusterBitMap;            // bit map of accepted MC clusters
  Int_t                       fMinMCLabel;                 // minimum MC label value for the tracks/clusters being considered MC particles

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

  ClassDef(AliClusterContainer,1);

};

#endif
Commit	Line	Data
e58333e0	1	#ifndef AliClusterContainer_H
	2	#define AliClusterContainer_H
	3
9239b066	4	// $Id$
e58333e0	5
6421eeb0	6	class TLorentzVector;
e58333e0	7
e58333e0	8	class AliVEvent;
e58333e0	9	class AliVCluster;
e58333e0	10
e58333e0	11	#include "AliEmcalContainer.h"
e58333e0	12
e58333e0	13	class AliClusterContainer : public AliEmcalContainer {
	14	public:
	15	AliClusterContainer();
	16	AliClusterContainer(const char *name);
6421eeb0	17	virtual ~AliClusterContainer(){;}
e58333e0	18
e58333e0	19	void SetClusPtCut(Double_t cut) { fClusPtCut = cut ; }
	20	void SetClusTimeCut(Double_t min, Double_t max) { fClusTimeCutLow = min ; fClusTimeCutUp = max ; }
	21	void SetClusterBitMap(UInt_t m) { fClusterBitMap = m ; }
	22	void SetMCClusterBitMap(UInt_t m) { fMCClusterBitMap = m ; }
	23	void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; }
	24
b6f970ad	25	AliVCluster GetLeadingCluster(const char opt="") ;
6421eeb0	26	AliVCluster *GetCluster(Int_t i) const;
e58333e0	27	AliVCluster *GetAcceptCluster(Int_t i) const;
ef46ebe5	28	AliVCluster *GetClusterWithLabel(Int_t lab) const;
ef46ebe5	29	AliVCluster *GetAcceptClusterWithLabel(Int_t lab) const;
b6f970ad	30	AliVCluster *GetNextAcceptCluster(Int_t i=-1) ;
ef46ebe5	31	AliVCluster *GetNextCluster(Int_t i=-1) ;
6421eeb0	32	void GetMomentum(TLorentzVector &mom, Int_t i) const;
e58333e0	33	Bool_t AcceptCluster(AliVCluster *vp) const;
e58333e0	34	Int_t GetNClusters() const {return GetNEntries();}
b6f970ad	35	void SetClassName(const char *clname);
e58333e0	36
	37	protected:
	38	Double_t fClusPtCut; // cut on cluster pt
	39	Double_t fClusTimeCutLow; // low time cut for clusters
	40	Double_t fClusTimeCutUp; // up time cut for clusters
	41	UInt_t fClusterBitMap; // bit map of accepted clusters (non MC)
	42	UInt_t fMCClusterBitMap; // bit map of accepted MC clusters
	43	Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles
e58333e0	44
	45	private:
	46	AliClusterContainer(const AliClusterContainer& obj); // copy constructor
	47	AliClusterContainer& operator=(const AliClusterContainer& other); // assignment
	48
	49	ClassDef(AliClusterContainer,1);
	50
	51	};
	52
	53	#endif
	54