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

#ifndef AliReaderESD_H
#define AliReaderESD_H
//___________________________________________________________________________
/////////////////////////////////////////////////////////////////////////////
//                                                                         //
// Multi file reader for ESD                                               //
//                                                                         //
// This reader reads tracks from Event Summary Data                        //
// do not read particles                                                   //
// Piotr.Skowronski@cern.ch                                                //
// more info: http://aliweb.cern.ch/people/skowron/analyzer/index.html    //
//                                                                         //
/////////////////////////////////////////////////////////////////////////////

#include "AliReader.h"
#include "AliTrackPoints.h"
#include <TString.h>
class TFile;
class AliRunLoader;
class AliESD;
class AliESDtrack;

class AliReaderESD: public AliReader
{
  public:
    AliReaderESD(const Char_t* esdfilename = "AliESDs.root", const Char_t* galfilename = "galice.root");

    AliReaderESD(TObjArray* dirs,const Char_t* esdfilename = "AliESDs.root", const Char_t* galfilename = "galice.root");

    virtual ~AliReaderESD();
    
    void          Rewind();
    
    void          ReadSimulatedData(Bool_t flag){fReadSim = flag;}//switches reading MC data
    Bool_t        ReadsRec() const {return kTRUE;}
    Bool_t        ReadsSim() const {return fReadSim;}
    void          SetCheckParticlePID(Bool_t flag){fCheckParticlePID = flag;}
    void          SetReadMostProbableOnly(Bool_t flag){fReadMostProbableOnly = flag;}
            
    void          ReadDataTPC(){}
    void          ReadDataITS(){}

    void          SetTPCNClustersRange(Int_t min,Int_t max);
    void          SetTPCChi2PerCluserRange(Float_t min, Float_t max);
    
    void          SetChi2Range(Float_t min, Float_t max);
    void          SetC00Range(Float_t min, Float_t max);
    void          SetC11Range(Float_t min, Float_t max);
    void          SetC22Range(Float_t min, Float_t max);
    void          SetC33Range(Float_t min, Float_t max);
    void          SetC44Range(Float_t min, Float_t max);
    void          SetNumberOfTrackPoints(Int_t n = 5,Float_t dr = 30.0) {fNTrackPoints = n; fdR = dr;}
    Int_t         GetNumberOfTrackPoints() const {return fNTrackPoints;}
    void          SetClusterMap(Bool_t flag = kTRUE){fClusterMap = flag;}
    void          SetITSTrackPoints(Bool_t flag, AliTrackPoints::ETypes type)
                                   {fITSTrackPoints = flag; fITSTrackPointsType = type;}
    void          MustTPC(Bool_t flag){fMustTPC = flag;}

    void          SetReadCentralBarrel(Bool_t flag){fReadCentralBarrel = flag;}
    void          SetReadMuon(Bool_t flag){fReadMuon = flag;}
    void          SetReadPHOS(Bool_t flag){fReadPHOS = flag;}

    enum ESpecies {kESDElectron = 0, kESDMuon, kESDPion, kESDKaon, kESDProton, kNSpecies};
    static Int_t  GetSpeciesPdgCode(ESpecies spec);
    
    Int_t         ReadESD(AliESD* esd);
    Int_t         ReadESDCentral(AliESD* esd);
    Int_t         ReadESDMuon(AliESD* esd);
    Int_t         ReadESDPHOS(AliESD* /*esd*/){return 0;}

  protected:
    virtual Int_t         ReadNext();
 
    virtual TFile*        OpenFile(Int_t evno);//opens files to be read for given event

    Bool_t        CheckTrack(AliESDtrack* t) const;
    
    TString       fESDFileName;//name of the file with tracks
    TString       fGAlFileName;//name of the file with tracks
    TFile*        fFile;//! pointer to current ESD file
    AliRunLoader* fRunLoader;//!Run Loader
    TIter*        fKeyIterator;//!iterator over keys in ESD file
    Bool_t        fReadSim;//flag indicating wether to read particles from kinematics
    Bool_t        fCheckParticlePID;//flag indicating to perform the check on PID of simulated particle - usefull in resoluion analysis
    Bool_t        fReadMostProbableOnly;//flag indicating to read ony one incarnation with the highest probability
    Int_t         fNTrackPoints;//number of track points; if==0 track points are not created
    Float_t       fdR;//spacing between points (along radius) in cm
                      //Track Points are needed for Anti-Merging Cut
    
    Bool_t        fClusterMap;//Flag indicating if Claster Map should be created for each track
                              //Claster map is needed for Anti-Splitting Cut

    
    Bool_t        fITSTrackPoints; //Flag indicalting if track positions in ITS are to be read
                                  //currently we use only position at first pixels wich are
	              //used by anti-merging cut in non-id analysis
    AliTrackPoints::ETypes  fITSTrackPointsType;//defines the way track points are calculated
    
    Bool_t        fMustTPC;// must be reconstructed in TPC -> reject tracks reconstructed ITS stand alone

    Bool_t        fReadCentralBarrel; // Flag for reading ESD central track 
    Bool_t        fReadMuon;// Flag for reading ESD Muon track 
    Bool_t        fReadPHOS;// Flag for reading ESD Phos 

    //Cut Parameters specific to TPC tracks
        
    Int_t         fNTPCClustMin;//Number of clusters min value
    Int_t         fNTPCClustMax;//Number of clusters max value
    
    Float_t       fTPCChi2PerClustMin;//Chi^2 per number of clusters min value
    Float_t       fTPCChi2PerClustMax;//Chi^2 per number of clusters max value


    // Required parameters at vertex
    Float_t       fChi2Min;//Chi^2 min value
    Float_t       fChi2Max;//Chi^2 max value

    Float_t       fC00Min;//C00 (0th diagonal element of covariance matrix) min value
    Float_t       fC00Max;//C00 (0th diagonal element of covariance matrix) max value
            
    Float_t       fC11Min;//C11 (1th diagonal element of covariance matrix) min value
    Float_t       fC11Max;//C11 (1th diagonal element of covariance matrix) max value
    
    Float_t       fC22Min;//C22 (2th diagonal element of covariance matrix) min value
    Float_t       fC22Max;//C22 (2th diagonal element of covariance matrix) max value
    
    Float_t       fC33Min;//C33 (3th diagonal element of covariance matrix) min value
    Float_t       fC33Max;//C33 (3th diagonal element of covariance matrix) max value
    
    Float_t       fC44Min;//C44 (4th diagonal element of covariance matrix) min value
    Float_t       fC44Max;//C44 (4th diagonal element of covariance matrix) max value

    // Required parameters at TPC Inner Layer
    Float_t       fTPCC00Min;//C00 (0th diagonal element of covariance matrix) min value
    Float_t       fTPCC00Max;//C00 (0th diagonal element of covariance matrix) max value
            
    Float_t       fTPCC11Min;//C11 (1th diagonal element of covariance matrix) min value
    Float_t       fTPCC11Max;//C11 (1th diagonal element of covariance matrix) max value
    
    Float_t       fTPCC22Min;//C22 (2th diagonal element of covariance matrix) min value
    Float_t       fTPCC22Max;//C22 (2th diagonal element of covariance matrix) max value
    
    Float_t       fTPCC33Min;//C33 (3th diagonal element of covariance matrix) min value
    Float_t       fTPCC33Max;//C33 (3th diagonal element of covariance matrix) max value
    
    Float_t       fTPCC44Min;//C44 (4th diagonal element of covariance matrix) min value
    Float_t       fTPCC44Max;//C44 (4th diagonal element of covariance matrix) max value
    
  private:
    ClassDef(AliReaderESD,1)
};


#endif
Commit	Line	Data
a5556ea5	1	#ifndef AliReaderESD_H
	2	#define AliReaderESD_H
	3	//___________________________________________________________________________
	4	/////////////////////////////////////////////////////////////////////////////
	5	// //
	6	// Multi file reader for ESD //
	7	// //
	8	// This reader reads tracks from Event Summary Data //
	9	// do not read particles //
	10	// Piotr.Skowronski@cern.ch //
c7ffd78f	11	// more info: http://aliweb.cern.ch/people/skowron/analyzer/index.html //
a5556ea5	12	// //
	13	/////////////////////////////////////////////////////////////////////////////
	14
	15	#include "AliReader.h"
6ceac6ac	16	#include "AliTrackPoints.h"
a5556ea5	17	#include <TString.h>
	18	class TFile;
	19	class AliRunLoader;
	20	class AliESD;
	21	class AliESDtrack;
	22
	23	class AliReaderESD: public AliReader
	24	{
	25	public:
	26	AliReaderESD(const Char_t* esdfilename = "AliESDs.root", const Char_t* galfilename = "galice.root");
	27
	28	AliReaderESD(TObjArray* dirs,const Char_t* esdfilename = "AliESDs.root", const Char_t* galfilename = "galice.root");
	29
	30	virtual ~AliReaderESD();
	31
	32	void Rewind();
	33
	34	void ReadSimulatedData(Bool_t flag){fReadSim = flag;}//switches reading MC data
	35	Bool_t ReadsRec() const {return kTRUE;}
	36	Bool_t ReadsSim() const {return fReadSim;}
	37	void SetCheckParticlePID(Bool_t flag){fCheckParticlePID = flag;}
	38	void SetReadMostProbableOnly(Bool_t flag){fReadMostProbableOnly = flag;}
	39
	40	void ReadDataTPC(){}
	41	void ReadDataITS(){}
	42
	43	void SetTPCNClustersRange(Int_t min,Int_t max);
	44	void SetTPCChi2PerCluserRange(Float_t min, Float_t max);
	45
	46	void SetChi2Range(Float_t min, Float_t max);
	47	void SetC00Range(Float_t min, Float_t max);
	48	void SetC11Range(Float_t min, Float_t max);
	49	void SetC22Range(Float_t min, Float_t max);
	50	void SetC33Range(Float_t min, Float_t max);
	51	void SetC44Range(Float_t min, Float_t max);
	52	void SetNumberOfTrackPoints(Int_t n = 5,Float_t dr = 30.0) {fNTrackPoints = n; fdR = dr;}
	53	Int_t GetNumberOfTrackPoints() const {return fNTrackPoints;}
	54	void SetClusterMap(Bool_t flag = kTRUE){fClusterMap = flag;}
6ceac6ac	55	void SetITSTrackPoints(Bool_t flag, AliTrackPoints::ETypes type)
6ceac6ac	56	{fITSTrackPoints = flag; fITSTrackPointsType = type;}
7814b246	57	void MustTPC(Bool_t flag){fMustTPC = flag;}
a5556ea5	58
beb1c41d	59	void SetReadCentralBarrel(Bool_t flag){fReadCentralBarrel = flag;}
	60	void SetReadMuon(Bool_t flag){fReadMuon = flag;}
	61	void SetReadPHOS(Bool_t flag){fReadPHOS = flag;}
	62
a5556ea5	63	enum ESpecies {kESDElectron = 0, kESDMuon, kESDPion, kESDKaon, kESDProton, kNSpecies};
afa8b37b	64	static Int_t GetSpeciesPdgCode(ESpecies spec);
a5556ea5	65
a5556ea5	66	Int_t ReadESD(AliESD* esd);
beb1c41d	67	Int_t ReadESDCentral(AliESD* esd);
	68	Int_t ReadESDMuon(AliESD* esd);
	69	Int_t ReadESDPHOS(AliESD* /esd/){return 0;}
	70
a5556ea5	71	protected:
beb1c41d	72	virtual Int_t ReadNext();
	73
	74	virtual TFile* OpenFile(Int_t evno);//opens files to be read for given event
	75
a5556ea5	76	Bool_t CheckTrack(AliESDtrack* t) const;
	77
	78	TString fESDFileName;//name of the file with tracks
	79	TString fGAlFileName;//name of the file with tracks
	80	TFile* fFile;//! pointer to current ESD file
	81	AliRunLoader* fRunLoader;//!Run Loader
	82	TIter* fKeyIterator;//!iterator over keys in ESD file
	83	Bool_t fReadSim;//flag indicating wether to read particles from kinematics
	84	Bool_t fCheckParticlePID;//flag indicating to perform the check on PID of simulated particle - usefull in resoluion analysis
	85	Bool_t fReadMostProbableOnly;//flag indicating to read ony one incarnation with the highest probability
	86	Int_t fNTrackPoints;//number of track points; if==0 track points are not created
	87	Float_t fdR;//spacing between points (along radius) in cm
	88	//Track Points are needed for Anti-Merging Cut
	89
	90	Bool_t fClusterMap;//Flag indicating if Claster Map should be created for each track
	91	//Claster map is needed for Anti-Splitting Cut
	92
6ceac6ac	93
6ceac6ac	94	Bool_t fITSTrackPoints; //Flag indicalting if track positions in ITS are to be read
7814b246	95	//currently we use only position at first pixels wich are
7814b246	96	//used by anti-merging cut in non-id analysis
6ceac6ac	97	AliTrackPoints::ETypes fITSTrackPointsType;//defines the way track points are calculated
6ceac6ac	98
7814b246	99	Bool_t fMustTPC;// must be reconstructed in TPC -> reject tracks reconstructed ITS stand alone
beb1c41d	100
	101	Bool_t fReadCentralBarrel; // Flag for reading ESD central track
	102	Bool_t fReadMuon;// Flag for reading ESD Muon track
	103	Bool_t fReadPHOS;// Flag for reading ESD Phos
	104
a5556ea5	105	//Cut Parameters specific to TPC tracks
	106
	107	Int_t fNTPCClustMin;//Number of clusters min value
	108	Int_t fNTPCClustMax;//Number of clusters max value
	109
	110	Float_t fTPCChi2PerClustMin;//Chi^2 per number of clusters min value
	111	Float_t fTPCChi2PerClustMax;//Chi^2 per number of clusters max value
	112
	113
	114	// Required parameters at vertex
	115	Float_t fChi2Min;//Chi^2 min value
	116	Float_t fChi2Max;//Chi^2 max value
	117
	118	Float_t fC00Min;//C00 (0th diagonal element of covariance matrix) min value
	119	Float_t fC00Max;//C00 (0th diagonal element of covariance matrix) max value
	120
	121	Float_t fC11Min;//C11 (1th diagonal element of covariance matrix) min value
	122	Float_t fC11Max;//C11 (1th diagonal element of covariance matrix) max value
	123
	124	Float_t fC22Min;//C22 (2th diagonal element of covariance matrix) min value
	125	Float_t fC22Max;//C22 (2th diagonal element of covariance matrix) max value
	126
	127	Float_t fC33Min;//C33 (3th diagonal element of covariance matrix) min value
	128	Float_t fC33Max;//C33 (3th diagonal element of covariance matrix) max value
	129
	130	Float_t fC44Min;//C44 (4th diagonal element of covariance matrix) min value
	131	Float_t fC44Max;//C44 (4th diagonal element of covariance matrix) max value
	132
	133	// Required parameters at TPC Inner Layer
	134	Float_t fTPCC00Min;//C00 (0th diagonal element of covariance matrix) min value
	135	Float_t fTPCC00Max;//C00 (0th diagonal element of covariance matrix) max value
	136
	137	Float_t fTPCC11Min;//C11 (1th diagonal element of covariance matrix) min value
	138	Float_t fTPCC11Max;//C11 (1th diagonal element of covariance matrix) max value
	139
	140	Float_t fTPCC22Min;//C22 (2th diagonal element of covariance matrix) min value
	141	Float_t fTPCC22Max;//C22 (2th diagonal element of covariance matrix) max value
	142
	143	Float_t fTPCC33Min;//C33 (3th diagonal element of covariance matrix) min value
	144	Float_t fTPCC33Max;//C33 (3th diagonal element of covariance matrix) max value
	145
	146	Float_t fTPCC44Min;//C44 (4th diagonal element of covariance matrix) min value
	147	Float_t fTPCC44Max;//C44 (4th diagonal element of covariance matrix) max value
	148
	149	private:
	150	ClassDef(AliReaderESD,1)
	151	};
	152
	153
	154	#endif