Validator.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include <iostream>
#include <math.h>
#include <float.h>
#include "JetSubStructureMomentTools/Validator.h"
#include <TFile.h>
#include <TH1.h>
 
using namespace std;
 
Validator::Validator(const std::string& name) : 
  AsgTool(name),
  m_histSvc("THistSvc", name)
{
  declareProperty("InputContainer", m_InputContainer = "");
  declareProperty("FloatMoments", m_FloatMoments);
}
 
 
StatusCode Validator::initialize()
{
  ATH_CHECK( m_histSvc.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
int Validator::execute() const
{
    // Get leading jet
    const xAOD::JetContainer* jets = nullptr;
  if(!evtStore()->contains<xAOD::JetContainer>(m_InputContainer)) {
    ATH_MSG_ERROR("Unable to retrieve jets from collection: " << m_InputContainer);
    return 1;
  }
    jets = evtStore()->retrieve<const xAOD::JetContainer>(m_InputContainer);
    if(jets->empty()) return 0;
    const xAOD::Jet *jet = jets->at(0); // This assumes the container is sorted
 
    // Loop over float moments
    for(unsigned int i=0; i<m_FloatMoments.size(); i++) {
        TH1 *outputHist;
        if(m_histSvc->exists("/JetSubstructureMoments/" + m_FloatMoments[i])) {
            m_histSvc->getHist("/JetSubstructureMoments/" + m_FloatMoments[i], outputHist).ignore();
        }
        else {
      unsigned int nbins = 100;
      float xlow = 0, xhigh = 1.0;
      if(m_FloatMoments[i].find("Split") != string::npos ||
         m_FloatMoments[i] == "pt") {
        nbins = 1000;
        xhigh = 2000000;
      }
      else if(m_FloatMoments[i].find("Pull") != string::npos) {
        if(m_FloatMoments[i].find("PullMag") != string::npos) {
          xhigh = 0.1;
          nbins = 100;
        }
        else {
          xlow = -10;
          xhigh = 10;
          nbins = 200;
        }
      }
            else if(m_FloatMoments[i].find("ShowerDeconstruction") != string::npos) {
                xlow = -10;
                xhigh = 10;
                nbins = 100;
            }
 
            outputHist = new TH1F(m_FloatMoments[i].c_str(), "", nbins, xlow, xhigh);
            StatusCode sc = m_histSvc->regHist("/JetSubstructureMoments/" + m_FloatMoments[i], outputHist);
            if(sc.isFailure()) {
                ATH_MSG_ERROR("Unable to register histogram");
                return 1;
            }
        }
 
    if(m_FloatMoments[i] == "pt") {
      outputHist->Fill(jet->pt());
    }
    else {
        outputHist->Fill(jet->getAttribute<float>(m_FloatMoments[i].c_str()));
    }
    }
 
    return 0;
}