PWGHF/vertexingHF/AliAnalysisTaskCombinHF.h

   1 #ifndef ALIANALYSISTASKCOMBINHF_H
   2 #define ALIANALYSISTASKCOMBINHF_H
   3
   4 /* Copyright(c) 1998-2018, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id: $ */
   8
   9 //*************************************************************************
  10 // Class AliAnalysisTaskCombinHF
  11 // AliAnalysisTaskSE to build D meson candidates by combining tracks
  12 //  background is computed LS and track rotations is
  13 // Authors: F. Prino, A. Rossi
  14 /////////////////////////////////////////////////////////////
  15
  16 #include <TH1F.h>
  17 #include <TH3F.h>
  18 #include <TObjString.h>
  19 #include <THnSparse.h>
  20 #include "AliAnalysisTaskSE.h"
  21 #include "AliAODTrack.h"
  22 #include "AliNormalizationCounter.h"
  23 #include "AliRDHFCuts.h"
  24
  25 class AliAnalysisTaskCombinHF : public AliAnalysisTaskSE
  26 {
  27 public:
  28
  29   AliAnalysisTaskCombinHF();
  30   AliAnalysisTaskCombinHF(Int_t meson, AliRDHFCuts* analysiscuts);
  31   virtual ~AliAnalysisTaskCombinHF();
  32
  33   virtual void UserCreateOutputObjects();
  34   virtual void Init(){};
  35   virtual void LocalInit() {Init();}
  36   virtual void UserExec(Option_t *option);
  37   virtual void Terminate(Option_t *option);
  38   virtual void FinishTaskOutput();
  39
  40   void SetReadMC(Bool_t read){fReadMC=read;}
  41
  42   void SetEventMixingWithCuts(Double_t maxDeltaVz, Double_t maxDeltaMult){
  43     fDoEventMixing=2; fMaxzVertDistForMix=maxDeltaVz; fMaxMultDiffForMix=maxDeltaMult;
  44   }
  45   void SetEventMixingWithPools(){fDoEventMixing=1;}
  46   void SetEventMixingOff(){fDoEventMixing=0;}
  47   void SetNumberOfEventsForMixing(Int_t minn){fNumberOfEventsForMixing=minn;}
  48
  49   void ConfigureZVertPools(Int_t nPools, Double_t*  zVertLimits);
  50   void ConfigureMultiplicityPools(Int_t nPools, Double_t*  multLimits);
  51   void SelectPromptD(){fPromptFeeddown=kPrompt;}
  52   void SelectFeeddownD(){fPromptFeeddown=kFeeddown;}
  53   void SelectPromptAndFeeddownD(){fPromptFeeddown=kBoth;}
  54   void SetGoUpToQuark(Bool_t opt){fGoUpToQuark=opt;}
  55   void SetKeepNegIDtracks(Bool_t nid){fKeepNegID=nid;}//set it to kTRUE only if you know what you are doing
  56   void SetTrackCuts(AliESDtrackCuts* cuts){
  57     if(fTrackCutsAll) delete fTrackCutsAll;
  58     fTrackCutsAll=new AliESDtrackCuts(*cuts);
  59   }
  60   void SetPionTrackCuts(AliESDtrackCuts* cuts){
  61     if(fTrackCutsPion) delete fTrackCutsPion;
  62     fTrackCutsPion=new AliESDtrackCuts(*cuts);
  63   }
  64   void SetKaonTrackCuts(AliESDtrackCuts* cuts){
  65     if(fTrackCutsKaon) delete fTrackCutsKaon;
  66     fTrackCutsKaon=new AliESDtrackCuts(*cuts);
  67   }
  68   void SetPIDHF(AliAODPidHF* pid){
  69     if(fPidHF) delete fPidHF;
  70     fPidHF=new AliAODPidHF(*pid);
  71   }
  72   void SetRDHFCuts(AliRDHFCuts* cuts){
  73     fAnalysisCuts=cuts;
  74   }
  75   void SetFilterMask(UInt_t mask=16){fFilterMask=mask;}
  76   void SetAnalysisLevel(Int_t level){fFullAnalysis=level;}
  77   void ConfigureRotation(Int_t n, Double_t phimin, Double_t phimax){
  78     fNRotations=n;
  79     fMinAngleForRot=phimin;
  80     fMaxAngleForRot=phimax;
  81   }
  82   void SetMassWindow(Double_t minMass, Double_t maxMass){fMinMass=minMass; fMaxMass=maxMass;}
  83   void SetMaxPt(Double_t maxPt){fMaxPt=maxPt;}
  84   void SetPtBinWidth(Double_t binw){fPtBinWidth=binw;}
  85   void SetEtaAccCut(Double_t etacut){fEtaAccCut=etacut;}
  86   void SetPtAccCut(Double_t ptcut){fPtAccCut=ptcut;}
  87   void SetMultiplicityRange(Double_t mmin=-0.5, Double_t mmax=199.5){
  88     fMinMultiplicity=mmin;
  89     fMaxMultiplicity=mmax;
  90   }
  91
  92   void SetPIDstrategy(Int_t strat){fPIDstrategy=strat;}
  93   void SetMaxPforIDPion(Double_t maxpIdPion){fmaxPforIDPion=maxpIdPion;}
  94   void SetMaxPforIDKaon(Double_t maxpIdKaon){fmaxPforIDKaon=maxpIdKaon;}
  95   void SetPIDselCaseZero(Int_t strat){fPIDselCaseZero=strat;}
  96   void SetBayesThres(Double_t thresKaon, Double_t thresPion){
  97     fBayesThresKaon=thresKaon;
  98     fBayesThresPion=thresPion;
  99   }
 100
 101   Bool_t IsTrackSelected(AliAODTrack* track);
 102   Bool_t IsKaon(AliAODTrack* track);
 103   Bool_t IsPion(AliAODTrack* track);
 104   Bool_t SelectAODTrack(AliAODTrack *track, AliESDtrackCuts *cuts);
 105
 106   Bool_t FillHistos(Int_t pdgD,Int_t nProngs, AliAODRecoDecay* tmpRD, Double_t* px, Double_t* py, Double_t* pz, UInt_t *pdgdau, TClonesArray *arrayMC, Int_t* dgLabels);
 107   void FillLSHistos(Int_t pdgD,Int_t nProngs, AliAODRecoDecay* tmpRD, Double_t* px, Double_t* py, Double_t* pz, UInt_t *pdgdau, Int_t charge);
 108   void FillMEHistos(Int_t pdgD,Int_t nProngs, AliAODRecoDecay* tmpRD, Double_t* px, Double_t* py, Double_t* pz, UInt_t *pdgdau);
 109   void FillMEHistosLS(Int_t pdgD,Int_t nProngs, AliAODRecoDecay* tmpRD, Double_t* px, Double_t* py, Double_t* pz, UInt_t *pdgdau, Int_t charge);
 110   void FillGenHistos(TClonesArray* arrayMC);
 111   Bool_t CheckAcceptance(TClonesArray* arrayMC, Int_t nProng, Int_t *labDau);
 112   Int_t GetPoolIndex(Double_t zvert, Double_t mult);
 113   void ResetPool(Int_t poolIndex);
 114   void DoMixingWithPools(Int_t poolIndex);
 115   void DoMixingWithCuts();
 116   Bool_t CanBeMixed(Double_t zv1, Double_t zv2, Double_t mult1, Double_t mult2);
 117   enum EMesonSpecies {kDzero, kDplus, kDstar, kDs};
 118   enum EPrompFd {kNone,kPrompt,kFeeddown,kBoth};
 119   enum EPIDstrategy {knSigma, kBayesianMaxProb, kBayesianThres};
 120
 121 private:
 122
 123   AliAnalysisTaskCombinHF(const AliAnalysisTaskCombinHF &source);
 124   AliAnalysisTaskCombinHF& operator=(const AliAnalysisTaskCombinHF& source);
 125
 126   TList   *fOutput; //! list send on output slot 0
 127   TH1F *fHistNEvents;         //!hist. for No. of events
 128   TH1F *fHistTrackStatus;     //!hist. of status of tracks
 129   TH1F *fHistCheckOrigin;     //!hist. of origin (c/b) of D meson
 130   TH1F *fHistCheckOriginSel;  //!hist. of origin (c/b) of D meson
 131   TH1F *fHistCheckDecChan;    //!hist. of decay channel of D meson
 132   TH1F *fHistCheckDecChanAcc; //!hist. of decay channel of D meson in acc.
 133   TH3F *fPtVsYVsMultGen;        //! hist. of Y vs. Pt vs. Mult generated (all D)
 134   TH3F *fPtVsYVsMultGenLargeAcc; //! hist. of Y vs. Pt vs. Mult generated (|y|<0.9)
 135   TH3F *fPtVsYVsMultGenLimAcc;  //! hist. of Y vs. Pt vs. Mult generated (|y|<0.5)
 136   TH3F *fPtVsYVsMultGenAcc;     //! hist. of Y vs. Pt vs. Mult generated (D in acc)
 137   TH3F *fPtVsYVsMultReco;       //! hist. of Y vs. Pt vs. Mult generated (Reco D)
 138   TH3F *fMassVsPtVsY;     //! hist. of Y vs. Pt vs. Mass (all cand)
 139   TH3F *fMassVsPtVsYRot;   //! hist. of Y vs. Pt vs. Mass (rotations)
 140   TH3F *fMassVsPtVsYLSpp;  //! hist. of Y vs. Pt vs. Mass (like sign ++)
 141   TH3F *fMassVsPtVsYLSmm;  //! hist. of Y vs. Pt vs. Mass (like sign --)
 142   TH3F *fMassVsPtVsYSig;   //! hist. of Y vs. Pt vs. Mass (signal)
 143   TH3F *fMassVsPtVsYRefl;  //! hist. of Y vs. Pt vs. Mass (reflections)
 144   TH3F *fMassVsPtVsYBkg;   //! hist. of Y vs. Pt vs. Mass (background)
 145   TH1F *fNSelected;        //! hist. of n. of selected D+
 146   TH1F *fNormRotated;      //! hist. rotated/selected D+
 147   TH1F *fDeltaMass;        //! hist. mass difference after rotations
 148   THnSparse *fDeltaMassFullAnalysis; //! hist. mass difference after rotations with more details
 149   TH3F *fMassVsPtVsYME;   //! hist. of Y vs. Pt vs. Mass (mixedevents)
 150   TH3F *fMassVsPtVsYMELSpp;   //! hist. of Y vs. Pt vs. Mass (mixedevents)
 151   TH3F *fMassVsPtVsYMELSmm;   //! hist. of Y vs. Pt vs. Mass (mixedevents)
 152   TH2F* fEventsPerPool;   //! hist with number of events per pool
 153   TH2F* fMixingsPerPool;    //! hist with number of mixings per pool
 154   UInt_t fFilterMask; // FilterMask
 155   AliESDtrackCuts* fTrackCutsAll; // track selection
 156   AliESDtrackCuts* fTrackCutsPion; // pion track selection
 157   AliESDtrackCuts* fTrackCutsKaon; // kaon track selection
 158   AliAODPidHF* fPidHF; // PID configuration
 159   AliRDHFCuts *fAnalysisCuts; // Cuts for candidates
 160
 161   Double_t fMinMass; // minimum value of invariant mass
 162   Double_t fMaxMass; // maximum value of invariant mass
 163   Double_t fMaxPt;   // maximum pT value for inv. mass histograms
 164   Double_t fPtBinWidth; // width of pt bin (GeV/c)
 165   Double_t fEtaAccCut; // eta limits for acceptance step
 166   Double_t fPtAccCut; // pt limits for acceptance step
 167
 168   Int_t fNRotations; // number of rotations
 169   Double_t fMinAngleForRot; // minimum angle for track rotation
 170   Double_t fMaxAngleForRot; // maximum angle for track rotation
 171   Double_t fMinAngleForRot3; // minimum angle for track rotation (3rd prong)
 172   Double_t fMaxAngleForRot3; // maximum angle for track rotation (3rd prong)
 173
 174   AliNormalizationCounter *fCounter;//!Counter for normalization
 175
 176   Int_t fMeson;          // mesonSpecies (see enum)
 177   Bool_t  fReadMC;       //  flag for access to MC
 178   Int_t fPromptFeeddown; // flag to select prompt (1), feeddown (2) or all (3)
 179   Bool_t fGoUpToQuark;   // flag for definition of c,b origin
 180   Int_t fFullAnalysis;   // flag to set analysis level (0 is the fastest)
 181
 182   Int_t    fPIDstrategy;     // knSigma, kBayesianMaxProb, kBayesianThres
 183   Double_t fmaxPforIDPion; // flag for upper p limit for id band for pion
 184   Double_t fmaxPforIDKaon; // flag for upper p limit for id band for kaon
 185   Bool_t   fKeepNegID;    // flag to keep also track with negative ID (default kFALSE, change it only if you know what you are doing)
 186   Int_t    fPIDselCaseZero;  // flag to change PID strategy
 187   Double_t fBayesThresKaon;  // threshold for kaon identification via Bayesian PID
 188   Double_t fBayesThresPion;  // threshold for pion identification via Bayesian PID
 189
 190   Int_t fDoEventMixing; // flag for event mixing
 191   Int_t  fNumberOfEventsForMixing; // maximum number of events to be used in event mixing
 192   Double_t fMaxzVertDistForMix; // cut on zvertex distance for event mixing with cuts
 193   Double_t fMaxMultDiffForMix; // cut on multiplicity difference for event mixing with cuts
 194   Int_t fNzVertPools; // number of pools in z vertex for event mixing
 195   Int_t fNzVertPoolsLimSize; // number of pools in z vertex for event mixing +1
 196   Double_t* fzVertPoolLims; //[fNzVertPoolsLimSize] limits of the pools in zVertex
 197   Int_t fNMultPools; // number of pools in multiplicity for event mixing
 198   Int_t fNMultPoolsLimSize; // number of pools in multiplicity for event mixing +1
 199   Double_t* fMultPoolLims; //[fNMultPoolsLimSize] limits of the pools in multiplicity
 200   Int_t  fNOfPools; // number of pools
 201   TTree** fEventBuffer;   //! structure for event mixing
 202   TObjString* fEventInfo;  // unique event Id for event mixing checks
 203   Double_t fVtxZ;         // zVertex
 204   Double_t fMultiplicity; // multiplicity
 205   Double_t fMinMultiplicity;  // lower limit for multiplcities in MC histos
 206   Double_t fMaxMultiplicity; // upper limit for multiplcities in MC histos
 207   TObjArray* fKaonTracks; // array of kaon-compatible tracks (TLorentzVectors)
 208   TObjArray* fPionTracks; // array of pion-compatible tracks (TLorentzVectors)
 209   ClassDef(AliAnalysisTaskCombinHF,10); // D0D+ task from AOD tracks
 210 };
 211
 212 #endif