---

TesIO.h

Go to the documentation of this file.

//
#ifndef ATLFAST_TESIO_H
#define ATLFAST_TESIO_H

// STL
#include <vector>
#include <string>

// Other
//#include "AtlfastCode/HepMC_helper.h"
#include "AtlfastCode/MCparticleCollection.h"
//#include "AtlfastCode/Fallible.h"

//Generator includes
#include "GeneratorObjects/McEventCollection.h"

#// Gaudi includes
//#include "GaudiKernel/SmartDataPtr.h"
//#include "GaudiKernel/DataSvc.h"
//#include "GaudiKernel/IDataProviderSvc.h"
#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"

namespace HepMC_helper{
  class IMCselector;
}

namespace Atlfast {
  class TesIoStat{
  public:
    TesIoStat(): m_status(StatusCode::SUCCESS),m_errMsg(""){}
    TesIoStat(StatusCode sc, std::string string): m_status(sc),
      m_errMsg(string){}
    TesIoStat(StatusCode s): m_status(s), m_errMsg(""){}
    
    StatusCode status()   const {return m_status;}
    std::string message() const {return m_errMsg;}

    bool isValid()        const {return m_status.isSuccess();}
    bool isNotValid()     const {return !isValid();}
    bool operator !()     const {return isNotValid();}
    operator bool()       const {return isValid();}
    operator StatusCode() const {return m_status;}
  private:
    StatusCode m_status;
    std::string m_errMsg;
  };
  
  class TesIO {
  public:
    typedef std::string String;
    //
    TesIO();
    TesIoStat getMC   (MCparticleCollection&)                            const;
    TesIoStat getMC   (MCparticleCollection&, 
                       const HepMC_helper::IMCselector* )                const;
    //
    template<class T> TesIoStat copy  (std::vector<T*>&)                 const;
    template<class T> TesIoStat copy  (std::vector<T*>&,     String key) const;
    template<class T> TesIoStat getDH (const DataHandle<T>&)             const;
    template<class T> TesIoStat getDH (const DataHandle<T>&, String key) const;
    template<class T> TesIoStat store (T*)                               const;
    template<class T> TesIoStat store (T*,                   String key) const;
    
  private:
    //    IDataProviderSvc* m_dataSvc;
    StoreGateSvc* m_sgSvc;
  };


  //**************************************************************************
  // TesIO()                                                                 *
  //**************************************************************************
  inline TesIO::TesIO():m_sgSvc(0){
    ISvcLocator* svcLoc = Gaudi::svcLocator( );
    if( (svcLoc->service( "StoreGateSvc", m_sgSvc )).isFailure() ){
      throw TesIoStat(StatusCode::FAILURE, "TesIO could not find StoreGate");
    }
  }    
  //**************************************************************************
  // TesIoStat TesIO::copy(std::vector<T*>& localCopy) const                 *
  //**************************************************************************
  template<class T> inline 
    TesIoStat TesIO::copy(std::vector<T*>& localCopy) const {
    const DataHandle<ObjectVector<T> > objects;
    TesIoStat stat = this->getDH(objects);
    if(!stat) {
      return TesIoStat(StatusCode::FAILURE,
                       "Could not copy collection: "+stat.message());
    }
    
    std::copy(objects->begin(),objects->end(),back_inserter(localCopy));
    return TesIoStat(StatusCode::SUCCESS, "Copied Collection from TDS");
    
  }
  //**************************************************************************
  //TesIoStat TesIO::copy(std::vector<T*>& localCopy, String key) const      *
  //**************************************************************************
  template<class T> inline 
    TesIoStat TesIO::copy(std::vector<T*>& localCopy, String key) const {
    const DataHandle<ObjectVector<T> > objects;
    TesIoStat stat = this->getDH(objects, key);
    if(!stat){
      return TesIoStat(StatusCode::FAILURE,
                       "Could not copy keyed collection: "+stat.message());
    }

    std::copy(objects->begin(),objects->end(),back_inserter(localCopy));
    return TesIoStat(StatusCode::SUCCESS, "Copied Collection from TDS");
    
  }
  //**************************************************************************
  //*TesIoStat TesIO::getDH(const DataHandle<T>& dh) const                   *
  //**************************************************************************
  template<class T> inline 
    TesIoStat TesIO::getDH(const DataHandle<T>& dh) const {
    if((m_sgSvc->retrieve(dh)).isFailure() ){
      return TesIoStat(StatusCode::FAILURE,
                       "Could not find a valid DataHandle");
    }
    return TesIoStat(StatusCode::SUCCESS, "Found unkeyed DH in TDS");
  }
  //**************************************************************************
  //*TesIoStat TesIO::getDH(const DataHandle<T>& dh, String key) const       *
  //**************************************************************************
  template<class T> inline 
    TesIoStat TesIO::getDH(const DataHandle<T>& dh, String key) const {
    if( (m_sgSvc->retrieve(dh, key)).isFailure() ){
      return TesIoStat(StatusCode::FAILURE,
                       "Could not find a valid keyed DataHandle");
    }
    return TesIoStat(StatusCode::SUCCESS, "Found keyed DH in TDS");
  }
  //**************************************************************************
  //template<class T> inline TesIoStat TesIO::store(T* pObject) const        *
  //**************************************************************************
  template<class T> inline TesIoStat TesIO::store(T* pObject) const { 
    return (TesIoStat(m_sgSvc->record(pObject)) ) ;
  }
  //**************************************************************************
  //TesIoStat TesIO::store(T* storee, String key) const                      *
  //**************************************************************************
  template<class T> inline 
    TesIoStat TesIO::store(T* storee, String key) const { 
    return (TesIoStat(m_sgSvc->record(storee, key)) ) ;
  }
  
}
#endif

---