TBEventInfoStreamerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TBEventInfoStreamerTool.h"
#include "TBEvent/TBEventInfo.h"
 
#include <string>
#include <ios>
#include <algorithm>
 
TBEventInfoStreamerTool::TBEventInfoStreamerTool(const std::string& type,
                         const std::string& name,
                             const IInterface* parent)
  : TBEventStreamerTool(type,name,parent),
    m_selected_events(0)
{
   declareInterface<TBEventStreamerTool>(this);
  m_acceptTypes.resize(1,"Physics");
  declareProperty("AcceptedTypes",m_acceptTypes);
}
 
TBEventInfoStreamerTool::~TBEventInfoStreamerTool()
{ }

// Initialize //
 
StatusCode TBEventInfoStreamerTool::initializeTool()
{
  m_selected_events = 0;
 
  for ( unsigned int i=0; i<m_acceptTypes.size(); i++ )
    {
      if ( m_acceptTypes[i] == "Physics" )
    {
      m_acceptCodes.push_back(1);
    }
      else if ( m_acceptTypes[i] == "Special" )
    {
      m_acceptCodes.push_back(0);
    }
      else if ( m_acceptTypes[i] == "f/e_Calibration" )
    {
      m_acceptCodes.push_back(2);
    }
      else if ( m_acceptTypes[i] == "Random" )
    {
      m_acceptCodes.push_back(3);
    }
      else if ( m_acceptTypes[i] == "BPC_Calibration" )
    {
      m_acceptCodes.push_back(4);
    }
      else
    {
      ATH_MSG_FATAL ( "Detector event type not existing " << m_acceptTypes );
      ATH_MSG_FATAL ( "Possible types are: Special, Physics, f/e_calibration, Random, BPC_Calibration" );
      return StatusCode::FAILURE;
    }
     
    }
 
  if (m_acceptCodes.size()==0) {
    ATH_MSG_ERROR ( "No detector event type specified! Please specify at least one in the JobOptions" );
    ATH_MSG_ERROR ( "Possible types are: Special, Physics, f/e_Calibration, Random, BPC_Calibration");
  }
  else {
    for ( unsigned int i=0; i<m_acceptCodes.size(); i++ )
      {
    ATH_MSG_DEBUG ( "Detector event code  found " << m_acceptCodes[i] );
      }
  }
 
  return StatusCode::SUCCESS;
}

// Accept/Reject //
 
StatusCode TBEventInfoStreamerTool::accept()
{
  // retrieve Event Info
  const TBEventInfo* theEventInfo;
  ATH_CHECK( evtStore()->retrieve(theEventInfo,"TBEventInfo") );
 
  int evtType = theEventInfo->getEventType();
  ATH_MSG_DEBUG ( "Event Type found " << evtType );
 
  return (std::find(m_acceptCodes.begin(),m_acceptCodes.end(),evtType) != 
      m_acceptCodes.end())  
    ? StatusCode::SUCCESS 
    : StatusCode::FAILURE;
 
}