o small fixes

[u/mrichter/AliRoot.git] / TPC / Upgrade / ToyMCEvent.h

#ifndef ToyMCEvent_H
#define ToyMCEvent_H


#include <TClonesArray.h>
#include "ToyMCTrack.h"

class ToyMCEvent : public TObject {
 public:
  ToyMCEvent();
  ToyMCEvent(const ToyMCEvent &event);
  virtual ~ToyMCEvent() {}
  ToyMCEvent& operator = (const ToyMCEvent &event);
  

  ToyMCTrack* AddTrack(const ToyMCTrack &track);

  Int_t GetNumberOfTracks() const { return fTracks.GetEntriesFast(); }
  const ToyMCTrack* GetTrack(Int_t track) const { return static_cast<const ToyMCTrack*>(fTracks.At(track)); }
    
  void SetT0 (Float_t time)              { fT0   = time;          }
  void SetX(Float_t var)                 { fX = var;              }
  void SetY(Float_t var)                 { fY = var;              }
  void SetZ(Float_t var)                 { fZ = var;              }

  UInt_t GetEventNumber()    const {return fEventNumber;  }
  Float_t GetT0()            const {return fT0;           }
  Float_t GetX()             const {return fX;            }
  Float_t GetY()             const {return fY;            }
  Float_t GetZ()             const {return fZ;            }
  
 private:
  static Int_t evCounter;
    
  UInt_t fEventNumber;
    
  Float_t fT0;
  Float_t fX;
  Float_t fY;
  Float_t fZ;

  TClonesArray fTracks;

  ClassDef(ToyMCEvent, 1);

};

#endif