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

#ifndef ALIEMCALPICOTRACKMAKER_H
#define ALIEMCALPICOTRACKMAKER_H

// $Id$

class TClonesArray;
class AliVEvent;
class AliVTrack;
class AliESDtrackCuts;

#include "AliAnalysisTaskSE.h"

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

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

  void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)      { fAODfilterBits[0] = b0  ; fAODfilterBits[1] = b1  ; }
  void SetESDtrackCuts(AliESDtrackCuts *cuts)            { fESDtrackCuts     = cuts; }
  void SetTracksInName(const char *name)                 { fTracksInName     = name; }
  void SetTracksOutName(const char *name)                { fTracksOutName    = name; }
  void SetTrackPtLimits(Double_t min, Double_t max)      { fMaxTrackPt       = max ; fMinTrackPt       = min ; }
  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 SetTrackEfficiency(Double_t eff = 0.95)           { fTrackEfficiency  = eff ; }
  void SetIncludeNoITS(Bool_t f)                         { fIncludeNoITS     = f   ; }
  void SetUseNegativeLabels(Bool_t f)                    { fUseNegativeLabels= f   ; }
  void SetMC(Bool_t a)                                   { fIsMC             = a   ; }
  void SetCutMaxFractionSharedTPCClusters(Double_t c = 0.4)        { fCutMaxFractionSharedTPCClusters  = c ; }

 protected:
  Int_t              fAODfilterBits[2];     // AOD track filter bit map
  AliESDtrackCuts   *fESDtrackCuts;         // ESD track cuts
  TString            fTracksOutName;        // name of output track array
  TString            fTracksInName;         // name of input track array
  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             fIncludeNoITS;         // includes tracks with failed ITS refit
  Bool_t             fUseNegativeLabels;    // whether or not should use negative MC labels
  Bool_t             fIsMC;                 // whether it is a MC event or not
  Double_t           fCutMaxFractionSharedTPCClusters;  // max fraction of shared TPC clusters
  TClonesArray      *fTracksIn;             //!track array in
  TClonesArray      *fTracksOut;            //!track array out

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

  ClassDef(AliEmcalPicoTrackMaker, 6); // Task to make PicoTracks in AOD/ESD events
};
#endif
Commit	Line	Data
680855e5	1	#ifndef ALIEMCALPICOTRACKMAKER_H
	2	#define ALIEMCALPICOTRACKMAKER_H
	3
980821ba	4	// $Id$
680855e5	5
	6	class TClonesArray;
	7	class AliVEvent;
	8	class AliVTrack;
	9	class AliESDtrackCuts;
	10
	11	#include "AliAnalysisTaskSE.h"
	12
	13	class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
	14	public:
	15	AliEmcalPicoTrackMaker();
	16	AliEmcalPicoTrackMaker(const char *name);
	17	virtual ~AliEmcalPicoTrackMaker();
	18
	19	void UserCreateOutputObjects();
	20	void UserExec(Option_t *option);
	21
9733b37f	22	void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0) { fAODfilterBits[0] = b0 ; fAODfilterBits[1] = b1 ; }
	23	void SetESDtrackCuts(AliESDtrackCuts *cuts) { fESDtrackCuts = cuts; }
	24	void SetTracksInName(const char *name) { fTracksInName = name; }
	25	void SetTracksOutName(const char *name) { fTracksOutName = name; }
	26	void SetTrackPtLimits(Double_t min, Double_t max) { fMaxTrackPt = max ; fMinTrackPt = min ; }
	27	void SetTrackEtaLimits(Double_t min, Double_t max) { fMaxTrackEta = max ; fMinTrackEta = min ; }
	28	void SetTrackPhiLimits(Double_t min, Double_t max) { fMaxTrackPhi = max ; fMinTrackPhi = min ; }
7f76e479	29	void SetTrackEfficiency(Double_t eff = 0.95) { fTrackEfficiency = eff ; }
7f76e479	30	void SetIncludeNoITS(Bool_t f) { fIncludeNoITS = f ; }
86f2216c	31	void SetUseNegativeLabels(Bool_t f) { fUseNegativeLabels= f ; }
7f76e479	32	void SetMC(Bool_t a) { fIsMC = a ; }
69d71b2d	33	void SetCutMaxFractionSharedTPCClusters(Double_t c = 0.4) { fCutMaxFractionSharedTPCClusters = c ; }
680855e5	34
680855e5	35	protected:
c4a9715d	36	Int_t fAODfilterBits[2]; // AOD track filter bit map
680855e5	37	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts
a070db7d	38	TString fTracksOutName; // name of output track array
a070db7d	39	TString fTracksInName; // name of input track array
9733b37f	40	Double_t fMinTrackPt; // mix pt of tracks
	41	Double_t fMaxTrackPt; // max pt of tracks
	42	Double_t fMinTrackEta; // cut on track eta
	43	Double_t fMaxTrackEta; // cut on track eta
	44	Double_t fMinTrackPhi; // cut on track phi
	45	Double_t fMaxTrackPhi; // cut on track phi
11d18b51	46	Double_t fTrackEfficiency; // track efficiency
bb99fbec	47	Bool_t fIncludeNoITS; // includes tracks with failed ITS refit
7f76e479	48	Bool_t fUseNegativeLabels; // whether or not should use negative MC labels
7f76e479	49	Bool_t fIsMC; // whether it is a MC event or not
69d71b2d	50	Double_t fCutMaxFractionSharedTPCClusters; // max fraction of shared TPC clusters
a070db7d	51	TClonesArray *fTracksIn; //!track array in
a070db7d	52	TClonesArray *fTracksOut; //!track array out
680855e5	53
	54	private:
	55	AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&); // not implemented
	56	AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented
	57
43032ce2	58	ClassDef(AliEmcalPicoTrackMaker, 6); // Task to make PicoTracks in AOD/ESD events
680855e5	59	};
680855e5	60	#endif