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) {}


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

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

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

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

    virtual 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