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

#ifndef ALIEMCALESDTRACKFILTERTASK_H
#define ALIEMCALESDTRACKFILTERTASK_H

class TClonesArray;
class AliESDEvent;
class AliESDtrackCuts;

#include <TF1.h>

#include "AliAnalysisTaskSE.h"
#include "AliESDtrackCuts.h"

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

  void SetDist(Double_t d)                       { fDist             = d;    }
  void SetDoPropagation(Bool_t b)                { fDoPropagation    = b;    }
  void SetDoSpdVtxConstrain(Bool_t b)            { fDoSpdVtxCon      = b;    }
  void SetHybridTrackCuts(AliESDtrackCuts *cuts) { fHybridTrackCuts  = cuts; }
  void SetIncludeNoITS(Bool_t f)                 { fIncludeNoITS     = f;    }
  void SetTrackCuts(AliESDtrackCuts *cuts)       { fEsdTrackCuts     = cuts; }
  void SetTracksName(const char *name)           { fTracksName       = name; }
  void SetTrackEfficiency(Double_t eff = 0.95)   { fTrackEfficiency  = new TF1("eff", "[0]", 0, 500); fTrackEfficiency->FixParameter(0,eff); }
  void SetTrackEfficiency(TF1* eff)              { fTrackEfficiency  = eff;  }
  void SetMC(Bool_t b)                           { fIsMC             = b  ;  }

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

  AliESDtrackCuts   *fEsdTrackCuts;      // esd track cuts
  Bool_t             fDoSpdVtxCon;       // if true then do vertex constraint
  AliESDtrackCuts   *fHybridTrackCuts;   // hybrid track cuts
  TString            fTracksName;        // name of tracks 
  Bool_t             fIncludeNoITS;      // includes tracks with failed ITS refit
  Bool_t             fDoPropagation;     // propagate all hybrid tracks to EMCal surface
  Double_t           fDist;              // distance to surface (440cm default)
  TF1               *fTrackEfficiency;   // track efficiency
  Bool_t             fIsMC;              // whether it is a MC event or not
  AliESDEvent       *fEsdEv;             //!esd event
  TClonesArray      *fTracks;            //!track array

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

  ClassDef(AliEmcalEsdTrackFilterTask, 4); // Class to constrain TPC tracks to SPD vertex
};

#endif
Commit	Line	Data
	1	#ifndef ALIEMCALESDTRACKFILTERTASK_H
	2	#define ALIEMCALESDTRACKFILTERTASK_H
	3
	4	class TClonesArray;
	5	class AliESDEvent;
	6	class AliESDtrackCuts;
	7
	8	#include <TF1.h>
	9
	10	#include "AliAnalysisTaskSE.h"
	11	#include "AliESDtrackCuts.h"
	12
	13	class AliEmcalEsdTrackFilterTask : public AliAnalysisTaskSE {
	14	public:
	15	AliEmcalEsdTrackFilterTask();
	16	AliEmcalEsdTrackFilterTask(const char *name);
	17	virtual ~AliEmcalEsdTrackFilterTask();
	18
	19	void SetDist(Double_t d) { fDist = d; }
	20	void SetDoPropagation(Bool_t b) { fDoPropagation = b; }
	21	void SetDoSpdVtxConstrain(Bool_t b) { fDoSpdVtxCon = b; }
	22	void SetHybridTrackCuts(AliESDtrackCuts *cuts) { fHybridTrackCuts = cuts; }
	23	void SetIncludeNoITS(Bool_t f) { fIncludeNoITS = f; }
	24	void SetTrackCuts(AliESDtrackCuts *cuts) { fEsdTrackCuts = cuts; }
	25	void SetTracksName(const char *name) { fTracksName = name; }
	26	void SetTrackEfficiency(Double_t eff = 0.95) { fTrackEfficiency = new TF1("eff", "[0]", 0, 500); fTrackEfficiency->FixParameter(0,eff); }
	27	void SetTrackEfficiency(TF1* eff) { fTrackEfficiency = eff; }
	28	void SetMC(Bool_t b) { fIsMC = b ; }
	29
	30	protected:
	31	void UserCreateOutputObjects();
	32	void UserExec(Option_t *option);
	33
	34	AliESDtrackCuts *fEsdTrackCuts; // esd track cuts
	35	Bool_t fDoSpdVtxCon; // if true then do vertex constraint
	36	AliESDtrackCuts *fHybridTrackCuts; // hybrid track cuts
	37	TString fTracksName; // name of tracks
	38	Bool_t fIncludeNoITS; // includes tracks with failed ITS refit
	39	Bool_t fDoPropagation; // propagate all hybrid tracks to EMCal surface
	40	Double_t fDist; // distance to surface (440cm default)
	41	TF1 *fTrackEfficiency; // track efficiency
	42	Bool_t fIsMC; // whether it is a MC event or not
	43	AliESDEvent *fEsdEv; //!esd event
	44	TClonesArray *fTracks; //!track array
	45
	46	private:
	47	AliEmcalEsdTrackFilterTask(const AliEmcalEsdTrackFilterTask&); // not implemented
	48	AliEmcalEsdTrackFilterTask &operator=(const AliEmcalEsdTrackFilterTask&); // not implemented
	49
	50	ClassDef(AliEmcalEsdTrackFilterTask, 4); // Class to constrain TPC tracks to SPD vertex
	51	};
	52
	53	#endif