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

#ifndef AliD0toKpiAnalysis_H
#define AliD0toKpiAnalysis_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//-------------------------------------------------------------------------
//                      Class AliD0toKpiAnalysis
//             Reconstruction and analysis D0 -> K^- pi^+
//      
//         Origin: A. Dainese    andrea.dainese@pd.infn.it                  
//-------------------------------------------------------------------------

#include <TString.h>
#include <TNamed.h>
#include "AliESD.h"
#include "AliITStrackV2.h"

//-----------------------------------------------------------------------------
class AliD0toKpiAnalysis : public TNamed {
 public:
  //
  AliD0toKpiAnalysis();
  virtual ~AliD0toKpiAnalysis();

  void ApplySelection(const Char_t *inName="AliD0toKpi.root",
		      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 PrintStatus() const;
  void SetBz(Double_t bz=-9999.) { fBz=bz; }
  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(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)
  TString  fPID;           // PID scheme

  Double_t fV1[3]; // primary vertex position (in cm)
  Double_t fPtCut;   // minimum track pt (in GeV/c)
  Double_t fd0Cut;   // minimum track |rphi impact parameter| (in micron) 
  Double_t fMassCut; // maximum of |InvMass-MD0| (in GeV)
  Double_t fD0Cuts[9]; // cuts on D0 candidates (see SetD0Cuts())
                       // (to be passed to function AliD0toKpi::Select())
                       // 0 = inv. mass half width [GeV]   
                       // 1 = dca [micron]
                       // 2 = cosThetaStar 
                       // 3 = pTK [GeV/c]
                       // 4 = pTPi [GeV/c]
                       // 5 = d0K [micron]   upper limit!
                       // 6 = d0Pi [micron]  upper limit!
                       // 7 = d0d0 [micron^2]
                       // 8 = cosThetaPoint

  //
  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     SetVertex1(Double_t x=0.,Double_t y=0.,Double_t z=0.) 
    { fV1[0]=x;fV1[1]=y;fV1[2]=z; }
  void     SimulationInfo(TTree *treeD0in,TTree *treeD0out) const;
  Bool_t   SingleTrkCuts(const AliITStrackV2& trk) const;
  //
  ClassDef(AliD0toKpiAnalysis,2)  // Reconstruction of D0 candidates class
};


#endif
Commit	Line	Data
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"
	16	#include "AliITStrackV2.h"
3a9a3487	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;
7b84ea57	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; }
3a9a3487	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]);
7b84ea57	44	void SetPID(const Char_t * pid="TOFparam_PbPb") { fPID=pid; }
3a9a3487	45	//
	46	private:
	47	//
	48	Double_t fBz; // value of the magnetic field
	49	Bool_t fVertexOnTheFly; // flag for primary vertex reco on the fly
	50	Bool_t fSim; // flag for the analysis of simulated events
	51	Bool_t fOnlySignal; // write to file only signal candidates (for sim)
	52	TString fPID; // PID scheme
	53
	54	Double_t fV1[3]; // primary vertex position (in cm)
	55	Double_t fPtCut; // minimum track pt (in GeV/c)
	56	Double_t fd0Cut; // minimum track \|rphi impact parameter\| (in micron)
	57	Double_t fMassCut; // maximum of \|InvMass-MD0\| (in GeV)
	58	Double_t fD0Cuts[9]; // cuts on D0 candidates (see SetD0Cuts())
	59	// (to be passed to function AliD0toKpi::Select())
	60	// 0 = inv. mass half width [GeV]
	61	// 1 = dca [micron]
	62	// 2 = cosThetaStar
	63	// 3 = pTK [GeV/c]
	64	// 4 = pTPi [GeV/c]
	65	// 5 = d0K [micron] upper limit!
	66	// 6 = d0Pi [micron] upper limit!
	67	// 7 = d0d0 [micron^2]
	68	// 8 = cosThetaPoint
	69
3a9a3487	70	//
	71	Double_t CalculateTOFmass(Double_t mom,Double_t length,Double_t time) const;
	72	void MakeTracksRefFile(Int_t evFirst=0,Int_t evLast=0) const;
	73	void MakeTracksRefFileESD() const;
	74	Bool_t SelectInvMass(const Double_t p[6]) const;
	75	void SelectTracks(TTree &trkTree,
7b84ea57	76	TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
7b84ea57	77	TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
3a9a3487	78	void SelectTracksESD(AliESD &event,
7b84ea57	79	TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
7b84ea57	80	TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
3a9a3487	81	void SelectTracksESDvtx(AliESD &event,TTree *trkTree,
7b84ea57	82	TObjArray &trksP,Int_t *trkEntryP,Int_t &nTrksP,
7b84ea57	83	TObjArray &trksN,Int_t *trkEntryN,Int_t &nTrksN) const;
3a9a3487	84	void SetVertex1(Double_t x=0.,Double_t y=0.,Double_t z=0.)
7b84ea57	85	{ fV1[0]=x;fV1[1]=y;fV1[2]=z; }
3a9a3487	86	void SimulationInfo(TTree treeD0in,TTree treeD0out) const;
	87	Bool_t SingleTrkCuts(const AliITStrackV2& trk) const;
	88	//
59c6807d	89	ClassDef(AliD0toKpiAnalysis,2) // Reconstruction of D0 candidates class
3a9a3487	90	};
	91
	92
	93	#endif
	94
	95
	96
	97
	98
	99
	100
	101