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