PyDataHeader.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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 "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/SmartIF.h"
#include "AthenaKernel/getMessageSvc.h"
 
#include <vector>
 
class PyDataHeader
{
public:
  PyDataHeader () : m_DataHeaderElementV(0)
  {
    MsgStream log(Athena::getMessageSvc(), "PyDataHeader");
 
    // get StoreGate
    SmartIF<StoreGateSvc> pSvc{ Gaudi::svcLocator()->service("StoreGateSvc") };
    if (!pSvc)
      log << MSG::ERROR << "could not get StoreGateSvc" << endmsg;
    else
      {
    // retrieve DataHeader
        SG::ConstIterator<DataHeader> beg; 
        SG::ConstIterator<DataHeader> ending; 
    StatusCode 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);
          }
      }
      }
 
    if (m_classIDSvc.retrieve().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;

  ServiceHandle<IClassIDSvc> m_classIDSvc{"ClassIDSvc", "PyDataHeader"};
};
 
#endif