#include <stdlib.h>
#include "CbmKFTrackInterface.h"
#include "CbmKFParticleDatabase.h"

CbmKFTrackInterface::CbmKFTrackInterface(): fMass(0), fChi2(0), fIsElectron(0), fNDF(0){
  for (Int_t i=0; i<6; i++) fT[i]=0.;
  for (Int_t i=0; i<15; i++) fC[i]=0.;
}

void CbmKFTrackInterface::SetTrack( const CbmL1Track &track ){
  for( Int_t i=0; i<6;  i++) fT[i] = track.GetTrack()[i];
  for( Int_t i=0; i<15; i++) fC[i] = track.GetCovMatrix()[i];
  fMass = track.GetMass();
  fIsElectron = track.IsElectron();
  fChi2 = track.GetRefChi2();
  fNDF  = track.GetRefNDF();
}

void CbmKFTrackInterface::SetPID( Int_t pidHypo )
{
  fMass = CbmKFParticleDatabase::Instance()->GetMass(pidHypo);
  fIsElectron = (abs(pidHypo)==11); 
}