MuonBlueprintNodeBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSMUONDETECTOR_MUONBLUEPRINTNODEBUILDER_H
#define ACTSMUONDETECTOR_MUONBLUEPRINTNODEBUILDER_H
 
 
 
#include <GaudiKernel/MsgStream.h>
#include <AthenaBaseComps/AthMessaging.h>
 
#include <MuonReadoutGeometryR4/MuonDetectorManager.h>
#include "MuonStationIndex/MuonStationIndex.h"
#include "ActsGeometryInterfaces/IBlueprintNodeBuilder.h"
 
#include "Acts/Utilities/BoundFactory.hpp"
#include "Acts/Surfaces/PlanarBounds.hpp"
#include "Acts/Surfaces/Surface.hpp"
 
 
 
 
namespace MuonGMR4 {
  class Chamber;
}
 
namespace Acts {
  class GeometryContext;
  namespace Experimental {
    class StaticBlueprintNode;
    
}
 
}  // namespace Acts
 
 
namespace ActsTrk {
 
  using staticNodePtr = std::shared_ptr<Acts::Experimental::StaticBlueprintNode>;
  using surfacePtr = std::shared_ptr<Acts::Surface>;
  using MuonChamberSet = MuonGMR4::MuonDetectorManager::MuonChamberSet;
  using MuonSectorSet = MuonGMR4::MuonDetectorManager::MuonSectorSet;
  using StIdx = Muon::MuonStationIndex::StIndex;

class MuonBlueprintNodeBuilder : public extends<AthAlgTool, IBlueprintNodeBuilder> {
 
public:
 
 
  enum class EndcapSide {
      A,
      C,
      Both
  };
 
  StatusCode initialize() override;
  using base_class::base_class;
  
  std::shared_ptr<Acts::Experimental::BlueprintNode> buildBlueprintNode(const Acts::GeometryContext& gctx,
                                              std::shared_ptr<Acts::Experimental::BlueprintNode>&& childNode) override;
                                
 
private:
 
  const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};
 
  Gaudi::Property<bool> m_dumpVolumes{this, "dumpVolumes", false}; // Flag to control if we want to visualize each chamber volume individually
 
  Gaudi::Property<bool> m_useSectors{this, "UseSectors", false}; // Flag to control if we want to build the muon node from sectors or chambers

  template<typename T>
  std::shared_ptr<Acts::Surface> blendMaterial(const T& element) const;

   template<typename T>
   std::pair<std::vector<staticNodePtr>, std::vector<surfacePtr>> getSensitiveElements(const ActsTrk::GeometryContext& gctx,
                                                                                  const T& element,
                                                                                  const Acts::GeometryIdentifier& chId,
                                                                                  Acts::VolumeBoundFactory& boundsFactory) const;

  template<typename T>
  bool isElementInTheStation(const T& element, const std::vector<StIdx>& stationNames, const EndcapSide& side) const;

  template<typename MuonElementsSet>
  std::shared_ptr<Acts::Experimental::StaticBlueprintNode> buildMuonNode(const Acts::GeometryContext& gctx,
    const MuonElementsSet& elements,
    const std::string& name,
    const Acts::GeometryIdentifier& id,
    Acts::VolumeBoundFactory& boundsFactory) const;
};
 
} //namespace ActsTrk
 
#endif