MuonRegionHough.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONREGIONHOUGH_H
#define MUONREGIONHOUGH_H
 
#include <cmath>
#include <iostream>
#include <vector>
 
#include "MuonLayerHough/MuonLayerHough.h"
#include "MuonLayerHough/MuonPhiLayerHough.h"
#include "MuonStationIndex/MuonStationIndex.h"
 
namespace MuonHough {
 
    class MuonDetectorDescription {
    public:
        
        using DetRegIdx = Muon::MuonStationIndex::DetectorRegionIndex;
        using LayIdx = Muon::MuonStationIndex::LayerIndex;
        MuonDetectorDescription();
        
 
        RegionDescriptor getDescriptor(int sector, DetRegIdx region, LayIdx layer) const;
 
    private:
        void initDefaultRegions();
 
        RegionDescriptionVec m_regionDescriptions;  
    };
 
    class MuonSectorHough {
    public:
        MuonSectorHough(int sector, const MuonDetectorDescription& regionDescriptions);
 
        ~MuonSectorHough();
        
        using DetRegIdx = Muon::MuonStationIndex::DetectorRegionIndex;
        using LayIdx = Muon::MuonStationIndex::LayerIndex;
        MuonLayerHough& hough(DetRegIdx region, LayIdx layer) {
            int index = Muon::MuonStationIndex::sectorLayerHash(region, layer);
            return *m_transforms[index];
        }
 
        void reset();
 
    private:
        std::vector<std::unique_ptr<MuonLayerHough>> m_transforms;  
                                                    // int m_sector;                                /// sector number
    };
 
    class MuonDetectorHough {
    public:
        using DetRegIdx = Muon::MuonStationIndex::DetectorRegionIndex;
        using LayIdx = Muon::MuonStationIndex::LayerIndex;
        MuonPhiLayerHough& phiHough(DetRegIdx region) {
             return *m_phiTransforms[static_cast<int>(region)];
        }
 
        MuonLayerHough& hough(int sector, DetRegIdx region, LayIdx layer) {
            return m_sectors[sector - 1]->hough(region, layer);
        }
 
        void reset();
 
        MuonDetectorHough();
 
        MuonDetectorHough(const RegionDescriptionVec& regionDescriptors);
 
        ~MuonDetectorHough();
 
    private:
        void init();
 
        std::vector<std::unique_ptr<MuonSectorHough>> m_sectors;          
        std::vector<std::unique_ptr<MuonPhiLayerHough>> m_phiTransforms;  
    };
 
    
}  // namespace MuonHough
#endif