DetHitAccess.h

Go to the documentation of this file.

#ifndef RoadSearch_DetHitAccess_h
#define RoadSearch_DetHitAccess_h

#include <string>
#include <vector>
#include <algorithm>

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "DataFormats/SiStripDetId/interface/TECDetId.h"

class DetHitAccess {

 public:

  enum accessMode {standard,rphi,rphi_stereo};

  DetHitAccess();

  ~DetHitAccess();

  DetHitAccess(const SiStripRecHit2DCollection* rphiRecHits,
           const SiStripRecHit2DCollection* stereoRecHits,
           const SiStripMatchedRecHit2DCollection* matchedRecHits,
           const SiPixelRecHitCollection* pixelRecHits);
   
  void setCollections(const SiStripRecHit2DCollection* rphiRecHits,
              const SiStripRecHit2DCollection* stereoRecHits,
              const SiStripMatchedRecHit2DCollection* matchedRecHits,
              const SiPixelRecHitCollection* pixelRecHits);

  std::vector<TrackingRecHit*> getHitVector(const DetId* detid);

  inline void setMode(accessMode input) { accessMode_ = input; }
  inline void use_rphiRecHits(bool input) {use_rphiRecHits_ = input;}
  inline void use_stereoRecHits(bool input) {use_stereoRecHits_ = input;}

 private:

  accessMode accessMode_;

  bool use_rphiRecHits_;
  bool use_stereoRecHits_;

  const SiStripRecHit2DCollection* rphiHits_;
  const SiStripRecHit2DCollection* stereoHits_;
  const SiStripMatchedRecHit2DCollection * matchedHits_;
  const SiPixelRecHitCollection *pixelHits_;

};

#endif