PWG4/JCORRAN/AliPhJMCTrackList.h

   1 // $Id: AliPhJMCTrackList.h,v 1.4 2008/05/08 13:44:45 djkim Exp $
   2
   3 ////////////////////////////////////////////////////
   4 /*!
   5   \file AliPhJMCTrackList.hh
   6   \brief
   7   \author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla)
   8   \email: djkim@jyu.fi
   9   \version $Revision: 1.4 $
  10   \date $Date: 2008/05/08 13:44:45 $
  11 */
  12 ////////////////////////////////////////////////////
  13
  14 #ifndef ALIPHJMCTRACKLIST_H
  15 #define ALIPHJMCTRACKLIST_H
  16
  17 #ifndef ROOT_TObject
  18 #include <TObject.h>
  19 #endif
  20
  21 #include "TClonesArray.h"
  22 #include <iostream>
  23
  24 #include "JConst.h"
  25 #include "AliJMCTrack.h"
  26
  27 class AliJMCTrack;
  28 class TClonesArray;
  29
  30 class AliPhJMCTrackList : public TObject {
  31
  32 public:
  33   AliPhJMCTrackList();
  34   AliPhJMCTrackList(expName exp);
  35   AliPhJMCTrackList(const AliPhJMCTrackList& a);
  36   virtual ~AliPhJMCTrackList();
  37
  38   void Reset();
  39   //getters
  40   unsigned short GetNTracks() const { return fTracks; }
  41   AliJMCTrack*        GetTrack(const unsigned int itrk);
  42   //setters
  43   void SetNTracks(const unsigned short ntrk) { fTracks = ntrk; }
  44   int  SetTClonesArraySize(const unsigned int ntrk);
  45   // add tracks
  46   void AddJMCTrack(const unsigned int itrk);     // MC add
  47
  48   AliPhJMCTrackList& operator=(const AliPhJMCTrackList&  list);
  49
  50 protected:
  51   TClonesArray *GetList() const { return fMcTrackList; }
  52   TClonesArray *fMcTrackList;    //list of MC tracks
  53   unsigned short fTracks;        //number of objects in the list
  54
  55 private:
  56   ClassDef(AliPhJMCTrackList,1)
  57
  58 };
  59
  60 #endif