TrapezoidBoundsExpander.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSMUONDETECTOR_TRAPEZOIDBOUNDSEXPANDER_H
#define ACTSMUONDETECTOR_TRAPEZOIDBOUNDSEXPANDER_H
#ifndef SIMULATIONBASE
 
#include "MuonReadoutGeometryR4/BoundsExpander.h"
 
#include <iostream>
 
namespace MuonGMR4 {
    class TrapezoidBoundsExpander : public BoundsExpander {
        public:
            struct TrapezoidPatch{
                explicit TrapezoidPatch(const Acts::Transform3& trf,
                                        std::shared_ptr<const Acts::VolumeBounds> volBounds);
                void print(std::ostream& ostr) const;
                inline friend std::ostream& operator<<(std::ostream& ostr, const TrapezoidPatch& patch) {
                    patch.print(ostr);
                    return ostr;
                }

                double area() const;
                Acts::Transform3 toCenter{Acts::Transform3::Identity()};
                std::shared_ptr<const Acts::VolumeBounds> bounds{}; 
            };

            enum class Side : std::int8_t{left = -1, right = 1};
            enum class Level : std::int8_t { bottom= -1, top = 1 };
            friend std::ostream& operator<<(std::ostream& ostr, const Side side){
                switch (side) {
                    case Side::left: return ostr<<"left";
                    case Side::right: return ostr<<"right";
                }
                return ostr;
            }

            friend std::ostream& operator<<(std::ostream& ostr, const Level lvl) {
                switch (lvl) {
                    case Level::bottom: return ostr<<"bottom";
                    case Level::top: return ostr<<"top";
                }
                return ostr;
            }

            static Amg::Vector3D makeVertex(const TrapezoidPatch& trapezoid,
                                            const Side side,
                                            const Level lvl);
            static double distanceToTrapezoid(const Amg::Vector3D& vertex,
                                              const Amg::Vector3D& edge,
                                              const Amg::Vector3D& testMe,
                                              const Side side);
            static std::uint8_t toIndex(const Side side, const Level lvl);

            using BoundsExpander::BoundsExpander;
            using BoundsExpander::expand;
            using BoundsExpander::makeBounds;
            virtual void expand(const Acts::GeometryContext& tgContext,
                                 const Acts::Volume& volume) override final;

            virtual std::shared_ptr<Acts::VolumeBounds> 
                    makeBounds(Acts::VolumeBoundFactory& factory) override final;      
            void setMargin(const double margin);
            void setMargin(const double marginXlow, const double marginXhigh,
                           const double marginY, const double marginZ);
        private:
            void findTrapezoidEdge(Amg::Vector3D& botCorner,
                                   Amg::Vector3D& topCorner,
                                   const Side side,
                                   const std::size_t lowPatch,
                                   const std::size_t highPatch) const;  
            std::vector<TrapezoidPatch> m_constituents{};
            std::shared_ptr<Acts::VolumeBounds> m_finalBounds{};
            std::array<double, 4> m_margin{Acts::filledArray<double, 4>(0.)};
    };
}
#endif
#endif