MuonServiceProxy.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackingTools_MuonServiceProxy_H
#define RecoMuon_TrackingTools_MuonServiceProxy_H

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include "FWCore/Utilities/interface/ESGetToken.h"

// EventSetup data types
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "RecoMuon/DetLayers/interface/MuonDetLayerGeometry.h"
#include "TrackingTools/GeomPropagators/interface/Propagator.h"

// EventSetup record types
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoMuon/Records/interface/MuonRecoGeometryRecord.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"

#include "RecoMuon/Navigation/interface/MuonNavigationSchool.h"

#include <map>
#include <string>

class MuonServiceProxy {
public:
  enum class UseEventSetupIn { Run, Event, RunAndEvent };

  MuonServiceProxy(const edm::ParameterSet&,
                   edm::ConsumesCollector&&,
                   UseEventSetupIn useEventSetupIn = UseEventSetupIn::Event);

  virtual ~MuonServiceProxy();

  // Operations

  void update(const edm::EventSetup& setup, bool duringEvent = true);

  edm::ESHandle<MagneticField> magneticField() const { return theMGField; }

  edm::ESHandle<GlobalTrackingGeometry> trackingGeometry() const { return theTrackingGeometry; }

  edm::ESHandle<MuonDetLayerGeometry> detLayerGeometry() const { return theDetLayerGeometry; }

  edm::ESHandle<Propagator> propagator(std::string propagatorName) const;

  const edm::EventSetup& eventSetup() const { return *theEventSetup; }

  bool isTrackingComponentsRecordChanged() const { return theChangeInTrackingComponentsRecord; }

  const MuonNavigationSchool* muonNavigationSchool() const { return theSchool; }

private:
  class PropagatorInfo {
  public:
    edm::ESHandle<Propagator> esHandle_;
    edm::ESGetToken<Propagator, TrackingComponentsRecord> eventToken_;
    edm::ESGetToken<Propagator, TrackingComponentsRecord> runToken_;
  };

  using PropagatorMap = std::map<std::string, PropagatorInfo>;

  edm::ESHandle<GlobalTrackingGeometry> theTrackingGeometry;
  edm::ESHandle<MagneticField> theMGField;
  edm::ESHandle<MuonDetLayerGeometry> theDetLayerGeometry;

  edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> globalTrackingGeometryEventToken_;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldEventToken_;
  edm::ESGetToken<MuonDetLayerGeometry, MuonRecoGeometryRecord> muonDetLayerGeometryEventToken_;

  edm::ESGetToken<GlobalTrackingGeometry, GlobalTrackingGeometryRecord> globalTrackingGeometryRunToken_;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldRunToken_;
  edm::ESGetToken<MuonDetLayerGeometry, MuonRecoGeometryRecord> muonDetLayerGeometryRunToken_;

  const edm::EventSetup* theEventSetup;
  bool theMuonNavigationFlag;
  bool theRPCLayer;
  bool theCSCLayer;
  bool theGEMLayer;
  bool theME0Layer;
  const MuonNavigationSchool* theSchool;

  PropagatorMap thePropagators;

  unsigned long long theCacheId_GTG;
  unsigned long long theCacheId_MG;
  unsigned long long theCacheId_DG;
  unsigned long long theCacheId_P;

  bool theChangeInTrackingComponentsRecord;
};
#endif