1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
17 // Several studies done on clean samples of electrons, pions and kaons
19 // More information can be found in the implementation file
33 class TMultiLayerPerceptron;
40 class AliExternalTrackParam;
44 class AliHFEcollection;
50 class AliHFEpidQA : public TObject{
54 AliHFEpidQA(const AliHFEpidQA &ref);
55 AliHFEpidQA &operator=(const AliHFEpidQA &ref);
56 virtual void Copy(TObject &o) const;
65 Bool_t HasV0pidQA() const { return TestBit(kV0pidQA); };
66 Bool_t HasRecalculateTRDpid() const { return TestBit(kRecalculateTRDpid); };
68 void SetEvent(AliVEvent* const ev) { fEvent = ev; };
69 void SetMCEvent(AliMCEvent * const mc) { fMC = mc; };
70 void SetV0pidQA(Bool_t v0pidQA = kTRUE) { SetBit(kV0pidQA, v0pidQA); };
71 void SetRecalculateTRDpid(Bool_t recal = kTRUE) { SetBit(kRecalculateTRDpid, recal); };
73 void SetESDpid(AliESDpid* const pid) { fESDpid = pid; }
74 Float_t TOFbeta(AliESDtrack* const track) const;
77 void SetNNref(TFile *f) { fNNref = f; };
79 AliHFEtrdPIDqa *GetTRDQA() const { return fTRDpidQA; }
84 kRecalculateTRDpid = BIT(15)
86 enum{ // detectors for histogram names
93 TObjArray *MakeTrackList(TObjArray *tracks) const;
94 TObjArray *MakeCleanListElectrons(TObjArray *tracks) const;
96 void MakePurity(TObjArray *tracks, Int_t species);
97 TObjArray *MakeCleanListForTRD(TObjArray * const track, Int_t species);
98 void FillElectronLikelihoods(TObjArray * const particles, Int_t species);
99 void FillPIDresponse(TObjArray * const particles, Int_t species);
100 void FillIllumination(TObjArray *const particles, Int_t species);
101 void FillTPCinfo(AliESDtrack * const track, Int_t species);
102 void RecalculateTRDpid(AliESDtrack *track, Double_t *pidProbs) const;
103 void RecalculateTRDpid(AliAODTrack *track, Double_t *pidProbs) const;
104 void CheckTenderV0pid(TObjArray * const particles, Int_t species);
105 Int_t GetTenderV0pid(AliESDtrack * const track);
107 Double_t TRDlikeTracklet(Int_t layer, AliESDtrack * const track, Double_t *likelihood);
108 Int_t TRDmomBin(Double_t p);
111 Int_t GetMaxPID(Double_t *pidProbs) const;
112 Int_t GetPDG(Int_t index);
115 AliVEvent *fEvent; // event pointer
116 AliMCEvent *fMC; // MC Event
117 AliHFEV0pid *fV0pid; // V0 PID
118 AliHFEV0pidMC *fV0pidMC; // V0 MC PID
119 AliHFEtrdPIDqa *fTRDpidQA; //! TRD PID QA object
120 AliHFEcollection *fOutput; // Output container
121 AliESDpid *fESDpid; // ESD PID object
123 TFile *fNNref; // reference file for NN pid
124 TMultiLayerPerceptron *fNet[11]; // reference networks
126 ClassDef(AliHFEpidQA, 1) // PID QA tool