[u/mrichter/AliRoot.git] / PWG4 / JetTasks / 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 AliAnalyseLeadingTrackUE : public TObject {

 public: 

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

  // Setters
  void  SetParticleSelectionCriteria(Int_t filterbit, Bool_t onlyhadrons, Double_t etacut) { fFilterBit   = filterbit;
  											     fOnlyHadrons = onlyhadrons;
										             fTrackEtaCut = etacut;     }
  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         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

 private:
  Int_t          fDebug;             // debug flag
  Int_t          fFilterBit;         // track selection cuts
  Bool_t         fOnlyHadrons;       // consider only charged Pions, Protons and Kaons 
  Double_t       fTrackEtaCut;       // pseudo-rapidity limit of transverse regions     
  AliESDtrackCuts *fEsdTrackCuts;    // set of cuts when reading ESD
  AliESDtrackCuts *fEsdTrackCutsSPD;    // set of cuts when reading ESD
  AliESDtrackCuts *fEsdTrackCutsSDD;    // set of cuts when reading ESD
  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;
	32
	33	class AliAnalyseLeadingTrackUE : public TObject {
	34
	35	public:
	36
	37	AliAnalyseLeadingTrackUE(); //constructor
	38	AliAnalyseLeadingTrackUE(const AliAnalyseLeadingTrackUE & g); //copy constructor
	39	AliAnalyseLeadingTrackUE & operator = (const AliAnalyseLeadingTrackUE & g); //assignment operator
	40	virtual ~AliAnalyseLeadingTrackUE(); //virtual destructor
	41
	42	// Setters
	43	void SetParticleSelectionCriteria(Int_t filterbit, Bool_t onlyhadrons, Double_t etacut) { fFilterBit = filterbit;
	44	fOnlyHadrons = onlyhadrons;
	45	fTrackEtaCut = etacut; }
	46	void SetDebug(Int_t debug) { fDebug = debug; }
3712ba26	47	Bool_t ApplyCuts(TObject* track); // Reproduces the cuts of the corresponding bit in the ESD->AOD filtering
3712ba26	48	void DefineESDCuts(Int_t filterbit); // Emulate filterbit
a75aacd6	49	TObjArray* FindLeadingObjects(TObject* obj); // Looks for leading track or MC particle
	50	TObjArray* GetMinMaxRegion(TList* transv1, TList* transv2); // Sorts the transverse regions in MIN and MAX
	51	Int_t NParticles(TObject *obj); // Counts tracks or MC particles
b1831bcb	52	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	53	void QSortTracks(TObjArray &a, Int_t first, Int_t last); // Sort by pT an array of AliVParticles
a75aacd6	54	TObjArray* SortRegions(const AliVParticle* leading, TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE); // Assign particles to towards, away or transverse regions
b1831bcb	55	TObjArray* GetAcceptedParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1);
a75aacd6	56	Bool_t TriggerSelection(const TObject* obj); // Select good triggers with AliPhysicsSelection class
	57	Bool_t VertexSelection(const TObject* obj, Int_t ntracks, Double_t zed); // Vertex selection: see implementation
	58
	59	private:
	60	Int_t fDebug; // debug flag
	61	Int_t fFilterBit; // track selection cuts
	62	Bool_t fOnlyHadrons; // consider only charged Pions, Protons and Kaons
	63	Double_t fTrackEtaCut; // pseudo-rapidity limit of transverse regions
3712ba26	64	AliESDtrackCuts *fEsdTrackCuts; // set of cuts when reading ESD
	65	AliESDtrackCuts *fEsdTrackCutsSPD; // set of cuts when reading ESD
	66	AliESDtrackCuts *fEsdTrackCutsSDD; // set of cuts when reading ESD
	67	ClassDef(AliAnalyseLeadingTrackUE,0)
a75aacd6	68	};
a75aacd6	69	#endif