MultiComponentStateAssembler.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef MultiComponentStateAssembler_H
#define MultiComponentStateAssembler_H
 
#include "TrkGaussianSumFilterUtils/GsfConstants.h"
#include "TrkParameters/ComponentParameters.h"
#include <vector>
 
namespace Trk {

namespace MultiComponentStateAssembler {
struct Cache {
  Cache() {
    multiComponentState.reserve(GSFConstants::maxNumberofStateComponents);
  }
  Trk::MultiComponentState multiComponentState{};
  double validWeightSum = 0;
  double invalidWeightSum = 0;
  static constexpr double minimumValidFraction = 0.01;
  static constexpr double minimumFractionalWeight = 1e-09;
};

void
addComponent(
  MultiComponentStateAssembler::Cache& cache,
  ComponentParameters&& multiComponentState);

void
addMultiState(
  MultiComponentStateAssembler::Cache& cache,
  Trk::MultiComponentState&& multiComponentState);

MultiComponentState
assembledState(MultiComponentStateAssembler::Cache&& cache);
 
} // End MultiComponentStateAssembler namespace
 
} // End Trk namepace
 
// inline methods
inline void Trk::MultiComponentStateAssembler::addComponent(
    Cache& cache, ComponentParameters&& componentParameters) {
 
  cache.validWeightSum += componentParameters.weight;
  cache.multiComponentState.push_back({std::move(componentParameters.params),
                                         componentParameters.weight});
}
 
inline void Trk::MultiComponentStateAssembler::addMultiState(
    Cache& cache, Trk::MultiComponentState&& multiComponentState) {
 
  double sumW(0.);
  for (auto& component : multiComponentState) {
    sumW += component.weight;
    cache.multiComponentState.push_back({std::move(component.params),
                                           component.weight});
  }
  cache.validWeightSum += sumW;
}
 
#endif