#ifndef ALIUA1JETFINDERUNIT_H
#define ALIUA1JETFINDERUNIT_H

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

/* $Id$ */

//======================================================================
// Revised Version for JETAN - 08/11/2006
// Functions added
// Author: magali.estienne@ires.in2p3.fr
//======================================================================
//  Unit used by UA1 algorithm
//
//*-- Author: Sarah Blyth (LBL/UCT)

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

class AliJetUnitArray : public TObject
{
 public:
  AliJetUnitArray();
  ~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 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; 
  } 

  // 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   GetUnitEntries() const           {return fUnitNum;}
  Int_t   GetUnitClusterID() const         {return fUnitClusterID;}
  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           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

  ClassDef(AliJetUnitArray,3)
};

#endif