// -------------------------------------------------------------------------
// -----                    CbmTrdFitTracks source file                -----
// -----                  Created 28/11/05  by D. Kresan               -----
// -----                 according to the CbmStsFitTracks              -----
// -------------------------------------------------------------------------


#include "iostream.h"
#include "TClonesArray.h"
#include "CbmRootManager.h"
#include "CbmRunAna.h"
#include "CbmRuntimeDb.h"
#include "CbmTrdFitTracks.h"
#include "CbmTrdTrackFitter.h"



// -----   Default constructor   -------------------------------------------
CbmTrdFitTracks::CbmTrdFitTracks() {
  fFitter        = NULL;
  fTrackArray    = NULL;
  fNofTracks     = 0;
}
// -------------------------------------------------------------------------



// -----   Standard constructor   ------------------------------------------
CbmTrdFitTracks::CbmTrdFitTracks(const char* name,
				 const char* title,
				 CbmTrdTrackFitter* fitter)
  : CbmTask(name) {
  fFitter        = fitter;
  fTrackArray    = NULL;
  fNofTracks     = 0;
}
// -------------------------------------------------------------------------



// -----   Destructor   ----------------------------------------------------
CbmTrdFitTracks::~CbmTrdFitTracks() { }
// -------------------------------------------------------------------------



// -----  SetParContainers -------------------------------------------------
void CbmTrdFitTracks::SetParContainers()
{
    CbmRunAna* ana = CbmRunAna::Instance();
    CbmRuntimeDb* rtdb = ana->GetRuntimeDb();
    rtdb->getContainer("CbmGeoPassivePar");
    rtdb->getContainer("CbmGeoStsPar");
    rtdb->getContainer("CbmGeoTrdPar");
}
// -------------------------------------------------------------------------



// -----   Public method Init (abstract in base class)  --------------------
InitStatus CbmTrdFitTracks::Init() {

  // Check for Track fitter
  if (! fFitter) {
    cout << "-E- CbmTrdFitTracks: No track fitter selected!" << endl;
    Fatal("CbmTrdFitTracks","No fitter selected");
    return kERROR;
  }

  // Get and check CbmRootManager
  CbmRootManager* ioman = CbmRootManager::Instance();
  if (! ioman) {
    cout << "-E- CbmTrdFitTracks::Init: "
	 << "RootManager not instantised!" << endl;
    return kFATAL;
  }

  // Get TrdTrack array
  fTrackArray  = (TClonesArray*) ioman->GetObject("TRDTrack"); //=>SG
  if ( ! fTrackArray) {
    cout << "-E- CbmTrdFitTracks::Init: No TrdTrack array!"
	 << endl;
    return kERROR;
  }

  // Call the Init method of the track fitter
  fFitter->Init();

  return kSUCCESS;

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



// -----   Public method Exec   --------------------------------------------
void CbmTrdFitTracks::Exec(Option_t* opt) {
  if ( !fTrackArray ) return; // =>SG
  Int_t nTracks = fTrackArray->GetEntriesFast();
  for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
    CbmTrdTrack* pTrack = (CbmTrdTrack*)fTrackArray->At(iTrack);
    fFitter->DoFit(pTrack);
  }
}
// -------------------------------------------------------------------------



// -----   Public method Finish   ------------------------------------------
void CbmTrdFitTracks::Finish() { }
// -------------------------------------------------------------------------




ClassImp(CbmTrdFitTracks)