CbmLitAllHitsTofMerger.cxx

Go to the documentation of this file.

 
#include "finder/CbmLitAllHitsTofMerger.h"
 
#include "data/CbmLitHit.h"
#include "data/CbmLitTofTrack.h"
#include "data/CbmLitTrack.h"
#include "data/CbmLitTrackParam.h"
#include "interface/CbmLitTrackPropagator.h"
#include "interface/CbmLitTrackUpdate.h"
#include "utils/CbmLitMath.h"
 
#include <iostream>
#include <map>
 
using std::map;
 
CbmLitAllHitsTofMerger::CbmLitAllHitsTofMerger()
  : fFieldPropagator()
  , fLinePropagator()
  , fFilter()
  , fPDG(211)
  , fChiSqCut(25.) {}
 
CbmLitAllHitsTofMerger::~CbmLitAllHitsTofMerger() {}
 
LitStatus CbmLitAllHitsTofMerger::DoMerge(HitPtrVector& hits,
                                          TrackPtrVector& tracks,
                                          TofTrackPtrVector& tofTracks) {
  // First find hit with minimum Z position and build map from Z hit position
  // to track parameter to improve the calculation speed.
  litfloat zMin = std::numeric_limits<litfloat>::max();
  map<litfloat, CbmLitTrackParam> zParamMap;
  for (HitPtrIterator it = hits.begin(); it != hits.end(); it++) {
    const CbmLitHit* hit   = *it;
    zMin                   = std::min(zMin, hit->GetZ());
    zParamMap[hit->GetZ()] = CbmLitTrackParam();
  }
 
  // Loop over input tracks
  for (TrackPtrIterator it = tracks.begin(); it != tracks.end(); it++) {
    CbmLitTrack* track = *it;
    if (track->GetQuality() != kLITGOODMERGE) { continue; }
    CbmLitTrackParam par(*track->GetParamLast());
 
    // Extrapolate track minimum Z position of hit using magnetic field propagator.
    if (fFieldPropagator->Propagate(&par, zMin, fPDG) == kLITERROR) {
      continue;  // skip this track if its propagation failed
    }
 
    // Extrapolate track parameters to each Z position in the map.
    // This is done to improve calculation speed.
    // In case of planar TOF geometry only 1 track extrapolation is required,
    // since all hits located at the same Z.
    for (map<litfloat, CbmLitTrackParam>::iterator it2 = zParamMap.begin();
         it2 != zParamMap.end();
         it2++) {
      (*it2).second = par;
      fLinePropagator->Propagate(&(*it2).second, (*it2).first, fPDG);
    }
 
    // Loop over hits
    for (HitPtrIterator it = hits.begin(); it != hits.end(); it++) {
      const CbmLitHit* hit = *it;
      if (zParamMap.find(hit->GetZ())
          == zParamMap.end()) {  // This should never happen
        std::cout << "-E- CbmLitAllHitsTofMerger::DoMerge: Z position "
                  << hit->GetZ() << " not found in map. Something is wrong.\n";
      }
      CbmLitTrackParam tpar(zParamMap[hit->GetZ()]);
      litfloat chi = 0.;
      fFilter->Update(&tpar, hit, chi);
      if (
        chi
        < fChiSqCut) {  // Check if hit is inside validation gate and closer to the track.
                        //Create new TOF track
        CbmLitTofTrack* tofTrack = new CbmLitTofTrack();
        tofTrack->SetTrack(track);
        tofTrack->SetHit(hit);
        tofTrack->SetTrackParam(&tpar);
        tofTrack->SetDistance(chi);
        tofTracks.push_back(tofTrack);
      }
    }
  }
  return kLITSUCCESS;
}