// -------------------------------------------------------------------------
// -----                      CbmStsTrack source file                  -----
// -----                  Created 26/01/05  by V. Friese               -----
// -------------------------------------------------------------------------


#include "iostream.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"


// -----   Default constructor   -------------------------------------------
CbmStsTrack::CbmStsTrack() {
  fPidHypo = 0;
  fFlag    = 0;
  fChi2    = 0.;
}
// -------------------------------------------------------------------------



// -----   Destructor   ----------------------------------------------------
CbmStsTrack::~CbmStsTrack() {
  fHitMap.clear();
}
// -------------------------------------------------------------------------



// -----   Public method AddHit   ------------------------------------------
void CbmStsTrack::AddHit(Int_t hitId, CbmStsHit* hit) {
  fHitMap[hit->GetZ()] = hitId;
}
// -------------------------------------------------------------------------



// -----   Public method Print   -------------------------------------------
void CbmStsTrack::Print() {
  cout << " Number of attached STS hits : " 
       << fHits.GetSize()  << endl;
  fParamFirst.Print();
  cout << " Chi2: " << fChi2 << ", Quality flag " << fFlag << endl;
}
// -------------------------------------------------------------------------



// -----   Public method SortHits   ----------------------------------------
void CbmStsTrack::SortHits() {

  Int_t index=0;
  map<Double_t, Int_t>::iterator it;

  fHits.Reset();
  fHits.Set(fHitMap.size());
  index = 0;
  for (it = fHitMap.begin(); it != fHitMap.end(); it++) {
    fHits[index] = it->second;
    index++;
  }

}
// -------------------------------------------------------------------------



ClassImp(CbmStsTrack)