[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(){;}

  Bool_t                      AcceptCluster(AliVCluster         *vp) const;
  AliVCluster                *GetAcceptCluster(Int_t i)              const;
  AliVCluster                *GetAcceptClusterWithLabel(Int_t lab)   const;
  AliVCluster                *GetCluster(Int_t i)                    const;
  AliVCluster                *GetClusterWithLabel(Int_t lab)         const;
  AliVCluster                *GetLeadingCluster(const char* opt="")       ;
  void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
  AliVCluster                *GetNextAcceptCluster(Int_t i=-1)            ;
  AliVCluster                *GetNextCluster(Int_t i=-1)                  ;
  Int_t                       GetNClusters()                         const { return GetNEntries();   }
  void                        SetClassName(const char *clname);
  void                        SetClusECut(Double_t cut)                    { fClusECut      = cut ; }
  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 ; }

 protected:
  Double_t                    fClusPtCut;                  // cut on cluster pt
  Double_t                    fClusECut;                   // cut on cluster E
  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,2);
};

#endif
Commit	Line	Data
e58333e0	1	#ifndef AliClusterContainer_H
	2	#define AliClusterContainer_H
	3
29462c2a	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
29462c2a	19	Bool_t AcceptCluster(AliVCluster *vp) const;
e58333e0	20	AliVCluster *GetAcceptCluster(Int_t i) const;
ef46ebe5	21	AliVCluster *GetAcceptClusterWithLabel(Int_t lab) const;
29462c2a	22	AliVCluster *GetCluster(Int_t i) const;
	23	AliVCluster *GetClusterWithLabel(Int_t lab) const;
	24	AliVCluster GetLeadingCluster(const char opt="") ;
	25	void GetMomentum(TLorentzVector &mom, Int_t i) const;
b6f970ad	26	AliVCluster *GetNextAcceptCluster(Int_t i=-1) ;
ef46ebe5	27	AliVCluster *GetNextCluster(Int_t i=-1) ;
29462c2a	28	Int_t GetNClusters() const { return GetNEntries(); }
b6f970ad	29	void SetClassName(const char *clname);
9ce14666	30	void SetClusECut(Double_t cut) { fClusECut = cut ; }
29462c2a	31	void SetClusPtCut(Double_t cut) { fClusPtCut = cut ; }
	32	void SetClusTimeCut(Double_t min, Double_t max) { fClusTimeCutLow = min ; fClusTimeCutUp = max ; }
	33	void SetClusterBitMap(UInt_t m) { fClusterBitMap = m ; }
	34	void SetMCClusterBitMap(UInt_t m) { fMCClusterBitMap = m ; }
	35	void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; }
e58333e0	36
	37	protected:
	38	Double_t fClusPtCut; // cut on cluster pt
29462c2a	39	Double_t fClusECut; // cut on cluster E
e58333e0	40	Double_t fClusTimeCutLow; // low time cut for clusters
	41	Double_t fClusTimeCutUp; // up time cut for clusters
	42	UInt_t fClusterBitMap; // bit map of accepted clusters (non MC)
	43	UInt_t fMCClusterBitMap; // bit map of accepted MC clusters
	44	Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles
e58333e0	45
	46	private:
	47	AliClusterContainer(const AliClusterContainer& obj); // copy constructor
	48	AliClusterContainer& operator=(const AliClusterContainer& other); // assignment
	49
29462c2a	50	ClassDef(AliClusterContainer,2);
e58333e0	51	};
	52
	53	#endif
	54