EventCollectionSelector.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2025 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#pragma once
 
#include <GaudiKernel/GenericAddress.h>
#include <GaudiKernel/IEvtSelector.h>
#include <GaudiKernel/NTuple.h>
#include <GaudiKernel/Service.h>
 
#include <optional>
 
class INTuple;
class INTupleSvc;
class IAddressCreator;
class EventSelectorDataStream;
template <class TYPE>
class EventIterator;

class EventCollectionSelector : public extends<Service, IEvtSelector> {
public:
  class MyContextType : public IEvtSelector::Context {
  public:
    std::string                                  criteria;
    NTuple::Tuple*                               tuple;
    std::optional<NTuple::Item<IOpaqueAddress*>> item{};
    IOpaqueAddress*                              addressBuffer;
    MyContextType( MyContextType* ctxt = nullptr ) {
      addressBuffer = new GenericAddress();
      addressBuffer->addRef();
      tuple = ( ctxt ) ? ctxt->tuple : nullptr;
      if ( ctxt ) item.emplace( *ctxt->item );
    }
    MyContextType( const MyContextType& ctxt ) : IEvtSelector::Context( ctxt ) {
      addressBuffer = new GenericAddress();
      addressBuffer->addRef();
      tuple = ctxt.tuple;
      item.emplace( *ctxt.item );
    }
    ~MyContextType() override { addressBuffer->release(); }
    MyContextType& operator=( const MyContextType& ) = delete;
 
    void* identifier() const override { return (void*)addressBuffer; }
    void  setAddress( IOpaqueAddress* pAddr );
  };
 
protected:
  mutable SmartIF<INTupleSvc>      m_tupleSvc;
  mutable SmartIF<IAddressCreator> m_pAddrCreator;
 
  // Properties
  Gaudi::Property<std::string> m_tupleSvcName{ this, "CnvService", "EvtTupleSvc",
                                               "name of the event collection service" };
  Gaudi::Property<std::string> m_authentication{ this, "Authentication", "", "authentication string (if needed)" };
  Gaudi::Property<std::string> m_cntName{ this, "Container", "B2PiPi", "container name" };
  Gaudi::Property<std::string> m_itemName{ this, "Item", "Address", "item name" };
  Gaudi::Property<std::string> m_criteria{ this, "Criteria", "", "criteria" };
  Gaudi::Property<std::string> m_database{ this, "DB", "", "datafile name" };
  Gaudi::Property<std::string> m_dbType{ this, "DbType", "", "database type identifier" };
  Gaudi::Property<std::string> m_dbSvc{ this, "DbService", "", "database service (exclusive property with db type)" };
  Gaudi::Property<std::string> m_statement{ this, "Function", "NTuple::Selector", "selector name" };
 
public:
  StatusCode initialize() override;
  StatusCode finalize() override;


  StatusCode createContext( Context*& refpCtxt ) const override;


  StatusCode next( Context& refCtxt ) const override;


  StatusCode next( Context& refCtxt, int jump ) const override;


  StatusCode previous( Context& refCtxt ) const override;


  StatusCode previous( Context& refCtxt, int jump ) const override;


  StatusCode rewind( Context& refCtxt ) const override;


  StatusCode createAddress( const Context& refCtxt, IOpaqueAddress*& refpAddr ) const override;


  StatusCode releaseContext( Context*& refCtxt ) const override;

  StatusCode resetCriteria( const std::string& cr, Context& c ) const override;

  StatusCode last( Context& refCtxt ) const override;

  virtual StatusCode connectCollection( MyContextType* ctxt ) const;

  virtual StatusCode connectDataSource( const std::string& db, const std::string& typ ) const;
  virtual StatusCode connectTuple( const std::string& nam, const std::string& itName, NTuple::Tuple*& tup,
                                   std::optional<NTuple::Item<IOpaqueAddress*>>& item ) const;
  virtual StatusCode connectStatement( const std::string& typ, const std::string& crit, INTuple* tuple ) const;
  virtual StatusCode getNextRecord( NTuple::Tuple* tuple ) const;
  virtual StatusCode getPreviousRecord( NTuple::Tuple* tuple ) const;

  using extends::extends;
};