ConstantContentBinMaker.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonCalibMath/ConstantContentBinMaker.h"
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
 
#include <iostream>
 
using namespace MuonCalib;
 
//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
//:: IMPLEMENTATION OF METHODS DEFINED IN THE CLASS ConstantContentBinMaker ::
//::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
 
//*****************************************************************************
 
//:::::::::::::::::
//:: CONSTRUCTOR ::
//:::::::::::::::::
 
ConstantContentBinMaker::ConstantContentBinMaker(
                            const std::vector<DataPoint> & points,
                            const double  epsilon)
  : m_points (points),
    m_epsilon (std::abs(epsilon))
{
}
 
 
ConstantContentBinMaker::~ConstantContentBinMaker()
{
  for (DataBin* b : m_bins) {
    delete b;
  }
}
 
 
//*****************************************************************************
 
//::::::::::::::::::::::::::
//:: METHOD binDataPoints ::
//::::::::::::::::::::::::::
 
bool ConstantContentBinMaker::binDataPoints(const unsigned int & bin_content,
                                       std::vector<unsigned int> & ref_coord) {
 
// VARIABLES //
 
    unsigned int n_bins(static_cast<unsigned int>(
                        std::log(m_points.size()/static_cast<double>(bin_content))
                                                                    /std::log(2.0)));
    std::vector<unsigned int> splitting_axis(ref_coord);
    unsigned int axis_counter(0);
 
// RESET //
    for (DataBin* b : m_bins) {
      delete b;
    }
    m_bins.clear();
 
// CHECK WHETHER THE IS SUFFICIENT DATA POINTS //
 
    if (m_points.empty() || n_bins<2) {
        MsgStream log(Athena::getMessageSvc(), "ConstantContentBinMaker");
        log<< MSG::WARNING << "Class ConstantContentBinMaker, method binDataPoints: Too few data points for binning!"<<endmsg;
        return false;
    }
 
// CHECK REFERENCE COORDINATE //
 
    if (ref_coord.empty()) {
        splitting_axis = std::vector<unsigned int>(
                                        m_points[0].dataVector().rows());
        for (unsigned int k=0; k<splitting_axis.size(); k++) {
            splitting_axis[k] = k;
        }
    }
 
// FILL A MOTHER BIN AND BIN THE DATA UNTIL THE REQUIRED BIN CONTENT IS //
// REACHED.                                                             //
 
// mother bin //
    m_bins.push_back(new DataBin(m_points, m_epsilon));
 
    for (unsigned int split=0; split<n_bins; split++) {
        unsigned int nb_bins_so_far(m_bins.size());
        for (unsigned int k=0; k<nb_bins_so_far; k++) {
                m_bins.push_back(m_bins[k]->splitBin(
                                            splitting_axis[axis_counter]));
                axis_counter++;
                if (axis_counter==splitting_axis.size()) {
                    axis_counter = 0;
                }
        }
    }
 
    return true;
 
}
 
//*****************************************************************************
 
//::::::::::::::::::::
//:: METHOD getBins ::
//::::::::::::::::::::
 
const std::vector<DataBin *> & ConstantContentBinMaker::getBins() const {
 
    return m_bins;
 
}