#ifndef ALIESDTRACKCUTS_H
#define ALIESDTRACKCUTS_H
-#include <TF1.h>
-#include <TH2.h>
#include "AliAnalysisCuts.h"
class AliESDEvent;
class AliESDtrack;
class AliLog;
class TTree;
+class TH1;
+class TH1F;
+class TH2F;
+class TF1;
+class TCollection;
class AliESDtrackCuts : public AliAnalysisCuts
{
public:
+ enum ITSClusterRequirement { kOff = 0, kNone, kAny, kFirst, kOnlyFirst, kSecond, kOnlySecond, kBoth };
+ enum Detector { kSPD = 0, kSDD, kSSD };
+
AliESDtrackCuts(const Char_t* name = "AliESDtrackCuts", const Char_t* title = "");
virtual ~AliESDtrackCuts();
Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
Bool_t AcceptTrack(AliESDtrack* esdTrack);
- TObjArray* GetAcceptedTracks(AliESDEvent* esd,Bool_t bTPC = kFALSE);
+ TObjArray* GetAcceptedTracks(AliESDEvent* esd, Bool_t bTPC = kFALSE);
Int_t CountAcceptedTracks(AliESDEvent* esd);
static AliESDtrack* GetTPCOnlyTrack(AliESDEvent* esd, Int_t iTrack);
// track quality cut setters
void SetMinNClustersTPC(Int_t min=-1) {fCutMinNClusterTPC=min;}
void SetMinNClustersITS(Int_t min=-1) {fCutMinNClusterITS=min;}
+ void SetClusterRequirementITS(Detector det, ITSClusterRequirement req = kOff) { fCutClusterRequirementITS[det] = req; }
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 SetAcceptKingDaughters(Bool_t b=kFALSE) {fCutAcceptKinkDaughters=b;}
+ void SetRequireITSStandAlone(Bool_t b) {fCutRequireITSStandAlone = b;}
+ void SetAcceptKinkDaughters(Bool_t b=kTRUE) {fCutAcceptKinkDaughters=b;}
void SetMaxCovDiagonalElements(Float_t c1=1e10, Float_t c2=1e10, Float_t c3=1e10, Float_t c4=1e10, Float_t c5=1e10)
{fCutMaxC11=c1; fCutMaxC22=c2; fCutMaxC33=c3; fCutMaxC44=c4; fCutMaxC55=c5;}
// track to vertex cut setters
- void SetMinNsigmaToVertex(Float_t sigma=1e10) {fCutNsigmaToVertex = sigma;}
- void SetRequireSigmaToVertex(Bool_t b=kTRUE ) {fCutSigmaToVertexRequired = b;}
- void SetDCAToVertex(Float_t dist=1e10) {fCutDCAToVertex = dist;}
- void SetDCAToVertexXY(Float_t dist=1e10) {fCutDCAToVertexXY = dist;}
+ void SetMaxNsigmaToVertex(Float_t sigma=1e10) {fCutNsigmaToVertex = sigma; SetRequireSigmaToVertex(kTRUE);}
+ void SetRequireSigmaToVertex(Bool_t b=kTRUE) {fCutSigmaToVertexRequired = b;}
+ void SetMaxDCAToVertexXY(Float_t dist=1e10) {fCutMaxDCAToVertexXY = dist;}
+ void SetMaxDCAToVertexZ(Float_t dist=1e10) {fCutMaxDCAToVertexZ = dist;}
+ void SetMinDCAToVertexXY(Float_t dist=0.) {fCutMinDCAToVertexXY = dist;}
+ void SetMinDCAToVertexZ(Float_t dist=0.) {fCutMinDCAToVertexZ = dist;}
+ void SetDCAToVertex2D(Bool_t b=kFALSE) {fCutDCAToVertex2D = b;}
+
+ // deprecated, will be removed in next release
+ void SetMinNsigmaToVertex(Float_t sigma=1e10);
+ Float_t GetMinNsigmaToVertex() const;
+ void SetAcceptKingDaughters(Bool_t b=kFALSE);
+ Bool_t GetAcceptKingDaughters() const;
// getters
Int_t GetMinNClusterTPC() const { return fCutMinNClusterTPC;}
Int_t GetMinNClustersITS() const { return fCutMinNClusterITS;}
+ ITSClusterRequirement GetClusterRequirementITS(Detector det) const { return fCutClusterRequirementITS[det]; }
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 GetAcceptKingDaughters() const { return fCutAcceptKinkDaughters;}
+ Bool_t GetRequireITSStandAlone() const { return fCutRequireITSStandAlone; }
+ Bool_t GetAcceptKinkDaughters() const { return fCutAcceptKinkDaughters;}
void GetMaxCovDiagonalElements(Float_t& c1, Float_t& c2, Float_t& c3, Float_t& c4, Float_t& c5)
{c1 = fCutMaxC11; c2 = fCutMaxC22; c3 = fCutMaxC33; c4 = fCutMaxC44; c5 = fCutMaxC55;}
- Float_t GetMinNsigmaToVertex() const { return fCutNsigmaToVertex;}
- Float_t GetDCAToVertex() const { return fCutDCAToVertex;}
- Float_t GetDCAToVertexXY() const { return fCutDCAToVertexXY;}
+ Float_t GetMaxNsigmaToVertex() const { return fCutNsigmaToVertex;}
+ Float_t GetMaxDCAToVertexXY() const { return fCutMaxDCAToVertexXY;}
+ Float_t GetMaxDCAToVertexZ() const { return fCutMaxDCAToVertexZ;}
+ Float_t GetMinDCAToVertexXY() const { return fCutMinDCAToVertexXY;}
+ Float_t GetMinDCAToVertexZ() const { return fCutMinDCAToVertexZ;}
+ Bool_t GetDCAToVertex2D() const { return fCutDCAToVertex2D;}
Bool_t GetRequireSigmaToVertex( ) const { return fCutSigmaToVertexRequired;}
void GetPRange(Float_t& r1, Float_t& r2) {r1=fPMin; r2=fPMax;}
void GetPzRange(Float_t& r1, Float_t& r2) {r1=fPzMin; r2=fPzMax;}
void GetEtaRange(Float_t& r1, Float_t& r2) {r1=fEtaMin; r2=fEtaMax;}
void GetRapRange(Float_t& r1, Float_t& r2) {r1=fRapMin; r2=fRapMax;}
-
+
// track kinmatic cut setters
void SetPRange(Float_t r1=0, Float_t r2=1e10) {fPMin=r1; fPMax=r2;}
void SetPtRange(Float_t r1=0, Float_t r2=1e10) {fPtMin=r1; fPtMax=r2;}
// void SaveQualityCuts(Char_t* file)
// void LoadQualityCuts(Char_t* file)
- TH1* GetDZNormalized(Int_t i) const { return fhDZNormalized[i]; }
+ TH1F* GetDZNormalized(Int_t i) const { return fhDZNormalized[i]; }
protected:
void Init(); // sets everything to 0
-
- enum { kNCuts = 23 };
+ Bool_t CheckITSClusterRequirement(ITSClusterRequirement req, Bool_t clusterL1, Bool_t clusterL2);
+
+ enum { kNCuts = 31 };
//######################################################
// esd track quality cuts
Int_t fCutMinNClusterTPC; // min number of tpc clusters
Int_t fCutMinNClusterITS; // min number of its clusters
+
+ ITSClusterRequirement fCutClusterRequirementITS[3]; // detailed ITS cluster requirements for (SPD, SDD, SSD)
Float_t fCutMaxChi2PerClusterTPC; // max tpc fit chi2 per tpc cluster
Float_t fCutMaxChi2PerClusterITS; // max its fit chi2 per its cluster
Bool_t fCutAcceptKinkDaughters; // accepting kink daughters?
Bool_t fCutRequireTPCRefit; // require TPC refit
Bool_t fCutRequireITSRefit; // require ITS refit
+ Bool_t fCutRequireITSStandAlone; // require ITS standalone tracks
// track to vertex cut
Float_t fCutNsigmaToVertex; // max number of estimated sigma from track-to-vertex
Bool_t fCutSigmaToVertexRequired; // cut track if sigma from track-to-vertex could not be calculated
- Float_t fCutDCAToVertex; // track-to-vertex cut in absolute distance
- Float_t fCutDCAToVertexXY; // track-to-vertex cut in absolute distance in xy-plane
+ 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
+ Float_t fCutMinDCAToVertexXY; // track-to-vertex cut on min absolute distance in xy-plane
+ Float_t fCutMinDCAToVertexZ; // track-to-vertex cut on min 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
// esd kinematics cuts
Float_t fPMin, fPMax; // definition of the range of the P
TH1F* fhDXY[2]; //->
TH1F* fhDZ[2]; //->
- TH1F* fhDXYDZ[2]; //-> absolute distance sqrt(dxy**2 + dz**2) to vertex
+ TH1F* fhDXYDZ[2]; //-> absolute distance sqrt(dxy**2 + dz**2) to vertex; if 2D cut is set, normalized to given values
TH2F* fhDXYvsDZ[2]; //->
TH1F* fhDXYNormalized[2]; //->
TH1F* fhCutStatistics; //-> statistics of what cuts the tracks did not survive
TH2F* fhCutCorrelation; //-> 2d statistics plot
- ClassDef(AliESDtrackCuts, 2)
+ ClassDef(AliESDtrackCuts, 6)
};
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff