[u/mrichter/AliRoot.git] / HLT / src / AliL3Evaluate.h

#ifndef ALIL3_Evaluate
#define ALIL3_Evaluate

#include <TObject.h>
#include <TH1.h>
#include <TTree.h>
#include "AliSimDigits.h"

class AliL3Track;
class TClonesArray;
class TFile;
class AliL3TrackArray;
class AliL3SpacePointData;
class TH1F;
class AliL3Transform;
class AliTPCParam;
class TTree;
class AliSimDigits;
class TObjArray;

class AliL3Evaluate : public TObject {

 private:

  TFile *fMCFile;
  TFile *fMCclusterfile;  //If you run the fast simulator.
  TClonesArray *fParticles;
  AliL3TrackArray *fTracks; //!
  AliTPCParam *fParam;
  AliL3SpacePointData *fClusters[36][5]; //!
  AliL3Transform *fTransform; //!
  AliSimDigits *fDigits;
  TTree *fDigitsTree;
  Int_t fMinSlice;
  Int_t fMaxSlice;
  UInt_t fNcl[36][5];
  Int_t fRowid[36][174];
  Int_t fMinPointsOnTrack;  //Minimum points on track to be considered.
  Bool_t fIsSlow;
  Int_t fNFastPoints;
  UInt_t fMcIndex;
  Int_t fMcId;
  
  //Histograms
  TH1F *fPtRes;
  TH1F *fNGoodTracksPt;
  TH1F *fNFoundTracksPt;
  TH1F *fNFakeTracksPt;
  TH1F *fTrackEffPt;
  TH1F *fFakeTrackEffPt;
  TH1F *fNGoodTracksEta;
  TH1F *fNFoundTracksEta;
  TH1F *fNFakeTracksEta;
  TH1F *fTrackEffEta;
  TH1F *fFakeTrackEffEta;
  
  void FillEffHistos(TObjArray *good_particles,Int_t *particle_id);
  void AssignIDs();
  Bool_t SetDigitsTree();
  Bool_t SetMCParticleArray();
  TObjArray *DefineGoodTracks(Int_t slice,Int_t *padrow,Int_t good_number,Int_t *particle_id);
  Int_t GetMCTrackLabel(AliL3Track *track,UInt_t *index=0,Int_t *pID=0,Int_t npoints=0);
  Int_t **GetClusterIDs(AliL3Track *track,UInt_t *index=0,Int_t *pID=0,Int_t npoints=0);
  Int_t *GetFastIDs(UInt_t &tmp_ind,Int_t &npoints);
 
 public:
  AliL3Evaluate();
  AliL3Evaluate(Char_t *mcfile,Int_t *slice);

  virtual ~AliL3Evaluate();

  void Setup(Char_t *trackfile,Char_t *clustfile);
  void SetupFast(Char_t *trackfile,Char_t *clustfile,Char_t *mcClusterfile);
  void CreateHistos(Int_t nbin=20,Int_t xlow=0,Int_t xup=4);
  void EvaluatePatch(Int_t slice,Int_t patch,Int_t min_points,Int_t good_number);
  void EvaluateSlice(Int_t slice,Int_t min_points,Int_t good_number);
  void EvaluateGlobal();
  void Write2File(Char_t *outputfile);
    
  TH1F *GetTrackEffPt() {return fTrackEffPt;}
  TH1F *GetTrackEffEta() {return fTrackEffEta;}
  TH1F *GetPtRes() {return fPtRes;}

  void SetMinPoints(Int_t f) {fMinPointsOnTrack = f;}

  ClassDef(AliL3Evaluate,1) 
};

#endif
Commit	Line	Data
108615fc	1	#ifndef ALIL3_Evaluate
	2	#define ALIL3_Evaluate
	3
	4	#include <TObject.h>
	5	#include <TH1.h>
	6	#include <TTree.h>
	7	#include "AliSimDigits.h"
	8
	9	class AliL3Track;
	10	class TClonesArray;
	11	class TFile;
	12	class AliL3TrackArray;
	13	class AliL3SpacePointData;
	14	class TH1F;
	15	class AliL3Transform;
	16	class AliTPCParam;
	17	class TTree;
	18	class AliSimDigits;
	19	class TObjArray;
	20
	21	class AliL3Evaluate : public TObject {
	22
	23	private:
	24
	25	TFile *fMCFile;
	26	TFile *fMCclusterfile; //If you run the fast simulator.
	27	TClonesArray *fParticles;
	28	AliL3TrackArray *fTracks; //!
	29	AliTPCParam *fParam;
	30	AliL3SpacePointData *fClusters[36][5]; //!
	31	AliL3Transform *fTransform; //!
	32	AliSimDigits *fDigits;
	33	TTree *fDigitsTree;
	34	Int_t fMinSlice;
	35	Int_t fMaxSlice;
	36	UInt_t fNcl[36][5];
	37	Int_t fRowid[36][174];
	38	Int_t fMinPointsOnTrack; //Minimum points on track to be considered.
	39	Bool_t fIsSlow;
	40	Int_t fNFastPoints;
	41	UInt_t fMcIndex;
	42	Int_t fMcId;
	43
	44	//Histograms
	45	TH1F *fPtRes;
	46	TH1F *fNGoodTracksPt;
	47	TH1F *fNFoundTracksPt;
	48	TH1F *fNFakeTracksPt;
	49	TH1F *fTrackEffPt;
	50	TH1F *fFakeTrackEffPt;
	51	TH1F *fNGoodTracksEta;
	52	TH1F *fNFoundTracksEta;
	53	TH1F *fNFakeTracksEta;
	54	TH1F *fTrackEffEta;
	55	TH1F *fFakeTrackEffEta;
	56
	57	void FillEffHistos(TObjArray good_particles,Int_t particle_id);
	58	void AssignIDs();
	59	Bool_t SetDigitsTree();
	60	Bool_t SetMCParticleArray();
	61	TObjArray DefineGoodTracks(Int_t slice,Int_t padrow,Int_t good_number,Int_t *particle_id);
	62	Int_t GetMCTrackLabel(AliL3Track track,UInt_t index=0,Int_t *pID=0,Int_t npoints=0);
	63	Int_t *GetClusterIDs(AliL3Track track,UInt_t index=0,Int_t pID=0,Int_t npoints=0);
	64	Int_t *GetFastIDs(UInt_t &tmp_ind,Int_t &npoints);
65
66	public:
67	AliL3Evaluate();
68	AliL3Evaluate(Char_t mcfile,Int_t slice);
69
70	virtual ~AliL3Evaluate();
71
72	void Setup(Char_t trackfile,Char_t clustfile);
73	void SetupFast(Char_t trackfile,Char_t clustfile,Char_t *mcClusterfile);
74	void CreateHistos(Int_t nbin=20,Int_t xlow=0,Int_t xup=4);
75	void EvaluatePatch(Int_t slice,Int_t patch,Int_t min_points,Int_t good_number);
76	void EvaluateSlice(Int_t slice,Int_t min_points,Int_t good_number);
77	void EvaluateGlobal();
78	void Write2File(Char_t *outputfile);
79
80	TH1F *GetTrackEffPt() {return fTrackEffPt;}
81	TH1F *GetTrackEffEta() {return fTrackEffEta;}
82	TH1F *GetPtRes() {return fPtRes;}
83
84	void SetMinPoints(Int_t f) {fMinPointsOnTrack = f;}
85
86	ClassDef(AliL3Evaluate,1)
87	};
88
89	#endif