MuonTkNavigationSchool.h

Go to the documentation of this file.

#ifndef RecoMuon_GlobalTrackingTools_MuonTkNavigationSchool_H
#define RecoMuon_GlobalTrackingTools_MuonTkNavigationSchool_H

#include "TrackingTools/DetLayers/interface/NavigationSchool.h"
#include "RecoMuon/Navigation/interface/MuonDetLayerMap.h"
#include <vector>

class BarrelDetLayer;
class ForwardDetLayer;
class NavigableLayer;
class SimpleBarrelNavigableLayer;
class SimpleForwardNavigableLayer;
class MuonBarrelNavigableLayer;
class MuonForwardNavigableLayer;
class GeometricSearchTracker;
class MuonDetLayerGeometry;
class MagneticField;

//              ---------------------
//              -- Class Interface --
//              ---------------------

class MuonTkNavigationSchool : public NavigationSchool {

  public:

    MuonTkNavigationSchool(const MuonDetLayerGeometry*, 
                           const GeometricSearchTracker*, 
                           const MagneticField*);

    ~MuonTkNavigationSchool();

    virtual std::vector<NavigableLayer*> navigableLayers() const;

  private:

    void addBarrelLayer(BarrelDetLayer*);

    void addEndcapLayer(ForwardDetLayer*);

    void linkBarrelLayers();

    void linkEndcapLayers(const MapE&, 
                          std::vector<MuonForwardNavigableLayer*>&,
                          std::vector<SimpleForwardNavigableLayer*>&);

    float barrelLength() const;

    float calculateEta(float r, float z) const;

  private:

    struct delete_layer {
      template <typename T>
      void operator()(T*& p) {
        if (p) {
          delete p;
          p = 0;
        }
      }
    };

    typedef std::vector<BarrelDetLayer*>   BDLC;
    typedef std::vector<ForwardDetLayer*>  FDLC;

    MapB theBarrelLayers;
    MapE theForwardLayers;   // +z
    MapE theBackwardLayers;  // -z

    std::vector<SimpleBarrelNavigableLayer*>  theTkBarrelNLC;
    std::vector<SimpleForwardNavigableLayer*> theTkForwardNLC;
    std::vector<SimpleForwardNavigableLayer*> theTkBackwardNLC;

    std::vector<MuonBarrelNavigableLayer*>  theMuonBarrelNLC;
    std::vector<MuonForwardNavigableLayer*> theMuonForwardNLC;
    std::vector<MuonForwardNavigableLayer*> theMuonBackwardNLC;

    const MuonDetLayerGeometry* theMuonDetLayerGeometry;
    const GeometricSearchTracker* theGeometricSearchTracker;
    const MagneticField* theMagneticField;

};

#endif