TriggerTowerThinningAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
//  ***************************************************************************
//  *   Author: John Morris (john.morris@cern.ch)                             *
//  *           Queen Mary University of London                               *
//  *                                                                         *
 
#include "TrigT1CaloCalibTools/TriggerTowerThinningAlg.h"
// TrigT1 common definitions
#include "TrigT1Interfaces/TrigT1CaloDefs.h"
#include "StoreGate/ThinningHandle.h"
#include "StoreGate/ReadDecorHandle.h"
#include "AthenaKernel/RNGWrapper.h"
#include "GaudiKernel/ThreadLocalContext.h"
#include "CLHEP/Random/RandomEngine.h"
#include "CLHEP/Random/RandFlat.h"
 
namespace DerivationFramework {
 
  TriggerTowerThinningAlg::TriggerTowerThinningAlg(const std::string& t, const std::string& n, const IInterface* p) :
    base_class(t,n,p),
    m_minCaloCellET(0.4),
    m_minADC(32),
    m_useRandom(false),
    m_minRandom(0.03),
    m_nEventsProcessed(0),
    m_nEventsAllTriggerTowersKeptByRandom(0),
    m_nTriggerTowersProcessed(0),
    m_nTriggerTowersKept(0),
    m_nTriggerTowersRejected(0)
  {
    declareProperty("MinCaloCellET",m_minCaloCellET);
    declareProperty("MinADC",m_minADC);
    declareProperty("UseRandom",m_useRandom);
    declareProperty("MinRandom",m_minRandom);
  }
 
  TriggerTowerThinningAlg::~TriggerTowerThinningAlg()
  {
  }
 
  StatusCode TriggerTowerThinningAlg::initialize(){
    ATH_MSG_INFO("L1Calo TriggerTowerThinningAlg::initialize()");
 
    ATH_CHECK( m_triggerTowerLocation.initialize (m_streamName) );
    ATH_CHECK( m_caloCellETByLayerKey.initialize(SG::AllowEmpty) );
 
    ATH_CHECK( m_rndmSvc.retrieve() );
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode TriggerTowerThinningAlg::doThinning() const{
    const EventContext& ctx = Gaudi::Hive::currentContext();
 
    // Create the mask to be used for thinning
    std::vector<bool> mask;
 
    // Counters
    unsigned long nKeep(0),nReject(0),nTotal(0);
    ++m_nEventsProcessed;
    
    SG::ThinningHandle<xAOD::TriggerTowerContainer> tts
      (m_triggerTowerLocation, ctx);
    
    mask.assign(tts->size(),false); // default: don't keep any clusters
    
    // Should we save the trigger tower or not???
    // If this is calibration data, then yes - save the lot
    // Randomly save all trigger towers m_minRandom % of the time.
    //
    // If this is collisions data then only save if:
    // 1.) Any ADC value is greater than m_minADC  OR
    // 2.) The CaloCell ET is greater than m_minCaloCellET
 
    bool globalSaveMe(false);
    // Random number save all
    if(m_useRandom == true){
      if(globalSaveMe == false){
        ATHRNG::RNGWrapper* wrapper = m_rndmSvc->getEngine (this);
        wrapper->setSeed (this->name(), ctx);
        CLHEP::HepRandomEngine* engine = wrapper->getEngine (ctx);
        if (CLHEP::RandFlat::shoot (engine) < m_minRandom) {
          globalSaveMe = true;
          m_nEventsAllTriggerTowersKeptByRandom++;
        }
      }
    }
 
    SG::ReadDecorHandle<xAOD::TriggerTowerContainer, std::vector<float> > caloCellETByLayerHandle(m_caloCellETByLayerKey, ctx);
    bool isDecorAvailable = caloCellETByLayerHandle.isAvailable();
 
    // Iterate over all trigger towers
    for(auto tt : *tts){
 
      bool saveMe(false);
      if(globalSaveMe == true){saveMe = true;}
      
      // Test for Calo ET
      if(saveMe == false){
    if (isDecorAvailable) {
      const std::vector<float>& caloCellETByLayer = caloCellETByLayerHandle(*tt);
 
          float totalCaloCellET(0.);
          for (float c : caloCellETByLayer) {
            totalCaloCellET += c;
          }
          if (totalCaloCellET > m_minCaloCellET) {
            saveMe = true;
          }
        }
      }
      
 
      // Test for ADC values
      if (saveMe == false) {
        const std::vector<uint16_t>& ttADC = tt->adc();
        for (uint16_t i : ttADC) {
          if ( i > m_minADC) {
            saveMe = true;
            break;
          }
        }
      }
 
      // assign result to mask
      mask[nTotal] = saveMe;
      
      ++nTotal;
    } // End loop over trigger towers
 
 
    tts.keep (mask);
 
    // Counters
    m_nTriggerTowersProcessed += nTotal;
    // FIXME: nKeep is always 0
    m_nTriggerTowersKept += nKeep;
    m_nTriggerTowersRejected += nReject;
 
    ATH_MSG_DEBUG(" L1Calo Trigger Tower Thinning statistics: keeping " << nKeep << " cells"
          << " and rejecting " << nReject << " cells");
 
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode TriggerTowerThinningAlg::finalize(){
 
        ATH_MSG_INFO(
          "==> finalize " << name() << "...\n"
          << "***************************************************************\n"
          << "Results of " << name() << " thinning algorithm:\n"
          << "-------------");
        ATH_MSG_INFO(" Number of processed events:  " << m_nEventsProcessed);
 
        if(m_nTriggerTowersProcessed > 0) { 
          ATH_MSG_INFO(
            " Average percent of Trigger Towers kept = "
            << 100.0 * m_nTriggerTowersKept / (double)m_nTriggerTowersProcessed << " %");
        }
        if(m_useRandom == true && m_nEventsProcessed > 0){
          ATH_MSG_INFO(
            " Percentage of events where all Trigger Towers were kept (should be approx "<< 100.0*m_minRandom << ") = "
            << 100.0 * m_nEventsAllTriggerTowersKeptByRandom / (double)m_nEventsProcessed);
        }
      
 
    return StatusCode::SUCCESS;
  }
 
} // namespace