3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
7 ///////////////////////////////////////////////////////////////////////////////
9 // The TRD particle identification base class //
11 // Its main purposes are: //
12 // - Provide I/O framework for all neccessary files //
13 // - Assignment of a e/pi propability to a given track //
15 ///////////////////////////////////////////////////////////////////////////////
25 class AliTRDpid : public TNamed {
30 AliTRDpid(const char* name, const char* title);
31 AliTRDpid(const AliTRDpid &p);
33 AliTRDpid &operator=(const AliTRDpid &p);
35 virtual void Copy(TObject &p);
36 virtual Bool_t Init();
37 virtual Bool_t AssignLikelihood();
38 virtual Bool_t AssignLikelihood(TObjArray *tarray);
39 virtual Bool_t AssignLikelihood(AliTRDtrack *t) = 0;
40 virtual Bool_t FillSpectra();
41 virtual Bool_t FillSpectra(TObjArray *tarray);
42 virtual Bool_t FillSpectra(const AliTRDtrack *t) = 0;
43 virtual Bool_t Open(const Char_t *name, Int_t event = 0);
44 virtual Bool_t Open(const Char_t *namekine
45 , const Char_t *namecluster
46 , const Char_t *nametracks, Int_t event = 0);
47 virtual Int_t MCpid(const AliTRDtrack *t);
48 virtual Int_t MCpid(const AliTRDtrack *t, Int_t *pdg, Int_t *nFound, Int_t *indices);
49 virtual Bool_t ReadCluster(const Char_t *name);
50 virtual Bool_t ReadTracks(const Char_t *name);
51 virtual Bool_t ReadKine(const Char_t *name, Int_t event);
52 virtual Bool_t SumCharge(const AliTRDtrack *t, Float_t *charge, Int_t *nCluster);
54 virtual Int_t GetIndex(const AliTRDtrack *t) = 0;
56 void SetGeometry(AliTRDgeometry *geo) { fGeometry = geo; };
57 void SetTrackArray(TObjArray *tarray) { fTrackArray = tarray; };
58 void SetClusterArray(TObjArray *carray) { fClusterArray = carray; };
60 void SetPIDratioMin(Float_t min) { fPIDratioMin = min; };
61 void SetPIDpurePoints(Bool_t pure) { fPIDpurePoints = pure; };
62 void SetPIDindexMin(Int_t min) { fPIDindexMin = min; };
63 void SetPIDindexMax(Int_t max) { fPIDindexMax = max; };
65 void SetThreePadOnly(Bool_t only) { fThreePadOnly = only; };
66 void SetEvent(Int_t event) { fEvent = event; };
68 TObjArray *GetTrackArray() const { return fTrackArray; };
69 TObjArray *GetClusterArray() const { return fClusterArray; };
71 Float_t GetPIDratioMin() const { return fPIDratioMin; };
72 Bool_t GetPIDpurePoints() const { return fPIDpurePoints; };
73 Float_t GetPIDindexMin() const { return fPIDindexMin; };
74 Float_t GetPIDindexMax() const { return fPIDindexMax; };
76 Bool_t GetThreePadOnly() const { return fThreePadOnly; };
81 kNpid = 2, // Number of pid types (pion + electron)
82 kElectron = 0, // Electron pid
86 Float_t fPIDratioMin; // Minimum fraction of cluster from one particle
87 Bool_t fPIDpurePoints; // Require pure (nono overlapping) cluster
88 Int_t fPIDindexMin; // Lower index MC particles to be considered
89 Int_t fPIDindexMax; // Upper index MC particles to be considered
91 Bool_t fThreePadOnly; // Use only three pad cluster in the charge sum
93 Int_t fEvent; // Event number
95 TObjArray *fTrackArray; //! Array containing the tracks
96 TObjArray *fClusterArray; //! Array containing the cluster
97 AliTRDgeometry *fGeometry; //! The TRD geometry
98 TFile *fFileKine; //! The kine input file
100 ClassDef(AliTRDpid,1) // Assigns the e/pi propability to the tracks