// $Id$
-#include <AliAnalysisTaskEmcalJetDev.h>
+#include <AliAnalysisTaskEmcalJet.h>
#include <AliEmcalJet.h>
#include <AliVEvent.h>
#include <AliVTrack.h>
class TProfile;
class AliLocalRhoParameter;
-class AliAnalysisTaskLocalRho : public AliAnalysisTaskEmcalJetDev {
- 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
- void ExecOnce();
- 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; }
- // 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 SetUseScaledRho(Bool_t s) {fUseScaledRho = s;}
- 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);
- 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(AliVTrack* track) const { return AcceptTrack(track, 0);}
- /* inline */ Bool_t PassesCuts(AliEmcalJet* jet) { return AcceptJet(jet, 0);}
- virtual void Terminate(Option_t* option);
+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
+ void ExecOnce();
+ 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; }
+ // 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 SetUseScaledRho(Bool_t s) {fUseScaledRho = s;}
+ 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);
+ 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 SetUsePtWeightErrorPropagation(Bool_t w) {fUsePtWeightErrorPropagation = 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(AliVTrack* track) const { return AcceptTrack(track, 0);}
+ /* inline */ Bool_t PassesCuts(AliEmcalJet* jet) { return AcceptJet(jet, 0);}
+ 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
- Bool_t fUseScaledRho; // use scaled 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
- // 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
+ 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
+ Bool_t fUseScaledRho; // use scaled 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
+ Bool_t fUsePtWeightErrorPropagation; // recalculate the bin error on the dpt dphi histogram
+ 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
+ // 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
+ TH2F* fHistRhoStatusCent; //! status of rho vs centrality
+ // 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
+ AliAnalysisTaskLocalRho(const AliAnalysisTaskLocalRho&); // not implemented
+ AliAnalysisTaskLocalRho& operator=(const AliAnalysisTaskLocalRho&); // not implemented
- ClassDef(AliAnalysisTaskLocalRho, 4);
+ ClassDef(AliAnalysisTaskLocalRho, 5);
};
-
#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff