MuonForwardNavigableLayer.h

Go to the documentation of this file.

#ifndef Navigation_MuonForwardNavigableLayer_H
#define Navigation_MuonForwardNavigableLayer_H

/* Collaborating Class Declarations */
#include "RecoMuon/Navigation/interface/MuonDetLayerMap.h"
#include "RecoMuon/Navigation/interface/MuonEtaRange.h"

class DetLayer;
class ForwardDetLayer;

/* Base Class Headers */
#include "RecoMuon/Navigation/interface/MuonNavigableLayer.h"

/* C++ Headers */

/* ====================================================================== */

/* Class MuonForwardNavigableLayer Interface */

class MuonForwardNavigableLayer : public MuonNavigableLayer {
public:
  MuonForwardNavigableLayer(const ForwardDetLayer* fdl,
                            const MapB& innerBarrel,
                            const MapE& outerEndcap,
                            const MapE& innerEndcap,
                            const MapB& allInnerBarrel,
                            const MapE& allOuterEndcap,
                            const MapE& allInnerEndcap)
      : theDetLayer(fdl),
        theInnerBarrelLayers(innerBarrel),
        theOuterEndcapLayers(outerEndcap),
        theInnerEndcapLayers(innerEndcap),
        theAllInnerBarrelLayers(allInnerBarrel),
        theAllOuterEndcapLayers(allOuterEndcap),
        theAllInnerEndcapLayers(allInnerEndcap) {}

  MuonForwardNavigableLayer(const ForwardDetLayer* fdl, const MapE& outerEndcap)
      : theDetLayer(fdl), theOuterEndcapLayers(outerEndcap) {}
  MuonForwardNavigableLayer(const ForwardDetLayer* fdl, const MapE& outerEndcap, const MapE& allOuterEndcap)
      : theDetLayer(fdl), theOuterEndcapLayers(outerEndcap), theAllOuterEndcapLayers(allOuterEndcap) {}

  std::vector<const DetLayer*> nextLayers(NavigationDirection dir) const override;

  std::vector<const DetLayer*> nextLayers(const FreeTrajectoryState& fts, PropagationDirection dir) const override;

  std::vector<const DetLayer*> compatibleLayers(NavigationDirection dir) const override;

  std::vector<const DetLayer*> compatibleLayers(const FreeTrajectoryState& fts,
                                                PropagationDirection dir) const override;
  const DetLayer* detLayer() const override;

  void setDetLayer(const DetLayer*) override;

  MapE getOuterEndcapLayers() const { return theOuterEndcapLayers; }
  MapE getInnerEndcapLayers() const { return theInnerEndcapLayers; }
  MapB getInnerBarrelLayers() const { return theInnerBarrelLayers; }

  MapE getAllOuterEndcapLayers() const { return theAllOuterEndcapLayers; }
  MapE getAllInnerEndcapLayers() const { return theAllInnerEndcapLayers; }
  MapB getAllInnerBarrelLayers() const { return theAllInnerBarrelLayers; }

  void setInwardLinks(const MapB&, const MapE&);
  void setInwardCompatibleLinks(const MapB&, const MapE&);

private:
  void pushResult(std::vector<const DetLayer*>& result, const MapB& map) const;

  void pushResult(std::vector<const DetLayer*>& result, const MapE& map) const;

  void pushResult(std::vector<const DetLayer*>& result, const MapB& map, const FreeTrajectoryState& fts) const;

  void pushResult(std::vector<const DetLayer*>& result, const MapE& map, const FreeTrajectoryState& fts) const;

  void pushCompatibleResult(std::vector<const DetLayer*>& result,
                            const MapB& map,
                            const FreeTrajectoryState& fts) const;

  void pushCompatibleResult(std::vector<const DetLayer*>& result,
                            const MapE& map,
                            const FreeTrajectoryState& fts) const;

private:
  const ForwardDetLayer* theDetLayer;
  MapB theInnerBarrelLayers;
  MapE theOuterEndcapLayers;
  MapE theInnerEndcapLayers;
  MapB theAllInnerBarrelLayers;
  MapE theAllOuterEndcapLayers;
  MapE theAllInnerEndcapLayers;
};
#endif