Correct treatment of tracks with pT<pTmin

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

diff --git a/ITS/AliITSPid.h b/ITS/AliITSPid.h
index 8de43be4142da155c7644ded3222f6dd229955fb..e8e393537b387e7dd0229f0d97220d719a880a1e 100644 (file)
--- a/ITS/AliITSPid.h
+++ b/ITS/AliITSPid.h
@@ -1,61 +1,81 @@
 #ifndef ALIITSPID_H
 #define ALIITSPID_H
+/////////////////////////////////////////////////////////////////
+// Class for identification of pions,kaons and protons in ITS  //
+// Prior particles population (probabilities) are taken from   //
+// Hijing event generator.                                     //
+/////////////////////////////////////////////////////////////////
+
 #include <TObject.h>
-#include <TClonesArray.h>
-#include <TVector.h>
-#include "../TPC/AliTPCtrack.h"
-#include "AliITSIOTrack.h"
-#include "AliITStrackV2.h"
-#include <assert.h>
-//#include <stl.h>
-//___________________________________________________________________________
-class  AliITSPid :
-  public TObject {
+#include <TVectorfwd.h>
+
+class TClonesArray;
+class AliITSIOTrack;
+class AliKalmanTrack;
+class AliITStrackV2;
+class TF1;
+
+
+class  AliITSPid : public TObject {
 
 public:
                AliITSPid(Int_t ntrs=1000);
                virtual ~AliITSPid(){}
+
        void    SetEdep(Int_t track,Float_t Edep);
        void    SetPmom(Int_t track,Float_t Pmom);
        void    SetPcod(Int_t track,Int_t Pcod);
        void    Print(Int_t track);
+        virtual void Print(Option_t *option="") const {TObject::Print(option);}
        void    Tab(void);
        void    Reset(void);
-       void    SetVec(Int_t track,TVector info);
-       TVector* GetVec(Int_t track);
-       Int_t   GetPcode(TClonesArray*,Float_t);
-       Int_t   GetPcode(Float_t,Float_t);
-       Int_t   GetPcode(AliTPCtrack*);  // For PID TPC
-//        Int_t   GetPcode(AliITSIOTrack*); // For PID ITS tracking V1
-        Int_t   GetPcode(AliITStrackV2*); // For PID ITS tracking V2
-       void    SetCut(Int_t,Float_t,Float_t,Float_t,
-                           Float_t,Float_t,Float_t,Float_t);
-       Float_t GetWpi(){return fWpi;}
-       Float_t GetWk(){return fWk;}
-       Float_t GetWp(){return fWp;}
-       Int_t   GetPid(){return fPcode;};
+       void    SetVec(Int_t track,const TVector& info) const;
+       TVector* GetVec(Int_t track) const;
+       Int_t   GetPcode(TClonesArray* rps,Float_t pm);
+       Int_t   GetPcode(Float_t p,Float_t pm);
+       Int_t   GetPcode(AliKalmanTrack* track);
+        Int_t   GetPcode(AliITSIOTrack* track); 
+        Int_t   GetPcode(AliITStrackV2* track);
+       void    SetCut(Int_t n,Float_t pm,Float_t pilo,Float_t pihi,
+                      Float_t klo,Float_t khi,Float_t plo,Float_t phi);
+       void    SetAProb(Int_t ivar,Int_t icut,Float_t apro){ fAprob[ivar][icut]=apro; } 
+       Float_t GetAProb(Int_t ivar,Int_t icut) const { return fAprob[ivar][icut]; } 
+       Float_t GetWpi() const {return fWpi;}
+       Float_t GetWk() const {return fWk;}
+       Float_t GetWp() const {return fWp;}
+       Int_t   GetPid() const {return fPcode;};
 protected:
-public:
-       Float_t cut[13][7];
-       Int_t       mxtrs;
-       TClonesArray *trs;
-       Float_t qtot;
-       Float_t fWpi,fWk,fWp;
-       Float_t fRpik,fRppi,fRpka,fRp; 
-       Int_t   fPcode;
-//private:
-public:
-       Float_t qcorr(Float_t);
-       int     qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
-       Float_t qtrm(Int_t track);
-       Float_t qtrm(Float_t qarr[6],Int_t narr);
-       Int_t   wpik(Int_t,Float_t);
-       Int_t   wpikp(Int_t,Float_t);
-       Int_t   pion(){return fWpi=1.,fPcode=211;}
-       Int_t   kaon(){return fWk=1.,fPcode=321;}
-       Int_t   proton(){return fWp=1.,fPcode=2212;}
-public:
-  ClassDef(AliITSPid,1) // Class for ITS PID
+       // copy constructor and assignment operator are protected
+       // since they are not allowed
+        AliITSPid(const AliITSPid &source); // copy constructor. 
+        AliITSPid& operator=(const AliITSPid&  source); // = operator.
+
+       int     Qcomp(Float_t* qa,Float_t* qb) const {return qa[0]>qb[0]?1:0;}
+       Float_t Qtrm(Int_t track);
+       Float_t Qtrm(Float_t qarr[6],Int_t narr) const;
+       Int_t   Wpik(Float_t pm,Float_t q);
+        Int_t  Wpikp(Float_t pm,Float_t q);
+       Int_t   Pion(){return fWpi=1.,fPcode=211;}
+       Int_t   Kaon(){return fWk=1.,fPcode=321;}
+       Int_t   Proton(){return fWp=1.,fPcode=2212;}
+       //================ Data members ========================
+       Float_t fCut[13][7],fAprob[3][8]; //Cuts and prior probs tables
+       Int_t       fMxtrs; //Maximum tracks limit
+       TClonesArray *fTrs; //Tracks set under investigation
+       Float_t fWpi,fWk,fWp; //Probabilities for pions,kaons,protons        
+       Float_t fRpik,fRppi,fRpka,fRp; //Signal ratios
+       Int_t   fPcode;  //Particle code
+       Float_t fSigmin; // Tuning parameter
+        Int_t   fSilent; // Output suppresion flag
+       TF1*    fCutKa;  // Pions-kaons cut function
+       TF1*    fCutPr;  // Kaons-protons cut function
+       TF1*    fGGpi[6];// Pions signal parametrization for Hijing
+       TF1*    fGGka[3];// Kaons          --//--
+        TF1*    fGGpr[3];// Protons        --//--
+       TF1*    fggpi;   // Pions signal for given momentum
+       TF1*    fggka;   // Kaons          --//--
+       TF1*    fggpr;   // Protons        --//--
+  ClassDef(AliITSPid,2) // Class for ITS PID
 };
 
 #endif