TrackingVolumeDisplayer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

// TrackingVolumeDisplayer.h, (c) ATLAS Detector software
 
#ifndef TRKDETDESCRTOOLS_TRACKINGVOLUMEDISPLAYER_H
#define TRKDETDESCRTOOLS_TRACKINGVOLUMEDISPLAYER_H
 
// Trk
#include "TrkDetDescrTestTools/RecursiveGeometryProcessor.h"
// STL
#include <string>
#include <fstream>
 
#include "CxxUtils/checker_macros.h" 
namespace Trk {
 
    class TrackingVolume;
    class Layer;
    class Surface;
    class MaterialProperties;

 
    class ATLAS_NOT_THREAD_SAFE TrackingVolumeDisplayer : //mutable
      virtual public RecursiveGeometryProcessor {
 
      public:

        using RecursiveGeometryProcessor::RecursiveGeometryProcessor;

        StatusCode initialize();
        
        StatusCode finalize();
 
      private:

        StatusCode processNode(const TrackingVolume& tvol, size_t level=0) const;

        StatusCode processNode(const Layer& lay, size_t level=0) const;

        StatusCode processNode(const Surface&, size_t level=0) const;
 
        static void openFile(std::ofstream& output, const std::string& filename) ;  
        static void closeFile(std::ofstream& output) ;                              

        static int colorCodeFromMaterial(const Trk::MaterialProperties* prop, std::ofstream& output) ;
 
        mutable int                         m_volumeCounter = 0;                      
 
        mutable std::ofstream               m_fileVolumeOutput;                   
        Gaudi::Property<std::string> m_fileVolumeOutputName
          {this, "TrackingVolumeOutputFile", "TrackingGeometryVolumeDisplay.C",
           "file name for visualization action"};
        Gaudi::Property<bool> m_fileVolumeOutputMode
          {this, "TrackingVolumeOutput", true, "steer writing"};
 
        mutable std::ofstream               m_fileLayerOutput;                    
        Gaudi::Property<std::string> m_fileLayerOutputName
          {this, "LayerOutputFile", "TrackingGeometryLayerDisplay.C",
           "file name for visualization action"};
        Gaudi::Property<bool> m_fileLayerOutputMode
          {this, "LayerOutput", true, "steer writing"};
 
        mutable std::ofstream               m_fileSurfaceOutput;                  
        Gaudi::Property<std::string> m_fileSurfaceOutputName
          {this, "SurfaceOutputFile", "TrackingGeometrySurfaceDisplay.C",
           "file name for visualization action"};
        Gaudi::Property<bool> m_fileSurfaceOutputMode
          {this, "SurfaceOutput", true, "steer writing"};
        Gaudi::Property<bool> m_fileSurfaceOutputSplit
          {this, "SurfaceOutputSplit", false, "use one file for each layer"};
 
        static int                          s_displaySurfaces;                    
 
    };
 
}
 
#endif