TrackingVolumeDisplayer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 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:

        TrackingVolumeDisplayer(const std::string&,const std::string&,const IInterface*);
        virtual ~TrackingVolumeDisplayer();

        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;                      
 
        mutable std::ofstream               m_fileVolumeOutput;                   
        mutable std::string                 m_fileVolumeOutputName;               
        mutable bool                        m_fileVolumeOutputMode;               
 
        mutable std::ofstream               m_fileLayerOutput;                    
        mutable std::string                 m_fileLayerOutputName;                
        mutable bool                        m_fileLayerOutputMode;                
 
        mutable std::ofstream               m_fileSurfaceOutput;                  
        mutable std::string                 m_fileSurfaceOutputName;              
        mutable bool                        m_fileSurfaceOutputMode;              
        mutable bool                        m_fileSurfaceOutputSplit;             
 
        static int                          s_displaySurfaces;                    
 
    };
 
}
 
#endif