[u/mrichter/AliRoot.git] / JETAN / AliJetUnitArray.h

#ifndef ALIJETUNITARRAY_H
#define ALIJETUNITARRAY_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *  *  * See cxx source for full Copyright notice     */

//------------------------------------------------------------------
//  Unit used by UA1 algorithm
//  Authors: Sarah Blyth (LBL/UCT)
//           Magali Estienne (IReS) (new version for JETAN)
//------------------------------------------------------------------


#include <TObject.h>
#include "AliJetFinderTypes.h"

class AliJetUnitArray : public TObject
{
 public:
  AliJetUnitArray();
  AliJetUnitArray(Int_t absId, Int_t esdId, Float_t eta, Float_t phi, Float_t en, Float_t px, Float_t py, Float_t pz, Float_t Deta, Float_t Dphi, AliJetFinderUnitDetectorFlagType_t det, AliJetFinderUnitFlagType_t inout, AliJetFinderUnitCutFlagType_t cut, Float_t mass, Int_t clusId);
  ~AliJetUnitArray();

  // Setter
  void SetUnitEnergy(Float_t energy)  {fUnitEnergy = energy;}
  void SetUnitEta(Float_t eta)        {fUnitEta = eta;} 
  void SetUnitPhi(Float_t phi)        {fUnitPhi = phi;}
  void SetUnitDeta(Float_t deta)      {fUnitDeta = deta;} 
  void SetUnitDphi(Float_t dphi)      {fUnitDphi = dphi;}
  void SetUnitID(Int_t id)            {fUnitID = id;}
  void SetUnitTrackID(Int_t esdid)    {fUnitTrackID = esdid;}
  void SetUnitEntries(Int_t num)      {fUnitNum = num;}
  void SetUnitClusterID(Int_t id)     {fUnitClusterID = id;}
  void SetUnitFlag(AliJetFinderUnitFlagType_t flag)    
  { 
    fUnitFlag = flag; 
  } 
  void SetUnitCutFlag(AliJetFinderUnitCutFlagType_t cutFlag)    
  { 
    fUnitCutFlag = cutFlag; 
  } 
  void SetUnitSignalFlag(AliJetFinderUnitSignalFlagType_t signalFlag)    
  { 
    fUnitSignalFlag = signalFlag; 
  } 
  void SetUnitDetectorFlag(AliJetFinderUnitDetectorFlagType_t detectorflag)    
  { 
    fUnitDetectorFlag = detectorflag; 
  } 
  void SetUnitPxPyPz(Double_t *pxyz)  {fUnitPx = pxyz[0]; fUnitPy = pxyz[1]; fUnitPz = pxyz[2];}
  void SetUnitMass(Float_t mass) {fUnitMass = mass;}

  // Getter
  Float_t GetUnitEnergy() const            {return fUnitEnergy;}
  Float_t GetUnitEta() const               {return fUnitEta;}
  Float_t GetUnitPhi() const               {return fUnitPhi;}         
  Float_t GetUnitDeta() const              {return fUnitDeta;}
  Float_t GetUnitDphi() const              {return fUnitDphi;}         
  Int_t   GetUnitID() const                {return fUnitID;}
  Int_t   GetUnitTrackID() const             {return fUnitTrackID;}
  Int_t   GetUnitEntries() const           {return fUnitNum;}
  Int_t   GetUnitClusterID() const         {return fUnitClusterID;}
  Float_t GetUnitMass() const              {return fUnitMass;}
  Bool_t  GetUnitPxPyPz(Double_t* p) const {p[0]=fUnitPx; p[1]=fUnitPy; p[2]=fUnitPz; return kTRUE;}

  AliJetFinderUnitFlagType_t GetUnitFlag() const     
  {
	  return fUnitFlag;
  }
  AliJetFinderUnitCutFlagType_t GetUnitCutFlag() const     
  {
	  return fUnitCutFlag;
  }
  AliJetFinderUnitSignalFlagType_t GetUnitSignalFlag() const     
  {
	  return fUnitSignalFlag;
  }
  AliJetFinderUnitDetectorFlagType_t GetUnitDetectorFlag() const     
  {
	  return fUnitDetectorFlag;
  }


  Bool_t operator>  ( AliJetUnitArray unit1) const;
  Bool_t operator<  ( AliJetUnitArray unit1) const;
  Bool_t operator== ( AliJetUnitArray unit1) const;

 protected:
  Float_t         fUnitEnergy;                          // Energy of the unit 
  Float_t         fUnitEta;                             // Eta of the unit
  Float_t         fUnitPhi;                             // Phi of the unit
  Float_t         fUnitDeta;                            // Delta Eta of the unit
  Float_t         fUnitDphi;                            // Delta Phi of the unit
  Int_t           fUnitID;                              // ID of the unit
  Int_t           fUnitTrackID;                         // ID of the unit
  Int_t           fUnitNum;                             // number of units
  Int_t           fUnitClusterID;                       // ID of the unit
  AliJetFinderUnitFlagType_t         fUnitFlag;         // Flag of the unit
  AliJetFinderUnitCutFlagType_t      fUnitCutFlag;      // Flag of the unit
  AliJetFinderUnitSignalFlagType_t   fUnitSignalFlag;   // Flag of the unit
  AliJetFinderUnitDetectorFlagType_t fUnitDetectorFlag; // Detector flag of the unit
  Float_t         fUnitPx;                              // Px of charged track
  Float_t         fUnitPy;                              // Py of charged track
  Float_t         fUnitPz;                              // Pz of charged track
  Float_t         fUnitMass;                            // Mass of charged particle

  ClassDef(AliJetUnitArray,1)
};

#endif
Commit	Line	Data
ee7de0dd	1	#ifndef ALIJETUNITARRAY_H
ee7de0dd	2	#define ALIJETUNITARRAY_H
4a01bb2c	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* * * See cxx source for full Copyright notice */
	6
ee7de0dd	7	//------------------------------------------------------------------
4a01bb2c	8	// Unit used by UA1 algorithm
ee7de0dd	9	// Authors: Sarah Blyth (LBL/UCT)
	10	// Magali Estienne (IReS) (new version for JETAN)
	11	//------------------------------------------------------------------
	12
4a01bb2c	13
	14	#include <TObject.h>
	15	#include "AliJetFinderTypes.h"
	16
	17	class AliJetUnitArray : public TObject
	18	{
	19	public:
	20	AliJetUnitArray();
ee7de0dd	21	AliJetUnitArray(Int_t absId, Int_t esdId, Float_t eta, Float_t phi, Float_t en, Float_t px, Float_t py, Float_t pz, Float_t Deta, Float_t Dphi, AliJetFinderUnitDetectorFlagType_t det, AliJetFinderUnitFlagType_t inout, AliJetFinderUnitCutFlagType_t cut, Float_t mass, Int_t clusId);
4a01bb2c	22	~AliJetUnitArray();
	23
	24	// Setter
	25	void SetUnitEnergy(Float_t energy) {fUnitEnergy = energy;}
	26	void SetUnitEta(Float_t eta) {fUnitEta = eta;}
	27	void SetUnitPhi(Float_t phi) {fUnitPhi = phi;}
	28	void SetUnitDeta(Float_t deta) {fUnitDeta = deta;}
	29	void SetUnitDphi(Float_t dphi) {fUnitDphi = dphi;}
	30	void SetUnitID(Int_t id) {fUnitID = id;}
ee7de0dd	31	void SetUnitTrackID(Int_t esdid) {fUnitTrackID = esdid;}
4a01bb2c	32	void SetUnitEntries(Int_t num) {fUnitNum = num;}
	33	void SetUnitClusterID(Int_t id) {fUnitClusterID = id;}
	34	void SetUnitFlag(AliJetFinderUnitFlagType_t flag)
	35	{
	36	fUnitFlag = flag;
	37	}
	38	void SetUnitCutFlag(AliJetFinderUnitCutFlagType_t cutFlag)
	39	{
	40	fUnitCutFlag = cutFlag;
	41	}
	42	void SetUnitSignalFlag(AliJetFinderUnitSignalFlagType_t signalFlag)
	43	{
	44	fUnitSignalFlag = signalFlag;
	45	}
	46	void SetUnitDetectorFlag(AliJetFinderUnitDetectorFlagType_t detectorflag)
	47	{
	48	fUnitDetectorFlag = detectorflag;
	49	}
ee7de0dd	50	void SetUnitPxPyPz(Double_t *pxyz) {fUnitPx = pxyz[0]; fUnitPy = pxyz[1]; fUnitPz = pxyz[2];}
ee7de0dd	51	void SetUnitMass(Float_t mass) {fUnitMass = mass;}
4a01bb2c	52
	53	// Getter
	54	Float_t GetUnitEnergy() const {return fUnitEnergy;}
	55	Float_t GetUnitEta() const {return fUnitEta;}
	56	Float_t GetUnitPhi() const {return fUnitPhi;}
	57	Float_t GetUnitDeta() const {return fUnitDeta;}
	58	Float_t GetUnitDphi() const {return fUnitDphi;}
	59	Int_t GetUnitID() const {return fUnitID;}
ee7de0dd	60	Int_t GetUnitTrackID() const {return fUnitTrackID;}
4a01bb2c	61	Int_t GetUnitEntries() const {return fUnitNum;}
4a01bb2c	62	Int_t GetUnitClusterID() const {return fUnitClusterID;}
ee7de0dd	63	Float_t GetUnitMass() const {return fUnitMass;}
	64	Bool_t GetUnitPxPyPz(Double_t* p) const {p[0]=fUnitPx; p[1]=fUnitPy; p[2]=fUnitPz; return kTRUE;}
	65
4a01bb2c	66	AliJetFinderUnitFlagType_t GetUnitFlag() const
	67	{
	68	return fUnitFlag;
	69	}
	70	AliJetFinderUnitCutFlagType_t GetUnitCutFlag() const
	71	{
	72	return fUnitCutFlag;
	73	}
	74	AliJetFinderUnitSignalFlagType_t GetUnitSignalFlag() const
	75	{
	76	return fUnitSignalFlag;
	77	}
	78	AliJetFinderUnitDetectorFlagType_t GetUnitDetectorFlag() const
	79	{
	80	return fUnitDetectorFlag;
	81	}
	82
ee7de0dd	83
4a01bb2c	84	Bool_t operator> ( AliJetUnitArray unit1) const;
	85	Bool_t operator< ( AliJetUnitArray unit1) const;
	86	Bool_t operator== ( AliJetUnitArray unit1) const;
	87
	88	protected:
ee7de0dd	89	Float_t fUnitEnergy; // Energy of the unit
	90	Float_t fUnitEta; // Eta of the unit
	91	Float_t fUnitPhi; // Phi of the unit
	92	Float_t fUnitDeta; // Delta Eta of the unit
	93	Float_t fUnitDphi; // Delta Phi of the unit
	94	Int_t fUnitID; // ID of the unit
	95	Int_t fUnitTrackID; // ID of the unit
	96	Int_t fUnitNum; // number of units
	97	Int_t fUnitClusterID; // ID of the unit
	98	AliJetFinderUnitFlagType_t fUnitFlag; // Flag of the unit
	99	AliJetFinderUnitCutFlagType_t fUnitCutFlag; // Flag of the unit
	100	AliJetFinderUnitSignalFlagType_t fUnitSignalFlag; // Flag of the unit
	101	AliJetFinderUnitDetectorFlagType_t fUnitDetectorFlag; // Detector flag of the unit
	102	Float_t fUnitPx; // Px of charged track
	103	Float_t fUnitPy; // Py of charged track
	104	Float_t fUnitPz; // Pz of charged track
	105	Float_t fUnitMass; // Mass of charged particle
4a01bb2c	106
ee7de0dd	107	ClassDef(AliJetUnitArray,1)
4a01bb2c	108	};
	109
	110	#endif
	111