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/AlignedDynArray.h"
#include "TrkGaussianSumFilterUtils/GsfConstants.h"
//
#include <vector>
#include <cstdint>
 
namespace GSFUtils {

struct Component1D
{
  double mean = 0.;
  double cov = 0.;
  double invCov = 1e10;
  double weight = 0.;
};
using Component1DArray = AlignedDynArray<Component1D, GSFConstants::alignment>;
struct Merge {
  int To = 0;
  int From = 0;
};
using MergeArray = std::vector<Merge>;

MergeArray
findMerges(Component1DArray&& componentsIn, const int reducedSize);
 
} // namespace KLGaussianMixtureReduction
 
#endif