[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliAnalysisTaskLocalRho.h

#ifndef ALIANALYSISTASKLOCALRHO_H
#define ALIANALYSISTASKLOCALRHO_H

// $Id$

#include <AliAnalysisTaskEmcalJet.h>
#include <AliEmcalJet.h>
#include <AliVEvent.h>
#include <AliVTrack.h>
#include <AliVCluster.h>
#include <TClonesArray.h>
#include <TMath.h>
#include <TRandom3.h>

class TF1;
class THF1;
class THF2;
class TProfile;
class AliLocalRhoParameter;

class AliAnalysisTaskLocalRho : public AliAnalysisTaskEmcalJet
{
    public:
         // enumerators
        enum fitModulationType  { kNoFit, kV2, kV3, kCombined, kFourierSeries, kIntegratedFlow, kQC2, kQC4 }; // fit type
        enum detectorType       { kTPC, kVZEROA, kVZEROC, kVZEROComb};  // detector that was used
        enum qcRecovery         { kFixedRho, kNegativeVn, kTryFit };    // how to deal with negative cn value for qcn value
        enum runModeType        { kLocal, kGrid };                      // run mode type
        // constructors, destructor
                                AliAnalysisTaskLocalRho();
                                AliAnalysisTaskLocalRho(const char *name, runModeType type);
        virtual                 ~AliAnalysisTaskLocalRho();
        // setting up the task and technical aspects
        Bool_t                  InitializeAnalysis();
        virtual void            UserCreateOutputObjects();
        TH1F*                   BookTH1F(const char* name, const char* x, Int_t bins, Double_t min, Double_t max, Int_t c = -1, Bool_t append = kTRUE);
        TH2F*                   BookTH2F(const char* name, const char* x, const char* y, Int_t binsx, Double_t minx, Double_t maxx, Int_t binsy, Double_t miny, Double_t maxy, Int_t c = -1, Bool_t append = kTRUE);
        virtual Bool_t          Run();
        /* inline */    Double_t PhaseShift(Double_t x) const {  
            while (x>=TMath::TwoPi())x-=TMath::TwoPi();
            while (x<0.)x+=TMath::TwoPi();
            return x; }
        /* inline */    Double_t PhaseShift(Double_t x, Double_t n) const {
            x = PhaseShift(x);
            if(TMath::Nint(n)==2) while (x>TMath::Pi()) x-=TMath::Pi();
            if(TMath::Nint(n)==3) {
                if(x>2.*TMath::TwoPi()/n) x = TMath::TwoPi() - x;
                if(x>TMath::TwoPi()/n) x = TMath::TwoPi()-(x+TMath::TwoPi()/n);
            }
            return x; }
        /* inline */    Double_t ChiSquarePDF(Int_t ndf, Double_t x) const {
            Double_t n(ndf/2.), denom(TMath::Power(2, n)*TMath::Gamma(n));
            if (denom!=0)  return ((1./denom)*TMath::Power(x, n-1)*TMath::Exp(-x/2.)); 
            return -999; }
        // note that the cdf of the chisquare distribution is the normalized lower incomplete gamma function
        /* inline */    Double_t ChiSquareCDF(Int_t ndf, Double_t x) const { return TMath::Gamma(ndf/2., x/2.); }
        // setters - setup how to run
        void                    SetDebugMode(Int_t d)                           {fDebug = d;}
        void                    SetCentralityClasses(TArrayI* c)                {fCentralityClasses = c;}
        void                    SetAttachToEvent(Bool_t a)                      {fAttachToEvent = a;}
        void                    SetLocalRhoName(TString n)                      {fLocalRhoName = n;}
        void                    SetFillHistograms(Bool_t b)                     {fFillHistograms = b;}
        // setters - analysis details
        void                    SetNoEventWeightsForQC(Bool_t e)                {fNoEventWeightsForQC = e;}
        void                    SetIntegratedFlow(TH1F* i, TH1F* j)             {fUserSuppliedV2 = i;
                                                                                 fUserSuppliedV3 = j; }
        void                    SetOnTheFlyResCorrection(TH1F* r2, TH1F* r3)    {fUserSuppliedR2 = r2;
                                                                                 fUserSuppliedR3 = r3; }
        void                    SetModulationFit(TF1* fit)                      {if (fFitModulation) delete fFitModulation;
                                                                                 fFitModulation = fit; }
        void                    SetModulationFitMinMaxP(Float_t m, Float_t n)   {fMinPvalue = m; fMaxPvalue = n; }
        void                    SetModulationFitType(fitModulationType type)    {fFitModulationType = type; }
        void                    SetQCnRecoveryType(qcRecovery type)             {fQCRecovery = type; }
        void                    SetModulationFitOptions(TString opt)            {fFitModulationOptions = opt; }
        void                    SetReferenceDetector(detectorType type)         {fDetectorType = type; }
        void                    SetUsePtWeight(Bool_t w)                        {fUsePtWeight = w; }
        void                    SetRunModeType(runModeType type)                {fRunModeType = type; }
        void                    SetForceAbsVnHarmonics(Bool_t f)                {fAbsVnHarmonics = f; }
        void                    SetExcludeLeadingJetsFromFit(Float_t n)         {fExcludeLeadingJetsFromFit = n; }
        void                    SetRebinSwapHistoOnTheFly(Bool_t r)             {fRebinSwapHistoOnTheFly = r; }
        void                    SetSaveThisPercentageOfFits(Float_t p)          {fPercentageOfFits = p; }
        void                    SetUseV0EventPlaneFromHeader(Bool_t h)          {fUseV0EventPlaneFromHeader = h;}
        void                    SetSoftTrackMinMaxPt(Float_t min, Float_t max)  {fSoftTrackMinPt = min; fSoftTrackMaxPt = max;}
        // getters
        TString                 GetLocalRhoName() const                         {return fLocalRhoName; }
        // numerical evaluations
        void                    CalculateEventPlaneVZERO(Double_t vzero[2][2]) const;
        void                    CalculateEventPlaneTPC(Double_t* tpc);
        void                    CalculateEventPlaneCombinedVZERO(Double_t* comb) const;
        Double_t                CalculateQC2(Int_t harm);
        Double_t                CalculateQC4(Int_t harm);
        // helper calculations for the q-cumulant analysis, also used by AliAnalyisTaskJetFlow
        void                    QCnQnk(Int_t n, Int_t k, Double_t &reQ, Double_t &imQ);
        Double_t                QCnS(Int_t i, Int_t j);
        Double_t                QCnM();
        Double_t                QCnM11();
        Double_t                QCnM1111();
        Bool_t                  QCnRecovery(Double_t psi2, Double_t psi3);
        // analysis details
        Bool_t                  CorrectRho(Double_t psi2, Double_t psi3);
        void                    FillEventPlaneHistograms(Double_t psi2, Double_t psi3) const;
        void                    FillAnalysisSummaryHistogram() const;
        // track selection
        /* inline */    Bool_t PassesCuts(const AliVTrack* track) const {
            if(!track) return kFALSE;
            return (track->Pt() < fTrackPtCut || track->Eta() < fTrackMinEta || track->Eta() > fTrackMaxEta || track->Phi() < fTrackMinPhi || track->Phi() > fTrackMaxPhi) ? kFALSE : kTRUE; }
        /* inline */    Bool_t PassesCuts(AliEmcalJet* jet) const {
            if(!jet || fJetRadius <= 0) return kFALSE;
            return (jet->Pt() < fJetPtCut || jet->Area()/(fJetRadius*fJetRadius*TMath::Pi()) < fPercAreaCut || jet->Eta() < fJetMinEta || jet->Eta() > fJetMaxEta || jet->Phi() < fJetMinPhi || jet->Phi() > fJetMaxPhi) ? kFALSE : kTRUE; }
        // filling histograms
        virtual void            Terminate(Option_t* option);

    private: 
        Int_t                   fDebug;                 // debug level (0 none, 1 fcn calls, 2 verbose)
        Bool_t                  fInitialized;           //! is the analysis initialized?
        Bool_t                  fAttachToEvent;         // attach local rho to the event
        Bool_t                  fFillHistograms;        // fill qa histograms
        Bool_t                  fNoEventWeightsForQC;   // don't store event weights for qc analysis
        TString                 fLocalRhoName;          // name for local rho
        TArrayI*                fCentralityClasses;     //-> centrality classes (maximum 10) used for QA
        TH1F*                   fUserSuppliedV2;        // histo with integrated v2
        TH1F*                   fUserSuppliedV3;        // histo with integrated v3
        TH1F*                   fUserSuppliedR2;        // correct the extracted v2 with this r
        TH1F*                   fUserSuppliedR3;        // correct the extracted v3 with this r
        Int_t                   fNAcceptedTracks;       //! number of accepted tracks
        Int_t                   fNAcceptedTracksQCn;    //! accepted tracks for QCn
        Int_t                   fInCentralitySelection; //! centrality bin, only for QA plots
        fitModulationType       fFitModulationType;     // fit modulation type
        qcRecovery              fQCRecovery;            // recovery type for e-by-e qc method
        Bool_t                  fUsePtWeight;           // use dptdphi instead of dndphi
        detectorType            fDetectorType;          // type of detector used for modulation fit
        TString                 fFitModulationOptions;  // fit options for modulation fit
        runModeType             fRunModeType;           // run mode type 
        TF1*                    fFitModulation;         //-> modulation fit for rho
        Float_t                 fMinPvalue;             // minimum value of p
        Float_t                 fMaxPvalue;             // maximum value of p
        AliLocalRhoParameter*   fLocalRho;              //! local rho
        // additional jet cuts (most are inherited)
        Float_t                 fLocalJetMinEta;        // local eta cut for jets
        Float_t                 fLocalJetMaxEta;        // local eta cut for jets
        Float_t                 fLocalJetMinPhi;        // local phi cut for jets
        Float_t                 fLocalJetMaxPhi;        // local phi cut for jets
        Float_t                 fSoftTrackMinPt;        // min pt for soft tracks
        Float_t                 fSoftTrackMaxPt;        // max pt for soft tracks
        // general qa histograms
        TH1F*                   fHistPvalueCDF;         //! cdf value of chisquare p
        // general settings
        Bool_t                  fAbsVnHarmonics;        // force postive local rho
        Float_t                 fExcludeLeadingJetsFromFit;    // exclude n leading jets from fit
        Bool_t                  fRebinSwapHistoOnTheFly;       // rebin swap histo on the fly
        Float_t                 fPercentageOfFits;      // save this percentage of fits
        Bool_t                  fUseV0EventPlaneFromHeader;    // use the vzero event plane from the header
        // transient object pointers
        TList*                  fOutputList;            //! output list
        TList*                  fOutputListGood;        //! output list for local analysis
        TList*                  fOutputListBad;         //! output list for local analysis
        TH1F*                   fHistSwap;              //! swap histogram
        TH1F*                   fHistAnalysisSummary;   //! flags
        TProfile*               fProfV2;                //! extracted v2
        TProfile*               fProfV2Cumulant;        //! v2 cumulant
        TProfile*               fProfV3;                //! extracted v3
        TProfile*               fProfV3Cumulant;        //! v3 cumulant
        TH1F*                   fHistPsi2[10];          //! psi 2
        TH1F*                   fHistPsi3[10];          //! psi 3

        AliAnalysisTaskLocalRho(const AliAnalysisTaskLocalRho&);                  // not implemented
        AliAnalysisTaskLocalRho& operator=(const AliAnalysisTaskLocalRho&);       // not implemented

        ClassDef(AliAnalysisTaskLocalRho, 1);
};

#endif
Commit	Line	Data
5ce0dc6d	1	#ifndef ALIANALYSISTASKLOCALRHO_H
	2	#define ALIANALYSISTASKLOCALRHO_H
	3
	4	// $Id$
	5
	6	#include <AliAnalysisTaskEmcalJet.h>
	7	#include <AliEmcalJet.h>
	8	#include <AliVEvent.h>
	9	#include <AliVTrack.h>
	10	#include <AliVCluster.h>
	11	#include <TClonesArray.h>
	12	#include <TMath.h>
	13	#include <TRandom3.h>
	14
	15	class TF1;
	16	class THF1;
	17	class THF2;
	18	class TProfile;
	19	class AliLocalRhoParameter;
	20
	21	class AliAnalysisTaskLocalRho : public AliAnalysisTaskEmcalJet
	22	{
	23	public:
	24	// enumerators
	25	enum fitModulationType { kNoFit, kV2, kV3, kCombined, kFourierSeries, kIntegratedFlow, kQC2, kQC4 }; // fit type
	26	enum detectorType { kTPC, kVZEROA, kVZEROC, kVZEROComb}; // detector that was used
	27	enum qcRecovery { kFixedRho, kNegativeVn, kTryFit }; // how to deal with negative cn value for qcn value
	28	enum runModeType { kLocal, kGrid }; // run mode type
	29	// constructors, destructor
	30	AliAnalysisTaskLocalRho();
	31	AliAnalysisTaskLocalRho(const char *name, runModeType type);
	32	virtual ~AliAnalysisTaskLocalRho();
	33	// setting up the task and technical aspects
	34	Bool_t InitializeAnalysis();
	35	virtual void UserCreateOutputObjects();
	36	TH1F* BookTH1F(const char* name, const char* x, Int_t bins, Double_t min, Double_t max, Int_t c = -1, Bool_t append = kTRUE);
	37	TH2F* BookTH2F(const char* name, const char* x, const char* y, Int_t binsx, Double_t minx, Double_t maxx, Int_t binsy, Double_t miny, Double_t maxy, Int_t c = -1, Bool_t append = kTRUE);
	38	virtual Bool_t Run();
	39	/* inline */ Double_t PhaseShift(Double_t x) const {
	40	while (x>=TMath::TwoPi())x-=TMath::TwoPi();
	41	while (x<0.)x+=TMath::TwoPi();
	42	return x; }
	43	/* inline */ Double_t PhaseShift(Double_t x, Double_t n) const {
	44	x = PhaseShift(x);
	45	if(TMath::Nint(n)==2) while (x>TMath::Pi()) x-=TMath::Pi();
	46	if(TMath::Nint(n)==3) {
	47	if(x>2.*TMath::TwoPi()/n) x = TMath::TwoPi() - x;
	48	if(x>TMath::TwoPi()/n) x = TMath::TwoPi()-(x+TMath::TwoPi()/n);
	49	}
	50	return x; }
	51	/* inline */ Double_t ChiSquarePDF(Int_t ndf, Double_t x) const {
	52	Double_t n(ndf/2.), denom(TMath::Power(2, n)*TMath::Gamma(n));
	53	if (denom!=0) return ((1./denom)TMath::Power(x, n-1)TMath::Exp(-x/2.));
	54	return -999; }
	55	// note that the cdf of the chisquare distribution is the normalized lower incomplete gamma function
	56	/* inline */ Double_t ChiSquareCDF(Int_t ndf, Double_t x) const { return TMath::Gamma(ndf/2., x/2.); }
	57	// setters - setup how to run
	58	void SetDebugMode(Int_t d) {fDebug = d;}
	59	void SetCentralityClasses(TArrayI* c) {fCentralityClasses = c;}
	60	void SetAttachToEvent(Bool_t a) {fAttachToEvent = a;}
	61	void SetLocalRhoName(TString n) {fLocalRhoName = n;}
	62	void SetFillHistograms(Bool_t b) {fFillHistograms = b;}
	63	// setters - analysis details
	64	void SetNoEventWeightsForQC(Bool_t e) {fNoEventWeightsForQC = e;}
65	void SetIntegratedFlow(TH1F* i, TH1F* j) {fUserSuppliedV2 = i;
66	fUserSuppliedV3 = j; }
67	void SetOnTheFlyResCorrection(TH1F* r2, TH1F* r3) {fUserSuppliedR2 = r2;
68	fUserSuppliedR3 = r3; }
69	void SetModulationFit(TF1* fit) {if (fFitModulation) delete fFitModulation;
70	fFitModulation = fit; }
71	void SetModulationFitMinMaxP(Float_t m, Float_t n) {fMinPvalue = m; fMaxPvalue = n; }
72	void SetModulationFitType(fitModulationType type) {fFitModulationType = type; }
73	void SetQCnRecoveryType(qcRecovery type) {fQCRecovery = type; }
74	void SetModulationFitOptions(TString opt) {fFitModulationOptions = opt; }
75	void SetReferenceDetector(detectorType type) {fDetectorType = type; }
76	void SetUsePtWeight(Bool_t w) {fUsePtWeight = w; }
77	void SetRunModeType(runModeType type) {fRunModeType = type; }
78	void SetForceAbsVnHarmonics(Bool_t f) {fAbsVnHarmonics = f; }
79	void SetExcludeLeadingJetsFromFit(Float_t n) {fExcludeLeadingJetsFromFit = n; }
80	void SetRebinSwapHistoOnTheFly(Bool_t r) {fRebinSwapHistoOnTheFly = r; }
81	void SetSaveThisPercentageOfFits(Float_t p) {fPercentageOfFits = p; }
82	void SetUseV0EventPlaneFromHeader(Bool_t h) {fUseV0EventPlaneFromHeader = h;}
83	void SetSoftTrackMinMaxPt(Float_t min, Float_t max) {fSoftTrackMinPt = min; fSoftTrackMaxPt = max;}
84	// getters
85	TString GetLocalRhoName() const {return fLocalRhoName; }
86	// numerical evaluations
87	void CalculateEventPlaneVZERO(Double_t vzero[2][2]) const;
88	void CalculateEventPlaneTPC(Double_t* tpc);
89	void CalculateEventPlaneCombinedVZERO(Double_t* comb) const;
90	Double_t CalculateQC2(Int_t harm);
91	Double_t CalculateQC4(Int_t harm);
92	// helper calculations for the q-cumulant analysis, also used by AliAnalyisTaskJetFlow
93	void QCnQnk(Int_t n, Int_t k, Double_t &reQ, Double_t &imQ);
94	Double_t QCnS(Int_t i, Int_t j);
95	Double_t QCnM();
96	Double_t QCnM11();
97	Double_t QCnM1111();
98	Bool_t QCnRecovery(Double_t psi2, Double_t psi3);
99	// analysis details
100	Bool_t CorrectRho(Double_t psi2, Double_t psi3);
101	void FillEventPlaneHistograms(Double_t psi2, Double_t psi3) const;
102	void FillAnalysisSummaryHistogram() const;
103	// track selection
104	/* inline / Bool_t PassesCuts(const AliVTrack track) const {
105	if(!track) return kFALSE;
106	return (track->Pt() < fTrackPtCut \|\| track->Eta() < fTrackMinEta \|\| track->Eta() > fTrackMaxEta \|\| track->Phi() < fTrackMinPhi \|\| track->Phi() > fTrackMaxPhi) ? kFALSE : kTRUE; }
107	/* inline / Bool_t PassesCuts(AliEmcalJet jet) const {
108	if(!jet \|\| fJetRadius <= 0) return kFALSE;
109	return (jet->Pt() < fJetPtCut \|\| jet->Area()/(fJetRadiusfJetRadiusTMath::Pi()) < fPercAreaCut \|\| jet->Eta() < fJetMinEta \|\| jet->Eta() > fJetMaxEta \|\| jet->Phi() < fJetMinPhi \|\| jet->Phi() > fJetMaxPhi) ? kFALSE : kTRUE; }
110	// filling histograms
111	virtual void Terminate(Option_t* option);
112
113	private:
114	Int_t fDebug; // debug level (0 none, 1 fcn calls, 2 verbose)
115	Bool_t fInitialized; //! is the analysis initialized?
116	Bool_t fAttachToEvent; // attach local rho to the event
117	Bool_t fFillHistograms; // fill qa histograms
118	Bool_t fNoEventWeightsForQC; // don't store event weights for qc analysis
119	TString fLocalRhoName; // name for local rho
120	TArrayI* fCentralityClasses; //-> centrality classes (maximum 10) used for QA
121	TH1F* fUserSuppliedV2; // histo with integrated v2
122	TH1F* fUserSuppliedV3; // histo with integrated v3
123	TH1F* fUserSuppliedR2; // correct the extracted v2 with this r
124	TH1F* fUserSuppliedR3; // correct the extracted v3 with this r
125	Int_t fNAcceptedTracks; //! number of accepted tracks
126	Int_t fNAcceptedTracksQCn; //! accepted tracks for QCn
127	Int_t fInCentralitySelection; //! centrality bin, only for QA plots
128	fitModulationType fFitModulationType; // fit modulation type
129	qcRecovery fQCRecovery; // recovery type for e-by-e qc method
130	Bool_t fUsePtWeight; // use dptdphi instead of dndphi
131	detectorType fDetectorType; // type of detector used for modulation fit
132	TString fFitModulationOptions; // fit options for modulation fit
133	runModeType fRunModeType; // run mode type
134	TF1* fFitModulation; //-> modulation fit for rho
135	Float_t fMinPvalue; // minimum value of p
136	Float_t fMaxPvalue; // maximum value of p
137	AliLocalRhoParameter* fLocalRho; //! local rho
138	// additional jet cuts (most are inherited)
139	Float_t fLocalJetMinEta; // local eta cut for jets
140	Float_t fLocalJetMaxEta; // local eta cut for jets
141	Float_t fLocalJetMinPhi; // local phi cut for jets
142	Float_t fLocalJetMaxPhi; // local phi cut for jets
143	Float_t fSoftTrackMinPt; // min pt for soft tracks
144	Float_t fSoftTrackMaxPt; // max pt for soft tracks
145	// general qa histograms
146	TH1F* fHistPvalueCDF; //! cdf value of chisquare p
147	// general settings
148	Bool_t fAbsVnHarmonics; // force postive local rho
149	Float_t fExcludeLeadingJetsFromFit; // exclude n leading jets from fit
150	Bool_t fRebinSwapHistoOnTheFly; // rebin swap histo on the fly
151	Float_t fPercentageOfFits; // save this percentage of fits
152	Bool_t fUseV0EventPlaneFromHeader; // use the vzero event plane from the header
153	// transient object pointers
154	TList* fOutputList; //! output list
155	TList* fOutputListGood; //! output list for local analysis
156	TList* fOutputListBad; //! output list for local analysis
157	TH1F* fHistSwap; //! swap histogram
158	TH1F* fHistAnalysisSummary; //! flags
159	TProfile* fProfV2; //! extracted v2
160	TProfile* fProfV2Cumulant; //! v2 cumulant
161	TProfile* fProfV3; //! extracted v3
162	TProfile* fProfV3Cumulant; //! v3 cumulant
163	TH1F* fHistPsi2[10]; //! psi 2
164	TH1F* fHistPsi3[10]; //! psi 3
165
166	AliAnalysisTaskLocalRho(const AliAnalysisTaskLocalRho&); // not implemented
167	AliAnalysisTaskLocalRho& operator=(const AliAnalysisTaskLocalRho&); // not implemented
168
169	ClassDef(AliAnalysisTaskLocalRho, 1);
170	};
171
172	#endif