MuonSpectrometer/MuonPhaseII/Event/MuonSpacePoint/MuonSpacePoint/SpacePointContainer.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONSPACEPOINT_MUONSPACEPOINTCONTAINER_H
#define MUONSPACEPOINT_MUONSPACEPOINTCONTAINER_H
 
#include "MuonSpacePoint/SpacePoint.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
#include "GaudiKernel/SystemOfUnits.h"
#include "AthContainers/DataVector.h"
#include <vector>
namespace MuonR4{
    class SpacePointBucket : 
        public std::vector<std::shared_ptr<SpacePoint>> {
         public:
            using chamberLocation = MuonGMR4::SpectrometerSector::chamberLocation;
            using std::vector<std::shared_ptr<SpacePoint>>::vector;
            void setCoveredRange(double min, double max){
                m_min = min;
                m_max = max;
            }
            double coveredMin() const { return m_min; }
            double coveredMax() const { return m_max; }
            const MuonGMR4::SpectrometerSector* msSector() const {
                return empty() ? nullptr : front()->msSector();
            }
            void setBucketId(unsigned int id) {
                m_bucketId = id;
            }
            unsigned int bucketId() const { return m_bucketId; }
            bool operator<(const SpacePointBucket& other) const {
                using ChamberSorter = MuonGMR4::MuonDetectorManager::MSEnvelopeSorter;
                static const ChamberSorter sorter{};
                int chambCompare = -sorter(msSector(), other.msSector()) + 
                                    sorter(other.msSector(), msSector());
                if (chambCompare) return chambCompare < 0;
                return bucketId() < other.bucketId();
            }
            void populateChamberLocations(){
                if (!msSector()){
                    std::cerr << "SpacePointContainer::populateChamberLocations can only be called once we have a valid hit"<<std::endl;
                    return; 
                }
                chamberLocation closestRight{1e8,1e8,1e8,1e8}; 
                // loop over all chambers in the sector
                for (auto & chamber : msSector()->chamberLocations()){
                    // truncate to the bucket volume 
                    double left = std::max(m_min, chamber.yLeft); 
                    double right = std::min(m_max, chamber.yRight);
                    // only keep one chamber outside the bucket - the right-hand side 
                    // neighbour (for shallow tracks)  
                    if (left > right){
                        if (chamber.yLeft - m_max < closestRight.yLeft - m_max){
                            closestRight = chamber; 
                        }
                    } 
                    // keep all chambers inside the bucket 
                    else{
                        m_chamberLocs.push_back(chamber);
                    }
                }
                // add the closest right hand side chamber, if there is one 
                if (closestRight.yLeft < 1e8) m_chamberLocs.push_back(closestRight); 
            }
            const std::vector<chamberLocation> & chamberLocations() const{
                return m_chamberLocs; 
            }
        private:
            unsigned int m_bucketId{0};
            double m_min{-20. *Gaudi::Units::m};
            double m_max{20. * Gaudi::Units::m};
            std::vector<chamberLocation> m_chamberLocs; 
    };
 
    using SpacePointContainer = DataVector<SpacePointBucket>;
}
#include "AthenaKernel/CLASS_DEF.h"
CLASS_DEF( MuonR4::SpacePointContainer , 1138818003 , 1 );
 
#endif