LitCudaConverter.h

Go to the documentation of this file.

 
#ifndef LITCUDACONVERTER_H_
#define LITCUDACONVERTER_H_
 
#include "CbmLitPixelHit.h"
#include "CbmLitTrack.h"
#include "CbmLitTrackParam.h"
 
#include "LitCudaHit.h"
#include "LitCudaTrack.h"
#include "LitCudaTrackParam.h"
 
/* Converts CbmLitPixelHit to LitCudaPixelHit.
 * @param hit Pointer to CbmLitPixelHit.
 * @param lhit Pointer to LitScalPixelHit.
 */
void CbmLitPixelHitToLitCudaPixelHit(const CbmLitPixelHit* hit,
                                     LitCudaPixelHit* lhit) {
  lhit->X       = hit->GetX();
  lhit->Y       = hit->GetY();
  lhit->Dx      = hit->GetDx();
  lhit->Dy      = hit->GetDy();
  lhit->Dxy     = hit->GetDxy();
  lhit->planeId = hit->GetPlaneId();
  lhit->refId   = hit->GetRefId();
  lhit->Z       = hit->GetZ();
}
 
/* Converts CbmLitTrackParam to LitCudaTrackParam.
 * @param par Pointer to CbmLitTrackParam.
 * @param lpar Pointer to LitCudaTrackParam.
 */
void CbmLitTrackParamToLitCudaTrackParam(const CbmLitTrackParam* par,
                                         LitCudaTrackParam* lpar) {
  lpar->X   = par->GetX();
  lpar->Y   = par->GetY();
  lpar->Tx  = par->GetTx();
  lpar->Ty  = par->GetTy();
  lpar->Qp  = par->GetQp();
  lpar->Z   = par->GetZ();
  lpar->C0  = par->GetCovariance(0);
  lpar->C1  = par->GetCovariance(1);
  lpar->C2  = par->GetCovariance(2);
  lpar->C3  = par->GetCovariance(3);
  lpar->C4  = par->GetCovariance(4);
  lpar->C5  = par->GetCovariance(5);
  lpar->C6  = par->GetCovariance(6);
  lpar->C7  = par->GetCovariance(7);
  lpar->C8  = par->GetCovariance(8);
  lpar->C9  = par->GetCovariance(9);
  lpar->C10 = par->GetCovariance(10);
  lpar->C11 = par->GetCovariance(11);
  lpar->C12 = par->GetCovariance(12);
  lpar->C13 = par->GetCovariance(13);
  lpar->C14 = par->GetCovariance(14);
}
 
/* Converts LitCudaTrackParam to CbmLitTrackParam.
 * @param lpar Pointer to LitCudaTrackParam.
 * @param par Pointer to CbmLitTrackParam.
 */
void LitCudaTrackParamToCbmLitTrackParam(const LitCudaTrackParam* lpar,
                                         CbmLitTrackParam* par) {
  par->SetX(lpar->X);
  par->SetY(lpar->Y);
  par->SetTx(lpar->Tx);
  par->SetTy(lpar->Ty);
  par->SetQp(lpar->Qp);
  par->SetZ(lpar->Z);
  par->SetCovariance(0, lpar->C0);
  par->SetCovariance(1, lpar->C1);
  par->SetCovariance(2, lpar->C2);
  par->SetCovariance(3, lpar->C3);
  par->SetCovariance(4, lpar->C4);
  par->SetCovariance(5, lpar->C5);
  par->SetCovariance(6, lpar->C6);
  par->SetCovariance(7, lpar->C7);
  par->SetCovariance(8, lpar->C8);
  par->SetCovariance(9, lpar->C9);
  par->SetCovariance(10, lpar->C10);
  par->SetCovariance(11, lpar->C11);
  par->SetCovariance(12, lpar->C12);
  par->SetCovariance(13, lpar->C13);
  par->SetCovariance(14, lpar->C14);
}
 
/* Concerts LitCudaTrack to CbmLitTrack.
 * @param ltrack Pointer to LitCudaTrack.
 * @param track Pointer to CbmLitTrack.
 */
inline void LitCudaTrackToCbmLitTrack(LitCudaTrack* ltrack,
                                      CbmLitTrack* track) {
  // Convert track parameters
  CbmLitTrackParam parFirst, parLast;
  LitCudaTrackParamToCbmLitTrackParam(&(ltrack->paramFirst), &parFirst);
  LitCudaTrackParamToCbmLitTrackParam(&(ltrack->paramLast), &parLast);
  track->SetParamLast(&parLast);
  track->SetParamFirst(&parFirst);
 
  // Copy other track parameters
  track->SetChi2(ltrack->chiSq);
  track->SetNDF(ltrack->NDF);
  track->SetPreviousTrackId(ltrack->previouseTrackId);
  track->SetQuality(kLITGOOD);
  // track->SetRefId(std::distance(ltracks.begin(), it));
 
  // Convert hits
  for (int i = 0; i < ltrack->nofHits; i++) {
    CbmLitPixelHit* newHit = new CbmLitPixelHit;
    newHit->SetX(ltrack->hits[i]->X);
    newHit->SetY(ltrack->hits[i]->Y);
    newHit->SetDx(ltrack->hits[i]->Dx);
    newHit->SetDy(ltrack->hits[i]->Dy);
    newHit->SetDxy(ltrack->hits[i]->Dxy);
    newHit->SetPlaneId(ltrack->hits[i]->planeId);
    newHit->SetHitType(kLITPIXELHIT);
    newHit->SetDetectorId(kLITMUCH);
    newHit->SetRefId(ltrack->hits[i]->refId);
    newHit->SetZ(ltrack->hits[i]->Z);
    //    std::cout << ltrack->hits[i];
    track->AddHit(newHit);
  }
}
 
/* Converts CbmLitTrack to LitCudaTrack.
 * @param track Pointer to CbmLitTrack.
 * @param ltrack Pointer to LitCudaTrack.
 */
inline void CbmLitTrackToLitCudaTrack(CbmLitTrack* track,
                                      LitCudaTrack* ltrack) {
  // Convert track parameters
  // CbmLitTrackParamToLitTrackParamScal(track->GetParamFirst(), &(ltrack->paramFirst));
  CbmLitTrackParamToLitCudaTrackParam(track->GetParamLast(),
                                      &(ltrack->paramLast));
 
  // Copy other track parameters
  // ltrack->chiSq = track->GetChi2();
  // ltrack->NDF = track->GetNDF();
  // ltrack->previouseTrackId = track->GetPreviousTrackId();
 
  // Convert hits
  for (int i = 0; i < track->GetNofHits(); i++) {
    LitCudaPixelHit* newHit = new LitCudaPixelHit;
    CbmLitPixelHit* hit     = (CbmLitPixelHit*) track->GetHit(i);
    newHit->X               = hit->GetX();
    newHit->Y               = hit->GetY();
    newHit->Dx              = hit->GetDx();
    newHit->Dy              = hit->GetDy();
    newHit->Dxy             = hit->GetDxy();
    //    newHit->planeId = hit->GetPlaneId();
    //    newHit->refId = hit->GetRefId();
    //    newHit->Z = hit->GetZ();
    ltrack->AddHit(newHit);
  }
}
#endif /* LITCUDACONVERTER_H_ */