[u/mrichter/AliRoot.git] / ITS / AliITSTrackleterSPDEff.h

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

/* $Id$ */

//____________________________________________________________________
// 
// AliITSTrackleterSPDEff - find SPD chips efficiencies by using tracklets.
// 
// This class has been derived from AliITSMultReconstructor (see
// it for more details). It is the class for the Trackleter used to estimate
// SPD plane efficiency.
// The trackleter prediction is built using the vertex and 1 cluster.

//
// 
//  Author :  Giuseppe Eugenio Bruno, based on the skeleton of Reconstruct method  provided by Tiziano Virgili
//  email:    giuseppe.bruno@ba.infn.it
//  
//____________________________________________________________________

#include "AliITSMultReconstructor.h"
#include "AliITSPlaneEffSPD.h"

class AliStack;

class AliITSTrackleterSPDEff : public AliITSMultReconstructor 
{
public:
  AliITSTrackleterSPDEff();
  virtual ~AliITSTrackleterSPDEff();

  void Reconstruct(TTree* tree, Float_t* vtx, Float_t* vtxRes, AliStack* pStack=0x0);

  void SetPhiWindowL1(Float_t w=0.08) {fPhiWindowL1=w;}
  void SetZetaWindowL1(Float_t w=1.) {fZetaWindowL1=w;}
  void SetOnlyOneTrackletPerC1(Bool_t b = kTRUE) {fOnlyOneTrackletPerC1 = b;}
  
  AliITSPlaneEffSPD* GetPlaneEff() const {return fPlaneEffSPD;}
  
  void SetMC(Bool_t mc=kTRUE) {fMC=mc; InitPredictionMC(); return;}
  Bool_t GetMC() const {return fMC;}
  void SetUseOnlyPrimaryForPred(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyPrimaryForPred = flag; }
  void SetUseOnlySecondaryForPred(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlySecondaryForPred = flag;}
  void SetUseOnlySameParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlySameParticle = flag;}
  void SetUseOnlyDifferentParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyDifferentParticle = flag;}
  void SetUseOnlyStableParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyStableParticle = flag;}
  Bool_t GetUseOnlyPrimaryForPred() const {CallWarningMC(); return fUseOnlyPrimaryForPred; }
  Bool_t GetUseOnlySecondaryForPred() const {CallWarningMC(); return fUseOnlySecondaryForPred;}
  Bool_t GetUseOnlySameParticle() const {CallWarningMC(); return fUseOnlySameParticle;}
  Bool_t GetUseOnlyDifferentParticle() const {CallWarningMC(); return fUseOnlyDifferentParticle;}
  Bool_t GetUseOnlyStableParticle() const {CallWarningMC(); return fUseOnlyStableParticle;}
  Int_t GetPredictionPrimary(const UInt_t key) const;
  Int_t GetPredictionSecondary(const UInt_t key) const;
  Int_t GetClusterPrimary(const UInt_t key) const;
  Int_t GetClusterSecondary(const UInt_t key) const;
  Int_t GetPredictionPrimary(const UInt_t mod, const UInt_t chip) const
        {return GetPredictionPrimary(fPlaneEffSPD->GetKey(mod,chip));};
  Int_t GetPredictionSecondary(const UInt_t mod, const UInt_t chip) const
        {return GetPredictionSecondary(fPlaneEffSPD->GetKey(mod,chip));};
  Int_t GetClusterPrimary(const UInt_t mod, const UInt_t chip) const
        {return GetClusterPrimary(fPlaneEffSPD->GetKey(mod,chip));};
  Int_t GetClusterSecondary(const UInt_t mod, const UInt_t chip) const
        {return GetClusterSecondary(fPlaneEffSPD->GetKey(mod,chip));};
  void SavePredictionMC(TString filename="TrackletsMCpred.txt") const;
  void ReadPredictionMC(TString filename="TrackletsMCpred.txt");
  // Print some class info in ascii form to stream (cut values and MC statistics)
  virtual void PrintAscii(ostream *os)const;
  // Read some class info in ascii form from stream (cut values and MC statistics)
  virtual void ReadAscii(istream *is);
  Bool_t GetHistOn() const {return fHistOn;}; // return status of histograms
  Bool_t WriteHistosToFile(TString filename="TrackleterSPDHistos.root",Option_t* option = "RECREATE");
  void SetHistOn(Bool_t his=kTRUE) {AliITSMultReconstructor::SetHistOn(his); 
         if(GetHistOn()) {DeleteHistos(); BookHistos();} else DeleteHistos(); return;}

protected:
  AliITSTrackleterSPDEff(const AliITSTrackleterSPDEff& mr);
  AliITSTrackleterSPDEff& operator=(const AliITSTrackleterSPDEff& mr);

  Bool_t*       fAssociationFlag1;    // flag for the associations (Layer 1)
  UInt_t*       fChipPredOnLay2;      // prediction for the chip traversed by the tracklet 
                                      // based on vtx and ClusterLay1 (to be used in extrapolation)
  UInt_t*       fChipPredOnLay1;      // prediction for the chip traversed by the tracklet 
                                      // based on vtx and ClusterLay2 (to be used in interpolation)
  Int_t         fNTracklets1;   // Number of tracklets layer 1
  Float_t       fPhiWindowL1;     // Search window in phi (Layer 1)
  Float_t       fZetaWindowL1;    // SEarch window in zeta (Layer 1)
  Bool_t        fOnlyOneTrackletPerC1; // only one tracklet per cluster in L. 1
  AliITSPlaneEffSPD* fPlaneEffSPD; // pointer to SPD plane efficiency class
  Bool_t   fMC; // Boolean to access Kinematics (only for MC events )
  Bool_t   fUseOnlyPrimaryForPred; // Only for MC: if this is true, build tracklet prediction using only primary particles
  Bool_t   fUseOnlySecondaryForPred; // Only for MC: if this is true build tracklet prediction using only secondary particles
  Bool_t   fUseOnlySameParticle; // Only for MC: if this is true, assign a success only if clusters from same particles 
                                 // (i.e. PP or SS) otherwise ignore the combination
  Bool_t   fUseOnlyDifferentParticle; // Only for MC: if this is true, assign a success only if clusters from different particles 
                                      // (i.e. PP' or PS or SS') otherwise ignore the combination
  Bool_t   fUseOnlyStableParticle; // Only for MC: if this is kTRUE then method PrimaryTrackChecker return kTRUE only 
                                //              for particles decaying (eventually) after pixel layers
  Int_t *fPredictionPrimary;  // those for correction of bias from secondaries
  Int_t *fPredictionSecondary; // chip_by_chip: number of Prediction built with primaries/secondaries
  Int_t *fClusterPrimary;  //   number of clusters on a given chip fired by (at least) a primary
  Int_t *fClusterSecondary; //  number of clusters on a given chip fired by (only) secondaries
 // extra histograms with respect to the base class AliITSMultReconstructor
  TH1F*         fhClustersDPhiInterpAcc;   // Phi2 - Phi1 for tracklets (interpolation phase)
  TH1F*         fhClustersDThetaInterpAcc; // Theta2 - Theta1 for tracklets (interpolation phase)
  TH1F*         fhClustersDZetaInterpAcc;  // z2 - z1projected for tracklets (interpolation phase)
  TH1F*         fhClustersDPhiInterpAll;   // Phi2 - Phi1 all the combinations (interpolation phase)
  TH1F*         fhClustersDThetaInterpAll; // Theta2 - Theta1 all the combinations (interpolation phase)
  TH1F*         fhClustersDZetaInterpAll;  // z2 - z1projected all the combinations (interpolation phase)
  TH2F*         fhDPhiVsDThetaInterpAll; // 2D plot for all the combinations
  TH2F*         fhDPhiVsDThetaInterpAcc; // same plot for tracklets
  TH2F*         fhDPhiVsDZetaInterpAll;  // 2d plot for all the combination
  TH2F*         fhDPhiVsDZetaInterpAcc;  // same plot for tracklets
  TH1F*         fhetaClustersLay2; // Pseudorapidity distr. for Clusters L. 2
  TH1F*         fhphiClustersLay2; // Azimuthal (Phi) distr. for Clusters L. 2
//
  Double_t GetRLayer(Int_t layer); // return average radius of layer (0,1) from Geometry
  Bool_t PrimaryTrackChecker(Int_t ipart,AliStack* stack=0x0);  // check if a MC particle is primary (need AliStack)
  Int_t DecayingTrackChecker(Int_t ipart,AliStack* stack=0x0);  // For a primary particle, check if it is stable (see cxx)
  void InitPredictionMC();
  // method to locate a chip using current vtx and polar coordinate od tracklet w.r.t. to vtx (zVtx may not be given)
  Bool_t FindChip(UInt_t &key, Int_t layer,  Float_t* vtx, Float_t thetaVtx, Float_t phiVtx, Float_t zVtx=999.); 
  // method to transform from Global Cilindrical coordinate to local (module) Cartesian coordinate
  Bool_t FromGloCilToLocCart(Int_t ilayer,Int_t idet, Double_t r, Double_t phi, Double_t z,
                           Float_t &xloc, Float_t &zloc);
  // method to obtain the module (detector) index using global coordinates
  Int_t FindDetectorIndex(Int_t layer, Double_t phi, Double_t z);
  // this method gives you the intersections between a line and a circle (centred in the origin) 
  // using polar coordinates
  Bool_t FindIntersectionPolar(Double_t vtx[2],Double_t phiVtx, Double_t R,Double_t &phi);
  Bool_t SetAngleRange02Pi(Double_t &angle); // set the range of angle in [0,2pi[ 
  Bool_t SetAngleRange02Pi(Float_t  &angle) 
  {Double_t tmp=(Double_t)angle; Bool_t ret=SetAngleRange02Pi(tmp);angle=(Float_t)tmp;return ret;};  
  void CallWarningMC() const {if(!fMC) AliWarning("You can use this method only for MC! Call SetMC() first");}
  Bool_t SaveHists();
  void BookHistos(); // booking of extra histograms w.r.t. base class
  void DeleteHistos(); //delete histos from memory

  ClassDef(AliITSTrackleterSPDEff,1)
};
// Input and output function for standard C++ input/output (for the cut values and MC statistics).
ostream &operator<<(ostream &os,const AliITSTrackleterSPDEff &s);
istream &operator>>(istream &is, AliITSTrackleterSPDEff &s);
#endif
Commit	Line	Data
275a301c	1	#ifndef ALIITSTRACKLETERSPDEFF_H
	2	#define ALIITSTRACKLETERSPDEFF_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	//____________________________________________________________________
	9	//
	10	// AliITSTrackleterSPDEff - find SPD chips efficiencies by using tracklets.
	11	//
	12	// This class has been derived from AliITSMultReconstructor (see
	13	// it for more details). It is the class for the Trackleter used to estimate
	14	// SPD plane efficiency.
	15	// The trackleter prediction is built using the vertex and 1 cluster.
	16
	17	//
	18	//
	19	// Author : Giuseppe Eugenio Bruno, based on the skeleton of Reconstruct method provided by Tiziano Virgili
	20	// email: giuseppe.bruno@ba.infn.it
	21	//
	22	//____________________________________________________________________
	23
	24	#include "AliITSMultReconstructor.h"
	25	#include "AliITSPlaneEffSPD.h"
	26
	27	class AliStack;
	28
	29	class AliITSTrackleterSPDEff : public AliITSMultReconstructor
	30	{
	31	public:
	32	AliITSTrackleterSPDEff();
	33	virtual ~AliITSTrackleterSPDEff();
	34
	35	void Reconstruct(TTree* tree, Float_t* vtx, Float_t* vtxRes, AliStack* pStack=0x0);
	36
	37	void SetPhiWindowL1(Float_t w=0.08) {fPhiWindowL1=w;}
	38	void SetZetaWindowL1(Float_t w=1.) {fZetaWindowL1=w;}
	39	void SetOnlyOneTrackletPerC1(Bool_t b = kTRUE) {fOnlyOneTrackletPerC1 = b;}
	40
	41	AliITSPlaneEffSPD* GetPlaneEff() const {return fPlaneEffSPD;}
	42
	43	void SetMC(Bool_t mc=kTRUE) {fMC=mc; InitPredictionMC(); return;}
	44	Bool_t GetMC() const {return fMC;}
	45	void SetUseOnlyPrimaryForPred(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyPrimaryForPred = flag; }
	46	void SetUseOnlySecondaryForPred(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlySecondaryForPred = flag;}
	47	void SetUseOnlySameParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlySameParticle = flag;}
	48	void SetUseOnlyDifferentParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyDifferentParticle = flag;}
	49	void SetUseOnlyStableParticle(Bool_t flag=kTRUE) {CallWarningMC(); fUseOnlyStableParticle = flag;}
	50	Bool_t GetUseOnlyPrimaryForPred() const {CallWarningMC(); return fUseOnlyPrimaryForPred; }
	51	Bool_t GetUseOnlySecondaryForPred() const {CallWarningMC(); return fUseOnlySecondaryForPred;}
	52	Bool_t GetUseOnlySameParticle() const {CallWarningMC(); return fUseOnlySameParticle;}
	53	Bool_t GetUseOnlyDifferentParticle() const {CallWarningMC(); return fUseOnlyDifferentParticle;}
	54	Bool_t GetUseOnlyStableParticle() const {CallWarningMC(); return fUseOnlyStableParticle;}
	55	Int_t GetPredictionPrimary(const UInt_t key) const;
	56	Int_t GetPredictionSecondary(const UInt_t key) const;
	57	Int_t GetClusterPrimary(const UInt_t key) const;
	58	Int_t GetClusterSecondary(const UInt_t key) const;
	59	Int_t GetPredictionPrimary(const UInt_t mod, const UInt_t chip) const
	60	{return GetPredictionPrimary(fPlaneEffSPD->GetKey(mod,chip));};
	61	Int_t GetPredictionSecondary(const UInt_t mod, const UInt_t chip) const
	62	{return GetPredictionSecondary(fPlaneEffSPD->GetKey(mod,chip));};
	63	Int_t GetClusterPrimary(const UInt_t mod, const UInt_t chip) const
	64	{return GetClusterPrimary(fPlaneEffSPD->GetKey(mod,chip));};
65	Int_t GetClusterSecondary(const UInt_t mod, const UInt_t chip) const
66	{return GetClusterSecondary(fPlaneEffSPD->GetKey(mod,chip));};
67	void SavePredictionMC(TString filename="TrackletsMCpred.txt") const;
68	void ReadPredictionMC(TString filename="TrackletsMCpred.txt");
69	// Print some class info in ascii form to stream (cut values and MC statistics)
70	virtual void PrintAscii(ostream *os)const;
71	// Read some class info in ascii form from stream (cut values and MC statistics)
72	virtual void ReadAscii(istream *is);
73	Bool_t GetHistOn() const {return fHistOn;}; // return status of histograms
74	Bool_t WriteHistosToFile(TString filename="TrackleterSPDHistos.root",Option_t* option = "RECREATE");
75	void SetHistOn(Bool_t his=kTRUE) {AliITSMultReconstructor::SetHistOn(his);
76	if(GetHistOn()) {DeleteHistos(); BookHistos();} else DeleteHistos(); return;}
77
78	protected:
79	AliITSTrackleterSPDEff(const AliITSTrackleterSPDEff& mr);
80	AliITSTrackleterSPDEff& operator=(const AliITSTrackleterSPDEff& mr);
81
82	Bool_t* fAssociationFlag1; // flag for the associations (Layer 1)
83	UInt_t* fChipPredOnLay2; // prediction for the chip traversed by the tracklet
84	// based on vtx and ClusterLay1 (to be used in extrapolation)
85	UInt_t* fChipPredOnLay1; // prediction for the chip traversed by the tracklet
86	// based on vtx and ClusterLay2 (to be used in interpolation)
87	Int_t fNTracklets1; // Number of tracklets layer 1
88	Float_t fPhiWindowL1; // Search window in phi (Layer 1)
89	Float_t fZetaWindowL1; // SEarch window in zeta (Layer 1)
90	Bool_t fOnlyOneTrackletPerC1; // only one tracklet per cluster in L. 1
91	AliITSPlaneEffSPD* fPlaneEffSPD; // pointer to SPD plane efficiency class
92	Bool_t fMC; // Boolean to access Kinematics (only for MC events )
93	Bool_t fUseOnlyPrimaryForPred; // Only for MC: if this is true, build tracklet prediction using only primary particles
94	Bool_t fUseOnlySecondaryForPred; // Only for MC: if this is true build tracklet prediction using only secondary particles
95	Bool_t fUseOnlySameParticle; // Only for MC: if this is true, assign a success only if clusters from same particles
96	// (i.e. PP or SS) otherwise ignore the combination
97	Bool_t fUseOnlyDifferentParticle; // Only for MC: if this is true, assign a success only if clusters from different particles
98	// (i.e. PP' or PS or SS') otherwise ignore the combination
99	Bool_t fUseOnlyStableParticle; // Only for MC: if this is kTRUE then method PrimaryTrackChecker return kTRUE only
100	// for particles decaying (eventually) after pixel layers
101	Int_t *fPredictionPrimary; // those for correction of bias from secondaries
102	Int_t *fPredictionSecondary; // chip_by_chip: number of Prediction built with primaries/secondaries
103	Int_t *fClusterPrimary; // number of clusters on a given chip fired by (at least) a primary
104	Int_t *fClusterSecondary; // number of clusters on a given chip fired by (only) secondaries
105	// extra histograms with respect to the base class AliITSMultReconstructor
106	TH1F* fhClustersDPhiInterpAcc; // Phi2 - Phi1 for tracklets (interpolation phase)
107	TH1F* fhClustersDThetaInterpAcc; // Theta2 - Theta1 for tracklets (interpolation phase)
108	TH1F* fhClustersDZetaInterpAcc; // z2 - z1projected for tracklets (interpolation phase)
109	TH1F* fhClustersDPhiInterpAll; // Phi2 - Phi1 all the combinations (interpolation phase)
110	TH1F* fhClustersDThetaInterpAll; // Theta2 - Theta1 all the combinations (interpolation phase)
111	TH1F* fhClustersDZetaInterpAll; // z2 - z1projected all the combinations (interpolation phase)
112	TH2F* fhDPhiVsDThetaInterpAll; // 2D plot for all the combinations
113	TH2F* fhDPhiVsDThetaInterpAcc; // same plot for tracklets
114	TH2F* fhDPhiVsDZetaInterpAll; // 2d plot for all the combination
115	TH2F* fhDPhiVsDZetaInterpAcc; // same plot for tracklets
116	TH1F* fhetaClustersLay2; // Pseudorapidity distr. for Clusters L. 2
117	TH1F* fhphiClustersLay2; // Azimuthal (Phi) distr. for Clusters L. 2
118	//
119	Double_t GetRLayer(Int_t layer); // return average radius of layer (0,1) from Geometry
120	Bool_t PrimaryTrackChecker(Int_t ipart,AliStack* stack=0x0); // check if a MC particle is primary (need AliStack)
121	Int_t DecayingTrackChecker(Int_t ipart,AliStack* stack=0x0); // For a primary particle, check if it is stable (see cxx)
122	void InitPredictionMC();
123	// method to locate a chip using current vtx and polar coordinate od tracklet w.r.t. to vtx (zVtx may not be given)
124	Bool_t FindChip(UInt_t &key, Int_t layer, Float_t* vtx, Float_t thetaVtx, Float_t phiVtx, Float_t zVtx=999.);
125	// method to transform from Global Cilindrical coordinate to local (module) Cartesian coordinate
126	Bool_t FromGloCilToLocCart(Int_t ilayer,Int_t idet, Double_t r, Double_t phi, Double_t z,
127	Float_t &xloc, Float_t &zloc);
128	// method to obtain the module (detector) index using global coordinates
129	Int_t FindDetectorIndex(Int_t layer, Double_t phi, Double_t z);
130	// this method gives you the intersections between a line and a circle (centred in the origin)
131	// using polar coordinates
132	Bool_t FindIntersectionPolar(Double_t vtx[2],Double_t phiVtx, Double_t R,Double_t &phi);
133	Bool_t SetAngleRange02Pi(Double_t &angle); // set the range of angle in [0,2pi[
134	Bool_t SetAngleRange02Pi(Float_t &angle)
135	{Double_t tmp=(Double_t)angle; Bool_t ret=SetAngleRange02Pi(tmp);angle=(Float_t)tmp;return ret;};
136	void CallWarningMC() const {if(!fMC) AliWarning("You can use this method only for MC! Call SetMC() first");}
137	Bool_t SaveHists();
138	void BookHistos(); // booking of extra histograms w.r.t. base class
139	void DeleteHistos(); //delete histos from memory
140
141	ClassDef(AliITSTrackleterSPDEff,1)
142	};
143	// Input and output function for standard C++ input/output (for the cut values and MC statistics).
144	ostream &operator<<(ostream &os,const AliITSTrackleterSPDEff &s);
145	istream &operator>>(istream &is, AliITSTrackleterSPDEff &s);
146	#endif