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