InputCopyStream.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2019 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.                                             *
\***********************************************************************************/
// Framework include files
#include "InputCopyStream.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/DataStoreItem.h"
#include "GaudiKernel/IDataManagerSvc.h"
#include "GaudiKernel/IDataProviderSvc.h"
#include "GaudiKernel/IDataStoreLeaves.h"
#include "GaudiKernel/IOpaqueAddress.h"
#include "GaudiKernel/IRegistry.h"
#include "GaudiKernel/MsgStream.h"

// Define the algorithm factory for the standard output data writer
DECLARE_COMPONENT( InputCopyStream )

// Standard Constructor
InputCopyStream::InputCopyStream( const std::string& name, ISvcLocator* pSvcLocator )
    : OutputStream( name, pSvcLocator ) {
  m_doPreLoad    = false;
  m_doPreLoadOpt = false;
}

// Initialize the instance.
StatusCode InputCopyStream::initialize() {
  StatusCode sc = OutputStream::initialize();
  if ( sc.isFailure() ) return sc;

  sc = toolSvc()->retrieveTool( "DataSvcFileEntriesTool", "InputCopyStreamTool", m_leavesTool );
  return sc;
}

// Finalize the instance.
StatusCode InputCopyStream::finalize() {
  toolSvc()->releaseTool( m_leavesTool ).ignore();
  m_leavesTool = nullptr;
  return OutputStream::finalize();
}

StatusCode InputCopyStream::collectObjects() {
  // Get the objects in the same file as the root node
  try {
    // Get all the leaves on the input stream
    const IDataStoreLeaves::LeavesList& leaves = m_leavesTool->leaves();

    // Do we need to veto anything ?
    if ( UNLIKELY( !m_tesVetoList.empty() ) ) {
      // Veto out locations
      IDataStoreLeaves::LeavesList filteredLeaves;
      filteredLeaves.reserve( leaves.size() );
      std::copy_if( leaves.begin(), leaves.end(), std::back_inserter( filteredLeaves ),
                    [&]( IDataStoreLeaves::LeavesList::const_reference i ) {
                      return i && i->registry() &&
                             std::find( m_tesVetoList.begin(), m_tesVetoList.end(), i->registry()->identifier() ) ==
                                 m_tesVetoList.end();
                    } );
      // save the veto'ed list
      m_objects.assign( filteredLeaves.begin(), filteredLeaves.end() );

    } else {
      // no veto'ing, so save everything
      m_objects.assign( leaves.begin(), leaves.end() );
    }
  } catch ( const GaudiException& e ) {
    error() << e.message() << endmsg;
    return StatusCode::FAILURE;
  }

  // Collect the other objects from the transient store (mandatory and optional)
  return OutputStream::collectObjects();
}