//_________________________________________________________________________
// Class for PID selection with calorimeters
-// The Output of the 2 main methods GetPdg is a PDG number identifying the cluster,
+// The Output of the main method GetIdentifiedParticleType is a PDG number identifying the cluster,
// being kPhoton, kElectron, kPi0 ... as defined in the header file
-// - GetPdg(const TString calo, const Double_t * pid, const Float_t energy)
-// Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle
-// - GetPdg(const TString calo,const TLorentzVector mom, const AliVCluster * cluster)
-// Recalcultes PID, the bayesian or any new one to be implemented in the future
-// Right now only the possibility to recalculate EMCAL with bayesian and simple PID.
+// - GetIdentifiedParticleType(const TString calo, const TLorentzVector mom, const AliVCluster * cluster)
+// Assignes a PID tag to the cluster, right now there is the possibility to : use bayesian weights from reco,
+// recalculate them (EMCAL) or use other procedures not used in reco.
// In order to recalculate Bayesian, it is necessary to load the EMCALUtils library
// and do SwitchOnBayesianRecalculation().
// To change the PID parameters from Low to High like the ones by default, use the constructor
// where flux is AliCaloPID::kLow or AliCaloPID::kHigh
// If it is necessary to change the parameters use the constructor
// AliCaloPID(AliEMCALPIDUtils *utils) and set the parameters before.
-// - SetPIDBits: Simple PID, depending on the thresholds fDispCut fTOFCut and even the
+
+// - GetGetIdentifiedParticleTypeFromBayesian(const TString calo, const Double_t * pid, const Float_t energy)
+// Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle,
+// executed when bayesian is ON by GetIdentifiedParticleType(const TString calo, const TLorentzVector mom, const AliVCluster * cluster)
+// - SetPIDBits: Simple PID, depending on the thresholds fLOCut fTOFCut and even the
// result of the PID bayesian a different PID bit is set.
//
-// All these methods can be called in the analysis you are interested.
//
//*-- Author: Gustavo Conesa (INFN-LNF)
#include <TObject.h>
class TString ;
class TLorentzVector ;
-//class TFormula ;
-class TTask;
+#include <TFormula.h>
+class TList;
class TH2F ;
//--- AliRoot system ---
class AliAODPWG4Particle;
class AliEMCALPIDUtils;
class AliCalorimeterUtils;
+class AliVEvent;
class AliCaloPID : public TObject {
AliCaloPID() ; // ctor
AliCaloPID(const Int_t particleFlux) ; // ctor, to be used when recalculating bayesian PID
- AliCaloPID(const TTask * emcalpid) ; // ctor, to be used when recalculating bayesian PID and need different parameters
+ AliCaloPID(const TNamed * emcalpid) ; // ctor, to be used when recalculating bayesian PID and need different parameters
virtual ~AliCaloPID() ;//virtual dtor
-
-private:
- AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
- AliCaloPID(const AliCaloPID & g) ; // cpy ctor
-
-public:
-
+
enum PidType {
- kPhoton = 22,
- kPi0 = 111,
- kEta = 221,
- kElectron = 11,
- kEleCon = -11,
- kNeutralHadron = 2112,
- kChargedHadron = 211,
+ kPhoton = 22,
+ kPi0 = 111,
+ kEta = 221,
+ kElectron = 11,
+ kEleCon =-11,
+ kNeutralHadron = 2112,
+ kChargedHadron = 211,
kNeutralUnknown = 130,
- kChargedUnknown=321
+ kChargedUnknown = 321
};
enum TagType {kPi0Decay, kEtaDecay, kOtherDecay, kConversion, kNoTag = -1};
- TList * GetCreateOutputObjects();
-
- void InitParameters();
+ // Main methods
+
+ TList * GetCreateOutputObjects();
- Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
+ void InitParameters();
+
+ Int_t GetIdentifiedParticleTypeFromBayesWeights(const TString calo, const Double_t * pid, const Float_t energy) ;
+
+ Int_t GetIdentifiedParticleType(const TString calo, const TLorentzVector mom, const AliVCluster * cluster) ;
+
+ TString GetPIDParametersList();
- Int_t GetPdg(const TString calo,const TLorentzVector mom, const AliVCluster * cluster) const ;
+ Bool_t IsTrackMatched(AliVCluster * cluster, AliCalorimeterUtils* cu, AliVEvent* event) const ;
- TString GetPIDParametersList();
+ void SetPIDBits(const TString calo, AliVCluster * cluster, AliAODPWG4Particle *aodph,
+ AliCalorimeterUtils* cu, AliVEvent* event);
- void SetPIDBits(const TString calo, const AliVCluster * cluster, AliAODPWG4Particle *aodph, const AliCalorimeterUtils* cu);
+ void Print(const Option_t * opt)const;
- void Print(const Option_t * opt)const;
+ //Check if cluster photon-like. Uses photon cluster parameterization in real pp data
+ //Returns distance in sigmas. Recommended cut 2.5
+ Float_t TestPHOSDispersion(const Double_t pt, const Double_t m20, const Double_t m02) const ;
+ //Checks distance to the closest track. Takes into account
+ //non-perpendicular incidence of tracks.
+ Float_t TestPHOSChargedVeto(const Double_t dx, const Double_t dz, const Double_t ptTrack,
+ const Int_t chargeTrack, const Double_t mf) const ;
+
+ // Setters, getters
+
+ void SetDebug(Int_t deb) { fDebug = deb ; }
+ Int_t GetDebug() const { return fDebug ; }
+
+ enum eventType{kLow,kHigh};
+ void SetLowParticleFlux() { fParticleFlux = kLow ; }
+ void SetHighParticleFlux() { fParticleFlux = kHigh ; }
+ // not really used, only for bayesian recalculation in EMCAL, but could be useful in future
+
+ // Bayesian
+
+ void SwitchOnBayesian() { fUseBayesianWeights = kTRUE ; }
+ void SwitchOffBayesian() { fUseBayesianWeights = kFALSE; }
+ void SwitchOnBayesianRecalculation() { fRecalculateBayesian = kTRUE ; fUseBayesianWeights = kTRUE ;} // EMCAL
+ void SwitchOffBayesianRecalculation() { fRecalculateBayesian = kFALSE; } // EMCAL
+
+ AliEMCALPIDUtils * GetEMCALPIDUtils() ;
//Weight getters
- Float_t GetEMCALPhotonWeight() const { return fEMCALPhotonWeight ; }
- Float_t GetEMCALPi0Weight() const { return fEMCALPi0Weight ; }
- Float_t GetEMCALElectronWeight() const { return fEMCALElectronWeight ; }
- Float_t GetEMCALChargeWeight() const { return fEMCALChargeWeight ; }
- Float_t GetEMCALNeutralWeight() const { return fEMCALNeutralWeight ; }
- Float_t GetPHOSPhotonWeight() const { return fPHOSPhotonWeight ; }
- Float_t GetPHOSPi0Weight() const { return fPHOSPi0Weight ; }
- Float_t GetPHOSElectronWeight() const { return fPHOSElectronWeight ; }
- Float_t GetPHOSChargeWeight() const { return fPHOSChargeWeight ; }
- Float_t GetPHOSNeutralWeight() const { return fPHOSNeutralWeight ; }
-
-// Bool_t IsPHOSPIDWeightFormulaOn() const { return fPHOSWeightFormula ; }
-// TFormula * GetPHOSPhotonWeightFormula() const { return fPHOSPhotonWeightFormula ; }
-// TFormula * GetPHOSPi0WeightFormula() const { return fPHOSPi0WeightFormula ; }
-//
+ Float_t GetEMCALPhotonWeight() const { return fEMCALPhotonWeight ; }
+ Float_t GetEMCALPi0Weight() const { return fEMCALPi0Weight ; }
+ Float_t GetEMCALElectronWeight() const { return fEMCALElectronWeight ; }
+ Float_t GetEMCALChargeWeight() const { return fEMCALChargeWeight ; }
+ Float_t GetEMCALNeutralWeight() const { return fEMCALNeutralWeight ; }
+ Float_t GetPHOSPhotonWeight() const { return fPHOSPhotonWeight ; }
+ Float_t GetPHOSPi0Weight() const { return fPHOSPi0Weight ; }
+ Float_t GetPHOSElectronWeight() const { return fPHOSElectronWeight ; }
+ Float_t GetPHOSChargeWeight() const { return fPHOSChargeWeight ; }
+ Float_t GetPHOSNeutralWeight() const { return fPHOSNeutralWeight ; }
+
+ Bool_t IsPHOSPIDWeightFormulaOn() const { return fPHOSWeightFormula ; }
+
+ TFormula * GetPHOSPhotonWeightFormula() {
+ if(!fPHOSPhotonWeightFormula)
+ fPHOSPhotonWeightFormula = new TFormula("phos_photon_weight",
+ fPHOSPhotonWeightFormulaExpression);
+ return fPHOSPhotonWeightFormula ; }
+
+ TFormula * GetPHOSPi0WeightFormula() {
+ if(!fPHOSPi0WeightFormula)
+ fPHOSPi0WeightFormula = new TFormula("phos_pi0_weight",
+ fPHOSPi0WeightFormulaExpression);
+ return fPHOSPi0WeightFormula ; }
+
+ TString GetPHOSPhotonWeightFormulaExpression() const { return fPHOSPhotonWeightFormulaExpression ; }
+ TString GetPHOSPi0WeightFormulaExpression() const { return fPHOSPi0WeightFormulaExpression ; }
+
//Weight setters
- void SetEMCALPhotonWeight(Float_t w){ fEMCALPhotonWeight = w ; }
- void SetEMCALPi0Weight(Float_t w){ fEMCALPi0Weight = w ; }
- void SetEMCALElectronWeight(Float_t w){ fEMCALElectronWeight = w ; }
- void SetEMCALChargeWeight(Float_t w){ fEMCALChargeWeight = w ; }
- void SetEMCALNeutralWeight(Float_t w){ fEMCALNeutralWeight = w ; }
- void SetPHOSPhotonWeight(Float_t w){ fPHOSPhotonWeight = w ; }
- void SetPHOSPi0Weight(Float_t w){ fPHOSPi0Weight = w ; }
- void SetPHOSElectronWeight(Float_t w){ fPHOSElectronWeight = w ; }
- void SetPHOSChargeWeight(Float_t w){ fPHOSChargeWeight = w ; }
- void SetPHOSNeutralWeight(Float_t w){ fPHOSNeutralWeight = w ; }
+ void SetEMCALPhotonWeight (Float_t w) { fEMCALPhotonWeight = w ; }
+ void SetEMCALPi0Weight (Float_t w) { fEMCALPi0Weight = w ; }
+ void SetEMCALElectronWeight(Float_t w) { fEMCALElectronWeight = w ; }
+ void SetEMCALChargeWeight (Float_t w) { fEMCALChargeWeight = w ; }
+ void SetEMCALNeutralWeight (Float_t w) { fEMCALNeutralWeight = w ; }
+ void SetPHOSPhotonWeight (Float_t w) { fPHOSPhotonWeight = w ; }
+ void SetPHOSPi0Weight (Float_t w) { fPHOSPi0Weight = w ; }
+ void SetPHOSElectronWeight (Float_t w) { fPHOSElectronWeight = w ; }
+ void SetPHOSChargeWeight (Float_t w) { fPHOSChargeWeight = w ; }
+ void SetPHOSNeutralWeight (Float_t w) { fPHOSNeutralWeight = w ; }
-// void UsePHOSPIDWeightFormula(Bool_t par) { fPHOSWeightFormula = par; }
-// void SetPHOSPhotonWeightFormula(TFormula * const photon) { fPHOSPhotonWeightFormula = photon; }
-// void SetPHOSPi0WeightFormula(TFormula * const pi0) { fPHOSPi0WeightFormula = pi0; }
+ void UsePHOSPIDWeightFormula (Bool_t ok ) { fPHOSWeightFormula = ok ; }
+ void SetPHOSPhotonWeightFormulaExpression(TString ph) { fPHOSPhotonWeightFormulaExpression = ph ; }
+ void SetPHOSPi0WeightFormulaExpression (TString pi) { fPHOSPi0WeightFormulaExpression = pi ; }
- //PID bits setters and getters
+ //PID cuts
- Bool_t IsTrackMatched(const AliVCluster * cluster, const AliCalorimeterUtils* cu) const ;
+ void SetEMCALLambda0CutMax(Float_t lcut ) { fEMCALL0CutMax = lcut ; }
+ Float_t GetEMCALLambda0CutMax() const { return fEMCALL0CutMax ; }
- void SetDispersionCut(Float_t dcut ) {fDispCut = dcut; }
- Float_t GetDispersionCut() const {return fDispCut ;}
+ void SetEMCALLambda0CutMin(Float_t lcut ) { fEMCALL0CutMin = lcut ; }
+ Float_t GetEMCALLambda0CutMin() const { return fEMCALL0CutMin ; }
- void SetTOFCut(Float_t tcut ) {fTOFCut = tcut; }
- Float_t GetTOFCut() const {return fTOFCut ;}
+ void SetEMCALDEtaCut(Float_t dcut ) { fEMCALDEtaCut = dcut ; }
+ Float_t GetEMCALDEtaCut() const { return fEMCALDEtaCut ; }
- void SetDebug(Int_t deb) {fDebug=deb;}
- Int_t GetDebug() const {return fDebug;}
+ void SetEMCALDPhiCut(Float_t dcut ) { fEMCALDPhiCut = dcut ; }
+ Float_t GetEMCALDPhiCut() const { return fEMCALDPhiCut ; }
- void SwitchOnBayesianRecalculation() {fRecalculateBayesian = kTRUE ; }
- void SwitchOffBayesianRecalculation() {fRecalculateBayesian = kFALSE; }
- enum eventType{kLow,kHigh};
- void SetLowParticleFlux() {fParticleFlux = kLow;}
- void SetHighParticleFlux() {fParticleFlux = kHigh;}
-
+ void SetTOFCut(Float_t tcut ) { fTOFCut = tcut ; }
+ Float_t GetTOFCut() const { return fTOFCut ; }
+ void SetPHOSRCut(Float_t rcut ) { fPHOSRCut = rcut ; }
+ Float_t GetPHOSRCut() const { return fPHOSRCut ; }
+
+ void SetPHOSDispersionCut(Float_t dcut ) { fPHOSDispersionCut = dcut ; }
+ Float_t GetPHOSDispersionCut() const { return fPHOSDispersionCut ; }
- //Histogrammes setters and getters
+ // Track matching histogrammes setters and getters
virtual void SetHistoERangeAndNBins(Float_t min, Float_t max, Int_t n) {
- fHistoNEBins = n ;
- fHistoEMax = max ;
- fHistoEMin = min ;
- }
+ fHistoNEBins = n ; fHistoEMax = max ; fHistoEMin = min ; }
virtual void SetHistoDEtaRangeAndNBins(Float_t min, Float_t max, Int_t n) {
- fHistoNDEtaBins = n ;
- fHistoDEtaMax = max ;
- fHistoDEtaMin = min ;
- }
+ fHistoNDEtaBins = n ; fHistoDEtaMax = max ; fHistoDEtaMin = min ; }
-
virtual void SetHistoDPhiRangeAndNBins(Float_t min, Float_t max, Int_t n) {
- fHistoNDPhiBins = n ;
- fHistoDPhiMax = max ;
- fHistoDPhiMin = min ;
- }
-
-
+ fHistoNDPhiBins = n ; fHistoDPhiMax = max ; fHistoDPhiMin = min ; }
+
private:
- Float_t fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL
- Float_t fEMCALPi0Weight; //Bayesian PID weight for pi0 in EMCAL
- Float_t fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL
- Float_t fEMCALChargeWeight; //Bayesian PID weight for charged hadrons in EMCAL
- Float_t fEMCALNeutralWeight; //Bayesian PID weight for neutral hadrons in EMCAL
- Float_t fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS
- Float_t fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS
- Float_t fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS
- Float_t fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS
- Float_t fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS
-
-// Bool_t fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
-// TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
-// TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight
-
- Float_t fDispCut; //Cut on dispersion, used in PID evaluation
- Float_t fTOFCut; //Cut on TOF, used in PID evaluation
-
- Int_t fDebug; //Debug level
-
- Bool_t fRecalculateBayesian; // Recalculate PID bayesian or use simple PID?
- Int_t fParticleFlux; // Particle flux for setting PID parameters
- AliEMCALPIDUtils * fEMCALPIDUtils; // Pointer to EMCALPID to redo the PID Bayesian calculation
-
- //Histograms
- Int_t fHistoNEBins ; // Number of bins in cluster E axis
- Float_t fHistoEMax ; // Maximum value of cluster E histogram range
- Float_t fHistoEMin ; // Minimum value of cluster E histogram range
- Int_t fHistoNDEtaBins ; // Number of bins in dEta (cluster-track) axis
- Float_t fHistoDEtaMax ; // Maximum value of dEta (cluster-track) histogram range
- Float_t fHistoDEtaMin ; // Minimum value of dEta (cluster-track) histogram range
- Int_t fHistoNDPhiBins ; // Number of bins in dPhi axis
- Float_t fHistoDPhiMax ; // Maximum value of dPhi (cluster-track) histogram range
- Float_t fHistoDPhiMin ; // Minimum value of dPhi (cluster-track) histogram range
-
- TH2F * fhTrackMatchedDEta ; //! Eta distance between track and cluster vs cluster E
- TH2F * fhTrackMatchedDPhi ; //! Phi distance between track and cluster vs cluster E
- TH2F * fhTrackMatchedDEtaDPhi ; //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
-
-
- ClassDef(AliCaloPID,5)
+ Int_t fDebug; // Debug level
+ Int_t fParticleFlux; // Particle flux for setting PID parameters
+
+ // Bayesian
+ AliEMCALPIDUtils * fEMCALPIDUtils; // Pointer to EMCALPID to redo the PID Bayesian calculation
+ Bool_t fUseBayesianWeights; // Select clusters based on weights calculated in reconstruction
+ Bool_t fRecalculateBayesian; // Recalculate PID bayesian or use simple PID?
+
+ Float_t fEMCALPhotonWeight; // Bayesian PID weight for photons in EMCAL
+ Float_t fEMCALPi0Weight; // Bayesian PID weight for pi0 in EMCAL
+ Float_t fEMCALElectronWeight; // Bayesian PID weight for electrons in EMCAL
+ Float_t fEMCALChargeWeight; // Bayesian PID weight for charged hadrons in EMCAL
+ Float_t fEMCALNeutralWeight; // Bayesian PID weight for neutral hadrons in EMCAL
+ Float_t fPHOSPhotonWeight; // Bayesian PID weight for photons in PHOS
+ Float_t fPHOSPi0Weight; // Bayesian PID weight for pi0 in PHOS
+ Float_t fPHOSElectronWeight; // Bayesian PID weight for electrons in PHOS
+ Float_t fPHOSChargeWeight; // Bayesian PID weight for charged hadrons in PHOS
+ Float_t fPHOSNeutralWeight; // Bayesian PID weight for neutral hadrons in PHOS
+
+ Bool_t fPHOSWeightFormula ; // Use parametrized weight threshold, function of energy
+ TFormula *fPHOSPhotonWeightFormula ; // Formula for photon weight
+ TFormula *fPHOSPi0WeightFormula ; // Formula for pi0 weight
+ TString fPHOSPhotonWeightFormulaExpression; // Photon weight formula in string
+ TString fPHOSPi0WeightFormulaExpression; // Pi0 weight formula in string
+
+ // PID calculation
+ Float_t fEMCALL0CutMax; // Max Cut on shower shape lambda0, used in PID evaluation, only EMCAL
+ Float_t fEMCALL0CutMin; // Min Cut on shower shape lambda0, used in PID evaluation, only EMCAL
+ Float_t fEMCALDEtaCut; // Track matching cut on Dz
+ Float_t fEMCALDPhiCut; // Track matching cut on Dx
+
+ Float_t fTOFCut; // Cut on TOF, used in PID evaluation
+
+ Float_t fPHOSDispersionCut; // Shower shape elipse radious cut
+ Float_t fPHOSRCut; // Track-Cluster distance cut for track matching in PHOS
+
+ // Track matching control histograms
+ Int_t fHistoNEBins ; // Number of bins in cluster E axis
+ Float_t fHistoEMax ; // Maximum value of cluster E histogram range
+ Float_t fHistoEMin ; // Minimum value of cluster E histogram range
+ Int_t fHistoNDEtaBins ; // Number of bins in dEta (cluster-track) axis
+ Float_t fHistoDEtaMax ; // Maximum value of dEta (cluster-track) histogram range
+ Float_t fHistoDEtaMin ; // Minimum value of dEta (cluster-track) histogram range
+ Int_t fHistoNDPhiBins ; // Number of bins in dPhi axis
+ Float_t fHistoDPhiMax ; // Maximum value of dPhi (cluster-track) histogram range
+ Float_t fHistoDPhiMin ; // Minimum value of dPhi (cluster-track) histogram range
+
+ TH2F *fhTrackMatchedDEta ; //! Eta distance between track and cluster vs cluster E
+ TH2F *fhTrackMatchedDPhi ; //! Phi distance between track and cluster vs cluster E
+ TH2F *fhTrackMatchedDEtaDPhi ; //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
+
+ AliCaloPID & operator = (const AliCaloPID & g) ; // cpy assignment
+ AliCaloPID(const AliCaloPID & g) ; // cpy ctor
+
+ ClassDef(AliCaloPID,10)
} ;
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff