//
//
// Author: Gustavo Conesa (LPSC- Grenoble)
-///////////////////////////////////////////////////////////////////////////////
+// Track matching part: Rongrong Ma (Yale)
+///////////////////////////////////////////////////////////////////////////////
// --- standard c ---
// standard C++ includes
#include <TGeoManager.h>
#include <TGeoMatrix.h>
#include <TGeoBBox.h>
+#include <TH2F.h>
+#include <TArrayI.h>
+#include <TArrayF.h>
+#include "TObjArray.h"
// STEER includes
-#include "AliEMCALRecoUtils.h"
-#include "AliEMCALGeometry.h"
#include "AliVCluster.h"
#include "AliVCaloCells.h"
#include "AliVEvent.h"
#include "AliLog.h"
-#include "AliEMCALPIDUtils.h"
#include "AliPID.h"
#include "AliESDEvent.h"
+#include "AliAODEvent.h"
#include "AliESDtrack.h"
+#include "AliAODTrack.h"
+#include "AliExternalTrackParam.h"
+#include "AliESDfriendTrack.h"
+#include "AliTrackerBase.h"
+
+// EMCAL includes
+#include "AliEMCALRecoUtils.h"
+#include "AliEMCALGeometry.h"
#include "AliEMCALTrack.h"
+#include "AliEMCALCalibTimeDepCorrection.h"
+#include "AliEMCALPIDUtils.h"
ClassImp(AliEMCALRecoUtils)
//______________________________________________
AliEMCALRecoUtils::AliEMCALRecoUtils():
fNonLinearityFunction (kNoCorrection), fParticleType(kPhoton),
- fPosAlgo(kUnchanged),fW0(4.),
+ fPosAlgo(kUnchanged),fW0(4.), fNonLinearThreshold(30),
fRecalibration(kFALSE), fEMCALRecalibrationFactors(),
- fRemoveBadChannels(kFALSE), fEMCALBadChannelMap(),
+ fRemoveBadChannels(kFALSE), fRecalDistToBadChannels(kFALSE), fEMCALBadChannelMap(),
fNCellsFromEMCALBorder(0), fNoEMCALBorderAtEta0(kTRUE),
- fMatchedClusterIndex(0x0), fResidualZ(0x0), fResidualR(0x0), fCutR(40), fCutZ(20),
- fCutMinNClusterTPC(0), fCutMinNClusterITS(0), fCutMaxChi2PerClusterTPC(0), fCutMaxChi2PerClusterITS(0),
- fCutRequireTPCRefit(0), fCutRequireITSRefit(0), fCutAcceptKinkDaughters(0),
- fCutMaxDCAToVertexXY(0), fCutMaxDCAToVertexZ(0),fCutDCAToVertex2D(0),
- fPIDUtils()
+ fAODFilterMask(32),
+ fMatchedTrackIndex(0x0), fMatchedClusterIndex(0x0),
+ fResidualEta(0x0), fResidualPhi(0x0), fCutEtaPhiSum(kTRUE), fCutEtaPhiSeparate(kFALSE), fCutR(0.1), fCutEta(0.02), fCutPhi(0.04), fMass(0.139), fStep(50),
+ fRejectExoticCluster(kFALSE),
+ fTrackCutsType(kTPCOnlyCut), fCutMinTrackPt(0), fCutMinNClusterTPC(-1), fCutMinNClusterITS(-1), fCutMaxChi2PerClusterTPC(1e10), fCutMaxChi2PerClusterITS(1e10),
+ fCutRequireTPCRefit(kFALSE), fCutRequireITSRefit(kFALSE), fCutAcceptKinkDaughters(kFALSE),
+ fCutMaxDCAToVertexXY(1e10), fCutMaxDCAToVertexZ(1e10),fCutDCAToVertex2D(kFALSE),fPIDUtils(),
+ fUseTimeCorrectionFactors(kFALSE), fTimeCorrectionFactorsSet(kFALSE)
{
//
// Constructor.
// during Reco algorithm execution
//
+ //Misalignment matrices
for(Int_t i = 0; i < 15 ; i++) {
fMisalTransShift[i] = 0.;
- fMisalRotShift[i] = 0.;
+ fMisalRotShift[i] = 0.;
}
- for(Int_t i = 0; i < 6 ; i++) fNonLinearityParams[i] = 0.;
- //For kPi0GammaGamma case, but default is no correction
- fNonLinearityParams[0] = 0.1457/0.1349766/1.038;
- fNonLinearityParams[1] = -0.02024/0.1349766/1.038;
- fNonLinearityParams[2] = 1.046;
-
- fMatchedClusterIndex = new TArrayI();
- fResidualZ = new TArrayF();
- fResidualR = new TArrayF();
- fPIDUtils = new AliEMCALPIDUtils();
-
+ //Non linearity
+ for(Int_t i = 0; i < 7 ; i++) fNonLinearityParams[i] = 0.;
+
+ //For kBeamTestCorrected case, but default is no correction
+ fNonLinearityParams[0] = 0.99078;
+ fNonLinearityParams[1] = 0.161499;
+ fNonLinearityParams[2] = 0.655166;
+ fNonLinearityParams[3] = 0.134101;
+ fNonLinearityParams[4] = 163.282;
+ fNonLinearityParams[5] = 23.6904;
+ fNonLinearityParams[6] = 0.978;
+
+ //For kPi0GammaGamma case
+ //fNonLinearityParams[0] = 0.1457/0.1349766/1.038;
+ //fNonLinearityParams[1] = -0.02024/0.1349766/1.038;
+ //fNonLinearityParams[2] = 1.046;
+
+ //Track matching
+ fMatchedTrackIndex = new TArrayI();
+ fMatchedClusterIndex = new TArrayI();
+ fResidualPhi = new TArrayF();
+ fResidualEta = new TArrayF();
+
InitTrackCuts();
-
+ fPIDUtils = new AliEMCALPIDUtils();
}
//______________________________________________________________________
AliEMCALRecoUtils::AliEMCALRecoUtils(const AliEMCALRecoUtils & reco)
: TNamed(reco), fNonLinearityFunction(reco.fNonLinearityFunction),
- fParticleType(reco.fParticleType), fPosAlgo(reco.fPosAlgo), fW0(reco.fW0),
+ fParticleType(reco.fParticleType), fPosAlgo(reco.fPosAlgo), fW0(reco.fW0), fNonLinearThreshold(reco.fNonLinearThreshold),
fRecalibration(reco.fRecalibration),fEMCALRecalibrationFactors(reco.fEMCALRecalibrationFactors),
- fRemoveBadChannels(reco.fRemoveBadChannels),fEMCALBadChannelMap(reco.fEMCALBadChannelMap),
+ fRemoveBadChannels(reco.fRemoveBadChannels),fRecalDistToBadChannels(reco.fRecalDistToBadChannels),
+ fEMCALBadChannelMap(reco.fEMCALBadChannelMap),
fNCellsFromEMCALBorder(reco.fNCellsFromEMCALBorder),fNoEMCALBorderAtEta0(reco.fNoEMCALBorderAtEta0),
+ fAODFilterMask(reco.fAODFilterMask),
+ fMatchedTrackIndex(reco.fMatchedTrackIndex?new TArrayI(*reco.fMatchedTrackIndex):0x0),
fMatchedClusterIndex(reco.fMatchedClusterIndex?new TArrayI(*reco.fMatchedClusterIndex):0x0),
- fResidualZ(reco.fResidualZ?new TArrayF(*reco.fResidualZ):0x0),
- fResidualR(reco.fResidualR?new TArrayF(*reco.fResidualR):0x0),
- fCutR(reco.fCutR),fCutZ(reco.fCutZ),
- fCutMinNClusterTPC(reco.fCutMinNClusterTPC), fCutMinNClusterITS(reco.fCutMinNClusterITS),
+ fResidualEta(reco.fResidualEta?new TArrayF(*reco.fResidualEta):0x0),
+ fResidualPhi(reco.fResidualPhi?new TArrayF(*reco.fResidualPhi):0x0),
+ fCutEtaPhiSum(reco.fCutEtaPhiSum), fCutEtaPhiSeparate(reco.fCutEtaPhiSeparate), fCutR(reco.fCutR), fCutEta(reco.fCutEta), fCutPhi(reco.fCutPhi),
+ fMass(reco.fMass), fStep(reco.fStep),
+ fRejectExoticCluster(reco.fRejectExoticCluster),
+ fTrackCutsType(reco.fTrackCutsType), fCutMinTrackPt(reco.fCutMinTrackPt), fCutMinNClusterTPC(reco.fCutMinNClusterTPC), fCutMinNClusterITS(reco.fCutMinNClusterITS),
fCutMaxChi2PerClusterTPC(reco.fCutMaxChi2PerClusterTPC), fCutMaxChi2PerClusterITS(reco.fCutMaxChi2PerClusterITS),
fCutRequireTPCRefit(reco.fCutRequireTPCRefit), fCutRequireITSRefit(reco.fCutRequireITSRefit),
fCutAcceptKinkDaughters(reco.fCutAcceptKinkDaughters),
fCutMaxDCAToVertexXY(reco.fCutMaxDCAToVertexXY), fCutMaxDCAToVertexZ(reco.fCutMaxDCAToVertexZ),fCutDCAToVertex2D(reco.fCutDCAToVertex2D),
- fPIDUtils(reco.fPIDUtils)
-
+ fPIDUtils(reco.fPIDUtils),
+ fUseTimeCorrectionFactors(reco.fUseTimeCorrectionFactors), fTimeCorrectionFactorsSet(reco.fTimeCorrectionFactorsSet)
{
//Copy ctor
fMisalRotShift[i] = reco.fMisalRotShift[i];
fMisalTransShift[i] = reco.fMisalTransShift[i];
}
- for(Int_t i = 0; i < 6 ; i++) fNonLinearityParams[i] = reco.fNonLinearityParams[i];
+ for(Int_t i = 0; i < 7 ; i++) fNonLinearityParams[i] = reco.fNonLinearityParams[i];
}
if(this == &reco)return *this;
((TNamed *)this)->operator=(reco);
- fNonLinearityFunction = reco.fNonLinearityFunction;
- fParticleType = reco.fParticleType;
- fPosAlgo = reco.fPosAlgo;
- fW0 = reco.fW0;
- fRecalibration = reco.fRecalibration;
+ fNonLinearityFunction = reco.fNonLinearityFunction;
+ fParticleType = reco.fParticleType;
+ fPosAlgo = reco.fPosAlgo;
+ fW0 = reco.fW0;
+ fNonLinearThreshold = reco.fNonLinearThreshold;
+ fRecalibration = reco.fRecalibration;
fEMCALRecalibrationFactors = reco.fEMCALRecalibrationFactors;
- fRemoveBadChannels = reco.fRemoveBadChannels;
- fEMCALBadChannelMap = reco.fEMCALBadChannelMap;
- fNCellsFromEMCALBorder = reco.fNCellsFromEMCALBorder;
- fNoEMCALBorderAtEta0 = reco.fNoEMCALBorderAtEta0;
+ fRemoveBadChannels = reco.fRemoveBadChannels;
+ fRecalDistToBadChannels = reco.fRecalDistToBadChannels;
+ fEMCALBadChannelMap = reco.fEMCALBadChannelMap;
+ fNCellsFromEMCALBorder = reco.fNCellsFromEMCALBorder;
+ fNoEMCALBorderAtEta0 = reco.fNoEMCALBorderAtEta0;
for(Int_t i = 0; i < 15 ; i++) {fMisalTransShift[i] = reco.fMisalTransShift[i]; fMisalRotShift[i] = reco.fMisalRotShift[i];}
- for(Int_t i = 0; i < 6 ; i++) fNonLinearityParams[i] = reco.fNonLinearityParams[i];
-
- fCutR = reco.fCutR;
- fCutZ = reco.fCutZ;
-
- fCutMinNClusterTPC = reco.fCutMinNClusterTPC;
- fCutMinNClusterITS = reco.fCutMinNClusterITS;
- fCutMaxChi2PerClusterTPC = reco.fCutMaxChi2PerClusterTPC;
- fCutMaxChi2PerClusterITS = reco.fCutMaxChi2PerClusterITS;
- fCutRequireTPCRefit = reco.fCutRequireTPCRefit;
- fCutRequireITSRefit = reco.fCutRequireITSRefit;
- fCutAcceptKinkDaughters = reco.fCutAcceptKinkDaughters;
- fCutMaxDCAToVertexXY = reco.fCutMaxDCAToVertexXY;
- fCutMaxDCAToVertexZ = reco.fCutMaxDCAToVertexZ;
- fCutDCAToVertex2D = reco.fCutDCAToVertex2D;
+ for(Int_t i = 0; i < 7 ; i++) fNonLinearityParams[i] = reco.fNonLinearityParams[i];
- fPIDUtils = reco.fPIDUtils;
+ fAODFilterMask = reco.fAODFilterMask;
+ fCutEtaPhiSum = reco.fCutEtaPhiSum;
+ fCutEtaPhiSeparate = reco.fCutEtaPhiSeparate;
+ fCutR = reco.fCutR;
+ fCutEta = reco.fCutEta;
+ fCutPhi = reco.fCutPhi;
+ fMass = reco.fMass;
+ fStep = reco.fStep;
+ fRejectExoticCluster = reco.fRejectExoticCluster;
+
+ fTrackCutsType = reco.fTrackCutsType;
+ fCutMinTrackPt = reco.fCutMinTrackPt;
+ fCutMinNClusterTPC = reco.fCutMinNClusterTPC;
+ fCutMinNClusterITS = reco.fCutMinNClusterITS;
+ fCutMaxChi2PerClusterTPC = reco.fCutMaxChi2PerClusterTPC;
+ fCutMaxChi2PerClusterITS = reco.fCutMaxChi2PerClusterITS;
+ fCutRequireTPCRefit = reco.fCutRequireTPCRefit;
+ fCutRequireITSRefit = reco.fCutRequireITSRefit;
+ fCutAcceptKinkDaughters = reco.fCutAcceptKinkDaughters;
+ fCutMaxDCAToVertexXY = reco.fCutMaxDCAToVertexXY;
+ fCutMaxDCAToVertexZ = reco.fCutMaxDCAToVertexZ;
+ fCutDCAToVertex2D = reco.fCutDCAToVertex2D;
+
+ fPIDUtils = reco.fPIDUtils;
- if(reco.fResidualR){
+ fUseTimeCorrectionFactors = reco.fUseTimeCorrectionFactors;
+ fTimeCorrectionFactorsSet = reco.fTimeCorrectionFactorsSet;
+
+
+ if(reco.fResidualEta){
// assign or copy construct
- if(fResidualR){
- *fResidualR = *reco.fResidualR;
+ if(fResidualEta){
+ *fResidualEta = *reco.fResidualEta;
}
- else fResidualR = new TArrayF(*reco.fResidualR);
+ else fResidualEta = new TArrayF(*reco.fResidualEta);
}
else{
- if(fResidualR)delete fResidualR;
- fResidualR = 0;
+ if(fResidualEta)delete fResidualEta;
+ fResidualEta = 0;
}
- if(reco.fResidualZ){
+ if(reco.fResidualPhi){
// assign or copy construct
- if(fResidualZ){
- *fResidualZ = *reco.fResidualZ;
+ if(fResidualPhi){
+ *fResidualPhi = *reco.fResidualPhi;
}
- else fResidualZ = new TArrayF(*reco.fResidualZ);
+ else fResidualPhi = new TArrayF(*reco.fResidualPhi);
}
else{
- if(fResidualZ)delete fResidualZ;
- fResidualZ = 0;
+ if(fResidualPhi)delete fResidualPhi;
+ fResidualPhi = 0;
}
+ if(reco.fMatchedTrackIndex){
+ // assign or copy construct
+ if(fMatchedTrackIndex){
+ *fMatchedTrackIndex = *reco.fMatchedTrackIndex;
+ }
+ else fMatchedTrackIndex = new TArrayI(*reco.fMatchedTrackIndex);
+ }
+ else{
+ if(fMatchedTrackIndex)delete fMatchedTrackIndex;
+ fMatchedTrackIndex = 0;
+ }
if(reco.fMatchedClusterIndex){
// assign or copy construct
if(fMatchedClusterIndex)delete fMatchedClusterIndex;
fMatchedClusterIndex = 0;
}
-
-
+
return *this;
}
delete fEMCALBadChannelMap;
}
- if(fMatchedClusterIndex) {delete fMatchedClusterIndex; fMatchedClusterIndex=0;}
- if(fResidualR) {delete fResidualR; fResidualR=0;}
- if(fResidualZ) {delete fResidualZ; fResidualZ=0;}
+ delete fMatchedTrackIndex ;
+ delete fMatchedClusterIndex ;
+ delete fResidualEta ;
+ delete fResidualPhi ;
}
// Given the list of AbsId of the cluster, get the maximum cell and
// check if there are fNCellsFromBorder from the calorimeter border
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return kFALSE;
+ }
+
//If the distance to the border is 0 or negative just exit accept all clusters
if(cells->GetType()==AliVCaloCells::kEMCALCell && fNCellsFromEMCALBorder <= 0 ) return kTRUE;
- Int_t absIdMax = -1, iSM =-1, ieta = -1, iphi = -1;
- GetMaxEnergyCell(geom, cells, cluster, absIdMax, iSM, ieta, iphi);
+ Int_t absIdMax = -1, iSM =-1, ieta = -1, iphi = -1;
+ Bool_t shared = kFALSE;
+ GetMaxEnergyCell(geom, cells, cluster, absIdMax, iSM, ieta, iphi, shared);
AliDebug(2,Form("Cluster Max AbsId %d, Cell Energy %2.2f, Cluster Energy %2.2f, Ncells from border %d, EMCAL eta=0 %d\n",
absIdMax, cells->GetCellAmplitude(absIdMax), cluster->E(), fNCellsFromEMCALBorder, fNoEMCALBorderAtEta0));
//_________________________________________________________________________________________________________
-Bool_t AliEMCALRecoUtils::ClusterContainsBadChannel(AliEMCALGeometry* geom, UShort_t* cellList, Int_t nCells){
+Bool_t AliEMCALRecoUtils::ClusterContainsBadChannel(AliEMCALGeometry* geom, UShort_t* cellList, const Int_t nCells){
// Check that in the cluster cells, there is no bad channel of those stored
// in fEMCALBadChannelMap or fPHOSBadChannelMap
}
+//_________________________________________________
+Bool_t AliEMCALRecoUtils::IsExoticCluster(AliVCluster *cluster) const {
+ // Check if the cluster has high energy but small number of cells
+ // The criteria comes from Gustavo's study
+ //
+
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return kFALSE;
+ }
+
+ Int_t nc = cluster->GetNCells() ;
+
+ if ( nc > 8 ) return kFALSE ; // Good cluster, needed for 3x3 clusterizer
+ else if ( nc < 1 + cluster->E()/3. ) return kTRUE ; // Bad cluster
+ else return kFALSE ; // Good cluster
+
+}
+
//__________________________________________________
Float_t AliEMCALRecoUtils::CorrectClusterEnergyLinearity(AliVCluster* cluster){
// Correct cluster energy from non linearity functions
+
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return 0;
+ }
+
+
Float_t energy = cluster->E();
switch (fNonLinearityFunction) {
case kPi0MC:
+ {
//Non-Linearity correction (from MC with function ([0]*exp(-[1]/E))+(([2]/([3]*2.*TMath::Pi())*exp(-(E-[4])^2/(2.*[3]^2)))))
- //Double_t par0 = 1.001;
- //Double_t par1 = -0.01264;
- //Double_t par2 = -0.03632;
- //Double_t par3 = 0.1798;
- //Double_t par4 = -0.522;
- energy /= (fNonLinearityParams[0]*exp(-fNonLinearityParams[1]/energy))+
+ //Double_t fNonLinearityParams[0] = 1.014;
+ //Double_t fNonLinearityParams[1] = -0.03329;
+ //Double_t fNonLinearityParams[2] = -0.3853;
+ //Double_t fNonLinearityParams[3] = 0.5423;
+ //Double_t fNonLinearityParams[4] = -0.4335;
+ energy *= (fNonLinearityParams[0]*exp(-fNonLinearityParams[1]/energy))+
((fNonLinearityParams[2]/(fNonLinearityParams[3]*2.*TMath::Pi())*
exp(-(energy-fNonLinearityParams[4])*(energy-fNonLinearityParams[4])/(2.*fNonLinearityParams[3]*fNonLinearityParams[3]))));
break;
-
+ }
+
case kPi0GammaGamma:
-
+ {
//Non-Linearity correction (from Olga Data with function p0+p1*exp(-p2*E))
- //Double_t par0 = 0.1457;
- //Double_t par1 = -0.02024;
- //Double_t par2 = 1.046;
+ //Double_t fNonLinearityParams[0] = 1.04;
+ //Double_t fNonLinearityParams[1] = -0.1445;
+ //Double_t fNonLinearityParams[2] = 1.046;
energy /= (fNonLinearityParams[0]+fNonLinearityParams[1]*exp(-fNonLinearityParams[2]*energy)); //Olga function
break;
+ }
case kPi0GammaConversion:
-
+ {
//Non-Linearity correction (Nicolas from Dimitri Data with function C*[1-a*exp(-b*E)])
- //Double_t C = 0.139393/0.1349766;
- //Double_t a = 0.0566186;
- //Double_t b = 0.982133;
+ //fNonLinearityParams[0] = 0.139393/0.1349766;
+ //fNonLinearityParams[1] = 0.0566186;
+ //fNonLinearityParams[2] = 0.982133;
energy /= fNonLinearityParams[0]*(1-fNonLinearityParams[1]*exp(-fNonLinearityParams[2]*energy));
break;
+ }
+
+ case kBeamTest:
+ {
+ //From beam test, Alexei's results, for different ZS thresholds
+ // th=30 MeV; th = 45 MeV; th = 75 MeV
+ //fNonLinearityParams[0] = 1.007; 1.003; 1.002
+ //fNonLinearityParams[1] = 0.894; 0.719; 0.797
+ //fNonLinearityParams[2] = 0.246; 0.334; 0.358
+ //Rescale the param[0] with 1.03
+ energy /= fNonLinearityParams[0]/(1+fNonLinearityParams[1]*exp(-energy/fNonLinearityParams[2]));
+
+ break;
+ }
+
+ case kBeamTestCorrected:
+ {
+ //From beam test, corrected for material between beam and EMCAL
+ //fNonLinearityParams[0] = 0.99078
+ //fNonLinearityParams[1] = 0.161499;
+ //fNonLinearityParams[2] = 0.655166;
+ //fNonLinearityParams[3] = 0.134101;
+ //fNonLinearityParams[4] = 163.282;
+ //fNonLinearityParams[5] = 23.6904;
+ //fNonLinearityParams[6] = 0.978;
+ energy *= fNonLinearityParams[6]/(fNonLinearityParams[0]*(1./(1.+fNonLinearityParams[1]*exp(-energy/fNonLinearityParams[2]))*1./(1.+fNonLinearityParams[3]*exp((energy-fNonLinearityParams[4])/fNonLinearityParams[5]))));
+
+ break;
+ }
case kNoCorrection:
AliDebug(2,"No correction on the energy\n");
return energy;
}
+//__________________________________________________
+void AliEMCALRecoUtils::InitNonLinearityParam()
+{
+ //Initialising Non Linearity Parameters
+
+ if(fNonLinearityFunction == kPi0MC)
+ {
+ fNonLinearityParams[0] = 1.014;
+ fNonLinearityParams[1] = -0.03329;
+ fNonLinearityParams[2] = -0.3853;
+ fNonLinearityParams[3] = 0.5423;
+ fNonLinearityParams[4] = -0.4335;
+ }
+
+ if(fNonLinearityFunction == kPi0GammaGamma)
+ {
+ fNonLinearityParams[0] = 1.04;
+ fNonLinearityParams[1] = -0.1445;
+ fNonLinearityParams[2] = 1.046;
+ }
+
+ if(fNonLinearityFunction == kPi0GammaConversion)
+ {
+ fNonLinearityParams[0] = 0.139393;
+ fNonLinearityParams[1] = 0.0566186;
+ fNonLinearityParams[2] = 0.982133;
+ }
+
+ if(fNonLinearityFunction == kBeamTest)
+ {
+ if(fNonLinearThreshold == 30)
+ {
+ fNonLinearityParams[0] = 1.007;
+ fNonLinearityParams[1] = 0.894;
+ fNonLinearityParams[2] = 0.246;
+ }
+ if(fNonLinearThreshold == 45)
+ {
+ fNonLinearityParams[0] = 1.003;
+ fNonLinearityParams[1] = 0.719;
+ fNonLinearityParams[2] = 0.334;
+ }
+ if(fNonLinearThreshold == 75)
+ {
+ fNonLinearityParams[0] = 1.002;
+ fNonLinearityParams[1] = 0.797;
+ fNonLinearityParams[2] = 0.358;
+ }
+ }
+
+ if(fNonLinearityFunction == kBeamTestCorrected)
+ {
+ fNonLinearityParams[0] = 0.99078;
+ fNonLinearityParams[1] = 0.161499;
+ fNonLinearityParams[2] = 0.655166;
+ fNonLinearityParams[3] = 0.134101;
+ fNonLinearityParams[4] = 163.282;
+ fNonLinearityParams[5] = 23.6904;
+ fNonLinearityParams[6] = 0.978;
+ }
+}
//__________________________________________________
Float_t AliEMCALRecoUtils::GetDepth(const Float_t energy, const Int_t iParticle, const Int_t iSM) const
//Calculate shower depth for a given cluster energy and particle type
// parameters
- Float_t x0 = 1.23;
+ Float_t x0 = 1.31;
Float_t ecr = 8;
Float_t depth = 0;
}
//__________________________________________________
-void AliEMCALRecoUtils::GetMaxEnergyCell(AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu, Int_t & absId, Int_t& iSupMod, Int_t& ieta, Int_t& iphi)
+void AliEMCALRecoUtils::GetMaxEnergyCell(AliEMCALGeometry *geom, AliVCaloCells* cells, AliVCluster* clu,
+ Int_t & absId, Int_t& iSupMod, Int_t& ieta, Int_t& iphi, Bool_t &shared)
{
//For a given CaloCluster gets the absId of the cell
//with maximum energy deposit.
Int_t iTower = -1;
Int_t iIphi = -1;
Int_t iIeta = -1;
- //printf("---Max?\n");
+ Int_t iSupMod0= -1;
+
+ if(!clu){
+ AliInfo("Cluster pointer null!");
+ absId=-1; iSupMod0=-1, ieta = -1; iphi = -1; shared = -1;
+ return;
+ }
+
for (Int_t iDig=0; iDig< clu->GetNCells(); iDig++) {
cellAbsId = clu->GetCellAbsId(iDig);
fraction = clu->GetCellAmplitudeFraction(iDig);
//printf("a Cell %d, id, %d, amp %f, fraction %f\n",iDig,cellAbsId,cells->GetCellAmplitude(cellAbsId),fraction);
if(fraction < 1e-4) fraction = 1.; // in case unfolding is off
+ geom->GetCellIndex(cellAbsId,iSupMod,iTower,iIphi,iIeta);
+ geom->GetCellPhiEtaIndexInSModule(iSupMod,iTower,iIphi, iIeta,iphi,ieta);
+ if(iDig==0) iSupMod0=iSupMod;
+ else if(iSupMod0!=iSupMod) {
+ shared = kTRUE;
+ //printf("AliEMCALRecoUtils::GetMaxEnergyCell() - SHARED CLUSTER\n");
+ }
if(IsRecalibrationOn()) {
- geom->GetCellIndex(cellAbsId,iSupMod,iTower,iIphi,iIeta);
- geom->GetCellPhiEtaIndexInSModule(iSupMod,iTower,iIphi, iIeta,iphi,ieta);
recalFactor = GetEMCALChannelRecalibrationFactor(iSupMod,ieta,iphi);
}
eCell = cells->GetCellAmplitude(cellAbsId)*fraction*recalFactor;
Bool_t oldStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE);
- fEMCALRecalibrationFactors = new TObjArray(12);
- for (int i = 0; i < 12; i++) fEMCALRecalibrationFactors->Add(new TH2F(Form("EMCALRecalFactors_SM%d",i),Form("EMCALRecalFactors_SM%d",i), 48, 0, 48, 24, 0, 24));
+ fEMCALRecalibrationFactors = new TObjArray(10);
+ for (int i = 0; i < 10; i++) fEMCALRecalibrationFactors->Add(new TH2F(Form("EMCALRecalFactors_SM%d",i),Form("EMCALRecalFactors_SM%d",i), 48, 0, 48, 24, 0, 24));
//Init the histograms with 1
- for (Int_t sm = 0; sm < 12; sm++) {
+ for (Int_t sm = 0; sm < 10; sm++) {
for (Int_t i = 0; i < 48; i++) {
for (Int_t j = 0; j < 24; j++) {
SetEMCALChannelRecalibrationFactor(sm,i,j,1.);
Bool_t oldStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE);
- fEMCALBadChannelMap = new TObjArray(12);
+ fEMCALBadChannelMap = new TObjArray(10);
//TH2F * hTemp = new TH2I("EMCALBadChannelMap","EMCAL SuperModule bad channel map", 48, 0, 48, 24, 0, 24);
- for (int i = 0; i < 12; i++) {
+ for (int i = 0; i < 10; i++) {
fEMCALBadChannelMap->Add(new TH2I(Form("EMCALBadChannelMap_Mod%d",i),Form("EMCALBadChannelMap_Mod%d",i), 48, 0, 48, 24, 0, 24));
}
void AliEMCALRecoUtils::RecalibrateClusterEnergy(AliEMCALGeometry* geom, AliVCluster * cluster, AliVCaloCells * cells){
// Recalibrate the cluster energy, considering the recalibration map and the energy of the cells that compose the cluster.
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return;
+ }
+
//Get the cluster number of cells and list of absId, check what kind of cluster do we have.
UShort_t * index = cluster->GetCellsAbsId() ;
Double_t * fraction = cluster->GetCellsAmplitudeFraction() ;
{
//For a given CaloCluster recalculates the position for a given set of misalignment shifts and puts it again in the CaloCluster.
+ if(!clu){
+ AliInfo("Cluster pointer null!");
+ return;
+ }
+
if (fPosAlgo==kPosTowerGlobal) RecalculateClusterPositionFromTowerGlobal( geom, cells, clu);
else if(fPosAlgo==kPosTowerIndex) RecalculateClusterPositionFromTowerIndex ( geom, cells, clu);
else AliDebug(2,"Algorithm to recalculate position not selected, do nothing.");
Float_t weight = 0., totalWeight=0.;
Float_t newPos[3] = {0,0,0};
Double_t pLocal[3], pGlobal[3];
-
+ Bool_t shared = kFALSE;
+
Float_t clEnergy = clu->E(); //Energy already recalibrated previously
- GetMaxEnergyCell(geom, cells, clu, absId, iSupModMax, ieta, iphi);
+ GetMaxEnergyCell(geom, cells, clu, absId, iSupModMax, ieta, iphi,shared);
Double_t depth = GetDepth(clEnergy,fParticleType,iSupModMax) ;
//printf("** Cluster energy %f, ncells %d, depth %f\n",clEnergy,clu->GetNCells(),depth);
Int_t iIphi = -1, iIeta = -1;
Int_t iSupMod = -1, iSupModMax = -1;
Int_t iphi = -1, ieta =-1;
-
+ Bool_t shared = kFALSE;
+
Float_t clEnergy = clu->E(); //Energy already recalibrated previously.
- GetMaxEnergyCell(geom, cells, clu, absId, iSupModMax, ieta, iphi);
+ GetMaxEnergyCell(geom, cells, clu, absId, iSupModMax, ieta, iphi,shared);
Float_t depth = GetDepth(clEnergy,fParticleType,iSupMod) ;
Float_t weight = 0., weightedCol = 0., weightedRow = 0., totalWeight=0.;
}
+//____________________________________________________________________________
+void AliEMCALRecoUtils::RecalculateClusterDistanceToBadChannel(AliEMCALGeometry * geom, AliVCaloCells* cells, AliVCluster * cluster){
+
+ //re-evaluate distance to bad channel with updated bad map
+
+ if(!fRecalDistToBadChannels) return;
+
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return;
+ }
+
+ //Get channels map of the supermodule where the cluster is.
+ Int_t absIdMax = -1, iSupMod =-1, icolM = -1, irowM = -1;
+ Bool_t shared = kFALSE;
+ GetMaxEnergyCell(geom, cells, cluster, absIdMax, iSupMod, icolM, irowM, shared);
+ TH2D* hMap = (TH2D*)fEMCALBadChannelMap->At(iSupMod);
+
+ Int_t dRrow, dRcol;
+ Float_t minDist = 10000.;
+ Float_t dist = 0.;
+
+ //Loop on tower status map
+ for(Int_t irow = 0; irow < AliEMCALGeoParams::fgkEMCALRows; irow++){
+ for(Int_t icol = 0; icol < AliEMCALGeoParams::fgkEMCALCols; icol++){
+ //Check if tower is bad.
+ if(hMap->GetBinContent(icol,irow)==0) continue;
+ //printf("AliEMCALRecoUtils::RecalculateDistanceToBadChannels() - \n \t Bad channel in SM %d, col %d, row %d, \n \t Cluster max in col %d, row %d\n",
+ // iSupMod,icol, irow, icolM,irowM);
+
+ dRrow=TMath::Abs(irowM-irow);
+ dRcol=TMath::Abs(icolM-icol);
+ dist=TMath::Sqrt(dRrow*dRrow+dRcol*dRcol);
+ if(dist < minDist){
+ //printf("MIN DISTANCE TO BAD %2.2f\n",dist);
+ minDist = dist;
+ }
+
+ }
+ }
+
+ //In case the cluster is shared by 2 SuperModules, need to check the map of the second Super Module
+ if (shared) {
+ TH2D* hMap2 = 0;
+ Int_t iSupMod2 = -1;
+
+ //The only possible combinations are (0,1), (2,3) ... (8,9)
+ if(iSupMod%2) iSupMod2 = iSupMod-1;
+ else iSupMod2 = iSupMod+1;
+ hMap2 = (TH2D*)fEMCALBadChannelMap->At(iSupMod2);
+
+ //Loop on tower status map of second super module
+ for(Int_t irow = 0; irow < AliEMCALGeoParams::fgkEMCALRows; irow++){
+ for(Int_t icol = 0; icol < AliEMCALGeoParams::fgkEMCALCols; icol++){
+ //Check if tower is bad.
+ if(hMap2->GetBinContent(icol,irow)==0) continue;
+ //printf("AliEMCALRecoUtils::RecalculateDistanceToBadChannels(shared) - \n \t Bad channel in SM %d, col %d, row %d \n \t Cluster max in SM %d, col %d, row %d\n",
+ // iSupMod2,icol, irow,iSupMod,icolM,irowM);
+
+ dRrow=TMath::Abs(irow-irowM);
+
+ if(iSupMod%2) {
+ dRcol=TMath::Abs(icol-(AliEMCALGeoParams::fgkEMCALCols+icolM));
+ }
+ else {
+ dRcol=TMath::Abs(AliEMCALGeoParams::fgkEMCALCols+icol-icolM);
+ }
+
+ dist=TMath::Sqrt(dRrow*dRrow+dRcol*dRcol);
+ if(dist < minDist) minDist = dist;
+
+ }
+ }
+
+ }// shared cluster in 2 SuperModules
+
+ AliDebug(2,Form("Max cluster cell (SM,col,row)=(%d %d %d) - Distance to Bad Channel %2.2f",iSupMod, icolM, irowM, minDist));
+ cluster->SetDistanceToBadChannel(minDist);
+
+}
+
//____________________________________________________________________________
void AliEMCALRecoUtils::RecalculateClusterPID(AliVCluster * cluster){
//re-evaluate identification parameters with bayesian
-
+
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return;
+ }
+
if ( cluster->GetM02() != 0)
fPIDUtils->ComputePID(cluster->E(),cluster->GetM02());
// and tranfers into global ALICE coordinates
// Calculates Dispersion and main axis
+ if(!cluster){
+ AliInfo("Cluster pointer null!");
+ return;
+ }
+
Int_t nstat = 0;
Float_t wtot = 0. ;
Double_t eCell = 0.;
cluster->SetDispersion(TMath::Sqrt(d)) ;
else
cluster->SetDispersion(0) ;
-
}
-//__________________________________________________
-void AliEMCALRecoUtils::FindMatches(AliVEvent *event)
+//____________________________________________________________________________
+void AliEMCALRecoUtils::FindMatches(AliVEvent *event,TObjArray * clusterArr, AliEMCALGeometry *geom)
{
//This function should be called before the cluster loop
//Before call this function, please recalculate the cluster positions
//Given the input event, loop over all the tracks, select the closest cluster as matched with fCutR
//Store matched cluster indexes and residuals
- //It only works with ESDs, not AODs
+ fMatchedTrackIndex->Reset();
fMatchedClusterIndex->Reset();
- fResidualZ->Reset();
- fResidualR->Reset();
+ fResidualPhi->Reset();
+ fResidualEta->Reset();
+
+ fMatchedTrackIndex->Set(500);
+ fMatchedClusterIndex->Set(500);
+ fResidualPhi->Set(500);
+ fResidualEta->Set(500);
- fMatchedClusterIndex->Set(100);
- fResidualZ->Set(100);
- fResidualR->Set(100);
+ AliESDEvent* esdevent = dynamic_cast<AliESDEvent*> (event);
+ AliAODEvent* aodevent = dynamic_cast<AliAODEvent*> (event);
Int_t matched=0;
- Float_t clsPos[3];
- Double_t trkPos[3];
+ Double_t cv[21];
+ for (Int_t i=0; i<21;i++) cv[i]=0;
for(Int_t itr=0; itr<event->GetNumberOfTracks(); itr++)
{
- AliESDtrack *track = ((AliESDEvent*)event)->GetTrack(itr);
- if(!track || !IsAccepted(track)) continue;
+ AliExternalTrackParam *trackParam = 0;
- Float_t dRMax = fCutR, dZMax = fCutZ;
- Int_t index = -1;
- AliEMCALTrack *emctrack = new AliEMCALTrack(*track);
- for(Int_t icl=0; icl<event->GetNumberOfCaloClusters(); icl++)
+ //If the input event is ESD, the starting point for extrapolation is TPCOut, if available, or TPCInner
+ if(esdevent)
{
- AliVCluster *cluster = (AliVCluster*) event->GetCaloCluster(icl);
- if(!cluster->IsEMCAL()) continue;
- cluster->GetPosition(clsPos); //Has been recalculated
- if(!emctrack->PropagateToGlobal(clsPos[0],clsPos[1],clsPos[2],0.,0.) ) continue;
- emctrack->GetXYZ(trkPos);
- Float_t tmpR = TMath::Sqrt( TMath::Power(clsPos[0]-trkPos[0],2)+TMath::Power(clsPos[1]-trkPos[1],2)+TMath::Power(clsPos[2]-trkPos[2],2) );
- Float_t tmpZ = TMath::Abs(clsPos[2]-trkPos[2]);
-
- if(tmpR<dRMax)
- {
- dRMax=tmpR;
- dZMax=tmpZ;
- index=icl;
- }
-
- }//cluser loop
+ AliESDtrack *esdTrack = esdevent->GetTrack(itr);
+ if(!esdTrack || !IsAccepted(esdTrack)) continue;
+ if(esdTrack->Pt()<fCutMinTrackPt) continue;
+ const AliESDfriendTrack* friendTrack = esdTrack->GetFriendTrack();
+ if(friendTrack && friendTrack->GetTPCOut())
+ {
+ //Use TPC Out as starting point if it is available
+ trackParam= const_cast<AliExternalTrackParam*>(friendTrack->GetTPCOut());
+ }
+ else
+ {
+ //Otherwise use TPC inner
+ trackParam = const_cast<AliExternalTrackParam*>(esdTrack->GetInnerParam());
+ }
+ }
+
+ //If the input event is AOD, the starting point for extrapolation is at vertex
+ //AOD tracks are selected according to its bit.
+ else if(aodevent)
+ {
+ AliAODTrack *aodTrack = aodevent->GetTrack(itr);
+ if(!aodTrack) continue;
+ if(!aodTrack->TestFilterMask(fAODFilterMask)) continue; //Select AOD tracks that fulfill GetStandardITSTPCTrackCuts2010()
+ if(aodTrack->Pt()<fCutMinTrackPt) continue;
+ Double_t pos[3],mom[3];
+ aodTrack->GetXYZ(pos);
+ aodTrack->GetPxPyPz(mom);
+ AliDebug(5,Form("aod track: i=%d | pos=(%5.4f,%5.4f,%5.4f) | mom=(%5.4f,%5.4f,%5.4f) | charge=%d\n",itr,pos[0],pos[1],pos[2],mom[0],mom[1],mom[2],aodTrack->Charge()));
+ trackParam= new AliExternalTrackParam(pos,mom,cv,aodTrack->Charge());
+ }
+
+ //Return if the input data is not "AOD" or "ESD"
+ else
+ {
+ printf("Wrong input data type! Should be \"AOD\" or \"ESD\"\n");
+ return;
+ }
+
+ if(!trackParam) continue;
+
+ Float_t dRMax = fCutR, dEtaMax=fCutEta, dPhiMax=fCutPhi;
+ Int_t index = -1;
+ if(!clusterArr){// get clusters from event
+ for(Int_t icl=0; icl<event->GetNumberOfCaloClusters(); icl++)
+ {
+ AliVCluster *cluster = (AliVCluster*) event->GetCaloCluster(icl);
+ if(geom && !IsGoodCluster(cluster,geom,(AliVCaloCells*)event->GetEMCALCells())) continue;
+ AliExternalTrackParam trkPamTmp(*trackParam);//Retrieve the starting point every time before the extrapolation
+ Float_t tmpEta=-999, tmpPhi=-999;
+ if(!ExtrapolateTrackToCluster(&trkPamTmp, cluster, tmpEta, tmpPhi)) continue;
+ if(fCutEtaPhiSum)
+ {
+ Float_t tmpR=TMath::Sqrt(tmpEta*tmpEta + tmpPhi*tmpPhi);
+ if(tmpR<dRMax)
+ {
+ dRMax=tmpR;
+ dEtaMax=tmpEta;
+ dPhiMax=tmpPhi;
+ index=icl;
+ }
+ }
+ else if(fCutEtaPhiSeparate)
+ {
+ if(TMath::Abs(tmpEta)<TMath::Abs(dEtaMax) && TMath::Abs(tmpPhi)<TMath::Abs(dPhiMax))
+ {
+ dEtaMax = tmpEta;
+ dPhiMax = tmpPhi;
+ index=icl;
+ }
+ }
+ else
+ {
+ printf("Error: please specify your cut criteria\n");
+ printf("To cut on sqrt(dEta^2+dPhi^2), use: SwitchOnCutEtaPhiSum()\n");
+ printf("To cut on dEta and dPhi separately, use: SwitchOnCutEtaPhiSeparate()\n");
+ if(aodevent && trackParam) delete trackParam;
+ return;
+ }
+ }//cluster loop
+ }
+ else { // external cluster array, not from ESD event
+ for(Int_t icl=0; icl<clusterArr->GetEntriesFast(); icl++)
+ {
+ AliVCluster *cluster = dynamic_cast<AliVCluster*> (clusterArr->At(icl)) ;
+ if(!cluster){
+ AliInfo("Cluster not found!!!");
+ continue;
+ }
+ if(!cluster->IsEMCAL()) continue;
+ AliExternalTrackParam trkPamTmp (*trackParam);//Retrieve the starting point every time before the extrapolation
+ Float_t tmpEta=-999, tmpPhi=-999;
+ if(!ExtrapolateTrackToCluster(&trkPamTmp, cluster, tmpEta, tmpPhi)) continue;
+ if(fCutEtaPhiSum)
+ {
+ Float_t tmpR=TMath::Sqrt(tmpEta*tmpEta + tmpPhi*tmpPhi);
+ if(tmpR<dRMax)
+ {
+ dRMax=tmpR;
+ dEtaMax=tmpEta;
+ dPhiMax=tmpPhi;
+ index=icl;
+ }
+ }
+ else if(fCutEtaPhiSeparate)
+ {
+ if(TMath::Abs(tmpEta)<TMath::Abs(dEtaMax) && TMath::Abs(tmpPhi)<TMath::Abs(dPhiMax))
+ {
+ dEtaMax = tmpEta;
+ dPhiMax = tmpPhi;
+ index=icl;
+ }
+ }
+ else
+ {
+ printf("Error: please specify your cut criteria\n");
+ printf("To cut on sqrt(dEta^2+dPhi^2), use: SwitchOnCutEtaPhiSum()\n");
+ printf("To cut on dEta and dPhi separately, use: SwitchOnCutEtaPhiSeparate()\n");
+ if(aodevent && trackParam) delete trackParam;
+ return;
+ }
+ }//cluster loop
+ }// external list of clusters
if(index>-1)
{
+ fMatchedTrackIndex ->AddAt(itr,matched);
fMatchedClusterIndex->AddAt(index,matched);
- fResidualZ->AddAt(dZMax,matched);
- fResidualR->AddAt(dRMax,matched);
+ fResidualEta ->AddAt(dEtaMax,matched);
+ fResidualPhi ->AddAt(dPhiMax,matched);
matched++;
}
- delete emctrack;
+ if(aodevent && trackParam) delete trackParam;
}//track loop
+
+ AliDebug(2,Form("Number of matched pairs = %d !\n",matched));
+
+ fMatchedTrackIndex ->Set(matched);
fMatchedClusterIndex->Set(matched);
- fResidualZ->Set(matched);
- fResidualR->Set(matched);
+ fResidualPhi ->Set(matched);
+ fResidualEta ->Set(matched);
+}
+
+//________________________________________________________________________________
+Int_t AliEMCALRecoUtils::FindMatchedCluster(AliESDtrack *track, AliVEvent *event, AliEMCALGeometry *geom)
+{
+ //
+ // This function returns the index of matched cluster to input track
+ // Returns -1 if no match is found
+
- //printf("Number of matched pairs: %d\n",matched);
+ Float_t dRMax = fCutR, dEtaMax = fCutEta, dPhiMax = fCutPhi;
+ Int_t index = -1;
+
+ AliExternalTrackParam *trackParam=0;
+ const AliESDfriendTrack* friendTrack = track->GetFriendTrack();
+ if(friendTrack && friendTrack->GetTPCOut())
+ trackParam= const_cast<AliExternalTrackParam*>(friendTrack->GetTPCOut());
+ else
+ trackParam = const_cast<AliExternalTrackParam*>(track->GetInnerParam());
+
+ if(!trackParam) return index;
+ for(Int_t icl=0; icl<event->GetNumberOfCaloClusters(); icl++)
+ {
+ AliVCluster *cluster = (AliVCluster*) event->GetCaloCluster(icl);
+ if(geom && !IsGoodCluster(cluster,geom,(AliVCaloCells*)event->GetEMCALCells())) continue;
+ AliExternalTrackParam trkPamTmp (*trackParam);//Retrieve the starting point every time before the extrapolation
+ Float_t tmpEta=-999, tmpPhi=-999;
+ if(!ExtrapolateTrackToCluster(&trkPamTmp, cluster, tmpEta, tmpPhi)) continue;
+ if(fCutEtaPhiSum)
+ {
+ Float_t tmpR=TMath::Sqrt(tmpEta*tmpEta + tmpPhi*tmpPhi);
+ if(tmpR<dRMax)
+ {
+ dRMax=tmpR;
+ dEtaMax=tmpEta;
+ dPhiMax=tmpPhi;
+ index=icl;
+ }
+ }
+ else if(fCutEtaPhiSeparate)
+ {
+ if(TMath::Abs(tmpEta)<TMath::Abs(dEtaMax) && TMath::Abs(tmpPhi)<TMath::Abs(dPhiMax))
+ {
+ dEtaMax = tmpEta;
+ dPhiMax = tmpPhi;
+ index=icl;
+ }
+ }
+ else
+ {
+ printf("Error: please specify your cut criteria\n");
+ printf("To cut on sqrt(dEta^2+dPhi^2), use: SwitchOnCutEtaPhiSum()\n");
+ printf("To cut on dEta and dPhi separately, use: SwitchOnCutEtaPhiSeparate()\n");
+ return -1;
+ }
+ }//cluster loop
+ return index;
}
-//__________________________________________________
-void AliEMCALRecoUtils::GetMatchedResiduals(Int_t index, Float_t &dR, Float_t &dZ)
+//________________________________________________________________________________
+Bool_t AliEMCALRecoUtils::ExtrapolateTrackToCluster(AliExternalTrackParam *trkParam, AliVCluster *cluster, Float_t &tmpEta, Float_t &tmpPhi)
{
- //Given a cluster index as in AliESDEvent::GetCaloCluster(index)
- //Get the residuals dR and dZ for this cluster
- //It only works with ESDs, not AODs
+ //
+ //Return the residual by extrapolating a track to a cluster
+ //
+ if(!trkParam || !cluster) return kFALSE;
+ Float_t clsPos[3];
+ Double_t trkPos[3];
+ cluster->GetPosition(clsPos); //Has been recalculated
+ TVector3 vec(clsPos[0],clsPos[1],clsPos[2]);
+ Double_t alpha = ((int)(vec.Phi()*TMath::RadToDeg()/20)+0.5)*20*TMath::DegToRad();
+ vec.RotateZ(-alpha); //Rotate the cluster to the local extrapolation coordinate system
+ trkParam->Rotate(alpha); //Rotate the track to the same local extrapolation system
+ if(!AliTrackerBase::PropagateTrackToBxByBz(trkParam, vec.X(), fMass, fStep,kFALSE, 0.8, -1)) return kFALSE;
+ trkParam->GetXYZ(trkPos); //Get the extrapolated global position
+
+ TVector3 clsPosVec(clsPos[0],clsPos[1],clsPos[2]);
+ TVector3 trkPosVec(trkPos[0],trkPos[1],trkPos[2]);
+
+ Float_t clsPhi = (Float_t)clsPosVec.Phi();
+ if(clsPhi<0) clsPhi+=2*TMath::Pi();
+ Float_t trkPhi = (Float_t)trkPosVec.Phi();
+ if(trkPhi<0) trkPhi+=2*TMath::Pi();
+ tmpPhi = clsPhi-trkPhi; // track cluster matching
+ tmpEta = clsPosVec.Eta()-trkPosVec.Eta(); // track cluster matching
+
+ return kTRUE;
+}
- if( FindMatchedPos(index)==-1 )
+//________________________________________________________________________________
+void AliEMCALRecoUtils::GetMatchedResiduals(Int_t clsIndex, Float_t &dEta, Float_t &dPhi)
+{
+ //Given a cluster index as in AliESDEvent::GetCaloCluster(clsIndex)
+ //Get the residuals dEta and dPhi for this cluster to the closest track
+ //Works with ESDs and AODs
+
+ if( FindMatchedPosForCluster(clsIndex) >= 999 )
{
AliDebug(2,"No matched tracks found!\n");
- dR=999.;
- dZ=999.;
+ dEta=999.;
+ dPhi=999.;
+ return;
+ }
+ dEta = fResidualEta->At(FindMatchedPosForCluster(clsIndex));
+ dPhi = fResidualPhi->At(FindMatchedPosForCluster(clsIndex));
+}
+//________________________________________________________________________________
+void AliEMCALRecoUtils::GetMatchedClusterResiduals(Int_t trkIndex, Float_t &dEta, Float_t &dPhi)
+{
+ //Given a track index as in AliESDEvent::GetTrack(trkIndex)
+ //Get the residuals dEta and dPhi for this track to the closest cluster
+ //Works with ESDs and AODs
+
+ if( FindMatchedPosForTrack(trkIndex) >= 999 )
+ {
+ AliDebug(2,"No matched cluster found!\n");
+ dEta=999.;
+ dPhi=999.;
return;
}
- dR = fResidualR->At(FindMatchedPos(index));
- dZ = fResidualZ->At(FindMatchedPos(index));
+ dEta = fResidualEta->At(FindMatchedPosForTrack(trkIndex));
+ dPhi = fResidualPhi->At(FindMatchedPosForTrack(trkIndex));
+}
+
+//__________________________________________________________
+Int_t AliEMCALRecoUtils::GetMatchedTrackIndex(Int_t clsIndex)
+{
+ //Given a cluster index as in AliESDEvent::GetCaloCluster(clsIndex)
+ //Get the index of matched track to this cluster
+ //Works with ESDs and AODs
+
+ if(IsClusterMatched(clsIndex))
+ return fMatchedTrackIndex->At(FindMatchedPosForCluster(clsIndex));
+ else
+ return -1;
+}
+
+//__________________________________________________________
+Int_t AliEMCALRecoUtils::GetMatchedClusterIndex(Int_t trkIndex)
+{
+ //Given a track index as in AliESDEvent::GetTrack(trkIndex)
+ //Get the index of matched cluster to this track
+ //Works with ESDs and AODs
+
+ if(IsTrackMatched(trkIndex))
+ return fMatchedClusterIndex->At(FindMatchedPosForTrack(trkIndex));
+ else
+ return -1;
}
//__________________________________________________
-Bool_t AliEMCALRecoUtils::IsMatched(Int_t index)
+Bool_t AliEMCALRecoUtils::IsClusterMatched(Int_t clsIndex) const
{
- //Given a cluster index as in AliESDEvent::GetCaloCluster(index)
- //Returns if cluster has a match
- if(FindMatchedPos(index)>-1) return kTRUE;
+ //Given a cluster index as in AliESDEvent::GetCaloCluster(clsIndex)
+ //Returns if the cluster has a match
+ if(FindMatchedPosForCluster(clsIndex) < 999)
+ return kTRUE;
else
return kFALSE;
}
+
//__________________________________________________
-Int_t AliEMCALRecoUtils::FindMatchedPos(Int_t index) const
+Bool_t AliEMCALRecoUtils::IsTrackMatched(Int_t trkIndex) const
+{
+ //Given a track index as in AliESDEvent::GetTrack(trkIndex)
+ //Returns if the track has a match
+ if(FindMatchedPosForTrack(trkIndex) < 999)
+ return kTRUE;
+ else
+ return kFALSE;
+}
+
+//__________________________________________________________
+UInt_t AliEMCALRecoUtils::FindMatchedPosForCluster(Int_t clsIndex) const
{
- //Given a cluster index as in AliESDEvent::GetCaloCluster(index)
+ //Given a cluster index as in AliESDEvent::GetCaloCluster(clsIndex)
//Returns the position of the match in the fMatchedClusterIndex array
Float_t tmpR = fCutR;
- Int_t pos=-1;
-
+ UInt_t pos = 999;
+
for(Int_t i=0; i<fMatchedClusterIndex->GetSize(); i++)
- {
- if(fMatchedClusterIndex->At(i)==index && fResidualR->At(i)<tmpR)
- {
- pos=i;
- tmpR=fResidualR->At(i);
- }
- //printf("Matched cluster pos: %d, index: %d, dR: %2.4f, dZ: %2.4f.\n",i,fMatchedClusterIndex->At(i),fResidualR->At(i),fResidualZ->At(i));
- }
+ {
+ if(fMatchedClusterIndex->At(i)==clsIndex)
+ {
+ Float_t r = TMath::Sqrt(fResidualEta->At(i)*fResidualEta->At(i) + fResidualPhi->At(i)*fResidualPhi->At(i));
+ if(r<tmpR)
+ {
+ pos=i;
+ tmpR=r;
+ AliDebug(3,Form("Matched cluster index: index: %d, dEta: %2.4f, dPhi: %2.4f.\n",fMatchedClusterIndex->At(i),fResidualEta->At(i),fResidualPhi->At(i)));
+ }
+ }
+ }
return pos;
}
+//__________________________________________________________
+UInt_t AliEMCALRecoUtils::FindMatchedPosForTrack(Int_t trkIndex) const
+{
+ //Given a track index as in AliESDEvent::GetTrack(trkIndex)
+ //Returns the position of the match in the fMatchedTrackIndex array
+ Float_t tmpR = fCutR;
+ UInt_t pos = 999;
+
+ for(Int_t i=0; i<fMatchedTrackIndex->GetSize(); i++)
+ {
+ if(fMatchedTrackIndex->At(i)==trkIndex)
+ {
+ Float_t r = TMath::Sqrt(fResidualEta->At(i)*fResidualEta->At(i) + fResidualPhi->At(i)*fResidualPhi->At(i));
+ if(r<tmpR)
+ {
+ pos=i;
+ tmpR=r;
+ AliDebug(3,Form("Matched track index: index: %d, dEta: %2.4f, dPhi: %2.4f.\n",fMatchedTrackIndex->At(i),fResidualEta->At(i),fResidualPhi->At(i)));
+ }
+ }
+ }
+ return pos;
+}
+
+//__________________________________________________________
+Bool_t AliEMCALRecoUtils::IsGoodCluster(AliVCluster *cluster, AliEMCALGeometry *geom, AliVCaloCells* cells)
+{
+ // check if the cluster survives some quality cut
+ //
+ //
+ Bool_t isGood=kTRUE;
+ if(!cluster || !cluster->IsEMCAL()) return kFALSE;
+ if(ClusterContainsBadChannel(geom,cluster->GetCellsAbsId(),cluster->GetNCells())) return kFALSE;
+ if(!CheckCellFiducialRegion(geom,cluster,cells)) return kFALSE;
+ if(fRejectExoticCluster && IsExoticCluster(cluster)) return kFALSE;
+
+ return isGood;
+}
+
+//__________________________________________________________
Bool_t AliEMCALRecoUtils::IsAccepted(AliESDtrack *esdTrack)
{
// Given a esd track, return whether the track survive all the cuts
chi2PerClusterITS = esdTrack->GetITSchi2()/Float_t(nClustersITS);
if (nClustersTPC!=0)
chi2PerClusterTPC = esdTrack->GetTPCchi2()/Float_t(nClustersTPC);
-
+
+
+ //DCA cuts
+ Float_t maxDCAToVertexXYPtDep = 0.0182 + 0.0350/TMath::Power(esdTrack->Pt(),1.01); //This expression comes from AliESDtrackCuts::GetStandardITSTPCTrackCuts2010()
+ //AliDebug(3,Form("Track pT = %f, DCAtoVertexXY = %f",esdTrack->Pt(),MaxDCAToVertexXYPtDep));
+ SetMaxDCAToVertexXY(maxDCAToVertexXYPtDep); //Set pT dependent DCA cut to vertex in x-y plane
+
+
Float_t b[2];
Float_t bCov[3];
esdTrack->GetImpactParameters(b,bCov);
if (!fCutDCAToVertex2D && TMath::Abs(dcaToVertexZ) > fCutMaxDCAToVertexZ)
cuts[9] = kTRUE;
+ //Require at least one SPD point + anything else in ITS
+ if( (esdTrack->HasPointOnITSLayer(0) || esdTrack->HasPointOnITSLayer(1)) == kFALSE)
+ cuts[10] = kTRUE;
+
Bool_t cut=kFALSE;
for (Int_t i=0; i<kNCuts; i++)
if (cuts[i]) {cut = kTRUE;}
//Intilize the track cut criteria
//By default these cuts are set according to AliESDtrackCuts::GetStandardTPCOnlyTrackCuts()
//Also you can customize the cuts using the setters
-
- SetMinNClustersTPC(50);
- SetMaxChi2PerClusterTPC(4);
- SetAcceptKinkDaughters(kFALSE);
- SetMaxDCAToVertexZ(3.2);
- SetMaxDCAToVertexXY(2.4);
- SetDCAToVertex2D(kTRUE);
+
+ switch (fTrackCutsType)
+ {
+ case kTPCOnlyCut:
+ {
+ AliInfo(Form("Track cuts for matching: GetStandardTPCOnlyTrackCuts()\n"));
+ //TPC
+ SetMinNClustersTPC(70);
+ SetMaxChi2PerClusterTPC(4);
+ SetAcceptKinkDaughters(kFALSE);
+ SetRequireTPCRefit(kFALSE);
+
+ //ITS
+ SetRequireITSRefit(kFALSE);
+ SetMaxDCAToVertexZ(3.2);
+ SetMaxDCAToVertexXY(2.4);
+ SetDCAToVertex2D(kTRUE);
+
+ break;
+ }
+
+ case kGlobalCut:
+ {
+ AliInfo(Form("Track cuts for matching: GetStandardITSTPCTrackCuts2010(kTURE)\n"));
+ //TPC
+ SetMinNClustersTPC(70);
+ SetMaxChi2PerClusterTPC(4);
+ SetAcceptKinkDaughters(kFALSE);
+ SetRequireTPCRefit(kTRUE);
+
+ //ITS
+ SetRequireITSRefit(kTRUE);
+ SetMaxDCAToVertexZ(2);
+ SetMaxDCAToVertexXY();
+ SetDCAToVertex2D(kFALSE);
+
+ break;
+ }
+ }
}
-//__________________________________________________
+//___________________________________________________
void AliEMCALRecoUtils::Print(const Option_t *) const
{
// Print Parameters
printf("Position Recalculation option %d, Particle Type %d, fW0 %2.2f, Recalibrate Data %d \n",fPosAlgo,fParticleType,fW0, fRecalibration);
- printf("Matching criteria: dR < %2.2f[cm], dZ < %2.2f[cm]\n",fCutR,fCutZ);
+ printf("Matching criteria: ");
+ if(fCutEtaPhiSum)
+ {
+ printf("sqrt(dEta^2+dPhi^2)<%2.2f\n",fCutR);
+ }
+ else if(fCutEtaPhiSeparate)
+ {
+ printf("dEta<%2.2f, dPhi<%2.2f\n",fCutEta,fCutPhi);
+ }
+ else
+ {
+ printf("Error\n");
+ printf("please specify your cut criteria\n");
+ printf("To cut on sqrt(dEta^2+dPhi^2), use: SwitchOnCutEtaPhiSum()\n");
+ printf("To cut on dEta and dPhi separately, use: SwitchOnCutEtaPhiSeparate()\n");
+ }
+
+ printf("Mass hypothesis = %2.3f [GeV/c^2], extrapolation step = %2.2f[cm]\n",fMass,fStep);
printf("Track cuts: \n");
+ printf("Minimum track pT: %1.2f\n",fCutMinTrackPt);
+ printf("AOD track selection mask: %d\n",fAODFilterMask);
printf("TPCRefit = %d, ITSRefit = %d\n",fCutRequireTPCRefit,fCutRequireITSRefit);
printf("AcceptKinks = %d\n",fCutAcceptKinkDaughters);
printf("MinNCulsterTPC = %d, MinNClusterITS = %d\n",fCutMinNClusterTPC,fCutMinNClusterITS);
printf("MaxChi2TPC = %2.2f, MaxChi2ITS = %2.2f\n",fCutMaxChi2PerClusterTPC,fCutMaxChi2PerClusterITS);
printf("DCSToVertex2D = %d, MaxDCAToVertexXY = %2.2f, MaxDCAToVertexZ = %2.2f\n",fCutDCAToVertex2D,fCutMaxDCAToVertexXY,fCutMaxDCAToVertexZ);
-
}
+
+//_____________________________________________________________________
+void AliEMCALRecoUtils::SetTimeDependentCorrections(Int_t runnumber){
+ //Get EMCAL time dependent corrections from file and put them in the recalibration histograms
+ //Do it only once and only if it is requested
+
+ if(!fUseTimeCorrectionFactors) return;
+ if(fTimeCorrectionFactorsSet) return;
+
+ printf("AliEMCALRecoUtils::GetTimeDependentCorrections() - Get Correction Factors for Run number %d\n",runnumber);
+
+ AliEMCALCalibTimeDepCorrection *corr = new AliEMCALCalibTimeDepCorrection();
+ corr->ReadRootInfo(Form("CorrectionFiles/Run%d_Correction.root",runnumber));
+
+ SwitchOnRecalibration();
+ for(Int_t ism = 0; ism < 4; ism++){
+ for(Int_t icol = 0; icol < 48; icol++){
+ for(Int_t irow = 0; irow < 24; irow++){
+ Float_t orgRecalFactor = GetEMCALChannelRecalibrationFactors(ism)->GetBinContent(icol,irow);
+ Float_t newRecalFactor = orgRecalFactor*corr->GetCorrection(ism, icol,irow,0);
+ GetEMCALChannelRecalibrationFactors(ism)->SetBinContent(icol,irow,newRecalFactor);
+ //printf("ism %d, icol %d, irow %d, corrections : org %f, time dep %f, final %f (org*time %f)\n",ism, icol, irow,
+ // orgRecalFactor, corr->GetCorrection(ism, icol,irow,0),
+ // (GetEMCALChannelRecalibrationFactors(ism))->GetBinContent(icol,irow),newRecalFactor);
+ }
+ }
+ }
+ fTimeCorrectionFactorsSet = kTRUE;
+}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff