// -------------------------------------------------------------------------
// -----                      CbmSttTrack source file                  -----
// -----                  Created 28/03/06  by R. Castelijns           -----
// -------------------------------------------------------------------------


#include "iostream.h"
#include "CbmSttTrack.h"
#include "CbmSttVertex.h"

// -----   Default constructor   -------------------------------------------
CbmSttVertex::CbmSttVertex() 
{
  fTracks.Set(2);
  fMCTracks.Set(2);
  fNumberOfTracks = 0;
}
// -------------------------------------------------------------------------

// -----   Destructor   ----------------------------------------------------
CbmSttVertex::~CbmSttVertex() 
{
  fTracks.Reset();
}
// -------------------------------------------------------------------------
void CbmSttVertex::SetVertex(Double_t x, Double_t y, Double_t z)
{
  fX = x;
  fY = y;
  fZ = z;
}

// -----   Public method AddTrack   --------------------------------------
void CbmSttVertex::AddTrack(Int_t recoID, Int_t mcID) 
{
  if (fNumberOfTracks == 0)
    {
      fTracks[0] = recoID;
      fMCTracks[0] = mcID;
      fNumberOfTracks++;
    }
  else if (fNumberOfTracks == 1)
    {
      fTracks[1] = recoID;
      fMCTracks[1] = mcID;
      fNumberOfTracks++;
    }
}
// -------------------------------------------------------------------------

// -----   Public method GetTrack   --------------------------------------
Int_t CbmSttVertex::GetTrack(Int_t index) 
{
  Int_t
    retval = -1;

  if (index < 2)
    {
      retval = fTracks[index];
    }

  return retval;
}
// -------------------------------------------------------------------------

// -----   Public method GetMCTrack   --------------------------------------
Int_t CbmSttVertex::GetMCTrack(Int_t index) 
{
  Int_t
    retval = -1;

  if (index < 2)
    {
      retval = fMCTracks[index];
    }

  return retval;
}
// -------------------------------------------------------------------------

// -----   Public method Print   -------------------------------------------
void CbmSttVertex::Print() 
{
  cout << "CbmSttVertex::Print() not yet implemented" << endl;
}
// -------------------------------------------------------------------------

ClassImp(CbmSttVertex)