#ifndef CBMLITCONVERTERFAIRTRACKPARAM_H_
#define CBMLITCONVERTERFAIRTRACKPARAM_H_

#include "base/CbmLitTypes.h"
#include "base/CbmLitEnums.h"

//#include "data/CbmLitHit.h"
//#include "data/CbmLitPixelHit.h"
//#include "data/CbmLitStripHit.h"
//#include "data/CbmLitTrack.h"
#include "data/CbmLitTrackParam.h"
//#include "data/CbmLitFitNode.h"
//#include "data/CbmLitTofTrack.h"

//#include "CbmTrack.h"
//#include "CbmTofTrack.h"
//#include "CbmHit.h"
//#include "CbmPixelHit.h"
//#include "CbmStripHit.h"
#include "CbmTrackParam.h"
//#include "CbmStsTrack.h"
//#include "CbmTrdTrack.h"
//#include "CbmMuchStrawHit.h"
//#include "CbmMuchGeoScheme.h"
//#include "CbmMvdHit.h"
//#include "CbmMuchTrack.h"

//#include "CbmGlobalTrack.h"
//#include "CbmTrdAddress.h"

#include "TClonesArray.h"

#include <iostream>
#include <cmath>
#include <set>
#include <cassert>


class CbmLitConverterFairTrackParam
{
public:
    
    static void FairTrackParamToCbmLitTrackParam(
                                                 const FairTrackParam* par,
                                                 CbmLitTrackParam* litPar)
    {
        litPar->SetX((litfloat)par->GetX());
        litPar->SetY((litfloat)par->GetY());
        litPar->SetZ((litfloat)par->GetZ());
        litPar->SetTx((litfloat)par->GetTx());
        litPar->SetTy((litfloat)par->GetTy());
        litPar->SetQp((litfloat)par->GetQp());
        litPar->SetTime(0);
        Double_t cov[15];
        par->CovMatrix(cov);
        std::vector<litfloat> covVec(21);
        for (UInt_t i = 0; i < 5; ++i) { covVec[i] = (litfloat) cov[i]; }
        for (UInt_t i = 5; i < 9; ++i) { covVec[i + 1] = (litfloat) cov[i]; }
        for (UInt_t i = 9; i < 12; ++i) { covVec[i + 2] = (litfloat) cov[i]; }
        for (UInt_t i = 12; i < 14; ++i) { covVec[i + 3] = (litfloat) cov[i]; }
        covVec[18] = (litfloat) cov[14];
        covVec[5] = 0;
        covVec[10] = 0;
        covVec[5] = 0;
        covVec[14] = 0;
        covVec[17] = 0;
        covVec[19] = 0;
        covVec[20] = 4 * 4;
        litPar->SetCovMatrix(covVec);
    }
    
    static void FairTrackParamToCbmLitTrackParam(
                                                 const CbmTrackParam* par,
                                                 CbmLitTrackParam* litPar)
    {
        litPar->SetX((litfloat)par->GetX());
        litPar->SetY((litfloat)par->GetY());
        litPar->SetZ((litfloat)par->GetZ());
        litPar->SetTx((litfloat)par->GetTx());
        litPar->SetTy((litfloat)par->GetTy());
        litPar->SetQp((litfloat)par->GetQp());
        litPar->SetTime((litfloat)par->GetTime());
        Double_t cov[15];
        par->CovMatrix(cov);
        std::vector<litfloat> covVec(21);
        for (UInt_t i = 0; i < 5; ++i) { covVec[i] = (litfloat) cov[i]; }
        for (UInt_t i = 5; i < 9; ++i) { covVec[i + 1] = (litfloat) cov[i]; }
        for (UInt_t i = 9; i < 12; ++i) { covVec[i + 2] = (litfloat) cov[i]; }
        for (UInt_t i = 12; i < 14; ++i) { covVec[i + 3] = (litfloat) cov[i]; }
        covVec[18] = (litfloat) cov[14];
        covVec[5] = 0;
        covVec[10] = 0;
        covVec[5] = 0;
        covVec[14] = 0;
        covVec[17] = 0;
        covVec[19] = 0;
        covVec[20] = (litfloat)(par->GetDTime() * par->GetDTime());
        litPar->SetCovMatrix(covVec);
    }
    
    static void CbmLitTrackParamToFairTrackParam(
                                                 const CbmLitTrackParam* litPar,
                                                 FairTrackParam* par)
    {
        par->SetX(litPar->GetX());
        par->SetY(litPar->GetY());
        par->SetZ(litPar->GetZ());
        par->SetTx(litPar->GetTx());
        par->SetTy(litPar->GetTy());
        par->SetQp(litPar->GetQp());
        Double_t cov[15];
        std::vector<litfloat> covVec(litPar->GetCovMatrix());
        for (UInt_t i = 0; i < 5; ++i) { cov[i] = (Double_t) covVec[i]; }
        for (UInt_t i = 5; i < 9; ++i) { cov[i] = (Double_t) covVec[i + 1]; }
        for (UInt_t i = 9; i < 12; ++i) { cov[i] = (Double_t) covVec[i + 2]; }
        for (UInt_t i = 12; i < 14; ++i) { cov[i] = (Double_t) covVec[i + 3 ]; }
        cov[14] = (Double_t) covVec[18];
        par->SetCovMatrix(cov);
    }

};

#endif /*CBMLITCONVERTERFAIRTRACKPARAM_H_*/