MicroGMTIsolationUnit.h

Go to the documentation of this file.

#ifndef __l1microgmtisolationunit_h
#define __l1microgmtisolationunit_h

#include "MicroGMTConfiguration.h"
#include "MicroGMTExtrapolationLUT.h"
#include "MicroGMTRelativeIsolationCheckLUT.h"
#include "MicroGMTAbsoluteIsolationCheckLUT.h"
#include "MicroGMTCaloIndexSelectionLUT.h"

#include "L1Trigger/L1TMuon/interface/L1TMuonGlobalParamsHelper.h"
#include "L1Trigger/L1TMuon/interface/MicroGMTLUTFactories.h"

namespace l1t {
  class MicroGMTIsolationUnit {
    public:
      MicroGMTIsolationUnit ();
      virtual ~MicroGMTIsolationUnit ();

      void initialise(L1TMuonGlobalParamsHelper*);
      // returns the index corresponding to the calo tower sum using the LUT
      int getCaloIndex(MicroGMTConfiguration::InterMuon&) const;
      // copies the energy values to the m_towerEnergies map for consistent access
      void setTowerSums(const MicroGMTConfiguration::CaloInputCollection& inputs, int bx);
      // First step done for calo input preparation, calculates strip sums
      void calculate5by1Sums(const MicroGMTConfiguration::CaloInputCollection&, int bx);
      // Second step, only done for the sums needed for final iso requirement
      int calculate5by5Sum(unsigned index) const;

      // Checks with LUT isolation for all muons in list, assuming input calo is non-summed
      void isolate(MicroGMTConfiguration::InterMuonList&) const;
      // Checks with LUT isolation for all muons in list, assuming input calo is pre-summed
      void isolatePreSummed(MicroGMTConfiguration::InterMuonList& muons) const;
      // Uses *Extrapolation LUTs to project trajectory to the vertex and adds info to muon
      void extrapolateMuons(MicroGMTConfiguration::InterMuonList&) const;

    private:
      int m_fwVersion;

      std::shared_ptr<MicroGMTExtrapolationLUT> m_BEtaExtrapolation;
      std::shared_ptr<MicroGMTExtrapolationLUT> m_BPhiExtrapolation;
      std::shared_ptr<MicroGMTExtrapolationLUT> m_OEtaExtrapolation;
      std::shared_ptr<MicroGMTExtrapolationLUT> m_OPhiExtrapolation;
      std::shared_ptr<MicroGMTExtrapolationLUT> m_FEtaExtrapolation;
      std::shared_ptr<MicroGMTExtrapolationLUT> m_FPhiExtrapolation;

      std::map<tftype, std::shared_ptr<MicroGMTExtrapolationLUT>> m_phiExtrapolationLUTs;
      std::map<tftype, std::shared_ptr<MicroGMTExtrapolationLUT>> m_etaExtrapolationLUTs;

      std::shared_ptr<MicroGMTCaloIndexSelectionLUT> m_IdxSelMemEta;
      std::shared_ptr<MicroGMTCaloIndexSelectionLUT> m_IdxSelMemPhi;

      std::shared_ptr<MicroGMTRelativeIsolationCheckLUT> m_RelIsoCheckMem;
      std::shared_ptr<MicroGMTAbsoluteIsolationCheckLUT> m_AbsIsoCheckMem;

      std::bitset<28> m_caloInputsToDisable;
      std::bitset<28> m_maskedCaloInputs;

      std::vector<int> m_5by1TowerSums;
      std::map<int, int> m_towerEnergies;
      bool m_initialSums;
  };
}

#endif /* defined(__l1microgmtisolationunit_h) */