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

#ifndef ALIEMCALPICOTRACKMAKER_H
#define ALIEMCALPICOTRACKMAKER_H

class TClonesArray;
class AliVParticle;

#include "AliAnalysisTaskSE.h"

class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
 public:
  AliEmcalPicoTrackMaker();
  AliEmcalPicoTrackMaker(const char *name);
  virtual ~AliEmcalPicoTrackMaker();

  void               SetTrackEfficiency(Double_t eff = 0.95)           { fTrackEfficiency   = eff ; }
  void               SetTrackEtaLimits(Double_t min, Double_t max)     { fMaxTrackEta       = max ; fMinTrackEta      = min ; }
  void               SetTrackPhiLimits(Double_t min, Double_t max)     { fMaxTrackPhi       = max ; fMinTrackPhi      = min ; }
  void               SetTrackPtLimits(Double_t min, Double_t max)      { fMaxTrackPt        = max ; fMinTrackPt       = min ; }
  void               SetTracksInName(const char *name)                 { fTracksInName      = name; }
  void               SetTracksOutName(const char *name)                { fTracksOutName     = name; }
  void               SetMCParticlesName(const char *name)              { fMCParticlesName   = name; }
  void               SetCopyMCFlag(Bool_t c, const char* name)         { fCopyMCFlag        = c   ; fMCParticlesName  = name; }
  

 protected:
  void               UserCreateOutputObjects();
  void               UserExec(Option_t *option);

  AliVParticle*      GetMCParticle(Int_t label);

  Int_t              fAODfilterBits[2];     // AOD track filter bit map
  TString            fTracksOutName;        // name of output track array
  TString            fTracksInName;         // name of input track array
  TString            fMCParticlesName;      // name of MC particle array, used by IsHIJINGParticle
  Double_t           fMinTrackPt;           // mix pt of tracks
  Double_t           fMaxTrackPt;           // max pt of tracks
  Double_t           fMinTrackEta;          // cut on track eta
  Double_t           fMaxTrackEta;          // cut on track eta
  Double_t           fMinTrackPhi;          // cut on track phi
  Double_t           fMaxTrackPhi;          // cut on track phi
  Double_t           fTrackEfficiency;      // track efficiency
  Bool_t             fCopyMCFlag;           // copy MC flag
  TClonesArray      *fTracksIn;             //!track array in
  TClonesArray      *fTracksOut;            //!track array out
  TClonesArray      *fMCParticles;          //!MC particle array, used by IsHIJINGParticle
  Bool_t             fInit;                 //!true = task initialized

 private:
  AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&);            // not implemented
  AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented

  ClassDef(AliEmcalPicoTrackMaker, 8); // Task to make PicoTracks in AOD/ESD events
};
#endif
Commit	Line	Data
680855e5	1	#ifndef ALIEMCALPICOTRACKMAKER_H
	2	#define ALIEMCALPICOTRACKMAKER_H
	3
680855e5	4	class TClonesArray;
eb556d0c	5	class AliVParticle;
680855e5	6
	7	#include "AliAnalysisTaskSE.h"
	8
	9	class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
	10	public:
	11	AliEmcalPicoTrackMaker();
	12	AliEmcalPicoTrackMaker(const char *name);
	13	virtual ~AliEmcalPicoTrackMaker();
	14
33a8056d	15	void SetTrackEfficiency(Double_t eff = 0.95) { fTrackEfficiency = eff ; }
	16	void SetTrackEtaLimits(Double_t min, Double_t max) { fMaxTrackEta = max ; fMinTrackEta = min ; }
	17	void SetTrackPhiLimits(Double_t min, Double_t max) { fMaxTrackPhi = max ; fMinTrackPhi = min ; }
	18	void SetTrackPtLimits(Double_t min, Double_t max) { fMaxTrackPt = max ; fMinTrackPt = min ; }
	19	void SetTracksInName(const char *name) { fTracksInName = name; }
	20	void SetTracksOutName(const char *name) { fTracksOutName = name; }
eb556d0c	21	void SetMCParticlesName(const char *name) { fMCParticlesName = name; }
	22	void SetCopyMCFlag(Bool_t c, const char* name) { fCopyMCFlag = c ; fMCParticlesName = name; }
	23
680855e5	24
680855e5	25	protected:
33a8056d	26	void UserCreateOutputObjects();
	27	void UserExec(Option_t *option);
	28
eb556d0c	29	AliVParticle* GetMCParticle(Int_t label);
eb556d0c	30
c4a9715d	31	Int_t fAODfilterBits[2]; // AOD track filter bit map
a070db7d	32	TString fTracksOutName; // name of output track array
a070db7d	33	TString fTracksInName; // name of input track array
eb556d0c	34	TString fMCParticlesName; // name of MC particle array, used by IsHIJINGParticle
9733b37f	35	Double_t fMinTrackPt; // mix pt of tracks
	36	Double_t fMaxTrackPt; // max pt of tracks
	37	Double_t fMinTrackEta; // cut on track eta
	38	Double_t fMaxTrackEta; // cut on track eta
	39	Double_t fMinTrackPhi; // cut on track phi
	40	Double_t fMaxTrackPhi; // cut on track phi
11d18b51	41	Double_t fTrackEfficiency; // track efficiency
eb556d0c	42	Bool_t fCopyMCFlag; // copy MC flag
a070db7d	43	TClonesArray *fTracksIn; //!track array in
a070db7d	44	TClonesArray *fTracksOut; //!track array out
eb556d0c	45	TClonesArray *fMCParticles; //!MC particle array, used by IsHIJINGParticle
eb556d0c	46	Bool_t fInit; //!true = task initialized
680855e5	47
	48	private:
	49	AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&); // not implemented
	50	AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented
	51
eb556d0c	52	ClassDef(AliEmcalPicoTrackMaker, 8); // Task to make PicoTracks in AOD/ESD events
680855e5	53	};
680855e5	54	#endif