Acts/Geometry/TrapezoidPortalShell.hpp

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 #pragma once
0010
0011 #include "Acts/Definitions/Algebra.hpp"
0012 #include "Acts/Geometry/PortalShell.hpp"
0013 #include "Acts/Geometry/TrackingVolume.hpp"
0014 #include "Acts/Geometry/TrapezoidVolumeBounds.hpp"
0015 #include "Acts/Utilities/AxisDefinitions.hpp"
0016 #include "Acts/Utilities/Logger.hpp"
0017
0018 namespace Acts {
0019
0020 /// @class TrapezoidPortalShell
0021 /// Base class for trapezoid shaped portal shells, e.g. shells for trapezoid
0022 /// volumes
0023 class TrapezoidPortalShell : public PortalShellBase {
0024  public:
0025   using Face = TrapezoidVolumeBounds::Face;
0026
0027   using enum TrapezoidVolumeBounds::Face;
0028
0029   /// Retrieve the portal associated to the given face. Can be nullptr if unset.
0030   /// @param face The face to retrieve the portal for
0031   /// @return The portal associated to the face
0032   virtual Portal* portal(Face face) = 0;
0033
0034   /// Retrieve a shared_ptr for the portal associated to the given face. Can be
0035   /// nullptr if unset.
0036   /// @param face The face to retrieve the portal for
0037   /// @return The portal associated to the face
0038   virtual std::shared_ptr<Portal> portalPtr(Face face) = 0;
0039
0040   /// Set the portal associated to the given face.
0041   /// @param portal The portal to set
0042   /// @param face The face to set the portal
0043   virtual void setPortal(std::shared_ptr<Portal> portal, Face face) = 0;
0044
0045   /// @copydoc PortalShellBase::fill
0046   void fill(TrackingVolume& volume) override;
0047
0048   virtual const Transform3& transform() const = 0;
0049 };
0050
0051 /// Output stream operator for the TrapezoidPortalShell::Face enum
0052 /// @param os The output stream
0053 /// @param face The face to output
0054 /// @return The output stream
0055 std::ostream& operator<<(std::ostream& os, TrapezoidPortalShell::Face face);
0056
0057 /// @class SingleTrapezoidPortalShell
0058 /// This class describes a trapezoid shell containing a single volume.
0059 class SingleTrapezoidPortalShell : public TrapezoidPortalShell {
0060  public:
0061   // constructs a single trapezoid shell for the given tracking volume
0062   // @param volume The volume to create the shell for
0063   explicit SingleTrapezoidPortalShell(TrackingVolume& volume);
0064
0065   /// @copydoc PortalShellBase::size
0066   std::size_t size() const override;
0067
0068   /// @copydoc TrapezoidPortalShell::portal
0069   Portal* portal(Face face) override;
0070
0071   /// @copydoc TrapezoidPortalShell::portalPtr
0072   std::shared_ptr<Portal> portalPtr(Face face) override;
0073
0074   /// @copydoc TrapezoidPortalShell::setPortal
0075   void setPortal(std::shared_ptr<Portal> portal, Face face) override;
0076
0077   /// @copydoc PortalShellBase::applyToVolume
0078   void applyToVolume() override;
0079
0080   /// @copydoc PortalShellBase::isValid
0081   bool isValid() const override;
0082
0083   /// @copydoc PortalShellBase::label
0084   std::string label() const override;
0085
0086   const Transform3& transform() const override {
0087     return m_volume->transform();
0088   };
0089
0090  private:
0091   std::array<std::shared_ptr<Portal>, 6> m_portals{};
0092
0093   TrackingVolume* m_volume;
0094 };
0095
0096 }  // namespace Acts