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 "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
#include "RecoMuon/MeasurementDet/interface/MuonDetLayerMeasurements.h"

#include <vector>

class MuonDetLayerGeometry;

class TrajectoryStateTransform;

namespace edm {class ParameterSet; class Event; class EventSetup;}

class CosmicMuonSeedGenerator: public edm::stream::EDProducer<> {
 public:

  CosmicMuonSeedGenerator(const edm::ParameterSet&);
  
  virtual ~CosmicMuonSeedGenerator();
  
  // Operations

  virtual 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;

  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;
  edm::ESHandle<MuonDetLayerGeometry> theMuonLayers;
  edm::ESHandle<MagneticField> theField;

  TrajectoryStateTransform* theTSTransform;

  std::map<std::string, float> theParameters;

  MuonDetLayerMeasurements* muonMeasurements;

};
#endif