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); };
72 void SetTRDTotalChargeInSlice0() { fTRDTotalChargeInSlice0 = kTRUE; }
74 void SetPIDResponse(AliPIDResponse* const pid) { fESDpid = pid; }
75 Float_t TOFbeta(const AliESDtrack* const track) const;
78 void SetNNref(TFile *f) { fNNref = f; };
80 AliHFEtrdPIDqa *GetTRDQA() const { return fTRDpidQA; }
82 Int_t GetCentrality(AliVEvent* const fInputEvent);
83 Int_t GetMultiplicityITS(AliVEvent* const fInputEvent) const;
84 void SetPbPb() { fIsPbPb = kTRUE; };
85 void SetPP() { fIsPbPb = kFALSE; };
86 void SetPPMultiBin() { fIsppMultiBin = kFALSE; };
87 Bool_t IsPbPb() const { return fIsPbPb; };
88 Bool_t IsPP() const { return !fIsPbPb; };
89 Bool_t IsPPMultiBin() const { return fIsppMultiBin; };
94 kRecalculateTRDpid = BIT(15)
96 enum{ // detectors for histogram names
103 TObjArray *MakeTrackList(const TObjArray *tracks) const;
104 TObjArray *MakeCleanListElectrons(const TObjArray *tracks) const;
106 void MakePurity(const TObjArray *tracks, Int_t species);
107 TObjArray *MakeCleanListForTRD(const TObjArray * const track, Int_t species);
108 void FillElectronLikelihoods(const TObjArray * const particles, Int_t species);
109 void FillPIDresponse(const TObjArray * const particles, Int_t species);
110 void FillIllumination(const TObjArray *const particles, Int_t species);
111 void FillTPCinfo(AliESDtrack * const track, Int_t species);
112 void TestTRDResponse(const TObjArray * const tracks, Int_t species);
113 void RecalculateTRDpid(AliESDtrack *track, Double_t *pidProbs) const;
114 void RecalculateTRDpid(AliAODTrack *track, Double_t *pidProbs) const;
115 void CheckTenderV0pid(const TObjArray * const particles, Int_t species);
116 Int_t GetTenderV0pid(AliESDtrack * const track);
118 Double_t TRDlikeTracklet(Int_t layer, AliESDtrack * const track, Double_t *likelihood);
119 Int_t TRDmomBin(Double_t p) const;
125 Int_t GetMaxPID(const Double_t *pidProbs) const;
126 Int_t GetPDG(Int_t index);
129 AliVEvent *fEvent; // event pointer
130 AliMCEvent *fMC; // MC Event
131 AliHFEV0pid *fV0pid; // V0 PID
132 AliHFEV0pidMC *fV0pidMC; // V0 MC PID
133 AliHFEtrdPIDqa *fTRDpidQA; //! TRD PID QA object
134 AliHFEcollection *fOutput; // Output container
135 AliPIDResponse *fESDpid; // ESD PID object
137 TFile *fNNref; // reference file for NN pid
138 TMultiLayerPerceptron *fNet[11]; // reference networks
139 Bool_t fTRDTotalChargeInSlice0; // Fix for Foreward/Backward compatibility
141 Bool_t fIsPbPb; // Analysis Type: pp or PbPb
142 Bool_t fIsppMultiBin; // pp multiplicity bin analysis
144 ClassDef(AliHFEpidQA, 1) // PID QA tool