GeoGetIds.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
/*
 * Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration.
 */
 
 
#ifndef GEOMODELUTILITIES_GEOGETIDS_H
#define GEOMODELUTILITIES_GEOGETIDS_H

#include "GeoPrimitives/GeoPrimitives.h"
 
#include "GeoModelKernel/GeoNodeAction.h"
 

class IGeoGetIdsAction
{
public:
  virtual ~IGeoGetIdsAction() = default;
  virtual void operator() (int id) = 0;
};
 

class GeoGetIds
  : public GeoNodeAction
{
public:
  GeoGetIds (IGeoGetIdsAction& action, int depthLimit = 1);
  virtual void handleSerialTransformer (const GeoSerialTransformer  *sT) override;
  virtual void handleIdentifierTag (const GeoIdentifierTag *idTag) override;
  virtual void handleSerialIdentifier(const GeoSerialIdentifier *sI) override;
 
 
private:
  IGeoGetIdsAction& m_action;
  const GeoSerialIdentifier *m_serialIdentifier = nullptr;
};
 

template <class FUNCTION>
class GeoGetIdsAction
  : public IGeoGetIdsAction
{
public:
  GeoGetIdsAction (FUNCTION f) : m_f (f) {}
  virtual void operator() (int id) override
  {
    m_f (id);
  }
  FUNCTION m_f;
};
 
 

template <class FUNCTION>
void geoGetIds (FUNCTION f, const GeoGraphNode* node, int depthLimit = 1)
{
  GeoGetIdsAction<FUNCTION> act (f);
  GeoGetIds visitor (act, depthLimit);
  node->exec (&visitor);
}
 
 
#endif // not GEOMODELUTILITIES_GEOGETIDS_H