MTDNavigationSchool.h

Go to the documentation of this file.

#ifndef Navigation_MTDNavigationSchool_H
#define Navigation_MTDNavigationSchool_H

#include "RecoMTD/Navigation/interface/MTDDetLayerMap.h"
#include "TrackingTools/DetLayers/interface/NavigationSchool.h"
#include "RecoMTD/DetLayers/interface/MTDDetLayerGeometry.h"
#include <vector>
#include <map>

class BTLNavigableLayer;
class ETLNavigableLayer;
class MTDEtaRange;
class BarrelDetLayer;
class ForwardDetLayer;

class MTDNavigationSchool : public NavigationSchool {
public:
  MTDNavigationSchool(const MTDDetLayerGeometry*, bool enableBTL = true, bool enableETL = true);
  ~MTDNavigationSchool() override;
  StateType navigableLayers() override;

private:
  void addBarrelLayer(const BarrelDetLayer*);
  void addEndcapLayer(const ForwardDetLayer*);
  void linkBarrelLayers();
  void linkEndcapLayers(const MapE&, std::vector<ETLNavigableLayer*>&);
  void createInverseLinks();
  float calculateEta(const float&, const float&) const;

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

  MapB theBarrelLayers;    
  MapE theForwardLayers;   
  MapE theBackwardLayers;  

  std::vector<BTLNavigableLayer*> theBarrelNLC;
  std::vector<ETLNavigableLayer*> theForwardNLC;
  std::vector<ETLNavigableLayer*> theBackwardNLC;

  const MTDDetLayerGeometry* theMTDDetLayerGeometry;
};
#endif