GenAccessIO.h

Go to the documentation of this file.

//
#ifndef TRUTHHELPER_GENACCESSIO_H
#define TRUTHHELPER_GENACCESSIO_H
// STL
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/ObjectVector.h"
#include <string>
// StoreGate
#include "StoreGate/StoreGateSvc.h"
// #include "StoreGate/DataHandle.h"
 
#include "GeneratorObject/McEventCollection.h"
#include "HepMC/GenParticle.h"
#include <vector>
typedef std::vector<const HepMC::GenParticle*>                 MCparticleCollection;
typedef std::vector<const HepMC::GenParticle*>::const_iterator MCparticleCollectionCIter;
 
class GenIMCselector;
 
class GenAccessIO {
public:
  typedef std::string String;
  //
  GenAccessIO();
  StatusCode getMC( MCparticleCollection& ) const;
  StatusCode getMC( MCparticleCollection&, const GenIMCselector* ) const;
  //
  StatusCode getDH( const McEventCollection*& ) const;
  StatusCode getDH( const McEventCollection*&, String key ) const;
  StatusCode store( McEventCollection* ) const;
  StatusCode store( McEventCollection*, String key ) const;
 
private:
  //    IDataProviderSvc* m_dataSvc;
  StoreGateSvc* m_sgSvc;
};
 
inline GenAccessIO::GenAccessIO() : m_sgSvc( 0 ) {
  ISvcLocator* svcLoc = Gaudi::svcLocator();
  if ( ( svcLoc->service( "StoreGateSvc", m_sgSvc ) ).isFailure() )
  { throw StatusCode::FAILURE; }
}
 
inline StatusCode GenAccessIO::getDH( const McEventCollection*& dh ) const {
  if ( ( m_sgSvc->retrieve( dh ) ).isFailure() ) { return StatusCode::FAILURE; }
  return StatusCode::SUCCESS;
}
 
inline StatusCode GenAccessIO::getDH( const McEventCollection*& dh, String key ) const {
  if ( ( m_sgSvc->retrieve( dh, key ) ).isFailure() ) { return StatusCode::FAILURE; }
  return StatusCode::SUCCESS;
}
 
inline StatusCode GenAccessIO::store( McEventCollection* pObject ) const {
  return ( m_sgSvc->record( pObject ) );
}
 
inline StatusCode GenAccessIO::store( McEventCollection* storee, String key ) const {
  return ( m_sgSvc->record( storee, key ) );
}
 
#endif