G4CosmicAndFilter.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#include "G4CosmicFilter/G4CosmicAndFilter.h"
#include "MCTruth/TrackHelper.h"
#include "TrackRecord/TrackRecordCollection.h"
#include "G4RunManager.hh"
#include "G4Event.hh"
 
#include "StoreGate/ReadHandle.h"
 
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IMessageSvc.h"
 
namespace G4UA
{
 
  G4CosmicAndFilter::G4CosmicAndFilter(const Config& config)
    : AthMessaging(Gaudi::svcLocator()->service< IMessageSvc >( "MessageSvc" ), "G4CosmicAndFilter"),
      m_config(config), m_report()
  {
  }
 
  void G4CosmicAndFilter::EndOfEventAction(const G4Event*)
  {
 
    m_report.ntot++;
    int counter(0);
    SG::ReadHandle <TrackRecordCollection> coll(m_config.collectionName);
    if (! coll.isValid())
      {
        ATH_MSG_WARNING( "Cannot retrieve TrackRecordCollection " << m_config.collectionName);
      }
    else
      {
        counter = coll->size();
      }
 
    if (counter==0)
      {
        ATH_MSG_INFO("aborting event due to failing AND filter");
        G4RunManager::GetRunManager()->AbortEvent();
        return;
      }
 
    SG::ReadHandle <TrackRecordCollection> coll2(m_config.collectionName2);
    if (! coll2.isValid())
      {
        ATH_MSG_INFO( "Cannot retrieve TrackRecordCollection " << m_config.collectionName2 );
      }
    else
      {
        counter = coll2->size();
      }
 
    if (counter==0)
      {
        ATH_MSG_INFO("aborting event due to failing AND filter");
        G4RunManager::GetRunManager()->AbortEvent();
        return;
      }
 
    m_report.npass++;
    return;
 
  }
 
} // namespace G4UA