PyDataHeader.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PYANALYSISCORE_PYDATAHEADER_H
#define PYANALYSISCORE_PYDATAHEADER_H
 
#include "PersistentDataModel/DataHeader.h"
#include "StoreGate/StoreGateSvc.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/MsgStream.h"
#include "AthenaKernel/getMessageSvc.h"
 
#include <vector>
 
class PyDataHeader
{
public:
  PyDataHeader () : m_DataHeaderElementV(0)
  {
    MsgStream log(Athena::getMessageSvc(), "PyDataHeader");
 
    // get StoreGate
    StoreGateSvc *pSvc = 0;
    StatusCode sc = Gaudi::svcLocator()->service("StoreGateSvc", pSvc);
    if (sc.isFailure())
      log << MSG::ERROR << "could not get StoreGateSvc" << endmsg;
    else
      {
    // retrieve DataHeader
        SG::ConstIterator<DataHeader> beg; 
        SG::ConstIterator<DataHeader> ending; 
    sc = pSvc->retrieve(beg,ending);
    if (sc.isFailure() || beg==ending)
      log << MSG::ERROR << "could not get DataHeader" << endmsg;
    else
      {
        // cache DataHeaderElement
        for (; beg != ending; ++beg)
          {
        const DataHeader &dh = *beg;
        std::vector<DataHeaderElement>::const_iterator it = dh.begin();
        for (; it != dh.end(); ++it)
          m_DataHeaderElementV.push_back(&*it);
          }
      }
      }
 
    m_classIDSvc = 0;
    sc = Gaudi::svcLocator()->service("ClassIDSvc", m_classIDSvc);
    if (sc.isFailure())
      log << MSG::ERROR << "could not get ClassIDSvc" << endmsg;
  }
 
  virtual ~PyDataHeader () {}
 
  int size() { return m_DataHeaderElementV.size(); }
 
  std::string getName (int index)
  {
    if (index >= size()) return "";
    std::vector<const DataHeaderElement *>::iterator it = m_DataHeaderElementV.begin();
    it += index;
 
    // look for non-symlink name
    std::string name="";
    for (CLID clid : (*it)->getClassIDs())
      {
    // convert CLID to class name
    std::string localName;
    StatusCode sc = m_classIDSvc->getTypeNameOfID (clid, localName);
    if (sc.isFailure()) 
      {
        MsgStream log(Athena::getMessageSvc(), "PyDataHeader");
        log << MSG::ERROR << "could not get TypeName for " << clid << endmsg;
        return "";
      }
    // select non-symlink
    if ((name == "") || 
        ((localName != "ParticleBaseContainer") && 
         (localName != "IParticleContainer") &&
         (localName != "INavigable4MomentumCollection")))
      name = localName;
      }
    return name;
  }
 
  std::string getKey (int index)
  {
    if (index >= size()) return "";
    std::vector<const DataHeaderElement *>::iterator it = m_DataHeaderElementV.begin();
    it += index;
    return (*it)->getKey();
  }
 
private:
  std::vector<const DataHeaderElement *> m_DataHeaderElementV;
 
  IClassIDSvc *m_classIDSvc;
};
 
#endif