BaseSelector.cxx

Go to the documentation of this file.

#include "BaseSelector.h"
#include <fstream>
#include <iostream>
 
BaseSelector::BaseSelector() { 
    BaseSelector("default");
    m_cfgFile = "";
    debug_ = false;
    passSelection = true;
}
 
BaseSelector::BaseSelector(const std::string& inputName) { 
    m_name = inputName;
    m_cfgFile = "";
    debug_ = false;
    passSelection = true;
}
 
BaseSelector::BaseSelector(const std::string& inputName, const std::string& cfgFile) { 
    m_name = inputName;
    m_cfgFile = cfgFile;
    debug_ = false;
    passSelection=true;
}
 
void BaseSelector::setCfgFile(const std::string& cfgFile) {
    m_cfgFile = cfgFile;
}
 
void BaseSelector::setDebug(bool val) { 
    debug_ = val;
}
 
BaseSelector::~BaseSelector(){}
 
bool BaseSelector::LoadSelection(){
    if (m_cfgFile.empty()) {
        std::cout<<"ERROR BaseSelector::Configuration File not specified";
        return false;
    }
    
    std::ifstream i_file(m_cfgFile);
    i_file >> _h_selections;
    if (debug_) {
        for (auto& el : _h_selections.items())
            std::cout<<el.key() << " : " << el.value() << "\n";
    }
    
    for (auto cut : _h_selections.items()) {
        cuts[cut.key()].first  = cut.value().at("cut");
        cuts[cut.key()].second = cut.value().at("id");
        labels[cut.key()] = cut.value().at("info");
        ncuts_++;
    }
 
    if (debug_) {
        for (cut_it it = cuts.begin(); it != cuts.end(); ++it) {
            std::cout<<it->first<<" [value:]=" <<it->second.first<<" [id:] ="<<it->second.second<<std::endl;
        }
    }
 
    makeCutFlowHisto();
    return true;
}
    
void BaseSelector::makeCutFlowHisto() {
 
    h_cf_ = std::make_shared<TH1F>((m_name+"_cutflow").c_str(),(m_name+"_cutflow").c_str(),ncuts_+1,0,ncuts_+1);
    h_cf_->GetXaxis()->SetCanExtend(true);
    h_cf_->Sumw2();
    h_cf_->GetXaxis()->SetBinLabel(1,"no-cuts");
    
    for (cut_it it = cuts.begin(); it != cuts.end(); ++it) {
        h_cf_->GetXaxis()->SetBinLabel((it->second.second+2),labels[(it->first)].c_str());
    }
}
 
 
//TODO Clean up logic
bool BaseSelector::passCutEq(const std::string& cutname, double val, double w) {
    
    if (hasCut(cutname)) {
        if (val != cuts[cutname].first) { 
            passSelection = passSelection && false;
            return false;
        }
        else {
            h_cf_->Fill((double)(cuts[cutname].second + 1), w);
            passSelection = passSelection && true;
        }
    }
    return true;
}
 
 
bool BaseSelector::passCutLt(const std::string& cutname, double val, double w) {
    
    if (hasCut(cutname)) {
        if (val > cuts[cutname].first) {
            passSelection = passSelection && false;
            return false;
        }
        else {
            h_cf_->Fill((double)(cuts[cutname].second + 1), w);
            passSelection = passSelection && true;
        }
    }
    return true;
}
 
bool BaseSelector::passCutGt(const std::string& cutname, double val, double w) {
    
    if (hasCut(cutname)) {
        if (val < cuts[cutname].first) {
            passSelection = passSelection && false;
            return false;
        }
        else {
            h_cf_->Fill((double)(cuts[cutname].second + 1), w);
            passSelection = passSelection && true;
        }
    }
    return true;
}