#ifndef CBMSTTVERTEX_H
#define CBMSTTVERTEX_H 

#include "TArrayI.h"
#include "TObject.h"

class CbmSttTrack;

class CbmSttVertex : public TObject
{
 public:
  /** Default constructor **/
  CbmSttVertex();

  /** Destructor **/
  virtual ~CbmSttVertex();

  /** Public methods AddHit
   ** Adds the hit index to the index array
   **/
  void AddTrack(Int_t trackID, Int_t mcID);
  Int_t GetTrack(Int_t index); 
  Int_t GetMCTrack(Int_t index); 
  void SetVertex(Double_t x, Double_t y, Double_t z);
  Double_t GetVertexX() const {return fX;};
  Double_t GetVertexY() const {return fY;};
  Double_t GetVertexZ() const {return fZ;};

  /** Public method Print
   ** Output to screen 
   **/
  void Print();

  /** Accessors  **/
  Int_t GetNofTracks()                const { return fTracks.GetSize(); };

 private:
  /** Arrays containg the indices of the tracks attached to the vertex **/
  TArrayI fTracks;
  TArrayI fMCTracks;

  Double_t fX, fY, fZ;

  Int_t fNumberOfTracks;
  
  ClassDef(CbmSttVertex,1);
};

#endif