CutBinning.C

Go to the documentation of this file.

#include "jbltools/sfh/CutBinning.h"
#include "jbltools/sfh/FillIterator.h"
#include "jbltools/sfh/BaseCutBase.h"

#include <iostream>
#include <cstring>

static const char *ident="@(#)$Id: CutBinning.C,v 1.4 2005/07/20 10:51:32 blist Exp $";

 
CutBinning::CutBinning (int nbins, const BaseCut *cuts[], const char *name) 
: BinningFun (nbins, -0.5, nbins-0.5, (name ? std::string(name) : "")),
  theCuts(nbins), binNames(nbins), binTitles(nbins),
  theFillIterator (0)
{
  for (int i = 0; i < nbins; ++i) {
    const BaseCut *cut = cuts[i];
    theCuts[i] = cut;
    binNames[i] = str(i, 3);
    binTitles[i] = cut ? std::string(cut->getName()) : "";
  }
  genDefinition();
}

CutBinning::CutBinning (int nbins, BaseCut *cuts[], const char *name) 
: BinningFun (nbins, -0.5, nbins-0.5, (name ? std::string(name) : "")),
  theCuts(nbins), binNames(nbins), binTitles(nbins),
  theFillIterator (0)
{
  for (int i = 0; i < nbins; ++i) {
    const BaseCut *cut = cuts[i];
    theCuts[i] = cut;
    binNames[i] = str(i, 3);
    binTitles[i] = cut ? std::string(cut->getName()) : "";
  }
  genDefinition();
}
 
CutBinning::CutBinning (int nbins, const BaseCut *cuts[], const char *titles[], const char *name) 
: BinningFun (nbins, -0.5, nbins-0.5, (name ? std::string(name) : "")),
  theCuts(nbins), binNames(nbins), binTitles(nbins),
  theFillIterator (0)
{
  for (int i = 0; i < nbins; ++i) {
    const BaseCut *cut = cuts[i];
    const char *title = titles[i];
    theCuts[i] = cut;
    binNames[i] = str(i, 3);
    binTitles[i] = title ? std::string (title) : "";
  }
  genDefinition();
}
CutBinning::CutBinning (int nbins, BaseCut *cuts[], const char *titles[], const char *name) 
: BinningFun (nbins, -0.5, nbins-0.5, (name ? std::string(name) : "")),
  theCuts(nbins), binNames(nbins), binTitles(nbins),
  theFillIterator (0)
{
  for (int i = 0; i < nbins; ++i) {
    const BaseCut *cut = cuts[i];
    const char *title = titles[i];
    theCuts[i] = cut;
    binNames[i] = str(i, 3);
    binTitles[i] = title ? std::string (title) : "";
  }
  genDefinition();
}

bool CutBinning::operator() (int i) const {
      const BaseCut *cut = (i >= 0 && static_cast<unsigned int>(i) < theCuts.size()) ? theCuts[i] : 0;                 
      return cut ? (*cut)() : true; 
    }    
                    
const char *CutBinning::getBinName(int i) const {
  if (i < 0 || i >= getNBins()) return 0;
  size_t len = binNames[i].size()+1;
  char *result = new char[len];
  strncpy (result, binNames[i].c_str(), len);
  return result;
}

const char *CutBinning::getBinTitle(int i) const {
  if (i < 0 || i >= getNBins()) return 0;
  size_t len = binTitles[i].size()+1;
  char *result = new char[len];
  strncpy (result, binTitles[i].c_str(), len);
  return result;
}

void CutBinning::genDefinition() {
  std::string definition_("(");
  int i = 0;
  while (i < getNBins()) {
    definition_ += binTitles[i];
    if (++i >= getNBins()) break;
    definition_ += std::string(", ");
  }
  definition_ += ")";
  setDefinition (definition_);
}

void CutBinning::checkIterators() {
  theFillIterator = 0;
  const FillIterator *fi = 0;
  for (int i = 0; i < getNBins(); ++i) {
    const BaseCut *cut = theCuts[i];
    if (cut && (fi = cut->getIterator())) {
      if (!theFillIterator) {
        theFillIterator = fi;
      } 
      else if (theFillIterator != fi) {
        std::cerr << "CutBinning::checkIterators: cut " << i 
                  << " depends on iterator " << fi->getName()
                  << ", other cuts depend on " << theFillIterator->getName()
                  << "!\n  (If bothe have the same name, it is a different instance!)\n";
      }
    }
  }
}

bool CutBinning::hasBinLabels() const {
  return true;
}
    
const char *CutBinning::getBinLabel(int i) const {
  return getBinTitle(i);
}

---