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