]>
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 | |
41 | AliAnalyseLeadingTrackUE(const AliAnalyseLeadingTrackUE & g); //copy constructor | |
42 | AliAnalyseLeadingTrackUE & operator = (const AliAnalyseLeadingTrackUE & g); //assignment operator | |
43 | virtual ~AliAnalyseLeadingTrackUE(); //virtual destructor | |
44 | ||
45 | // Setters | |
e0331fd9 | 46 | void SetParticleSelectionCriteria(Int_t filterbit, Bool_t onlyhadrons, Double_t etacut, Double_t ptMin = 0) { fFilterBit = filterbit; |
a75aacd6 | 47 | fOnlyHadrons = onlyhadrons; |
e0331fd9 | 48 | fTrackEtaCut = etacut; |
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 |
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 |
60 | TObjArray* SortRegions(const AliVParticle* leading, TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE); // Assign particles to towards, away or transverse regions | |
59375a30 | 61 | TObjArray* GetAcceptedParticles(TObject* obj, TObject* arrayMC, Bool_t onlyprimaries = kTRUE, Int_t particleSpecies = -1, Bool_t useEtaPtCuts = kFALSE, Bool_t speciesOnTracks = kTRUE); |
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 |
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; } |
68 | void SetHelperPID(AliHelperPID* pid) { fHelperPID = pid; } | |
418b56c5 | 69 | void SetTrackStatus(UInt_t status) { fTrackStatus = status; } |
70 | UInt_t GetTrackStatus() { return fTrackStatus; } | |
9e35c487 | 71 | void SetCheckMotherPDG(Bool_t checkpdg) { fCheckMotherPDG = checkpdg; } |
72 | Bool_t GetCheckMotherPDG() { return fCheckMotherPDG; } | |
416af868 | 73 | void NextEvent() { fEventCounter++; } |
97270ee2 | 74 | void SetDCAXYCut(TFormula* value) { fDCAXYCut = value; } |
418b56c5 | 75 | |
76 | protected: | |
77 | Bool_t CheckTrack(AliVParticle * part); | |
78 | ||
32e49607 | 79 | private: |
a75aacd6 | 80 | Int_t fDebug; // debug flag |
81 | Int_t fFilterBit; // track selection cuts | |
418b56c5 | 82 | UInt_t fTrackStatus; // if non-0, the bits set in this variable are required for each track |
a75aacd6 | 83 | Bool_t fOnlyHadrons; // consider only charged Pions, Protons and Kaons |
9e35c487 | 84 | Bool_t fCheckMotherPDG; // Check the PDG code of mother for secondaries |
a75aacd6 | 85 | Double_t fTrackEtaCut; // pseudo-rapidity limit of transverse regions |
e0331fd9 | 86 | Double_t fTrackPtMin; // pt limit for selecting particles |
5cabd2cd | 87 | UInt_t fEventSelection; // bit for physics selection |
97270ee2 | 88 | TFormula* fDCAXYCut; // additional pt dependent cut on DCA XY (only for AOD) |
3712ba26 | 89 | AliESDtrackCuts *fEsdTrackCuts; // set of cuts when reading ESD |
8b43ff78 | 90 | AliESDtrackCuts *fEsdTrackCutsExtra1; // set of cuts when reading ESD |
91 | AliESDtrackCuts *fEsdTrackCutsExtra2; // set of cuts when reading ESD | |
32e49607 | 92 | AliHelperPID *fHelperPID; // PID Helper object |
416af868 | 93 | UInt_t fEventCounter; //! processed events (counter) |
3f3f12d9 | 94 | |
3712ba26 | 95 | ClassDef(AliAnalyseLeadingTrackUE,0) |
a75aacd6 | 96 | }; |
97 | #endif |