MuonDetLayerMeasurements.h

Go to the documentation of this file.

#ifndef MeasurementDet_MuonDetLayerMeasurements_H
#define MeasurementDet_MuonDetLayerMeasurements_H

#include "FWCore/Framework/interface/Event.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/DetLayers/interface/MeasurementEstimator.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
//#include "TrackingTools/ementDet/interface/TrajectoryMeasurement.h"
#include "TrackingTools/MeasurementDet/interface/TrajectoryMeasurementGroup.h"
#include "DataFormats/DTRecHit/interface/DTRecSegment4DCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
#include "DataFormats/RPCRecHit/interface/RPCRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/ME0SegmentCollection.h"

#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"



#include <vector>

class DetLayer;
class GeomDet;
class TrajectoryMeasurement;


//FIXME: these typedefs MUST GO inside the scope of MuonDetLayerMeasurements
typedef std::vector<TrajectoryMeasurement>          MeasurementContainer;
typedef std::pair<const GeomDet*,TrajectoryStateOnSurface> DetWithState;


class MuonDetLayerMeasurements {
 public:
  typedef MuonTransientTrackingRecHit::MuonRecHitContainer MuonRecHitContainer;

  MuonDetLayerMeasurements(edm::InputTag dtlabel,
               edm::InputTag csclabel,
               edm::InputTag rpclabel,
               edm::InputTag gemlabel,
               edm::InputTag me0label,
               edm::ConsumesCollector& iC,
               bool enableDT = true,
               bool enableCSC = true,
               bool enableRPC = true,
               bool enableGEM = true,
               bool enableME0 = true);
  
  virtual ~MuonDetLayerMeasurements();
  
  // for a given det and state.  Not clear when the fastMeasurements below
  //  should be used, since it isn't passed a GeomDet
  MeasurementContainer
    measurements( const DetLayer* layer,
                  const GeomDet * det,
                  const TrajectoryStateOnSurface& stateOnDet,
                  const MeasurementEstimator& est,
                  const edm::Event& iEvent);

  MeasurementContainer
    measurements( const DetLayer* layer,
          const TrajectoryStateOnSurface& startingState,
          const Propagator& prop,
          const MeasurementEstimator& est,
          const edm::Event& iEvent);

  MeasurementContainer
    fastMeasurements( const DetLayer* layer,
              const TrajectoryStateOnSurface& theStateOnDet,
              const TrajectoryStateOnSurface& startingState,
              const Propagator& prop,
              const MeasurementEstimator& est,
              const edm::Event& iEvent);

  MeasurementContainer
    measurements( const DetLayer* layer,
          const TrajectoryStateOnSurface& startingState,
          const Propagator& prop,
          const MeasurementEstimator& est);

  MeasurementContainer
    fastMeasurements( const DetLayer* layer,
              const TrajectoryStateOnSurface& theStateOnDet,
              const TrajectoryStateOnSurface& startingState,
              const Propagator& prop,
              const MeasurementEstimator& est);

  std::vector<TrajectoryMeasurementGroup>
    groupedMeasurements( const DetLayer* layer,
                  const TrajectoryStateOnSurface& startingState,
                  const Propagator& prop,
                  const MeasurementEstimator& est,
                  const edm::Event& iEvent);

  std::vector<TrajectoryMeasurementGroup>
    groupedMeasurements( const DetLayer* layer,
                  const TrajectoryStateOnSurface& startingState,
                  const Propagator& prop,
                  const MeasurementEstimator& est);
 
  void setEvent(const edm::Event &);  

  MuonRecHitContainer recHits(const DetLayer* layer, const edm::Event& iEvent);

  MuonRecHitContainer recHits(const DetLayer* layer);


 private:

  MuonRecHitContainer recHits(const GeomDet*, const edm::Event& iEvent);

  void checkEvent() const;


  edm::EDGetTokenT<DTRecSegment4DCollection> dtToken_;
  edm::EDGetTokenT<CSCSegmentCollection> cscToken_;
  edm::EDGetTokenT<RPCRecHitCollection> rpcToken_;
  edm::EDGetTokenT<GEMRecHitCollection> gemToken_;
  edm::EDGetTokenT<ME0SegmentCollection> me0Token_;


  bool enableDTMeasurement;
  bool enableCSCMeasurement;
  bool enableRPCMeasurement;
  bool enableGEMMeasurement;
  bool enableME0Measurement;
  
  // caches that should get filled once per event
  edm::Handle<DTRecSegment4DCollection> theDTRecHits;
  edm::Handle<CSCSegmentCollection>     theCSCRecHits;
  edm::Handle<RPCRecHitCollection>      theRPCRecHits;
  edm::Handle<GEMRecHitCollection>      theGEMRecHits;
  edm::Handle<ME0SegmentCollection>      theME0RecHits;

  void checkDTRecHits();
  void checkCSCRecHits();
  void checkRPCRecHits();
  void checkGEMRecHits();
  void checkME0RecHits();

  // keeps track of which event the cache holds
  edm::Event::CacheIdentifier_t theDTEventCacheID;
  edm::Event::CacheIdentifier_t theCSCEventCacheID;
  edm::Event::CacheIdentifier_t theRPCEventCacheID;
  edm::Event::CacheIdentifier_t theGEMEventCacheID;
  edm::Event::CacheIdentifier_t theME0EventCacheID;

  const edm::Event* theEvent;   

};
#endif