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 <format>
 
#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) {
            using namespace Muon::MuonStationIndex;
            int index = sectorLayerHash(region, layer);
            return *m_transforms.at(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.at(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