Loading [MathJax]/extensions/tex2jax.js
The Gaudi Framework  v31r0 (aeb156f0)
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
EventCollectionSelector.cpp
Go to the documentation of this file.
1 // Include files
10 #include "GaudiKernel/NTuple.h"
12 #include "GaudiKernel/SmartIF.h"
13 
15 
16 #include <memory>
17 
19 
20 
26 public:
28 
29 private:
30  GenericAddress* m_pAddressBuffer = nullptr;
31  const EventCollectionSelector* m_pSelector = nullptr;
32  ListName m_files;
34  ListName::const_iterator m_fileIterator;
36 
37 public:
41  ~EventCollectionContext() override;
42  const std::string& currentInput() const { return m_currentInput; }
43  void setCurrentInput( const std::string& v ) { m_currentInput = v; }
44  ListName& files() { return m_files; }
45  void* identifier() const override { return (void*)m_pSelector; }
46  void setCriteria( const std::string& crit ) { m_criteria = crit; }
47  ListName::const_iterator fileIterator() { return m_fileIterator; }
48  void setFileIterator( ListName::const_iterator new_iter ) { m_fileIterator = new_iter; }
49 };
50 
52 
56 }
57 
58 // IService implementation: Db event selector override
60  // Initialize base class
62  if ( !status.isSuccess() ) {
63  error() << "Error initializing base class Service!" << endmsg;
64  return status;
65  }
66  m_pAddrCreator = serviceLocator()->service( "EventPersistencySvc" );
67  if ( !m_pAddrCreator ) {
68  error() << "Unable to locate IAddressCreator interface of "
69  << "EventPersistencySvc" << endmsg;
70  return status;
71  }
72  m_tupleSvc = serviceLocator()->service( m_tupleSvcName );
73  if ( !m_tupleSvc ) {
74  error() << "Unable to locate INTupleSvc interface of " << m_tupleSvcName << endmsg;
75  return status;
76  }
77  return status;
78 }
79 
80 // Connect collection to selector
83  SmartIF<IDataSourceMgr> svc( m_tupleSvc );
84  if ( svc && !db.empty() ) {
85  std::string ident = name() + ' ';
86  ident += "DATAFILE='" + m_database.value().substr( 5 ) + "' ";
87  if ( !m_dbSvc.empty() )
88  ident += "SVC='" + m_dbSvc + "' ";
89  else
90  ident += "TYP='" + m_dbType + "' ";
91  ident += "OPT='READ' ";
92  if ( m_authentication.length() > 0 ) { ident += "AUTH='" + m_authentication + "' "; }
93  status = svc->connect( ident );
94  }
95  return status;
96 }
97 
100  NTuple::Tuple*& tup, NTuple::Item<IOpaqueAddress*>*& item ) const {
101  std::string top = "/NTUPLES/" + name() + '/' + nam;
102  StatusCode status = m_tupleSvc->retrieveObject( top, (DataObject*&)tup );
103  if ( status.isSuccess() ) {
104  item = new NTuple::Item<IOpaqueAddress*>();
105  status = tup->item( itName, *item );
106  if ( status.isSuccess() ) return status;
107  error() << "Item " << itName << " is not part of the collection:" << top << endmsg;
108  delete item;
109  item = nullptr;
110  } else {
111  error() << "Cannot connect to collection:" << top << endmsg;
112  }
113  tup = nullptr;
114  return status;
115 }
116 
119  INTuple* tuple ) const {
120  std::string seltyp = typ;
121  if ( !seltyp.empty() || !crit.empty() ) {
122  if ( !crit.empty() && seltyp.length() == 0 ) seltyp = "NTuple::Selector";
123  SmartIF<ISelectStatement> stmt( ObjFactory::create( seltyp, serviceLocator() ).release() );
124  if ( stmt ) {
125  if ( !crit.empty() ) stmt->setCriteria( crit );
126  tuple->attachSelector( stmt ).ignore();
127  return StatusCode::SUCCESS;
128  }
129  return StatusCode::FAILURE;
130  }
131  return StatusCode::SUCCESS;
132 }
133 
137  if ( tuple ) {
138  do {
139  status = m_tupleSvc->readRecord( tuple );
140  if ( status.isSuccess() ) {
141  ISelectStatement* statement = tuple->selector();
142  bool use_it = ( statement ) ? ( *statement )( tuple ) : true;
143  if ( use_it ) { return status; }
144  }
145  } while ( status.isSuccess() );
146  }
147  return status;
148 }
149 
153  if ( tuple ) {
154  IRegistry* pReg = tuple->registry();
155  if ( pReg ) {
156  IOpaqueAddress* pAddr = pReg->address();
157  if ( pAddr ) {
158  long* ip = (long*)pAddr->ipar();
159  do {
160  if ( ip[1] > 1 ) {
161  ip[1] -= 2;
162  status = m_tupleSvc->readRecord( tuple );
163  if ( status.isSuccess() ) {
164  ISelectStatement* statement = tuple->selector();
165  bool use_it = ( statement ) ? ( *statement )( tuple ) : true;
166  if ( use_it ) { return status; }
167  }
168  } else {
169  return StatusCode::FAILURE;
170  }
171  } while ( status.isSuccess() );
172  }
173  }
174  }
175  return StatusCode::FAILURE;
176 }
177 
178 // Connect collection to selector
180  if ( ctxt ) {
181  StatusCode status = connectDataSource( m_database, m_dbType );
182  if ( status.isSuccess() ) {
183  status = connectTuple( m_cntName, m_itemName, ctxt->tuple, ctxt->item );
184  if ( status.isSuccess() ) {
185  status = connectStatement( m_statement, m_criteria, ctxt->tuple );
186  if ( status.isSuccess() ) {
187  *( ctxt->item ) = 0;
188  return status;
189  }
190  }
191  }
192  return status;
193  }
194  return StatusCode::FAILURE;
195 }
196 
197 // Finalize service
199  // release services
200  m_pAddrCreator = nullptr;
201  m_tupleSvc = nullptr;
202  return Service::finalize();
203 }
204 
207  refpCtxt = nullptr;
208  auto ctxt = std::make_unique<MyContextType>();
209  StatusCode status = connectCollection( ctxt.get() );
210  if ( !status.isSuccess() ) {
211  error() << "Unable to connect Collection file \"" << m_database << "\"" << endmsg;
212  return status;
213  }
214  refpCtxt = ctxt.release();
215  return StatusCode::SUCCESS;
216 }
217 
219 StatusCode EventCollectionSelector::next( Context& refCtxt ) const { return next( refCtxt, 1 ); }
220 
222 StatusCode EventCollectionSelector::next( Context& refCtxt, int jump ) const {
223  MyContextType* ctxt = dynamic_cast<MyContextType*>( &refCtxt );
224  if ( ctxt ) {
225  *( ctxt->item ) = ctxt->addressBuffer;
227  for ( int i = 0; i < jump && sc.isSuccess(); ++i ) { sc = getNextRecord( ctxt->tuple ); }
228  return sc;
229  }
230  return StatusCode::FAILURE;
231 }
232 
234 StatusCode EventCollectionSelector::previous( Context& refCtxt ) const { return previous( refCtxt, 1 ); }
235 
237 StatusCode EventCollectionSelector::previous( Context& refCtxt, int jump ) const {
238  MyContextType* ctxt = dynamic_cast<MyContextType*>( &refCtxt );
239  if ( ctxt ) {
240  *( ctxt->item ) = ctxt->addressBuffer;
242  for ( int i = 0; i < jump && sc.isSuccess(); ++i ) { sc = getPreviousRecord( ctxt->tuple ); }
243  return sc;
244  }
245  return StatusCode::FAILURE;
246 }
247 
249 StatusCode EventCollectionSelector::rewind( Context& /* refCtxt */ ) const { return StatusCode::FAILURE; }
250 
252 StatusCode EventCollectionSelector::createAddress( const Context& refCtxt, IOpaqueAddress*& refpAddr ) const {
253  const MyContextType* ctxt = dynamic_cast<const MyContextType*>( &refCtxt );
254  if ( ctxt ) {
255  IOpaqueAddress* pA = *( ctxt->item );
256  if ( pA ) {
257  IOpaqueAddress* pAddress = nullptr;
258  StatusCode status = m_pAddrCreator->createAddress( pA->svcType(), pA->clID(), pA->par(), pA->ipar(), pAddress );
259  if ( status.isSuccess() ) {
260  refpAddr = pAddress;
261  return StatusCode::SUCCESS;
262  } else {
263  error() << "Failed to access " << pA->par()[0] << ":" << pA->par()[1] << " SvcTyp:" << long( pA->svcType() )
264  << " CLID:" << pA->clID() << endmsg;
265  }
266  }
267  }
268  return StatusCode::FAILURE;
269 }
270 
273  MyContextType* ctxt = dynamic_cast<MyContextType*>( refCtxt );
274  if ( ctxt ) {
275  delete ctxt;
276  ctxt = nullptr;
277  return StatusCode::SUCCESS;
278  }
279  return StatusCode::FAILURE;
280 }
281 
284 StatusCode EventCollectionSelector::resetCriteria( const std::string& cr, Context& refCtxt ) const {
285  MyContextType* ctxt = dynamic_cast<MyContextType*>( &refCtxt );
286  if ( ctxt ) {
287  ctxt->criteria = cr;
288  return StatusCode::SUCCESS;
289  }
290  return StatusCode::FAILURE;
291 }
292 
294 StatusCode EventCollectionSelector::last( Context& /* refCtxt */ ) const { return StatusCode::FAILURE; }
StatusCode initialize() override
Definition: Service.cpp:60
Small smart pointer class with automatic reference counting for IInterface.
Definition: IConverter.h:15
T empty(T...args)
StatusCode createContext(Context *&refpCtxt) const override
Create a new event loop context.
virtual const CLID & clID() const =0
Retrieve class information from link.
virtual StatusCode getNextRecord(NTuple::Tuple *tuple) const
Read next record of the N-tuple.
StatusCode resetCriteria(const std::string &cr, Context &c) const override
Will set a new criteria for the selection of the next list of events and will change the state of the...
StatusCode finalize() override
Definition: Service.cpp:164
A select statement can either contain.
EventCollectionContext(const EventCollectionSelector *pSelector)
Standard constructor.
The Event Selector Interface.
Definition: IEvtSelector.h:18
unsigned long addRef() override
Add reference to object.
bool isSuccess() const
Definition: StatusCode.h:267
unsigned long release() override
release reference to object
virtual StatusCode connectDataSource(const std::string &db, const std::string &typ) const
Connect collection&#39;s data source to selector.
~EventCollectionContext() override
Standard destructor.
constexpr static const auto SUCCESS
Definition: StatusCode.h:85
#define class
Generic Transient Address.
virtual ISelectStatement * selector()=0
Access selector.
virtual const std::string * par() const =0
Retrieve String parameters.
IRegistry * registry() const
Get pointer to Registry.
Definition: DataObject.h:72
StatusCode createAddress(const Context &refCtxt, IOpaqueAddress *&refpAddr) const override
Create new Opaque address corresponding to the current record.
STL class.
#define DECLARE_COMPONENT(type)
NTuple interface class definition.
Definition: INTuple.h:81
StatusCode finalize() override
Service override: Finalize Service.
virtual StatusCode connect(const std::string &logon, std::string &identifier)=0
Connect data source.
ListName::const_iterator fileIterator()
const std::string & currentInput() const
virtual const unsigned long * ipar() const =0
Access to generic link parameters.
This class is used for returning status codes from appropriate routines.
Definition: StatusCode.h:50
void setFileIterator(ListName::const_iterator new_iter)
The IRegistry represents the entry door to the environment any data object residing in a transient da...
Definition: IRegistry.h:22
StatusCode last(Context &refCtxt) const override
Access last item in the iteration.
virtual long svcType() const =0
Retrieve service type.
Abstract base class which allows the user to interact with the actual N tuple implementation.
Definition: NTuple.h:375
void setCurrentInput(const std::string &v)
T length(T...args)
const EventCollectionSelector * m_pSelector
virtual StatusCode getPreviousRecord(NTuple::Tuple *tuple) const
Read next record of the N-tuple.
Definition of class EventCollectionSelector.
StatusCode previous(Context &refCtxt) const override
Get previous iteration item from the event loop context.
const StatusCode & ignore() const
Ignore/check StatusCode.
Definition: StatusCode.h:153
StatusCode releaseContext(Context *&refCtxt) const override
Release existing event iteration context.
constexpr static const auto FAILURE
Definition: StatusCode.h:86
NTuple::Item< IOpaqueAddress * > * item
StatusCode initialize() override
Service override: Initialize service.
T substr(T...args)
virtual StatusCode connectTuple(const std::string &nam, const std::string &itName, NTuple::Tuple *&tup, NTuple::Item< IOpaqueAddress * > *&item) const
Connect to existing N-tuple.
std::list< std::string > ListName
Opaque address interface definition.
StatusCode rewind(Context &refCtxt) const override
Rewind the dataset.
virtual StatusCode attachSelector(ISelectStatement *sel)=0
Attach selector.
virtual StatusCode connectCollection(MyContextType *ctxt) const
Connect collection to selector.
ListName::const_iterator m_fileIterator
virtual IOpaqueAddress * address() const =0
Retrieve opaque storage address.
virtual void setCriteria(const std::string &crit)=0
Set the type.
A DataObject is the base class of any identifiable object on any data store.
Definition: DataObject.h:30
StatusCode next(Context &refCtxt) const override
Get next iteration item from the event loop context.
MsgStream & endmsg(MsgStream &s)
MsgStream Modifier: endmsg. Calls the output method of the MsgStream.
Definition: MsgStream.h:192
void setCriteria(const std::string &crit)
void * identifier() const override
virtual StatusCode connectStatement(const std::string &typ, const std::string &crit, INTuple *tuple) const
Connect selection statement to refine data access.