VolumeTreeNavigator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
//                              //
//          VolumeTreeNavigator.h           //
//  Class containing functions to allow a general       //
//  navigation of uniquely specified GEANT4 volumes     //
//  and collection of their tree patterns and locations.    //
//                              //
//  Written by  Kevin Sapp              //
//          University of Pittsburgh        //
//          kcs34@pitt.edu              //
//  Last update 02.02.10                //
//                                                              //
//                                                              //
 
#ifndef VolumeTreeNavigator_H
#define VolumeTreeNavigator_H
 
#include <string>
#include <vector>
#include <stdarg.h>
 
#include "AthenaBaseComps/AthMessaging.h"
 
class G4Step;
class G4Track;
class G4StepPoint;
class G4TouchableHistory;
class G4VPhysicalVolume;
 
typedef std::pair<G4VPhysicalVolume*,int> VolID;
typedef std::vector<VolID> VolTree;
typedef VolTree::iterator VolNav;
 
class VolumeTreeNavigator : public AthMessaging {
  public:
      VolumeTreeNavigator(const G4Step*);
 
      void               SetDepthCutSimple(const int, const int, const int, const int);
      void               SetDepthCutDetail(const char*);
      VolTree            Extract();
      bool               Ascend(int levels = 1);
      bool               Descend(int levels = 1);
      bool               KillProcesses(const int, const char*, ...);
 
      int                GetCurrentDepth();
      G4VPhysicalVolume* GetVolume(int rel = 0) const;
      int                GetCopyNumber(int rel = 0) const;
      const G4Track*     GetTrack() const;
      const G4StepPoint* GetPreStepPoint() const;
      const G4StepPoint* GetPostStepPoint() const;
      int                GetStepNumber() const;
      const VolTree&     GetHistory() const;
      int                GetFullDepth() const;
 
  private:
      G4Track* m_track;             
      G4StepPoint* m_preStepPoint;      
      G4StepPoint* m_postStepPoint;     
      const G4TouchableHistory* m_preHistory;       
      int m_preDepth, m_stepNo;         
 
      VolTree m_history;                
      VolNav m_h_end;               
      VolNav m_h_nav;               
};
 
template <class T>
inline std::string stringify(T obj)
{
  std::ostringstream ss;
  ss << obj;
  return ss.str();
}
 
template<class T>
inline std::string cleanstr(T obj)
{
  std::string s = stringify(obj);
  std::string::size_type col = s.find_first_of(":");
  while (col != std::string::npos) {
      (s.substr(col,2) == "::") ? s.replace(col,2,"_") : s.replace(col,1,"-");
      col = s.find_first_of(":");
  }
  std::string::size_type sp = s.find_first_of(" ");
  while (sp != std::string::npos) {
      s.erase(sp,1);
      sp = s.find_first_of(" ");
  }
  return s;
}
 
#include "VolumeTreeNavigator.icc"
#endif