ThinNegativeEnergyCaloClustersAlg.cxx

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// ThinNegativeEnergyCaloClustersAlg.cxx
// Author: James Catmore <James.Catmore@cern.ch>
// based on similar code by Karsten Koeneke <karsten.koeneke@cern.ch>
// Uses thinning service to remove CaloClusters with negative energy
// Intended for use in ESD->AOD in reconstruction - use other tools
// for analysis (Expression evaluation is not used here)
 
// EventUtils includes
#include "ThinNegativeEnergyCaloClustersAlg.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
 
// STL includes
#include <algorithm>
 
// FrameWork includes
#include "StoreGate/ThinningHandle.h"

// Public methods:
 
// Constructors
ThinNegativeEnergyCaloClustersAlg::ThinNegativeEnergyCaloClustersAlg(
  const std::string& name,
  ISvcLocator* pSvcLocator)
  : AthReentrantAlgorithm(name, pSvcLocator)
  , m_nEventsProcessed(0)
  , m_nCaloClustersProcessed(0)
  , m_nCaloClustersThinned(0)
{}
 
// Athena Algorithm's Hooks
StatusCode
ThinNegativeEnergyCaloClustersAlg::initialize()
{
  ATH_MSG_DEBUG("Initializing " << name() << "...");
 
  // Print out the used configuration
  ATH_MSG_DEBUG(" using = " << m_streamName);
 
  // Is truth thinning required?
  if (!m_doThinning) {
    ATH_MSG_INFO("Negative energy CaloClusters thinning not required");
  } else {
    ATH_MSG_INFO("Negative energy CaloClusters will be thinned");
  }
 
  if (m_doThinning && m_streamName.empty()) {
    ATH_MSG_ERROR("StreamName property has not been initialized.");
    return StatusCode::FAILURE;
  }
  ATH_CHECK(m_caloClustersKey.initialize(m_streamName, m_doThinning));
 
  // Initialize the counters to zero
  m_nEventsProcessed = 0;
  m_nCaloClustersProcessed = 0;
  m_nCaloClustersThinned = 0;
 
  ATH_MSG_DEBUG("==> done with initialize " << name() << "...");
 
  return StatusCode::SUCCESS;
}
 
StatusCode
ThinNegativeEnergyCaloClustersAlg::finalize()
{
  ATH_MSG_DEBUG("Finalizing " << name() << "...");
  ATH_MSG_INFO("Processed " << m_nEventsProcessed << " events containing "
                            << m_nCaloClustersProcessed << " CaloClusters");
  ATH_MSG_INFO("Removed " << m_nCaloClustersThinned
                          << " negative energy CaloClusters ");
  return StatusCode::SUCCESS;
}
 
StatusCode
ThinNegativeEnergyCaloClustersAlg::execute(const EventContext& ctx) const
{
  // Increase the event counter
  m_nEventsProcessed.fetch_add(1, std::memory_order_relaxed);
 
  // Is thinning required?
  if (!m_doThinning) {
    return StatusCode::SUCCESS;
  }
 
  // Retrieve truth and vertex containers
  SG::ThinningHandle<xAOD::CaloClusterContainer> caloClusters(m_caloClustersKey,
                                                              ctx);
 
  // Set up masks
  std::vector<bool> mask;
  int nCaloClusters = caloClusters->size();
  m_nCaloClustersProcessed.fetch_add(nCaloClusters, std::memory_order_relaxed);
  mask.assign(nCaloClusters, false);
 
  // Loop over CaloClusters and update mask
  unsigned long int nCaloClustersThinned = 0;
  for (int i = 0; i < nCaloClusters; ++i) {
    const xAOD::CaloCluster* caloCluster = (*caloClusters)[i];
    // Retain postive energy clusters
    if (caloCluster->rawE() > 0.0) {
      mask[i] = true;
    } else {
      ++nCaloClustersThinned;
    }
  }
  m_nCaloClustersThinned.fetch_add(nCaloClustersThinned,
                                   std::memory_order_relaxed);
 
  // Apply masks to thinning service
  caloClusters.keep(mask);
 
  return StatusCode::SUCCESS;
}