[u/mrichter/AliRoot.git] / PWG0 / dNdEta / AlidNdEtaCorrectionTask.h

/* $Id$ */

#ifndef AlidNdEtaCorrectionTask_H
#define AlidNdEtaCorrectionTask_H

#include "AliAnalysisTask.h"
#include <TString.h>

class AliESDtrackCuts;
class dNdEtaAnalysis;
class AlidNdEtaCorrection;
class TH1F;
class AliESDEvent;
class TParticlePDG;

class AlidNdEtaCorrectionTask : public AliAnalysisTask {
  public:
    enum AnalysisMethod { kSPD = 0, kTPC };

    AlidNdEtaCorrectionTask(const char* opt = "");
    virtual ~AlidNdEtaCorrectionTask();

    virtual void   ConnectInputData(Option_t *);
    virtual void   CreateOutputObjects();
    virtual void   Exec(Option_t*);
    virtual void   Terminate(Option_t *);

    void SetTrackCuts(AliESDtrackCuts* cuts) { fEsdTrackCuts = cuts; }
    void SetAnalysisMode(AnalysisMethod mode) { fAnalysisMode = mode; }

 protected:
    Bool_t SignOK(TParticlePDG* particle);

    AliESDEvent *fESD;               //! ESD object
    TList* fOutput;                  //! list send on output slot 0

    TString fOption;                 // option string
    AnalysisMethod fAnalysisMode;    // detector that is used for analysis
    Int_t fSignMode;                 // if 0 process all particles, if +-1 process only particles with that sign

    AliESDtrackCuts*  fEsdTrackCuts;             // Object containing the parameters of the esd track cuts

    AlidNdEtaCorrection* fdNdEtaCorrection;      //! contains the intermediate histograms (on each slave)

    dNdEtaAnalysis* fdNdEtaAnalysisMC;           //! analysis from MC (only triggered, vertex events)
    dNdEtaAnalysis* fdNdEtaAnalysisESD;          //! analysis from ESD (not yet corrected!)

    // control histograms
    TH1F* fPIDParticles;                         //! pid of primary particles
    TH1F* fPIDTracks;                            //! pid of reconstructed tracks

    // histograms for systematic studies (must be enabled with option)

    TH1F* fSigmaVertexTracks;                    //! (accepted tracks) vs (n of sigma to vertex cut)
    TH1F* fSigmaVertexPrim;                      //! (accepted primaries) vs (n of sigma to vertex cut)
                                                 // enable with option: sigma-vertex

    AlidNdEtaCorrection* fdNdEtaCorrectionProcessType[3]; //! correction for specific process type (ND, SD, DD)
                                                          // enable with option: process-types

 private:
    AlidNdEtaCorrectionTask(const AlidNdEtaCorrectionTask&);
    AlidNdEtaCorrectionTask& operator=(const AlidNdEtaCorrectionTask&);

  ClassDef(AlidNdEtaCorrectionTask, 1);
};

#endif
Commit	Line	Data
0f67a57c	1	/* $Id$ */
	2
	3	#ifndef AlidNdEtaCorrectionTask_H
	4	#define AlidNdEtaCorrectionTask_H
	5
	6	#include "AliAnalysisTask.h"
	7	#include <TString.h>
	8
	9	class AliESDtrackCuts;
	10	class dNdEtaAnalysis;
	11	class AlidNdEtaCorrection;
	12	class TH1F;
	13	class AliESDEvent;
	14	class TParticlePDG;
	15
	16	class AlidNdEtaCorrectionTask : public AliAnalysisTask {
	17	public:
	18	enum AnalysisMethod { kSPD = 0, kTPC };
	19
	20	AlidNdEtaCorrectionTask(const char* opt = "");
	21	virtual ~AlidNdEtaCorrectionTask();
	22
	23	virtual void ConnectInputData(Option_t *);
	24	virtual void CreateOutputObjects();
	25	virtual void Exec(Option_t*);
	26	virtual void Terminate(Option_t *);
	27
	28	void SetTrackCuts(AliESDtrackCuts* cuts) { fEsdTrackCuts = cuts; }
	29	void SetAnalysisMode(AnalysisMethod mode) { fAnalysisMode = mode; }
	30
	31	protected:
	32	Bool_t SignOK(TParticlePDG* particle);
	33
	34	AliESDEvent *fESD; //! ESD object
	35	TList* fOutput; //! list send on output slot 0
	36
	37	TString fOption; // option string
	38	AnalysisMethod fAnalysisMode; // detector that is used for analysis
	39	Int_t fSignMode; // if 0 process all particles, if +-1 process only particles with that sign
	40
	41	AliESDtrackCuts* fEsdTrackCuts; // Object containing the parameters of the esd track cuts
	42
	43	AlidNdEtaCorrection* fdNdEtaCorrection; //! contains the intermediate histograms (on each slave)
	44
	45	dNdEtaAnalysis* fdNdEtaAnalysisMC; //! analysis from MC (only triggered, vertex events)
	46	dNdEtaAnalysis* fdNdEtaAnalysisESD; //! analysis from ESD (not yet corrected!)
	47
	48	// control histograms
	49	TH1F* fPIDParticles; //! pid of primary particles
	50	TH1F* fPIDTracks; //! pid of reconstructed tracks
	51
96fcc8a7	52	// histograms for systematic studies (must be enabled with option)
	53
	54	TH1F* fSigmaVertexTracks; //! (accepted tracks) vs (n of sigma to vertex cut)
	55	TH1F* fSigmaVertexPrim; //! (accepted primaries) vs (n of sigma to vertex cut)
	56	// enable with option: sigma-vertex
	57
	58	AlidNdEtaCorrection* fdNdEtaCorrectionProcessType[3]; //! correction for specific process type (ND, SD, DD)
	59	// enable with option: process-types
	60
0f67a57c	61	private:
	62	AlidNdEtaCorrectionTask(const AlidNdEtaCorrectionTask&);
	63	AlidNdEtaCorrectionTask& operator=(const AlidNdEtaCorrectionTask&);
	64
	65	ClassDef(AlidNdEtaCorrectionTask, 1);
	66	};
	67
	68	#endif