[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskESDfilter.h

#ifndef ALIANALYSISTASKESDFILTER_H
#define ALIANALYSISTASKESDFILTER_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

#include <TList.h> 
#include <TF1.h> 
#include "AliAnalysisTaskSE.h"
#include "AliESDtrack.h"
#include "AliAODTrack.h"
#include "AliAODPid.h"
#include "AliESDpid.h"

class AliAnalysisFilter;
class AliStack;
class AliESDtrack;
class AliMCEventHandler;
class TRefArray;
class AliAODHeader;
class AliESDtrackCuts;

class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskESDfilter();
    AliAnalysisTaskESDfilter(const char* name);
    virtual ~AliAnalysisTaskESDfilter();
    // Implementation of interface methods
    virtual void   UserCreateOutputObjects();
    virtual void   Init();
    virtual void   LocalInit() {Init();}
    virtual void   UserExec(Option_t *option);
    virtual void   Terminate(Option_t *option);

    virtual void ConvertESDtoAOD();
    // Setters
    virtual void SetTrackFilter   (AliAnalysisFilter*   trackF) {fTrackFilter    =   trackF;}
    virtual void SetTPCOnlyFilterMask (UInt_t filterMask)       {SetTPCConstrainedFilterMask(filterMask);}

    virtual void SetTPCConstrainedFilterMask (UInt_t filterMask)       {fTPCConstrainedFilterMask    =  filterMask;}
    virtual void SetHybridFilterMaskTPCConstrainedGlobal(UInt_t filterMask)  {fHybridFilterMaskTPCCG    =  filterMask;}
    virtual void SetWriteHybridTPCConstrainedOnly(bool b){fWriteHybridTPCCOnly = b;}

    virtual void SetGlobalConstrainedFilterMask (UInt_t filterMask)    {fGlobalConstrainedFilterMask    =  filterMask;}
    virtual void SetHybridFilterMaskGlobalConstrainedGlobal(UInt_t filterMask)  {fHybridFilterMaskGCG    =  filterMask;}
    virtual void SetWriteHybridGlobalConstrainedOnly(bool b){fWriteHybridGCOnly = b;}


    virtual void SetKinkFilter    (AliAnalysisFilter*    KinkF) {fKinkFilter     =    KinkF;}
    virtual void SetV0Filter      (AliAnalysisFilter*      V0F) {fV0Filter       =      V0F;}
    virtual void SetCascadeFilter (AliAnalysisFilter* CascadeF) {fCascadeFilter  = CascadeF;}
    virtual void SetPthreshold    (Double_t p)                  {fHighPthreshold =        p;}
    virtual void SetPshape        (TF1 *func)                   {fPtshape        =     func;}
    virtual void SetEnableFillAOD (Bool_t b)                    {fEnableFillAOD  =     b;}

    virtual void SetAODPID(AliESDtrack *esdtrack, AliAODTrack *aodtrack, AliAODPid *detpid);
    void SetDetectorRawSignals(AliAODPid *aodpid, AliESDtrack *track);

    void SetV0Cuts(const Double_t cuts[7]) {for (Int_t icut = 0; icut<7; icut++) fV0Cuts[icut] = cuts[icut];}

    void SetCascadeCuts(const Double_t cuts[8]) {for (Int_t icut = 0; icut<8; icut++) fCascadeCuts[icut] = cuts[icut];} 

    void GetV0Cuts(Double_t cuts[7]) const {for (Int_t icut = 0; icut<7; icut++) cuts[icut] = fV0Cuts[icut];}

    void GetCascadeCuts(Double_t cuts[8]) const {for (Int_t icut = 0; icut<8; icut++) cuts[icut] = fCascadeCuts[icut];}


  void PrintTask(Option_t *option="all", Int_t indent=0) const;
  
  void DisableVZERO() { fIsVZEROEnabled = kFALSE; }
  void DisableTZERO() { fIsTZEROEnabled = kFALSE; }
  void DisableZDC()   { fIsZDCEnabled   = kFALSE; }
  void DisableCascades() { fAreCascadesEnabled = kFALSE; }
  void DisableV0s() { fAreV0sEnabled = kFALSE; }
  void DisableKinks() { fAreKinksEnabled = kFALSE; }
  void DisableTracks() { fAreTracksEnabled = kFALSE; }
  void DisablePmdClusters() { fArePmdClustersEnabled = kFALSE; }
  void DisableCaloClusters() { fAreCaloClustersEnabled = kFALSE; }
  void DisableCells() { fAreEMCALCellsEnabled = fArePHOSCellsEnabled = kFALSE; }
  void DisableCaloTrigger(TString calo = "PHOS") { if (calo.Contains("EMCAL")) fAreEMCALTriggerEnabled = kFALSE; else fArePHOSTriggerEnabled = kFALSE; }
  void DisableTracklets() { fAreTrackletsEnabled = kFALSE; }
  void DisableHMPID()  { fIsHMPIDEnabled = kFALSE; } 

  void EnableV0CascadeVerticesReco() { fIsV0CascadeRecoEnabled = kTRUE; }

  void SetPropagateTrackToEMCal(Bool_t propagate) {fDoPropagateTrackToEMCal = propagate;}

  virtual void SetTimeZeroType(AliESDpid::EStartTimeType_t tofTimeZeroType) {fTimeZeroType = tofTimeZeroType;}
  
private:
    AliAnalysisTaskESDfilter(const AliAnalysisTaskESDfilter&);
    AliAnalysisTaskESDfilter& operator=(const AliAnalysisTaskESDfilter&);
    void PrintMCInfo(AliStack *pStack,Int_t label); // for debugging
    Double_t Chi2perNDF(AliESDtrack* track);
    
  AliAODHeader* ConvertHeader(const AliESDEvent& esd);
  void ConvertCascades(const AliESDEvent& esd);
  void ConvertV0s(const AliESDEvent& esd);
  void ConvertKinks(const AliESDEvent& esd);
  void ConvertPrimaryVertices(const AliESDEvent& esd);
  void ConvertTracks(const AliESDEvent& esd);
  void ConvertPmdClusters(const AliESDEvent& esd);
  void ConvertCaloClusters(const AliESDEvent& esd);
  void ConvertEMCALCells(const AliESDEvent& esd);
  void ConvertPHOSCells(const AliESDEvent& esd);
  void ConvertCaloTrigger(TString calo, const AliESDEvent& esd);
  void ConvertTracklets(const AliESDEvent& esd);
  void ConvertTPCOnlyTracks(const AliESDEvent& esd);
  void ConvertGlobalConstrainedTracks(const AliESDEvent& esd);
  void ConvertVZERO(const AliESDEvent& esd);
  void ConvertTZERO(const AliESDEvent& esd);
  void ConvertZDC(const AliESDEvent& esd);
  Int_t ConvertHMPID(const AliESDEvent& esd);
  void PropagateTrackToEMCal(AliESDtrack *esdTrack);

  TClonesArray& Tracks();
  TClonesArray& V0s();
  TClonesArray& Vertices();
  TClonesArray& Cascades();
  
  // Filtering
  AliAnalysisFilter* fTrackFilter;      //  Track   Filter
  AliAnalysisFilter* fKinkFilter;       //  Kink    Filter
  AliAnalysisFilter* fV0Filter;         //  V0      Filter
  AliAnalysisFilter* fCascadeFilter;    //  Cascade Filter
  // PID
  Double_t     fHighPthreshold;    //  Pt threshold for detector signal setting
  TF1 *        fPtshape;           //  Pt spectrum distribution
  Bool_t       fEnableFillAOD;     //  value that decides if this task activates AOD filling
  Bool_t* fUsedTrack; //! indices of used tracks
  Bool_t* fUsedKink; //! indices of used kinks
  Bool_t* fUsedV0; //! indices of used V0s
  TRefArray* fAODTrackRefs; // array of track references
  TRefArray* fAODV0VtxRefs; // array of v0 vertices references
  TRefArray* fAODV0Refs ; // array of v0s references
  AliMCEventHandler* fMChandler; // pointer to MC handler (if any)
  Int_t fNumberOfTracks; // current number of tracks
  Int_t fNumberOfPositiveTracks; // current number of positive tracks
  Int_t fNumberOfV0s; // current number of v0s
  Int_t fNumberOfVertices; // current number of vertices
  Int_t fNumberOfCascades; // current number of cascades
  Int_t fNumberOfKinks; // current number of kinks
  Bool_t fOldESDformat; // is the ESD in old format ?
  AliAODVertex* fPrimaryVertex; // pointer to primary vertex of the event
  UInt_t fTPCConstrainedFilterMask;      //  Filter Mask used to select and store refitted TPC only tracks
  UInt_t fHybridFilterMaskTPCCG; //  Filter Mask used to mark global tracks as hybrid
  Bool_t fWriteHybridTPCCOnly;// write only the complent tracks not all global constrained

  UInt_t fGlobalConstrainedFilterMask; //  Filter Mask used to select and store refitted TPC only tracks
  UInt_t fHybridFilterMaskGCG;  //  Filter Mask used to mark global tracks as hybrid
  Bool_t fWriteHybridGCOnly;// write only the complent tracks not all global constrained

  Bool_t fIsVZEROEnabled; // whether or not to fill the vzero branch (true by default)
  Bool_t fIsTZEROEnabled; // whether or not to fill the tzero branch (true by default)
  Bool_t fIsZDCEnabled; // whether or not to fill the zdc branch (true by default)
  Bool_t fIsHMPIDEnabled; // whether or not to fill the hmpid branch (true by default) 
  Bool_t fIsV0CascadeRecoEnabled; // whether or not to reconstruct again V0s and cascades (false by default)
  Bool_t fAreCascadesEnabled; // whether or not to fill the cascades branch (true by default)
  Bool_t fAreV0sEnabled; // whether or not to fill the v0 branch (true by default)
  Bool_t fAreKinksEnabled; // whether or not to fill the kinks (true by default)
  Bool_t fAreTracksEnabled; // whether or not to fill the (central) tracks branch (true by default)
  Bool_t fArePmdClustersEnabled; // whether or not to fill the pmd clusters (true by default)
  Bool_t fAreCaloClustersEnabled; // whether or not to fill the calo clusters (true by default)
  Bool_t fAreEMCALCellsEnabled; // whether or not to fill the emcal cells (true by default)
  Bool_t fArePHOSCellsEnabled; // whether or not to fill the phos cells (true by default)
  Bool_t fAreEMCALTriggerEnabled; // whether or not to fill the emcal trigger (true by default)
  Bool_t fArePHOSTriggerEnabled; // whether or not to fill the phos trigger (true by default)
  Bool_t fAreTrackletsEnabled; // whether or not to fill the tracklets (true by default)
  AliESDpid* fESDpid; // esd pid
  Bool_t fIsPidOwner; // whether we own fESDpid
  Int_t fTimeZeroType;  //  time zero type 
  AliESDtrackCuts* fTPCaloneTrackCuts; // TPC stand-alone track cuts
  Double_t        fV0Cuts[7];       // Array to store the values for the different reco selections V0 related
  Double_t        fCascadeCuts[8];  // Array to store the values for the different reco selections cascades related
  Bool_t fDoPropagateTrackToEMCal;  // whether or not to propagate the tracks to EMCal surface (430cm) -- true by default
  
  
  ClassDef(AliAnalysisTaskESDfilter, 15); // Analysis task for standard ESD filtering
};

#endif
Commit	Line	Data
c7ef1c47	1	#ifndef ALIANALYSISTASKESDFILTER_H
	2	#define ALIANALYSISTASKESDFILTER_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8
	9	#include <TList.h>
	10	#include <TF1.h>
	11	#include "AliAnalysisTaskSE.h"
	12	#include "AliESDtrack.h"
	13	#include "AliAODTrack.h"
	14	#include "AliAODPid.h"
	15	#include "AliESDpid.h"
	16
	17	class AliAnalysisFilter;
	18	class AliStack;
	19	class AliESDtrack;
	20	class AliMCEventHandler;
	21	class TRefArray;
	22	class AliAODHeader;
2f5ee3b7	23	class AliESDtrackCuts;
c7ef1c47	24
	25	class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
	26	{
	27	public:
	28	AliAnalysisTaskESDfilter();
	29	AliAnalysisTaskESDfilter(const char* name);
75a996ab	30	virtual ~AliAnalysisTaskESDfilter();
c7ef1c47	31	// Implementation of interface methods
	32	virtual void UserCreateOutputObjects();
	33	virtual void Init();
	34	virtual void LocalInit() {Init();}
	35	virtual void UserExec(Option_t *option);
	36	virtual void Terminate(Option_t *option);
	37
	38	virtual void ConvertESDtoAOD();
	39	// Setters
	40	virtual void SetTrackFilter (AliAnalysisFilter* trackF) {fTrackFilter = trackF;}
67abcce9	41	virtual void SetTPCOnlyFilterMask (UInt_t filterMask) {SetTPCConstrainedFilterMask(filterMask);}
	42
	43	virtual void SetTPCConstrainedFilterMask (UInt_t filterMask) {fTPCConstrainedFilterMask = filterMask;}
	44	virtual void SetHybridFilterMaskTPCConstrainedGlobal(UInt_t filterMask) {fHybridFilterMaskTPCCG = filterMask;}
	45	virtual void SetWriteHybridTPCConstrainedOnly(bool b){fWriteHybridTPCCOnly = b;}
	46
	47	virtual void SetGlobalConstrainedFilterMask (UInt_t filterMask) {fGlobalConstrainedFilterMask = filterMask;}
	48	virtual void SetHybridFilterMaskGlobalConstrainedGlobal(UInt_t filterMask) {fHybridFilterMaskGCG = filterMask;}
	49	virtual void SetWriteHybridGlobalConstrainedOnly(bool b){fWriteHybridGCOnly = b;}
	50
	51
c7ef1c47	52	virtual void SetKinkFilter (AliAnalysisFilter* KinkF) {fKinkFilter = KinkF;}
	53	virtual void SetV0Filter (AliAnalysisFilter* V0F) {fV0Filter = V0F;}
	54	virtual void SetCascadeFilter (AliAnalysisFilter* CascadeF) {fCascadeFilter = CascadeF;}
	55	virtual void SetPthreshold (Double_t p) {fHighPthreshold = p;}
	56	virtual void SetPshape (TF1 *func) {fPtshape = func;}
	57	virtual void SetEnableFillAOD (Bool_t b) {fEnableFillAOD = b;}
	58
1a3579a6	59	virtual void SetAODPID(AliESDtrack esdtrack, AliAODTrack aodtrack, AliAODPid *detpid);
1a3579a6	60	void SetDetectorRawSignals(AliAODPid aodpid, AliESDtrack track);
c7ef1c47	61
b8d29a28	62	void SetV0Cuts(const Double_t cuts[7]) {for (Int_t icut = 0; icut<7; icut++) fV0Cuts[icut] = cuts[icut];}
	63
	64	void SetCascadeCuts(const Double_t cuts[8]) {for (Int_t icut = 0; icut<8; icut++) fCascadeCuts[icut] = cuts[icut];}
	65
	66	void GetV0Cuts(Double_t cuts[7]) const {for (Int_t icut = 0; icut<7; icut++) cuts[icut] = fV0Cuts[icut];}
	67
	68	void GetCascadeCuts(Double_t cuts[8]) const {for (Int_t icut = 0; icut<8; icut++) cuts[icut] = fCascadeCuts[icut];}
	69
	70
c7ef1c47	71	void PrintTask(Option_t *option="all", Int_t indent=0) const;
	72
	73	void DisableVZERO() { fIsVZEROEnabled = kFALSE; }
a1f96974	74	void DisableTZERO() { fIsTZEROEnabled = kFALSE; }
8c87d22c	75	void DisableZDC() { fIsZDCEnabled = kFALSE; }
c7ef1c47	76	void DisableCascades() { fAreCascadesEnabled = kFALSE; }
	77	void DisableV0s() { fAreV0sEnabled = kFALSE; }
	78	void DisableKinks() { fAreKinksEnabled = kFALSE; }
	79	void DisableTracks() { fAreTracksEnabled = kFALSE; }
	80	void DisablePmdClusters() { fArePmdClustersEnabled = kFALSE; }
	81	void DisableCaloClusters() { fAreCaloClustersEnabled = kFALSE; }
	82	void DisableCells() { fAreEMCALCellsEnabled = fArePHOSCellsEnabled = kFALSE; }
720f7306	83	void DisableCaloTrigger(TString calo = "PHOS") { if (calo.Contains("EMCAL")) fAreEMCALTriggerEnabled = kFALSE; else fArePHOSTriggerEnabled = kFALSE; }
c7ef1c47	84	void DisableTracklets() { fAreTrackletsEnabled = kFALSE; }
08b38f3f	85	void DisableHMPID() { fIsHMPIDEnabled = kFALSE; }
c7ef1c47	86
b8d29a28	87	void EnableV0CascadeVerticesReco() { fIsV0CascadeRecoEnabled = kTRUE; }
b8d29a28	88
ed15417e	89	void SetPropagateTrackToEMCal(Bool_t propagate) {fDoPropagateTrackToEMCal = propagate;}
ed15417e	90
c7ef1c47	91	virtual void SetTimeZeroType(AliESDpid::EStartTimeType_t tofTimeZeroType) {fTimeZeroType = tofTimeZeroType;}
	92
	93	private:
	94	AliAnalysisTaskESDfilter(const AliAnalysisTaskESDfilter&);
	95	AliAnalysisTaskESDfilter& operator=(const AliAnalysisTaskESDfilter&);
	96	void PrintMCInfo(AliStack *pStack,Int_t label); // for debugging
	97	Double_t Chi2perNDF(AliESDtrack* track);
	98
	99	AliAODHeader* ConvertHeader(const AliESDEvent& esd);
	100	void ConvertCascades(const AliESDEvent& esd);
	101	void ConvertV0s(const AliESDEvent& esd);
	102	void ConvertKinks(const AliESDEvent& esd);
	103	void ConvertPrimaryVertices(const AliESDEvent& esd);
	104	void ConvertTracks(const AliESDEvent& esd);
	105	void ConvertPmdClusters(const AliESDEvent& esd);
	106	void ConvertCaloClusters(const AliESDEvent& esd);
	107	void ConvertEMCALCells(const AliESDEvent& esd);
	108	void ConvertPHOSCells(const AliESDEvent& esd);
720f7306	109	void ConvertCaloTrigger(TString calo, const AliESDEvent& esd);
c7ef1c47	110	void ConvertTracklets(const AliESDEvent& esd);
c7ef1c47	111	void ConvertTPCOnlyTracks(const AliESDEvent& esd);
67abcce9	112	void ConvertGlobalConstrainedTracks(const AliESDEvent& esd);
c7ef1c47	113	void ConvertVZERO(const AliESDEvent& esd);
a1f96974	114	void ConvertTZERO(const AliESDEvent& esd);
8c87d22c	115	void ConvertZDC(const AliESDEvent& esd);
08b38f3f	116	Int_t ConvertHMPID(const AliESDEvent& esd);
37b92631	117	void PropagateTrackToEMCal(AliESDtrack *esdTrack);
37b92631	118
c7ef1c47	119	TClonesArray& Tracks();
	120	TClonesArray& V0s();
	121	TClonesArray& Vertices();
	122	TClonesArray& Cascades();
	123
	124	// Filtering
	125	AliAnalysisFilter* fTrackFilter; // Track Filter
	126	AliAnalysisFilter* fKinkFilter; // Kink Filter
	127	AliAnalysisFilter* fV0Filter; // V0 Filter
	128	AliAnalysisFilter* fCascadeFilter; // Cascade Filter
	129	// PID
	130	Double_t fHighPthreshold; // Pt threshold for detector signal setting
	131	TF1 * fPtshape; // Pt spectrum distribution
	132	Bool_t fEnableFillAOD; // value that decides if this task activates AOD filling
	133	Bool_t* fUsedTrack; //! indices of used tracks
	134	Bool_t* fUsedKink; //! indices of used kinks
	135	Bool_t* fUsedV0; //! indices of used V0s
	136	TRefArray* fAODTrackRefs; // array of track references
	137	TRefArray* fAODV0VtxRefs; // array of v0 vertices references
	138	TRefArray* fAODV0Refs ; // array of v0s references
	139	AliMCEventHandler* fMChandler; // pointer to MC handler (if any)
	140	Int_t fNumberOfTracks; // current number of tracks
	141	Int_t fNumberOfPositiveTracks; // current number of positive tracks
	142	Int_t fNumberOfV0s; // current number of v0s
	143	Int_t fNumberOfVertices; // current number of vertices
	144	Int_t fNumberOfCascades; // current number of cascades
	145	Int_t fNumberOfKinks; // current number of kinks
	146	Bool_t fOldESDformat; // is the ESD in old format ?
	147	AliAODVertex* fPrimaryVertex; // pointer to primary vertex of the event
67abcce9	148	UInt_t fTPCConstrainedFilterMask; // Filter Mask used to select and store refitted TPC only tracks
	149	UInt_t fHybridFilterMaskTPCCG; // Filter Mask used to mark global tracks as hybrid
	150	Bool_t fWriteHybridTPCCOnly;// write only the complent tracks not all global constrained
	151
	152	UInt_t fGlobalConstrainedFilterMask; // Filter Mask used to select and store refitted TPC only tracks
	153	UInt_t fHybridFilterMaskGCG; // Filter Mask used to mark global tracks as hybrid
	154	Bool_t fWriteHybridGCOnly;// write only the complent tracks not all global constrained
25a60d41	155
c7ef1c47	156	Bool_t fIsVZEROEnabled; // whether or not to fill the vzero branch (true by default)
a1f96974	157	Bool_t fIsTZEROEnabled; // whether or not to fill the tzero branch (true by default)
8c87d22c	158	Bool_t fIsZDCEnabled; // whether or not to fill the zdc branch (true by default)
08b38f3f	159	Bool_t fIsHMPIDEnabled; // whether or not to fill the hmpid branch (true by default)
b8d29a28	160	Bool_t fIsV0CascadeRecoEnabled; // whether or not to reconstruct again V0s and cascades (false by default)
c7ef1c47	161	Bool_t fAreCascadesEnabled; // whether or not to fill the cascades branch (true by default)
	162	Bool_t fAreV0sEnabled; // whether or not to fill the v0 branch (true by default)
	163	Bool_t fAreKinksEnabled; // whether or not to fill the kinks (true by default)
	164	Bool_t fAreTracksEnabled; // whether or not to fill the (central) tracks branch (true by default)
	165	Bool_t fArePmdClustersEnabled; // whether or not to fill the pmd clusters (true by default)
	166	Bool_t fAreCaloClustersEnabled; // whether or not to fill the calo clusters (true by default)
	167	Bool_t fAreEMCALCellsEnabled; // whether or not to fill the emcal cells (true by default)
	168	Bool_t fArePHOSCellsEnabled; // whether or not to fill the phos cells (true by default)
720f7306	169	Bool_t fAreEMCALTriggerEnabled; // whether or not to fill the emcal trigger (true by default)
720f7306	170	Bool_t fArePHOSTriggerEnabled; // whether or not to fill the phos trigger (true by default)
c7ef1c47	171	Bool_t fAreTrackletsEnabled; // whether or not to fill the tracklets (true by default)
	172	AliESDpid* fESDpid; // esd pid
	173	Bool_t fIsPidOwner; // whether we own fESDpid
	174	Int_t fTimeZeroType; // time zero type
2f5ee3b7	175	AliESDtrackCuts* fTPCaloneTrackCuts; // TPC stand-alone track cuts
b8d29a28	176	Double_t fV0Cuts[7]; // Array to store the values for the different reco selections V0 related
b8d29a28	177	Double_t fCascadeCuts[8]; // Array to store the values for the different reco selections cascades related
ed15417e	178	Bool_t fDoPropagateTrackToEMCal; // whether or not to propagate the tracks to EMCal surface (430cm) -- true by default
b8d29a28	179
c7ef1c47	180
08b38f3f	181	ClassDef(AliAnalysisTaskESDfilter, 15); // Analysis task for standard ESD filtering
c7ef1c47	182	};
b8d29a28	183
c7ef1c47	184	#endif