ANALYSIS/AliD0toKpiAnalysis.h

   1 #ifndef AliD0toKpiAnalysis_H
   2 #define AliD0toKpiAnalysis_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 //-------------------------------------------------------------------------
   7 //                      Class AliD0toKpiAnalysis
   8 //             Reconstruction and analysis D0 -> K^- pi^+
   9 //
  10 //         Origin: A. Dainese    andrea.dainese@pd.infn.it
  11 //-------------------------------------------------------------------------
  12
  13 #include <TString.h>
  14 #include <TNamed.h>
  15 #include "AliESD.h"
  16 #include "AliITStrackV2.h"
  17
  18 //-----------------------------------------------------------------------------
  19 class AliD0toKpiAnalysis : public TNamed {
  20  public:
  21   //
  22   AliD0toKpiAnalysis();
  23   virtual ~AliD0toKpiAnalysis();
  24
  25   void ApplySelection(const Char_t *inName="AliD0toKpi.root",
  26                       const Char_t *outName="AliD0toKpi_sele.root") const;
  27   void FindCandidates(Int_t evFirst=0,Int_t evLast=0,
  28                       const Char_t *outName="AliD0toKpi.root");
  29   void FindCandidatesESD(Int_t evFirst=0,Int_t evLast=0,
  30                          const Char_t *outName="AliD0toKpi.root");
  31   void PrintStatus() const;
  32   void SetBz(Double_t bz=-9999.) { fBz=bz; }
  33   void SetVertexOnTheFly() { fVertexOnTheFly=kTRUE; }
  34   void SetSimulation() { fSim=kTRUE; }
  35   void SetOnlySignal() { fOnlySignal=kTRUE; }
  36   void SetPtCut(Double_t pt=0.) { fPtCut=pt; }
  37   void Setd0Cut(Double_t d0=0.) { fd0Cut=d0; }
  38   void SetMassCut(Double_t deltaM=1000.) { fMassCut=deltaM; }
  39   void SetD0Cuts(Double_t cut0=1000.,Double_t cut1=100000.,
  40                  Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
  41                  Double_t cut5=100000.,Double_t cut6=100000.,
  42                  Double_t cut7=100000000.,Double_t cut8=-1.1);
  43   void SetD0Cuts(const Double_t cuts[9]);
  44   void SetPID(const Char_t * pid="TOFparam_PbPb") { fPID=pid; }
  45   void SetDebug() { fDebug=kTRUE; }
  46   //
  47  private:
  48   //
  49   Double_t fBz;             // value of the magnetic field
  50   Bool_t   fVertexOnTheFly; // flag for primary vertex reco on the fly
  51   Bool_t   fSim;            // flag for the analysis of simulated events
  52   Bool_t   fOnlySignal;     // write to file only signal candidates (for sim)
  53   TString  fPID;           // PID scheme
  54
  55   Double_t fV1[3]; // primary vertex position (in cm)
  56   Double_t fPtCut;   // minimum track pt (in GeV/c)
  57   Double_t fd0Cut;   // minimum track |rphi impact parameter| (in micron)
  58   Double_t fMassCut; // maximum of |InvMass-MD0| (in GeV)
  59   Double_t fD0Cuts[9]; // cuts on D0 candidates (see SetD0Cuts())
  60                        // (to be passed to function AliD0toKpi::Select())
  61                        // 0 = inv. mass half width [GeV]
  62                        // 1 = dca [micron]
  63                        // 2 = cosThetaStar
  64                        // 3 = pTK [GeV/c]
  65                        // 4 = pTPi [GeV/c]
  66                        // 5 = d0K [micron]   upper limit!
  67                        // 6 = d0Pi [micron]  upper limit!
  68                        // 7 = d0d0 [micron^2]
  69                        // 8 = cosThetaPoint
  70
  71   Bool_t fDebug;       // debug option
  72   //
  73   Double_t CalculateTOFmass(Double_t mom,Double_t length,Double_t time) const;
  74   void     MakeTracksRefFile(Int_t evFirst=0,Int_t evLast=0) const;
  75   void     MakeTracksRefFileESD() const;
  76   Bool_t   SelectInvMass(const Double_t p[6]) const;
  77   void     SelectTracks(TTree &trkTree,
  78                         TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
  79                         TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
  80   void     SelectTracksESD(AliESD &event,
  81                            TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
  82                            TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
  83   void     SelectTracksESDvtx(AliESD &event,TTree *trkTree,
  84                               TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
  85                               TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
  86   void     SetVertex1(Double_t x=0.,Double_t y=0.,Double_t z=0.)
  87     { fV1[0]=x;fV1[1]=y;fV1[2]=z; }
  88   void     SimulationInfo(TTree *treeD0in,TTree *treeD0out) const;
  89   Bool_t   SingleTrkCuts(const AliITStrackV2& trk) const;
  90   //
  91   ClassDef(AliD0toKpiAnalysis,1)  // Reconstruction of D0 candidates class
  92 };
  93
  94
  95 #endif
  96
  97
  98
  99
 100
 101
 102
 103