X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=EMCAL%2FAliEMCALRecoUtils.h;h=03945c28fad44c120f15576d65a2a877798369f7;hb=0f7547f17c20abb826b907a88a51513512283ad1;hp=0465ba7efafa2589f1aa11e3d21de5345e48cbec;hpb=bd8c7aeff7a58a644927b3d8ac5916c7ff8a733e;p=u%2Fmrichter%2FAliRoot.git diff --git a/EMCAL/AliEMCALRecoUtils.h b/EMCAL/AliEMCALRecoUtils.h index 0465ba7efaf..03945c28fad 100644 --- a/EMCAL/AliEMCALRecoUtils.h +++ b/EMCAL/AliEMCALRecoUtils.h @@ -10,24 +10,29 @@ // // // Author: Gustavo Conesa (LPSC- Grenoble) +// Track matching part: Rongrong Ma (Yale) /////////////////////////////////////////////////////////////////////////////// //Root includes -#include "TNamed.h" -#include "TMath.h" -#include "TObjArray.h" -#include "TArrayI.h" -#include "TArrayF.h" -#include "TH2F.h" +#include +#include +class TObjArray; +class TArrayI; +class TArrayF; +#include +class TH2F; +#include //AliRoot includes class AliVCluster; class AliVCaloCells; class AliVEvent; -#include "AliLog.h" + +// EMCAL includes class AliEMCALGeometry; class AliEMCALPIDUtils; class AliESDtrack; +class AliExternalTrackParam; class AliEMCALRecoUtils : public TNamed { @@ -36,213 +41,329 @@ public: AliEMCALRecoUtils(); AliEMCALRecoUtils(const AliEMCALRecoUtils&); AliEMCALRecoUtils& operator=(const AliEMCALRecoUtils&); - virtual ~AliEMCALRecoUtils() ; - - enum NonlinearityFunctions{kPi0MC=0,kPi0GammaGamma=1,kPi0GammaConversion=2,kNoCorrection=3}; - enum PositionAlgorithms{kUnchanged=-1,kPosTowerIndex=0, kPosTowerGlobal=1}; - enum ParticleType{kPhoton=0, kElectron=1,kHadron =2, kUnknown=-1}; - + virtual ~AliEMCALRecoUtils() ; + void Print(const Option_t*) const; + + //enums + enum NonlinearityFunctions{kPi0MC=0,kPi0GammaGamma=1,kPi0GammaConversion=2,kNoCorrection=3,kBeamTest=4,kBeamTestCorrected=5}; + enum PositionAlgorithms{kUnchanged=-1,kPosTowerIndex=0, kPosTowerGlobal=1}; + enum ParticleType{kPhoton=0, kElectron=1,kHadron =2, kUnknown=-1}; + enum { kNCuts = 11 }; //track matching + enum TrackCutsType{kTPCOnlyCut=0, kGlobalCut=1}; + + //----------------------------------------------------- //Position recalculation - void RecalculateClusterPosition(AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu); + //----------------------------------------------------- + + void RecalculateClusterPosition (AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu); void RecalculateClusterPositionFromTowerIndex (AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu); void RecalculateClusterPositionFromTowerGlobal(AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu); - Float_t GetCellWeight(const Float_t eCell, const Float_t eCluster) const { return TMath::Max( 0., fW0 + TMath::Log( eCell / eCluster ));} + Float_t GetCellWeight(const Float_t eCell, const Float_t eCluster) const { return TMath::Max( 0., fW0 + TMath::Log( eCell / eCluster )) ; } Float_t GetDepth(const Float_t eCluster, const Int_t iParticle, const Int_t iSM) const ; void GetMaxEnergyCell(AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu, - Int_t & absId, Int_t& iSupMod, Int_t& ieta, Int_t& iphi); + Int_t & absId, Int_t& iSupMod, Int_t& ieta, Int_t& iphi, Bool_t &shared); - Float_t GetMisalTransShift(const Int_t i) const { - if(i < 15 ){return fMisalTransShift[i]; } - else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)); return 0.;} - } - Float_t* GetMisalTransShiftArray() {return fMisalTransShift; } + Float_t GetMisalTransShift(const Int_t i) const { if(i < 15 ) { return fMisalTransShift[i] ; } + else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)) ; + return 0. ; } } + Float_t* GetMisalTransShiftArray() { return fMisalTransShift ; } void SetMisalTransShift(const Int_t i, const Float_t shift) { - if(i < 15 ){fMisalTransShift[i] = shift; } - else { AliInfo(Form("Index %d larger than 15, do nothing\n",i));} - } - void SetMisalTransShiftArray(Float_t * misal) - { for(Int_t i = 0; i < 15; i++)fMisalTransShift[i] = misal[i]; } - - Float_t GetMisalRotShift(const Int_t i) const { - if(i < 15 ){return fMisalRotShift[i]; } - else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)); return 0.;} - } - Float_t* GetMisalRotShiftArray() {return fMisalRotShift; } + if(i < 15 ) { fMisalTransShift[i] = shift ; } + else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)) ; } } + void SetMisalTransShiftArray(Float_t * misal) { for(Int_t i = 0; i < 15; i++) fMisalTransShift[i] = misal[i] ; } + + Float_t GetMisalRotShift(const Int_t i) const { if(i < 15 ) { return fMisalRotShift[i] ; } + else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)) ; + return 0. ; } } + + Float_t* GetMisalRotShiftArray() { return fMisalRotShift ; } void SetMisalRotShift(const Int_t i, const Float_t shift) { - if(i < 15 ){fMisalRotShift[i] = shift; } - else { AliInfo(Form("Index %d larger than 15, do nothing\n",i));} - } - void SetMisalRotShiftArray(Float_t * misal) - { for(Int_t i = 0; i < 15; i++)fMisalRotShift[i] = misal[i]; } + if(i < 15 ) { fMisalRotShift[i] = shift ; } + else { AliInfo(Form("Index %d larger than 15, do nothing\n",i)) ; } } + + void SetMisalRotShiftArray(Float_t * misal) { for(Int_t i = 0; i < 15; i++)fMisalRotShift[i] = misal[i] ; } - Int_t GetParticleType() const {return fParticleType ;} - void SetParticleType(Int_t particle) {fParticleType = particle ;} + Int_t GetParticleType() const { return fParticleType ; } + void SetParticleType(Int_t particle) { fParticleType = particle ; } - Int_t GetPositionAlgorithm() const {return fPosAlgo;} - void SetPositionAlgorithm(Int_t alg) {fPosAlgo = alg ;} + Int_t GetPositionAlgorithm() const { return fPosAlgo ; } + void SetPositionAlgorithm(Int_t alg) { fPosAlgo = alg ; } - Float_t GetW0() const {return fW0;} - void SetW0(Float_t w0) {fW0 = w0 ;} + Float_t GetW0() const { return fW0 ; } + void SetW0(Float_t w0) { fW0 = w0 ; } + //----------------------------------------------------- //Non Linearity + //----------------------------------------------------- + + Float_t CorrectClusterEnergyLinearity(AliVCluster* clu) ; - Float_t CorrectClusterEnergyLinearity(AliVCluster* clu); - - Float_t GetNonLinearityParam(const Int_t i) const { - if(i < 6 ){return fNonLinearityParams[i]; } - else { AliInfo(Form("Index %d larger than 6, do nothing\n",i)); return 0.;} - } + Float_t GetNonLinearityParam(const Int_t i) const { if(i < 7 ){ return fNonLinearityParams[i] ; } + else { AliInfo(Form("Index %d larger than 7, do nothing\n",i)) ; + return 0. ; } } void SetNonLinearityParam(const Int_t i, const Float_t param) { - if(i < 6 ){fNonLinearityParams[i] = param; } - else { AliInfo(Form("Index %d larger than 6, do nothing\n",i));} - } - - Int_t GetNonLinearityFunction() const {return fNonLinearityFunction;} - void SetNonLinearityFunction(Int_t fun) {fNonLinearityFunction = fun ;} + if(i < 7 ){fNonLinearityParams[i] = param ; } + else { AliInfo(Form("Index %d larger than 7, do nothing\n",i)) ; } } + void InitNonLinearityParam(); + + Int_t GetNonLinearityFunction() const { return fNonLinearityFunction ; } + void SetNonLinearityFunction(Int_t fun) { fNonLinearityFunction = fun ; InitNonLinearityParam() ; } + + void SetNonLinearityThreshold(Int_t threshold) { fNonLinearThreshold = threshold ; } //only for Alexie's non linearity correction + Int_t GetNonLinearityThreshold() const { return fNonLinearThreshold ; } +// + //----------------------------------------------------- + // MC clusters energy smearing + //----------------------------------------------------- - void Print(const Option_t*) const; + Float_t SmearClusterEnergy(AliVCluster* clu) ; + void SwitchOnClusterEnergySmearing() { fSmearClusterEnergy = kTRUE ; } + void SwitchOffClusterEnergySmearing() { fSmearClusterEnergy = kFALSE ; } + Bool_t IsClusterEnergySmeared() const { return fSmearClusterEnergy ; } + void SetSmearingParameters(Int_t i, Float_t param) { if(i < 3){ fSmearClusterParam[i] = param ; } + else { AliInfo(Form("Index %d larger than 2, do nothing\n",i)) ; } } + //----------------------------------------------------- //Recalibration - void RecalibrateClusterEnergy(AliEMCALGeometry* geom, AliVCluster* cluster, AliVCaloCells * cells); + //----------------------------------------------------- - Bool_t IsRecalibrationOn() const { return fRecalibration ; } - void SwitchOnRecalibration() {fRecalibration = kTRUE ; InitEMCALRecalibrationFactors();} - void SwitchOffRecalibration() {fRecalibration = kFALSE ; } - - void InitEMCALRecalibrationFactors() ; - - Float_t GetEMCALChannelRecalibrationFactor(Int_t iSM , Int_t iCol, Int_t iRow) const { - if(fEMCALRecalibrationFactors) return (Float_t) ((TH2F*)fEMCALRecalibrationFactors->At(iSM))->GetBinContent(iCol,iRow); - else return 1;} + void RecalibrateClusterEnergy(AliEMCALGeometry* geom, AliVCluster* cluster, AliVCaloCells * cells) ; + + Bool_t IsRecalibrationOn() const { return fRecalibration ; } + void SwitchOffRecalibration() { fRecalibration = kFALSE ; } + void SwitchOnRecalibration() { fRecalibration = kTRUE ; + if(!fEMCALRecalibrationFactors)InitEMCALRecalibrationFactors() ; } + void InitEMCALRecalibrationFactors() ; + + //Recalibrate channels with time dependent corrections + void SwitchOffTimeDepCorrection() { fUseTimeCorrectionFactors = kFALSE ; } + void SwitchOnTimeDepCorrection() { fUseTimeCorrectionFactors = kTRUE ; + SwitchOnRecalibration() ; } + void SetTimeDependentCorrections(Int_t runnumber); + + Float_t GetEMCALChannelRecalibrationFactor(Int_t iSM , Int_t iCol, Int_t iRow) const { + if(fEMCALRecalibrationFactors) + return (Float_t) ((TH2F*)fEMCALRecalibrationFactors->At(iSM))->GetBinContent(iCol,iRow); + else return 1 ; } - void SetEMCALChannelRecalibrationFactor(Int_t iSM , Int_t iCol, Int_t iRow, Double_t c = 1) { - if(!fEMCALRecalibrationFactors) InitEMCALRecalibrationFactors(); - ((TH2F*)fEMCALRecalibrationFactors->At(iSM))->SetBinContent(iCol,iRow,c);} + void SetEMCALChannelRecalibrationFactor(Int_t iSM , Int_t iCol, Int_t iRow, Double_t c = 1) { + if(!fEMCALRecalibrationFactors) InitEMCALRecalibrationFactors() ; + ((TH2F*)fEMCALRecalibrationFactors->At(iSM))->SetBinContent(iCol,iRow,c) ; } - TH2F * GetEMCALChannelRecalibrationFactors(Int_t iSM) const {return (TH2F*)fEMCALRecalibrationFactors->At(iSM);} - void SetEMCALChannelRecalibrationFactors(TObjArray *map) {fEMCALRecalibrationFactors = map;} - void SetEMCALChannelRecalibrationFactors(Int_t iSM , TH2F* h) {fEMCALRecalibrationFactors->AddAt(h,iSM);} + TH2F * GetEMCALChannelRecalibrationFactors(Int_t iSM) const { return (TH2F*)fEMCALRecalibrationFactors->At(iSM) ; } + void SetEMCALChannelRecalibrationFactors(TObjArray *map) { fEMCALRecalibrationFactors = map ; } + void SetEMCALChannelRecalibrationFactors(Int_t iSM , TH2F* h) { fEMCALRecalibrationFactors->AddAt(h,iSM) ; } + //----------------------------------------------------- //Modules fiducial region, remove clusters in borders - Bool_t CheckCellFiducialRegion(AliEMCALGeometry* geom, AliVCluster* cluster, AliVCaloCells* cells) ; - void SetNumberOfCellsFromEMCALBorder(Int_t n) {fNCellsFromEMCALBorder = n; } - Int_t GetNumberOfCellsFromEMCALBorder() const {return fNCellsFromEMCALBorder; } + //----------------------------------------------------- + + Bool_t CheckCellFiducialRegion(AliEMCALGeometry* geom, AliVCluster* cluster, AliVCaloCells* cells) ; + void SetNumberOfCellsFromEMCALBorder(Int_t n) { fNCellsFromEMCALBorder = n ; } + Int_t GetNumberOfCellsFromEMCALBorder() const { return fNCellsFromEMCALBorder ; } - void SwitchOnNoFiducialBorderInEMCALEta0() {fNoEMCALBorderAtEta0 = kTRUE; } - void SwitchOffNoFiducialBorderInEMCALEta0() {fNoEMCALBorderAtEta0 = kFALSE; } - Bool_t IsEMCALNoBorderAtEta0() {return fNoEMCALBorderAtEta0;} + void SwitchOnNoFiducialBorderInEMCALEta0() { fNoEMCALBorderAtEta0 = kTRUE ; } + void SwitchOffNoFiducialBorderInEMCALEta0() { fNoEMCALBorderAtEta0 = kFALSE ; } + Bool_t IsEMCALNoBorderAtEta0() const { return fNoEMCALBorderAtEta0 ; } + //----------------------------------------------------- // Bad channels - Bool_t IsBadChannelsRemovalSwitchedOn() const { return fRemoveBadChannels ; } - void SwitchOnBadChannelsRemoval () {fRemoveBadChannels = kTRUE ; InitEMCALBadChannelStatusMap();} - void SwitchOffBadChannelsRemoval() {fRemoveBadChannels = kFALSE ; } + //----------------------------------------------------- + + Bool_t IsBadChannelsRemovalSwitchedOn() const { return fRemoveBadChannels ; } + void SwitchOffBadChannelsRemoval() { fRemoveBadChannels = kFALSE ; } + void SwitchOnBadChannelsRemoval () { fRemoveBadChannels = kTRUE ; + if(!fEMCALBadChannelMap)InitEMCALBadChannelStatusMap() ; } - void InitEMCALBadChannelStatusMap() ; + Bool_t IsDistanceToBadChannelRecalculated() const { return fRecalDistToBadChannels ; } + void SwitchOffDistToBadChannelRecalculation() { fRecalDistToBadChannels = kFALSE ; } + void SwitchOnDistToBadChannelRecalculation() { fRecalDistToBadChannels = kTRUE ; + if(!fEMCALBadChannelMap)InitEMCALBadChannelStatusMap() ; } + + void InitEMCALBadChannelStatusMap() ; - Int_t GetEMCALChannelStatus(Int_t iSM , Int_t iCol, Int_t iRow) const { + Int_t GetEMCALChannelStatus(Int_t iSM , Int_t iCol, Int_t iRow) const { if(fEMCALBadChannelMap) return (Int_t) ((TH2I*)fEMCALBadChannelMap->At(iSM))->GetBinContent(iCol,iRow); else return 0;}//Channel is ok by default - void SetEMCALChannelStatus(Int_t iSM , Int_t iCol, Int_t iRow, Double_t c = 1) { - if(!fEMCALBadChannelMap)InitEMCALBadChannelStatusMap() ; - ((TH2I*)fEMCALBadChannelMap->At(iSM))->SetBinContent(iCol,iRow,c);} - - TH2I * GetEMCALChannelStatusMap(Int_t iSM) const {return (TH2I*)fEMCALBadChannelMap->At(iSM);} - void SetEMCALChannelStatusMap(TObjArray *map) {fEMCALBadChannelMap = map;} + void SetEMCALChannelStatus(Int_t iSM , Int_t iCol, Int_t iRow, Double_t c = 1) { + if(!fEMCALBadChannelMap)InitEMCALBadChannelStatusMap() ; + ((TH2I*)fEMCALBadChannelMap->At(iSM))->SetBinContent(iCol,iRow,c) ; } - Bool_t ClusterContainsBadChannel(AliEMCALGeometry* geom, UShort_t* cellList, Int_t nCells); + TH2I * GetEMCALChannelStatusMap(Int_t iSM) const { return (TH2I*)fEMCALBadChannelMap->At(iSM) ; } + void SetEMCALChannelStatusMap(TObjArray *map) { fEMCALBadChannelMap = map ; } + void SetEMCALChannelStatusMap(Int_t iSM , TH2I* h) { fEMCALBadChannelMap->AddAt(h,iSM) ; } + + Bool_t ClusterContainsBadChannel(AliEMCALGeometry* geom, UShort_t* cellList, const Int_t nCells); - //Recalculate other cluster parameters - void RecalculateClusterPID(AliVCluster * cluster); + //----------------------------------------------------- + // Recalculate other cluster parameters + //----------------------------------------------------- + + void RecalculateClusterDistanceToBadChannel (AliEMCALGeometry * geom, AliVCaloCells* cells, AliVCluster * cluster); + void RecalculateClusterShowerShapeParameters(AliEMCALGeometry * geom, AliVCaloCells* cells, AliVCluster * cluster); + void RecalculateClusterPID(AliVCluster * cluster); + AliEMCALPIDUtils * GetPIDUtils() { return fPIDUtils;} - void RecalculateClusterShowerShapeParameters(AliEMCALGeometry * geom, AliVCaloCells* cells, AliVCluster * cluster); - //Track matching - void FindMatches(AliVEvent *event); - void GetMatchedResiduals(Int_t index, Float_t &dR, Float_t &dZ); - Bool_t IsMatched(Int_t index); - Int_t FindMatchedPos(Int_t index) const; + //---------------------------------------------------- + // Track matching + //---------------------------------------------------- + + Bool_t ExtrapolateTrackToCluster(AliExternalTrackParam *trkParam, AliVCluster *cluster, Float_t &tmpEta, Float_t &tmpPhi); + + void FindMatches(AliVEvent *event, TObjArray * clusterArr=0x0, AliEMCALGeometry *geom=0x0); + Int_t FindMatchedCluster(AliESDtrack *track, AliVEvent *event, AliEMCALGeometry *geom); + UInt_t FindMatchedPosForCluster(Int_t clsIndex) const; + UInt_t FindMatchedPosForTrack(Int_t trkIndex) const; + + void GetMatchedResiduals(Int_t clsIndex, Float_t &dEta, Float_t &dPhi); + void GetMatchedClusterResiduals(Int_t trkIndex, Float_t &dEta, Float_t &dPhi); + Int_t GetMatchedTrackIndex(Int_t clsIndex); + Int_t GetMatchedClusterIndex(Int_t trkIndex); + + Bool_t IsClusterMatched(Int_t clsIndex) const; + Bool_t IsTrackMatched(Int_t trkIndex) const; + + + void SwitchOnCutEtaPhiSum() { fCutEtaPhiSum = kTRUE ; + fCutEtaPhiSeparate = kFALSE ; } + void SwitchOnCutEtaPhiSeparate() { fCutEtaPhiSeparate = kTRUE ; + fCutEtaPhiSum = kFALSE ; } + + Float_t GetCutR() const { return fCutR ; } + Float_t GetCutEta() const { return fCutEta ; } + Float_t GetCutPhi() const { return fCutPhi ; } + void SetCutR(Float_t cutR) { fCutR = cutR ; } + void SetCutEta(Float_t cutEta) { fCutEta = cutEta ; } + void SetCutPhi(Float_t cutPhi) { fCutPhi = cutPhi ; } + void SetCutZ(Float_t cutZ) { printf("Obsolete fucntion of cutZ=%1.1f\n",cutZ) ; } //Obsolete + + Double_t GetMass() const { return fMass ; } + Double_t GetStep() const { return fStep ; } + void SetMass(Double_t mass) { fMass = mass ; } + void SetStep(Double_t step) { fStep = step ; } + + //Cluster cut + Bool_t IsGoodCluster(AliVCluster *cluster, AliEMCALGeometry *geom, AliVCaloCells* cells); + Bool_t IsExoticCluster(AliVCluster *cluster) const ; - Float_t GetCutR() const { return fCutR; } - Float_t GetCutZ() const { return fCutZ; } + void SwitchOnRejectExoticCluster() { fRejectExoticCluster=kTRUE ; } + void SwitchOffRejectExoticCluster() { fRejectExoticCluster=kFALSE ; } + Bool_t IsRejectExoticCluster() const { return fRejectExoticCluster ; } - void SetCutR(Float_t CutR) { fCutR=CutR; } - void SetCutZ(Float_t CutZ) { fCutZ=CutZ; } //Track Cuts - Bool_t IsAccepted(AliESDtrack *track); - void InitTrackCuts(); + Bool_t IsAccepted(AliESDtrack *track); + void InitTrackCuts(); + void SetTrackCutsType(Int_t type) { fTrackCutsType = type ; + InitTrackCuts() ; } + Int_t GetTrackCutsType() const { return fTrackCutsType; } // track quality cut setters - void SetMinNClustersTPC(Int_t min=-1) {fCutMinNClusterTPC=min;} - void SetMinNClustersITS(Int_t min=-1) {fCutMinNClusterITS=min;} - void SetMaxChi2PerClusterTPC(Float_t max=1e10) {fCutMaxChi2PerClusterTPC=max;} - void SetMaxChi2PerClusterITS(Float_t max=1e10) {fCutMaxChi2PerClusterITS=max;} - void SetRequireTPCRefit(Bool_t b=kFALSE) {fCutRequireTPCRefit=b;} - void SetRequireITSRefit(Bool_t b=kFALSE) {fCutRequireITSRefit=b;} - void SetAcceptKinkDaughters(Bool_t b=kTRUE) {fCutAcceptKinkDaughters=b;} - void SetMaxDCAToVertexXY(Float_t dist=1e10) {fCutMaxDCAToVertexXY = dist;} - void SetMaxDCAToVertexZ(Float_t dist=1e10) {fCutMaxDCAToVertexZ = dist;} - void SetDCAToVertex2D(Bool_t b=kFALSE) {fCutDCAToVertex2D = b;} - - // getters - - Int_t GetMinNClusterTPC() const { return fCutMinNClusterTPC;} - Int_t GetMinNClustersITS() const { return fCutMinNClusterITS;} - Float_t GetMaxChi2PerClusterTPC() const { return fCutMaxChi2PerClusterTPC;} - Float_t GetMaxChi2PerClusterITS() const { return fCutMaxChi2PerClusterITS;} - Bool_t GetRequireTPCRefit() const { return fCutRequireTPCRefit;} - Bool_t GetRequireITSRefit() const { return fCutRequireITSRefit;} - Bool_t GetAcceptKinkDaughters() const { return fCutAcceptKinkDaughters;} - Float_t GetMaxDCAToVertexXY() const { return fCutMaxDCAToVertexXY;} - Float_t GetMaxDCAToVertexZ() const { return fCutMaxDCAToVertexZ;} - Bool_t GetDCAToVertex2D() const { return fCutDCAToVertex2D;} + void SetMinTrackPt(Double_t pt=0) { fCutMinTrackPt = pt ; } + void SetMinNClustersTPC(Int_t min=-1) { fCutMinNClusterTPC = min ; } + void SetMinNClustersITS(Int_t min=-1) { fCutMinNClusterITS = min ; } + void SetMaxChi2PerClusterTPC(Float_t max=1e10) { fCutMaxChi2PerClusterTPC = max ; } + void SetMaxChi2PerClusterITS(Float_t max=1e10) { fCutMaxChi2PerClusterITS = max ; } + void SetRequireTPCRefit(Bool_t b=kFALSE) { fCutRequireTPCRefit = b ; } + void SetRequireITSRefit(Bool_t b=kFALSE) { fCutRequireITSRefit = b ; } + void SetAcceptKinkDaughters(Bool_t b=kTRUE) { fCutAcceptKinkDaughters = b ; } + void SetMaxDCAToVertexXY(Float_t dist=1e10) { fCutMaxDCAToVertexXY = dist ; } + void SetMaxDCAToVertexZ(Float_t dist=1e10) { fCutMaxDCAToVertexZ = dist ; } + void SetDCAToVertex2D(Bool_t b=kFALSE) { fCutDCAToVertex2D = b ; } + + // getters + Double_t GetMinTrackPt() const { return fCutMinTrackPt ; } + Int_t GetMinNClusterTPC() const { return fCutMinNClusterTPC ; } + Int_t GetMinNClustersITS() const { return fCutMinNClusterITS ; } + Float_t GetMaxChi2PerClusterTPC() const { return fCutMaxChi2PerClusterTPC ; } + Float_t GetMaxChi2PerClusterITS() const { return fCutMaxChi2PerClusterITS ; } + Bool_t GetRequireTPCRefit() const { return fCutRequireTPCRefit ; } + Bool_t GetRequireITSRefit() const { return fCutRequireITSRefit ; } + Bool_t GetAcceptKinkDaughters() const { return fCutAcceptKinkDaughters ; } + Float_t GetMaxDCAToVertexXY() const { return fCutMaxDCAToVertexXY ; } + Float_t GetMaxDCAToVertexZ() const { return fCutMaxDCAToVertexZ ; } + Bool_t GetDCAToVertex2D() const { return fCutDCAToVertex2D ; } private: - Float_t fMisalTransShift[15]; // Shift parameters - Float_t fMisalRotShift[15]; // Shift parameters - Int_t fNonLinearityFunction; // Non linearity function choice - Float_t fNonLinearityParams[6]; // Parameters for the non linearity function - Int_t fParticleType; // Particle type for depth calculation - Int_t fPosAlgo; // Position recalculation algorithm - Float_t fW0; // Weight0 + //Position recalculation + Float_t fMisalTransShift[15]; // Shift parameters + Float_t fMisalRotShift[15]; // Shift parameters + Int_t fParticleType; // Particle type for depth calculation + Int_t fPosAlgo; // Position recalculation algorithm + Float_t fW0; // Weight0 + + // Non linearity + Int_t fNonLinearityFunction; // Non linearity function choice + Float_t fNonLinearityParams[7]; // Parameters for the non linearity function + Int_t fNonLinearThreshold; // Non linearity threshold value for kBeamTesh non linearity function + // Energy smearing for MC + Bool_t fSmearClusterEnergy; // Smear cluster energy, to be done only for simulated data to match real data + Float_t fSmearClusterParam[3]; // Smearing parameters + TRandom3 fRandom; // Random generator + + // Recalibration Bool_t fRecalibration; // Switch on or off the recalibration TObjArray* fEMCALRecalibrationFactors; // Array of histograms with map of recalibration factors, EMCAL + + // Recalibrate with run dependent corrections + Bool_t fUseTimeCorrectionFactors; // Use Time Dependent Correction + Bool_t fTimeCorrectionFactorsSet; // Time Correction set at leat once + + // Bad Channels Bool_t fRemoveBadChannels; // Check the channel status provided and remove clusters with bad channels + Bool_t fRecalDistToBadChannels; // Calculate distance from highest energy tower of cluster to closes bad channel TObjArray* fEMCALBadChannelMap; // Array of histograms with map of bad channels, EMCAL + + // Border cells Int_t fNCellsFromEMCALBorder; // Number of cells from EMCAL border the cell with maximum amplitude has to be. Bool_t fNoEMCALBorderAtEta0; // Do fiducial cut in EMCAL region eta = 0? + + // Cluster cuts + Bool_t fRejectExoticCluster; // Switch on or off exotic cluster rejection + + // PID + AliEMCALPIDUtils * fPIDUtils; // Recalculate PID parameters + + //Track matching + UInt_t fAODFilterMask; // Filter mask to select AOD tracks. Refer to $ALICE_ROOT/ANALYSIS/macros/AddTaskESDFilter.C + TArrayI * fMatchedTrackIndex; // Array that stores indexes of matched tracks + TArrayI * fMatchedClusterIndex; // Array that stores indexes of matched clusters + TArrayF * fResidualEta; // Array that stores the residual eta + TArrayF * fResidualPhi; // Array that stores the residual phi + Bool_t fCutEtaPhiSum; // Place cut on sqrt(dEta^2+dPhi^2) + Bool_t fCutEtaPhiSeparate; // Cut on dEta and dPhi separately + Float_t fCutR; // sqrt(dEta^2+dPhi^2) cut on matching + Float_t fCutEta; // dEta cut on matching + Float_t fCutPhi; // dPhi cut on matching + Double_t fMass; // Mass hypothesis of the track + Double_t fStep; // Length of each step used in extrapolation in the unit of cm. - TArrayI *fMatchedClusterIndex; //Array that stores indexes of matched clusters - TArrayF *fResidualZ; //Array that stores the residual z - TArrayF *fResidualR; //Array that stores the residual r - Float_t fCutR; //dR cut on matching - Float_t fCutZ; //dZ cut on matching - - enum { kNCuts = 10 }; - Int_t fCutMinNClusterTPC; // min number of tpc clusters - Int_t fCutMinNClusterITS; // min number of its clusters - Float_t fCutMaxChi2PerClusterTPC; // max tpc fit chi2 per tpc cluster - Float_t fCutMaxChi2PerClusterITS; // max its fit chi2 per its cluster - Bool_t fCutRequireTPCRefit; // require TPC refit - Bool_t fCutRequireITSRefit; // require ITS refit - Bool_t fCutAcceptKinkDaughters; // accepting kink daughters? - Float_t fCutMaxDCAToVertexXY; // track-to-vertex cut in max absolute distance in xy-plane - Float_t fCutMaxDCAToVertexZ; // track-to-vertex cut in max absolute distance in z-plane - Bool_t fCutDCAToVertex2D; // if true a 2D DCA cut is made. Tracks are accepted if sqrt((DCAXY / fCutMaxDCAToVertexXY)^2 + (DCAZ / fCutMaxDCAToVertexZ)^2) < 1 AND sqrt((DCAXY / fCutMinDCAToVertexXY)^2 + (DCAZ / fCutMinDCAToVertexZ)^2) > 1 - - AliEMCALPIDUtils * fPIDUtils; // Recalculate PID parameters - - ClassDef(AliEMCALRecoUtils, 6) + // Track cuts + Int_t fTrackCutsType; // Esd track cuts type for matching + Double_t fCutMinTrackPt; // Cut on track pT + Int_t fCutMinNClusterTPC; // Min number of tpc clusters + Int_t fCutMinNClusterITS; // Min number of its clusters + Float_t fCutMaxChi2PerClusterTPC; // Max tpc fit chi2 per tpc cluster + Float_t fCutMaxChi2PerClusterITS; // Max its fit chi2 per its cluster + Bool_t fCutRequireTPCRefit; // Require TPC refit + Bool_t fCutRequireITSRefit; // Require ITS refit + Bool_t fCutAcceptKinkDaughters; // Accepting kink daughters? + Float_t fCutMaxDCAToVertexXY; // Track-to-vertex cut in max absolute distance in xy-plane + Float_t fCutMaxDCAToVertexZ; // Track-to-vertex cut in max absolute distance in z-plane + Bool_t fCutDCAToVertex2D; // If true a 2D DCA cut is made. Tracks are accepted if sqrt((DCAXY / fCutMaxDCAToVertexXY)^2 + (DCAZ / fCutMaxDCAToVertexZ)^2) < 1 AND sqrt((DCAXY / fCutMinDCAToVertexXY)^2 + (DCAZ / fCutMinDCAToVertexZ)^2) > 1 + + ClassDef(AliEMCALRecoUtils, 13) };