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 **************************************************************************/
16 // Steering class for electron identification
17 // Combines detector PID objects
18 // For more information please check the implementation file
27 #ifndef ALIHFEPIDBASE_H
28 #include "AliHFEpidBase.h"
32 //#include "AliPIDResponse.h"
34 class AliHFEcontainer;
35 class AliHFEvarManager;
43 class AliHFEpid : public TNamed{
59 AliHFEpid(const Char_t *name);
60 AliHFEpid(const AliHFEpid &c);
61 AliHFEpid &operator=(const AliHFEpid &c);
62 void Copy(TObject &o) const;
65 Bool_t InitializePID(Int_t run = 0);
66 Bool_t IsSelected(const AliHFEpidObject * const track, AliHFEcontainer *cont = NULL, const Char_t *contname = "trackContainer", AliHFEpidQAmanager *qa = NULL);
68 Bool_t HasMCData() const { return TestBit(kHasMCData); };
70 void AddDetector(TString detector, UInt_t position);
71 void SetDetectorsForAnalysis(TString detectors);
72 void SetPIDResponse(const AliPIDResponse * const pid);
73 void SetVarManager(AliHFEvarManager *vm) { fVarManager = vm; }
74 void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
76 const AliPIDResponse *GetPIDResponse() const;
77 UInt_t GetNumberOfPIDdetectors() const { return fNPIDdetectors; }
78 Bool_t HasDetector(EDETtype_t det) const { return IsDetectorOn(det); }
79 Bool_t IsInitialized() const { return TestBit(kIsInit); }
81 AliHFEpidBase *GetDetPID(EDETtype_t det) const { return det < kNdetectorPID ? fDetectorPID[det] : NULL; }
83 void PrintStatus() const;
84 const Char_t *SortedDetectorName(Int_t det) {
85 if(!TestBit(kDetectorsSorted)) SortDetectors();
86 if(det < kNdetectorPID) return fgkDetectorName[fSortedOrder[det]];
87 else return fgkDetectorName[kNdetectorPID];
89 //-----Configure PID detectors with predefined stettings------
90 void ConfigureTOF(Float_t TOFcut = 3.);
91 void ConfigureTPCasymmetric(Double_t pmin = 0.1, Double_t pmax = 20., Double_t sigmamin = -0.2, Double_t sigmamax = 5.);
92 void ConfigureTPCrejectionSimple();
93 void ConfigureTPCcentralityCut(Int_t centralityBin, const char *lowerCutParam = NULL, const Double_t * const params = NULL, Float_t upperTPCCut=3.0);
94 void ConfigureTPCdefaultCut(const char *lowerCutParam = NULL, const Double_t * const params = NULL, Float_t upperTPCCut=3.0);
95 void ConfigureBayesDetectorMask(Int_t detmask = 10);
96 void ConfigureBayesPIDThreshold(Float_t pidthres = 0.9);
97 //------------------------------------------------------------
100 Bool_t MakePidTpcTof(AliHFEpidObject *track);
103 kHasMCData = BIT(14),
105 kDetectorsSorted = BIT(16)
115 void AddCommonObject(TObject * const o);
116 void ClearCommonObjects();
117 //-----Switch on/off detectors in PID sequence------
118 void SwitchOnDetector(UInt_t det){
119 if(det < kNdetectorPID) SETBIT(fEnabledDetectors, det);
121 void SwitchOffDetector(UInt_t det){
122 if(det < kNdetectorPID) CLRBIT(fEnabledDetectors, det);
124 Bool_t IsDetectorOn(UInt_t det) const {
125 return det < kNdetectorPID ? TESTBIT(fEnabledDetectors, det): kFALSE;
127 //--------------------------------------------------
128 void ConfigureTPCcut(Int_t centralityBin, const char *lowerCutParam, const Double_t * const params, Float_t upperTPCCut);
130 static const Char_t *fgkDetectorName[kNdetectorPID + 1]; // PID Detector Names
131 AliHFEpidBase *fDetectorPID[kNdetectorPID]; // Detector PID classes
132 UInt_t fDetectorOrder[kNdetectorPID]; // Position requested by the user
133 UInt_t fSortedOrder[kNdetectorPID]; // Sorted array of detectorIDs
134 UInt_t fEnabledDetectors; // Enabled Detectors
135 UInt_t fNPIDdetectors; // Number of PID detectors
136 AliHFEvarManager *fVarManager; //! HFE Var Manager
137 TObjArray *fCommonObjects; // Garbage Collector
139 ClassDef(AliHFEpid, 1) // Steering class for Electron ID