[u/mrichter/AliRoot.git] / PWG2 / FLOW / AliFlowTasks / AliFlowTrackCuts.h

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. */
/* See cxx source for full Copyright notice */
/* $Id$ */

// AliFlowTrackESDCuts:
// A cut class for ESD, AOD and MC particles for the flow framework
// author: Mikolaj Krzewicki (mikolaj.krzewicki@cern.ch)

#ifndef ALIFLOWTRACKCUTS_H
#define ALIFLOWTRACKCUTS_H

#include <TMatrix.h>
#include <TList.h>
#include "AliFlowTrackSimpleCuts.h"
#include "AliESDtrackCuts.h"
#include "TMCProcess.h"
#include "AliESDtrack.h"
#include "AliPID.h"
#include "AliESDpid.h"

class TBrowser;
class AliVParticle;
class AliMCParticle;
class AliFlowTrack;
class AliMCEvent;
class AliVEvent;
class AliMultiplicity; 
class AliAODTrack;
class AliESDtrack;
class AliESDPmdTrack;
class AliESDVZERO;

class AliFlowTrackCuts : public AliFlowTrackSimpleCuts {

 public:
  AliFlowTrackCuts();
  AliFlowTrackCuts(const char* name);
  AliFlowTrackCuts(const AliFlowTrackCuts& someCuts);
  AliFlowTrackCuts& operator=(const AliFlowTrackCuts& someCuts);
  virtual ~AliFlowTrackCuts();

  static AliFlowTrackCuts* GetStandardTPCStandaloneTrackCuts();
  static AliFlowTrackCuts* GetStandardTPCStandaloneTrackCuts2010();
  static AliFlowTrackCuts* GetStandardGlobalTrackCuts2010();
  static AliFlowTrackCuts* GetStandardITSTPCTrackCuts2009(Bool_t selPrimaries=kTRUE);
  static AliFlowTrackCuts* GetStandardVZEROOnlyTrackCuts();

  Int_t Count(AliVEvent* event=NULL);

  enum trackParameterType { kMC, 
                            kGlobal, 
                            kTPCstandalone, 
                            kSPDtracklet,
                            kPMD,
                            kV0
                          };
  enum trackParameterMix  { kPure, 
                            kTrackWithMCkine, 
                            kTrackWithMCPID, 
                            kTrackWithMCpt, 
                            kTrackWithPtFromFirstMother,
                            kTrackWithTPCInnerParams
                          };
  enum PIDsource {
                   kTPCpid,      // default TPC pid (via GetTPCpid)
                   kTOFpid,      // default TOF pid (via GetTOFpid)
                   kTOFbayesian, // TOF bayesian pid (F.Noferini)
                   kTOFbeta,     // asymmetric cuts of TOF beta signal
                   kTPCdedx,      // asymmetric cuts of TPC dedx signal
                   kTOFbetaSimple, //simple TOF only cut
                   kTPCbayesian //bayesian cutTPC
                 };

  //setters (interface to AliESDtrackCuts)
  void SetMinNClustersTPC( Int_t a ) {fCutNClustersTPC=kTRUE; fNClustersTPCMin=a;}
  void SetMinNClustersITS( Int_t a ) {fCutNClustersITS=kTRUE; fNClustersITSMin=a;}
  void SetClusterRequirementITS( AliESDtrackCuts::Detector det,
                                 AliESDtrackCuts::ITSClusterRequirement req = AliESDtrackCuts::kOff )
                                 { InitESDcuts(); fAliESDtrackCuts->SetClusterRequirementITS(det,req); } 
  void SetMaxChi2PerClusterTPC( Float_t a ) {fMaxChi2PerClusterTPC=a;fCutChi2PerClusterTPC=kTRUE;}
  void SetMinChi2PerClusterTPC( Float_t a ) {fMinChi2PerClusterTPC=a;fCutChi2PerClusterTPC=kTRUE;}
  void SetMaxChi2PerClusterITS( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxChi2PerClusterITS(a);}
  void SetRequireTPCRefit( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetRequireTPCRefit(a);}
  void SetRequireTPCStandAlone( Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireTPCStandAlone(a);}
  void SetRequireITSRefit( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetRequireITSRefit(a);}
  void SetRequireITSStandAlone( Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireITSStandAlone(a);}
  void SetAcceptKinkDaughters( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetAcceptKinkDaughters(a);}
  void SetMaxDCAToVertexZ( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexZ(a);fCutDCAToVertexZ=kTRUE;}
  void SetMaxDCAToVertexXY( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexXY(a);fCutDCAToVertexXY=kTRUE;}
  void SetMaxDCAToVertexXYPtDep( const char* a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexXYPtDep(a);}
  void SetRequireSigmaToVertex(Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireSigmaToVertex(a);}
  void SetMaxNsigmaToVertex(Float_t sigma=1e10) {InitESDcuts(); fAliESDtrackCuts->SetMaxNsigmaToVertex(sigma); }
  void SetDCAToVertex2D( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetDCAToVertex2D(a);}
  void SetEtaRange( Float_t r1, Float_t r2 ) { SetEtaMin(r1); SetEtaMax(r2); }
  void SetPtRange( Float_t r1, Float_t r2 ) { SetPtMin(r1); SetPtMax(r2); }
  void SetRequireCharge( Bool_t r ) {fRequireCharge=r;}
  void SetFakesAreOK( Bool_t b ) {fFakesAreOK=b;}
  void SetSPDtrackletDeltaPhiMax( Double_t m ) {fSPDtrackletDeltaPhiMax=m; fCutSPDtrackletDeltaPhi=kTRUE;}
  void SetSPDtrackletDeltaPhiMin( Double_t m ) {fSPDtrackletDeltaPhiMin=m; fCutSPDtrackletDeltaPhi=kTRUE;}
  void SetIgnoreTPCzRange( Double_t min, Double_t max ) 
                         { fIgnoreTPCzRange=kTRUE; fIgnoreTPCzRangeMin=min; fIgnoreTPCzRangeMax=max; }
  void SetAODfilterBit( UInt_t a ) {fAODFilterBit = a; fUseAODFilterBit = kTRUE;}  						 
  void SetMinimalTPCdedx(Double_t d=10.) {fMinimalTPCdedx=d; fCutMinimalTPCdedx=kTRUE;}
  void SetPmdDetPlane(Int_t pmdDet){fCutPmdDet=kTRUE; fPmdDet = pmdDet; }
  void SetPmdAdc(Float_t pmdAdc){fCutPmdAdc=kTRUE; fPmdAdc = pmdAdc; }
  void SetPmdNcell(Float_t pmdNcell) {fCutPmdNcell=kTRUE; fPmdNcell = pmdNcell; }						 
  void SetPriors(Float_t centr = 0); // set my favourite priors for Bayesian PID

  Int_t GetMinNClustersTPC() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClusterTPC();}
  Int_t GetMinNClustersITS() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClustersITS();}
  AliESDtrackCuts::ITSClusterRequirement GetClusterRequirementITS( AliESDtrackCuts::Detector det ) const
                                 {if (!fAliESDtrackCuts) return AliESDtrackCuts::kOff;  return fAliESDtrackCuts->GetClusterRequirementITS(det); } 
  Float_t GetMaxChi2PerClusterTPC() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxChi2PerClusterTPC();}
  Float_t GetMaxChi2PerClusterITS() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxChi2PerClusterITS();}
  Bool_t GetRequireTPCRefit() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireTPCRefit();}
  Bool_t GetRequireTPCStandAlone() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireTPCStandAlone();}
  Bool_t GetRequireITSRefit() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireITSRefit();}
  Bool_t GetRequireITSStandAlone() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireITSStandAlone();}
  Bool_t GetAcceptKinkDaughters() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetAcceptKinkDaughters();}
  Float_t GetMaxDCAToVertexZ() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexZ();}
  Float_t GetMaxDCAToVertexXY() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexXY();}
  const char* GetMaxDCAToVertexXYPtDep() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexXYPtDep();}
  Bool_t GetRequireSigmaToVertex() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireSigmaToVertex();}
  Float_t GetMaxNsigmaToVertex() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxNsigmaToVertex(); }
  Bool_t GetDCAToVertex2D() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetDCAToVertex2D();}
  void GetEtaRange( Float_t& r1, Float_t& r2 ) const { r1=GetEtaMin(); r2=GetEtaMax(); }
  void GetPtRange( Float_t& r1, Float_t& r2 ) const { r1=GetPtMin(); r2=GetPtMax(); }
  Bool_t GetRequireCharge() const {return fRequireCharge;}
  Bool_t GetFakesAreOK() const {return fFakesAreOK;}
  Double_t GetSPDtrackletDeltaPhiMax() const {return fSPDtrackletDeltaPhiMax;}
  Double_t GetSPDtrackletDeltaPhiMin() const {return fSPDtrackletDeltaPhiMin;}
  UInt_t GetAODFilterBit() const {if (!fUseAODFilterBit) return 0; return fAODFilterBit;}
  Double_t GetMinimalTPCdedx() const {return fMinimalTPCdedx;}
  Int_t GetPmdDetPlane()const {return fPmdDet; }
  Float_t GetPmdAdc()const {return fPmdAdc;}
  Float_t GetPmdNcell() const {return fPmdNcell; }
  Float_t GetBeta(const AliESDtrack* t);
  Float_t Getdedx(const AliESDtrack* t);
 
  void SetQA(Bool_t b=kTRUE) {if (b) DefineHistograms();}
  TList* GetQA() const {return fQA;}
  TH1* QAbefore(Int_t i) {return static_cast<TH1*>(static_cast<TList*>(fQA->At(0))->At(i));}
  TH1* QAafter(Int_t i) {return static_cast<TH1*>(static_cast<TList*>(fQA->At(1))->At(i));}  

  //MC stuff
  void SetIgnoreSignInMCPID( Bool_t b ) {fIgnoreSignInMCPID=b;}
  void SetCutMC( Bool_t b=kTRUE );
  void SetCutMChasTrackReferences(Bool_t b=kTRUE) {fCutMChasTrackReferences=b;}
  void SetMCprocessType( TMCProcess t ) { fMCprocessType = t; fCutMCprocessType=kTRUE; SetCutMC();}
  void SetMCisPrimary( Bool_t b ) { fMCisPrimary=b; fCutMCisPrimary=kTRUE; SetCutMC();}
  void SetMCPID( Int_t pid ) { fMCPID=pid; fCutMCPID=kTRUE; SetCutMC(); }
  TMCProcess GetMCprocessType() const { return fMCprocessType; }
  Bool_t GetMCisPrimary() const {return fMCisPrimary;}
  Int_t GetMCPID() const {return fMCPID;}
  void SetRequireTransportBitForPrimaries(Bool_t b) {fRequireTransportBitForPrimaries=b; SetCutMC();}

  void SetParamType(trackParameterType paramType) {fParamType=paramType;}
  trackParameterType GetParamType() const {return fParamType;}
  static const char* GetParamTypeName(trackParameterType type);
  void SetParamMix(trackParameterMix paramMix) {fParamMix=paramMix;}
  trackParameterMix GetParamMix() const {return fParamMix;}

  virtual Bool_t IsSelected(TObject* obj, Int_t id=-666);
  virtual Bool_t IsSelectedMCtruth(TObject* obj, Int_t id=-666);
  AliVParticle* GetTrack() const {return fTrack;}
  AliMCParticle* GetMCparticle() const {return fMCparticle;}
  AliFlowTrack* MakeFlowTrack() const;
  Bool_t FillFlowTrack(AliFlowTrack* track) const;
  Bool_t IsPhysicalPrimary() const; 
  static Bool_t IsPhysicalPrimary(AliMCEvent* p, Int_t label, Bool_t requiretransported=kTRUE); 
  
  void SetMCevent(AliMCEvent* mcEvent) {fMCevent=mcEvent;}
  AliMCEvent* GetMCevent() const {return fMCevent;}
  void SetEvent(AliVEvent* event, AliMCEvent* mcEvent=NULL);
  AliVEvent* GetEvent() const {return fEvent;}
  Int_t GetNumberOfInputObjects() const;
  TObject* GetInputObject(Int_t i);
  void Clear(Option_t* option="");

  Double_t GetPmdEta(Float_t xPos, Float_t yPos, Float_t zPos);
  Double_t GetPmdPhi(Float_t xPos, Float_t yPos);  

  //PID
  void SetPID(AliPID::EParticleType pid, PIDsource s=kTOFpid, Double_t prob=0.9)
             {fParticleID=pid; fPIDsource=s; fParticleProbability=prob; fCutPID=kTRUE; InitPIDcuts();}
  void SetTPCpidCuts(TMatrixF* mat) {fTPCpidCuts=new TMatrixF(*mat);}
  void SetTOFpidCuts(TMatrixF* mat) {fTOFpidCuts=new TMatrixF(*mat);}
  static const char* PIDsourceName(PIDsource s);
  AliESDpid& GetESDpid() {return fESDpid;}
  void SetAllowTOFmismatchFlag(Bool_t b=kTRUE) {fAllowTOFmismatchFlag=b;}
  Bool_t GetAllowTOFmismatchFlag() const {return fAllowTOFmismatchFlag;}
  void SetRequireStrictTOFTPCagreement(Bool_t b=kTRUE) {fRequireStrictTOFTPCagreement=b;}
  Bool_t GetRequireStrictTOFTPCagreement() const {return fRequireStrictTOFTPCagreement;}
  void SetRejectElectronsWithTPCpid(Bool_t b=kTRUE) {fCutRejectElectronsWithTPCpid=b;}

  //these should maybe be protected
  Bool_t PassesCuts(AliVParticle* track);
  Bool_t PassesESDcuts(AliESDtrack* track);
  Bool_t PassesAODcuts(const AliAODTrack* track);
  Bool_t PassesPMDcuts(AliESDPmdTrack* track);
  Bool_t PassesV0cuts(AliESDVZERO* track, Int_t id);
  Bool_t PassesCuts(const AliFlowTrackSimple* track);
  Bool_t PassesCuts(const AliMultiplicity* track, Int_t id);
  Bool_t PassesMCcuts();
  Bool_t PassesMCcuts(AliMCEvent* mcevent, Int_t label);
  Bool_t PassesTPCdedxCut(const AliESDtrack* track);
  Bool_t PassesTPCbayesianCut(const AliESDtrack* track);
  Bool_t PassesTPCpidCut(const AliESDtrack* track) const;
  Bool_t PassesTOFbetaCut(const AliESDtrack* track);  
  Bool_t PassesTOFbetaSimpleCut(const AliESDtrack* track);  
  Bool_t PassesTOFpidCut(const AliESDtrack* track) const;  

  void Browse(TBrowser* b);
  Long64_t Merge(TCollection* list);

 protected:
  AliFlowTrack* MakeFlowTrackSPDtracklet() const;
  AliFlowTrack* MakeFlowTrackPMDtrack() const;
  AliFlowTrack* MakeFlowTrackV0() const;
  AliFlowTrack* MakeFlowTrackVParticle() const;
  Bool_t FillFlowTrackVParticle(AliFlowTrack* t) const;
  Bool_t FillFlowTrackGeneric(AliFlowTrack* t) const;
  void HandleESDtrack(AliESDtrack* track);
  void HandleVParticle(AliVParticle* track);
  void DefineHistograms();
  void InitPIDcuts();
  void InitESDcuts() {if (!fAliESDtrackCuts) {fAliESDtrackCuts=new AliESDtrackCuts();}}
  // part added by F. Noferini
  Bool_t PassesTOFbayesianCut(const AliESDtrack* track); 
  Int_t GetESDPdg(const AliESDtrack *track,Option_t *option="bayesianTOF",Int_t ipart=2,Float_t cPi=-1.0,Float_t cKa=0.0,Float_t cPr=0.0); // 3sigma cut ipart=0(el),1(mu),2(pi),3(K),4(p)
  Bool_t TPCTOFagree(const AliESDtrack *track);
  // end part added by F. Noferini

  //the cuts
  AliESDtrackCuts* fAliESDtrackCuts; //alianalysis cuts
  TList* fQA;                        //qa histograms go here
  Bool_t fCutMC;                     //do we cut on MC?
  Bool_t fCutMChasTrackReferences;   //did we leave a trace in the detector?
  Bool_t fCutMCprocessType;          //do we cut on mc process type?
  TMCProcess fMCprocessType;         //mc process type
  Bool_t fCutMCPID;                  //cut on MC pid?
  Int_t fMCPID;                      //MC PID
  Bool_t fIgnoreSignInMCPID;           //when MC PID cut is set, pass also the antiparticle
  Bool_t fCutMCisPrimary;            //do we cut on primaryness?
  Bool_t fRequireTransportBitForPrimaries; //require the transport bit to be set for primaries
  Bool_t fMCisPrimary;               //is MC primary
  Bool_t fRequireCharge;          //is charged?
  Bool_t fFakesAreOK;             //are fakes (negative labels) ok?
  Bool_t fCutSPDtrackletDeltaPhi; //are we cutting on the trcklet deltaphi?
  Double_t fSPDtrackletDeltaPhiMax; //maximal deltaphi for tracklets
  Double_t fSPDtrackletDeltaPhiMin; //minimal deltaphi for tracklets
  Bool_t fIgnoreTPCzRange;   //ignore tracks going close to central membrane
  Double_t fIgnoreTPCzRangeMax; //max z to ignore
  Double_t fIgnoreTPCzRangeMin; //min z to ignore
  Bool_t fCutChi2PerClusterTPC; //cut on tpc chi2
  Float_t fMaxChi2PerClusterTPC; //max chi2 tpc/cluster
  Float_t fMinChi2PerClusterTPC; //min chi2 tpc/cluster
  Bool_t fCutNClustersTPC;       //cut on clusters?
  Int_t fNClustersTPCMax;        //max tpc ncls
  Int_t fNClustersTPCMin;        //min tpc clusters  
  Bool_t fCutNClustersITS;       //cut on clusters?
  Int_t fNClustersITSMax;        //max tpc ncls
  Int_t fNClustersITSMin;        //min tpc clusters  
  Bool_t fUseAODFilterBit;       //use AOD filter bit selection?
  UInt_t fAODFilterBit;          //AOD filter bit to select
  Bool_t fCutDCAToVertexXY;      //dca xy cut
  Bool_t fCutDCAToVertexZ;       //dca z cut
  Bool_t fCutMinimalTPCdedx;    //cut on minimal dedx in TPC to reject noise tracks
  Double_t fMinimalTPCdedx;       //value for minimal TPC dedx
  
  Bool_t  fCutPmdDet;   //cut on PMD detector plane 
  Int_t   fPmdDet;      // value of PMD detector plane
  Bool_t  fCutPmdAdc;   //cut on cluster ADC
  Float_t fPmdAdc;      //value of cluster ADC
  Bool_t  fCutPmdNcell; //cut on cluster ncell
  Float_t fPmdNcell;    //value of cluster ncell
   
  trackParameterType fParamType;     //parameter type tu cut on
  trackParameterMix fParamMix;       //parameter mixing
  AliVParticle* fTrack;              //!the track to apply cuts on
  Double_t fTrackPhi;                //!track phi
  Double_t fTrackEta;                //!track eta
  Double_t fTrackWeight;             //!track weight
  Int_t fTrackLabel;                 //!track label, or its absolute value if FakesAreOK
  AliMCEvent* fMCevent;              //!mc event
  AliMCParticle* fMCparticle;        //!mc particle
  AliVEvent* fEvent;                 //!placeholder for current event
  AliESDtrack fTPCtrack;             //!placeholder for TPC only track to avoid new/delete on every track

  //PID
  AliESDpid fESDpid; //pid obj
  PIDsource fPIDsource; //pid source
  TMatrixF* fTPCpidCuts; //tpc pid cuts
  TMatrixF* fTOFpidCuts; //tof pid cuts
  AliPID::EParticleType fParticleID; //alipid
  Double_t fParticleProbability; //desired prob for a particle type
  Bool_t fAllowTOFmismatchFlag; //allow TOFmismatch flag=1 in ESD
  Bool_t fRequireStrictTOFTPCagreement; //require stricter than TOFmismatch flag TOF-TPC agreement
  Bool_t fCutRejectElectronsWithTPCpid; //reject electrons with TPC pid

  // part added by F. Noferini
  static const Int_t fgkPIDptBin = 20; // pT bins for priors
  Float_t fC[fgkPIDptBin][5],fBinLimitPID[fgkPIDptBin]; // pt bin limit and priors
  Float_t fProbBayes[5]; // bayesian probability
  // end part added by F. Noferini

  static const Int_t fgkNumberOfV0tracks=64; //number of V0 channels

  ClassDef(AliFlowTrackCuts,8)
};

#endif
Commit	Line	Data
daf66719	1	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. */
	2	/* See cxx source for full Copyright notice */
	3	/* $Id$ */
	4
	5	// AliFlowTrackESDCuts:
	6	// A cut class for ESD, AOD and MC particles for the flow framework
	7	// author: Mikolaj Krzewicki (mikolaj.krzewicki@cern.ch)
	8
	9	#ifndef ALIFLOWTRACKCUTS_H
	10	#define ALIFLOWTRACKCUTS_H
	11
32b846cd	12	#include <TMatrix.h>
499fe731	13	#include <TList.h>
daf66719	14	#include "AliFlowTrackSimpleCuts.h"
	15	#include "AliESDtrackCuts.h"
	16	#include "TMCProcess.h"
1ff4bda1	17	#include "AliESDtrack.h"
32b846cd	18	#include "AliPID.h"
aab6527a	19	#include "AliESDpid.h"
daf66719	20
499fe731	21	class TBrowser;
daf66719	22	class AliVParticle;
	23	class AliMCParticle;
	24	class AliFlowTrack;
	25	class AliMCEvent;
	26	class AliVEvent;
12b2b8bc	27	class AliMultiplicity;
42655d16	28	class AliAODTrack;
42655d16	29	class AliESDtrack;
d148af7e	30	class AliESDPmdTrack;
22289738	31	class AliESDVZERO;
daf66719	32
	33	class AliFlowTrackCuts : public AliFlowTrackSimpleCuts {
	34
	35	public:
	36	AliFlowTrackCuts();
441ea1cf	37	AliFlowTrackCuts(const char* name);
daf66719	38	AliFlowTrackCuts(const AliFlowTrackCuts& someCuts);
	39	AliFlowTrackCuts& operator=(const AliFlowTrackCuts& someCuts);
	40	virtual ~AliFlowTrackCuts();
	41
1a80f9f6	42	static AliFlowTrackCuts* GetStandardTPCStandaloneTrackCuts();
1a80f9f6	43	static AliFlowTrackCuts* GetStandardTPCStandaloneTrackCuts2010();
a0241c3a	44	static AliFlowTrackCuts* GetStandardGlobalTrackCuts2010();
daf66719	45	static AliFlowTrackCuts* GetStandardITSTPCTrackCuts2009(Bool_t selPrimaries=kTRUE);
1a80f9f6	46	static AliFlowTrackCuts* GetStandardVZEROOnlyTrackCuts();
daf66719	47
a0241c3a	48	Int_t Count(AliVEvent* event=NULL);
a0241c3a	49
2b1eaa10	50	enum trackParameterType { kMC,
2b1eaa10	51	kGlobal,
1a80f9f6	52	kTPCstandalone,
1a80f9f6	53	kSPDtracklet,
22289738	54	kPMD,
22289738	55	kV0
2b1eaa10	56	};
	57	enum trackParameterMix { kPure,
	58	kTrackWithMCkine,
	59	kTrackWithMCPID,
	60	kTrackWithMCpt,
d79d61fb	61	kTrackWithPtFromFirstMother,
d79d61fb	62	kTrackWithTPCInnerParams
2b1eaa10	63	};
	64	enum PIDsource {
	65	kTPCpid, // default TPC pid (via GetTPCpid)
	66	kTOFpid, // default TOF pid (via GetTOFpid)
	67	kTOFbayesian, // TOF bayesian pid (F.Noferini)
	68	kTOFbeta, // asymmetric cuts of TOF beta signal
1a80f9f6	69	kTPCdedx, // asymmetric cuts of TPC dedx signal
1a80f9f6	70	kTOFbetaSimple, //simple TOF only cut
cd62a2a8	71	kTPCbayesian //bayesian cutTPC
2b1eaa10	72	};
daf66719	73
daf66719	74	//setters (interface to AliESDtrackCuts)
42655d16	75	void SetMinNClustersTPC( Int_t a ) {fCutNClustersTPC=kTRUE; fNClustersTPCMin=a;}
42655d16	76	void SetMinNClustersITS( Int_t a ) {fCutNClustersITS=kTRUE; fNClustersITSMin=a;}
daf66719	77	void SetClusterRequirementITS( AliESDtrackCuts::Detector det,
daf66719	78	AliESDtrackCuts::ITSClusterRequirement req = AliESDtrackCuts::kOff )
441ea1cf	79	{ InitESDcuts(); fAliESDtrackCuts->SetClusterRequirementITS(det,req); }
2948ac5a	80	void SetMaxChi2PerClusterTPC( Float_t a ) {fMaxChi2PerClusterTPC=a;fCutChi2PerClusterTPC=kTRUE;}
2948ac5a	81	void SetMinChi2PerClusterTPC( Float_t a ) {fMinChi2PerClusterTPC=a;fCutChi2PerClusterTPC=kTRUE;}
441ea1cf	82	void SetMaxChi2PerClusterITS( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxChi2PerClusterITS(a);}
	83	void SetRequireTPCRefit( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetRequireTPCRefit(a);}
	84	void SetRequireTPCStandAlone( Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireTPCStandAlone(a);}
	85	void SetRequireITSRefit( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetRequireITSRefit(a);}
	86	void SetRequireITSStandAlone( Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireITSStandAlone(a);}
	87	void SetAcceptKinkDaughters( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetAcceptKinkDaughters(a);}
a63303bd	88	void SetMaxDCAToVertexZ( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexZ(a);fCutDCAToVertexZ=kTRUE;}
a63303bd	89	void SetMaxDCAToVertexXY( Float_t a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexXY(a);fCutDCAToVertexXY=kTRUE;}
441ea1cf	90	void SetMaxDCAToVertexXYPtDep( const char* a ) {InitESDcuts(); fAliESDtrackCuts->SetMaxDCAToVertexXYPtDep(a);}
	91	void SetRequireSigmaToVertex(Bool_t a) {InitESDcuts(); fAliESDtrackCuts->SetRequireSigmaToVertex(a);}
	92	void SetMaxNsigmaToVertex(Float_t sigma=1e10) {InitESDcuts(); fAliESDtrackCuts->SetMaxNsigmaToVertex(sigma); }
	93	void SetDCAToVertex2D( Bool_t a ) {InitESDcuts(); fAliESDtrackCuts->SetDCAToVertex2D(a);}
daf66719	94	void SetEtaRange( Float_t r1, Float_t r2 ) { SetEtaMin(r1); SetEtaMax(r2); }
daf66719	95	void SetPtRange( Float_t r1, Float_t r2 ) { SetPtMin(r1); SetPtMax(r2); }
127a5825	96	void SetRequireCharge( Bool_t r ) {fRequireCharge=r;}
127a5825	97	void SetFakesAreOK( Bool_t b ) {fFakesAreOK=b;}
9a0783cc	98	void SetSPDtrackletDeltaPhiMax( Double_t m ) {fSPDtrackletDeltaPhiMax=m; fCutSPDtrackletDeltaPhi=kTRUE;}
9a0783cc	99	void SetSPDtrackletDeltaPhiMin( Double_t m ) {fSPDtrackletDeltaPhiMin=m; fCutSPDtrackletDeltaPhi=kTRUE;}
1ff4bda1	100	void SetIgnoreTPCzRange( Double_t min, Double_t max )
5ba02b32	101	{ fIgnoreTPCzRange=kTRUE; fIgnoreTPCzRangeMin=min; fIgnoreTPCzRangeMax=max; }
5ba02b32	102	void SetAODfilterBit( UInt_t a ) {fAODFilterBit = a; fUseAODFilterBit = kTRUE;}
2b1eaa10	103	void SetMinimalTPCdedx(Double_t d=10.) {fMinimalTPCdedx=d; fCutMinimalTPCdedx=kTRUE;}
1a80f9f6	104	void SetPmdDetPlane(Int_t pmdDet){fCutPmdDet=kTRUE; fPmdDet = pmdDet; }
	105	void SetPmdAdc(Float_t pmdAdc){fCutPmdAdc=kTRUE; fPmdAdc = pmdAdc; }
	106	void SetPmdNcell(Float_t pmdNcell) {fCutPmdNcell=kTRUE; fPmdNcell = pmdNcell; }
875ee12a	107	void SetPriors(Float_t centr = 0); // set my favourite priors for Bayesian PID
daf66719	108
441ea1cf	109	Int_t GetMinNClustersTPC() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClusterTPC();}
441ea1cf	110	Int_t GetMinNClustersITS() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMinNClustersITS();}
daf66719	111	AliESDtrackCuts::ITSClusterRequirement GetClusterRequirementITS( AliESDtrackCuts::Detector det ) const
441ea1cf	112	{if (!fAliESDtrackCuts) return AliESDtrackCuts::kOff; return fAliESDtrackCuts->GetClusterRequirementITS(det); }
	113	Float_t GetMaxChi2PerClusterTPC() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxChi2PerClusterTPC();}
	114	Float_t GetMaxChi2PerClusterITS() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxChi2PerClusterITS();}
	115	Bool_t GetRequireTPCRefit() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireTPCRefit();}
	116	Bool_t GetRequireTPCStandAlone() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireTPCStandAlone();}
	117	Bool_t GetRequireITSRefit() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireITSRefit();}
	118	Bool_t GetRequireITSStandAlone() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireITSStandAlone();}
	119	Bool_t GetAcceptKinkDaughters() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetAcceptKinkDaughters();}
	120	Float_t GetMaxDCAToVertexZ() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexZ();}
	121	Float_t GetMaxDCAToVertexXY() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexXY();}
	122	const char* GetMaxDCAToVertexXYPtDep() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxDCAToVertexXYPtDep();}
	123	Bool_t GetRequireSigmaToVertex() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetRequireSigmaToVertex();}
	124	Float_t GetMaxNsigmaToVertex() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetMaxNsigmaToVertex(); }
	125	Bool_t GetDCAToVertex2D() const {if (!fAliESDtrackCuts) return 0; return fAliESDtrackCuts->GetDCAToVertex2D();}
daf66719	126	void GetEtaRange( Float_t& r1, Float_t& r2 ) const { r1=GetEtaMin(); r2=GetEtaMax(); }
daf66719	127	void GetPtRange( Float_t& r1, Float_t& r2 ) const { r1=GetPtMin(); r2=GetPtMax(); }
957517fa	128	Bool_t GetRequireCharge() const {return fRequireCharge;}
9a0783cc	129	Bool_t GetFakesAreOK() const {return fFakesAreOK;}
	130	Double_t GetSPDtrackletDeltaPhiMax() const {return fSPDtrackletDeltaPhiMax;}
	131	Double_t GetSPDtrackletDeltaPhiMin() const {return fSPDtrackletDeltaPhiMin;}
5ba02b32	132	UInt_t GetAODFilterBit() const {if (!fUseAODFilterBit) return 0; return fAODFilterBit;}
2b1eaa10	133	Double_t GetMinimalTPCdedx() const {return fMinimalTPCdedx;}
1a80f9f6	134	Int_t GetPmdDetPlane()const {return fPmdDet; }
1a80f9f6	135	Float_t GetPmdAdc()const {return fPmdAdc;}
499fe731	136	Float_t GetPmdNcell() const {return fPmdNcell; }
	137	Float_t GetBeta(const AliESDtrack* t);
	138	Float_t Getdedx(const AliESDtrack* t);
5ba02b32	139
a14b8f3c	140	void SetQA(Bool_t b=kTRUE) {if (b) DefineHistograms();}
499fe731	141	TList* GetQA() const {return fQA;}
	142	TH1* QAbefore(Int_t i) {return static_cast<TH1>(static_cast<TList>(fQA->At(0))->At(i));}
	143	TH1* QAafter(Int_t i) {return static_cast<TH1>(static_cast<TList>(fQA->At(1))->At(i));}
924b02b0	144
daf66719	145	//MC stuff
a14b8f3c	146	void SetIgnoreSignInMCPID( Bool_t b ) {fIgnoreSignInMCPID=b;}
aab6527a	147	void SetCutMC( Bool_t b=kTRUE );
f21bdf48	148	void SetCutMChasTrackReferences(Bool_t b=kTRUE) {fCutMChasTrackReferences=b;}
aab6527a	149	void SetMCprocessType( TMCProcess t ) { fMCprocessType = t; fCutMCprocessType=kTRUE; SetCutMC();}
	150	void SetMCisPrimary( Bool_t b ) { fMCisPrimary=b; fCutMCisPrimary=kTRUE; SetCutMC();}
	151	void SetMCPID( Int_t pid ) { fMCPID=pid; fCutMCPID=kTRUE; SetCutMC(); }
daf66719	152	TMCProcess GetMCprocessType() const { return fMCprocessType; }
daf66719	153	Bool_t GetMCisPrimary() const {return fMCisPrimary;}
1ff4bda1	154	Int_t GetMCPID() const {return fMCPID;}
a14b8f3c	155	void SetRequireTransportBitForPrimaries(Bool_t b) {fRequireTransportBitForPrimaries=b; SetCutMC();}
daf66719	156
	157	void SetParamType(trackParameterType paramType) {fParamType=paramType;}
	158	trackParameterType GetParamType() const {return fParamType;}
12b2b8bc	159	static const char* GetParamTypeName(trackParameterType type);
daf66719	160	void SetParamMix(trackParameterMix paramMix) {fParamMix=paramMix;}
	161	trackParameterMix GetParamMix() const {return fParamMix;}
	162
9a0783cc	163	virtual Bool_t IsSelected(TObject* obj, Int_t id=-666);
1ff4bda1	164	virtual Bool_t IsSelectedMCtruth(TObject* obj, Int_t id=-666);
127a5825	165	AliVParticle* GetTrack() const {return fTrack;}
127a5825	166	AliMCParticle* GetMCparticle() const {return fMCparticle;}
41dc4195	167	AliFlowTrack* MakeFlowTrack() const;
7d27a354	168	Bool_t FillFlowTrack(AliFlowTrack* track) const;
127a5825	169	Bool_t IsPhysicalPrimary() const;
441ea1cf	170	static Bool_t IsPhysicalPrimary(AliMCEvent* p, Int_t label, Bool_t requiretransported=kTRUE);
daf66719	171
	172	void SetMCevent(AliMCEvent* mcEvent) {fMCevent=mcEvent;}
	173	AliMCEvent* GetMCevent() const {return fMCevent;}
aab6527a	174	void SetEvent(AliVEvent* event, AliMCEvent* mcEvent=NULL);
9a0783cc	175	AliVEvent* GetEvent() const {return fEvent;}
	176	Int_t GetNumberOfInputObjects() const;
	177	TObject* GetInputObject(Int_t i);
a1c43d26	178	void Clear(Option_t* option="");
daf66719	179
1a80f9f6	180	Double_t GetPmdEta(Float_t xPos, Float_t yPos, Float_t zPos);
	181	Double_t GetPmdPhi(Float_t xPos, Float_t yPos);
	182
32b846cd	183	//PID
2b1eaa10	184	void SetPID(AliPID::EParticleType pid, PIDsource s=kTOFpid, Double_t prob=0.9)
2b1eaa10	185	{fParticleID=pid; fPIDsource=s; fParticleProbability=prob; fCutPID=kTRUE; InitPIDcuts();}
32b846cd	186	void SetTPCpidCuts(TMatrixF* mat) {fTPCpidCuts=new TMatrixF(*mat);}
32b846cd	187	void SetTOFpidCuts(TMatrixF* mat) {fTOFpidCuts=new TMatrixF(*mat);}
2b1eaa10	188	static const char* PIDsourceName(PIDsource s);
aab6527a	189	AliESDpid& GetESDpid() {return fESDpid;}
875ee12a	190	void SetAllowTOFmismatchFlag(Bool_t b=kTRUE) {fAllowTOFmismatchFlag=b;}
	191	Bool_t GetAllowTOFmismatchFlag() const {return fAllowTOFmismatchFlag;}
	192	void SetRequireStrictTOFTPCagreement(Bool_t b=kTRUE) {fRequireStrictTOFTPCagreement=b;}
	193	Bool_t GetRequireStrictTOFTPCagreement() const {return fRequireStrictTOFTPCagreement;}
0b14d0f4	194	void SetRejectElectronsWithTPCpid(Bool_t b=kTRUE) {fCutRejectElectronsWithTPCpid=b;}
32b846cd	195
1a80f9f6	196	//these should maybe be protected
daf66719	197	Bool_t PassesCuts(AliVParticle* track);
42655d16	198	Bool_t PassesESDcuts(AliESDtrack* track);
1a80f9f6	199	Bool_t PassesAODcuts(const AliAODTrack* track);
d148af7e	200	Bool_t PassesPMDcuts(AliESDPmdTrack* track);
22289738	201	Bool_t PassesV0cuts(AliESDVZERO* track, Int_t id);
1a80f9f6	202	Bool_t PassesCuts(const AliFlowTrackSimple* track);
1a80f9f6	203	Bool_t PassesCuts(const AliMultiplicity* track, Int_t id);
12b2b8bc	204	Bool_t PassesMCcuts();
1ff4bda1	205	Bool_t PassesMCcuts(AliMCEvent* mcevent, Int_t label);
1a80f9f6	206	Bool_t PassesTPCdedxCut(const AliESDtrack* track);
cd62a2a8	207	Bool_t PassesTPCbayesianCut(const AliESDtrack* track);
1a80f9f6	208	Bool_t PassesTPCpidCut(const AliESDtrack* track) const;
	209	Bool_t PassesTOFbetaCut(const AliESDtrack* track);
	210	Bool_t PassesTOFbetaSimpleCut(const AliESDtrack* track);
	211	Bool_t PassesTOFpidCut(const AliESDtrack* track) const;
	212
499fe731	213	void Browse(TBrowser* b);
	214	Long64_t Merge(TCollection* list);
	215
1a80f9f6	216	protected:
d148af7e	217	AliFlowTrack* MakeFlowTrackSPDtracklet() const;
d148af7e	218	AliFlowTrack* MakeFlowTrackPMDtrack() const;
22289738	219	AliFlowTrack* MakeFlowTrackV0() const;
d148af7e	220	AliFlowTrack* MakeFlowTrackVParticle() const;
7d27a354	221	Bool_t FillFlowTrackVParticle(AliFlowTrack* t) const;
7d27a354	222	Bool_t FillFlowTrackGeneric(AliFlowTrack* t) const;
daf66719	223	void HandleESDtrack(AliESDtrack* track);
daf66719	224	void HandleVParticle(AliVParticle* track);
924b02b0	225	void DefineHistograms();
32b846cd	226	void InitPIDcuts();
7288f660	227	void InitESDcuts() {if (!fAliESDtrackCuts) {fAliESDtrackCuts=new AliESDtrackCuts();}}
3abf7ecc	228	// part added by F. Noferini
cd62a2a8	229	Bool_t PassesTOFbayesianCut(const AliESDtrack* track);
cd62a2a8	230	Int_t GetESDPdg(const AliESDtrack track,Option_t option="bayesianTOF",Int_t ipart=2,Float_t cPi=-1.0,Float_t cKa=0.0,Float_t cPr=0.0); // 3sigma cut ipart=0(el),1(mu),2(pi),3(K),4(p)
875ee12a	231	Bool_t TPCTOFagree(const AliESDtrack *track);
3abf7ecc	232	// end part added by F. Noferini
daf66719	233
9a0783cc	234	//the cuts
daf66719	235	AliESDtrackCuts* fAliESDtrackCuts; //alianalysis cuts
499fe731	236	TList* fQA; //qa histograms go here
1ff4bda1	237	Bool_t fCutMC; //do we cut on MC?
f21bdf48	238	Bool_t fCutMChasTrackReferences; //did we leave a trace in the detector?
daf66719	239	Bool_t fCutMCprocessType; //do we cut on mc process type?
	240	TMCProcess fMCprocessType; //mc process type
	241	Bool_t fCutMCPID; //cut on MC pid?
	242	Int_t fMCPID; //MC PID
a14b8f3c	243	Bool_t fIgnoreSignInMCPID; //when MC PID cut is set, pass also the antiparticle
daf66719	244	Bool_t fCutMCisPrimary; //do we cut on primaryness?
441ea1cf	245	Bool_t fRequireTransportBitForPrimaries; //require the transport bit to be set for primaries
daf66719	246	Bool_t fMCisPrimary; //is MC primary
127a5825	247	Bool_t fRequireCharge; //is charged?
127a5825	248	Bool_t fFakesAreOK; //are fakes (negative labels) ok?
9a0783cc	249	Bool_t fCutSPDtrackletDeltaPhi; //are we cutting on the trcklet deltaphi?
	250	Double_t fSPDtrackletDeltaPhiMax; //maximal deltaphi for tracklets
	251	Double_t fSPDtrackletDeltaPhiMin; //minimal deltaphi for tracklets
1ff4bda1	252	Bool_t fIgnoreTPCzRange; //ignore tracks going close to central membrane
	253	Double_t fIgnoreTPCzRangeMax; //max z to ignore
	254	Double_t fIgnoreTPCzRangeMin; //min z to ignore
2948ac5a	255	Bool_t fCutChi2PerClusterTPC; //cut on tpc chi2
	256	Float_t fMaxChi2PerClusterTPC; //max chi2 tpc/cluster
	257	Float_t fMinChi2PerClusterTPC; //min chi2 tpc/cluster
32b846cd	258	Bool_t fCutNClustersTPC; //cut on clusters?
	259	Int_t fNClustersTPCMax; //max tpc ncls
	260	Int_t fNClustersTPCMin; //min tpc clusters
42655d16	261	Bool_t fCutNClustersITS; //cut on clusters?
	262	Int_t fNClustersITSMax; //max tpc ncls
	263	Int_t fNClustersITSMin; //min tpc clusters
5ba02b32	264	Bool_t fUseAODFilterBit; //use AOD filter bit selection?
5ba02b32	265	UInt_t fAODFilterBit; //AOD filter bit to select
a63303bd	266	Bool_t fCutDCAToVertexXY; //dca xy cut
a63303bd	267	Bool_t fCutDCAToVertexZ; //dca z cut
2b1eaa10	268	Bool_t fCutMinimalTPCdedx; //cut on minimal dedx in TPC to reject noise tracks
2b1eaa10	269	Double_t fMinimalTPCdedx; //value for minimal TPC dedx
1a80f9f6	270
	271	Bool_t fCutPmdDet; //cut on PMD detector plane
	272	Int_t fPmdDet; // value of PMD detector plane
	273	Bool_t fCutPmdAdc; //cut on cluster ADC
	274	Float_t fPmdAdc; //value of cluster ADC
	275	Bool_t fCutPmdNcell; //cut on cluster ncell
	276	Float_t fPmdNcell; //value of cluster ncell
	277
daf66719	278	trackParameterType fParamType; //parameter type tu cut on
daf66719	279	trackParameterMix fParamMix; //parameter mixing
daf66719	280	AliVParticle* fTrack; //!the track to apply cuts on
12b2b8bc	281	Double_t fTrackPhi; //!track phi
	282	Double_t fTrackEta; //!track eta
	283	Double_t fTrackWeight; //!track weight
127a5825	284	Int_t fTrackLabel; //!track label, or its absolute value if FakesAreOK
957517fa	285	AliMCEvent* fMCevent; //!mc event
daf66719	286	AliMCParticle* fMCparticle; //!mc particle
9a0783cc	287	AliVEvent* fEvent; //!placeholder for current event
1ff4bda1	288	AliESDtrack fTPCtrack; //!placeholder for TPC only track to avoid new/delete on every track
daf66719	289
32b846cd	290	//PID
aab6527a	291	AliESDpid fESDpid; //pid obj
32b846cd	292	PIDsource fPIDsource; //pid source
	293	TMatrixF* fTPCpidCuts; //tpc pid cuts
	294	TMatrixF* fTOFpidCuts; //tof pid cuts
2b1eaa10	295	AliPID::EParticleType fParticleID; //alipid
2b1eaa10	296	Double_t fParticleProbability; //desired prob for a particle type
875ee12a	297	Bool_t fAllowTOFmismatchFlag; //allow TOFmismatch flag=1 in ESD
875ee12a	298	Bool_t fRequireStrictTOFTPCagreement; //require stricter than TOFmismatch flag TOF-TPC agreement
0b14d0f4	299	Bool_t fCutRejectElectronsWithTPCpid; //reject electrons with TPC pid
32b846cd	300
3abf7ecc	301	// part added by F. Noferini
1a80f9f6	302	static const Int_t fgkPIDptBin = 20; // pT bins for priors
1a80f9f6	303	Float_t fC[fgkPIDptBin][5],fBinLimitPID[fgkPIDptBin]; // pt bin limit and priors
3abf7ecc	304	Float_t fProbBayes[5]; // bayesian probability
	305	// end part added by F. Noferini
	306
1a80f9f6	307	static const Int_t fgkNumberOfV0tracks=64; //number of V0 channels
1a80f9f6	308
f21bdf48	309	ClassDef(AliFlowTrackCuts,8)
daf66719	310	};
	311
	312	#endif
	313