CutFlowSvc.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// Implementation file for class CutFlowSvc
// Authors:
//   Tadej Novak <tadej@cern.ch>
//   Joao Firmino da Costa <joao.costa@cern.ch> and David Cote <david.cote@cern.ch>
 
// Not dual use at the moment
#ifndef XAOD_STANDALONE
 
#include <EventBookkeeperTools/CutFlowSvc.h>
 
#include <GaudiKernel/Incident.h>
#include <GaudiKernel/FileIncident.h>
#include <GaudiKernel/IIncidentSvc.h>
 
// EDM includes
#include <EventInfo/EventStreamInfo.h>
#include <xAODMetaData/FileMetaData.h>
 
CutFlowSvc::CutFlowSvc(const std::string& name,
                       ISvcLocator* pSvcLocator ) :
  base_class(name, pSvcLocator)
{
  assert( pSvcLocator );
}
 
 
StatusCode
CutFlowSvc::initialize()
{
  ATH_MSG_DEBUG( "Initializing " << name() );
 
  // Only run if explicitly configured
  if (m_configured.value() == false) {
    ATH_MSG_ERROR("CutFlowSvc should be explicitly configured!");
    return StatusCode::FAILURE;
  }
 
  //Get input MetaData StoreGate
  ATH_CHECK( m_inMetaDataStore.retrieve() );
 
  //Get IncidentSvc
  ServiceHandle<IIncidentSvc> incSvc("IncidentSvc", this->name());
  ATH_CHECK( incSvc.retrieve() );
  incSvc->addListener(this, IncidentType::BeginInputFile, 60); // pri has to be < 100 to be after MetaDataSvc.
  incSvc->addListener(this, "MetaDataStop", 30);
 
  // Create initial bookkeeper container for bookkeepers in _this_ processing
  ATH_CHECK(createContainers(1));
 
  // Determine the skimming cycle number that we should use now from the input file
  ATH_MSG_VERBOSE("Have currently the cycle number = " << m_skimmingCycle );
  ATH_CHECK( this->determineCycleNumberFromInput(m_completeCollName) );
  ATH_MSG_VERBOSE("Will use cycle number = " << m_skimmingCycle );
 
  return StatusCode::SUCCESS;
}
 
 
 
CutIdentifier CutFlowSvc::registerFilter( const std::string& name,
                                          const std::string& description,
                                          bool nominalOnly )
{
  ATH_MSG_DEBUG("Registering filter with name '" << name << "' and description '" << description << "'");
 
  auto newCbk = std::make_unique<xAOD::CutBookkeeper>();
  newCbk->setName(name);
  newCbk->setDescription(description);
  newCbk->setInputStream(m_inputStream);
  newCbk->setCycle(m_skimmingCycle);
  CutIdentifier cutID = newCbk->uniqueIdentifier();
 
  // Let's see if an CutBookkeeper of this name already exists
  xAOD::CutBookkeeperContainer *container = m_containers.at(0);
  ATH_MSG_VERBOSE("Searching if this CutBookkeeper already exists");
  for (xAOD::CutBookkeeper* cbk : *container) {
    if (newCbk->isEqualTo(cbk)) {
      ATH_MSG_DEBUG("The CutBookkeeper with name '" << name << "' already exists"
                    << " and has CutID " << cbk->uniqueIdentifier() << "... Not adding!" );
      // Return the existing cut ID
      return cbk->uniqueIdentifier();
    }
  }
 
  // If it is a new CutBookkeeper, add it to the container
  ATH_MSG_DEBUG("Declaring a new filter with name '" << name << "' and CutID " << cutID );
  container->push_back(std::move(newCbk));
 
  if (nominalOnly) {
    m_nominalOnlyCuts.emplace(cutID);
  }
 
  return cutID;
}
 
 
 
// This method is probably only called by the DecisionSvc
CutIdentifier CutFlowSvc::registerTopFilter ATLAS_NOT_THREAD_SAFE ( const std::string& name,
                                             const std::string& description,
                                             unsigned int logic,
                                             const std::string& outputStream,
                                             bool nominalOnly ) 
{
  ATH_MSG_DEBUG("Registering top filter with name '" << name << "' and description '" << description << "'"
                << ", logic=" << logic << ", outputStream=" << outputStream << ")");
 
  // Call the registerFilter method and get the correct CutBookkeeper
  // from the returned cutID
  CutIdentifier cutID = registerFilter(name, description, nominalOnly);
  xAOD::CutBookkeeper* cbk = getCutBookkeeper(cutID, 0);
  if (cbk == nullptr) {
    ATH_MSG_ERROR("Could not find CutBookkeeper with CutID " << cutID);
    throw std::runtime_error("Could not find CutBookkeeper");
  }
 
  // Then set the logic and outputStream of the relevant CutBookkeeper
  cbk->setCutLogic(xAOD::CutBookkeeper::CutLogic(logic));
  cbk->setTopFilter(true);
  cbk->addOutputStreamForAllUsed(outputStream);
 
  return cutID;
}
 
 
CutIdentifier CutFlowSvc::registerCut( const std::string& name,
                                       const std::string& description,
                                       CutIdentifier parentCutID,
                                       bool nominalOnly )
{
  ATH_MSG_DEBUG("Registering cut with name '" << name << "', description '" << description
                << "' and original CutID " << parentCutID);
 
  // Get the CutBookkeeper of the origin Filter Algorithm/Tool
  xAOD::CutBookkeeper* parentCbk = getCutBookkeeper(parentCutID, 0);
  if (parentCbk == nullptr) {
    ATH_MSG_ERROR("Could not find parent CutBookkeeper with CutID " << parentCutID);
    throw std::runtime_error("Could not find CutBookkeeper");
  }
 
  // Call the registerFilter method and get the correct CutBookkeeper
  // from the returned cutID
  CutIdentifier cutID = registerFilter(name, description, nominalOnly);
  xAOD::CutBookkeeper* cbk = getCutBookkeeper(cutID, 0);
  if (cbk == nullptr) {
    ATH_MSG_ERROR("Could not find CutBookkeeper with CutID " << cutID);
    throw std::runtime_error("Could not find CutBookkeeper");
  }
 
  // Add child to parent
  parentCbk->addChild(cbk);
 
  return cutID;
}
 
 
void
CutFlowSvc::setFilterDescription( CutIdentifier cutID,
                                  const std::string& descr )
{
  ATH_MSG_DEBUG("calling setFilterDescription(" << cutID << ", " << descr << ")" );
 
  for (size_t i = 0; i < m_containers.size(); ++i) {
    xAOD::CutBookkeeper* cbk = getCutBookkeeper(cutID, i);
    if (cbk != nullptr) {
      cbk->setDescription(descr);
    }
  }
  return;
}
 
 
void
CutFlowSvc::addEvent( CutIdentifier cutID,
                      const std::vector<float>& weights )
{
  if (weights.size() != m_containers.size()) {
    ATH_MSG_ERROR("Inconsistent weights and variation sizes " << weights.size() << " and " << m_containers.size());
    throw std::runtime_error("Inconsistent weights and variation sizes");
  }
 
  for (size_t i = 0; i < m_containers.size(); ++i) {
    addEvent(cutID, i, weights[i]);
  }
}
 
 
void
CutFlowSvc::addEvent( CutIdentifier cutID,
                      double weight )
{
  if (m_nominalOnlyCuts.count(cutID) == 1) {
    addEvent(cutID, 0, weight);
    return;
  }
 
  for (size_t i = 0; i < m_containers.size(); ++i) {
    addEvent(cutID, i, weight);
  }
}
 
 
 
void
CutFlowSvc::addEvent( CutIdentifier cutID,
                      size_t index,
                      double weight )
{
  std::lock_guard<std::recursive_mutex> lock(m_addEventMutex);
 
  ATH_MSG_VERBOSE("Adding event with weight " << weight << " to cut " << cutID << " for variation " << index);
 
  xAOD::CutBookkeeper* cbk = getCutBookkeeper(cutID, index);
  if (cbk == nullptr) {
    ATH_MSG_ERROR("Could not find CutBookkeeper for CutID " << cutID << " and variation " << index);
    throw std::runtime_error("Could not find CutBookkeeper");
  }
 
  cbk->addNAcceptedEvents(1);
  cbk->addSumOfEventWeights(weight);
  cbk->addSumOfEventWeightsSquared(weight*weight);
}
 
 
 
//__________________________________________________________________________
void CutFlowSvc::handle( const Incident& inc )
{
  ATH_MSG_VERBOSE( "Start incident " << inc.type() );
 
  if (inc.type() == IncidentType::BeginInputFile) {
    // Look up input stream name from FileMetaData
    std::string inputstream = "";
    if (m_inMetaDataStore->contains<xAOD::FileMetaData>("FileMetaData")) {
      const xAOD::FileMetaData* fmd = nullptr;
      if (m_inMetaDataStore->retrieve(fmd).isFailure()) {
        ATH_MSG_ERROR("Failed to retrieve input FileMetaData");
      } else if (fmd &&
                 !fmd->value(xAOD::FileMetaData::dataType, inputstream)) {
        ATH_MSG_WARNING("Failed to get input stream name from FileMetaData");
      }
    } else {
      ATH_MSG_DEBUG("No FileMetaData in input, trying EventStreamInfo");
      // determine key for EventStreamInfo
      std::vector<std::string> vKeys;
      m_inMetaDataStore->keys<EventStreamInfo>(vKeys);
      // eliminate duplicate keys
      std::set<std::string> keys(vKeys.begin(), vKeys.end());
      if (keys.size() == 1) {
        std::string key = *keys.begin();  // get the one key
        auto esi = m_inMetaDataStore->tryConstRetrieve<EventStreamInfo>(key);
        if (esi && !esi->getProcessingTags().empty()) {
          // use the first tag
          inputstream = *esi->getProcessingTags().begin();
        }
      }
    }
 
    if (inputstream.empty()) {
      if (m_inputStream.empty()) {
        ATH_MSG_FATAL("Cannot determine input stream name");
        return;
      }
      ATH_MSG_DEBUG(
          "Failed to parse stream name from metadata, using "
          "property \"InputStream\"");
      inputstream = m_inputStream;
    }
 
    // Check that input stream is consistent with previous file
    ATH_MSG_DEBUG("Input stream name: \"" << inputstream << '"');
    if (m_inputStream.empty()) {
      m_inputStream = inputstream;
    } else if (m_inputStream != inputstream) {
      const FileIncident* finc = dynamic_cast<const FileIncident*>(&inc);
      if (m_inputStream != "N/A" && m_inputStream != "unknownStream") {
        ATH_MSG_FATAL("File " << finc->fileName() << " stream " << inputstream
                              << " does not match previous file "
                              << m_inputStream);
        return;
      }
    }
  }
 
  // Clear the local bookkeepers
  if (inc.type() == "MetaDataStop") {
    if (!m_containers.empty()) {
      // Reset existing container
      for (size_t i = 0; i < m_containers.size(); ++i) {
        for (xAOD::CutBookkeeper* cbk : *m_containers.at(i)) {
          cbk->setNAcceptedEvents(0);
          cbk->setSumOfEventWeights(0);
          cbk->setSumOfEventWeightsSquared(0);
        }
      }
    }
  }
 
  ATH_MSG_VERBOSE( "End incident " << inc.type() );
  return;
}
 
 
 
StatusCode CutFlowSvc::determineCycleNumberFromInput( const std::string& collName )
{
  ATH_MSG_DEBUG("calling determineCycleNumberFromInput('" << collName
                  << "')... have currently cycle number = " << m_skimmingCycle );
 
  // Try to get CutBookkeepers from the input file
  if ( m_inMetaDataStore->contains<xAOD::CutBookkeeperContainer>(collName) ) {
    ATH_MSG_VERBOSE("Found xAOD::CutBookkeeperContainer in input MetaStore with name: " << collName);
 
    const xAOD::CutBookkeeperContainer* inputContainer{};
    ATH_CHECK( m_inMetaDataStore->retrieve( inputContainer, collName ) );
    // Now, iterate over all CutBookkeepers and search for the highest cycle number
    int maxCycle{};
    for (const xAOD::CutBookkeeper* cbk : *inputContainer) {
      int inCycle = cbk->cycle();
      if (inCycle > maxCycle) maxCycle = inCycle;
    }
    m_skimmingCycle = std::max(m_skimmingCycle.value(), maxCycle + 1);
  }
 
  ATH_MSG_DEBUG("done calling determineCycleNumberFromInput('" << collName
                  << "')... have now cycle number = " << m_skimmingCycle );
  return StatusCode::SUCCESS;
}
 
 
 
StatusCode CutFlowSvc::createContainers(size_t count)
{
  // Expect that this should only be called once for now
  if (!m_containers.empty() && m_containers.size() == count) {
    return StatusCode::SUCCESS;
  }
 
  if (m_containers.size() > count) {
    ATH_MSG_ERROR("Inconsistent variation count");
    return StatusCode::FAILURE;
  }
 
  ATH_MSG_VERBOSE("Create containers with count " << count);
 
  ATH_MSG_VERBOSE("Existing size " << m_containers.size());
 
  // Create the containers
  CutBookkeepersLocalCache::prepareContainers(m_containers, count, true);
  if (count == 1) {
    return StatusCode::SUCCESS;
  }
 
  // Copy existing CutBookkeepers for consistency
  xAOD::CutBookkeeperContainer *first = m_containers.at(0);
  if (first->empty()) {
    return StatusCode::SUCCESS;
  }
  for (size_t i = 1; i < count; ++i) {
    xAOD::CutBookkeeperContainer *cont = m_containers.at(i);
    for (const xAOD::CutBookkeeper *cbk : *first) {
      xAOD::CutBookkeeper *newCbk = new xAOD::CutBookkeeper();
      cont->push_back(newCbk);
      *newCbk = *cbk;
    }
  }
 
  return StatusCode::SUCCESS;
}
 
 
 
StatusCode CutFlowSvc::setNumberOfWeightVariations(size_t count)
{
  ATH_CHECK(createContainers(count));
  return StatusCode::SUCCESS;
}
 
 
 
const CutBookkeepersLocalCache& CutFlowSvc::getCutBookkeepers() const
{
  return m_containers;
}
 
 
 
xAOD::CutBookkeeper* CutFlowSvc::getCutBookkeeper(const CutIdentifier cutID,
                                                  size_t index) const
{
  if (index >= m_containers.size()) {
    return nullptr;
  }
 
  for (xAOD::CutBookkeeper* cbk : *m_containers.at(index)) {
    if (cbk->uniqueIdentifier() == cutID) {
      return cbk;
    }
  }
  return nullptr;
}
 
 
 
uint64_t CutFlowSvc::getNAcceptedEvents(const CutIdentifier cutID) const
{
  const xAOD::CutBookkeeper* cbk = getCutBookkeeper(cutID, 0);
  if (cbk == nullptr) {
    return std::numeric_limits<uint64_t>::max();
  }
  return cbk->nAcceptedEvents();
}
 
 
#endif // dual use