Add fast merging option (Diego)

[u/mrichter/AliRoot.git] / PWG3 / hfe / AliAnalysisTaskHFEFlow.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
// Flow task class for the ALICE HFE group\r
//\r
//\r
#ifndef ALIANALYSISTASKHFEFLOW_H\r
#define ALIANALYSISTASKHFEFLOW_H\r
\r
\r
\r
\r
#include <AliAnalysisTaskSE.h>\r
\r
class TList;\r
class AliFlowTrackCuts;\r
class AliFlowCandidateTrack;\r
class AliHFEcuts;\r
class AliHFEpid;\r
class TH1D;\r
class TH2D;\r
class THnSparse;\r
class AliHFEpidQAmanager;\r
class AliFlowEvent;\r
\r
class AliAnalysisTaskHFEFlow: public AliAnalysisTaskSE {\r
  \r
public:\r
  AliAnalysisTaskHFEFlow();\r
  AliAnalysisTaskHFEFlow(const char *name);\r
  //TODO: if we get the ESDpid object from somewhere else, we should not delete it anymore!!!\r
  virtual ~AliAnalysisTaskHFEFlow(){;}\r
  \r
  virtual void  UserExec(Option_t */*option*/);\r
  virtual void  UserCreateOutputObjects();\r
\r
  AliHFEpid *GetPID() const { return fPID; }\r
  void SetHFECuts(AliHFEcuts * const cuts) { fHFECuts = cuts; };\r
  void SetSubEtaGapTPC(Bool_t  subEtaGapTPC) { fSubEtaGapTPC = subEtaGapTPC; };\r
  void SetEtaGap(Double_t  etaGap) { fEtaGap = etaGap; };\r
\r
  void SetNbBinsPtQCumulant(Int_t nbBinsPtQCumulant) { fNbBinsPtQCumulant = nbBinsPtQCumulant; };\r
  void SetMinPtQCumulant(Double_t minPtQCumulant) { fMinPtQCumulant = minPtQCumulant; };\r
  void SetMaxPtQCumulant(Double_t maxPtQCumulant) { fMaxPtQCumulant = maxPtQCumulant; };\r
\r
  void SetAfterBurnerOn(Bool_t afterBurnerOn)     { fAfterBurnerOn = afterBurnerOn; };\r
  void SetNonFlowNumberOfTrackClones(Int_t nonFlowNumberOfTrackClones) { fNonFlowNumberOfTrackClones = nonFlowNumberOfTrackClones; };\r
  void SetV1V2V3V4V5(Double_t v1,Double_t v2,Double_t v3,Double_t v4,Double_t v5) {fV1 = v1; fV2 = v2; fV3 = v3; fV4 = v4; fV5 = v5; };\r
  void SetMaxNumberOfIterations(Int_t maxNumberOfIterations) { fMaxNumberOfIterations = maxNumberOfIterations; };\r
  void SetPrecisionPhi(Double_t precisionPhi) { fPrecisionPhi = precisionPhi;};\r
  void SetUseMCReactionPlane(Bool_t useMCReactionPlane) { fUseMCReactionPlane = useMCReactionPlane;};\r
  void SetMCPID(Bool_t mcPID) { fMCPID = mcPID;};\r
\r
  \r
  AliFlowCandidateTrack *MakeTrack( Double_t mass, Double_t pt, Double_t phi, Double_t eta) ;\r
  Double_t GetPhiAfterAddV2(Double_t phi,Double_t reactionPlaneAngle) const;\r
  \r
private:\r
  TList       *fListHist;  //TH list\r
\r
  Bool_t  fSubEtaGapTPC;  // bool to fill with eta gap\r
  Double_t fEtaGap;       // Value of the eta gap\r
\r
  Int_t   fNbBinsPtQCumulant; // Nbbinspt QCumulant method\r
  Double_t fMinPtQCumulant;   // Min pt QCumulant method\r
  Double_t fMaxPtQCumulant;   // Max pt QCumulant method\r
  Bool_t   fAfterBurnerOn;    // Add flow to all tracks\r
  Int_t     fNonFlowNumberOfTrackClones; // number of times to clone the particles (nonflow) \r
  Double_t  fV1;        // Add Flow. Must be in range [0,0.5].\r
  Double_t  fV2;        // Add Flow. Must be in range [0,0.5].\r
  Double_t  fV3;        // Add Flow. Must be in range [0,0.5].\r
  Double_t  fV4;        // Add Flow. Must be in range [0,0.5].\r
  Double_t  fV5;        // Add Flow. Must be in range [0,0.5].\r
  Int_t     fMaxNumberOfIterations; // Max number of iteration for adding v2\r
  Double_t  fPrecisionPhi;  // precision phi for adding v2\r
  Bool_t    fUseMCReactionPlane; // use MC reaction plane\r
\r
  Bool_t    fMCPID; // MC PID for electrons\r
  \r
\r
  // Cuts for FLOW PWG2\r
  AliFlowTrackCuts* fcutsRP;  // Reference particle cut\r
  AliFlowTrackCuts* fcutsPOI; // Particle Of Interest cut\r
  \r
  // Cuts for HFE\r
  AliHFEcuts *fHFECuts;       // HFE cuts\r
  AliHFEpid  *fPID;           // PID cuts \r
  AliHFEpidQAmanager *fPIDqa;   // QA Manager\r
  AliFlowEvent *fflowEvent;     // Flow event   \r
  \r
\r
  // Histos\r
  TH2D *fHistEV;               // Number of events\r
  \r
  // A Event plane as function of phiep, centrality \r
  THnSparseF *fEventPlane;     // Event plane\r
  \r
  // B Event Plane after subtraction as function of phiep, centrality \r
  THnSparseF *fEventPlaneaftersubtraction; // Event plane\r
\r
  // E Monitoring Event plane after subtraction of the track: cos, centrality, pt, eta\r
  THnSparseF *fCos2phie;  // Monitoring\r
  THnSparseF *fSin2phie;  // Monitoring\r
  THnSparseF *fCos2phiep;  // Monitoring\r
  THnSparseF *fSin2phiep;  // Monitoring\r
  THnSparseF *fSin2phiephiep;  // Monitoring\r
  \r
  // F Resolution as function of cosres, centrality \r
  THnSparseF *fCosRes; // Res\r
  \r
  // G Maps delta phi as function of deltaphi, centrality, pt\r
  THnSparseF *fDeltaPhiMaps; // Delta phi\r
  \r
  // H Maps cos phi : cos, centrality, pt\r
  THnSparseF *fCosPhiMaps;  //Cos\r
\r
         \r
  \r
  \r
  AliAnalysisTaskHFEFlow(const AliAnalysisTaskHFEFlow&); // not implemented\r
  AliAnalysisTaskHFEFlow& operator=(const AliAnalysisTaskHFEFlow&); // not implemented\r
  \r
  ClassDef(AliAnalysisTaskHFEFlow, 1); // analysisclass\r
};\r
\r
#endif\r