#ifndef TMBEmcl_H
#define TMBEmcl_H

//////////////////////////////////////////////////////////////////////////
//                                                                      //
// TMBTree Emcl class                                                   //
//  E. Nagy and E. Thomas                                                           //
//  Revised by E. Nagy October 30th 2003                                //
//                                                                      //
//////////////////////////////////////////////////////////////////////////

#include "tmb_tree/TPhysObj.hpp"
#ifndef ROOT_TRef
#include "TRef.h"
#endif
#ifndef ROOT_TRefArray
#include "TRefArray.h"
#endif
#include "TArrayI.h"
#include "TArrayF.h"
#include "TString.h"


class TMBTrks;
class TMBCps;
class TMBFps;
class TMBEmCells;
class TMBVrts;

class TMBEmcl : public TPhysObj {

private:
  Float_t     _E;
  Float_t     _pT;
  Float_t     _eta;
  Float_t     _phi;
  Float_t     _pxyz[3];             //px py pz
  Float_t     _q;             //charge 
  Int_t       _id;
  Float_t     _iso;
  Float_t     _EisoCore;
  Int_t       _isol;
  Int_t       _nb_CPS;
  Int_t       _nb_FPS;
  Float_t     _floorE[6];
  Float_t     _floorX[6];
  Float_t     _floorY[6];
  Float_t     _floorZ[6];
  Float_t     _flrS1[6];
  Float_t     _flrS2[6];
  Float_t     _flrC[6];
  Int_t       _flrNch[6];
  Float_t     _EMfrac;
  Float_t     _HA;
  Float_t     _HMx7;
  Float_t     _HMx8;
  Float_t     _HMx41;
  Float_t     _elk8;
  Float_t     _elk41;
  Float_t     _Excess;
  Float_t     _rRC[5];
  Float_t     _rR1[5];
  Float_t     _rR2[5];
  TString     _algoname;
  TRef        _chptr;
  TRef        _vtxref;
  TRefArray   _cpsptr;
  TRefArray   _fpsptr;
  TRefArray   _ecells;
  // New variables from p14.05.00
  Bool_t      _is_in_fiducial;
  Bool_t      _is_in_phi_fiducial;
  Bool_t      _is_in_eta_fiducial;
  Float_t     _Calpx;
  Float_t     _Calpy;
  Float_t     _Calpz;
  Float_t     _Calpt;
  Float_t     _CalE;
  Float_t     _CalPhi;
  Float_t     _CalEta;
  Float_t     _CalDetectorPhi;
  Float_t     _CalDetectorEta;
  Float_t     _sigE;
  TRef        _CpsBest;
  Float_t     _CpsFitZVtx;
  Float_t     _CpsFitZVtxErr;
  Float_t     _CpsFitDCA;
  Float_t     _CpsFitDCAErr;
  Float_t     _CpsChi2Match;
  Float_t     _LhoodIso8;
  Float_t     _Lhood8;
  Bool_t      _has_track_match;
  Float_t     _TrMatchChi2ProbBest;
  Float_t     _SpatialTrMatchChi2ProbBest;
  TRef        _chptrBest;
  Int_t       _nChPart;
  TRefArray   _ChPart;
  TArrayF     _TrMatchChi2Prob;
  TArrayF     _SpatialTrMatchChi2Prob;

  TArrayF     _CPSFitChi2Match;
  TArrayF     _CPSFitZVtx;
  TArrayF     _CPSFitZVtxErr;
  TArrayF     _CPSFitDCA;
  TArrayF     _CPSFitDCAErr;
  // New variables for p17 photon/electron ID
  Float_t     _bestPSz;
  TArrayF     _PSz;
  // return true if EM object is matched with CPS object
  Int_t       _has_cps_match;
  TArrayI     _cps_match;
  // return distance in resolutions (fi,z) between CPS and EM, projected onto CPS
  Float_t     _bestcps_match_chi2;
  TArrayF     _cps_match_chi2;
  // number of EM-like CPS clusters
  Int_t       _nEMLikeCPSClusters;
  // 4 vector of di-photon candidate (E, px, py, pz)
  Float_t     _diphoton_p4v[4];

public:

TMBEmcl &operator=( TMBEmcl &orig) {
     return *this;
  }

TMBEmcl &operator=(const TMBEmcl &orig) {

    _E       = orig._E;
    _pT       = orig._pT;
    _eta      = orig._eta;
    _phi      = orig._phi;
 for(int j=0;j<3;j++)
    _pxyz[j]  = orig._pxyz[j];
    _q        = orig._q;
    _id       = orig._id;
    _iso      = orig._iso;
    _EisoCore = orig._EisoCore;
    _isol     = orig._isol;
    _nb_CPS   = orig._nb_CPS;
    _nb_FPS   = orig._nb_FPS;

 for(int j=0;j<6;j++){
    _floorE[j] = orig._floorE[j];
    _floorX[j] = orig._floorX[j];
    _floorY[j] = orig._floorY[j];
    _floorZ[j] = orig._floorZ[j];
    _flrS1[j] = orig._flrS1[j];
    _flrS2[j] = orig._flrS2[j];
    _flrC[j] = orig._flrC[j];
    _flrNch[j] = orig._flrNch[j];
  } 
    _EMfrac   = orig._EMfrac;
    _HA       = orig._HA;
    _HMx7     = orig._HMx7;
    _HMx8     = orig._HMx8;
    _HMx41    = orig._HMx41;
    _elk8     = orig._elk8; 
    _elk41    = orig._elk41;
    _Excess   = orig._Excess;
    _algoname = orig._algoname;
    _chptr    = orig._chptr;
    _vtxref   = orig._vtxref;
  for(int j=0;j<5;j++){
    _rRC[j] = orig._rRC[j];
    _rR1[j] = orig._rR1[j];
    _rR2[j] = orig._rR2[j];
  } 
    _cpsptr.Clear();
  for (Int_t i=0; i<orig._cpsptr.GetLast()+1; i++) {
       _cpsptr.Add( orig._cpsptr.At(i) );
  }

    _fpsptr.Clear();
  for (Int_t i=0; i<orig._fpsptr.GetLast()+1; i++) {
       _fpsptr.Add( orig._fpsptr.At(i) );
  }

    _ecells.Clear();
  for (Int_t i=0; i<orig._ecells.GetLast()+1; i++) {
       _ecells.Add( orig._ecells.At(i) );
  }

  _is_in_fiducial = orig._is_in_fiducial;
  _is_in_phi_fiducial = orig._is_in_phi_fiducial;
  _is_in_eta_fiducial = orig._is_in_eta_fiducial;
  _Calpx = orig._Calpx;
  _Calpy = orig._Calpy;
  _Calpz = orig._Calpz;
  _Calpt = orig._Calpt;
  _CalE = orig._CalE;
  _CalPhi = orig._CalPhi;
  _CalEta = orig._CalEta;
  _CalDetectorPhi = orig._CalDetectorPhi;
  _CalDetectorEta = orig._CalDetectorEta;
  _sigE = orig._sigE;
  _CpsBest = orig._CpsBest;
  _CpsFitZVtx = orig._CpsFitZVtx;
  _CpsFitZVtxErr = orig._CpsFitZVtxErr;
  _CpsFitDCA = orig._CpsFitDCA;
  _CpsFitDCAErr = orig._CpsFitDCAErr;
  _CpsChi2Match = orig._CpsChi2Match;
  _LhoodIso8 = orig._LhoodIso8;
  _Lhood8 = orig._Lhood8; 
  _has_track_match = orig._has_track_match;
  _TrMatchChi2ProbBest = orig._TrMatchChi2ProbBest;
  _SpatialTrMatchChi2ProbBest = orig._SpatialTrMatchChi2ProbBest;
  _chptrBest = orig._chptrBest;
  _nChPart = orig._nChPart;
  
  _ChPart.Clear();
  for (Int_t i=0; i<orig._ChPart.GetLast()+1; i++) {
    _ChPart.Add( orig._ChPart.At(i) );
  }

  _TrMatchChi2Prob = orig._TrMatchChi2Prob; 
  _SpatialTrMatchChi2Prob = orig._SpatialTrMatchChi2Prob; 
  
  _CPSFitChi2Match = orig._CPSFitChi2Match;
  _CPSFitZVtx      = orig._CPSFitZVtx;
  _CPSFitZVtxErr   = orig._CPSFitZVtxErr;
  _CPSFitDCA       = orig._CPSFitDCA;
  _CPSFitDCAErr    = orig._CPSFitDCAErr;
  
  // New photon/electron ID tools for p17
  _bestPSz            = orig._bestPSz;
  _PSz                = orig._PSz;
  _has_cps_match      = orig._has_cps_match;
  _cps_match          = orig._cps_match;
  _bestcps_match_chi2 = orig._bestcps_match_chi2;
  _cps_match_chi2     = orig._cps_match_chi2;
  _nEMLikeCPSClusters = orig._nEMLikeCPSClusters;
  for (Int_t i=0; i!=4; ++i) 
    _diphoton_p4v[i]       = orig._diphoton_p4v[i];

     return *this;
}

TMBEmcl (const TMBEmcl& orig):TPhysObj() {
    *this = orig;
}

TMBEmcl() {_vtxref =0;}

TMBEmcl(Float_t E, Float_t pT, Float_t eta, Float_t phi,
      Float_t* pxyz, Float_t q,
      Int_t id, Float_t iso, Float_t EisoCore, Int_t isol,
      Int_t nb_CPS, Int_t nb_FPS,
      Float_t* floorE, Float_t* floorX, Float_t* floorY, Float_t* floorZ, 
      Float_t* flrS1, Float_t* flrS2, Float_t* flrC, Int_t* flrNch,
      Float_t EMfrac, Float_t HA, Float_t HMx7, Float_t HMx8, Float_t HMx41,
      Float_t elk8, Float_t elk41, Float_t Excess, 
      Float_t* rRC, Float_t* rR1, Float_t* rR2, 
      const char *algoname,
      TRef chptr, TRef vtxref, 
      TRefArray* cpsptr, TRefArray* fpsptr, TRefArray* ecells);
    
void Set(Float_t E, Float_t pT, Float_t eta, Float_t phi,
	 Float_t* pxyz, Float_t q,
	 Int_t id, Float_t iso, Float_t EisoCore, Int_t isol,
	 Int_t nb_CPS, Int_t nb_FPS,
	 Float_t* floorE, Float_t* floorX, Float_t* floorY, Float_t* floorZ, 
	 Float_t* flrS1, Float_t* flrS2, Float_t* flrC, Int_t* flrNch,
	 Float_t EMfrac, Float_t HA, Float_t HMx7, Float_t HMx8, Float_t HMx41,
	 Float_t elk8, Float_t elk41, Float_t Excess, 
	 Float_t* rRC, Float_t* rR1, Float_t* rR2, 
	 const char *algoname,
	 TRef chptr, TRef vtxref, 
	 TRefArray* cpsptr, TRefArray* fpsptr, TRefArray* ecells);

void Set1(Bool_t is_in_fiducial, Bool_t is_in_phi_fiducial,
	  Bool_t is_in_eta_fiducial, Float_t Calpx,
	  Float_t Calpy, Float_t Calpz, Float_t Calpt, Float_t CalE,
	  Float_t CalPhi, Float_t CalEta, Float_t CalDetectorPhi,
	  Float_t CalDetectorEta, Float_t sigE, TRef CpsBest,
	  Float_t CpsFitZVtx, Float_t CpsFitZVtxErr,
	  Float_t CpsFitDCA, Float_t CpsFitDCAErr,
	  Float_t CpsChi2Match, Float_t LhoodIso8,
	  Float_t Lhood8, Bool_t has_track_match, Float_t TrMatchChi2ProbBest,
	  Float_t SpatialTrMatchChi2ProbBest, TRef chptrBest,
	  Int_t nChPart, TRefArray* ChPart, 
	  TArrayF* TrMatchChi2Prob, TArrayF* SpatialTrMatchChi2Prob,
	  TArrayF* CPSFitChi2Match, TArrayF* CPSFitZVtx, TArrayF* CPSFitZVtxErr,
	  TArrayF* CPSFitDCA, TArrayF* CPSFitDCAErr);
	  

void Set2(Float_t bestPSz, TArrayF* PSz, Int_t has_cps_match, TArrayI* cps_match, 
	  Float_t bestcps_match_chi2, TArrayF* cps_match_chi2, 
	  Int_t nEMLikeCPSClusters, Float_t* diphoton_p4v);

  Float_t       E()        const  {return _E;}
  Float_t       pT()       const  {return _pT;}
  Float_t       eta()      const  {return _eta;}
  Float_t       phi()      const  {return _phi;}
  Float_t       px()       const  {return _pxyz[0];}
  Float_t       py()       const  {return _pxyz[1];}
  Float_t       pz()       const  {return _pxyz[2];}
  Float_t       charge()   const  {return _q;}
  Int_t         id()       const  {return _id;}
  Int_t         typeID()   const  {return _id;}
  Float_t       iso()      const  {return _iso;}
  Float_t       EisoCore() const  {return _EisoCore;}
  Int_t         isol()     const  {return _isol;}
  Int_t         isolation() const  {return _isol;}
  Int_t         nb_CPS()   const  {return _nb_CPS;}
  Int_t         nb_FPS()   const  {return _nb_FPS;}
  Float_t*      floorE(Int_t nfloor)  {return &_floorE[nfloor];}
  Float_t*      floorX(Int_t nfloor)  {return &_floorX[nfloor];}
  Float_t*      floorY(Int_t nfloor)  {return &_floorY[nfloor];}
  Float_t*      floorZ(Int_t nfloor)  {return &_floorZ[nfloor];}
  Float_t*      flrS1(Int_t nfloor)   {return &_flrS1[nfloor];}
  Float_t*      flrS2(Int_t nfloor)   {return &_flrS2[nfloor];}
  Float_t*      flrC(Int_t nfloor)    {return &_flrC[nfloor];}
  Int_t*        flrNch(Int_t nfloor)  {return &_flrNch[nfloor];}
  Float_t       EMfrac()      const        {return _EMfrac;}
  Float_t       emfrac()      const        {return _EMfrac;}
  Float_t       HA()          const        {return _HA;}
  Float_t       HMx7()        const        {return _HMx7;}
  Float_t       HMx8()        const        {return _HMx8;}
  Float_t       HMx41()       const        {return _HMx41;}
  Float_t       elk8()        const        {return _elk8;}
  Float_t       elk41()       const        {return _elk41;}
  Float_t       Excess()      const        {return _Excess;}
  Float_t*      rRC(Int_t i)                {return &_rRC[i];}
  Float_t*      rR1(Int_t i)                {return &_rR1[i];}
  Float_t*      rR2(Int_t i)                {return &_rR2[i];}
  const char*   algoname()    const         {return _algoname;}
  TMBTrks*      GetChargedTrack() const    {return (TMBTrks*)_chptr.GetObject();}
  TMBVrts*      GetVertex()     const      {return (TMBVrts*)_vtxref.GetObject();}
  TMBCps*       GetCps(Int_t icps)const    {return (TMBCps*)_cpsptr.At(icps);}
  Int_t         ncps()         const       {return _cpsptr.GetLast()+1;}
  TMBFps*       GetFps(Int_t ifps) const   {return (TMBFps*)_fpsptr.At(ifps);}
  Int_t         nfps()          const      {return _fpsptr.GetLast()+1;}
  TMBEmCells*   GetEmCell(Int_t icell) const   {return (TMBEmCells*)_ecells.At(icell);}
  Int_t         ncells() const {return _ecells.GetLast()+1;}

  // New methods for  p14.05.00

  Bool_t       is_in_fiducial()        const        {return _is_in_fiducial;}
  Bool_t       is_in_phi_fiducial()    const        {return _is_in_phi_fiducial;}
  Bool_t       is_in_eta_fiducial()    const        {return _is_in_eta_fiducial;}
  Float_t      Calpx()                 const        {return _Calpx;}
  Float_t      Calpy()                 const        {return _Calpy;}
  Float_t      Calpz()                 const        {return _Calpz;}
  Float_t      Calpt()                 const        {return _Calpt;}
  Float_t      CalE()                  const        {return _CalE;}
  Float_t      CalPhi()                const        {return _CalPhi;}
  Float_t      CalEta()                const        {return _CalEta;}
  Float_t      CalDetectorPhi()        const        {return _CalDetectorPhi;}
  Float_t      CalDetectorEta()        const        {return _CalDetectorEta;}
  Float_t      sigE()                  const        {return _sigE;}
  TMBCps*      getPtrCPS()             const        {return (TMBCps*)_CpsBest.GetObject();}
  Float_t      CpsFitZVtx()            const        {return _CpsFitZVtx;}
  Float_t      CpsFitZVtxErr()         const        {return _CpsFitZVtxErr;}
  Float_t      CpsFitDCA()             const        {return _CpsFitDCA;}
  Float_t      CpsFitDCAErr()          const        {return _CpsFitDCAErr;}
  Float_t      CpsChi2Match()          const        {return _CpsChi2Match;}
  Float_t      LhoodIso8()             const        {return _LhoodIso8;}
  Float_t      Lhood8()                const        {return _Lhood8;}
  Bool_t       has_track_match()       const        {return  _has_track_match;}
  Float_t      track_match_chi2prob()  const        {return _TrMatchChi2ProbBest;}
  Float_t      track_match_spatialchi2prob()  const {return _SpatialTrMatchChi2ProbBest;}
  TMBTrks*     getPtrChp()             const        {return (TMBTrks*)_chptrBest.GetObject();}
  Int_t        nChPart()               const        {return _ChPart.GetLast()+1;}
  TMBTrks*     GetChPart(Int_t itr)    const        {return (TMBTrks*)_ChPart.At(itr);}
  Float_t      track_match_chi2prob(Int_t itr) const {return (Float_t)_TrMatchChi2Prob.At(itr);}
  Float_t      track_match_spatialchi2prob(Int_t itr) const {return (Float_t)_SpatialTrMatchChi2Prob.At(itr);}
  Float_t      CPSFitChi2Match(Int_t icps) const    {return (Float_t)_CPSFitChi2Match.At(icps);}
  Float_t      CPSFitZvtx(Int_t icps)  const        {return (Float_t)_CPSFitZVtx.At(icps);}
  Float_t      CPSFitZvtxErr(Int_t icps)const       {return (Float_t)_CPSFitZVtxErr.At(icps);}
  Float_t      CPSFitDCA(Int_t icps)   const        {return (Float_t)_CPSFitDCA.At(icps);}
  Float_t      CPSFitDCAErr(Int_t icps)const        {return (Float_t)_CPSFitDCAErr.At(icps);}
  // New photon/electron ID tools for p17
  Float_t      bestPSz()                  const {return (Float_t) _bestPSz;}
  Float_t      PSz(Int_t icps)            const {return (Float_t) _PSz.At(icps);}
  Int_t        has_cps_match()            const {return (Int_t)   _has_cps_match;}
  Int_t        cps_match(Int_t icps)      const {return (Int_t)   _cps_match.At(icps);}
  Float_t      bestcps_match_chi2()       const {return (Float_t) _bestcps_match_chi2;}
  Float_t      cps_match_chi2(Int_t icps) const {return (Float_t) _cps_match_chi2.At(icps);}
  Int_t        nEMLikeCPSClusters()       const {return (Int_t)   _nEMLikeCPSClusters;}
  Float_t*     diphoton_p4v()                   {return  _diphoton_p4v;}



  ~TMBEmcl() {;}

  ClassDef(TMBEmcl, 5)   //TMBTree Emcl class
};

#endif