//modified by Elisabetta Prencipe, 19/05/2014

#include"PndGenfitAdapters.h"

#include <iostream>

#include"Track.h"
#include"AbsTrackRep.h"
#include"TrackCand.h"
#include"PndTrack.h"
#include"PndTrackCand.h"
#include"DetPlane.h"
#include"Exception.h"
#include"TMatrixT.h"
#include"FairTrackParP.h"

//#include"GeaneTrackRep.h"
#include"RKTrackRep.h"
#include <cmath>

PndTrackCand* GenfitTrackCand2PndTrackCand(const genfit::TrackCand* cand){
  PndTrackCand* retVal = new PndTrackCand();
  unsigned int nhits = cand->getNHits();
  unsigned detId,hitId;
  double rho;
  for(unsigned int i=0;i<nhits;++i){
    cand->getHit(i,detId,hitId,rho);
    retVal->AddHit(detId,hitId,rho);
  }
  retVal->setMcTrackId(cand->getMcTrackId());
  retVal->setTrackSeed(cand->getPosSeed(),
		       cand->getDirSeed(),
		       cand->getQoverPseed());
  return retVal;
}

genfit::TrackCand* PndTrackCand2GenfitTrackCand(PndTrackCand* cand){
  genfit::TrackCand* retVal = new genfit::TrackCand();
  unsigned int nhits = cand->GetNHits();
  for(unsigned int i=0;i<nhits;++i){
    PndTrackCandHit candHit = cand->GetSortedHit(i);
    retVal->addHit(candHit.GetDetId(),candHit.GetHitId(),candHit.GetRho(),i);
  }
  retVal->setMcTrackId(cand->getMcTrackId());
  retVal->setTrackSeed(cand->getPosSeed(),
		       cand->getDirSeed(),
		       cand->getQoverPseed());
  return retVal;
}

PndTrack* GenfitTrack2PndTrack(const genfit::Track* tr){
  genfit::AbsTrackRep* clone = tr->getCardinalRep()->clone();
  TMatrixT<double> firstState = clone->getFirstState();
  TMatrixT<double> lastState = clone->getLastState();
  TMatrixT<double> firstCov = clone->getFirstCov();
  TMatrixT<double> lastCov = clone->getLastCov();
  genfit::DetPlane firstPlane = clone->getFirstPlane();
  genfit::DetPlane lastPlane = clone->getLastPlane();
  
  genfit::AbsTrackRep* gtr;
  if (dynamic_cast<RKTrackRep*>(clone)!=NULL)
	  gtr = dynamic_cast<RKTrackRep*>(clone);
  else {
    std::cerr << " GenfitAbsTrackRep2PndTrack() can currently only handle GeaneTrackRep and RKTrackRep" << std::endl;
    throw;
  }

  //make the FairTrackParP for first and last hit
  double firstCova[15];
  int count=0;;
  for(int i=0; i<5;++i){
     for(int j=i;j<5;++j){
	   firstCova[count++]=firstCov[i][j];
     }
  }
  double lastCova[15];
  count=0;;
  for(int i=0; i<5;++i){
    for(int j=i;j<5;++j){
	  lastCova[count++]=lastCov[i][j];
    }
  }

  //  calculation of spu = sign[p(DJ x DK)]
  double first_pro(0), last_pro(0), first_spu, last_spu;
  bool exc(false);

  try{
    first_pro = gtr->getMom(firstPlane).Dot(firstPlane.getNormal());
    last_pro = gtr->getMom(lastPlane).Dot(lastPlane.getNormal());
  }
  catch (genfit::Exception& e){
    exc=true;
    std::cerr<<"could not convert GenfitTrack to PndTrack"<<std::endl;
    e.what();
  }

  first_spu = (first_pro>=0) ? 1 : -1;
  last_spu = (last_pro>=0) ? 1 : -1;
    
  FairTrackParP first(firstState[3][0],firstState[4][0],firstState[1][0],firstState[2][0],firstState[0][0],firstCova,firstPlane.getO(),firstPlane.getU(),firstPlane.getV(),first_spu);
  FairTrackParP last(lastState[3][0],lastState[4][0],lastState[1][0],lastState[2][0],lastState[0][0],lastCova,lastPlane.getO(),lastPlane.getU(),lastPlane.getV(),last_spu);
    
  //copy the trackCand
  genfit::TrackCand genfitCand = tr->getCand();
  PndTrackCand* pndCand = GenfitTrackCand2PndTrackCand(&genfitCand);
  PndTrack* retVal =  new PndTrack(first,last,*pndCand);
  retVal->SetChi2(tr->getChiSqu());
  retVal->SetNDF(tr->getNDF());
  if (tr->getNDF()==0 || exc) {
    retVal->SetFlag(-1);
  }
  else {
	  retVal->SetFlag(1);
  }

  delete pndCand;
  return retVal;
  
}