SurfaceHelper.cpp

#include "DDRec/SurfaceHelper.h"

#include "DDRec/DetectorSurfaces.h"
#include "DD4hep/Detector.h"
#include "DD4hep/LCDD.h"
#include "DD4hep/VolumeManager.h"

namespace DD4hep {
  
  using namespace Geometry ;

  namespace DDRec {
    

    SurfaceHelper::SurfaceHelper(DD4hep::Geometry::DetElement const& e) : _det(e) {

      initialize() ;
    }
    
    SurfaceHelper::~SurfaceHelper(){
      // nothing to do
    }
    
    
    void SurfaceHelper::initialize() {
      
      // have to populate the volume manager once in order to have 
      // the volumeIDs attached to the DetElements
      LCDD& lcdd = LCDD::getInstance();
      /* VolumeManager volMgr = */ VolumeManager::getVolumeManager(lcdd);

      //------------------ breadth first tree traversal ---------
      std::list< DetElement > dets ;
      std::list< DetElement > daugs ; 
      std::list< DetElement > gdaugs ; 
      
      daugs.push_back( _det ) ;
      
      while( ! daugs.empty() ) {
	
        for( std::list< DetElement >::iterator li=daugs.begin() ; li != daugs.end() ; ++li ){
          DetElement dau = *li ;
          DetElement::Children chMap = dau.children() ;
          for ( DetElement::Children::const_iterator it=chMap.begin() ; it != chMap.end() ; ++it ){
            DetElement de = (*it).second ;
            gdaugs.push_back( de ) ;
          }  
        }
        dets.splice( dets.end() , daugs ) ;
        daugs.splice( daugs.end() , gdaugs ) ;
      }
      //------------------ end tree traversal ---------
      
      //      std::cout << " **** SurfaceHelper::initialize() : # DetElements found  " << dets.size() << std::endl ;
      
      for( std::list< DetElement >::iterator li=dets.begin() ; li != dets.end() ; ++li ) {
	
        DetElement det = (*li) ;
	


        // create surfaces
        DetectorSurfaces ds( det ) ;
	
        const SurfaceList& detSL = ds.surfaceList() ;


        // // ---------------------- debug printout
        // std::cout << " ---- DetElement id: " << det.volumeID() << " name : " << det.name() << "  #surfaces : " << detSL.size() << std::endl ; 
        // PlacedVolume pv = det.placement() ;
        // if( pv.isValid() ) {
        //   try{ // needed as above is also true for world whcih has invalid placment ...
        //     PlacedVolume::VolIDs volIDs = pv.volIDs() ;
        //     for(unsigned i=0,n=volIDs.size(); i<n ; ++i){
        //       std::cout << "        " << volIDs[i].first << " : " << volIDs[i].second << std::endl ;
        //     }
        //   }catch(...){}
        // }else{
        //   std::cout << "        invalid placement for DetElement ??? !! " << std::endl ;
        // }
        // // ------------------------- end debug printout


        // and add copy them to this list
        _sL.insert( _sL.end(), detSL.begin(), detSL.end() );
      }
      
    }
    


  } // namespace
}// namespace