MuonSeedCleaner.h

Go to the documentation of this file.

#ifndef RecoMuon_MuonSeedCleaner_H
#define RecoMuon_MuonSeedCleaner_H

#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"
#include <FWCore/ParameterSet/interface/ParameterSet.h>
#include <RecoMuon/MeasurementDet/interface/MuonDetLayerMeasurements.h>
#include <DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h>

//muon service
#include <RecoMuon/TrackingTools/interface/MuonServiceProxy.h>

#include <vector>

class DetLayer;
class MuonDetLayerGeometry;
class MagneticField;

typedef std::vector<TrajectorySeed> SeedContainer;

class MuonSeedCleaner {
public:
  typedef MuonTransientTrackingRecHit::MuonRecHitContainer SegmentContainer;
  typedef std::deque<bool> BoolContainer;

  explicit MuonSeedCleaner(const edm::ParameterSet&, edm::ConsumesCollector&&);

  ~MuonSeedCleaner();

  // Operations

  //void setGeometry( const MuonDetLayerGeometry* lgeom ) {muonLayers = lgeom;}

  //void setBField( const MagneticField* theField ) {BField = theField;}

  std::vector<TrajectorySeed> seedCleaner(const edm::EventSetup& eventSetup, std::vector<TrajectorySeed>& seeds);

  int NRecHitsFromSegment(const TrackingRecHit& rhit);
  int NRecHitsFromSegment(MuonTransientTrackingRecHit* rhit);
  //int NRecHitsFromSegment( const MuonTransientTrackingRecHit& rhit );

  std::vector<int> badSeedLayer;

private:
  std::vector<SeedContainer> GroupSeeds(std::vector<TrajectorySeed>& seeds);

  TrajectorySeed Chi2LengthSelection(std::vector<TrajectorySeed>& seeds);
  TrajectorySeed LeanHighMomentum(std::vector<TrajectorySeed>& seeds);
  TrajectorySeed BiggerCone(std::vector<TrajectorySeed>& seeds);
  TrajectorySeed MoreRecHits(std::vector<TrajectorySeed>& seeds);

  bool MomentumFilter(std::vector<TrajectorySeed>& seeds);
  SeedContainer SeedCandidates(std::vector<TrajectorySeed>& seeds, bool good);
  unsigned int OverlapSegments(const TrajectorySeed& seed1, const TrajectorySeed& seed2);

  double NChi2OfSegment(const TrackingRecHit& rhit);

  double SeedChi2(const TrajectorySeed& seed);
  int SeedLength(const TrajectorySeed& seed);

  GlobalPoint SeedPosition(const TrajectorySeed& seed);
  GlobalVector SeedMomentum(const TrajectorySeed& seed);

  // obsoleted functions
  SeedContainer LengthFilter(std::vector<TrajectorySeed>& seeds);

  // This Producer private debug flag
  bool debug;

  // Number of Segments from a shower
  SegmentContainer ShoweringSegments;
  std::vector<int> ShoweringLayers;

  // muon service
  MuonServiceProxy* theService;

  // Minimum separation when we can distinguish between 2 muon seeds
  // (to suppress combinatorics)
};
#endif