TFCSParametrizationFloatSelectChain.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
#include "ISF_FastCaloSimEvent/TFCSParametrizationFloatSelectChain.h"
#include <algorithm>
#include <iterator>
#include <iostream>
 
//=============================================
//======= TFCSParametrizationFloatSelectChain =========
//=============================================
 
int TFCSParametrizationFloatSelectChain::push_back_in_bin(
    TFCSParametrizationBase *param, float low, float up) {
  if (up < low) {
    // can't handle wrong order of bounds
    ATH_MSG_ERROR("Cannot add " << param->GetName() << ": range [" << low << ","
                                << up << ") not well defined");
    return -1;
  }
  if (get_number_of_bins() == 0) {
    // special case of adding the first bin
    push_back_in_bin(param, 0);
    m_bin_low_edge[0] = low;
    m_bin_low_edge[1] = up;
    return 0;
  }
  int ilow = val_to_bin(low);
  int iup = val_to_bin(up);
  if (ilow < 0 && iup < 0 && m_bin_low_edge[get_number_of_bins()] == low) {
    // special case of adding bin at the end of existing bins
    int endbin = get_number_of_bins();
    push_back_in_bin(param, endbin);
    m_bin_low_edge[endbin] = low;
    m_bin_low_edge[endbin + 1] = up;
    return endbin;
  }
 
  if (ilow < 0 && iup < 0) {
    // can't handle disjunct ranges
    ATH_MSG_ERROR("Cannot add " << param->GetName() << ": range [" << low << ","
                                << up << ") which is outside existing range ["
                                << m_bin_low_edge[0] << ","
                                << m_bin_low_edge[get_number_of_bins()] << ")");
    return -1;
  }
 
  if (iup >= 0 && ilow >= 0 && iup - ilow == 1 && m_bin_low_edge[ilow] == low &&
      m_bin_low_edge[iup] == up) {
    // Case of adding to an existing bin
    push_back_in_bin(param, ilow);
    return ilow;
  }
 
  if (ilow < 0 && iup == 0 && m_bin_low_edge[iup] == up) {
    // Case of adding a new first bin before existing bins
    int newbin = iup;
    int oldsize = m_bin_start.size();
    m_bin_start.resize(oldsize + 1, m_bin_start.back());
    m_bin_start.shrink_to_fit();
    m_bin_low_edge.resize(oldsize + 1, m_bin_low_edge.back());
    m_bin_low_edge.shrink_to_fit();
    for (int i = oldsize; i > newbin; --i) {
      m_bin_start[i] = m_bin_start[i - 1];
      m_bin_low_edge[i] = m_bin_low_edge[i - 1];
    }
    m_bin_low_edge[newbin] = low;
    m_bin_low_edge[newbin + 1] = up;
    push_back_in_bin(param, newbin);
    return newbin;
  }
 
  ATH_MSG_ERROR("Cannot add "
                << param->GetName() << ": range [" << low << "," << up
                << ") covers more than one bin in existing range ["
                << m_bin_low_edge[0] << ","
                << m_bin_low_edge[get_number_of_bins()]
                << ") or splits an existing bin");
  return -1;
}
 
void TFCSParametrizationFloatSelectChain::push_back_in_bin(
    TFCSParametrizationBase *param, unsigned int bin) {
  TFCSParametrizationBinnedChain::push_back_in_bin(param, bin);
  if (m_bin_low_edge.size() < m_bin_start.size()) {
    m_bin_low_edge.resize(m_bin_start.size(), m_bin_low_edge.back());
    m_bin_low_edge.shrink_to_fit();
  }
}
 
int TFCSParametrizationFloatSelectChain::val_to_bin(float val) const {
  if (val < m_bin_low_edge[0]) {
    ATH_MSG_VERBOSE("val_to_bin(" << val << ")=-1: " << val << " < "
                                  << m_bin_low_edge[0]);
    return -1;
  }
  if (val >= m_bin_low_edge[get_number_of_bins()]) {
    ATH_MSG_VERBOSE("val_to_bin(" << val << ")=-1: " << val << " >= "
                                  << m_bin_low_edge[get_number_of_bins()]);
    return -1;
  }
 
  auto it = std::upper_bound(m_bin_low_edge.begin(), m_bin_low_edge.end(), val);
  int dist = std::distance(m_bin_low_edge.begin(), it) - 1;
  ATH_MSG_VERBOSE("val_to_bin(" << val << ")=" << dist);
  return dist;
}
 
void TFCSParametrizationFloatSelectChain::unit_test(
    TFCSSimulationState *, TFCSTruthState *, const TFCSExtrapolationState *) {
  ISF_FCS::MLogging logger;
  ATH_MSG_NOCLASS(
      logger,
      "==================================================================");
  ATH_MSG_NOCLASS(
      logger,
      "= Please call TFCSParametrizationEkinSelectChain::unit_test(...) =");
  ATH_MSG_NOCLASS(
      logger,
      "= or          TFCSParametrizationEtaSelectChain ::unit_test(...) =");
  ATH_MSG_NOCLASS(
      logger,
      "=================================================================="
          << std::endl);
}