covariance_helper.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file covariance_helper.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Definitions/TrackParametrization.hpp"

#include <type_traits>

namespace Acts {
namespace detail {
template <typename CovMatrix_t, signed int NumIter = 1>
struct covariance_helper {
  static bool validate(CovMatrix_t& covariance) {
    if (covariance.hasNaN()) {
      return false;
    }
    size_t nIteration = 0;
    while (nIteration < NumIter) {
      if (isSemiPositive(covariance)) {
        return true;
      } else {
        Eigen::JacobiSVD<CovMatrix_t> svdCov(
            covariance, Eigen::ComputeFullU | Eigen::ComputeFullV);
        CovMatrix_t S = svdCov.singularValues().asDiagonal();
        CovMatrix_t V = svdCov.matrixV();
        CovMatrix_t H = V * S * V.transpose();
        covariance = (covariance + H) / 2;
        nIteration++;
      }
    }
    return isSemiPositive(covariance);
  }

  static bool isSemiPositive(const CovMatrix_t& covariance) {
    if (covariance.hasNaN()) {
      return false;
    }
    Eigen::LDLT<CovMatrix_t> ldltCov(covariance);
    return ldltCov.isPositive();
  }

  static bool isPositive(const CovMatrix_t& covariance) {
    if (covariance.hasNaN()) {
      return false;
    }
    Eigen::LLT<CovMatrix_t> lltCov(covariance);
    return lltCov.info() == Eigen::Success ? true : false;
  }
};

}  // namespace detail
}  // namespace Acts