[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFE.h

/**************************************************************************\r
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
*                                                                        *\r
* Author: The ALICE Off-line Project.                                    *\r
* Contributors are mentioned in the code where appropriate.              *\r
*                                                                        *\r
* Permission to use, copy, modify and distribute this software and its   *\r
* documentation strictly for non-commercial purposes is hereby granted   *\r
* without fee, provided that the above copyright notice appears in all   *\r
* copies and that both the copyright notice and this permission notice   *\r
* appear in the supporting documentation. The authors make no claims     *\r
* about the suitability of this software for any purpose. It is          *\r
* provided "as is" without express or implied warranty.                  *\r
**************************************************************************/\r
//\r
// Task for Heavy Flavour Electron Analysis\r
// Fills a single-inclusive electron pt-spectrum\r
// For further information see implementation file\r
//\r
#ifndef ALIANALYSISTASKHFE_H\r
#define ALIANALYSISTASKHFE_H\r
\r
#ifndef ALIANALYSISTASKSE_H\r
#include "AliAnalysisTaskSE.h"\r
#endif\r
\r
#ifndef ROOT_TString\r
#include <TString.h>\r
#endif\r
\r
#ifndef ROOT_TBits\r
#include <TBits.h>\r
#endif\r
\r
class AliAnalysisUtils;\r
class AliESDtrackCuts;\r
class AliHFEcontainer;\r
class AliHFEcollection;\r
class AliHFEcuts;\r
class AliHFEextraCuts;\r
class AliHFEelecbackground;\r
class AliHFENonPhotonicElectron;\r
class AliHFEmcQA;\r
class AliHFEpid;\r
class AliHFEpidQAmanager;\r
class AliHFEsecVtx;\r
class AliHFEsignalCuts;\r
class AliHFEvarManager;\r
class AliHFEtaggedTrackAnalysis;\r
class AliCFManager;\r
class AliMCEvent;\r
class AliOADBContainer;\r
class AliAODMCHeader;\r
class AliVEvent;\r
class AliVParticle;\r
class AliTriggerAnalysis;\r
class TH1I; \r
class TList;\r
class TClonesArray;\r
\r
class AliAnalysisTaskHFE : public AliAnalysisTaskSE{\r
  public:\r
    enum{\r
      kPIDqa = 0,\r
      kMCqa =1 \r
    };\r
    enum{\r
      kPriVtx = 0,\r
      kSecVtx = 1,\r
      kIsElecBackGround = 2,\r
      kPostProcess = 3,\r
      kDEstep = 4,\r
      kTaggedTrackAnalysis = 5,\r
      kNonPhotonicElectron = 6\r
    };\r
    enum CreationProcess_t{\r
      kSignalCharm = 0,\r
      kSignalBeauty = 1,\r
      kGammaConv = 2,\r
      kOther = 3\r
    };\r
    enum{\r
      kBgPtBins = 44,\r
      kElecBgSpecies = 6,\r
      kCentBins = 11,\r
      kBgLevels = 3\r
    };\r
    typedef enum{\r
      kpp = 0,\r
      kpPb = 1,\r
      kPbPb = 2\r
    } ECollisionSystem_t;\r
\r
    AliAnalysisTaskHFE();\r
    AliAnalysisTaskHFE(const char * name);\r
    AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref);\r
    AliAnalysisTaskHFE& operator=(const AliAnalysisTaskHFE &ref);\r
    virtual void Copy(TObject &o) const;\r
    virtual ~AliAnalysisTaskHFE();\r
\r
    virtual void UserCreateOutputObjects();\r
    virtual void UserExec(Option_t *);\r
    virtual void Terminate(Option_t *);\r
\r
    virtual Bool_t IsEventInBinZero();\r
\r
    Bool_t IsQAOn(Int_t qaLevel) const { return TESTBIT(fQAlevel, qaLevel); };\r
    Bool_t IsAODanalysis() const { return TestBit(kAODanalysis); };\r
    Bool_t IsESDanalysis() const { return !TestBit(kAODanalysis); };\r
    Bool_t HasMCData() const { return TestBit(kHasMCdata); }\r
    Bool_t Ispp() const { return fCollisionSystem.TestBitNumber(kpp); }\r
    Bool_t IsPbPb() const { return fCollisionSystem.TestBitNumber(kPbPb); }\r
    Bool_t IspPb() const { return fCollisionSystem.TestBitNumber(kpPb); }\r
    Bool_t IsHeavyIon() const { return IsPbPb() || IspPb(); }\r
    Bool_t GetPlugin(Int_t plug) const { return TESTBIT(fPlugins, plug); };\r
\r
    // Get Components for configuration\r
    AliHFEvarManager *GetVarManager() const { return fVarManager; }\r
    AliHFEpidQAmanager *GetPIDQAManager() const { return fPIDqa; }\r
    AliHFEpid *GetPID() const { return fPID; }\r
    AliHFENonPhotonicElectron *GetHFEBackgroundSubtraction() const { return fBackgroundSubtraction; }\r
\r
    void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };\r
    void SetTaggedTrackCuts(AliHFEcuts * const cuts) { fTaggedTrackCuts = cuts; }\r
    void SetCleanTaggedTrack(Bool_t clean) { fCleanTaggedTrack = clean; };\r
    void SetVariablesTRDTaggedTrack(Bool_t variablesTRD) { fVariablesTRDTaggedTrack = variablesTRD; };\r
    void SetHFECutsPreselect(AliESDtrackCuts * const cuts) { fCutspreselect = cuts; };\r
    void SetHFEElecBackGround(AliHFEelecbackground * const elecBackGround) { fElecBackGround = elecBackGround; };\r
    void SetHFEBackgroundSubtraction(AliHFENonPhotonicElectron * const backgroundSubtraction) { fBackgroundSubtraction = backgroundSubtraction; };\r
    void SetQAOn(Int_t qaLevel) { SETBIT(fQAlevel, qaLevel); };\r
    void SwitchOnPlugin(Int_t plug);\r
    void SetHasMCData(Bool_t hasMC = kTRUE) { SetBit(kHasMCdata, hasMC); };\r
    void SetFillSignalOnly(Bool_t signalOnly) { fFillSignalOnly = signalOnly; }\r
   \r
    void SetFillNoCuts(Bool_t fillNoCuts) { fFillNoCuts = fillNoCuts; }\r
    void SetApplyCutAOD(Bool_t applyCutAOD)   { fApplyCutAOD = applyCutAOD; }\r
    void SetRemovePileUp(Bool_t removePileUp) { fRemovePileUp = removePileUp; }\r
    void SetRemoveFirstEventInChunk() {fRemoveFirstEvent = kTRUE;}\r
    void SetPIDPreselect(AliHFEpid * const cuts) { fPIDpreselect = cuts; };\r
    void SetAODAnalysis() { SetBit(kAODanalysis, kTRUE); };\r
    void SetESDAnalysis() { SetBit(kAODanalysis, kFALSE); };\r
    void SetTRDTrigger(Bool_t activateTRDTrigger, Int_t trdtrigger) {fTRDTrigger=activateTRDTrigger; fWhichTRDTrigger=trdtrigger;};\r
    void SetCollisionSystem(ECollisionSystem_t system){\r
      fCollisionSystem.Clear();\r
      fCollisionSystem.SetBitNumber(system, kTRUE);\r
    }\r
    void SetppAnalysis(){\r
      fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
      fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
      fCollisionSystem.SetBitNumber(kpp, kTRUE); \r
    }\r
    void SetpPbAnalysis() {\r
      fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
      fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
      fCollisionSystem.SetBitNumber(kpPb, kTRUE); \r
    }\r
    void SetPbPbAnalysis() { \r
      fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
      fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
      fCollisionSystem.SetBitNumber(kPbPb, kTRUE); \r
    };\r
    void SetCentralityEstimator(const char *estimator) { fCentralityEstimator = estimator; }\r
    void SetPbPbUserCentralityLimit(Bool_t isPbPbUserBinning = kFALSE){fPbPbUserCentralityBinning = isPbPbUserBinning; };\r
    void SetPbPbUserCentralityArray(Int_t icentr, Float_t valuecentr) {fCentralityLimits[icentr] = valuecentr;};\r
    void SetPPMultiBinAnalysis(Bool_t isppMultiBin) { fisppMultiBin = isppMultiBin; };\r
    void SetNonHFEsystematics(Bool_t isSystematics) {fisNonHFEsystematics = isSystematics; };\r
    void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };\r
    void SetBackGroundFactorsFunction(const TF1 * const backGroundFactorsFunction, Int_t centralitybin=0){  \r
      fkBackGroundFactorArray[centralitybin]=backGroundFactorsFunction;\r
      fBackGroundFactorApply=kTRUE;\r
      SetBit(kBackgroundInitialized);\r
    };\r
    void SetElecBackGroundFactors(Int_t iPt, Int_t iType, Int_t iCent, Int_t iError, Double_t elecBackGroundFactor) {fElecBackgroundFactor[iError][iCent][iType][iPt] = elecBackGroundFactor; };\r
    void SetBinLimits(Int_t iPt, Double_t momentum){fBinLimit[iPt] = momentum;};\r
    void PrintStatus() const;\r
    Bool_t ReadCentrality();\r
    void RejectionPileUpVertexRangeEventCut();  \r
    void SelectSpecialTrigger(const Char_t *trgclust, Int_t runMin = 0, Int_t runMax = 999999999); \r
    void SetDebugStreaming() {SetBit(kTreeStream);};\r
    Bool_t CheckTRDTrigger(AliESDEvent *ev);\r
    void DrawTRDTrigger(AliESDEvent *ev);
\r
  private:\r
    enum{\r
      kHasMCdata = BIT(19),\r
      kAODanalysis = BIT(20),\r
      kBackgroundInitialized = BIT(21),\r
      kTreeStream = BIT(22)\r
    };\r
\r
    Bool_t FillProductionVertex(const AliVParticle * const track) const;\r
    void MakeParticleContainer();\r
    void MakeEventContainer();\r
    void InitHistoITScluster();\r
    void InitContaminationQA();\r
    const Char_t *GetSpecialTrigger(Int_t run);\r
    void ProcessMC();\r
    void ProcessESD();\r
    void ProcessAOD();\r
    Int_t GetITSMultiplicity(AliVEvent *ev);\r
    Bool_t PreSelectTrack(AliESDtrack *track) const;\r
    Bool_t ProcessMCtrack(AliVParticle *track);\r
    Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);\r
    AliAODMCHeader *fAODMCHeader;         // ! MC info AOD\r
    TClonesArray *fAODArrayMCInfo;        // ! MC info particle AOD\r
    ULong_t fQAlevel;                     // QA level\r
    UShort_t fPlugins;                    // Enabled Plugins\r
    TBits fCollisionSystem;              // Collision System;\r
    Bool_t fFillSignalOnly;               // Fill container only with MC Signal Tracks\r
    Bool_t fFillNoCuts;                   // Fill container before any cut\r
    Bool_t fApplyCutAOD;                  // Apply the analysis cut for AOD tracks\r
    Bool_t fBackGroundFactorApply;        // Apply Background Function Subtraction,   MF: To be removed when transition to OADB container is finished\r
    Bool_t fRemovePileUp;                 // Remove Pile Up\r
    Bool_t fIdentifiedAsPileUp;           // Identified as pile-up\r
    Bool_t fIdentifiedAsOutInz;           // Out Of Range in z\r
    Bool_t fPassTheEventCut;              // Pass The Event Cut\r
    Bool_t fRejectKinkMother;             // Reject Kink Mother\r
    Bool_t fisppMultiBin;                 // pp Multiplicity Bin analysis\r
    Bool_t fPbPbUserCentralityBinning;    // PbPb user centrality binning\r
    Bool_t fRemoveFirstEvent;             // Remove first event from chunk\r
    Bool_t fisNonHFEsystematics;          // Non-HFE background systematics analysis\r
    AliOADBContainer *fSpecialTrigger;    // Special trigger selection\r
    Int_t   fCentralityF;                 // Centrality bin\r
    Float_t fCentralityPercent;           // Centrality percentile\r
    TString fCentralityEstimator;         // Centrality Estimator\r
    Float_t fContributors;                // Contributors\r
    Double_t fWeightBackGround;            // weight background function\r
    Double_t fVz;                         // z position of the primary vertex\r
    const TF1  *fkBackGroundFactorArray[12];   // Array of BackGround factors for each centrality bin, bin0 = min bias\r
    Double_t fElecBackgroundFactor[kBgLevels][kCentBins][kElecBgSpecies][kBgPtBins];     // Electron background factors\r
    Double_t fBinLimit[kBgPtBins+1];      // Electron pt bin edges\r
    Float_t fCentralityLimits[12];        // Limits for centrality bins\r
    AliHFEcontainer *fContainer;          //! The HFE container\r
    AliHFEvarManager *fVarManager;        // The var manager as the backbone of the analysis\r
    AliHFEsignalCuts *fSignalCuts;        //! MC true signal (electron coming from certain source) \r
    AliCFManager *fCFM;                   //! Correction Framework Manager\r
    AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis for Normalisation\r
    AliHFEpid *fPID;                      // PID\r
    AliHFEpidQAmanager *fPIDqa;           // PID QA\r
    AliHFEpid *fPIDpreselect;             // PID oject for pre-selected tracks (without QA)\r
    AliHFEcuts *fCuts;                    // Cut Collection\r
    AliHFEcuts *fTaggedTrackCuts;         // Cut Collection for V0 tagged tracks\r
    Bool_t fCleanTaggedTrack;             // Loose cleaning of the V0 tagged tracks electron\r
    Bool_t fVariablesTRDTaggedTrack;      // Take the variables at the TRD for the V0 tagged tracks electron\r
    AliAnalysisUtils *fAnalysisUtils;     // Utility object to remove the first event of a chunk from the analysis\r
    AliESDtrackCuts *fCutspreselect;      // Cut Collection for pre-selected tracks\r
    AliHFEsecVtx *fSecVtx;                //! Secondary Vertex Analysis\r
    AliHFEelecbackground *fElecBackGround;//! Background analysis\r
    AliHFEmcQA *fMCQA;                    //! MC QA\r
    AliHFEtaggedTrackAnalysis *fTaggedTrackAnalysis;     //!Analyse V0-tagged tracks\r
    AliHFEextraCuts *fExtraCuts;          //! temporary implementation for IP QA\r
    AliHFENonPhotonicElectron *fBackgroundSubtraction; // Background subtraction\r
    Bool_t fTRDTrigger;                   // Check if event is TRD triggered event\r
    Int_t  fWhichTRDTrigger;               // Select type of TRD trigger\r
    //-----------QA and output---------------\r
    TList *fQA;                           //! QA histos for the cuts\r
    TList *fOutput;                       //! Container for Task Output\r
    TList *fHistMCQA;                     //! Output container for MC QA histograms \r
    TList *fHistSECVTX;                   //! Output container for sec. vertexing results\r
    TList *fHistELECBACKGROUND;           //! Output container for electron background analysis\r
    AliHFEcollection *fQACollection;      //! Tasks own QA collection\r
    //---------------------------------------\r
\r
    ClassDef(AliAnalysisTaskHFE, 2)       // The electron Analysis Task\r
};\r
#endif\r
\r
Commit	Line	Data
051db1cf	1	/**************************************************************************\r
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
	3	* *\r
	4	* Author: The ALICE Off-line Project. *\r
	5	* Contributors are mentioned in the code where appropriate. *\r
	6	* *\r
	7	* Permission to use, copy, modify and distribute this software and its *\r
	8	* documentation strictly for non-commercial purposes is hereby granted *\r
	9	* without fee, provided that the above copyright notice appears in all *\r
	10	* copies and that both the copyright notice and this permission notice *\r
	11	* appear in the supporting documentation. The authors make no claims *\r
	12	* about the suitability of this software for any purpose. It is *\r
	13	* provided "as is" without express or implied warranty. *\r
	14	**************************************************************************/\r
	15	//\r
	16	// Task for Heavy Flavour Electron Analysis\r
	17	// Fills a single-inclusive electron pt-spectrum\r
	18	// For further information see implementation file\r
	19	//\r
	20	#ifndef ALIANALYSISTASKHFE_H\r
	21	#define ALIANALYSISTASKHFE_H\r
	22	\r
	23	#ifndef ALIANALYSISTASKSE_H\r
	24	#include "AliAnalysisTaskSE.h"\r
	25	#endif\r
	26	\r
	27	#ifndef ROOT_TString\r
	28	#include <TString.h>\r
	29	#endif\r
	30	\r
	31	#ifndef ROOT_TBits\r
	32	#include <TBits.h>\r
	33	#endif\r
	34	\r
	35	class AliAnalysisUtils;\r
	36	class AliESDtrackCuts;\r
	37	class AliHFEcontainer;\r
	38	class AliHFEcollection;\r
	39	class AliHFEcuts;\r
	40	class AliHFEextraCuts;\r
	41	class AliHFEelecbackground;\r
	42	class AliHFENonPhotonicElectron;\r
	43	class AliHFEmcQA;\r
	44	class AliHFEpid;\r
	45	class AliHFEpidQAmanager;\r
	46	class AliHFEsecVtx;\r
	47	class AliHFEsignalCuts;\r
	48	class AliHFEvarManager;\r
	49	class AliHFEtaggedTrackAnalysis;\r
	50	class AliCFManager;\r
	51	class AliMCEvent;\r
	52	class AliOADBContainer;\r
	53	class AliAODMCHeader;\r
	54	class AliVEvent;\r
	55	class AliVParticle;\r
	56	class AliTriggerAnalysis;\r
	57	class TH1I; \r
	58	class TList;\r
	59	class TClonesArray;\r
	60	\r
	61	class AliAnalysisTaskHFE : public AliAnalysisTaskSE{\r
	62	public:\r
	63	enum{\r
	64	kPIDqa = 0,\r
65	kMCqa =1 \r
66	};\r
67	enum{\r
68	kPriVtx = 0,\r
69	kSecVtx = 1,\r
70	kIsElecBackGround = 2,\r
71	kPostProcess = 3,\r
72	kDEstep = 4,\r
73	kTaggedTrackAnalysis = 5,\r
74	kNonPhotonicElectron = 6\r
75	};\r
76	enum CreationProcess_t{\r
77	kSignalCharm = 0,\r
78	kSignalBeauty = 1,\r
79	kGammaConv = 2,\r
80	kOther = 3\r
81	};\r
82	enum{\r
83	kBgPtBins = 44,\r
84	kElecBgSpecies = 6,\r
85	kCentBins = 11,\r
86	kBgLevels = 3\r
87	};\r
88	typedef enum{\r
89	kpp = 0,\r
90	kpPb = 1,\r
91	kPbPb = 2\r
92	} ECollisionSystem_t;\r
93	\r
94	AliAnalysisTaskHFE();\r
95	AliAnalysisTaskHFE(const char * name);\r
96	AliAnalysisTaskHFE(const AliAnalysisTaskHFE &ref);\r
97	AliAnalysisTaskHFE& operator=(const AliAnalysisTaskHFE &ref);\r
98	virtual void Copy(TObject &o) const;\r
99	virtual ~AliAnalysisTaskHFE();\r
100	\r
101	virtual void UserCreateOutputObjects();\r
102	virtual void UserExec(Option_t *);\r
103	virtual void Terminate(Option_t *);\r
104	\r
105	virtual Bool_t IsEventInBinZero();\r
106	\r
107	Bool_t IsQAOn(Int_t qaLevel) const { return TESTBIT(fQAlevel, qaLevel); };\r
108	Bool_t IsAODanalysis() const { return TestBit(kAODanalysis); };\r
109	Bool_t IsESDanalysis() const { return !TestBit(kAODanalysis); };\r
110	Bool_t HasMCData() const { return TestBit(kHasMCdata); }\r
111	Bool_t Ispp() const { return fCollisionSystem.TestBitNumber(kpp); }\r
112	Bool_t IsPbPb() const { return fCollisionSystem.TestBitNumber(kPbPb); }\r
113	Bool_t IspPb() const { return fCollisionSystem.TestBitNumber(kpPb); }\r
114	Bool_t IsHeavyIon() const { return IsPbPb() \|\| IspPb(); }\r
115	Bool_t GetPlugin(Int_t plug) const { return TESTBIT(fPlugins, plug); };\r
116	\r
117	// Get Components for configuration\r
118	AliHFEvarManager *GetVarManager() const { return fVarManager; }\r
119	AliHFEpidQAmanager *GetPIDQAManager() const { return fPIDqa; }\r
120	AliHFEpid *GetPID() const { return fPID; }\r
121	AliHFENonPhotonicElectron *GetHFEBackgroundSubtraction() const { return fBackgroundSubtraction; }\r
122	\r
123	void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };\r
124	void SetTaggedTrackCuts(AliHFEcuts * const cuts) { fTaggedTrackCuts = cuts; }\r
125	void SetCleanTaggedTrack(Bool_t clean) { fCleanTaggedTrack = clean; };\r
126	void SetVariablesTRDTaggedTrack(Bool_t variablesTRD) { fVariablesTRDTaggedTrack = variablesTRD; };\r
127	void SetHFECutsPreselect(AliESDtrackCuts * const cuts) { fCutspreselect = cuts; };\r
128	void SetHFEElecBackGround(AliHFEelecbackground * const elecBackGround) { fElecBackGround = elecBackGround; };\r
129	void SetHFEBackgroundSubtraction(AliHFENonPhotonicElectron * const backgroundSubtraction) { fBackgroundSubtraction = backgroundSubtraction; };\r
130	void SetQAOn(Int_t qaLevel) { SETBIT(fQAlevel, qaLevel); };\r
131	void SwitchOnPlugin(Int_t plug);\r
132	void SetHasMCData(Bool_t hasMC = kTRUE) { SetBit(kHasMCdata, hasMC); };\r
133	void SetFillSignalOnly(Bool_t signalOnly) { fFillSignalOnly = signalOnly; }\r
134	\r
135	void SetFillNoCuts(Bool_t fillNoCuts) { fFillNoCuts = fillNoCuts; }\r
136	void SetApplyCutAOD(Bool_t applyCutAOD) { fApplyCutAOD = applyCutAOD; }\r
137	void SetRemovePileUp(Bool_t removePileUp) { fRemovePileUp = removePileUp; }\r
138	void SetRemoveFirstEventInChunk() {fRemoveFirstEvent = kTRUE;}\r
139	void SetPIDPreselect(AliHFEpid * const cuts) { fPIDpreselect = cuts; };\r
140	void SetAODAnalysis() { SetBit(kAODanalysis, kTRUE); };\r
141	void SetESDAnalysis() { SetBit(kAODanalysis, kFALSE); };\r
142	void SetTRDTrigger(Bool_t activateTRDTrigger, Int_t trdtrigger) {fTRDTrigger=activateTRDTrigger; fWhichTRDTrigger=trdtrigger;};\r
143	void SetCollisionSystem(ECollisionSystem_t system){\r
144	fCollisionSystem.Clear();\r
145	fCollisionSystem.SetBitNumber(system, kTRUE);\r
146	}\r
147	void SetppAnalysis(){\r
148	fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
149	fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
150	fCollisionSystem.SetBitNumber(kpp, kTRUE); \r
151	}\r
152	void SetpPbAnalysis() {\r
153	fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
154	fCollisionSystem.SetBitNumber(kPbPb, kFALSE); \r
155	fCollisionSystem.SetBitNumber(kpPb, kTRUE); \r
156	}\r
157	void SetPbPbAnalysis() { \r
158	fCollisionSystem.SetBitNumber(kpp, kFALSE); \r
159	fCollisionSystem.SetBitNumber(kpPb, kFALSE); \r
160	fCollisionSystem.SetBitNumber(kPbPb, kTRUE); \r
161	};\r
162	void SetCentralityEstimator(const char *estimator) { fCentralityEstimator = estimator; }\r
163	void SetPbPbUserCentralityLimit(Bool_t isPbPbUserBinning = kFALSE){fPbPbUserCentralityBinning = isPbPbUserBinning; };\r
164	void SetPbPbUserCentralityArray(Int_t icentr, Float_t valuecentr) {fCentralityLimits[icentr] = valuecentr;};\r
165	void SetPPMultiBinAnalysis(Bool_t isppMultiBin) { fisppMultiBin = isppMultiBin; };\r
166	void SetNonHFEsystematics(Bool_t isSystematics) {fisNonHFEsystematics = isSystematics; };\r
167	void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };\r
168	void SetBackGroundFactorsFunction(const TF1 * const backGroundFactorsFunction, Int_t centralitybin=0){ \r
169	fkBackGroundFactorArray[centralitybin]=backGroundFactorsFunction;\r
170	fBackGroundFactorApply=kTRUE;\r
171	SetBit(kBackgroundInitialized);\r
172	};\r
173	void SetElecBackGroundFactors(Int_t iPt, Int_t iType, Int_t iCent, Int_t iError, Double_t elecBackGroundFactor) {fElecBackgroundFactor[iError][iCent][iType][iPt] = elecBackGroundFactor; };\r
174	void SetBinLimits(Int_t iPt, Double_t momentum){fBinLimit[iPt] = momentum;};\r
175	void PrintStatus() const;\r
176	Bool_t ReadCentrality();\r
177	void RejectionPileUpVertexRangeEventCut(); \r
178	void SelectSpecialTrigger(const Char_t *trgclust, Int_t runMin = 0, Int_t runMax = 999999999); \r
179	void SetDebugStreaming() {SetBit(kTreeStream);};\r
180	Bool_t CheckTRDTrigger(AliESDEvent *ev);\r
8c07fb0d	181	void DrawTRDTrigger(AliESDEvent *ev);
051db1cf	182	\r
	183	private:\r
	184	enum{\r
	185	kHasMCdata = BIT(19),\r
	186	kAODanalysis = BIT(20),\r
	187	kBackgroundInitialized = BIT(21),\r
	188	kTreeStream = BIT(22)\r
	189	};\r
	190	\r
	191	Bool_t FillProductionVertex(const AliVParticle * const track) const;\r
	192	void MakeParticleContainer();\r
	193	void MakeEventContainer();\r
	194	void InitHistoITScluster();\r
	195	void InitContaminationQA();\r
	196	const Char_t *GetSpecialTrigger(Int_t run);\r
	197	void ProcessMC();\r
	198	void ProcessESD();\r
	199	void ProcessAOD();\r
	200	Int_t GetITSMultiplicity(AliVEvent *ev);\r
	201	Bool_t PreSelectTrack(AliESDtrack *track) const;\r
	202	Bool_t ProcessMCtrack(AliVParticle *track);\r
	203	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);\r
	204	AliAODMCHeader *fAODMCHeader; // ! MC info AOD\r
	205	TClonesArray *fAODArrayMCInfo; // ! MC info particle AOD\r
	206	ULong_t fQAlevel; // QA level\r
	207	UShort_t fPlugins; // Enabled Plugins\r
	208	TBits fCollisionSystem; // Collision System;\r
	209	Bool_t fFillSignalOnly; // Fill container only with MC Signal Tracks\r
	210	Bool_t fFillNoCuts; // Fill container before any cut\r
	211	Bool_t fApplyCutAOD; // Apply the analysis cut for AOD tracks\r
	212	Bool_t fBackGroundFactorApply; // Apply Background Function Subtraction, MF: To be removed when transition to OADB container is finished\r
	213	Bool_t fRemovePileUp; // Remove Pile Up\r
	214	Bool_t fIdentifiedAsPileUp; // Identified as pile-up\r
	215	Bool_t fIdentifiedAsOutInz; // Out Of Range in z\r
	216	Bool_t fPassTheEventCut; // Pass The Event Cut\r
	217	Bool_t fRejectKinkMother; // Reject Kink Mother\r
	218	Bool_t fisppMultiBin; // pp Multiplicity Bin analysis\r
	219	Bool_t fPbPbUserCentralityBinning; // PbPb user centrality binning\r
	220	Bool_t fRemoveFirstEvent; // Remove first event from chunk\r
	221	Bool_t fisNonHFEsystematics; // Non-HFE background systematics analysis\r
	222	AliOADBContainer *fSpecialTrigger; // Special trigger selection\r
	223	Int_t fCentralityF; // Centrality bin\r
	224	Float_t fCentralityPercent; // Centrality percentile\r
	225	TString fCentralityEstimator; // Centrality Estimator\r
	226	Float_t fContributors; // Contributors\r
	227	Double_t fWeightBackGround; // weight background function\r
	228	Double_t fVz; // z position of the primary vertex\r
	229	const TF1 *fkBackGroundFactorArray[12]; // Array of BackGround factors for each centrality bin, bin0 = min bias\r
	230	Double_t fElecBackgroundFactor[kBgLevels][kCentBins][kElecBgSpecies][kBgPtBins]; // Electron background factors\r
	231	Double_t fBinLimit[kBgPtBins+1]; // Electron pt bin edges\r
	232	Float_t fCentralityLimits[12]; // Limits for centrality bins\r
	233	AliHFEcontainer *fContainer; //! The HFE container\r
	234	AliHFEvarManager *fVarManager; // The var manager as the backbone of the analysis\r
	235	AliHFEsignalCuts *fSignalCuts; //! MC true signal (electron coming from certain source) \r
	236	AliCFManager *fCFM; //! Correction Framework Manager\r
	237	AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis for Normalisation\r
	238	AliHFEpid *fPID; // PID\r
	239	AliHFEpidQAmanager *fPIDqa; // PID QA\r
	240	AliHFEpid *fPIDpreselect; // PID oject for pre-selected tracks (without QA)\r
	241	AliHFEcuts *fCuts; // Cut Collection\r
	242	AliHFEcuts *fTaggedTrackCuts; // Cut Collection for V0 tagged tracks\r
	243	Bool_t fCleanTaggedTrack; // Loose cleaning of the V0 tagged tracks electron\r
	244	Bool_t fVariablesTRDTaggedTrack; // Take the variables at the TRD for the V0 tagged tracks electron\r
	245	AliAnalysisUtils *fAnalysisUtils; // Utility object to remove the first event of a chunk from the analysis\r
246	AliESDtrackCuts *fCutspreselect; // Cut Collection for pre-selected tracks\r
247	AliHFEsecVtx *fSecVtx; //! Secondary Vertex Analysis\r
248	AliHFEelecbackground *fElecBackGround;//! Background analysis\r
249	AliHFEmcQA *fMCQA; //! MC QA\r
250	AliHFEtaggedTrackAnalysis *fTaggedTrackAnalysis; //!Analyse V0-tagged tracks\r
251	AliHFEextraCuts *fExtraCuts; //! temporary implementation for IP QA\r
252	AliHFENonPhotonicElectron *fBackgroundSubtraction; // Background subtraction\r
253	Bool_t fTRDTrigger; // Check if event is TRD triggered event\r
254	Int_t fWhichTRDTrigger; // Select type of TRD trigger\r
255	//-----------QA and output---------------\r
256	TList *fQA; //! QA histos for the cuts\r
257	TList *fOutput; //! Container for Task Output\r
258	TList *fHistMCQA; //! Output container for MC QA histograms \r
259	TList *fHistSECVTX; //! Output container for sec. vertexing results\r
260	TList *fHistELECBACKGROUND; //! Output container for electron background analysis\r
261	AliHFEcollection *fQACollection; //! Tasks own QA collection\r
262	//---------------------------------------\r
263	\r
264	ClassDef(AliAnalysisTaskHFE, 2) // The electron Analysis Task\r
265	};\r
266	#endif\r
267	\r