#include <AANTEventSelector.h>

Inheritance diagram for AANTEventSelector:

Collaboration diagram for AANTEventSelector:

Public Member Functions
	AANTEventSelector (const std::string &name, ISvcLocator *svcloc)

	~AANTEventSelector ()

virtual StatusCode	initialize ()

virtual StatusCode	createContext (Context *&it) const

virtual StatusCode	next (Context &it) const

virtual StatusCode	next (Context &it, int jump) const

virtual StatusCode	previous (Context &it) const

virtual StatusCode	previous (Context &it, int jump) const

virtual StatusCode	last (Context &it) const

virtual StatusCode	rewind (Context &it) const

virtual StatusCode	createAddress (const Context &it, IOpaqueAddress *&iop) const

virtual StatusCode	releaseContext (Context *&it) const

virtual StatusCode	resetCriteria (const std::string &criteria, Context &context) const

virtual StatusCode	queryInterface (const InterfaceID &riid, void **ppvInterface)

MsgStream &	msg () const

MsgStream &	msg (const MSG::Level lvl) const

bool	msgLvl (const MSG::Level lvl) const

Private Attributes
StoreGateSvc *	m_storeGate

StringArrayProperty	m_inputCollectionsProp

int	m_skipEvents

long	m_numEvents

long	m_totalNEvents

std::string	m_strConverter

std::string	m_strSelection

TChain *	m_tree

Long64_t	m_runNumber

Long64_t	m_eventNumber

PyObject *	m_convFunc

PyObject *	m_selectionFunc

Detailed Description

Definition at line 34 of file AANTEventSelector.h.

Constructor & Destructor Documentation

◆ AANTEventSelector()

AANTEventSelector::AANTEventSelector	(	const std::string &	name,
		ISvcLocator *	svcloc
	)

Definition at line 30 of file AANTEventSelector.cxx.

   : AthService(name, svcloc),
     m_storeGate(0),
     m_skipEvents(0), m_numEvents(0), m_totalNEvents(0),
     m_tree(0),
     m_runNumber(0),
     m_eventNumber(0),
     m_convFunc(0), m_selectionFunc(0)
 {
   declareProperty("InputCollections",          m_inputCollectionsProp);
   declareProperty("SkipEvents",                m_skipEvents = 0);
   declareProperty("Converter",                 m_strConverter = "");
   declareProperty("Selection",                 m_strSelection = "");
 }

◆ ~AANTEventSelector()

AANTEventSelector::~AANTEventSelector ( )

Definition at line 47 of file AANTEventSelector.cxx.

 {
   if (m_tree) 
     delete m_tree;
 }

Member Function Documentation

◆ createAddress()

StatusCode AANTEventSelector::createAddress	(	const Context &	it,
		IOpaqueAddress *&	iop
	)		const

virtual

Definition at line 259 of file AANTEventSelector.cxx.

 {
   return StatusCode::SUCCESS;
 }

◆ createContext()

StatusCode AANTEventSelector::createContext ( IEvtSelector::Context *& it ) const

virtual

Definition at line 141 of file AANTEventSelector.cxx.

 {
   it = new AANTEventContext(this);
   return StatusCode::SUCCESS;
 }

◆ initialize()

StatusCode AANTEventSelector::initialize ( )

virtual

Definition at line 55 of file AANTEventSelector.cxx.

 {
   // Initialize the Service base class.
   StatusCode sc = Service::initialize();
   
   // Create a message stream.
   MsgStream log(msgSvc(), name());
  
   log << MSG::DEBUG << "initialize()" << endmsg;
  
   if (sc.isFailure())
     {
       log << MSG::ERROR << "Unable to initialize Service base class" << endmsg;
       return sc;
     }
  
   sc = service("StoreGateSvc", m_storeGate);
   if (sc.isFailure())
     {
       log << MSG::ERROR << "Unable to retrieve pointer to StoreGateSvc" << endmsg;
       return sc;
     }
   
  
   // parse jobO
   if (m_inputCollectionsProp.value().size() == 0)
     {
       log << MSG::ERROR << "Use the property:"
       << " EventSelector.InputCollections = [ \"<collectionName>\" ] (list of collections)"
       << endmsg;
       return StatusCode::FAILURE;
     }
   // create TChain
   m_tree = new TChain("CollectionTree");
   std::vector<std::string> inputColl = m_inputCollectionsProp.value();
   std::vector<std::string>::iterator it  = inputColl.begin();
   std::vector<std::string>::iterator itE = inputColl.end();
   for (; it!=itE; ++it)
     {
       log << MSG::DEBUG << "Add : " << *it << endmsg;      
       m_tree->Add(it->c_str());
     }
  
   // get total number of events
   m_totalNEvents = m_tree->GetEntries();
   
   log << MSG::DEBUG << "Total Events : " << m_totalNEvents << endmsg;      
   
   // RunNumber and EventNumber
   m_tree->SetBranchAddress("EventNumber",&m_eventNumber);
   m_tree->SetBranchAddress("RunNumber",  &m_runNumber);
  
   // set TTree to AANTTreeGate
   AANTTreeGate::setTree(m_tree);
  
   // selection criteria
   log << MSG::DEBUG << "Load Sel: " << m_strSelection << " from __main__" << endmsg;      
   char smain[] = "__main__";
 #if PY_MAJOR_VERSION < 3
   m_selectionFunc = PyObject_GetAttr(PyImport_AddModule(smain),PyString_FromString(m_strSelection.c_str()));
 #else
   m_selectionFunc = PyObject_GetAttr(PyImport_AddModule(smain),PyUnicode_FromString(m_strSelection.c_str()));
 #endif
   if (m_selectionFunc == NULL)
     {
       log << MSG::ERROR << "Could not load sel : " << m_strSelection << endmsg;
       return StatusCode::FAILURE;
     }
  
   log << MSG::DEBUG << "Load Cnv: " << m_strConverter << " from __main__" << endmsg;      
 #if PY_MAJOR_VERSION < 3
   m_convFunc = PyObject_GetAttr(PyImport_AddModule(smain),PyString_FromString(m_strConverter.c_str()));
 #else
   m_convFunc = PyObject_GetAttr(PyImport_AddModule(smain),PyUnicode_FromString(m_strConverter.c_str()));
 #endif
   if (m_convFunc == NULL)
     {
       log << MSG::ERROR << "Could not load conv : " << m_strConverter << endmsg;
       return StatusCode::FAILURE;
     }
  
   return  StatusCode::SUCCESS;
 }

◆ last()

StatusCode AANTEventSelector::last ( IEvtSelector::Context & it ) const

virtual

Definition at line 232 of file AANTEventSelector.cxx.

 {
   MsgStream log(msgSvc(), name());
   log << MSG::ERROR << "AANTEventSelector::last() not implemented" << endmsg;
   return StatusCode::FAILURE;
 }

◆ msg() [1/2]

MsgStream& AthCommonMsg< Service >::msg

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< Service >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

bool AthCommonMsg< Service >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                                {
     return this->msgLevel(lvl);
   }

◆ next() [1/2]

StatusCode AANTEventSelector::next ( IEvtSelector::Context & it ) const

virtual

Definition at line 149 of file AANTEventSelector.cxx.

 {
   return this->next(it,0);
 }

◆ next() [2/2]

StatusCode AANTEventSelector::next	(	IEvtSelector::Context &	it,
		int	jump
	)		const

virtual

Definition at line 156 of file AANTEventSelector.cxx.

 {
   MsgStream log(msgSvc(), name());
   log << MSG::DEBUG << "next(" << jump << ") : iEvt " << m_numEvents << endmsg;
  
   // get EventContext
   AANTEventContext* ct = dynamic_cast<AANTEventContext*>(&it);
   if (ct == 0)
     {
       log << MSG::ERROR << "Could not dcast to AANTEventContext" << endmsg;
       return StatusCode::FAILURE;
     }
   // jump
   if (((m_numEvents + jump) >= 0) and ((m_numEvents + jump) < m_totalNEvents))
     {
       // move pointer
       m_numEvents += jump;
       // get entry
       m_tree->GetEntry(m_numEvents);
       // increment pointer
       ++m_numEvents;
       // convert C++ obj to Python obj
       RootUtils::PyGILStateEnsure gil;
       PyObject *pyObj = PyObject_CallObject(m_convFunc,NULL); 
       // execute Python code fragment
       PyObject *tup = Py_BuildValue((char*)"(O)",pyObj);
       PyObject *ret = PyObject_CallObject(m_selectionFunc,tup); 
       // decrement reference counters
       Py_DECREF(pyObj); 
       Py_DECREF(tup); 
       if (ret != NULL and PyObject_IsTrue(ret))
     {
       // decrement reference counters
       Py_DECREF(ret); 
       // EventInfo
       EventType *peT = new EventType();
       EventInfo *evtInfo = new EventInfo( new EventID(m_runNumber,m_eventNumber,0),peT);
       StatusCode sc = m_storeGate->record(evtInfo, "AANTEventInfo");
       if (sc.isFailure())
         {
           log << MSG::ERROR << "Could not record AANTEventInfo" << endmsg;
           return sc;
         }
       // return
       return StatusCode::SUCCESS;
     }
       else
     {
       // decrement reference counters
       Py_XDECREF(ret); 
       // next
       return this->next(it,0);
     }
     }
  
   // EOF
   //ct = new AANTEventContext(0);
   return StatusCode::FAILURE; 
 }

◆ previous() [1/2]

StatusCode AANTEventSelector::previous ( IEvtSelector::Context & it ) const

virtual

Definition at line 218 of file AANTEventSelector.cxx.

 {
   return this->next(it,-1);
 }

◆ previous() [2/2]

StatusCode AANTEventSelector::previous	(	IEvtSelector::Context &	it,
		int	jump
	)		const

virtual

Definition at line 225 of file AANTEventSelector.cxx.

 {
   return this->next(it,-jump);
 }

◆ queryInterface()

StatusCode AANTEventSelector::queryInterface	(	const InterfaceID &	riid,
		void **	ppvInterface
	)

virtual

Definition at line 276 of file AANTEventSelector.cxx.

 {
   if (riid == IEvtSelector::interfaceID())  
      {
        *ppvInterface = (IEvtSelector*)this;
      }
   else if (riid == IProperty::interfaceID())
      {
        *ppvInterface = (IProperty*)this;
      }
    else   
      {
        return Service::queryInterface(riid, ppvInterface);
      }
    
    addRef();
    
    return StatusCode::SUCCESS; 
 }

◆ releaseContext()

StatusCode AANTEventSelector::releaseContext ( IEvtSelector::Context *& it ) const

virtual

Definition at line 267 of file AANTEventSelector.cxx.

 {
   MsgStream log(msgSvc(), name());
   log << MSG::ERROR << "AANTEventSelector::releaseContext() not implemented" << endmsg;
   return StatusCode::FAILURE;
 }

◆ resetCriteria()

StatusCode AANTEventSelector::resetCriteria	(	const std::string &	criteria,
		IEvtSelector::Context &	context
	)		const

virtual

Definition at line 241 of file AANTEventSelector.cxx.

 {
   MsgStream log(msgSvc(), name());
   log << MSG::ERROR << "AANTEventSelector::resetCriteria() not implemented" << endmsg;
   return StatusCode::FAILURE;
 }

◆ rewind()

StatusCode AANTEventSelector::rewind ( IEvtSelector::Context & it ) const

virtual

Definition at line 250 of file AANTEventSelector.cxx.

 {
   MsgStream log(msgSvc(), name());
   log << MSG::ERROR << "AANTEventSelector::rewind() not implemented" << endmsg;
   return StatusCode::FAILURE;
 }

Member Data Documentation

◆ m_convFunc

PyObject* AANTEventSelector::m_convFunc

private

Definition at line 95 of file AANTEventSelector.h.

◆ m_eventNumber

Long64_t AANTEventSelector::m_eventNumber

private

Definition at line 92 of file AANTEventSelector.h.

◆ m_inputCollectionsProp

StringArrayProperty AANTEventSelector::m_inputCollectionsProp

private

Definition at line 71 of file AANTEventSelector.h.

◆ m_numEvents

long AANTEventSelector::m_numEvents

mutableprivate

Definition at line 77 of file AANTEventSelector.h.

◆ m_runNumber

Long64_t AANTEventSelector::m_runNumber

private

Definition at line 91 of file AANTEventSelector.h.

◆ m_selectionFunc

PyObject* AANTEventSelector::m_selectionFunc

private

Definition at line 96 of file AANTEventSelector.h.

◆ m_skipEvents

int AANTEventSelector::m_skipEvents

private

Definition at line 74 of file AANTEventSelector.h.

◆ m_storeGate

StoreGateSvc* AANTEventSelector::m_storeGate

private

Definition at line 68 of file AANTEventSelector.h.

◆ m_strConverter

std::string AANTEventSelector::m_strConverter

private

Definition at line 83 of file AANTEventSelector.h.

◆ m_strSelection

std::string AANTEventSelector::m_strSelection

private

Definition at line 86 of file AANTEventSelector.h.

◆ m_totalNEvents

long AANTEventSelector::m_totalNEvents

private

Definition at line 80 of file AANTEventSelector.h.

◆ m_tree

TChain* AANTEventSelector::m_tree

private

Definition at line 89 of file AANTEventSelector.h.

The documentation for this class was generated from the following files:

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ AANTEventSelector()

◆ ~AANTEventSelector()

Member Function Documentation

◆ createAddress()

◆ createContext()

◆ initialize()

◆ last()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ next() [1/2]

◆ next() [2/2]

◆ previous() [1/2]

◆ previous() [2/2]

◆ queryInterface()

◆ releaseContext()

◆ resetCriteria()

◆ rewind()

Member Data Documentation

◆ m_convFunc

◆ m_eventNumber

◆ m_inputCollectionsProp

◆ m_numEvents

◆ m_runNumber

◆ m_selectionFunc

◆ m_skipEvents

◆ m_storeGate

◆ m_strConverter

◆ m_strSelection

◆ m_totalNEvents

◆ m_tree