AnnealingUtility.cpp

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

// 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/.

#include "Acts/Utilities/AnnealingUtility.hpp"

#include "Acts/Utilities/AlgebraHelpers.hpp"

namespace {
double computeAnnealingWeight(double chi2, double invTemp) {
  return Acts::safeExp(-chi2 * invTemp);
}
}  // namespace

Acts::AnnealingUtility::Config::Config() = default;

void Acts::AnnealingUtility::anneal(State& state) const {
  if (state.currentTemperatureIndex < m_cfg.setOfTemperatures.size() - 1) {
    ++state.currentTemperatureIndex;
  } else {
    state.equilibriumReached = true;
  }
}

double Acts::AnnealingUtility::getWeight(
    State& state, double chi2, const std::vector<double>& allChi2) const {
  unsigned int idx = state.currentTemperatureIndex;
  // Calculate 1/denominator in exp function already here
  const double currentInvTemp = 1. / (2. * m_cfg.setOfTemperatures[idx]);

  double num = computeAnnealingWeight(chi2, currentInvTemp);

  double denom = m_gaussCutTempVec[idx];

  for (double val : allChi2) {
    denom += computeAnnealingWeight(val, currentInvTemp);
  }

  return num / denom;
}

double Acts::AnnealingUtility::getWeight(State& state, double chi2) const {
  // Calculate 1/denominator in exp function
  const double currentInvTemp =
      1. / (2 * m_cfg.setOfTemperatures[state.currentTemperatureIndex]);

  return 1. /
         (1. + computeAnnealingWeight(m_cfg.cutOff - chi2, currentInvTemp));
}