[u/mrichter/AliRoot.git] / PWGCF / Correlations / Base / AliAnalyseLeadingTrackUE.h

//-*- Mode: C++ -*-
#ifndef ALIANALYSELEADINGTRACKUE_H
#define ALIANALYSELEADINGTRACKUE_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

////////////////////////////////////////////////
//--------------------------------------------- 
// Class  for transverse regions analysis w.r.t leading track
//---------------------------------------------
////////////////////////////////////////////////

// --- ROOT system ---
#include <TObject.h> 

class AliAODEvent;
class AliAODInputHandler;
class AliESDEvent;
class AliESDtrackCuts;
class AliAODTrack;
class AliESDTrack;
class AliGenPythiaEventHeader;
class AliInputEventHandler;
class AliLog;
class AliMCEvent;
class AliStack;
class AliVParticle;
class TClonesArray;
class TObject;
class TROOT;
class TVector3;
class AliVTrack;
class AliHelperPID;
class TFormula;

class AliAnalyseLeadingTrackUE : public TObject {

 public: 

  AliAnalyseLeadingTrackUE();                                                   //constructor
  AliAnalyseLeadingTrackUE & operator = (const AliAnalyseLeadingTrackUE & g);   //assignment operator
  virtual ~AliAnalyseLeadingTrackUE();                                          //virtual destructor

  // Setters
  void  SetParticleSelectionCriteria(Int_t filterbit, Bool_t onlyhadrons, Double_t etacut, Double_t etacutMin=-1., Double_t ptMin = 0) { fFilterBit   = filterbit;
  											     fOnlyHadrons = onlyhadrons;
										             fTrackEtaCut = etacut;     
										             fTrackEtaCutMin = etacutMin;     
										             fTrackPtMin = ptMin;}

  void SetEventSelection(UInt_t bits) { fEventSelection = bits; }
  void  SetDebug(Int_t debug) { fDebug = debug; } 
  Bool_t         ApplyCuts(TObject* track);                       // Reproduces the cuts of the corresponding bit in the ESD->AOD filtering
  void           DefineESDCuts(Int_t filterbit);                                 // Emulate filterbit
  TObjArray*     FindLeadingObjects(TObject* obj);                                 // Looks for leading track or MC particle
  TObjArray*     GetMinMaxRegion(TList* transv1, TList* transv2);                  // Sorts the transverse regions in MIN and MAX
  Int_t          NParticles(TObject *obj);                                         // Counts tracks or MC particles
  AliVParticle*  ParticleWithCuts(TObject* obj, Int_t ipart, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1);                     // Returns track or MC particle at position "ipart" if passes selection criteria
  void  	 QSortTracks(TObjArray &a, Int_t first, Int_t last);               // Sort by pT an array of AliVParticles 
  TObjArray*     SortRegions(const AliVParticle* leading, TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE); // Assign particles to towards, away or transverse regions
  TObjArray*     GetAcceptedParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1, Bool_t useEtaPtCuts = kFALSE, Bool_t speciesOnTracks = kTRUE, Double_t evtPlane = -999.); 
  TObjArray*     GetFakeParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries, Int_t particleSpecies, Bool_t useEtaPtCuts);
  Bool_t         TriggerSelection(const TObject* obj);                                   // Select good triggers with AliPhysicsSelection class
  Bool_t         VertexSelection(const TObject* obj, Int_t ntracks, Double_t zed);       // Vertex selection: see implementation
  void 		 RemoveInjectedSignals(TObjArray* tracks, TObject* arrayMC, Int_t maxLabel);
  void 		 RemoveWeakDecays(TObjArray* tracks, TObject* mcObj);
  AliHelperPID*  GetHelperPID() { return fHelperPID; }
  void 		 SetHelperPID(AliHelperPID* pid) { fHelperPID = pid; }
  void		 SetTrackStatus(UInt_t status) { fTrackStatus = status; }
  UInt_t	 GetTrackStatus() { return fTrackStatus; }
  void		 SetCheckMotherPDG(Bool_t checkpdg) { fCheckMotherPDG = checkpdg; }
  Bool_t	 GetCheckMotherPDG() { return fCheckMotherPDG; }
  void		 NextEvent() { fEventCounter++; }
  UInt_t	 GetEventCounter() { return fEventCounter; }
  void		 SetDCAXYCut(TFormula* value) { fDCAXYCut = value; }
  void		 SetSharedClusterCut(Double_t value) { fSharedClusterCut = value; }
  void		 SetCrossedRowsCut(Int_t value) { fCrossedRowsCut = value; }
  void		 SetFoundFractionCut(Double_t value) { fFoundFractionCut = value; }
  void           SetParticlePhiCutEventPlane(Double_t min, Double_t max) { fTrackPhiCutEvPlMin = min; fTrackPhiCutEvPlMax = max; }

protected:
  Bool_t CheckTrack(AliVParticle * part);

private:
  AliAnalyseLeadingTrackUE(const AliAnalyseLeadingTrackUE & g);                 //copy constructor, not implemented
  Int_t          fDebug;             // debug flag
  Int_t          fFilterBit;         // track selection cuts
  UInt_t         fTrackStatus;       // if non-0, the bits set in this variable are required for each track
  Bool_t         fOnlyHadrons;       // consider only charged Pions, Protons and Kaons 
  Bool_t         fCheckMotherPDG;     // Check the PDG code of mother for secondaries 
  Double_t       fTrackEtaCut;       // pseudo-rapidity limit of transverse regions     
  Double_t       fTrackEtaCutMin;       // minimum of the pseudo-rapidity limit of transverse regions     
  Double_t       fTrackPhiCutEvPlMin;   // Minimum Phi cut on particles with respect to the Event Plane (values between 0 and Pi/2)
  Double_t       fTrackPhiCutEvPlMax;   // Maximum Phi cut on particles with respect to the Event Plane (values between 0 and Pi/2)
  Double_t       fTrackPtMin;        // pt limit for selecting particles
  UInt_t         fEventSelection;    // bit for physics selection
  TFormula*      fDCAXYCut;          // additional pt dependent cut on DCA XY (only for AOD)
  Double_t       fSharedClusterCut;  // cut on shared clusters (only for AOD)
  Int_t		 fCrossedRowsCut;   // cut on crossed rows (only for AOD)
  Double_t	 fFoundFractionCut;     // cut on crossed rows/findable clusters (only for AOD)
  
  AliESDtrackCuts *fEsdTrackCuts;    // set of cuts when reading ESD
  AliESDtrackCuts *fEsdTrackCutsExtra1;    // set of cuts when reading ESD
  AliESDtrackCuts *fEsdTrackCutsExtra2;    // set of cuts when reading ESD
  AliHelperPID   *fHelperPID;    // PID Helper object
  UInt_t fEventCounter;		  //! processed events (counter)
  
  ClassDef(AliAnalyseLeadingTrackUE,0)
};
#endif
Commit	Line	Data
a75aacd6	1	//-- Mode: C++ --
	2	#ifndef ALIANALYSELEADINGTRACKUE_H
	3	#define ALIANALYSELEADINGTRACKUE_H
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////////
	8	//---------------------------------------------
	9	// Class for transverse regions analysis w.r.t leading track
	10	//---------------------------------------------
	11	////////////////////////////////////////////////
	12
	13	// --- ROOT system ---
	14	#include <TObject.h>
	15
	16	class AliAODEvent;
144bd037	17	class AliAODInputHandler;
a75aacd6	18	class AliESDEvent;
3712ba26	19	class AliESDtrackCuts;
a75aacd6	20	class AliAODTrack;
	21	class AliESDTrack;
	22	class AliGenPythiaEventHeader;
	23	class AliInputEventHandler;
144bd037	24	class AliLog;
a75aacd6	25	class AliMCEvent;
	26	class AliStack;
	27	class AliVParticle;
	28	class TClonesArray;
	29	class TObject;
144bd037	30	class TROOT;
a75aacd6	31	class TVector3;
39c2cb38	32	class AliVTrack;
32e49607	33	class AliHelperPID;
97270ee2	34	class TFormula;
a75aacd6	35
	36	class AliAnalyseLeadingTrackUE : public TObject {
	37
	38	public:
	39
	40	AliAnalyseLeadingTrackUE(); //constructor
a75aacd6	41	AliAnalyseLeadingTrackUE & operator = (const AliAnalyseLeadingTrackUE & g); //assignment operator
	42	virtual ~AliAnalyseLeadingTrackUE(); //virtual destructor
	43
	44	// Setters
97dc2b4e	45	void SetParticleSelectionCriteria(Int_t filterbit, Bool_t onlyhadrons, Double_t etacut, Double_t etacutMin=-1., Double_t ptMin = 0) { fFilterBit = filterbit;
a75aacd6	46	fOnlyHadrons = onlyhadrons;
e0331fd9	47	fTrackEtaCut = etacut;
97dc2b4e	48	fTrackEtaCutMin = etacutMin;
e0331fd9	49	fTrackPtMin = ptMin;}
418b56c5	50
5cabd2cd	51	void SetEventSelection(UInt_t bits) { fEventSelection = bits; }
a75aacd6	52	void SetDebug(Int_t debug) { fDebug = debug; }
3712ba26	53	Bool_t ApplyCuts(TObject* track); // Reproduces the cuts of the corresponding bit in the ESD->AOD filtering
3712ba26	54	void DefineESDCuts(Int_t filterbit); // Emulate filterbit
a75aacd6	55	TObjArray* FindLeadingObjects(TObject* obj); // Looks for leading track or MC particle
	56	TObjArray* GetMinMaxRegion(TList* transv1, TList* transv2); // Sorts the transverse regions in MIN and MAX
	57	Int_t NParticles(TObject *obj); // Counts tracks or MC particles
b1831bcb	58	AliVParticle* ParticleWithCuts(TObject* obj, Int_t ipart, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1); // Returns track or MC particle at position "ipart" if passes selection criteria
a75aacd6	59	void QSortTracks(TObjArray &a, Int_t first, Int_t last); // Sort by pT an array of AliVParticles
a75aacd6	60	TObjArray* SortRegions(const AliVParticle* leading, TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE); // Assign particles to towards, away or transverse regions
774ef124	61	TObjArray* GetAcceptedParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1, Bool_t useEtaPtCuts = kFALSE, Bool_t speciesOnTracks = kTRUE, Double_t evtPlane = -999.);
b591fb9c	62	TObjArray* GetFakeParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries, Int_t particleSpecies, Bool_t useEtaPtCuts);
a75aacd6	63	Bool_t TriggerSelection(const TObject* obj); // Select good triggers with AliPhysicsSelection class
a75aacd6	64	Bool_t VertexSelection(const TObject* obj, Int_t ntracks, Double_t zed); // Vertex selection: see implementation
1ccd8a0a	65	void RemoveInjectedSignals(TObjArray* tracks, TObject* arrayMC, Int_t maxLabel);
f613255f	66	void RemoveWeakDecays(TObjArray* tracks, TObject* mcObj);
32e49607	67	AliHelperPID* GetHelperPID() { return fHelperPID; }
32e49607	68	void SetHelperPID(AliHelperPID* pid) { fHelperPID = pid; }
418b56c5	69	void SetTrackStatus(UInt_t status) { fTrackStatus = status; }
418b56c5	70	UInt_t GetTrackStatus() { return fTrackStatus; }
9e35c487	71	void SetCheckMotherPDG(Bool_t checkpdg) { fCheckMotherPDG = checkpdg; }
9e35c487	72	Bool_t GetCheckMotherPDG() { return fCheckMotherPDG; }
416af868	73	void NextEvent() { fEventCounter++; }
677dc831	74	UInt_t GetEventCounter() { return fEventCounter; }
97270ee2	75	void SetDCAXYCut(TFormula* value) { fDCAXYCut = value; }
132b6963	76	void SetSharedClusterCut(Double_t value) { fSharedClusterCut = value; }
	77	void SetCrossedRowsCut(Int_t value) { fCrossedRowsCut = value; }
	78	void SetFoundFractionCut(Double_t value) { fFoundFractionCut = value; }
774ef124	79	void SetParticlePhiCutEventPlane(Double_t min, Double_t max) { fTrackPhiCutEvPlMin = min; fTrackPhiCutEvPlMax = max; }
418b56c5	80
	81	protected:
	82	Bool_t CheckTrack(AliVParticle * part);
	83
32e49607	84	private:
0c61c973	85	AliAnalyseLeadingTrackUE(const AliAnalyseLeadingTrackUE & g); //copy constructor, not implemented
a75aacd6	86	Int_t fDebug; // debug flag
a75aacd6	87	Int_t fFilterBit; // track selection cuts
418b56c5	88	UInt_t fTrackStatus; // if non-0, the bits set in this variable are required for each track
a75aacd6	89	Bool_t fOnlyHadrons; // consider only charged Pions, Protons and Kaons
9e35c487	90	Bool_t fCheckMotherPDG; // Check the PDG code of mother for secondaries
a75aacd6	91	Double_t fTrackEtaCut; // pseudo-rapidity limit of transverse regions
97dc2b4e	92	Double_t fTrackEtaCutMin; // minimum of the pseudo-rapidity limit of transverse regions
774ef124	93	Double_t fTrackPhiCutEvPlMin; // Minimum Phi cut on particles with respect to the Event Plane (values between 0 and Pi/2)
774ef124	94	Double_t fTrackPhiCutEvPlMax; // Maximum Phi cut on particles with respect to the Event Plane (values between 0 and Pi/2)
e0331fd9	95	Double_t fTrackPtMin; // pt limit for selecting particles
5cabd2cd	96	UInt_t fEventSelection; // bit for physics selection
97270ee2	97	TFormula* fDCAXYCut; // additional pt dependent cut on DCA XY (only for AOD)
7d756130	98	Double_t fSharedClusterCut; // cut on shared clusters (only for AOD)
132b6963	99	Int_t fCrossedRowsCut; // cut on crossed rows (only for AOD)
132b6963	100	Double_t fFoundFractionCut; // cut on crossed rows/findable clusters (only for AOD)
7d756130	101
3712ba26	102	AliESDtrackCuts *fEsdTrackCuts; // set of cuts when reading ESD
8b43ff78	103	AliESDtrackCuts *fEsdTrackCutsExtra1; // set of cuts when reading ESD
8b43ff78	104	AliESDtrackCuts *fEsdTrackCutsExtra2; // set of cuts when reading ESD
32e49607	105	AliHelperPID *fHelperPID; // PID Helper object
416af868	106	UInt_t fEventCounter; //! processed events (counter)
3f3f12d9	107
3712ba26	108	ClassDef(AliAnalyseLeadingTrackUE,0)
a75aacd6	109	};
a75aacd6	110	#endif