PWGHF/vertexingHF/AliAnalysisVertexingHF.h

   1 #ifndef ALIANALYSISVERTEXINGHF_H
   2 #define ALIANALYSISVERTEXINGHF_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //-------------------------------------------------------------------------
   9 //                      Class AliAnalysisVertexingHF
  10 //            Reconstruction of heavy-flavour decay candidates
  11 //
  12 //  Origin: E.Bruna, G.E.Bruno, A.Dainese, F.Prino, R.Romita, X.M.Zhang
  13 //  Contact: andrea.dainese@pd.infn.it
  14 //-------------------------------------------------------------------------
  15
  16 #include <TNamed.h>
  17 #include <TList.h>
  18
  19 #include "AliAnalysisFilter.h"
  20 #include "AliESDtrackCuts.h"
  21
  22 class AliESDVertex;
  23 class AliAODRecoDecay;
  24 class AliAODRecoDecayHF;
  25 class AliAODRecoDecayHF2Prong;
  26 class AliAODRecoDecayHF3Prong;
  27 class AliAODRecoDecayHF4Prong;
  28 class AliAODRecoCascadeHF;
  29 class AliAnalysisFilter;
  30 class AliRDHFCuts;
  31 class AliRDHFCutsD0toKpi;
  32 class AliRDHFCutsJpsitoee;
  33 class AliRDHFCutsDplustoKpipi;
  34 class AliRDHFCutsDstoKKpi;
  35 class AliRDHFCutsLctopKpi;
  36 class AliRDHFCutsLctoV0;
  37 class AliRDHFCutsD0toKpipipi;
  38 class AliRDHFCutsDStartoKpipi;
  39 class AliESDtrack;
  40 class AliVEvent;
  41 class AliAODVertex;
  42 class AliVertexerTracks;
  43 class AliESDv0;
  44 class AliAODv0;
  45
  46 //-----------------------------------------------------------------------------
  47 class AliAnalysisVertexingHF : public TNamed {
  48  public:
  49   //
  50   AliAnalysisVertexingHF();
  51   AliAnalysisVertexingHF(const AliAnalysisVertexingHF& source);
  52   AliAnalysisVertexingHF& operator=(const AliAnalysisVertexingHF& source);
  53   virtual ~AliAnalysisVertexingHF();
  54
  55   void FindCandidates(AliVEvent *event,
  56                       TClonesArray *aodVerticesHFTClArr,
  57                       TClonesArray *aodD0toKpiTClArr,
  58                       TClonesArray *aodJPSItoEleTClArr,
  59                       TClonesArray *aodCharm3ProngTClArr,
  60                       TClonesArray *aodCharm4ProngTClArr,
  61                       TClonesArray *aodDstarTClArr,
  62                       TClonesArray *aodCascadesTClArr,
  63                       TClonesArray *aodLikeSign2ProngTClArr,
  64                       TClonesArray *aodLikeSign3ProngTClArr);
  65
  66   TList* FillListOfCuts();
  67   void FixReferences(AliAODEvent *aod);
  68   void PrintStatus() const;
  69   void SetSecVtxWithKF() { fSecVtxWithKF=kTRUE; }
  70   void SetD0toKpiOn() { fD0toKpi=kTRUE; }
  71   void SetD0toKpiOff() { fD0toKpi=kFALSE; }
  72   void SetJPSItoEleOn() { fJPSItoEle=kTRUE; }
  73   void SetJPSItoEleOff() { fJPSItoEle=kFALSE; }
  74   void Set3ProngOn() { f3Prong=kTRUE; }
  75   void Set3ProngOff() { f3Prong=kFALSE; }
  76   void Set4ProngOn() { f4Prong=kTRUE; }
  77   void Set4ProngOff() { f4Prong=kFALSE; }
  78   void SetDstarOn() { fDstar=kTRUE; }
  79   void SetDstarOff() { fDstar=kFALSE; }
  80   void SetCascadesOn() { fCascades=kTRUE; }
  81   void SetCascadesOff() { fCascades=kFALSE; }
  82   void SetLikeSignOn() { fLikeSign=kTRUE; }
  83   void SetLikeSignOff() {fLikeSign=kFALSE; fLikeSign3prong=kFALSE;}
  84   void SetLikeSign3prongOn() { fLikeSign=kTRUE; fLikeSign3prong=kTRUE; }
  85   void SetLikeSign3prongOff() { fLikeSign3prong=kFALSE; }
  86   void SetMixEventOn() { fMixEvent=kTRUE; }
  87   void SetMixEventOff() { fMixEvent=kFALSE; }
  88   void SetInputAOD() { fInputAOD=kTRUE; }
  89   Bool_t GetD0toKpi() const { return fD0toKpi; }
  90   Bool_t GetJPSItoEle() const { return fJPSItoEle; }
  91   Bool_t Get3Prong() const { return f3Prong; }
  92   Bool_t Get4Prong() const { return f4Prong; }
  93   Bool_t GetDstar()  const { return fDstar; }
  94   Bool_t GetCascades() const { return fCascades; }
  95   Bool_t GetLikeSign() const { return fLikeSign; }
  96   Bool_t GetLikeSign3prong() const { return fLikeSign3prong; }
  97   Bool_t GetMixEvent() const { return fMixEvent; }
  98   Bool_t GetInputAOD() const { return fInputAOD; }
  99   Bool_t GetRecoPrimVtxSkippingTrks() const {return fRecoPrimVtxSkippingTrks;}
 100   Bool_t GetRmTrksFromPrimVtx() const {return fRmTrksFromPrimVtx;}
 101   void SetFindVertexForDstar(Bool_t vtx=kTRUE) { fFindVertexForDstar=vtx; }
 102   void SetFindVertexForCascades(Bool_t vtx=kTRUE) { fFindVertexForCascades=vtx; }
 103   void SetRecoPrimVtxSkippingTrks()
 104     { fRecoPrimVtxSkippingTrks=kTRUE; fRmTrksFromPrimVtx=kFALSE;}
 105   void UnsetRecoPrimVtxSkippingTrks()
 106     { fRecoPrimVtxSkippingTrks=kFALSE; fRmTrksFromPrimVtx=kFALSE;}
 107   void SetRmTrksFromPrimVtx()
 108     {fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
 109   void SetTrackFilter(AliAnalysisFilter* trackF) {
 110     // switch off the TOF selection that cannot be applied with AODTracks
 111     TList *l = (TList*)trackF->GetCuts();
 112     AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
 113     if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
 114     fTrackFilter = trackF;
 115   }
 116   void SetTrackFilter2prongPbCentral(Float_t maxPercentile, AliAnalysisFilter* trackF) {
 117     // switch off the TOF selection that cannot be applied with AODTracks
 118     TList *l = (TList*)trackF->GetCuts();
 119     AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
 120     if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
 121     fTrackFilter2prongCentral = trackF;
 122     fMaxCentPercentileForTightCuts=maxPercentile;
 123   }
 124   void SetTrackFilter3prongPbCentral(Float_t maxPercentile, AliAnalysisFilter* trackF) {
 125     // switch off the TOF selection that cannot be applied with AODTracks
 126     TList *l = (TList*)trackF->GetCuts();
 127     AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
 128     if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
 129     fTrackFilter3prongCentral = trackF;
 130     fMaxCentPercentileForTightCuts=maxPercentile;
 131   }
 132   void SetTrackFilterSoftPi(AliAnalysisFilter* trackF) {
 133     // switch off the TOF selection that cannot be applied with AODTracks
 134     TList *l = (TList*)trackF->GetCuts();
 135     AliESDtrackCuts *tcuts = (AliESDtrackCuts*)l->FindObject("AliESDtrackCuts");
 136     if(tcuts->GetFlagCutTOFdistance()) tcuts->SetFlagCutTOFdistance(kFALSE);
 137     fTrackFilterSoftPi = trackF;
 138   }
 139   AliAnalysisFilter* GetTrackFilter() const { return fTrackFilter; }
 140   AliAnalysisFilter* GetTrackFilterSoftPi() const { return fTrackFilterSoftPi; }
 141   void SetCutsD0toKpi(AliRDHFCutsD0toKpi* cuts) { fCutsD0toKpi = cuts; }
 142   AliRDHFCutsD0toKpi* GetCutsD0toKpi() const { return fCutsD0toKpi; }
 143   void SetCutsJpsitoee(AliRDHFCutsJpsitoee* cuts) { fCutsJpsitoee = cuts; }
 144   AliRDHFCutsJpsitoee* GetCutsJpsitoee() const { return fCutsJpsitoee; }
 145   void SetCutsDplustoKpipi(AliRDHFCutsDplustoKpipi* cuts) { fCutsDplustoKpipi = cuts; }
 146   AliRDHFCutsDplustoKpipi* GetCutsDplustoKpipi() const { return fCutsDplustoKpipi; }
 147   void SetCutsDstoKKpi(AliRDHFCutsDstoKKpi* cuts) { fCutsDstoKKpi = cuts; }
 148   AliRDHFCutsDstoKKpi* GetCutsDstoKKpi() const { return fCutsDstoKKpi; }
 149   void SetCutsLctopKpi(AliRDHFCutsLctopKpi* cuts) { fCutsLctopKpi = cuts; }
 150   AliRDHFCutsLctopKpi* GetCutsLctopKpi() const { return fCutsLctopKpi; }
 151   void SetCutsLctoV0(AliRDHFCutsLctoV0* cuts) { fCutsLctoV0 = cuts; }
 152   AliRDHFCutsLctoV0* GetCutsLctoV0() const { return fCutsLctoV0; }
 153   void SetCutsD0toKpipipi(AliRDHFCutsD0toKpipipi* cuts) { fCutsD0toKpipipi = cuts; }
 154   AliRDHFCutsD0toKpipipi* GetCutsD0toKpipipi() const { return fCutsD0toKpipipi; }
 155   void SetCutsDStartoKpipi(AliRDHFCutsDStartoKpipi* cuts) { fCutsDStartoKpipi = cuts; }
 156   AliRDHFCutsDStartoKpipi* GetCutsDStartoKpipi() const { return fCutsDStartoKpipi; }
 157   void SetMassCutBeforeVertexing(Bool_t flag) { fMassCutBeforeVertexing=flag; }
 158
 159   void SetMasses();
 160
 161   //
 162  private:
 163   //
 164   enum { kBitDispl = 0, kBitSoftPi = 1, kBit3Prong = 2 };
 165
 166   Bool_t fInputAOD; // input from AOD (kTRUE) or ESD (kFALSE)
 167   Int_t fAODMapSize; // size of fAODMap
 168   Int_t *fAODMap; //[fAODMapSize] map between index and ID for AOD tracks
 169
 170   AliVertexerTracks* fVertexerTracks; // vertexer, to compute secondary vertices
 171   Double_t fBzkG; // z componenent of field in kG
 172
 173   Bool_t fSecVtxWithKF; // if kTRUE use KF vertexer, else AliVertexerTracks
 174
 175   Bool_t fRecoPrimVtxSkippingTrks; // flag for primary vertex reco on the fly
 176                                    // for each candidate, w/o its daughters
 177   Bool_t fRmTrksFromPrimVtx; // flag for fast removal of daughters from
 178                              // the primary vertex
 179
 180   AliESDVertex *fV1; // primary vertex
 181
 182   // flag to enable candidates production
 183   Bool_t fD0toKpi;   // D0->Kpi
 184   Bool_t fJPSItoEle; // Jpsi->ee
 185   Bool_t f3Prong;    // D+,Ds,Lc
 186   Bool_t f4Prong;    // D0->Kpipipi
 187   Bool_t fDstar;     // D*->D0pi
 188   Bool_t fCascades;  // cascades, Lc --> v0+track
 189   Bool_t fLikeSign;  // Like-sign pairs
 190   Bool_t fLikeSign3prong;  // Like-sign triplets
 191   Bool_t fMixEvent; // event mixing
 192
 193   Float_t fMaxCentPercentileForTightCuts; //max. centrality percentile for using tight cuts
 194
 195   // single-track cuts
 196   AliAnalysisFilter *fTrackFilter; //  Track Filter for displaced vertices
 197   AliAnalysisFilter *fTrackFilter2prongCentral; //  Track Filter for displaced vertices in PbPb central events (tighter cuts) for 2 prong (D0->Kpi)
 198   AliAnalysisFilter *fTrackFilter3prongCentral; //  Track Filter for displaced vertices in PbPb central events (tighter cuts) for 3 prong (D+, Ds, Lc)
 199   AliAnalysisFilter *fTrackFilterSoftPi; //  Track Filter for D* soft pion
 200   // candidates cuts
 201   AliRDHFCutsD0toKpi *fCutsD0toKpi; // D0->Kpi cuts
 202   AliRDHFCutsJpsitoee *fCutsJpsitoee; // J/psi->ee cuts
 203   AliRDHFCutsDplustoKpipi *fCutsDplustoKpipi; // D+->Kpipi cuts
 204   AliRDHFCutsDstoKKpi *fCutsDstoKKpi; // Ds->KKpi cuts
 205   AliRDHFCutsLctopKpi *fCutsLctopKpi; // Lc->pKpi cuts
 206   AliRDHFCutsLctoV0 *fCutsLctoV0; // Lc --> v0 + bachelor cuts
 207   AliRDHFCutsD0toKpipipi *fCutsD0toKpipipi; // D0->Kpipipi cuts
 208   AliRDHFCutsDStartoKpipi *fCutsDStartoKpipi; // Dstar->D0pi cuts
 209
 210   TList *fListOfCuts;    // pointer to list of cuts for output file
 211   Bool_t fFindVertexForDstar; // reconstruct a secondary vertex or assume it's from the primary vertex
 212   Bool_t fFindVertexForCascades;  // reconstruct a secondary vertex or assume it's from the primary vertex
 213   Bool_t fMassCutBeforeVertexing; // to go faster in PbPb
 214   // dummies for invariant mass calculation
 215   AliAODRecoDecay *fMassCalc2; // for 2 prong
 216   AliAODRecoDecay *fMassCalc3; // for 3 prong
 217   AliAODRecoDecay *fMassCalc4; // for 4 prong
 218   Bool_t fOKInvMassD0; // pair fullfilling D0 inv mass selection
 219   Bool_t fOKInvMassJpsi; // pair fullfilling Jpsi inv mass selection
 220   Bool_t fOKInvMassDplus; // triplet fullfilling D+ inv mass selection
 221   Bool_t fOKInvMassDs; // triplet fullfilling Ds inv mass selection
 222   Bool_t fOKInvMassLc; // triplet fullfilling Lc inv mass selection
 223   Bool_t fOKInvMassDstar; // combination fullfilling D* inv mass selection
 224   Bool_t fOKInvMassD0to4p; // 4tracks fullfilling D0 inv mass selection
 225   Bool_t fOKInvMassLctoV0; // triplet fullfilling Lc inv mass selection
 226
 227   Int_t fnTrksTotal;
 228   Int_t fnSeleTrksTotal;
 229
 230   Double_t fMassDzero;
 231   Double_t fMassDplus;
 232   Double_t fMassDs;
 233   Double_t fMassLambdaC;
 234   Double_t fMassDstar;
 235   Double_t fMassJpsi;
 236
 237
 238   //
 239   void AddRefs(AliAODVertex *v,AliAODRecoDecayHF *rd,const AliVEvent *event,
 240                const TObjArray *trkArray) const;
 241   void AddDaughterRefs(AliAODVertex *v,const AliVEvent *event,
 242                        const TObjArray *trkArray) const;
 243   AliAODRecoDecayHF2Prong* Make2Prong(TObjArray *twoTrackArray1,AliVEvent *event,
 244                                       AliAODVertex *secVert,Double_t dcap1n1,
 245                                       Bool_t &okD0,Bool_t &okJPSI,Bool_t &okD0fromDstar);
 246   AliAODRecoDecayHF3Prong* Make3Prong(TObjArray *threeTrackArray,AliVEvent *event,
 247                                       AliAODVertex *secVert,
 248                                       Double_t dispersion,
 249                                       const AliAODVertex *vertexp1n1,
 250                                       const AliAODVertex *vertexp2n1,
 251                                       Double_t dcap1n1,Double_t dcap2n1,Double_t dcap1p2,
 252                                       Bool_t &ok3Prong);
 253   AliAODRecoDecayHF4Prong* Make4Prong(TObjArray *fourTrackArray,AliVEvent *event,
 254                                       AliAODVertex *secVert,
 255                                       const AliAODVertex *vertexp1n1,
 256                                       const AliAODVertex *vertexp1n1p2,
 257                                       Double_t dcap1n1,Double_t dcap1n2,
 258                                       Double_t dcap2n1,Double_t dcap2n2,
 259                                       Bool_t &ok4Prong);
 260   AliAODRecoCascadeHF* MakeCascade(TObjArray *twoTrackArray,AliVEvent *event,
 261                                    AliAODVertex *secVert,
 262                                    AliAODRecoDecayHF2Prong *rd2Prong,
 263                                    Double_t dca,
 264                                    Bool_t &okDstar);
 265   AliAODRecoCascadeHF* MakeCascade(TObjArray *twoTrackArray,AliVEvent *event,
 266                                    AliAODVertex *secVert,
 267                                    AliAODv0 *v0,
 268                                    Double_t dca,
 269                                    Bool_t &okCascades);
 270
 271   AliAODVertex* PrimaryVertex(const TObjArray *trkArray=0x0,AliVEvent *event=0x0) const;
 272   AliAODVertex* ReconstructSecondaryVertex(TObjArray *trkArray,Double_t &dispersion,Bool_t useTRefArray=kTRUE) const;
 273
 274   Bool_t SelectInvMassAndPt3prong(Double_t *px,Double_t *py,Double_t *pz);
 275   Bool_t SelectInvMassAndPt4prong(Double_t *px,Double_t *py,Double_t *pz);
 276   Bool_t SelectInvMassAndPtD0Kpi(Double_t *px,Double_t *py,Double_t *pz);
 277   Bool_t SelectInvMassAndPtJpsiee(Double_t *px,Double_t *py,Double_t *pz);
 278   Bool_t SelectInvMassAndPtDstarD0pi(Double_t *px,Double_t *py,Double_t *pz);
 279   Bool_t SelectInvMassAndPtCascade(Double_t *px,Double_t *py,Double_t *pz);
 280
 281   Bool_t SelectInvMassAndPt3prong(TObjArray *trkArray);
 282   Bool_t SelectInvMassAndPt4prong(TObjArray *trkArray);
 283   Bool_t SelectInvMassAndPtDstarD0pi(TObjArray *trkArray);
 284
 285   void   SelectTracksAndCopyVertex(const AliVEvent *event,Int_t trkEntries,
 286                                    TObjArray &seleTrksArray,
 287                                    TObjArray &tracksAtVertex,
 288                                    Int_t &nSeleTrks,
 289                                    UChar_t *seleFlags,Int_t *evtNumber);
 290   void SetParametersAtVertex(AliESDtrack* esdt, const AliExternalTrackParam* extpar) const;
 291
 292   Bool_t SingleTrkCuts(AliESDtrack *trk,Float_t centralityperc, Bool_t &okDisplaced,Bool_t &okSoftPi, Bool_t &ok3prong) const;
 293
 294   void   SetSelectionBitForPID(AliRDHFCuts *cuts,AliAODRecoDecayHF *rd,Int_t bit);
 295
 296   AliAODv0* TransformESDv0toAODv0(AliESDv0 *esdv0,
 297                                   TObjArray *twoTrackArrayV0);
 298
 299   //
 300   ClassDef(AliAnalysisVertexingHF,20);  // Reconstruction of HF decay candidates
 301 };
 302
 303
 304 #endif
 305
 306
 307
 308
 309
 310
 311
 312