GetESD.sh becomes runtest.sh

[u/mrichter/AliRoot.git] / ANALYSIS / AliD0toKpiAnalysis.h

diff --git a/ANALYSIS/AliD0toKpiAnalysis.h b/ANALYSIS/AliD0toKpiAnalysis.h
index 326d085c1e8b4c27a85d26dea441d521e6c0c7f2..bc6ebcff56b7c240aa6b0722a19b48068dd7717a 100644 (file)
--- a/ANALYSIS/AliD0toKpiAnalysis.h
+++ b/ANALYSIS/AliD0toKpiAnalysis.h
@@ -7,19 +7,13 @@
 //                      Class AliD0toKpiAnalysis
 //             Reconstruction and analysis D0 -> K^- pi^+
 //      
-//         Origin: A. Dainese    andrea.dainese@pd.infn.it                  
+//         Origin: A. Dainese    andrea.dainese@lnl.infn.it                  
 //-------------------------------------------------------------------------
 
-#include <Riostream.h>
 #include <TString.h>
 #include <TNamed.h>
 #include "AliESD.h"
-#include "AliITStrackV2.h"
-#include "AliITSVertexerTracks.h"
-#include "AliESDVertex.h"
-#include "AliV0vertexer.h"
-#include "AliV0vertex.h"
-#include "AliD0toKpi.h"
+#include "AliRun.h"
 
 //-----------------------------------------------------------------------------
 class AliD0toKpiAnalysis : public TNamed {
@@ -32,27 +26,23 @@ class AliD0toKpiAnalysis : public TNamed {
                      const Char_t *outName="AliD0toKpi_sele.root") const;
   void FindCandidates(Int_t evFirst=0,Int_t evLast=0,
                      const Char_t *outName="AliD0toKpi.root");
-  void FindCandidatesESD(Int_t evFirst=0,Int_t evLast=0,
-                        const Char_t *outName="AliD0toKpi.root");
+  void MakeTracksRefFile(AliRun *gAlice,Int_t evFirst=0,Int_t evLast=0) const;
   void PrintStatus() const;
-  void SetBz(Double_t bz=-9999.) { fBz=bz; return; }
-  void SetVertexOnTheFly() { fVertexOnTheFly=kTRUE; return; }
-  void SetSimulation() { fSim=kTRUE; return; }
-  void SetOnlySignal() { fOnlySignal=kTRUE; return; }
-  void SetPtCut(Double_t pt=0.) { fPtCut=pt; return; }
-  void Setd0Cut(Double_t d0=0.) { fd0Cut=d0; return; } 
-  void SetMassCut(Double_t deltaM=1000.) { fMassCut=deltaM; return; }
+  void SetVertexOnTheFly() { fVertexOnTheFly=kTRUE; }
+  void SetSimulation() { fSim=kTRUE; }
+  void SetOnlySignal() { fOnlySignal=kTRUE; }
+  void SetPtCut(Double_t pt=0.) { fPtCut=pt; }
+  void Setd0Cut(Double_t d0=0.) { fd0Cut=d0; } 
+  void SetMassCut(Double_t deltaM=1000.) { fMassCut=deltaM; }
   void SetD0Cuts(Double_t cut0=1000.,Double_t cut1=100000.,
                 Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
                 Double_t cut5=100000.,Double_t cut6=100000.,
                 Double_t cut7=100000000.,Double_t cut8=-1.1); 
   void SetD0Cuts(const Double_t cuts[9]); 
-  void SetPID(TString pid="TOFparam_PbPb") { fPID=pid; return; }
-  void SetDebug() { fDebug=kTRUE; return; }
+  void SetPID(const Char_t * pid="TOFparam_PbPb") { fPID=pid; }
   //
  private:
   //
-  Double_t fBz;             // value of the magnetic field
   Bool_t   fVertexOnTheFly; // flag for primary vertex reco on the fly
   Bool_t   fSim;            // flag for the analysis of simulated events
   Bool_t   fOnlySignal;     // write to file only signal candidates (for sim)
@@ -74,27 +64,18 @@ class AliD0toKpiAnalysis : public TNamed {
                        // 7 = d0d0 [micron^2]
                        // 8 = cosThetaPoint
 
-  Bool_t fDebug;       // debug option
   //
   Double_t CalculateTOFmass(Double_t mom,Double_t length,Double_t time) const;
-  void     MakeTracksRefFile(Int_t evFirst=0,Int_t evLast=0) const;
-  void     MakeTracksRefFileESD() const;
   Bool_t   SelectInvMass(const Double_t p[6]) const;
-  void     SelectTracks(TTree &trkTree,
-                     TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
-                     TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const; 
-  void     SelectTracksESD(AliESD &event,
-       TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
-       TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const; 
-  void     SelectTracksESDvtx(AliESD &event,TTree *trkTree,
-       TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
-       TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const; 
+  void     SelectTracks(AliESD *event,
+                       TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
+                       TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
   void     SetVertex1(Double_t x=0.,Double_t y=0.,Double_t z=0.) 
-    { fV1[0]=x;fV1[1]=y;fV1[2]=z; return; }
+    { fV1[0]=x;fV1[1]=y;fV1[2]=z; }
   void     SimulationInfo(TTree *treeD0in,TTree *treeD0out) const;
-  Bool_t   SingleTrkCuts(const AliITStrackV2& trk) const;
+  Bool_t   SingleTrkCuts(const AliESDtrack& trk, Double_t b) const;
   //
-  ClassDef(AliD0toKpiAnalysis,1)  // Reconstruction of D0 candidates class
+  ClassDef(AliD0toKpiAnalysis,3)  // Reconstruction of D0 candidates class
 };