// Panda Headers ----------------------

// This Class' Header ------------------
#include "PndSttKalmanTask.h"

// C/C++ Headers ----------------------
#include <algorithm>
#include <iostream>
#include <assert.h>

// Collaborating Class Headers --------
#include "FairRootManager.h"
#include "TClonesArray.h"
#include "GFTrack.h"
#include "GFRecoHitFactory.h"
#include "TGeoTrack.h"
#include "TGeoManager.h"
#include "PndSttRecoHit.h"
#include "PndSttHit.h"
#include "GeaneTrackRep.h"
#include "GFException.h"
#include "GFKalman.h"

using namespace std;

// Class Member definitions -----------


PndSttKalmanTask::PndSttKalmanTask(){}

PndSttKalmanTask::~PndSttKalmanTask(){}

InitStatus
PndSttKalmanTask::Init()
{
  cout << "PndSttKalmanTask::Init()" << endl;

  FairRootManager *ioman = FairRootManager::Instance();
  
  if (!ioman) 
  {
      cout << "-E- PndSttRecoHirProducer: "
	   << "RootManager not instantised!" << endl;
      return kFATAL;
  }

  // open STTTrack array
  fTrackArray=(TClonesArray*) ioman->GetObject("Track");
  if(fTrackArray==0){
    Error("PndSttKalmanTask::Init","track-array not found!");
    return kERROR;
  }
  
  // Build hit factory -----------------------------
  _theRecoHitFactory = new GFRecoHitFactory();
  
  std::map<unsigned int,TString>::iterator iter=_hitBranchMap.begin();

  while(iter!=_hitBranchMap.end()){
    TClonesArray* ar=(TClonesArray*) ioman->GetObject(iter->second);
    if(ar==0){
      Error("PndSttKalmanTask::Init","point-array %s not found!",iter->second.Data());
    }
    else{ 
      _theRecoHitFactory->addProducer(iter->first,new GFRecoHitProducer<PndSttHit,PndSttRecoHit>(ar));
    }
    ++iter;
  }//end loops over hit types
  
 return kSUCCESS;
}


void
PndSttKalmanTask::Exec(Option_t* opt)
{
//   std::cout<<"PndSttKalmanTask::Exec"<<  std::endl;
  if(fTrackArray)
    {

      Int_t ntracks=fTrackArray->GetEntriesFast();
      //      cout << "ntracks " << ntracks << endl;

      for(Int_t itr=0;itr<ntracks;++itr){
	GFTrack* trk = (GFTrack*) fTrackArray->At(itr);
	trk->addHitVector(_theRecoHitFactory->createMany(trk->getCand()));
	std::cout<<trk->getNumHits()<<" hits in track " <<itr<<std::endl;
	GFKalman k;
        k.setLazy(1);
	k.setNumIterations(1);
	k.processTrack(trk);
      }
    }
  return;
}

void 
PndSttKalmanTask::AddHitBranch(unsigned int detId, const TString& m){
  
  std::cout << "hit branch " << m << " " << detId  << std::endl;
  _hitBranchMap[detId]=m;
}


ClassImp(PndSttKalmanTask)