src/Geometry/TrapezoidPortalShell.cpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #include "Acts/Geometry/TrapezoidPortalShell.hpp"
0010
0011 #include "Acts/Geometry/BoundarySurfaceFace.hpp"
0012 #include "Acts/Geometry/Portal.hpp"
0013 #include "Acts/Geometry/PortalLinkBase.hpp"
0014 #include "Acts/Geometry/TrapezoidVolumeBounds.hpp"
0015
0016 namespace Acts {
0017
0018 void TrapezoidPortalShell::fill(TrackingVolume& volume) {
0019   for (Face face : {NegativeZFaceXY, PositiveZFaceXY, TrapezoidFaceAlpha,
0020                     TrapezoidFaceBeta, NegativeYFaceZX, PositiveYFaceZX}) {
0021     const auto& portalAtFace = portalPtr(face);
0022     if (portalAtFace != nullptr) {
0023       portalAtFace->fill(volume);
0024       volume.addPortal(portalAtFace);
0025     }
0026   }
0027 }
0028
0029 SingleTrapezoidPortalShell::SingleTrapezoidPortalShell(TrackingVolume& volume)
0030     : m_volume{&volume} {
0031   if (m_volume->volumeBounds().type() != VolumeBounds::BoundsType::eTrapezoid) {
0032     throw std::invalid_argument("Invalid volume bounds - not trapezoid");
0033   }
0034
0035   const auto& bounds =
0036       dynamic_cast<const TrapezoidVolumeBounds&>(m_volume->volumeBounds());
0037
0038   std::vector<OrientedSurface> orientedSurfaces =
0039       bounds.orientedSurfaces(m_volume->transform());
0040
0041   for (Face face : {NegativeZFaceXY, PositiveZFaceXY, TrapezoidFaceAlpha,
0042                     TrapezoidFaceBeta, NegativeYFaceZX, PositiveYFaceZX}) {
0043     const auto& portalSurface = orientedSurfaces.at(toUnderlying(face));
0044
0045     m_portals.at(toUnderlying(face)) = std::make_shared<Portal>(
0046         portalSurface.direction, portalSurface.surface, *m_volume);
0047   }
0048 }
0049
0050 Portal* SingleTrapezoidPortalShell::portal(Face face) {
0051   return portalPtr(face).get();
0052 }
0053
0054 std::shared_ptr<Portal> SingleTrapezoidPortalShell::portalPtr(Face face) {
0055   return m_portals.at(toUnderlying(face));
0056 }
0057
0058 void SingleTrapezoidPortalShell::setPortal(std::shared_ptr<Portal> portal,
0059                                            Face face) {
0060   assert(portal != nullptr);
0061   assert(portal->isValid());
0062   m_portals.at(toUnderlying(face)) = std::move(portal);
0063 }
0064
0065 std::size_t SingleTrapezoidPortalShell::size() const {
0066   std::size_t count = 0;
0067   std::ranges::for_each(
0068       m_portals, [&count](const auto& portal) { count += portal ? 1 : 0; });
0069   return count;
0070 }
0071
0072 void SingleTrapezoidPortalShell::applyToVolume() {
0073   for (std::size_t i = 0; i < m_portals.size(); i++) {
0074     const auto& portal = m_portals.at(i);
0075     if (portal != nullptr) {
0076       if (!portal->isValid()) {
0077         std::stringstream ss;
0078         ss << static_cast<Face>(i);
0079         throw std::runtime_error{"Invalid portal found in shell at " +
0080                                  ss.str()};
0081       }
0082       m_volume->addPortal(portal);
0083     }
0084   }
0085 }
0086
0087 bool SingleTrapezoidPortalShell::isValid() const {
0088   return std::ranges::all_of(m_portals, [](const auto& portal) {
0089     return portal == nullptr || portal->isValid();
0090   });
0091 }
0092
0093 std::string SingleTrapezoidPortalShell::label() const {
0094   std::stringstream ss;
0095   ss << "TrapezoidShell(vol=" << m_volume->volumeName() << ")";
0096   return ss.str();
0097 }
0098
0099 std::ostream& operator<<(std::ostream& os, TrapezoidPortalShell::Face face) {
0100   switch (face) {
0101     case TrapezoidPortalShell::NegativeZFaceXY:
0102       return os << "NegativeZFaceXY";
0103     case TrapezoidPortalShell::PositiveZFaceXY:
0104       return os << "PositiveZFaceXY";
0105     case TrapezoidPortalShell::TrapezoidFaceAlpha:
0106       return os << "TrapezoidFaceAlpha";
0107     case TrapezoidPortalShell::TrapezoidFaceBeta:
0108       return os << "TrapezoidFaceBeta";
0109     case TrapezoidPortalShell::NegativeYFaceZX:
0110       return os << "NegativeYFaceZX";
0111     case TrapezoidPortalShell::PositiveYFaceZX:
0112       return os << "PositiveYFaceZX";
0113     default:
0114       return os << "Invalid face";
0115   }
0116 }
0117
0118 }  // namespace Acts