NavigableLayer.h

Go to the documentation of this file.

#ifndef DetLayers_NavigableLayer_h
#define DetLayers_NavigableLayer_h

#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"
#include "TrackingTools/DetLayers/interface/NavigationDirection.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <vector>

class DetLayer;
class FreeTrajectoryState;
class NavigationSchool;

class NavigableLayer {
public:
  virtual ~NavigableLayer() {}

  virtual std::vector<const DetLayer*> nextLayers(NavigationDirection direction) const = 0;

  virtual std::vector<const DetLayer*> nextLayers(const FreeTrajectoryState& fts,
                                                  PropagationDirection timeDirection) const = 0;

  virtual std::vector<const DetLayer*> compatibleLayers(NavigationDirection direction) const = 0;

  virtual std::vector<const DetLayer*> compatibleLayers(const FreeTrajectoryState& fts,
                                                        PropagationDirection timeDirection) const {
    int counter = 0;
    return compatibleLayers(fts, timeDirection, counter);
  };

  virtual std::vector<const DetLayer*> compatibleLayers(const FreeTrajectoryState& fts,
                                                        PropagationDirection timeDirection,
                                                        int& counter) const {
    edm::LogWarning("DetLayers") << "compatibleLayers(fts,dir,counter) not implemented. returning empty vector";
    return std::vector<const DetLayer*>();
  }

  virtual DetLayer const* detLayer() const = 0;
  virtual void setDetLayer(DetLayer const* dl) = 0;

  void setSchool(NavigationSchool const* sh) { school = sh; }

protected:
  NavigationSchool const* school = nullptr;
};

#endif