KLGaussianMixtureReduction.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef KLGaussianMixReductionUtils_H
#define KLGaussianMixReductionUtils_H
 
#include "TrkGaussianSumFilterUtils/GsfConstants.h"
//
#include <array>
#include <cstdint>
 
namespace GSFUtils {
 
struct Component1D
{
  double mean = 0.;
  double cov = 0.;
  double invCov = 1e10;
  double weight = 0.;
};
struct Component1DArray
{
  alignas(GSFConstants::alignment)
    std::array<Component1D,
               GSFConstants::maxComponentsAfterConvolution> components{};
  int numComponents = 0;
};
struct MergeArray
{
  struct merge
  {
    int8_t To = 0;
    int8_t From = 0;
  };
  std::array<merge, GSFConstants::maxComponentsAfterConvolution> merges{};
  int numMerges = 0;
};
 
MergeArray
findMerges(const Component1DArray& componentsIn, const int8_t reducedSize);
 
} // namespace KLGaussianMixtureReduction
 
#endif