new module for hadron correlations

[u/mrichter/AliRoot.git] / PWG4 / JetCorrel / CorrelList.h

#ifndef __CORRELLIST_H__
#define __CORRELLIST_H__
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */
/* $Id:  $ */

//__________________________________________
// Three classes: a particle list class CorrelList_t with its
// node class CorrelListNode_t and iterator class CorrelListIter_t
//-- Author: Paul Constantin

#include "CorrelParticle.h"
#include "CorrelTrack.h"
#include "CorrelRecoParent.h"

namespace JetCorrelHD {

  class CorrelList_t;
  class CorrelListIter_t;
  
  class CorrelListNode_t {
    public:
      CorrelListNode_t(CorrelParticle_t* p, CorrelListNode_t* n) : fPart(p), fNext(n) {}
      ~CorrelListNode_t() {if(fPart) delete fPart; fNext = NULL;}
      const CorrelListNode_t& operator=(const CorrelListNode_t& rhs);
      CorrelParticle_t* GetData() const {return fPart;}
      CorrelListNode_t* GetNext() const {return fNext;}

    private:
      CorrelParticle_t* fPart;
      CorrelListNode_t* fNext;      
  };

  class CorrelListIter_t {
    public:
      CorrelListIter_t() : fCurr(NULL) {}
      CorrelListIter_t(CorrelListNode_t* theNode) : fCurr(theNode) {}
      ~CorrelListIter_t(){;}
      CorrelListIter_t(const CorrelListIter_t& rhs) : fCurr(rhs.fCurr) {}
      const CorrelListIter_t& operator=(const CorrelListIter_t& rhs);

      Bool_t HasEnded() const {return (fCurr==NULL);}
      void Check();
      void Move() {Check(); fCurr=fCurr->GetNext();}
      CorrelListNode_t* Node() {Check(); return fCurr;}
      CorrelParticle_t* Data() {Check(); return fCurr->GetData();}

    private:
      CorrelListNode_t* fCurr; // iterator "current node"
  };

  class CorrelList_t : public TObject {
    public:
      CorrelList_t();
      virtual ~CorrelList_t() {Reset();}
      void Reset();
      const CorrelList_t& operator=(const CorrelList_t& rhs); // makes shallow copy
      CorrelList_t* DeepCopy();                     // use this method to get a deep copy

      void Push(CorrelParticle_t* p);
      UInt_t Size() const {return fSize;}
      UInt_t EvtID() const {return fEvtID;}
      PartType_t PartID() const {return fPartID;}
      PoolType_t PoolID() const {return fPoolID;}
      Bool_t Filled() const {return fFilled;}
      void SetFilled(Bool_t f) {fFilled=f;}
      void Label(PartType_t p, PoolType_t l, UInt_t e) {fPartID=p; fPoolID=l; fEvtID=e;}
      CorrelListIter_t Head() const {return CorrelListIter_t(fHead);}
      void ShowHead();
      void Show();

    private:
      UInt_t fSize;             // list size
      UInt_t fEvtID;            // event ID
      Bool_t fFilled;           // is filled
      PartType_t fPartID;       // particle ID
      PoolType_t fPoolID;       // pool type
      CorrelListNode_t* fHead;  // list head

      CorrelList_t(const CorrelList_t& rhs);        // forbid copy constructor
  };

} // namespace declaration

#endif