NTupleImplementation.cpp

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.                                             *
\***********************************************************************************/
#include <GaudiKernel/IDataSelector.h>
#include <GaudiKernel/INTupleSvc.h>
#include <GaudiKernel/ISelectStatement.h>
#include <GaudiKernel/NTupleImplementation.h>
 
namespace NTuple {
  TupleImp::TupleImp( std::string title ) : m_title( std::move( title ) ) {}
  TupleImp::~TupleImp() {
    for ( auto& i : m_items ) i->release();
  }

  StatusCode TupleImp::attachSelector( ISelectStatement* sel ) {
    m_pSelector = sel;
    return StatusCode::SUCCESS;
  }

  ISelectStatement* TupleImp::selector() { return m_pSelector.get(); }

  void TupleImp::reset() {
    for ( auto& i : m_items ) i->reset();
  }

  INTupleItem* TupleImp::i_find( const std::string& name ) const {
    auto i = std::find_if( std::begin( m_items ), std::end( m_items ),
                           [&]( ItemContainer::const_reference j ) { return j->name() == name; } );
    return i != std::end( m_items ) ? const_cast<INTupleItem*>( *i ) : nullptr;
  }

  StatusCode TupleImp::add( INTupleItem* item ) {
    if ( item ) {
      INTupleItem* i = i_find( item->name() );
      if ( !i ) {
        m_items.push_back( item );
        return StatusCode::SUCCESS;
      }
    }
    return StatusCode::FAILURE;
  }

  StatusCode TupleImp::remove( const std::string& name ) {
    INTupleItem* i = i_find( name );
    return i ? remove( i ) : StatusCode::FAILURE;
  }

  StatusCode TupleImp::remove( INTupleItem* item ) {
    auto i = std::find( std::begin( m_items ), std::end( m_items ), item );
    if ( i == std::end( m_items ) ) return StatusCode::FAILURE;
    m_items.erase( i );
    item->release();
    return StatusCode::SUCCESS;
  }

  char* TupleImp::setBuffer( std::unique_ptr<char[]>&& buff ) {
    m_buffer = std::move( buff );
    return m_buffer.get();
  }

  char* TupleImp::setBuffer( char* buff ) {
    m_buffer.reset( buff );
    return m_buffer.get();
  }

  StatusCode TupleImp::write() { return m_ntupleSvc->writeRecord( this ); }
  StatusCode TupleImp::writeRecord() { return m_ntupleSvc->writeRecord( this ); }
  StatusCode TupleImp::read() { return m_ntupleSvc->readRecord( this ); }
  StatusCode TupleImp::readRecord() { return m_ntupleSvc->readRecord( this ); }
  StatusCode TupleImp::save() { return m_ntupleSvc->save( this ); }
} // namespace NTuple