fTRDsignal is added to AliAODPid class (copied from AliESDtrack).
[u/mrichter/AliRoot.git] / STEER / AOD / AliAODPWG4ParticleCorrelation.h
CommitLineData
1c5acb87 1#ifndef ALIAODPWG4PARTICLECORRELATION_H
2#define ALIAODPWG4PARTICLECORRELATION_H
3/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id: AliAODPWG4ParticleCorrelation.h $ */
7
8//-------------------------------------------------------------------------
9// Copy of AOD photon class, adapted for particle identification
10// and correlations analysis
11// Author: Yves Schutz, CERN, Gustavo Conesa, INFN
12//-------------------------------------------------------------------------
13
14//-- ROOT system --
a3aebfff 15#include "TList.h"
5e6a3170 16#include "AliAODJet.h"
1c5acb87 17
18//-- Analysis system
5e6a3170 19
20
1c5acb87 21#include "AliAODPWG4Particle.h"
22
23class AliAODPWG4ParticleCorrelation : public AliAODPWG4Particle {
24
25 public:
477d6cee 26 AliAODPWG4ParticleCorrelation();
27 AliAODPWG4ParticleCorrelation(Double_t px, Double_t py, Double_t pz, Double_t e);
28 AliAODPWG4ParticleCorrelation(TLorentzVector & p);
29 AliAODPWG4ParticleCorrelation(AliAODPWG4Particle & p);
30
31 virtual ~AliAODPWG4ParticleCorrelation();
4695da48 32 virtual void Clear(const Option_t* /*opt*/);
33
477d6cee 34 AliAODPWG4ParticleCorrelation(const AliAODPWG4ParticleCorrelation& photon);
82679d84 35private:
477d6cee 36 AliAODPWG4ParticleCorrelation& operator=(const AliAODPWG4ParticleCorrelation& photon);
37
82679d84 38public:
4695da48 39 virtual TObjArray* GetObjArray(TString refname) const { if(fListOfObjArrays) return (TObjArray*) fListOfObjArrays->FindObject(refname);
40 else return 0x0;}
591cc579 41 virtual TList* GetObjArrayList() const { return fListOfObjArrays; }
42 virtual void AddObjArray(TObjArray * refarray) { fListOfObjArrays->Add(refarray); }
a3aebfff 43
477d6cee 44 virtual TString GetLeadingDetector() const {return fLeadingDetector ; }
45 virtual void SetLeadingDetector(TString d) {fLeadingDetector = d ; }
46
47 virtual TLorentzVector GetLeading() const { return fLeading;}
48 virtual void SetLeading(TLorentzVector lead) {fLeading = lead;}
49
50 virtual TLorentzVector GetCorrelatedJet() const { return fCorrJet;}
51 virtual void SetCorrelatedJet(TLorentzVector jet) {fCorrJet = jet;}
52
53 virtual TLorentzVector GetCorrelatedBackground() const { return fCorrBkg;}
54 virtual void SetCorrelatedBackground(TLorentzVector bkg) {fCorrBkg = bkg;}
55
56 virtual void SetRefJet(AliAODJet* jet) { fRefJet = jet;}
57 virtual AliAODJet* GetJet() const {return ((AliAODJet*) fRefJet.GetObject());}
58 virtual TRef GetRefJet() const {return fRefJet;}
59
a3aebfff 60 virtual Bool_t IsIsolated() const { return fIsolated ; }
61 virtual void SetIsolated(Bool_t iso) { fIsolated = iso ; }
62
f284616d 63 virtual Bool_t IsLeadingParticle() const { return fLeadingParticle ; }
64 virtual void SetLeadingParticle(Bool_t leadPart) { fLeadingParticle = leadPart ; }
477d6cee 65 virtual void Print(Option_t* /*option*/) const;
66
1c5acb87 67 private:
f284616d 68 Bool_t fIsolated ; //Particle is isolated or not
69 Bool_t fLeadingParticle ; //Particle is leading or not
a3aebfff 70 TString fLeadingDetector; // Detector where leading particle was measured.
71 TLorentzVector fLeading; // Leading Particle 4-momentum vector
72 TLorentzVector fCorrJet; // Jet 4-momentum vector
73 TLorentzVector fCorrBkg; // Background 4-momentum vector
74 TRef fRefJet; // Reference to jet found with JETAN and correlated with particle
591cc579 75 TList * fListOfObjArrays ; // List with correlation reference arrays
a3aebfff 76
f284616d 77 ClassDef(AliAODPWG4ParticleCorrelation, 4);
1c5acb87 78};
79
80
81#endif //ALIAODPWG4PARTICLECORRELATION_H