GeoVisitVolumes.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-2024 CERN for the benefit of the ATLAS collaboration.
 */
#ifndef GEOMODELUTILITIES_GEOVISITVOLUMES_H
#define GEOMODELUTILITIES_GEOVISITVOLUMES_H
 
#include "GeoPrimitives/GeoPrimitives.h"
 
#include "GeoModelKernel/GeoNodeAction.h"
#include "GeoModelKernel/GeoDefinitions.h"
#include <vector>
#include <variant>
 
 
class GeoVPhysVol;
class GeoTransform;
class GeoVAlignmentStore;
 
 
class IGeoVisitVolumesAction
{
public:
  virtual ~IGeoVisitVolumesAction() = default;
  virtual void operator() (GeoNodeAction& action,
                           int id,
                           const std::string& name,
                           const GeoVPhysVol* volume,
                           const GeoTrf::Transform3D& transform,
                           const GeoTrf::Transform3D& defTransform) = 0;
};
 
 
class IGeoVisitVolumesNoXformAction
{
public:
  virtual ~IGeoVisitVolumesNoXformAction() = default;
  virtual void operator() (GeoNodeAction& action,
                           int id,
                           const std::string& name,
                           const GeoVPhysVol* volume) = 0;
};
 
 
class GeoVisitVolumes : public GeoNodeAction
{
public:
  GeoVisitVolumes (IGeoVisitVolumesAction& action, int depthLimit = 1);
 
  GeoVisitVolumes (IGeoVisitVolumesNoXformAction& action, int depthLimit = 1);
 
  virtual void handleTransform (const GeoTransform *xform) override;
 
  virtual void handlePhysVol (const GeoPhysVol *vol) override;
 
  virtual void handleFullPhysVol (const GeoFullPhysVol *vol) override;
 
  virtual void handleNameTag (const GeoNameTag *nameTag) override;
 
  virtual void handleSerialDenominator (const GeoSerialDenominator *sD) override;
 
  virtual void handleSerialTransformer (const GeoSerialTransformer  *sT) override;
 
  virtual void handleIdentifierTag (const GeoIdentifierTag *idTag) override;
 
  virtual void handleSerialIdentifier(const GeoSerialIdentifier *sI) override;
 
 
private:
  GeoVisitVolumes (const GeoVisitVolumes &right) = delete;
  GeoVisitVolumes & operator=(const GeoVisitVolumes &right) = delete;
 
  void handleVol (const GeoVPhysVol *vol);
 
  std::variant<IGeoVisitVolumesAction*,
               IGeoVisitVolumesNoXformAction*> m_action;
 
  const GeoNameTag *m_nameTag;
 
  const GeoSerialDenominator *m_serialDenominator;
 
  const GeoIdentifierTag *m_idTag;
 
  std::vector<const GeoTransform *>  m_pendingTransformList;
 
  const GeoSerialIdentifier *m_serialIdentifier;
};
 
 
template <class FUNCTION>
class GeoVisitVolumesAction
  : public IGeoVisitVolumesAction
{
public:
  GeoVisitVolumesAction (FUNCTION f) : m_f (f) {}
  virtual void operator() (GeoNodeAction& /*action*/,
                           int id,
                           const std::string& name,
                           const GeoVPhysVol* volume,
                           const GeoTrf::Transform3D& transform,
                           const GeoTrf::Transform3D& defTransform) override
  {
    m_f (id, name, volume, transform, defTransform);
  }
  FUNCTION m_f;
};
 
 
template <class FUNCTION>
class GeoVisitVolumesNoXformAction
  : public IGeoVisitVolumesNoXformAction
{
public:
  GeoVisitVolumesNoXformAction (FUNCTION f) : m_f (f) {}
  virtual void operator() (GeoNodeAction& /*action*/,
                           int id,
                           const std::string& name,
                           const GeoVPhysVol* volume) override
  {
    m_f (id, name, volume);
  }
  FUNCTION m_f;
};
 
 
 
template <class FUNCTION>
void geoVisitVolumes (FUNCTION f, const GeoGraphNode* node, int depthLimit = 1)
{
  GeoVisitVolumesAction<FUNCTION> act (f);
  GeoVisitVolumes visitor (act, depthLimit);
  node->exec (&visitor);
}
 
 
template <class FUNCTION>
void geoVisitVolumesNoXform (FUNCTION f, const GeoGraphNode* node, int depthLimit = 1)
{
  GeoVisitVolumesNoXformAction<FUNCTION> act (f);
  GeoVisitVolumes visitor (act, depthLimit);
  node->exec (&visitor);
}
 
 
typedef std::vector<std::pair<const GeoVPhysVol*, GeoTrf::Transform3D> >
  GeoVolumeVec_t;
GeoVolumeVec_t geoGetVolumes (const GeoGraphNode* node,
                              int depthLimit = 1,
                              int sizeHint = 20);
 
 
std::vector<const GeoVPhysVol*>
geoGetVolumesNoXform (const GeoGraphNode* node,
                      int depthLimit = 1,
                      int sizeHint = 20);
 
 
#endif // not GEOMODELUTILITIES_GEOVISITVOLUMES_H