FPGATrackSimSectorBank.h

Go to the documentation of this file.

// Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATrackSimSECTORBANK_H
#define FPGATrackSimSECTORBANK_H
 
#include <iostream>
#include <map>
#include <fstream>
#include <vector>
#include <memory>
 
#include "FPGATrackSimObjects/FPGATrackSimTypes.h"
 
class FPGATrackSimHit;
 
 
class FPGATrackSimSectorBank 
{
    public:
 
        FPGATrackSimSectorBank(std::string const & filepath);
 
        size_t getNLayers() const { return m_nLayers; }
        size_t getNSectors() const { return m_s2mMap.size(); }
 
        std::vector<module_t> const & getModules(sector_t sector) const { return m_s2mMap[sector]; }
        std::vector<sector_t> getSectors(unsigned layer, module_t module) const; // TODO rename find?
 
        // Returns the sector matching 'modules' exactly (WC must be matched with WC), or -1 if none.
        sector_t findSector(std::vector<module_t> const & modules) const;
 
        // Returns the sector matching 'hits' exactly (WC must be matched with WC), or -1 if none.
        // If multiple hits have different hashIDs, will prefer most popular, then random.
        sector_t findSector(std::vector<std::vector<std::shared_ptr<const FPGATrackSimHit>>> const & hits) const;
 
 
        // Helper functions to set and retrieve information about the sector q/pt binning.
    const std::vector<double>& getQOverPtBins() const { return m_qOverPtBins; }
        size_t getNQOverPtBins() const {return m_qOverPtBins.size(); }
        bool isAbsQOverPtBinning() const { return m_absQOverPtBinning; }
        void storeQOverPtBinning(const std::vector<double>& qOverPtBins, bool absBinning);
 
 
    private:
 
        size_t m_nLayers = 0UL;
        std::vector<std::vector<module_t>> m_s2mMap; // sector to module map. index by (sector, layer)
        std::vector<std::multimap<module_t, sector_t>> m_m2sMap; // module to sector map. index by (layer, module).
 
        // Store these as part of the sector bank.
        std::vector<double> m_qOverPtBins;
        bool m_absQOverPtBinning = false;
 
        // Helpers
        void readHeader(std::ifstream & fin);
        void readSectors(std::ifstream & fin);
        void makeInverseMap();
};
 
 
 
#endif // FPGATrackSimSECTORBANK_H