[u/mrichter/AliRoot.git] / STEER / AOD / AliAODPWG4ParticleCorrelation.h

#ifndef ALIAODPWG4PARTICLECORRELATION_H
#define ALIAODPWG4PARTICLECORRELATION_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliAODPWG4ParticleCorrelation.h  $ */

//-------------------------------------------------------------------------
//     AOD objects class in use in the CaloTrackCorrelations
//     analysis pacackge ($ALICE_ROOT/PWGGA/CaloTrackCorrelations)
//     Includes correlation objects
//
//     Author: Gustavo Conesa Balbastre - CNRS-LPSC-Grenoble
//-------------------------------------------------------------------------

//-- ROOT system --
#include "TList.h"
#include "AliAODJet.h"

//-- Analysis system


#include "AliAODPWG4Particle.h"

class AliAODPWG4ParticleCorrelation : public AliAODPWG4Particle {

 public:
  AliAODPWG4ParticleCorrelation();
  AliAODPWG4ParticleCorrelation(Double_t px, Double_t py, Double_t pz, Double_t e);
  AliAODPWG4ParticleCorrelation(TLorentzVector & p);  
  AliAODPWG4ParticleCorrelation(AliAODPWG4Particle & p);  
  
  virtual ~AliAODPWG4ParticleCorrelation();
  virtual void Clear(const Option_t* /*opt*/);
  
  AliAODPWG4ParticleCorrelation(const AliAODPWG4ParticleCorrelation& photon); 
private:
  AliAODPWG4ParticleCorrelation& operator=(const AliAODPWG4ParticleCorrelation& photon);
  
public:
  virtual TObjArray* GetObjArray(TString refname)  const { if(fListOfObjArrays) return (TObjArray*) fListOfObjArrays->FindObject(refname); 
                                                           else return 0x0;} 
  virtual TList*     GetObjArrayList()             const { return  fListOfObjArrays; } 
  virtual void       AddObjArray(TObjArray * refarray)  { fListOfObjArrays->Add(refarray); }

  virtual TString GetLeadingDetector()   const {return fLeadingDetector ; }
  virtual void SetLeadingDetector(TString d)   {fLeadingDetector = d ; }
  
  virtual TLorentzVector  GetLeading()               const { return  fLeading;}
  virtual void  SetLeading(TLorentzVector lead) {fLeading = lead;}
  
  virtual TLorentzVector  GetCorrelatedJet()         const { return  fCorrJet;}
  virtual void  SetCorrelatedJet(TLorentzVector jet) {fCorrJet = jet;}
  
  virtual TLorentzVector  GetCorrelatedBackground()  const { return  fCorrBkg;}
  virtual void  SetCorrelatedBackground(TLorentzVector bkg) {fCorrBkg = bkg;}
  
  virtual void SetRefJet(AliAODJet* jet)  { fRefJet = jet;}
  virtual      AliAODJet* GetJet() const {return ((AliAODJet*) fRefJet.GetObject());}
  virtual TRef GetRefJet()         const {return fRefJet;}
  
  virtual void   Print(Option_t* /*option*/) const;
  
 private:
  TString        fLeadingDetector;  // Detector where leading particle was measured.
  TLorentzVector fLeading;          // Leading Particle 4-momentum vector
  TLorentzVector fCorrJet;          // Jet  4-momentum vector
  TLorentzVector fCorrBkg;          // Background  4-momentum vector
  TRef           fRefJet;           // Reference to jet found with JETAN and correlated with particle
  TList   *      fListOfObjArrays ; // List with correlation reference arrays
  
  ClassDef(AliAODPWG4ParticleCorrelation, 5);
};


#endif //ALIAODPWG4PARTICLECORRELATION_H
Commit	Line	Data
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	//-------------------------------------------------------------------------
bfa9c596	9	// AOD objects class in use in the CaloTrackCorrelations
	10	// analysis pacackge ($ALICE_ROOT/PWGGA/CaloTrackCorrelations)
	11	// Includes correlation objects
	12	//
	13	// Author: Gustavo Conesa Balbastre - CNRS-LPSC-Grenoble
1c5acb87	14	//-------------------------------------------------------------------------
	15
	16	//-- ROOT system --
a3aebfff	17	#include "TList.h"
5e6a3170	18	#include "AliAODJet.h"
1c5acb87	19
1c5acb87	20	//-- Analysis system
5e6a3170	21
5e6a3170	22
1c5acb87	23	#include "AliAODPWG4Particle.h"
	24
	25	class AliAODPWG4ParticleCorrelation : public AliAODPWG4Particle {
	26
	27	public:
477d6cee	28	AliAODPWG4ParticleCorrelation();
	29	AliAODPWG4ParticleCorrelation(Double_t px, Double_t py, Double_t pz, Double_t e);
	30	AliAODPWG4ParticleCorrelation(TLorentzVector & p);
	31	AliAODPWG4ParticleCorrelation(AliAODPWG4Particle & p);
	32
	33	virtual ~AliAODPWG4ParticleCorrelation();
4695da48	34	virtual void Clear(const Option_t* /opt/);
4695da48	35
477d6cee	36	AliAODPWG4ParticleCorrelation(const AliAODPWG4ParticleCorrelation& photon);
82679d84	37	private:
477d6cee	38	AliAODPWG4ParticleCorrelation& operator=(const AliAODPWG4ParticleCorrelation& photon);
477d6cee	39
82679d84	40	public:
4695da48	41	virtual TObjArray* GetObjArray(TString refname) const { if(fListOfObjArrays) return (TObjArray*) fListOfObjArrays->FindObject(refname);
4695da48	42	else return 0x0;}
591cc579	43	virtual TList* GetObjArrayList() const { return fListOfObjArrays; }
591cc579	44	virtual void AddObjArray(TObjArray * refarray) { fListOfObjArrays->Add(refarray); }
a3aebfff	45
477d6cee	46	virtual TString GetLeadingDetector() const {return fLeadingDetector ; }
	47	virtual void SetLeadingDetector(TString d) {fLeadingDetector = d ; }
	48
	49	virtual TLorentzVector GetLeading() const { return fLeading;}
	50	virtual void SetLeading(TLorentzVector lead) {fLeading = lead;}
	51
	52	virtual TLorentzVector GetCorrelatedJet() const { return fCorrJet;}
	53	virtual void SetCorrelatedJet(TLorentzVector jet) {fCorrJet = jet;}
	54
	55	virtual TLorentzVector GetCorrelatedBackground() const { return fCorrBkg;}
	56	virtual void SetCorrelatedBackground(TLorentzVector bkg) {fCorrBkg = bkg;}
	57
	58	virtual void SetRefJet(AliAODJet* jet) { fRefJet = jet;}
	59	virtual AliAODJet* GetJet() const {return ((AliAODJet*) fRefJet.GetObject());}
	60	virtual TRef GetRefJet() const {return fRefJet;}
	61
	62	virtual void Print(Option_t* /option/) const;
	63
1c5acb87	64	private:
a3aebfff	65	TString fLeadingDetector; // Detector where leading particle was measured.
	66	TLorentzVector fLeading; // Leading Particle 4-momentum vector
	67	TLorentzVector fCorrJet; // Jet 4-momentum vector
	68	TLorentzVector fCorrBkg; // Background 4-momentum vector
	69	TRef fRefJet; // Reference to jet found with JETAN and correlated with particle
591cc579	70	TList * fListOfObjArrays ; // List with correlation reference arrays
a3aebfff	71
bfa9c596	72	ClassDef(AliAODPWG4ParticleCorrelation, 5);
1c5acb87	73	};
	74
	75
	76	#endif //ALIAODPWG4PARTICLECORRELATION_H