MTDDetLayerGeometry.h

Go to the documentation of this file.

#ifndef RecoMTD_DetLayers_MTDDetLayerGeometry_h
#define RecoMTD_DetLayers_MTDDetLayerGeometry_h

#include "DataFormats/DetId/interface/DetId.h"
#include "TrackingTools/DetLayers/interface/DetLayerGeometry.h"
#include "Geometry/MTDGeometryBuilder/interface/MTDGeometry.h"
#include "Geometry/MTDNumberingBuilder/interface/MTDTopology.h"

#include <vector>
#include <map>

class DetLayer;

class MTDDetLayerGeometry : public DetLayerGeometry {
public:
  MTDDetLayerGeometry();

  friend class MTDDetLayerGeometryESProducer;

  ~MTDDetLayerGeometry() override;

  const std::vector<const DetLayer*>& allBarrelLayers() const;

  const std::vector<const DetLayer*>& allEndcapLayers() const;

  const std::vector<const DetLayer*>& allForwardLayers() const;

  const std::vector<const DetLayer*>& allBackwardLayers() const;

  const std::vector<const DetLayer*>& allBTLLayers() const;

  const std::vector<const DetLayer*>& allETLLayers() const;

  const std::vector<const DetLayer*>& allLayers() const;

  const DetLayer* idToLayer(const DetId& detId) const override;

  // Added to allow building the MTDDetLayer from external plugins
  void buildLayers(const MTDGeometry* geo, const MTDTopology* mtopo);

  void sortLayers();

private:
  void addETLLayers(const std::pair<std::vector<DetLayer*>, std::vector<DetLayer*> >& etllayers);

  //. Add BTL layers; dtlayers is ASSUMED to be sorted inside-out
  void addBTLLayers(const std::vector<DetLayer*>& btllayers);

  DetId makeDetLayerId(const DetLayer* detLayer) const;

  std::vector<const DetLayer*> etlLayers_fw;
  std::vector<const DetLayer*> etlLayers_bk;
  std::vector<const DetLayer*> etlLayers_all;

  std::vector<const DetLayer*> btlLayers;
  std::vector<const DetLayer*> allForward;
  std::vector<const DetLayer*> allBackward;
  std::vector<const DetLayer*> allEndcap;
  std::vector<const DetLayer*> allBarrel;
  std::vector<const DetLayer*> allDetLayers;

  std::map<DetId, const DetLayer*> detLayersMap;
};
#endif