GenAccessIO.h

Go to the documentation of this file.

//
#ifndef TRUTHHELPER_GENACCESSIO_H
#define TRUTHHELPER_GENACCESSIO_H
// STL
#include <string>
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ObjectVector.h"
// StoreGate
#include "StoreGate/StoreGateSvc.h"
// #include "StoreGate/DataHandle.h"
 
#include "GeneratorObjects/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