coding/naming rule fixes

[u/mrichter/AliRoot.git] / PWGCF / EBYE / BalanceFunctions / AliAnalysisTaskEventMixingBF.h

#ifndef ALIANALYSISTASKEVENTMIXINGBF_CXX\r
#define ALIANALYSISTASKEVENTMIXINGBF_CXX\r
\r
// Analysis task for the EventMixingBF code\r
// Authors: Panos Cristakoglou@cern.ch, m.weber@cern.ch\r
\r
class TList;\r
class TH1F;\r
class TH2F;\r
class TF1;\r
\r
class AliMixInputEventHandler;\r
class AliBalanceEventMixing;\r
class AliESDtrackCuts;\r
\r
#include "AliAnalysisTaskSE.h"\r
#include "AliBalanceEventMixing.h"\r
\r
#include "AliPID.h"  \r
#include "AliPIDResponse.h"\r
#include "AliPIDCombined.h"\r
 \r
\r
class AliAnalysisTaskEventMixingBF : public AliAnalysisTaskSE {\r
 public:\r
  AliAnalysisTaskEventMixingBF(const char *name = "AliAnalysisTaskEventMixingBF");\r
  virtual ~AliAnalysisTaskEventMixingBF(); \r
  \r
  \r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   UserExecMix(Option_t*);\r
  virtual void   FinishTaskOutput();\r
  virtual void   Terminate(Option_t *);\r
\r
  void SetAnalysisObject(AliBalanceEventMixing *const analysis) {\r
    fBalance         = analysis;\r
    }\r
  void SetShufflingObject(AliBalanceEventMixing *const analysisShuffled) {\r
    fRunShuffling = kTRUE;\r
    fShuffledBalance = analysisShuffled;\r
  }\r
  void SetAnalysisCutObject(AliESDtrackCuts *const trackCuts) {\r
    fESDtrackCuts = trackCuts;}\r
  void SetVertexDiamond(Double_t vx, Double_t vy, Double_t vz) {\r
    fVxMax = vx;\r
    fVyMax = vy;\r
    fVzMax = vz;\r
  }\r
\r
  //==============AOD analysis==============//\r
  void SetAODtrackCutBit(Int_t bit){\r
    nAODtrackCutBit = bit;\r
  }\r
\r
  void SetKinematicsCutsAOD(Double_t ptmin, Double_t ptmax, Double_t etamin, Double_t etamax){\r
    fPtMin  = ptmin;  fPtMax  = ptmax;\r
    fEtaMin = etamin; fEtaMax = etamax;\r
\r
  }\r
\r
  void SetExtraDCACutsAOD(Double_t DCAxy, Double_t DCAz){\r
    fDCAxyCut  = DCAxy;\r
    fDCAzCut = DCAz;\r
  }\r
\r
   void SetExtraTPCCutsAOD(Double_t maxTPCchi2, Int_t minNClustersTPC){\r
    fTPCchi2Cut      = maxTPCchi2;\r
    fNClustersTPCCut = minNClustersTPC;\r
  }\r
\r
  //==============MC analysis==============//\r
  void SetKinematicsCutsMC(Double_t ptmin, Double_t ptmax,\r
                           Double_t etamin, Double_t etamax){\r
    fPtMin  = ptmin; fPtMax  = ptmax;\r
    fEtaMin = etamin; fEtaMax = etamax;\r
  }\r
  void UseFlowAfterBurner(TF1 *gDifferentialV2) {\r
    fDifferentialV2 = gDifferentialV2;\r
    fUseFlowAfterBurner = kTRUE;\r
  }\r
  void ExcludeResonancesInMC() {fExcludeResonancesInMC = kTRUE;}\r
\r
  void SetPDGCode(Int_t gPdgCode) {\r
    fUseMCPdgCode = kTRUE;\r
    fPDGCodeToBeAnalyzed = gPdgCode;\r
  }\r
\r
  //Centrality\r
  void SetCentralityEstimator(const char* centralityEstimator) {fCentralityEstimator = centralityEstimator;}\r
  const char* GetCentralityEstimator(void)    const            {return fCentralityEstimator;}\r
  void SetCentralityPercentileRange(Double_t min, Double_t max) { \r
    fUseCentrality = kTRUE;\r
    fCentralityPercentileMin=min;\r
    fCentralityPercentileMax=max;\r
  }\r
  void SetImpactParameterRange(Double_t min, Double_t max) { \r
    fUseCentrality = kTRUE;\r
    fImpactParameterMin=min;\r
    fImpactParameterMax=max;\r
  }\r
\r
  //multiplicity\r
  void SetMultiplicityRange(Int_t min, Int_t max) {\r
    fUseMultiplicity = kTRUE;\r
    fNumberOfAcceptedTracksMin = min;\r
    fNumberOfAcceptedTracksMax = max;}\r
  \r
  void UseOfflineTrigger() {fUseOfflineTrigger = kTRUE;}\r
  \r
  //Acceptance filter\r
  void SetAcceptanceParameterization(TF1 *parameterization) {\r
    fAcceptanceParameterization = parameterization;}\r
\r
  //pid\r
  enum kDetectorUsedForPID { kTPCpid, kTOFpid, kTPCTOF }; // default TPC & TOF pid (via GetTPCpid & GetTOFpid)  \r
  enum kParticleOfInterest { kMuon, kElectron, kPion, kKaon, kProton };  \r
\r
  void SetUseBayesianPID(Double_t gMinProbabilityValue) {\r
    fUsePID = kTRUE; fUsePIDnSigma = kFALSE; fUsePIDPropabilities = kTRUE;\r
    fMinAcceptedPIDProbability = gMinProbabilityValue; }\r
\r
  void SetUseNSigmaPID(Double_t gMaxNSigma) {\r
    fUsePID = kTRUE; fUsePIDPropabilities = kFALSE; fUsePIDnSigma = kTRUE;\r
    fPIDNSigma = gMaxNSigma; }\r
\r
  void SetParticleOfInterest(kParticleOfInterest poi) {\r
    fParticleOfInterest = poi;}\r
  void SetDetectorUsedForPID(kDetectorUsedForPID detConfig) {\r
    fPidDetectorConfig = detConfig;}\r
\r
 private:\r
  AliBalanceEventMixing *fBalance; //EventMixingBF object\r
  Bool_t fRunShuffling;//run shuffling or not\r
  AliBalanceEventMixing *fShuffledBalance; //EventMixingBF object (shuffled)\r
  TList *fList; //fList object\r
  TList *fListEventMixingBF; //fList object\r
  TList *fListEventMixingBFS; //fList object\r
  TList *fHistListPIDQA;  //! list of histograms\r
\r
  TH1F *fHistEventStats; //event stats\r
  TH2F *fHistCentStats; //centrality stats\r
  TH1F *fHistTriggerStats; //trigger stats\r
  TH1F *fHistTrackStats; //Track filter bit stats\r
  TH1F *fHistVx; //x coordinate of the primary vertex\r
  TH1F *fHistVy; //y coordinate of the primary vertex\r
  TH1F *fHistVz; //z coordinate of the primary vertex\r
\r
  TH2F *fHistClus;//number of clusters (QA histogram)\r
  TH2F *fHistDCA;//DCA  (QA histogram)\r
  TH1F *fHistChi2;//track chi2 (QA histogram)\r
  TH1F *fHistPt;//transverse momentum (QA histogram)\r
  TH1F *fHistEta;//pseudorapidity (QA histogram)\r
  TH1F *fHistPhi;//phi (QA histogram)\r
  TH1F *fHistPhiBefore;//phi before v2 afterburner (QA histogram)\r
  TH1F *fHistPhiAfter;//phi after v2 afterburner (QA histogram)\r
  TH2F *fHistV0M;//V0 multiplicities (QA histogram)\r
  TH2F *fHistRefTracks;//reference track multiplicities (QA histogram)\r
\r
  //============PID============//\r
  TH2D *fHistdEdxVsPTPCbeforePID;//TPC dEdx vs momentum before PID cuts (QA histogram)\r
  TH2D *fHistBetavsPTOFbeforePID;//beta vs momentum before PID cuts (QA histogram)\r
  TH2D *fHistProbTPCvsPtbeforePID; //TPC probability vs pT before PID cuts (QA histogram)\r
  TH2D *fHistProbTOFvsPtbeforePID;//TOF probability vs pT before PID cuts (QA histogram)\r
  TH2D *fHistProbTPCTOFvsPtbeforePID;//TOF/TPC probability vs pT before PID cuts (QA histogram)\r
  TH2D *fHistNSigmaTPCvsPtbeforePID;//TPC nsigma vs pT before PID cuts (QA histogram)\r
  TH2D *fHistNSigmaTOFvsPtbeforePID;//TOF nsigma vs pT before PID cuts (QA histogram)\r
  TH2D *fHistdEdxVsPTPCafterPID;//TPC dEdx vs momentum after PID cuts (QA histogram)\r
  TH2D *fHistBetavsPTOFafterPID;//beta vs momentum after PID cuts (QA histogram)\r
  TH2D *fHistProbTPCvsPtafterPID; //TPC probability vs pT after PID cuts (QA histogram)\r
  TH2D *fHistProbTOFvsPtafterPID;//TOF probability vs pT after PID cuts (QA histogram)\r
  TH2D *fHistProbTPCTOFvsPtafterPID;//TOF/TPC probability vs pT after PID cuts (QA histogram)\r
  TH2D *fHistNSigmaTPCvsPtafterPID;//TPC nsigma vs pT after PID cuts (QA histogram)\r
  TH2D *fHistNSigmaTOFvsPtafterPID;//TOF nsigma vs pT after PID cuts (QA histogram)\r
\r
  AliPIDResponse *fPIDResponse;     //! PID response object\r
  AliPIDCombined       *fPIDCombined;     //! combined PID object\r
  \r
  kParticleOfInterest  fParticleOfInterest;//analyzed particle\r
  kDetectorUsedForPID   fPidDetectorConfig;//used detector for PID\r
\r
  Bool_t fUsePID; //flag to use PID \r
  Bool_t fUsePIDnSigma;//flag to use nsigma method for PID\r
  Bool_t fUsePIDPropabilities;//flag to use probability method for PID\r
  Double_t fPIDNSigma;//nsigma cut for PID\r
  Double_t fMinAcceptedPIDProbability;//probability cut for PID\r
  //============PID============//\r
\r
  AliESDtrackCuts *fESDtrackCuts; //ESD track cuts\r
\r
  TString fCentralityEstimator;      //"V0M","TRK","TKL","ZDC","FMD"\r
  Bool_t fUseCentrality;//use the centrality (PbPb) or not (pp)\r
  Double_t fCentralityPercentileMin;//centrality percentile min\r
  Double_t fCentralityPercentileMax;//centrality percentile max\r
  Double_t fImpactParameterMin;//impact parameter min (used for MC)\r
  Double_t fImpactParameterMax;//impact parameter max (used for MC)\r
\r
  Bool_t fUseMultiplicity;//use the multiplicity cuts\r
  Int_t fNumberOfAcceptedTracksMin;//min. number of number of accepted tracks (used for the multiplicity dependence study - pp)\r
  Int_t fNumberOfAcceptedTracksMax;//max. number of number of accepted tracks (used for the multiplicity dependence study - pp)\r
  TH1F *fHistNumberOfAcceptedTracks;//hisot to store the number of accepted tracks\r
\r
  Bool_t fUseOfflineTrigger;//Usage of the offline trigger selection\r
\r
  Double_t fVxMax;//vxmax\r
  Double_t fVyMax;//vymax\r
  Double_t fVzMax;//vzmax\r
\r
  Int_t nAODtrackCutBit;//track cut bit from track selection (only used for AODs)\r
\r
  Double_t fPtMin;//only used for AODs\r
  Double_t fPtMax;//only used for AODs\r
  Double_t fEtaMin;//only used for AODs\r
  Double_t fEtaMax;//only used for AODs\r
\r
  Double_t fDCAxyCut;//only used for AODs\r
  Double_t fDCAzCut;//only used for AODs\r
\r
  Double_t fTPCchi2Cut;//only used for AODs\r
  Int_t fNClustersTPCCut;//only used for AODs\r
\r
  TF1 *fAcceptanceParameterization;//acceptance filter used for MC\r
\r
  TF1 *fDifferentialV2;//pt-differential v2 (from real data)\r
  Bool_t fUseFlowAfterBurner;//Usage of a flow after burner\r
\r
  Bool_t fExcludeResonancesInMC;//flag to exclude the resonances' decay products from the MC analysis\r
  Bool_t fUseMCPdgCode; //Boolean to analyze a set of particles in MC\r
  Int_t fPDGCodeToBeAnalyzed; //Analyze a set of particles in MC\r
\r
  // Event Mixing\r
  AliVEvent       *fMainEvent;//main event in the event mixing loop\r
  AliVEvent       *fMixEvent;//second event in the event mixing loop\r
\r
  AliMixInputEventHandler *SetupEventsForMixing();  \r
\r
  AliAnalysisTaskEventMixingBF(const AliAnalysisTaskEventMixingBF&); // not implemented\r
  AliAnalysisTaskEventMixingBF& operator=(const AliAnalysisTaskEventMixingBF&); // not implemented\r
  \r
  ClassDef(AliAnalysisTaskEventMixingBF, 1); // example of analysis\r
};\r
\r
#endif\r