CosmicMuonSeedGenerator.h

Go to the documentation of this file.

#ifndef MuonSeedGenerator_CosmicMuonSeedGenerator_H
#define MuonSeedGenerator_CosmicMuonSeedGenerator_H
 
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
 
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
#include "RecoMuon/MeasurementDet/interface/MuonDetLayerMeasurements.h"
#include "RecoMuon/DetLayers/interface/MuonDetLayerGeometry.h"
#include "RecoMuon/Records/interface/MuonRecoGeometryRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
 
#include <vector>
 
class MuonDetLayerGeometry;
 
struct TrajectoryStateTransform;
 
namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm
 
class CosmicMuonSeedGenerator : public edm::stream::EDProducer<> {
public:
  CosmicMuonSeedGenerator(const edm::ParameterSet&);
 
  ~CosmicMuonSeedGenerator() override;
 
  // Operations
 
  void produce(edm::Event&, const edm::EventSetup&) override;
 
private:
  struct MuonRecHitPair {
    MuonRecHitPair(const MuonTransientTrackingRecHit::MuonRecHitPointer& a,
                   const MuonTransientTrackingRecHit::MuonRecHitPointer& b,
                   std::string c = "")
        : first(a), second(b), type(c) {}
 
    MuonTransientTrackingRecHit::MuonRecHitPointer first;
    MuonTransientTrackingRecHit::MuonRecHitPointer second;
    std::string type;
  };
 
  typedef std::vector<MuonRecHitPair> MuonRecHitPairVector;
 
  void createSeeds(TrajectorySeedCollection& results,
                   const MuonTransientTrackingRecHit::MuonRecHitContainer& hits,
                   const edm::EventSetup& eSetup) const;
 
  void createSeeds(TrajectorySeedCollection& results,
                   const CosmicMuonSeedGenerator::MuonRecHitPairVector& hits,
                   const edm::EventSetup& eSetup) const;
 
  bool checkQuality(const MuonTransientTrackingRecHit::MuonRecHitPointer&) const;
 
  MuonTransientTrackingRecHit::MuonRecHitContainer selectSegments(
      const MuonTransientTrackingRecHit::MuonRecHitContainer&) const;
 
  std::vector<TrajectorySeed> createSeed(const MuonTransientTrackingRecHit::MuonRecHitPointer&,
                                         const edm::EventSetup&) const;
 
  std::vector<MuonRecHitPair> makeSegPairs(const MuonTransientTrackingRecHit::MuonRecHitContainer&,
                                           const MuonTransientTrackingRecHit::MuonRecHitContainer&,
                                           std::string) const;
 
  std::vector<TrajectorySeed> createSeed(const MuonRecHitPair&, const edm::EventSetup&) const;
 
  TrajectorySeed tsosToSeed(const TrajectoryStateOnSurface&, uint32_t) const;
  TrajectorySeed tsosToSeed(const TrajectoryStateOnSurface&, uint32_t, edm::OwnVector<TrackingRecHit>&) const;
 
  bool areCorrelated(const MuonTransientTrackingRecHit::MuonRecHitPointer&,
                     const MuonTransientTrackingRecHit::MuonRecHitPointer&) const;
 
  bool leftIsBetter(const MuonTransientTrackingRecHit::MuonRecHitPointer&,
                    const MuonTransientTrackingRecHit::MuonRecHitPointer&) const;
 
  struct DecreasingGlobalY {
    bool operator()(const MuonTransientTrackingRecHit::ConstMuonRecHitPointer& lhs,
                    const MuonTransientTrackingRecHit::ConstMuonRecHitPointer& rhs) const {
      return lhs->globalPosition().y() > rhs->globalPosition().y();
    }
  };
 
private:
  bool theEnableDTFlag;
 
  bool theEnableCSCFlag;
 
  edm::InputTag theDTRecSegmentLabel;
 
  edm::InputTag theCSCRecSegmentLabel;
 
  unsigned int theMaxSeeds;
 
  double theMaxDTChi2;
  double theMaxCSCChi2;
  bool theForcePointDownFlag;
  edm::ESHandle<MuonDetLayerGeometry> theMuonLayers;
  edm::ESHandle<MagneticField> theField;
  edm::ESGetToken<MuonDetLayerGeometry, MuonRecoGeometryRecord> muonLayersToken;
  edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magFieldToken;
 
  std::map<std::string, float> theParameters;
 
  MuonDetLayerMeasurements* muonMeasurements;
};
#endif