KfTrajectory.cxx

Go to the documentation of this file.

/* Copyright (C) 2007-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergey Gorbunov [committer] */
 
#include "KfTrajectory.h"
 
#include "AlgoFairloggerCompat.h"
 
 
using namespace cbm::algo::kf;
 
 
template<typename T>
inline void Trajectory<T>::MakeConsistent()
{
  // sort the nodes in Z, add missing material layers and set fFirstMeasurementNodeId and fLastMeasurementNodeId
 
  if (!fKfSetup) {
    LOG(fatal) << "KfTrajectory::MakeConsistent: fKfSetup is not set!";
  }
 
  fFirstMeasurementNodeId = -1;
  fLastMeasurementNodeId  = -1;
  fNmeasurements          = 0;
  fIsFitted               = false;
  fIsFullyExtrapolated    = false;
  fQaChi2Downstream       = -1.;
 
  if (fNodes.empty()) {
    return;
  }
 
  std::sort(fNodes.begin(), fNodes.end());
 
  std::vector<char> isMaterialLayerPresent(fKfSetup->GetNofLayers(), false);
 
  // mark existing material layers
  for (const auto& n : fNodes) {
    if (n.materialLayer >= 0 && n.materialLayer < fKfSetup->GetNofLayers()) {
      isMaterialLayerPresent[n.materialLayer] = true;
    }
  }
 
  // add missing material layers between the first and the last nodes
 
  double zMin          = fNodes.front().z;
  double zMax          = fNodes.back().z;
  unsigned int sizeOld = fNodes.size();
 
  for (int ml = 0; ml < fKfSetup->GetNofLayers(); ++ml) {
    const auto& mat = fKfSetup->GetMaterial(ml);
    if (mat.GetZref() < zMin) {
      continue;
    }
    if (mat.GetZref() > zMax) {
      break;
    }
    if (isMaterialLayerPresent[ml]) {
      continue;
    }
    Node matNode;
    matNode.materialLayer = ml;
    matNode.z             = mat.GetZref();
    fNodes.push_back(matNode);
  }
  // sort again after adding material layers
  if (fNodes.size() > sizeOld) {
    std::sort(fNodes.begin(), fNodes.end());
  }
 
  // set first and last measurement nodes
  for (unsigned int i = 0; i < fNodes.size(); i++) {
    const auto& n = fNodes[i];
    if (n.isXySet) {
      fNmeasurements++;
      if (fFirstMeasurementNodeId < 0) {
        fFirstMeasurementNodeId = i;
      }
      fLastMeasurementNodeId = i;
    }
  }
}  // MakeConsistent
 
 
template class cbm::algo::kf::Trajectory<float>;
template class cbm::algo::kf::Trajectory<double>;