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

#ifndef ALIL3_Evaluate
#define ALIL3_Evaluate

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


struct GoodTrack 
{
  Int_t label;
  Double_t eta;
  Int_t code;
  Double_t px,py,pz;
  Double_t x,y,z;
  Int_t nhits;
  Int_t sector;
};
typedef struct GoodTrack GoodTrack;


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

class AliL3Evaluate {

 private:

  AliL3TrackArray *fTracks; //!
  AliTPCParam *fParam;
  AliL3SpacePointData *fClusters[36][6]; //!
  AliSimDigits *fDigits;
  TTree *fDigitsTree;
  Int_t fMinSlice;
  Int_t fMaxSlice;
  UInt_t fNcl[36][6];
  Int_t fRowid[36][176];
  Int_t fMinPointsOnTrack;  //Minimum points on track to be considered.
  Int_t fMinHitsFromParticle;
  GoodTrack *fGoodTracks; //!
  Float_t fMaxFalseClusters;
  
  Int_t fNFastPoints;
  UInt_t *fMcIndex;//!
  Int_t *fMcId;//!
  Int_t fGoodFound;
  Int_t fGoodGen;
  Double_t fMinGoodPt;

  //Histograms
  TNtuple *fNtuppel;
  TH1F *fPtRes;
  TH1F *fNGoodTracksPt;
  TH1F *fNFoundTracksPt;
  TH1F *fNFakeTracksPt;
  TH1F *fTrackEffPt;
  TH1F *fFakeTrackEffPt;
  TH1F *fNGoodTracksEta;
  TH1F *fNFoundTracksEta;
  TH1F *fNFakeTracksEta;
  TH1F *fTrackEffEta;
  TH1F *fFakeTrackEffEta;
  
  
 public:
  AliL3Evaluate();
  AliL3Evaluate(Char_t *path,Int_t min_clusters,Int_t minhits,Double_t minpt=0.1,Int_t *slice=0);
  virtual ~AliL3Evaluate();

  void CreateHistos(Int_t nbin=20,Float_t xlow=0,Float_t xup=4);
  void Write2File(Char_t *outputfile);
  void FillEffHistos();
  void FillEffHistosNAIVE();
  void CalcEffHistos();
  void AssignIDs();
  void GetGoodParticles(Char_t *particle_file,Bool_t sector=kFALSE);
  void GetFastClusterIDs(Char_t *path);
  void GetCFeff(Char_t *outfile);
  Int_t GetMCTrackLabel(AliL3Track *track);
  TNtupleD *CalculateResiduals(Char_t *datapath);
  TNtuple *EvaluatePoints(Char_t *rootfile);
  
  void SetMinPoints(Int_t f) {fMinPointsOnTrack = f;}
  void SetMinGoodPt(Double_t f) {fMinGoodPt = f;}
  void SetMaxFalseClusters(Float_t f) {fMaxFalseClusters = f;}

  Int_t GetNGoodTracks() {return fGoodGen;}
  Int_t GetNFoundTracks() {return fGoodFound;}
  TH1F *GetTrackEffPt() {return fTrackEffPt;}
  TH1F *GetTrackEffEta() {return fTrackEffEta;}
  TH1F *GetFakeEffEta() {return fFakeTrackEffEta;}
  TH1F *GetFakeEffPt() {return fFakeTrackEffPt;}
  TH1F *GetPtRes() {return fPtRes;}

  
  ClassDef(AliL3Evaluate,1) //Tracking evaluation class
};

#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>
1f79afc0	7	#include <TNtuple.h>
108615fc	8	#include "AliSimDigits.h"
108615fc	9
e1a32fa5	10
	11	struct GoodTrack
	12	{
e1a32fa5	13	Int_t label;
40896357	14	Double_t eta;
e1a32fa5	15	Int_t code;
e1a32fa5	16	Double_t px,py,pz;
e0f4d6b2	17	Double_t x,y,z;
e0f4d6b2	18	Int_t nhits;
a8a3f5c2	19	Int_t sector;
e1a32fa5	20	};
	21	typedef struct GoodTrack GoodTrack;
	22
	23
108615fc	24	class AliL3Track;
	25	class TClonesArray;
	26	class TFile;
	27	class AliL3TrackArray;
	28	class AliL3SpacePointData;
	29	class TH1F;
108615fc	30	class AliTPCParam;
	31	class TTree;
	32	class AliSimDigits;
	33	class TObjArray;
78127e35	34	class TParticle;
a8a3f5c2	35	class TNtupleD;
108615fc	36
e0f4d6b2	37	class AliL3Evaluate {
108615fc	38
	39	private:
	40
108615fc	41	AliL3TrackArray *fTracks; //!
108615fc	42	AliTPCParam *fParam;
40896357	43	AliL3SpacePointData *fClusters[36][6]; //!
108615fc	44	AliSimDigits *fDigits;
	45	TTree *fDigitsTree;
	46	Int_t fMinSlice;
	47	Int_t fMaxSlice;
40896357	48	UInt_t fNcl[36][6];
ae97a0b9	49	Int_t fRowid[36][176];
108615fc	50	Int_t fMinPointsOnTrack; //Minimum points on track to be considered.
e0f4d6b2	51	Int_t fMinHitsFromParticle;
	52	GoodTrack *fGoodTracks; //!
	53	Float_t fMaxFalseClusters;
	54
108615fc	55	Int_t fNFastPoints;
c0217812	56	UInt_t *fMcIndex;//!
c0217812	57	Int_t *fMcId;//!
23908b9b	58	Int_t fGoodFound;
23908b9b	59	Int_t fGoodGen;
e170e0f9	60	Double_t fMinGoodPt;
23908b9b	61
108615fc	62	//Histograms
1f79afc0	63	TNtuple *fNtuppel;
108615fc	64	TH1F *fPtRes;
	65	TH1F *fNGoodTracksPt;
	66	TH1F *fNFoundTracksPt;
	67	TH1F *fNFakeTracksPt;
	68	TH1F *fTrackEffPt;
	69	TH1F *fFakeTrackEffPt;
	70	TH1F *fNGoodTracksEta;
	71	TH1F *fNFoundTracksEta;
	72	TH1F *fNFakeTracksEta;
	73	TH1F *fTrackEffEta;
	74	TH1F *fFakeTrackEffEta;
	75
e0f4d6b2	76
e1a32fa5	77
108615fc	78	public:
108615fc	79	AliL3Evaluate();
e0f4d6b2	80	AliL3Evaluate(Char_t path,Int_t min_clusters,Int_t minhits,Double_t minpt=0.1,Int_t slice=0);
108615fc	81	virtual ~AliL3Evaluate();
108615fc	82
e0f4d6b2	83	void CreateHistos(Int_t nbin=20,Float_t xlow=0,Float_t xup=4);
108615fc	84	void Write2File(Char_t *outputfile);
e0f4d6b2	85	void FillEffHistos();
	86	void FillEffHistosNAIVE();
	87	void CalcEffHistos();
	88	void AssignIDs();
a8a3f5c2	89	void GetGoodParticles(Char_t *particle_file,Bool_t sector=kFALSE);
e0f4d6b2	90	void GetFastClusterIDs(Char_t *path);
	91	void GetCFeff(Char_t *outfile);
	92	Int_t GetMCTrackLabel(AliL3Track *track);
a8a3f5c2	93	TNtupleD CalculateResiduals(Char_t datapath);
e0f4d6b2	94	TNtuple EvaluatePoints(Char_t rootfile);
e0f4d6b2	95
108615fc	96	void SetMinPoints(Int_t f) {fMinPointsOnTrack = f;}
e170e0f9	97	void SetMinGoodPt(Double_t f) {fMinGoodPt = f;}
e0f4d6b2	98	void SetMaxFalseClusters(Float_t f) {fMaxFalseClusters = f;}
e0f4d6b2	99
23908b9b	100	Int_t GetNGoodTracks() {return fGoodGen;}
23908b9b	101	Int_t GetNFoundTracks() {return fGoodFound;}
e0f4d6b2	102	TH1F *GetTrackEffPt() {return fTrackEffPt;}
	103	TH1F *GetTrackEffEta() {return fTrackEffEta;}
	104	TH1F *GetFakeEffEta() {return fFakeTrackEffEta;}
	105	TH1F *GetFakeEffPt() {return fFakeTrackEffPt;}
	106	TH1F *GetPtRes() {return fPtRes;}
78127e35	107
e0f4d6b2	108
b661165c	109	ClassDef(AliL3Evaluate,1) //Tracking evaluation class
108615fc	110	};
	111
	112	#endif